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