Fixes for James

[mirror/userdir-ldap.git] / ud-fingerserv

diff --git a/ud-fingerserv b/ud-fingerserv
index 07466de..17c64a7 100755 (executable)
--- a/ud-fingerserv
+++ b/ud-fingerserv
@@ -1,12 +1,11 @@
 #!/usr/bin/perl
-# $Id: ud-fingerserv,v 1.10 1999/10/17 02:21:48 tausq Exp $
+# $Id: ud-fingerserv,v 1.14 2000/02/21 06:08:32 tausq Exp $
 
 # (c) 1999 Randolph Chung. Licensed under the GPL. <tausq@debian.org>
 
 use lib '/var/www/userdir-ldap/';
 #use lib '/home/randolph/projects/userdir-ldap/web';
 use strict vars;
-#use Apache::Registry;
 use IO::Handle;
 use IO::Socket;
 use POSIX qw(:sys_wait_h);
@@ -28,10 +27,11 @@ my %attrs = (
   'email' => 'Email',
   'keyfingerprint' => 'Fingerprint',
   'key' => 'Key block',
-  'ircnick' => 'IRC nickname'
+  'ircnick' => 'IRC nickname',
+  'labeledurl' => 'URL'
 );
 
-my @summarykeys = ('cn', 'mn', 'sn', 'email', 'ircnick', 'keyfingerprint', 'key');
+my @summarykeys = ('cn', 'mn', 'sn', 'email', 'labeledurl', 'ircnick', 'keyfingerprint', 'key');
 
 $SIG{__DIE__} = \&DieHandler;
 $SIG{INT} = \&DieHandler;
@@ -65,8 +65,8 @@ if (!$use_inetd) {
     $client->autoflush(1);
     my $hostinfo = gethostbyaddr($client->peeraddr, AF_INET);
     &log(sprintf("[Connect from %s]", $hostinfo || $client->peerhost));
-    my $query = <$client>;
-    &ProcessQuery($client, $query);
+    my $query = &readdata($client);
+    &ProcessQuery($client, $query) if (defined($query));
     $client->close;
     exit;
   } continue {
@@ -77,8 +77,8 @@ if (!$use_inetd) {
   my $sockaddr = getpeername(STDIN);
   my ($port, $addr) = unpack_sockaddr_in(getpeername(STDIN));
   &log(sprintf("[Connect from %s (%s)]", gethostbyaddr($addr, AF_INET), inet_ntoa($addr)));
-  my $query = <STDIN>;
-  &ProcessQuery(\*STDOUT, $query);
+  my $query = &readdata(\*STDIN);
+  &ProcessQuery(\*STDOUT, $query) if (defined($query));
   exit;
 }
 
@@ -103,7 +103,7 @@ sub ProcessQuery {
   $query =~ s/[^\/,0-9a-z]//gi; # be paranoid about input
   my ($uid, $fields) = split(/\//, $query, 2);
   
-  if ($uid =~ /^help$/i) {
+  if (($uid eq "") || ($uid =~ /^help$/i)) {
     &sendhelp($client);
     return;
   }
@@ -113,7 +113,12 @@ sub ProcessQuery {
   $mesg = $ldap->search(base  => $config{basedn}, filter => "uid=$uid");
   $mesg->code && die $mesg->error;
   $entries = $mesg->as_struct;
-  
+
+  if ($mesg->count == 0) {
+    print $client "$uid not found at db.debian.org\n";
+    exit 0;
+  }
+
   foreach $dn (sort {$entries->{$a}->{sn}->[0] <=> $entries->{$b}->{sn}->[0]} keys(%$entries)) {
     $data = $entries->{$dn};
 
@@ -165,6 +170,29 @@ sub log {
   print STDERR "$time $msg\n";
 }
 
+sub readdata {
+  my $fh = shift;
+  my $in = undef;
+  my $out = undef;
+  my $bytesread = 0;
+  my $ret;
+
+  my $flags= fcntl($fh, F_GETFL, 0)
+     or die "Can't get flags for socket: $!\n";
+  fcntl($fh, F_SETFL, $flags | O_NONBLOCK)
+     or die "Can't make socket nonblocking: $!\n";
+                                               
+  while (($bytesread < 1024) && ($out !~ /\n/)) {
+    $ret = sysread($fh, $in, 1024);
+    return undef if (!defined($ret) || ($ret == 0));
+    $bytesread += $ret;
+    $out .= $in;
+  }
+
+  $out =~ /(.*?)\n/;
+  return $1;
+}
+
 sub sendhelp {
   my $client = shift;
   
@@ -174,7 +202,7 @@ sub sendhelp {
   print $client "  where uid is the user id of the user\n";
   print $client "  the optional attributes parameter specifies what to return\n";
   print $client "    if nothing is specified, all attributes are returned.\n";
-  print $client "    The following attributes are current supported:\n";
+  print $client "    The following attributes are currently supported:\n";
   foreach (@summarykeys) {
     print $client "      $_ : $attrs{$_}\n";
   }