git.sur5r.net Git - openldap/blob - clients/finger/main.c

   1 /*
   2  * Copyright (c) 1990,1994 Regents of the University of Michigan.
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms are permitted
   6  * provided that this notice is preserved and that due credit is given
   7  * to the University of Michigan at Ann Arbor. The name of the University
   8  * may not be used to endorse or promote products derived from this
   9  * software without specific prior written permission. This software
  10  * is provided ``as is'' without express or implied warranty.
  11  */
  12
  13 #include "portable.h"
  14
  15 #include <stdio.h>
  16 #include <ctype.h>
  17 #include <signal.h>
  18
  19 #include <ac/socket.h>
  20 #include <ac/string.h>
  21 #include <ac/syslog.h>
  22 #include <ac/time.h>
  23 #include <ac/unistd.h>
  24 #include <ac/wait.h>
  25
  26 #include <sys/resource.h>
  27
  28 #include "lber.h"
  29 #include "ldap.h"
  30
  31 #include "disptmpl.h"
  32
  33 #include "ldapconfig.h"
  34
  35
  36 int     dosyslog = 1;
  37 char    *ldaphost = NULL;
  38 int     ldapport = 0;
  39 char    *base = NULL;
  40 int     deref;
  41 char    *filterfile = FILTERFILE;
  42 char    *templatefile = TEMPLATEFILE;
  43 int     rdncount = FINGER_RDNCOUNT;
  44
  45 static do_query();
  46 static do_search();
  47 static do_read();
  48 static print_attr();
  49
  50 static usage( name )
  51 char    *name;
  52 {
  53         fprintf( stderr, "usage: %s [-l] [-x ldaphost] [-p ldapport] [-f filterfile] [-t templatefile] [-c rdncount]\r\n", name );
  54         exit( 1 );
  55 }
  56
  57 main (argc, argv)
  58 int     argc;
  59 char    **argv;
  60 {
  61         int                     i;
  62         char                    *myname;
  63         unsigned long           mypeer = -1;
  64         struct hostent          *hp;
  65         struct sockaddr_in      peername;
  66         int                     peernamelen;
  67         int                     interactive = 0;
  68         extern char             *optarg;
  69
  70         deref = FINGER_DEREF;
  71         while ( (i = getopt( argc, argv, "f:ilp:t:x:p:c:" )) != EOF ) {
  72                 switch( i ) {
  73                 case 'f':       /* ldap filter file */
  74                         filterfile = strdup( optarg );
  75                         break;
  76
  77                 case 'i':       /* interactive */
  78                         interactive = 1;
  79                         break;
  80
  81                 case 'l':       /* don't do syslogging */
  82                         dosyslog = 0;
  83                         break;
  84
  85                 case 't':       /* ldap template file */
  86                         templatefile = strdup( optarg );
  87                         break;
  88
  89                 case 'x':       /* specify ldap host */
  90                         ldaphost = strdup( optarg );
  91                         break;
  92
  93                 case 'p':       /* specify ldap port */
  94                         ldapport = atoi( optarg );
  95                         break;
  96
  97                 case 'c':       /* specify number of DN components to show */
  98                         rdncount = atoi( optarg );
  99                         break;
 100
 101                 default:
 102                         usage( argv[0] );
 103                 }
 104         }
 105
 106         if ( !interactive ) {
 107                 peernamelen = sizeof(peername);
 108                 if ( getpeername( 0, (struct sockaddr *)&peername,
 109                     &peernamelen ) != 0 ) {
 110                         perror( "getpeername" );
 111                         exit( 1 );
 112                 }
 113                 mypeer = (unsigned long) peername.sin_addr.s_addr;
 114         }
 115
 116 #ifdef FINGER_BANNER
 117         if ( FINGER_BANNER != NULL && strcmp( FINGER_BANNER, "" ) != 0 ) {
 118                 printf( FINGER_BANNER );
 119                 fflush( stdout );
 120         }
 121 #endif
 122
 123         if ( (myname = strrchr( argv[0], '/' )) == NULL )
 124                 myname = strdup( argv[0] );
 125         else
 126                 myname = strdup( myname + 1 );
 127
 128         if ( dosyslog ) {
 129 #ifdef LOG_LOCAL4
 130                 openlog( myname, OPENLOG_OPTIONS, LOG_LOCAL4 );
 131 #else
 132                 openlog( myname, OPENLOG_OPTIONS );
 133 #endif
 134         }
 135
 136         if ( dosyslog && mypeer != -1 ) {
 137                 struct in_addr  addr;
 138
 139                 hp = gethostbyaddr( (char *) &mypeer, sizeof(mypeer), AF_INET );
 140                 addr.s_addr = mypeer;
 141                 syslog( LOG_INFO, "connection from %s (%s)", (hp == NULL) ?
 142                     "unknown" : hp->h_name, inet_ntoa( addr ) );
 143         }
 144
 145         do_query();
 146
 147         return( 0 );
 148 }
 149
 150 static do_query()
 151 {
 152         char            buf[256];
 153         int             len, rc, tblsize;
 154         struct timeval  timeout;
 155         fd_set          readfds;
 156         LDAP            *ld;
 157
 158         if ( (ld = ldap_open( ldaphost, ldapport )) == NULL ) {
 159                 fprintf( stderr, FINGER_UNAVAILABLE );
 160                 perror( "ldap_open" );
 161                 exit( 1 );
 162         }
 163
 164         {
 165                 int limit = FINGER_SIZELIMIT;
 166                 ldap_set_option(ld, LDAP_OPT_SIZELIMIT, &limit);
 167         }
 168         ldap_set_option(ld, LDAP_OPT_DEREF, &deref);
 169
 170         if ( ldap_simple_bind_s( ld, NULL, NULL )
 171                 != LDAP_SUCCESS )
 172         {
 173                 fprintf( stderr, FINGER_UNAVAILABLE );
 174                 ldap_perror( ld, "ldap_simple_bind_s" );
 175                 exit( 1 );
 176         }
 177
 178 #ifdef HAVE_SYSCONF
 179         tblsize = sysconf( _SC_OPEN_MAX );
 180 #elif HAVE_GETDTABLESIZE
 181         tblsize = getdtablesize();
 182 #else
 183         tblsize = FD_SETSIZE;
 184 #endif
 185
 186 #ifdef FD_SETSIZE
 187         if (tblsize > FD_SETSIZE) {
 188                 tblsize = FD_SETSIZE;
 189         }
 190 #endif  /* FD_SETSIZE*/
 191
 192         timeout.tv_sec = FINGER_TIMEOUT;
 193         timeout.tv_usec = 0;
 194         FD_ZERO( &readfds );
 195         FD_SET( fileno( stdin ), &readfds );
 196
 197         if ( (rc = select( tblsize, &readfds, 0, 0, &timeout )) <= 0 ) {
 198                 if ( rc < 0 )
 199                         perror( "select" );
 200                 else
 201                         fprintf( stderr, "connection timed out on input\r\n" );
 202                 exit( 1 );
 203         }
 204
 205         if ( fgets( buf, sizeof(buf), stdin ) == NULL )
 206                 exit( 1 );
 207
 208         len = strlen( buf );
 209
 210         /* strip off \r \n */
 211         if ( buf[len - 1] == '\n' ) {
 212                 buf[len - 1] = '\0';
 213                 len--;
 214         }
 215         if ( buf[len - 1] == '\r' ) {
 216                 buf[len - 1] = '\0';
 217                 len--;
 218         }
 219
 220         if ( len == 0 ) {
 221                 printf( "No campus-wide login information available.  Info for this machine only:\r\n" );
 222                 fflush( stdout );
 223                 execl( FINGER_CMD, FINGER_CMD, NULL );
 224         } else {
 225                 char    *p;
 226
 227                 /* skip and ignore stinking /w */
 228                 if ( strncmp( buf, "/W ", 2 ) == 0 ) {
 229                         p = buf + 2;
 230                 } else {
 231                         p = buf;
 232                 }
 233
 234                 for ( ; *p && isspace( *p ); p++ )
 235                         ;       /* NULL */
 236
 237                 do_search( ld, p );
 238         }
 239 }
 240
 241 static void
 242 spaces2dots( s )
 243     char        *s;
 244 {
 245         for ( ; *s; s++ ) {
 246                 if ( *s == ' ' ) {
 247                         *s = '.';
 248                 }
 249         }
 250 }
 251
 252 static do_search( ld, buf )
 253 LDAP    *ld;
 254 char    *buf;
 255 {
 256         char            *dn, *rdn;
 257         char            **title;
 258         int             rc, matches, i, ufn;
 259         struct timeval  tv;
 260         LDAPFiltDesc    *fd;
 261         LDAPFiltInfo    *fi;
 262         LDAPMessage     *result, *e;
 263         static char     *attrs[] = { "cn", "title", "objectClass", "joinable",
 264 #ifdef FINGER_SORT_ATTR
 265                                         FINGER_SORT_ATTR,
 266 #endif
 267                                         0 };
 268         extern int      strcasecmp();
 269
 270         ufn = 0;
 271 #ifdef FINGER_UFN
 272         if ( strchr( buf, ',' ) != NULL ) {
 273                 ldap_ufn_setprefix( ld, base );
 274                 tv.tv_sec = FINGER_TIMEOUT;
 275                 tv.tv_usec = 0;
 276                 ldap_ufn_timeout( (void *) &tv );
 277
 278                 if ( (rc = ldap_ufn_search_s( ld, buf, attrs, 0, &result ))
 279                     != LDAP_SUCCESS && rc != LDAP_SIZELIMIT_EXCEEDED ) {
 280                         fprintf( stderr, FINGER_UNAVAILABLE );
 281                         ldap_perror( ld, "ldap_search_st" );
 282                         exit( 1 );
 283                 }
 284
 285                 matches = ldap_count_entries( ld, result );
 286                 ufn = 1;
 287         } else {
 288 #endif
 289                 if ( (fd = ldap_init_getfilter( filterfile ))
 290                     == NULL ) {
 291                         fprintf( stderr, "Cannot open filter file (%s)\n",
 292                             filterfile );
 293                         exit( 1 );
 294                 }
 295
 296                 for ( fi = ldap_getfirstfilter( fd, "finger", buf );
 297                     fi != NULL;
 298                     fi = ldap_getnextfilter( fd ) )
 299                 {
 300                         tv.tv_sec = FINGER_TIMEOUT;
 301                         tv.tv_usec = 0;
 302                         if ( (rc = ldap_search_st( ld, base, LDAP_SCOPE_SUBTREE,
 303                             fi->lfi_filter, attrs, 0, &tv, &result ))
 304                             != LDAP_SUCCESS && rc != LDAP_SIZELIMIT_EXCEEDED
 305                             && rc != LDAP_TIMELIMIT_EXCEEDED )
 306                         {
 307                                 fprintf( stderr, FINGER_UNAVAILABLE );
 308                                 ldap_perror( ld, "ldap_search_st" );
 309                                 exit( 1 );
 310                         }
 311
 312                         if ( (matches = ldap_count_entries( ld, result )) != 0 )
 313                                 break;
 314
 315                         ldap_msgfree( result );
 316                         result = NULL;
 317                 }
 318 #ifdef FINGER_UFN
 319         }
 320 #endif
 321
 322         if ( rc == LDAP_SIZELIMIT_EXCEEDED ) {
 323                 printf( "(Partial results - a size limit was exceeded)\r\n" );
 324         } else if ( rc == LDAP_TIMELIMIT_EXCEEDED ) {
 325                 printf( "(Partial results - a time limit was exceeded)\r\n" );
 326         }
 327
 328         if ( matches == 0 ) {
 329                 printf( FINGER_NOMATCH );
 330                 fflush( stdout );
 331         } else if ( matches < 0 ) {
 332                 fprintf( stderr, "error return from ldap_count_entries\r\n" );
 333                 exit( 1 );
 334         } else if ( matches <= FINGER_LISTLIMIT ) {
 335                 printf( "%d %s match%s found for \"%s\":\r\n", matches,
 336                     ufn ? "UFN" : fi->lfi_desc, matches > 1 ? "es" : "", buf );
 337                 fflush( stdout );
 338
 339                 for ( e = ldap_first_entry( ld, result ); e != NULL; ) {
 340                         do_read( ld, e );
 341                         e = ldap_next_entry( ld, e );
 342                         if ( e != NULL ) {
 343                                 printf( "--------------------\r\n" );
 344                         }
 345                 }
 346         } else {
 347                 printf( "%d %s matches for \"%s\":\r\n", matches,
 348                     ufn ? "UFN" : fi->lfi_desc, buf );
 349                 fflush( stdout );
 350
 351 #ifdef FINGER_SORT_ATTR
 352                 ldap_sort_entries( ld, &result, FINGER_SORT_ATTR, strcasecmp );
 353 #endif
 354
 355                 for ( e = ldap_first_entry( ld, result ); e != NULL;
 356                     e = ldap_next_entry( ld, e ) ) {
 357                         char    *p;
 358
 359                         dn = ldap_get_dn( ld, e );
 360                         rdn = dn;
 361                         if ( (p = strchr( dn, ',' )) != NULL )
 362                                 *p = '\0';
 363                         while ( *rdn && *rdn != '=' )
 364                                 rdn++;
 365                         if ( *rdn )
 366                                 rdn++;
 367
 368                         /* hack attack */
 369                         for ( i = 0; buf[i] != '\0'; i++ ) {
 370                                 if ( buf[i] == '.' || buf[i] == '_' )
 371                                         buf[i] = ' ';
 372                         }
 373                         if ( strcasecmp( rdn, buf ) == 0 ) {
 374                                 char    **cn;
 375                                 int     i, last;
 376
 377                                 cn = ldap_get_values( ld, e, "cn" );
 378                                 for ( i = 0; cn[i] != NULL; i++ ) {
 379                                         last = strlen( cn[i] ) - 1;
 380                                         if ( isdigit( cn[i][last] ) ) {
 381                                                 rdn = strdup( cn[i] );
 382                                                 break;
 383                                         }
 384                                 }
 385                         }
 386
 387                         title = ldap_get_values( ld, e, "title" );
 388
 389                         spaces2dots( rdn );
 390                         printf( "  %-20s    %s\r\n", rdn,
 391                             title ? title[0] : "" );
 392                         if ( title != NULL ) {
 393                                 for ( i = 1; title[i] != NULL; i++ )
 394                                         printf( "  %-20s    %s\r\n", "",
 395                                             title[i] );
 396                         }
 397                         fflush( stdout );
 398
 399                         if ( title != NULL )
 400                                 ldap_value_free( title );
 401
 402                         free( dn );
 403                 }
 404         }
 405
 406         if ( result != NULL ) {
 407                 ldap_msgfree( result );
 408         }
 409         ldap_unbind( ld );
 410 }
 411
 412
 413 static int
 414 entry2textwrite( void *fp, char *buf, int len )
 415 {
 416         return( fwrite( buf, len, 1, (FILE *)fp ) == 0 ? -1 : len );
 417 }
 418
 419
 420 static do_read( ld, e )
 421 LDAP            *ld;
 422 LDAPMessage     *e;
 423 {
 424         static struct ldap_disptmpl *tmpllist;
 425         static char     *defattrs[] = { "mail", NULL };
 426         static char     *mailvals[] = FINGER_NOEMAIL;
 427         static char     **defvals[] = { mailvals, NULL };
 428
 429         ldap_init_templates( templatefile, &tmpllist );
 430
 431         if ( ldap_entry2text_search( ld, NULL, base, e, tmpllist, defattrs,
 432             defvals, entry2textwrite, (void *)stdout, "\r\n", rdncount,
 433             LDAP_DISP_OPT_DOSEARCHACTIONS ) != LDAP_SUCCESS ) {
 434                 ldap_perror( ld, "ldap_entry2text_search" );
 435                 exit( 1 );
 436         }
 437
 438         if ( tmpllist != NULL ) {
 439             ldap_free_templates( tmpllist );
 440         }
 441 }