#!/usr/bin/env python
# -*- mode: python -*-
# Generates passwd, shadow and group files from the ldap directory.

import string, re, time, ldap, getopt, sys, os, pwd, posix;
from userdir_ldap import *;

PasswdAttrs = None;
GroupIDMap = {};
Allowed = None;
CurrentHost = "";

EmailCheck = re.compile("^([^ <>@]+@[^ ,<>@]+)?$");
BSMTPCheck = re.compile(".*mx 0 klecker\.debian\.org\..*",re.DOTALL);
DNSZone = ".debian.net"

def Sanitize(Str):
  return string.translate(Str,string.maketrans("\n\r\t","$$$"));

def DoLink(From,To,File):
   try: posix.remove(To+File);
   except: pass;
   posix.link(From+File,To+File);

# See if this user is in the group list
def IsInGroup(DnRecord):
  global Allowed,CurrentHost;
  if Allowed == None:
     return 1;

  # See if the primary group is in the list
  if Allowed.has_key(GetAttr(DnRecord,"gidnumber")) != 0:
     return 1;

  # Check the host based ACL
  if DnRecord[1].has_key("allowedhosts") != 0:
     for I in DnRecord[1]["allowedhosts"]:
        if CurrentHost == I:
           return 1;

  # See if there are supplementary groups
  if DnRecord[1].has_key("supplementarygid") == 0:
     return 0;

  # Check the supplementary groups
  for I in DnRecord[1]["supplementarygid"]:
     if Allowed.has_key(I):
        return 1;
  return 0;

def Die(File,F,Fdb):
   if F != None:
      F.close();
   if Fdb != None:
      Fdb.close();
   try: os.remove(File + ".tmp");
   except: pass;
   try: os.remove(File + ".tdb.tmp");
   except: pass;

def Done(File,F,Fdb):
  if F != None:
    F.close();
    os.rename(File + ".tmp",File);
  if Fdb != None:
    Fdb.close();
    os.rename(File + ".tdb.tmp",File+".tdb");
  
# Generate the password list
def GenPasswd(l,File,HomePrefix):
  F = None;
  Fdb = None;
  try:
   F = open(File + ".tmp","w");
   Fdb = open(File + ".tdb.tmp","w");

   # Fetch all the users
   global PasswdAttrs;
   if PasswdAttrs == None:
      raise "No Users";

   I = 0;
   for x in PasswdAttrs:
      if x[1].has_key("uidnumber") == 0 or IsInGroup(x) == 0:
         continue;

      # Do not let people try to buffer overflow some busted passwd parser.
      if len(GetAttr(x,"gecos")) > 100 or len(GetAttr(x,"loginshell")) > 50:
         continue;

      Line = "%s:x:%s:%s:%s:%s%s:%s\n" % (GetAttr(x,"uid"),\
              GetAttr(x,"uidnumber"),GetAttr(x,"gidnumber"),\
              GetAttr(x,"gecos"),HomePrefix,GetAttr(x,"uid"),\
              GetAttr(x,"loginshell"));
      F.write(Line);
      Fdb.write("0%u %s" % (I,Line));
      Fdb.write(".%s %s" % (GetAttr(x,"uid"),Line));
      Fdb.write("=%s %s" % (GetAttr(x,"uidnumber"),Line));
      I = I + 1;

  # Oops, something unspeakable happened.
  except:
   Die(File,F,Fdb);
   raise;
  Done(File,F,Fdb);

# Generate the shadow list
def GenShadow(l,File):
  F = None;
  Fdb = None;
  try:
   OldMask = os.umask(0077);
   F = open(File + ".tmp","w",0600);
   Fdb = open(File + ".tdb.tmp","w",0600);
   os.umask(OldMask);

   # Fetch all the users
   global PasswdAttrs;
   if PasswdAttrs == None:
      raise "No Users";

   I = 0;
   for x in PasswdAttrs:
      if x[1].has_key("uidnumber") == 0 or IsInGroup(x) == 0:
         continue;
	 
      Pass = GetAttr(x,"userpassword");
      if Pass[0:7] != "{crypt}" or len(Pass) > 50:
         Pass = '*';
      else:
         Pass = Pass[7:];
      Line = "%s:%s:%s:%s:%s:%s:%s:%s:" % (GetAttr(x,"uid"),\
              Pass,GetAttr(x,"shadowlastchange"),\
              GetAttr(x,"shadowmin"),GetAttr(x,"shadowmax"),\
              GetAttr(x,"shadowwarning"),GetAttr(x,"shadowinactive"),\
              GetAttr(x,"shadowexpire"));
      Line = Sanitize(Line) + "\n";
      F.write(Line);
      Fdb.write("0%u %s" % (I,Line));
      Fdb.write(".%s %s" % (GetAttr(x,"uid"),Line));
      I = I + 1;

  # Oops, something unspeakable happened.
  except:
   Die(File,F,Fdb);
   raise;
  Done(File,F,Fdb);

# Generate the shadow list
def GenSSHShadow(l,File):
  F = None;
  Fdb = None;
  try:
   OldMask = os.umask(0077);
   F = open(File + ".tmp","w",0600);
   Fdb = os.popen("cdbmake %s.cdb %s.cdb.tmp"%(File,File),"w");
   os.umask(OldMask);

   # Fetch all the users
   global PasswdAttrs;
   if PasswdAttrs == None:
      raise "No Users";

   I = 0;
   for x in PasswdAttrs:
      if x[1].has_key("uidnumber") == 0 or \
         x[1].has_key("sshrsaauthkey") == 0:
         continue;
      for I in x[1]["sshrsaauthkey"]:
         User = GetAttr(x,"uid");
         Line = "%s: %s" %(User,I);
         Line = Sanitize(Line) + "\n";
         F.write(Line);
         Fdb.write("+%d,%d:%s->%s\n"%(len(User),len(I),User,I));
   Fdb.write("\n");
  # Oops, something unspeakable happened.
  except:
   Die(File,F,Fdb);
   raise;
  if Fdb.close() != None:
     raise "cdbmake gave an error";
  Done(File,F,None);

# Generate the group list
def GenGroup(l,File):
  F = None;
  Fdb = None;
  try:
   F = open(File + ".tmp","w");
   Fdb = open(File + ".tdb.tmp","w");

   # Generate the GroupMap
   GroupMap = {};
   for x in GroupIDMap.keys():
      GroupMap[x] = [];
      
   # Fetch all the users
   global PasswdAttrs;
   if PasswdAttrs == None:
      raise "No Users";

   # Sort them into a list of groups having a set of users
   for x in PasswdAttrs:
      if x[1].has_key("uidnumber") == 0 or IsInGroup(x) == 0:
         continue;
      if x[1].has_key("supplementarygid") == 0:
         continue;
	 
      for I in x[1]["supplementarygid"]:
         if GroupMap.has_key(I):
	    GroupMap[I].append(GetAttr(x,"uid"));
	 else:
            print "Group does not exist ",I,"but",GetAttr(x,"uid"),"is in it";
	    
   # Output the group file.
   Counter = 0; 
   for x in GroupMap.keys():
      if GroupIDMap.has_key(x) == 0:
         continue;
      Line = "%s:x:%u:" % (x,GroupIDMap[x]);
      Comma = '';
      for I in GroupMap[x]:
        Line = Line + ("%s%s" % (Comma,I));
        Comma = ',';
      Line = Sanitize(Line) + "\n";
      F.write(Line);
      Fdb.write("0%u %s" % (Counter,Line));
      Fdb.write(".%s %s" % (x,Line));
      Fdb.write("=%u %s" % (GroupIDMap[x],Line));
      Counter = Counter + 1;
      
  # Oops, something unspeakable happened.
  except:
   Die(File,F,Fdb);
   raise;
  Done(File,F,Fdb);

# Generate the email forwarding list
def GenForward(l,File):
  F = None;
  Fdb = None;
  try:
   OldMask = os.umask(0022);
   F = open(File + ".tmp","w",0644);
   os.umask(OldMask);

   # Fetch all the users
   global PasswdAttrs;
   if PasswdAttrs == None:
      raise "No Users";

   # Write out the email address for each user
   for x in PasswdAttrs:
      if x[1].has_key("emailforward") == 0 or IsInGroup(x) == 0:
         continue;
      
      # Do not allow people to try to buffer overflow busted parsers
      if len(GetAttr(x,"emailforward")) > 200:
         continue;

      # Check the forwarding address
      if EmailCheck.match(GetAttr(x,"emailforward")) == None:
         continue;
      Line = "%s: %s" % (GetAttr(x,"uid"),GetAttr(x,"emailforward"));
      Line = Sanitize(Line) + "\n";
      F.write(Line);
      
  # Oops, something unspeakable happened.
  except:
   Die(File,F,Fdb);
   raise;
  Done(File,F,Fdb);

def GenAllForward(l,File):
  Fdb = None;
  try:
   OldMask = os.umask(0022);
   Fdb = os.popen("cdbmake %s %s.tmp"%(File,File),"w");
   os.umask(OldMask);

   # Fetch all the users
   global PasswdAttrs;
   if PasswdAttrs == None:
      raise "No Users";

   # Write out the email address for each user
   for x in PasswdAttrs:
      if x[1].has_key("emailforward") == 0:
         continue;
      
      # Do not allow people to try to buffer overflow busted parsers
      Forward = GetAttr(x,"emailforward");
      if len(Forward) > 200:
         continue;

      # Check the forwarding address
      if EmailCheck.match(Forward) == None:
         continue;
	 
      User = GetAttr(x,"uid");
      Fdb.write("+%d,%d:%s->%s\n"%(len(User),len(Forward),User,Forward));
   Fdb.write("\n");
  # Oops, something unspeakable happened.
  except:
    Fdb.close();
    raise;
  if Fdb.close() != None:
    raise "cdbmake gave an error";

# Generate the anon XEarth marker file 
def GenMarkers(l,File):
  F = None;
  Fdb = None;
  try:
   F = open(File + ".tmp","w");
   Fdb = None;

   # Fetch all the users
   global PasswdAttrs;
   if PasswdAttrs == None:
      raise "No Users";

   # Write out the position for each user
   for x in PasswdAttrs:
      if x[1].has_key("latitude") == 0 or x[1].has_key("longitude") == 0:
         continue;	 
      try:
         Line = "%8s %8s \"\""%(DecDegree(GetAttr(x,"latitude"),1),DecDegree(GetAttr(x,"longitude"),1));
         Line = Sanitize(Line) + "\n";
         F.write(Line);
      except:
         pass;
      
  # Oops, something unspeakable happened.
  except:
   Die(File,F,Fdb);
   raise;
  Done(File,F,Fdb);

# Generate the debian-private subscription list
def GenPrivate(l,File):
  F = None;
  Fdb = None;
  try:
   F = open(File + ".tmp","w");
   Fdb = None;

   # Fetch all the users
   global PasswdAttrs;
   if PasswdAttrs == None:
      raise "No Users";

   # Write out the position for each user
   for x in PasswdAttrs:
      if x[1].has_key("privatesub") == 0:
         continue;

      # If the account is locked, do not write it
      if (string.find(GetAttr(x,"userpassword"),"*LK*")  != -1):
         continue;

      # If the account has no PGP key, do not write it
      if x[1].has_key("keyfingerprint") == 0:
         continue;

      # Must be in the Debian group (yuk, hard coded for now)
      if GetAttr(x,"gidnumber") != "800":
	 continue;

      try:
         Line = "%s"%(GetAttr(x,"privatesub"));
         Line = Sanitize(Line) + "\n";
         F.write(Line);
      except:
         pass;
      
  # Oops, something unspeakable happened.
  except:
   Die(File,F,Fdb);
   raise;
  Done(File,F,Fdb);

# Generate the DNS Zone file
def GenDNS(l,File,BSMTPFile,HomePrefix):
  F = None;
  FB = None;
  try:
   F = open(File + ".tmp","w");
   FB = open(BSMTPFile + ".tmp","w");
   
   # Fetch all the users
   global PasswdAttrs;
   if PasswdAttrs == None:
      raise "No Users";

   # Write out the zone file entry for each user
   for x in PasswdAttrs:
      if x[1].has_key("dnszoneentry") == 0:
         continue;
      try:
         F.write("; %s\n"%(EmailAddress(x)));
         for z in x[1]["dnszoneentry"]:
            Split = string.split(string.lower(z));
	    if string.lower(Split[1]) == 'in':
               for y in range(0,len(Split)):
                  if Split[y] == "$":
                     Split[y] = "\n\t";
               Line = string.join(Split," ") + "\n";
               F.write(Line);
	       
	       Host = Split[0] + DNSZone;
	       if BSMTPCheck.match(Line) != None:
		   F.write("; Has BSMTP\n");
		   FB.write("%s: user=%s group=Debian file=%s%s/bsmtp/%s\n"%(Host,
 		               GetAttr(x,"uid"),HomePrefix,GetAttr(x,"uid"),Host));
			       
	       # Write some identification information
               if string.lower(Split[2]) == "a":
   	          Line = "%s IN TXT \"%s\"\n"%(Split[0],EmailAddress(x));
                  for y in x[1]["keyfingerprint"]:
  	             Line = Line + "%s IN TXT \"PGP %s\"\n"%(Split[0],FormatPGPKey(y));
                  F.write(Line);
	    else:
               Line = "; Err %s"%(str(Split));
               F.write(Line);

         F.write("\n");
      except:
	 F.write("; Errors\n");
         pass;
      
  # Oops, something unspeakable happened.
  except:
   Die(File,F,None);
   Die(BSMTPFile,FB,None);
   raise;
  Done(File,F,None);
  Done(BSMTPFile,FB,None);

# Generate the shadow list
def GenSSHKnown(l,File):
  F = None;
  Fdb = None;
  try:
   OldMask = os.umask(0022);
   F = open(File + ".tmp","w",0644);
   Fdb = os.popen("cdbmake %s.cdb %s.cdb.tmp"%(File,File),"w");
   os.umask(OldMask);

   # Fetch all the hosts
   HostKeys = l.search_s(HostBaseDn,ldap.SCOPE_ONELEVEL,"sshrsahostkey=*",\
                ["hostname","sshrsahostkey"]);
   
   if HostKeys == None:
      raise "No Hosts";

   I = 0;
   for x in HostKeys:
      if x[1].has_key("hostname") == 0 or \
         x[1].has_key("sshrsahostkey") == 0:
         continue;
      Host = GetAttr(x,"hostname");
      SHost = string.find(Host,".");
      for I in x[1]["sshrsahostkey"]:
         if SHost == None:
            Line = "%s %s" %(Host,I);
         else:
            Line = "%s,%s %s" %(Host,Host[0:SHost],I);
         Line = Sanitize(Line) + "\n";
         F.write(Line);
         Fdb.write("+%d,%d:%s->%s\n"%(len(Host),len(I),Host,I));
   Fdb.write("\n");
  # Oops, something unspeakable happened.
  except:
   Die(File,F,Fdb);
   raise;
  if Fdb.close() != None:
     raise "cdbmake gave an error";
  Done(File,F,None);


# Connect to the ldap server
l = ldap.open(LDAPServer);
F = open(PassDir+"/pass-"+pwd.getpwuid(os.getuid())[0],"r");
Pass = string.split(string.strip(F.readline())," ");
F.close();
l.simple_bind_s("uid="+Pass[0]+","+BaseDn,Pass[1]);

# Fetch all the groups
GroupIDMap = {};
Attrs = l.search_s(BaseDn,ldap.SCOPE_ONELEVEL,"gid=*",\
                  ["gid","gidnumber"]);

# Generate the GroupMap and GroupIDMap
for x in Attrs:
   if x[1].has_key("gidnumber") == 0:
      continue;
   GroupIDMap[x[1]["gid"][0]] = int(x[1]["gidnumber"][0]);

# Fetch all the users
PasswdAttrs = l.search_s(BaseDn,ldap.SCOPE_ONELEVEL,"uid=*",\
                ["uid","uidnumber","gidnumber","supplementarygid",\
                 "gecos","loginshell","userpassword","shadowlastchange",\
                 "shadowmin","shadowmax","shadowwarning","shadowinactive",
	         "shadowexpire","emailforward","latitude","longitude",\
                 "allowedhosts","sshrsaauthkey","dnszoneentry","cn","sn",\
	         "keyfingerprint","privatesub"]);

# Open the control file
if len(sys.argv) == 1:
   F = open(GenerateConf,"r");
else:
   F = open(sys.argv[1],"r")

# Generate global things
GlobalDir = GenerateDir+"/";
GenSSHShadow(l,GlobalDir+"ssh-rsa-shadow");
GenAllForward(l,GlobalDir+"mail-forward.cdb");
GenMarkers(l,GlobalDir+"markers");
GenPrivate(l,GlobalDir+"debian-private");
GenSSHKnown(l,GlobalDir+"ssh_known_hosts");

# Compatibility.
GenForward(l,GlobalDir+"forward-alias");
   
while(1):
   Line = F.readline();
   if Line == "":
      break;
   Line = string.strip(Line);
   if Line == "":
      continue;
   if Line[0] == '#':
      continue;

   Split = string.split(Line," ");
   OutDir = GenerateDir + '/' + Split[0] + '/';
   try: os.mkdir(OutDir);
   except: pass;

   # Get the group list and convert any named groups to numerics
   GroupList = {};
   ExtraList = {};
   for I in Split[2:]:
      if I[0] == '[':
         ExtraList[I] = None;
         continue;
      GroupList[I] = None;
      if GroupIDMap.has_key(I):
         GroupList[str(GroupIDMap[I])] = None;

   global Allowed,CurrentHost;
   Allowed = GroupList;
   CurrentHost = Split[0];

   sys.stdout.flush();
   GenPasswd(l,OutDir+"passwd",Split[1]);
   sys.stdout.flush();
   GenGroup(l,OutDir+"group");
   GenShadow(l,OutDir+"shadow");
	
   # Link in global things   
   DoLink(GlobalDir,OutDir,"ssh-rsa-shadow");
   DoLink(GlobalDir,OutDir,"ssh-rsa-shadow.cdb");
   DoLink(GlobalDir,OutDir,"markers");
   DoLink(GlobalDir,OutDir,"mail-forward.cdb");
   DoLink(GlobalDir,OutDir,"ssh_known_hosts");

   # Compatibility.
   DoLink(GlobalDir,OutDir,"forward-alias");

   if ExtraList.has_key("[DNS]"):
      GenDNS(l,OutDir+"dns-zone",OutDir+"bsmtp",Split[1]);
      
   if ExtraList.has_key("[PRIVATE]"):
      DoLink(GlobalDir,OutDir,"debian-private");