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