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