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