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) James Peach                    2007
   8
   9    This program is free software; you can redistribute it and/or modify
  10    it under the terms of the GNU General Public License as published by
  11    the Free Software Foundation; either version 2 of the License, or
  12    (at your option) any later version.
  13
  14    This program is distributed in the hope that it will be useful,
  15    but WITHOUT ANY WARRANTY; without even the implied warranty of
  16    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17    GNU General Public License for more details.
  18
  19    You should have received a copy of the GNU General Public License
  20    along with this program; if not, write to the Free Software
  21    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  22 */
  23
  24 #include "includes.h"
  25
  26 static_decl_rpc;
  27
  28 static int am_parent = 1;
  29
  30 /* the last message the was processed */
  31 int last_message = -1;
  32
  33 /* a useful macro to debug the last message processed */
  34 #define LAST_MESSAGE() smb_fn_name(last_message)
  35
  36 extern struct auth_context *negprot_global_auth_context;
  37 extern pstring user_socket_options;
  38 extern SIG_ATOMIC_T got_sig_term;
  39 extern SIG_ATOMIC_T reload_after_sighup;
  40 static SIG_ATOMIC_T got_sig_cld;
  41
  42 #ifdef WITH_DFS
  43 extern int dcelogin_atmost_once;
  44 #endif /* WITH_DFS */
  45
  46 /* really we should have a top level context structure that has the
  47    client file descriptor as an element. That would require a major rewrite :(
  48
  49    the following 2 functions are an alternative - they make the file
  50    descriptor private to smbd
  51  */
  52 static int server_fd = -1;
  53
  54 int smbd_server_fd(void)
  55 {
  56         return server_fd;
  57 }
  58
  59 static void smbd_set_server_fd(int fd)
  60 {
  61         server_fd = fd;
  62         client_setfd(fd);
  63 }
  64
  65 struct event_context *smbd_event_context(void)
  66 {
  67         static struct event_context *ctx;
  68
  69         if (!ctx && !(ctx = event_context_init(NULL))) {
  70                 smb_panic("Could not init smbd event context\n");
  71         }
  72         return ctx;
  73 }
  74
  75 struct messaging_context *smbd_messaging_context(void)
  76 {
  77         static struct messaging_context *ctx;
  78
  79         if (!ctx && !(ctx = messaging_init(NULL, server_id_self(),
  80                                            smbd_event_context()))) {
  81                 smb_panic("Could not init smbd messaging context\n");
  82         }
  83         return ctx;
  84 }
  85
  86 /*******************************************************************
  87  What to do when smb.conf is updated.
  88  ********************************************************************/
  89
  90 static void smb_conf_updated(struct messaging_context *msg,
  91                              void *private_data,
  92                              uint32_t msg_type,
  93                              struct server_id server_id,
  94                              DATA_BLOB *data)
  95 {
  96         DEBUG(10,("smb_conf_updated: Got message saying smb.conf was "
  97                   "updated. Reloading.\n"));
  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  Terminate signal.
 119 ****************************************************************************/
 120
 121 static void sig_term(void)
 122 {
 123         got_sig_term = 1;
 124         sys_select_signal(SIGTERM);
 125 }
 126
 127 /****************************************************************************
 128  Catch a sighup.
 129 ****************************************************************************/
 130
 131 static void sig_hup(int sig)
 132 {
 133         reload_after_sighup = 1;
 134         sys_select_signal(SIGHUP);
 135 }
 136
 137 /****************************************************************************
 138  Catch a sigcld
 139 ****************************************************************************/
 140 static void sig_cld(int sig)
 141 {
 142         got_sig_cld = 1;
 143         sys_select_signal(SIGCLD);
 144 }
 145
 146 /****************************************************************************
 147   Send a SIGTERM to our process group.
 148 *****************************************************************************/
 149
 150 static void  killkids(void)
 151 {
 152         if(am_parent) kill(0,SIGTERM);
 153 }
 154
 155 /****************************************************************************
 156  Process a sam sync message - not sure whether to do this here or
 157  somewhere else.
 158 ****************************************************************************/
 159
 160 static void msg_sam_sync(struct messaging_context *msg,
 161                          void *private_data,
 162                          uint32_t msg_type,
 163                          struct server_id server_id,
 164                          DATA_BLOB *data)
 165 {
 166         DEBUG(10, ("** sam sync message received, ignoring\n"));
 167 }
 168
 169 /****************************************************************************
 170  Process a sam sync replicate message - not sure whether to do this here or
 171  somewhere else.
 172 ****************************************************************************/
 173
 174 static void msg_sam_repl(struct messaging_context *msg,
 175                          void *private_data,
 176                          uint32_t msg_type,
 177                          struct server_id server_id,
 178                          DATA_BLOB *data)
 179 {
 180         uint32 low_serial;
 181
 182         if (data->length != sizeof(low_serial))
 183                 return;
 184
 185         low_serial = *((uint32 *)data->data);
 186
 187         DEBUG(3, ("received sam replication message, serial = 0x%04x\n",
 188                   low_serial));
 189 }
 190
 191 /****************************************************************************
 192  Open the socket communication - inetd.
 193 ****************************************************************************/
 194
 195 static BOOL open_sockets_inetd(void)
 196 {
 197         /* Started from inetd. fd 0 is the socket. */
 198         /* We will abort gracefully when the client or remote system
 199            goes away */
 200         smbd_set_server_fd(dup(0));
 201
 202         /* close our standard file descriptors */
 203         close_low_fds(False); /* Don't close stderr */
 204
 205         set_socket_options(smbd_server_fd(),"SO_KEEPALIVE");
 206         set_socket_options(smbd_server_fd(), user_socket_options);
 207
 208         return True;
 209 }
 210
 211 static void msg_exit_server(struct messaging_context *msg,
 212                             void *private_data,
 213                             uint32_t msg_type,
 214                             struct server_id server_id,
 215                             DATA_BLOB *data)
 216 {
 217         DEBUG(3, ("got a SHUTDOWN message\n"));
 218         exit_server_cleanly(NULL);
 219 }
 220
 221 #ifdef DEVELOPER
 222 static void msg_inject_fault(struct messaging_context *msg,
 223                              void *private_data,
 224                              uint32_t msg_type,
 225                              struct server_id src,
 226                              DATA_BLOB *data)
 227 {
 228         int sig;
 229
 230         if (data->length != sizeof(sig)) {
 231
 232                 DEBUG(0, ("Process %s sent bogus signal injection request\n",
 233                           procid_str_static(&src)));
 234                 return;
 235         }
 236
 237         sig = *(int *)data->data;
 238         if (sig == -1) {
 239                 exit_server("internal error injected");
 240                 return;
 241         }
 242
 243 #if HAVE_STRSIGNAL
 244         DEBUG(0, ("Process %s requested injection of signal %d (%s)\n",
 245                   procid_str_static(&src), sig, strsignal(sig)));
 246 #else
 247         DEBUG(0, ("Process %s requested injection of signal %d\n",
 248                   procid_str_static(&src), sig));
 249 #endif
 250
 251         kill(sys_getpid(), sig);
 252 }
 253 #endif /* DEVELOPER */
 254
 255 struct child_pid {
 256         struct child_pid *prev, *next;
 257         pid_t pid;
 258 };
 259
 260 static struct child_pid *children;
 261 static int num_children;
 262
 263 static void add_child_pid(pid_t pid)
 264 {
 265         struct child_pid *child;
 266
 267         if (lp_max_smbd_processes() == 0) {
 268                 /* Don't bother with the child list if we don't care anyway */
 269                 return;
 270         }
 271
 272         child = SMB_MALLOC_P(struct child_pid);
 273         if (child == NULL) {
 274                 DEBUG(0, ("Could not add child struct -- malloc failed\n"));
 275                 return;
 276         }
 277         child->pid = pid;
 278         DLIST_ADD(children, child);
 279         num_children += 1;
 280 }
 281
 282 static void remove_child_pid(pid_t pid)
 283 {
 284         struct child_pid *child;
 285
 286         if (lp_max_smbd_processes() == 0) {
 287                 /* Don't bother with the child list if we don't care anyway */
 288                 return;
 289         }
 290
 291         for (child = children; child != NULL; child = child->next) {
 292                 if (child->pid == pid) {
 293                         struct child_pid *tmp = child;
 294                         DLIST_REMOVE(children, child);
 295                         SAFE_FREE(tmp);
 296                         num_children -= 1;
 297                         return;
 298                 }
 299         }
 300
 301         DEBUG(0, ("Could not find child %d -- ignoring\n", (int)pid));
 302 }
 303
 304 /****************************************************************************
 305  Have we reached the process limit ?
 306 ****************************************************************************/
 307
 308 static BOOL allowable_number_of_smbd_processes(void)
 309 {
 310         int max_processes = lp_max_smbd_processes();
 311
 312         if (!max_processes)
 313                 return True;
 314
 315         return num_children < max_processes;
 316 }
 317
 318 /****************************************************************************
 319  Open the socket communication.
 320 ****************************************************************************/
 321
 322 static BOOL open_sockets_smbd(enum smb_server_mode server_mode, const char *smb_ports)
 323 {
 324         int num_sockets = 0;
 325         int fd_listenset[FD_SETSIZE];
 326         fd_set listen_set;
 327         int s;
 328         int maxfd = 0;
 329         int i;
 330         struct timeval idle_timeout = timeval_zero();
 331
 332         if (server_mode == SERVER_MODE_INETD) {
 333                 return open_sockets_inetd();
 334         }
 335
 336 #ifdef HAVE_ATEXIT
 337         {
 338                 static int atexit_set;
 339                 if(atexit_set == 0) {
 340                         atexit_set=1;
 341                         atexit(killkids);
 342                 }
 343         }
 344 #endif
 345
 346         /* Stop zombies */
 347         CatchSignal(SIGCLD, sig_cld);
 348
 349         FD_ZERO(&listen_set);
 350
 351         /* At this point, it doesn't matter what daemon mode we are in, we
 352          * need some sockets to listen on.
 353          */
 354         num_sockets = smbd_sockinit(smb_ports, fd_listenset, &idle_timeout);
 355         if (num_sockets == 0) {
 356                 return False;
 357         }
 358
 359         for (i = 0; i < num_sockets; ++i) {
 360                 FD_SET(fd_listenset[i], &listen_set);
 361                 maxfd = MAX(maxfd, fd_listenset[i]);
 362         }
 363
 364         /* Listen to messages */
 365
 366         messaging_register(smbd_messaging_context(), NULL,
 367                            MSG_SMB_SAM_SYNC, msg_sam_sync);
 368         messaging_register(smbd_messaging_context(), NULL,
 369                            MSG_SMB_SAM_REPL, msg_sam_repl);
 370         messaging_register(smbd_messaging_context(), NULL,
 371                            MSG_SHUTDOWN, msg_exit_server);
 372         messaging_register(smbd_messaging_context(), NULL,
 373                            MSG_SMB_FILE_RENAME, msg_file_was_renamed);
 374         messaging_register(smbd_messaging_context(), NULL,
 375                            MSG_SMB_CONF_UPDATED, smb_conf_updated);
 376         messaging_register(smbd_messaging_context(), NULL,
 377                            MSG_SMB_STAT_CACHE_DELETE, smb_stat_cache_delete);
 378
 379 #ifdef DEVELOPER
 380         messaging_register(smbd_messaging_context(), NULL,
 381                            MSG_SMB_INJECT_FAULT, msg_inject_fault);
 382 #endif
 383
 384         /* now accept incoming connections - forking a new process
 385            for each incoming connection */
 386         DEBUG(2,("waiting for a connection\n"));
 387         while (1) {
 388                 fd_set lfds;
 389                 int num;
 390
 391                 /* Free up temporary memory from the main smbd. */
 392                 lp_TALLOC_FREE();
 393
 394                 /* Ensure we respond to PING and DEBUG messages from the main smbd. */
 395                 message_dispatch();
 396
 397                 if (got_sig_cld) {
 398                         pid_t pid;
 399                         got_sig_cld = False;
 400
 401                         while ((pid = sys_waitpid(-1, NULL, WNOHANG)) > 0) {
 402                                 remove_child_pid(pid);
 403                         }
 404                 }
 405
 406                 memcpy((char *)&lfds, (char *)&listen_set,
 407                        sizeof(listen_set));
 408
 409                 num = sys_select(maxfd+1,&lfds,NULL,NULL,
 410                                  timeval_is_zero(&idle_timeout) ?
 411                                  NULL : &idle_timeout);
 412
 413                 if (num == -1 && errno == EINTR) {
 414                         if (got_sig_term) {
 415                                 exit_server_cleanly(NULL);
 416                         }
 417
 418                         /* check for sighup processing */
 419                         if (reload_after_sighup) {
 420                                 change_to_root_user();
 421                                 DEBUG(1,("Reloading services after SIGHUP\n"));
 422                                 reload_services(False);
 423                                 reload_after_sighup = 0;
 424                         }
 425
 426                         continue;
 427                 }
 428
 429                 /* If the idle timeout fired and we don't have any connected
 430                  * users, exit gracefully. We should be running under a process
 431                  * controller that will restart us if necessry.
 432                  */
 433                 if (num == 0 && count_all_current_connections() == 0) {
 434                         exit_server_cleanly("idle timeout");
 435                 }
 436
 437                 /* check if we need to reload services */
 438                 check_reload(time(NULL));
 439
 440                 /* Find the sockets that are read-ready -
 441                    accept on these. */
 442                 for( ; num > 0; num--) {
 443                         struct sockaddr addr;
 444                         socklen_t in_addrlen = sizeof(addr);
 445                         pid_t child = 0;
 446
 447                         s = -1;
 448                         for(i = 0; i < num_sockets; i++) {
 449                                 if(FD_ISSET(fd_listenset[i],&lfds)) {
 450                                         s = fd_listenset[i];
 451                                         /* Clear this so we don't look
 452                                            at it again. */
 453                                         FD_CLR(fd_listenset[i],&lfds);
 454                                         break;
 455                                 }
 456                         }
 457
 458                         smbd_set_server_fd(accept(s,&addr,&in_addrlen));
 459
 460                         if (smbd_server_fd() == -1 && errno == EINTR)
 461                                 continue;
 462
 463                         if (smbd_server_fd() == -1) {
 464                                 DEBUG(0,("open_sockets_smbd: accept: %s\n",
 465                                          strerror(errno)));
 466                                 continue;
 467                         }
 468
 469                         /* Ensure child is set to blocking mode */
 470                         set_blocking(smbd_server_fd(),True);
 471
 472                         /* In interactive mode, return with a connected socket.
 473                          * Foreground and daemon modes should fork worker
 474                          * processes.
 475                          */
 476                         if (server_mode == SERVER_MODE_INTERACTIVE) {
 477                                 return True;
 478                         }
 479
 480                         if (allowable_number_of_smbd_processes() &&
 481                             smbd_server_fd() != -1 &&
 482                             ((child = sys_fork())==0)) {
 483                                 /* Child code ... */
 484
 485                                 /* Stop zombies, the parent explicitly handles
 486                                  * them, counting worker smbds. */
 487                                 CatchChild();
 488
 489                                 /* close the listening socket(s) */
 490                                 for(i = 0; i < num_sockets; i++)
 491                                         close(fd_listenset[i]);
 492
 493                                 /* close our standard file
 494                                    descriptors */
 495                                 close_low_fds(False);
 496                                 am_parent = 0;
 497
 498                                 set_socket_options(smbd_server_fd(),"SO_KEEPALIVE");
 499                                 set_socket_options(smbd_server_fd(),user_socket_options);
 500
 501                                 /* this is needed so that we get decent entries
 502                                    in smbstatus for port 445 connects */
 503                                 set_remote_machine_name(get_peer_addr(smbd_server_fd()),
 504                                                         False);
 505
 506                                 /* Reset the state of the random
 507                                  * number generation system, so
 508                                  * children do not get the same random
 509                                  * numbers as each other */
 510
 511                                 set_need_random_reseed();
 512                                 /* tdb needs special fork handling - remove
 513                                  * CLEAR_IF_FIRST flags */
 514                                 if (tdb_reopen_all(1) == -1) {
 515                                         DEBUG(0,("tdb_reopen_all failed.\n"));
 516                                         smb_panic("tdb_reopen_all failed.");
 517                                 }
 518
 519                                 return True;
 520                         }
 521                         /* The parent doesn't need this socket */
 522                         close(smbd_server_fd());
 523
 524                         /* Sun May 6 18:56:14 2001 ackley@cs.unm.edu:
 525                                 Clear the closed fd info out of server_fd --
 526                                 and more importantly, out of client_fd in
 527                                 util_sock.c, to avoid a possible
 528                                 getpeername failure if we reopen the logs
 529                                 and use %I in the filename.
 530                         */
 531
 532                         smbd_set_server_fd(-1);
 533
 534                         if (child != 0) {
 535                                 add_child_pid(child);
 536                         }
 537
 538                         /* Force parent to check log size after
 539                          * spawning child.  Fix from
 540                          * klausr@ITAP.Physik.Uni-Stuttgart.De.  The
 541                          * parent smbd will log to logserver.smb.  It
 542                          * writes only two messages for each child
 543                          * started/finished. But each child writes,
 544                          * say, 50 messages also in logserver.smb,
 545                          * begining with the debug_count of the
 546                          * parent, before the child opens its own log
 547                          * file logserver.client. In a worst case
 548                          * scenario the size of logserver.smb would be
 549                          * checked after about 50*50=2500 messages
 550                          * (ca. 100kb).
 551                          * */
 552                         force_check_log_size();
 553
 554                 } /* end for num */
 555         } /* end while 1 */
 556
 557 /* NOTREACHED   return True; */
 558 }
 559
 560 /****************************************************************************
 561  Reload printers
 562 **************************************************************************/
 563 void reload_printers(void)
 564 {
 565         int snum;
 566         int n_services = lp_numservices();
 567         int pnum = lp_servicenumber(PRINTERS_NAME);
 568         const char *pname;
 569
 570         pcap_cache_reload();
 571
 572         /* remove stale printers */
 573         for (snum = 0; snum < n_services; snum++) {
 574                 /* avoid removing PRINTERS_NAME or non-autoloaded printers */
 575                 if (snum == pnum || !(lp_snum_ok(snum) && lp_print_ok(snum) &&
 576                                       lp_autoloaded(snum)))
 577                         continue;
 578
 579                 pname = lp_printername(snum);
 580                 if (!pcap_printername_ok(pname)) {
 581                         DEBUG(3, ("removing stale printer %s\n", pname));
 582
 583                         if (is_printer_published(NULL, snum, NULL))
 584                                 nt_printer_publish(NULL, snum, SPOOL_DS_UNPUBLISH);
 585                         del_a_printer(pname);
 586                         lp_killservice(snum);
 587                 }
 588         }
 589
 590         load_printers();
 591 }
 592
 593 /****************************************************************************
 594  Reload the services file.
 595 **************************************************************************/
 596
 597 BOOL reload_services(BOOL test)
 598 {
 599         BOOL ret;
 600
 601         if (lp_loaded()) {
 602                 pstring fname;
 603                 pstrcpy(fname,lp_configfile());
 604                 if (file_exist(fname, NULL) &&
 605                     !strcsequal(fname, dyn_CONFIGFILE)) {
 606                         pstrcpy(dyn_CONFIGFILE, fname);
 607                         test = False;
 608                 }
 609         }
 610
 611         reopen_logs();
 612
 613         if (test && !lp_file_list_changed())
 614                 return(True);
 615
 616         lp_killunused(conn_snum_used);
 617
 618         ret = lp_load(dyn_CONFIGFILE, False, False, True, True);
 619
 620         reload_printers();
 621
 622         /* perhaps the config filename is now set */
 623         if (!test)
 624                 reload_services(True);
 625
 626         reopen_logs();
 627
 628         load_interfaces();
 629
 630         if (smbd_server_fd() != -1) {
 631                 set_socket_options(smbd_server_fd(),"SO_KEEPALIVE");
 632                 set_socket_options(smbd_server_fd(), user_socket_options);
 633         }
 634
 635         mangle_reset_cache();
 636         reset_stat_cache();
 637
 638         /* this forces service parameters to be flushed */
 639         set_current_service(NULL,0,True);
 640
 641         return(ret);
 642 }
 643
 644 /****************************************************************************
 645  Exit the server.
 646 ****************************************************************************/
 647
 648 /* Reasons for shutting down a server process. */
 649 enum server_exit_reason { SERVER_EXIT_NORMAL, SERVER_EXIT_ABNORMAL };
 650
 651 static void exit_server_common(enum server_exit_reason how,
 652         const char *const reason) NORETURN_ATTRIBUTE;
 653
 654 static void exit_server_common(enum server_exit_reason how,
 655         const char *const reason)
 656 {
 657         static int firsttime=1;
 658
 659         if (!firsttime)
 660                 exit(0);
 661         firsttime = 0;
 662
 663         change_to_root_user();
 664
 665         if (negprot_global_auth_context) {
 666                 (negprot_global_auth_context->free)(&negprot_global_auth_context);
 667         }
 668
 669         conn_close_all();
 670
 671         invalidate_all_vuids();
 672
 673         /* 3 second timeout. */
 674         print_notify_send_messages(smbd_messaging_context(), 3);
 675
 676         /* delete our entry in the connections database. */
 677         yield_connection(NULL,"");
 678
 679         respond_to_all_remaining_local_messages();
 680
 681 #ifdef WITH_DFS
 682         if (dcelogin_atmost_once) {
 683                 dfs_unlogin();
 684         }
 685 #endif
 686
 687         locking_end();
 688         printing_end();
 689
 690         server_encryption_shutdown();
 691
 692         if (how != SERVER_EXIT_NORMAL) {
 693                 int oldlevel = DEBUGLEVEL;
 694                 char *last_inbuf = get_InBuffer();
 695
 696                 DEBUGLEVEL = 10;
 697
 698                 DEBUGSEP(0);
 699                 DEBUG(0,("Abnormal server exit: %s\n",
 700                         reason ? reason : "no explanation provided"));
 701                 DEBUGSEP(0);
 702
 703                 log_stack_trace();
 704                 if (last_inbuf) {
 705                         DEBUG(0,("Last message was %s\n", LAST_MESSAGE()));
 706                         show_msg(last_inbuf);
 707                 }
 708
 709                 DEBUGLEVEL = oldlevel;
 710                 dump_core();
 711
 712         } else {
 713                 DEBUG(3,("Server exit (%s)\n",
 714                         (reason ? reason : "normal exit")));
 715         }
 716
 717         exit(0);
 718 }
 719
 720 void exit_server(const char *const explanation)
 721 {
 722         exit_server_common(SERVER_EXIT_ABNORMAL, explanation);
 723 }
 724
 725 void exit_server_cleanly(const char *const explanation)
 726 {
 727         exit_server_common(SERVER_EXIT_NORMAL, explanation);
 728 }
 729
 730 void exit_server_fault(void)
 731 {
 732         exit_server("critical server fault");
 733 }
 734
 735 /****************************************************************************
 736  Initialise connect, service and file structs.
 737 ****************************************************************************/
 738
 739 static BOOL init_structs(void )
 740 {
 741         /*
 742          * Set the machine NETBIOS name if not already
 743          * set from the config file.
 744          */
 745
 746         if (!init_names())
 747                 return False;
 748
 749         conn_init();
 750
 751         file_init();
 752
 753         /* for RPC pipes */
 754         init_rpc_pipe_hnd();
 755
 756         init_dptrs();
 757
 758         secrets_init();
 759
 760         return True;
 761 }
 762
 763 /*
 764  * Send keepalive packets to our client
 765  */
 766 static BOOL keepalive_fn(const struct timeval *now, void *private_data)
 767 {
 768         if (!send_keepalive(smbd_server_fd())) {
 769                 DEBUG( 2, ( "Keepalive failed - exiting.\n" ) );
 770                 return False;
 771         }
 772         return True;
 773 }
 774
 775 /*
 776  * Do the recurring check if we're idle
 777  */
 778 static BOOL deadtime_fn(const struct timeval *now, void *private_data)
 779 {
 780         if ((conn_num_open() == 0)
 781             || (conn_idle_all(now->tv_sec))) {
 782                 DEBUG( 2, ( "Closing idle connection\n" ) );
 783                 messaging_send(smbd_messaging_context(), procid_self(),
 784                                MSG_SHUTDOWN, &data_blob_null);
 785                 return False;
 786         }
 787
 788         return True;
 789 }
 790
 791
 792 /****************************************************************************
 793  main program.
 794 ****************************************************************************/
 795
 796 /* Declare prototype for build_options() to avoid having to run it through
 797    mkproto.h.  Mixing $(builddir) and $(srcdir) source files in the current
 798    prototype generation system is too complicated. */
 799
 800 extern void build_options(BOOL screen);
 801
 802  int main(int argc,const char *argv[])
 803 {
 804         /* shall I run as a daemon */
 805         BOOL no_process_group = False;
 806         BOOL log_stdout = False;
 807         const char *ports = NULL;
 808         const char *profile_level = NULL;
 809         int opt;
 810         poptContext pc;
 811
 812         enum smb_server_mode server_mode = SERVER_MODE_DAEMON;
 813
 814         struct poptOption long_options[] = {
 815         POPT_AUTOHELP
 816         {"daemon", 'D', POPT_ARG_VAL, &server_mode, SERVER_MODE_DAEMON,
 817                 "Become a daemon (default)" },
 818         {"interactive", 'i', POPT_ARG_VAL, &server_mode, SERVER_MODE_INTERACTIVE,
 819                 "Run interactive (not a daemon)"},
 820         {"foreground", 'F', POPT_ARG_VAL, &server_mode, SERVER_MODE_FOREGROUND,
 821                 "Run daemon in foreground (for daemontools, etc.)" },
 822         {"no-process-group", '\0', POPT_ARG_VAL, &no_process_group, True,
 823                 "Don't create a new process group" },
 824         {"log-stdout", 'S', POPT_ARG_VAL, &log_stdout, True, "Log to stdout" },
 825         {"build-options", 'b', POPT_ARG_NONE, NULL, 'b', "Print build options" },
 826         {"port", 'p', POPT_ARG_STRING, &ports, 0, "Listen on the specified ports"},
 827         {"profiling-level", 'P', POPT_ARG_STRING, &profile_level, 0, "Set profiling level","PROFILE_LEVEL"},
 828         POPT_COMMON_SAMBA
 829         POPT_COMMON_DYNCONFIG
 830         POPT_TABLEEND
 831         };
 832
 833         load_case_tables();
 834
 835         TimeInit();
 836
 837 #ifdef HAVE_SET_AUTH_PARAMETERS
 838         set_auth_parameters(argc,argv);
 839 #endif
 840
 841         pc = poptGetContext("smbd", argc, argv, long_options, 0);
 842
 843         while((opt = poptGetNextOpt(pc)) != -1) {
 844                 switch (opt)  {
 845                 case 'b':
 846                         build_options(True); /* Display output to screen as well as debug */
 847                         exit(0);
 848                         break;
 849                 }
 850         }
 851
 852         poptFreeContext(pc);
 853
 854 #ifdef HAVE_SETLUID
 855         /* needed for SecureWare on SCO */
 856         setluid(0);
 857 #endif
 858
 859         sec_init();
 860
 861         set_remote_machine_name("smbd", False);
 862
 863         if (server_mode == SERVER_MODE_INTERACTIVE) {
 864                 log_stdout = True;
 865                 if (DEBUGLEVEL >= 9) {
 866                         talloc_enable_leak_report();
 867                 }
 868         }
 869
 870         if (log_stdout && server_mode == SERVER_MODE_DAEMON) {
 871                 DEBUG(0,("ERROR: Can't log to stdout (-S) unless daemon is in foreground (-F) or interactive (-i)\n"));
 872                 exit(1);
 873         }
 874
 875         setup_logging(argv[0],log_stdout);
 876
 877         /* we want to re-seed early to prevent time delays causing
 878            client problems at a later date. (tridge) */
 879         generate_random_buffer(NULL, 0);
 880
 881         /* make absolutely sure we run as root - to handle cases where people
 882            are crazy enough to have it setuid */
 883
 884         gain_root_privilege();
 885         gain_root_group_privilege();
 886
 887         fault_setup((void (*)(void *))exit_server_fault);
 888         dump_core_setup("smbd");
 889
 890         CatchSignal(SIGTERM , SIGNAL_CAST sig_term);
 891         CatchSignal(SIGHUP,SIGNAL_CAST sig_hup);
 892
 893         /* we are never interested in SIGPIPE */
 894         BlockSignals(True,SIGPIPE);
 895
 896 #if defined(SIGFPE)
 897         /* we are never interested in SIGFPE */
 898         BlockSignals(True,SIGFPE);
 899 #endif
 900
 901 #if defined(SIGUSR2)
 902         /* We are no longer interested in USR2 */
 903         BlockSignals(True,SIGUSR2);
 904 #endif
 905
 906         /* POSIX demands that signals are inherited. If the invoking process has
 907          * these signals masked, we will have problems, as we won't recieve them. */
 908         BlockSignals(False, SIGHUP);
 909         BlockSignals(False, SIGUSR1);
 910         BlockSignals(False, SIGTERM);
 911
 912         /* we want total control over the permissions on created files,
 913            so set our umask to 0 */
 914         umask(0);
 915
 916         init_sec_ctx();
 917
 918         reopen_logs();
 919
 920         DEBUG(0,( "smbd version %s started.\n", SAMBA_VERSION_STRING));
 921         DEBUGADD( 0, ( "%s\n", COPYRIGHT_STARTUP_MESSAGE ) );
 922
 923         DEBUG(2,("uid=%d gid=%d euid=%d egid=%d\n",
 924                  (int)getuid(),(int)getgid(),(int)geteuid(),(int)getegid()));
 925
 926         /* Output the build options to the debug log */
 927         build_options(False);
 928
 929         if (sizeof(uint16) < 2 || sizeof(uint32) < 4) {
 930                 DEBUG(0,("ERROR: Samba is not configured correctly for the word size on your machine\n"));
 931                 exit(1);
 932         }
 933
 934         /*
 935          * Do this before reload_services.
 936          */
 937
 938         if (!reload_services(False))
 939                 return(-1);
 940
 941         init_structs();
 942
 943 #ifdef WITH_PROFILE
 944         if (!profile_setup(smbd_messaging_context(), False)) {
 945                 DEBUG(0,("ERROR: failed to setup profiling\n"));
 946                 return -1;
 947         }
 948         if (profile_level != NULL) {
 949                 int pl = atoi(profile_level);
 950                 struct server_id src;
 951
 952                 DEBUG(1, ("setting profiling level: %s\n",profile_level));
 953                 src.pid = getpid();
 954                 set_profile_level(pl, src);
 955         }
 956 #endif
 957
 958         DEBUG(3,( "loaded services\n"));
 959
 960         if (is_a_socket(0)) {
 961                 if (server_mode == SERVER_MODE_DAEMON) {
 962                         DEBUG(0,("standard input is a socket, "
 963                                     "assuming -F option\n"));
 964                 }
 965                 server_mode = SERVER_MODE_INETD;
 966         }
 967
 968         if (server_mode == SERVER_MODE_DAEMON) {
 969                 DEBUG( 3, ( "Becoming a daemon.\n" ) );
 970                 become_daemon(True, no_process_group);
 971         } else if (server_mode == SERVER_MODE_FOREGROUND) {
 972                 become_daemon(False, no_process_group);
 973         }
 974
 975 #if HAVE_SETPGID
 976         /*
 977          * If we're interactive we want to set our own process group for
 978          * signal management.
 979          */
 980         if (server_mode == SERVER_MODE_INTERACTIVE && !no_process_group) {
 981                 setpgid( (pid_t)0, (pid_t)0);
 982         }
 983 #endif
 984
 985         if (!directory_exist(lp_lockdir(), NULL))
 986                 mkdir(lp_lockdir(), 0755);
 987
 988         if (server_mode != SERVER_MODE_INETD &&
 989             server_mode != SERVER_MODE_INTERACTIVE) {
 990                 pidfile_create("smbd");
 991         }
 992
 993         /* Setup all the TDB's - including CLEAR_IF_FIRST tdb's. */
 994         if (smbd_messaging_context() == NULL)
 995                 exit(1);
 996
 997         /* Initialise the password backed before the global_sam_sid
 998            to ensure that we fetch from ldap before we make a domain sid up */
 999
1000         if(!initialize_password_db(False, smbd_event_context()))
1001                 exit(1);
1002
1003         if (!secrets_init()) {
1004                 DEBUG(0, ("ERROR: smbd can not open secrets.tdb\n"));
1005                 exit(1);
1006         }
1007
1008         if(!get_global_sam_sid()) {
1009                 DEBUG(0,("ERROR: Samba cannot create a SAM SID.\n"));
1010                 exit(1);
1011         }
1012
1013         if (!session_init())
1014                 exit(1);
1015
1016         if (!connections_init(True))
1017                 exit(1);
1018
1019         if (!locking_init(0))
1020                 exit(1);
1021
1022         namecache_enable();
1023
1024         if (!init_registry())
1025                 exit(1);
1026
1027 #if 0
1028         if (!init_svcctl_db())
1029                 exit(1);
1030 #endif
1031
1032         if (!print_backend_init())
1033                 exit(1);
1034
1035         if (!init_guest_info()) {
1036                 DEBUG(0,("ERROR: failed to setup guest info.\n"));
1037                 return -1;
1038         }
1039
1040         /* Setup the main smbd so that we can get messages. */
1041         /* don't worry about general printing messages here */
1042
1043         claim_connection(NULL,"",0,True,FLAG_MSG_GENERAL|FLAG_MSG_SMBD);
1044
1045         /* only start the background queue daemon if we are
1046            running as a daemon -- bad things will happen if
1047            smbd is launched via inetd and we fork a copy of
1048            ourselves here */
1049         if (server_mode != SERVER_MODE_INETD &&
1050             server_mode != SERVER_MODE_INTERACTIVE) {
1051                 start_background_queue();
1052         }
1053
1054         /* Always attempt to initialize DMAPI. We will only use it later if
1055          * lp_dmapi_support is set on the share, but we need a single global
1056          * session to work with.
1057          */
1058         dmapi_init_session();
1059
1060         if (!open_sockets_smbd(server_mode, ports)) {
1061                 exit(1);
1062         }
1063
1064         /*
1065          * everything after this point is run after the fork()
1066          */
1067
1068         static_init_rpc;
1069
1070         init_modules();
1071
1072         /* Possibly reload the services file. Only worth doing in
1073          * daemon mode. In inetd mode, we know we only just loaded this.
1074          */
1075         if (server_mode != SERVER_MODE_INETD &&
1076             server_mode != SERVER_MODE_INTERACTIVE) {
1077                 reload_services(True);
1078         }
1079
1080         if (!init_account_policy()) {
1081                 DEBUG(0,("Could not open account policy tdb.\n"));
1082                 exit(1);
1083         }
1084
1085         if (*lp_rootdir()) {
1086                 if (sys_chroot(lp_rootdir()) == 0)
1087                         DEBUG(2,("Changed root to %s\n", lp_rootdir()));
1088         }
1089
1090         /* Setup oplocks */
1091         if (!init_oplocks(smbd_messaging_context()))
1092                 exit(1);
1093
1094         /* Setup aio signal handler. */
1095         initialize_async_io_handler();
1096
1097         /* register our message handlers */
1098         messaging_register(smbd_messaging_context(), NULL,
1099                            MSG_SMB_FORCE_TDIS, msg_force_tdis);
1100
1101         if ((lp_keepalive() != 0)
1102             && !(event_add_idle(smbd_event_context(), NULL,
1103                                 timeval_set(lp_keepalive(), 0),
1104                                 "keepalive", keepalive_fn,
1105                                 NULL))) {
1106                 DEBUG(0, ("Could not add keepalive event\n"));
1107                 exit(1);
1108         }
1109
1110         if (!(event_add_idle(smbd_event_context(), NULL,
1111                              timeval_set(IDLE_CLOSED_TIMEOUT, 0),
1112                              "deadtime", deadtime_fn, NULL))) {
1113                 DEBUG(0, ("Could not add deadtime event\n"));
1114                 exit(1);
1115         }
1116
1117         smbd_process();
1118
1119         namecache_shutdown();
1120
1121         exit_server_cleanly(NULL);
1122         return(0);
1123 }