r23272: Fix main event loop - reviewed by Volker.
[sfrench/samba-autobuild/.git] / 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         brl_register_msgs(smbd_messaging_context());
379
380 #ifdef DEVELOPER
381         messaging_register(smbd_messaging_context(), NULL,
382                            MSG_SMB_INJECT_FAULT, msg_inject_fault);
383 #endif
384
385         /* now accept incoming connections - forking a new process
386            for each incoming connection */
387         DEBUG(2,("waiting for a connection\n"));
388         while (1) {
389                 struct timeval now;
390                 fd_set r_fds, w_fds;
391                 int num;
392                 
393                 /* Free up temporary memory from the main smbd. */
394                 lp_TALLOC_FREE();
395
396                 /* Ensure we respond to PING and DEBUG messages from the main smbd. */
397                 message_dispatch(smbd_messaging_context());
398
399                 if (got_sig_cld) {
400                         pid_t pid;
401                         got_sig_cld = False;
402
403                         while ((pid = sys_waitpid(-1, NULL, WNOHANG)) > 0) {
404                                 remove_child_pid(pid);
405                         }
406                 }
407
408                 memcpy((char *)&r_fds, (char *)&listen_set, 
409                        sizeof(listen_set));
410                 FD_ZERO(&w_fds);
411                 GetTimeOfDay(&now);
412
413                 event_add_to_select_args(smbd_event_context(), &now,
414                                          &r_fds, &w_fds, &idle_timeout,
415                                          &maxfd);
416
417                 num = sys_select(maxfd+1,&r_fds,&w_fds,NULL,
418                                  timeval_is_zero(&idle_timeout) ?
419                                  NULL : &idle_timeout);
420                 
421                 if (num == -1 && errno == EINTR) {
422                         if (got_sig_term) {
423                                 exit_server_cleanly(NULL);
424                         }
425
426                         /* check for sighup processing */
427                         if (reload_after_sighup) {
428                                 change_to_root_user();
429                                 DEBUG(1,("Reloading services after SIGHUP\n"));
430                                 reload_services(False);
431                                 reload_after_sighup = 0;
432                         }
433
434                         continue;
435                 }
436
437                 if (run_events(smbd_event_context(), num, &r_fds, &w_fds)) {
438                         continue;
439                 }
440
441 #if 0
442                 Deactivated for now, this needs to become a timed event
443                 vl
444
445                 /* If the idle timeout fired and we don't have any connected
446                  * users, exit gracefully. We should be running under a process
447                  * controller that will restart us if necessry.
448                  */
449                 if (num == 0 && count_all_current_connections() == 0) {
450                         exit_server_cleanly("idle timeout");
451                 }
452 #endif
453
454                 /* check if we need to reload services */
455                 check_reload(time(NULL));
456
457                 /* Find the sockets that are read-ready -
458                    accept on these. */
459                 for( ; num > 0; num--) {
460                         struct sockaddr addr;
461                         socklen_t in_addrlen = sizeof(addr);
462                         pid_t child = 0;
463
464                         s = -1;
465                         for(i = 0; i < num_sockets; i++) {
466                                 if(FD_ISSET(fd_listenset[i],&r_fds)) {
467                                         s = fd_listenset[i];
468                                         /* Clear this so we don't look
469                                            at it again. */
470                                         FD_CLR(fd_listenset[i],&r_fds);
471                                         break;
472                                 }
473                         }
474
475                         smbd_set_server_fd(accept(s,&addr,&in_addrlen));
476                         
477                         if (smbd_server_fd() == -1 && errno == EINTR)
478                                 continue;
479                         
480                         if (smbd_server_fd() == -1) {
481                                 DEBUG(0,("open_sockets_smbd: accept: %s\n",
482                                          strerror(errno)));
483                                 continue;
484                         }
485
486                         /* Ensure child is set to blocking mode */
487                         set_blocking(smbd_server_fd(),True);
488
489                         /* In interactive mode, return with a connected socket.
490                          * Foreground and daemon modes should fork worker
491                          * processes.
492                          */
493                         if (server_mode == SERVER_MODE_INTERACTIVE) {
494                                 return True;
495                         }
496                         
497                         if (allowable_number_of_smbd_processes() &&
498                             smbd_server_fd() != -1 &&
499                             ((child = sys_fork())==0)) {
500                                 /* Child code ... */
501
502                                 /* Stop zombies, the parent explicitly handles
503                                  * them, counting worker smbds. */
504                                 CatchChild();
505                                 
506                                 /* close the listening socket(s) */
507                                 for(i = 0; i < num_sockets; i++)
508                                         close(fd_listenset[i]);
509                                 
510                                 /* close our standard file
511                                    descriptors */
512                                 close_low_fds(False);
513                                 am_parent = 0;
514                                 
515                                 set_socket_options(smbd_server_fd(),"SO_KEEPALIVE");
516                                 set_socket_options(smbd_server_fd(),user_socket_options);
517                                 
518                                 /* this is needed so that we get decent entries
519                                    in smbstatus for port 445 connects */
520                                 set_remote_machine_name(get_peer_addr(smbd_server_fd()),
521                                                         False);
522                                 
523                                 /* Reset the state of the random
524                                  * number generation system, so
525                                  * children do not get the same random
526                                  * numbers as each other */
527
528                                 set_need_random_reseed();
529                                 /* tdb needs special fork handling - remove
530                                  * CLEAR_IF_FIRST flags */
531                                 if (tdb_reopen_all(1) == -1) {
532                                         DEBUG(0,("tdb_reopen_all failed.\n"));
533                                         smb_panic("tdb_reopen_all failed.");
534                                 }
535
536                                 return True; 
537                         }
538                         /* The parent doesn't need this socket */
539                         close(smbd_server_fd()); 
540
541                         /* Sun May 6 18:56:14 2001 ackley@cs.unm.edu:
542                                 Clear the closed fd info out of server_fd --
543                                 and more importantly, out of client_fd in
544                                 util_sock.c, to avoid a possible
545                                 getpeername failure if we reopen the logs
546                                 and use %I in the filename.
547                         */
548
549                         smbd_set_server_fd(-1);
550
551                         if (child != 0) {
552                                 add_child_pid(child);
553                         }
554
555                         /* Force parent to check log size after
556                          * spawning child.  Fix from
557                          * klausr@ITAP.Physik.Uni-Stuttgart.De.  The
558                          * parent smbd will log to logserver.smb.  It
559                          * writes only two messages for each child
560                          * started/finished. But each child writes,
561                          * say, 50 messages also in logserver.smb,
562                          * begining with the debug_count of the
563                          * parent, before the child opens its own log
564                          * file logserver.client. In a worst case
565                          * scenario the size of logserver.smb would be
566                          * checked after about 50*50=2500 messages
567                          * (ca. 100kb).
568                          * */
569                         force_check_log_size();
570  
571                 } /* end for num */
572         } /* end while 1 */
573
574 /* NOTREACHED   return True; */
575 }
576
577 /****************************************************************************
578  Reload printers
579 **************************************************************************/
580 void reload_printers(void)
581 {
582         int snum;
583         int n_services = lp_numservices();
584         int pnum = lp_servicenumber(PRINTERS_NAME);
585         const char *pname;
586
587         pcap_cache_reload();
588
589         /* remove stale printers */
590         for (snum = 0; snum < n_services; snum++) {
591                 /* avoid removing PRINTERS_NAME or non-autoloaded printers */
592                 if (snum == pnum || !(lp_snum_ok(snum) && lp_print_ok(snum) &&
593                                       lp_autoloaded(snum)))
594                         continue;
595
596                 pname = lp_printername(snum);
597                 if (!pcap_printername_ok(pname)) {
598                         DEBUG(3, ("removing stale printer %s\n", pname));
599
600                         if (is_printer_published(NULL, snum, NULL))
601                                 nt_printer_publish(NULL, snum, SPOOL_DS_UNPUBLISH);
602                         del_a_printer(pname);
603                         lp_killservice(snum);
604                 }
605         }
606
607         load_printers();
608 }
609
610 /****************************************************************************
611  Reload the services file.
612 **************************************************************************/
613
614 BOOL reload_services(BOOL test)
615 {
616         BOOL ret;
617         
618         if (lp_loaded()) {
619                 pstring fname;
620                 pstrcpy(fname,lp_configfile());
621                 if (file_exist(fname, NULL) &&
622                     !strcsequal(fname, dyn_CONFIGFILE)) {
623                         pstrcpy(dyn_CONFIGFILE, fname);
624                         test = False;
625                 }
626         }
627
628         reopen_logs();
629
630         if (test && !lp_file_list_changed())
631                 return(True);
632
633         lp_killunused(conn_snum_used);
634
635         ret = lp_load(dyn_CONFIGFILE, False, False, True, True);
636
637         reload_printers();
638
639         /* perhaps the config filename is now set */
640         if (!test)
641                 reload_services(True);
642
643         reopen_logs();
644
645         load_interfaces();
646
647         if (smbd_server_fd() != -1) {      
648                 set_socket_options(smbd_server_fd(),"SO_KEEPALIVE");
649                 set_socket_options(smbd_server_fd(), user_socket_options);
650         }
651
652         mangle_reset_cache();
653         reset_stat_cache();
654
655         /* this forces service parameters to be flushed */
656         set_current_service(NULL,0,True);
657
658         return(ret);
659 }
660
661 /****************************************************************************
662  Exit the server.
663 ****************************************************************************/
664
665 /* Reasons for shutting down a server process. */
666 enum server_exit_reason { SERVER_EXIT_NORMAL, SERVER_EXIT_ABNORMAL };
667
668 static void exit_server_common(enum server_exit_reason how,
669         const char *const reason) NORETURN_ATTRIBUTE;
670
671 static void exit_server_common(enum server_exit_reason how,
672         const char *const reason)
673 {
674         static int firsttime=1;
675
676         if (!firsttime)
677                 exit(0);
678         firsttime = 0;
679
680         change_to_root_user();
681
682         if (negprot_global_auth_context) {
683                 (negprot_global_auth_context->free)(&negprot_global_auth_context);
684         }
685
686         conn_close_all();
687
688         invalidate_all_vuids();
689
690         /* 3 second timeout. */
691         print_notify_send_messages(smbd_messaging_context(), 3);
692
693         /* delete our entry in the connections database. */
694         yield_connection(NULL,"");
695
696         respond_to_all_remaining_local_messages();
697
698 #ifdef WITH_DFS
699         if (dcelogin_atmost_once) {
700                 dfs_unlogin();
701         }
702 #endif
703
704         locking_end();
705         printing_end();
706
707         server_encryption_shutdown();
708
709         if (how != SERVER_EXIT_NORMAL) {
710                 int oldlevel = DEBUGLEVEL;
711                 char *last_inbuf = get_InBuffer();
712
713                 DEBUGLEVEL = 10;
714
715                 DEBUGSEP(0);
716                 DEBUG(0,("Abnormal server exit: %s\n",
717                         reason ? reason : "no explanation provided"));
718                 DEBUGSEP(0);
719
720                 log_stack_trace();
721                 if (last_inbuf) {
722                         DEBUG(0,("Last message was %s\n", LAST_MESSAGE()));
723                         show_msg(last_inbuf);
724                 }
725
726                 DEBUGLEVEL = oldlevel;
727                 dump_core();
728
729         } else {    
730                 DEBUG(3,("Server exit (%s)\n",
731                         (reason ? reason : "normal exit")));
732         }
733
734         exit(0);
735 }
736
737 void exit_server(const char *const explanation)
738 {
739         exit_server_common(SERVER_EXIT_ABNORMAL, explanation);
740 }
741
742 void exit_server_cleanly(const char *const explanation)
743 {
744         exit_server_common(SERVER_EXIT_NORMAL, explanation);
745 }
746
747 void exit_server_fault(void)
748 {
749         exit_server("critical server fault");
750 }
751
752 /****************************************************************************
753  Initialise connect, service and file structs.
754 ****************************************************************************/
755
756 static BOOL init_structs(void )
757 {
758         /*
759          * Set the machine NETBIOS name if not already
760          * set from the config file.
761          */
762
763         if (!init_names())
764                 return False;
765
766         conn_init();
767
768         file_init();
769
770         /* for RPC pipes */
771         init_rpc_pipe_hnd();
772
773         init_dptrs();
774
775         secrets_init();
776
777         return True;
778 }
779
780 /*
781  * Send keepalive packets to our client
782  */
783 static BOOL keepalive_fn(const struct timeval *now, void *private_data)
784 {
785         if (!send_keepalive(smbd_server_fd())) {
786                 DEBUG( 2, ( "Keepalive failed - exiting.\n" ) );
787                 return False;
788         }
789         return True;
790 }
791
792 /*
793  * Do the recurring check if we're idle
794  */
795 static BOOL deadtime_fn(const struct timeval *now, void *private_data)
796 {
797         if ((conn_num_open() == 0)
798             || (conn_idle_all(now->tv_sec))) {
799                 DEBUG( 2, ( "Closing idle connection\n" ) );
800                 messaging_send(smbd_messaging_context(), procid_self(),
801                                MSG_SHUTDOWN, &data_blob_null);
802                 return False;
803         }
804
805         return True;
806 }
807
808
809 /****************************************************************************
810  main program.
811 ****************************************************************************/
812
813 /* Declare prototype for build_options() to avoid having to run it through
814    mkproto.h.  Mixing $(builddir) and $(srcdir) source files in the current
815    prototype generation system is too complicated. */
816
817 extern void build_options(BOOL screen);
818
819  int main(int argc,const char *argv[])
820 {
821         /* shall I run as a daemon */
822         BOOL no_process_group = False;
823         BOOL log_stdout = False;
824         const char *ports = NULL;
825         const char *profile_level = NULL;
826         int opt;
827         poptContext pc;
828
829         enum smb_server_mode server_mode = SERVER_MODE_DAEMON;
830
831         struct poptOption long_options[] = {
832         POPT_AUTOHELP
833         {"daemon", 'D', POPT_ARG_VAL, &server_mode, SERVER_MODE_DAEMON,
834                 "Become a daemon (default)" },
835         {"interactive", 'i', POPT_ARG_VAL, &server_mode, SERVER_MODE_INTERACTIVE,
836                 "Run interactive (not a daemon)"},
837         {"foreground", 'F', POPT_ARG_VAL, &server_mode, SERVER_MODE_FOREGROUND,
838                 "Run daemon in foreground (for daemontools, etc.)" },
839         {"no-process-group", '\0', POPT_ARG_VAL, &no_process_group, True,
840                 "Don't create a new process group" },
841         {"log-stdout", 'S', POPT_ARG_VAL, &log_stdout, True, "Log to stdout" },
842         {"build-options", 'b', POPT_ARG_NONE, NULL, 'b', "Print build options" },
843         {"port", 'p', POPT_ARG_STRING, &ports, 0, "Listen on the specified ports"},
844         {"profiling-level", 'P', POPT_ARG_STRING, &profile_level, 0, "Set profiling level","PROFILE_LEVEL"},
845         POPT_COMMON_SAMBA
846         POPT_COMMON_DYNCONFIG
847         POPT_TABLEEND
848         };
849
850         load_case_tables();
851
852         TimeInit();
853
854 #ifdef HAVE_SET_AUTH_PARAMETERS
855         set_auth_parameters(argc,argv);
856 #endif
857
858         pc = poptGetContext("smbd", argc, argv, long_options, 0);
859         
860         while((opt = poptGetNextOpt(pc)) != -1) {
861                 switch (opt)  {
862                 case 'b':
863                         build_options(True); /* Display output to screen as well as debug */ 
864                         exit(0);
865                         break;
866                 }
867         }
868
869         poptFreeContext(pc);
870
871 #ifdef HAVE_SETLUID
872         /* needed for SecureWare on SCO */
873         setluid(0);
874 #endif
875
876         sec_init();
877
878         set_remote_machine_name("smbd", False);
879
880         if (server_mode == SERVER_MODE_INTERACTIVE) {
881                 log_stdout = True;
882                 if (DEBUGLEVEL >= 9) {
883                         talloc_enable_leak_report();
884                 }
885         }
886
887         if (log_stdout && server_mode == SERVER_MODE_DAEMON) {
888                 DEBUG(0,("ERROR: Can't log to stdout (-S) unless daemon is in foreground (-F) or interactive (-i)\n"));
889                 exit(1);
890         }
891
892         setup_logging(argv[0],log_stdout);
893
894         /* we want to re-seed early to prevent time delays causing
895            client problems at a later date. (tridge) */
896         generate_random_buffer(NULL, 0);
897
898         /* make absolutely sure we run as root - to handle cases where people
899            are crazy enough to have it setuid */
900
901         gain_root_privilege();
902         gain_root_group_privilege();
903
904         fault_setup((void (*)(void *))exit_server_fault);
905         dump_core_setup("smbd");
906
907         CatchSignal(SIGTERM , SIGNAL_CAST sig_term);
908         CatchSignal(SIGHUP,SIGNAL_CAST sig_hup);
909         
910         /* we are never interested in SIGPIPE */
911         BlockSignals(True,SIGPIPE);
912
913 #if defined(SIGFPE)
914         /* we are never interested in SIGFPE */
915         BlockSignals(True,SIGFPE);
916 #endif
917
918 #if defined(SIGUSR2)
919         /* We are no longer interested in USR2 */
920         BlockSignals(True,SIGUSR2);
921 #endif
922
923         /* POSIX demands that signals are inherited. If the invoking process has
924          * these signals masked, we will have problems, as we won't recieve them. */
925         BlockSignals(False, SIGHUP);
926         BlockSignals(False, SIGUSR1);
927         BlockSignals(False, SIGTERM);
928
929         /* we want total control over the permissions on created files,
930            so set our umask to 0 */
931         umask(0);
932
933         init_sec_ctx();
934
935         reopen_logs();
936
937         DEBUG(0,( "smbd version %s started.\n", SAMBA_VERSION_STRING));
938         DEBUGADD( 0, ( "%s\n", COPYRIGHT_STARTUP_MESSAGE ) );
939
940         DEBUG(2,("uid=%d gid=%d euid=%d egid=%d\n",
941                  (int)getuid(),(int)getgid(),(int)geteuid(),(int)getegid()));
942
943         /* Output the build options to the debug log */ 
944         build_options(False);
945
946         if (sizeof(uint16) < 2 || sizeof(uint32) < 4) {
947                 DEBUG(0,("ERROR: Samba is not configured correctly for the word size on your machine\n"));
948                 exit(1);
949         }
950
951         /*
952          * Do this before reload_services.
953          */
954
955         if (!reload_services(False))
956                 return(-1);     
957
958         init_structs();
959
960 #ifdef WITH_PROFILE
961         if (!profile_setup(smbd_messaging_context(), False)) {
962                 DEBUG(0,("ERROR: failed to setup profiling\n"));
963                 return -1;
964         }
965         if (profile_level != NULL) {
966                 int pl = atoi(profile_level);
967                 struct server_id src;
968
969                 DEBUG(1, ("setting profiling level: %s\n",profile_level));
970                 src.pid = getpid();
971                 set_profile_level(pl, src);
972         }
973 #endif
974
975         DEBUG(3,( "loaded services\n"));
976
977         if (is_a_socket(0)) {
978                 if (server_mode == SERVER_MODE_DAEMON) {
979                         DEBUG(0,("standard input is a socket, "
980                                     "assuming -F option\n"));
981                 }
982                 server_mode = SERVER_MODE_INETD;
983         }
984
985         if (server_mode == SERVER_MODE_DAEMON) {
986                 DEBUG( 3, ( "Becoming a daemon.\n" ) );
987                 become_daemon(True, no_process_group);
988         } else if (server_mode == SERVER_MODE_FOREGROUND) {
989                 become_daemon(False, no_process_group);
990         }
991
992 #if HAVE_SETPGID
993         /*
994          * If we're interactive we want to set our own process group for
995          * signal management.
996          */
997         if (server_mode == SERVER_MODE_INTERACTIVE && !no_process_group) {
998                 setpgid( (pid_t)0, (pid_t)0);
999         }
1000 #endif
1001
1002         if (!directory_exist(lp_lockdir(), NULL))
1003                 mkdir(lp_lockdir(), 0755);
1004
1005         if (server_mode != SERVER_MODE_INETD &&
1006             server_mode != SERVER_MODE_INTERACTIVE) {
1007                 pidfile_create("smbd");
1008         }
1009
1010         /* Setup all the TDB's - including CLEAR_IF_FIRST tdb's. */
1011
1012         if (smbd_messaging_context() == NULL)
1013                 exit(1);
1014
1015         /* Initialise the password backed before the global_sam_sid
1016            to ensure that we fetch from ldap before we make a domain sid up */
1017
1018         if(!initialize_password_db(False, smbd_event_context()))
1019                 exit(1);
1020
1021         if (!secrets_init()) {
1022                 DEBUG(0, ("ERROR: smbd can not open secrets.tdb\n"));
1023                 exit(1);
1024         }
1025
1026         if(!get_global_sam_sid()) {
1027                 DEBUG(0,("ERROR: Samba cannot create a SAM SID.\n"));
1028                 exit(1);
1029         }
1030
1031         if (!session_init())
1032                 exit(1);
1033
1034         if (!connections_init(True))
1035                 exit(1);
1036
1037         if (!locking_init(0))
1038                 exit(1);
1039
1040         namecache_enable();
1041
1042         if (!init_registry())
1043                 exit(1);
1044
1045 #if 0
1046         if (!init_svcctl_db())
1047                 exit(1);
1048 #endif
1049
1050         if (!print_backend_init(smbd_messaging_context()))
1051                 exit(1);
1052
1053         if (!init_guest_info()) {
1054                 DEBUG(0,("ERROR: failed to setup guest info.\n"));
1055                 return -1;
1056         }
1057
1058         /* Setup the main smbd so that we can get messages. */
1059         /* don't worry about general printing messages here */
1060
1061         claim_connection(NULL,"",FLAG_MSG_GENERAL|FLAG_MSG_SMBD);
1062
1063         /* only start the background queue daemon if we are 
1064            running as a daemon -- bad things will happen if
1065            smbd is launched via inetd and we fork a copy of 
1066            ourselves here */
1067         if (server_mode != SERVER_MODE_INETD &&
1068             server_mode != SERVER_MODE_INTERACTIVE) {
1069                 start_background_queue(); 
1070         }
1071
1072         /* Always attempt to initialize DMAPI. We will only use it later if
1073          * lp_dmapi_support is set on the share, but we need a single global
1074          * session to work with.
1075          */
1076         dmapi_init_session();
1077
1078         if (!open_sockets_smbd(server_mode, ports)) {
1079                 exit(1);
1080         }
1081
1082         /*
1083          * everything after this point is run after the fork()
1084          */ 
1085
1086         static_init_rpc;
1087
1088         init_modules();
1089
1090         /* Possibly reload the services file. Only worth doing in
1091          * daemon mode. In inetd mode, we know we only just loaded this.
1092          */
1093         if (server_mode != SERVER_MODE_INETD &&
1094             server_mode != SERVER_MODE_INTERACTIVE) {
1095                 reload_services(True);
1096         }
1097
1098         if (!init_account_policy()) {
1099                 DEBUG(0,("Could not open account policy tdb.\n"));
1100                 exit(1);
1101         }
1102
1103         if (*lp_rootdir()) {
1104                 if (sys_chroot(lp_rootdir()) == 0)
1105                         DEBUG(2,("Changed root to %s\n", lp_rootdir()));
1106         }
1107
1108         /* Setup oplocks */
1109         if (!init_oplocks(smbd_messaging_context()))
1110                 exit(1);
1111         
1112         /* Setup aio signal handler. */
1113         initialize_async_io_handler();
1114
1115         /* register our message handlers */
1116         messaging_register(smbd_messaging_context(), NULL,
1117                            MSG_SMB_FORCE_TDIS, msg_force_tdis);
1118
1119         if ((lp_keepalive() != 0)
1120             && !(event_add_idle(smbd_event_context(), NULL,
1121                                 timeval_set(lp_keepalive(), 0),
1122                                 "keepalive", keepalive_fn,
1123                                 NULL))) {
1124                 DEBUG(0, ("Could not add keepalive event\n"));
1125                 exit(1);
1126         }
1127
1128         if (!(event_add_idle(smbd_event_context(), NULL,
1129                              timeval_set(IDLE_CLOSED_TIMEOUT, 0),
1130                              "deadtime", deadtime_fn, NULL))) {
1131                 DEBUG(0, ("Could not add deadtime event\n"));
1132                 exit(1);
1133         }
1134
1135         smbd_process();
1136
1137         namecache_shutdown();
1138
1139         exit_server_cleanly(NULL);
1140         return(0);
1141 }