ud-emailmatcher

   1 #!/usr/bin/env python
   2 # -*- mode: python -*-
   3 # This script tries to match a list of email addresses to the ldap database
   4 # uids. It makes use of the PGP key ring to determine matches
   5
   6 import string, re, time, ldap, getopt, sys;
   7 from userdir_ldap import *;
   8 from userdir_gpg import *;
   9
  10 AddressSplit = re.compile("(.*).*<([^@]*)@([^>]*)>");
  11
  12 # Import an an forward file
  13 def ImportForward(File,EmailMap):
  14    F = open(File,"r");
  15    while(1):
  16       Line = string.strip(F.readline());
  17       if Line == "":
  18          break;
  19       Split = string.split(Line,":");
  20       if len(Split) != 2:
  21          continue;
  22
  23       Addr = string.strip(Split[1]);
  24       if EmailMap.has_key(Addr) and  EmailMap[Addr] != Split[0]:
  25          print "Dup Over Emap",Line,Split
  26       else:
  27          EmailMap[Addr] = Split[0];
  28    F.close();
  29
  30 # Import an override file
  31 def ImportOverride(File,OverMap):
  32    F = open(File,"r");
  33    while(1):
  34       Line = F.readline();
  35       if Line == "":
  36          break;
  37       Line = string.strip(Line);
  38
  39       Split = string.split(Line,":");
  40       if len(Split) != 2:
  41          continue;
  42       OverMap[Split[0]] = string.strip(Split[1]);
  43    F.close();
  44
  45 (options, arguments) = getopt.getopt(sys.argv[1:], "o:f:")
  46
  47 # Popen GPG with the correct magic special options
  48 Args = [GPGPath] + GPGBasicOptions + GPGKeyRings;
  49 for x in arguments:
  50    Args.append("--keyring");
  51    Args.append(x);
  52 Args = Args + GPGSearchOptions + [" 2> /dev/null"]
  53 Keys = os.popen(string.join(Args," "),"r");
  54
  55 l = ldap.open(LDAPServer);
  56 l.simple_bind_s("","");
  57
  58 # Fetch the key list and map to email address
  59 PasswdAttrs = l.search_s(BaseDn,ldap.SCOPE_ONELEVEL,"keyfingerprint=*",\
  60                 ["uid","keyfingerprint"]);
  61 KFMap = {}
  62 for x in PasswdAttrs:
  63    if x[1].has_key("keyfingerprint") == 0 or x[1].has_key("uid") == 0:
  64       continue;
  65    for I in x[1]["keyfingerprint"]:
  66       KFMap[I] = x[1]["uid"][0];
  67
  68 # Loop over the GPG key file mapping addresses to uids
  69 Outstanding = 0;
  70 Ignored = 0;
  71 Emails = [];
  72 EmailMap = {};
  73 UIDMap = {};
  74 UID = None;
  75 FingerPrint = None;
  76 print "Reading keyrings",
  77 sys.stdout.flush();
  78 while(1):
  79    Line = Keys.readline();
  80    if Line == "":
  81       break;
  82
  83    Split = string.split(Line,":");
  84    if len(Split) >= 8 and Split[0] == "pub":
  85       if FingerPrint != None and UID != None:
  86          for x in Emails:
  87             Match = AddressSplit.match(x);
  88             if Match == None:
  89               continue;
  90             Groups = Match.groups();
  91             Email = Groups[1]+'@'+Groups[2];
  92             if UIDMap.has_key(Groups[1]):
  93                UIDMap[Groups[1]].append(Email);
  94             else:
  95                UIDMap[Groups[1]] = [Email];
  96             if EmailMap.has_key(Email) and EmailMap[Email] != UID:
  97                print "Dup Emap",Email
  98             else:
  99                EmailMap[Email] = UID;
 100       Emails = [Split[9]];
 101       continue;
 102    if len(Split) >= 11 and Split[0] == "fpr":
 103       FingerPrint = Split[9];
 104       if KFMap.has_key(FingerPrint) == 0:
 105          print "Failed",FingerPrint;
 106          UID = None;
 107          continue;
 108       UID = KFMap[FingerPrint];
 109    if len(Split) >= 9 and Split[0] == "uid":
 110       Emails.append(Split[9]);
 111 print;
 112
 113 # Process the override files
 114 for (switch, val) in options:
 115    if (switch == '-f'):
 116       ImportForward(val,EmailMap);
 117       BindUser = val;
 118    elif (switch == '-o'):
 119       ImportOverride(val,EmailMap);
 120
 121 # Map the input
 122 FinalMap = {};
 123 while(1):
 124    Line = sys.stdin.readline();
 125    if Line == "":
 126       break;
 127    Line = string.strip(Line);
 128
 129    Split = string.split(Line,"@");
 130    if len(Split) != 2:
 131       continue;
 132
 133    # The address is in our domain, go directly
 134    if Split[1] == EmailAppend:
 135       if FinalMap.has_key(Line):
 136         print "Dup",Line
 137       Split2 = string.split(Split[0],"-");
 138       FinalMap[Line] = Split2[0];
 139       continue;
 140
 141    # Exists in the email map..
 142    if EmailMap.has_key(Line):
 143       if FinalMap.has_key(Line):
 144         print "Dup",Line
 145       FinalMap[Line] = EmailMap[Line];
 146       continue;
 147
 148    # Try again splitting off common address appendage modes
 149    Split2 = string.split(Split[0],"-");
 150    Addr = Split2[0]+'@'+Split[1];
 151    if EmailMap.has_key(Addr):
 152       if FinalMap.has_key(Addr):
 153         print "Dup",Addr
 154       FinalMap[Line] = EmailMap[Addr];
 155       continue;
 156
 157     # Failed
 158    if UIDMap.has_key(Split[0]):
 159       print Line,UIDMap[Split[0]];
 160    print Line;
 161 print "-----";
 162
 163 # Generate a reverse map and check for duplicates
 164 Back = {};
 165 for x in FinalMap.keys():
 166    if Back.has_key(FinalMap[x]):
 167       print "Dup",x,FinalMap[x],Back[FinalMap[x]];
 168    Back[FinalMap[x]] = x;
 169
 170 # Print the forward map
 171 for x in Back.keys():
 172    print "%s: %s" % (x,Back[x]);