diff -r 7e659a6870de -r dcd195f3e52c refresh-auth --- a/refresh-auth Tue Apr 15 18:30:43 2008 +0100 +++ b/refresh-auth Wed Apr 16 12:43:21 2008 +0100 @@ -1,14 +1,20 @@ #!/usr/bin/python # WARNING -# This script completely destroys your .ssh/authorized_keys +# This script completely destroys your ~/.ssh/authorized_keys # file every time it is run # WARNING +import sys import os import os.path import re +if len(sys.argv) != 2: + sys.stderr.write("refresh-auth: wrong number of arguments (%s)" % sys.argv) + sys.exit(-1) + +wrappercommand = sys.argv[1] akeyfile = os.path.expanduser("~/.ssh/authorized_keys") allowedchars = "A-Za-z0-9_.-" @@ -22,11 +28,10 @@ continue keyname = ffn[5:] prefix=('command="%s",no-pty,no-port-forwarding,no-X11-forwarding,no-agent-forwarding' - % ('./.hg-ssh-wrapper %s' % keyname)) + % ('%s %s' % (wrappercommand, keyname)) kf = open(ffn) try: for l in kf: - #prefix='no-port-forwarding,no-X11-forwarding,no-agent-forwarding' akeys.write("%s %s\n" % (prefix, l.strip())) finally: kf.close()