X-Git-Url: https://git.adam-barratt.org.uk/?p=mirror%2Fuserdir-ldap.git;a=blobdiff_plain;f=ud-gpgimport;h=4e26a5e8bb9905a9ba95821392792ab062c326bc;hp=8c7ad738fa21ceeeb99f3baa154fdba5f1d2592a;hb=HEAD;hpb=9e1e7b0639b1265bf39acd330122f40123451c56 diff --git a/ud-gpgimport b/ud-gpgimport index 8c7ad73..4e26a5e 100755 --- a/ud-gpgimport +++ b/ud-gpgimport @@ -1,5 +1,25 @@ #!/usr/bin/env python # -*- mode: python -*- + +# Copyright (c) 1999-2000 Jason Gunthorpe +# Copyright (c) 2004 Joey Schulze +# Copyright (c) 2008, 2009, 2010 Peter Palfrader +# Copyright (c) 2010 Martin Zobel-Helas +# +# This program is free software; you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation; either version 2 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program; if not, write to the Free Software +# Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. + # This script tries to match key fingerprints from a keyring with user # name in a directory. When an unassigned key is found a heuristic match # against the keys given cn/sn and the directory is performed to try to get @@ -14,7 +34,7 @@ # in the directory but not in the key ring will be removed from the # directory. -import string, re, time, ldap, getopt, sys, pwd, os; +import re, time, ldap, getopt, sys, pwd, os; from userdir_ldap import *; from userdir_gpg import *; @@ -32,29 +52,86 @@ def LoadOverride(File): if Line == "": break; Split = re.split("[:\n]",Line); - UnknownMap[Split[0]] = string.strip(Split[1]); + UnknownMap[Split[0]] = Split[1].strip() + + +def load_keys_from_gpg(keyrings): + keys = {} + + # Popen GPG with the correct magic special options + ClearKeyrings() + SetKeyrings(keyrings) + + Args = [GPGPath] + GPGBasicOptions + GPGKeyRings + GPGSearchOptions + [" 2> /dev/null"] + Keys = os.popen(" ".join(Args),"r"); + + # Loop over the GPG key file + Outstanding = 0; + while(1): + Line = Keys.readline(); + if Line == "": + break; + + Split = Line.split(":") + if len(Split) < 8 or Split[0] != "pub": + continue; + + while (1): + Line2 = Keys.readline(); + if Line2 == "": + break; + Split2 = Line2.split(":"); + if len(Split2) < 11 or Split2[0] != "fpr": + continue; + break; + if Line2 == "": + break; + + pgp_uid = Split[9] + fingerprint = Split2[9] + + if fingerprint in keys: + print "Duplicate key in keyrings: %s, belonging to %s"%(fingerprint, pgp_uid) + continue + keys[fingerprint] = pgp_uid + + if Keys.close() != None: + raise "Error","GPG failed" + + return keys + + + + + # Process options AdminUser = pwd.getpwuid(os.getuid())[0]; -(options, arguments) = getopt.getopt(sys.argv[1:], "au:m:n") +(options, arguments) = getopt.getopt(sys.argv[1:], "ahu:m:") for (switch, val) in options: if (switch == '-u'): AdminUser = val elif (switch == '-m'): LoadOverride(val); + elif (switch == '-h'): + print "Usage: ud-gpgimport " + print "Available options:" + print " -h Show this help" + print " -u= Admin user (defaults to current username)" + print " -m= Override file to use" + print " -a actually do changes, not dry-run" + sys.exit(0) elif (switch == '-a'): NoAct = 0; -if len(arguments) == 0: - print "Give some keyrings to probe"; - sys.exit(0); + # Main program starts here # Connect to the ldap server if NoAct == 0: - l = passwdAccessLDAP(LDAPServer, BaseDn, AdminUser) + l = passwdAccessLDAP(BaseDn, AdminUser) else: - l = ldap.open(LDAPServer); + l = connectLDAP() l.simple_bind_s("",""); # Download the existing key list and put it into a map @@ -86,69 +163,36 @@ for x in Attrs: Attrs = None; print; -# Popen GPG with the correct magic special options -Args = [GPGPath] + GPGBasicOptions; -for x in arguments: - Args.append("--keyring"); - if string.find(x,"/") == -1: - Args.append("./"+x); - else: - Args.append(x); -Args = Args + GPGSearchOptions + [" 2> /dev/null"] -Keys = os.popen(string.join(Args," "),"r"); - -# Loop over the GPG key file -Outstanding = 0; -Ignored = 0; -SeenKeys = {}; -while(1): - Line = Keys.readline(); - if Line == "": - break; - - Split = string.split(Line,":"); - if len(Split) < 8 or Split[0] != "pub": - continue; - while (1): - Line2 = Keys.readline(); - if Line2 == "": - break; - Split2 = string.split(Line2,":"); - if len(Split2) < 11 or Split2[0] != "fpr": - continue; - break; - if Line2 == "": - break; - - if SeenKeys.has_key(Split2[9]): - print "Dup key 0x",Split2[9],"belonging to",KeyMap[Split2[9]][0]; - continue; - SeenKeys[Split2[9]] = None; +pgpkeys = load_keys_from_gpg( ConfModule.add_keyrings.split(":") ) +pgpkeys_extra = load_keys_from_gpg( ConfModule.add_keyrings_guest.split(":") ) - if KeyMap.has_key(Split2[9]): +Ignored = 0; +for fpr in pgpkeys: + pgp_uid = pgpkeys[fpr] + if fpr in KeyMap: Ignored = Ignored + 1; - # print "Ignoring keyID",Split2[9],"belonging to",KeyMap[Split2[9]][0]; - KeyMap[Split2[9]][1] = 1; + # print "Ignoring keyID",fpr,"belonging to",KeyMap[fpr][0]; + KeyMap[fpr][1] = 1; continue; - - UID = GetUID(l,SplitEmail(Split[9]),UnknownMap); + + UID = GetUID(l,SplitEmail(pgp_uid),UnknownMap); if UID[0] == None: - print "None for",SplitEmail(Split[9]),"'%s'"%(Split[9]); - if UID[1] != None: + print "Unassigned key in keyrings: %s, belonging to %s"%(fpr, pgp_uid) + if UID[1] != None: for x in UID[1]: print x; - print "MISSING 0x" + Split2[9]; + print "MISSING " + fpr; continue; UID = UID[0] - Rec = [(ldap.MOD_ADD,"keyFingerPrint",Split2[9])]; + Rec = [(ldap.MOD_ADD,"keyFingerPrint",fpr)]; Dn = "uid=" + UID + "," + BaseDn; - print "Adding key 0x"+Split2[9],"to",UID; + print "Adding key "+fpr,"to",UID; if KeyCount.has_key(UID): KeyCount[UID] = KeyCount[UID] + 1; else: KeyCount[UID] = 1; - + if NoAct == 1: continue; @@ -161,15 +205,12 @@ while(1): if NoAct == 0: FlushOutstanding(l,Outstanding); -if Keys.close() != None: - raise "Error","GPG failed" - print Ignored,"keys already in the directory (ignored)"; # Look for unmatched keys for x in KeyMap.keys(): - if KeyMap[x][1] == 0: - print "key 0x%s belonging to %s removed"%(x,KeyMap[x][0]); + if KeyMap[x][1] == 0 and not x in pgpkeys_extra: + print "key %s belonging to %s removed"%(x,KeyMap[x][0]); if KeyCount.has_key(KeyMap[x][0]) : KeyCount[KeyMap[x][0]] = KeyCount[KeyMap[x][0]] - 1 if KeyCount[KeyMap[x][0]] <= 0: @@ -177,4 +218,7 @@ for x in KeyMap.keys(): if NoAct == 0: l.modify_s("uid="+KeyMap[x][0]+","+BaseDn,\ [(ldap.MOD_DELETE,"keyFingerPrint",x)]); - + +# vim:set et: +# vim:set ts=3: +# vim:set shiftwidth=3: