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