source4/rpc_server/service_rpc.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3
   4    smbd-specific dcerpc server code
   5
   6    Copyright (C) Andrew Tridgell 2003-2005
   7    Copyright (C) Stefan (metze) Metzmacher 2004-2005
   8    Copyright (C) Jelmer Vernooij <jelmer@samba.org> 2004,2007
   9
  10    This program is free software; you can redistribute it and/or modify
  11    it under the terms of the GNU General Public License as published by
  12    the Free Software Foundation; either version 3 of the License, or
  13    (at your option) any later version.
  14
  15    This program is distributed in the hope that it will be useful,
  16    but WITHOUT ANY WARRANTY; without even the implied warranty of
  17    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18    GNU General Public License for more details.
  19
  20    You should have received a copy of the GNU General Public License
  21    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  22 */
  23
  24 #include "includes.h"
  25 #include "librpc/gen_ndr/ndr_dcerpc.h"
  26 #include "auth/auth.h"
  27 #include "auth/gensec/gensec.h"
  28 #include "lib/util/dlinklist.h"
  29 #include "rpc_server/dcerpc_server.h"
  30 #include "lib/events/events.h"
  31 #include "smbd/service_task.h"
  32 #include "smbd/service_stream.h"
  33 #include "smbd/service.h"
  34 #include "system/filesys.h"
  35 #include "libcli/security/security.h"
  36 #include "lib/socket/socket.h"
  37 #include "lib/messaging/irpc.h"
  38 #include "system/network.h"
  39 #include "lib/socket/netif.h"
  40 #include "build.h"
  41 #include "param/param.h"
  42
  43 struct dcesrv_socket_context {
  44         const struct dcesrv_endpoint *endpoint;
  45         struct dcesrv_context *dcesrv_ctx;
  46 };
  47
  48 /*
  49   write_fn callback for dcesrv_output()
  50 */
  51 static NTSTATUS dcerpc_write_fn(void *private_data, DATA_BLOB *out, size_t *nwritten)
  52 {
  53         NTSTATUS status;
  54         struct socket_context *sock = talloc_get_type(private_data, struct socket_context);
  55         size_t sendlen;
  56
  57         status = socket_send(sock, out, &sendlen);
  58         NT_STATUS_IS_ERR_RETURN(status);
  59
  60         *nwritten = sendlen;
  61         return status;
  62 }
  63
  64 static void dcesrv_terminate_connection(struct dcesrv_connection *dce_conn, const char *reason)
  65 {
  66         struct stream_connection *srv_conn;
  67         srv_conn = talloc_get_type(dce_conn->transport.private_data,
  68                                    struct stream_connection);
  69
  70         stream_terminate_connection(srv_conn, reason);
  71 }
  72
  73 static void dcesrv_sock_report_output_data(struct dcesrv_connection *dcesrv_conn)
  74 {
  75         struct stream_connection *srv_conn;
  76         srv_conn = talloc_get_type(dcesrv_conn->transport.private_data,
  77                                    struct stream_connection);
  78
  79         if (srv_conn && srv_conn->event.fde) {
  80                 EVENT_FD_WRITEABLE(srv_conn->event.fde);
  81         }
  82 }
  83
  84 static struct socket_address *dcesrv_sock_get_my_addr(struct dcesrv_connection *dcesrv_conn, TALLOC_CTX *mem_ctx)
  85 {
  86         struct stream_connection *srv_conn;
  87         srv_conn = talloc_get_type(dcesrv_conn->transport.private_data,
  88                                    struct stream_connection);
  89
  90         return socket_get_my_addr(srv_conn->socket, mem_ctx);
  91 }
  92
  93 static struct socket_address *dcesrv_sock_get_peer_addr(struct dcesrv_connection *dcesrv_conn, TALLOC_CTX *mem_ctx)
  94 {
  95         struct stream_connection *srv_conn;
  96         srv_conn = talloc_get_type(dcesrv_conn->transport.private_data,
  97                                    struct stream_connection);
  98
  99         return socket_get_peer_addr(srv_conn->socket, mem_ctx);
 100 }
 101
 102 static void dcesrv_sock_accept(struct stream_connection *srv_conn)
 103 {
 104         NTSTATUS status;
 105         struct dcesrv_socket_context *dcesrv_sock =
 106                 talloc_get_type(srv_conn->private, struct dcesrv_socket_context);
 107         struct dcesrv_connection *dcesrv_conn = NULL;
 108         struct auth_session_info *session_info = NULL;
 109
 110         status = auth_anonymous_session_info(srv_conn, &session_info);
 111         if (!NT_STATUS_IS_OK(status)) {
 112                 DEBUG(0,("dcesrv_sock_accept: auth_anonymous_session_info failed: %s\n",
 113                         nt_errstr(status)));
 114                 stream_terminate_connection(srv_conn, nt_errstr(status));
 115                 return;
 116         }
 117
 118         status = dcesrv_endpoint_connect(dcesrv_sock->dcesrv_ctx,
 119                                          srv_conn,
 120                                          dcesrv_sock->endpoint,
 121                                          session_info,
 122                                          srv_conn->event.ctx,
 123                                          srv_conn->msg_ctx,
 124                                          srv_conn->server_id,
 125                                          DCESRV_CALL_STATE_FLAG_MAY_ASYNC,
 126                                          &dcesrv_conn);
 127         if (!NT_STATUS_IS_OK(status)) {
 128                 DEBUG(0,("dcesrv_sock_accept: dcesrv_endpoint_connect failed: %s\n",
 129                         nt_errstr(status)));
 130                 stream_terminate_connection(srv_conn, nt_errstr(status));
 131                 return;
 132         }
 133
 134         dcesrv_conn->transport.private_data             = srv_conn;
 135         dcesrv_conn->transport.report_output_data       = dcesrv_sock_report_output_data;
 136         dcesrv_conn->transport.get_my_addr              = dcesrv_sock_get_my_addr;
 137         dcesrv_conn->transport.get_peer_addr            = dcesrv_sock_get_peer_addr;
 138
 139         srv_conn->private = dcesrv_conn;
 140
 141         irpc_add_name(srv_conn->msg_ctx, "rpc_server");
 142
 143         return;
 144 }
 145
 146 static void dcesrv_sock_recv(struct stream_connection *conn, uint16_t flags)
 147 {
 148         NTSTATUS status;
 149         struct dcesrv_connection *dce_conn = talloc_get_type(conn->private, struct dcesrv_connection);
 150         DATA_BLOB tmp_blob;
 151         size_t nread;
 152
 153         if (dce_conn->processing) {
 154                 EVENT_FD_NOT_READABLE(conn->event.fde);
 155                 return;
 156         }
 157
 158         tmp_blob = data_blob_talloc(conn->socket, NULL, 0x1000);
 159         if (tmp_blob.data == NULL) {
 160                 dcesrv_terminate_connection(dce_conn, "out of memory");
 161                 return;
 162         }
 163
 164         status = socket_recv(conn->socket, tmp_blob.data, tmp_blob.length, &nread);
 165         if (NT_STATUS_IS_ERR(status)) {
 166                 dcesrv_terminate_connection(dce_conn, nt_errstr(status));
 167                 return;
 168         }
 169         if (nread == 0) {
 170                 talloc_free(tmp_blob.data);
 171                 return;
 172         }
 173
 174         tmp_blob.length = nread;
 175
 176         dce_conn->processing = True;
 177         status = dcesrv_input(dce_conn, &tmp_blob);
 178         dce_conn->processing = False;
 179         talloc_free(tmp_blob.data);
 180
 181         EVENT_FD_READABLE(conn->event.fde);
 182
 183         if (!NT_STATUS_IS_OK(status)) {
 184                 dcesrv_terminate_connection(dce_conn, nt_errstr(status));
 185                 return;
 186         }
 187
 188         if (dce_conn->call_list && dce_conn->call_list->replies) {
 189                 EVENT_FD_WRITEABLE(conn->event.fde);
 190         }
 191 }
 192
 193 static void dcesrv_sock_send(struct stream_connection *conn, uint16_t flags)
 194 {
 195         struct dcesrv_connection *dce_conn = talloc_get_type(conn->private, struct dcesrv_connection);
 196         NTSTATUS status;
 197
 198         status = dcesrv_output(dce_conn, conn->socket, dcerpc_write_fn);
 199         if (NT_STATUS_IS_ERR(status)) {
 200                 dcesrv_terminate_connection(dce_conn, "eof on socket");
 201                 return;
 202         }
 203
 204         if (!dce_conn->call_list || !dce_conn->call_list->replies) {
 205                 EVENT_FD_NOT_WRITEABLE(conn->event.fde);
 206         }
 207 }
 208
 209
 210 static const struct stream_server_ops dcesrv_stream_ops = {
 211         .name                   = "rpc",
 212         .accept_connection      = dcesrv_sock_accept,
 213         .recv_handler           = dcesrv_sock_recv,
 214         .send_handler           = dcesrv_sock_send,
 215 };
 216
 217
 218
 219 NTSTATUS dcesrv_add_ep_unix(struct dcesrv_context *dce_ctx, struct dcesrv_endpoint *e,
 220                             struct event_context *event_ctx, const struct model_ops *model_ops)
 221 {
 222         struct dcesrv_socket_context *dcesrv_sock;
 223         uint16_t port = 1;
 224         NTSTATUS status;
 225
 226         dcesrv_sock = talloc(event_ctx, struct dcesrv_socket_context);
 227         NT_STATUS_HAVE_NO_MEMORY(dcesrv_sock);
 228
 229         /* remember the endpoint of this socket */
 230         dcesrv_sock->endpoint           = e;
 231         dcesrv_sock->dcesrv_ctx         = talloc_reference(dcesrv_sock, dce_ctx);
 232
 233         status = stream_setup_socket(event_ctx, model_ops, &dcesrv_stream_ops,
 234                                      "unix", e->ep_description->endpoint, &port,
 235                                      dcesrv_sock);
 236         if (!NT_STATUS_IS_OK(status)) {
 237                 DEBUG(0,("service_setup_stream_socket(path=%s) failed - %s\n",
 238                          e->ep_description->endpoint, nt_errstr(status)));
 239         }
 240
 241         return status;
 242 }
 243
 244 NTSTATUS dcesrv_add_ep_ncalrpc(struct dcesrv_context *dce_ctx, struct dcesrv_endpoint *e,
 245                                struct event_context *event_ctx, const struct model_ops *model_ops)
 246 {
 247         struct dcesrv_socket_context *dcesrv_sock;
 248         uint16_t port = 1;
 249         char *full_path;
 250         NTSTATUS status;
 251
 252         if (!e->ep_description->endpoint) {
 253                 /* No identifier specified: use DEFAULT.
 254                  * DO NOT hardcode this value anywhere else. Rather, specify
 255                  * no endpoint and let the epmapper worry about it. */
 256                 e->ep_description->endpoint = talloc_strdup(dce_ctx, "DEFAULT");
 257         }
 258
 259         full_path = talloc_asprintf(dce_ctx, "%s/%s", lp_ncalrpc_dir(), e->ep_description->endpoint);
 260
 261         dcesrv_sock = talloc(event_ctx, struct dcesrv_socket_context);
 262         NT_STATUS_HAVE_NO_MEMORY(dcesrv_sock);
 263
 264         /* remember the endpoint of this socket */
 265         dcesrv_sock->endpoint           = e;
 266         dcesrv_sock->dcesrv_ctx         = talloc_reference(dcesrv_sock, dce_ctx);
 267
 268         status = stream_setup_socket(event_ctx, model_ops, &dcesrv_stream_ops,
 269                                      "unix", full_path, &port, dcesrv_sock);
 270         if (!NT_STATUS_IS_OK(status)) {
 271                 DEBUG(0,("service_setup_stream_socket(identifier=%s,path=%s) failed - %s\n",
 272                          e->ep_description->endpoint, full_path, nt_errstr(status)));
 273         }
 274         return status;
 275 }
 276
 277
 278 /*
 279   add a socket address to the list of events, one event per dcerpc endpoint
 280 */
 281 static NTSTATUS add_socket_rpc_pipe_iface(struct dcesrv_context *dce_ctx, struct dcesrv_endpoint *e,
 282                                          struct event_context *event_ctx, const struct model_ops *model_ops)
 283 {
 284         struct dcesrv_socket_context *dcesrv_sock;
 285         NTSTATUS status;
 286
 287         if (e->ep_description->endpoint == NULL) {
 288                 DEBUG(0, ("Endpoint mandatory for named pipes\n"));
 289                 return NT_STATUS_INVALID_PARAMETER;
 290         }
 291
 292         dcesrv_sock = talloc(event_ctx, struct dcesrv_socket_context);
 293         NT_STATUS_HAVE_NO_MEMORY(dcesrv_sock);
 294
 295         /* remember the endpoint of this socket */
 296         dcesrv_sock->endpoint           = e;
 297         dcesrv_sock->dcesrv_ctx         = talloc_reference(dcesrv_sock, dce_ctx);
 298
 299         status = NT_STATUS_OK;
 300 #if 0
 301
 302         status = stream_setup_smb_pipe(event_ctx, model_ops, &dcesrv_stream_ops,
 303                                      e->ep_description->endpoint, dcesrv_sock);
 304         if (!NT_STATUS_IS_OK(status)) {
 305                 DEBUG(0,("service_setup_stream_socket(path=%s) failed - %s\n",
 306                          e->ep_description->endpoint, nt_errstr(status)));
 307         }
 308 #endif
 309         return status;
 310 }
 311
 312 NTSTATUS dcesrv_add_ep_np(struct dcesrv_context *dce_ctx, struct dcesrv_endpoint *e,
 313                                    struct event_context *event_ctx, const struct model_ops *model_ops)
 314 {
 315         NTSTATUS status;
 316
 317         status = add_socket_rpc_pipe_iface(dce_ctx, e, event_ctx, model_ops);
 318         NT_STATUS_NOT_OK_RETURN(status);
 319
 320         return status;
 321 }
 322
 323 /*
 324   add a socket address to the list of events, one event per dcerpc endpoint
 325 */
 326 static NTSTATUS add_socket_rpc_tcp_iface(struct dcesrv_context *dce_ctx, struct dcesrv_endpoint *e,
 327                                          struct event_context *event_ctx, const struct model_ops *model_ops,
 328                                          const char *address)
 329 {
 330         struct dcesrv_socket_context *dcesrv_sock;
 331         uint16_t port = 0;
 332         NTSTATUS status;
 333
 334         if (e->ep_description->endpoint) {
 335                 port = atoi(e->ep_description->endpoint);
 336         }
 337
 338         dcesrv_sock = talloc(event_ctx, struct dcesrv_socket_context);
 339         NT_STATUS_HAVE_NO_MEMORY(dcesrv_sock);
 340
 341         /* remember the endpoint of this socket */
 342         dcesrv_sock->endpoint           = e;
 343         dcesrv_sock->dcesrv_ctx         = talloc_reference(dcesrv_sock, dce_ctx);
 344
 345         status = stream_setup_socket(event_ctx, model_ops, &dcesrv_stream_ops,
 346                                      "ipv4", address, &port, dcesrv_sock);
 347         if (!NT_STATUS_IS_OK(status)) {
 348                 DEBUG(0,("service_setup_stream_socket(address=%s,port=%u) failed - %s\n",
 349                          address, port, nt_errstr(status)));
 350         }
 351
 352         if (e->ep_description->endpoint == NULL) {
 353                 e->ep_description->endpoint = talloc_asprintf(dce_ctx, "%d", port);
 354         }
 355
 356         return status;
 357 }
 358
 359 NTSTATUS dcesrv_add_ep_tcp(struct dcesrv_context *dce_ctx, struct dcesrv_endpoint *e,
 360                                    struct event_context *event_ctx, const struct model_ops *model_ops)
 361 {
 362         NTSTATUS status;
 363
 364         /* Add TCP/IP sockets */
 365         if (lp_interfaces() && lp_bind_interfaces_only()) {
 366                 int num_interfaces = iface_count();
 367                 int i;
 368                 for(i = 0; i < num_interfaces; i++) {
 369                         const char *address = iface_n_ip(i);
 370                         status = add_socket_rpc_tcp_iface(dce_ctx, e, event_ctx, model_ops, address);
 371                         NT_STATUS_NOT_OK_RETURN(status);
 372                 }
 373         } else {
 374                 status = add_socket_rpc_tcp_iface(dce_ctx, e, event_ctx, model_ops, lp_socket_address());
 375                 NT_STATUS_NOT_OK_RETURN(status);
 376         }
 377
 378         return NT_STATUS_OK;
 379 }
 380
 381
 382 NTSTATUS dcesrv_add_ep(struct dcesrv_context *dce_ctx, struct dcesrv_endpoint *e,
 383                           struct event_context *event_ctx, const struct model_ops *model_ops)
 384 {
 385         switch (e->ep_description->transport) {
 386         case NCACN_UNIX_STREAM:
 387                 return dcesrv_add_ep_unix(dce_ctx, e, event_ctx, model_ops);
 388
 389         case NCALRPC:
 390                 return dcesrv_add_ep_ncalrpc(dce_ctx, e, event_ctx, model_ops);
 391
 392         case NCACN_IP_TCP:
 393                 return dcesrv_add_ep_tcp(dce_ctx, e, event_ctx, model_ops);
 394
 395         case NCACN_NP:
 396                 return dcesrv_add_ep_np(dce_ctx, e, event_ctx, model_ops);
 397
 398         default:
 399                 return NT_STATUS_NOT_SUPPORTED;
 400         }
 401 }
 402
 403 /*
 404   open the dcerpc server sockets
 405 */
 406 static void dcesrv_task_init(struct task_server *task)
 407 {
 408         NTSTATUS status;
 409         struct dcesrv_context *dce_ctx;
 410         struct dcesrv_endpoint *e;
 411
 412         task_server_set_title(task, "task[dcesrv]");
 413
 414         status = dcesrv_init_context(task->event_ctx,
 415                                      lp_dcerpc_endpoint_servers(),
 416                                      &dce_ctx);
 417         if (!NT_STATUS_IS_OK(status)) goto failed;
 418
 419         /* Make sure the directory for NCALRPC exists */
 420         if (!directory_exist(lp_ncalrpc_dir())) {
 421                 mkdir(lp_ncalrpc_dir(), 0755);
 422         }
 423
 424         for (e=dce_ctx->endpoint_list;e;e=e->next) {
 425                 status = dcesrv_add_ep(dce_ctx, e, task->event_ctx, task->model_ops);
 426                 if (!NT_STATUS_IS_OK(status)) goto failed;
 427         }
 428
 429         return;
 430 failed:
 431         task_server_terminate(task, "Failed to startup dcerpc server task");
 432 }
 433
 434 /*
 435   called on startup of the smb server service It's job is to start
 436   listening on all configured sockets
 437 */
 438 static NTSTATUS dcesrv_init(struct event_context *event_context,
 439                             const struct model_ops *model_ops)
 440 {
 441         return task_server_startup(event_context, model_ops, dcesrv_task_init);
 442 }
 443
 444 NTSTATUS server_service_rpc_init(void)
 445 {
 446         init_module_fn static_init[] = STATIC_dcerpc_server_MODULES;
 447         init_module_fn *shared_init = load_samba_modules(NULL, "dcerpc_server");
 448
 449         run_init_functions(static_init);
 450         run_init_functions(shared_init);
 451
 452         talloc_free(shared_init);
 453
 454         return register_server_service("rpc", dcesrv_init);
 455 }
 456
 457