s4-server: kill main daemon if a task fails to initialise

When one of our core tasks fails to initialise it can now ask for the
server as a whole to die, rather than limping along in a degraded
state.

1 files changed, 46 insertions, 0 deletions

diff --git a/source4/smbd/server.c b/source4/smbd/server.c
index 73dbec0120..d150161e05 100644
--- a/source4/smbd/server.c
+++ b/source4/smbd/server.c
@@ -40,6 +40,9 @@
 #include "param/param.h"
 #include "dsdb/samdb/samdb.h"
 #include "auth/session.h"
+#include "lib/messaging/irpc.h"
+#include "librpc/gen_ndr/ndr_irpc.h"
+#include "cluster/cluster.h"
 
 /*
   recursively delete a directory tree
@@ -192,6 +195,43 @@ static void prime_samdb_schema(struct tevent_context *event_ctx)
 	talloc_free(samdb_context);
 }
 
+
+/*
+  called when a fatal condition occurs in a child task
+ */
+static NTSTATUS samba_terminate(struct irpc_message *msg, 
+				struct samba_terminate *r)
+{
+	DEBUG(0,("samba_terminate: %s\n", r->in.reason));
+	exit(1);
+}
+
+/*
+  setup messaging for the top level samba (parent) task
+ */
+static NTSTATUS setup_parent_messaging(struct tevent_context *event_ctx, 
+				       struct loadparm_context *lp_ctx)
+{
+	struct messaging_context *msg;
+	NTSTATUS status;
+
+	msg = messaging_init(talloc_autofree_context(), 
+			     lp_messaging_path(event_ctx, lp_ctx),
+			     cluster_id(0, SAMBA_PARENT_TASKID),
+			     lp_iconv_convenience(lp_ctx),
+			     event_ctx);
+	NT_STATUS_HAVE_NO_MEMORY(msg);
+
+	irpc_add_name(msg, "samba");
+
+	status = IRPC_REGISTER(msg, irpc, SAMBA_TERMINATE,
+			       samba_terminate, NULL);
+
+	return status;
+}
+
+
+
 /*
  main server.
 */
@@ -363,6 +403,12 @@ static int binary_smbd_main(const char *binary_name, int argc, const char *argv[
 
 	prime_samdb_schema(event_ctx);
 
+	status = setup_parent_messaging(event_ctx, cmdline_lp_ctx);
+	if (!NT_STATUS_IS_OK(status)) {
+		DEBUG(0,("Failed to setup parent messaging - %s\n", nt_errstr(status)));
+		return 1;
+	}
+
 	DEBUG(0,("%s: using '%s' process model\n", binary_name, model));
 	status = server_service_startup(event_ctx, cmdline_lp_ctx, model, 
 					lp_server_services(cmdline_lp_ctx));