s4: Add TALLOC_CTX * to register_server_service().

[garming/samba-autobuild/.git] / source4 / nbt_server / nbt_server.c

diff --git a/source4/nbt_server/nbt_server.c b/source4/nbt_server/nbt_server.c
index 1a8872b458c88baa1efd30d548523d8aa3c1ddf9..2196bb045e9a8046a204dcc4b9761ea0e59c808d 100644 (file)
--- a/source4/nbt_server/nbt_server.c
+++ b/source4/nbt_server/nbt_server.c
@@ -7,7 +7,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,
@@ -16,59 +16,89 @@
    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 "lib/events/events.h"
 #include "smbd/service_task.h"
+#include "smbd/service.h"
 #include "nbt_server/nbt_server.h"
+#include "nbt_server/wins/winsserver.h"
+#include "system/network.h"
+#include "lib/socket/netif.h"
+#include "auth/auth.h"
+#include "dsdb/samdb/samdb.h"
+#include "param/param.h"
 
+NTSTATUS server_service_nbtd_init(TALLOC_CTX *);
 
 /*
   startup the nbtd task
 */
 static void nbtd_task_init(struct task_server *task)
 {
-       struct nbt_server *nbtsrv;
+       struct nbtd_server *nbtsrv;
        NTSTATUS status;
+       struct interface *ifaces;
 
-       nbtsrv = talloc(task, struct nbt_server);
+       load_interface_list(task, task->lp_ctx, &ifaces);
+
+       if (iface_list_count(ifaces) == 0) {
+               task_server_terminate(task, "nbtd: no network interfaces configured", false);
+               return;
+       }
+
+       if (lpcfg_disable_netbios(task->lp_ctx)) {
+               task_server_terminate(task, "nbtd: 'disable netbios = yes' set in smb.conf, shutting down nbt server", false);
+               return;
+       }
+
+       task_server_set_title(task, "task[nbtd]");
+
+       nbtsrv = talloc(task, struct nbtd_server);
        if (nbtsrv == NULL) {
-               task_terminate(task, "nbtd: out of memory");
+               task_server_terminate(task, "nbtd: out of memory", true);
                return;
        }
 
        nbtsrv->task            = task;
        nbtsrv->interfaces      = NULL;
        nbtsrv->bcast_interface = NULL;
+       nbtsrv->wins_interface  = NULL;
 
        /* start listening on the configured network interfaces */
-       status = nbt_startup_interfaces(nbtsrv);
+       status = nbtd_startup_interfaces(nbtsrv, task->lp_ctx, ifaces);
        if (!NT_STATUS_IS_OK(status)) {
-               task_terminate(task, "nbtd failed to setup interfaces");
+               task_server_terminate(task, "nbtd failed to setup interfaces", true);
                return;
        }
 
-       /* start the process of registering our names on all interfaces */
-       nbt_register_names(nbtsrv);
-}
+       nbtsrv->sam_ctx = samdb_connect(nbtsrv, task->event_ctx, task->lp_ctx, system_session(task->lp_ctx), 0);
+       if (nbtsrv->sam_ctx == NULL) {
+               task_server_terminate(task, "nbtd failed to open samdb", true);
+               return;
+       }
+
+       /* start the WINS server, if appropriate */
+       status = nbtd_winsserver_init(nbtsrv);
+       if (!NT_STATUS_IS_OK(status)) {
+               task_server_terminate(task, "nbtd failed to start WINS server", true);
+               return;
+       }
 
+       nbtd_register_irpc(nbtsrv);
 
-/*
-  initialise the nbt server
- */
-static NTSTATUS nbtd_init(struct event_context *event_ctx, const struct model_ops *model_ops)
-{
-       return task_server_startup(event_ctx, model_ops, nbtd_task_init);
+       /* start the process of registering our names on all interfaces */
+       nbtd_register_names(nbtsrv);
+
+       irpc_add_name(task->msg_ctx, "nbt_server");
 }
 
 
 /*
   register ourselves as a available server
 */
-NTSTATUS server_service_nbtd_init(void)
+NTSTATUS server_service_nbtd_init(TALLOC_CTX *ctx)
 {
-       return register_server_service("nbt", nbtd_init);
+       return register_server_service(ctx, "nbt", nbtd_task_init);
 }