#include "includes.h"
pstring servicesf = CONFIGFILE;
int ClientNMB = -1;
int ClientDGRAM = -1;
int global_nmb_port = -1;
static pstring host_file;
extern pstring global_myname;
extern fstring global_myworkgroup;
extern char **my_netbios_names;
extern BOOL global_in_nmbd;
static BOOL is_daemon = False;
BOOL found_lm_clients = False;
time_t StartupTime = 0;
extern struct in_addr ipzero;
static void sig_term(int sig)
{
BlockSignals(True,SIGTERM);
DEBUG(0,("Got SIGTERM: going down...\n"));
wins_write_database(False);
release_my_names();
announce_my_servers_removed();
kill_async_dns_child();
exit(0);
}
static VOLATILE sig_atomic_t reload_after_sighup = False;
static void sig_hup(int sig)
{
BlockSignals( True, SIGHUP );
DEBUG( 0, ( "Got SIGHUP dumping debug info.\n" ) );
write_browse_list( 0, True );
dump_all_namelists();
reload_after_sighup = True;
BlockSignals(False,SIGHUP);
}
#if DUMP_CORE
static BOOL dump_core(void)
{
char *p;
pstring dname;
pstrcpy( dname, lp_logfile() );
if ((p=strrchr(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
static void fault_continue(void)
{
#if DUMP_CORE
dump_core();
#endif
}
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;
extern BOOL rescan_listen_set;
extern struct in_addr loopback_ip;
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 (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;
extern fstring remote_machine;
fstrcpy( remote_machine, "nmbd" );
if ( lp_loaded() )
{
pstring fname;
pstrcpy( fname,lp_configfile());
if (file_exist(fname,NULL) && !strcsequal(fname,servicesf))
{
pstrcpy(servicesf,fname);
test = False;
}
}
if ( test && !lp_file_list_changed() )
return(True);
ret = lp_load( servicesf, True , False, False);
if ( !test )
{
DEBUG( 3, ( "services not loaded\n" ) );
reload_nmbd_services( True );
}
if( lp_wins_support() && *lp_wins_server() )
{
DEBUG(0,("ERROR: both 'wins support = true' and 'wins server = <server>' \
cannot be set in the smb.conf file. nmbd aborting.\n"));
exit(10);
}
return(ret);
}
static void process(void)
{
BOOL run_election;
while( True )
{
time_t t = time(NULL);
message_dispatch();
run_election = check_elections();
if(listen_for_packets(run_election))
return;
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) {
reload_nmbd_services( True );
reopen_logs();
if(reload_interfaces(0))
return;
reload_after_sighup = False;
}
if(reload_interfaces(t))
return;
lp_talloc_free();
}
}
static BOOL open_sockets(BOOL isdaemon, int port)
{
if ( isdaemon )
ClientNMB = open_socket_in(SOCK_DGRAM, port,0,0,True);
else
ClientNMB = 0;
ClientDGRAM = open_socket_in(SOCK_DGRAM,DGRAM_PORT,3,0,True);
if ( ClientNMB == -1 )
return( False );
BlockSignals(True,SIGPIPE);
set_socket_options( ClientNMB, "SO_BROADCAST" );
set_socket_options( ClientDGRAM, "SO_BROADCAST" );
DEBUG( 3, ( "open_sockets: Broadcast sockets opened.\n" ) );
return( True );
}
static BOOL init_structs(void)
{
extern fstring local_machine;
char *p, *ptr;
int namecount;
int n;
int nodup;
pstring nbname;
if (! *global_myname)
{
fstrcpy( global_myname, myhostname() );
p = strchr( global_myname, '.' );
if (p)
*p = 0;
}
strupper( global_myname );
ptr = lp_netbios_aliases();
for( namecount=0; next_token(&ptr,nbname,NULL, sizeof(nbname)); namecount++ )
;
if ( *global_myname )
namecount++;
my_netbios_names = (char **)malloc( sizeof(char *) * (namecount+1) );
if( NULL == my_netbios_names )
{
DEBUG( 0, ( "init_structs: malloc fail.\n" ) );
return( False );
}
namecount=0;
if ( *global_myname )
my_netbios_names[namecount++] = global_myname;
ptr = lp_netbios_aliases();
while ( next_token( &ptr, nbname, NULL, sizeof(nbname) ) )
{
strupper( nbname );
nodup=1;
for( n=0; n<namecount; n++ )
{
if( 0 == strcmp( nbname, my_netbios_names[n] ) )
nodup=0;
}
if (nodup)
my_netbios_names[namecount++] = strdup( nbname );
}
for( n = 0; n < namecount; n++ )
if( NULL == my_netbios_names[n] )
{
DEBUG(0,("init_structs: malloc fail when allocating names.\n"));
return False;
}
my_netbios_names[namecount++] = NULL;
fstrcpy( local_machine, global_myname );
trim_string( local_machine, " ", " " );
p = strchr( local_machine, ' ' );
if (p)
*p = 0;
strlower( local_machine );
DEBUG( 5, ("Netbios name list:-\n") );
for( n=0; my_netbios_names[n]; n++ )
DEBUGADD( 5, ( "my_netbios_names[%d]=\"%s\"\n", n, my_netbios_names[n] ) );
return( True );
}
static void usage(char *pname)
{
printf( "Usage: %s [-DaiohV] [-H lmhosts file] [-d debuglevel] [-l log basename]\n", pname );
printf( " [-n name] [-p port] [-s configuration file]\n" );
printf( "\t-D Become a daemon (default)\n" );
printf( "\t-a Append to log file (default)\n" );
printf( "\t-i Run interactive (not a daemon)\n" );
printf( "\t-o Overwrite log file, don't append\n" );
printf( "\t-h Print usage\n" );
printf( "\t-V Print version\n" );
printf( "\t-H hosts file Load a netbios hosts file\n" );
printf( "\t-d debuglevel Set the debuglevel\n" );
printf( "\t-l log basename. Basename for log/debug files\n" );
printf( "\t-n netbiosname. Primary netbios name\n" );
printf( "\t-p port Listen on the specified port\n" );
printf( "\t-s configuration file Configuration file name\n" );
printf( "\n");
}
int main(int argc,char *argv[])
{
int opt;
extern FILE *dbf;
extern char *optarg;
extern BOOL append_log;
BOOL opt_interactive = False;
pstring logfile;
append_log = True;
global_nmb_port = NMB_PORT;
*host_file = 0;
global_in_nmbd = True;
StartupTime = time(NULL);
sys_srandom(time(NULL) ^ sys_getpid());
TimeInit();
slprintf(logfile, sizeof(logfile)-1, "%s/log.nmbd", LOGFILEBASE);
lp_set_logfile(logfile);
charset_initialise();
#ifdef LMHOSTSFILE
pstrcpy( host_file, LMHOSTSFILE );
#endif
while (argc > 1 && (*argv[1] != '-'))
{
argv++;
argc--;
}
fault_setup((void (*)(void *))fault_continue );
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
while( EOF !=
(opt = getopt( argc, argv, "Vaos:T:I:C:bAB:N:Rn:l:d:Dip:hSH:G:f:" )) )
{
switch (opt)
{
case 's':
pstrcpy(servicesf,optarg);
break;
case 'N':
case 'B':
case 'I':
case 'C':
case 'G':
DEBUG(0,("Obsolete option '%c' used\n",opt));
break;
case 'H':
pstrcpy(host_file,optarg);
break;
case 'n':
pstrcpy(global_myname,optarg);
strupper(global_myname);
break;
case 'l':
slprintf(logfile, sizeof(logfile)-1, "%s/log.nmbd", optarg);
lp_set_logfile(logfile);
break;
case 'a':
append_log = True;
break;
case 'o':
append_log = False;
break;
case 'i':
opt_interactive = True;
break;
case 'D':
is_daemon = True;
break;
case 'd':
DEBUGLEVEL = atoi(optarg);
break;
case 'p':
global_nmb_port = atoi(optarg);
break;
case 'h':
usage(argv[0]);
exit(0);
break;
case 'V':
printf( "Version %s\n", VERSION );
exit(0);
break;
default:
if( !is_a_socket(0) )
{
DEBUG(0,("Incorrect program usage - is the command line correct?\n"));
usage(argv[0]);
exit(0);
}
break;
}
}
setup_logging( argv[0], opt_interactive );
reopen_logs();
DEBUG( 0, ( "Netbios nameserver version %s started.\n", VERSION ) );
DEBUGADD( 0, ( "Copyright Andrew Tridgell and the Samba Team 1994-2002\n" ) );
if ( !reload_nmbd_services(False) )
return(-1);
#ifdef WITH_PROFILE
if (!profile_setup(False)) {
DEBUG(0,("ERROR: failed to setup profiling shared memory\n"));
return -1;
}
#endif
codepage_initialise(lp_client_code_page());
if(!init_structs())
return -1;
reload_nmbd_services( True );
fstrcpy( global_myworkgroup, lp_workgroup() );
if (strequal(global_myworkgroup,"*"))
{
DEBUG(0,("ERROR: a workgroup name of * is no longer supported\n"));
exit(1);
}
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 && !opt_interactive)
{
DEBUG( 2, ( "Becoming a daemon.\n" ) );
become_daemon();
}
#if HAVE_SETPGID
if (opt_interactive)
setpgid( (pid_t)0, (pid_t)0 );
#endif
#ifndef SYNC_DNS
if(lp_we_are_a_wins_server()) {
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);
DEBUG( 3, ( "Opening sockets %d\n", global_nmb_port ) );
if ( !open_sockets( is_daemon, global_nmb_port ) )
return 1;
load_interfaces();
if( False == create_subnets() )
{
DEBUG(0,("ERROR: Failed when creating subnet lists. Exiting.\n"));
exit(1);
}
if ( *host_file )
{
load_lmhosts_file(host_file);
DEBUG(3,("Loaded hosts file\n"));
}
if( !initialise_wins() )
{
DEBUG( 0, ( "nmbd: Failed when initialising WINS server.\n" ) );
exit(1);
}
if( False == register_my_workgroup_and_names() )
{
DEBUG(0,("ERROR: Failed when creating my my workgroup. Exiting.\n"));
exit(1);
}
BlockSignals( True, SIGTERM );
process();
if (dbf)
fclose(dbf);
return(0);
}