source4 smdb: Add a post fork hook to the service API
[garming/samba-autobuild/.git] / source4 / smbd / process_single.c
index 86f6a38cd10d6e4641ac4adb65fe645d753e2152..242622b3b8fc7d071f3ea7240646b130445608d6 100644 (file)
@@ -9,7 +9,7 @@
    
    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
-   the Free Software Foundation; either version 2 of the License, or
+   the Free Software Foundation; either version 3 of the License, or
    (at your option) any later version.
    
    This program is distributed in the hope that it will be useful,
    GNU General Public License for more details.
    
    You should have received a copy of the GNU General Public License
-   along with this program; if not, write to the Free Software
-   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+   along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
 #include "includes.h"
 #include "smbd/process_model.h"
+#include "system/filesys.h"
+#include "cluster/cluster.h"
 
-/* For sepecifiying event context to GSSAPI below */
-#include "system/kerberos.h"
-#include "heimdal/lib/gssapi/gssapi_locl.h"
-
+NTSTATUS process_model_single_init(TALLOC_CTX *);
 
 /*
   called when the process model is selected
 */
-static void single_model_init(struct event_context *ev)
+static void single_model_init(void)
 {
-       /* Hack to ensure that GSSAPI uses the right event context */
-       gssapi_krb5_init_ev(ev);
 }
 
 /*
   called when a listening socket becomes readable. 
 */
-static void single_accept_connection(struct event_context *ev, 
-                                    struct socket_context *sock,
-                                    void (*new_conn)(struct event_context *, struct socket_context *, 
-                                                     uint32_t , void *), 
-                                    void *private)
+static void single_accept_connection(struct tevent_context *ev, 
+                                    struct loadparm_context *lp_ctx,
+                                    struct socket_context *listen_socket,
+                                    void (*new_conn)(struct tevent_context *, 
+                                                     struct loadparm_context *,
+                                                     struct socket_context *, 
+                                                     struct server_id, void *,
+                                                     void *),
+                                    void *private_data,
+                                    void *process_context)
 {
        NTSTATUS status;
-       struct socket_context *sock2;
+       struct socket_context *connected_socket;
+       pid_t pid = getpid();
 
        /* accept an incoming connection. */
-       status = socket_accept(sock, &sock2);
+       status = socket_accept(listen_socket, &connected_socket);
        if (!NT_STATUS_IS_OK(status)) {
-               DEBUG(0,("accept_connection_single: 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. If we don't
-                  then we will spin filling the log and causing more
-                  problems. We don't panic as this is probably a
-                  temporary resource constraint */
+               DBG_ERR("single_accept_connection: accept: %s\n",
+                       nt_errstr(status));
+               /* this looks strange, but is correct. 
+
+                  We can only be here if woken up from select, due to
+                  an incoming connection.
+
+                  We need to throttle things until the system clears
+                  enough resources to handle this new socket. 
+
+                  If we don't then we will spin filling the log and
+                  causing more problems. We don't panic as this is
+                  probably a temporary resource constraint */
                sleep(1);
                return;
        }
 
-       talloc_steal(private, sock);
-
-       new_conn(ev, sock2, socket_get_fd(sock2), private);
+       talloc_steal(private_data, connected_socket);
+
+       /*
+        * We use the PID so we cannot collide in with cluster ids
+        * generated in other single mode tasks, and, and won't
+        * collide with PIDs from process model standard because a the
+        * combination of pid/fd should be unique system-wide
+        */
+       new_conn(ev, lp_ctx, connected_socket,
+                cluster_id(pid, socket_get_fd(connected_socket)), private_data,
+                           process_context);
 }
 
 /*
   called to startup a new task
 */
-static void single_new_task(struct event_context *ev, 
-                           void (*new_task)(struct event_context *, uint32_t, void *), 
-                           void *private)
+static void single_new_task(struct tevent_context *ev,
+                           struct loadparm_context *lp_ctx,
+                           const char *service_name,
+                           struct task_server *(*new_task)(struct tevent_context *,
+                                            struct loadparm_context *,
+                                            struct server_id, void *, void *),
+                           void *private_data,
+                           const struct service_details *service_details,
+                           int from_parent_fd)
 {
-       static uint32_t taskid = 0x10000000;
-       new_task(ev, taskid++, private);
+       pid_t pid = getpid();
+       /* start our taskids at MAX_INT32, the first 2^31 tasks are is reserved for fd numbers */
+       static uint32_t taskid = INT32_MAX;
+       struct task_server *task = NULL;
+       /*
+        * We use the PID so we cannot collide in with cluster ids
+        * generated in other single mode tasks, and, and won't
+        * collide with PIDs from process model starndard because a the
+        * combination of pid/task_id should be unique system-wide
+        *
+        * Using the pid unaltered makes debugging of which process
+        * owns the messaging socket easier.
+        */
+       task = new_task(ev, lp_ctx, cluster_id(pid, taskid++), private_data, NULL);
+       if (task != NULL && service_details->post_fork != NULL) {
+               service_details->post_fork(task);
+       }
 }
 
 
 /* called when a task goes down */
-static void single_terminate(struct event_context *ev, const char *reason) 
+static void single_terminate(struct tevent_context *ev,
+                            struct loadparm_context *lp_ctx,
+                            const char *reason,
+                            void *process_context)
 {
-       DEBUG(2,("single_terminate: reason[%s]\n",reason));
+       DBG_NOTICE("single_terminate: reason[%s]\n",reason);
 }
 
 /* called to set a title of a task or connection */
-static void single_set_title(struct event_context *ev, const char *title) 
+static void single_set_title(struct tevent_context *ev, const char *title) 
 {
 }
 
-static const struct model_ops single_ops = {
+const struct model_ops single_ops = {
        .name                   = "single",
        .model_init             = single_model_init,
        .new_task               = single_new_task,
@@ -106,7 +146,7 @@ static const struct model_ops single_ops = {
   initialise the single process model, registering ourselves with the
   process model subsystem
  */
-NTSTATUS process_model_single_init(void)
+NTSTATUS process_model_single_init(TALLOC_CTX *ctx)
 {
        return register_process_model(&single_ops);
 }