refresh-auth
author Paul Crowley <paul@lshift.net>
Wed, 28 May 2008 18:14:15 +0100
changeset 31 d54720d47ca2
parent 30 98dbde5b13a1
child 32 4059dbe9f26a
permissions -rwxr-xr-x
start to move towards things living where they should and new break-in system. Big change.
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
16
9fac559c3d55 don't assume Python path for refresh-auth
Paul Crowley <paul@ciphergoth.org>
parents: 15
diff changeset
     1
#!/usr/bin/env python
0
41ecb5a3172c separate out executables and data
Paul Crowley <paul@lshift.net>
parents:
diff changeset
     2
3
7e659a6870de make more robus and less crufty
Paul Crowley <paul@lshift.net>
parents: 1
diff changeset
     3
# WARNING
4
dcd195f3e52c move config out of Python files; don't make hg-ssh-wrapper a dotfile;
Paul Crowley <paul@lshift.net>
parents: 3
diff changeset
     4
# This script completely destroys your ~/.ssh/authorized_keys
3
7e659a6870de make more robus and less crufty
Paul Crowley <paul@lshift.net>
parents: 1
diff changeset
     5
# file every time it is run
7e659a6870de make more robus and less crufty
Paul Crowley <paul@lshift.net>
parents: 1
diff changeset
     6
# WARNING
7e659a6870de make more robus and less crufty
Paul Crowley <paul@lshift.net>
parents: 1
diff changeset
     7
4
dcd195f3e52c move config out of Python files; don't make hg-ssh-wrapper a dotfile;
Paul Crowley <paul@lshift.net>
parents: 3
diff changeset
     8
import sys
0
41ecb5a3172c separate out executables and data
Paul Crowley <paul@lshift.net>
parents:
diff changeset
     9
import os
41ecb5a3172c separate out executables and data
Paul Crowley <paul@lshift.net>
parents:
diff changeset
    10
import os.path
29
87279134a212 Convert PuTTY-style public keys automatically
Paul Crowley <paul@lshift.net>
parents: 16
diff changeset
    11
import ruleset
87279134a212 Convert PuTTY-style public keys automatically
Paul Crowley <paul@lshift.net>
parents: 16
diff changeset
    12
import subprocess
0
41ecb5a3172c separate out executables and data
Paul Crowley <paul@lshift.net>
parents:
diff changeset
    13
30
98dbde5b13a1 refresh-auth now takes ~/.ssh/authorized_keys as an argument, and
Paul Crowley <paul@lshift.net>
parents: 29
diff changeset
    14
if len(sys.argv) != 3:
11
f3c73c9fc0ff add newline to error message
Paul Crowley <paul@lshift.net>
parents: 6
diff changeset
    15
    sys.stderr.write("refresh-auth: wrong number of arguments (%s)\n" % sys.argv)
4
dcd195f3e52c move config out of Python files; don't make hg-ssh-wrapper a dotfile;
Paul Crowley <paul@lshift.net>
parents: 3
diff changeset
    16
    sys.exit(-1)
dcd195f3e52c move config out of Python files; don't make hg-ssh-wrapper a dotfile;
Paul Crowley <paul@lshift.net>
parents: 3
diff changeset
    17
30
98dbde5b13a1 refresh-auth now takes ~/.ssh/authorized_keys as an argument, and
Paul Crowley <paul@lshift.net>
parents: 29
diff changeset
    18
akeyfile = sys.argv[1]
98dbde5b13a1 refresh-auth now takes ~/.ssh/authorized_keys as an argument, and
Paul Crowley <paul@lshift.net>
parents: 29
diff changeset
    19
wrappercommand = sys.argv[2]
98dbde5b13a1 refresh-auth now takes ~/.ssh/authorized_keys as an argument, and
Paul Crowley <paul@lshift.net>
parents: 29
diff changeset
    20
prefix='no-pty,no-port-forwarding,no-X11-forwarding,no-agent-forwarding,command='
98dbde5b13a1 refresh-auth now takes ~/.ssh/authorized_keys as an argument, and
Paul Crowley <paul@lshift.net>
parents: 29
diff changeset
    21
98dbde5b13a1 refresh-auth now takes ~/.ssh/authorized_keys as an argument, and
Paul Crowley <paul@lshift.net>
parents: 29
diff changeset
    22
if os.path.exists(akeyfile):
98dbde5b13a1 refresh-auth now takes ~/.ssh/authorized_keys as an argument, and
Paul Crowley <paul@lshift.net>
parents: 29
diff changeset
    23
    f = open(akeyfile)
98dbde5b13a1 refresh-auth now takes ~/.ssh/authorized_keys as an argument, and
Paul Crowley <paul@lshift.net>
parents: 29
diff changeset
    24
    try:
98dbde5b13a1 refresh-auth now takes ~/.ssh/authorized_keys as an argument, and
Paul Crowley <paul@lshift.net>
parents: 29
diff changeset
    25
        for l in f:
98dbde5b13a1 refresh-auth now takes ~/.ssh/authorized_keys as an argument, and
Paul Crowley <paul@lshift.net>
parents: 29
diff changeset
    26
            if not l.startswith(prefix):
98dbde5b13a1 refresh-auth now takes ~/.ssh/authorized_keys as an argument, and
Paul Crowley <paul@lshift.net>
parents: 29
diff changeset
    27
                raise Exception("Safety check failed, delete %s to continue" % akeyfile)
98dbde5b13a1 refresh-auth now takes ~/.ssh/authorized_keys as an argument, and
Paul Crowley <paul@lshift.net>
parents: 29
diff changeset
    28
    finally:
98dbde5b13a1 refresh-auth now takes ~/.ssh/authorized_keys as an argument, and
Paul Crowley <paul@lshift.net>
parents: 29
diff changeset
    29
        f.close()
0
41ecb5a3172c separate out executables and data
Paul Crowley <paul@lshift.net>
parents:
diff changeset
    30
41ecb5a3172c separate out executables and data
Paul Crowley <paul@lshift.net>
parents:
diff changeset
    31
akeys = open(akeyfile + "_new", "w")
41ecb5a3172c separate out executables and data
Paul Crowley <paul@lshift.net>
parents:
diff changeset
    32
for root, dirs, files in os.walk("keys"):
41ecb5a3172c separate out executables and data
Paul Crowley <paul@lshift.net>
parents:
diff changeset
    33
    for fn in files:
41ecb5a3172c separate out executables and data
Paul Crowley <paul@lshift.net>
parents:
diff changeset
    34
        ffn = os.path.join(root, fn)
29
87279134a212 Convert PuTTY-style public keys automatically
Paul Crowley <paul@lshift.net>
parents: 16
diff changeset
    35
        if not ruleset.goodpath(ffn):
3
7e659a6870de make more robus and less crufty
Paul Crowley <paul@lshift.net>
parents: 1
diff changeset
    36
            # ignore any path that contains dodgy characters
7e659a6870de make more robus and less crufty
Paul Crowley <paul@lshift.net>
parents: 1
diff changeset
    37
            continue
0
41ecb5a3172c separate out executables and data
Paul Crowley <paul@lshift.net>
parents:
diff changeset
    38
        keyname = ffn[5:]
29
87279134a212 Convert PuTTY-style public keys automatically
Paul Crowley <paul@lshift.net>
parents: 16
diff changeset
    39
        p = subprocess.Popen(("ssh-keygen", "-i", "-f", ffn), 
87279134a212 Convert PuTTY-style public keys automatically
Paul Crowley <paul@lshift.net>
parents: 16
diff changeset
    40
            stdout=subprocess.PIPE, stderr=subprocess.PIPE)
87279134a212 Convert PuTTY-style public keys automatically
Paul Crowley <paul@lshift.net>
parents: 16
diff changeset
    41
        newkey = p.communicate()[0]
87279134a212 Convert PuTTY-style public keys automatically
Paul Crowley <paul@lshift.net>
parents: 16
diff changeset
    42
        if p.wait() == 0:
87279134a212 Convert PuTTY-style public keys automatically
Paul Crowley <paul@lshift.net>
parents: 16
diff changeset
    43
            klines = [l.strip() for l in newkey.split("\n")]
87279134a212 Convert PuTTY-style public keys automatically
Paul Crowley <paul@lshift.net>
parents: 16
diff changeset
    44
        else:
87279134a212 Convert PuTTY-style public keys automatically
Paul Crowley <paul@lshift.net>
parents: 16
diff changeset
    45
            # Conversion failed, read it directly.
87279134a212 Convert PuTTY-style public keys automatically
Paul Crowley <paul@lshift.net>
parents: 16
diff changeset
    46
            kf = open(ffn)
87279134a212 Convert PuTTY-style public keys automatically
Paul Crowley <paul@lshift.net>
parents: 16
diff changeset
    47
            try:
87279134a212 Convert PuTTY-style public keys automatically
Paul Crowley <paul@lshift.net>
parents: 16
diff changeset
    48
                klines = [l.strip() for l in kf]
87279134a212 Convert PuTTY-style public keys automatically
Paul Crowley <paul@lshift.net>
parents: 16
diff changeset
    49
            finally:
87279134a212 Convert PuTTY-style public keys automatically
Paul Crowley <paul@lshift.net>
parents: 16
diff changeset
    50
                kf.close()
87279134a212 Convert PuTTY-style public keys automatically
Paul Crowley <paul@lshift.net>
parents: 16
diff changeset
    51
        for l in klines:
87279134a212 Convert PuTTY-style public keys automatically
Paul Crowley <paul@lshift.net>
parents: 16
diff changeset
    52
            if len(l):
30
98dbde5b13a1 refresh-auth now takes ~/.ssh/authorized_keys as an argument, and
Paul Crowley <paul@lshift.net>
parents: 29
diff changeset
    53
                akeys.write('%s"%s %s" %s\n' % (prefix, wrappercommand, keyname, l))
29
87279134a212 Convert PuTTY-style public keys automatically
Paul Crowley <paul@lshift.net>
parents: 16
diff changeset
    54
15
f3654416d178 minor changes to README and script
Hubert Plociniczak <hubert@lshift.net>
parents: 11
diff changeset
    55
akeys.close()
0
41ecb5a3172c separate out executables and data
Paul Crowley <paul@lshift.net>
parents:
diff changeset
    56
41ecb5a3172c separate out executables and data
Paul Crowley <paul@lshift.net>
parents:
diff changeset
    57
os.rename(akeyfile + "_new", akeyfile)
41ecb5a3172c separate out executables and data
Paul Crowley <paul@lshift.net>
parents:
diff changeset
    58