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 "../librpc/gen_ndr/messaging.h"
  26 #include "../librpc/gen_ndr/srv_epmapper.h"
  27 #include "rpc_server/rpc_server.h"
  28
  29 #define DAEMON_NAME "epmd"
  30
  31 void start_epmd(struct tevent_context *ev_ctx,
  32                 struct messaging_context *msg_ctx);
  33
  34 static bool epmd_open_sockets(struct tevent_context *ev_ctx,
  35                               struct messaging_context *msg_ctx)
  36 {
  37         uint32_t num_ifs = iface_count();
  38         uint16_t port;
  39         uint32_t i;
  40
  41         if (lp_interfaces() && lp_bind_interfaces_only()) {
  42                 /*
  43                  * We have been given an interfaces line, and been told to only
  44                  * bind to those interfaces. Create a socket per interface and
  45                  * bind to only these.
  46                  */
  47
  48                 /* Now open a listen socket for each of the interfaces. */
  49                 for(i = 0; i < num_ifs; i++) {
  50                         const struct sockaddr_storage *ifss =
  51                                         iface_n_sockaddr_storage(i);
  52
  53                         port = setup_dcerpc_ncacn_tcpip_socket(ev_ctx,
  54                                                                msg_ctx,
  55                                                                ndr_table_epmapper.syntax_id,
  56                                                                ifss,
  57                                                                135);
  58                         if (port == 0) {
  59                                 return false;
  60                         }
  61                 }
  62         } else {
  63                 const char *sock_addr = lp_socket_address();
  64                 const char *sock_ptr;
  65                 char *sock_tok;
  66
  67                 if (strequal(sock_addr, "0.0.0.0") ||
  68                     strequal(sock_addr, "::")) {
  69 #if HAVE_IPV6
  70                         sock_addr = "::";
  71 #else
  72                         sock_addr = "0.0.0.0";
  73 #endif
  74                 }
  75
  76                 for (sock_ptr = sock_addr;
  77                      next_token_talloc(talloc_tos(), &sock_ptr, &sock_tok, " \t,");
  78                     ) {
  79                         struct sockaddr_storage ss;
  80
  81                         /* open an incoming socket */
  82                         if (!interpret_string_addr(&ss,
  83                                                    sock_tok,
  84                                                    AI_NUMERICHOST|AI_PASSIVE)) {
  85                                 continue;
  86                         }
  87
  88                         port = setup_dcerpc_ncacn_tcpip_socket(ev_ctx,
  89                                                                msg_ctx,
  90                                                                ndr_table_epmapper.syntax_id,
  91                                                                &ss,
  92                                                                135);
  93                         if (port == 0) {
  94                                 return false;
  95                         }
  96                 }
  97         }
  98
  99         return true;
 100 }
 101
 102 static void epmd_reopen_logs(void)
 103 {
 104         char *lfile = lp_logfile();
 105         int rc;
 106
 107         if (lfile == NULL || lfile[0] == '\0') {
 108                 rc = asprintf(&lfile, "%s/log.%s", get_dyn_LOGFILEBASE(), DAEMON_NAME);
 109                 if (rc > 0) {
 110                         lp_set_logfile(lfile);
 111                         SAFE_FREE(lfile);
 112                 }
 113         } else {
 114                 if (strstr(lfile, DAEMON_NAME) == NULL) {
 115                         rc = asprintf(&lfile, "%s.%s", lp_logfile(), DAEMON_NAME);
 116                         if (rc > 0) {
 117                                 lp_set_logfile(lfile);
 118                                 SAFE_FREE(lfile);
 119                         }
 120                 }
 121         }
 122
 123         reopen_logs();
 124 }
 125
 126 static void epmd_smb_conf_updated(struct messaging_context *msg,
 127                                   void *private_data,
 128                                   uint32_t msg_type,
 129                                   struct server_id server_id,
 130                                   DATA_BLOB *data)
 131 {
 132         struct tevent_context *ev_ctx =
 133                 talloc_get_type_abort(private_data, struct tevent_context);
 134
 135         DEBUG(10, ("Got message saying smb.conf was updated. Reloading.\n"));
 136         change_to_root_user();
 137         epmd_reopen_logs();
 138 }
 139
 140 static void epmd_sig_term_handler(struct tevent_context *ev,
 141                                   struct tevent_signal *se,
 142                                   int signum,
 143                                   int count,
 144                                   void *siginfo,
 145                                   void *private_data)
 146 {
 147         exit_server_cleanly("termination signal");
 148 }
 149
 150 static void epmd_setup_sig_term_handler(struct tevent_context *ev_ctx)
 151 {
 152         struct tevent_signal *se;
 153
 154         se = tevent_add_signal(ev_ctx,
 155                                ev_ctx,
 156                                SIGTERM, 0,
 157                                epmd_sig_term_handler,
 158                                NULL);
 159         if (se == NULL) {
 160                 exit_server("failed to setup SIGTERM handler");
 161         }
 162 }
 163
 164 static void epmd_sig_hup_handler(struct tevent_context *ev,
 165                                     struct tevent_signal *se,
 166                                     int signum,
 167                                     int count,
 168                                     void *siginfo,
 169                                     void *private_data)
 170 {
 171         struct messaging_context *msg_ctx = talloc_get_type_abort(private_data,
 172                                                                   struct messaging_context);
 173
 174         change_to_root_user();
 175
 176         DEBUG(1,("Reloading printers after SIGHUP\n"));
 177         epmd_reopen_logs();
 178 }
 179
 180 static void epmd_setup_sig_hup_handler(struct tevent_context *ev_ctx,
 181                                        struct messaging_context *msg_ctx)
 182 {
 183         struct tevent_signal *se;
 184
 185         se = tevent_add_signal(ev_ctx,
 186                                ev_ctx,
 187                                SIGHUP, 0,
 188                                epmd_sig_hup_handler,
 189                                msg_ctx);
 190         if (se == NULL) {
 191                 exit_server("failed to setup SIGHUP handler");
 192         }
 193 }
 194
 195 void start_epmd(struct tevent_context *ev_ctx,
 196                 struct messaging_context *msg_ctx)
 197 {
 198         NTSTATUS status;
 199         pid_t pid;
 200         bool ok;
 201         int rc;
 202
 203         DEBUG(1, ("Forking Endpoint Mapper Daemon\n"));
 204
 205         pid = sys_fork();
 206
 207         if (pid == -1) {
 208                 DEBUG(0, ("Failed to fork Endpoint Mapper [%s], aborting ...\n",
 209                           strerror(errno)));
 210                 exit(1);
 211         }
 212
 213         if (pid) {
 214                 /* parent */
 215                 return;
 216         }
 217
 218         /* child */
 219         close_low_fds(false);
 220
 221         status = reinit_after_fork(msg_ctx,
 222                                    ev_ctx,
 223                                    procid_self(),
 224                                    true);
 225         if (!NT_STATUS_IS_OK(status)) {
 226                 DEBUG(0,("reinit_after_fork() failed\n"));
 227                 smb_panic("reinit_after_fork() failed");
 228         }
 229
 230         epmd_reopen_logs();
 231
 232         epmd_setup_sig_term_handler(ev_ctx);
 233         epmd_setup_sig_hup_handler(ev_ctx, msg_ctx);
 234
 235         ok = serverid_register(procid_self(),
 236                                FLAG_MSG_GENERAL|FLAG_MSG_SMBD
 237                                |FLAG_MSG_PRINT_GENERAL);
 238         if (!ok) {
 239                 DEBUG(0, ("Failed to register serverid in epmd!\n"));
 240                 exit(1);
 241         }
 242
 243         messaging_register(msg_ctx,
 244                            ev_ctx,
 245                            MSG_SMB_CONF_UPDATED,
 246                            epmd_smb_conf_updated);
 247
 248         status = rpc_epmapper_init(NULL);
 249         if (!NT_STATUS_IS_OK(status)) {
 250                 DEBUG(0, ("Failed to register epmd rpc inteface! (%s)\n",
 251                           nt_errstr(status)));
 252                 exit(1);
 253         }
 254
 255         ok = setup_dcerpc_ncalrpc_socket(ev_ctx,
 256                                          msg_ctx,
 257                                          ndr_table_epmapper.syntax_id,
 258                                          "EPMAPPER");
 259         if (!ok) {
 260                 DEBUG(0, ("Failed to open epmd ncalrpc pipe!\n"));
 261                 exit(1);
 262         }
 263
 264         ok = epmd_open_sockets(ev_ctx, msg_ctx);
 265         if (!ok) {
 266                 DEBUG(0, ("Failed to open epmd tcpip sockets!\n"));
 267                 exit(1);
 268         }
 269
 270         ok = setup_named_pipe_socket("epmapper", ev_ctx);
 271         if (!ok) {
 272                 DEBUG(0, ("Failed to open epmd named pipe!\n"));
 273                 exit(1);
 274         }
 275
 276         DEBUG(1, ("Endpoint Mapper Daemon Started (%d)\n", getpid()));
 277
 278         /* loop forever */
 279         rc = tevent_loop_wait(ev_ctx);
 280
 281         /* should not be reached */
 282         DEBUG(0,("background_queue: tevent_loop_wait() exited with %d - %s\n",
 283                  rc, (rc == 0) ? "out of events" : strerror(errno)));
 284
 285         exit(1);
 286 }
 287
 288 /* vim: set ts=8 sw=8 noet cindent ft=c.doxygen: */