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