r14079: I just found the setproctitle library from alt linux:-)
[kai/samba-autobuild/.git] / source4 / smbd / process_standard.c
index 505c2aafbf7cf764397313014c1021ffd56b9e9b..798d0883c8ef97038afd33ef2db72a995c057617 100644 (file)
@@ -1,8 +1,11 @@
 /* 
    Unix SMB/CIFS implementation.
+
    process model: standard (1 process per client connection)
-   Copyright (C) Andrew Tridgell 1992-2003
+
+   Copyright (C) Andrew Tridgell 1992-2005
    Copyright (C) James J Myers 2003 <myersjj@samba.org>
+   Copyright (C) Stefan (metze) Metzmacher 2004
    
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
 */
 
 #include "includes.h"
+#include "lib/events/events.h"
+#include "lib/tdb/include/tdb.h"
+#include "smb_server/smb_server.h"
+#include "lib/socket/socket.h"
+
+/* For specifiying event context to GSSAPI below */
+#include "system/kerberos.h"
+#include "heimdal/lib/gssapi/gssapi_locl.h"
+
+#include "passdb/secrets.h"
+
+#ifdef HAVE_SETPROCTITLE
+#include <setproctitle.h>
+#define SETPROCTITLE(x) setproctitle x
+#else
+#define SETPROCTITLE(x)
+#endif
 
 /*
   called when the process model is selected
 */
-static void model_startup(void)
+static void standard_model_init(struct event_context *ev)
 {
+       signal(SIGCHLD, SIG_IGN);
 }
 
 /*
-  called when a listening socket becomes readable
+  called when a listening socket becomes readable
 */
-static void accept_connection(struct event_context *ev, struct fd_event *fde, time_t t, uint16 flags)
+static void standard_accept_connection(struct event_context *ev, 
+                                      struct socket_context *sock, 
+                                      void (*new_conn)(struct event_context *, struct socket_context *, 
+                                                       uint32_t , void *), 
+                                      void *private)
 {
-       int accepted_fd;
-       struct sockaddr addr;
-       socklen_t in_addrlen = sizeof(addr);
+       NTSTATUS status;
+       struct socket_context *sock2;
        pid_t pid;
-       struct model_ops *model_ops = fde->private;
-
-       accepted_fd = accept(fde->fd,&addr,&in_addrlen);
-       if (accepted_fd == -1) {
-               DEBUG(0,("accept_connection_standard: accept: %s\n",
-                        strerror(errno)));
+       struct event_context *ev2;
+       struct socket_address *c, *s;
+
+       /* accept an incoming connection. */
+       status = socket_accept(sock, &sock2);
+       if (!NT_STATUS_IS_OK(status)) {
+               DEBUG(0,("standard_accept_connection: accept: %s\n",
+                        nt_errstr(status)));
+               /* this looks strange, but is correct. We need to throttle things until
+                  the system clears enough resources to handle this new socket */
+               sleep(1);
                return;
        }
 
@@ -50,107 +79,159 @@ static void accept_connection(struct event_context *ev, struct fd_event *fde, ti
 
        if (pid != 0) {
                /* parent or error code ... */
-
-               close(accepted_fd);
+               talloc_free(sock2);
                /* go back to the event loop */
                return;
        }
 
-       /* Child code ... */
+       pid = getpid();
+
+       /* This is now the child code. We need a completely new event_context to work with */
+       ev2 = event_context_init(NULL);
+
+       /* the service has given us a private pointer that
+          encapsulates the context it needs for this new connection -
+          everything else will be freed */
+       talloc_steal(ev2, private);
+       talloc_steal(private, sock2);
 
-       /* close all the listening sockets */
-       event_remove_fd_all_handler(ev, model_ops->accept_connection);
-       event_remove_fd_all_handler(ev, model_ops->accept_rpc_connection);
+       /* this will free all the listening sockets and all state that
+          is not associated with this new connection */
+       talloc_free(sock);
+       talloc_free(ev);
+
+       /* we don't care if the dup fails, as its only a select()
+          speed optimisation */
+       socket_dup(sock2);
                        
        /* tdb needs special fork handling */
        if (tdb_reopen_all() == -1) {
-               DEBUG(0,("accept_connection_standard: tdb_reopen_all failed.\n"));
+               DEBUG(0,("standard_accept_connection: tdb_reopen_all failed.\n"));
        }
 
-       /* Load DSO's */
-       init_modules();
-               
-       /* initialize new process */
-       smbd_process_init();
-               
-       init_smbsession(ev, model_ops, accepted_fd, smbd_read_handler);
+       /* Hack to ensure that GSSAPI uses the right event context */
+       gssapi_krb5_init_ev(ev2);
+
+       /* Ensure that the forked children do not expose identical random streams */
+       set_need_random_reseed();
 
-       /* return to the event loop */
+       /* setup the process title */
+       c = socket_get_peer_addr(sock2, ev2);
+       s = socket_get_my_addr(sock2, ev2);
+       if (s && c) {
+               SETPROCTITLE(("conn c[%s:%u] s[%s:%u] server_id[%d]",
+                             c->addr, c->port, s->addr, s->port, pid));
+       }
+       talloc_free(c);
+       talloc_free(s);
+
+       /* setup this new connection */
+       new_conn(ev2, sock2, pid, private);
+
+       /* we can't return to the top level here, as that event context is gone,
+          so we now process events in the new event context until there are no
+          more to process */      
+       event_loop_wait(ev2);
+
+       talloc_free(ev2);
+       exit(0);
 }
 
 /*
-  called when a rpc listening socket becomes readable
+  called to create a new server task
 */
-static void accept_rpc_connection(struct event_context *ev, struct fd_event *fde, time_t t, uint16 flags)
+static void standard_new_task(struct event_context *ev, 
+                             void (*new_task)(struct event_context *, uint32_t , void *), 
+                             void *private)
 {
-       int accepted_fd;
-       struct sockaddr addr;
-       socklen_t in_addrlen = sizeof(addr);
        pid_t pid;
-
-       accepted_fd = accept(fde->fd,&addr,&in_addrlen);
-       if (accepted_fd == -1) {
-               DEBUG(0,("accept_connection_standard: accept: %s\n",
-                        strerror(errno)));
-               return;
-       }
+       struct event_context *ev2;
 
        pid = fork();
 
        if (pid != 0) {
-               /* parent or error code ... */
-               close(accepted_fd);
-               /* go back to the event loop */
+               /* parent or error code ... go back to the event loop */
                return;
        }
 
-       /* Child code ... */
+       pid = getpid();
 
-       /* close all the listening sockets */
-       event_remove_fd_all_handler(ev, accept_connection);
-       event_remove_fd_all_handler(ev, accept_rpc_connection);
-                       
-       init_rpc_session(ev, fde->private, accepted_fd); 
-}
+       /* This is now the child code. We need a completely new event_context to work with */
+       ev2 = event_context_init(NULL);
 
-/* called when a SMB connection goes down */
-static void terminate_connection(struct server_context *server, const char *reason) 
-{
-       server_terminate(server);
-       /* terminate this process */
+       /* the service has given us a private pointer that
+          encapsulates the context it needs for this new connection -
+          everything else will be freed */
+       talloc_steal(ev2, private);
+
+       /* this will free all the listening sockets and all state that
+          is not associated with this new connection */
+       talloc_free(ev);
+
+       /* tdb needs special fork handling */
+       if (tdb_reopen_all() == -1) {
+               DEBUG(0,("standard_accept_connection: tdb_reopen_all failed.\n"));
+       }
+
+       /* Ensure that the forked children do not expose identical random streams */
+       set_need_random_reseed();
+
+       SETPROCTITLE(("task server_id[%d]", pid));
+
+       /* setup this new connection */
+       new_task(ev2, pid, private);
+
+       /* we can't return to the top level here, as that event context is gone,
+          so we now process events in the new event context until there are no
+          more to process */      
+       event_loop_wait(ev2);
+
+       talloc_free(ev2);
        exit(0);
 }
 
-/* called when a rpc connection goes down */
-static void terminate_rpc_connection(void *r, const char *reason) 
+
+/* called when a task goes down */
+static void standard_terminate(struct event_context *ev, const char *reason) 
 {
-       rpc_server_terminate(r);
+       DEBUG(2,("standard_terminate: reason[%s]\n",reason));
+
+       /* this init_iconv() has the effect of freeing the iconv context memory,
+          which makes leak checking easier */
+       init_iconv();
+
+       /* the secrets db should really hang off the connection structure */
+       secrets_shutdown();
+
+       talloc_free(ev);
+
        /* terminate this process */
        exit(0);
 }
 
-static int get_id(struct request_context *req)
+/* called to set a title of a task or connection */
+static void standard_set_title(struct event_context *ev, const char *title) 
 {
-       return (int)req->smb->pid;
+       if (title) {
+               SETPROCTITLE(("%s", title));
+       } else {
+               SETPROCTITLE((NULL));
+       }
 }
 
+static const struct model_ops standard_ops = {
+       .name                   = "standard",
+       .model_init             = standard_model_init,
+       .accept_connection      = standard_accept_connection,
+       .new_task               = standard_new_task,
+       .terminate              = standard_terminate,
+       .set_title              = standard_set_title,
+};
+
 /*
-  initialise the standard process model, registering ourselves with the model subsystem
+  initialise the standard process model, registering ourselves with the process model subsystem
  */
-void process_model_standard_init(void)
+NTSTATUS process_model_standard_init(void)
 {
-       struct model_ops ops;
-
-       ZERO_STRUCT(ops);
-       
-       /* fill in all the operations */
-       ops.model_startup = model_startup;
-       ops.accept_connection = accept_connection;
-       ops.accept_rpc_connection = accept_rpc_connection;
-       ops.terminate_connection = terminate_connection;
-       ops.terminate_rpc_connection = terminate_rpc_connection;
-       ops.get_id = get_id;
-
-       /* register ourselves with the process model subsystem. We register under the name 'standard'. */
-       register_process_model("standard", &ops);
+       return register_process_model(&standard_ops);
 }