#include "includes.h"
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;
static BOOL log_stdout;
BOOL found_lm_clients = False;
time_t StartupTime = 0;
static void terminate(void)
{
DEBUG(0,("Got SIGTERM: going down...\n"));
wins_write_database(0,False);
release_wins_names();
announce_my_servers_removed();
kill_async_dns_child();
exit(0);
}
static void nmbd_terminate(int msg_type, struct process_id src,
void *buf, size_t len, void *private_data)
{
terminate();
}
static SIG_ATOMIC_T got_sig_term;
static void sig_term(int sig)
{
got_sig_term = 1;
sys_select_signal(SIGTERM);
}
static SIG_ATOMIC_T reload_after_sighup;
static void sig_hup(int sig)
{
reload_after_sighup = 1;
sys_select_signal(SIGHUP);
}
static void fault_continue(void)
{
dump_core();
}
static void expire_names_and_servers(time_t t)
{
static time_t lastrun = 0;
if ( !lastrun )
lastrun = t;
if ( t < (lastrun + 5) )
return;
lastrun = t;
expire_names(t);
expire_workgroups_and_servers(t);
}
static BOOL reload_interfaces(time_t t)
{
static time_t lastt;
int n;
struct subnet_record *subrec;
if (t && ((t - lastt) < NMBD_INTERFACES_RELOAD)) return False;
lastt = t;
if (!interfaces_changed()) return False;
load_interfaces();
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;
}
if (ip_equal(iface->ip, loopback_ip)) {
DEBUG(2,("reload_interfaces: Ignoring loopback interface %s\n", inet_ntoa(iface->ip)));
continue;
}
for (subrec=subnetlist; subrec; subrec=subrec->next) {
if (ip_equal(iface->ip, subrec->myip) &&
ip_equal(iface->nmask, subrec->mask_ip)) break;
}
if (!subrec) {
DEBUG(2,("Found new interface %s\n",
inet_ntoa(iface->ip)));
subrec = make_normal_subnet(iface);
if (subrec)
register_my_workgroup_one_subnet(subrec);
}
}
for (subrec=subnetlist; subrec; subrec=subrec->next) {
for (n=iface_count() - 1; n >= 0; n--) {
struct interface *iface = get_interface(n);
if (ip_equal(iface->ip, subrec->myip) &&
ip_equal(iface->nmask, subrec->mask_ip)) break;
}
if (n == -1) {
DEBUG(2,("Deleting dead interface %s\n",
inet_ntoa(subrec->myip)));
close_subnet(subrec);
}
}
rescan_listen_set = True;
if (FIRST_SUBNET == NULL) {
DEBUG(0,("reload_interfaces: No subnets to listen to. Shutting down...\n"));
return True;
}
return False;
}
static BOOL reload_nmbd_services(BOOL test)
{
BOOL ret;
set_remote_machine_name("nmbd", False);
if ( lp_loaded() ) {
pstring fname;
pstrcpy( fname,lp_configfile());
if (file_exist(fname,NULL) && !strcsequal(fname,dyn_CONFIGFILE)) {
pstrcpy(dyn_CONFIGFILE,fname);
test = False;
}
}
if ( test && !lp_file_list_changed() )
return(True);
ret = lp_load( dyn_CONFIGFILE, True , False, False, True);
if ( !test ) {
DEBUG( 3, ( "services not loaded\n" ) );
reload_nmbd_services( True );
}
return(ret);
}
static void msg_reload_nmbd_services(int msg_type, struct process_id src,
void *buf, size_t len, void *private_data)
{
write_browse_list( 0, True );
dump_all_namelists();
reload_nmbd_services( True );
reopen_logs();
if(buf) {
*((BOOL*)buf) = reload_interfaces(0);
}
}
static void msg_nmbd_send_packet(int msg_type, struct process_id src,
void *buf, size_t len, void *private_data)
{
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);
}
static void process(void)
{
BOOL run_election;
BOOL no_subnets;
while( True ) {
time_t t = time(NULL);
message_dispatch();
run_election = check_elections();
if(listen_for_packets(run_election))
return;
if (got_sig_term) {
got_sig_term = 0;
terminate();
}
run_packet_queue();
run_elections(t);
announce_my_server_names(t);
announce_my_lm_server_names(t);
announce_myself_to_domain_master_browser(t);
announce_remote(t);
browse_sync_remote(t);
refresh_my_names(t);
expire_names_and_servers(t);
write_browse_list(t, False);
dmb_expire_and_sync_browser_lists(t);
check_master_browser_exists(t);
add_domain_names(t);
initiate_wins_processing(t);
if (lp_enhanced_browsing())
collect_all_workgroup_names_from_wins_server(t);
retransmit_or_expire_response_records(t);
sync_check_completion();
if (lp_enhanced_browsing())
sync_all_dmbs(t);
clear_unexpected(t);
if(reload_after_sighup) {
DEBUG( 0, ( "Got SIGHUP dumping debug info.\n" ) );
msg_reload_nmbd_services(MSG_SMB_CONF_UPDATED,
pid_to_procid(0), (void*) &no_subnets, 0, NULL);
if(no_subnets)
return;
reload_after_sighup = 0;
}
if(reload_interfaces(t))
return;
lp_TALLOC_FREE();
}
}
static BOOL open_sockets(enum smb_server_mode server_mode, int port)
{
if ( server_mode == SERVER_MODE_INETD ) {
ClientNMB = 0;
} else {
ClientNMB = open_socket_in(SOCK_DGRAM, port,
0, interpret_addr(lp_socket_address()),
True);
}
ClientDGRAM = open_socket_in(SOCK_DGRAM, DGRAM_PORT,
3, interpret_addr(lp_socket_address()),
True);
if ( ClientNMB == -1 )
return( False );
BlockSignals(True,SIGPIPE);
set_socket_options( ClientNMB, "SO_BROADCAST" );
set_socket_options( ClientDGRAM, "SO_BROADCAST" );
set_blocking( ClientNMB, False);
set_blocking( ClientDGRAM, False);
DEBUG( 3, ( "open_sockets: Broadcast sockets opened.\n" ) );
return( True );
}
int main(int argc, const char *argv[])
{
pstring logfile;
poptContext pc;
static char *p_lmhosts = dyn_LMHOSTSFILE;
static BOOL no_process_group = False;
enum smb_server_mode server_mode = SERVER_MODE_DAEMON;
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)" },
{"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" },
POPT_COMMON_SAMBA
{ NULL }
};
load_case_tables();
global_nmb_port = NMB_PORT;
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 );
dump_core_setup("nmbd");
BlockSignals(False, SIGHUP);
BlockSignals(False, SIGUSR1);
BlockSignals(False, SIGTERM);
CatchSignal( SIGHUP, SIGNAL_CAST sig_hup );
CatchSignal( SIGTERM, SIGNAL_CAST sig_term );
#if defined(SIGFPE)
BlockSignals(True,SIGFPE);
#endif
#if defined(SIGUSR2)
BlockSignals(True, SIGUSR2);
#endif
if (server_mode == SERVER_MODE_INTERACTIVE) {
log_stdout = True;
}
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);
}
setup_logging( argv[0], log_stdout );
reopen_logs();
DEBUG( 0, ( "Netbios nameserver version %s started.\n", SAMBA_VERSION_STRING) );
DEBUGADD( 0, ( "%s\n", COPYRIGHT_STARTUP_MESSAGE ) );
if ( !reload_nmbd_services(False) )
return(-1);
if(!init_names())
return -1;
reload_nmbd_services( True );
if (strequal(lp_workgroup(),"*")) {
DEBUG(0,("ERROR: a workgroup name of * is no longer supported\n"));
exit(1);
}
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 (server_mode == SERVER_MODE_DAEMON) {
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);
}
#if HAVE_SETPGID
if (server_mode == SERVER_MODE_INTERACTIVE && !no_process_group)
setpgid( (pid_t)0, (pid_t)0 );
#endif
#ifndef SYNC_DNS
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, NULL);
#if 0
message_register(MSG_WINS_NEW_ENTRY, nmbd_wins_new_entry, NULL);
#endif
message_register(MSG_SHUTDOWN, nmbd_terminate, NULL);
message_register(MSG_SMB_CONF_UPDATED, msg_reload_nmbd_services, NULL);
message_register(MSG_SEND_PACKET, msg_nmbd_send_packet, NULL);
TimeInit();
DEBUG( 3, ( "Opening sockets %d\n", global_nmb_port ) );
if ( !open_sockets( server_mode, global_nmb_port ) ) {
kill_async_dns_child();
return 1;
}
load_interfaces();
if( False == create_subnets() ) {
DEBUG(0,("ERROR: Failed when creating subnet lists. Exiting.\n"));
kill_async_dns_child();
exit(1);
}
load_lmhosts_file(p_lmhosts);
DEBUG(3,("Loaded hosts file %s\n", p_lmhosts));
if( !initialise_wins() ) {
DEBUG( 0, ( "nmbd: Failed when initialising WINS server.\n" ) );
kill_async_dns_child();
exit(1);
}
if( False == register_my_workgroup_and_names() ) {
DEBUG(0,("ERROR: Failed when creating my my workgroup. Exiting.\n"));
kill_async_dns_child();
exit(1);
}
BlockSignals( True, SIGTERM );
process();
if (dbf)
x_fclose(dbf);
kill_async_dns_child();
return(0);
}