s3:smbd: make mdns registration event driven.
[metze/samba/wip.git] / source3 / smbd / server.c
1 /*
2    Unix SMB/CIFS implementation.
3    Main SMB server routines
4    Copyright (C) Andrew Tridgell                1992-1998
5    Copyright (C) Martin Pool                    2002
6    Copyright (C) Jelmer Vernooij                2002-2003
7    Copyright (C) Volker Lendecke                1993-2007
8    Copyright (C) Jeremy Allison                 1993-2007
9
10    This program is free software; you can redistribute it and/or modify
11    it under the terms of the GNU General Public License as published by
12    the Free Software Foundation; either version 3 of the License, or
13    (at your option) any later version.
14
15    This program is distributed in the hope that it will be useful,
16    but WITHOUT ANY WARRANTY; without even the implied warranty of
17    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18    GNU General Public License for more details.
19
20    You should have received a copy of the GNU General Public License
21    along with this program.  If not, see <http://www.gnu.org/licenses/>.
22 */
23
24 #include "includes.h"
25 #include "smbd/globals.h"
26
27 static_decl_rpc;
28
29 #ifdef WITH_DFS
30 extern int dcelogin_atmost_once;
31 #endif /* WITH_DFS */
32
33 int smbd_server_fd(void)
34 {
35         return server_fd;
36 }
37
38 static void smbd_set_server_fd(int fd)
39 {
40         server_fd = fd;
41 }
42
43 int get_client_fd(void)
44 {
45         return server_fd;
46 }
47
48 struct event_context *smbd_event_context(void)
49 {
50         if (!smbd_event_ctx) {
51                 smbd_event_ctx = event_context_init(talloc_autofree_context());
52         }
53         if (!smbd_event_ctx) {
54                 smb_panic("Could not init smbd event context");
55         }
56         return smbd_event_ctx;
57 }
58
59 struct messaging_context *smbd_messaging_context(void)
60 {
61         if (smbd_msg_ctx == NULL) {
62                 smbd_msg_ctx = messaging_init(talloc_autofree_context(),
63                                               server_id_self(),
64                                               smbd_event_context());
65         }
66         if (smbd_msg_ctx == NULL) {
67                 DEBUG(0, ("Could not init smbd messaging context.\n"));
68         }
69         return smbd_msg_ctx;
70 }
71
72 struct memcache *smbd_memcache(void)
73 {
74         if (!smbd_memcache_ctx) {
75                 smbd_memcache_ctx = memcache_init(talloc_autofree_context(),
76                                                   lp_max_stat_cache_size()*1024);
77         }
78         if (!smbd_memcache_ctx) {
79                 smb_panic("Could not init smbd memcache");
80         }
81
82         return smbd_memcache_ctx;
83 }
84
85 /*******************************************************************
86  What to do when smb.conf is updated.
87  ********************************************************************/
88
89 static void smb_conf_updated(struct messaging_context *msg,
90                              void *private_data,
91                              uint32_t msg_type,
92                              struct server_id server_id,
93                              DATA_BLOB *data)
94 {
95         DEBUG(10,("smb_conf_updated: Got message saying smb.conf was "
96                   "updated. Reloading.\n"));
97         reload_services(False);
98 }
99
100
101 /*******************************************************************
102  Delete a statcache entry.
103  ********************************************************************/
104
105 static void smb_stat_cache_delete(struct messaging_context *msg,
106                                   void *private_data,
107                                   uint32_t msg_tnype,
108                                   struct server_id server_id,
109                                   DATA_BLOB *data)
110 {
111         const char *name = (const char *)data->data;
112         DEBUG(10,("smb_stat_cache_delete: delete name %s\n", name));
113         stat_cache_delete(name);
114 }
115
116 /****************************************************************************
117   Send a SIGTERM to our process group.
118 *****************************************************************************/
119
120 static void  killkids(void)
121 {
122         if(am_parent) kill(0,SIGTERM);
123 }
124
125 /****************************************************************************
126  Process a sam sync message - not sure whether to do this here or
127  somewhere else.
128 ****************************************************************************/
129
130 static void msg_sam_sync(struct messaging_context *msg,
131                          void *private_data,
132                          uint32_t msg_type,
133                          struct server_id server_id,
134                          DATA_BLOB *data)
135 {
136         DEBUG(10, ("** sam sync message received, ignoring\n"));
137 }
138
139 static void msg_exit_server(struct messaging_context *msg,
140                             void *private_data,
141                             uint32_t msg_type,
142                             struct server_id server_id,
143                             DATA_BLOB *data)
144 {
145         DEBUG(3, ("got a SHUTDOWN message\n"));
146         exit_server_cleanly(NULL);
147 }
148
149 #ifdef DEVELOPER
150 static void msg_inject_fault(struct messaging_context *msg,
151                              void *private_data,
152                              uint32_t msg_type,
153                              struct server_id src,
154                              DATA_BLOB *data)
155 {
156         int sig;
157
158         if (data->length != sizeof(sig)) {
159                 
160                 DEBUG(0, ("Process %s sent bogus signal injection request\n",
161                           procid_str_static(&src)));
162                 return;
163         }
164
165         sig = *(int *)data->data;
166         if (sig == -1) {
167                 exit_server("internal error injected");
168                 return;
169         }
170
171 #if HAVE_STRSIGNAL
172         DEBUG(0, ("Process %s requested injection of signal %d (%s)\n",
173                   procid_str_static(&src), sig, strsignal(sig)));
174 #else
175         DEBUG(0, ("Process %s requested injection of signal %d\n",
176                   procid_str_static(&src), sig));
177 #endif
178
179         kill(sys_getpid(), sig);
180 }
181 #endif /* DEVELOPER */
182
183 struct child_pid {
184         struct child_pid *prev, *next;
185         pid_t pid;
186 };
187
188 static void add_child_pid(pid_t pid)
189 {
190         struct child_pid *child;
191
192         if (lp_max_smbd_processes() == 0) {
193                 /* Don't bother with the child list if we don't care anyway */
194                 return;
195         }
196
197         child = SMB_MALLOC_P(struct child_pid);
198         if (child == NULL) {
199                 DEBUG(0, ("Could not add child struct -- malloc failed\n"));
200                 return;
201         }
202         child->pid = pid;
203         DLIST_ADD(children, child);
204         num_children += 1;
205 }
206
207 static void remove_child_pid(pid_t pid, bool unclean_shutdown)
208 {
209         struct child_pid *child;
210
211         if (unclean_shutdown) {
212                 /* a child terminated uncleanly so tickle all processes to see 
213                    if they can grab any of the pending locks
214                 */
215                 DEBUG(3,(__location__ " Unclean shutdown of pid %u\n", pid));
216                 messaging_send_buf(smbd_messaging_context(), procid_self(), 
217                                    MSG_SMB_BRL_VALIDATE, NULL, 0);
218                 message_send_all(smbd_messaging_context(), 
219                                  MSG_SMB_UNLOCK, NULL, 0, NULL);
220         }
221
222         if (lp_max_smbd_processes() == 0) {
223                 /* Don't bother with the child list if we don't care anyway */
224                 return;
225         }
226
227         for (child = children; child != NULL; child = child->next) {
228                 if (child->pid == pid) {
229                         struct child_pid *tmp = child;
230                         DLIST_REMOVE(children, child);
231                         SAFE_FREE(tmp);
232                         num_children -= 1;
233                         return;
234                 }
235         }
236
237         DEBUG(0, ("Could not find child %d -- ignoring\n", (int)pid));
238 }
239
240 /****************************************************************************
241  Have we reached the process limit ?
242 ****************************************************************************/
243
244 static bool allowable_number_of_smbd_processes(void)
245 {
246         int max_processes = lp_max_smbd_processes();
247
248         if (!max_processes)
249                 return True;
250
251         return num_children < max_processes;
252 }
253
254 static void smbd_sig_chld_handler(struct tevent_context *ev,
255                                   struct tevent_signal *se,
256                                   int signum,
257                                   int count,
258                                   void *siginfo,
259                                   void *private_data)
260 {
261         pid_t pid;
262         int status;
263
264         while ((pid = sys_waitpid(-1, &status, WNOHANG)) > 0) {
265                 bool unclean_shutdown = False;
266
267                 /* If the child terminated normally, assume
268                    it was an unclean shutdown unless the
269                    status is 0
270                 */
271                 if (WIFEXITED(status)) {
272                         unclean_shutdown = WEXITSTATUS(status);
273                 }
274                 /* If the child terminated due to a signal
275                    we always assume it was unclean.
276                 */
277                 if (WIFSIGNALED(status)) {
278                         unclean_shutdown = True;
279                 }
280                 remove_child_pid(pid, unclean_shutdown);
281         }
282 }
283
284 static void smbd_setup_sig_chld_handler(void)
285 {
286         struct tevent_signal *se;
287
288         se = tevent_add_signal(smbd_event_context(),
289                                smbd_event_context(),
290                                SIGCHLD, 0,
291                                smbd_sig_chld_handler,
292                                NULL);
293         if (!se) {
294                 exit_server("failed to setup SIGCHLD handler");
295         }
296 }
297
298 /****************************************************************************
299  Open the socket communication.
300 ****************************************************************************/
301
302 static bool open_sockets_smbd(bool interactive, const char *smb_ports)
303 {
304         int num_interfaces = iface_count();
305         int num_sockets = 0;
306         int fd_listenset[FD_SETSIZE];
307         fd_set listen_set;
308         int s;
309         int maxfd = 0;
310         int i;
311         char *ports;
312         TALLOC_CTX *dns_ctx = NULL;
313         unsigned dns_port = 0;
314
315 #ifdef HAVE_ATEXIT
316         atexit(killkids);
317 #endif
318
319         dns_ctx = talloc_new(NULL);
320
321         /* Stop zombies */
322         smbd_setup_sig_chld_handler();
323
324         FD_ZERO(&listen_set);
325
326         /* use a reasonable default set of ports - listing on 445 and 139 */
327         if (!smb_ports) {
328                 ports = lp_smb_ports();
329                 if (!ports || !*ports) {
330                         ports = smb_xstrdup(SMB_PORTS);
331                 } else {
332                         ports = smb_xstrdup(ports);
333                 }
334         } else {
335                 ports = smb_xstrdup(smb_ports);
336         }
337
338         if (lp_interfaces() && lp_bind_interfaces_only()) {
339                 /* We have been given an interfaces line, and been
340                    told to only bind to those interfaces. Create a
341                    socket per interface and bind to only these.
342                 */
343
344                 /* Now open a listen socket for each of the
345                    interfaces. */
346                 for(i = 0; i < num_interfaces; i++) {
347                         TALLOC_CTX *frame = NULL;
348                         const struct sockaddr_storage *ifss =
349                                         iface_n_sockaddr_storage(i);
350                         char *tok;
351                         const char *ptr;
352
353                         if (ifss == NULL) {
354                                 DEBUG(0,("open_sockets_smbd: "
355                                         "interface %d has NULL IP address !\n",
356                                         i));
357                                 continue;
358                         }
359
360                         frame = talloc_stackframe();
361                         for (ptr=ports;
362                                         next_token_talloc(frame,&ptr, &tok, " \t,");) {
363                                 unsigned port = atoi(tok);
364                                 if (port == 0 || port > 0xffff) {
365                                         continue;
366                                 }
367
368                                 /* Keep the first port for mDNS service
369                                  * registration.
370                                  */
371                                 if (dns_port == 0) {
372                                         dns_port = port;
373                                 }
374
375                                 s = fd_listenset[num_sockets] =
376                                         open_socket_in(SOCK_STREAM,
377                                                         port,
378                                                         num_sockets == 0 ? 0 : 2,
379                                                         ifss,
380                                                         true);
381                                 if(s == -1) {
382                                         continue;
383                                 }
384
385                                 /* ready to listen */
386                                 set_socket_options(s,"SO_KEEPALIVE");
387                                 set_socket_options(s,lp_socket_options());
388
389                                 /* Set server socket to
390                                  * non-blocking for the accept. */
391                                 set_blocking(s,False);
392
393                                 if (listen(s, SMBD_LISTEN_BACKLOG) == -1) {
394                                         DEBUG(0,("open_sockets_smbd: listen: "
395                                                 "%s\n", strerror(errno)));
396                                         close(s);
397                                         TALLOC_FREE(frame);
398                                         return False;
399                                 }
400                                 FD_SET(s,&listen_set);
401                                 maxfd = MAX( maxfd, s);
402
403                                 num_sockets++;
404                                 if (num_sockets >= FD_SETSIZE) {
405                                         DEBUG(0,("open_sockets_smbd: Too "
406                                                 "many sockets to bind to\n"));
407                                         TALLOC_FREE(frame);
408                                         return False;
409                                 }
410                         }
411                         TALLOC_FREE(frame);
412                 }
413         } else {
414                 /* Just bind to 0.0.0.0 - accept connections
415                    from anywhere. */
416
417                 TALLOC_CTX *frame = talloc_stackframe();
418                 char *tok;
419                 const char *ptr;
420                 const char *sock_addr = lp_socket_address();
421                 char *sock_tok;
422                 const char *sock_ptr;
423
424                 if (strequal(sock_addr, "0.0.0.0") ||
425                     strequal(sock_addr, "::")) {
426 #if HAVE_IPV6
427                         sock_addr = "::,0.0.0.0";
428 #else
429                         sock_addr = "0.0.0.0";
430 #endif
431                 }
432
433                 for (sock_ptr=sock_addr;
434                                 next_token_talloc(frame, &sock_ptr, &sock_tok, " \t,"); ) {
435                         for (ptr=ports; next_token_talloc(frame, &ptr, &tok, " \t,"); ) {
436                                 struct sockaddr_storage ss;
437
438                                 unsigned port = atoi(tok);
439                                 if (port == 0 || port > 0xffff) {
440                                         continue;
441                                 }
442
443                                 /* Keep the first port for mDNS service
444                                  * registration.
445                                  */
446                                 if (dns_port == 0) {
447                                         dns_port = port;
448                                 }
449
450                                 /* open an incoming socket */
451                                 if (!interpret_string_addr(&ss, sock_tok,
452                                                 AI_NUMERICHOST|AI_PASSIVE)) {
453                                         continue;
454                                 }
455
456                                 s = open_socket_in(SOCK_STREAM,
457                                                 port,
458                                                 num_sockets == 0 ? 0 : 2,
459                                                 &ss,
460                                                 true);
461                                 if (s == -1) {
462                                         continue;
463                                 }
464
465                                 /* ready to listen */
466                                 set_socket_options(s,"SO_KEEPALIVE");
467                                 set_socket_options(s,lp_socket_options());
468
469                                 /* Set server socket to non-blocking
470                                  * for the accept. */
471                                 set_blocking(s,False);
472
473                                 if (listen(s, SMBD_LISTEN_BACKLOG) == -1) {
474                                         DEBUG(0,("open_sockets_smbd: "
475                                                 "listen: %s\n",
476                                                  strerror(errno)));
477                                         close(s);
478                                         TALLOC_FREE(frame);
479                                         return False;
480                                 }
481
482                                 fd_listenset[num_sockets] = s;
483                                 FD_SET(s,&listen_set);
484                                 maxfd = MAX( maxfd, s);
485
486                                 num_sockets++;
487
488                                 if (num_sockets >= FD_SETSIZE) {
489                                         DEBUG(0,("open_sockets_smbd: Too "
490                                                 "many sockets to bind to\n"));
491                                         TALLOC_FREE(frame);
492                                         return False;
493                                 }
494                         }
495                 }
496                 TALLOC_FREE(frame);
497         }
498
499         SAFE_FREE(ports);
500
501         if (num_sockets == 0) {
502                 DEBUG(0,("open_sockets_smbd: No "
503                         "sockets available to bind to.\n"));
504                 return false;
505         }
506
507         /* Setup the main smbd so that we can get messages. Note that
508            do this after starting listening. This is needed as when in
509            clustered mode, ctdb won't allow us to start doing database
510            operations until it has gone thru a full startup, which
511            includes checking to see that smbd is listening. */
512         claim_connection(NULL,"",
513                          FLAG_MSG_GENERAL|FLAG_MSG_SMBD|FLAG_MSG_DBWRAP);
514
515         /* Listen to messages */
516
517         messaging_register(smbd_messaging_context(), NULL,
518                            MSG_SMB_SAM_SYNC, msg_sam_sync);
519         messaging_register(smbd_messaging_context(), NULL,
520                            MSG_SHUTDOWN, msg_exit_server);
521         messaging_register(smbd_messaging_context(), NULL,
522                            MSG_SMB_FILE_RENAME, msg_file_was_renamed);
523         messaging_register(smbd_messaging_context(), NULL,
524                            MSG_SMB_CONF_UPDATED, smb_conf_updated);
525         messaging_register(smbd_messaging_context(), NULL,
526                            MSG_SMB_STAT_CACHE_DELETE, smb_stat_cache_delete);
527         brl_register_msgs(smbd_messaging_context());
528
529 #ifdef CLUSTER_SUPPORT
530         if (lp_clustering()) {
531                 ctdbd_register_reconfigure(messaging_ctdbd_connection());
532         }
533 #endif
534
535 #ifdef DEVELOPER
536         messaging_register(smbd_messaging_context(), NULL,
537                            MSG_SMB_INJECT_FAULT, msg_inject_fault);
538 #endif
539
540         if (dns_port != 0) {
541                 smbd_setup_mdns_registration(smbd_event_context(),
542                                              dns_ctx, dns_port);
543         }
544
545         /* now accept incoming connections - forking a new process
546            for each incoming connection */
547         DEBUG(2,("waiting for a connection\n"));
548         while (1) {
549                 struct timeval now, idle_timeout;
550                 fd_set r_fds, w_fds;
551                 int num;
552
553                 if (run_events(smbd_event_context(), 0, NULL, NULL)) {
554                         continue;
555                 }
556
557                 idle_timeout = timeval_zero();
558
559                 memcpy((char *)&r_fds, (char *)&listen_set,
560                        sizeof(listen_set));
561                 FD_ZERO(&w_fds);
562                 GetTimeOfDay(&now);
563
564                 event_add_to_select_args(smbd_event_context(), &now,
565                                          &r_fds, &w_fds, &idle_timeout,
566                                          &maxfd);
567
568                 num = sys_select(maxfd+1,&r_fds,&w_fds,NULL,
569                                  timeval_is_zero(&idle_timeout) ?
570                                  NULL : &idle_timeout);
571
572                 if (run_events(smbd_event_context(), num, &r_fds, &w_fds)) {
573                         continue;
574                 }
575
576                 /* If the idle timeout fired and we don't have any connected
577                  * users, exit gracefully. We should be running under a process
578                  * controller that will restart us if necessry.
579                  */
580                 if (num == 0 && count_all_current_connections() == 0) {
581                         exit_server_cleanly("idle timeout");
582                 }
583
584                 /* check if we need to reload services */
585                 check_reload(time(NULL));
586
587                 /* Find the sockets that are read-ready -
588                    accept on these. */
589                 for( ; num > 0; num--) {
590                         struct sockaddr addr;
591                         socklen_t in_addrlen = sizeof(addr);
592                         pid_t child = 0;
593
594                         s = -1;
595                         for(i = 0; i < num_sockets; i++) {
596                                 if(FD_ISSET(fd_listenset[i],&r_fds)) {
597                                         s = fd_listenset[i];
598                                         /* Clear this so we don't look
599                                            at it again. */
600                                         FD_CLR(fd_listenset[i],&r_fds);
601                                         break;
602                                 }
603                         }
604
605                         smbd_set_server_fd(accept(s,&addr,&in_addrlen));
606
607                         if (smbd_server_fd() == -1 && errno == EINTR)
608                                 continue;
609
610                         if (smbd_server_fd() == -1) {
611                                 DEBUG(2,("open_sockets_smbd: accept: %s\n",
612                                          strerror(errno)));
613                                 continue;
614                         }
615
616                         if (interactive)
617                                 return True;
618
619                         if (allowable_number_of_smbd_processes() &&
620                             ((child = sys_fork())==0)) {
621                                 /* Child code ... */
622
623                                 TALLOC_FREE(dns_ctx);
624
625                                 /* Stop zombies, the parent explicitly handles
626                                  * them, counting worker smbds. */
627                                 CatchChild();
628
629                                 /* close the listening socket(s) */
630                                 for(i = 0; i < num_sockets; i++)
631                                         close(fd_listenset[i]);
632
633                                 /* close our standard file
634                                    descriptors */
635                                 close_low_fds(False);
636                                 am_parent = 0;
637
638                                 if (!reinit_after_fork(
639                                             smbd_messaging_context(),
640                                             smbd_event_context(),
641                                             true)) {
642                                         DEBUG(0,("reinit_after_fork() failed\n"));
643                                         smb_panic("reinit_after_fork() failed");
644                                 }
645
646                                 smbd_setup_sig_term_handler();
647                                 smbd_setup_sig_hup_handler();
648
649                                 return True;
650                         }
651                         /* The parent doesn't need this socket */
652                         close(smbd_server_fd());
653
654                         /* Sun May 6 18:56:14 2001 ackley@cs.unm.edu:
655                                 Clear the closed fd info out of server_fd --
656                                 and more importantly, out of client_fd in
657                                 util_sock.c, to avoid a possible
658                                 getpeername failure if we reopen the logs
659                                 and use %I in the filename.
660                         */
661
662                         smbd_set_server_fd(-1);
663
664                         if (child != 0) {
665                                 add_child_pid(child);
666                         }
667
668                         /* Force parent to check log size after
669                          * spawning child.  Fix from
670                          * klausr@ITAP.Physik.Uni-Stuttgart.De.  The
671                          * parent smbd will log to logserver.smb.  It
672                          * writes only two messages for each child
673                          * started/finished. But each child writes,
674                          * say, 50 messages also in logserver.smb,
675                          * begining with the debug_count of the
676                          * parent, before the child opens its own log
677                          * file logserver.client. In a worst case
678                          * scenario the size of logserver.smb would be
679                          * checked after about 50*50=2500 messages
680                          * (ca. 100kb).
681                          * */
682                         force_check_log_size();
683
684                 } /* end for num */
685         } /* end while 1 */
686
687 /* NOTREACHED   return True; */
688 }
689
690 /****************************************************************************
691  Reload printers
692 **************************************************************************/
693 void reload_printers(void)
694 {
695         int snum;
696         int n_services = lp_numservices();
697         int pnum = lp_servicenumber(PRINTERS_NAME);
698         const char *pname;
699
700         pcap_cache_reload();
701
702         /* remove stale printers */
703         for (snum = 0; snum < n_services; snum++) {
704                 /* avoid removing PRINTERS_NAME or non-autoloaded printers */
705                 if (snum == pnum || !(lp_snum_ok(snum) && lp_print_ok(snum) &&
706                                       lp_autoloaded(snum)))
707                         continue;
708
709                 pname = lp_printername(snum);
710                 if (!pcap_printername_ok(pname)) {
711                         DEBUG(3, ("removing stale printer %s\n", pname));
712
713                         if (is_printer_published(NULL, snum, NULL))
714                                 nt_printer_publish(NULL, snum, SPOOL_DS_UNPUBLISH);
715                         del_a_printer(pname);
716                         lp_killservice(snum);
717                 }
718         }
719
720         load_printers();
721 }
722
723 /****************************************************************************
724  Reload the services file.
725 **************************************************************************/
726
727 bool reload_services(bool test)
728 {
729         bool ret;
730
731         if (lp_loaded()) {
732                 char *fname = lp_configfile();
733                 if (file_exist(fname) &&
734                     !strcsequal(fname, get_dyn_CONFIGFILE())) {
735                         set_dyn_CONFIGFILE(fname);
736                         test = False;
737                 }
738         }
739
740         reopen_logs();
741
742         if (test && !lp_file_list_changed())
743                 return(True);
744
745         lp_killunused(conn_snum_used);
746
747         ret = lp_load(get_dyn_CONFIGFILE(), False, False, True, True);
748
749         reload_printers();
750
751         /* perhaps the config filename is now set */
752         if (!test)
753                 reload_services(True);
754
755         reopen_logs();
756
757         load_interfaces();
758
759         if (smbd_server_fd() != -1) {
760                 set_socket_options(smbd_server_fd(),"SO_KEEPALIVE");
761                 set_socket_options(smbd_server_fd(), lp_socket_options());
762         }
763
764         mangle_reset_cache();
765         reset_stat_cache();
766
767         /* this forces service parameters to be flushed */
768         set_current_service(NULL,0,True);
769
770         return(ret);
771 }
772
773 /****************************************************************************
774  Exit the server.
775 ****************************************************************************/
776
777 /* Reasons for shutting down a server process. */
778 enum server_exit_reason { SERVER_EXIT_NORMAL, SERVER_EXIT_ABNORMAL };
779
780 static void exit_server_common(enum server_exit_reason how,
781         const char *const reason) _NORETURN_;
782
783 static void exit_server_common(enum server_exit_reason how,
784         const char *const reason)
785 {
786         bool had_open_conn;
787
788         if (!exit_firsttime)
789                 exit(0);
790         exit_firsttime = false;
791
792         change_to_root_user();
793
794         if (negprot_global_auth_context) {
795                 (negprot_global_auth_context->free)(&negprot_global_auth_context);
796         }
797
798         had_open_conn = conn_close_all();
799
800         invalidate_all_vuids();
801
802         /* 3 second timeout. */
803         print_notify_send_messages(smbd_messaging_context(), 3);
804
805         /* delete our entry in the connections database. */
806         yield_connection(NULL,"");
807
808 #ifdef WITH_DFS
809         if (dcelogin_atmost_once) {
810                 dfs_unlogin();
811         }
812 #endif
813
814 #ifdef USE_DMAPI
815         /* Destroy Samba DMAPI session only if we are master smbd process */
816         if (am_parent) {
817                 if (!dmapi_destroy_session()) {
818                         DEBUG(0,("Unable to close Samba DMAPI session\n"));
819                 }
820         }
821 #endif
822
823         locking_end();
824         printing_end();
825
826         if (how != SERVER_EXIT_NORMAL) {
827                 int oldlevel = DEBUGLEVEL;
828
829                 DEBUGLEVEL = 10;
830
831                 DEBUGSEP(0);
832                 DEBUG(0,("Abnormal server exit: %s\n",
833                         reason ? reason : "no explanation provided"));
834                 DEBUGSEP(0);
835
836                 log_stack_trace();
837
838                 DEBUGLEVEL = oldlevel;
839                 dump_core();
840
841         } else {    
842                 DEBUG(3,("Server exit (%s)\n",
843                         (reason ? reason : "normal exit")));
844         }
845
846         /* if we had any open SMB connections when we exited then we
847            need to tell the parent smbd so that it can trigger a retry
848            of any locks we may have been holding or open files we were
849            blocking */
850         if (had_open_conn) {
851                 exit(1);
852         } else {
853                 exit(0);
854         }
855 }
856
857 void exit_server(const char *const explanation)
858 {
859         exit_server_common(SERVER_EXIT_ABNORMAL, explanation);
860 }
861
862 void exit_server_cleanly(const char *const explanation)
863 {
864         exit_server_common(SERVER_EXIT_NORMAL, explanation);
865 }
866
867 void exit_server_fault(void)
868 {
869         exit_server("critical server fault");
870 }
871
872 /****************************************************************************
873  Initialise connect, service and file structs.
874 ****************************************************************************/
875
876 static bool init_structs(void )
877 {
878         /*
879          * Set the machine NETBIOS name if not already
880          * set from the config file.
881          */
882
883         if (!init_names())
884                 return False;
885
886         conn_init();
887
888         file_init();
889
890         init_dptrs();
891
892         if (!secrets_init())
893                 return False;
894
895         return True;
896 }
897
898 /****************************************************************************
899  main program.
900 ****************************************************************************/
901
902 /* Declare prototype for build_options() to avoid having to run it through
903    mkproto.h.  Mixing $(builddir) and $(srcdir) source files in the current
904    prototype generation system is too complicated. */
905
906 extern void build_options(bool screen);
907
908  int main(int argc,const char *argv[])
909 {
910         /* shall I run as a daemon */
911         bool is_daemon = false;
912         bool interactive = false;
913         bool Fork = true;
914         bool no_process_group = false;
915         bool log_stdout = false;
916         char *ports = NULL;
917         char *profile_level = NULL;
918         int opt;
919         poptContext pc;
920         bool print_build_options = False;
921         enum {
922                 OPT_DAEMON = 1000,
923                 OPT_INTERACTIVE,
924                 OPT_FORK,
925                 OPT_NO_PROCESS_GROUP,
926                 OPT_LOG_STDOUT
927         };
928         struct poptOption long_options[] = {
929         POPT_AUTOHELP
930         {"daemon", 'D', POPT_ARG_NONE, NULL, OPT_DAEMON, "Become a daemon (default)" },
931         {"interactive", 'i', POPT_ARG_NONE, NULL, OPT_INTERACTIVE, "Run interactive (not a daemon)"},
932         {"foreground", 'F', POPT_ARG_NONE, NULL, OPT_FORK, "Run daemon in foreground (for daemontools, etc.)" },
933         {"no-process-group", '\0', POPT_ARG_NONE, NULL, OPT_NO_PROCESS_GROUP, "Don't create a new process group" },
934         {"log-stdout", 'S', POPT_ARG_NONE, NULL, OPT_LOG_STDOUT, "Log to stdout" },
935         {"build-options", 'b', POPT_ARG_NONE, NULL, 'b', "Print build options" },
936         {"port", 'p', POPT_ARG_STRING, &ports, 0, "Listen on the specified ports"},
937         {"profiling-level", 'P', POPT_ARG_STRING, &profile_level, 0, "Set profiling level","PROFILE_LEVEL"},
938         POPT_COMMON_SAMBA
939         POPT_COMMON_DYNCONFIG
940         POPT_TABLEEND
941         };
942         TALLOC_CTX *frame = talloc_stackframe(); /* Setup tos. */
943
944         smbd_init_globals();
945
946         TimeInit();
947
948 #ifdef HAVE_SET_AUTH_PARAMETERS
949         set_auth_parameters(argc,argv);
950 #endif
951
952         pc = poptGetContext("smbd", argc, argv, long_options, 0);
953         while((opt = poptGetNextOpt(pc)) != -1) {
954                 switch (opt)  {
955                 case OPT_DAEMON:
956                         is_daemon = true;
957                         break;
958                 case OPT_INTERACTIVE:
959                         interactive = true;
960                         break;
961                 case OPT_FORK:
962                         Fork = false;
963                         break;
964                 case OPT_NO_PROCESS_GROUP:
965                         no_process_group = true;
966                         break;
967                 case OPT_LOG_STDOUT:
968                         log_stdout = true;
969                         break;
970                 case 'b':
971                         print_build_options = True;
972                         break;
973                 default:
974                         d_fprintf(stderr, "\nInvalid option %s: %s\n\n",
975                                   poptBadOption(pc, 0), poptStrerror(opt));
976                         poptPrintUsage(pc, stderr, 0);
977                         exit(1);
978                 }
979         }
980         poptFreeContext(pc);
981
982         if (interactive) {
983                 Fork = False;
984                 log_stdout = True;
985         }
986
987         setup_logging(argv[0],log_stdout);
988
989         if (print_build_options) {
990                 build_options(True); /* Display output to screen as well as debug */
991                 exit(0);
992         }
993
994         load_case_tables();
995
996 #ifdef HAVE_SETLUID
997         /* needed for SecureWare on SCO */
998         setluid(0);
999 #endif
1000
1001         sec_init();
1002
1003         set_remote_machine_name("smbd", False);
1004
1005         if (interactive && (DEBUGLEVEL >= 9)) {
1006                 talloc_enable_leak_report();
1007         }
1008
1009         if (log_stdout && Fork) {
1010                 DEBUG(0,("ERROR: Can't log to stdout (-S) unless daemon is in foreground (-F) or interactive (-i)\n"));
1011                 exit(1);
1012         }
1013
1014         /* we want to re-seed early to prevent time delays causing
1015            client problems at a later date. (tridge) */
1016         generate_random_buffer(NULL, 0);
1017
1018         /* make absolutely sure we run as root - to handle cases where people
1019            are crazy enough to have it setuid */
1020
1021         gain_root_privilege();
1022         gain_root_group_privilege();
1023
1024         fault_setup((void (*)(void *))exit_server_fault);
1025         dump_core_setup("smbd");
1026
1027         /* we are never interested in SIGPIPE */
1028         BlockSignals(True,SIGPIPE);
1029
1030 #if defined(SIGFPE)
1031         /* we are never interested in SIGFPE */
1032         BlockSignals(True,SIGFPE);
1033 #endif
1034
1035 #if defined(SIGUSR2)
1036         /* We are no longer interested in USR2 */
1037         BlockSignals(True,SIGUSR2);
1038 #endif
1039
1040         /* POSIX demands that signals are inherited. If the invoking process has
1041          * these signals masked, we will have problems, as we won't recieve them. */
1042         BlockSignals(False, SIGHUP);
1043         BlockSignals(False, SIGUSR1);
1044         BlockSignals(False, SIGTERM);
1045
1046         /* we want total control over the permissions on created files,
1047            so set our umask to 0 */
1048         umask(0);
1049
1050         init_sec_ctx();
1051
1052         reopen_logs();
1053
1054         DEBUG(0,("smbd version %s started.\n", samba_version_string()));
1055         DEBUGADD(0,("%s\n", COPYRIGHT_STARTUP_MESSAGE));
1056
1057         DEBUG(2,("uid=%d gid=%d euid=%d egid=%d\n",
1058                  (int)getuid(),(int)getgid(),(int)geteuid(),(int)getegid()));
1059
1060         /* Output the build options to the debug log */ 
1061         build_options(False);
1062
1063         if (sizeof(uint16) < 2 || sizeof(uint32) < 4) {
1064                 DEBUG(0,("ERROR: Samba is not configured correctly for the word size on your machine\n"));
1065                 exit(1);
1066         }
1067
1068         if (!lp_load_initial_only(get_dyn_CONFIGFILE())) {
1069                 DEBUG(0, ("error opening config file\n"));
1070                 exit(1);
1071         }
1072
1073         if (smbd_messaging_context() == NULL)
1074                 exit(1);
1075
1076         if (!reload_services(False))
1077                 return(-1);     
1078
1079         init_structs();
1080
1081 #ifdef WITH_PROFILE
1082         if (!profile_setup(smbd_messaging_context(), False)) {
1083                 DEBUG(0,("ERROR: failed to setup profiling\n"));
1084                 return -1;
1085         }
1086         if (profile_level != NULL) {
1087                 int pl = atoi(profile_level);
1088                 struct server_id src;
1089
1090                 DEBUG(1, ("setting profiling level: %s\n",profile_level));
1091                 src.pid = getpid();
1092                 set_profile_level(pl, src);
1093         }
1094 #endif
1095
1096         DEBUG(3,( "loaded services\n"));
1097
1098         if (!is_daemon && !is_a_socket(0)) {
1099                 if (!interactive)
1100                         DEBUG(0,("standard input is not a socket, assuming -D option\n"));
1101
1102                 /*
1103                  * Setting is_daemon here prevents us from eventually calling
1104                  * the open_sockets_inetd()
1105                  */
1106
1107                 is_daemon = True;
1108         }
1109
1110         if (is_daemon && !interactive) {
1111                 DEBUG( 3, ( "Becoming a daemon.\n" ) );
1112                 become_daemon(Fork, no_process_group);
1113         }
1114
1115 #if HAVE_SETPGID
1116         /*
1117          * If we're interactive we want to set our own process group for
1118          * signal management.
1119          */
1120         if (interactive && !no_process_group)
1121                 setpgid( (pid_t)0, (pid_t)0);
1122 #endif
1123
1124         if (!directory_exist(lp_lockdir()))
1125                 mkdir(lp_lockdir(), 0755);
1126
1127         if (is_daemon)
1128                 pidfile_create("smbd");
1129
1130         if (!reinit_after_fork(smbd_messaging_context(),
1131                                smbd_event_context(), false)) {
1132                 DEBUG(0,("reinit_after_fork() failed\n"));
1133                 exit(1);
1134         }
1135
1136         smbd_setup_sig_term_handler();
1137         smbd_setup_sig_hup_handler();
1138
1139         /* Setup all the TDB's - including CLEAR_IF_FIRST tdb's. */
1140
1141         if (smbd_memcache() == NULL) {
1142                 exit(1);
1143         }
1144
1145         memcache_set_global(smbd_memcache());
1146
1147         /* Initialise the password backed before the global_sam_sid
1148            to ensure that we fetch from ldap before we make a domain sid up */
1149
1150         if(!initialize_password_db(False, smbd_event_context()))
1151                 exit(1);
1152
1153         if (!secrets_init()) {
1154                 DEBUG(0, ("ERROR: smbd can not open secrets.tdb\n"));
1155                 exit(1);
1156         }
1157
1158         if(!get_global_sam_sid()) {
1159                 DEBUG(0,("ERROR: Samba cannot create a SAM SID.\n"));
1160                 exit(1);
1161         }
1162
1163         if (!session_init())
1164                 exit(1);
1165
1166         if (!connections_init(True))
1167                 exit(1);
1168
1169         if (!locking_init())
1170                 exit(1);
1171
1172         namecache_enable();
1173
1174         if (!W_ERROR_IS_OK(registry_init_full()))
1175                 exit(1);
1176
1177 #if 0
1178         if (!init_svcctl_db())
1179                 exit(1);
1180 #endif
1181
1182         if (!print_backend_init(smbd_messaging_context()))
1183                 exit(1);
1184
1185         if (!init_guest_info()) {
1186                 DEBUG(0,("ERROR: failed to setup guest info.\n"));
1187                 return -1;
1188         }
1189
1190         /* only start the background queue daemon if we are 
1191            running as a daemon -- bad things will happen if
1192            smbd is launched via inetd and we fork a copy of 
1193            ourselves here */
1194
1195         if (is_daemon && !interactive
1196             && lp_parm_bool(-1, "smbd", "backgroundqueue", true)) {
1197                 start_background_queue();
1198         }
1199
1200         if (!is_daemon) {
1201                 /* inetd mode */
1202                 TALLOC_FREE(frame);
1203
1204                 /* Started from inetd. fd 0 is the socket. */
1205                 /* We will abort gracefully when the client or remote system
1206                    goes away */
1207                 smbd_set_server_fd(dup(0));
1208
1209                 /* close our standard file descriptors */
1210                 close_low_fds(False); /* Don't close stderr */
1211
1212                 smbd_process();
1213
1214                 exit_server_cleanly(NULL);
1215                 return(0);
1216         }
1217
1218         if (!open_sockets_smbd(interactive, ports))
1219                 exit(1);
1220
1221         TALLOC_FREE(frame);
1222
1223         smbd_process();
1224
1225         exit_server_cleanly(NULL);
1226         return(0);
1227 }