ud-mailgate: remove exception for münchen.debian.net
[mirror/userdir-ldap.git] / ud-mailgate
index 952a4dc..30af583 100755 (executable)
@@ -7,55 +7,60 @@
 #   Copyright (c) 2008 Joerg Jaspert <joerg@debian.org>
 #   Copyright (c) 2010 Helmut Grohne <helmut@subdivi.de>
 
-import userdir_gpg, userdir_ldap, sys, traceback, time, ldap, os, commands
-import pwd, tempfile
+import userdir_gpg
+import userdir_ldap
+import sys
+import traceback
+import time
+import ldap
+import os
+import commands
+import pwd
+import tempfile
 import subprocess
+import email
+import email.parser
+import binascii
 
 from userdir_gpg import *
 from userdir_ldap import *
 from userdir_exceptions import *
 
 # Error codes from /usr/include/sysexits.h
-ReplyTo = ConfModule.replyto;
-PingFrom = ConfModule.pingfrom;
-ChPassFrom = ConfModule.chpassfrom;
-ChangeFrom = ConfModule.changefrom;
-ReplayCacheFile = ConfModule.replaycachefile;
+ReplyTo = ConfModule.replyto
+PingFrom = ConfModule.pingfrom
+ChPassFrom = ConfModule.chpassfrom
+ChangeFrom = ConfModule.changefrom
+ReplayCacheFile = ConfModule.replaycachefile
 SSHFingerprintFile = ConfModule.fingerprintfile
+TOTPTicketDirectory = ConfModule.totpticketdirectory
+WebUILocation = ConfModule.webuilocation
 
 UUID_FORMAT = '[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}'
+machine_regex = re.compile("^[0-9a-zA-Z.-]+$")
 
-EX_TEMPFAIL = 75;
-EX_PERMFAIL = 65;      # EX_DATAERR
-Error = 'Message Error';
-SeenKey = 0;
-SeenDNS = 0;
+EX_TEMPFAIL = 75
+EX_PERMFAIL = 65      # EX_DATAERR
+Error = 'Message Error'
+SeenKey = 0
+SeenDNS = 0
 mailRBL = {}
 mailRHSBL = {}
 mailWhitelist = {}
 SeenList = {}
 DNS = {}
-SudoPasswd = {}
-ValidHostNames = [] # will be initialized in later
-
-SSHFingerprint = re.compile('^(\d+) ([0-9a-f\:]{47}) (.+)$')
-SSHRSA1Match = re.compile('^^(.* )?\d+ \d+ \d+')
+ValidHostNames = []  # will be initialized in later
 
-GenderTable = {"male": '1',
-              "1": '1',
-              "female": '2',
-              "2": '2',
-              "unspecified": '9',
-              "9": '9',
-};
+SSHFingerprint = re.compile(r'^(\d+) ([0-9a-f\:]{47}|SHA256:[0-9A-Za-z/+]{43}) (.+)$')
+SSHRSA1Match = re.compile(r'^^(.* )?\d+ \d+ \d+')
 
 ArbChanges = {"c": "..",
-             "l": ".*",
-             "facsimileTelephoneNumber": ".*",
-             "telephoneNumber": ".*",
-             "postalAddress": ".*",
-         "bATVToken": ".*",
-             "postalCode": ".*",
+              "l": ".*",
+              "facsimileTelephoneNumber": ".*",
+              "telephoneNumber": ".*",
+              "postalAddress": ".*",
+              "bATVToken": ".*",
+              "postalCode": ".*",
               "loginShell": ".*",
               "emailForward": "^([^<>@]+@.+)?$",
               "jabberJID": "^([^<>@]+@.+)?$",
@@ -65,13 +70,11 @@ ArbChanges = {"c": "..",
               "labeledURI": ".*",
               "birthDate": "^([0-9]{4})([01][0-9])([0-3][0-9])$",
               "mailDisableMessage": ".*",
-             "mailGreylisting": "^(TRUE|FALSE)$",
-             "mailCallout": "^(TRUE|FALSE)$",
-             "mailDefaultOptions": "^(TRUE|FALSE)$",
-             "VoIP": ".*",
-             "gender": "^(1|2|9|male|female|unspecified)$",
-         "mailContentInspectionAction": "^(reject|blackhole|markup)$",
-};
+              "mailGreylisting": "^(TRUE|FALSE)$",
+              "mailCallout": "^(TRUE|FALSE)$",
+              "mailDefaultOptions": "^(TRUE|FALSE)$",
+              "VoIP": ".*",
+              "mailContentInspectionAction": "^(reject|blackhole|markup)$", }
 
 DelItems = {"c": None,
             "l": None,
@@ -84,14 +87,14 @@ DelItems = {"c": None,
             "ircNick": None,
             "onVacation": None,
             "labeledURI": None,
-           "latitude": None,
-           "longitude": None,
+            "latitude": None,
+            "longitude": None,
             "icqUin": None,
             "jabberJID": None,
             "jpegPhoto": None,
             "dnsZoneEntry": None,
-           "sshRSAAuthKey": None,
-            "birthDate" : None,
+            "sshRSAAuthKey": None,
+            "birthDate": None,
             "mailGreylisting": None,
             "mailCallout": None,
             "mailRBL": None,
@@ -101,788 +104,801 @@ DelItems = {"c": None,
             "mailDefaultOptions": None,
             "VoIP": None,
             "mailContentInspectionAction": None,
-            };
+            }
 
 
 # Decode a GPS location from some common forms
-def LocDecode(Str,Dir):
-   # Check for Decimal degrees, DGM, or DGMS
-   if re.match("^[+-]?[\d.]+$",Str) != None:
-      return Str;
-
-   Deg = '0'; Min = None; Sec = None; Dr = Dir[0];
-   
-   # Check for DDDxMM.MMMM where x = [nsew]
-   Match = re.match("^(\d+)(["+Dir+"])([\d.]+)$",Str);
-   if Match != None:
-      G = Match.groups();
-      Deg = G[0]; Min = G[2]; Dr = G[1];
-
-   # Check for DD.DD x 
-   Match = re.match("^([\d.]+) ?(["+Dir+"])$",Str);
-   if Match != None:
-      G = Match.groups();
-      Deg = G[0]; Dr = G[1];
-
-   # Check for DD:MM.MM x 
-   Match = re.match("^(\d+):([\d.]+) ?(["+Dir+"])$",Str);
-   if Match != None:
-      G = Match.groups();
-      Deg = G[0]; Min = G[1]; Dr = G[2];
-
-   # Check for DD:MM:SS.SS x
-   Match = re.match("^(\d+):(\d+):([\d.]+) ?(["+Dir+"])$",Str);
-   if Match != None:
-      G = Match.groups();
-      Deg = G[0]; Min = G[1]; Sec = G[2]; Dr = G[3];
-      
-   # Some simple checks
-   if float(Deg) > 180:
-      raise UDFormatError, "Bad degrees";
-   if Min != None and float(Min) > 60:
-      raise UDFormatError, "Bad minutes";
-   if Sec != None and float(Sec) > 60:
-      raise UDFormatError, "Bad seconds";
-      
-   # Pad on an extra leading 0 to disambiguate small numbers
-   if len(Deg) <= 1 or Deg[1] == '.':
-      Deg = '0' + Deg;
-   if Min != None and (len(Min) <= 1 or Min[1] == '.'):
-      Min = '0' + Min;
-   if Sec != None and (len(Sec) <= 1 or Sec[1] == '.'):
-      Sec = '0' + Sec;
-   
-   # Construct a DGM/DGMS type value from the components.
-   Res = "+"
-   if Dr == Dir[1]:
-      Res = "-";
-   Res = Res + Deg;
-   if Min != None:
-      Res = Res + Min;
-   if Sec != None:
-      Res = Res + Sec;
-   return Res;
-             
+def LocDecode(Str, Dir):
+    # Check for Decimal degrees, DGM, or DGMS
+    if re.match(r"^[+-]?[\d.]+$", Str) is not None:
+        return Str
+
+    Deg = '0'
+    Min = None
+    Sec = None
+    Dr = Dir[0]
+
+    # Check for DDDxMM.MMMM where x = [nsew]
+    Match = re.match(r"^(\d+)([" + Dir + r"])([\d.]+)$", Str)
+    if Match:
+        G = Match.groups()
+        Deg = G[0]
+        Min = G[2]
+        Dr = G[1]
+
+    # Check for DD.DD x
+    Match = re.match(r"^([\d.]+) ?([" + Dir + r"])$", Str)
+    if Match:
+        G = Match.groups()
+        Deg = G[0]
+        Dr = G[1]
+
+    # Check for DD:MM.MM x
+    Match = re.match(r"^(\d+):([\d.]+) ?([" + Dir + "])$", Str)
+    if Match:
+        G = Match.groups()
+        Deg = G[0]
+        Min = G[1]
+        Dr = G[2]
+
+    # Check for DD:MM:SS.SS x
+    Match = re.match(r"^(\d+):(\d+):([\d.]+) ?([" + Dir + "])$", Str)
+    if Match:
+        G = Match.groups()
+        Deg = G[0]
+        Min = G[1]
+        Sec = G[2]
+        Dr = G[3]
+
+    # Some simple checks
+    if float(Deg) > 180:
+        raise UDFormatError("Bad degrees")
+    if Min is not None and float(Min) > 60:
+        raise UDFormatError("Bad minutes")
+    if Sec is not None and float(Sec) > 60:
+        raise UDFormatError("Bad seconds")
+
+    # Pad on an extra leading 0 to disambiguate small numbers
+    if len(Deg) <= 1 or Deg[1] == '.':
+        Deg = '0' + Deg
+    if Min is not None and (len(Min) <= 1 or Min[1] == '.'):
+        Min = '0' + Min
+    if Sec is not None and (len(Sec) <= 1 or Sec[1] == '.'):
+        Sec = '0' + Sec
+
+    # Construct a DGM/DGMS type value from the components.
+    Res = "+"
+    if Dr == Dir[1]:
+        Res = "-"
+    Res = Res + Deg
+    if Min is not None:
+        Res = Res + Min
+    if Sec is not None:
+        Res = Res + Sec
+    return Res
+
+
 # Handle changing a set of arbitary fields
 #  <field>: value
-def DoArbChange(Str,Attrs):
-   Match = re.match("^([^ :]+): (.*)$",Str);
-   if Match == None:
-      return None;
-   G = Match.groups();
-
-   attrName = G[0].lower();
-   for i in ArbChanges.keys():
-      if i.lower() == attrName:
-         attrName = i;
-         break;
-   if ArbChanges.has_key(attrName) == 0:
-      return None;
-
-   if re.match(ArbChanges[attrName],G[1]) == None:
-      raise UDFormatError, "Item does not match the required format"+ArbChanges[attrName];
-
-   value = G[1];
-   if attrName == 'gender':
-      if G[1] not in GenderTable:
-         raise UDFormatError, "Gender not found in table"
-      value = GenderTable[G[1]]
-
-#   if attrName == 'birthDate':
-#      (re.match("^([0-9]{4})([01][0-9])([0-3][0-9])$",G[1]) {
-#    $bd_yr = $1; $bd_mo = $2; $bd_day = $3;
-#    if ($bd_mo > 0 and $bd_mo <= 12 and $bd_day > 0) {
-#      if ($bd_mo == 2) {
-#       if ($bd_day == 29 and ($bd_yr == 0 or ($bd_yr % 4 == 0 && ($bd_yr % 100 != 0 || $bd_yr % 400 == 0)))) {
-#         $bd_ok = 1;
-#       } elsif ($bd_day <= 28) {
-#         $bd_ok = 1;
-#       }
-#      } elsif ($bd_mo == 4 or $bd_mo == 6 or $bd_mo == 9 or $bd_mo == 11) {
-#      if ($bd_day <= 30) {
-#        $bd_ok = 1;
-#      }
-#      } else {
-#      if ($bd_day <= 31) {
-#        $bd_ok = 1;
-#      }
-#      }
-#    }
-#  } elsif (not defined($query->param('birthdate')) or $query->param('birthdate') =~ /^\s*$/) {
-#    $bd_ok = 1;
-#  }
-   Attrs.append((ldap.MOD_REPLACE,attrName,value));
-   return "Changed entry %s to %s"%(attrName,value);
+def DoArbChange(Str, Attrs):
+    Match = re.match("^([^ :]+): (.*)$", Str)
+    if Match is None:
+        return None
+    G = Match.groups()
+
+    attrName = G[0].lower()
+    for i in ArbChanges.keys():
+        if i.lower() == attrName:
+            attrName = i
+            break
+    if attrName not in ArbChanges:
+        return None
+
+    value = G[1]
+    if re.match(ArbChanges[attrName], value) is None:
+        raise UDFormatError("Item does not match the required format" + ArbChanges[attrName])
+
+    Attrs.append((ldap.MOD_REPLACE, attrName, value))
+    return "Changed entry %s to %s" % (attrName, value)
+
 
 # Handle changing a set of arbitary fields
 #  <field>: value
-def DoDel(Str,Attrs):
-   Match = re.match("^del (.*)$",Str);
-   if Match == None:
-      return None;
-   G = Match.groups();
-
-   attrName = G[0].lower();
-   for i in DelItems.keys():
-      if i.lower() == attrName:
-         attrName = i;
-         break;
-   if DelItems.has_key(attrName) == 0:
-      return "Cannot erase entry %s"%(attrName);
-
-   Attrs.append((ldap.MOD_DELETE,attrName,None));
-   return "Removed entry %s"%(attrName);
+def DoDel(Str, Attrs):
+    Match = re.match("^del (.*)$", Str)
+    if Match is None:
+        return None
+    G = Match.groups()
+
+    attrName = G[0].lower()
+    for i in DelItems.keys():
+        if i.lower() == attrName:
+            attrName = i
+            break
+    if attrName not in DelItems:
+        return "Cannot erase entry %s" % (attrName,)
+
+    Attrs.append((ldap.MOD_DELETE, attrName, None))
+    return "Removed entry %s" % (attrName)
+
 
 # Handle a position change message, the line format is:
 #  Lat: -12412.23 Long: +12341.2342
-def DoPosition(Str,Attrs):
-   Match = re.match("^lat: ([+\-]?[\d:.ns]+(?: ?[ns])?) long: ([+\-]?[\d:.ew]+(?: ?[ew])?)$", Str.lower())
-   if Match == None:
-      return None;
-
-   G = Match.groups();
-   try:
-      sLat = LocDecode(G[0],"ns");
-      sLong = LocDecode(G[1],"ew");
-      Lat = DecDegree(sLat,1);
-      Long = DecDegree(sLong,1);
-   except:
-      raise UDFormatError, "Positions were found, but they are not correctly formed";
-
-   Attrs.append((ldap.MOD_REPLACE,"latitude",sLat));
-   Attrs.append((ldap.MOD_REPLACE,"longitude",sLong));
-   return "Position set to %s/%s (%s/%s decimal degrees)"%(sLat,sLong,Lat,Long);
+def DoPosition(Str, Attrs):
+    Match = re.match(r"^lat: ([+\-]?[\d:.ns]+(?: ?[ns])?) long: ([+\-]?[\d:.ew]+(?: ?[ew])?)$", Str.lower())
+    if Match is None:
+        return None
+
+    G = Match.groups()
+    try:
+        sLat = LocDecode(G[0], "ns")
+        sLong = LocDecode(G[1], "ew")
+        Lat = DecDegree(sLat, 1)
+        Long = DecDegree(sLong, 1)
+    except Exception:
+        raise UDFormatError("Positions were found, but they are not correctly formed")
+
+    Attrs.append((ldap.MOD_REPLACE, "latitude", sLat))
+    Attrs.append((ldap.MOD_REPLACE, "longitude", sLong))
+    return "Position set to %s/%s (%s/%s decimal degrees)" % (sLat, sLong, Lat, Long)
+
 
 # Load bad ssh fingerprints
 def LoadBadSSH():
-   f = open(SSHFingerprintFile, "r")
-   bad = []
-   FingerprintLine = re.compile('^([0-9a-f\:]{47}).*$')
-   for line in f.readlines():
-      Match = FingerprintLine.match(line)
-      if Match is not None:
-         g = Match.groups()
-         bad.append(g[0])
-   return bad
+    f = open(SSHFingerprintFile, "r")
+    bad = []
+    FingerprintLine = re.compile(r'^([0-9a-f:]{47}).*$')
+    for line in f.readlines():
+        Match = FingerprintLine.match(line)
+        if Match is not None:
+            g = Match.groups()
+            bad.append(g[0])
+    return bad
+
 
 # Handle an SSH authentication key, the line format is:
 #  [options] 1024 35 13188913666680[..] [comment]
 # maybe it really should be:
 # [allowed_hosts=machine1,machine2 ][options ]ssh-rsa keybytes [comment]
-machine_regex = re.compile("^[0-9a-zA-Z.-]+$")
 def DoSSH(Str, Attrs, badkeys, uid):
-   Match = SSH2AuthSplit.match(Str);
-   if Match == None:
-      return None;
-   g = Match.groups()
-   typekey = g[1]
-   if Match == None:
-      Match = SSHRSA1Match.match(Str)
-      if Match is not None:
-         return "RSA1 keys not supported anymore"
-      return None;
-
-   # lines can now be prepended with "allowed_hosts=machine1,machine2 "
-   machines = []
-   if Str.startswith("allowed_hosts="):
-      Str = Str.split("=", 1)[1]
-      if ' ' not in Str:
-         return "invalid ssh key syntax with machine specification"
-      machines, Str = Str.split(' ', 1)
-      machines = machines.split(",")
-      for m in machines:
-         if not m:
-            return "empty machine specification for ssh key"
-         if not machine_regex.match(m):
-            return "machine specification for ssh key contains invalid characters"
-         if m not in ValidHostNames:
-            return "unknown machine used in allowed_hosts stanza for ssh keys"
-
-   (fd, path) = tempfile.mkstemp(".pub", "sshkeytry", "/tmp")
-   f = open(path, "w")
-   f.write("%s\n" % (Str))
-   f.close()
-   cmd = "/usr/bin/ssh-keygen -l -f %s < /dev/null" % (path)
-   (result, output) = commands.getstatusoutput(cmd)
-   os.remove(path)
-   if (result != 0):
-      raise UDExecuteError, "ssh-keygen -l invocation failed!\n%s\n" % (output)
-
-   # format the string again for ldap:
-   if machines:
-      Str = "allowed_hosts=%s %s" % (",".join(machines), Str)
-
-
-   # Head
-   Date = time.strftime("%a, %d %b %Y %H:%M:%S +0000",time.gmtime(time.time()))
-   ErrReplyHead = "From: %s\nCc: %s\nReply-To: %s\nDate: %s\n" % (os.environ['SENDER'],os.environ['SENDER'],ReplyTo,Date)
-   Subst = {}
-   Subst["__ADMIN__"] = ReplyTo
-   Subst["__USER__"] = uid
-
-   Match = SSHFingerprint.match(output)
-   g = Match.groups()
-
-   if int(g[0]) < 1024:
-      try:
-         # Body
-         Subst["__ERROR__"] = "SSH keysize %s is below limit 1024" % (g[0])
-         ErrReply = TemplateSubst(Subst,open(TemplatesDir+"admin-info","r").read())
-
-         Child = os.popen("/usr/sbin/sendmail -t","w")
-         Child.write(ErrReplyHead)
-         Child.write(ErrReply)
-         if Child.close() != None:
-            raise UDExecuteError, "Sendmail gave a non-zero return code"
-      except:
-         sys.exit(EX_TEMPFAIL)
-
-      # And now break and stop processing input, which sends a reply to the user.
-      raise UDFormatError, "SSH keys must have at least 1024 bits, processing halted, NOTHING MODIFIED AT ALL"
-   elif g[1] in badkeys:
-      try:
-         # Body
-         Subst["__ERROR__"] = "SSH key with fingerprint %s known as bad key" % (g[1])
-         ErrReply = TemplateSubst(Subst,open(TemplatesDir+"admin-info","r").read())
-
-         Child = os.popen("/usr/sbin/sendmail -t","w")
-         Child.write(ErrReplyHead)
-         Child.write(ErrReply)
-         if Child.close() != None:
-            raise UDExecuteError, "Sendmail gave a non-zero return code"
-      except:
-         sys.exit(EX_TEMPFAIL)
-
-      # And now break and stop processing input, which sends a reply to the user.
-      raise UDFormatError, "Submitted SSH Key known to be bad and insecure, processing halted, NOTHING MODIFIED AT ALL"
-
-   if (typekey == "dss"):
-      return "DSA keys not accepted anymore"
-
-   global SeenKey;
-   if SeenKey:
-     Attrs.append((ldap.MOD_ADD,"sshRSAAuthKey",Str));
-     return "SSH Key added "+FormatSSHAuth(Str);
-      
-   Attrs.append((ldap.MOD_REPLACE,"sshRSAAuthKey",Str));
-   SeenKey = 1;
-   return "SSH Keys replaced with "+FormatSSHAuth(Str);
+    Match = SSH2AuthSplit.match(Str)
+    if Match is None:
+        return None
+    g = Match.groups()
+    typekey = g[1]
+    if Match is None:
+        Match = SSHRSA1Match.match(Str)
+        if Match is not None:
+            return "RSA1 keys not supported anymore"
+        return None
+
+    # lines can now be prepended with "allowed_hosts=machine1,machine2 "
+    machines = []
+    if Str.startswith("allowed_hosts="):
+        Str = Str.split("=", 1)[1]
+        if ' ' not in Str:
+            return "invalid ssh key syntax with machine specification"
+        machines, Str = Str.split(' ', 1)
+        machines = machines.split(",")
+        for m in machines:
+            if not m:
+                return "empty machine specification for ssh key"
+            if not machine_regex.match(m):
+                return "machine specification for ssh key contains invalid characters"
+            if m not in ValidHostNames:
+                return "unknown machine {} used in allowed_hosts stanza for ssh keys".format(m)
+
+    (fd, path) = tempfile.mkstemp(".pub", "sshkeytry", "/tmp")
+    f = open(path, "w")
+    f.write("%s\n" % (Str))
+    f.close()
+    cmd = "/usr/bin/ssh-keygen -l -f %s < /dev/null" % (path)
+    (result, output) = commands.getstatusoutput(cmd)
+    os.remove(path)
+    if (result != 0):
+        raise UDExecuteError("ssh-keygen -l invocation failed!\n%s\n" % (output))
+
+    # format the string again for ldap:
+    if machines:
+        Str = "allowed_hosts=%s %s" % (",".join(machines), Str)
+
+    # Head
+    Date = time.strftime("%a, %d %b %Y %H:%M:%S +0000", time.gmtime(time.time()))
+    ErrReplyHead = "From: %s\nCc: %s\nReply-To: %s\nDate: %s\n" % (os.environ['SENDER'], os.environ['SENDER'], ReplyTo, Date)
+    Subst = {}
+    Subst["__ADMIN__"] = ReplyTo
+    Subst["__USER__"] = uid
+
+    Match = SSHFingerprint.match(output)
+    if Match is None:
+        return "Failed to match SSH fingerprint, has the output of ssh-keygen changed?"
+    g = Match.groups()
+    key_size = g[0]
+    fingerprint = g[1]
+
+    if typekey == "rsa":
+        key_size_ok = (int(key_size) >= 2048)
+    elif typekey == "ed25519":
+        key_size_ok = True
+    else:
+        key_size_ok = False
+
+    if not key_size_ok:
+        return "SSH key fails formal criteria, not added.  We only accept RSA keys (>= 2048 bits) or ed25519 keys."
+    elif fingerprint in badkeys:
+        try:
+            # Body
+            Subst["__ERROR__"] = "SSH key with fingerprint %s known as bad key" % (g[1])
+            ErrReply = TemplateSubst(Subst, open(TemplatesDir + "admin-info", "r").read())
+
+            Child = os.popen("/usr/sbin/sendmail -t", "w")
+            Child.write(ErrReplyHead)
+            Child.write(ErrReply)
+            if Child.close() is not None:
+                raise UDExecuteError("Sendmail gave a non-zero return code")
+        except Exception:
+            sys.exit(EX_TEMPFAIL)
+
+        # And now break and stop processing input, which sends a reply to the user.
+        raise UDFormatError("Submitted SSH key known to be bad and insecure, processing halted, NOTHING MODIFIED AT ALL")
+
+    global SeenKey
+    if SeenKey:
+        Attrs.append((ldap.MOD_ADD, "sshRSAAuthKey", Str))
+        return "SSH Key added: %s %s [%s]" % (key_size, fingerprint, FormatSSHAuth(Str))
+
+    Attrs.append((ldap.MOD_REPLACE, "sshRSAAuthKey", Str))
+    SeenKey = 1
+    return "SSH Keys replaced with: %s %s [%s]" % (key_size, fingerprint, FormatSSHAuth(Str))
+
 
 # Handle changing a dns entry
 #  host IN A     12.12.12.12
 #  host IN AAAA  1234::5678
 #  host IN CNAME foo.bar.    <- Trailing dot is required
 #  host IN MX    foo.bar.    <- Trailing dot is required
-def DoDNS(Str,Attrs,DnRecord):
-   cnamerecord = re.match("^[-\w]+\s+IN\s+CNAME\s+([-\w.]+\.)$",Str,re.IGNORECASE)
-   arecord     = re.match('^[-\w]+\s+IN\s+A\s+(\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3})$',Str,re.IGNORECASE)
-   mxrecord    = re.match("^[-\w]+\s+IN\s+MX\s+(\d{1,3})\s+([-\w.]+\.)$",Str,re.IGNORECASE)
-   txtrecord   = re.match("^[-\w]+\s+IN\s+TXT\s+([-\d. a-z\t<>@]+)", Str, re.IGNORECASE)
-   #aaaarecord  = re.match('^[-\w]+\s+IN\s+AAAA\s+((?:[0-9a-f]{1,4})(?::[0-9a-f]{1,4})*(?::(?:(?::[0-9a-f]{1,4})*|:))?)$',Str,re.IGNORECASE)
-   aaaarecord  = re.match('^[-\w]+\s+IN\s+AAAA\s+([A-F0-9:]{2,39})$',Str,re.IGNORECASE)
-
-   if cnamerecord is None and\
-      arecord is None and\
-      mxrecord is None and\
-      txtrecord is None and\
-      aaaarecord is None:
-     return None;
-
-   # Check if the name is already taken
-   G = re.match('^([-\w+]+)\s',Str)
-   if G is None:
-     raise UDFormatError, "Hostname not found although we already passed record syntax checks"
-   hostname = G.group(1)
-
-   # Check for collisions
-   global l;
-   # [JT 20070409 - search for both tab and space suffixed hostnames
-   #  since we accept either.  It'd probably be better to parse the
-   #  incoming string in order to construct what we feed LDAP rather
-   #  than just passing it through as is.]
-   filter = "(|(dnsZoneEntry=%s        *)(dnsZoneEntry=%s *))" % (hostname, hostname)
-   Rec = l.search_s(BaseDn,ldap.SCOPE_ONELEVEL,filter,["uid"]);
-   for x in Rec:
-      if GetAttr(x,"uid") != GetAttr(DnRecord,"uid"):
-         return "DNS entry is already owned by " + GetAttr(x,"uid")
-
-   global SeenDNS;
-   global DNS;
-
-   if cnamerecord:
-     if DNS.has_key(hostname):
-       return "CNAME and other RR types not allowed: "+Str
-     else:
-       DNS[hostname] = 2
-   else:
-     if DNS.has_key(hostname) and DNS[hostname] == 2:
-       return "CNAME and other RR types not allowed: "+Str
-     else:
-       DNS[hostname] = 1
-
-   if cnamerecord is not None:
-     sanitized = "%s IN CNAME %s" % (hostname, cnamerecord.group(1))
-   elif txtrecord is not None:
-      sanitized = "%s IN TXT %s" % (hostname, txtrecord.group(1))
-   elif arecord is not None:
-     ipaddress = arecord.group(1)
-     for quad in ipaddress.split('.'):
-       if not (int(quad) >=0 and int(quad) <= 255):
-         return "Invalid quad %s in IP address %s in line %s" %(quad, ipaddress, Str)
-     sanitized = "%s IN A %s"% (hostname, ipaddress)
-   elif mxrecord is not None:
-     priority = mxrecord.group(1)
-     mx = mxrecord.group(2)
-     sanitized = "%s IN MX %s %s" % (hostname, priority, mx)
-   elif aaaarecord is not None:
-     ipv6address = aaaarecord.group(1)
-     parts = ipv6address.split(':')
-     if len(parts) > 8:
-       return "Invalid IPv6 address (%s): too many parts"%(ipv6address)
-     if len(parts) <= 2:
-       return "Invalid IPv6 address (%s): too few parts"%(ipv6address)
-     if parts[0] == "":
-       parts.pop(0)
-     if parts[-1] == "":
-       parts.pop(-1)
-     seenEmptypart = False
-     for p in parts:
-       if len(p) > 4:
-         return "Invalid IPv6 address (%s): part %s is longer than 4 characters"%(ipv6address, p)
-       if p == "":
-         if seenEmptypart:
-           return "Invalid IPv6 address (%s): more than one :: (nothing in between colons) is not allowed"%(ipv6address)
-         seenEmptypart = True
-     sanitized = "%s IN AAAA %s" % (hostname, ipv6address)
-   else:
-     raise UDFormatError, "None of the types I recognize was it.  I shouldn't be here.  confused."
-
-   if SeenDNS:
-     Attrs.append((ldap.MOD_ADD,"dnsZoneEntry",sanitized));
-     return "DNS Entry added "+sanitized;
-
-   Attrs.append((ldap.MOD_REPLACE,"dnsZoneEntry",sanitized));
-   SeenDNS = 1;
-   return "DNS Entry replaced with "+sanitized;
+def DoDNS(Str, Attrs, DnRecord):
+    cnamerecord = re.match(r"^[-\w]+\s+IN\s+CNAME\s+([-\w.]+\.)$", Str, re.IGNORECASE)
+    arecord = re.match(r'^[-\w]+\s+IN\s+A\s+(\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3})$', Str, re.IGNORECASE)
+    mxrecord = re.match(r"^[-\w]+\s+IN\s+MX\s+(\d{1,3})\s+([-\w.]+\.)$", Str, re.IGNORECASE)
+    txtrecord = re.match(r"^[-\w]+\s+IN\s+TXT\s+([-\d. a-z\t<>@:]+)", Str, re.IGNORECASE)
+    aaaarecord = re.match(r'^[-\w]+\s+IN\s+AAAA\s+([A-F0-9:]{2,39})$', Str, re.IGNORECASE)
+
+    if cnamerecord is None and\
+       arecord is None and\
+       mxrecord is None and\
+       txtrecord is None and\
+       aaaarecord is None:
+        return None
+
+    # Check for punycode.  We ought to validate it before we allow it in our zone.
+    if Str.lower().startswith('xn--'):
+        return "Punycode not allowed: " + Str
+
+    # Check if the name is already taken
+    G = re.match(r'^([-\w+]+)\s', Str)
+    if G is None:
+        raise UDFormatError("Hostname not found although we already passed record syntax checks")
+    hostname = G.group(1)
+
+    # Check for collisions
+    global lc
+    # [JT 20070409 - search for both tab and space suffixed hostnames
+    #  since we accept either.  It'd probably be better to parse the
+    #  incoming string in order to construct what we feed LDAP rather
+    #  than just passing it through as is.]
+    filter = "(|(dnsZoneEntry=%s       *)(dnsZoneEntry=%s *))" % (hostname, hostname)
+    Rec = lc.search_s(BaseDn, ldap.SCOPE_ONELEVEL, filter, ["uid"])
+    for x in Rec:
+        if GetAttr(x, "uid") != GetAttr(DnRecord, "uid"):
+            return "DNS entry is already owned by " + GetAttr(x, "uid")
+
+    global SeenDNS
+    global DNS
+
+    if cnamerecord:
+        if hostname in DNS:
+            return "CNAME and other RR types not allowed: " + Str
+        else:
+            DNS[hostname] = 2
+    else:
+        if DNS.get(hostname) == 2:
+            return "CNAME and other RR types not allowed: " + Str
+        else:
+            DNS[hostname] = 1
+
+    if cnamerecord is not None:
+        sanitized = "%s IN CNAME %s" % (hostname, cnamerecord.group(1))
+    elif txtrecord is not None:
+        sanitized = "%s IN TXT %s" % (hostname, txtrecord.group(1))
+    elif arecord is not None:
+        ipaddress = arecord.group(1)
+        for quad in ipaddress.split('.'):
+            if not (int(quad) >= 0 and int(quad) <= 255):
+                return "Invalid quad %s in IP address %s in line %s" % (quad, ipaddress, Str)
+        sanitized = "%s IN A %s" % (hostname, ipaddress)
+    elif mxrecord is not None:
+        priority = mxrecord.group(1)
+        mx = mxrecord.group(2)
+        sanitized = "%s IN MX %s %s" % (hostname, priority, mx)
+    elif aaaarecord is not None:
+        ipv6address = aaaarecord.group(1)
+        parts = ipv6address.split(':')
+        if len(parts) > 8:
+            return "Invalid IPv6 address (%s): too many parts" % (ipv6address)
+        if len(parts) <= 2:
+            return "Invalid IPv6 address (%s): too few parts" % (ipv6address)
+        if parts[0] == "":
+            parts.pop(0)
+        if parts[-1] == "":
+            parts.pop(-1)
+        seenEmptypart = False
+        for p in parts:
+            if len(p) > 4:
+                return "Invalid IPv6 address (%s): part %s is longer than 4 characters" % (ipv6address, p)
+            if p == "":
+                if seenEmptypart:
+                    return "Invalid IPv6 address (%s): more than one :: (nothing in between colons) is not allowed" % (ipv6address)
+                seenEmptypart = True
+            sanitized = "%s IN AAAA %s" % (hostname, ipv6address)
+    else:
+        raise UDFormatError("None of the types I recognize was it.  I shouldn't be here.  confused.")
+
+    if SeenDNS:
+        Attrs.append((ldap.MOD_ADD, "dnsZoneEntry", sanitized))
+        return "DNS Entry added " + sanitized
+
+    Attrs.append((ldap.MOD_REPLACE, "dnsZoneEntry", sanitized))
+    SeenDNS = 1
+    return "DNS Entry replaced with " + sanitized
+
 
 # Handle an RBL list (mailRBL, mailRHSBL, mailWhitelist)
-def DoRBL(Str,Attrs):
-   Match = re.compile('^mail(rbl|rhsbl|whitelist) ([-a-z0-9.]+)$').match(Str.lower())
-   if Match == None:
-      return None
-   
-   if Match.group(1) == "rbl":
-      Key = "mailRBL"
-   if Match.group(1) == "rhsbl":
-      Key = "mailRHSBL"
-   if Match.group(1) == "whitelist":
-      Key = "mailWhitelist"
-   Host = Match.group(2)
-
-   global SeenList
-   if SeenList.has_key(Key):
-     Attrs.append((ldap.MOD_ADD,Key,Host))
-     return "%s added %s" % (Key,Host)
-      
-   Attrs.append((ldap.MOD_REPLACE,Key,Host))
-   SeenList[Key] = 1;
-   return "%s replaced with %s" % (Key,Host)
+def DoRBL(Str, Attrs):
+    Match = re.compile('^mail(rbl|rhsbl|whitelist) ([-a-z0-9.]+)$').match(Str.lower())
+    if Match is None:
+        return None
+
+    if Match.group(1) == "rbl":
+        Key = "mailRBL"
+    if Match.group(1) == "rhsbl":
+        Key = "mailRHSBL"
+    if Match.group(1) == "whitelist":
+        Key = "mailWhitelist"
+    Host = Match.group(2)
+
+    global SeenList
+    if Key in SeenList:
+        Attrs.append((ldap.MOD_ADD, Key, Host))
+        return "%s added %s" % (Key, Host)
+
+    Attrs.append((ldap.MOD_REPLACE, Key, Host))
+    SeenList[Key] = 1
+    return "%s replaced with %s" % (Key, Host)
+
 
 # Handle a ConfirmSudoPassword request
-def DoConfirmSudopassword(Str):
-   Match = re.compile('^confirm sudopassword ('+UUID_FORMAT+') ([a-z0-9.,*]+) ([0-9a-f]{40})$').match(Str)
-   if Match == None:
-      return None
-
-   uuid = Match.group(1)
-   hosts = Match.group(2)
-   hmac = Match.group(3)
-
-   global SudoPasswd
-   SudoPasswd[uuid] = (hosts, hmac)
-   return "got confirm for sudo password %s on host(s) %s, auth code %s" % (uuid,hosts, hmac)
-
-def FinishConfirmSudopassword(l, uid, Attrs):
-   global SudoPasswd
-   result = "\n"
-
-   res = l.search_s(BaseDn,ldap.SCOPE_ONELEVEL,"uid="+uid, ['sudoPassword']);
-   if len(res) != 1:
-      raise UDFormatError, "Not exactly one hit when searching for user"
-   if res[0][1].has_key('sudoPassword'):
-      inldap = res[0][1]['sudoPassword']
-   else:
-      inldap = []
-
-   newldap = []
-   for entry in inldap:
-      Match = re.compile('^('+UUID_FORMAT+') (confirmed:[0-9a-f]{40}|unconfirmed) ([a-z0-9.,*]+) ([^ ]+)$').match(entry)
-      if Match == None:
-         raise UDFormatError, "Could not parse existing sudopasswd entry"
-      uuid = Match.group(1)
-      status = Match.group(2)
-      hosts = Match.group(3)
-      cryptedpass = Match.group(4)
-
-      if SudoPasswd.has_key(uuid):
-         confirmedHosts = SudoPasswd[uuid][0]
-         confirmedHmac = SudoPasswd[uuid][1]
-         if status.startswith('confirmed:'):
-            if status == 'confirmed:'+make_passwd_hmac('password-is-confirmed', 'sudo', uid, uuid, hosts, cryptedpass):
-               result = result + "Entry %s for sudo password on hosts %s already confirmed.\n"%(uuid, hosts)
+def DoConfirmSudopassword(Str, SudoPasswd):
+    Match = re.compile('^confirm sudopassword (' + UUID_FORMAT + ') ([a-z0-9.,*-]+) ([0-9a-f]{40})$').match(Str)
+    if Match is None:
+        return None
+
+    uuid = Match.group(1)
+    hosts = Match.group(2)
+    hmac = Match.group(3)
+
+    SudoPasswd[uuid] = (hosts, hmac)
+    return "got confirm for sudo password %s on host(s) %s, auth code %s" % (uuid, hosts, hmac)
+
+
+def FinishConfirmSudopassword(lc, uid, Attrs, SudoPasswd):
+    result = "\n"
+
+    if len(SudoPasswd) == 0:
+        return None
+
+    res = lc.search_s(BaseDn, ldap.SCOPE_ONELEVEL, "uid=" + uid, ['sudoPassword'])
+    if len(res) != 1:
+        raise UDFormatError("Not exactly one hit when searching for user")
+    if 'sudoPassword' in res[0][1]:
+        inldap = res[0][1]['sudoPassword']
+    else:
+        inldap = []
+
+    newldap = []
+    for entry in inldap:
+        Match = re.compile('^(' + UUID_FORMAT + ') (confirmed:[0-9a-f]{40}|unconfirmed) ([a-z0-9.,*-]+) ([^ ]+)$').match(entry)
+        if Match is None:
+            raise UDFormatError("Could not parse existing sudopasswd entry")
+        uuid = Match.group(1)
+        status = Match.group(2)
+        hosts = Match.group(3)
+        cryptedpass = Match.group(4)
+
+        if uuid in SudoPasswd:
+            confirmedHosts = SudoPasswd[uuid][0]
+            confirmedHmac = SudoPasswd[uuid][1]
+            if status.startswith('confirmed:'):
+                if status == 'confirmed:' + make_passwd_hmac('password-is-confirmed', 'sudo', uid, uuid, hosts, cryptedpass):
+                    result += "Entry %s for sudo password on hosts %s already confirmed.\n" % (uuid, hosts)
+                else:
+                    result += "Entry %s for sudo password on hosts %s is listed as confirmed, but HMAC does not verify.\n" % (uuid, hosts)
+            elif confirmedHosts != hosts:
+                result += "Entry %s hostlist mismatch (%s vs. %s).\n" % (uuid, hosts, confirmedHosts)
+            elif make_passwd_hmac('confirm-new-password', 'sudo', uid, uuid, hosts, cryptedpass) == confirmedHmac:
+                result += "Entry %s for sudo password on hosts %s now confirmed.\n" % (uuid, hosts)
+                status = 'confirmed:' + make_passwd_hmac('password-is-confirmed', 'sudo', uid, uuid, hosts, cryptedpass)
             else:
-               result = result + "Entry %s for sudo password on hosts %s is listed as confirmed, but HMAC does not verify.\n"%(uuid, hosts)
-         elif confirmedHosts != hosts:
-            result = result + "Entry %s hostlist mismatch (%s vs. %s).\n"%(uuid, hosts, confirmedHosts)
-         elif make_passwd_hmac('confirm-new-password', 'sudo', uid, uuid, hosts, cryptedpass) == confirmedHmac:
-            result = result + "Entry %s for sudo password on hosts %s now confirmed.\n"%(uuid, hosts)
-            status = 'confirmed:'+make_passwd_hmac('password-is-confirmed', 'sudo', uid, uuid, hosts, cryptedpass)
-         else:
-            result = result + "Entry %s for sudo password on hosts %s HMAC verify failed.\n"%(uuid, hosts)
-         del SudoPasswd[uuid]
-
-      newentry = " ".join([uuid, status, hosts, cryptedpass])
-      if len(newldap) == 0:
-         newldap.append((ldap.MOD_REPLACE,"sudoPassword",newentry))
-      else:
-         newldap.append((ldap.MOD_ADD,"sudoPassword",newentry))
-
-   for entry in SudoPasswd:
-      result = result + "Entry %s that you confirm is not listed in ldap."%(entry)
-
-   for entry in newldap:
-      Attrs.append(entry)
-
-   return result
+                result += "Entry %s for sudo password on hosts %s HMAC verify failed.\n" % (uuid, hosts)
+            del SudoPasswd[uuid]
+
+        newentry = " ".join([uuid, status, hosts, cryptedpass])
+        if len(newldap) == 0:
+            newldap.append((ldap.MOD_REPLACE, "sudoPassword", newentry))
+        else:
+            newldap.append((ldap.MOD_ADD, "sudoPassword", newentry))
+
+    for entry in SudoPasswd:
+        result += "Entry %s that you confirm is not listed in ldap." % (entry,)
+
+    for entry in newldap:
+        Attrs.append(entry)
+
+    return result
+
 
 def connect_to_ldap_and_check_if_locked(DnRecord):
-   # Connect to the ldap server
-   l = connectLDAP()
-   F = open(PassDir+"/pass-"+pwd.getpwuid(os.getuid())[0],"r");
-   AccessPass = F.readline().strip().split(" ")
-   F.close();
-   l.simple_bind_s("uid="+AccessPass[0]+","+BaseDn,AccessPass[1]);
+    # Connect to the ldap server
+    lc = connectLDAP()
+    F = open(PassDir + "/pass-" + pwd.getpwuid(os.getuid())[0], "r")
+    AccessPass = F.readline().strip().split(" ")
+    F.close()
+    lc.simple_bind_s("uid={},{}".format(AccessPass[0], BaseDn), AccessPass[1])
 
-   # Check for a locked account
-   Attrs = l.search_s(BaseDn,ldap.SCOPE_ONELEVEL,"uid="+GetAttr(DnRecord,"uid"));
-   if (GetAttr(Attrs[0],"userPassword").find("*LK*") != -1) \
-             or GetAttr(Attrs[0],"userPassword").startswith("!"):
-      raise UDNotAllowedError, "This account is locked";
+    # Check for a locked account
+    Attrs = lc.search_s(BaseDn, ldap.SCOPE_ONELEVEL, "uid=" + GetAttr(DnRecord, "uid"))
+    if (GetAttr(Attrs[0], "userPassword").find("*LK*") != -1) \
+       or GetAttr(Attrs[0], "userPassword").startswith("!"):
+        raise UDNotAllowedError("This account is locked")
+    return lc
 
-   return l
 
 # Handle an [almost] arbitary change
-def HandleChange(Reply,DnRecord,Key):
-   global PlainText;
-   Lines = re.split("\n *\r?",PlainText);
-
-   Result = "";
-   Attrs = [];
-   Show = 0;
-   CommitChanges = 1
-   for Line in Lines: 
-      Line = Line.strip()
-      if Line == "":
-         continue;
-
-      # Try to process a command line
-      Result = Result + "> "+Line+"\n";
-      try:
-         if Line == "show":
-            Show = 1;
-            Res = "OK";
-         else:
-            badkeys = LoadBadSSH()
-            Res = DoPosition(Line,Attrs) or DoDNS(Line,Attrs,DnRecord) or \
-                  DoArbChange(Line,Attrs) or DoSSH(Line,Attrs,badkeys,GetAttr(DnRecord,"uid")) or \
-                  DoDel(Line,Attrs) or DoRBL(Line,Attrs) or DoConfirmSudopassword(Line)
-      except:
-         Res = None;
-         Result = Result + "==> %s: %s\n" %(sys.exc_type,sys.exc_value);
-
-      # Fail, if someone tries to send someone elses signed email to the
-      # daemon then we want to abort ASAP.
-      if Res == None:
-         CommitChanges = 0
-         Result = Result + "Command is not understood. Halted - no changes committed\n";
-         break;
-      Result = Result + Res + "\n";
-
-   # Connect to the ldap server
-   l = connect_to_ldap_and_check_if_locked(DnRecord)
-
-   if CommitChanges == 1: # only if we are still good to go
-      try:
-         Res = FinishConfirmSudopassword(l, GetAttr(DnRecord,"uid"), Attrs)
-         Result = Result + Res + "\n";
-      except Error, e:
-         CommitChanges = 0
-         Result = Result + "FinishConfirmSudopassword raised an error (%s) - no changes committed\n"%(e);
-
-   # Modify the record
-   if CommitChanges == 1:
-      Dn = "uid=" + GetAttr(DnRecord,"uid") + "," + BaseDn;
-      l.modify_s(Dn,Attrs);
-
-   Attribs = "";
-   if Show == 1:
-      Attrs = l.search_s(BaseDn,ldap.SCOPE_ONELEVEL,"uid="+GetAttr(DnRecord,"uid"));
-      if len(Attrs) == 0:
-         raise UDNotAllowedError, "User not found"
-      Attribs = GPGEncrypt(PrettyShow(Attrs[0])+"\n","0x"+Key[1],Key[4]);
-
-   Subst = {};
-   Subst["__FROM__"] = ChangeFrom;
-   Subst["__EMAIL__"] = EmailAddress(DnRecord);
-   Subst["__ADMIN__"] = ReplyTo;
-   Subst["__RESULT__"] = Result;
-   Subst["__ATTR__"] = Attribs;
-
-   return Reply + TemplateSubst(Subst,open(TemplatesDir+"change-reply","r").read());
-   
+def HandleChange(Reply, DnRecord, Key):
+    global PlainText
+    Lines = re.split("\n *\r?", PlainText)
+
+    Result = ""
+    Attrs = []
+    SudoPasswd = {}
+    Show = 0
+    CommitChanges = 1
+    for Line in Lines:
+        Line = Line.strip()
+        if Line == "":
+            continue
+
+        # Try to process a command line
+        Result += "> " + Line + "\n"
+        try:
+            if Line == "show":
+                Show = 1
+                Res = "OK"
+            else:
+                badkeys = LoadBadSSH()
+                Res = DoPosition(Line, Attrs) or DoDNS(Line, Attrs, DnRecord) or \
+                    DoArbChange(Line, Attrs) or DoSSH(Line, Attrs, badkeys, GetAttr(DnRecord, "uid")) or \
+                    DoDel(Line, Attrs) or DoRBL(Line, Attrs) or DoConfirmSudopassword(Line, SudoPasswd)
+        except Exception:
+            Res = None
+            Result += "==> %s: %s\n" % (sys.exc_type, sys.exc_value)
+
+        # Fail, if someone tries to send someone elses signed email to the
+        # daemon then we want to abort ASAP.
+        if Res is None:
+            CommitChanges = 0
+            Result = Result + "Command is not understood. Halted - no changes committed\n"
+            break
+        Result += Res + "\n"
+
+    # Connect to the ldap server
+    lc = connect_to_ldap_and_check_if_locked(DnRecord)
+
+    if CommitChanges == 1 and len(SudoPasswd) > 0:  # only if we are still good to go
+        try:
+            Res = FinishConfirmSudopassword(lc, GetAttr(DnRecord, "uid"), Attrs, SudoPasswd)
+            if Res is not None:
+                Result += Res + "\n"
+        except Error, e:
+            CommitChanges = 0
+            Result = Result + "FinishConfirmSudopassword raised an error (%s) - no changes committed\n" % (e)
+
+    if CommitChanges == 1 and len(Attrs) > 0:
+        Dn = "uid=" + GetAttr(DnRecord, "uid") + "," + BaseDn
+        lc.modify_s(Dn, Attrs)
+
+    Attribs = ""
+    if Show == 1:
+        Attrs = lc.search_s(BaseDn, ldap.SCOPE_ONELEVEL, "uid=" + GetAttr(DnRecord, "uid"))
+        if len(Attrs) == 0:
+            raise UDNotAllowedError("User not found")
+        Attribs = GPGEncrypt(PrettyShow(Attrs[0]) + "\n", "0x" + Key[1], Key[4])
+
+    Subst = {}
+    Subst["__FROM__"] = ChangeFrom
+    Subst["__EMAIL__"] = EmailAddress(DnRecord)
+    Subst["__ADMIN__"] = ReplyTo
+    Subst["__RESULT__"] = Result
+    Subst["__ATTR__"] = Attribs
+
+    return Reply + TemplateSubst(Subst, open(TemplatesDir + "change-reply", "r").read())
+
+
 # Handle ping handles an email sent to the 'ping' address (ie this program
 # called with a ping argument) It replies with a dump of the public records.
-def HandlePing(Reply,DnRecord,Key):
-   Subst = {};
-   Subst["__FROM__"] = PingFrom;
-   Subst["__EMAIL__"] = EmailAddress(DnRecord);
-   Subst["__LDAPFIELDS__"] = PrettyShow(DnRecord);
-   Subst["__ADMIN__"] = ReplyTo;
-
-   return Reply + TemplateSubst(Subst,open(TemplatesDir+"ping-reply","r").read());
+def HandlePing(Reply, DnRecord, Key):
+    Subst = {}
+    Subst["__FROM__"] = PingFrom
+    Subst["__EMAIL__"] = EmailAddress(DnRecord)
+    Subst["__LDAPFIELDS__"] = PrettyShow(DnRecord)
+    Subst["__ADMIN__"] = ReplyTo
 
+    return Reply + TemplateSubst(Subst, open(TemplatesDir + "ping-reply", "r").read())
 
 
 def get_crypttype_preamble(key):
-   if (key[4] == 1):
-      type = "Your message was encrypted using PGP 2.x\ncompatibility mode.";
-   else:
-      type = "Your message was encrypted using GPG (OpenPGP)\ncompatibility "\
-             "mode, without IDEA. This message cannot be decoded using PGP 2.x";
-   return type
+    if (key[4] == 1):
+        type = "Your message was encrypted using PGP 2.x\ncompatibility mode."
+    else:
+        type = "Your message was encrypted using GPG (OpenPGP)\ncompatibility "\
+               "mode, without IDEA. This message cannot be decoded using PGP 2.x"
+    return type
+
 
 # Handle a change password email sent to the change password address
 # (this program called with the chpass argument)
-def HandleChPass(Reply,DnRecord,Key):
-   # Generate a random password
-   Password = GenPass();
-   Pass = HashPass(Password);
-
-   # Use GPG to encrypt it      
-   Message = GPGEncrypt("Your new password is '" + Password + "'\n",\
-                        "0x"+Key[1],Key[4]);
-   Password = None;
-
-   if Message == None:
-      raise UDFormatError, "Unable to generate the encrypted reply, gpg failed.";
-
-   Subst = {};
-   Subst["__FROM__"] = ChPassFrom;
-   Subst["__EMAIL__"] = EmailAddress(DnRecord);
-   Subst["__CRYPTTYPE__"] = get_crypttype_preamble(Key)
-   Subst["__PASSWORD__"] = Message;
-   Subst["__ADMIN__"] = ReplyTo;
-   Reply = Reply + TemplateSubst(Subst,open(TemplatesDir+"passwd-changed","r").read());
-
-   l = connect_to_ldap_and_check_if_locked(DnRecord)
-   # Modify the password
-   Rec = [(ldap.MOD_REPLACE,"userPassword","{crypt}"+Pass),
-          (ldap.MOD_REPLACE,"shadowLastChange",str(int(time.time()/24/60/60)))];
-   Dn = "uid=" + GetAttr(DnRecord,"uid") + "," + BaseDn;
-   l.modify_s(Dn,Rec);
-
-   return Reply;
-
-def HandleChKrbPass(Reply,DnRecord,Key):
-   # Connect to the ldap server, will throw an exception if account locked.
-   l = connect_to_ldap_and_check_if_locked(DnRecord)
-
-   user = GetAttr(DnRecord,"uid")
-   krb_proc = subprocess.Popen( ('ud-krb-reset', user), stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
-   krb_proc.stdin.close()
-   out = krb_proc.stdout.readlines()
-   krb_proc.wait()
-   exitcode = krb_proc.returncode
-
-   # Use GPG to encrypt it
-   m = "Tried to reset your kerberos principal's password.\n"
-   if exitcode == 0:
-      m += "The exitcode of the reset script was zero, indicating that everything\n"
-      m += "worked.  However, this being software who knows.  Script's output below."
-   else:
-      m += "The exitcode of the reset script was %d, indicating that something\n"%(exitcode)
-      m += "went terribly, terribly wrong.  Please consult the script's output below\n"
-      m += "for more information.  Contact the admins if you have any questions or\n"
-      m += "require assitance."
-
-   m += "\n"+''.join( map(lambda x: "| "+x, out)  )
-
-   Message = GPGEncrypt(m, "0x"+Key[1],Key[4]);
-   if Message == None:
-      raise UDFormatError, "Unable to generate the encrypted reply, gpg failed.";
-
-   Subst = {};
-   Subst["__FROM__"] = ChPassFrom;
-   Subst["__EMAIL__"] = EmailAddress(DnRecord);
-   Subst["__CRYPTTYPE__"] = get_crypttype_preamble(Key)
-   Subst["__PASSWORD__"] = Message;
-   Subst["__ADMIN__"] = ReplyTo;
-   Reply = Reply + TemplateSubst(Subst,open(TemplatesDir+"passwd-changed","r").read());
-
-   return Reply;
+def HandleChPass(Reply, DnRecord, Key):
+    # Generate a random password
+    Password = GenPass()
+    Pass = HashPass(Password)
+
+    # Use GPG to encrypt it
+    Message = GPGEncrypt("Your new password is '" + Password + "'\n", "0x" + Key[1], Key[4])
+    Password = None
+
+    if Message is None:
+        raise UDFormatError("Unable to generate the encrypted reply, gpg failed.")
+
+    Subst = {}
+    Subst["__FROM__"] = ChPassFrom
+    Subst["__EMAIL__"] = EmailAddress(DnRecord)
+    Subst["__CRYPTTYPE__"] = get_crypttype_preamble(Key)
+    Subst["__PASSWORD__"] = Message
+    Subst["__ADMIN__"] = ReplyTo
+    Reply = Reply + TemplateSubst(Subst, open(TemplatesDir + "passwd-changed", "r").read())
+
+    lc = connect_to_ldap_and_check_if_locked(DnRecord)
+    # Modify the password
+    Rec = [(ldap.MOD_REPLACE, "userPassword", "{crypt}" + Pass),
+           (ldap.MOD_REPLACE, "shadowLastChange", str(int(time.time() / (24 * 60 * 60))))]
+    Dn = "uid=" + GetAttr(DnRecord, "uid") + "," + BaseDn
+    lc.modify_s(Dn, Rec)
+
+    return Reply
+
+
+def HandleChTOTPSeed(Reply, DnRecord, Key):
+    # Generate a random seed
+    seed = binascii.hexlify(open("/dev/urandom", "r").read(32))
+    random_id = binascii.hexlify(open("/dev/urandom", "r").read(32))
+    totp_file_name = "%d-%s" % (time.time(), random_id,)
+
+    msg = GPGEncrypt("Please go to %s/fetch-totp-seed.cgi?id=%s\n to fetch your TOTP seed" % (WebUILocation, totp_file_name), "0x" + Key[1], Key[4])
+
+    if msg is None:
+        raise UDFormatError("Unable to generate the encrypted reply, gpg failed.")
+
+    Subst = {}
+    Subst["__FROM__"] = ChPassFrom
+    Subst["__EMAIL__"] = EmailAddress(DnRecord)
+    Subst["__PASSWORD__"] = msg
+    Subst["__ADMIN__"] = ReplyTo
+    Reply += TemplateSubst(Subst, open(TemplatesDir + "totp-seed-changed", "r").read())
+
+    lc = connect_to_ldap_and_check_if_locked(DnRecord)
+    # Save the seed so the user can pick it up.
+    f = open(os.path.join(TOTPTicketDirectory, totp_file_name), os.O_WRONLY | os.O_CREAT)
+    print >> f, seed
+    print >> f, GetAttr(DnRecord, "uid")
+    f.close()
+
+    # Modify the password
+    Rec = [(ldap.MOD_REPLACE, "totpSeed", seed)]
+    Dn = "uid=" + GetAttr(DnRecord, "uid") + "," + BaseDn
+    lc.modify_s(Dn, Rec)
+    return Reply
+
+
+def HandleChKrbPass(Reply, DnRecord, Key):
+    # Connect to the ldap server, will throw an exception if account locked.
+    lc = connect_to_ldap_and_check_if_locked(DnRecord)
+
+    user = GetAttr(DnRecord, "uid")
+    krb_proc = subprocess.Popen(('ud-krb-reset', user), stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
+    krb_proc.stdin.close()
+    out = krb_proc.stdout.readlines()
+    krb_proc.wait()
+    exitcode = krb_proc.returncode
+
+    # Use GPG to encrypt it
+    m = "Tried to reset your kerberos principal's password.\n"
+    if exitcode == 0:
+        m += "The exitcode of the reset script was zero, indicating that everything\n"
+        m += "worked.  However, this being software who knows.  Script's output below."
+    else:
+        m += "The exitcode of the reset script was %d, indicating that something\n" % (exitcode,)
+        m += "went terribly, terribly wrong.  Please consult the script's output below\n"
+        m += "for more information.  Contact the admins if you have any questions or\n"
+        m += "require assitance."
+
+    m += "\n" + ''.join(map(lambda x: "| " + x, out))
+
+    Message = GPGEncrypt(m, "0x" + Key[1], Key[4])
+    if Message is None:
+        raise UDFormatError("Unable to generate the encrypted reply, gpg failed.")
+
+    Subst = {}
+    Subst["__FROM__"] = ChPassFrom
+    Subst["__EMAIL__"] = EmailAddress(DnRecord)
+    Subst["__CRYPTTYPE__"] = get_crypttype_preamble(Key)
+    Subst["__PASSWORD__"] = Message
+    Subst["__ADMIN__"] = ReplyTo
+    Reply += TemplateSubst(Subst, open(TemplatesDir + "passwd-changed", "r").read())
+
+    return Reply
 
 # Start of main program
 
+
 # Drop messages from a mailer daemon.
-if os.environ.has_key('SENDER') == 0 or len(os.environ['SENDER']) == 0:
-   sys.exit(0);
+if not os.environ.get('SENDER'):
+    sys.exit(0)
 
-ErrMsg = "Indeterminate Error";
-ErrType = EX_TEMPFAIL;
+ErrMsg = "Indeterminate Error"
+ErrType = EX_TEMPFAIL
 try:
-   # Startup the replay cache
-   ErrType = EX_TEMPFAIL;
-   ErrMsg = "Failed to initialize the replay cache:";
-
-   # Get the email 
-   ErrType = EX_PERMFAIL;
-   ErrMsg = "Failed to understand the email or find a signature:";
-   Email = mimetools.Message(sys.stdin,0);
-   Msg = GetClearSig(Email);
-
-   ErrMsg = "Message is not PGP signed:"
-   if Msg[0].find("-----BEGIN PGP SIGNED MESSAGE-----") == -1 and \
-      Msg[0].find("-----BEGIN PGP MESSAGE-----") == -1:
-      raise UDFormatError, "No PGP signature";
-   
-   # Check the signature
-   ErrMsg = "Unable to check the signature or the signature was invalid:";
-   pgp = GPGCheckSig2(Msg[0])
-
-   if not pgp.ok:
-      raise UDFormatError, pgp.why
-      
-   if pgp.text is None:
-      raise UDFormatError, "Null signature text"
-
-   # Extract the plain message text in the event of mime encoding
-   global PlainText;
-   ErrMsg = "Problem stripping MIME headers from the decoded message"
-   if Msg[1] == 1:
-      try:
-         Index = pgp.text.index("\n\n") + 2
-      except ValueError:
-         Index = pgp.text.index("\n\r\n") + 3
-      PlainText = pgp.text[Index:]
-   else:
-      PlainText = pgp.text
-
-   # Connect to the ldap server
-   ErrType = EX_TEMPFAIL;
-   ErrMsg = "An error occured while performing the LDAP lookup";
-   global l;
-   l = connectLDAP()
-   l.simple_bind_s("","");
-
-   # Search for the matching key fingerprint
-   Attrs = l.search_s(BaseDn,ldap.SCOPE_ONELEVEL,"keyFingerPrint=" + pgp.key_fpr)
-
-   ErrType = EX_PERMFAIL;
-   if len(Attrs) == 0:
-      raise UDFormatError, "Key not found"
-   if len(Attrs) != 1:
-      raise UDFormatError, "Oddly your key fingerprint is assigned to more than one account.."
-
-
-   # Check the signature against the replay cache
-   RC = ReplayCache(ReplayCacheFile);
-   RC.Clean();
-   ErrMsg = "The replay cache rejected your message. Check your clock!";
-   Rply = RC.Check(pgp.sig_info);
-   if Rply != None:
-      RC.close()
-      raise UDNotAllowedError, Rply;
-   RC.Add(pgp.sig_info);
-   RC.close()
-
-   # Determine the sender address
-   ErrMsg = "A problem occured while trying to formulate the reply";
-   Sender = Email.getheader("Reply-To");
-   if Sender == None:
-      Sender = Email.getheader("From");
-   if Sender == None:
-      raise UDFormatError, "Unable to determine the sender's address";
-
-   # Formulate a reply
-   Date = time.strftime("%a, %d %b %Y %H:%M:%S +0000",time.gmtime(time.time()));
-   Reply = "To: %s\nReply-To: %s\nDate: %s\n" % (Sender,ReplyTo,Date);
-
-   Res = l.search_s(HostBaseDn, ldap.SCOPE_SUBTREE, '(objectClass=debianServer)', ['hostname'] )
-   # Res is a list of tuples.
-   # The tuples contain a dn (str) and a dictionary.
-   # The dictionaries map the key "hostname" to a list.
-   # These lists contain a single hostname (str).
-   ValidHostNames = reduce(lambda a,b: a+b, [value.get("hostname", []) for (dn, value) in Res], [])
-
-   # Dispatch
-   if sys.argv[1] == "ping":
-      Reply = HandlePing(Reply,Attrs[0],pgp.key_info);
-   elif sys.argv[1] == "chpass":
-      if PlainText.strip().find("Please change my Debian password") >= 0:
-         Reply = HandleChPass(Reply,Attrs[0],pgp.key_info);
-      elif PlainText.strip().find("Please change my Kerberos password") >= 0:
-         Reply = HandleChKrbPass(Reply,Attrs[0],pgp.key_info);
-      else:
-         raise UDFormatError,"Please send a signed message where the first line of text is the string 'Please change my Debian password' or some other string we accept here.";
-   elif sys.argv[1] == "change":
-      Reply = HandleChange(Reply,Attrs[0],pgp.key_info);
-   else:
-      print sys.argv;
-      raise UDFormatError, "Incorrect Invokation";
-
-   # Send the message through sendmail      
-   ErrMsg = "A problem occured while trying to send the reply";
-   Child = os.popen("/usr/sbin/sendmail -t","w");
-#   Child = os.popen("cat","w");
-   Child.write(Reply);
-   if Child.close() != None:
-      raise UDExecuteError, "Sendmail gave a non-zero return code";
-
-except:
-   # Error Reply Header
-   Date = time.strftime("%a, %d %b %Y %H:%M:%S +0000",time.gmtime(time.time()));
-   ErrReplyHead = "To: %s\nReply-To: %s\nDate: %s\n" % (os.environ['SENDER'],ReplyTo,Date);
-
-   # Error Body
-   Subst = {};
-   Subst["__ERROR__"] = ErrMsg;
-   Subst["__ADMIN__"] = ReplyTo;
-
-   Trace = "==> %s: %s\n" %(sys.exc_type,sys.exc_value);
-   List = traceback.extract_tb(sys.exc_traceback);
-   if len(List) > 1:
-      Trace = Trace + "Python Stack Trace:\n";
-      for x in List:
-         Trace = Trace +  "   %s %s:%u: %s\n" %(x[2],x[0],x[1],x[3]);
-
-   Subst["__TRACE__"] = Trace;
-
-   # Try to send the bounce
-   try:
-      ErrReply = TemplateSubst(Subst,open(TemplatesDir+"error-reply","r").read());
-
-      Child = os.popen("/usr/sbin/sendmail -t -oi -f ''","w");
-      Child.write(ErrReplyHead);
-      Child.write(ErrReply);
-      if Child.close() != None:
-         raise UDExecuteError, "Sendmail gave a non-zero return code";
-   except:
-      sys.exit(EX_TEMPFAIL);
-      
-   if ErrType != EX_PERMFAIL:
-      sys.exit(ErrType);
-   sys.exit(0);
+    # Startup the replay cache
+    ErrType = EX_TEMPFAIL
+    ErrMsg = "Failed to initialize the replay cache:"
+
+    # Get the email
+    ErrType = EX_PERMFAIL
+    ErrMsg = "Failed to understand the email or find a signature:"
+    mail = email.parser.Parser().parse(sys.stdin)
+    Msg = GetClearSig(mail)
+
+    ErrMsg = "Message is not PGP signed:"
+    if Msg[0].find("-----BEGIN PGP SIGNED MESSAGE-----") == -1 and \
+       Msg[0].find("-----BEGIN PGP MESSAGE-----") == -1:
+        raise UDFormatError("No PGP signature")
+
+    # Check the signature
+    ErrMsg = "Unable to check the signature or the signature was invalid:"
+    pgp = GPGCheckSig2(Msg[0])
+
+    if not pgp.ok:
+        raise UDFormatError(pgp.why)
+
+    if pgp.text is None:
+        raise UDFormatError("Null signature text")
+
+    # Extract the plain message text in the event of mime encoding
+    global PlainText
+    ErrMsg = "Problem stripping MIME headers from the decoded message"
+    if Msg[1] == 1:
+        e = email.parser.Parser().parsestr(pgp.text)
+        PlainText = e.get_payload(decode=True)
+    else:
+        PlainText = pgp.text
+
+    # Connect to the ldap server
+    ErrType = EX_TEMPFAIL
+    ErrMsg = "An error occured while performing the LDAP lookup"
+    global lc
+    lc = connectLDAP()
+    lc.simple_bind_s("", "")
+
+    # Search for the matching key fingerprint
+    Attrs = lc.search_s(BaseDn, ldap.SCOPE_ONELEVEL, "keyFingerPrint={}".format(pgp.key_fpr))
+
+    ErrType = EX_PERMFAIL
+    if len(Attrs) == 0:
+        raise UDFormatError("Key not found")
+    if len(Attrs) != 1:
+        raise UDFormatError("Oddly your key fingerprint is assigned to more than one account..")
+
+    # Check the signature against the replay cache
+    RC = ReplayCache(ReplayCacheFile)
+    RC.process(pgp.sig_info)
+
+    # Determine the sender address
+    ErrMsg = "A problem occured while trying to formulate the reply"
+    Sender = mail.get('Reply-To', mail.get('From'))
+    if not Sender:
+        raise UDFormatError("Unable to determine the sender's address")
+
+    # Formulate a reply
+    Date = time.strftime("%a, %d %b %Y %H:%M:%S +0000", time.gmtime(time.time()))
+    Reply = "To: %s\nReply-To: %s\nDate: %s\n" % (Sender, ReplyTo, Date)
+
+    Res = lc.search_s(HostBaseDn, ldap.SCOPE_SUBTREE, '(objectClass=debianServer)', ['hostname'])
+    # Res is a list of tuples.
+    # The tuples contain a dn (str) and a dictionary.
+    # The dictionaries map the key "hostname" to a list.
+    # These lists contain a single hostname (str).
+    ValidHostNames = reduce(lambda a, b: a + b, [value.get("hostname", []) for (dn, value) in Res], [])
+
+    # Dispatch
+    if sys.argv[1] == "ping":
+        Reply = HandlePing(Reply, Attrs[0], pgp.key_info)
+    elif sys.argv[1] == "chpass":
+        if PlainText.strip().find("Please change my Debian password") >= 0:
+            Reply = HandleChPass(Reply, Attrs[0], pgp.key_info)
+        elif PlainText.strip().find("Please change my Kerberos password") >= 0:
+            Reply = HandleChKrbPass(Reply, Attrs[0], pgp.key_info)
+        elif PlainText.strip().find("Please change my TOTP seed") >= 0:
+            Reply = HandleChTOTPSeed(Reply, Attrs[0], pgp.key_info)
+        else:
+            raise UDFormatError("Please send a signed message where the first line of text is the string 'Please change my Debian password' or some other string we accept here.")
+    elif sys.argv[1] == "change":
+        Reply = HandleChange(Reply, Attrs[0], pgp.key_info)
+    else:
+        print sys.argv
+        raise UDFormatError("Incorrect Invokation")
+
+    # Send the message through sendmail
+    ErrMsg = "A problem occured while trying to send the reply"
+    Child = os.popen("/usr/sbin/sendmail -t", "w")
+    #   Child = os.popen("cat","w")
+    Child.write(Reply)
+    if Child.close() is not None:
+        raise UDExecuteError("Sendmail gave a non-zero return code")
+
+except Exception:
+    # Error Reply Header
+    Date = time.strftime("%a, %d %b %Y %H:%M:%S +0000", time.gmtime(time.time()))
+    ErrReplyHead = "To: %s\nReply-To: %s\nDate: %s\n" % (os.environ['SENDER'], ReplyTo, Date)
+
+    # Error Body
+    Subst = {}
+    Subst["__ERROR__"] = ErrMsg
+    Subst["__ADMIN__"] = ReplyTo
+
+    Trace = "==> %s: %s\n" % (sys.exc_type, sys.exc_value)
+    List = traceback.extract_tb(sys.exc_traceback)
+    if len(List) > 1:
+        Trace = Trace + "Python Stack Trace:\n"
+        for x in List:
+            Trace = Trace + "   %s %s:%u: %s\n" % (x[2], x[0], x[1], x[3])
+
+    Subst["__TRACE__"] = Trace
+
+    # Try to send the bounce
+    try:
+        ErrReply = TemplateSubst(Subst, open(TemplatesDir + "error-reply", "r").read())
+
+        Child = os.popen("/usr/sbin/sendmail -t -oi -f ''", "w")
+        Child.write(ErrReplyHead)
+        Child.write(ErrReply)
+        if Child.close() is not None:
+            raise UDExecuteError("Sendmail gave a non-zero return code")
+    except Exception:
+        sys.exit(EX_TEMPFAIL)
+
+    if ErrType != EX_PERMFAIL:
+        sys.exit(ErrType)
+    sys.exit(0)
 
 # vim:set et:
-# vim:set ts=3:
-# vim:set shiftwidth=3:
+# vim:set ts=4:
+# vim:set shiftwidth=4: