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