source3/rpc_server/rpc_modules.c

   1 /*
   2  *  Unix SMB/CIFS implementation.
   3  *
   4  *  SMBD RPC modules
   5  *
   6  *  Copyright (c) 2015 Ralph Boehme <slow@samba.org>
   7  *
   8  *  This program is free software; you can redistribute it and/or modify
   9  *  it under the terms of the GNU General Public License as published by
  10  *  the Free Software Foundation; either version 3 of the License, or
  11  *  (at your option) any later version.
  12  *
  13  *  This program is distributed in the hope that it will be useful,
  14  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16  *  GNU General Public License for more details.
  17  *
  18  *  You should have received a copy of the GNU General Public License
  19  *  along with this program; if not, see <http://www.gnu.org/licenses/>.
  20  */
  21
  22 #include "includes.h"
  23 #include "rpc_server/rpc_modules.h"
  24
  25 #undef DBGC_CLASS
  26 #define DBGC_CLASS DBGC_RPC_SRV
  27
  28 static struct rpc_module *rpc_modules;
  29
  30 struct rpc_module {
  31         struct rpc_module *prev, *next;
  32         char *name;
  33         struct rpc_module_fns *fns;
  34 };
  35
  36 static struct rpc_module *find_rpc_module(const char *name)
  37 {
  38         struct rpc_module *module = NULL;
  39
  40         for (module = rpc_modules; module != NULL; module = module->next) {
  41                 if (strequal(module->name, name)) {
  42                         return module;
  43                 }
  44         }
  45
  46         return NULL;
  47 }
  48
  49 NTSTATUS register_rpc_module(struct rpc_module_fns *fns,
  50                              const char *name)
  51 {
  52         struct rpc_module *module = find_rpc_module(name);
  53
  54         if (module != NULL) {
  55                 DBG_ERR("RPC module %s already loaded!\n", name);
  56                 return NT_STATUS_OBJECT_NAME_COLLISION;
  57         }
  58
  59         module = SMB_XMALLOC_P(struct rpc_module);
  60         module->name = smb_xstrdup(name);
  61         module->fns = fns;
  62
  63         DLIST_ADD(rpc_modules, module);
  64         DBG_NOTICE("Successfully added RPC module '%s'\n", name);
  65
  66         return NT_STATUS_OK;
  67 }
  68
  69 bool setup_rpc_module(struct tevent_context *ev_ctx,
  70                       struct messaging_context *msg_ctx,
  71                       const char *name)
  72 {
  73         bool ok;
  74         struct rpc_module *module = find_rpc_module(name);
  75
  76         if (module == NULL) {
  77                 return false;
  78         }
  79
  80         ok = module->fns->setup(ev_ctx, msg_ctx);
  81         if (!ok) {
  82                 DBG_ERR("calling setup for %s failed\n", name);
  83         }
  84
  85         return true;
  86 }
  87
  88 bool setup_rpc_modules(struct tevent_context *ev_ctx,
  89                        struct messaging_context *msg_ctx)
  90 {
  91         bool ok;
  92         struct rpc_module *module = rpc_modules;
  93
  94         for (module = rpc_modules; module; module = module->next) {
  95                 ok = module->fns->setup(ev_ctx, msg_ctx);
  96                 if (!ok) {
  97                         DBG_ERR("calling setup for %s failed\n", module->name);
  98                 }
  99         }
 100
 101         return true;
 102 }
 103
 104 bool init_rpc_module(const char *name,
 105                      const struct rpc_srv_callbacks *rpc_srv_cb)
 106 {
 107         struct rpc_module *module = find_rpc_module(name);
 108         NTSTATUS status;
 109
 110         if (module == NULL) {
 111                 return false;
 112         }
 113
 114         status = module->fns->init(rpc_srv_cb);
 115         if (!NT_STATUS_IS_OK(status)) {
 116                 DBG_ERR("calling init for %s failed %s\n",
 117                         name, nt_errstr(status));
 118                 return false;
 119         }
 120
 121         return true;
 122 }
 123
 124 bool shutdown_rpc_module(const char *name)
 125 {
 126         struct rpc_module *module = find_rpc_module(name);
 127         NTSTATUS status;
 128
 129         if (module == NULL) {
 130                 return false;
 131         }
 132
 133         status = module->fns->shutdown();
 134         if (!NT_STATUS_IS_OK(status)) {
 135                 DBG_ERR("calling shutdown for %s failed %s\n",
 136                         name, nt_errstr(status));
 137                 return false;
 138         }
 139
 140         return true;
 141 }