Update from samosa: Only use SSH2AuthSplit.match(), maybe it's superior?
[mirror/userdir-ldap.git] / ud-mailgate
index 6764c88..ebb5d45 100755 (executable)
@@ -15,38 +15,38 @@ ReplayCacheFile = ConfModule.replaycachefile;
 EX_TEMPFAIL = 75;
 EX_PERMFAIL = 65;      # EX_DATAERR
 Error = 'Message Error';
-SeenRSA = 0;
-SeenDSA = 0;
+SeenKey = 0;
 SeenDNS = 0;
+DNS = {}
 
 ArbChanges = {"c": "..",
              "l": ".*",
-             "facsimiletelephonenumber": ".*",
-             "telephonenumber": ".*",
-             "postaladdress": ".*",
-             "postalcode": ".*",
-              "loginshell": ".*",
-              "emailforward": "^([^<>@]+@.+)?$",
-              "ircnick": ".*",
-              "icquin": "^[0-9]*$",
-              "onvacation": ".*",
-              "labeledurl": ".*"};
+             "facsimileTelephoneNumber": ".*",
+             "telephoneNumber": ".*",
+             "postalAddress": ".*",
+             "postalCode": ".*",
+              "loginShell": ".*",
+              "emailForward": "^([^<>@]+@.+)?$",
+              "ircNick": ".*",
+              "icqUin": "^[0-9]*$",
+              "onVacation": ".*",
+              "labledURI": ".*"};
 
 DelItems = {"c": None,
             "l": None,
-            "facsimiletelephonenumber": None,
-            "telephonenumber": None,
-            "postaladdress": None,
-            "postalcode": None,
-            "emailforward": None,
-            "ircnick": None,
-            "onvacation": None,
-            "labeledurl": None,
+            "facsimileTelephoneNumber": None,
+            "telephoneNumber": None,
+            "postalAddress": None,
+            "postalCode": None,
+            "emailForward": None,
+            "ircNick": None,
+            "onVacation": None,
+            "labledURI": None,
            "latitude": None,
            "longitude": None,
-            "icquin": None,
-           "sshrsaauthkey": None,
-           "sshdsaauthkey": None};
+            "icqUin": None,
+           "sshRSAAuthKey": None,
+           "sshDSAAuthKey": None};
 
 # Decode a GPS location from some common forms
 def LocDecode(Str,Dir):
@@ -158,64 +158,61 @@ def DoPosition(Str,Attrs):
    Attrs.append((ldap.MOD_REPLACE,"longitude",sLong));
    return "Position set to %s/%s (%s/%s decimal degrees)"%(sLat,sLong,Lat,Long);
 
-# Handle a SSH RSA authentication key, the line format is:
+# Handle an SSH authentication key, the line format is:
 #  [options] 1024 35 13188913666680[..] [comment]
 def DoSSH(Str,Attrs):
-   Match = SSHAuthSplit.match(Str);
+   Match = SSH2AuthSplit.match(Str);
    if Match == None:
       return None;
    
-   global SeenRSA;
-   if SeenRSA:
-     Attrs.append((ldap.MOD_ADD,"sshrsaauthkey",Str));
+   global SeenKey;
+   if SeenKey:
+     Attrs.append((ldap.MOD_ADD,"sshRSAAuthKey",Str));
      return "SSH Key added "+FormatSSHAuth(Str);
       
-   Attrs.append((ldap.MOD_REPLACE,"sshrsaauthkey",Str));
-   SeenRSA = 1;
+   Attrs.append((ldap.MOD_REPLACE,"sshRSAAuthKey",Str));
+   SeenKey = 1;
    return "SSH Keys replaced with "+FormatSSHAuth(Str);
 
-# Handle a SSH DSA authentication key, the line format is:
-#  ssh-dss [key] [comment]
-def DoSSH2(Str,Attrs):
-   Match = SSHDSAAuthSplit.match(Str);
-   if Match == None:
-      return None;
-
-   global SeenDSA;
-   if SeenDSA:
-     Attrs.append((ldap.MOD_ADD,"sshdsaauthkey",Str));
-     return "SSH2 Key added "+FormatSSH2Auth(Str);
-      
-   Attrs.append((ldap.MOD_REPLACE,"sshdsaauthkey",Str));
-   SeenDSA = 1;
-   return "SSH2 Keys replaced with "+FormatSSH2Auth(Str);
-
 # Handle changing a dns entry
 #  host in a 12.12.12.12
 #  host in cname foo.bar.    <- Trailing dot is required
 def DoDNS(Str,Attrs,DnRecord):
-   if re.match('^[\w-]+\s+in\s+a\s+\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}$',\
-        Str,re.IGNORECASE) == None and \
-      re.match("^[\w-]+\s+in\s+cname\s+[\w.\-]+\.$",Str,re.IGNORECASE) == None and \
-      re.match("^[\w-]+\s+in\s+mx\s+\d{1,3}\s+[\w.\-]+\.$",Str,re.IGNORECASE) == None:
+   cname = re.match("^[-\w]+\s+in\s+cname\s+[-\w.]+\.$",Str,re.IGNORECASE);
+   if re.match('^[-\w]+\s+in\s+a\s+\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}$',\
+        Str,re.IGNORECASE) == None and cname == None and \
+      re.match("^[-\w]+\s+in\s+mx\s+\d{1,3}\s+[-\w.]+\.$",Str,re.IGNORECASE) == None:
      return None;     
 
    # Check if the name is already taken
-   G = re.match('^([\w-+]+)\s',Str).groups();
+   G = re.match('^([-\w+]+)\s',Str).groups();
 
    # Check for collisions
    global l;
-   Rec = l.search_s(BaseDn,ldap.SCOPE_ONELEVEL,"dnszoneentry="+G[0]+" *",["uid"]);
+   Rec = l.search_s(BaseDn,ldap.SCOPE_ONELEVEL,"dnsZoneEntry="+G[0]+" *",["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 cname:
+     if DNS.has_key(G[0]):
+       return "CNAME and other RR types not allowed: "+Str
+     else:
+       DNS[G[0]] = 2
+   else:
+     if DNS.has_key(G[0]) and DNS[G[0]] == 2:
+       return "CNAME and other RR types not allowed: "+Str
+     else:
+       DNS[G[0]] = 1
+     
    if SeenDNS:
-     Attrs.append((ldap.MOD_ADD,"dnszoneentry",Str));
+     Attrs.append((ldap.MOD_ADD,"dnsZoneEntry",Str));
      return "DNS Entry added "+Str;
       
-   Attrs.append((ldap.MOD_REPLACE,"dnszoneentry",Str));
+   Attrs.append((ldap.MOD_REPLACE,"dnsZoneEntry",Str));
    SeenDNS = 1;
    return "DNS Entry replaced with "+Str;
 
@@ -241,7 +238,7 @@ def HandleChange(Reply,DnRecord,Key):
          else:
            Res = DoPosition(Line,Attrs) or DoDNS(Line,Attrs,DnRecord) or \
                  DoArbChange(Line,Attrs) or DoSSH(Line,Attrs) or \
-                DoSSH2(Line,Attrs) or DoDel(Line,Attrs);
+                DoDel(Line,Attrs);
       except:
          Res = None;
          Result = Result + "==> %s: %s\n" %(sys.exc_type,sys.exc_value);
@@ -329,7 +326,7 @@ def HandleChPass(Reply,DnRecord,Key):
 
    # Check for a locked account
    Attrs = l.search_s(BaseDn,ldap.SCOPE_ONELEVEL,"uid="+GetAttr(DnRecord,"uid"));
-   if (string.find(GetAttr(Attrs[0],"userpassword"),"*LK*")  != -1):
+   if (string.find(GetAttr(Attrs[0],"userPassword"),"*LK*")  != -1):
       raise Error, "This account is locked";
 
    # Modify the password
@@ -361,7 +358,8 @@ try:
    Msg = GetClearSig(Email);
 
    ErrMsg = "Message is not PGP signed:"
-   if string.find(Msg[0],"-----BEGIN PGP SIGNED MESSAGE-----") == -1:
+   if string.find(Msg[0],"-----BEGIN PGP SIGNED MESSAGE-----") == -1 and \
+      string.find(Msg[0],"-----BEGIN PGP MESSAGE-----") == -1:
       raise Error, "No PGP signature";
    
    # Check the signature
@@ -401,7 +399,7 @@ try:
    l.simple_bind_s("","");
 
    # Search for the matching key fingerprint
-   Attrs = l.search_s(BaseDn,ldap.SCOPE_ONELEVEL,"keyfingerprint=" + Res[2][1]);
+   Attrs = l.search_s(BaseDn,ldap.SCOPE_ONELEVEL,"keyFingerPrint=" + Res[2][1]);
    if len(Attrs) == 0:
       raise Error, "Key not found"
    if len(Attrs) != 1:
@@ -416,7 +414,7 @@ try:
    if Sender == None:
       raise Error, "Unable to determine the sender's address";
 
-   if (string.find(GetAttr(Attrs[0],"userpassword"),"*LK*")  != -1):
+   if (string.find(GetAttr(Attrs[0],"userPassword"),"*LK*")  != -1):
       raise Error, "This account is locked";
 
    # Formulate a reply