#include "includes.h"
#ifdef HAVE_VPROC_H
#include <vproc.h>
static vproc_transaction_t smbd_vproc_trans;
static void smbd_vproc_start(void)
{
smbd_vproc_trans = vproc_transaction_begin(NULL);
}
static void smbd_vproc_end(void)
{
vproc_transaction_end(NULL, smbd_vproc_trans);
}
#else
#define smbd_vproc_start()
#define smbd_vproc_end()
#endif
static_decl_rpc;
static int am_parent = 1;
int last_message = -1;
#define LAST_MESSAGE() (last_message != -1 ? smb_fn_name(last_message) : "")
extern struct auth_context *negprot_global_auth_context;
extern pstring user_socket_options;
extern SIG_ATOMIC_T got_sig_term;
extern SIG_ATOMIC_T reload_after_sighup;
static SIG_ATOMIC_T got_sig_cld;
#ifdef WITH_DFS
extern int dcelogin_atmost_once;
#endif
static int server_fd = -1;
int smbd_server_fd(void)
{
return server_fd;
}
static void smbd_set_server_fd(int fd)
{
server_fd = fd;
client_setfd(fd);
}
struct event_context *smbd_event_context(void)
{
static struct event_context *ctx;
if (!ctx && !(ctx = event_context_init(NULL))) {
smb_panic("Could not init smbd event context\n");
}
return ctx;
}
struct messaging_context *smbd_messaging_context(void)
{
static struct messaging_context *ctx;
if (!ctx && !(ctx = messaging_init(NULL, server_id_self(),
smbd_event_context()))) {
smb_panic("Could not init smbd messaging context\n");
}
return ctx;
}
static void smb_conf_updated(int msg_type, struct process_id src,
void *buf, size_t len, void *private_data)
{
DEBUG(10,("smb_conf_updated: Got message saying smb.conf was updated. Reloading.\n"));
reload_services(False);
}
static void smb_stat_cache_delete(int msg_type, struct process_id src,
void *buf, size_t len, void *private_data)
{
const char *name = (const char *)buf;
DEBUG(10,("smb_stat_cache_delete: delete name %s\n", name));
stat_cache_delete(name);
}
static void sig_term(void)
{
got_sig_term = 1;
sys_select_signal(SIGTERM);
}
static void sig_hup(int sig)
{
reload_after_sighup = 1;
sys_select_signal(SIGHUP);
}
static void sig_cld(int sig)
{
got_sig_cld = 1;
sys_select_signal(SIGCLD);
}
static void killkids(void)
{
if(am_parent) kill(0,SIGTERM);
}
static void msg_sam_sync(int UNUSED(msg_type), struct process_id UNUSED(pid),
void *UNUSED(buf), size_t UNUSED(len),
void *private_data)
{
DEBUG(10, ("** sam sync message received, ignoring\n"));
}
static void msg_sam_repl(int msg_type, struct process_id pid,
void *buf, size_t len, void *private_data)
{
uint32 low_serial;
if (len != sizeof(uint32))
return;
low_serial = *((uint32 *)buf);
DEBUG(3, ("received sam replication message, serial = 0x%04x\n",
low_serial));
}
#ifdef WITH_DARWIN_STATS
static void msg_usr_stat(int msg_type, struct process_id src,
void *buf, size_t len, void * UNUSED(context))
{
if (len != 0) {
DEBUG(10,("Received NON 0 USTAT message from PID %u \n", src));
return;
} else {
pid_t pid = procid_to_pid(&src);
u_int64_t data[2] = {user_op_count, user_byte_count};
DEBUG(1,("USTAT message send to PID %u for size %d\n",
(unsigned)pid, sizeof(u_int64_t) * 2));
become_root();
message_send_pid(src, msg_type, &data[0],
sizeof(u_int64_t) * 2, False);
unbecome_root();
}
}
#endif
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);
return True;
}
static void msg_exit_server(int msg_type, struct process_id src,
void *buf, size_t len, void *private_data)
{
DEBUG(3, ("got a SHUTDOWN message\n"));
exit_server_cleanly(NULL);
}
#ifdef DEVELOPER
static void msg_inject_fault(int msg_type, struct process_id src,
void *buf, size_t len, void *private_data)
{
int sig;
if (len != sizeof(int)) {
DEBUG(0, ("Process %llu sent bogus signal injection request\n",
(unsigned long long)src.pid));
return;
}
sig = *(int *)buf;
if (sig == -1) {
exit_server("internal error injected");
return;
}
#if HAVE_STRSIGNAL
DEBUG(0, ("Process %llu requested injection of signal %d (%s)\n",
(unsigned long long)src.pid, sig, strsignal(sig)));
#else
DEBUG(0, ("Process %llu requested injection of signal %d\n",
(unsigned long long)src.pid, sig));
#endif
kill(sys_getpid(), sig);
}
#endif
struct child_pid {
struct child_pid *prev, *next;
pid_t pid;
};
static struct child_pid *children;
static int num_children;
static void add_child_pid(pid_t pid)
{
struct child_pid *child;
if (lp_max_smbd_processes() == 0) {
return;
}
child = SMB_MALLOC_P(struct child_pid);
if (child == NULL) {
DEBUG(0, ("Could not add child struct -- malloc failed\n"));
return;
}
child->pid = pid;
DLIST_ADD(children, child);
num_children += 1;
}
static void remove_child_pid(pid_t pid)
{
struct child_pid *child;
if (lp_max_smbd_processes() == 0) {
return;
}
for (child = children; child != NULL; child = child->next) {
if (child->pid == pid) {
struct child_pid *tmp = child;
DLIST_REMOVE(children, child);
SAFE_FREE(tmp);
num_children -= 1;
return;
}
}
DEBUG(0, ("Could not find child %d -- ignoring\n", (int)pid));
}
static BOOL allowable_number_of_smbd_processes(void)
{
int max_processes = lp_max_smbd_processes();
if (!max_processes)
return True;
return num_children < max_processes;
}
static BOOL open_sockets_smbd(enum smb_server_mode server_mode, const char *smb_ports)
{
int num_sockets = 0;
int fd_listenset[FD_SETSIZE];
fd_set listen_set;
int s;
int maxfd = 0;
int i;
struct timeval idle_timeout = {0, 0};
if (server_mode == SERVER_MODE_INETD) {
return open_sockets_inetd();
}
#ifdef HAVE_ATEXIT
{
static int atexit_set;
if(atexit_set == 0) {
atexit_set=1;
atexit(killkids);
}
}
#endif
CatchSignal(SIGCLD, sig_cld);
FD_ZERO(&listen_set);
num_sockets = smbd_sockinit(smb_ports, fd_listenset, &idle_timeout);
if (num_sockets == 0) {
return False;
}
for (i = 0; i < num_sockets; ++i) {
FD_SET(fd_listenset[i], &listen_set);
maxfd = MAX(maxfd, fd_listenset[i]);
}
message_register(MSG_SMB_SAM_SYNC, msg_sam_sync, NULL);
message_register(MSG_SMB_SAM_REPL, msg_sam_repl, NULL);
message_register(MSG_SHUTDOWN, msg_exit_server, NULL);
message_register(MSG_SMB_FILE_RENAME, msg_file_was_renamed, NULL);
message_register(MSG_SMB_CONF_UPDATED, smb_conf_updated, NULL);
message_register(MSG_SMB_STAT_CACHE_DELETE, smb_stat_cache_delete,
NULL);
#ifdef DEVELOPER
message_register(MSG_SMB_INJECT_FAULT, msg_inject_fault, NULL);
#endif
DEBUG(2,("waiting for a connection\n"));
while (1) {
fd_set lfds;
int num;
lp_TALLOC_FREE();
message_dispatch();
if (got_sig_cld) {
pid_t pid;
got_sig_cld = False;
while ((pid = sys_waitpid(-1, NULL, WNOHANG)) > 0) {
remove_child_pid(pid);
}
}
memcpy((char *)&lfds, (char *)&listen_set,
sizeof(listen_set));
smbd_vproc_end();
num = sys_select(maxfd+1,&lfds,NULL,NULL,
idle_timeout.tv_sec ? &idle_timeout : NULL);
{
int errsav = errno;
smbd_vproc_start();
errno = errsav;
}
if (num == -1 && errno == EINTR) {
if (got_sig_term) {
exit_server_cleanly(NULL);
}
if (reload_after_sighup) {
change_to_root_user();
DEBUG(1,("Reloading services after SIGHUP\n"));
reload_services(False);
reload_after_sighup = 0;
}
continue;
}
if (num == 0 && count_all_current_connections() == 0) {
exit_server_cleanly("idle timeout");
}
check_reload(time(NULL));
for( ; num > 0; num--) {
struct sockaddr addr;
socklen_t in_addrlen = sizeof(addr);
pid_t child = 0;
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 (server_mode == SERVER_MODE_INTERACTIVE) {
return True;
}
if (allowable_number_of_smbd_processes() &&
smbd_server_fd() != -1 &&
((child = sys_fork())==0)) {
CatchChild();
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) == -1) {
DEBUG(0,("tdb_reopen_all failed.\n"));
smb_panic("tdb_reopen_all failed.");
}
return True;
}
close(smbd_server_fd());
smbd_set_server_fd(-1);
if (child != 0) {
add_child_pid(child);
}
force_check_log_size();
}
}
}
void reload_printers(void)
{
int snum;
int n_services = lp_numservices();
int pnum = lp_servicenumber(PRINTERS_NAME);
const char *pname;
pcap_cache_reload();
for (snum = 0; snum < n_services; snum++) {
if (snum == pnum || !(lp_snum_ok(snum) && lp_print_ok(snum) &&
lp_autoloaded(snum)))
continue;
pname = lp_printername(snum);
if (!pcap_printername_ok(pname)) {
DEBUG(3, ("removing stale printer %s\n", pname));
if (is_printer_published(NULL, snum, NULL))
nt_printer_publish(NULL, snum, SPOOL_DS_UNPUBLISH);
del_a_printer(pname);
lp_killservice(snum);
}
}
load_printers();
}
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, True);
reload_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);
}
enum server_exit_reason { SERVER_EXIT_NORMAL, SERVER_EXIT_ABNORMAL };
static void exit_server_common(enum server_exit_reason how,
const char *const reason) NORETURN_ATTRIBUTE;
static void exit_server_common(enum server_exit_reason how,
const char *const reason)
{
static int firsttime=1;
if (!firsttime)
exit(0);
firsttime = 0;
change_to_root_user();
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);
yield_connection(NULL,"");
respond_to_all_remaining_local_messages();
#ifdef WITH_DFS
if (dcelogin_atmost_once) {
dfs_unlogin();
}
#endif
locking_end();
printing_end();
if (how != SERVER_EXIT_NORMAL) {
int oldlevel = DEBUGLEVEL;
char *last_inbuf = get_InBuffer();
DEBUGLEVEL = 10;
DEBUGSEP(0);
DEBUG(0,("Abnormal server exit: %s\n",
reason ? reason : "no explanation provided"));
DEBUGSEP(0);
log_stack_trace();
if (last_inbuf) {
DEBUG(0,("Last message was %s\n", LAST_MESSAGE()));
show_msg(last_inbuf);
}
DEBUGLEVEL = oldlevel;
dump_core();
} else {
DEBUG(3,("Server exit (%s)\n",
(reason ? reason : "normal exit")));
}
exit(0);
}
void exit_server(const char *const explanation)
{
exit_server_common(SERVER_EXIT_ABNORMAL, explanation);
}
void exit_server_cleanly(const char *const explanation)
{
exit_server_common(SERVER_EXIT_NORMAL, explanation);
}
void exit_server_fault(void)
{
exit_server("critical server fault");
}
static BOOL init_structs(void )
{
if (!init_names())
return False;
conn_init();
file_init();
init_rpc_pipe_hnd();
init_dptrs();
secrets_init();
return True;
}
extern void build_options(BOOL screen);
int main(int argc,const char *argv[])
{
static BOOL no_process_group = False;
static BOOL log_stdout = False;
static char *ports = NULL;
static char *profile_level = NULL;
int opt;
poptContext pc;
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" },
{"build-options", 'b', POPT_ARG_NONE, NULL, 'b', "Print build options" },
{"port", 'p', POPT_ARG_STRING, &ports, 0, "Listen on the specified ports"},
{"profiling-level", 'P', POPT_ARG_STRING, &profile_level, 0, "Set profiling level","PROFILE_LEVEL"},
POPT_COMMON_SAMBA
POPT_COMMON_DYNCONFIG
POPT_TABLEEND
};
smbd_vproc_start();
load_case_tables();
TimeInit();
#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();
set_remote_machine_name("smbd", False);
if (server_mode == SERVER_MODE_INTERACTIVE) {
log_stdout = True;
if (DEBUGLEVEL >= 9) {
talloc_enable_leak_report();
}
}
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);
generate_random_buffer(NULL, 0);
gain_root_privilege();
gain_root_group_privilege();
fault_setup((void (*)(void *))exit_server_fault);
dump_core_setup("smbd");
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, ( "%s\n", COPYRIGHT_STARTUP_MESSAGE ) );
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;
}
if (profile_level != NULL) {
int pl = atoi(profile_level);
struct process_id src;
DEBUG(1, ("setting profiling level: %s\n",profile_level));
src.pid = getpid();
set_profile_level(pl, src);
}
#endif
#ifdef WITH_DARWIN_STATS
if (!service_stats_setup(False)) {
DEBUG(0,("ERROR: failed to setup service stats\n"));
return -1;
}
#endif
DEBUG(3,( "loaded services\n"));
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( 3, ( "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
if (!directory_exist(lp_lockdir(), NULL))
mkdir(lp_lockdir(), 0755);
if (server_mode != SERVER_MODE_INETD &&
server_mode != SERVER_MODE_INTERACTIVE) {
pidfile_create("smbd");
}
if (!message_init())
exit(1);
#ifdef WITH_DARWIN_STATS
message_register(MSG_USR_STATS, msg_usr_stat, NULL);
#endif
if(!initialize_password_db(False))
exit(1);
if (!secrets_init()) {
DEBUG(0, ("ERROR: smbd can not open secrets.tdb\n"));
exit(1);
}
if(!get_global_sam_sid()) {
DEBUG(0,("ERROR: Samba cannot create a SAM SID.\n"));
exit(1);
}
if (!session_init())
exit(1);
if (conn_tdb_ctx() == NULL)
exit(1);
if (!locking_init(0))
exit(1);
namecache_enable();
if (!init_registry())
exit(1);
#if 0
if (!init_svcctl_db())
exit(1);
#endif
if (!print_backend_init())
exit(1);
if (!init_guest_info()) {
DEBUG(0,("ERROR: failed to setup guest info.\n"));
return -1;
}
claim_connection(NULL,"",0,True,FLAG_MSG_GENERAL|FLAG_MSG_SMBD);
if (server_mode != SERVER_MODE_INETD &&
server_mode != SERVER_MODE_INTERACTIVE) {
start_background_queue();
}
dmapi_init_session();
if (!open_sockets_smbd(server_mode, ports)) {
exit(1);
}
static_init_rpc;
init_modules();
if (server_mode != SERVER_MODE_INETD &&
server_mode != SERVER_MODE_INTERACTIVE) {
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);
initialize_async_io_handler();
message_register(MSG_SMB_FORCE_TDIS, msg_force_tdis, NULL);
smbd_process();
namecache_shutdown();
exit_server_cleanly(NULL);
return(0);
}