libuv-svace-build

Форк
0
/
test-ipc.c 
857 строк · 21.0 Кб
1
/* Copyright Joyent, Inc. and other Node contributors. All rights reserved.
2
 *
3
 * Permission is hereby granted, free of charge, to any person obtaining a copy
4
 * of this software and associated documentation files (the "Software"), to
5
 * deal in the Software without restriction, including without limitation the
6
 * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
7
 * sell copies of the Software, and to permit persons to whom the Software is
8
 * furnished to do so, subject to the following conditions:
9
 *
10
 * The above copyright notice and this permission notice shall be included in
11
 * all copies or substantial portions of the Software.
12
 *
13
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
14
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
15
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
16
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
17
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
18
 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
19
 * IN THE SOFTWARE.
20
 */
21

22
#include "uv.h"
23
#include "task.h"
24

25
#include <stdio.h>
26
#include <string.h>
27

28
static uv_pipe_t channel;
29
static uv_tcp_t tcp_server;
30
static uv_tcp_t tcp_server2;
31
static uv_tcp_t tcp_connection;
32

33
static int exit_cb_called;
34
static int read_cb_called;
35
static int tcp_write_cb_called;
36
static int tcp_read_cb_called;
37
static int on_pipe_read_called;
38
static int local_conn_accepted;
39
static int remote_conn_accepted;
40
static int tcp_server_listening;
41
static uv_write_t write_req;
42
static uv_write_t write_req2;
43
static uv_write_t conn_notify_req;
44
static int close_cb_called;
45
static int connection_accepted;
46
static int tcp_conn_read_cb_called;
47
static int tcp_conn_write_cb_called;
48
static int send_zero_write;
49

50
typedef struct {
51
  uv_connect_t conn_req;
52
  uv_write_t tcp_write_req;
53
  uv_tcp_t conn;
54
} tcp_conn;
55

56
#define CONN_COUNT 100
57
#define BACKLOG 128
58

59

60
static void close_server_conn_cb(uv_handle_t* handle) {
61
  free(handle);
62
}
63

64

65
static void on_connection(uv_stream_t* server, int status) {
66
  uv_tcp_t* conn;
67
  int r;
68

69
  if (!local_conn_accepted) {
70
    /* Accept the connection and close it.  Also and close the server. */
71
    ASSERT_OK(status);
72
    ASSERT_PTR_EQ(&tcp_server, server);
73

74
    conn = malloc(sizeof(*conn));
75
    ASSERT_NOT_NULL(conn);
76
    r = uv_tcp_init(server->loop, conn);
77
    ASSERT_OK(r);
78

79
    r = uv_accept(server, (uv_stream_t*)conn);
80
    ASSERT_OK(r);
81

82
    uv_close((uv_handle_t*)conn, close_server_conn_cb);
83
    uv_close((uv_handle_t*)server, NULL);
84
    local_conn_accepted = 1;
85
  }
86
}
87

88

89
static void exit_cb(uv_process_t* process,
90
                    int64_t exit_status,
91
                    int term_signal) {
92
  printf("exit_cb\n");
93
  exit_cb_called++;
94
  ASSERT_OK(exit_status);
95
  ASSERT_OK(term_signal);
96
  uv_close((uv_handle_t*)process, NULL);
97
}
98

99

100
static void on_alloc(uv_handle_t* handle,
101
                     size_t suggested_size,
102
                     uv_buf_t* buf) {
103
  buf->base = malloc(suggested_size);
104
  buf->len = suggested_size;
105
}
106

107

108
static void close_client_conn_cb(uv_handle_t* handle) {
109
  tcp_conn* p = (tcp_conn*)handle->data;
110
  free(p);
111
}
112

113

114
static void connect_cb(uv_connect_t* req, int status) {
115
  uv_close((uv_handle_t*)req->handle, close_client_conn_cb);
116
}
117

118

119
static void make_many_connections(void) {
120
  tcp_conn* conn;
121
  struct sockaddr_in addr;
122
  int r, i;
123

124
  for (i = 0; i < CONN_COUNT; i++) {
125
    conn = malloc(sizeof(*conn));
126
    ASSERT_NOT_NULL(conn);
127

128
    r = uv_tcp_init(uv_default_loop(), &conn->conn);
129
    ASSERT_OK(r);
130
    ASSERT_OK(uv_ip4_addr("127.0.0.1", TEST_PORT, &addr));
131

132
    r = uv_tcp_connect(&conn->conn_req,
133
                       (uv_tcp_t*) &conn->conn,
134
                       (const struct sockaddr*) &addr,
135
                       connect_cb);
136
    ASSERT_OK(r);
137

138
    conn->conn.data = conn;
139
  }
140
}
141

142

143
static void on_read(uv_stream_t* handle,
144
                    ssize_t nread,
145
                    const uv_buf_t* buf) {
146
  int r;
147
  uv_pipe_t* pipe;
148
  uv_handle_type pending;
149
  uv_buf_t outbuf;
150

151
  pipe = (uv_pipe_t*) handle;
152

153
  if (nread == 0) {
154
    /* Everything OK, but nothing read. */
155
    free(buf->base);
156
    return;
157
  }
158

159
  if (nread < 0) {
160
    if (nread == UV_EOF) {
161
      free(buf->base);
162
      return;
163
    }
164

165
    printf("error recving on channel: %s\n", uv_strerror(nread));
166
    abort();
167
  }
168

169
  fprintf(stderr, "got %d bytes\n", (int)nread);
170

171
  pending = uv_pipe_pending_type(pipe);
172
  if (!tcp_server_listening) {
173
    ASSERT_EQ(1, uv_pipe_pending_count(pipe));
174
    ASSERT_GT(nread, 0);
175
    ASSERT_NOT_NULL(buf->base);
176
    ASSERT_NE(pending, UV_UNKNOWN_HANDLE);
177
    read_cb_called++;
178

179
    /* Accept the pending TCP server, and start listening on it. */
180
    ASSERT_EQ(pending, UV_TCP);
181
    r = uv_tcp_init(uv_default_loop(), &tcp_server);
182
    ASSERT_OK(r);
183

184
    r = uv_accept((uv_stream_t*)pipe, (uv_stream_t*)&tcp_server);
185
    ASSERT_OK(r);
186

187
    r = uv_listen((uv_stream_t*)&tcp_server, BACKLOG, on_connection);
188
    ASSERT_OK(r);
189

190
    tcp_server_listening = 1;
191

192
    /* Make sure that the expected data is correctly multiplexed. */
193
    ASSERT_MEM_EQ("hello\n", buf->base, nread);
194

195
    outbuf = uv_buf_init("foobar\n", 7);
196
    r = uv_write(&write_req, (uv_stream_t*)pipe, &outbuf, 1, NULL);
197
    ASSERT_OK(r);
198

199
    /* Create a bunch of connections to get both servers to accept. */
200
    make_many_connections();
201
  } else if (memcmp("accepted_connection\n", buf->base, nread) == 0) {
202
    /* Remote server has accepted a connection.  Close the channel. */
203
    ASSERT_OK(uv_pipe_pending_count(pipe));
204
    ASSERT_EQ(pending, UV_UNKNOWN_HANDLE);
205
    remote_conn_accepted = 1;
206
    uv_close((uv_handle_t*)&channel, NULL);
207
  }
208

209
  free(buf->base);
210
}
211

212
#ifdef _WIN32
213
static void on_read_listen_after_bound_twice(uv_stream_t* handle,
214
                                             ssize_t nread,
215
                                             const uv_buf_t* buf) {
216
  int r;
217
  uv_pipe_t* pipe;
218
  uv_handle_type pending;
219

220
  pipe = (uv_pipe_t*) handle;
221

222
  if (nread == 0) {
223
    /* Everything OK, but nothing read. */
224
    free(buf->base);
225
    return;
226
  }
227

228
  if (nread < 0) {
229
    if (nread == UV_EOF) {
230
      free(buf->base);
231
      return;
232
    }
233

234
    printf("error recving on channel: %s\n", uv_strerror(nread));
235
    abort();
236
  }
237

238
  fprintf(stderr, "got %d bytes\n", (int)nread);
239

240
  ASSERT_GT(uv_pipe_pending_count(pipe), 0);
241
  pending = uv_pipe_pending_type(pipe);
242
  ASSERT_GT(nread, 0);
243
  ASSERT_NOT_NULL(buf->base);
244
  ASSERT_NE(pending, UV_UNKNOWN_HANDLE);
245
  read_cb_called++;
246

247
  if (read_cb_called == 1) {
248
    /* Accept the first TCP server, and start listening on it. */
249
    ASSERT_EQ(pending, UV_TCP);
250
    r = uv_tcp_init(uv_default_loop(), &tcp_server);
251
    ASSERT_OK(r);
252

253
    r = uv_accept((uv_stream_t*)pipe, (uv_stream_t*)&tcp_server);
254
    ASSERT_OK(r);
255

256
    r = uv_listen((uv_stream_t*)&tcp_server, BACKLOG, on_connection);
257
    ASSERT_OK(r);
258
  } else if (read_cb_called == 2) {
259
    /* Accept the second TCP server, and start listening on it. */
260
    ASSERT_EQ(pending, UV_TCP);
261
    r = uv_tcp_init(uv_default_loop(), &tcp_server2);
262
    ASSERT_OK(r);
263

264
    r = uv_accept((uv_stream_t*)pipe, (uv_stream_t*)&tcp_server2);
265
    ASSERT_OK(r);
266

267
    r = uv_listen((uv_stream_t*)&tcp_server2, BACKLOG, on_connection);
268
    ASSERT_EQ(r, UV_EADDRINUSE);
269

270
    uv_close((uv_handle_t*)&tcp_server, NULL);
271
    uv_close((uv_handle_t*)&tcp_server2, NULL);
272
    ASSERT_OK(uv_pipe_pending_count(pipe));
273
    uv_close((uv_handle_t*)&channel, NULL);
274
  }
275

276
  free(buf->base);
277
}
278
#endif
279

280
void spawn_helper(uv_pipe_t* channel,
281
                  uv_process_t* process,
282
                  const char* helper) {
283
  uv_process_options_t options;
284
  size_t exepath_size;
285
  char exepath[1024];
286
  char* args[3];
287
  int r;
288
  uv_stdio_container_t stdio[3];
289

290
  r = uv_pipe_init(uv_default_loop(), channel, 1);
291
  ASSERT_OK(r);
292
  ASSERT_NE(0, channel->ipc);
293

294
  exepath_size = sizeof(exepath);
295
  r = uv_exepath(exepath, &exepath_size);
296
  ASSERT_OK(r);
297

298
  exepath[exepath_size] = '\0';
299
  args[0] = exepath;
300
  args[1] = (char*)helper;
301
  args[2] = NULL;
302

303
  memset(&options, 0, sizeof(options));
304
  options.file = exepath;
305
  options.args = args;
306
  options.exit_cb = exit_cb;
307
  options.stdio = stdio;
308
  options.stdio_count = ARRAY_SIZE(stdio);
309

310
  stdio[0].flags = UV_CREATE_PIPE | UV_READABLE_PIPE | UV_WRITABLE_PIPE;
311
  stdio[0].data.stream = (uv_stream_t*) channel;
312
  stdio[1].flags = UV_INHERIT_FD;
313
  stdio[1].data.fd = 1;
314
  stdio[2].flags = UV_INHERIT_FD;
315
  stdio[2].data.fd = 2;
316

317
  r = uv_spawn(uv_default_loop(), process, &options);
318
  ASSERT_OK(r);
319
}
320

321

322
static void on_tcp_write(uv_write_t* req, int status) {
323
  ASSERT_OK(status);
324
  ASSERT_PTR_EQ(req->handle, &tcp_connection);
325
  tcp_write_cb_called++;
326
}
327

328

329
static void on_read_alloc(uv_handle_t* handle,
330
                          size_t suggested_size,
331
                          uv_buf_t* buf) {
332
  buf->base = malloc(suggested_size);
333
  buf->len = suggested_size;
334
}
335

336

337
static void on_tcp_read(uv_stream_t* tcp, ssize_t nread, const uv_buf_t* buf) {
338
  ASSERT_GT(nread, 0);
339
  ASSERT_MEM_EQ("hello again\n", buf->base, nread);
340
  ASSERT_PTR_EQ(tcp, &tcp_connection);
341
  free(buf->base);
342

343
  tcp_read_cb_called++;
344

345
  uv_close((uv_handle_t*)tcp, NULL);
346
  uv_close((uv_handle_t*)&channel, NULL);
347
}
348

349

350
static void on_read_connection(uv_stream_t* handle,
351
                               ssize_t nread,
352
                               const uv_buf_t* buf) {
353
  int r;
354
  uv_buf_t outbuf;
355
  uv_pipe_t* pipe;
356
  uv_handle_type pending;
357

358
  pipe = (uv_pipe_t*) handle;
359
  if (nread == 0) {
360
    /* Everything OK, but nothing read. */
361
    free(buf->base);
362
    return;
363
  }
364

365
  if (nread < 0) {
366
    if (nread == UV_EOF) {
367
      free(buf->base);
368
      return;
369
    }
370

371
    printf("error recving on channel: %s\n", uv_strerror(nread));
372
    abort();
373
  }
374

375
  fprintf(stderr, "got %d bytes\n", (int)nread);
376

377
  ASSERT_EQ(1, uv_pipe_pending_count(pipe));
378
  pending = uv_pipe_pending_type(pipe);
379

380
  ASSERT_GT(nread, 0);
381
  ASSERT_NOT_NULL(buf->base);
382
  ASSERT_NE(pending, UV_UNKNOWN_HANDLE);
383
  read_cb_called++;
384

385
  /* Accept the pending TCP connection */
386
  ASSERT_EQ(pending, UV_TCP);
387
  r = uv_tcp_init(uv_default_loop(), &tcp_connection);
388
  ASSERT_OK(r);
389

390
  r = uv_accept(handle, (uv_stream_t*)&tcp_connection);
391
  ASSERT_OK(r);
392

393
  /* Make sure that the expected data is correctly multiplexed. */
394
  ASSERT_MEM_EQ("hello\n", buf->base, nread);
395

396
  /* Write/read to/from the connection */
397
  outbuf = uv_buf_init("world\n", 6);
398
  r = uv_write(&write_req, (uv_stream_t*)&tcp_connection, &outbuf, 1,
399
    on_tcp_write);
400
  ASSERT_OK(r);
401

402
  r = uv_read_start((uv_stream_t*)&tcp_connection, on_read_alloc, on_tcp_read);
403
  ASSERT_OK(r);
404

405
  free(buf->base);
406
}
407

408

409
static void on_read_send_zero(uv_stream_t* handle,
410
                              ssize_t nread,
411
                              const uv_buf_t* buf) {
412
  ASSERT(nread == 0 || nread == UV_EOF);
413
  free(buf->base);
414
}
415

416

417
static int run_ipc_test(const char* helper, uv_read_cb read_cb) {
418
  uv_process_t process;
419
  int r;
420

421
  spawn_helper(&channel, &process, helper);
422
  uv_read_start((uv_stream_t*)&channel, on_alloc, read_cb);
423

424
  r = uv_run(uv_default_loop(), UV_RUN_DEFAULT);
425
  ASSERT_OK(r);
426

427
  MAKE_VALGRIND_HAPPY(uv_default_loop());
428
  return 0;
429
}
430

431

432
TEST_IMPL(ipc_listen_before_write) {
433
#if defined(NO_SEND_HANDLE_ON_PIPE)
434
  RETURN_SKIP(NO_SEND_HANDLE_ON_PIPE);
435
#endif
436
  int r = run_ipc_test("ipc_helper_listen_before_write", on_read);
437
  ASSERT_EQ(1, local_conn_accepted);
438
  ASSERT_EQ(1, remote_conn_accepted);
439
  ASSERT_EQ(1, read_cb_called);
440
  ASSERT_EQ(1, exit_cb_called);
441
  return r;
442
}
443

444

445
TEST_IMPL(ipc_listen_after_write) {
446
#if defined(NO_SEND_HANDLE_ON_PIPE)
447
  RETURN_SKIP(NO_SEND_HANDLE_ON_PIPE);
448
#endif
449
  int r = run_ipc_test("ipc_helper_listen_after_write", on_read);
450
  ASSERT_EQ(1, local_conn_accepted);
451
  ASSERT_EQ(1, remote_conn_accepted);
452
  ASSERT_EQ(1, read_cb_called);
453
  ASSERT_EQ(1, exit_cb_called);
454
  return r;
455
}
456

457

458
TEST_IMPL(ipc_tcp_connection) {
459
#if defined(NO_SEND_HANDLE_ON_PIPE)
460
  RETURN_SKIP(NO_SEND_HANDLE_ON_PIPE);
461
#endif
462
  int r = run_ipc_test("ipc_helper_tcp_connection", on_read_connection);
463
  ASSERT_EQ(1, read_cb_called);
464
  ASSERT_EQ(1, tcp_write_cb_called);
465
  ASSERT_EQ(1, tcp_read_cb_called);
466
  ASSERT_EQ(1, exit_cb_called);
467
  return r;
468
}
469

470

471
#ifdef _WIN32
472
TEST_IMPL(listen_with_simultaneous_accepts) {
473
  uv_tcp_t server;
474
  int r;
475
  struct sockaddr_in addr;
476

477
  ASSERT_OK(uv_ip4_addr("0.0.0.0", TEST_PORT, &addr));
478

479
  r = uv_tcp_init(uv_default_loop(), &server);
480
  ASSERT_OK(r);
481

482
  r = uv_tcp_bind(&server, (const struct sockaddr*) &addr, 0);
483
  ASSERT_OK(r);
484

485
  r = uv_tcp_simultaneous_accepts(&server, 1);
486
  ASSERT_OK(r);
487

488
  r = uv_listen((uv_stream_t*)&server, SOMAXCONN, NULL);
489
  ASSERT_OK(r);
490
  ASSERT_EQ(32, server.reqs_pending);
491

492
  MAKE_VALGRIND_HAPPY(uv_default_loop());
493
  return 0;
494
}
495

496

497
TEST_IMPL(listen_no_simultaneous_accepts) {
498
  uv_tcp_t server;
499
  int r;
500
  struct sockaddr_in addr;
501

502
  ASSERT_OK(uv_ip4_addr("0.0.0.0", TEST_PORT, &addr));
503

504
  r = uv_tcp_init(uv_default_loop(), &server);
505
  ASSERT_OK(r);
506

507
  r = uv_tcp_bind(&server, (const struct sockaddr*) &addr, 0);
508
  ASSERT_OK(r);
509

510
  r = uv_tcp_simultaneous_accepts(&server, 0);
511
  ASSERT_OK(r);
512

513
  r = uv_listen((uv_stream_t*)&server, SOMAXCONN, NULL);
514
  ASSERT_OK(r);
515
  ASSERT_EQ(1, server.reqs_pending);
516

517
  MAKE_VALGRIND_HAPPY(uv_default_loop());
518
  return 0;
519
}
520

521
TEST_IMPL(ipc_listen_after_bind_twice) {
522
#if defined(NO_SEND_HANDLE_ON_PIPE)
523
  RETURN_SKIP(NO_SEND_HANDLE_ON_PIPE);
524
#endif
525
  int r = run_ipc_test("ipc_helper_bind_twice", on_read_listen_after_bound_twice);
526
  ASSERT_EQ(2, read_cb_called);
527
  ASSERT_EQ(1, exit_cb_called);
528
  return r;
529
}
530
#endif
531

532
TEST_IMPL(ipc_send_zero) {
533
  int r;
534
  r = run_ipc_test("ipc_helper_send_zero", on_read_send_zero);
535
  ASSERT_OK(r);
536
  return 0;
537
}
538

539

540
/* Everything here runs in a child process. */
541

542
static tcp_conn conn;
543

544

545
static void close_cb(uv_handle_t* handle) {
546
  close_cb_called++;
547
}
548

549

550
static void conn_notify_write_cb(uv_write_t* req, int status) {
551
  uv_close((uv_handle_t*)&tcp_server, close_cb);
552
  uv_close((uv_handle_t*)&channel, close_cb);
553
}
554

555

556
static void tcp_connection_write_cb(uv_write_t* req, int status) {
557
  ASSERT_PTR_EQ(&conn.conn, req->handle);
558
  uv_close((uv_handle_t*)req->handle, close_cb);
559
  uv_close((uv_handle_t*)&channel, close_cb);
560
  uv_close((uv_handle_t*)&tcp_server, close_cb);
561
  tcp_conn_write_cb_called++;
562
}
563

564

565
static void send_zero_write_cb(uv_write_t* req, int status) {
566
  ASSERT_OK(status);
567
  send_zero_write++;
568
}
569

570
static void on_tcp_child_process_read(uv_stream_t* tcp,
571
                                      ssize_t nread,
572
                                      const uv_buf_t* buf) {
573
  uv_buf_t outbuf;
574
  int r;
575

576
  if (nread < 0) {
577
    if (nread == UV_EOF) {
578
      free(buf->base);
579
      return;
580
    }
581

582
    printf("error recving on tcp connection: %s\n", uv_strerror(nread));
583
    abort();
584
  }
585

586
  ASSERT_GT(nread, 0);
587
  ASSERT_MEM_EQ("world\n", buf->base, nread);
588
  on_pipe_read_called++;
589
  free(buf->base);
590

591
  /* Write to the socket */
592
  outbuf = uv_buf_init("hello again\n", 12);
593
  r = uv_write(&conn.tcp_write_req, tcp, &outbuf, 1, tcp_connection_write_cb);
594
  ASSERT_OK(r);
595

596
  tcp_conn_read_cb_called++;
597
}
598

599

600
static void connect_child_process_cb(uv_connect_t* req, int status) {
601
  int r;
602

603
  ASSERT_OK(status);
604
  r = uv_read_start(req->handle, on_read_alloc, on_tcp_child_process_read);
605
  ASSERT_OK(r);
606
}
607

608

609
static void ipc_on_connection(uv_stream_t* server, int status) {
610
  int r;
611
  uv_buf_t buf;
612

613
  if (!connection_accepted) {
614
    /*
615
     * Accept the connection and close it.  Also let the other
616
     * side know.
617
     */
618
    ASSERT_OK(status);
619
    ASSERT_PTR_EQ(&tcp_server, server);
620

621
    r = uv_tcp_init(server->loop, &conn.conn);
622
    ASSERT_OK(r);
623

624
    r = uv_accept(server, (uv_stream_t*)&conn.conn);
625
    ASSERT_OK(r);
626

627
    uv_close((uv_handle_t*)&conn.conn, close_cb);
628

629
    buf = uv_buf_init("accepted_connection\n", 20);
630
    r = uv_write2(&conn_notify_req, (uv_stream_t*)&channel, &buf, 1,
631
      NULL, conn_notify_write_cb);
632
    ASSERT_OK(r);
633

634
    connection_accepted = 1;
635
  }
636
}
637

638

639
static void close_and_free_cb(uv_handle_t* handle) {
640
  close_cb_called++;
641
  free(handle);
642
}
643

644
static void ipc_on_connection_tcp_conn(uv_stream_t* server, int status) {
645
  int r;
646
  uv_buf_t buf;
647
  uv_tcp_t* conn;
648

649
  ASSERT_OK(status);
650
  ASSERT_PTR_EQ(&tcp_server, server);
651

652
  conn = malloc(sizeof(*conn));
653
  ASSERT_NOT_NULL(conn);
654

655
  r = uv_tcp_init(server->loop, conn);
656
  ASSERT_OK(r);
657

658
  r = uv_accept(server, (uv_stream_t*)conn);
659
  ASSERT_OK(r);
660

661
  /* Send the accepted connection to the other process */
662
  buf = uv_buf_init("hello\n", 6);
663
  r = uv_write2(&conn_notify_req, (uv_stream_t*)&channel, &buf, 1,
664
    (uv_stream_t*)conn, NULL);
665
  ASSERT_OK(r);
666

667
  r = uv_read_start((uv_stream_t*) conn,
668
                    on_read_alloc,
669
                    on_tcp_child_process_read);
670
  ASSERT_OK(r);
671

672
  uv_close((uv_handle_t*)conn, close_and_free_cb);
673
}
674

675

676
int ipc_helper(int listen_after_write) {
677
  /*
678
   * This is launched from test-ipc.c. stdin is a duplex channel that we
679
   * over which a handle will be transmitted.
680
   */
681
  struct sockaddr_in addr;
682
  int r;
683
  uv_buf_t buf;
684

685
  ASSERT_OK(uv_ip4_addr("0.0.0.0", TEST_PORT, &addr));
686

687
  r = uv_pipe_init(uv_default_loop(), &channel, 1);
688
  ASSERT_OK(r);
689

690
  uv_pipe_open(&channel, 0);
691

692
  ASSERT_EQ(1, uv_is_readable((uv_stream_t*) &channel));
693
  ASSERT_EQ(1, uv_is_writable((uv_stream_t*) &channel));
694
  ASSERT_OK(uv_is_closing((uv_handle_t*) &channel));
695

696
  r = uv_tcp_init(uv_default_loop(), &tcp_server);
697
  ASSERT_OK(r);
698

699
  r = uv_tcp_bind(&tcp_server, (const struct sockaddr*) &addr, 0);
700
  ASSERT_OK(r);
701

702
  if (!listen_after_write) {
703
    r = uv_listen((uv_stream_t*)&tcp_server, BACKLOG, ipc_on_connection);
704
    ASSERT_OK(r);
705
  }
706

707
  buf = uv_buf_init("hello\n", 6);
708
  r = uv_write2(&write_req, (uv_stream_t*)&channel, &buf, 1,
709
      (uv_stream_t*)&tcp_server, NULL);
710
  ASSERT_OK(r);
711

712
  if (listen_after_write) {
713
    r = uv_listen((uv_stream_t*)&tcp_server, BACKLOG, ipc_on_connection);
714
    ASSERT_OK(r);
715
  }
716

717
  notify_parent_process();
718
  r = uv_run(uv_default_loop(), UV_RUN_DEFAULT);
719
  ASSERT_OK(r);
720

721
  ASSERT_EQ(1, connection_accepted);
722
  ASSERT_EQ(3, close_cb_called);
723

724
  MAKE_VALGRIND_HAPPY(uv_default_loop());
725
  return 0;
726
}
727

728

729
int ipc_helper_tcp_connection(void) {
730
  /*
731
   * This is launched from test-ipc.c. stdin is a duplex channel
732
   * over which a handle will be transmitted.
733
   */
734

735
  int r;
736
  struct sockaddr_in addr;
737

738
  r = uv_pipe_init(uv_default_loop(), &channel, 1);
739
  ASSERT_OK(r);
740

741
  uv_pipe_open(&channel, 0);
742

743
  ASSERT_EQ(1, uv_is_readable((uv_stream_t*) &channel));
744
  ASSERT_EQ(1, uv_is_writable((uv_stream_t*) &channel));
745
  ASSERT_OK(uv_is_closing((uv_handle_t*) &channel));
746

747
  r = uv_tcp_init(uv_default_loop(), &tcp_server);
748
  ASSERT_OK(r);
749

750
  ASSERT_OK(uv_ip4_addr("0.0.0.0", TEST_PORT, &addr));
751

752
  r = uv_tcp_bind(&tcp_server, (const struct sockaddr*) &addr, 0);
753
  ASSERT_OK(r);
754

755
  r = uv_listen((uv_stream_t*)&tcp_server, BACKLOG, ipc_on_connection_tcp_conn);
756
  ASSERT_OK(r);
757

758
  /* Make a connection to the server */
759
  r = uv_tcp_init(uv_default_loop(), &conn.conn);
760
  ASSERT_OK(r);
761

762
  ASSERT_OK(uv_ip4_addr("127.0.0.1", TEST_PORT, &addr));
763

764
  r = uv_tcp_connect(&conn.conn_req,
765
                     (uv_tcp_t*) &conn.conn,
766
                     (const struct sockaddr*) &addr,
767
                     connect_child_process_cb);
768
  ASSERT_OK(r);
769

770
  r = uv_run(uv_default_loop(), UV_RUN_DEFAULT);
771
  ASSERT_OK(r);
772

773
  ASSERT_EQ(1, tcp_conn_read_cb_called);
774
  ASSERT_EQ(1, tcp_conn_write_cb_called);
775
  ASSERT_EQ(4, close_cb_called);
776

777
  MAKE_VALGRIND_HAPPY(uv_default_loop());
778
  return 0;
779
}
780

781
int ipc_helper_bind_twice(void) {
782
  /*
783
   * This is launched from test-ipc.c. stdin is a duplex channel
784
   * over which two handles will be transmitted.
785
   */
786
  struct sockaddr_in addr;
787
  int r;
788
  uv_buf_t buf;
789

790
  ASSERT_OK(uv_ip4_addr("0.0.0.0", TEST_PORT, &addr));
791

792
  r = uv_pipe_init(uv_default_loop(), &channel, 1);
793
  ASSERT_OK(r);
794

795
  uv_pipe_open(&channel, 0);
796

797
  ASSERT_EQ(1, uv_is_readable((uv_stream_t*) &channel));
798
  ASSERT_EQ(1, uv_is_writable((uv_stream_t*) &channel));
799
  ASSERT_OK(uv_is_closing((uv_handle_t*) &channel));
800

801
  buf = uv_buf_init("hello\n", 6);
802

803
  r = uv_tcp_init(uv_default_loop(), &tcp_server);
804
  ASSERT_OK(r);
805
  r = uv_tcp_init(uv_default_loop(), &tcp_server2);
806
  ASSERT_OK(r);
807

808
  r = uv_tcp_bind(&tcp_server, (const struct sockaddr*) &addr, 0);
809
  ASSERT_OK(r);
810
  r = uv_tcp_bind(&tcp_server2, (const struct sockaddr*) &addr, 0);
811
  ASSERT_OK(r);
812

813
  r = uv_write2(&write_req, (uv_stream_t*)&channel, &buf, 1,
814
                (uv_stream_t*)&tcp_server, NULL);
815
  ASSERT_OK(r);
816
  r = uv_write2(&write_req2, (uv_stream_t*)&channel, &buf, 1,
817
                (uv_stream_t*)&tcp_server2, NULL);
818
  ASSERT_OK(r);
819

820
  r = uv_run(uv_default_loop(), UV_RUN_DEFAULT);
821
  ASSERT_OK(r);
822

823
  MAKE_VALGRIND_HAPPY(uv_default_loop());
824
  return 0;
825
}
826

827
int ipc_helper_send_zero(void) {
828
  int r;
829
  uv_buf_t zero_buf;
830

831
  zero_buf = uv_buf_init(0, 0);
832

833
  r = uv_pipe_init(uv_default_loop(), &channel, 0);
834
  ASSERT_OK(r);
835

836
  uv_pipe_open(&channel, 0);
837

838
  ASSERT_EQ(1, uv_is_readable((uv_stream_t*) &channel));
839
  ASSERT_EQ(1, uv_is_writable((uv_stream_t*) &channel));
840
  ASSERT_OK(uv_is_closing((uv_handle_t*) &channel));
841

842
  r = uv_write(&write_req,
843
               (uv_stream_t*)&channel,
844
               &zero_buf,
845
               1,
846
               send_zero_write_cb);
847

848
  ASSERT_OK(r);
849

850
  r = uv_run(uv_default_loop(), UV_RUN_DEFAULT);
851
  ASSERT_OK(r);
852

853
  ASSERT_EQ(1, send_zero_write);
854

855
  MAKE_VALGRIND_HAPPY(uv_default_loop());
856
  return 0;
857
}
858

Использование cookies

Мы используем файлы cookie в соответствии с Политикой конфиденциальности и Политикой использования cookies.

Нажимая кнопку «Принимаю», Вы даете АО «СберТех» согласие на обработку Ваших персональных данных в целях совершенствования нашего веб-сайта и Сервиса GitVerse, а также повышения удобства их использования.

Запретить использование cookies Вы можете самостоятельно в настройках Вашего браузера.