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