ud-mailgate

   1 #!/usr/bin/env python
   2 # -*- mode: python -*-
   3 import userdir_gpg, userdir_ldap, sys, traceback, time, ldap, os;
   4 import string, pwd
   5 from userdir_gpg import *;
   6 from userdir_ldap import *;
   7
   8 # Error codes from /usr/include/sysexits.h
   9 ReplyTo = ConfModule.replyto;
  10 PingFrom = ConfModule.pingfrom;
  11 ChPassFrom = ConfModule.chpassfrom;
  12 ChangeFrom = ConfModule.changefrom;
  13 ReplayCacheFile = ConfModule.replaycachefile;
  14
  15 EX_TEMPFAIL = 75;
  16 EX_PERMFAIL = 65;      # EX_DATAERR
  17 Error = 'Message Error';
  18 SeenRSA = 0;
  19 SeenDSA = 0;
  20 SeenDNS = 0;
  21
  22 ArbChanges = {"c": "..",
  23               "l": ".*",
  24               "facsimiletelephonenumber": ".*",
  25               "telephonenumber": ".*",
  26               "postaladdress": ".*",
  27               "postalcode": ".*",
  28               "loginshell": ".*",
  29               "emailforward": "^([^<>@]+@.+)?$",
  30               "ircnick": ".*",
  31               "icquin": "^[0-9]*$",
  32               "onvacation": ".*",
  33               "labeledurl": ".*"};
  34
  35 DelItems = {"c": None,
  36             "l": None,
  37             "facsimiletelephonenumber": None,
  38             "telephonenumber": None,
  39             "postaladdress": None,
  40             "postalcode": None,
  41             "emailforward": None,
  42             "ircnick": None,
  43             "onvacation": None,
  44             "labeledurl": None,
  45             "latitude": None,
  46             "longitude": None,
  47             "icquin": None,
  48             "sshrsaauthkey": None,
  49             "sshdsaauthkey": None};
  50
  51 # Decode a GPS location from some common forms
  52 def LocDecode(Str,Dir):
  53    # Check for Decimal degrees, DGM, or DGMS
  54    if re.match("^[+-]?[\d.]+$",Str) != None:
  55       return Str;
  56
  57    Deg = '0'; Min = None; Sec = None; Dr = Dir[0];
  58
  59    # Check for DDDxMM.MMMM where x = [nsew]
  60    Match = re.match("^(\d+)(["+Dir+"])([\d.]+)$",Str);
  61    if Match != None:
  62       G = Match.groups();
  63       Deg = G[0]; Min = G[2]; Dr = G[1];
  64
  65    # Check for DD.DD x
  66    Match = re.match("^([\d.]+) ?(["+Dir+"])$",Str);
  67    if Match != None:
  68       G = Match.groups();
  69       Deg = G[0]; Dr = G[1];
  70
  71    # Check for DD:MM.MM x
  72    Match = re.match("^(\d+):([\d.]+) ?(["+Dir+"])$",Str);
  73    if Match != None:
  74       G = Match.groups();
  75       Deg = G[0]; Min = G[1]; Dr = G[2];
  76
  77    # Check for DD:MM:SS.SS x
  78    Match = re.match("^(\d+):(\d+):([\d.]+) ?(["+Dir+"])$",Str);
  79    if Match != None:
  80       G = Match.groups();
  81       Deg = G[0]; Min = G[1]; Sec = G[2]; Dr = G[3];
  82
  83    # Some simple checks
  84    if float(Deg) > 180:
  85       raise "Failed","Bad degrees";
  86    if Min != None and float(Min) > 60:
  87       raise "Failed","Bad minutes";
  88    if Sec != None and float(Sec) > 60:
  89       raise "Failed","Bad seconds";
  90
  91    # Pad on an extra leading 0 to disambiguate small numbers
  92    if len(Deg) <= 1 or Deg[1] == '.':
  93       Deg = '0' + Deg;
  94    if Min != None and (len(Min) <= 1 or Min[1] == '.'):
  95       Min = '0' + Min;
  96    if Sec != None and (len(Sec) <= 1 or Sec[1] == '.'):
  97       Sec = '0' + Sec;
  98
  99    # Construct a DGM/DGMS type value from the components.
 100    Res = "+"
 101    if Dr == Dir[1]:
 102       Res = "-";
 103    Res = Res + Deg;
 104    if Min != None:
 105       Res = Res + Min;
 106    if Sec != None:
 107       Res = Res + Sec;
 108    return Res;
 109
 110 # Handle changing a set of arbitary fields
 111 #  <field>: value
 112 def DoArbChange(Str,Attrs):
 113    Match = re.match("^([^ :]+): (.*)$",Str);
 114    if Match == None:
 115       return None;
 116    G = Match.groups();
 117
 118    if ArbChanges.has_key(G[0]) == 0:
 119       return None;
 120
 121    if re.match(ArbChanges[G[0]],G[1]) == None:
 122       raise Error, "Item does not match the required format"+ArbChanges[G[0]];
 123
 124    Attrs.append((ldap.MOD_REPLACE,G[0],G[1]));
 125    return "Changed entry %s to %s"%(G[0],G[1]);
 126
 127 # Handle changing a set of arbitary fields
 128 #  <field>: value
 129 def DoDel(Str,Attrs):
 130    Match = re.match("^del (.*)$",Str);
 131    if Match == None:
 132       return None;
 133    G = Match.groups();
 134
 135    if DelItems.has_key(G[0]) == 0:
 136       return "Cannot erase entry %s"%(G[0]);
 137
 138    Attrs.append((ldap.MOD_DELETE,G[0],None));
 139    return "Removed entry %s"%(G[0]);
 140
 141 # Handle a position change message, the line format is:
 142 #  Lat: -12412.23 Long: +12341.2342
 143 def DoPosition(Str,Attrs):
 144    Match = re.match("^lat: ([+\-]?[\d:.ns]+(?: ?[ns])?) long: ([+\-]?[\d:.ew]+(?: ?[ew])?)$",string.lower(Str));
 145    if Match == None:
 146       return None;
 147
 148    G = Match.groups();
 149    try:
 150       sLat = LocDecode(G[0],"ns");
 151       sLong = LocDecode(G[1],"ew");
 152       Lat = DecDegree(sLat,1);
 153       Long = DecDegree(sLong,1);
 154    except:
 155       raise Error, "Positions were found, but they are not correctly formed";
 156
 157    Attrs.append((ldap.MOD_REPLACE,"latitude",sLat));
 158    Attrs.append((ldap.MOD_REPLACE,"longitude",sLong));
 159    return "Position set to %s/%s (%s/%s decimal degrees)"%(sLat,sLong,Lat,Long);
 160
 161 # Handle a SSH RSA authentication key, the line format is:
 162 #  [options] 1024 35 13188913666680[..] [comment]
 163 def DoSSH(Str,Attrs):
 164    Match = SSHAuthSplit.match(Str);
 165    if Match == None:
 166       return None;
 167
 168    global SeenRSA;
 169    if SeenRSA:
 170      Attrs.append((ldap.MOD_ADD,"sshrsaauthkey",Str));
 171      return "SSH Key added "+FormatSSHAuth(Str);
 172
 173    Attrs.append((ldap.MOD_REPLACE,"sshrsaauthkey",Str));
 174    SeenRSA = 1;
 175    return "SSH Keys replaced with "+FormatSSHAuth(Str);
 176
 177 # Handle a SSH DSA authentication key, the line format is:
 178 #  ssh-dss [key] [comment]
 179 def DoSSH2(Str,Attrs):
 180    Match = SSHDSAAuthSplit.match(Str);
 181    if Match == None:
 182       return None;
 183
 184    global SeenDSA;
 185    if SeenDSA:
 186      Attrs.append((ldap.MOD_ADD,"sshdsaauthkey",Str));
 187      return "SSH2 Key added "+FormatSSH2Auth(Str);
 188
 189    Attrs.append((ldap.MOD_REPLACE,"sshdsaauthkey",Str));
 190    SeenDSA = 1;
 191    return "SSH2 Keys replaced with "+FormatSSH2Auth(Str);
 192
 193 # Handle changing a dns entry
 194 #  host in a 12.12.12.12
 195 #  host in cname foo.bar.    <- Trailing dot is required
 196 def DoDNS(Str,Attrs,DnRecord):
 197    if re.match('^[\w-]+\s+in\s+a\s+\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}$',\
 198         Str,re.IGNORECASE) == None and \
 199       re.match("^[\w-]+\s+in\s+cname\s+[\w.\-]+\.$",Str,re.IGNORECASE) == None and \
 200       re.match("^[\w-]+\s+in\s+mx\s+\d{1,3}\s+[\w.\-]+\.$",Str,re.IGNORECASE) == None:
 201      return None;
 202
 203    # Check if the name is already taken
 204    G = re.match('^([\w-+]+)\s',Str).groups();
 205
 206    # Check for collisions
 207    global l;
 208    Rec = l.search_s(BaseDn,ldap.SCOPE_ONELEVEL,"dnszoneentry="+G[0]+" *",["uid"]);
 209    for x in Rec:
 210       if GetAttr(x,"uid") != GetAttr(DnRecord,"uid"):
 211          return "DNS entry is already owned by " + GetAttr(x,"uid")
 212
 213    global SeenDNS;
 214    if SeenDNS:
 215      Attrs.append((ldap.MOD_ADD,"dnszoneentry",Str));
 216      return "DNS Entry added "+Str;
 217
 218    Attrs.append((ldap.MOD_REPLACE,"dnszoneentry",Str));
 219    SeenDNS = 1;
 220    return "DNS Entry replaced with "+Str;
 221
 222 # Handle an [almost] arbitary change
 223 def HandleChange(Reply,DnRecord,Key):
 224    global PlainText;
 225    Lines = re.split("\n *\r?",PlainText);
 226
 227    Result = "";
 228    Attrs = [];
 229    Show = 0;
 230    for Line in Lines:
 231       Line = string.strip(Line);
 232       if Line == "":
 233          continue;
 234
 235       # Try to process a command line
 236       Result = Result + "> "+Line+"\n";
 237       try:
 238          if Line == "show":
 239            Show = 1;
 240            Res = "OK";
 241          else:
 242            Res = DoPosition(Line,Attrs) or DoDNS(Line,Attrs,DnRecord) or \
 243                  DoArbChange(Line,Attrs) or DoSSH(Line,Attrs) or \
 244                  DoSSH2(Line,Attrs) or DoDel(Line,Attrs);
 245       except:
 246          Res = None;
 247          Result = Result + "==> %s: %s\n" %(sys.exc_type,sys.exc_value);
 248
 249       # Fail, if someone tries to send someone elses signed email to the
 250       # daemon then we want to abort ASAP.
 251       if Res == None:
 252          Result = Result + "Command is not understood. Halted\n";
 253          break;
 254       Result = Result + Res + "\n";
 255
 256    # Connect to the ldap server
 257    l = ldap.open(LDAPServer);
 258    F = open(PassDir+"/pass-"+pwd.getpwuid(os.getuid())[0],"r");
 259    AccessPass = string.split(string.strip(F.readline())," ");
 260    F.close();
 261
 262    # Modify the record
 263    l.simple_bind_s("uid="+AccessPass[0]+","+BaseDn,AccessPass[1]);
 264    Dn = "uid=" + GetAttr(DnRecord,"uid") + "," + BaseDn;
 265    l.modify_s(Dn,Attrs);
 266
 267    Attribs = "";
 268    if Show == 1:
 269       Attrs = l.search_s(BaseDn,ldap.SCOPE_ONELEVEL,"uid="+GetAttr(DnRecord,"uid"));
 270       if len(Attrs) == 0:
 271          raise Error, "User not found"
 272       Attribs = GPGEncrypt(PrettyShow(Attrs[0])+"\n","0x"+Key[1],Key[4]);
 273
 274    Subst = {};
 275    Subst["__FROM__"] = ChangeFrom;
 276    Subst["__EMAIL__"] = EmailAddress(DnRecord);
 277    Subst["__ADMIN__"] = ReplyTo;
 278    Subst["__RESULT__"] = Result;
 279    Subst["__ATTR__"] = Attribs;
 280
 281    return Reply + TemplateSubst(Subst,open(TemplatesDir+"change-reply","r").read());
 282
 283 # Handle ping handles an email sent to the 'ping' address (ie this program
 284 # called with a ping argument) It replies with a dump of the public records.
 285 def HandlePing(Reply,DnRecord,Key):
 286    Subst = {};
 287    Subst["__FROM__"] = PingFrom;
 288    Subst["__EMAIL__"] = EmailAddress(DnRecord);
 289    Subst["__LDAPFIELDS__"] = PrettyShow(DnRecord);
 290    Subst["__ADMIN__"] = ReplyTo;
 291
 292    return Reply + TemplateSubst(Subst,open(TemplatesDir+"ping-reply","r").read());
 293
 294 # Handle a change password email sent to the change password address
 295 # (this program called with the chpass argument)
 296 def HandleChPass(Reply,DnRecord,Key):
 297    # Generate a random password
 298    Password = GenPass();
 299    Pass = HashPass(Password);
 300
 301    # Use GPG to encrypt it
 302    Message = GPGEncrypt("Your new password is '" + Password + "'\n",\
 303                         "0x"+Key[1],Key[4]);
 304    Password = None;
 305
 306    if Message == None:
 307       raise Error, "Unable to generate the encrypted reply, gpg failed.";
 308
 309    if (Key[4] == 1):
 310       Type = "Your message was encrypted using PGP 2.x\ncompatibility mode.";
 311    else:
 312       Type = "Your message was encrypted using GPG (OpenPGP)\ncompatibility "\
 313              "mode, without IDEA. This message cannot be decoded using PGP 2.x";
 314
 315    Subst = {};
 316    Subst["__FROM__"] = ChPassFrom;
 317    Subst["__EMAIL__"] = EmailAddress(DnRecord);
 318    Subst["__CRYPTTYPE__"] = Type;
 319    Subst["__PASSWORD__"] = Message;
 320    Subst["__ADMIN__"] = ReplyTo;
 321    Reply = Reply + TemplateSubst(Subst,open(TemplatesDir+"passwd-changed","r").read());
 322
 323    # Connect to the ldap server
 324    l = ldap.open(LDAPServer);
 325    F = open(PassDir+"/pass-"+pwd.getpwuid(os.getuid())[0],"r");
 326    AccessPass = string.split(string.strip(F.readline())," ");
 327    F.close();
 328    l.simple_bind_s("uid="+AccessPass[0]+","+BaseDn,AccessPass[1]);
 329
 330    # Check for a locked account
 331    Attrs = l.search_s(BaseDn,ldap.SCOPE_ONELEVEL,"uid="+GetAttr(DnRecord,"uid"));
 332    if (string.find(GetAttr(Attrs[0],"userpassword"),"*LK*")  != -1):
 333       raise Error, "This account is locked";
 334
 335    # Modify the password
 336    Rec = [(ldap.MOD_REPLACE,"userPassword","{crypt}"+Pass)];
 337    Dn = "uid=" + GetAttr(DnRecord,"uid") + "," + BaseDn;
 338    l.modify_s(Dn,Rec);
 339
 340    return Reply;
 341
 342 # Start of main program
 343
 344 # Drop messages from a mailer daemon.
 345 if os.environ.has_key('SENDER') == 0 or len(os.environ['SENDER']) == 0:
 346    sys.exit(0);
 347
 348 ErrMsg = "Indeterminate Error";
 349 ErrType = EX_TEMPFAIL;
 350 try:
 351    # Startup the replay cache
 352    ErrType = EX_TEMPFAIL;
 353    ErrMsg = "Failed to initialize the replay cache:";
 354    RC = ReplayCache(ReplayCacheFile);
 355    RC.Clean();
 356
 357    # Get the email
 358    ErrType = EX_PERMFAIL;
 359    ErrMsg = "Failed to understand the email or find a signature:";
 360    Email = mimetools.Message(sys.stdin,0);
 361    Msg = GetClearSig(Email);
 362
 363    ErrMsg = "Message is not PGP signed:"
 364    if string.find(Msg[0],"-----BEGIN PGP SIGNED MESSAGE-----") == -1:
 365       raise Error, "No PGP signature";
 366
 367    # Check the signature
 368    ErrMsg = "Unable to check the signature or the signature was invalid:";
 369    Res = GPGCheckSig(Msg[0]);
 370
 371    if Res[0] != None:
 372       raise Error, Res[0];
 373
 374    if Res[3] == None:
 375       raise Error, "Null signature text";
 376
 377    # Extract the plain message text in the event of mime encoding
 378    global PlainText;
 379    ErrMsg = "Problem stripping MIME headers from the decoded message"
 380    if Msg[1] == 1:
 381       try:
 382          Index = string.index(Res[3],"\n\n") + 2;
 383       except ValueError:
 384          Index = string.index(Res[3],"\n\r\n") + 3;
 385       PlainText = Res[3][Index:];
 386    else:
 387       PlainText = Res[3];
 388
 389    # Check the signature against the replay cache
 390    ErrMsg = "The replay cache rejected your message. Check your clock!";
 391    Rply = RC.Check(Res[1]);
 392    if Rply != None:
 393       raise Error, Rply;
 394    RC.Add(Res[1]);
 395
 396    # Connect to the ldap server
 397    ErrType = EX_TEMPFAIL;
 398    ErrMsg = "An error occured while performing the LDAP lookup";
 399    global l;
 400    l = ldap.open(LDAPServer);
 401    l.simple_bind_s("","");
 402
 403    # Search for the matching key fingerprint
 404    Attrs = l.search_s(BaseDn,ldap.SCOPE_ONELEVEL,"keyfingerprint=" + Res[2][1]);
 405    if len(Attrs) == 0:
 406       raise Error, "Key not found"
 407    if len(Attrs) != 1:
 408       raise Error, "Oddly your key fingerprint is assigned to more than one account.."
 409
 410    # Determine the sender address
 411    ErrType = EX_PERMFAIL;
 412    ErrMsg = "A problem occured while trying to formulate the reply";
 413    Sender = Email.getheader("Reply-To");
 414    if Sender == None:
 415       Sender = Email.getheader("From");
 416    if Sender == None:
 417       raise Error, "Unable to determine the sender's address";
 418
 419    if (string.find(GetAttr(Attrs[0],"userpassword"),"*LK*")  != -1):
 420       raise Error, "This account is locked";
 421
 422    # Formulate a reply
 423    Date = time.strftime("%a, %d %b %Y %H:%M:%S +0000",time.gmtime(time.time()));
 424    Reply = "To: %s\nReply-To: %s\nDate: %s\n" % (Sender,ReplyTo,Date);
 425
 426    # Dispatch
 427    if sys.argv[1] == "ping":
 428       Reply = HandlePing(Reply,Attrs[0],Res[2]);
 429    elif sys.argv[1] == "chpass":
 430       if string.find(string.strip(PlainText),"Please change my Debian password") != 0:
 431          raise Error,"Please send a signed message where the first line of text is the string 'Please change my Debian password'";
 432       Reply = HandleChPass(Reply,Attrs[0],Res[2]);
 433    elif sys.argv[1] == "change":
 434       Reply = HandleChange(Reply,Attrs[0],Res[2]);
 435    else:
 436       print sys.argv;
 437       raise Error, "Incorrect Invokation";
 438
 439    # Send the message through sendmail
 440    ErrMsg = "A problem occured while trying to send the reply";
 441    Child = os.popen("/usr/sbin/sendmail -t","w");
 442 #   Child = os.popen("cat","w");
 443    Child.write(Reply);
 444    if Child.close() != None:
 445       raise Error, "Sendmail gave a non-zero return code";
 446
 447 except:
 448    # Error Reply Header
 449    Date = time.strftime("%a, %d %b %Y %H:%M:%S +0000",time.gmtime(time.time()));
 450    ErrReplyHead = "To: %s\nReply-To: %s\nDate: %s\n" % (os.environ['SENDER'],ReplyTo,Date);
 451
 452    # Error Body
 453    Subst = {};
 454    Subst["__ERROR__"] = ErrMsg;
 455    Subst["__ADMIN__"] = ReplyTo;
 456
 457    Trace = "==> %s: %s\n" %(sys.exc_type,sys.exc_value);
 458    List = traceback.extract_tb(sys.exc_traceback);
 459    if len(List) > 1:
 460       Trace = Trace + "Python Stack Trace:\n";
 461       for x in List:
 462          Trace = Trace +  "   %s %s:%u: %s\n" %(x[2],x[0],x[1],x[3]);
 463
 464    Subst["__TRACE__"] = Trace;
 465
 466    # Try to send the bounce
 467    try:
 468       ErrReply = TemplateSubst(Subst,open(TemplatesDir+"error-reply","r").read());
 469
 470       Child = os.popen("/usr/sbin/sendmail -t","w");
 471       Child.write(ErrReplyHead);
 472       Child.write(ErrReply);
 473       if Child.close() != None:
 474          raise Error, "Sendmail gave a non-zero return code";
 475    except:
 476       sys.exit(EX_TEMPFAIL);
 477
 478    if ErrType != EX_PERMFAIL:
 479       sys.exit(ErrType);
 480    sys.exit(0);
 481