source3/torture/test_messaging_fd_passing.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Test for fd passing with messaging
   4
   5    Copyright (C) Michael Adam 2014
   6
   7    This program is free software; you can redistribute it and/or modify
   8    it under the terms of the GNU General Public License as published by
   9    the Free Software Foundation; either version 3 of the License, or
  10    (at your option) any later version.
  11
  12    This program is distributed in the hope that it will be useful,
  13    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15    GNU General Public License for more details.
  16
  17    You should have received a copy of the GNU General Public License
  18    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19 */
  20
  21 #include "includes.h"
  22 #include "torture/proto.h"
  23 #include "lib/util/tevent_unix.h"
  24 #include "messages.h"
  25
  26 /**
  27  * test fdpass1:
  28  *
  29  * Try to pass an fd to the sending process - fails.
  30  */
  31 bool run_messaging_fdpass1(int dummy)
  32 {
  33         struct tevent_context *ev = NULL;
  34         struct messaging_context *msg_ctx = NULL;
  35         bool retval = false;
  36         int pipe_fds[2];
  37         int pass_fds[1] = { 0 };
  38         int ret;
  39         NTSTATUS status;
  40         struct server_id dst;
  41         TALLOC_CTX *frame = talloc_stackframe();
  42
  43         ev = samba_tevent_context_init(frame);
  44         if (ev == NULL) {
  45                 fprintf(stderr, "tevent_context_init failed\n");
  46                 goto fail;
  47         }
  48         msg_ctx = messaging_init(ev, ev);
  49         if (msg_ctx == NULL) {
  50                 fprintf(stderr, "messaging_init failed\n");
  51                 goto fail;
  52         }
  53
  54         dst = messaging_server_id(msg_ctx);
  55
  56         ret = pipe(pipe_fds);
  57         if (ret != 0) {
  58                 perror("pipe failed");
  59                 goto fail;
  60         }
  61
  62         pass_fds[0] = pipe_fds[0];
  63
  64         status = messaging_send_iov(msg_ctx, dst, MSG_PING, NULL, 0,
  65                                     pass_fds, 1);
  66         if (!NT_STATUS_EQUAL(status, NT_STATUS_OK)) {
  67                 fprintf(stderr,
  68                         "messaging_send_iov gave: %s\n", nt_errstr(status));
  69                 goto fail;
  70         }
  71
  72         retval = true;
  73
  74 fail:
  75         TALLOC_FREE(frame);
  76         return retval;
  77 }
  78
  79 /**
  80  * test fdpass2:
  81  *
  82  * - parent: create a child
  83  * - parent: create a two pipes in the parent: up and down
  84  * - parent: pass the up pipe's reading end and the down pipe's writing
  85  *   end to the child and close them
  86  * - parent: write a number into the up pipe's writing end
  87  * - child: read number from the passed reading fd (up)
  88  * - child: write the read number to the passed writing fd (down)
  89  * - parent: read number from the down pipe's reading end and compare with
  90  *   original number
  91  */
  92
  93 #define MSG_TORTURE_FDPASS2 0xF002
  94
  95 static bool fdpass2_filter(struct messaging_rec *rec, void *private_data)
  96 {
  97         if (rec->msg_type != MSG_TORTURE_FDPASS2) {
  98                 return false;
  99         }
 100
 101         if (rec->num_fds != 2) {
 102                 return false;
 103         }
 104
 105         return true;
 106 }
 107
 108 static bool fdpass2_child(int ready_fd)
 109 {
 110         struct tevent_context *ev = NULL;
 111         struct messaging_context *msg_ctx = NULL;
 112         TALLOC_CTX *frame = talloc_stackframe();
 113         bool retval = false;
 114         uint8_t c = 1;
 115         struct tevent_req *subreq;
 116         int ret;
 117         ssize_t bytes;
 118         int up_fd, down_fd;
 119         struct messaging_rec *rec;
 120         bool ok;
 121
 122         ev = samba_tevent_context_init(frame);
 123         if (ev == NULL) {
 124                 fprintf(stderr, "child: tevent_context_init failed\n");
 125                 goto done;
 126         }
 127
 128         msg_ctx = messaging_init(ev, ev);
 129         if (msg_ctx == NULL) {
 130                 fprintf(stderr, "child: messaging_init failed\n");
 131                 goto done;
 132         }
 133
 134         /* Tell the parent we are ready to receive mesages. */
 135         bytes = write(ready_fd, &c, 1);
 136         if (bytes != 1) {
 137                 perror("child: failed to write to ready_fd");
 138                 goto done;
 139         }
 140
 141         subreq = messaging_filtered_read_send(frame, /* TALLOC_CTX */
 142                                               ev, msg_ctx,
 143                                               fdpass2_filter, NULL);
 144         if (subreq == NULL) {
 145                 fprintf(stderr, "child: messaging_filtered_read_send failed\n");
 146                 goto done;
 147         }
 148
 149         ok = tevent_req_poll(subreq, ev);
 150         if (!ok) {
 151                 fprintf(stderr, "child: tevent_req_poll failed\n");
 152                 goto done;
 153         }
 154
 155         ret = messaging_filtered_read_recv(subreq, frame, &rec);
 156         TALLOC_FREE(subreq);
 157         if (ret != 0) {
 158                 fprintf(stderr, "child: messaging_filtered_read_recv failed\n");
 159                 goto done;
 160         }
 161
 162         SMB_ASSERT(rec->num_fds == 2);
 163
 164         /* Tell the parent we are done. */
 165         bytes = write(ready_fd, &c, 1);
 166         if (bytes != 1) {
 167                 perror("child: failed to write to ready_fd");
 168                 goto done;
 169         }
 170
 171         up_fd = rec->fds[0];
 172         down_fd = rec->fds[1];
 173
 174         bytes = read(up_fd, &c, 1);
 175         if (bytes != 1) {
 176                 perror("child: read from up_fd failed");
 177                 goto done;
 178         }
 179
 180         bytes = write(down_fd, &c, 1);
 181         if (bytes != 1) {
 182                 perror("child: write to down_fd failed");
 183         }
 184
 185         printf("child: done\n");
 186
 187         retval = true;
 188
 189 done:
 190         TALLOC_FREE(frame);
 191         return retval;
 192 }
 193
 194 struct child_done_state {
 195         int fd;
 196         bool done;
 197 };
 198
 199 static void child_done_cb(struct tevent_context *ev,
 200                           struct tevent_fd *fde,
 201                           uint16_t flags,
 202                           void *private_data)
 203 {
 204         struct child_done_state *state =
 205                         (struct child_done_state *)private_data;
 206         char c = 0;
 207         ssize_t bytes;
 208
 209         bytes = read(state->fd, &c, 1);
 210         if (bytes != 1) {
 211                 perror("parent: read from ready_fd failed");
 212         }
 213
 214         state->done = true;
 215 }
 216
 217 static bool fdpass2_parent(pid_t child_pid, int ready_fd)
 218 {
 219         struct tevent_context *ev = NULL;
 220         struct messaging_context *msg_ctx = NULL;
 221         bool retval = false;
 222         int up_pipe[2];
 223         int down_pipe[2];
 224         int pass_fds[2] = { 0 };
 225         int ret;
 226         NTSTATUS status;
 227         struct server_id dst;
 228         TALLOC_CTX *frame = talloc_stackframe();
 229         uint8_t c1 = 1, c2, c;
 230         ssize_t bytes;
 231         struct iovec iov;
 232         DATA_BLOB blob;
 233         struct tevent_fd *child_done_fde;
 234         struct child_done_state child_state;
 235
 236         ev = samba_tevent_context_init(frame);
 237         if (ev == NULL) {
 238                 fprintf(stderr, "parent: tevent_context_init failed\n");
 239                 goto done;
 240         }
 241
 242         msg_ctx = messaging_init(ev, ev);
 243         if (msg_ctx == NULL) {
 244                 fprintf(stderr, "parent: messaging_init failed\n");
 245                 goto done;
 246         }
 247
 248         /* wait util the child is ready to receive messages */
 249         bytes = read(ready_fd, &c, 1);
 250         if (bytes != 1) {
 251                 perror("parent: read from ready_fd failed");
 252                 goto done;
 253         }
 254
 255         ret = pipe(up_pipe);
 256         if (ret != 0) {
 257                 perror("parent: pipe failed for up_pipe");
 258                 goto done;
 259         }
 260
 261         ret = pipe(down_pipe);
 262         if (ret != 0) {
 263                 perror("parent: pipe failed for down_pipe");
 264                 goto done;
 265         }
 266
 267         child_state.fd = ready_fd;
 268         child_state.done = false;
 269
 270         child_done_fde = tevent_add_fd(ev, ev, ready_fd, TEVENT_FD_READ,
 271                                        child_done_cb, &child_state);
 272         if (child_done_fde == NULL) {
 273                 fprintf(stderr,
 274                         "parent: failed tevent_add_fd for child done\n");
 275                 goto done;
 276         }
 277
 278         pass_fds[0] = up_pipe[0];
 279         pass_fds[1] = down_pipe[1];
 280
 281         dst = messaging_server_id(msg_ctx);
 282         dst.pid = child_pid;
 283
 284         /*
 285          * Send a certain payload with the fds, to test to test
 286          * that fd-passing works when we have fragmentation and
 287          * re-assembly of the datagrams.
 288          */
 289         blob = data_blob_talloc_zero(frame, 1000*1000);
 290         iov.iov_base = blob.data;
 291         iov.iov_len  = blob.length;
 292
 293         status = messaging_send_iov(msg_ctx, dst, MSG_TORTURE_FDPASS2, &iov, 1,
 294                                     pass_fds, 2);
 295         if (!NT_STATUS_IS_OK(status)) {
 296                 fprintf(stderr, "parent: messaging_send_iov failed: %s\n",
 297                         nt_errstr(status));
 298                 goto done;
 299         }
 300
 301         printf("parent: waiting for child to confirm\n");
 302
 303         while (!child_state.done) {
 304                 ret = tevent_loop_once(ev);
 305                 if (ret != 0) {
 306                         fprintf(stderr, "parent: tevent_loop_once failed\n");
 307                         goto done;
 308                 }
 309         }
 310
 311         printf("parent: child confirmed\n");
 312
 313         close(up_pipe[0]);
 314         close(down_pipe[1]);
 315
 316         bytes = write(up_pipe[1], &c1, 1);
 317         if (bytes != 1) {
 318                 perror("parent: write to up pipe failed");
 319                 goto done;
 320         }
 321
 322         bytes = read(down_pipe[0], &c2, 1);
 323         if (bytes != 1) {
 324                 perror("parent: read from down pipe failed");
 325                 goto done;
 326         }
 327
 328         if (c1 != c2) {
 329                 fprintf(stderr, "parent: c1[%d] != c2[%d]\n", c1, c2);
 330                 goto done;
 331         }
 332
 333         ret = waitpid(child_pid, NULL, 0);
 334         if (ret == -1) {
 335                 perror("parent: waitpid failed");
 336                 goto done;
 337         }
 338
 339         retval = true;
 340
 341 done:
 342         TALLOC_FREE(frame);
 343         return retval;
 344 }
 345
 346 bool run_messaging_fdpass2(int dummy)
 347 {
 348         bool retval = false;
 349         pid_t child_pid;
 350         int ready_pipe[2];
 351         int ret;
 352
 353         ret = pipe(ready_pipe);
 354         if (ret != 0) {
 355                 perror("parent: pipe failed for ready_pipe");
 356                 return retval;
 357         }
 358
 359         child_pid = fork();
 360         if (child_pid == -1) {
 361                 perror("fork failed");
 362         } else if (child_pid == 0) {
 363                 close(ready_pipe[0]);
 364                 retval = fdpass2_child(ready_pipe[1]);
 365         } else {
 366                 close(ready_pipe[1]);
 367                 retval = fdpass2_parent(child_pid, ready_pipe[0]);
 368         }
 369
 370         return retval;
 371 }