r23784: use the GPLv3 boilerplate as recommended by the FSF and the license text
[tprouty/samba.git] / source / nmbd / nmbd.c
index 38183159a2b9d0ffffe7050147a4892117019a6d..b14e5a0b612b275e91614c15e701aba338b29fde 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,15 +32,6 @@ 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;
 
@@ -49,6 +39,27 @@ BOOL found_lm_clients = False;
 
 time_t StartupTime = 0;
 
+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");
+       }
+       return ctx;
+}
+
+struct messaging_context *nmbd_messaging_context(void)
+{
+       static struct messaging_context *ctx;
+
+       if (!ctx && !(ctx = messaging_init(NULL, server_id_self(),
+                                          nmbd_event_context()))) {
+               smb_panic("Could not init nmbd messaging context");
+       }
+       return ctx;
+}
+
 /**************************************************************************** **
  Handle a SIGTERM in band.
  **************************************************************************** */
@@ -76,8 +87,11 @@ static void terminate(void)
  Handle a SHUTDOWN message from smbcontrol.
  **************************************************************************** */
 
-static void nmbd_terminate(int msg_type, struct process_id src,
-                          void *buf, size_t len)
+static void nmbd_terminate(struct messaging_context *msg,
+                          void *private_data,
+                          uint32_t msg_type,
+                          struct server_id server_id,
+                          DATA_BLOB *data)
 {
        terminate();
 }
@@ -106,55 +120,13 @@ static void sig_hup(int sig)
        sys_select_signal(SIGHUP);
 }
 
-#if DUMP_CORE
-/**************************************************************************** **
- Prepare to dump a core file - carefully!
- **************************************************************************** */
-
-static BOOL dump_core(void)
-{
-       char *p;
-       pstring dname;
-       pstrcpy( dname, lp_logfile() );
-       if ((p=strrchr_m(dname,'/')))
-               *p=0;
-       pstrcat( dname, "/corefiles" );
-       mkdir( dname, 0700 );
-       sys_chown( dname, getuid(), getgid() );
-       chmod( dname, 0700 );
-       if ( chdir(dname) )
-               return( False );
-       umask( ~(0700) );
-
-#ifdef HAVE_GETRLIMIT
-#ifdef RLIMIT_CORE
-       {
-               struct rlimit rlp;
-               getrlimit( RLIMIT_CORE, &rlp );
-               rlp.rlim_cur = MAX( 4*1024*1024, rlp.rlim_cur );
-               setrlimit( RLIMIT_CORE, &rlp );
-               getrlimit( RLIMIT_CORE, &rlp );
-               DEBUG( 3, ( "Core limits now %d %d\n", (int)rlp.rlim_cur, (int)rlp.rlim_max ) );
-       }
-#endif
-#endif
-
-
-       DEBUG(0,("Dumping core in %s\n",dname));
-       abort();
-       return( True );
-}
-#endif
-
 /**************************************************************************** **
  Possibly continue after a fault.
  **************************************************************************** */
 
 static void fault_continue(void)
 {
-#if DUMP_CORE
        dump_core();
-#endif
 }
 
 /**************************************************************************** **
@@ -213,6 +185,11 @@ static BOOL reload_interfaces(time_t t)
        for (n=iface_count() - 1; n >= 0; n--) {
                struct interface *iface = get_interface(n);
 
+               if (!iface) {
+                       DEBUG(2,("reload_interfaces: failed to get interface %d\n", n));
+                       continue;
+               }
+
                /*
                 * We don't want to add a loopback interface, in case
                 * someone has added 127.0.0.1 for smbd, nmbd needs to
@@ -308,33 +285,39 @@ static BOOL reload_nmbd_services(BOOL test)
  * detects that there are no subnets.
  **************************************************************************** */
 
-static void msg_reload_nmbd_services(int msg_type, struct process_id src,
-                                    void *buf, size_t len)
+static void msg_reload_nmbd_services(struct messaging_context *msg,
+                                    void *private_data,
+                                    uint32_t msg_type,
+                                    struct server_id server_id,
+                                    DATA_BLOB *data)
 {
        write_browse_list( 0, True );
        dump_all_namelists();
        reload_nmbd_services( True );
        reopen_logs();
        
-       if(buf) {
+       if (data->data) {
                /* We were called from process() */
                /* If reload_interfaces() returned True */
                /* we need to shutdown if there are no subnets... */
                /* pass this info back to process() */
-               *((BOOL*)buf) = reload_interfaces(0);  
+               *((BOOL*)data->data) = reload_interfaces(0);  
        }
 }
 
-static void msg_nmbd_send_packet(int msg_type, struct process_id src,
-                                void *buf, size_t len)
+static void msg_nmbd_send_packet(struct messaging_context *msg,
+                                void *private_data,
+                                uint32_t msg_type,
+                                struct server_id src,
+                                DATA_BLOB *data)
 {
-       struct packet_struct *p = (struct packet_struct *)buf;
+       struct packet_struct *p = (struct packet_struct *)data->data;
        struct subnet_record *subrec;
        struct in_addr *local_ip;
 
        DEBUG(10, ("Received send_packet from %d\n", procid_to_pid(&src)));
 
-       if (len != sizeof(struct packet_struct)) {
+       if (data->length != sizeof(struct packet_struct)) {
                DEBUG(2, ("Discarding invalid packet length from %d\n",
                          procid_to_pid(&src)));
                return;
@@ -392,7 +375,7 @@ static void process(void)
 
                /* Check for internal messages */
 
-               message_dispatch();
+               message_dispatch(nmbd_messaging_context());
 
                /*
                 * Check all broadcast subnets to see if
@@ -593,9 +576,13 @@ static void process(void)
                 */
 
                if(reload_after_sighup) {
+                       DATA_BLOB blob = data_blob_const(&no_subnets,
+                                                        sizeof(no_subnets));
                        DEBUG( 0, ( "Got SIGHUP dumping debug info.\n" ) );
-                       msg_reload_nmbd_services(MSG_SMB_CONF_UPDATED,
-                                                pid_to_procid(0), (void*) &no_subnets, 0);
+                       msg_reload_nmbd_services(nmbd_messaging_context(),
+                                                NULL, MSG_SMB_CONF_UPDATED,
+                                                procid_self(), &blob);
+
                        if(no_subnets)
                                return;
                        reload_after_sighup = 0;
@@ -615,7 +602,7 @@ static void process(void)
  Open the socket communication.
  **************************************************************************** */
 
-static BOOL open_sockets(BOOL isdaemon, int port)
+static BOOL open_sockets(enum smb_server_mode server_mode, int port)
 {
        /*
         * The sockets opened here will be used to receive broadcast
@@ -625,12 +612,13 @@ static BOOL open_sockets(BOOL isdaemon, int port)
         * now deprecated.
         */
 
-       if ( isdaemon )
+       if ( server_mode == SERVER_MODE_INETD ) {
+               ClientNMB = 0;
+       } else {
                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()),
@@ -645,6 +633,10 @@ static BOOL open_sockets(BOOL isdaemon, int port)
        set_socket_options( ClientNMB,   "SO_BROADCAST" );
        set_socket_options( ClientDGRAM, "SO_BROADCAST" );
 
+       /* Ensure we're non-blocking. */
+       set_blocking( ClientNMB, False);
+       set_blocking( ClientDGRAM, False);
+
        DEBUG( 3, ( "open_sockets: Broadcast sockets opened.\n" ) );
        return( True );
 }
@@ -655,14 +647,21 @@ static BOOL open_sockets(BOOL isdaemon, int port)
  int main(int argc, const char *argv[])
 {
        pstring logfile;
-       static BOOL opt_interactive;
        poptContext pc;
-       static char *p_lmhosts = dyn_LMHOSTSFILE;
+       const char *p_lmhosts = dyn_LMHOSTSFILE;
+       BOOL no_process_group = False;
+       BOOL log_stdout = False;
+       enum smb_server_mode server_mode = SERVER_MODE_DAEMON;
+
        struct poptOption long_options[] = {
        POPT_AUTOHELP
-       {"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)" },
+       {"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)" },
+       {"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"},
        {"port", 'p', POPT_ARG_INT, &global_nmb_port, NMB_PORT, "Listen on the specified port" },
@@ -690,6 +689,7 @@ static BOOL open_sockets(BOOL isdaemon, int port)
        }
        
        fault_setup((void (*)(void *))fault_continue );
+       dump_core_setup("nmbd");
        
        /* POSIX demands that signals are inherited. If the invoking process has
         * these signals masked, we will have problems, as we won't receive them. */
@@ -710,12 +710,11 @@ static BOOL open_sockets(BOOL isdaemon, int port)
        BlockSignals(True, SIGUSR2);
 #endif
 
-       if ( opt_interactive ) {
-               Fork = False;
+       if (server_mode == SERVER_MODE_INTERACTIVE) {
                log_stdout = True;
        }
 
-       if ( log_stdout && Fork ) {
+       if (log_stdout && server_mode == SERVER_MODE_DAEMON) {
                DEBUG(0,("ERROR: Can't log to stdout (-S) unless daemon is in foreground (-F) or interactive (-i)\n"));
                exit(1);
        }
@@ -742,14 +741,19 @@ static BOOL open_sockets(BOOL isdaemon, int port)
 
        set_samba_nb_type();
 
-       if (!is_daemon && !is_a_socket(0)) {
-               DEBUG(0,("standard input is not a socket, assuming -D option\n"));
-               is_daemon = True;
+       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 && !opt_interactive) {
+
+       if (server_mode == SERVER_MODE_DAEMON) {
                DEBUG( 2, ( "Becoming a daemon.\n" ) );
-               become_daemon(Fork);
+               become_daemon(True, no_process_group);
+       } else if (server_mode == SERVER_MODE_FOREGROUND) {
+               become_daemon(False, no_process_group);
        }
 
 #if HAVE_SETPGID
@@ -757,10 +761,14 @@ static BOOL open_sockets(BOOL isdaemon, int port)
         * If we're interactive we want to set our own process group for 
         * signal management.
         */
-       if (opt_interactive)
+       if (server_mode == SERVER_MODE_INTERACTIVE && !no_process_group)
                setpgid( (pid_t)0, (pid_t)0 );
 #endif
 
+       if (nmbd_messaging_context() == NULL) {
+               return 1;
+       }
+
 #ifndef SYNC_DNS
        /* Setup the async dns. We do it here so it doesn't have all the other
                stuff initialised and thus chewing memory and sockets */
@@ -774,21 +782,25 @@ static BOOL open_sockets(BOOL isdaemon, int port)
        }
 
        pidfile_create("nmbd");
-       message_init();
-       message_register(MSG_FORCE_ELECTION, nmbd_message_election);
+       messaging_register(nmbd_messaging_context(), NULL,
+                          MSG_FORCE_ELECTION, nmbd_message_election);
 #if 0
        /* Until winsrepl is done. */
-       message_register(MSG_WINS_NEW_ENTRY, nmbd_wins_new_entry);
+       messaging_register(nmbd_messaging_context(), NULL,
+                          MSG_WINS_NEW_ENTRY, nmbd_wins_new_entry);
 #endif
-       message_register(MSG_SHUTDOWN, nmbd_terminate);
-       message_register(MSG_SMB_CONF_UPDATED, msg_reload_nmbd_services);
-       message_register(MSG_SEND_PACKET, msg_nmbd_send_packet);
+       messaging_register(nmbd_messaging_context(), NULL,
+                          MSG_SHUTDOWN, nmbd_terminate);
+       messaging_register(nmbd_messaging_context(), NULL,
+                          MSG_SMB_CONF_UPDATED, msg_reload_nmbd_services);
+       messaging_register(nmbd_messaging_context(), NULL,
+                          MSG_SEND_PACKET, msg_nmbd_send_packet);
 
        TimeInit();
 
        DEBUG( 3, ( "Opening sockets %d\n", global_nmb_port ) );
 
-       if ( !open_sockets( is_daemon, global_nmb_port ) ) {
+       if ( !open_sockets( server_mode, global_nmb_port ) ) {
                kill_async_dns_child();
                return 1;
        }