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