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