X-Git-Url: https://git.adam-barratt.org.uk/?a=blobdiff_plain;f=ud-generate;h=56ce0366ad7317197b7e0608daa4909294fff5e2;hb=a3437bcd185d533199ebd7c3e0f69b816cc904af;hp=f22b8de62bfdc5ce92c83ecacb2d3436b4b12fea;hpb=b22a2118b438dfeb8dedce430b4fc0d4ac2d132d;p=mirror%2Fuserdir-ldap.git diff --git a/ud-generate b/ud-generate index f22b8de..56ce036 100755 --- a/ud-generate +++ b/ud-generate @@ -11,6 +11,7 @@ # Copyright (c) 2008 Mark Hymers # Copyright (c) 2008 Luk Claes # Copyright (c) 2008 Thomas Viehmann +# Copyright (c) 2009 Stephen Gran # # 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 @@ -34,8 +35,8 @@ global Allowed global CurrentHost PasswdAttrs = None +DebianUsers = None DisabledUsers = [] -RetiredUsers = [] GroupIDMap = {} SubGroupMap = {} Allowed = None @@ -44,8 +45,8 @@ CurrentHost = "" UUID_FORMAT = '[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}' EmailCheck = re.compile("^([^ <>@]+@[^ ,<>@]+)?$") -BSMTPCheck = re.compile(".*mx 0 (gluck)\.debian\.org\..*",re.DOTALL) -PurposeHostField = re.compile(r"\[\[([\*\-]?[a-z0-9.\-]*)(?:\|.*)?\]\]") +BSMTPCheck = re.compile(".*mx 0 (gluck|master)\.debian\.org\..*",re.DOTALL) +PurposeHostField = re.compile(r".*\[\[([\*\-]?[a-z0-9.\-]*)(?:\|.*)?\]\]") DNSZone = ".debian.net" Keyrings = ConfModule.sync_keyrings.split(":") @@ -68,14 +69,14 @@ def safe_rmtree(dir): raise e def Sanitize(Str): - return Str.translate(string.maketrans("\n\r\t","$$$")) + return Str.translate(string.maketrans("\n\r\t", "$$$")) -def DoLink(From,To,File): +def DoLink(From, To, File): try: - posix.remove(To+File) + posix.remove(To + File) except: pass - posix.link(From+File,To+File) + posix.link(From + File, To + File) def IsRetired(DnRecord): """ @@ -83,7 +84,7 @@ def IsRetired(DnRecord): match it against one of the known retired statuses """ - status = GetAttr(DnRecord,"accountStatus", None) + status = GetAttr(DnRecord, "accountStatus", None) if status is None: return False @@ -100,40 +101,46 @@ def IsRetired(DnRecord): # We'll give them a few extra days over what we said age = 6 * 31 * 24 * 60 * 60 try: - if (time.time() - time.mktime(time.strptime(line[1], "%Y-%m-%d"))) > age: - return True + return (time.time() - time.mktime(time.strptime(line[1], "%Y-%m-%d"))) > age except IndexError: return False + except ValueError: + return False return False +def IsGidDebian(x): + try: + return int(GetAttr(x, "gidNumber", 0)) == 800 + except ValueError: + return False + # See if this user is in the group list def IsInGroup(DnRecord): - if Allowed == None: - return 1 + if Allowed is None: + return True # See if the primary group is in the list - if Allowed.has_key(GetAttr(DnRecord,"gidNumber")) != 0: - return 1 + if Allowed.has_key(GetAttr(DnRecord, "gidNumber")) != 0: + return True # Check the host based ACL if DnRecord[1].has_key("allowedHost") != 0: - for I in DnRecord[1]["allowedHost"]: - if CurrentHost == I: - return 1 + if CurrentHost in DnRecord[1]["allowedHost"]: + return True # See if there are supplementary groups if DnRecord[1].has_key("supplementaryGid") == 0: - return 0 + return False supgroups=[] - addGroups(supgroups, DnRecord[1]["supplementaryGid"], GetAttr(DnRecord,"uid")) + addGroups(supgroups, DnRecord[1]["supplementaryGid"], GetAttr(DnRecord, "uid")) for g in supgroups: if Allowed.has_key(g): - return 1 - return 0 + return True + return False -def Die(File,F,Fdb): +def Die(File, F, Fdb): if F != None: F.close() if Fdb != None: @@ -147,19 +154,19 @@ def Die(File,F,Fdb): except: pass -def Done(File,F,Fdb): +def Done(File, F, Fdb): if F != None: F.close() - os.rename(File + ".tmp",File) + os.rename(File + ".tmp", File) if Fdb != None: Fdb.close() - os.rename(File + ".tdb.tmp",File+".tdb") + os.rename(File + ".tdb.tmp", File + ".tdb") # Generate the password list -def GenPasswd(l,File,HomePrefix,PwdMarker): +def GenPasswd(File, HomePrefix, PwdMarker): F = None try: - F = open(File + ".tdb.tmp","w") + F = open(File + ".tdb.tmp", "w") userlist = {} # Fetch all the users @@ -167,41 +174,41 @@ def GenPasswd(l,File,HomePrefix,PwdMarker): I = 0 for x in PasswdAttrs: - if x[1].has_key("uidNumber") == 0 or IsInGroup(x) == 0: + if x[1].has_key("uidNumber") == 0 or not IsInGroup(x): continue # Do not let people try to buffer overflow some busted passwd parser. - if len(GetAttr(x,"gecos")) > 100 or len(GetAttr(x,"loginShell")) > 50: + if len(GetAttr(x, "gecos")) > 100 or len(GetAttr(x, "loginShell")) > 50: continue userlist[GetAttr(x, "uid")] = int(GetAttr(x, "gidNumber")) - Line = "%s:%s:%s:%s:%s:%s%s:%s" % (GetAttr(x,"uid"),\ + Line = "%s:%s:%s:%s:%s:%s%s:%s" % (GetAttr(x, "uid"),\ PwdMarker,\ - GetAttr(x,"uidNumber"),GetAttr(x,"gidNumber"),\ - GetAttr(x,"gecos"),HomePrefix,GetAttr(x,"uid"),\ - GetAttr(x,"loginShell")) + GetAttr(x, "uidNumber"), GetAttr(x, "gidNumber"),\ + GetAttr(x, "gecos"), HomePrefix, GetAttr(x, "uid"),\ + GetAttr(x, "loginShell")) Line = Sanitize(Line) + "\n" - F.write("0%u %s" % (I,Line)) - F.write(".%s %s" % (GetAttr(x,"uid"),Line)) - F.write("=%s %s" % (GetAttr(x,"uidNumber"),Line)) + F.write("0%u %s" % (I, Line)) + F.write(".%s %s" % (GetAttr(x, "uid"), Line)) + F.write("=%s %s" % (GetAttr(x, "uidNumber"), Line)) I = I + 1 # Oops, something unspeakable happened. except: - Die(File,None,F) + Die(File, None, F) raise - Done(File,None,F) + Done(File, None, F) # Return the list of users so we know which keys to export return userlist # Generate the shadow list -def GenShadow(l,File): +def GenShadow(File): F = None try: OldMask = os.umask(0077) - F = open(File + ".tdb.tmp","w",0600) + F = open(File + ".tdb.tmp", "w", 0600) os.umask(OldMask) # Fetch all the users @@ -209,10 +216,10 @@ def GenShadow(l,File): I = 0 for x in PasswdAttrs: - if x[1].has_key("uidNumber") == 0 or IsInGroup(x) == 0: + if x[1].has_key("uidNumber") == 0 or not IsInGroup(x): continue - Pass = GetAttr(x,"userPassword") + Pass = GetAttr(x, "userPassword") if Pass[0:7] != "{crypt}" or len(Pass) > 50: Pass = '*' else: @@ -220,34 +227,34 @@ def GenShadow(l,File): # If the account is locked, mark it as such in shadow # See Debian Bug #308229 for why we set it to 1 instead of 0 - if (GetAttr(x,"userPassword").find("*LK*") != -1) \ - or GetAttr(x,"userPassword").startswith("!"): + if (GetAttr(x, "userPassword").find("*LK*") != -1) \ + or GetAttr(x, "userPassword").startswith("!"): ShadowExpire = '1' else: - ShadowExpire = GetAttr(x,"shadowExpire") + ShadowExpire = GetAttr(x, "shadowExpire") - Line = "%s:%s:%s:%s:%s:%s:%s:%s:" % (GetAttr(x,"uid"),\ - Pass,GetAttr(x,"shadowLastChange"),\ - GetAttr(x,"shadowMin"),GetAttr(x,"shadowMax"),\ - GetAttr(x,"shadowWarning"),GetAttr(x,"shadowInactive"),\ + Line = "%s:%s:%s:%s:%s:%s:%s:%s:" % (GetAttr(x, "uid"),\ + Pass, GetAttr(x, "shadowLastChange"),\ + GetAttr(x, "shadowMin"), GetAttr(x, "shadowMax"),\ + GetAttr(x, "shadowWarning"), GetAttr(x, "shadowInactive"),\ ShadowExpire) Line = Sanitize(Line) + "\n" - F.write("0%u %s" % (I,Line)) - F.write(".%s %s" % (GetAttr(x,"uid"),Line)) + F.write("0%u %s" % (I, Line)) + F.write(".%s %s" % (GetAttr(x, "uid"), Line)) I = I + 1 # Oops, something unspeakable happened. except: - Die(File,None,F) + Die(File, None, F) raise - Done(File,None,F) + Done(File, None, F) # Generate the sudo passwd file -def GenShadowSudo(l,File, untrusted): +def GenShadowSudo(File, untrusted): F = None try: OldMask = os.umask(0077) - F = open(File + ".tmp","w",0600) + F = open(File + ".tmp", "w", 0600) os.umask(OldMask) # Fetch all the users @@ -255,7 +262,7 @@ def GenShadowSudo(l,File, untrusted): for x in PasswdAttrs: Pass = '*' - if x[1].has_key("uidNumber") == 0 or IsInGroup(x) == 0: + if x[1].has_key("uidNumber") == 0 or not IsInGroup(x): continue if x[1].has_key('sudoPassword'): @@ -279,24 +286,23 @@ def GenShadowSudo(l,File, untrusted): continue Pass = cryptedpass if for_this_host: # this makes sure we take a per-host entry over the for-all entry - break + break if len(Pass) > 50: Pass = '*' - Line = "%s:%s" % (GetAttr(x,"uid"), Pass) + Line = "%s:%s" % (GetAttr(x, "uid"), Pass) Line = Sanitize(Line) + "\n" F.write("%s" % (Line)) # Oops, something unspeakable happened. except: - Die(File,F,None) + Die(File, F, None) raise - Done(File,F,None) + Done(File, F, None) # Generate the shadow list -def GenSSHShadow(l): +def GenSSHShadow(): # Fetch all the users - singlefile = None userfiles = [] global PasswdAttrs @@ -306,20 +312,17 @@ def GenSSHShadow(l): for x in PasswdAttrs: - if x in DisabledUsers: - continue - if x[1].has_key("uidNumber") == 0 or \ x[1].has_key("sshRSAAuthKey") == 0: continue - User = GetAttr(x,"uid") + User = GetAttr(x, "uid") F = None try: OldMask = os.umask(0077) File = os.path.join(GlobalDir, 'userkeys', User) - F = open(File + ".tmp","w",0600) + F = open(File + ".tmp", "w", 0600) os.umask(OldMask) for I in x[1]["sshRSAAuthKey"]: @@ -327,14 +330,14 @@ def GenSSHShadow(l): MultipleLine = Sanitize(MultipleLine) + "\n" F.write(MultipleLine) - Done(File,F,None) + Done(File, F, None) userfiles.append(os.path.basename(File)) # Oops, something unspeakable happened. except IOError: - Die(File,F,None) - Die(masterFileName,masterFile,None) - raise + Die(File, F, None) + Die(masterFileName, masterFile, None) + raise return userfiles @@ -399,7 +402,7 @@ def addGroups(existingGroups, newGroups, uid): # let's see if we handled this group already if group in existingGroups: - continue + continue if not GroupIDMap.has_key(group): print "Group", group, "does not exist but", uid, "is in it" @@ -411,11 +414,11 @@ def addGroups(existingGroups, newGroups, uid): addGroups(existingGroups, SubGroupMap[group], uid) # Generate the group list -def GenGroup(l,File): +def GenGroup(File): grouprevmap = {} F = None try: - F = open(File + ".tdb.tmp","w") + F = open(File + ".tdb.tmp", "w") # Generate the GroupMap GroupMap = {} @@ -427,8 +430,8 @@ def GenGroup(l,File): # Sort them into a list of groups having a set of users for x in PasswdAttrs: - uid = GetAttr(x,"uid") - if x[1].has_key("uidNumber") == 0 or IsInGroup(x) == 0: + uid = GetAttr(x, "uid") + if x[1].has_key("uidNumber") == 0 or not IsInGroup(x): continue if x[1].has_key("supplementaryGid") == 0: continue @@ -444,85 +447,88 @@ def GenGroup(l,File): grouprevmap[GroupIDMap[x]] = x if GroupIDMap.has_key(x) == 0: continue - Line = "%s:x:%u:" % (x,GroupIDMap[x]) + Line = "%s:x:%u:" % (x, GroupIDMap[x]) Comma = '' for I in GroupMap[x]: - Line = Line + ("%s%s" % (Comma,I)) - Comma = ',' + Line = Line + ("%s%s" % (Comma, I)) + Comma = ',' Line = Sanitize(Line) + "\n" - F.write("0%u %s" % (J,Line)) - F.write(".%s %s" % (x,Line)) - F.write("=%u %s" % (GroupIDMap[x],Line)) + F.write("0%u %s" % (J, Line)) + F.write(".%s %s" % (x, Line)) + F.write("=%u %s" % (GroupIDMap[x], Line)) J = J + 1 # Oops, something unspeakable happened. except: - Die(File,None,F) + Die(File, None, F) raise - Done(File,None,F) + Done(File, None, F) return grouprevmap +def CheckForward(): + global DebianUsers + for x in DebianUsers: + if x[1].has_key("emailForward") == 0: + continue + + if not IsInGroup(x): + x[1].pop("emailForward") + continue + + # Do not allow people to try to buffer overflow busted parsers + if len(GetAttr(x, "emailForward")) > 200: + x[1].pop("emailForward") + continue + + # Check the forwarding address + if EmailCheck.match(GetAttr(x, "emailForward")) == None: + x[1].pop("emailForward") + # Generate the email forwarding list -def GenForward(l,File): +def GenForward(File): F = None try: OldMask = os.umask(0022) - F = open(File + ".tmp","w",0644) + F = open(File + ".tmp", "w", 0644) os.umask(OldMask) # Fetch all the users - global PasswdAttrs + global DebianUsers # Write out the email address for each user - for x in PasswdAttrs: - if x[1].has_key("emailForward") == 0 or IsInGroup(x) == 0: - continue - - # Do not allow people to try to buffer overflow busted parsers - if len(GetAttr(x,"emailForward")) > 200: + for x in DebianUsers: + if x[1].has_key("emailForward") == 0: continue - # Check the forwarding address - if EmailCheck.match(GetAttr(x,"emailForward")) == None: - continue - Line = "%s: %s" % (GetAttr(x,"uid"),GetAttr(x,"emailForward")) + Line = "%s: %s" % (GetAttr(x, "uid"), GetAttr(x, "emailForward")) Line = Sanitize(Line) + "\n" F.write(Line) # Oops, something unspeakable happened. except: - Die(File,F,None) + Die(File, F, None) raise - Done(File,F,None) + Done(File, F, None) -def GenAllForward(l,File): +def GenCDB(File, Key): Fdb = None try: OldMask = os.umask(0022) - Fdb = os.popen("cdbmake %s %s.tmp"%(File,File),"w") + Fdb = os.popen("cdbmake %s %s.tmp"%(File, File), "w") os.umask(OldMask) - + # Fetch all the users - global PasswdAttrs - + global DebianUsers + # Write out the email address for each user - for x in PasswdAttrs: - if x[1].has_key("emailForward") == 0: - continue - - # Do not allow people to try to buffer overflow busted parsers - Forward = GetAttr(x,"emailForward") - if len(Forward) > 200: + for x in DebianUsers: + if not Key in x[1]: continue - - # Check the forwarding address - if EmailCheck.match(Forward) == None: - continue - - User = GetAttr(x,"uid") - Fdb.write("+%d,%d:%s->%s\n"%(len(User),len(Forward),User,Forward)) - + Value = GetAttr(x, Key) + User = GetAttr(x, "uid") + Fdb.write("+%d,%d:%s->%s\n" % (len(User), len(Value), User, Value)) + Fdb.write("\n") # Oops, something unspeakable happened. except: @@ -532,20 +538,20 @@ def GenAllForward(l,File): raise "cdbmake gave an error" # Generate the anon XEarth marker file -def GenMarkers(l,File): +def GenMarkers(File): F = None try: - F = open(File + ".tmp","w") + F = open(File + ".tmp", "w") # Fetch all the users - global PasswdAttrs + global DebianUsers # Write out the position for each user - for x in PasswdAttrs: + for x in DebianUsers: if x[1].has_key("latitude") == 0 or x[1].has_key("longitude") == 0: continue try: - Line = "%8s %8s \"\""%(DecDegree(GetAttr(x,"latitude"),1),DecDegree(GetAttr(x,"longitude"),1)) + Line = "%8s %8s \"\""%(DecDegree(GetAttr(x, "latitude"), 1), DecDegree(GetAttr(x, "longitude"), 1)) Line = Sanitize(Line) + "\n" F.write(Line) except: @@ -553,21 +559,21 @@ def GenMarkers(l,File): # Oops, something unspeakable happened. except: - Die(File,F,None) + Die(File, F, None) raise - Done(File,F,None) + Done(File, F, None) # Generate the debian-private subscription list -def GenPrivate(l,File): +def GenPrivate(File): F = None try: - F = open(File + ".tmp","w") + F = open(File + ".tmp", "w") # Fetch all the users - global PasswdAttrs + global DebianUsers # Write out the position for each user - for x in PasswdAttrs: + for x in DebianUsers: if x[1].has_key("privateSub") == 0: continue @@ -575,12 +581,8 @@ def GenPrivate(l,File): if x[1].has_key("keyFingerPrint") == 0: continue - # Must be in the Debian group (yuk, hard coded for now) - if GetAttr(x,"gidNumber") != "800": - continue - try: - Line = "%s"%(GetAttr(x,"privateSub")) + Line = "%s"%(GetAttr(x, "privateSub")) Line = Sanitize(Line) + "\n" F.write(Line) except: @@ -588,15 +590,15 @@ def GenPrivate(l,File): # Oops, something unspeakable happened. except: - Die(File,F,None) + Die(File, F, None) raise - Done(File,F,None) + Done(File, F, None) # Generate a list of locked accounts -def GenDisabledAccounts(l,File): +def GenDisabledAccounts(File): F = None try: - F = open(File + ".tmp","w") + F = open(File + ".tmp", "w") # Fetch all the users global PasswdAttrs @@ -607,48 +609,44 @@ def GenDisabledAccounts(l,File): if x[1].has_key("uidNumber") == 0: continue - Pass = GetAttr(x,"userPassword") + Pass = GetAttr(x, "userPassword") Line = "" # *LK* is the reference value for a locked account # password starting with ! is also a locked account if Pass.find("*LK*") != -1 or Pass.startswith("!"): # Format is : - Line = "%s:%s" % (GetAttr(x,"uid"), "Account is locked") + Line = "%s:%s" % (GetAttr(x, "uid"), "Account is locked") + DisabledUsers.append(x) if Line != "": F.write(Sanitize(Line) + "\n") - DisabledUsers.append(x) # Oops, something unspeakable happened. except: - Die(File,F,None) + Die(File, F, None) raise - Done(File,F,None) + Done(File, F, None) # Generate the list of local addresses that refuse all mail -def GenMailDisable(l,File): +def GenMailDisable(File): F = None try: - F = open(File + ".tmp","w") + F = open(File + ".tmp", "w") # Fetch all the users - global PasswdAttrs + global DebianUsers - for x in PasswdAttrs: + for x in DebianUsers: Reason = None if x[1].has_key("mailDisableMessage"): - Reason = GetAttr(x,"mailDisableMessage") + Reason = GetAttr(x, "mailDisableMessage") else: continue - # Must be in the Debian group (yuk, hard coded for now) - if GetAttr(x,"gidNumber") != "800": - continue - try: - Line = "%s: %s"%(GetAttr(x,"uid"),Reason) + Line = "%s: %s"%(GetAttr(x, "uid"), Reason) Line = Sanitize(Line) + "\n" F.write(Line) except: @@ -656,34 +654,30 @@ def GenMailDisable(l,File): # Oops, something unspeakable happened. except: - Die(File,F,None) + Die(File, F, None) raise - Done(File,F,None) + Done(File, F, None) # Generate a list of uids that should have boolean affects applied -def GenMailBool(l,File,Key): +def GenMailBool(File, Key): F = None try: - F = open(File + ".tmp","w") + F = open(File + ".tmp", "w") # Fetch all the users - global PasswdAttrs + global DebianUsers - for x in PasswdAttrs: + for x in DebianUsers: Reason = None if x[1].has_key(Key) == 0: continue - # Must be in the Debian group (yuk, hard coded for now) - if GetAttr(x,"gidNumber") != "800": - continue - - if GetAttr(x,Key) != "TRUE": + if GetAttr(x, Key) != "TRUE": continue try: - Line = "%s"%(GetAttr(x,"uid")) + Line = "%s"%(GetAttr(x, "uid")) Line = Sanitize(Line) + "\n" F.write(Line) except: @@ -691,59 +685,55 @@ def GenMailBool(l,File,Key): # Oops, something unspeakable happened. except: - Die(File,F,None) + Die(File, F, None) raise - Done(File,F,None) + Done(File, F, None) # Generate a list of hosts for RBL or whitelist purposes. -def GenMailList(l,File,Key): +def GenMailList(File, Key): F = None try: - F = open(File + ".tmp","w") + F = open(File + ".tmp", "w") # Fetch all the users - global PasswdAttrs + global DebianUsers - for x in PasswdAttrs: + for x in DebianUsers: Reason = None if x[1].has_key(Key) == 0: continue - # Must be in the Debian group (yuk, hard coded for now) - if GetAttr(x,"gidNumber") != "800": - continue - try: found = 0 Line = None for z in x[1][Key]: if Key == "mailWhitelist": - if re.match('^[-\w.]+(/[\d]+)?$',z) == None: - continue + if re.match('^[-\w.]+(/[\d]+)?$', z) == None: + continue else: - if re.match('^[-\w.]+$',z) == None: - continue + if re.match('^[-\w.]+$', z) == None: + continue if found == 0: - found = 1 - Line = GetAttr(x,"uid") + found = 1 + Line = GetAttr(x, "uid") else: Line += " " Line += ": " + z if Key == "mailRHSBL": - Line += "/$sender_address_domain" + Line += "/$sender_address_domain" if Line != None: - Line = Sanitize(Line) + "\n" - F.write(Line) + Line = Sanitize(Line) + "\n" + F.write(Line) except: pass # Oops, something unspeakable happened. except: - Die(File,F,None) + Die(File, F, None) raise - Done(File,F,None) + Done(File, F, None) def isRoleAccount(pwEntry): if not pwEntry.has_key("objectClass"): @@ -756,10 +746,10 @@ def isRoleAccount(pwEntry): return False # Generate the DNS Zone file -def GenDNS(l,File,HomePrefix): +def GenDNS(File, HomePrefix): F = None try: - F = open(File + ".tmp","w") + F = open(File + ".tmp", "w") # Fetch all the users global PasswdAttrs @@ -777,7 +767,7 @@ def GenDNS(l,File,HomePrefix): for z in x[1]["dnsZoneEntry"]: Split = z.lower().split() if Split[1].lower() == 'in': - for y in range(0,len(Split)): + for y in range(0, len(Split)): if Split[y] == "$": Split[y] = "\n\t" Line = " ".join(Split) + "\n" @@ -785,13 +775,13 @@ def GenDNS(l,File,HomePrefix): Host = Split[0] + DNSZone if BSMTPCheck.match(Line) != None: - F.write("; Has BSMTP\n") + F.write("; Has BSMTP\n") # Write some identification information if Split[2].lower() == "a": - Line = "%s IN TXT \"%s\"\n"%(Split[0],EmailAddress(x)) + Line = "%s IN TXT \"%s\"\n"%(Split[0], EmailAddress(x)) for y in x[1]["keyFingerPrint"]: - Line = Line + "%s IN TXT \"PGP %s\"\n"%(Split[0],FormatPGPKey(y)) + Line = Line + "%s IN TXT \"PGP %s\"\n"%(Split[0], FormatPGPKey(y)) F.write(Line) else: Line = "; Err %s"%(str(Split)) @@ -804,15 +794,15 @@ def GenDNS(l,File,HomePrefix): # Oops, something unspeakable happened. except: - Die(File,F,None) + Die(File, F, None) raise - Done(File,F,None) + Done(File, F, None) # Generate the DNS SSHFP records -def GenSSHFP(l,File,HomePrefix): +def GenSSHFP(File, HomePrefix): F = None try: - F = open(File + ".tmp","w") + F = open(File + ".tmp", "w") # Fetch all the hosts global HostAttrs @@ -823,7 +813,7 @@ def GenSSHFP(l,File,HomePrefix): if x[1].has_key("hostname") == 0 or \ x[1].has_key("sshRSAHostKey") == 0: continue - Host = GetAttr(x,"hostname") + Host = GetAttr(x, "hostname") Algorithm = None for I in x[1]["sshRSAHostKey"]: Split = I.split() @@ -834,26 +824,26 @@ def GenSSHFP(l,File,HomePrefix): if Algorithm == None: continue Fingerprint = sha.new(base64.decodestring(Split[1])).hexdigest() - Line = "%s. IN SSHFP %u 1 %s" % (Host,Algorithm,Fingerprint) + Line = "%s. IN SSHFP %u 1 %s" % (Host, Algorithm, Fingerprint) Line = Sanitize(Line) + "\n" F.write(Line) # Oops, something unspeakable happened. except: - Die(File,F,None) + Die(File, F, None) raise - Done(File,F,None) + Done(File, F, None) # Generate the BSMTP file -def GenBSMTP(l,File,HomePrefix): +def GenBSMTP(File, HomePrefix): F = None try: - F = open(File + ".tmp","w") + F = open(File + ".tmp", "w") # Fetch all the users - global PasswdAttrs + global DebianUsers # Write out the zone file entry for each user - for x in PasswdAttrs: + for x in DebianUsers: if x[1].has_key("dnsZoneEntry") == 0: continue @@ -864,7 +854,7 @@ def GenBSMTP(l,File,HomePrefix): for z in x[1]["dnsZoneEntry"]: Split = z.lower().split() if Split[1].lower() == 'in': - for y in range(0,len(Split)): + for y in range(0, len(Split)): if Split[y] == "$": Split[y] = "\n\t" Line = " ".join(Split) + "\n" @@ -872,7 +862,7 @@ def GenBSMTP(l,File,HomePrefix): Host = Split[0] + DNSZone if BSMTPCheck.match(Line) != None: F.write("%s: user=%s group=Debian file=%s%s/bsmtp/%s\n"%(Host, - GetAttr(x,"uid"),HomePrefix,GetAttr(x,"uid"),Host)) + GetAttr(x, "uid"), HomePrefix, GetAttr(x, "uid"), Host)) except: F.write("; Errors\n") @@ -880,9 +870,9 @@ def GenBSMTP(l,File,HomePrefix): # Oops, something unspeakable happened. except: - Die(File,F,None) + Die(File, F, None) raise - Done(File,F,None) + Done(File, F, None) # cache IP adresses HostToIPCache = {} @@ -891,7 +881,7 @@ def HostToIP(Host): if not Host in HostToIPCache: IPAdressesT = None try: - IPAdressesT = list(set([ (a[0],a[4][0]) for a in socket.getaddrinfo(Host, None)])) + IPAdressesT = list(set([ (a[0], a[4][0]) for a in socket.getaddrinfo(Host, None)])) except socket.gaierror, (code): if code[0] != -2: raise @@ -906,32 +896,32 @@ def HostToIP(Host): return HostToIPCache[Host] # Generate the ssh known hosts file -def GenSSHKnown(l,File,mode=None): +def GenSSHKnown(File, mode=None): F = None try: OldMask = os.umask(0022) - F = open(File + ".tmp","w",0644) + F = open(File + ".tmp", "w", 0644) os.umask(OldMask) global HostAttrs - if HostAttrs == None: + if HostAttrs is None: raise UDEmptyList, "No Hosts" for x in HostAttrs: if x[1].has_key("hostname") == 0 or \ x[1].has_key("sshRSAHostKey") == 0: continue - Host = GetAttr(x,"hostname") + Host = GetAttr(x, "hostname") HostNames = [ Host ] if Host.endswith(HostDomain): - HostNames.append(Host[:-(len(HostDomain)+1)]) + HostNames.append(Host[:-(len(HostDomain) + 1)]) # in the purpose field [[host|some other text]] (where some other text is optional) # makes a hyperlink on the web thing. we now also add these hosts to the ssh known_hosts # file. But so that we don't have to add everything we link we can add an asterisk # and say [[*... to ignore it. In order to be able to add stuff to ssh without # http linking it we also support [[-hostname]] entries. - for i in x[1].get("purpose",[]): + for i in x[1].get("purpose", []): m = PurposeHostField.match(i) if m: m = m.group(1) @@ -943,7 +933,7 @@ def GenSSHKnown(l,File,mode=None): if m: HostNames.append(m) if m.endswith(HostDomain): - HostNames.append(m[:-(len(HostDomain)+1)]) + HostNames.append(m[:-(len(HostDomain) + 1)]) for I in x[1]["sshRSAHostKey"]: if mode and mode == 'authorized_keys': @@ -955,28 +945,28 @@ def GenSSHKnown(l,File,mode=None): F.write(Line) # Oops, something unspeakable happened. except: - Die(File,F,None) + Die(File, F, None) raise - Done(File,F,None) + Done(File, F, None) # Generate the debianhosts file (list of all IP addresses) -def GenHosts(l,File): +def GenHosts(l, File): F = None try: OldMask = os.umask(0022) - F = open(File + ".tmp","w",0644) + F = open(File + ".tmp", "w", 0644) os.umask(OldMask) # Fetch all the hosts hostnames = l.search_s(HostBaseDn, ldap.SCOPE_ONELEVEL, "hostname=*", ["hostname"]) - if hostnames == None: + if hostnames is None: raise UDEmptyList, "No Hosts" seen = set() for x in hostnames: - host = GetAttr(x,"hostname", None) + host = GetAttr(x, "hostname", None) if host: addrs = [] try: @@ -996,26 +986,25 @@ def GenHosts(l,File): seen.add(addr) # Oops, something unspeakable happened. except: - Die(File,F,None) - raise - Done(File,F,None) + Die(File, F, None) + raise + Done(File, F, None) -def GenKeyrings(l,OutDir): +def GenKeyrings(OutDir): for k in Keyrings: shutil.copy(k, OutDir) - # Connect to the ldap server l = connectLDAP() -F = open(PassDir+"/pass-"+pwd.getpwuid(os.getuid())[0],"r") +F = open(PassDir + "/pass-" + pwd.getpwuid(os.getuid())[0], "r") Pass = F.readline().strip().split(" ") F.close() -l.simple_bind_s("uid="+Pass[0]+","+BaseDn,Pass[1]) +l.simple_bind_s("uid=" + Pass[0] + "," + BaseDn, Pass[1]) # Fetch all the groups GroupIDMap = {} -Attrs = l.search_s(BaseDn,ldap.SCOPE_ONELEVEL,"gid=*",\ - ["gid","gidNumber","subGroup"]) +Attrs = l.search_s(BaseDn, ldap.SCOPE_ONELEVEL, "gid=*",\ + ["gid", "gidNumber", "subGroup"]) # Generate the SubGroupMap and GroupIDMap for x in Attrs: @@ -1026,59 +1015,62 @@ for x in Attrs: SubGroupMap.setdefault(x[1]["gid"][0], []).extend(x[1]["subGroup"]) # Fetch all the users -PasswdAttrs = l.search_s(BaseDn,ldap.SCOPE_ONELEVEL,"uid=*",\ - ["uid","uidNumber","gidNumber","supplementaryGid",\ - "gecos","loginShell","userPassword","shadowLastChange",\ - "shadowMin","shadowMax","shadowWarning","shadowInactive", - "shadowExpire","emailForward","latitude","longitude",\ - "allowedHost","sshRSAAuthKey","dnsZoneEntry","cn","sn",\ - "keyFingerPrint","privateSub","mailDisableMessage",\ - "mailGreylisting","mailCallout","mailRBL","mailRHSBL",\ - "mailWhitelist", "sudoPassword", "objectClass", "accountStatus"]) +PasswdAttrs = l.search_s(BaseDn, ldap.SCOPE_ONELEVEL, "uid=*",\ + ["uid", "uidNumber", "gidNumber", "supplementaryGid",\ + "gecos", "loginShell", "userPassword", "shadowLastChange",\ + "shadowMin", "shadowMax", "shadowWarning", "shadowInactive", + "shadowExpire", "emailForward", "latitude", "longitude",\ + "allowedHost", "sshRSAAuthKey", "dnsZoneEntry", "cn", "sn",\ + "keyFingerPrint", "privateSub", "mailDisableMessage",\ + "mailGreylisting", "mailCallout", "mailRBL", "mailRHSBL",\ + "mailWhitelist", "sudoPassword", "objectClass", "accountStatus",\ + "mailContentInspectionAction"]) if PasswdAttrs is None: raise UDEmptyList, "No Users" # Fetch all the hosts -HostAttrs = l.search_s(HostBaseDn,ldap.SCOPE_ONELEVEL,"sshRSAHostKey=*",\ - ["hostname","sshRSAHostKey","purpose"]) +HostAttrs = l.search_s(HostBaseDn, ldap.SCOPE_ONELEVEL, "sshRSAHostKey=*",\ + ["hostname", "sshRSAHostKey", "purpose"]) # Open the control file if len(sys.argv) == 1: - F = open(GenerateConf,"r") + F = open(GenerateConf, "r") else: - F = open(sys.argv[1],"r") + F = open(sys.argv[1], "r") # Generate global things -GlobalDir = GenerateDir+"/" -GenMailDisable(l,GlobalDir+"mail-disable") +GlobalDir = GenerateDir + "/" +GenDisabledAccounts(GlobalDir + "disabled-accounts") -for x in PasswdAttrs: - if IsRetired(x): - RetiredUsers.append(x) +PasswdAttrs = filter(lambda x: not IsRetired(x), PasswdAttrs) +#DebianUsers = filter(lambda x: IsGidDebian(x), PasswdAttrs) +DebianUsers = PasswdAttrs -PasswdAttrs = filter(lambda x: not x in RetiredUsers, PasswdAttrs) +CheckForward() -SSHFiles = GenSSHShadow(l) -GenAllForward(l,GlobalDir+"mail-forward.cdb") -GenMarkers(l,GlobalDir+"markers") -GenPrivate(l,GlobalDir+"debian-private") -GenDisabledAccounts(l,GlobalDir+"disabled-accounts") -GenSSHKnown(l,GlobalDir+"ssh_known_hosts") +GenMailDisable(GlobalDir + "mail-disable") +GenCDB(GlobalDir + "mail-forward.cdb", 'emailForward') +GenCDB(GlobalDir + "mail-contentinspectionaction.cdb", 'mailContentInspectionAction') +GenPrivate(GlobalDir + "debian-private") #GenSSHKnown(l,GlobalDir+"authorized_keys", 'authorized_keys') -GenHosts(l,GlobalDir+"debianhosts") -GenMailBool(l,GlobalDir+"mail-greylist","mailGreylisting") -GenMailBool(l,GlobalDir+"mail-callout","mailCallout") -GenMailList(l,GlobalDir+"mail-rbl","mailRBL") -GenMailList(l,GlobalDir+"mail-rhsbl","mailRHSBL") -GenMailList(l,GlobalDir+"mail-whitelist","mailWhitelist") -GenKeyrings(l,GlobalDir) +GenMailBool(GlobalDir + "mail-greylist", "mailGreylisting") +GenMailBool(GlobalDir + "mail-callout", "mailCallout") +GenMailList(GlobalDir + "mail-rbl", "mailRBL") +GenMailList(GlobalDir + "mail-rhsbl", "mailRHSBL") +GenMailList(GlobalDir + "mail-whitelist", "mailWhitelist") +GenKeyrings(GlobalDir) # Compatibility. -GenForward(l,GlobalDir+"forward-alias") +GenForward(GlobalDir + "forward-alias") PasswdAttrs = filter(lambda x: not x in DisabledUsers, PasswdAttrs) +SSHFiles = GenSSHShadow() +GenMarkers(GlobalDir + "markers") +GenSSHKnown(GlobalDir + "ssh_known_hosts") +GenHosts(l, GlobalDir + "debianhosts") + while(1): Line = F.readline() if Line == "": @@ -1109,21 +1101,21 @@ while(1): Allowed = GroupList if Allowed == {}: - Allowed = None + Allowed = None CurrentHost = Split[0] - DoLink(GlobalDir,OutDir,"debianhosts") - DoLink(GlobalDir,OutDir,"ssh_known_hosts") - DoLink(GlobalDir,OutDir,"disabled-accounts") + DoLink(GlobalDir, OutDir, "debianhosts") + DoLink(GlobalDir, OutDir, "ssh_known_hosts") + DoLink(GlobalDir, OutDir, "disabled-accounts") sys.stdout.flush() if ExtraList.has_key("[NOPASSWD]"): - userlist = GenPasswd(l,OutDir+"passwd",Split[1], "*") + userlist = GenPasswd(OutDir + "passwd", Split[1], "*") else: - userlist = GenPasswd(l,OutDir+"passwd",Split[1], "x") + userlist = GenPasswd(OutDir + "passwd", Split[1], "x") sys.stdout.flush() - grouprevmap = GenGroup(l,OutDir+"group") - GenShadowSudo(l, OutDir+"sudo-passwd", ExtraList.has_key("[UNTRUSTED]") or ExtraList.has_key("[NOPASSWD]")) + grouprevmap = GenGroup(OutDir + "group") + GenShadowSudo(OutDir + "sudo-passwd", ExtraList.has_key("[UNTRUSTED]") or ExtraList.has_key("[NOPASSWD]")) # Now we know who we're allowing on the machine, export # the relevant ssh keys @@ -1133,39 +1125,40 @@ while(1): print "[UNTRUSTED] tag is obsolete and may be removed in the future." continue if not ExtraList.has_key("[NOPASSWD]"): - GenShadow(l,OutDir+"shadow") + GenShadow(OutDir + "shadow") # Link in global things if not ExtraList.has_key("[NOMARKERS]"): - DoLink(GlobalDir,OutDir,"markers") - DoLink(GlobalDir,OutDir,"mail-forward.cdb") - DoLink(GlobalDir,OutDir,"mail-disable") - DoLink(GlobalDir,OutDir,"mail-greylist") - DoLink(GlobalDir,OutDir,"mail-callout") - DoLink(GlobalDir,OutDir,"mail-rbl") - DoLink(GlobalDir,OutDir,"mail-rhsbl") - DoLink(GlobalDir,OutDir,"mail-whitelist") + DoLink(GlobalDir, OutDir, "markers") + DoLink(GlobalDir, OutDir, "mail-forward.cdb") + DoLink(GlobalDir, OutDir, "mail-contentinspectionaction.cdb") + DoLink(GlobalDir, OutDir, "mail-disable") + DoLink(GlobalDir, OutDir, "mail-greylist") + DoLink(GlobalDir, OutDir, "mail-callout") + DoLink(GlobalDir, OutDir, "mail-rbl") + DoLink(GlobalDir, OutDir, "mail-rhsbl") + DoLink(GlobalDir, OutDir, "mail-whitelist") # Compatibility. - DoLink(GlobalDir,OutDir,"forward-alias") + DoLink(GlobalDir, OutDir, "forward-alias") if ExtraList.has_key("[DNS]"): - GenDNS(l,OutDir+"dns-zone",Split[1]) - GenSSHFP(l,OutDir+"dns-sshfp",Split[1]) + GenDNS(OutDir + "dns-zone", Split[1]) + GenSSHFP(OutDir + "dns-sshfp", Split[1]) if ExtraList.has_key("[BSMTP]"): - GenBSMTP(l,OutDir+"bsmtp",Split[1]) + GenBSMTP(OutDir + "bsmtp", Split[1]) if ExtraList.has_key("[PRIVATE]"): - DoLink(GlobalDir,OutDir,"debian-private") + DoLink(GlobalDir, OutDir, "debian-private") if ExtraList.has_key("[KEYRING]"): for k in Keyrings: - DoLink(GlobalDir,OutDir,os.path.basename(k)) + DoLink(GlobalDir, OutDir, os.path.basename(k)) else: for k in Keyrings: try: - posix.remove(OutDir+os.path.basename(k)) + posix.remove(OutDir + os.path.basename(k)) except: pass