ud-echelon

   1 #!/usr/bin/env python
   2 # -*- mode: python -*-
   3 import userdir_gpg, userdir_ldap, sys, traceback, time, ldap, posix;
   4 import string, pwd
   5 from userdir_gpg import *;
   6 from userdir_ldap import *;
   7
   8 EX_TEMPFAIL = 75;
   9 EX_PERMFAIL = 65;      # EX_DATAERR
  10 Debug = "";
  11
  12 # Try to extract a key fingerprint from a PGP siged message
  13 def TryGPG(Email):
  14    # Try to get a pgp text
  15    Msg = GetClearSig(Email);
  16    if string.find(Msg[0],"-----BEGIN PGP SIGNED MESSAGE-----") == -1:
  17       return None;
  18
  19    print Msg[0];
  20    Res = GPGCheckSig(Msg[0]);
  21
  22    # Failed to find a matching sig
  23    if Res[0] != None:
  24       return None;
  25
  26    # Search for the matching key fingerprint
  27    Attrs = l.search_s(BaseDn,ldap.SCOPE_ONELEVEL,"keyfingerprint=" + Res[2][1]);
  28    if len(Attrs) == 0:
  29       return None;
  30    if len(Attrs) != 1:
  31       raise Error, "Oddly your key fingerprint is assigned to more than one account.."
  32
  33    return (Attrs[0][1]["uid"][0],"PGP",FormatPGPKey(Res[2][1]));
  34
  35 # Try to guess the name from the email address
  36 def TryMatcher(Email):
  37    Sender = Email.getheader("From");
  38    if Sender == None:
  39       return None;
  40
  41    # Split up the address and invoke the matcher routine
  42    UID = GetUID(l,SplitEmail(Sender));
  43
  44    if UID[0] == None:
  45       if UID[1] == None or len(UID[1]) == 0:
  46          return None;
  47
  48       # Print out an error message
  49       S = "%s: %s -> Address matching failed '%s'\n" %(Now,MsgID,Sender);
  50       for x in UID[1]:
  51          S = S + " " + x + "\n";
  52       ErrLog.write(S);
  53       return None;
  54
  55    return (UID[0],"FROM",Sender);
  56
  57 # Open the log files
  58 if Debug == None:
  59    MainLog = open(Ech_MainLog,"a+",0);
  60    ErrLog = open(Ech_ErrorLog,"a+",0);
  61 else:
  62    MainLog = open("/dev/stdout","a+",0);
  63    ErrLog = open("/dev/stdout","a+",0);
  64
  65 # Start of main program
  66 ErrMsg = "Indeterminate Error";
  67 ErrType = EX_TEMPFAIL;
  68 Now = time.strftime("%a, %d %b %Y %H:%M:%S",time.gmtime(time.time()));
  69 MsgID = None;
  70 try:
  71    # Get the email
  72    ErrType = EX_PERMFAIL;
  73    ErrMsg = "Failed to understand the email or find a signature:";
  74    Email = mimetools.Message(sys.stdin,0);
  75    MsgID = Email.getheader("Message-ID");
  76
  77    # Connect to the ldap server
  78    ErrType = EX_TEMPFAIL;
  79    ErrMsg = "An error occured while performing the LDAP lookup";
  80    global l;
  81    l = ldap.open(LDAPServer);
  82    if Debug == None:
  83       F = open(PassDir+"/pass-"+pwd.getpwuid(posix.getuid())[0],"r");
  84       AccessPass = string.split(string.strip(F.readline())," ");
  85       l.simple_bind_s("uid="+AccessPass[0]+","+BaseDn,AccessPass[1]);
  86       F.close();
  87    else:
  88       l.simple_bind_s("","");
  89
  90    # Try to decode
  91    ErrType = EX_TEMPFAIL;
  92    ErrMsg = "An error occured while trying GPG decoding";
  93    User = TryGPG(Email);
  94    if User == None:
  95       ErrMsg = "An error occured while trying Matcher decoding";
  96       User = TryMatcher(Email);
  97
  98    # Get any mailing list information
  99    List = Email.getheader("X-Mailing-List");
 100    if List == None:
 101       List = "-";
 102
 103    # Tada, write a log message
 104    if User != None:
 105       Msg = "[%s] \"%s\" \"%s\" \"%s\""%(Now,User[2],List,MsgID);
 106       MainLog.write("%s %s %s\n"%(User[0],User[1],Msg));
 107       Dn = "uid=" + User[0] + "," + BaseDn;
 108       Rec = [(ldap.MOD_REPLACE,"activity-%s"%(User[1]),Msg)];
 109       if Debug == None:
 110          l.modify_s(Dn,Rec);
 111       else:
 112          print Rec;
 113    else:
 114       User = ("-","UKN",Email.getheader("From"));
 115       Msg = "[%s] \"%s\" \"%s\" \"%s\""%(Now,User[2],List,MsgID);
 116       MainLog.write("%s %s %s\n"%(User[0],User[1],Msg));
 117
 118 except:
 119    # Log an exception..
 120    S = "%s: %s -> %s\n" %(Now,MsgID,ErrMsg);
 121    S = S + "==> %s: %s\n" %(sys.exc_type,sys.exc_value);
 122    List = traceback.extract_tb(sys.exc_traceback);
 123    if len(List) > 1:
 124       for x in List:
 125          S = S + "   %s %s:%u: %s\n" %(x[2],x[0],x[1],x[3]);
 126    ErrLog.write(S);
 127    sys.exit(ErrType);
 128
 129 sys.exit(0);