source3/smbd/server.c

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