foolio 0.0.3

Sign up to get free protection for your applications and to get access to all the features.
Files changed (261) hide show
  1. data/.gitignore +22 -0
  2. data/Gemfile +4 -0
  3. data/LICENSE +22 -0
  4. data/README.md +29 -0
  5. data/Rakefile +21 -0
  6. data/examples/timer.rb +20 -0
  7. data/ext/foolio/extconf.rb +34 -0
  8. data/ext/foolio/foolio_ext.c +921 -0
  9. data/ext/foolio/gen.rb +50 -0
  10. data/ext/foolio/make_table.rb +12 -0
  11. data/ext/foolio/templ +243 -0
  12. data/ext/libuv/.gitignore +33 -0
  13. data/ext/libuv/.mailmap +13 -0
  14. data/ext/libuv/.travis.yml +9 -0
  15. data/ext/libuv/AUTHORS +61 -0
  16. data/ext/libuv/LICENSE +44 -0
  17. data/ext/libuv/Makefile +71 -0
  18. data/ext/libuv/README.md +90 -0
  19. data/ext/libuv/common.gypi +178 -0
  20. data/ext/libuv/gyp_uv +73 -0
  21. data/ext/libuv/include/uv-private/eio.h +403 -0
  22. data/ext/libuv/include/uv-private/ev.h +838 -0
  23. data/ext/libuv/include/uv-private/ngx-queue.h +108 -0
  24. data/ext/libuv/include/uv-private/tree.h +768 -0
  25. data/ext/libuv/include/uv-private/uv-unix.h +324 -0
  26. data/ext/libuv/include/uv-private/uv-win.h +517 -0
  27. data/ext/libuv/include/uv.h +1838 -0
  28. data/ext/libuv/src/fs-poll.c +235 -0
  29. data/ext/libuv/src/inet.c +293 -0
  30. data/ext/libuv/src/unix/async.c +148 -0
  31. data/ext/libuv/src/unix/core.c +696 -0
  32. data/ext/libuv/src/unix/cygwin.c +83 -0
  33. data/ext/libuv/src/unix/darwin.c +342 -0
  34. data/ext/libuv/src/unix/dl.c +83 -0
  35. data/ext/libuv/src/unix/eio/Changes +63 -0
  36. data/ext/libuv/src/unix/eio/LICENSE +36 -0
  37. data/ext/libuv/src/unix/eio/Makefile.am +15 -0
  38. data/ext/libuv/src/unix/eio/aclocal.m4 +8957 -0
  39. data/ext/libuv/src/unix/eio/autogen.sh +3 -0
  40. data/ext/libuv/src/unix/eio/config.h.in +86 -0
  41. data/ext/libuv/src/unix/eio/config_cygwin.h +80 -0
  42. data/ext/libuv/src/unix/eio/config_darwin.h +141 -0
  43. data/ext/libuv/src/unix/eio/config_freebsd.h +81 -0
  44. data/ext/libuv/src/unix/eio/config_linux.h +94 -0
  45. data/ext/libuv/src/unix/eio/config_netbsd.h +81 -0
  46. data/ext/libuv/src/unix/eio/config_openbsd.h +137 -0
  47. data/ext/libuv/src/unix/eio/config_sunos.h +84 -0
  48. data/ext/libuv/src/unix/eio/configure.ac +22 -0
  49. data/ext/libuv/src/unix/eio/demo.c +194 -0
  50. data/ext/libuv/src/unix/eio/ecb.h +370 -0
  51. data/ext/libuv/src/unix/eio/eio.3 +3428 -0
  52. data/ext/libuv/src/unix/eio/eio.c +2593 -0
  53. data/ext/libuv/src/unix/eio/eio.pod +969 -0
  54. data/ext/libuv/src/unix/eio/libeio.m4 +195 -0
  55. data/ext/libuv/src/unix/eio/xthread.h +164 -0
  56. data/ext/libuv/src/unix/error.c +105 -0
  57. data/ext/libuv/src/unix/ev/Changes +388 -0
  58. data/ext/libuv/src/unix/ev/LICENSE +36 -0
  59. data/ext/libuv/src/unix/ev/Makefile.am +18 -0
  60. data/ext/libuv/src/unix/ev/Makefile.in +771 -0
  61. data/ext/libuv/src/unix/ev/README +58 -0
  62. data/ext/libuv/src/unix/ev/aclocal.m4 +8957 -0
  63. data/ext/libuv/src/unix/ev/autogen.sh +6 -0
  64. data/ext/libuv/src/unix/ev/config.guess +1526 -0
  65. data/ext/libuv/src/unix/ev/config.h.in +125 -0
  66. data/ext/libuv/src/unix/ev/config.sub +1658 -0
  67. data/ext/libuv/src/unix/ev/config_cygwin.h +123 -0
  68. data/ext/libuv/src/unix/ev/config_darwin.h +122 -0
  69. data/ext/libuv/src/unix/ev/config_freebsd.h +120 -0
  70. data/ext/libuv/src/unix/ev/config_linux.h +141 -0
  71. data/ext/libuv/src/unix/ev/config_netbsd.h +120 -0
  72. data/ext/libuv/src/unix/ev/config_openbsd.h +126 -0
  73. data/ext/libuv/src/unix/ev/config_sunos.h +122 -0
  74. data/ext/libuv/src/unix/ev/configure +13037 -0
  75. data/ext/libuv/src/unix/ev/configure.ac +18 -0
  76. data/ext/libuv/src/unix/ev/depcomp +630 -0
  77. data/ext/libuv/src/unix/ev/ev++.h +816 -0
  78. data/ext/libuv/src/unix/ev/ev.3 +5311 -0
  79. data/ext/libuv/src/unix/ev/ev.c +3925 -0
  80. data/ext/libuv/src/unix/ev/ev.pod +5243 -0
  81. data/ext/libuv/src/unix/ev/ev_epoll.c +266 -0
  82. data/ext/libuv/src/unix/ev/ev_kqueue.c +235 -0
  83. data/ext/libuv/src/unix/ev/ev_poll.c +148 -0
  84. data/ext/libuv/src/unix/ev/ev_port.c +179 -0
  85. data/ext/libuv/src/unix/ev/ev_select.c +310 -0
  86. data/ext/libuv/src/unix/ev/ev_vars.h +203 -0
  87. data/ext/libuv/src/unix/ev/ev_win32.c +153 -0
  88. data/ext/libuv/src/unix/ev/ev_wrap.h +196 -0
  89. data/ext/libuv/src/unix/ev/event.c +402 -0
  90. data/ext/libuv/src/unix/ev/event.h +170 -0
  91. data/ext/libuv/src/unix/ev/install-sh +294 -0
  92. data/ext/libuv/src/unix/ev/libev.m4 +39 -0
  93. data/ext/libuv/src/unix/ev/ltmain.sh +8413 -0
  94. data/ext/libuv/src/unix/ev/missing +336 -0
  95. data/ext/libuv/src/unix/ev/mkinstalldirs +111 -0
  96. data/ext/libuv/src/unix/freebsd.c +326 -0
  97. data/ext/libuv/src/unix/fs.c +739 -0
  98. data/ext/libuv/src/unix/internal.h +188 -0
  99. data/ext/libuv/src/unix/kqueue.c +120 -0
  100. data/ext/libuv/src/unix/linux/inotify.c +239 -0
  101. data/ext/libuv/src/unix/linux/linux-core.c +557 -0
  102. data/ext/libuv/src/unix/linux/syscalls.c +388 -0
  103. data/ext/libuv/src/unix/linux/syscalls.h +124 -0
  104. data/ext/libuv/src/unix/loop-watcher.c +62 -0
  105. data/ext/libuv/src/unix/loop.c +94 -0
  106. data/ext/libuv/src/unix/netbsd.c +108 -0
  107. data/ext/libuv/src/unix/openbsd.c +295 -0
  108. data/ext/libuv/src/unix/pipe.c +259 -0
  109. data/ext/libuv/src/unix/poll.c +114 -0
  110. data/ext/libuv/src/unix/process.c +495 -0
  111. data/ext/libuv/src/unix/signal.c +269 -0
  112. data/ext/libuv/src/unix/stream.c +990 -0
  113. data/ext/libuv/src/unix/sunos.c +481 -0
  114. data/ext/libuv/src/unix/tcp.c +393 -0
  115. data/ext/libuv/src/unix/thread.c +251 -0
  116. data/ext/libuv/src/unix/timer.c +136 -0
  117. data/ext/libuv/src/unix/tty.c +145 -0
  118. data/ext/libuv/src/unix/udp.c +659 -0
  119. data/ext/libuv/src/unix/uv-eio.c +107 -0
  120. data/ext/libuv/src/unix/uv-eio.h +13 -0
  121. data/ext/libuv/src/uv-common.c +380 -0
  122. data/ext/libuv/src/uv-common.h +170 -0
  123. data/ext/libuv/src/win/async.c +100 -0
  124. data/ext/libuv/src/win/atomicops-inl.h +56 -0
  125. data/ext/libuv/src/win/core.c +278 -0
  126. data/ext/libuv/src/win/dl.c +86 -0
  127. data/ext/libuv/src/win/error.c +155 -0
  128. data/ext/libuv/src/win/fs-event.c +510 -0
  129. data/ext/libuv/src/win/fs.c +1948 -0
  130. data/ext/libuv/src/win/getaddrinfo.c +365 -0
  131. data/ext/libuv/src/win/handle-inl.h +149 -0
  132. data/ext/libuv/src/win/handle.c +154 -0
  133. data/ext/libuv/src/win/internal.h +343 -0
  134. data/ext/libuv/src/win/loop-watcher.c +122 -0
  135. data/ext/libuv/src/win/pipe.c +1672 -0
  136. data/ext/libuv/src/win/poll.c +616 -0
  137. data/ext/libuv/src/win/process-stdio.c +500 -0
  138. data/ext/libuv/src/win/process.c +1013 -0
  139. data/ext/libuv/src/win/req-inl.h +220 -0
  140. data/ext/libuv/src/win/req.c +25 -0
  141. data/ext/libuv/src/win/signal.c +57 -0
  142. data/ext/libuv/src/win/stream-inl.h +67 -0
  143. data/ext/libuv/src/win/stream.c +167 -0
  144. data/ext/libuv/src/win/tcp.c +1394 -0
  145. data/ext/libuv/src/win/thread.c +372 -0
  146. data/ext/libuv/src/win/threadpool.c +74 -0
  147. data/ext/libuv/src/win/timer.c +224 -0
  148. data/ext/libuv/src/win/tty.c +1799 -0
  149. data/ext/libuv/src/win/udp.c +716 -0
  150. data/ext/libuv/src/win/util.c +864 -0
  151. data/ext/libuv/src/win/winapi.c +132 -0
  152. data/ext/libuv/src/win/winapi.h +4452 -0
  153. data/ext/libuv/src/win/winsock.c +557 -0
  154. data/ext/libuv/src/win/winsock.h +171 -0
  155. data/ext/libuv/test/benchmark-async-pummel.c +97 -0
  156. data/ext/libuv/test/benchmark-async.c +137 -0
  157. data/ext/libuv/test/benchmark-fs-stat.c +135 -0
  158. data/ext/libuv/test/benchmark-getaddrinfo.c +94 -0
  159. data/ext/libuv/test/benchmark-list.h +127 -0
  160. data/ext/libuv/test/benchmark-loop-count.c +88 -0
  161. data/ext/libuv/test/benchmark-million-timers.c +65 -0
  162. data/ext/libuv/test/benchmark-ping-pongs.c +213 -0
  163. data/ext/libuv/test/benchmark-pound.c +324 -0
  164. data/ext/libuv/test/benchmark-pump.c +462 -0
  165. data/ext/libuv/test/benchmark-sizes.c +44 -0
  166. data/ext/libuv/test/benchmark-spawn.c +162 -0
  167. data/ext/libuv/test/benchmark-tcp-write-batch.c +140 -0
  168. data/ext/libuv/test/benchmark-thread.c +64 -0
  169. data/ext/libuv/test/benchmark-udp-packet-storm.c +247 -0
  170. data/ext/libuv/test/blackhole-server.c +118 -0
  171. data/ext/libuv/test/dns-server.c +321 -0
  172. data/ext/libuv/test/echo-server.c +378 -0
  173. data/ext/libuv/test/fixtures/empty_file +0 -0
  174. data/ext/libuv/test/fixtures/load_error.node +1 -0
  175. data/ext/libuv/test/run-benchmarks.c +64 -0
  176. data/ext/libuv/test/run-tests.c +138 -0
  177. data/ext/libuv/test/runner-unix.c +295 -0
  178. data/ext/libuv/test/runner-unix.h +36 -0
  179. data/ext/libuv/test/runner-win.c +285 -0
  180. data/ext/libuv/test/runner-win.h +42 -0
  181. data/ext/libuv/test/runner.c +355 -0
  182. data/ext/libuv/test/runner.h +159 -0
  183. data/ext/libuv/test/task.h +112 -0
  184. data/ext/libuv/test/test-async.c +118 -0
  185. data/ext/libuv/test/test-callback-order.c +76 -0
  186. data/ext/libuv/test/test-callback-stack.c +203 -0
  187. data/ext/libuv/test/test-connection-fail.c +148 -0
  188. data/ext/libuv/test/test-cwd-and-chdir.c +64 -0
  189. data/ext/libuv/test/test-delayed-accept.c +188 -0
  190. data/ext/libuv/test/test-dlerror.c +58 -0
  191. data/ext/libuv/test/test-error.c +59 -0
  192. data/ext/libuv/test/test-fail-always.c +29 -0
  193. data/ext/libuv/test/test-fs-event.c +474 -0
  194. data/ext/libuv/test/test-fs-poll.c +146 -0
  195. data/ext/libuv/test/test-fs.c +1843 -0
  196. data/ext/libuv/test/test-get-currentexe.c +63 -0
  197. data/ext/libuv/test/test-get-loadavg.c +36 -0
  198. data/ext/libuv/test/test-get-memory.c +38 -0
  199. data/ext/libuv/test/test-getaddrinfo.c +122 -0
  200. data/ext/libuv/test/test-getsockname.c +342 -0
  201. data/ext/libuv/test/test-hrtime.c +54 -0
  202. data/ext/libuv/test/test-idle.c +81 -0
  203. data/ext/libuv/test/test-ipc-send-recv.c +209 -0
  204. data/ext/libuv/test/test-ipc.c +620 -0
  205. data/ext/libuv/test/test-list.h +427 -0
  206. data/ext/libuv/test/test-loop-handles.c +336 -0
  207. data/ext/libuv/test/test-multiple-listen.c +102 -0
  208. data/ext/libuv/test/test-mutexes.c +63 -0
  209. data/ext/libuv/test/test-pass-always.c +28 -0
  210. data/ext/libuv/test/test-ping-pong.c +253 -0
  211. data/ext/libuv/test/test-pipe-bind-error.c +140 -0
  212. data/ext/libuv/test/test-pipe-connect-error.c +96 -0
  213. data/ext/libuv/test/test-platform-output.c +87 -0
  214. data/ext/libuv/test/test-poll-close.c +72 -0
  215. data/ext/libuv/test/test-poll.c +573 -0
  216. data/ext/libuv/test/test-process-title.c +49 -0
  217. data/ext/libuv/test/test-ref.c +338 -0
  218. data/ext/libuv/test/test-run-once.c +48 -0
  219. data/ext/libuv/test/test-semaphore.c +111 -0
  220. data/ext/libuv/test/test-shutdown-close.c +103 -0
  221. data/ext/libuv/test/test-shutdown-eof.c +183 -0
  222. data/ext/libuv/test/test-signal.c +162 -0
  223. data/ext/libuv/test/test-spawn.c +863 -0
  224. data/ext/libuv/test/test-stdio-over-pipes.c +246 -0
  225. data/ext/libuv/test/test-tcp-bind-error.c +191 -0
  226. data/ext/libuv/test/test-tcp-bind6-error.c +154 -0
  227. data/ext/libuv/test/test-tcp-close-while-connecting.c +80 -0
  228. data/ext/libuv/test/test-tcp-close.c +129 -0
  229. data/ext/libuv/test/test-tcp-connect-error-after-write.c +95 -0
  230. data/ext/libuv/test/test-tcp-connect-error.c +70 -0
  231. data/ext/libuv/test/test-tcp-connect-timeout.c +85 -0
  232. data/ext/libuv/test/test-tcp-connect6-error.c +68 -0
  233. data/ext/libuv/test/test-tcp-flags.c +51 -0
  234. data/ext/libuv/test/test-tcp-shutdown-after-write.c +131 -0
  235. data/ext/libuv/test/test-tcp-unexpected-read.c +113 -0
  236. data/ext/libuv/test/test-tcp-write-error.c +168 -0
  237. data/ext/libuv/test/test-tcp-write-to-half-open-connection.c +135 -0
  238. data/ext/libuv/test/test-tcp-writealot.c +170 -0
  239. data/ext/libuv/test/test-thread.c +183 -0
  240. data/ext/libuv/test/test-threadpool.c +57 -0
  241. data/ext/libuv/test/test-timer-again.c +141 -0
  242. data/ext/libuv/test/test-timer.c +152 -0
  243. data/ext/libuv/test/test-tty.c +110 -0
  244. data/ext/libuv/test/test-udp-dgram-too-big.c +86 -0
  245. data/ext/libuv/test/test-udp-ipv6.c +156 -0
  246. data/ext/libuv/test/test-udp-multicast-join.c +139 -0
  247. data/ext/libuv/test/test-udp-multicast-ttl.c +86 -0
  248. data/ext/libuv/test/test-udp-options.c +86 -0
  249. data/ext/libuv/test/test-udp-send-and-recv.c +208 -0
  250. data/ext/libuv/test/test-util.c +97 -0
  251. data/ext/libuv/test/test-walk-handles.c +77 -0
  252. data/ext/libuv/uv.gyp +375 -0
  253. data/ext/libuv/vcbuild.bat +105 -0
  254. data/foolio.gemspec +18 -0
  255. data/lib/foolio.rb +9 -0
  256. data/lib/foolio/handle.rb +27 -0
  257. data/lib/foolio/listener.rb +26 -0
  258. data/lib/foolio/loop.rb +79 -0
  259. data/lib/foolio/stream.rb +109 -0
  260. data/lib/foolio/version.rb +3 -0
  261. metadata +309 -0
@@ -0,0 +1,324 @@
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 "task.h"
23
+ #include "uv.h"
24
+
25
+ /* Update this is you're going to run > 1000 concurrent requests. */
26
+ #define MAX_CONNS 1000
27
+
28
+ #undef NANOSEC
29
+ #define NANOSEC ((uint64_t) 1e9)
30
+
31
+ #undef DEBUG
32
+ #define DEBUG 0
33
+
34
+ struct conn_rec_s;
35
+
36
+ typedef void (*setup_fn)(int num, void* arg);
37
+ typedef void (*make_connect_fn)(struct conn_rec_s* conn);
38
+ typedef int (*connect_fn)(int num, make_connect_fn make_connect, void* arg);
39
+
40
+ /* Base class for tcp_conn_rec and pipe_conn_rec.
41
+ * The ordering of fields matters!
42
+ */
43
+ typedef struct conn_rec_s {
44
+ int i;
45
+ uv_connect_t conn_req;
46
+ uv_write_t write_req;
47
+ make_connect_fn make_connect;
48
+ uv_stream_t stream;
49
+ } conn_rec;
50
+
51
+ typedef struct {
52
+ int i;
53
+ uv_connect_t conn_req;
54
+ uv_write_t write_req;
55
+ make_connect_fn make_connect;
56
+ uv_tcp_t stream;
57
+ } tcp_conn_rec;
58
+
59
+ typedef struct {
60
+ int i;
61
+ uv_connect_t conn_req;
62
+ uv_write_t write_req;
63
+ make_connect_fn make_connect;
64
+ uv_pipe_t stream;
65
+ } pipe_conn_rec;
66
+
67
+ static char buffer[] = "QS";
68
+
69
+ static uv_loop_t* loop;
70
+
71
+ static tcp_conn_rec tcp_conns[MAX_CONNS];
72
+ static pipe_conn_rec pipe_conns[MAX_CONNS];
73
+
74
+ static uint64_t start; /* in ms */
75
+ static int closed_streams;
76
+ static int conns_failed;
77
+
78
+ static uv_buf_t alloc_cb(uv_handle_t* handle, size_t suggested_size);
79
+ static void connect_cb(uv_connect_t* conn_req, int status);
80
+ static void read_cb(uv_stream_t* stream, ssize_t nread, uv_buf_t buf);
81
+ static void close_cb(uv_handle_t* handle);
82
+
83
+
84
+ static uv_buf_t alloc_cb(uv_handle_t* handle, size_t suggested_size) {
85
+ static char slab[65536];
86
+ uv_buf_t buf;
87
+ buf.base = slab;
88
+ buf.len = sizeof(slab);
89
+ return buf;
90
+ }
91
+
92
+
93
+ static void after_write(uv_write_t* req, int status) {
94
+ if (status != 0) {
95
+ fprintf(stderr, "write error %s\n", uv_err_name(uv_last_error(loop)));
96
+ uv_close((uv_handle_t*)req->handle, close_cb);
97
+ conns_failed++;
98
+ return;
99
+ }
100
+ }
101
+
102
+
103
+ static void connect_cb(uv_connect_t* req, int status) {
104
+ conn_rec* conn;
105
+ uv_buf_t buf;
106
+ int r;
107
+
108
+ if (status != 0) {
109
+ #if DEBUG
110
+ fprintf(stderr,
111
+ "connect error %s\n",
112
+ uv_err_name(uv_last_error(uv_default_loop())));
113
+ #endif
114
+ uv_close((uv_handle_t*)req->handle, close_cb);
115
+ conns_failed++;
116
+ return;
117
+ }
118
+
119
+ ASSERT(req != NULL);
120
+ ASSERT(status == 0);
121
+
122
+ conn = (conn_rec*)req->data;
123
+ ASSERT(conn != NULL);
124
+
125
+ #if DEBUG
126
+ printf("connect_cb %d\n", conn->i);
127
+ #endif
128
+
129
+ r = uv_read_start(&conn->stream, alloc_cb, read_cb);
130
+ ASSERT(r == 0);
131
+
132
+ buf.base = buffer;
133
+ buf.len = sizeof(buffer) - 1;
134
+
135
+ r = uv_write(&conn->write_req, &conn->stream, &buf, 1, after_write);
136
+ ASSERT(r == 0);
137
+ }
138
+
139
+
140
+ static void read_cb(uv_stream_t* stream, ssize_t nread, uv_buf_t buf) {
141
+ uv_err_t err = uv_last_error(loop);
142
+
143
+ ASSERT(stream != NULL);
144
+
145
+ #if DEBUG
146
+ printf("read_cb %d\n", p->i);
147
+ #endif
148
+
149
+ uv_close((uv_handle_t*)stream, close_cb);
150
+
151
+ if (nread == -1) {
152
+ if (err.code == UV_EOF) {
153
+ ;
154
+ } else if (err.code == UV_ECONNRESET) {
155
+ conns_failed++;
156
+ } else {
157
+ fprintf(stderr, "read error %s\n", uv_err_name(uv_last_error(loop)));
158
+ ASSERT(0);
159
+ }
160
+ }
161
+ }
162
+
163
+
164
+ static void close_cb(uv_handle_t* handle) {
165
+ conn_rec* p = (conn_rec*)handle->data;
166
+
167
+ ASSERT(handle != NULL);
168
+ closed_streams++;
169
+
170
+ #if DEBUG
171
+ printf("close_cb %d\n", p->i);
172
+ #endif
173
+
174
+ if (uv_now(loop) - start < 10000) {
175
+ p->make_connect(p);
176
+ }
177
+ }
178
+
179
+
180
+ static void tcp_do_setup(int num, void* arg) {
181
+ int i;
182
+
183
+ for (i = 0; i < num; i++) {
184
+ tcp_conns[i].i = i;
185
+ }
186
+ }
187
+
188
+
189
+ static void pipe_do_setup(int num, void* arg) {
190
+ int i;
191
+
192
+ for (i = 0; i < num; i++) {
193
+ pipe_conns[i].i = i;
194
+ }
195
+ }
196
+
197
+
198
+ static void tcp_make_connect(conn_rec* p) {
199
+ struct sockaddr_in addr;
200
+ int r;
201
+
202
+ r = uv_tcp_init(loop, (uv_tcp_t*)&p->stream);
203
+ ASSERT(r == 0);
204
+
205
+ addr = uv_ip4_addr("127.0.0.1", TEST_PORT);
206
+
207
+ r = uv_tcp_connect(&((tcp_conn_rec*)p)->conn_req, (uv_tcp_t*)&p->stream, addr, connect_cb);
208
+ if (r) {
209
+ fprintf(stderr, "uv_tcp_connect error %s\n",
210
+ uv_err_name(uv_last_error(loop)));
211
+ ASSERT(0);
212
+ }
213
+
214
+ #if DEBUG
215
+ printf("make connect %d\n", p->i);
216
+ #endif
217
+
218
+ p->conn_req.data = p;
219
+ p->write_req.data = p;
220
+ p->stream.data = p;
221
+ }
222
+
223
+
224
+ static void pipe_make_connect(conn_rec* p) {
225
+ int r;
226
+
227
+ r = uv_pipe_init(loop, (uv_pipe_t*)&p->stream, 0);
228
+ ASSERT(r == 0);
229
+
230
+ uv_pipe_connect(&((pipe_conn_rec*)p)->conn_req, (uv_pipe_t*)&p->stream, TEST_PIPENAME, connect_cb);
231
+
232
+ #if DEBUG
233
+ printf("make connect %d\n", p->i);
234
+ #endif
235
+
236
+ p->conn_req.data = p;
237
+ p->write_req.data = p;
238
+ p->stream.data = p;
239
+ }
240
+
241
+
242
+ static int tcp_do_connect(int num, make_connect_fn make_connect, void* arg) {
243
+ int i;
244
+
245
+ for (i = 0; i < num; i++) {
246
+ tcp_make_connect((conn_rec*)&tcp_conns[i]);
247
+ tcp_conns[i].make_connect = make_connect;
248
+ }
249
+
250
+ return 0;
251
+ }
252
+
253
+
254
+ static int pipe_do_connect(int num, make_connect_fn make_connect, void* arg) {
255
+ int i;
256
+
257
+ for (i = 0; i < num; i++) {
258
+ pipe_make_connect((conn_rec*)&pipe_conns[i]);
259
+ pipe_conns[i].make_connect = make_connect;
260
+ }
261
+
262
+ return 0;
263
+ }
264
+
265
+
266
+ static int pound_it(int concurrency,
267
+ const char* type,
268
+ setup_fn do_setup,
269
+ connect_fn do_connect,
270
+ make_connect_fn make_connect,
271
+ void* arg) {
272
+ double secs;
273
+ int r;
274
+ uint64_t start_time; /* in ns */
275
+ uint64_t end_time;
276
+
277
+ loop = uv_default_loop();
278
+
279
+ uv_update_time(loop);
280
+ start = uv_now(loop);
281
+
282
+ /* Run benchmark for at least five seconds. */
283
+ start_time = uv_hrtime();
284
+
285
+ do_setup(concurrency, arg);
286
+
287
+ r = do_connect(concurrency, make_connect, arg);
288
+ ASSERT(!r);
289
+
290
+ uv_run(loop);
291
+
292
+ end_time = uv_hrtime();
293
+
294
+ /* Number of fractional seconds it took to run the benchmark. */
295
+ secs = (double)(end_time - start_time) / NANOSEC;
296
+
297
+ LOGF("%s-conn-pound-%d: %.0f accepts/s (%d failed)\n",
298
+ type,
299
+ concurrency,
300
+ closed_streams / secs,
301
+ conns_failed);
302
+
303
+ return 0;
304
+ }
305
+
306
+
307
+ BENCHMARK_IMPL(tcp4_pound_100) {
308
+ return pound_it(100, "tcp", tcp_do_setup, tcp_do_connect, tcp_make_connect, NULL);
309
+ }
310
+
311
+
312
+ BENCHMARK_IMPL(tcp4_pound_1000) {
313
+ return pound_it(1000, "tcp", tcp_do_setup, tcp_do_connect, tcp_make_connect, NULL);
314
+ }
315
+
316
+
317
+ BENCHMARK_IMPL(pipe_pound_100) {
318
+ return pound_it(100, "pipe", pipe_do_setup, pipe_do_connect, pipe_make_connect, NULL);
319
+ }
320
+
321
+
322
+ BENCHMARK_IMPL(pipe_pound_1000) {
323
+ return pound_it(1000, "pipe", pipe_do_setup, pipe_do_connect, pipe_make_connect, NULL);
324
+ }
@@ -0,0 +1,462 @@
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 "task.h"
23
+ #include "uv.h"
24
+
25
+ #include <math.h>
26
+ #include <stdio.h>
27
+
28
+
29
+ static int TARGET_CONNECTIONS;
30
+ #define WRITE_BUFFER_SIZE 8192
31
+ #define MAX_SIMULTANEOUS_CONNECTS 100
32
+
33
+ #define PRINT_STATS 0
34
+ #define STATS_INTERVAL 1000 /* msec */
35
+ #define STATS_COUNT 5
36
+
37
+
38
+ static void do_write(uv_stream_t*);
39
+ static void maybe_connect_some();
40
+
41
+ static uv_req_t* req_alloc();
42
+ static void req_free(uv_req_t* uv_req);
43
+
44
+ static uv_buf_t buf_alloc(uv_handle_t*, size_t size);
45
+ static void buf_free(uv_buf_t uv_buf_t);
46
+
47
+ static uv_loop_t* loop;
48
+
49
+ static uv_tcp_t tcpServer;
50
+ static uv_pipe_t pipeServer;
51
+ static uv_stream_t* server;
52
+ static struct sockaddr_in listen_addr;
53
+ static struct sockaddr_in connect_addr;
54
+
55
+ static int64_t start_time;
56
+
57
+ static int max_connect_socket = 0;
58
+ static int max_read_sockets = 0;
59
+ static int read_sockets = 0;
60
+ static int write_sockets = 0;
61
+
62
+ static int64_t nrecv = 0;
63
+ static int64_t nrecv_total = 0;
64
+ static int64_t nsent = 0;
65
+ static int64_t nsent_total = 0;
66
+
67
+ static int stats_left = 0;
68
+
69
+ static char write_buffer[WRITE_BUFFER_SIZE];
70
+
71
+ /* Make this as large as you need. */
72
+ #define MAX_WRITE_HANDLES 1000
73
+
74
+ static stream_type type;
75
+
76
+ static uv_tcp_t tcp_write_handles[MAX_WRITE_HANDLES];
77
+ static uv_pipe_t pipe_write_handles[MAX_WRITE_HANDLES];
78
+
79
+ static uv_timer_t timer_handle;
80
+
81
+
82
+ static double gbit(int64_t bytes, int64_t passed_ms) {
83
+ double gbits = ((double)bytes / (1024 * 1024 * 1024)) * 8;
84
+ return gbits / ((double)passed_ms / 1000);
85
+ }
86
+
87
+
88
+ static void show_stats(uv_timer_t* handle, int status) {
89
+ int64_t diff;
90
+ int i;
91
+
92
+ #if PRINT_STATS
93
+ LOGF("connections: %d, write: %.1f gbit/s\n",
94
+ write_sockets,
95
+ gbit(nsent, STATS_INTERVAL));
96
+ #endif
97
+
98
+ /* Exit if the show is over */
99
+ if (!--stats_left) {
100
+
101
+ uv_update_time(loop);
102
+ diff = uv_now(loop) - start_time;
103
+
104
+ LOGF("%s_pump%d_client: %.1f gbit/s\n", type == TCP ? "tcp" : "pipe", write_sockets,
105
+ gbit(nsent_total, diff));
106
+
107
+ for (i = 0; i < write_sockets; i++) {
108
+ uv_close(type == TCP ? (uv_handle_t*)&tcp_write_handles[i] : (uv_handle_t*)&pipe_write_handles[i], NULL);
109
+ }
110
+
111
+ exit(0);
112
+ }
113
+
114
+ /* Reset read and write counters */
115
+ nrecv = 0;
116
+ nsent = 0;
117
+ }
118
+
119
+
120
+ static void read_show_stats() {
121
+ int64_t diff;
122
+
123
+ uv_update_time(loop);
124
+ diff = uv_now(loop) - start_time;
125
+
126
+ LOGF("%s_pump%d_server: %.1f gbit/s\n", type == TCP ? "tcp" : "pipe", max_read_sockets,
127
+ gbit(nrecv_total, diff));
128
+ }
129
+
130
+
131
+
132
+ void write_sockets_close_cb(uv_handle_t* handle) {
133
+ /* If any client closes, the process is done. */
134
+ exit(0);
135
+ }
136
+
137
+
138
+ void read_sockets_close_cb(uv_handle_t* handle) {
139
+ free(handle);
140
+ read_sockets--;
141
+
142
+ /* If it's past the first second and everyone has closed their connection
143
+ * Then print stats.
144
+ */
145
+ if (uv_now(loop) - start_time > 1000 && read_sockets == 0) {
146
+ read_show_stats();
147
+ uv_close((uv_handle_t*)server, NULL);
148
+ }
149
+ }
150
+
151
+
152
+ static void start_stats_collection() {
153
+ int r;
154
+
155
+ /* Show-stats timer */
156
+ stats_left = STATS_COUNT;
157
+ r = uv_timer_init(loop, &timer_handle);
158
+ ASSERT(r == 0);
159
+ r = uv_timer_start(&timer_handle, show_stats, STATS_INTERVAL, STATS_INTERVAL);
160
+ ASSERT(r == 0);
161
+
162
+ uv_update_time(loop);
163
+ start_time = uv_now(loop);
164
+ }
165
+
166
+
167
+ static void read_cb(uv_stream_t* stream, ssize_t bytes, uv_buf_t buf) {
168
+ if (nrecv_total == 0) {
169
+ ASSERT(start_time == 0);
170
+ uv_update_time(loop);
171
+ start_time = uv_now(loop);
172
+ }
173
+
174
+ if (bytes < 0) {
175
+ uv_close((uv_handle_t*)stream, read_sockets_close_cb);
176
+ return;
177
+ }
178
+
179
+ buf_free(buf);
180
+
181
+ nrecv += bytes;
182
+ nrecv_total += bytes;
183
+ }
184
+
185
+
186
+ static void write_cb(uv_write_t* req, int status) {
187
+ ASSERT(status == 0);
188
+
189
+ req_free((uv_req_t*) req);
190
+
191
+ nsent += sizeof write_buffer;
192
+ nsent_total += sizeof write_buffer;
193
+
194
+ do_write((uv_stream_t*) req->handle);
195
+ }
196
+
197
+
198
+ static void do_write(uv_stream_t* stream) {
199
+ uv_write_t* req;
200
+ uv_buf_t buf;
201
+ int r;
202
+
203
+ buf.base = (char*) &write_buffer;
204
+ buf.len = sizeof write_buffer;
205
+
206
+ while (stream->write_queue_size == 0) {
207
+ req = (uv_write_t*) req_alloc();
208
+ r = uv_write(req, stream, &buf, 1, write_cb);
209
+ ASSERT(r == 0);
210
+ }
211
+ }
212
+
213
+
214
+ static void connect_cb(uv_connect_t* req, int status) {
215
+ int i;
216
+
217
+ if (status) LOG(uv_strerror(uv_last_error(loop)));
218
+ ASSERT(status == 0);
219
+
220
+ write_sockets++;
221
+ req_free((uv_req_t*) req);
222
+
223
+ maybe_connect_some();
224
+
225
+ if (write_sockets == TARGET_CONNECTIONS) {
226
+ start_stats_collection();
227
+
228
+ /* Yay! start writing */
229
+ for (i = 0; i < write_sockets; i++) {
230
+ do_write(type == TCP ? (uv_stream_t*)&tcp_write_handles[i] : (uv_stream_t*)&pipe_write_handles[i]);
231
+ }
232
+ }
233
+ }
234
+
235
+
236
+ static void maybe_connect_some() {
237
+ uv_connect_t* req;
238
+ uv_tcp_t* tcp;
239
+ uv_pipe_t* pipe;
240
+ int r;
241
+
242
+ while (max_connect_socket < TARGET_CONNECTIONS &&
243
+ max_connect_socket < write_sockets + MAX_SIMULTANEOUS_CONNECTS) {
244
+ if (type == TCP) {
245
+ tcp = &tcp_write_handles[max_connect_socket++];
246
+
247
+ r = uv_tcp_init(loop, tcp);
248
+ ASSERT(r == 0);
249
+
250
+ req = (uv_connect_t*) req_alloc();
251
+ r = uv_tcp_connect(req, tcp, connect_addr, connect_cb);
252
+ ASSERT(r == 0);
253
+ } else {
254
+ pipe = &pipe_write_handles[max_connect_socket++];
255
+
256
+ r = uv_pipe_init(loop, pipe, 0);
257
+ ASSERT(r == 0);
258
+
259
+ req = (uv_connect_t*) req_alloc();
260
+ uv_pipe_connect(req, pipe, TEST_PIPENAME, connect_cb);
261
+ }
262
+ }
263
+ }
264
+
265
+
266
+ static void connection_cb(uv_stream_t* s, int status) {
267
+ uv_stream_t* stream;
268
+ int r;
269
+
270
+ ASSERT(server == s);
271
+ ASSERT(status == 0);
272
+
273
+ if (type == TCP) {
274
+ stream = (uv_stream_t*)malloc(sizeof(uv_tcp_t));
275
+ r = uv_tcp_init(loop, (uv_tcp_t*)stream);
276
+ ASSERT(r == 0);
277
+ } else {
278
+ stream = (uv_stream_t*)malloc(sizeof(uv_pipe_t));
279
+ r = uv_pipe_init(loop, (uv_pipe_t*)stream, 0);
280
+ ASSERT(r == 0);
281
+ }
282
+
283
+ r = uv_accept(s, stream);
284
+ ASSERT(r == 0);
285
+
286
+ r = uv_read_start(stream, buf_alloc, read_cb);
287
+ ASSERT(r == 0);
288
+
289
+ read_sockets++;
290
+ max_read_sockets++;
291
+ }
292
+
293
+
294
+ /*
295
+ * Request allocator
296
+ */
297
+
298
+ typedef struct req_list_s {
299
+ union uv_any_req uv_req;
300
+ struct req_list_s* next;
301
+ } req_list_t;
302
+
303
+
304
+ static req_list_t* req_freelist = NULL;
305
+
306
+
307
+ static uv_req_t* req_alloc() {
308
+ req_list_t* req;
309
+
310
+ req = req_freelist;
311
+ if (req != NULL) {
312
+ req_freelist = req->next;
313
+ return (uv_req_t*) req;
314
+ }
315
+
316
+ req = (req_list_t*) malloc(sizeof *req);
317
+ return (uv_req_t*) req;
318
+ }
319
+
320
+
321
+ static void req_free(uv_req_t* uv_req) {
322
+ req_list_t* req = (req_list_t*) uv_req;
323
+
324
+ req->next = req_freelist;
325
+ req_freelist = req;
326
+ }
327
+
328
+
329
+ /*
330
+ * Buffer allocator
331
+ */
332
+
333
+ typedef struct buf_list_s {
334
+ uv_buf_t uv_buf_t;
335
+ struct buf_list_s* next;
336
+ } buf_list_t;
337
+
338
+
339
+ static buf_list_t* buf_freelist = NULL;
340
+
341
+
342
+ static uv_buf_t buf_alloc(uv_handle_t* handle, size_t size) {
343
+ buf_list_t* buf;
344
+
345
+ buf = buf_freelist;
346
+ if (buf != NULL) {
347
+ buf_freelist = buf->next;
348
+ return buf->uv_buf_t;
349
+ }
350
+
351
+ buf = (buf_list_t*) malloc(size + sizeof *buf);
352
+ buf->uv_buf_t.len = (unsigned int)size;
353
+ buf->uv_buf_t.base = ((char*) buf) + sizeof *buf;
354
+
355
+ return buf->uv_buf_t;
356
+ }
357
+
358
+
359
+ static void buf_free(uv_buf_t uv_buf_t) {
360
+ buf_list_t* buf = (buf_list_t*) (uv_buf_t.base - sizeof *buf);
361
+
362
+ buf->next = buf_freelist;
363
+ buf_freelist = buf;
364
+ }
365
+
366
+
367
+ HELPER_IMPL(tcp_pump_server) {
368
+ int r;
369
+
370
+ type = TCP;
371
+ loop = uv_default_loop();
372
+
373
+ listen_addr = uv_ip4_addr("0.0.0.0", TEST_PORT);
374
+
375
+ /* Server */
376
+ server = (uv_stream_t*)&tcpServer;
377
+ r = uv_tcp_init(loop, &tcpServer);
378
+ ASSERT(r == 0);
379
+ r = uv_tcp_bind(&tcpServer, listen_addr);
380
+ ASSERT(r == 0);
381
+ r = uv_listen((uv_stream_t*)&tcpServer, MAX_WRITE_HANDLES, connection_cb);
382
+ ASSERT(r == 0);
383
+
384
+ uv_run(loop);
385
+
386
+ return 0;
387
+ }
388
+
389
+
390
+ HELPER_IMPL(pipe_pump_server) {
391
+ int r;
392
+ type = PIPE;
393
+
394
+ loop = uv_default_loop();
395
+
396
+ /* Server */
397
+ server = (uv_stream_t*)&pipeServer;
398
+ r = uv_pipe_init(loop, &pipeServer, 0);
399
+ ASSERT(r == 0);
400
+ r = uv_pipe_bind(&pipeServer, TEST_PIPENAME);
401
+ ASSERT(r == 0);
402
+ r = uv_listen((uv_stream_t*)&pipeServer, MAX_WRITE_HANDLES, connection_cb);
403
+ ASSERT(r == 0);
404
+
405
+ uv_run(loop);
406
+
407
+ return 0;
408
+ }
409
+
410
+
411
+ void tcp_pump(int n) {
412
+ ASSERT(n <= MAX_WRITE_HANDLES);
413
+ TARGET_CONNECTIONS = n;
414
+ type = TCP;
415
+
416
+ loop = uv_default_loop();
417
+
418
+ connect_addr = uv_ip4_addr("127.0.0.1", TEST_PORT);
419
+
420
+ /* Start making connections */
421
+ maybe_connect_some();
422
+
423
+ uv_run(loop);
424
+ }
425
+
426
+
427
+ void pipe_pump(int n) {
428
+ ASSERT(n <= MAX_WRITE_HANDLES);
429
+ TARGET_CONNECTIONS = n;
430
+ type = PIPE;
431
+
432
+ loop = uv_default_loop();
433
+
434
+ /* Start making connections */
435
+ maybe_connect_some();
436
+
437
+ uv_run(loop);
438
+ }
439
+
440
+
441
+ BENCHMARK_IMPL(tcp_pump100_client) {
442
+ tcp_pump(100);
443
+ return 0;
444
+ }
445
+
446
+
447
+ BENCHMARK_IMPL(tcp_pump1_client) {
448
+ tcp_pump(1);
449
+ return 0;
450
+ }
451
+
452
+
453
+ BENCHMARK_IMPL(pipe_pump100_client) {
454
+ pipe_pump(100);
455
+ return 0;
456
+ }
457
+
458
+
459
+ BENCHMARK_IMPL(pipe_pump1_client) {
460
+ pipe_pump(1);
461
+ return 0;
462
+ }