parse machine specifications for ssh keys
[mirror/userdir-ldap.git] / ud-mailgate
1 #!/usr/bin/env python
2 # -*- mode: python -*-
3
4 #   Prior copyright probably rmurray, troup, joey, jgg -- weasel 2008
5 #   Copyright (c) 2009 Stephen Gran <steve@lobefin.net>
6 #   Copyright (c) 2008 Peter Palfrader <peter@palfrader.org>
7 #   Copyright (c) 2008 Joerg Jaspert <joerg@debian.org>
8 #   Copyright (c) 2010 Helmut Grohne <helmut@subdivi.de>
9
10 import userdir_gpg, userdir_ldap, sys, traceback, time, ldap, os, commands
11 import pwd, tempfile
12
13 from userdir_gpg import *
14 from userdir_ldap import *
15 from userdir_exceptions import *
16
17 # Error codes from /usr/include/sysexits.h
18 ReplyTo = ConfModule.replyto;
19 PingFrom = ConfModule.pingfrom;
20 ChPassFrom = ConfModule.chpassfrom;
21 ChangeFrom = ConfModule.changefrom;
22 ReplayCacheFile = ConfModule.replaycachefile;
23 SSHFingerprintFile = ConfModule.fingerprintfile
24
25 UUID_FORMAT = '[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}'
26
27 EX_TEMPFAIL = 75;
28 EX_PERMFAIL = 65;      # EX_DATAERR
29 Error = 'Message Error';
30 SeenKey = 0;
31 SeenDNS = 0;
32 mailRBL = {}
33 mailRHSBL = {}
34 mailWhitelist = {}
35 SeenList = {}
36 DNS = {}
37 SudoPasswd = {}
38
39 SSHFingerprint = re.compile('^(\d+) ([0-9a-f\:]{47}) (.+)$')
40 SSHRSA1Match = re.compile('^^(.* )?\d+ \d+ \d+')
41
42 GenderTable = {"male": 1,
43                "1": 1,
44                "female": 2,
45                "2": 2,
46                "unspecified": 9,
47                "9": 9,
48 };
49
50 ArbChanges = {"c": "..",
51               "l": ".*",
52               "facsimileTelephoneNumber": ".*",
53               "telephoneNumber": ".*",
54               "postalAddress": ".*",
55          "bATVToken": ".*",
56               "postalCode": ".*",
57               "loginShell": ".*",
58               "emailForward": "^([^<>@]+@.+)?$",
59               "jabberJID": "^([^<>@]+@.+)?$",
60               "ircNick": ".*",
61               "icqUin": "^[0-9]*$",
62               "onVacation": ".*",
63               "labeledURI": ".*",
64               "birthDate": "^([0-9]{4})([01][0-9])([0-3][0-9])$",
65               "mailDisableMessage": ".*",
66               "mailGreylisting": "^(TRUE|FALSE)$",
67               "mailCallout": "^(TRUE|FALSE)$",
68               "mailDefaultOptions": "^(TRUE|FALSE)$",
69               "VoIP": ".*",
70               "gender": "^(1|2|9|male|female|unspecified)$",
71          "mailContentInspectionAction": "^(reject|blackhole|markup)$",
72 };
73
74 DelItems = {"c": None,
75             "l": None,
76             "facsimileTelephoneNumber": None,
77             "telephoneNumber": None,
78             "postalAddress": None,
79             "bATVToken": None,
80             "postalCode": None,
81             "emailForward": None,
82             "ircNick": None,
83             "onVacation": None,
84             "labeledURI": None,
85             "latitude": None,
86             "longitude": None,
87             "icqUin": None,
88             "jabberJID": None,
89             "jpegPhoto": None,
90             "dnsZoneEntry": None,
91             "sshRSAAuthKey": None,
92             "birthDate" : None,
93             "mailGreylisting": None,
94             "mailCallout": None,
95             "mailRBL": None,
96             "mailRHSBL": None,
97             "mailWhitelist": None,
98             "mailDisableMessage": None,
99             "mailDefaultOptions": None,
100             "VoIP": None,
101             "mailContentInspectionAction": None,
102             };
103
104
105 # Decode a GPS location from some common forms
106 def LocDecode(Str,Dir):
107    # Check for Decimal degrees, DGM, or DGMS
108    if re.match("^[+-]?[\d.]+$",Str) != None:
109       return Str;
110
111    Deg = '0'; Min = None; Sec = None; Dr = Dir[0];
112    
113    # Check for DDDxMM.MMMM where x = [nsew]
114    Match = re.match("^(\d+)(["+Dir+"])([\d.]+)$",Str);
115    if Match != None:
116       G = Match.groups();
117       Deg = G[0]; Min = G[2]; Dr = G[1];
118
119    # Check for DD.DD x 
120    Match = re.match("^([\d.]+) ?(["+Dir+"])$",Str);
121    if Match != None:
122       G = Match.groups();
123       Deg = G[0]; Dr = G[1];
124
125    # Check for DD:MM.MM x 
126    Match = re.match("^(\d+):([\d.]+) ?(["+Dir+"])$",Str);
127    if Match != None:
128       G = Match.groups();
129       Deg = G[0]; Min = G[1]; Dr = G[2];
130
131    # Check for DD:MM:SS.SS x
132    Match = re.match("^(\d+):(\d+):([\d.]+) ?(["+Dir+"])$",Str);
133    if Match != None:
134       G = Match.groups();
135       Deg = G[0]; Min = G[1]; Sec = G[2]; Dr = G[3];
136       
137    # Some simple checks
138    if float(Deg) > 180:
139       raise UDFormatError, "Bad degrees";
140    if Min != None and float(Min) > 60:
141       raise UDFormatError, "Bad minutes";
142    if Sec != None and float(Sec) > 60:
143       raise UDFormatError, "Bad seconds";
144       
145    # Pad on an extra leading 0 to disambiguate small numbers
146    if len(Deg) <= 1 or Deg[1] == '.':
147       Deg = '0' + Deg;
148    if Min != None and (len(Min) <= 1 or Min[1] == '.'):
149       Min = '0' + Min;
150    if Sec != None and (len(Sec) <= 1 or Sec[1] == '.'):
151       Sec = '0' + Sec;
152    
153    # Construct a DGM/DGMS type value from the components.
154    Res = "+"
155    if Dr == Dir[1]:
156       Res = "-";
157    Res = Res + Deg;
158    if Min != None:
159       Res = Res + Min;
160    if Sec != None:
161       Res = Res + Sec;
162    return Res;
163               
164 # Handle changing a set of arbitary fields
165 #  <field>: value
166 def DoArbChange(Str,Attrs):
167    Match = re.match("^([^ :]+): (.*)$",Str);
168    if Match == None:
169       return None;
170    G = Match.groups();
171
172    attrName = G[0].lower();
173    for i in ArbChanges.keys():
174       if i.lower() == attrName:
175          attrName = i;
176          break;
177    if ArbChanges.has_key(attrName) == 0:
178       return None;
179
180    if re.match(ArbChanges[attrName],G[1]) == None:
181       raise UDFormatError, "Item does not match the required format"+ArbChanges[attrName];
182
183    value = G[1];
184    if attrName == 'gender':
185       if G[1] not in GenderTable:
186          raise UDFormatError, "Gender not found in table"
187       value = GenderTable[G[1]]
188
189 #   if attrName == 'birthDate':
190 #      (re.match("^([0-9]{4})([01][0-9])([0-3][0-9])$",G[1]) {
191 #    $bd_yr = $1; $bd_mo = $2; $bd_day = $3;
192 #    if ($bd_mo > 0 and $bd_mo <= 12 and $bd_day > 0) {
193 #      if ($bd_mo == 2) {
194 #        if ($bd_day == 29 and ($bd_yr == 0 or ($bd_yr % 4 == 0 && ($bd_yr % 100 != 0 || $bd_yr % 400 == 0)))) {
195 #          $bd_ok = 1;
196 #        } elsif ($bd_day <= 28) {
197 #          $bd_ok = 1;
198 #        }
199 #      } elsif ($bd_mo == 4 or $bd_mo == 6 or $bd_mo == 9 or $bd_mo == 11) {
200 #       if ($bd_day <= 30) {
201 #         $bd_ok = 1;
202 #       }
203 #      } else {
204 #       if ($bd_day <= 31) {
205 #         $bd_ok = 1;
206 #       }
207 #      }
208 #    }
209 #  } elsif (not defined($query->param('birthdate')) or $query->param('birthdate') =~ /^\s*$/) {
210 #    $bd_ok = 1;
211 #  }
212    Attrs.append((ldap.MOD_REPLACE,attrName,value));
213    return "Changed entry %s to %s"%(attrName,value);
214
215 # Handle changing a set of arbitary fields
216 #  <field>: value
217 def DoDel(Str,Attrs):
218    Match = re.match("^del (.*)$",Str);
219    if Match == None:
220       return None;
221    G = Match.groups();
222
223    attrName = G[0].lower();
224    for i in DelItems.keys():
225       if i.lower() == attrName:
226          attrName = i;
227          break;
228    if DelItems.has_key(attrName) == 0:
229       return "Cannot erase entry %s"%(attrName);
230
231    Attrs.append((ldap.MOD_DELETE,attrName,None));
232    return "Removed entry %s"%(attrName);
233
234 # Handle a position change message, the line format is:
235 #  Lat: -12412.23 Long: +12341.2342
236 def DoPosition(Str,Attrs):
237    Match = re.match("^lat: ([+\-]?[\d:.ns]+(?: ?[ns])?) long: ([+\-]?[\d:.ew]+(?: ?[ew])?)$", Str.lower())
238    if Match == None:
239       return None;
240
241    G = Match.groups();
242    try:
243       sLat = LocDecode(G[0],"ns");
244       sLong = LocDecode(G[1],"ew");
245       Lat = DecDegree(sLat,1);
246       Long = DecDegree(sLong,1);
247    except:
248       raise UDFormatError, "Positions were found, but they are not correctly formed";
249
250    Attrs.append((ldap.MOD_REPLACE,"latitude",sLat));
251    Attrs.append((ldap.MOD_REPLACE,"longitude",sLong));
252    return "Position set to %s/%s (%s/%s decimal degrees)"%(sLat,sLong,Lat,Long);
253
254 # Load bad ssh fingerprints
255 def LoadBadSSH():
256    f = open(SSHFingerprintFile, "r")
257    bad = []
258    FingerprintLine = re.compile('^([0-9a-f\:]{47}).*$')
259    for line in f.readlines():
260       Match = FingerprintLine.match(line)
261       if Match is not None:
262          g = Match.groups()
263          bad.append(g[0])
264    return bad
265
266 # Handle an SSH authentication key, the line format is:
267 #  [options] 1024 35 13188913666680[..] [comment]
268 # maybe it really should be:
269 # [allowed_hosts=machine1,machine2 ][options ]ssh-rsa keybytes [comment]
270 machine_regex = re.compile("^[0-9a-zA-Z.-]+$")
271 def DoSSH(Str, Attrs, badkeys, uid):
272    Match = SSH2AuthSplit.match(Str);
273    if Match == None:
274       return None;
275    g = Match.groups()
276    typekey = g[1]
277    if Match == None:
278       Match = SSHRSA1Match.match(Str)
279       if Match is not None:
280          return "RSA1 keys not supported anymore"
281       return None;
282
283    # lines can now be prepended with "allowed_hosts=machine1,machine2 "
284    if Str.startswith("allowed_hosts="):
285       Str = Str.split("=", 1)[1]
286       if ' ' not in Str:
287          return "invalid ssh key syntax with machine specification"
288       machines, Str = Str.split(' ', 1)
289       machines = machines.split(",")
290       for m in machines:
291          if not m:
292             return "empty machine specification for ssh key"
293          if not machine_regex.match(m):
294             return "machine specification for ssh key contains invalid characters"
295
296    (fd, path) = tempfile.mkstemp(".pub", "sshkeytry", "/tmp")
297    f = open(path, "w")
298    f.write("%s\n" % (Str))
299    f.close()
300    cmd = "/usr/bin/ssh-keygen -l -f %s < /dev/null" % (path)
301    (result, output) = commands.getstatusoutput(cmd)
302    os.remove(path)
303    if (result != 0):
304       raise UDExecuteError, "ssh-keygen -l invocation failed!\n%s\n" % (output)
305
306
307    # Head
308    Date = time.strftime("%a, %d %b %Y %H:%M:%S +0000",time.gmtime(time.time()))
309    ErrReplyHead = "From: %s\nCc: %s\nReply-To: %s\nDate: %s\n" % (os.environ['SENDER'],os.environ['SENDER'],ReplyTo,Date)
310    Subst = {}
311    Subst["__ADMIN__"] = ReplyTo
312    Subst["__USER__"] = uid
313
314    Match = SSHFingerprint.match(output)
315    g = Match.groups()
316
317    if int(g[0]) < 1024:
318       try:
319          # Body
320          Subst["__ERROR__"] = "SSH keysize %s is below limit 1024" % (g[0])
321          ErrReply = TemplateSubst(Subst,open(TemplatesDir+"admin-info","r").read())
322
323          Child = os.popen("/usr/sbin/sendmail -t","w")
324          Child.write(ErrReplyHead)
325          Child.write(ErrReply)
326          if Child.close() != None:
327             raise UDExecuteError, "Sendmail gave a non-zero return code"
328       except:
329          sys.exit(EX_TEMPFAIL)
330
331       # And now break and stop processing input, which sends a reply to the user.
332       raise UDFormatError, "SSH keys must have at least 1024 bits, processing halted, NOTHING MODIFIED AT ALL"
333    elif g[1] in badkeys:
334       try:
335          # Body
336          Subst["__ERROR__"] = "SSH key with fingerprint %s known as bad key" % (g[1])
337          ErrReply = TemplateSubst(Subst,open(TemplatesDir+"admin-info","r").read())
338
339          Child = os.popen("/usr/sbin/sendmail -t","w")
340          Child.write(ErrReplyHead)
341          Child.write(ErrReply)
342          if Child.close() != None:
343             raise UDExecuteError, "Sendmail gave a non-zero return code"
344       except:
345          sys.exit(EX_TEMPFAIL)
346
347       # And now break and stop processing input, which sends a reply to the user.
348       raise UDFormatError, "Submitted SSH Key known to be bad and insecure, processing halted, NOTHING MODIFIED AT ALL"
349
350    if (typekey == "dss"):
351       return "DSA keys not accepted anymore"
352
353    global SeenKey;
354    if SeenKey:
355      Attrs.append((ldap.MOD_ADD,"sshRSAAuthKey",Str));
356      return "SSH Key added "+FormatSSHAuth(Str);
357       
358    Attrs.append((ldap.MOD_REPLACE,"sshRSAAuthKey",Str));
359    SeenKey = 1;
360    return "SSH Keys replaced with "+FormatSSHAuth(Str);
361
362 # Handle changing a dns entry
363 #  host IN A     12.12.12.12
364 #  host IN AAAA  1234::5678
365 #  host IN CNAME foo.bar.    <- Trailing dot is required
366 #  host IN MX    foo.bar.    <- Trailing dot is required
367 def DoDNS(Str,Attrs,DnRecord):
368    cnamerecord = re.match("^[-\w]+\s+IN\s+CNAME\s+([-\w.]+\.)$",Str,re.IGNORECASE)
369    arecord     = re.match('^[-\w]+\s+IN\s+A\s+(\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3})$',Str,re.IGNORECASE)
370    mxrecord    = re.match("^[-\w]+\s+IN\s+MX\s+(\d{1,3})\s+([-\w.]+\.)$",Str,re.IGNORECASE)
371    #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)
372    aaaarecord  = re.match('^[-\w]+\s+IN\s+AAAA\s+([A-F0-9:]{2,39})$',Str,re.IGNORECASE)
373
374    if cnamerecord == None and\
375       arecord == None and\
376       mxrecord == None and\
377       aaaarecord == None:
378      return None;
379
380    # Check if the name is already taken
381    G = re.match('^([-\w+]+)\s',Str)
382    if G == None:
383      raise UDFormatError, "Hostname not found although we already passed record syntax checks"
384    hostname = G.group(1)
385
386    # Check for collisions
387    global l;
388    # [JT 20070409 - search for both tab and space suffixed hostnames
389    #  since we accept either.  It'd probably be better to parse the
390    #  incoming string in order to construct what we feed LDAP rather
391    #  than just passing it through as is.]
392    filter = "(|(dnsZoneEntry=%s *)(dnsZoneEntry=%s *))" % (hostname, hostname)
393    Rec = l.search_s(BaseDn,ldap.SCOPE_ONELEVEL,filter,["uid"]);
394    for x in Rec:
395       if GetAttr(x,"uid") != GetAttr(DnRecord,"uid"):
396          return "DNS entry is already owned by " + GetAttr(x,"uid")
397
398    global SeenDNS;
399    global DNS;
400
401    if cnamerecord:
402      if DNS.has_key(hostname):
403        return "CNAME and other RR types not allowed: "+Str
404      else:
405        DNS[hostname] = 2
406    else:
407      if DNS.has_key(hostname) and DNS[hostname] == 2:
408        return "CNAME and other RR types not allowed: "+Str
409      else:
410        DNS[hostname] = 1
411
412    if cnamerecord != None:
413      sanitized = "%s IN CNAME %s" % (hostname, cnamerecord.group(1))
414    elif arecord != None:
415      ipaddress = arecord.group(1)
416      for quad in ipaddress.split('.'):
417        if not (int(quad) >=0 and int(quad) <= 255):
418          return "Invalid quad %s in IP address %s in line %s" %(quad, ipaddress, Str)
419      sanitized = "%s IN A %s"% (hostname, ipaddress)
420    elif mxrecord != None:
421      priority = mxrecord.group(1)
422      mx = mxrecord.group(2)
423      sanitized = "%s IN MX %s %s" % (hostname, priority, mx)
424    elif aaaarecord != None:
425      ipv6address = aaaarecord.group(1)
426      parts = ipv6address.split(':')
427      if len(parts) > 8:
428        return "Invalid IPv6 address (%s): too many parts"%(ipv6address)
429      if len(parts) <= 2:
430        return "Invalid IPv6 address (%s): too few parts"%(ipv6address)
431      if parts[0] == "":
432        parts.pop(0)
433      if parts[-1] == "":
434        parts.pop(-1)
435      seenEmptypart = False
436      for p in parts:
437        if len(p) > 4:
438          return "Invalid IPv6 address (%s): part %s is longer than 4 characters"%(ipv6address, p)
439        if p == "":
440          if seenEmptypart:
441            return "Invalid IPv6 address (%s): more than one :: (nothing in between colons) is not allowed"%(ipv6address)
442          seenEmptypart = True
443      sanitized = "%s IN AAAA %s" % (hostname, ipv6address)
444    else:
445      raise UDFormatError, "None of the types I recognize was it.  I shouldn't be here.  confused."
446
447    if SeenDNS:
448      Attrs.append((ldap.MOD_ADD,"dnsZoneEntry",sanitized));
449      return "DNS Entry added "+sanitized;
450
451    Attrs.append((ldap.MOD_REPLACE,"dnsZoneEntry",sanitized));
452    SeenDNS = 1;
453    return "DNS Entry replaced with "+sanitized;
454
455 # Handle an RBL list (mailRBL, mailRHSBL, mailWhitelist)
456 def DoRBL(Str,Attrs):
457    Match = re.compile('^mail(rbl|rhsbl|whitelist) ([-a-z0-9.]+)$').match(Str.lower())
458    if Match == None:
459       return None
460    
461    if Match.group(1) == "rbl":
462       Key = "mailRBL"
463    if Match.group(1) == "rhsbl":
464       Key = "mailRHSBL"
465    if Match.group(1) == "whitelist":
466       Key = "mailWhitelist"
467    Host = Match.group(2)
468
469    global SeenList
470    if SeenList.has_key(Key):
471      Attrs.append((ldap.MOD_ADD,Key,Host))
472      return "%s added %s" % (Key,Host)
473       
474    Attrs.append((ldap.MOD_REPLACE,Key,Host))
475    SeenList[Key] = 1;
476    return "%s replaced with %s" % (Key,Host)
477
478 # Handle a ConfirmSudoPassword request
479 def DoConfirmSudopassword(Str):
480    Match = re.compile('^confirm sudopassword ('+UUID_FORMAT+') ([a-z0-9.,*]+) ([0-9a-f]{40})$').match(Str)
481    if Match == None:
482       return None
483
484    uuid = Match.group(1)
485    hosts = Match.group(2)
486    hmac = Match.group(3)
487
488    global SudoPasswd
489    SudoPasswd[uuid] = (hosts, hmac)
490    return "got confirm for sudo password %s on host(s) %s, auth code %s" % (uuid,hosts, hmac)
491
492 def FinishConfirmSudopassword(l, uid, Attrs):
493    global SudoPasswd
494    result = "\n"
495
496    res = l.search_s(BaseDn,ldap.SCOPE_ONELEVEL,"uid="+uid, ['sudoPassword']);
497    if len(res) != 1:
498       raise UDFormatError, "Not exactly one hit when searching for user"
499    if res[0][1].has_key('sudoPassword'):
500       inldap = res[0][1]['sudoPassword']
501    else:
502       inldap = []
503
504    newldap = []
505    for entry in inldap:
506       Match = re.compile('^('+UUID_FORMAT+') (confirmed:[0-9a-f]{40}|unconfirmed) ([a-z0-9.,*]+) ([^ ]+)$').match(entry)
507       if Match == None:
508          raise UDFormatError, "Could not parse existing sudopasswd entry"
509       uuid = Match.group(1)
510       status = Match.group(2)
511       hosts = Match.group(3)
512       cryptedpass = Match.group(4)
513
514       if SudoPasswd.has_key(uuid):
515          confirmedHosts = SudoPasswd[uuid][0]
516          confirmedHmac = SudoPasswd[uuid][1]
517          if status.startswith('confirmed:'):
518             if status == 'confirmed:'+make_passwd_hmac('password-is-confirmed', 'sudo', uid, uuid, hosts, cryptedpass):
519                result = result + "Entry %s for sudo password on hosts %s already confirmed.\n"%(uuid, hosts)
520             else:
521                result = result + "Entry %s for sudo password on hosts %s is listed as confirmed, but HMAC does not verify.\n"%(uuid, hosts)
522          elif confirmedHosts != hosts:
523             result = result + "Entry %s hostlist mismatch (%s vs. %s).\n"%(uuid, hosts, confirmedHosts)
524          elif make_passwd_hmac('confirm-new-password', 'sudo', uid, uuid, hosts, cryptedpass) == confirmedHmac:
525             result = result + "Entry %s for sudo password on hosts %s now confirmed.\n"%(uuid, hosts)
526             status = 'confirmed:'+make_passwd_hmac('password-is-confirmed', 'sudo', uid, uuid, hosts, cryptedpass)
527          else:
528             result = result + "Entry %s for sudo password on hosts %s HMAC verify failed.\n"%(uuid, hosts)
529          del SudoPasswd[uuid]
530
531       newentry = " ".join([uuid, status, hosts, cryptedpass])
532       if len(newldap) == 0:
533          newldap.append((ldap.MOD_REPLACE,"sudoPassword",newentry))
534       else:
535          newldap.append((ldap.MOD_ADD,"sudoPassword",newentry))
536
537    for entry in SudoPasswd:
538       result = result + "Entry %s that you confirm is not listed in ldap."%(entry)
539
540    for entry in newldap:
541       Attrs.append(entry)
542
543    return result
544
545 # Handle an [almost] arbitary change
546 def HandleChange(Reply,DnRecord,Key):
547    global PlainText;
548    Lines = re.split("\n *\r?",PlainText);
549
550    Result = "";
551    Attrs = [];
552    Show = 0;
553    CommitChanges = 1
554    for Line in Lines: 
555       Line = Line.strip()
556       if Line == "":
557          continue;
558
559       # Try to process a command line
560       Result = Result + "> "+Line+"\n";
561       try:
562          if Line == "show":
563             Show = 1;
564             Res = "OK";
565          else:
566             badkeys = LoadBadSSH()
567             Res = DoPosition(Line,Attrs) or DoDNS(Line,Attrs,DnRecord) or \
568                   DoArbChange(Line,Attrs) or DoSSH(Line,Attrs,badkeys,GetAttr(DnRecord,"uid")) or \
569                   DoDel(Line,Attrs) or DoRBL(Line,Attrs) or DoConfirmSudopassword(Line)
570       except:
571          Res = None;
572          Result = Result + "==> %s: %s\n" %(sys.exc_type,sys.exc_value);
573
574       # Fail, if someone tries to send someone elses signed email to the
575       # daemon then we want to abort ASAP.
576       if Res == None:
577          CommitChanges = 0
578          Result = Result + "Command is not understood. Halted - no changes committed\n";
579          break;
580       Result = Result + Res + "\n";
581
582    # Connect to the ldap server
583    l = connectLDAP()
584    F = open(PassDir+"/pass-"+pwd.getpwuid(os.getuid())[0],"r");
585    AccessPass = F.readline().strip().split(" ")
586    F.close();
587
588    l.simple_bind_s("uid="+AccessPass[0]+","+BaseDn,AccessPass[1]);
589    oldAttrs = l.search_s(BaseDn,ldap.SCOPE_ONELEVEL,"uid="+GetAttr(DnRecord,"uid"));
590    if ((GetAttr(oldAttrs[0],"userPassword").find("*LK*") != -1) 
591        or GetAttr(oldAttrs[0],"userPassword").startswith("!")):
592       raise UDNotAllowedError, "This account is locked";
593
594    if CommitChanges == 1: # only if we are still good to go
595       try:
596          Res = FinishConfirmSudopassword(l, GetAttr(DnRecord,"uid"), Attrs)
597          Result = Result + Res + "\n";
598       except Error, e:
599          CommitChanges = 0
600          Result = Result + "FinishConfirmSudopassword raised an error (%s) - no changes committed\n"%(e);
601
602    # Modify the record
603    if CommitChanges == 1:
604       Dn = "uid=" + GetAttr(DnRecord,"uid") + "," + BaseDn;
605       l.modify_s(Dn,Attrs);
606
607    Attribs = "";
608    if Show == 1:
609       Attrs = l.search_s(BaseDn,ldap.SCOPE_ONELEVEL,"uid="+GetAttr(DnRecord,"uid"));
610       if len(Attrs) == 0:
611          raise UDNotAllowedError, "User not found"
612       Attribs = GPGEncrypt(PrettyShow(Attrs[0])+"\n","0x"+Key[1],Key[4]);
613
614    Subst = {};
615    Subst["__FROM__"] = ChangeFrom;
616    Subst["__EMAIL__"] = EmailAddress(DnRecord);
617    Subst["__ADMIN__"] = ReplyTo;
618    Subst["__RESULT__"] = Result;
619    Subst["__ATTR__"] = Attribs;
620
621    return Reply + TemplateSubst(Subst,open(TemplatesDir+"change-reply","r").read());
622    
623 # Handle ping handles an email sent to the 'ping' address (ie this program
624 # called with a ping argument) It replies with a dump of the public records.
625 def HandlePing(Reply,DnRecord,Key):
626    Subst = {};
627    Subst["__FROM__"] = PingFrom;
628    Subst["__EMAIL__"] = EmailAddress(DnRecord);
629    Subst["__LDAPFIELDS__"] = PrettyShow(DnRecord);
630    Subst["__ADMIN__"] = ReplyTo;
631
632    return Reply + TemplateSubst(Subst,open(TemplatesDir+"ping-reply","r").read());
633
634 # Handle a change password email sent to the change password address
635 # (this program called with the chpass argument)
636 def HandleChPass(Reply,DnRecord,Key):
637    # Generate a random password
638    Password = GenPass();
639    Pass = HashPass(Password);
640       
641    # Use GPG to encrypt it      
642    Message = GPGEncrypt("Your new password is '" + Password + "'\n",\
643                         "0x"+Key[1],Key[4]);
644    Password = None;
645
646    if Message == None:
647       raise UDFormatError, "Unable to generate the encrypted reply, gpg failed.";
648
649    if (Key[4] == 1):
650       Type = "Your message was encrypted using PGP 2.x\ncompatibility mode.";
651    else:
652       Type = "Your message was encrypted using GPG (OpenPGP)\ncompatibility "\
653              "mode, without IDEA. This message cannot be decoded using PGP 2.x";
654    
655    Subst = {};
656    Subst["__FROM__"] = ChPassFrom;
657    Subst["__EMAIL__"] = EmailAddress(DnRecord);
658    Subst["__CRYPTTYPE__"] = Type;
659    Subst["__PASSWORD__"] = Message;
660    Subst["__ADMIN__"] = ReplyTo;
661    Reply = Reply + TemplateSubst(Subst,open(TemplatesDir+"passwd-changed","r").read());
662    
663    # Connect to the ldap server
664    l = connectLDAP()
665    F = open(PassDir+"/pass-"+pwd.getpwuid(os.getuid())[0],"r");
666    AccessPass = F.readline().strip().split(" ")
667    F.close();
668    l.simple_bind_s("uid="+AccessPass[0]+","+BaseDn,AccessPass[1]);
669
670    # Check for a locked account
671    Attrs = l.search_s(BaseDn,ldap.SCOPE_ONELEVEL,"uid="+GetAttr(DnRecord,"uid"));
672    if (GetAttr(Attrs[0],"userPassword").find("*LK*") != -1) \
673              or GetAttr(Attrs[0],"userPassword").startswith("!"):
674       raise UDNotAllowedError, "This account is locked";
675
676    # Modify the password
677    Rec = [(ldap.MOD_REPLACE,"userPassword","{crypt}"+Pass),
678           (ldap.MOD_REPLACE,"shadowLastChange",str(int(time.time()/24/60/60)))];
679    Dn = "uid=" + GetAttr(DnRecord,"uid") + "," + BaseDn;
680    l.modify_s(Dn,Rec);
681
682    return Reply;
683       
684 # Start of main program
685
686 # Drop messages from a mailer daemon.
687 if os.environ.has_key('SENDER') == 0 or len(os.environ['SENDER']) == 0:
688    sys.exit(0);
689
690 ErrMsg = "Indeterminate Error";
691 ErrType = EX_TEMPFAIL;
692 try:
693    # Startup the replay cache
694    ErrType = EX_TEMPFAIL;
695    ErrMsg = "Failed to initialize the replay cache:";
696
697    # Get the email 
698    ErrType = EX_PERMFAIL;
699    ErrMsg = "Failed to understand the email or find a signature:";
700    Email = mimetools.Message(sys.stdin,0);
701    Msg = GetClearSig(Email);
702
703    ErrMsg = "Message is not PGP signed:"
704    if Msg[0].find("-----BEGIN PGP SIGNED MESSAGE-----") == -1 and \
705       Msg[0].find("-----BEGIN PGP MESSAGE-----") == -1:
706       raise UDFormatError, "No PGP signature";
707    
708    # Check the signature
709    ErrMsg = "Unable to check the signature or the signature was invalid:";
710    Res = GPGCheckSig(Msg[0]);
711
712    if Res[0] != None:
713       raise UDFormatError, Res[0];
714       
715    if Res[3] == None:
716       raise UDFormatError, "Null signature text";
717
718    # Extract the plain message text in the event of mime encoding
719    global PlainText;
720    ErrMsg = "Problem stripping MIME headers from the decoded message"
721    if Msg[1] == 1:
722       try:
723          Index = Res[3].index("\n\n") + 2;
724       except ValueError:
725          Index = Res[3].index("\n\r\n") + 3;
726       PlainText = Res[3][Index:];
727    else:
728       PlainText = Res[3];   
729
730    # Connect to the ldap server
731    ErrType = EX_TEMPFAIL;
732    ErrMsg = "An error occured while performing the LDAP lookup";
733    global l;
734    l = connectLDAP()
735    l.simple_bind_s("","");
736
737    # Search for the matching key fingerprint
738    Attrs = l.search_s(BaseDn,ldap.SCOPE_ONELEVEL,"keyFingerPrint=" + Res[2][1]);
739
740    ErrType = EX_PERMFAIL;
741    if len(Attrs) == 0:
742       raise UDFormatError, "Key not found"
743    if len(Attrs) != 1:
744       raise UDFormatError, "Oddly your key fingerprint is assigned to more than one account.."
745
746
747    # Check the signature against the replay cache
748    RC = ReplayCache(ReplayCacheFile);
749    RC.Clean();
750    ErrMsg = "The replay cache rejected your message. Check your clock!";
751    Rply = RC.Check(Res[1]);
752    if Rply != None:
753       RC.close()
754       raise UDNotAllowedError, Rply;
755    RC.Add(Res[1]);
756    RC.close()
757
758    # Determine the sender address
759    ErrMsg = "A problem occured while trying to formulate the reply";
760    Sender = Email.getheader("Reply-To");
761    if Sender == None:
762       Sender = Email.getheader("From");
763    if Sender == None:
764       raise UDFormatError, "Unable to determine the sender's address";
765
766    # Formulate a reply
767    Date = time.strftime("%a, %d %b %Y %H:%M:%S +0000",time.gmtime(time.time()));
768    Reply = "To: %s\nReply-To: %s\nDate: %s\n" % (Sender,ReplyTo,Date);
769
770    # Dispatch
771    if sys.argv[1] == "ping":
772       Reply = HandlePing(Reply,Attrs[0],Res[2]);
773    elif sys.argv[1] == "chpass":
774       if PlainText.strip().find("Please change my Debian password") != 0:
775          raise UDFormatError,"Please send a signed message where the first line of text is the string 'Please change my Debian password'";
776       Reply = HandleChPass(Reply,Attrs[0],Res[2]);
777    elif sys.argv[1] == "change":
778       Reply = HandleChange(Reply,Attrs[0],Res[2]);
779    else:
780       print sys.argv;
781       raise UDFormatError, "Incorrect Invokation";
782
783    # Send the message through sendmail      
784    ErrMsg = "A problem occured while trying to send the reply";
785    Child = os.popen("/usr/sbin/sendmail -t","w");
786 #   Child = os.popen("cat","w");
787    Child.write(Reply);
788    if Child.close() != None:
789       raise UDExecuteError, "Sendmail gave a non-zero return code";
790
791 except:
792    # Error Reply Header
793    Date = time.strftime("%a, %d %b %Y %H:%M:%S +0000",time.gmtime(time.time()));
794    ErrReplyHead = "To: %s\nReply-To: %s\nDate: %s\n" % (os.environ['SENDER'],ReplyTo,Date);
795
796    # Error Body
797    Subst = {};
798    Subst["__ERROR__"] = ErrMsg;
799    Subst["__ADMIN__"] = ReplyTo;
800
801    Trace = "==> %s: %s\n" %(sys.exc_type,sys.exc_value);
802    List = traceback.extract_tb(sys.exc_traceback);
803    if len(List) > 1:
804       Trace = Trace + "Python Stack Trace:\n";
805       for x in List:
806          Trace = Trace +  "   %s %s:%u: %s\n" %(x[2],x[0],x[1],x[3]);
807
808    Subst["__TRACE__"] = Trace;
809
810    # Try to send the bounce
811    try:
812       ErrReply = TemplateSubst(Subst,open(TemplatesDir+"error-reply","r").read());
813
814       Child = os.popen("/usr/sbin/sendmail -t -oi -f ''","w");
815       Child.write(ErrReplyHead);
816       Child.write(ErrReply);
817       if Child.close() != None:
818          raise UDExecuteError, "Sendmail gave a non-zero return code";
819    except:
820       sys.exit(EX_TEMPFAIL);
821       
822    if ErrType != EX_PERMFAIL:
823       sys.exit(ErrType);
824    sys.exit(0);
825
826 # vim:set et:
827 # vim:set ts=3:
828 # vim:set shiftwidth=3: