source4/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    Copyright (C) Rafal Szczesniak 2006
   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 "system/filesys.h"
  26 #include "lib/events/events.h"
  27 #include "lib/socket/socket.h"
  28 #include "lib/tsocket/tsocket.h"
  29 #include "libcli/composite/composite.h"
  30 #include "librpc/rpc/dcerpc.h"
  31 #include "librpc/rpc/dcerpc_proto.h"
  32 #include "libcli/resolve/resolve.h"
  33 #include "librpc/rpc/rpc_common.h"
  34
  35 struct pipe_open_socket_state {
  36         struct dcecli_connection *conn;
  37         struct socket_context *socket_ctx;
  38         struct socket_address *localaddr;
  39         struct socket_address *server;
  40         const char *target_hostname;
  41         enum dcerpc_transport_t transport;
  42         struct socket_address *client;
  43 };
  44
  45
  46 static void continue_socket_connect(struct composite_context *ctx)
  47 {
  48         struct dcecli_connection *conn;
  49         struct composite_context *c = talloc_get_type_abort(
  50                 ctx->async.private_data, struct composite_context);
  51         struct pipe_open_socket_state *s = talloc_get_type_abort(
  52                 c->private_data, struct pipe_open_socket_state);
  53         int rc;
  54         int sock_fd;
  55
  56         /* make it easier to write a function calls */
  57         conn = s->conn;
  58
  59         c->status = socket_connect_recv(ctx);
  60         if (!NT_STATUS_IS_OK(c->status)) {
  61                 DEBUG(0, ("Failed to connect host %s on port %d - %s\n",
  62                           s->server->addr, s->server->port,
  63                           nt_errstr(c->status)));
  64                 composite_error(c, c->status);
  65                 return;
  66         }
  67
  68         s->client = socket_get_my_addr(s->socket_ctx, s);
  69         if (s->client == NULL) {
  70                 TALLOC_FREE(s->socket_ctx);
  71                 composite_error(c, NT_STATUS_NO_MEMORY);
  72                 return;
  73         }
  74         sock_fd = socket_get_fd(s->socket_ctx);
  75         if (sock_fd == -1) {
  76                 TALLOC_FREE(s->socket_ctx);
  77                 composite_error(c, NT_STATUS_INVALID_HANDLE);
  78                 return;
  79         }
  80         socket_set_flags(s->socket_ctx, SOCKET_FLAG_NOCLOSE);
  81         TALLOC_FREE(s->socket_ctx);
  82
  83         /*
  84           fill in the transport methods
  85         */
  86         conn->transport.transport       = s->transport;
  87         conn->transport.private_data    = NULL;
  88
  89         /*
  90          * Windows uses 5840 for ncacn_ip_tcp,
  91          * so we also use it (for every transport which uses bsd sockets)
  92          */
  93         conn->srv_max_xmit_frag = 5840;
  94         conn->srv_max_recv_frag = 5840;
  95
  96         conn->transport.pending_reads = 0;
  97         conn->server_name   = strupper_talloc(conn, s->target_hostname);
  98
  99         rc = tstream_bsd_existing_socket(conn, sock_fd,
 100                                          &conn->transport.stream);
 101         if (rc < 0) {
 102                 close(sock_fd);
 103                 composite_error(c, NT_STATUS_NO_MEMORY);
 104                 return;
 105         }
 106
 107         conn->transport.write_queue =
 108                 tevent_queue_create(conn, "dcerpc sock write queue");
 109         if (conn->transport.write_queue == NULL) {
 110                 TALLOC_FREE(conn->transport.stream);
 111                 composite_error(c, NT_STATUS_NO_MEMORY);
 112                 return;
 113         }
 114
 115         /* ensure we don't get SIGPIPE */
 116         BlockSignals(true, SIGPIPE);
 117
 118         composite_done(c);
 119 }
 120
 121
 122 static struct composite_context *dcerpc_pipe_open_socket_send(TALLOC_CTX *mem_ctx,
 123                                                        struct dcecli_connection *cn,
 124                                                        struct socket_address *localaddr,
 125                                                        struct socket_address *server,
 126                                                        const char *target_hostname,
 127                                                        const char *full_path,
 128                                                        enum dcerpc_transport_t transport)
 129 {
 130         struct composite_context *c;
 131         struct pipe_open_socket_state *s;
 132         struct composite_context *conn_req;
 133
 134         c = composite_create(mem_ctx, cn->event_ctx);
 135         if (c == NULL) return NULL;
 136
 137         s = talloc_zero(c, struct pipe_open_socket_state);
 138         if (composite_nomem(s, c)) return c;
 139         c->private_data = s;
 140
 141         s->conn      = cn;
 142         s->transport = transport;
 143         if (localaddr) {
 144                 s->localaddr = socket_address_copy(s, localaddr);
 145                 if (composite_nomem(s->localaddr, c)) return c;
 146         }
 147         s->server = socket_address_copy(s, server);
 148         if (composite_nomem(s->server, c)) return c;
 149         if (target_hostname) {
 150                 s->target_hostname = talloc_strdup(s, target_hostname);
 151                 if (composite_nomem(s->target_hostname, c)) return c;
 152         }
 153
 154         c->status = socket_create(server->family, SOCKET_TYPE_STREAM, &s->socket_ctx, 0);
 155         if (!composite_is_ok(c)) return c;
 156
 157         talloc_steal(s, s->socket_ctx);
 158
 159         conn_req = socket_connect_send(s->socket_ctx, s->localaddr, s->server, 0,
 160                                        c->event_ctx);
 161         composite_continue(c, conn_req, continue_socket_connect, c);
 162         return c;
 163 }
 164
 165 static NTSTATUS dcerpc_pipe_open_socket_recv(struct composite_context *c,
 166                                              TALLOC_CTX *mem_ctx,
 167                                              struct socket_address **localaddr)
 168 {
 169         NTSTATUS status = composite_wait(c);
 170
 171         if (NT_STATUS_IS_OK(status)) {
 172                 struct pipe_open_socket_state *s =
 173                         talloc_get_type_abort(c->private_data,
 174                         struct pipe_open_socket_state);
 175
 176                 if (localaddr != NULL) {
 177                         *localaddr = talloc_move(mem_ctx, &s->client);
 178                 }
 179         }
 180
 181         talloc_free(c);
 182         return status;
 183 }
 184
 185 struct pipe_tcp_state {
 186         const char *server;
 187         const char *target_hostname;
 188         const char **addresses;
 189         uint32_t index;
 190         uint32_t port;
 191         struct socket_address *localaddr;
 192         struct socket_address *srvaddr;
 193         struct resolve_context *resolve_ctx;
 194         struct dcecli_connection *conn;
 195         struct nbt_name name;
 196         char *local_address;
 197         char *remote_address;
 198 };
 199
 200
 201 static void continue_ip_open_socket(struct composite_context *ctx);
 202 static void continue_ip_resolve_name(struct composite_context *ctx);
 203
 204 static void continue_ip_resolve_name(struct composite_context *ctx)
 205 {
 206         struct composite_context *c = talloc_get_type_abort(
 207                 ctx->async.private_data, struct composite_context);
 208         struct pipe_tcp_state *s = talloc_get_type_abort(
 209                 c->private_data, struct pipe_tcp_state);
 210         struct composite_context *sock_ip_req;
 211
 212         c->status = resolve_name_multiple_recv(ctx, s, &s->addresses);
 213         if (!composite_is_ok(c)) return;
 214
 215         /* prepare server address using host ip:port and transport name */
 216         s->srvaddr = socket_address_from_strings(s->conn, "ip", s->addresses[s->index], s->port);
 217         s->index++;
 218         if (composite_nomem(s->srvaddr, c)) return;
 219
 220         sock_ip_req = dcerpc_pipe_open_socket_send(c, s->conn, s->localaddr,
 221                                                      s->srvaddr, s->target_hostname,
 222                                                      NULL,
 223                                                      NCACN_IP_TCP);
 224         composite_continue(c, sock_ip_req, continue_ip_open_socket, c);
 225 }
 226
 227
 228 /*
 229   Stage 2 of dcerpc_pipe_open_tcp_send: receive result of pipe open request
 230   on IP transport.
 231 */
 232 static void continue_ip_open_socket(struct composite_context *ctx)
 233 {
 234         struct composite_context *c = talloc_get_type_abort(
 235                 ctx->async.private_data, struct composite_context);
 236         struct pipe_tcp_state *s = talloc_get_type_abort(
 237                 c->private_data, struct pipe_tcp_state);
 238         struct socket_address *localaddr = NULL;
 239
 240         /* receive result socket open request */
 241         c->status = dcerpc_pipe_open_socket_recv(ctx, s, &localaddr);
 242         if (!NT_STATUS_IS_OK(c->status)) {
 243                 /* something went wrong... */
 244                 DEBUG(0, ("Failed to connect host %s (%s) on port %d - %s.\n",
 245                           s->addresses[s->index - 1], s->target_hostname,
 246                           s->port, nt_errstr(c->status)));
 247                 if (s->addresses[s->index]) {
 248                         struct composite_context *sock_ip_req;
 249                         talloc_free(s->srvaddr);
 250                         /* prepare server address using host ip:port and transport name */
 251                         s->srvaddr = socket_address_from_strings(s->conn, "ip", s->addresses[s->index], s->port);
 252                         s->index++;
 253                         if (composite_nomem(s->srvaddr, c)) return;
 254
 255                         sock_ip_req = dcerpc_pipe_open_socket_send(c, s->conn, s->localaddr,
 256                                                                 s->srvaddr, s->target_hostname,
 257                                                                 NULL,
 258                                                                 NCACN_IP_TCP);
 259                         composite_continue(c, sock_ip_req, continue_ip_open_socket, c);
 260
 261                         return;
 262                 } else {
 263                         composite_error(c, c->status);
 264                         return;
 265                 }
 266         }
 267
 268         s->local_address = talloc_strdup(s, localaddr->addr);
 269         if (composite_nomem(s->local_address, c)) return;
 270         s->remote_address = talloc_strdup(s, s->addresses[s->index - 1]);
 271         if (composite_nomem(s->remote_address, c)) return;
 272
 273         composite_done(c);
 274 }
 275
 276 /*
 277   Send rpc pipe open request to given host:port using
 278   tcp/ip transport
 279 */
 280 struct composite_context* dcerpc_pipe_open_tcp_send(struct dcecli_connection *conn,
 281                                                     const char *localaddr,
 282                                                     const char *server,
 283                                                     const char *target_hostname,
 284                                                     uint32_t port,
 285                                                     struct resolve_context *resolve_ctx)
 286 {
 287         struct composite_context *c;
 288         struct pipe_tcp_state *s;
 289         struct composite_context *resolve_req;
 290
 291         /* composite context allocation and setup */
 292         c = composite_create(conn, conn->event_ctx);
 293         if (c == NULL) return NULL;
 294
 295         s = talloc_zero(c, struct pipe_tcp_state);
 296         if (composite_nomem(s, c)) return c;
 297         c->private_data = s;
 298
 299         /* store input parameters in state structure */
 300         s->server          = talloc_strdup(c, server);
 301         if (composite_nomem(s->server, c)) return c;
 302         if (target_hostname) {
 303                 s->target_hostname = talloc_strdup(c, target_hostname);
 304                 if (composite_nomem(s->target_hostname, c)) return c;
 305         }
 306         s->port            = port;
 307         s->conn            = conn;
 308         s->resolve_ctx     = resolve_ctx;
 309         if (localaddr) {
 310                 s->localaddr = socket_address_from_strings(s, "ip", localaddr, 0);
 311                 /* if there is no localaddr, we pass NULL for
 312                    s->localaddr, which is handled by the socket libraries as
 313                    meaning no local binding address specified */
 314         }
 315
 316         make_nbt_name_server(&s->name, s->server);
 317         resolve_req = resolve_name_send(resolve_ctx, s, &s->name, c->event_ctx);
 318         composite_continue(c, resolve_req, continue_ip_resolve_name, c);
 319         return c;
 320 }
 321
 322 /*
 323   Receive result of pipe open request on tcp/ip
 324 */
 325 NTSTATUS dcerpc_pipe_open_tcp_recv(struct composite_context *c,
 326                                    TALLOC_CTX *mem_ctx,
 327                                    char **localaddr,
 328                                    char **remoteaddr)
 329 {
 330         NTSTATUS status;
 331         status = composite_wait(c);
 332
 333         if (NT_STATUS_IS_OK(status)) {
 334                 struct pipe_tcp_state *s = talloc_get_type_abort(
 335                         c->private_data, struct pipe_tcp_state);
 336
 337                 if (localaddr != NULL) {
 338                         *localaddr = talloc_move(mem_ctx, &s->local_address);
 339                 }
 340                 if (remoteaddr != NULL) {
 341                         *remoteaddr = talloc_move(mem_ctx, &s->remote_address);
 342                 }
 343         }
 344
 345         talloc_free(c);
 346         return status;
 347 }
 348
 349
 350 struct pipe_unix_state {
 351         const char *path;
 352         struct socket_address *srvaddr;
 353         struct dcecli_connection *conn;
 354 };
 355
 356
 357 /*
 358   Stage 2 of dcerpc_pipe_open_unix_stream_send: receive result of pipe open
 359   request on unix socket.
 360 */
 361 static void continue_unix_open_socket(struct composite_context *ctx)
 362 {
 363         struct composite_context *c = talloc_get_type_abort(
 364                 ctx->async.private_data, struct composite_context);
 365
 366         c->status = dcerpc_pipe_open_socket_recv(ctx, NULL, NULL);
 367         if (NT_STATUS_IS_OK(c->status)) {
 368                 composite_done(c);
 369                 return;
 370         }
 371
 372         composite_error(c, c->status);
 373 }
 374
 375
 376 /*
 377   Send pipe open request on unix socket
 378 */
 379 struct composite_context *dcerpc_pipe_open_unix_stream_send(struct dcecli_connection *conn,
 380                                                             const char *path)
 381 {
 382         struct composite_context *c;
 383         struct composite_context *sock_unix_req;
 384         struct pipe_unix_state *s;
 385
 386         /* composite context allocation and setup */
 387         c = composite_create(conn, conn->event_ctx);
 388         if (c == NULL) return NULL;
 389
 390         s = talloc_zero(c, struct pipe_unix_state);
 391         if (composite_nomem(s, c)) return c;
 392         c->private_data = s;
 393
 394         /* store parameters in state structure */
 395         s->path = talloc_strdup(c, path);
 396         if (composite_nomem(s->path, c)) return c;
 397         s->conn = conn;
 398
 399         /* prepare server address using socket path and transport name */
 400         s->srvaddr = socket_address_from_strings(conn, "unix", s->path, 0);
 401         if (composite_nomem(s->srvaddr, c)) return c;
 402
 403         /* send socket open request */
 404         sock_unix_req = dcerpc_pipe_open_socket_send(c, s->conn, NULL,
 405                                                      s->srvaddr, NULL,
 406                                                      s->path,
 407                                                      NCALRPC);
 408         composite_continue(c, sock_unix_req, continue_unix_open_socket, c);
 409         return c;
 410 }
 411
 412
 413 /*
 414   Receive result of pipe open request on unix socket
 415 */
 416 NTSTATUS dcerpc_pipe_open_unix_stream_recv(struct composite_context *c)
 417 {
 418         NTSTATUS status = composite_wait(c);
 419
 420         talloc_free(c);
 421         return status;
 422 }
 423
 424
 425 /*
 426   Stage 2 of dcerpc_pipe_open_pipe_send: receive socket open request
 427 */
 428 static void continue_np_open_socket(struct composite_context *ctx)
 429 {
 430         struct composite_context *c = talloc_get_type_abort(
 431                 ctx->async.private_data, struct composite_context);
 432
 433         c->status = dcerpc_pipe_open_socket_recv(ctx, NULL, NULL);
 434         if (!composite_is_ok(c)) return;
 435
 436         composite_done(c);
 437 }
 438
 439
 440 /*
 441   Send pipe open request on ncalrpc
 442 */
 443 struct composite_context* dcerpc_pipe_open_pipe_send(struct dcecli_connection *conn,
 444                                                      const char *ncalrpc_dir,
 445                                                      const char *identifier)
 446 {
 447         char *canon = NULL;
 448
 449         struct composite_context *c;
 450         struct composite_context *sock_np_req;
 451         struct pipe_unix_state *s;
 452
 453         /* composite context allocation and setup */
 454         c = composite_create(conn, conn->event_ctx);
 455         if (c == NULL) return NULL;
 456
 457         s = talloc_zero(c, struct pipe_unix_state);
 458         if (composite_nomem(s, c)) return c;
 459         c->private_data = s;
 460
 461         /* store parameters in state structure */
 462         canon = talloc_strdup(s, identifier);
 463         if (composite_nomem(canon, c)) return c;
 464         s->conn = conn;
 465
 466         string_replace(canon, '/', '\\');
 467         s->path = talloc_asprintf(canon, "%s/%s", ncalrpc_dir, canon);
 468         if (composite_nomem(s->path, c)) return c;
 469
 470         /* prepare server address using path and transport name */
 471         s->srvaddr = socket_address_from_strings(conn, "unix", s->path, 0);
 472         if (composite_nomem(s->srvaddr, c)) return c;
 473
 474         /* send socket open request */
 475         sock_np_req = dcerpc_pipe_open_socket_send(c, s->conn, NULL, s->srvaddr, NULL, s->path, NCALRPC);
 476         composite_continue(c, sock_np_req, continue_np_open_socket, c);
 477         return c;
 478 }
 479
 480
 481 /*
 482   Receive result of pipe open request on ncalrpc
 483 */
 484 NTSTATUS dcerpc_pipe_open_pipe_recv(struct composite_context *c)
 485 {
 486         NTSTATUS status = composite_wait(c);
 487
 488         talloc_free(c);
 489         return status;
 490 }
 491
 492
 493 /*
 494   Open a rpc pipe on a named pipe - sync version
 495 */
 496 NTSTATUS dcerpc_pipe_open_pipe(struct dcecli_connection *conn, const char *ncalrpc_dir, const char *identifier)
 497 {
 498         struct composite_context *c = dcerpc_pipe_open_pipe_send(conn, ncalrpc_dir, identifier);
 499         return dcerpc_pipe_open_pipe_recv(c);
 500 }