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