source/librpc/rpc/dcerpc_sock.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3
   4    dcerpc over standard sockets transport
   5
   6    Copyright (C) Andrew Tridgell 2003
   7    Copyright (C) Jelmer Vernooij 2004
   8
   9    This program is free software; you can redistribute it and/or modify
  10    it under the terms of the GNU General Public License as published by
  11    the Free Software Foundation; either version 2 of the License, or
  12    (at your option) any later version.
  13
  14    This program is distributed in the hope that it will be useful,
  15    but WITHOUT ANY WARRANTY; without even the implied warranty of
  16    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17    GNU General Public License for more details.
  18
  19    You should have received a copy of the GNU General Public License
  20    along with this program; if not, write to the Free Software
  21    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  22 */
  23
  24 #include "includes.h"
  25 #include "lib/events/events.h"
  26 #include "lib/socket/socket.h"
  27 #include "lib/stream/packet.h"
  28 #include "libcli/composite/composite.h"
  29
  30 /* transport private information used by general socket pipe transports */
  31 struct sock_private {
  32         struct fd_event *fde;
  33         struct socket_context *sock;
  34         char *server_name;
  35
  36         struct packet_context *packet;
  37         uint32_t pending_reads;
  38 };
  39
  40
  41 /*
  42   mark the socket dead
  43 */
  44 static void sock_dead(struct dcerpc_connection *p, NTSTATUS status)
  45 {
  46         struct sock_private *sock = p->transport.private;
  47
  48         if (sock && sock->sock != NULL) {
  49                 talloc_free(sock->fde);
  50                 talloc_free(sock->sock);
  51                 sock->sock = NULL;
  52         }
  53
  54         if (!NT_STATUS_IS_OK(status)) {
  55                 p->transport.recv_data(p, NULL, status);
  56         }
  57 }
  58
  59
  60 /*
  61   handle socket recv errors
  62 */
  63 static void sock_error_handler(void *private, NTSTATUS status)
  64 {
  65         struct dcerpc_connection *p = talloc_get_type(private,
  66                                                       struct dcerpc_connection);
  67         sock_dead(p, status);
  68 }
  69
  70 /*
  71   check if a blob is a complete packet
  72 */
  73 static NTSTATUS sock_complete_packet(void *private, DATA_BLOB blob, size_t *size)
  74 {
  75         if (blob.length < DCERPC_FRAG_LEN_OFFSET+2) {
  76                 return STATUS_MORE_ENTRIES;
  77         }
  78         *size = dcerpc_get_frag_length(&blob);
  79         if (*size > blob.length) {
  80                 return STATUS_MORE_ENTRIES;
  81         }
  82         return NT_STATUS_OK;
  83 }
  84
  85 /*
  86   process recv requests
  87 */
  88 static NTSTATUS sock_process_recv(void *private, DATA_BLOB blob)
  89 {
  90         struct dcerpc_connection *p = talloc_get_type(private,
  91                                                       struct dcerpc_connection);
  92         struct sock_private *sock = p->transport.private;
  93         sock->pending_reads--;
  94         if (sock->pending_reads == 0) {
  95                 packet_recv_disable(sock->packet);
  96         }
  97         p->transport.recv_data(p, &blob, NT_STATUS_OK);
  98         return NT_STATUS_OK;
  99 }
 100
 101 /*
 102   called when a IO is triggered by the events system
 103 */
 104 static void sock_io_handler(struct event_context *ev, struct fd_event *fde,
 105                             uint16_t flags, void *private)
 106 {
 107         struct dcerpc_connection *p = talloc_get_type(private,
 108                                                       struct dcerpc_connection);
 109         struct sock_private *sock = p->transport.private;
 110
 111         if (flags & EVENT_FD_WRITE) {
 112                 packet_queue_run(sock->packet);
 113                 return;
 114         }
 115
 116         if (sock->sock == NULL) {
 117                 return;
 118         }
 119
 120         if (flags & EVENT_FD_READ) {
 121                 packet_recv(sock->packet);
 122         }
 123 }
 124
 125 /*
 126    initiate a read request - not needed for dcerpc sockets
 127 */
 128 static NTSTATUS sock_send_read(struct dcerpc_connection *p)
 129 {
 130         struct sock_private *sock = p->transport.private;
 131         sock->pending_reads++;
 132         if (sock->pending_reads == 1) {
 133                 packet_recv_enable(sock->packet);
 134         }
 135         return NT_STATUS_OK;
 136 }
 137
 138 /*
 139    send an initial pdu in a multi-pdu sequence
 140 */
 141 static NTSTATUS sock_send_request(struct dcerpc_connection *p, DATA_BLOB *data,
 142                                   BOOL trigger_read)
 143 {
 144         struct sock_private *sock = p->transport.private;
 145         DATA_BLOB blob;
 146         NTSTATUS status;
 147
 148         if (sock->sock == NULL) {
 149                 return NT_STATUS_CONNECTION_DISCONNECTED;
 150         }
 151
 152         blob = data_blob_talloc(sock->packet, data->data, data->length);
 153         if (blob.data == NULL) {
 154                 return NT_STATUS_NO_MEMORY;
 155         }
 156
 157         status = packet_send(sock->packet, blob);
 158         if (!NT_STATUS_IS_OK(status)) {
 159                 return status;
 160         }
 161
 162         if (trigger_read) {
 163                 sock_send_read(p);
 164         }
 165
 166         return NT_STATUS_OK;
 167 }
 168
 169 /*
 170    shutdown sock pipe connection
 171 */
 172 static NTSTATUS sock_shutdown_pipe(struct dcerpc_connection *p)
 173 {
 174         struct sock_private *sock = p->transport.private;
 175
 176         if (sock && sock->sock) {
 177                 sock_dead(p, NT_STATUS_OK);
 178         }
 179
 180         return NT_STATUS_OK;
 181 }
 182
 183 /*
 184   return sock server name
 185 */
 186 static const char *sock_peer_name(struct dcerpc_connection *p)
 187 {
 188         struct sock_private *sock = p->transport.private;
 189         return sock->server_name;
 190 }
 191
 192
 193 struct pipe_open_socket_state {
 194         struct dcerpc_connection *conn;
 195         struct socket_context *socket_ctx;
 196         struct sock_private *sock;
 197         struct socket_address *server;
 198         enum dcerpc_transport_t transport;
 199 };
 200
 201
 202 static void continue_socket_connect(struct composite_context *ctx)
 203 {
 204         struct dcerpc_connection *conn;
 205         struct sock_private *sock;
 206         struct composite_context *c = talloc_get_type(ctx->async.private_data,
 207                                                       struct composite_context);
 208         struct pipe_open_socket_state *s = talloc_get_type(c->private_data,
 209                                                            struct pipe_open_socket_state);
 210
 211         /* make it easier to write a function calls */
 212         conn = s->conn;
 213         sock = s->sock;
 214
 215         c->status = socket_connect_recv(ctx);
 216         if (!NT_STATUS_IS_OK(c->status)) {
 217                 DEBUG(0, ("Failed to connect host %s on port %d - %s\n",
 218                           s->server->addr, s->server->port,
 219                           nt_errstr(c->status)));
 220                 composite_error(c, c->status);
 221                 return;
 222         }
 223
 224         /*
 225           fill in the transport methods
 226         */
 227         conn->transport.transport     = s->transport;
 228         conn->transport.private       = NULL;
 229
 230         conn->transport.send_request  = sock_send_request;
 231         conn->transport.send_read     = sock_send_read;
 232         conn->transport.recv_data     = NULL;
 233
 234         conn->transport.shutdown_pipe = sock_shutdown_pipe;
 235         conn->transport.peer_name     = sock_peer_name;
 236
 237         sock->sock          = s->socket_ctx;
 238         sock->pending_reads = 0;
 239         sock->server_name   = strupper_talloc(sock, s->server->addr);
 240
 241         sock->fde = event_add_fd(conn->event_ctx, sock->sock, socket_get_fd(sock->sock),
 242                                  0, sock_io_handler, conn);
 243
 244         conn->transport.private = sock;
 245
 246         sock->packet = packet_init(sock);
 247         if (sock->packet == NULL) {
 248                 composite_error(c, NT_STATUS_NO_MEMORY);
 249                 talloc_free(sock);
 250                 return;
 251         }
 252
 253         packet_set_private(sock->packet, conn);
 254         packet_set_socket(sock->packet, sock->sock);
 255         packet_set_callback(sock->packet, sock_process_recv);
 256         packet_set_full_request(sock->packet, sock_complete_packet);
 257         packet_set_error_handler(sock->packet, sock_error_handler);
 258         packet_set_event_context(sock->packet, conn->event_ctx);
 259         packet_set_fde(sock->packet, sock->fde);
 260         packet_set_serialise(sock->packet);
 261         packet_recv_disable(sock->packet);
 262         packet_set_initial_read(sock->packet, 16);
 263
 264         /* ensure we don't get SIGPIPE */
 265         BlockSignals(True,SIGPIPE);
 266
 267         composite_done(c);
 268 }
 269
 270
 271 struct composite_context *dcerpc_pipe_open_socket_send(TALLOC_CTX *mem_ctx,
 272                                                        struct dcerpc_connection *cn,
 273                                                        struct socket_address *server,
 274                                                        enum dcerpc_transport_t transport)
 275 {
 276         NTSTATUS status;
 277         struct composite_context *c;
 278         struct pipe_open_socket_state *s;
 279         struct composite_context *conn_req;
 280
 281         c = talloc_zero(mem_ctx, struct composite_context);
 282         if (c == NULL) return NULL;
 283
 284         s = talloc_zero(c, struct pipe_open_socket_state);
 285         if (s == NULL) {
 286                 composite_error(c, NT_STATUS_NO_MEMORY);
 287                 goto done;
 288         }
 289
 290         c->state = COMPOSITE_STATE_IN_PROGRESS;
 291         c->private_data = s;
 292         c->event_ctx = cn->event_ctx;
 293
 294         s->conn      = cn;
 295         s->transport = transport;
 296         s->server    = talloc_reference(c, server);
 297         if (s->server == NULL) {
 298                 composite_error(c, NT_STATUS_NO_MEMORY);
 299                 goto done;
 300         }
 301
 302         s->sock = talloc(cn, struct sock_private);
 303         if (s->sock == NULL) {
 304                 composite_error(c, NT_STATUS_NO_MEMORY);
 305                 goto done;
 306         }
 307
 308         status = socket_create(server->family, SOCKET_TYPE_STREAM, &s->socket_ctx, 0);
 309         if (!NT_STATUS_IS_OK(status)) {
 310                 composite_error(c, status);
 311                 talloc_free(s->sock);
 312                 goto done;
 313         }
 314         talloc_steal(s->sock, s->socket_ctx);
 315
 316         conn_req = socket_connect_send(s->socket_ctx, NULL, s->server, 0, c->event_ctx);
 317         if (conn_req == NULL) {
 318                 composite_error(c, NT_STATUS_NO_MEMORY);
 319                 goto done;
 320         }
 321
 322         composite_continue(c, conn_req, continue_socket_connect, c);
 323
 324 done:
 325         return c;
 326 }
 327
 328
 329 NTSTATUS dcerpc_pipe_open_socket_recv(struct composite_context *c)
 330 {
 331         NTSTATUS status = composite_wait(c);
 332
 333         talloc_free(c);
 334         return status;
 335 }
 336
 337 /*
 338    open a rpc connection using the generic socket library
 339 */
 340 NTSTATUS dcerpc_pipe_open_socket(struct dcerpc_connection *conn,
 341                                  struct socket_address *server,
 342                                  enum dcerpc_transport_t transport)
 343 {
 344         struct composite_context *c;
 345
 346         c = dcerpc_pipe_open_socket_send(conn, conn, server, transport);
 347         return dcerpc_pipe_open_socket_recv(c);
 348 }
 349
 350
 351 struct pipe_tcp_state {
 352         const char *server;
 353         uint32_t port;
 354         struct socket_address *srvaddr;
 355         struct dcerpc_connection *conn;
 356 };
 357
 358
 359 void continue_ipv6_open_socket(struct composite_context *ctx);
 360 void continue_ipv4_open_socket(struct composite_context *ctx);
 361
 362
 363 void continue_ipv6_open_socket(struct composite_context *ctx)
 364 {
 365         struct composite_context *c = talloc_get_type(ctx->async.private_data,
 366                                                       struct composite_context);
 367         struct pipe_tcp_state *s = talloc_get_type(c->private_data,
 368                                                    struct pipe_tcp_state);
 369         struct composite_context *sock_ipv4_req;
 370
 371         c->status = dcerpc_pipe_open_socket_recv(ctx);
 372         if (NT_STATUS_IS_OK(c->status)) return;
 373
 374         talloc_free(s->srvaddr);
 375
 376         s->srvaddr = socket_address_from_strings(s->conn, "ipv4", s->server, s->port);
 377         if (!s->srvaddr) {
 378                 composite_error(c, NT_STATUS_NO_MEMORY);
 379                 return;
 380         }
 381
 382         /* Try IPv4 if IPv6 fails */
 383         sock_ipv4_req = dcerpc_pipe_open_socket_send(c, s->conn, s->srvaddr, NCACN_IP_TCP);
 384         if (sock_ipv4_req == NULL) {
 385                 composite_error(c, NT_STATUS_NO_MEMORY);
 386                 return;
 387         }
 388
 389         composite_continue(c, sock_ipv4_req, continue_ipv4_open_socket, c);
 390 }
 391
 392
 393 void continue_ipv4_open_socket(struct composite_context *ctx)
 394 {
 395         struct composite_context *c = talloc_get_type(ctx->async.private_data,
 396                                                       struct composite_context);
 397         struct pipe_tcp_state *s = talloc_get_type(c->private_data,
 398                                                    struct pipe_tcp_state);
 399
 400         c->status = dcerpc_pipe_open_socket_recv(ctx);
 401         if (NT_STATUS_IS_OK(c->status)) {
 402                 composite_done(c);
 403                 return;
 404         }
 405
 406         DEBUG(0, ("Failed to connect host %s on port %d - %s.\n",
 407                   s->server, s->port, nt_errstr(c->status)));
 408
 409         composite_error(c, c->status);
 410 }
 411
 412
 413 struct composite_context* dcerpc_pipe_open_tcp_send(struct dcerpc_connection *conn,
 414                                                     const char* server, uint32_t port)
 415 {
 416         struct composite_context *c;
 417         struct composite_context *sock_ipv6_req;
 418         struct pipe_tcp_state *s;
 419
 420         /* composite context allocation and setup */
 421         c = talloc_zero(conn, struct composite_context);
 422         if (c == NULL) return NULL;
 423
 424         s = talloc_zero(c, struct pipe_tcp_state);
 425         if (s == NULL) {
 426                 composite_error(c, NT_STATUS_NO_MEMORY);
 427                 goto done;
 428         }
 429
 430         c->state = COMPOSITE_STATE_IN_PROGRESS;
 431         c->private_data = s;
 432         c->event_ctx = conn->event_ctx;
 433
 434         s->server = talloc_strdup(c, server);
 435         s->port   = port;
 436         s->conn   = conn;
 437
 438         s->srvaddr = socket_address_from_strings(s->conn, "ipv6", s->server, s->port);
 439         if (!s->srvaddr) {
 440                 composite_error(c, NT_STATUS_NO_MEMORY);
 441                 goto done;
 442         }
 443
 444         /* Try IPv6 first */
 445         sock_ipv6_req = dcerpc_pipe_open_socket_send(c, s->conn, s->srvaddr, NCACN_IP_TCP);
 446         if (sock_ipv6_req == NULL) {
 447                 composite_error(c, NT_STATUS_NO_MEMORY);
 448                 goto done;
 449         }
 450
 451         composite_continue(c, sock_ipv6_req, continue_ipv6_open_socket, c);
 452 done:
 453         return c;
 454 }
 455
 456
 457 NTSTATUS dcerpc_pipe_open_tcp_recv(struct composite_context *c)
 458 {
 459         NTSTATUS status;
 460         status = composite_wait(c);
 461
 462         talloc_free(c);
 463         return status;
 464 }
 465
 466
 467 /*
 468    open a rpc connection using tcp - sync version
 469 */
 470 NTSTATUS dcerpc_pipe_open_tcp(struct dcerpc_connection *conn, const char *server,
 471                               uint32_t port)
 472 {
 473         struct composite_context *c;
 474
 475         c = dcerpc_pipe_open_tcp_send(conn, server, port);
 476         return dcerpc_pipe_open_tcp_recv(c);
 477 }
 478
 479
 480 struct pipe_unix_state {
 481         const char *path;
 482         struct socket_address *srvaddr;
 483         struct dcerpc_connection *conn;
 484 };
 485
 486
 487 void continue_unix_open_socket(struct composite_context *ctx)
 488 {
 489         struct composite_context *c = talloc_get_type(ctx->async.private_data,
 490                                                       struct composite_context);
 491
 492         c->status = dcerpc_pipe_open_socket_recv(ctx);
 493         if (NT_STATUS_IS_OK(c->status)) {
 494                 composite_done(c);
 495                 return;
 496         }
 497
 498         composite_error(c, c->status);
 499 }
 500
 501
 502 struct composite_context *dcerpc_pipe_open_unix_stream_send(struct dcerpc_connection *conn,
 503                                                             const char *path)
 504 {
 505         struct composite_context *c;
 506         struct composite_context *sock_unix_req;
 507         struct pipe_unix_state *s;
 508
 509         /* composite context allocation and setup */
 510         c = talloc_zero(conn, struct composite_context);
 511         if (c == NULL) return NULL;
 512
 513         s = talloc_zero(c, struct pipe_unix_state);
 514         if (s == NULL) {
 515                 composite_error(c, NT_STATUS_NO_MEMORY);
 516                 goto done;
 517         }
 518
 519         c->state = COMPOSITE_STATE_IN_PROGRESS;
 520         c->private_data = s;
 521         c->event_ctx = conn->event_ctx;
 522
 523         s->path = talloc_strdup(c, path);
 524         s->conn = conn;
 525
 526         s->srvaddr = socket_address_from_strings(conn, "unix", s->path, 0);
 527         if (s->srvaddr == NULL) {
 528                 composite_error(c, NT_STATUS_NO_MEMORY);
 529                 goto done;
 530         }
 531
 532         sock_unix_req = dcerpc_pipe_open_socket_send(c, s->conn, s->srvaddr, NCALRPC);
 533         if (sock_unix_req == NULL) {
 534                 composite_error(c, NT_STATUS_NO_MEMORY);
 535                 goto done;
 536         }
 537
 538         composite_continue(c, sock_unix_req, continue_unix_open_socket, c);
 539
 540 done:
 541         return c;
 542 }
 543
 544
 545 NTSTATUS dcerpc_pipe_open_unix_stream_recv(struct composite_context *c)
 546 {
 547         NTSTATUS status = composite_wait(c);
 548
 549         talloc_free(c);
 550         return status;
 551 }
 552
 553
 554 /*
 555    open a rpc connection to a unix socket - sync version
 556 */
 557 NTSTATUS dcerpc_pipe_open_unix_stream(struct dcerpc_connection *conn, const char *path)
 558 {
 559         struct composite_context *c = dcerpc_pipe_open_unix_stream_send(conn, path);
 560         return dcerpc_pipe_open_unix_stream_recv(c);
 561 }
 562
 563
 564 struct pipe_np_state {
 565         char *full_path;
 566         struct socket_address *srvaddr;
 567         struct dcerpc_connection *conn;
 568 };
 569
 570
 571 void continue_np_open_socket(struct composite_context *ctx)
 572 {
 573         struct composite_context *c = talloc_get_type(ctx->async.private_data,
 574                                                       struct composite_context);
 575
 576         c->status = dcerpc_pipe_open_socket_recv(ctx);
 577         if (NT_STATUS_IS_OK(c->status)) {
 578                 composite_done(c);
 579                 return;
 580         }
 581
 582         composite_error(c, c->status);
 583 }
 584
 585
 586 struct composite_context* dcerpc_pipe_open_pipe_send(struct dcerpc_connection *conn,
 587                                                      const char *identifier)
 588 {
 589         char *canon;
 590
 591         struct composite_context *c;
 592         struct composite_context *sock_np_req;
 593         struct pipe_np_state *s;
 594
 595         /* composite context allocation and setup */
 596         c = talloc_zero(conn, struct composite_context);
 597         if (c == NULL) return NULL;
 598
 599         s = talloc_zero(c, struct pipe_np_state);
 600         if (s == NULL) {
 601                 composite_error(c, NT_STATUS_NO_MEMORY);
 602                 goto done;
 603         }
 604
 605         c->state = COMPOSITE_STATE_IN_PROGRESS;
 606         c->private_data = s;
 607         c->event_ctx = conn->event_ctx;
 608
 609         canon = talloc_strdup(c, identifier);
 610         s->conn = conn;
 611
 612         string_replace(canon, '/', '\\');
 613         s->full_path = talloc_asprintf(canon, "%s/%s", lp_ncalrpc_dir(), canon);
 614
 615         s->srvaddr = socket_address_from_strings(conn, "unix", s->full_path, 0);
 616         if (s->srvaddr == NULL) {
 617                 composite_error(c, NT_STATUS_NO_MEMORY);
 618                 goto done;
 619         }
 620
 621         sock_np_req = dcerpc_pipe_open_socket_send(c, s->conn, s->srvaddr, NCALRPC);
 622         if (sock_np_req == NULL) {
 623                 composite_error(c, NT_STATUS_NO_MEMORY);
 624                 goto done;
 625         }
 626
 627         composite_continue(c, sock_np_req, continue_np_open_socket, c);
 628
 629 done:
 630         talloc_free(canon);
 631         return c;
 632 }
 633
 634
 635 NTSTATUS dcerpc_pipe_open_pipe_recv(struct composite_context *c)
 636 {
 637         NTSTATUS status = composite_wait(c);
 638
 639         talloc_free(c);
 640         return status;
 641 }
 642
 643
 644 /*
 645    open a rpc connection to a named pipe - sync version
 646 */
 647 NTSTATUS dcerpc_pipe_open_pipe(struct dcerpc_connection *conn, const char *identifier)
 648 {
 649         struct composite_context *c = dcerpc_pipe_open_pipe_send(conn, identifier);
 650         return dcerpc_pipe_open_pipe_recv(c);
 651 }