src/libuv/test/blackhole-server.c

   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 <stdlib.h>
  27
  28 typedef struct {
  29   uv_tcp_t handle;
  30   uv_shutdown_t shutdown_req;
  31 } conn_rec;
  32
  33 static uv_tcp_t tcp_server;
  34
  35 static void connection_cb(uv_stream_t* stream, int status);
  36 static uv_buf_t alloc_cb(uv_handle_t* handle, size_t suggested_size);
  37 static void read_cb(uv_stream_t* stream, ssize_t nread, uv_buf_t buf);
  38 static void shutdown_cb(uv_shutdown_t* req, int status);
  39 static void close_cb(uv_handle_t* handle);
  40
  41
  42 static void connection_cb(uv_stream_t* stream, int status) {
  43   conn_rec* conn;
  44   int r;
  45
  46   ASSERT(status == 0);
  47   ASSERT(stream == (uv_stream_t*)&tcp_server);
  48
  49   conn = malloc(sizeof *conn);
  50   ASSERT(conn != NULL);
  51
  52   r = uv_tcp_init(stream->loop, &conn->handle);
  53   ASSERT(r == 0);
  54
  55   r = uv_accept(stream, (uv_stream_t*)&conn->handle);
  56   ASSERT(r == 0);
  57
  58   r = uv_read_start((uv_stream_t*)&conn->handle, alloc_cb, read_cb);
  59   ASSERT(r == 0);
  60 }
  61
  62
  63 static uv_buf_t alloc_cb(uv_handle_t* handle, size_t suggested_size) {
  64   static char buf[65536];
  65   return uv_buf_init(buf, sizeof buf);
  66 }
  67
  68
  69 static void read_cb(uv_stream_t* stream, ssize_t nread, uv_buf_t buf) {
  70   conn_rec* conn;
  71   int r;
  72
  73   if (nread >= 0)
  74     return;
  75
  76   ASSERT(uv_last_error(stream->loop).code == UV_EOF);
  77
  78   conn = container_of(stream, conn_rec, handle);
  79
  80   r = uv_shutdown(&conn->shutdown_req, stream, shutdown_cb);
  81   ASSERT(r == 0);
  82 }
  83
  84
  85 static void shutdown_cb(uv_shutdown_t* req, int status) {
  86   conn_rec* conn = container_of(req, conn_rec, shutdown_req);
  87   uv_close((uv_handle_t*)&conn->handle, close_cb);
  88 }
  89
  90
  91 static void close_cb(uv_handle_t* handle) {
  92   conn_rec* conn = container_of(handle, conn_rec, handle);
  93   free(conn);
  94 }
  95
  96
  97 HELPER_IMPL(tcp4_blackhole_server) {
  98   struct sockaddr_in addr;
  99   uv_loop_t* loop;
 100   int r;
 101
 102   loop = uv_default_loop();
 103   addr = uv_ip4_addr("127.0.0.1", TEST_PORT);
 104
 105   r = uv_tcp_init(loop, &tcp_server);
 106   ASSERT(r == 0);
 107
 108   r = uv_tcp_bind(&tcp_server, addr);
 109   ASSERT(r == 0);
 110
 111   r = uv_listen((uv_stream_t*)&tcp_server, 128, connection_cb);
 112   ASSERT(r == 0);
 113
 114   r = uv_run(loop, UV_RUN_DEFAULT);
 115   ASSERT(0 && "Blackhole server dropped out of event loop.");
 116
 117   return 0;
 118 }