source3/smbd/server.c

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