#include "includes.h"
static int am_parent = 1;
int last_message = -1;
#define LAST_MESSAGE() smb_fn_name(last_message)
extern pstring user_socket_options;
extern SIG_ATOMIC_T got_sig_term;
extern SIG_ATOMIC_T reload_after_sighup;
#ifdef WITH_DFS
extern int dcelogin_atmost_once;
#endif
static int server_fd = -1;
#ifdef WITH_BRLM
BOOL brlm_init = False;
#endif
int smbd_server_fd(void)
{
return server_fd;
}
static void smbd_set_server_fd(int fd)
{
server_fd = fd;
client_setfd(fd);
}
static void sig_term(void)
{
got_sig_term = 1;
sys_select_signal();
}
static void sig_hup(int sig)
{
reload_after_sighup = 1;
sys_select_signal();
}
static void killkids(void)
{
if(am_parent) kill(0,SIGTERM);
}
static void msg_sam_sync(int UNUSED(msg_type), pid_t UNUSED(pid),
void *UNUSED(buf), size_t UNUSED(len))
{
DEBUG(10, ("** sam sync message received, ignoring\n"));
}
static void msg_sam_repl(int msg_type, pid_t pid, void *buf, size_t len)
{
uint32 low_serial;
if (len != sizeof(uint32))
return;
low_serial = *((uint32 *)buf);
DEBUG(3, ("received sam replication message, serial = 0x%04x\n",
low_serial));
}
static BOOL open_sockets_inetd(void)
{
smbd_set_server_fd(dup(0));
close_low_fds(False);
set_socket_options(smbd_server_fd(),"SO_KEEPALIVE");
set_socket_options(smbd_server_fd(), user_socket_options);
#ifdef WITH_BRLM
if (lp_BRLM() && !brlm_init) {
BRLMInit();
brlm_init = True;
}
#endif
return True;
}
static void msg_exit_server(int msg_type, pid_t src, void *buf, size_t len)
{
exit_server("Got a SHUTDOWN message");
}
static BOOL allowable_number_of_smbd_processes(void)
{
int max_processes = lp_max_smbd_processes();
if (!max_processes)
return True;
{
TDB_CONTEXT *tdb = conn_tdb_ctx();
int32 val;
if (!tdb) {
DEBUG(0,("allowable_number_of_smbd_processes: can't open connection tdb.\n" ));
return False;
}
val = tdb_fetch_int32(tdb, "INFO/total_smbds");
if (val == -1 && (tdb_error(tdb) != TDB_ERR_NOEXIST)) {
DEBUG(0,("allowable_number_of_smbd_processes: can't fetch INFO/total_smbds. Error %s\n",
tdb_errorstr(tdb) ));
return False;
}
if (val > max_processes) {
DEBUG(0,("allowable_number_of_smbd_processes: number of processes (%d) is over allowed limit (%d)\n",
val, max_processes ));
return False;
}
}
return True;
}
static BOOL open_sockets_smbd(BOOL is_daemon, BOOL interactive, const char *smb_ports)
{
int num_interfaces = iface_count();
int num_sockets = 0;
int fd_listenset[FD_SETSIZE];
fd_set listen_set;
int s;
int i;
char *ports;
if (!is_daemon) {
return open_sockets_inetd();
}
#ifdef HAVE_ATEXIT
{
static int atexit_set;
if(atexit_set == 0) {
atexit_set=1;
atexit(killkids);
}
}
#endif
CatchChild();
FD_ZERO(&listen_set);
if (!smb_ports) {
ports = lp_smb_ports();
if (!ports || !*ports) {
ports = smb_xstrdup(SMB_PORTS);
} else {
ports = smb_xstrdup(ports);
}
} else {
ports = smb_xstrdup(smb_ports);
}
if (lp_interfaces() && lp_bind_interfaces_only()) {
for(i = 0; i < num_interfaces; i++) {
struct in_addr *ifip = iface_n_ip(i);
fstring tok;
const char *ptr;
if(ifip == NULL) {
DEBUG(0,("open_sockets_smbd: interface %d has NULL IP address !\n", i));
continue;
}
for (ptr=ports; next_token(&ptr, tok, NULL, sizeof(tok)); ) {
unsigned port = atoi(tok);
if (port == 0) continue;
s = fd_listenset[num_sockets] = open_socket_in(SOCK_STREAM, port, 0, ifip->s_addr, True);
if(s == -1)
return False;
set_socket_options(s,"SO_KEEPALIVE");
set_socket_options(s,user_socket_options);
set_blocking(s,False);
if (listen(s, SMBD_LISTEN_BACKLOG) == -1) {
DEBUG(0,("listen: %s\n",strerror(errno)));
close(s);
return False;
}
FD_SET(s,&listen_set);
num_sockets++;
if (num_sockets >= FD_SETSIZE) {
DEBUG(0,("open_sockets_smbd: Too many sockets to bind to\n"));
return False;
}
}
}
} else {
fstring tok;
const char *ptr;
num_interfaces = 1;
for (ptr=ports; next_token(&ptr, tok, NULL, sizeof(tok)); ) {
unsigned port = atoi(tok);
if (port == 0) continue;
s = open_socket_in(SOCK_STREAM, port, 0,
interpret_addr(lp_socket_address()),True);
if (s == -1)
return(False);
set_socket_options(s,"SO_KEEPALIVE");
set_socket_options(s,user_socket_options);
set_blocking(s,False);
if (listen(s, SMBD_LISTEN_BACKLOG) == -1) {
DEBUG(0,("open_sockets_smbd: listen: %s\n",
strerror(errno)));
close(s);
return False;
}
fd_listenset[num_sockets] = s;
FD_SET(s,&listen_set);
num_sockets++;
if (num_sockets >= FD_SETSIZE) {
DEBUG(0,("open_sockets_smbd: Too many sockets to bind to\n"));
return False;
}
}
}
SAFE_FREE(ports);
message_register(MSG_SMB_SAM_SYNC, msg_sam_sync);
message_register(MSG_SMB_SAM_REPL, msg_sam_repl);
message_register(MSG_SHUTDOWN, msg_exit_server);
DEBUG(2,("waiting for a connection\n"));
while (1) {
fd_set lfds;
int num;
lp_talloc_free();
message_dispatch();
memcpy((char *)&lfds, (char *)&listen_set,
sizeof(listen_set));
num = sys_select(FD_SETSIZE,&lfds,NULL,NULL,NULL);
if (num == -1 && errno == EINTR) {
if (got_sig_term) {
exit_server("Caught TERM signal");
}
if (reload_after_sighup) {
change_to_root_user();
DEBUG(1,("Reloading services after SIGHUP\n"));
reload_services(False);
reload_after_sighup = 0;
}
continue;
}
check_reload(time(NULL));
for( ; num > 0; num--) {
struct sockaddr addr;
socklen_t in_addrlen = sizeof(addr);
s = -1;
for(i = 0; i < num_sockets; i++) {
if(FD_ISSET(fd_listenset[i],&lfds)) {
s = fd_listenset[i];
FD_CLR(fd_listenset[i],&lfds);
break;
}
}
smbd_set_server_fd(accept(s,&addr,&in_addrlen));
if (smbd_server_fd() == -1 && errno == EINTR)
continue;
if (smbd_server_fd() == -1) {
DEBUG(0,("open_sockets_smbd: accept: %s\n",
strerror(errno)));
continue;
}
set_blocking(smbd_server_fd(),True);
if (smbd_server_fd() != -1 && interactive) {
#ifdef WITH_BRLM
if (lp_BRLM() && !brlm_init) {
BRLMInit();
brlm_init = True;
}
#endif
return True;
}
if (allowable_number_of_smbd_processes() && smbd_server_fd() != -1 && sys_fork()==0) {
for(i = 0; i < num_sockets; i++)
close(fd_listenset[i]);
close_low_fds(False);
am_parent = 0;
set_socket_options(smbd_server_fd(),"SO_KEEPALIVE");
set_socket_options(smbd_server_fd(),user_socket_options);
set_remote_machine_name(get_peer_addr(smbd_server_fd()), False);
set_need_random_reseed();
if (tdb_reopen_all() == -1) {
DEBUG(0,("tdb_reopen_all failed.\n"));
smb_panic("tdb_reopen_all failed.");
}
#ifdef WITH_BRLM
if (lp_BRLM() && !brlm_init) {
BRLMInit();
brlm_init = True;
}
#endif
return True;
}
close(smbd_server_fd());
smbd_set_server_fd(-1);
force_check_log_size();
}
}
}
BOOL reload_services(BOOL test)
{
BOOL ret;
if (lp_loaded()) {
pstring fname;
pstrcpy(fname,lp_configfile());
if (file_exist(fname, NULL) &&
!strcsequal(fname, dyn_CONFIGFILE)) {
pstrcpy(dyn_CONFIGFILE, fname);
test = False;
}
}
reopen_logs();
if (test && !lp_file_list_changed())
return(True);
lp_killunused(conn_snum_used);
ret = lp_load(dyn_CONFIGFILE, False, False, True);
load_printers();
if (!test)
reload_services(True);
reopen_logs();
load_interfaces();
if (smbd_server_fd() != -1) {
set_socket_options(smbd_server_fd(),"SO_KEEPALIVE");
set_socket_options(smbd_server_fd(), user_socket_options);
}
mangle_reset_cache();
reset_stat_cache();
set_current_service(NULL,0,True);
return(ret);
}
#if DUMP_CORE
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));
#ifdef SIGABRT
CatchSignal(SIGABRT,SIGNAL_CAST SIG_DFL);
#endif
abort();
return(True);
}
#endif
void exit_server(const char *reason)
{
static int firsttime=1;
extern char *last_inbuf;
extern struct auth_context *negprot_global_auth_context;
if (!firsttime)
exit(0);
firsttime = 0;
change_to_root_user();
DEBUG(2,("Closing connections\n"));
if (negprot_global_auth_context) {
(negprot_global_auth_context->free)(&negprot_global_auth_context);
}
conn_close_all();
invalidate_all_vuids();
print_notify_send_messages(3);
smb_run_exit_events();
yield_connection(NULL,"");
respond_to_all_remaining_local_messages();
decrement_smbd_process_count();
#ifdef WITH_DFS
if (dcelogin_atmost_once) {
dfs_unlogin();
}
#endif
if (!reason) {
int oldlevel = SAMBA_DEBUGLEVEL;
SAMBA_DEBUGLEVEL = 10;
DEBUG(0,("Last message was %s\n",smb_fn_name(last_message)));
if (last_inbuf)
show_msg(last_inbuf);
SAMBA_DEBUGLEVEL = oldlevel;
DEBUG(0,("===============================================================\n"));
#if DUMP_CORE
if (dump_core()) return;
#endif
}
locking_end();
printing_end();
#ifdef WITH_BRLM
if(lp_BRLM() && brlm_init)
BRLMClose();
#endif
DEBUG(3,("Server exit (%s)\n", (reason ? reason : "")));
exit(0);
}
static BOOL init_structs(void )
{
if (!init_names())
return False;
conn_init();
file_init();
init_rpc_pipe_hnd();
init_dptrs();
secrets_init();
return True;
}
void build_options(BOOL screen);
int main(int argc,const char *argv[])
{
static BOOL is_daemon = False;
static BOOL interactive = False;
static BOOL Fork = True;
static BOOL log_stdout = False;
static char *ports = NULL;
int opt;
poptContext pc;
struct poptOption long_options[] = {
POPT_AUTOHELP
{"daemon", 'D', POPT_ARG_VAL, &is_daemon, True, "Become a daemon (default)" },
{"interactive", 'i', POPT_ARG_VAL, &interactive, True, "Run interactive (not a daemon)"},
{"foreground", 'F', POPT_ARG_VAL, &Fork, False, "Run daemon in foreground (for daemontools & etc)" },
{"log-stdout", 'S', POPT_ARG_VAL, &log_stdout, True, "Log to stdout" },
{"build-options", 'b', POPT_ARG_NONE, NULL, 'b', "Print build options" },
{"port", 'p', POPT_ARG_STRING, &ports, 0, "Listen on the specified ports"},
POPT_COMMON_SAMBA
{ NULL }
};
#ifdef HAVE_SET_AUTH_PARAMETERS
set_auth_parameters(argc,argv);
#endif
pc = poptGetContext("smbd", argc, argv, long_options, 0);
while((opt = poptGetNextOpt(pc)) != -1) {
switch (opt) {
case 'b':
build_options(True);
exit(0);
break;
}
}
poptFreeContext(pc);
#ifdef HAVE_SETLUID
setluid(0);
#endif
sec_init();
load_case_tables();
set_remote_machine_name("smbd", False);
if (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);
}
setup_logging(argv[0],log_stdout);
generate_random_buffer(NULL, 0);
gain_root_privilege();
gain_root_group_privilege();
fault_setup((void (*)(void *))exit_server);
CatchSignal(SIGTERM , SIGNAL_CAST sig_term);
CatchSignal(SIGHUP,SIGNAL_CAST sig_hup);
BlockSignals(True,SIGPIPE);
#if defined(SIGFPE)
BlockSignals(True,SIGFPE);
#endif
#if defined(SIGUSR2)
BlockSignals(True,SIGUSR2);
#endif
BlockSignals(False, SIGHUP);
BlockSignals(False, SIGUSR1);
BlockSignals(False, SIGTERM);
umask(0);
init_sec_ctx();
reopen_logs();
DEBUG(0,( "smbd version %s started.\n", SAMBA_VERSION_STRING));
DEBUGADD(0,( "Copyright Andrew Tridgell and the Samba Team 1992-2004\n"));
DEBUG(2,("uid=%d gid=%d euid=%d egid=%d\n",
(int)getuid(),(int)getgid(),(int)geteuid(),(int)getegid()));
build_options(False);
if (sizeof(uint16) < 2 || sizeof(uint32) < 4) {
DEBUG(0,("ERROR: Samba is not configured correctly for the word size on your machine\n"));
exit(1);
}
if (!reload_services(False))
return(-1);
init_structs();
#ifdef WITH_PROFILE
if (!profile_setup(False)) {
DEBUG(0,("ERROR: failed to setup profiling\n"));
return -1;
}
#endif
DEBUG(3,( "loaded services\n"));
if (!is_daemon && !is_a_socket(0)) {
if (!interactive)
DEBUG(0,("standard input is not a socket, assuming -D option\n"));
is_daemon = True;
}
if (is_daemon && !interactive) {
DEBUG( 3, ( "Becoming a daemon.\n" ) );
become_daemon(Fork);
}
#if HAVE_SETPGID
if (interactive)
setpgid( (pid_t)0, (pid_t)0);
#endif
if (!directory_exist(lp_lockdir(), NULL))
mkdir(lp_lockdir(), 0755);
if (is_daemon)
pidfile_create("smbd");
if (!message_init())
exit(1);
if (!session_init())
exit(1);
if (conn_tdb_ctx() == NULL)
exit(1);
if (!locking_init(0))
exit(1);
if (!share_info_db_init())
exit(1);
namecache_enable();
if (!init_registry())
exit(1);
if (!print_backend_init())
exit(1);
claim_connection(NULL,"",0,True,FLAG_MSG_GENERAL|FLAG_MSG_SMBD);
if ( is_daemon )
start_background_queue();
if (!open_sockets_smbd(is_daemon, interactive, ports))
exit(1);
if(!initialize_password_db(False))
exit(1);
if(!get_global_sam_sid()) {
DEBUG(0,("ERROR: Samba cannot create a SAM SID.\n"));
exit(1);
}
static_init_rpc;
init_modules();
reload_services(True);
if (!init_account_policy()) {
DEBUG(0,("Could not open account policy tdb.\n"));
exit(1);
}
if (*lp_rootdir()) {
if (sys_chroot(lp_rootdir()) == 0)
DEBUG(2,("Changed root to %s\n", lp_rootdir()));
}
if (!init_oplocks())
exit(1);
if (!init_change_notify())
exit(1);
TimeInit();
message_register(MSG_SMB_FORCE_TDIS, msg_force_tdis);
smbd_process();
namecache_shutdown();
if (interactive) {
TALLOC_CTX *mem_ctx = talloc_init("end_description");
char *description = talloc_describe_all(mem_ctx);
DEBUG(3, ("tallocs left:\n%s\n", description));
talloc_destroy(mem_ctx);
}
exit_server("normal exit");
return(0);
}