r10656: BIG merge from trunk. Features not copied over
[tprouty/samba.git] / source3 / nmbd / nmbd.c
index 013ef9ddb79ae9e76de86cf1065005bbc5f0b7e0..fcaba03b3dbaad5ce42eb4c53e45018c27dab2a9 100644 (file)
@@ -27,16 +27,20 @@ int ClientNMB       = -1;
 int ClientDGRAM     = -1;
 int global_nmb_port = -1;
 
+extern BOOL rescan_listen_set;
+extern struct in_addr loopback_ip;
 extern BOOL global_in_nmbd;
 
+extern BOOL override_logfile;
+
 /* are we running as a daemon ? */
-static BOOL is_daemon = False;
+static BOOL is_daemon;
 
 /* fork or run in foreground ? */
 static BOOL Fork = True;
 
 /* log to standard output ? */
-static BOOL log_stdout = False;
+static BOOL log_stdout;
 
 /* have we found LanMan clients yet? */
 BOOL found_lm_clients = False;
@@ -72,7 +76,8 @@ static void terminate(void)
  Handle a SHUTDOWN message from smbcontrol.
  **************************************************************************** */
 
-static void nmbd_terminate(int msg_type, pid_t src, void *buf, size_t len)
+static void nmbd_terminate(int msg_type, struct process_id src,
+                          void *buf, size_t len)
 {
        terminate();
 }
@@ -86,7 +91,7 @@ static SIG_ATOMIC_T got_sig_term;
 static void sig_term(int sig)
 {
        got_sig_term = 1;
-       sys_select_signal();
+       sys_select_signal(SIGTERM);
 }
 
 /**************************************************************************** **
@@ -98,26 +103,7 @@ static SIG_ATOMIC_T reload_after_sighup;
 static void sig_hup(int sig)
 {
        reload_after_sighup = 1;
-       sys_select_signal();
-}
-
-/*******************************************************************
- Print out all talloc memory info.
-********************************************************************/
-
-void return_all_talloc_info(int msg_type, pid_t src_pid, void *buf, size_t len)
-{
-       TALLOC_CTX *ctx = talloc_init("info context");
-       char *info = NULL;
-
-       if (!ctx)
-               return;
-
-       info = talloc_describe_all(ctx);
-       if (info)
-               DEBUG(10,(info));
-       message_send_pid(src_pid, MSG_TALLOC_USAGE, info, info ? strlen(info) + 1 : 0, True);
-       talloc_destroy(ctx);
+       sys_select_signal(SIGHUP);
 }
 
 #if DUMP_CORE
@@ -213,8 +199,6 @@ static BOOL reload_interfaces(time_t t)
        static time_t lastt;
        int n;
        struct subnet_record *subrec;
-       extern BOOL rescan_listen_set;
-       extern struct in_addr loopback_ip;
 
        if (t && ((t - lastt) < NMBD_INTERFACES_RELOAD)) return False;
        lastt = t;
@@ -250,7 +234,8 @@ static BOOL reload_interfaces(time_t t)
                        DEBUG(2,("Found new interface %s\n", 
                                 inet_ntoa(iface->ip)));
                        subrec = make_normal_subnet(iface);
-                       if (subrec) register_my_workgroup_one_subnet(subrec);
+                       if (subrec)
+                               register_my_workgroup_one_subnet(subrec);
                }
        }
 
@@ -317,6 +302,82 @@ static BOOL reload_nmbd_services(BOOL test)
        return(ret);
 }
 
+/**************************************************************************** **
+ * React on 'smbcontrol nmbd reload-config' in the same way as to SIGHUP
+ * We use buf here to return BOOL result to process() when reload_interfaces()
+ * detects that there are no subnets.
+ **************************************************************************** */
+
+static void msg_reload_nmbd_services(int msg_type, struct process_id src,
+                                    void *buf, size_t len)
+{
+       write_browse_list( 0, True );
+       dump_all_namelists();
+       reload_nmbd_services( True );
+       reopen_logs();
+       
+       if(buf) {
+               /* 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);  
+       }
+}
+
+static void msg_nmbd_send_packet(int msg_type, struct process_id src,
+                                void *buf, size_t len)
+{
+       struct packet_struct *p = (struct packet_struct *)buf;
+       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)) {
+               DEBUG(2, ("Discarding invalid packet length from %d\n",
+                         procid_to_pid(&src)));
+               return;
+       }
+
+       if ((p->packet_type != NMB_PACKET) &&
+           (p->packet_type != DGRAM_PACKET)) {
+               DEBUG(2, ("Discarding invalid packet type from %d: %d\n",
+                         procid_to_pid(&src), p->packet_type));
+               return;
+       }
+
+       local_ip = iface_ip(p->ip);
+
+       if (local_ip == NULL) {
+               DEBUG(2, ("Could not find ip for packet from %d\n",
+                         procid_to_pid(&src)));
+               return;
+       }
+
+       subrec = FIRST_SUBNET;
+
+       p->fd = (p->packet_type == NMB_PACKET) ?
+               subrec->nmb_sock : subrec->dgram_sock;
+
+       for (subrec = FIRST_SUBNET; subrec != NULL;
+            subrec = NEXT_SUBNET_EXCLUDING_UNICAST(subrec)) {
+               if (ip_equal(*local_ip, subrec->myip)) {
+                       p->fd = (p->packet_type == NMB_PACKET) ?
+                               subrec->nmb_sock : subrec->dgram_sock;
+                       break;
+               }
+       }
+
+       if (p->packet_type == DGRAM_PACKET) {
+               p->port = 138;
+               p->packet.dgram.header.source_ip.s_addr = local_ip->s_addr;
+               p->packet.dgram.header.source_port = 138;
+       }
+
+       send_packet(p);
+}
+
 /**************************************************************************** **
  The main select loop.
  **************************************************************************** */
@@ -324,6 +385,7 @@ static BOOL reload_nmbd_services(BOOL test)
 static void process(void)
 {
        BOOL run_election;
+       BOOL no_subnets;
 
        while( True ) {
                time_t t = time(NULL);
@@ -532,11 +594,9 @@ static void process(void)
 
                if(reload_after_sighup) {
                        DEBUG( 0, ( "Got SIGHUP dumping debug info.\n" ) );
-                       write_browse_list( 0, True );
-                       dump_all_namelists();
-                       reload_nmbd_services( True );
-                       reopen_logs();
-                       if(reload_interfaces(0))
+                       msg_reload_nmbd_services(MSG_SMB_CONF_UPDATED,
+                                                pid_to_procid(0), (void*) &no_subnets, 0);
+                       if(no_subnets)
                                return;
                        reload_after_sighup = 0;
                }
@@ -566,7 +626,9 @@ static BOOL open_sockets(BOOL isdaemon, int port)
         */
 
        if ( isdaemon )
-               ClientNMB = open_socket_in(SOCK_DGRAM, port,0,0,True);
+               ClientNMB = open_socket_in(SOCK_DGRAM, port,
+                                          0, interpret_addr(lp_socket_address()),
+                                          True);
        else
                ClientNMB = 0;
   
@@ -590,7 +652,8 @@ static BOOL open_sockets(BOOL isdaemon, int port)
  **************************************************************************** */
  int main(int argc, const char *argv[])
 {
-       static BOOL opt_interactive = False;
+       pstring logfile;
+       static BOOL opt_interactive;
        poptContext pc;
        struct poptOption long_options[] = {
        POPT_AUTOHELP
@@ -603,167 +666,165 @@ static BOOL open_sockets(BOOL isdaemon, int port)
        POPT_COMMON_SAMBA
        { NULL }
        };
-       pstring logfile;
-
-  global_nmb_port = NMB_PORT;
-  global_in_nmbd = True;
 
-  StartupTime = time(NULL);
+       global_nmb_port = NMB_PORT;
 
-  sys_srandom(time(NULL) ^ sys_getpid());
-
-  slprintf(logfile, sizeof(logfile)-1, "%s/log.nmbd", dyn_LOGFILEBASE);
-  lp_set_logfile(logfile);
-
-  fault_setup((void (*)(void *))fault_continue );
-
-  /* POSIX demands that signals are inherited. If the invoking process has
-   * these signals masked, we will have problems, as we won't receive them. */
-  BlockSignals(False, SIGHUP);
-  BlockSignals(False, SIGUSR1);
-  BlockSignals(False, SIGTERM);
-
-  CatchSignal( SIGHUP,  SIGNAL_CAST sig_hup );
-  CatchSignal( SIGTERM, SIGNAL_CAST sig_term );
+       pc = poptGetContext("nmbd", argc, argv, long_options, 0);
+       while (poptGetNextOpt(pc) != -1) {};
+       poptFreeContext(pc);
 
+       global_in_nmbd = True;
+       
+       StartupTime = time(NULL);
+       
+       sys_srandom(time(NULL) ^ sys_getpid());
+       
+       if (!override_logfile) {
+               slprintf(logfile, sizeof(logfile)-1, "%s/log.nmbd", dyn_LOGFILEBASE);
+               lp_set_logfile(logfile);
+       }
+       
+       fault_setup((void (*)(void *))fault_continue );
+       
+       /* POSIX demands that signals are inherited. If the invoking process has
+        * these signals masked, we will have problems, as we won't receive them. */
+       BlockSignals(False, SIGHUP);
+       BlockSignals(False, SIGUSR1);
+       BlockSignals(False, SIGTERM);
+       
+       CatchSignal( SIGHUP,  SIGNAL_CAST sig_hup );
+       CatchSignal( SIGTERM, SIGNAL_CAST sig_term );
+       
 #if defined(SIGFPE)
-  /* we are never interested in SIGFPE */
-  BlockSignals(True,SIGFPE);
+       /* we are never interested in SIGFPE */
+       BlockSignals(True,SIGFPE);
 #endif
 
-  /* We no longer use USR2... */
+       /* We no longer use USR2... */
 #if defined(SIGUSR2)
-  BlockSignals(True, SIGUSR2);
+       BlockSignals(True, SIGUSR2);
 #endif
-  pc = poptGetContext("nmbd", argc, argv, long_options, 0);
-  
-  poptFreeContext(pc);
 
-  if ( opt_interactive ) {
-    Fork = False;
-    log_stdout = True;
-  }
+       if ( opt_interactive ) {
+               Fork = False;
+               log_stdout = True;
+       }
 
-  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);
-  }
+       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);
+       }
 
-  setup_logging( argv[0], log_stdout );
+       setup_logging( argv[0], log_stdout );
 
-  reopen_logs();
+       reopen_logs();
 
-  DEBUG( 0, ( "Netbios nameserver version %s started.\n", VERSION ) );
-  DEBUGADD( 0, ( "Copyright Andrew Tridgell and the Samba Team 1994-2003\n" ) );
+       DEBUG( 0, ( "Netbios nameserver version %s started.\n", SAMBA_VERSION_STRING) );
+       DEBUGADD( 0, ( "Copyright Andrew Tridgell and the Samba Team 1994-2004\n" ) );
 
-  if ( !reload_nmbd_services(False) )
-    return(-1);
+       if ( !reload_nmbd_services(False) )
+               return(-1);
 
-  if(!init_names())
-    return -1;
+       if(!init_names())
+               return -1;
 
-  reload_nmbd_services( True );
+       reload_nmbd_services( True );
 
-  if (strequal(lp_workgroup(),"*"))
-  {
-    DEBUG(0,("ERROR: a workgroup name of * is no longer supported\n"));
-    exit(1);
-  }
+       if (strequal(lp_workgroup(),"*")) {
+               DEBUG(0,("ERROR: a workgroup name of * is no longer supported\n"));
+               exit(1);
+       }
 
-  set_samba_nb_type();
+       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_daemon && !is_a_socket(0)) {
+               DEBUG(0,("standard input is not a socket, assuming -D option\n"));
+               is_daemon = True;
+       }
   
-  if (is_daemon && !opt_interactive)
-  {
-    DEBUG( 2, ( "Becoming a daemon.\n" ) );
-    become_daemon(Fork);
-  }
+       if (is_daemon && !opt_interactive) {
+               DEBUG( 2, ( "Becoming a daemon.\n" ) );
+               become_daemon(Fork);
+       }
 
 #if HAVE_SETPGID
-  /*
-   * If we're interactive we want to set our own process group for 
-   * signal management.
-   */
-  if (opt_interactive)
-    setpgid( (pid_t)0, (pid_t)0 );
+       /*
+        * If we're interactive we want to set our own process group for 
+        * signal management.
+        */
+       if (opt_interactive)
+               setpgid( (pid_t)0, (pid_t)0 );
 #endif
 
 #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 */
-  if(lp_we_are_a_wins_server() && lp_dns_proxy()) {
-         start_async_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 */
+       if(lp_we_are_a_wins_server() && lp_dns_proxy()) {
+               start_async_dns();
+       }
 #endif
 
-  if (!directory_exist(lp_lockdir(), NULL)) {
-         mkdir(lp_lockdir(), 0755);
-  }
-
-  pidfile_create("nmbd");
-  message_init();
-  message_register(MSG_FORCE_ELECTION, nmbd_message_election);
-  message_register(MSG_WINS_NEW_ENTRY, nmbd_wins_new_entry);
-  message_register(MSG_SHUTDOWN, nmbd_terminate);
-  message_register(MSG_REQ_TALLOC_USAGE, return_all_talloc_info);
-
-  DEBUG( 3, ( "Opening sockets %d\n", global_nmb_port ) );
-
-  if ( !open_sockets( is_daemon, global_nmb_port ) ) {
-    kill_async_dns_child();
-    return 1;
-  }
-
-  /* Determine all the IP addresses we have. */
-  load_interfaces();
-
-  /* Create an nmbd subnet record for each of the above. */
-  if( False == create_subnets() )
-  {
-    DEBUG(0,("ERROR: Failed when creating subnet lists. Exiting.\n"));
-    kill_async_dns_child();
-    exit(1);
-  }
-
-  /* Load in any static local names. */ 
-  load_lmhosts_file(dyn_LMHOSTSFILE);
-  DEBUG(3,("Loaded hosts file %s\n", dyn_LMHOSTSFILE));
-
-  /* If we are acting as a WINS server, initialise data structures. */
-  if( !initialise_wins() )
-  {
-    DEBUG( 0, ( "nmbd: Failed when initialising WINS server.\n" ) );
-    kill_async_dns_child();
-    exit(1);
-  }
-
-  /* 
-   * Register nmbd primary workgroup and nmbd names on all
-   * the broadcast subnets, and on the WINS server (if specified).
-   * Also initiate the startup of our primary workgroup (start
-   * elections if we are setup as being able to be a local
-   * master browser.
-   */
-
-  if( False == register_my_workgroup_and_names() )
-  {
-    DEBUG(0,("ERROR: Failed when creating my my workgroup. Exiting.\n"));
-    kill_async_dns_child();
-    exit(1);
-  }
-
-  /* We can only take signals in the select. */
-  BlockSignals( True, SIGTERM );
-
-  process();
-
-  if (dbf)
-    x_fclose(dbf);
-  kill_async_dns_child();
-  return(0);
+       if (!directory_exist(lp_lockdir(), NULL)) {
+               mkdir(lp_lockdir(), 0755);
+       }
+
+       pidfile_create("nmbd");
+       message_init();
+       message_register(MSG_FORCE_ELECTION, nmbd_message_election);
+       message_register(MSG_WINS_NEW_ENTRY, nmbd_wins_new_entry);
+       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);
+
+       DEBUG( 3, ( "Opening sockets %d\n", global_nmb_port ) );
+
+       if ( !open_sockets( is_daemon, global_nmb_port ) ) {
+               kill_async_dns_child();
+               return 1;
+       }
+
+       /* Determine all the IP addresses we have. */
+       load_interfaces();
+
+       /* Create an nmbd subnet record for each of the above. */
+       if( False == create_subnets() ) {
+               DEBUG(0,("ERROR: Failed when creating subnet lists. Exiting.\n"));
+               kill_async_dns_child();
+               exit(1);
+       }
+
+       /* Load in any static local names. */ 
+       load_lmhosts_file(dyn_LMHOSTSFILE);
+       DEBUG(3,("Loaded hosts file %s\n", dyn_LMHOSTSFILE));
+
+       /* If we are acting as a WINS server, initialise data structures. */
+       if( !initialise_wins() ) {
+               DEBUG( 0, ( "nmbd: Failed when initialising WINS server.\n" ) );
+               kill_async_dns_child();
+               exit(1);
+       }
+
+       /* 
+        * Register nmbd primary workgroup and nmbd names on all
+        * the broadcast subnets, and on the WINS server (if specified).
+        * Also initiate the startup of our primary workgroup (start
+        * elections if we are setup as being able to be a local
+        * master browser.
+        */
+
+       if( False == register_my_workgroup_and_names() ) {
+               DEBUG(0,("ERROR: Failed when creating my my workgroup. Exiting.\n"));
+               kill_async_dns_child();
+               exit(1);
+       }
+
+       /* We can only take signals in the select. */
+       BlockSignals( True, SIGTERM );
+
+       process();
+
+       if (dbf)
+               x_fclose(dbf);
+       kill_async_dns_child();
+       return(0);
 }