ud-generate

   1 #!/usr/bin/env python
   2 # -*- mode: python -*-
   3 # Generates passwd, shadow and group files from the ldap directory.
   4
   5 #   Copyright (c) 2000-2001  Jason Gunthorpe <jgg@debian.org>
   6 #   Copyright (c) 2003-2004  James Troup <troup@debian.org>
   7 #   Copyright (c) 2004-2005,7  Joey Schulze <joey@infodrom.org>
   8 #   Copyright (c) 2001-2007  Ryan Murray <rmurray@debian.org>
   9 #   Copyright (c) 2008 Peter Palfrader <peter@palfrader.org>
  10 #   Copyright (c) 2008 Andreas Barth <aba@not.so.argh.org>
  11 #   Copyright (c) 2008 Mark Hymers <mhy@debian.org>
  12 #   Copyright (c) 2008 Luk Claes <luk@debian.org>
  13 #   Copyright (c) 2008 Thomas Viehmann <tv@beamnet.de>
  14 #
  15 #   This program is free software; you can redistribute it and/or modify
  16 #   it under the terms of the GNU General Public License as published by
  17 #   the Free Software Foundation; either version 2 of the License, or
  18 #   (at your option) any later version.
  19 #
  20 #   This program is distributed in the hope that it will be useful,
  21 #   but WITHOUT ANY WARRANTY; without even the implied warranty of
  22 #   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  23 #   GNU General Public License for more details.
  24 #
  25 #   You should have received a copy of the GNU General Public License
  26 #   along with this program; if not, write to the Free Software
  27 #   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  28
  29 import string, re, time, ldap, getopt, sys, os, pwd, posix, socket, base64, sha, shutil, errno, tarfile, grp
  30 from userdir_ldap import *;
  31
  32 global Allowed;
  33 global CurrentHost;
  34
  35 PasswdAttrs = None;
  36 GroupIDMap = {};
  37 SubGroupMap = {};
  38 Allowed = None;
  39 CurrentHost = "";
  40
  41 UUID_FORMAT = '[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}'
  42
  43 EmailCheck = re.compile("^([^ <>@]+@[^ ,<>@]+)?$");
  44 BSMTPCheck = re.compile(".*mx 0 (gluck)\.debian\.org\..*",re.DOTALL);
  45 PurposeHostField = re.compile(r"\[\[([\*\-]?[a-z0-9.\-]*)(?:\|.*)?\]\]")
  46 DNSZone = ".debian.net"
  47 Keyrings = ConfModule.sync_keyrings.split(":")
  48
  49 def safe_makedirs(dir):
  50     try:
  51         os.makedirs(dir)
  52     except OSError, e:
  53         if e.errno == errno.EEXIST:
  54             pass
  55         else:
  56             raise e
  57
  58 def safe_rmtree(dir):
  59     try:
  60         shutil.rmtree(dir)
  61     except OSError, e:
  62         if e.errno == errno.ENOENT:
  63             pass
  64         else:
  65             raise e
  66
  67 def Sanitize(Str):
  68   return Str.translate(string.maketrans("\n\r\t","$$$"))
  69
  70 def DoLink(From,To,File):
  71    try: posix.remove(To+File);
  72    except: pass;
  73    posix.link(From+File,To+File);
  74
  75 # See if this user is in the group list
  76 def IsInGroup(DnRecord):
  77   if Allowed == None:
  78      return 1;
  79
  80   # See if the primary group is in the list
  81   if Allowed.has_key(GetAttr(DnRecord,"gidNumber")) != 0:
  82      return 1;
  83
  84   # Check the host based ACL
  85   if DnRecord[1].has_key("allowedHost") != 0:
  86      for I in DnRecord[1]["allowedHost"]:
  87         if CurrentHost == I:
  88            return 1;
  89
  90   # See if there are supplementary groups
  91   if DnRecord[1].has_key("supplementaryGid") == 0:
  92      return 0;
  93
  94   supgroups=[]
  95   addGroups(supgroups, DnRecord[1]["supplementaryGid"], GetAttr(DnRecord,"uid"))
  96   for g in supgroups:
  97      if Allowed.has_key(g):
  98         return 1;
  99   return 0;
 100
 101 def Die(File,F,Fdb):
 102    if F != None:
 103       F.close();
 104    if Fdb != None:
 105       Fdb.close();
 106    try: os.remove(File + ".tmp");
 107    except: pass;
 108    try: os.remove(File + ".tdb.tmp");
 109    except: pass;
 110
 111 def Done(File,F,Fdb):
 112   if F != None:
 113     F.close();
 114     os.rename(File + ".tmp",File);
 115   if Fdb != None:
 116     Fdb.close();
 117     os.rename(File + ".tdb.tmp",File+".tdb");
 118
 119 # Generate the password list
 120 def GenPasswd(l,File,HomePrefix,PwdMarker):
 121   F = None;
 122   try:
 123    F = open(File + ".tdb.tmp","w");
 124
 125    userlist = {}
 126    # Fetch all the users
 127    global PasswdAttrs;
 128    if PasswdAttrs == None:
 129       raise "No Users";
 130
 131    I = 0;
 132    for x in PasswdAttrs:
 133       if x[1].has_key("uidNumber") == 0 or IsInGroup(x) == 0:
 134          continue;
 135
 136       # Do not let people try to buffer overflow some busted passwd parser.
 137       if len(GetAttr(x,"gecos")) > 100 or len(GetAttr(x,"loginShell")) > 50:
 138          continue;
 139
 140       userlist[GetAttr(x, "uid")] = int(GetAttr(x, "gidNumber"))
 141       Line = "%s:%s:%s:%s:%s:%s%s:%s" % (GetAttr(x,"uid"),\
 142               PwdMarker,\
 143               GetAttr(x,"uidNumber"),GetAttr(x,"gidNumber"),\
 144               GetAttr(x,"gecos"),HomePrefix,GetAttr(x,"uid"),\
 145               GetAttr(x,"loginShell"));
 146
 147       Line = Sanitize(Line) + "\n";
 148       F.write("0%u %s" % (I,Line));
 149       F.write(".%s %s" % (GetAttr(x,"uid"),Line));
 150       F.write("=%s %s" % (GetAttr(x,"uidNumber"),Line));
 151       I = I + 1;
 152
 153   # Oops, something unspeakable happened.
 154   except:
 155    Die(File,None,F);
 156    raise;
 157   Done(File,None,F);
 158
 159   # Return the list of users so we know which keys to export
 160   return userlist
 161
 162 # Generate the shadow list
 163 def GenShadow(l,File):
 164   F = None;
 165   try:
 166    OldMask = os.umask(0077);
 167    F = open(File + ".tdb.tmp","w",0600);
 168    os.umask(OldMask);
 169
 170    # Fetch all the users
 171    global PasswdAttrs;
 172    if PasswdAttrs == None:
 173       raise "No Users";
 174
 175    I = 0;
 176    for x in PasswdAttrs:
 177       if x[1].has_key("uidNumber") == 0 or IsInGroup(x) == 0:
 178          continue;
 179
 180       Pass = GetAttr(x,"userPassword");
 181       if Pass[0:7] != "{crypt}" or len(Pass) > 50:
 182          Pass = '*';
 183       else:
 184          Pass = Pass[7:];
 185
 186       # If the account is locked, mark it as such in shadow
 187       # See Debian Bug #308229 for why we set it to 1 instead of 0
 188       if (GetAttr(x,"userPassword").find("*LK*") != -1) \
 189           or GetAttr(x,"userPassword").startswith("!"):
 190          ShadowExpire = '1'
 191       else:
 192          ShadowExpire = GetAttr(x,"shadowExpire")
 193
 194       Line = "%s:%s:%s:%s:%s:%s:%s:%s:" % (GetAttr(x,"uid"),\
 195               Pass,GetAttr(x,"shadowLastChange"),\
 196               GetAttr(x,"shadowMin"),GetAttr(x,"shadowMax"),\
 197               GetAttr(x,"shadowWarning"),GetAttr(x,"shadowInactive"),\
 198               ShadowExpire);
 199       Line = Sanitize(Line) + "\n";
 200       F.write("0%u %s" % (I,Line));
 201       F.write(".%s %s" % (GetAttr(x,"uid"),Line));
 202       I = I + 1;
 203
 204   # Oops, something unspeakable happened.
 205   except:
 206    Die(File,None,F);
 207    raise;
 208   Done(File,None,F);
 209
 210 # Generate the sudo passwd file
 211 def GenShadowSudo(l,File, untrusted):
 212   F = None;
 213   try:
 214    OldMask = os.umask(0077);
 215    F = open(File + ".tmp","w",0600);
 216    os.umask(OldMask);
 217
 218    # Fetch all the users
 219    global PasswdAttrs;
 220    if PasswdAttrs == None:
 221       raise "No Users";
 222
 223    for x in PasswdAttrs:
 224       Pass = '*'
 225       if x[1].has_key("uidNumber") == 0 or IsInGroup(x) == 0:
 226          continue;
 227
 228       if x[1].has_key('sudoPassword'):
 229          for entry in x[1]['sudoPassword']:
 230             Match = re.compile('^('+UUID_FORMAT+') (confirmed:[0-9a-f]{40}|unconfirmed) ([a-z0-9.,*]+) ([^ ]+)$').match(entry)
 231             if Match == None:
 232                continue
 233             uuid = Match.group(1)
 234             status = Match.group(2)
 235             hosts = Match.group(3)
 236             cryptedpass = Match.group(4)
 237
 238             if status != 'confirmed:'+make_passwd_hmac('password-is-confirmed', 'sudo', x[1]['uid'][0], uuid, hosts, cryptedpass):
 239                continue
 240             for_all = hosts == "*"
 241             for_this_host = CurrentHost in hosts.split(',')
 242             if not (for_all or for_this_host):
 243                continue
 244             # ignore * passwords for untrusted hosts, but copy host specific passwords
 245             if for_all and untrusted:
 246                continue
 247             Pass = cryptedpass
 248             if for_this_host: # this makes sure we take a per-host entry over the for-all entry
 249               break
 250          if len(Pass) > 50:
 251             Pass = '*'
 252
 253       Line = "%s:%s" % (GetAttr(x,"uid"), Pass)
 254       Line = Sanitize(Line) + "\n";
 255       F.write("%s" % (Line));
 256
 257   # Oops, something unspeakable happened.
 258   except:
 259    Die(File,F,None);
 260    raise;
 261   Done(File,F,None);
 262
 263 # Generate the shadow list
 264 def GenSSHShadow(l):
 265    # Fetch all the users
 266    singlefile = None
 267    userfiles = []
 268
 269    global PasswdAttrs;
 270    if PasswdAttrs == None:
 271       raise "No Users";
 272
 273    safe_rmtree(os.path.join(GlobalDir, 'userkeys'))
 274    safe_makedirs(os.path.join(GlobalDir, 'userkeys'))
 275
 276    for x in PasswdAttrs:
 277       # If the account is locked, do not write it.
 278       # This is a partial stop-gap. The ssh also needs to change this
 279       # to ignore ~/.ssh/authorized* files.
 280       if (GetAttr(x,"userPassword").find("*LK*") != -1) \
 281              or GetAttr(x,"userPassword").startswith("!"):
 282          continue;
 283
 284       if x[1].has_key("uidNumber") == 0 or \
 285          x[1].has_key("sshRSAAuthKey") == 0:
 286          continue;
 287       User = GetAttr(x,"uid");
 288       F = None;
 289
 290       try:
 291          OldMask = os.umask(0077);
 292          File = os.path.join(GlobalDir, 'userkeys', User)
 293          F = open(File + ".tmp","w",0600);
 294          os.umask(OldMask);
 295
 296          for I in x[1]["sshRSAAuthKey"]:
 297             MultipleLine = "%s" % I
 298             MultipleLine = Sanitize(MultipleLine) + "\n"
 299             F.write(MultipleLine)
 300
 301          Done(File,F,None);
 302          userfiles.append(os.path.basename(File))
 303
 304       # Oops, something unspeakable happened.
 305       except IOError:
 306           Die(File,F,None)
 307           Die(masterFileName,masterFile,None)
 308           raise;
 309
 310    return userfiles
 311
 312 def GenSSHtarballs(userlist, SSHFiles, grouprevmap, target):
 313    OldMask = os.umask(0077);
 314    tf = tarfile.open(name=os.path.join(GlobalDir, 'ssh-keys-%s.tar.gz' % CurrentHost), mode='w:gz')
 315    os.umask(OldMask);
 316    for f in userlist.keys():
 317       if f not in SSHFiles:
 318          continue
 319       # If we're not exporting their primary group, don't export
 320       # the key and warn
 321       grname = None
 322       if userlist[f] in grouprevmap.keys():
 323          grname = grouprevmap[userlist[f]]
 324       else:
 325          try:
 326             if int(userlist[f]) <= 100:
 327                # In these cases, look it up in the normal way so we
 328                # deal with cases where, for instance, users are in group
 329                # users as their primary group.
 330                grname = grp.getgrgid(userlist[f])[0]
 331          except Exception, e:
 332             pass
 333
 334       if grname is None:
 335          print "User %s is supposed to have their key exported to host %s but their primary group (gid: %d) isn't in LDAP" % (f, CurrentHost, userlist[f])
 336          continue
 337
 338       to = tf.gettarinfo(os.path.join(GlobalDir, 'userkeys', f), f)
 339       # These will only be used where the username doesn't
 340       # exist on the target system for some reason; hence,
 341       # in those cases, the safest thing is for the file to
 342       # be owned by root but group nobody.  This deals with
 343       # the bloody obscure case where the group fails to exist
 344       # whilst the user does (in which case we want to avoid
 345       # ending up with a file which is owned user:root to avoid
 346       # a fairly obvious attack vector)
 347       to.uid = 0
 348       to.gid = 65534
 349       # Using the username / groupname fields avoids any need
 350       # to give a shit^W^W^Wcare about the UIDoffset stuff.
 351       to.uname = f
 352       to.gname = grname
 353       to.mode  = 0400
 354       tf.addfile(to, file(os.path.join(GlobalDir, 'userkeys', f)))
 355
 356    tf.close()
 357    os.rename(os.path.join(GlobalDir, 'ssh-keys-%s.tar.gz' % CurrentHost), target)
 358
 359 # add a list of groups to existing groups,
 360 # including all subgroups thereof, recursively.
 361 # basically this proceduces the transitive hull of the groups in
 362 # addgroups.
 363 def addGroups(existingGroups, newGroups, uid):
 364    for group in newGroups:
 365       # if it's a <group>@host, split it and verify it's on the current host.
 366       s = group.split('@', 1)
 367       if len(s) == 2 and s[1] != CurrentHost:
 368          continue;
 369       group = s[0]
 370
 371       # let's see if we handled this group already
 372       if group in existingGroups:
 373         continue
 374
 375       if not GroupIDMap.has_key(group):
 376          print "Group", group, "does not exist but", uid, "is in it"
 377          continue
 378
 379       existingGroups.append(group)
 380
 381       if SubGroupMap.has_key(group):
 382          addGroups(existingGroups, SubGroupMap[group], uid)
 383
 384 # Generate the group list
 385 def GenGroup(l,File):
 386   grouprevmap = {}
 387   F = None;
 388   try:
 389    F = open(File + ".tdb.tmp","w");
 390
 391    # Generate the GroupMap
 392    GroupMap = {};
 393    for x in GroupIDMap.keys():
 394       GroupMap[x] = [];
 395
 396    # Fetch all the users
 397    global PasswdAttrs;
 398    if PasswdAttrs == None:
 399       raise "No Users";
 400
 401    # Sort them into a list of groups having a set of users
 402    for x in PasswdAttrs:
 403       uid = GetAttr(x,"uid")
 404       if x[1].has_key("uidNumber") == 0 or IsInGroup(x) == 0:
 405          continue;
 406       if x[1].has_key("supplementaryGid") == 0:
 407          continue;
 408
 409       supgroups=[]
 410       addGroups(supgroups, x[1]["supplementaryGid"], uid)
 411       for g in supgroups:
 412          GroupMap[g].append(uid);
 413
 414    # Output the group file.
 415    J = 0;
 416    for x in GroupMap.keys():
 417       grouprevmap[GroupIDMap[x]] = x
 418       if GroupIDMap.has_key(x) == 0:
 419          continue;
 420       Line = "%s:x:%u:" % (x,GroupIDMap[x]);
 421       Comma = '';
 422       for I in GroupMap[x]:
 423         Line = Line + ("%s%s" % (Comma,I));
 424         Comma = ',';
 425       Line = Sanitize(Line) + "\n";
 426       F.write("0%u %s" % (J,Line));
 427       F.write(".%s %s" % (x,Line));
 428       F.write("=%u %s" % (GroupIDMap[x],Line));
 429       J = J + 1;
 430
 431   # Oops, something unspeakable happened.
 432   except:
 433    Die(File,None,F);
 434    raise;
 435   Done(File,None,F);
 436
 437   return grouprevmap
 438
 439 # Generate the email forwarding list
 440 def GenForward(l,File):
 441   F = None;
 442   try:
 443    OldMask = os.umask(0022);
 444    F = open(File + ".tmp","w",0644);
 445    os.umask(OldMask);
 446
 447    # Fetch all the users
 448    global PasswdAttrs;
 449    if PasswdAttrs == None:
 450       raise "No Users";
 451
 452    # Write out the email address for each user
 453    for x in PasswdAttrs:
 454       if x[1].has_key("emailForward") == 0 or IsInGroup(x) == 0:
 455          continue;
 456
 457       # Do not allow people to try to buffer overflow busted parsers
 458       if len(GetAttr(x,"emailForward")) > 200:
 459          continue;
 460
 461       # Check the forwarding address
 462       if EmailCheck.match(GetAttr(x,"emailForward")) == None:
 463          continue;
 464       Line = "%s: %s" % (GetAttr(x,"uid"),GetAttr(x,"emailForward"));
 465       Line = Sanitize(Line) + "\n";
 466       F.write(Line);
 467
 468   # Oops, something unspeakable happened.
 469   except:
 470    Die(File,F,None);
 471    raise;
 472   Done(File,F,None);
 473
 474 def GenAllForward(l,File):
 475   Fdb = None;
 476   try:
 477    OldMask = os.umask(0022);
 478    Fdb = os.popen("cdbmake %s %s.tmp"%(File,File),"w");
 479    os.umask(OldMask);
 480
 481    # Fetch all the users
 482    global PasswdAttrs;
 483    if PasswdAttrs == None:
 484       raise "No Users";
 485
 486    # Write out the email address for each user
 487    for x in PasswdAttrs:
 488       if x[1].has_key("emailForward") == 0:
 489          continue;
 490
 491       # Do not allow people to try to buffer overflow busted parsers
 492       Forward = GetAttr(x,"emailForward");
 493       if len(Forward) > 200:
 494          continue;
 495
 496       # Check the forwarding address
 497       if EmailCheck.match(Forward) == None:
 498          continue;
 499
 500       User = GetAttr(x,"uid");
 501       Fdb.write("+%d,%d:%s->%s\n"%(len(User),len(Forward),User,Forward));
 502    Fdb.write("\n");
 503   # Oops, something unspeakable happened.
 504   except:
 505     Fdb.close();
 506     raise;
 507   if Fdb.close() != None:
 508     raise "cdbmake gave an error";
 509
 510 # Generate the anon XEarth marker file
 511 def GenMarkers(l,File):
 512   F = None;
 513   try:
 514    F = open(File + ".tmp","w");
 515
 516    # Fetch all the users
 517    global PasswdAttrs;
 518    if PasswdAttrs == None:
 519       raise "No Users";
 520
 521    # Write out the position for each user
 522    for x in PasswdAttrs:
 523       if x[1].has_key("latitude") == 0 or x[1].has_key("longitude") == 0:
 524          continue;
 525       try:
 526          Line = "%8s %8s \"\""%(DecDegree(GetAttr(x,"latitude"),1),DecDegree(GetAttr(x,"longitude"),1));
 527          Line = Sanitize(Line) + "\n";
 528          F.write(Line);
 529       except:
 530          pass;
 531
 532   # Oops, something unspeakable happened.
 533   except:
 534    Die(File,F,None);
 535    raise;
 536   Done(File,F,None);
 537
 538 # Generate the debian-private subscription list
 539 def GenPrivate(l,File):
 540   F = None;
 541   try:
 542    F = open(File + ".tmp","w");
 543
 544    # Fetch all the users
 545    global PasswdAttrs;
 546    if PasswdAttrs == None:
 547       raise "No Users";
 548
 549    # Write out the position for each user
 550    for x in PasswdAttrs:
 551       if x[1].has_key("privateSub") == 0:
 552          continue;
 553
 554       # If the account is locked, do not write it
 555       if (GetAttr(x,"userPassword").find("*LK*") != -1) \
 556              or GetAttr(x,"userPassword").startswith("!"):
 557          continue;
 558
 559       # If the account has no PGP key, do not write it
 560       if x[1].has_key("keyFingerPrint") == 0:
 561          continue;
 562
 563       # Must be in the Debian group (yuk, hard coded for now)
 564       if GetAttr(x,"gidNumber") != "800":
 565          continue;
 566
 567       try:
 568          Line = "%s"%(GetAttr(x,"privateSub"));
 569          Line = Sanitize(Line) + "\n";
 570          F.write(Line);
 571       except:
 572          pass;
 573
 574   # Oops, something unspeakable happened.
 575   except:
 576    Die(File,F,None);
 577    raise;
 578   Done(File,F,None);
 579
 580 # Generate a list of locked accounts
 581 def GenDisabledAccounts(l,File):
 582   F = None;
 583   try:
 584    F = open(File + ".tmp","w");
 585
 586    # Fetch all the users
 587    global PasswdAttrs;
 588    if PasswdAttrs == None:
 589       raise "No Users";
 590
 591    I = 0;
 592    for x in PasswdAttrs:
 593       if x[1].has_key("uidNumber") == 0:
 594          continue;
 595
 596       Pass = GetAttr(x,"userPassword");
 597       Line = ""
 598       # *LK* is the reference value for a locked account
 599       # password starting with ! is also a locked account
 600       if Pass.find("*LK*") != -1 or Pass.startswith("!"):
 601          # Format is <login>:<reason>
 602          Line = "%s:%s" % (GetAttr(x,"uid"), "Account is locked")
 603
 604       if Line != "":
 605          F.write(Sanitize(Line) + "\n")
 606
 607   # Oops, something unspeakable happened.
 608   except:
 609    Die(File,F,None);
 610    raise;
 611   Done(File,F,None);
 612
 613 # Generate the list of local addresses that refuse all mail
 614 def GenMailDisable(l,File):
 615   F = None;
 616   try:
 617    F = open(File + ".tmp","w");
 618
 619    # Fetch all the users
 620    global PasswdAttrs;
 621    if PasswdAttrs == None:
 622       raise "No Users";
 623
 624    for x in PasswdAttrs:
 625       Reason = None
 626
 627       if x[1].has_key("mailDisableMessage"):
 628          Reason = GetAttr(x,"mailDisableMessage")
 629       else:
 630          continue
 631
 632       # Must be in the Debian group (yuk, hard coded for now)
 633       if GetAttr(x,"gidNumber") != "800":
 634          continue;
 635
 636       try:
 637          Line = "%s: %s"%(GetAttr(x,"uid"),Reason);
 638          Line = Sanitize(Line) + "\n";
 639          F.write(Line);
 640       except:
 641          pass;
 642
 643   # Oops, something unspeakable happened.
 644   except:
 645    Die(File,F,None);
 646    raise;
 647   Done(File,F,None);
 648
 649 # Generate a list of uids that should have boolean affects applied
 650 def GenMailBool(l,File,Key):
 651   F = None;
 652   try:
 653    F = open(File + ".tmp","w");
 654
 655    # Fetch all the users
 656    global PasswdAttrs;
 657    if PasswdAttrs == None:
 658       raise "No Users";
 659
 660    for x in PasswdAttrs:
 661       Reason = None
 662
 663       if x[1].has_key(Key) == 0:
 664          continue
 665
 666       # Must be in the Debian group (yuk, hard coded for now)
 667       if GetAttr(x,"gidNumber") != "800":
 668          continue
 669
 670       if GetAttr(x,Key) != "TRUE":
 671          continue
 672
 673       try:
 674          Line = "%s"%(GetAttr(x,"uid"));
 675          Line = Sanitize(Line) + "\n";
 676          F.write(Line);
 677       except:
 678          pass;
 679
 680   # Oops, something unspeakable happened.
 681   except:
 682    Die(File,F,None);
 683    raise;
 684   Done(File,F,None);
 685
 686 # Generate a list of hosts for RBL or whitelist purposes.
 687 def GenMailList(l,File,Key):
 688   F = None;
 689   try:
 690    F = open(File + ".tmp","w");
 691
 692    # Fetch all the users
 693    global PasswdAttrs;
 694    if PasswdAttrs == None:
 695       raise "No Users";
 696
 697    for x in PasswdAttrs:
 698       Reason = None
 699
 700       if x[1].has_key(Key) == 0:
 701          continue
 702
 703       # Must be in the Debian group (yuk, hard coded for now)
 704       if GetAttr(x,"gidNumber") != "800":
 705          continue
 706
 707       try:
 708          found = 0
 709          Line = None
 710          for z in x[1][Key]:
 711              if Key == "mailWhitelist":
 712                  if re.match('^[-\w.]+(/[\d]+)?$',z) == None:
 713                      continue
 714              else:
 715                  if re.match('^[-\w.]+$',z) == None:
 716                      continue
 717              if found == 0:
 718                  found = 1
 719                  Line = GetAttr(x,"uid")
 720              else:
 721                  Line += " "
 722              Line += ": " + z
 723              if Key == "mailRHSBL":
 724                  Line += "/$sender_address_domain"
 725
 726          if Line != None:
 727              Line = Sanitize(Line) + "\n";
 728              F.write(Line);
 729       except:
 730          pass;
 731
 732   # Oops, something unspeakable happened.
 733   except:
 734    Die(File,F,None);
 735    raise;
 736   Done(File,F,None);
 737
 738 def isRoleAccount(pwEntry):
 739    if not pwEntry.has_key("objectClass"):
 740       raise "pwEntry has no objectClass"
 741    oc =  pwEntry['objectClass']
 742    try:
 743       i = oc.index('debianRoleAccount')
 744       return True
 745    except ValueError:
 746       return False
 747
 748 # Generate the DNS Zone file
 749 def GenDNS(l,File,HomePrefix):
 750   F = None;
 751   try:
 752    F = open(File + ".tmp","w");
 753
 754    # Fetch all the users
 755    global PasswdAttrs;
 756    if PasswdAttrs == None:
 757       raise "No Users";
 758
 759    # Write out the zone file entry for each user
 760    for x in PasswdAttrs:
 761       if x[1].has_key("dnsZoneEntry") == 0:
 762          continue;
 763
 764       # If the account has no PGP key, do not write it
 765       if x[1].has_key("keyFingerPrint") == 0 and not isRoleAccount(x[1]):
 766          continue;
 767       try:
 768          F.write("; %s\n"%(EmailAddress(x)));
 769          for z in x[1]["dnsZoneEntry"]:
 770             Split = z.lower().split()
 771             if Split[1].lower() == 'in':
 772                for y in range(0,len(Split)):
 773                   if Split[y] == "$":
 774                      Split[y] = "\n\t";
 775                Line = " ".join(Split) + "\n";
 776                F.write(Line);
 777
 778                Host = Split[0] + DNSZone;
 779                if BSMTPCheck.match(Line) != None:
 780                    F.write("; Has BSMTP\n");
 781
 782                # Write some identification information
 783                if Split[2].lower() == "a":
 784                   Line = "%s IN TXT \"%s\"\n"%(Split[0],EmailAddress(x));
 785                   for y in x[1]["keyFingerPrint"]:
 786                      Line = Line + "%s IN TXT \"PGP %s\"\n"%(Split[0],FormatPGPKey(y));
 787                   F.write(Line);
 788             else:
 789                Line = "; Err %s"%(str(Split));
 790                F.write(Line);
 791
 792          F.write("\n");
 793       except:
 794          F.write("; Errors\n");
 795          pass;
 796
 797   # Oops, something unspeakable happened.
 798   except:
 799    Die(File,F,None);
 800    raise;
 801   Done(File,F,None);
 802
 803 # Generate the DNS SSHFP records
 804 def GenSSHFP(l,File,HomePrefix):
 805   F = None
 806   try:
 807    F = open(File + ".tmp","w")
 808
 809    # Fetch all the hosts
 810    global HostAttrs
 811    if HostAttrs == None:
 812       raise "No Hosts"
 813
 814    for x in HostAttrs:
 815       if x[1].has_key("hostname") == 0 or \
 816          x[1].has_key("sshRSAHostKey") == 0:
 817          continue
 818       Host = GetAttr(x,"hostname");
 819       Algorithm = None
 820       for I in x[1]["sshRSAHostKey"]:
 821          Split = I.split()
 822          if Split[0] == 'ssh-rsa':
 823             Algorithm = 1
 824          if Split[0] == 'ssh-dss':
 825             Algorithm = 2
 826          if Algorithm == None:
 827             continue
 828          Fingerprint = sha.new(base64.decodestring(Split[1])).hexdigest()
 829          Line = "%s. IN SSHFP %u 1 %s" % (Host,Algorithm,Fingerprint)
 830          Line = Sanitize(Line) + "\n"
 831          F.write(Line)
 832   # Oops, something unspeakable happened.
 833   except:
 834    Die(File,F,None)
 835    raise;
 836   Done(File,F,None)
 837
 838 # Generate the BSMTP file
 839 def GenBSMTP(l,File,HomePrefix):
 840   F = None;
 841   try:
 842    F = open(File + ".tmp","w");
 843
 844    # Fetch all the users
 845    global PasswdAttrs;
 846    if PasswdAttrs == None:
 847       raise "No Users";
 848
 849    # Write out the zone file entry for each user
 850    for x in PasswdAttrs:
 851       if x[1].has_key("dnsZoneEntry") == 0:
 852          continue;
 853
 854       # If the account has no PGP key, do not write it
 855       if x[1].has_key("keyFingerPrint") == 0:
 856          continue;
 857       try:
 858          for z in x[1]["dnsZoneEntry"]:
 859             Split = z.lower().split()
 860             if Split[1].lower() == 'in':
 861                for y in range(0,len(Split)):
 862                   if Split[y] == "$":
 863                      Split[y] = "\n\t";
 864                Line = " ".join(Split) + "\n";
 865
 866                Host = Split[0] + DNSZone;
 867                if BSMTPCheck.match(Line) != None:
 868                    F.write("%s: user=%s group=Debian file=%s%s/bsmtp/%s\n"%(Host,
 869                                GetAttr(x,"uid"),HomePrefix,GetAttr(x,"uid"),Host));
 870
 871       except:
 872          F.write("; Errors\n");
 873          pass;
 874
 875   # Oops, something unspeakable happened.
 876   except:
 877    Die(File,F,None);
 878    raise;
 879   Done(File,F,None);
 880
 881 # cache IP adresses
 882 HostToIPCache = {}
 883 def HostToIP(Host):
 884     global HostToIPCache
 885     if not Host in HostToIPCache:
 886         IPAdressesT = None
 887         try:
 888             IPAdressesT = list(set([ (a[0],a[4][0]) for a in socket.getaddrinfo(Host, None)]))
 889         except socket.gaierror, (code):
 890             if code[0] != -2: raise
 891         IPAdresses = []
 892         for addr in IPAdressesT:
 893             if addr[0] == socket.AF_INET: IPAdresses += [addr[1], "::ffff:"+addr[1]]
 894             else: IPAdresses += [addr[1]]
 895         HostToIPCache[Host] = IPAdresses
 896     return HostToIPCache[Host]
 897
 898
 899 # Generate the ssh known hosts file
 900 def GenSSHKnown(l,File,mode=None):
 901   F = None;
 902   try:
 903    OldMask = os.umask(0022);
 904    F = open(File + ".tmp","w",0644);
 905    os.umask(OldMask);
 906
 907    global HostAttrs
 908    if HostAttrs == None:
 909       raise "No Hosts";
 910
 911    for x in HostAttrs:
 912       if x[1].has_key("hostname") == 0 or \
 913          x[1].has_key("sshRSAHostKey") == 0:
 914          continue;
 915       Host = GetAttr(x,"hostname");
 916       HostNames = [ Host ]
 917       if Host.endswith(HostDomain):
 918          HostNames.append(Host[:-(len(HostDomain)+1)])
 919
 920       # in the purpose field [[host|some other text]] (where some other text is optional)
 921       # makes a hyperlink on the web thing. we now also add these hosts to the ssh known_hosts
 922       # file.  But so that we don't have to add everything we link we can add an asterisk
 923       # and say [[*... to ignore it.  In order to be able to add stuff to ssh without
 924       # http linking it we also support [[-hostname]] entries.
 925       for i in x[1].get("purpose",[]):
 926          m = PurposeHostField.match(i)
 927          if m:
 928             m = m.group(1)
 929             # we ignore [[*..]] entries
 930             if m.startswith('*'):
 931                continue;
 932             if m.startswith('-'):
 933                m = m[1:]
 934             if m:
 935                HostNames.append(m)
 936                if m.endswith(HostDomain):
 937                   HostNames.append(m[:-(len(HostDomain)+1)])
 938
 939       for I in x[1]["sshRSAHostKey"]:
 940          if mode and mode == 'authorized_keys':
 941             #Line = 'command="rsync --server --sender -pr . /var/cache/userdir-ldap/hosts/%s",no-port-forwarding,no-X11-forwarding,no-agent-forwarding,from="%s" %s' % (Host, ",".join(HNames + HostToIP(Host)), I)
 942             Line = 'command="rsync --server --sender -pr . /var/cache/userdir-ldap/hosts/%s",no-port-forwarding,no-X11-forwarding,no-agent-forwarding %s' % (Host,I)
 943          else:
 944             Line = "%s %s" %(",".join(HostNames + HostToIP(Host)), I);
 945          Line = Sanitize(Line) + "\n";
 946          F.write(Line);
 947   # Oops, something unspeakable happened.
 948   except:
 949    Die(File,F,None);
 950    raise;
 951   Done(File,F,None);
 952
 953 # Generate the debianhosts file (list of all IP addresses)
 954 def GenHosts(l,File):
 955   F = None
 956   try:
 957     OldMask = os.umask(0022)
 958     F = open(File + ".tmp","w",0644)
 959     os.umask(OldMask)
 960
 961     # Fetch all the hosts
 962     hostnames = l.search_s(HostBaseDn, ldap.SCOPE_ONELEVEL, "hostname=*",
 963                            ["hostname"])
 964
 965     if hostnames == None:
 966        raise "No Hosts"
 967
 968     seen = set()
 969     for x in hostnames:
 970       host = GetAttr(x,"hostname", None)
 971       if host:
 972         addrs = []
 973         try:
 974           addrs += socket.getaddrinfo(host, None, socket.AF_INET)
 975         except socket.error:
 976           pass
 977         try:
 978           addrs += socket.getaddrinfo(host, None, socket.AF_INET6)
 979         except socket.error:
 980           pass
 981
 982         for addrinfo in addrs:
 983           if addrinfo[0] in (socket.AF_INET, socket.AF_INET6):
 984             addr = addrinfo[4][0]
 985             if addr not in seen:
 986               print >> F, addrinfo[4][0]
 987               seen.add(addr)
 988   # Oops, something unspeakable happened.
 989   except:
 990     Die(File,F,None)
 991     raise
 992   Done(File,F,None)
 993
 994 def GenKeyrings(l,OutDir):
 995   for k in Keyrings:
 996     shutil.copy(k, OutDir)
 997
 998
 999 # Connect to the ldap server
1000 l = connectLDAP()
1001 F = open(PassDir+"/pass-"+pwd.getpwuid(os.getuid())[0],"r");
1002 Pass = F.readline().strip().split(" ")
1003 F.close();
1004 l.simple_bind_s("uid="+Pass[0]+","+BaseDn,Pass[1]);
1005
1006 # Fetch all the groups
1007 GroupIDMap = {};
1008 Attrs = l.search_s(BaseDn,ldap.SCOPE_ONELEVEL,"gid=*",\
1009                   ["gid","gidNumber","subGroup"]);
1010
1011 # Generate the SubGroupMap and GroupIDMap
1012 for x in Attrs:
1013    if x[1].has_key("gidNumber") == 0:
1014       continue;
1015    GroupIDMap[x[1]["gid"][0]] = int(x[1]["gidNumber"][0]);
1016    if x[1].has_key("subGroup") != 0:
1017       SubGroupMap.setdefault(x[1]["gid"][0], []).extend(x[1]["subGroup"]);
1018
1019 # Fetch all the users
1020 PasswdAttrs = l.search_s(BaseDn,ldap.SCOPE_ONELEVEL,"uid=*",\
1021                 ["uid","uidNumber","gidNumber","supplementaryGid",\
1022                  "gecos","loginShell","userPassword","shadowLastChange",\
1023                  "shadowMin","shadowMax","shadowWarning","shadowInactive",
1024                  "shadowExpire","emailForward","latitude","longitude",\
1025                  "allowedHost","sshRSAAuthKey","dnsZoneEntry","cn","sn",\
1026                  "keyFingerPrint","privateSub","mailDisableMessage",\
1027                  "mailGreylisting","mailCallout","mailRBL","mailRHSBL",\
1028                  "mailWhitelist", "sudoPassword", "objectClass"]);
1029 # Fetch all the hosts
1030 HostAttrs    = l.search_s(HostBaseDn,ldap.SCOPE_ONELEVEL,"sshRSAHostKey=*",\
1031                 ["hostname","sshRSAHostKey","purpose"]);
1032
1033 # Open the control file
1034 if len(sys.argv) == 1:
1035    F = open(GenerateConf,"r");
1036 else:
1037    F = open(sys.argv[1],"r")
1038
1039 # Generate global things
1040 GlobalDir = GenerateDir+"/";
1041 SSHFiles = GenSSHShadow(l);
1042 GenAllForward(l,GlobalDir+"mail-forward.cdb");
1043 GenMarkers(l,GlobalDir+"markers");
1044 GenPrivate(l,GlobalDir+"debian-private");
1045 GenDisabledAccounts(l,GlobalDir+"disabled-accounts");
1046 GenSSHKnown(l,GlobalDir+"ssh_known_hosts");
1047 #GenSSHKnown(l,GlobalDir+"authorized_keys", 'authorized_keys');
1048 GenHosts(l,GlobalDir+"debianhosts");
1049 GenMailDisable(l,GlobalDir+"mail-disable");
1050 GenMailBool(l,GlobalDir+"mail-greylist","mailGreylisting");
1051 GenMailBool(l,GlobalDir+"mail-callout","mailCallout");
1052 GenMailList(l,GlobalDir+"mail-rbl","mailRBL");
1053 GenMailList(l,GlobalDir+"mail-rhsbl","mailRHSBL");
1054 GenMailList(l,GlobalDir+"mail-whitelist","mailWhitelist");
1055 GenKeyrings(l,GlobalDir);
1056
1057 # Compatibility.
1058 GenForward(l,GlobalDir+"forward-alias");
1059
1060 while(1):
1061    Line = F.readline();
1062    if Line == "":
1063       break;
1064    Line = Line.strip()
1065    if Line == "":
1066       continue;
1067    if Line[0] == '#':
1068       continue;
1069
1070    Split = Line.split(" ")
1071    OutDir = GenerateDir + '/' + Split[0] + '/';
1072    try: os.mkdir(OutDir);
1073    except: pass;
1074
1075    # Get the group list and convert any named groups to numerics
1076    GroupList = {};
1077    ExtraList = {};
1078    for I in Split[2:]:
1079       if I[0] == '[':
1080          ExtraList[I] = None;
1081          continue;
1082       GroupList[I] = None;
1083       if GroupIDMap.has_key(I):
1084          GroupList[str(GroupIDMap[I])] = None;
1085
1086    Allowed = GroupList;
1087    if Allowed == {}:
1088      Allowed = None
1089    CurrentHost = Split[0];
1090
1091    DoLink(GlobalDir,OutDir,"debianhosts");
1092    DoLink(GlobalDir,OutDir,"ssh_known_hosts");
1093    DoLink(GlobalDir,OutDir,"disabled-accounts")
1094
1095    sys.stdout.flush();
1096    if ExtraList.has_key("[NOPASSWD]"):
1097       userlist = GenPasswd(l,OutDir+"passwd",Split[1], "*");
1098    else:
1099       userlist = GenPasswd(l,OutDir+"passwd",Split[1], "x");
1100    sys.stdout.flush();
1101    grouprevmap = GenGroup(l,OutDir+"group");
1102    GenShadowSudo(l, OutDir+"sudo-passwd", ExtraList.has_key("[UNTRUSTED]") or ExtraList.has_key("[NOPASSWD]"))
1103
1104    # Now we know who we're allowing on the machine, export
1105    # the relevant ssh keys
1106    GenSSHtarballs(userlist, SSHFiles, grouprevmap, os.path.join(OutDir, 'ssh-keys.tar.gz'))
1107
1108    if ExtraList.has_key("[UNTRUSTED]"):
1109      print "[UNTRUSTED] tag is obsolete and may be removed in the future."
1110      continue;
1111    if not ExtraList.has_key("[NOPASSWD]"):
1112      GenShadow(l,OutDir+"shadow");
1113
1114    # Link in global things
1115    if not ExtraList.has_key("[NOMARKERS]"):
1116      DoLink(GlobalDir,OutDir,"markers");
1117    DoLink(GlobalDir,OutDir,"mail-forward.cdb");
1118    DoLink(GlobalDir,OutDir,"mail-disable");
1119    DoLink(GlobalDir,OutDir,"mail-greylist");
1120    DoLink(GlobalDir,OutDir,"mail-callout");
1121    DoLink(GlobalDir,OutDir,"mail-rbl");
1122    DoLink(GlobalDir,OutDir,"mail-rhsbl");
1123    DoLink(GlobalDir,OutDir,"mail-whitelist");
1124
1125    # Compatibility.
1126    DoLink(GlobalDir,OutDir,"forward-alias");
1127
1128    if ExtraList.has_key("[DNS]"):
1129       GenDNS(l,OutDir+"dns-zone",Split[1]);
1130       GenSSHFP(l,OutDir+"dns-sshfp",Split[1])
1131
1132    if ExtraList.has_key("[BSMTP]"):
1133       GenBSMTP(l,OutDir+"bsmtp",Split[1])
1134
1135    if ExtraList.has_key("[PRIVATE]"):
1136       DoLink(GlobalDir,OutDir,"debian-private")
1137
1138    if ExtraList.has_key("[KEYRING]"):
1139       for k in Keyrings:
1140         DoLink(GlobalDir,OutDir,os.path.basename(k))
1141    else:
1142      for k in Keyrings:
1143        try: posix.remove(OutDir+os.path.basename(k));
1144        except: pass;
1145
1146 # vim:set et:
1147 # vim:set ts=3:
1148 # vim:set shiftwidth=3: