source3/rpc_server/epmd.c

   1 /*
   2  *  Unix SMB/CIFS implementation.
   3  *
   4  *  SMBD RPC service callbacks
   5  *
   6  *  Copyright (c) 2011      Andreas Schneider <asn@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
  24 #include "serverid.h"
  25 #include "ntdomain.h"
  26 #include "messages.h"
  27
  28 #include "lib/util/util_process.h"
  29
  30 #include "librpc/rpc/dcerpc_ep.h"
  31 #include "../librpc/gen_ndr/srv_epmapper.h"
  32 #include "rpc_server/rpc_server.h"
  33 #include "rpc_server/rpc_sock_helper.h"
  34 #include "rpc_server/epmapper/srv_epmapper.h"
  35
  36 #define DAEMON_NAME "epmd"
  37
  38 void start_epmd(struct tevent_context *ev_ctx,
  39                 struct messaging_context *msg_ctx);
  40
  41 static void epmd_reopen_logs(void)
  42 {
  43         char *lfile = lp_logfile(talloc_tos());
  44         int rc;
  45
  46         if (lfile == NULL || lfile[0] == '\0') {
  47                 rc = asprintf(&lfile, "%s/log.%s", get_dyn_LOGFILEBASE(), DAEMON_NAME);
  48                 if (rc > 0) {
  49                         lp_set_logfile(lfile);
  50                         SAFE_FREE(lfile);
  51                 }
  52         } else {
  53                 if (strstr(lfile, DAEMON_NAME) == NULL) {
  54                         rc = asprintf(&lfile, "%s.%s",
  55                                       lp_logfile(talloc_tos()), DAEMON_NAME);
  56                         if (rc > 0) {
  57                                 lp_set_logfile(lfile);
  58                                 SAFE_FREE(lfile);
  59                         }
  60                 }
  61         }
  62
  63         reopen_logs();
  64 }
  65
  66 static void epmd_smb_conf_updated(struct messaging_context *msg,
  67                                   void *private_data,
  68                                   uint32_t msg_type,
  69                                   struct server_id server_id,
  70                                   DATA_BLOB *data)
  71 {
  72         DEBUG(10, ("Got message saying smb.conf was updated. Reloading.\n"));
  73         change_to_root_user();
  74         epmd_reopen_logs();
  75 }
  76
  77 static void epmd_sig_term_handler(struct tevent_context *ev,
  78                                   struct tevent_signal *se,
  79                                   int signum,
  80                                   int count,
  81                                   void *siginfo,
  82                                   void *private_data)
  83 {
  84         rpc_epmapper_shutdown();
  85
  86         exit_server_cleanly("termination signal");
  87 }
  88
  89 static void epmd_setup_sig_term_handler(struct tevent_context *ev_ctx)
  90 {
  91         struct tevent_signal *se;
  92
  93         se = tevent_add_signal(ev_ctx,
  94                                ev_ctx,
  95                                SIGTERM, 0,
  96                                epmd_sig_term_handler,
  97                                NULL);
  98         if (se == NULL) {
  99                 exit_server("failed to setup SIGTERM handler");
 100         }
 101 }
 102
 103 static void epmd_sig_hup_handler(struct tevent_context *ev,
 104                                     struct tevent_signal *se,
 105                                     int signum,
 106                                     int count,
 107                                     void *siginfo,
 108                                     void *private_data)
 109 {
 110         change_to_root_user();
 111
 112         DEBUG(1,("Reloading printers after SIGHUP\n"));
 113         epmd_reopen_logs();
 114 }
 115
 116 static void epmd_setup_sig_hup_handler(struct tevent_context *ev_ctx,
 117                                        struct messaging_context *msg_ctx)
 118 {
 119         struct tevent_signal *se;
 120
 121         se = tevent_add_signal(ev_ctx,
 122                                ev_ctx,
 123                                SIGHUP, 0,
 124                                epmd_sig_hup_handler,
 125                                msg_ctx);
 126         if (se == NULL) {
 127                 exit_server("failed to setup SIGHUP handler");
 128         }
 129 }
 130
 131 static bool epmapper_shutdown_cb(void *ptr) {
 132         srv_epmapper_cleanup();
 133
 134         return true;
 135 }
 136
 137 void start_epmd(struct tevent_context *ev_ctx,
 138                 struct messaging_context *msg_ctx)
 139 {
 140         struct rpc_srv_callbacks epmapper_cb;
 141         NTSTATUS status;
 142         pid_t pid;
 143         bool ok;
 144         int rc;
 145
 146         epmapper_cb.init = NULL;
 147         epmapper_cb.shutdown = epmapper_shutdown_cb;
 148         epmapper_cb.private_data = NULL;
 149
 150         DEBUG(1, ("Forking Endpoint Mapper Daemon\n"));
 151
 152         pid = fork();
 153
 154         if (pid == -1) {
 155                 DEBUG(0, ("Failed to fork Endpoint Mapper [%s], aborting ...\n",
 156                           strerror(errno)));
 157                 exit(1);
 158         }
 159
 160         if (pid) {
 161                 /* parent */
 162                 return;
 163         }
 164
 165         status = smbd_reinit_after_fork(msg_ctx, ev_ctx, true);
 166         if (!NT_STATUS_IS_OK(status)) {
 167                 DEBUG(0,("reinit_after_fork() failed\n"));
 168                 smb_panic("reinit_after_fork() failed");
 169         }
 170
 171         prctl_set_comment("epmd");
 172
 173         epmd_reopen_logs();
 174
 175         epmd_setup_sig_term_handler(ev_ctx);
 176         epmd_setup_sig_hup_handler(ev_ctx, msg_ctx);
 177
 178         ok = serverid_register(messaging_server_id(msg_ctx),
 179                                FLAG_MSG_GENERAL |
 180                                FLAG_MSG_PRINT_GENERAL);
 181         if (!ok) {
 182                 DEBUG(0, ("Failed to register serverid in epmd!\n"));
 183                 exit(1);
 184         }
 185
 186         messaging_register(msg_ctx,
 187                            ev_ctx,
 188                            MSG_SMB_CONF_UPDATED,
 189                            epmd_smb_conf_updated);
 190
 191         status = rpc_epmapper_init(&epmapper_cb);
 192         if (!NT_STATUS_IS_OK(status)) {
 193                 DEBUG(0, ("Failed to register epmd rpc interface! (%s)\n",
 194                           nt_errstr(status)));
 195                 exit(1);
 196         }
 197
 198         status = rpc_setup_tcpip_sockets(ev_ctx,
 199                                          msg_ctx,
 200                                          &ndr_table_epmapper,
 201                                          NULL,
 202                                          135);
 203         if (!NT_STATUS_IS_OK(status)) {
 204                 DEBUG(0, ("Failed to open epmd tcpip sockets!\n"));
 205                 exit(1);
 206         }
 207
 208         ok = setup_dcerpc_ncalrpc_socket(ev_ctx,
 209                                          msg_ctx,
 210                                          "EPMAPPER",
 211                                          srv_epmapper_delete_endpoints);
 212         if (!ok) {
 213                 DEBUG(0, ("Failed to open epmd ncalrpc pipe!\n"));
 214                 exit(1);
 215         }
 216
 217         ok = setup_named_pipe_socket("epmapper", ev_ctx, msg_ctx);
 218         if (!ok) {
 219                 DEBUG(0, ("Failed to open epmd named pipe!\n"));
 220                 exit(1);
 221         }
 222
 223         DEBUG(1, ("Endpoint Mapper Daemon Started (%u)\n", (unsigned int)getpid()));
 224
 225         /* loop forever */
 226         rc = tevent_loop_wait(ev_ctx);
 227
 228         /* should not be reached */
 229         DEBUG(0,("background_queue: tevent_loop_wait() exited with %d - %s\n",
 230                  rc, (rc == 0) ? "out of events" : strerror(errno)));
 231
 232         exit(1);
 233 }
 234
 235 /* vim: set ts=8 sw=8 noet cindent ft=c.doxygen: */