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