[GLUE] Rsync SAMBA_3_2_0 SVN r25598 in order to create the v3-2-test branch.
[samba.git] / source / nmbd / nmbd.c
index 8c94ad842b13f8dc166e370c695ae49fce482a0e..dc03506194b278f422ab59ac2133675e8fec4149 100644 (file)
@@ -7,7 +7,7 @@
    
    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
-   the Free Software Foundation; either version 2 of the License, or
+   the Free Software Foundation; either version 3 of the License, or
    (at your option) any later version.
    
    This program is distributed in the hope that it will be useful,
@@ -16,8 +16,7 @@
    GNU General Public License for more details.
    
    You should have received a copy of the GNU General Public License
-   along with this program; if not, write to the Free Software
-   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+   along with this program.  If not, see <http://www.gnu.org/licenses/>.
    
 */
 
@@ -33,6 +32,15 @@ extern BOOL global_in_nmbd;
 
 extern BOOL override_logfile;
 
+/* are we running as a daemon ? */
+static BOOL is_daemon;
+
+/* fork or run in foreground ? */
+static BOOL Fork = True;
+
+/* log to standard output ? */
+static BOOL log_stdout;
+
 /* have we found LanMan clients yet? */
 BOOL found_lm_clients = False;
 
@@ -45,7 +53,7 @@ struct event_context *nmbd_event_context(void)
        static struct event_context *ctx;
 
        if (!ctx && !(ctx = event_context_init(NULL))) {
-               smb_panic("Could not init nmbd event context\n");
+               smb_panic("Could not init nmbd event context");
        }
        return ctx;
 }
@@ -56,7 +64,7 @@ struct messaging_context *nmbd_messaging_context(void)
 
        if (!ctx && !(ctx = messaging_init(NULL, server_id_self(),
                                           nmbd_event_context()))) {
-               smb_panic("Could not init nmbd messaging context\n");
+               smb_panic("Could not init nmbd messaging context");
        }
        return ctx;
 }
@@ -373,6 +381,7 @@ static void process(void)
 
        while( True ) {
                time_t t = time(NULL);
+               TALLOC_CTX *frame = talloc_stackframe();
 
                /* Check for internal messages */
 
@@ -391,8 +400,10 @@ static void process(void)
                 * (nmbd_packets.c)
                 */
 
-               if(listen_for_packets(run_election))
+               if(listen_for_packets(run_election)) {
+                       TALLOC_FREE(frame);
                        return;
+               }
 
                /*
                 * Handle termination inband.
@@ -584,18 +595,22 @@ static void process(void)
                                                 NULL, MSG_SMB_CONF_UPDATED,
                                                 procid_self(), &blob);
 
-                       if(no_subnets)
+                       if(no_subnets) {
+                               TALLOC_FREE(frame);
                                return;
+                       }
                        reload_after_sighup = 0;
                }
 
                /* check for new network interfaces */
 
-               if(reload_interfaces(t))
+               if(reload_interfaces(t)) {
+                       TALLOC_FREE(frame);
                        return;
+               }
 
                /* free up temp memory */
-                       lp_TALLOC_FREE();
+               TALLOC_FREE(frame);
        }
 }
 
@@ -603,7 +618,7 @@ static void process(void)
  Open the socket communication.
  **************************************************************************** */
 
-static BOOL open_sockets(enum smb_server_mode server_mode, int port)
+static BOOL open_sockets(BOOL isdaemon, int port)
 {
        /*
         * The sockets opened here will be used to receive broadcast
@@ -613,13 +628,12 @@ static BOOL open_sockets(enum smb_server_mode server_mode, int port)
         * now deprecated.
         */
 
-       if ( server_mode == SERVER_MODE_INETD ) {
-               ClientNMB = 0;
-       } else {
+       if ( isdaemon )
                ClientNMB = open_socket_in(SOCK_DGRAM, port,
                                           0, interpret_addr(lp_socket_address()),
                                           True);
-       }
+       else
+               ClientNMB = 0;
   
        ClientDGRAM = open_socket_in(SOCK_DGRAM, DGRAM_PORT,
                                           3, interpret_addr(lp_socket_address()),
@@ -648,20 +662,16 @@ static BOOL open_sockets(enum smb_server_mode server_mode, int port)
  int main(int argc, const char *argv[])
 {
        pstring logfile;
+       static BOOL opt_interactive;
        poptContext pc;
-       const char *p_lmhosts = dyn_LMHOSTSFILE;
-       BOOL no_process_group = False;
-       BOOL log_stdout = False;
-       enum smb_server_mode server_mode = SERVER_MODE_DAEMON;
-
+       static char *p_lmhosts = dyn_LMHOSTSFILE;
+       static BOOL no_process_group = False;
+       int opt;
        struct poptOption long_options[] = {
        POPT_AUTOHELP
-       {"daemon", 'D', POPT_ARG_VAL, &server_mode, SERVER_MODE_DAEMON,
-               "Become a daemon(default)" },
-       {"interactive", 'i', POPT_ARG_VAL, &server_mode,
-               SERVER_MODE_INTERACTIVE, "Run interactive (not a daemon)" },
-       {"foreground", 'F', POPT_ARG_VAL, &server_mode,
-               SERVER_MODE_FOREGROUND, "Run daemon in foreground (for daemontools & etc)" },
+       {"daemon", 'D', POPT_ARG_VAL, &is_daemon, True, "Become a daemon(default)" },
+       {"interactive", 'i', POPT_ARG_VAL, &opt_interactive, True, "Run interactive (not a daemon)" },
+       {"foreground", 'F', POPT_ARG_VAL, &Fork, False, "Run daemon in foreground (for daemontools & etc)" },
        {"no-process-group", 0, POPT_ARG_VAL, &no_process_group, True, "Don't create a new process group" },
        {"log-stdout", 'S', POPT_ARG_VAL, &log_stdout, True, "Log to stdout" },
        {"hosts", 'H', POPT_ARG_STRING, &p_lmhosts, 'H', "Load a netbios hosts file"},
@@ -675,7 +685,15 @@ static BOOL open_sockets(enum smb_server_mode server_mode, int port)
        global_nmb_port = NMB_PORT;
 
        pc = poptGetContext("nmbd", argc, argv, long_options, 0);
-       while (poptGetNextOpt(pc) != -1) {};
+       while ((opt = poptGetNextOpt(pc)) != -1) {
+               switch (opt) {
+               default:
+                       d_fprintf(stderr, "\nInvalid option %s: %s\n\n",
+                                 poptBadOption(pc, 0), poptStrerror(opt));
+                       poptPrintUsage(pc, stderr, 0);
+                       exit(1);
+               }
+       };
        poptFreeContext(pc);
 
        global_in_nmbd = True;
@@ -711,11 +729,12 @@ static BOOL open_sockets(enum smb_server_mode server_mode, int port)
        BlockSignals(True, SIGUSR2);
 #endif
 
-       if (server_mode == SERVER_MODE_INTERACTIVE) {
+       if ( opt_interactive ) {
+               Fork = False;
                log_stdout = True;
        }
 
-       if (log_stdout && server_mode == SERVER_MODE_DAEMON) {
+       if ( log_stdout && Fork ) {
                DEBUG(0,("ERROR: Can't log to stdout (-S) unless daemon is in foreground (-F) or interactive (-i)\n"));
                exit(1);
        }
@@ -742,19 +761,14 @@ static BOOL open_sockets(enum smb_server_mode server_mode, int port)
 
        set_samba_nb_type();
 
-       if (is_a_socket(0)) {
-               if (server_mode == SERVER_MODE_DAEMON) {
-                       DEBUG(0,("standard input is a socket, "
-                                   "assuming -F option\n"));
-               }
-               server_mode = SERVER_MODE_INETD;
+       if (!is_daemon && !is_a_socket(0)) {
+               DEBUG(0,("standard input is not a socket, assuming -D option\n"));
+               is_daemon = True;
        }
-
-       if (server_mode == SERVER_MODE_DAEMON) {
+  
+       if (is_daemon && !opt_interactive) {
                DEBUG( 2, ( "Becoming a daemon.\n" ) );
-               become_daemon(True, no_process_group);
-       } else if (server_mode == SERVER_MODE_FOREGROUND) {
-               become_daemon(False, no_process_group);
+               become_daemon(Fork, no_process_group);
        }
 
 #if HAVE_SETPGID
@@ -762,7 +776,7 @@ static BOOL open_sockets(enum smb_server_mode server_mode, int port)
         * If we're interactive we want to set our own process group for 
         * signal management.
         */
-       if (server_mode == SERVER_MODE_INTERACTIVE && !no_process_group)
+       if (opt_interactive && !no_process_group)
                setpgid( (pid_t)0, (pid_t)0 );
 #endif
 
@@ -801,7 +815,7 @@ static BOOL open_sockets(enum smb_server_mode server_mode, int port)
 
        DEBUG( 3, ( "Opening sockets %d\n", global_nmb_port ) );
 
-       if ( !open_sockets( server_mode, global_nmb_port ) ) {
+       if ( !open_sockets( is_daemon, global_nmb_port ) ) {
                kill_async_dns_child();
                return 1;
        }