/* nslcd.c - ldap local connection daemon Copyright (C) 2006 West Consulting Copyright (C) 2006, 2007, 2008 Arthur de Jong This library is free software; you can redistribute it and/or modify it under the terms of the GNU Lesser General Public License as published by the Free Software Foundation; either version 2.1 of the License, or (at your option) any later version. This library is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details. You should have received a copy of the GNU Lesser General Public License along with this library; if not, write to the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA */ #include "config.h" #include #include #include #ifdef HAVE_STDINT_H #include #endif /* HAVE_STDINT_H */ #include #include #include #ifdef HAVE_GETOPT_H #include #endif /* HAVE_GETOPT_H */ #include #include #include #include #include #include #include #include #ifdef HAVE_GRP_H #include #endif /* HAVE_GRP_H */ #ifdef HAVE_NSS_H #include #endif /* HAVE_NSS_H */ #include #ifndef HAVE_GETOPT_LONG #include "compat/getopt_long.h" #endif /* not HAVE_GETOPT_LONG */ #ifndef HAVE_DAEMON #include "compat/daemon.h" #endif /* not HAVE_DAEMON */ #include "nslcd.h" #include "log.h" #include "cfg.h" #include "common.h" #include "compat/attrs.h" #include "compat/getpeercred.h" /* buffer sizes for I/O */ #define READBUFFER_MINSIZE 32 #define READBUFFER_MAXSIZE 64 #define WRITEBUFFER_MINSIZE 64 #define WRITEBUFFER_MAXSIZE 1*1024*1024 /* flag to indictate if we are in debugging mode */ static int nslcd_debugging=0; /* the exit flag to indicate that a signal was received */ static volatile int nslcd_exitsignal=0; /* the server socket used for communication */ static int nslcd_serversocket=-1; /* thread ids of all running threads */ static pthread_t *nslcd_threads; /* display version information */ static void display_version(FILE *fp) { fprintf(fp,"%s\n",PACKAGE_STRING); fprintf(fp,"Written by Luke Howard and Arthur de Jong.\n\n"); fprintf(fp,"Copyright (C) 1997-2008 Luke Howard, Arthur de Jong and West Consulting\n" "This is free software; see the source for copying conditions. There is NO\n" "warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.\n"); } /* display usage information to stdout and exit(status) */ static void display_usage(FILE *fp,const char *program_name) { fprintf(fp,"Usage: %s [OPTION]...\n",program_name); fprintf(fp,"Name Service LDAP connection daemon.\n"); fprintf(fp," -d, --debug don't fork and print debugging to stderr\n"); fprintf(fp," --help display this help and exit\n"); fprintf(fp," --version output version information and exit\n"); fprintf(fp,"\n" "Report bugs to <%s>.\n",PACKAGE_BUGREPORT); } /* the definition of options for getopt(). see getopt(2) */ static struct option const nslcd_options[] = { { "debug", no_argument, NULL, 'd' }, { "help", no_argument, NULL, 'h' }, { "version", no_argument, NULL, 'V' }, { NULL, 0, NULL, 0 } }; #define NSLCD_OPTIONSTRING "dhV" /* parse command line options and save settings in struct */ static void parse_cmdline(int argc,char *argv[]) { int optc; while ((optc=getopt_long(argc,argv,NSLCD_OPTIONSTRING,nslcd_options,NULL))!=-1) { switch (optc) { case 'd': /* -d, --debug don't fork and print debugging to stderr */ nslcd_debugging++; log_setdefaultloglevel(LOG_DEBUG); break; case 'h': /* --help display this help and exit */ display_usage(stdout,argv[0]); exit(EXIT_SUCCESS); case 'V': /* --version output version information and exit */ display_version(stdout); exit(EXIT_SUCCESS); case ':': /* missing required parameter */ case '?': /* unknown option character or extraneous parameter */ default: fprintf(stderr,"Try '%s --help' for more information.\n", argv[0]); exit(EXIT_FAILURE); } } /* check for remaining arguments */ if (optindldc_threads;i++) if (pthread_cancel(nslcd_threads[i])) { /* TODO: figure out if we can actually log from within a signal handler */ log_log(LOG_WARNING,"failed to stop thread %d (ignored): %s",i,strerror(errno)); } } /* do some cleaning up before terminating */ static void exithandler(void) { /* close socket if it's still in use */ if (nslcd_serversocket >= 0) { if (close(nslcd_serversocket)) log_log(LOG_WARNING,"problem closing server socket (ignored): %s",strerror(errno)); } /* remove existing named socket */ if (unlink(NSLCD_SOCKET)<0) { log_log(LOG_DEBUG,"unlink() of "NSLCD_SOCKET" failed (ignored): %s", strerror(errno)); } /* remove pidfile */ if (unlink(NSLCD_PIDFILE)<0) { log_log(LOG_DEBUG,"unlink() of "NSLCD_PIDFILE" failed (ignored): %s", strerror(errno)); } /* log exit */ log_log(LOG_INFO,"version %s bailing out",VERSION); } /* returns a socket ready to answer requests from the client, exit()s on error */ static int create_socket(void) { int sock; struct sockaddr_un addr; /* create a socket */ if ( (sock=socket(PF_UNIX,SOCK_STREAM,0))<0 ) { log_log(LOG_ERR,"cannot create socket: %s",strerror(errno)); exit(EXIT_FAILURE); } /* remove existing named socket */ if (unlink(NSLCD_SOCKET)<0) { log_log(LOG_DEBUG,"unlink() of "NSLCD_SOCKET" failed (ignored): %s", strerror(errno)); } /* create socket address structure */ memset(&addr,0,sizeof(struct sockaddr_un)); addr.sun_family=AF_UNIX; strncpy(addr.sun_path,NSLCD_SOCKET,sizeof(addr.sun_path)); addr.sun_path[sizeof(addr.sun_path)-1]='\0'; /* bind to the named socket */ if (bind(sock,(struct sockaddr *)&addr,sizeof(struct sockaddr_un))) { log_log(LOG_ERR,"bind() to "NSLCD_SOCKET" failed: %s", strerror(errno)); if (close(sock)) log_log(LOG_WARNING,"problem closing socket: %s",strerror(errno)); exit(EXIT_FAILURE); } /* close the file descriptor on exit */ if (fcntl(sock,F_SETFD,FD_CLOEXEC)<0) { log_log(LOG_ERR,"fctnl(F_SETFL,O_NONBLOCK) failed: %s",strerror(errno)); if (close(sock)) log_log(LOG_WARNING,"problem closing socket: %s",strerror(errno)); exit(EXIT_FAILURE); } /* set permissions of socket so anybody can do requests */ /* Note: we use chmod() here instead of fchmod() because fchmod does not work on sockets http://www.opengroup.org/onlinepubs/009695399/functions/fchmod.html http://lkml.org/lkml/2005/5/16/11 */ if (chmod(NSLCD_SOCKET,(mode_t)0666)) { log_log(LOG_ERR,"chmod(0666) failed: %s",strerror(errno)); if (close(sock)) log_log(LOG_WARNING,"problem closing socket: %s",strerror(errno)); exit(EXIT_FAILURE); } /* start listening for connections */ if (listen(sock,SOMAXCONN)<0) { log_log(LOG_ERR,"listen() failed: %s",strerror(errno)); if (close(sock)) log_log(LOG_WARNING,"problem closing socket: %s",strerror(errno)); exit(EXIT_FAILURE); } /* we're done */ return sock; } /* read the version information and action from the stream this function returns the read action in location pointer to by action */ static int read_header(TFILE *fp,int32_t *action) { int32_t tmpint32; /* read the protocol version */ READ_TYPE(fp,tmpint32,int32_t); if (tmpint32 != (int32_t)NSLCD_VERSION) { log_log(LOG_DEBUG,"wrong nslcd version id (%d)",(int)tmpint32); return -1; } /* read the request type */ READ(fp,action,sizeof(int32_t)); return 0; } /* read a request message, returns <0 in case of errors, this function closes the socket */ static void handleconnection(int sock,MYLDAP_SESSION *session) { TFILE *fp; int32_t action; struct timeval readtimeout,writetimeout; uid_t uid; gid_t gid; pid_t pid; /* indicate new connection to logging module (genrates unique id) */ log_newsession(); /* log connection */ if (getpeercred(sock,&uid,&gid,&pid)) log_log(LOG_DEBUG,"connection from unknown client: %s",strerror(errno)); else log_log(LOG_DEBUG,"connection from pid=%d uid=%d gid=%d", (int)pid,(int)uid,(int)gid); /* set the timeouts */ readtimeout.tv_sec=0; /* clients should send their request quickly */ readtimeout.tv_usec=500000; writetimeout.tv_sec=60; /* clients could be taking some time to process the results */ writetimeout.tv_usec=0; /* create a stream object */ if ((fp=tio_fdopen(sock,&readtimeout,&writetimeout, READBUFFER_MINSIZE,READBUFFER_MAXSIZE, WRITEBUFFER_MINSIZE,WRITEBUFFER_MAXSIZE))==NULL) { log_log(LOG_WARNING,"cannot create stream for writing: %s",strerror(errno)); (void)close(sock); return; } /* read request */ if (read_header(fp,&action)) { (void)tio_close(fp); return; } /* handle request */ switch (action) { case NSLCD_ACTION_ALIAS_BYNAME: (void)nslcd_alias_byname(fp,session); break; case NSLCD_ACTION_ALIAS_ALL: (void)nslcd_alias_all(fp,session); break; case NSLCD_ACTION_ETHER_BYNAME: (void)nslcd_ether_byname(fp,session); break; case NSLCD_ACTION_ETHER_BYETHER: (void)nslcd_ether_byether(fp,session); break; case NSLCD_ACTION_ETHER_ALL: (void)nslcd_ether_all(fp,session); break; case NSLCD_ACTION_GROUP_BYNAME: (void)nslcd_group_byname(fp,session); break; case NSLCD_ACTION_GROUP_BYGID: (void)nslcd_group_bygid(fp,session); break; case NSLCD_ACTION_GROUP_BYMEMBER: (void)nslcd_group_bymember(fp,session); break; case NSLCD_ACTION_GROUP_ALL: (void)nslcd_group_all(fp,session); break; case NSLCD_ACTION_HOST_BYNAME: (void)nslcd_host_byname(fp,session); break; case NSLCD_ACTION_HOST_BYADDR: (void)nslcd_host_byaddr(fp,session); break; case NSLCD_ACTION_HOST_ALL: (void)nslcd_host_all(fp,session); break; case NSLCD_ACTION_NETGROUP_BYNAME: (void)nslcd_netgroup_byname(fp,session); break; case NSLCD_ACTION_NETWORK_BYNAME: (void)nslcd_network_byname(fp,session); break; case NSLCD_ACTION_NETWORK_BYADDR: (void)nslcd_network_byaddr(fp,session); break; case NSLCD_ACTION_NETWORK_ALL: (void)nslcd_network_all(fp,session); break; case NSLCD_ACTION_PASSWD_BYNAME: (void)nslcd_passwd_byname(fp,session); break; case NSLCD_ACTION_PASSWD_BYUID: (void)nslcd_passwd_byuid(fp,session); break; case NSLCD_ACTION_PASSWD_ALL: (void)nslcd_passwd_all(fp,session); break; case NSLCD_ACTION_PROTOCOL_BYNAME: (void)nslcd_protocol_byname(fp,session); break; case NSLCD_ACTION_PROTOCOL_BYNUMBER:(void)nslcd_protocol_bynumber(fp,session); break; case NSLCD_ACTION_PROTOCOL_ALL: (void)nslcd_protocol_all(fp,session); break; case NSLCD_ACTION_RPC_BYNAME: (void)nslcd_rpc_byname(fp,session); break; case NSLCD_ACTION_RPC_BYNUMBER: (void)nslcd_rpc_bynumber(fp,session); break; case NSLCD_ACTION_RPC_ALL: (void)nslcd_rpc_all(fp,session); break; case NSLCD_ACTION_SERVICE_BYNAME: (void)nslcd_service_byname(fp,session); break; case NSLCD_ACTION_SERVICE_BYNUMBER: (void)nslcd_service_bynumber(fp,session); break; case NSLCD_ACTION_SERVICE_ALL: (void)nslcd_service_all(fp,session); break; case NSLCD_ACTION_SHADOW_BYNAME: if (uid==0) (void)nslcd_shadow_byname(fp,session); break; case NSLCD_ACTION_SHADOW_ALL: if (uid==0) (void)nslcd_shadow_all(fp,session); break; default: log_log(LOG_WARNING,"invalid request id: %d",(int)action); break; } /* we're done with the request */ myldap_session_cleanup(session); (void)tio_close(fp); return; } /* accept a connection on the socket */ static void acceptconnection(MYLDAP_SESSION *session) { int csock; int j; struct sockaddr_storage addr; socklen_t alen; /* accept a new connection */ alen=(socklen_t)sizeof(struct sockaddr_storage); csock=accept(nslcd_serversocket,(struct sockaddr *)&addr,&alen); if (csock<0) { if ((errno==EINTR)||(errno==EAGAIN)||(errno==EWOULDBLOCK)) { log_log(LOG_DEBUG,"accept() failed (ignored): %s",strerror(errno)); return; } log_log(LOG_ERR,"accept() failed: %s",strerror(errno)); return; } /* make sure O_NONBLOCK is not inherited */ if ((j=fcntl(csock,F_GETFL,0))<0) { log_log(LOG_ERR,"fctnl(F_GETFL) failed: %s",strerror(errno)); if (close(csock)) log_log(LOG_WARNING,"problem closing socket: %s",strerror(errno)); return; } if (fcntl(csock,F_SETFL,j&~O_NONBLOCK)<0) { log_log(LOG_ERR,"fctnl(F_SETFL,~O_NONBLOCK) failed: %s",strerror(errno)); if (close(csock)) log_log(LOG_WARNING,"problem closing socket: %s",strerror(errno)); return; } /* handle the connection */ handleconnection(csock,session); } /* write the current process id to the specified file */ static void create_pidfile(const char *filename) { FILE *fp; if (filename!=NULL) { if ((fp=fopen(filename,"w"))==NULL) { log_log(LOG_ERR,"cannot open pid file (%s): %s",filename,strerror(errno)); exit(EXIT_FAILURE); } if (fprintf(fp,"%d\n",(int)getpid())<=0) { log_log(LOG_ERR,"error writing pid file (%s)",filename); exit(EXIT_FAILURE); } if (fclose(fp)) { log_log(LOG_ERR,"error writing pid file (%s): %s",filename,strerror(errno)); exit(EXIT_FAILURE); } } } /* try to install signal handler and check result */ static void install_sighandler(int signum,RETSIGTYPE (*handler) (int)) { struct sigaction act; memset(&act,0,sizeof(struct sigaction)); act.sa_handler=handler; sigemptyset(&act.sa_mask); act.sa_flags=SA_RESTART|SA_NOCLDSTOP; if (sigaction(signum,&act,NULL)!=0) { log_log(LOG_ERR,"error installing signal handler for '%s': %s",signame(signum),strerror(errno)); exit(EXIT_FAILURE); } } static void worker_cleanup(void *arg) { MYLDAP_SESSION *session=(MYLDAP_SESSION *)arg; myldap_session_close(session); } static void *worker(void UNUSED(*arg)) { MYLDAP_SESSION *session; /* create a new LDAP session */ session=myldap_create_session(); /* clean up the session if we're done */ pthread_cleanup_push(worker_cleanup,session); /* start waiting for incoming connections */ while (1) { /* wait for a new connection */ acceptconnection(session); } pthread_cleanup_pop(1); return NULL; } /* the main program... */ int main(int argc,char *argv[]) { int i; /* parse the command line */ parse_cmdline(argc,argv); /* clear the environment */ /* TODO:implement */ /* check if we are already running */ /* FIXME: implement (maybe pass along options or commands) */ /* disable ldap lookups of host names to avoid lookup loop and fall back to files dns (a sensible default) */ /* TODO: parse /etc/nsswitch ourselves and just remove ldap from the list */ #ifdef HAVE___NSS_CONFIGURE_LOOKUP if (__nss_configure_lookup("hosts","files dns")) log_log(LOG_ERR,"unable to override hosts lookup method: %s",strerror(errno)); #endif /* HAVE___NSS_CONFIGURE_LOOKUP */ /* FIXME: have some other mechanism for systems that don't have this */ /* read configuration file */ cfg_init(NSS_LDAP_PATH_CONF); nslcd_cfg->ldc_debug=nslcd_debugging?nslcd_debugging-1:0; /* daemonize */ if ((!nslcd_debugging)&&(daemon(0,0)<0)) { log_log(LOG_ERR,"unable to daemonize: %s",strerror(errno)); exit(EXIT_FAILURE); } /* set default mode for pidfile and socket */ (void)umask((mode_t)0022); /* intilialize logging */ if (!nslcd_debugging) log_startlogging(); log_log(LOG_INFO,"version %s starting",VERSION); /* install handler to close stuff off on exit and log notice */ if (atexit(exithandler)) { log_log(LOG_ERR,"atexit() failed: %s",strerror(errno)); exit(EXIT_FAILURE); } /* write pidfile */ create_pidfile(NSLCD_PIDFILE); /* create socket */ nslcd_serversocket=create_socket(); #ifdef HAVE_SETGROUPS /* drop all supplemental groups */ if (setgroups(0,NULL)<0) log_log(LOG_WARNING,"cannot setgroups(0,NULL) (ignored): %s",strerror(errno)); else log_log(LOG_DEBUG,"setgroups(0,NULL) done"); #else /* HAVE_SETGROUPS */ log_log(LOG_DEBUG,"setgroups() not available"); #endif /* not HAVE_SETGROUPS */ /* change to nslcd gid */ if (nslcd_cfg->ldc_gid!=NOGID) { if (setgid(nslcd_cfg->ldc_gid)!=0) { log_log(LOG_ERR,"cannot setgid(%d): %s",(int)nslcd_cfg->ldc_gid,strerror(errno)); exit(EXIT_FAILURE); } log_log(LOG_DEBUG,"setgid(%d) done",(int)nslcd_cfg->ldc_gid); } /* change to nslcd uid */ if (nslcd_cfg->ldc_uid!=NOUID) { if (setuid(nslcd_cfg->ldc_uid)!=0) { log_log(LOG_ERR,"cannot setuid(%d): %s",(int)nslcd_cfg->ldc_uid,strerror(errno)); exit(EXIT_FAILURE); } log_log(LOG_DEBUG,"setuid(%d) done",(int)nslcd_cfg->ldc_uid); } /* install signalhandlers for some signals */ install_sighandler(SIGHUP, sigexit_handler); install_sighandler(SIGINT, sigexit_handler); install_sighandler(SIGQUIT,sigexit_handler); install_sighandler(SIGABRT,sigexit_handler); install_sighandler(SIGPIPE,SIG_IGN); install_sighandler(SIGTERM,sigexit_handler); install_sighandler(SIGUSR1,sigexit_handler); install_sighandler(SIGUSR2,sigexit_handler); /* TODO: install signal handlers for reloading configuration */ log_log(LOG_INFO,"accepting connections"); /* start worker threads */ nslcd_threads=(pthread_t *)malloc(nslcd_cfg->ldc_threads*sizeof(pthread_t)); if (nslcd_threads==NULL) { log_log(LOG_CRIT,"main(): malloc() failed to allocate memory"); exit(EXIT_FAILURE); } for (i=0;ildc_threads;i++) { if (pthread_create(&nslcd_threads[i],NULL,worker,NULL)) { log_log(LOG_ERR,"unable to start worker thread %d: %s",i,strerror(errno)); exit(EXIT_FAILURE); } } /* wait for all threads to die */ /* BUG: this causes problems if for some reason we want to exit but one of our threads hangs (e.g. has one of the LDAP locks) Other than that it may be a good idea to keep this thread more or less alive to do general house keeping things (e.g. checking signals etc) */ /* it is also better to always do thread_cancel() here instead of in the signal handler */ for (i=0;ildc_threads;i++) { if (pthread_join(nslcd_threads[i],NULL)) { log_log(LOG_ERR,"unable to wait for worker thread %d: %s",i,strerror(errno)); exit(EXIT_FAILURE); } } free(nslcd_threads); /* print something about received signals */ if (nslcd_exitsignal!=0) { log_log(LOG_INFO,"caught signal %s (%d), shutting down", signame(nslcd_exitsignal),nslcd_exitsignal); } return EXIT_FAILURE; }