Add a better error message to wb_common.c when unable to connect to a pipe
[tprouty/samba.git] / source3 / nsswitch / winbindd.c
index 4bfec1afe49338387de64b90fcc42b561631b027..4f161604b5d3bd01cf1fff65bfa22ad4684a1636 100644 (file)
@@ -5,6 +5,7 @@
 
    Copyright (C) by Tim Potter 2000-2002
    Copyright (C) Andrew Tridgell 2002
+   Copyright (C) Jelmer Vernooij 2003
    
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
 #include "winbindd.h"
 
 BOOL opt_nocache = False;
-BOOL opt_dual_daemon = False;
+BOOL opt_dual_daemon = True;
 
 /* Reload configuration */
 
 static BOOL reload_services_file(BOOL test)
 {
        BOOL ret;
-       pstring logfile;
 
        if (lp_loaded()) {
                pstring fname;
@@ -43,21 +43,16 @@ static BOOL reload_services_file(BOOL test)
                }
        }
 
-       snprintf(logfile, sizeof(logfile), "%s/log.winbindd", dyn_LOGFILEBASE);
-       lp_set_logfile(logfile);
-
        reopen_logs();
        ret = lp_load(dyn_CONFIGFILE,False,False,True);
 
-       snprintf(logfile, sizeof(logfile), "%s/log.winbindd", dyn_LOGFILEBASE);
-       lp_set_logfile(logfile);
-
        reopen_logs();
        load_interfaces();
 
        return(ret);
 }
 
+
 #if DUMP_CORE
 
 /**************************************************************************** **
@@ -122,8 +117,8 @@ static void winbindd_status(void)
        if (DEBUGLEVEL >= 2 && winbindd_num_clients()) {
                DEBUG(2, ("\tclient list:\n"));
                for(tmp = winbindd_client_list(); tmp; tmp = tmp->next) {
-                       DEBUG(2, ("\t\tpid %d, sock %d, rbl %d, wbl %d\n",
-                                 tmp->pid, tmp->sock, tmp->read_buf_len, 
+                       DEBUG(2, ("\t\tpid %lu, sock %d, rbl %d, wbl %d\n",
+                                 (unsigned long)tmp->pid, tmp->sock, tmp->read_buf_len, 
                                  tmp->write_buf_len));
                }
        }
@@ -134,7 +129,6 @@ static void winbindd_status(void)
 static void print_winbindd_status(void)
 {
        winbindd_status();
-       winbindd_idmap_status();
        winbindd_cm_status();
 }
 
@@ -142,8 +136,17 @@ static void print_winbindd_status(void)
 
 static void flush_caches(void)
 {
+#if 0
        /* Clear cached user and group enumation info */        
-       wcache_flush_cache();
+       if (!opt_dual_daemon) /* Until we have coherent cache flush. */
+               wcache_flush_cache();
+#endif
+
+       /* We need to invalidate cached user list entries on a SIGHUP 
+           otherwise cached access denied errors due to restrict anonymous
+           hang around until the sequence number changes. */
+
+       wcache_invalidate_cache();
 }
 
 /* Handle the signal by unlinking socket and exiting */
@@ -152,10 +155,10 @@ static void terminate(void)
 {
        pstring path;
 
-       winbindd_idmap_close();
+       idmap_close();
        
        /* Remove socket file */
-       snprintf(path, sizeof(path), "%s/%s", 
+       pstr_sprintf(path, "%s/%s", 
                 WINBINDD_SOCKET_DIR, WINBINDD_SOCKET_NAME);
        unlink(path);
        exit(0);
@@ -185,10 +188,24 @@ static void sighup_handler(int signum)
        sys_select_signal();
 }
 
+/* React on 'smbcontrol winbindd reload-config' in the same way as on SIGHUP*/
+static void msg_reload_services(int msg_type, pid_t src, void *buf, size_t len)
+{
+        /* Flush various caches */
+       flush_caches();
+       reload_services_file(True);
+}
+
+/* React on 'smbcontrol winbindd shutdown' in the same way as on SIGTERM*/
+static void msg_shutdown(int msg_type, pid_t src, void *buf, size_t len)
+{
+       terminate();
+}
+
 struct dispatch_table {
        enum winbindd_cmd cmd;
        enum winbindd_result (*fn)(struct winbindd_cli_state *state);
-       char *winbindd_cmd_name;
+       const char *winbindd_cmd_name;
 };
 
 static struct dispatch_table dispatch_table[] = {
@@ -245,12 +262,23 @@ static struct dispatch_table dispatch_table[] = {
        { WINBINDD_INFO, winbindd_info, "INFO" },
        { WINBINDD_INTERFACE_VERSION, winbindd_interface_version, "INTERFACE_VERSION" },
        { WINBINDD_DOMAIN_NAME, winbindd_domain_name, "DOMAIN_NAME" },
+       { WINBINDD_NETBIOS_NAME, winbindd_netbios_name, "NETBIOS_NAME" },
+       { WINBINDD_PRIV_PIPE_DIR, winbindd_priv_pipe_dir, "WINBINDD_PRIV_PIPE_DIR" },
 
        /* WINS functions */
 
        { WINBINDD_WINS_BYNAME, winbindd_wins_byname, "WINS_BYNAME" },
        { WINBINDD_WINS_BYIP, winbindd_wins_byip, "WINS_BYIP" },
-
+       
+       /* UNIX account management functions */
+       { WINBINDD_CREATE_USER,                 winbindd_create_user,           "CREATE_USER"           },
+       { WINBINDD_CREATE_GROUP,                winbindd_create_group,          "CREATE_GROUP"          },
+       { WINBINDD_ADD_USER_TO_GROUP,           winbindd_add_user_to_group,     "ADD_USER_TO_GROUP"     },
+       { WINBINDD_REMOVE_USER_FROM_GROUP,      winbindd_remove_user_from_group,"REMOVE_USER_FROM_GROUP"},
+       { WINBINDD_SET_USER_PRIMARY_GROUP,      winbindd_set_user_primary_group,"SET_USER_PRIMARY_GROUP"},
+       { WINBINDD_DELETE_USER,                 winbindd_delete_user,           "DELETE_USER"           },
+       { WINBINDD_DELETE_GROUP,                winbindd_delete_group,          "DELETE_GROUP"          },
+       
        /* End of list */
 
        { WINBINDD_NUM_CMDS, NULL, "NONE" }
@@ -291,7 +319,7 @@ static void process_request(struct winbindd_cli_state *state)
 
 /* Process a new connection by adding it to the client connection list */
 
-static void new_connection(int listen_sock)
+static void new_connection(int listen_sock, BOOL privileged)
 {
        struct sockaddr_un sunaddr;
        struct winbindd_cli_state *state;
@@ -319,7 +347,11 @@ static void new_connection(int listen_sock)
        
        ZERO_STRUCTP(state);
        state->sock = sock;
-       
+
+       state->last_access = time(NULL);        
+
+       state->privileged = privileged;
+
        /* Add to connection list */
        
        winbindd_add_client(state);
@@ -355,6 +387,35 @@ static void remove_client(struct winbindd_cli_state *state)
 }
 
 
+/* Shutdown client connection which has been idle for the longest time */
+
+static BOOL remove_idle_client(void)
+{
+       struct winbindd_cli_state *state, *remove_state = NULL;
+       time_t last_access = 0;
+       int nidle = 0;
+
+       for (state = winbindd_client_list(); state; state = state->next) {
+               if (state->read_buf_len == 0 && state->write_buf_len == 0 &&
+                               !state->getpwent_state && !state->getgrent_state) {
+                       nidle++;
+                       if (!last_access || state->last_access < last_access) {
+                               last_access = state->last_access;
+                               remove_state = state;
+                       }
+               }
+       }
+
+       if (remove_state) {
+               DEBUG(5,("Found %d idle client connections, shutting down sock %d, pid %u\n",
+                       nidle, remove_state->sock, (unsigned int)remove_state->pid));
+               remove_client(remove_state);
+               return True;
+       }
+
+       return False;
+}
+
 /* Process a complete received packet from a client */
 
 void winbind_process_packet(struct winbindd_cli_state *state)
@@ -362,7 +423,7 @@ void winbind_process_packet(struct winbindd_cli_state *state)
        /* Process request */
        
        /* Ensure null termination of entire request */
-       state->request.domain[sizeof(state->request.domain)-1]='\0';
+       state->request.null_term = '\0';
 
        state->pid = state->request.pid;
        
@@ -396,8 +457,8 @@ void winbind_client_read(struct winbindd_cli_state *state)
        /* Read failed, kill client */
        
        if (n == -1 || n == 0) {
-               DEBUG(5,("read failed on sock %d, pid %d: %s\n",
-                        state->sock, state->pid, 
+               DEBUG(5,("read failed on sock %d, pid %lu: %s\n",
+                        state->sock, (unsigned long)state->pid, 
                         (n == -1) ? strerror(errno) : "EOF"));
                
                state->finished = True;
@@ -407,6 +468,7 @@ void winbind_client_read(struct winbindd_cli_state *state)
        /* Update client state */
        
        state->read_buf_len += n;
+       state->last_access = time(NULL);
 }
 
 /* Write some data to a client connection */
@@ -417,6 +479,13 @@ static void client_write(struct winbindd_cli_state *state)
        int num_written;
        
        /* Write some data */
+       /*
+        * The fancy calculation of data below allows us to handle the 
+        * case where write (sys_write) does not write all the data we 
+        * gave it. In that case, we will come back through here again
+        * because of the loop above us, and we want to pick up where
+        * we left off.
+        */
        
        if (!state->write_extra_data) {
 
@@ -443,8 +512,8 @@ static void client_write(struct winbindd_cli_state *state)
        
        if (num_written == -1 || num_written == 0) {
                
-               DEBUG(3,("write failed on sock %d, pid %d: %s\n",
-                        state->sock, state->pid, 
+               DEBUG(3,("write failed on sock %d, pid %lu: %s\n",
+                        state->sock, (unsigned long)state->pid, 
                         (num_written == -1) ? strerror(errno) : "EOF"));
                
                state->finished = True;
@@ -457,7 +526,8 @@ static void client_write(struct winbindd_cli_state *state)
        /* Update client state */
        
        state->write_buf_len -= num_written;
-       
+       state->last_access = time(NULL);
+
        /* Have we written all data? */
        
        if (state->write_buf_len == 0) {
@@ -488,7 +558,7 @@ static void client_write(struct winbindd_cli_state *state)
        }
 }
 
-/* Process incoming clients on accept_sock.  We use a tricky non-blocking,
+/* Process incoming clients on listen_sock.  We use a tricky non-blocking,
    non-forking, non-threaded model which allows us to handle many
    simultaneous connections while remaining impervious to many denial of
    service attacks. */
@@ -500,15 +570,15 @@ static void process_loop(void)
        while (1) {
                struct winbindd_cli_state *state;
                fd_set r_fds, w_fds;
-               int maxfd, listen_sock, selret;
+               int maxfd, listen_sock, listen_priv_sock, selret;
                struct timeval timeout;
 
                /* Handle messages */
 
                message_dispatch();
 
-               /* rescan the trusted domains list. This must be done
-                  regularly to cope with transitive trusts */
+               /* refresh the trusted domain cache */
+                  
                rescan_trusted_domains();
 
                /* Free up temporary memory */
@@ -519,11 +589,19 @@ static void process_loop(void)
                /* Initialise fd lists for select() */
 
                listen_sock = open_winbindd_socket();
-               maxfd = listen_sock;
+               listen_priv_sock = open_winbindd_priv_socket();
+
+               if (listen_sock == -1 || listen_priv_sock == -1) {
+                       perror("open_winbind_socket");
+                       exit(1);
+               }
+
+               maxfd = MAX(listen_sock, listen_priv_sock);
 
                FD_ZERO(&r_fds);
                FD_ZERO(&w_fds);
                FD_SET(listen_sock, &r_fds);
+               FD_SET(listen_priv_sock, &r_fds);
 
                timeout.tv_sec = WINBINDD_ESTABLISH_LOOP;
                timeout.tv_usec = 0;
@@ -582,7 +660,7 @@ static void process_loop(void)
                        exit(1);
                }
 
-               /* Create a new connection if accept_sock readable */
+               /* Create a new connection if listen_sock readable */
 
                if (selret > 0) {
 
@@ -590,8 +668,33 @@ static void process_loop(void)
                                dual_select(&w_fds);
                        }
 
-                       if (FD_ISSET(listen_sock, &r_fds))
-                               new_connection(listen_sock);
+                       if (FD_ISSET(listen_sock, &r_fds)) {
+                               while (winbindd_num_clients() > WINBINDD_MAX_SIMULTANEOUS_CLIENTS - 1) {
+                                       DEBUG(5,("winbindd: Exceeding %d client connections, removing idle connection.\n",
+                                               WINBINDD_MAX_SIMULTANEOUS_CLIENTS));
+                                       if (!remove_idle_client()) {
+                                               DEBUG(0,("winbindd: Exceeding %d client connections, no idle connection found\n",
+                                                       WINBINDD_MAX_SIMULTANEOUS_CLIENTS));
+                                               break;
+                                       }
+                               }
+                               /* new, non-privileged connection */
+                               new_connection(listen_sock, False);
+                       }
+            
+                       if (FD_ISSET(listen_priv_sock, &r_fds)) {
+                               while (winbindd_num_clients() > WINBINDD_MAX_SIMULTANEOUS_CLIENTS - 1) {
+                                       DEBUG(5,("winbindd: Exceeding %d client connections, removing idle connection.\n",
+                                               WINBINDD_MAX_SIMULTANEOUS_CLIENTS));
+                                       if (!remove_idle_client()) {
+                                               DEBUG(0,("winbindd: Exceeding %d client connections, no idle connection found\n",
+                                                       WINBINDD_MAX_SIMULTANEOUS_CLIENTS));
+                                               break;
+                                       }
+                               }
+                               /* new, privileged connection */
+                               new_connection(listen_priv_sock, True);
+                       }
             
                        /* Process activity on client connections */
             
@@ -616,8 +719,8 @@ static void process_loop(void)
 
                                        if (state->read_buf_len >= sizeof(uint32)
                                            && *(uint32 *) &state->request != sizeof(state->request)) {
-                                               DEBUG(0,("process_loop: Invalid request size from pid %d: %d bytes sent, should be %d\n",
-                                                               state->request.pid, *(uint32 *) &state->request, sizeof(state->request)));
+                                               DEBUG(0,("process_loop: Invalid request size from pid %lu: %d bytes sent, should be %d\n",
+                                                               (unsigned long)state->request.pid, *(uint32 *) &state->request, sizeof(state->request)));
 
                                                remove_client(state);
                                                break;
@@ -651,11 +754,8 @@ static void process_loop(void)
                if (do_sighup) {
 
                        DEBUG(3, ("got SIGHUP\n"));
-                        /* Flush various caches */
 
-                       flush_caches();
-                       reload_services_file(True);
+                       msg_reload_services(MSG_SMB_CONF_UPDATED, (pid_t) 0, NULL, 0);
                        do_sighup = False;
                }
 
@@ -666,86 +766,27 @@ static void process_loop(void)
        }
 }
 
-
-/*
-  these are split out from the main winbindd for use by the background daemon
- */
-BOOL winbind_setup_common(void)
-{
-       load_interfaces();
-
-       if (!secrets_init()) {
-
-               DEBUG(0,("Could not initialize domain trust account secrets. Giving up\n"));
-               return False;
-       }
-
-       namecache_enable();     /* Enable netbios namecache */
-
-       /* Check winbindd parameters are valid */
-
-       ZERO_STRUCT(server_state);
-
-       if (!winbindd_param_init())
-               return False;
-
-       /* Get list of domains we look up requests for.  This includes the
-          domain which we are a member of as well as any trusted
-          domains. */ 
-
-       init_domain_list();
-
-       /* Winbind daemon initialisation */
-
-       if (!winbindd_idmap_init())
-               return False;
-
-       /* Unblock all signals we are interested in as they may have been
-          blocked by the parent process. */
-
-       BlockSignals(False, SIGINT);
-       BlockSignals(False, SIGQUIT);
-       BlockSignals(False, SIGTERM);
-       BlockSignals(False, SIGUSR1);
-       BlockSignals(False, SIGUSR2);
-       BlockSignals(False, SIGHUP);
-
-       /* Setup signal handlers */
-       
-       CatchSignal(SIGINT, termination_handler);      /* Exit on these sigs */
-       CatchSignal(SIGQUIT, termination_handler);
-       CatchSignal(SIGTERM, termination_handler);
-
-       CatchSignal(SIGPIPE, SIG_IGN);                 /* Ignore sigpipe */
-
-       CatchSignal(SIGUSR2, sigusr2_handler);         /* Debugging sigs */
-       CatchSignal(SIGHUP, sighup_handler);
-
-       return True;
-}
-
-
 /* Main function */
 
 struct winbindd_state server_state;   /* Server state information */
 
-
-static void usage(void)
-{
-       printf("Usage: winbindd [options]\n");
-       printf("\t-i                interactive mode\n");
-       printf("\t-B                dual daemon mode\n");
-       printf("\t-n                disable cacheing\n");
-       printf("\t-d level          set debug level\n");
-       printf("\t-s configfile     choose smb.conf location\n");
-       printf("\t-h                show this help message\n");
-}
-
- int main(int argc, char **argv)
+int main(int argc, char **argv)
 {
-       extern BOOL AllowDebugChange;
        pstring logfile;
-       BOOL interactive = False;
+       static BOOL interactive = False;
+       static BOOL Fork = True;
+       static BOOL log_stdout = False;
+       struct poptOption long_options[] = {
+               POPT_AUTOHELP
+               { "stdout", 'S', POPT_ARG_VAL, &log_stdout, True, "Log to stdout" },
+               { "foreground", 'F', POPT_ARG_VAL, &Fork, False, "Daemon in foreground mode" },
+               { "interactive", 'i', POPT_ARG_NONE, NULL, 'i', "Interactive mode" },
+               { "single-daemon", 'Y', POPT_ARG_VAL, &opt_dual_daemon, False, "Single daemon mode" },
+               { "no-caching", 'n', POPT_ARG_VAL, &opt_nocache, False, "Disable caching" },
+               POPT_COMMON_SAMBA
+               POPT_TABLEEND
+       };
+       poptContext pc;
        int opt;
 
        /* glibc (?) likes to print "User defined signal 1" and exit if a
@@ -756,66 +797,47 @@ static void usage(void)
 
        fault_setup((void (*)(void *))fault_quit );
 
-       snprintf(logfile, sizeof(logfile), "%s/log.winbindd", dyn_LOGFILEBASE);
-       lp_set_logfile(logfile);
-
        /* Initialise for running in non-root mode */
 
        sec_init();
 
+       set_remote_machine_name("winbindd", False);
+
        /* Set environment variable so we don't recursively call ourselves.
           This may also be useful interactively. */
 
-       SETENV(WINBINDD_DONT_ENV, "1", 1);
+       setenv(WINBINDD_DONT_ENV, "1", 1);
 
        /* Initialise samba/rpc client stuff */
 
-       while ((opt = getopt(argc, argv, "id:s:nhB")) != EOF) {
-               switch (opt) {
+       pc = poptGetContext("winbindd", argc, (const char **)argv, long_options,
+                                               POPT_CONTEXT_KEEP_FIRST);
 
+       while ((opt = poptGetNextOpt(pc)) != -1) {
+               switch (opt) {
                        /* Don't become a daemon */
                case 'i':
                        interactive = True;
+                       log_stdout = True;
+                       Fork = False;
                        break;
+               }
+       }
 
-                       /* dual daemon system */
-               case 'B':
-                       opt_dual_daemon = True;
-                       break;
-
-                       /* disable cacheing */
-               case 'n':
-                       opt_nocache = True;
-                       break;
-
-                       /* Run with specified debug level */
-               case 'd':
-                       DEBUGLEVEL = atoi(optarg);
-                       AllowDebugChange = False;
-                       break;
-
-                       /* Load a different smb.conf file */
-               case 's':
-                       pstrcpy(dyn_CONFIGFILE,optarg);
-                       break;
-
-               case 'h':
-                       usage();
-                       exit(0);
 
-               default:
-                       printf("Unknown option %c\n", (char)opt);
-                       exit(1);
-               }
+       if (log_stdout && Fork) {
+               printf("Can't log to stdout (-S) unless daemon is in foreground +(-F) or interactive (-i)\n");
+               poptPrintUsage(pc, stderr, 0);
+               exit(1);
        }
 
-       snprintf(logfile, sizeof(logfile), "%s/log.winbindd", dyn_LOGFILEBASE);
+       pstr_sprintf(logfile, "%s/log.winbindd", dyn_LOGFILEBASE);
        lp_set_logfile(logfile);
-       setup_logging("winbindd", interactive);
+       setup_logging("winbindd", log_stdout);
        reopen_logs();
 
-       DEBUG(1, ("winbindd version %s started.\n", VERSION ) );
-       DEBUGADD( 1, ( "Copyright The Samba Team 2000-2001\n" ) );
+       DEBUG(1, ("winbindd version %s started.\n", SAMBA_VERSION_STRING) );
+       DEBUGADD( 1, ( "Copyright The Samba Team 2000-2003\n" ) );
 
        if (!reload_services_file(False)) {
                DEBUG(0, ("error opening config file\n"));
@@ -827,11 +849,58 @@ static void usage(void)
        if (!init_names())
                exit(1);
 
-       if (!interactive) {
-               become_daemon();
-               pidfile_create("winbindd");
+       load_interfaces();
+
+       if (!secrets_init()) {
+
+               DEBUG(0,("Could not initialize domain trust account secrets. Giving up\n"));
+               return False;
        }
 
+       /* Enable netbios namecache */
+
+       namecache_enable();
+
+       /* Check winbindd parameters are valid */
+
+       ZERO_STRUCT(server_state);
+
+       if (!winbindd_param_init())
+               return 1;
+
+       /* Winbind daemon initialisation */
+
+       if (!winbindd_upgrade_idmap())
+               return 1;
+
+       if (!idmap_init(lp_idmap_backend()))
+               return 1;
+
+       /* Unblock all signals we are interested in as they may have been
+          blocked by the parent process. */
+
+       BlockSignals(False, SIGINT);
+       BlockSignals(False, SIGQUIT);
+       BlockSignals(False, SIGTERM);
+       BlockSignals(False, SIGUSR1);
+       BlockSignals(False, SIGUSR2);
+       BlockSignals(False, SIGHUP);
+
+       /* Setup signal handlers */
+       
+       CatchSignal(SIGINT, termination_handler);      /* Exit on these sigs */
+       CatchSignal(SIGQUIT, termination_handler);
+       CatchSignal(SIGTERM, termination_handler);
+
+       CatchSignal(SIGPIPE, SIG_IGN);                 /* Ignore sigpipe */
+
+       CatchSignal(SIGUSR2, sigusr2_handler);         /* Debugging sigs */
+       CatchSignal(SIGHUP, sighup_handler);
+
+       if (!interactive)
+               become_daemon(Fork);
+
+       pidfile_create("winbindd");
 
 #if HAVE_SETPGID
        /*
@@ -842,10 +911,6 @@ static void usage(void)
                setpgid( (pid_t)0, (pid_t)0);
 #endif
 
-       if (!winbind_setup_common()) {
-               return 1;
-       }
-
        if (opt_dual_daemon) {
                do_dual_daemon();
        }
@@ -856,13 +921,21 @@ static void usage(void)
                DEBUG(0, ("unable to initialise messaging system\n"));
                exit(1);
        }
+       
+       /* React on 'smbcontrol winbindd reload-config' in the same way
+          as to SIGHUP signal */
+       message_register(MSG_SMB_CONF_UPDATED, msg_reload_services);
+       message_register(MSG_SHUTDOWN, msg_shutdown);
+       
+       poptFreeContext(pc);
 
-       register_msg_pool_usage();
-
+       netsamlogon_cache_init(); /* Non-critical */
+       
        /* Loop waiting for requests */
 
        process_loop();
 
-       uni_group_cache_shutdown();
+       trustdom_cache_shutdown();
+
        return 0;
 }