ud-fingerserv

   1 #!/usr/bin/perl
   2 # $Id: ud-fingerserv,v 1.5 1999/10/16 21:50:52 tausq Exp $
   3
   4 # (c) 1999 Randolph Chung. Licensed under the GPL. <tausq@debian.org>
   5
   6 use lib '/home/randolph/projects/userdir-ldap/web';
   7 use strict vars;
   8 #use Apache::Registry;
   9 use IO::Handle;
  10 use IO::Socket;
  11 use POSIX qw(:sys_wait_h);
  12 use Getopt::Std;
  13 use Util;
  14 use Net::LDAP qw(:all);
  15
  16 # Global settings...
  17 my %config = &Util::ReadConfigFile;
  18 my %opts;
  19 getopts("iqh", \%opts);
  20 my $use_inetd = $config{use_inetd} || $opts{i};
  21 $| = 1;
  22
  23 my %attrs = (
  24   'cn' => 'First name',
  25   'mn' => 'Middle name',
  26   'sn' => 'Last name',
  27   'keyfingerprint' => 'Fingerprint',
  28   'key' => 'Key block',
  29   'ircnick' => 'IRC nickname'
  30 );
  31
  32 my @summarykeys = ('cn', 'mn', 'sn', 'ircnick', 'keyfingerprint', 'key');
  33
  34 $SIG{__DIE__} = \&DieHandler;
  35 $SIG{INT} = \&DieHandler;
  36 $SIG{CHLD} = \&Reaper;
  37
  38 &help if (defined($opts{h}));
  39 #my $logfh = STDOUT; #TODO
  40
  41 &log("Binding to LDAP server at $config{ldaphost}") if (defined($opts{v}));
  42 my $ldap = Net::LDAP->new($config{ldaphost}) || die $1;
  43 $ldap->bind;
  44
  45 if ($use_inetd == 0) {
  46   &log("Binding to port 79") if (defined($opts{v}));
  47   my $server = IO::Socket::INET->new(Proto => 'tcp',
  48                                      LocalPort => 'finger(79)',
  49                                      Listen => SOMAXCONN,
  50                                      Reuse => 1);
  51
  52   die "Cannot listen on finger port" unless $server;
  53   &log("[Server listening for connections]");
  54
  55   my ($pid, $client, $hostinfo);
  56
  57   while ($client = $server->accept()) {
  58     &log("Forking to handle client request") if (defined($opts{v}));
  59     next if $pid = fork; # parent
  60     die "fork: $!" unless defined $pid;
  61
  62     # child
  63     $client->autoflush(1);
  64     my $hostinfo = gethostbyaddr($client->peeraddr, AF_INET);
  65     &log(sprintf("[Connect from %s]", $hostinfo || $client->peerhost));
  66     my $query = <$client>;
  67     &ProcessQuery($client, $query);
  68     $client->close;
  69     exit;
  70   } continue {
  71     $client->close;
  72   }
  73 } else { # inetd
  74   &log("inetd mode");
  75   my $sockaddr = getpeername(STDIN);
  76   my ($port, $addr) = unpack_sockaddr_in(getpeername(STDIN));
  77   &log(sprintf("[Connect from %s (%s)]", gethostbyaddr($addr, AF_INET), inet_ntoa($addr)));
  78   my $query = <STDIN>;
  79   &ProcessQuery(\*STDOUT, $query);
  80   exit;
  81 }
  82
  83 $ldap->unbind;
  84
  85 sub DieHandler {
  86   $ldap->unbind if (defined($ldap));
  87   exit 0;
  88 }
  89
  90 sub Reaper {
  91   1 until (-1 == waitpid(-1, WNOHANG));
  92   $SIG{CHLD} = \&Reaper;
  93 }
  94
  95 sub ProcessQuery {
  96   my $client = shift;
  97   my $query = shift;
  98
  99   my ($uid, $fields, $mesg, $entries, $dn, $key, $pid, $data);
 100
 101   $query =~ s/[^\/,0-9a-z]//gi; # be paranoid about input
 102   my ($uid, $fields) = split(/\//, $query, 2);
 103
 104   &log("Looking up $uid at $config{basedn}, uid=$uid");
 105
 106   $mesg = $ldap->search(base  => $config{basedn}, filter => "uid=$uid");
 107   $mesg->code && die $mesg->error;
 108   $entries = $mesg->as_struct;
 109
 110   foreach $dn (sort {$entries->{$a}->{sn}->[0] <=> $entries->{$b}->{sn}->[0]} keys(%$entries)) {
 111     $data = $entries->{$dn};
 112
 113     $data->{key} = [];
 114     foreach (@{$data->{keyfingerprint}}) {
 115       push (@{$data->{key}}, "\n".&Util::FetchKey($_));
 116     }
 117
 118     print $client "$dn\n";
 119     if (!$fields) {
 120       foreach $key (@summarykeys) {
 121         foreach (@{$data->{$key}}) {
 122           print $client "$attrs{$key}: ";
 123           print $client "$_\n";
 124         }
 125       }
 126     } else {
 127   #     print "$fields\n";
 128       foreach $key (split(/,/, $fields)) {
 129         foreach (@{$data->{$key}}) {
 130           print $client "$attrs{$key}: ";
 131           print $client "$_\n";
 132         }
 133       }
 134     }
 135   }
 136 }
 137
 138 sub help {
 139   print "fingerserv [-i | -q | -v | -h]\n";
 140   print "-i = inetd mode; otherwise runs standalone\n";
 141   print "-q = quiet mode; no output\n";
 142   print "-v = verbose mode\n";
 143   print "-h = this help message\n";
 144   exit 0;
 145 }
 146
 147 sub log {
 148   my $msg = shift;
 149   return if (defined($opts{q}));
 150
 151   my $time = localtime;
 152   print STDERR "$time $msg\n";
 153 }