rbuv 0.0.1 → 0.0.2

Sign up to get free protection for your applications and to get access to all the features.
Files changed (233) hide show
  1. checksums.yaml +4 -4
  2. data/.gitignore +5 -0
  3. data/.rspec +2 -0
  4. data/.travis.yml +5 -0
  5. data/Gemfile +0 -1
  6. data/README.md +6 -1
  7. data/Rakefile +42 -0
  8. data/deps/libuv/.gitignore +34 -0
  9. data/deps/libuv/.mailmap +16 -0
  10. data/deps/libuv/AUTHORS +81 -0
  11. data/deps/libuv/ChangeLog +45 -0
  12. data/deps/libuv/LICENSE +41 -0
  13. data/deps/libuv/Makefile +53 -0
  14. data/deps/libuv/README.md +118 -0
  15. data/deps/libuv/build.mk +164 -0
  16. data/deps/libuv/checksparse.sh +230 -0
  17. data/deps/libuv/common.gypi +197 -0
  18. data/deps/libuv/config-mingw.mk +48 -0
  19. data/deps/libuv/config-unix.mk +167 -0
  20. data/deps/libuv/gyp_uv +98 -0
  21. data/deps/libuv/include/uv-private/ngx-queue.h +129 -0
  22. data/deps/libuv/include/uv-private/stdint-msvc2008.h +247 -0
  23. data/deps/libuv/include/uv-private/tree.h +768 -0
  24. data/deps/libuv/include/uv-private/uv-bsd.h +34 -0
  25. data/deps/libuv/include/uv-private/uv-darwin.h +61 -0
  26. data/deps/libuv/include/uv-private/uv-linux.h +34 -0
  27. data/deps/libuv/include/uv-private/uv-sunos.h +44 -0
  28. data/deps/libuv/include/uv-private/uv-unix.h +332 -0
  29. data/deps/libuv/include/uv-private/uv-win.h +585 -0
  30. data/deps/libuv/include/uv.h +1987 -0
  31. data/deps/libuv/src/fs-poll.c +248 -0
  32. data/deps/libuv/src/inet.c +298 -0
  33. data/deps/libuv/src/unix/aix.c +393 -0
  34. data/deps/libuv/src/unix/async.c +281 -0
  35. data/deps/libuv/src/unix/core.c +714 -0
  36. data/deps/libuv/src/unix/cygwin.c +93 -0
  37. data/deps/libuv/src/unix/darwin-proctitle.m +78 -0
  38. data/deps/libuv/src/unix/darwin.c +431 -0
  39. data/deps/libuv/src/unix/dl.c +83 -0
  40. data/deps/libuv/src/unix/error.c +109 -0
  41. data/deps/libuv/src/unix/freebsd.c +343 -0
  42. data/deps/libuv/src/unix/fs.c +869 -0
  43. data/deps/libuv/src/unix/fsevents.c +299 -0
  44. data/deps/libuv/src/unix/getaddrinfo.c +159 -0
  45. data/deps/libuv/src/unix/internal.h +259 -0
  46. data/deps/libuv/src/unix/kqueue.c +347 -0
  47. data/deps/libuv/src/unix/linux-core.c +724 -0
  48. data/deps/libuv/src/unix/linux-inotify.c +236 -0
  49. data/deps/libuv/src/unix/linux-syscalls.c +388 -0
  50. data/deps/libuv/src/unix/linux-syscalls.h +150 -0
  51. data/deps/libuv/src/unix/loop-watcher.c +64 -0
  52. data/deps/libuv/src/unix/loop.c +114 -0
  53. data/deps/libuv/src/unix/netbsd.c +353 -0
  54. data/deps/libuv/src/unix/openbsd.c +304 -0
  55. data/deps/libuv/src/unix/pipe.c +261 -0
  56. data/deps/libuv/src/unix/poll.c +108 -0
  57. data/deps/libuv/src/unix/process.c +501 -0
  58. data/deps/libuv/src/unix/proctitle.c +103 -0
  59. data/deps/libuv/src/unix/signal.c +455 -0
  60. data/deps/libuv/src/unix/stream.c +1380 -0
  61. data/deps/libuv/src/unix/sunos.c +647 -0
  62. data/deps/libuv/src/unix/tcp.c +357 -0
  63. data/deps/libuv/src/unix/thread.c +431 -0
  64. data/deps/libuv/src/unix/threadpool.c +286 -0
  65. data/deps/libuv/src/unix/timer.c +153 -0
  66. data/deps/libuv/src/unix/tty.c +179 -0
  67. data/deps/libuv/src/unix/udp.c +715 -0
  68. data/deps/libuv/src/uv-common.c +431 -0
  69. data/deps/libuv/src/uv-common.h +204 -0
  70. data/deps/libuv/src/version.c +60 -0
  71. data/deps/libuv/src/win/async.c +99 -0
  72. data/deps/libuv/src/win/atomicops-inl.h +56 -0
  73. data/deps/libuv/src/win/core.c +310 -0
  74. data/deps/libuv/src/win/dl.c +86 -0
  75. data/deps/libuv/src/win/error.c +164 -0
  76. data/deps/libuv/src/win/fs-event.c +506 -0
  77. data/deps/libuv/src/win/fs.c +1951 -0
  78. data/deps/libuv/src/win/getaddrinfo.c +365 -0
  79. data/deps/libuv/src/win/handle-inl.h +164 -0
  80. data/deps/libuv/src/win/handle.c +153 -0
  81. data/deps/libuv/src/win/internal.h +346 -0
  82. data/deps/libuv/src/win/loop-watcher.c +124 -0
  83. data/deps/libuv/src/win/pipe.c +1656 -0
  84. data/deps/libuv/src/win/poll.c +615 -0
  85. data/deps/libuv/src/win/process-stdio.c +503 -0
  86. data/deps/libuv/src/win/process.c +1048 -0
  87. data/deps/libuv/src/win/req-inl.h +224 -0
  88. data/deps/libuv/src/win/req.c +25 -0
  89. data/deps/libuv/src/win/signal.c +354 -0
  90. data/deps/libuv/src/win/stream-inl.h +67 -0
  91. data/deps/libuv/src/win/stream.c +198 -0
  92. data/deps/libuv/src/win/tcp.c +1422 -0
  93. data/deps/libuv/src/win/thread.c +666 -0
  94. data/deps/libuv/src/win/threadpool.c +82 -0
  95. data/deps/libuv/src/win/timer.c +230 -0
  96. data/deps/libuv/src/win/tty.c +1857 -0
  97. data/deps/libuv/src/win/udp.c +744 -0
  98. data/deps/libuv/src/win/util.c +946 -0
  99. data/deps/libuv/src/win/winapi.c +152 -0
  100. data/deps/libuv/src/win/winapi.h +4476 -0
  101. data/deps/libuv/src/win/winsock.c +560 -0
  102. data/deps/libuv/src/win/winsock.h +171 -0
  103. data/deps/libuv/test/benchmark-async-pummel.c +119 -0
  104. data/deps/libuv/test/benchmark-async.c +139 -0
  105. data/deps/libuv/test/benchmark-fs-stat.c +136 -0
  106. data/deps/libuv/test/benchmark-getaddrinfo.c +91 -0
  107. data/deps/libuv/test/benchmark-list.h +163 -0
  108. data/deps/libuv/test/benchmark-loop-count.c +90 -0
  109. data/deps/libuv/test/benchmark-million-async.c +112 -0
  110. data/deps/libuv/test/benchmark-million-timers.c +77 -0
  111. data/deps/libuv/test/benchmark-multi-accept.c +432 -0
  112. data/deps/libuv/test/benchmark-ping-pongs.c +212 -0
  113. data/deps/libuv/test/benchmark-pound.c +325 -0
  114. data/deps/libuv/test/benchmark-pump.c +459 -0
  115. data/deps/libuv/test/benchmark-sizes.c +45 -0
  116. data/deps/libuv/test/benchmark-spawn.c +163 -0
  117. data/deps/libuv/test/benchmark-tcp-write-batch.c +141 -0
  118. data/deps/libuv/test/benchmark-thread.c +64 -0
  119. data/deps/libuv/test/benchmark-udp-pummel.c +238 -0
  120. data/deps/libuv/test/blackhole-server.c +118 -0
  121. data/deps/libuv/test/dns-server.c +329 -0
  122. data/deps/libuv/test/echo-server.c +384 -0
  123. data/deps/libuv/test/fixtures/empty_file +0 -0
  124. data/deps/libuv/test/fixtures/load_error.node +1 -0
  125. data/deps/libuv/test/run-benchmarks.c +64 -0
  126. data/deps/libuv/test/run-tests.c +159 -0
  127. data/deps/libuv/test/runner-unix.c +328 -0
  128. data/deps/libuv/test/runner-unix.h +36 -0
  129. data/deps/libuv/test/runner-win.c +318 -0
  130. data/deps/libuv/test/runner-win.h +43 -0
  131. data/deps/libuv/test/runner.c +394 -0
  132. data/deps/libuv/test/runner.h +165 -0
  133. data/deps/libuv/test/task.h +122 -0
  134. data/deps/libuv/test/test-active.c +83 -0
  135. data/deps/libuv/test/test-async.c +136 -0
  136. data/deps/libuv/test/test-barrier.c +98 -0
  137. data/deps/libuv/test/test-callback-order.c +77 -0
  138. data/deps/libuv/test/test-callback-stack.c +204 -0
  139. data/deps/libuv/test/test-condvar.c +173 -0
  140. data/deps/libuv/test/test-connection-fail.c +150 -0
  141. data/deps/libuv/test/test-cwd-and-chdir.c +64 -0
  142. data/deps/libuv/test/test-delayed-accept.c +189 -0
  143. data/deps/libuv/test/test-dlerror.c +58 -0
  144. data/deps/libuv/test/test-embed.c +136 -0
  145. data/deps/libuv/test/test-error.c +59 -0
  146. data/deps/libuv/test/test-fail-always.c +29 -0
  147. data/deps/libuv/test/test-fs-event.c +504 -0
  148. data/deps/libuv/test/test-fs-poll.c +148 -0
  149. data/deps/libuv/test/test-fs.c +1899 -0
  150. data/deps/libuv/test/test-get-currentexe.c +63 -0
  151. data/deps/libuv/test/test-get-loadavg.c +36 -0
  152. data/deps/libuv/test/test-get-memory.c +38 -0
  153. data/deps/libuv/test/test-getaddrinfo.c +120 -0
  154. data/deps/libuv/test/test-getsockname.c +344 -0
  155. data/deps/libuv/test/test-hrtime.c +54 -0
  156. data/deps/libuv/test/test-idle.c +82 -0
  157. data/deps/libuv/test/test-ipc-send-recv.c +218 -0
  158. data/deps/libuv/test/test-ipc.c +625 -0
  159. data/deps/libuv/test/test-list.h +492 -0
  160. data/deps/libuv/test/test-loop-handles.c +337 -0
  161. data/deps/libuv/test/test-loop-stop.c +73 -0
  162. data/deps/libuv/test/test-multiple-listen.c +103 -0
  163. data/deps/libuv/test/test-mutexes.c +63 -0
  164. data/deps/libuv/test/test-pass-always.c +28 -0
  165. data/deps/libuv/test/test-ping-pong.c +250 -0
  166. data/deps/libuv/test/test-pipe-bind-error.c +144 -0
  167. data/deps/libuv/test/test-pipe-connect-error.c +98 -0
  168. data/deps/libuv/test/test-platform-output.c +87 -0
  169. data/deps/libuv/test/test-poll-close.c +73 -0
  170. data/deps/libuv/test/test-poll.c +575 -0
  171. data/deps/libuv/test/test-process-title.c +49 -0
  172. data/deps/libuv/test/test-ref.c +415 -0
  173. data/deps/libuv/test/test-run-nowait.c +46 -0
  174. data/deps/libuv/test/test-run-once.c +49 -0
  175. data/deps/libuv/test/test-semaphore.c +111 -0
  176. data/deps/libuv/test/test-shutdown-close.c +105 -0
  177. data/deps/libuv/test/test-shutdown-eof.c +184 -0
  178. data/deps/libuv/test/test-signal-multiple-loops.c +270 -0
  179. data/deps/libuv/test/test-signal.c +152 -0
  180. data/deps/libuv/test/test-spawn.c +938 -0
  181. data/deps/libuv/test/test-stdio-over-pipes.c +250 -0
  182. data/deps/libuv/test/test-tcp-bind-error.c +198 -0
  183. data/deps/libuv/test/test-tcp-bind6-error.c +159 -0
  184. data/deps/libuv/test/test-tcp-close-while-connecting.c +81 -0
  185. data/deps/libuv/test/test-tcp-close.c +130 -0
  186. data/deps/libuv/test/test-tcp-connect-error-after-write.c +96 -0
  187. data/deps/libuv/test/test-tcp-connect-error.c +71 -0
  188. data/deps/libuv/test/test-tcp-connect-timeout.c +86 -0
  189. data/deps/libuv/test/test-tcp-connect6-error.c +69 -0
  190. data/deps/libuv/test/test-tcp-flags.c +52 -0
  191. data/deps/libuv/test/test-tcp-open.c +175 -0
  192. data/deps/libuv/test/test-tcp-read-stop.c +73 -0
  193. data/deps/libuv/test/test-tcp-shutdown-after-write.c +132 -0
  194. data/deps/libuv/test/test-tcp-unexpected-read.c +114 -0
  195. data/deps/libuv/test/test-tcp-write-to-half-open-connection.c +136 -0
  196. data/deps/libuv/test/test-tcp-writealot.c +171 -0
  197. data/deps/libuv/test/test-thread.c +183 -0
  198. data/deps/libuv/test/test-threadpool-cancel.c +311 -0
  199. data/deps/libuv/test/test-threadpool.c +77 -0
  200. data/deps/libuv/test/test-timer-again.c +142 -0
  201. data/deps/libuv/test/test-timer.c +266 -0
  202. data/deps/libuv/test/test-tty.c +111 -0
  203. data/deps/libuv/test/test-udp-dgram-too-big.c +87 -0
  204. data/deps/libuv/test/test-udp-ipv6.c +158 -0
  205. data/deps/libuv/test/test-udp-multicast-join.c +140 -0
  206. data/deps/libuv/test/test-udp-multicast-ttl.c +87 -0
  207. data/deps/libuv/test/test-udp-open.c +154 -0
  208. data/deps/libuv/test/test-udp-options.c +87 -0
  209. data/deps/libuv/test/test-udp-send-and-recv.c +210 -0
  210. data/deps/libuv/test/test-util.c +97 -0
  211. data/deps/libuv/test/test-walk-handles.c +78 -0
  212. data/deps/libuv/uv.gyp +431 -0
  213. data/deps/libuv/vcbuild.bat +128 -0
  214. data/ext/rbuv/debug.h +27 -0
  215. data/ext/rbuv/error.c +7 -0
  216. data/ext/rbuv/error.h +10 -0
  217. data/ext/rbuv/extconf.rb +35 -0
  218. data/ext/rbuv/handle.c +40 -0
  219. data/ext/rbuv/handle.h +14 -0
  220. data/ext/rbuv/libuv.mk +12 -0
  221. data/ext/rbuv/loop.c +50 -0
  222. data/ext/rbuv/loop.h +13 -0
  223. data/ext/rbuv/rbuv.c +15 -0
  224. data/ext/rbuv/rbuv.h +27 -0
  225. data/ext/rbuv/timer.c +133 -0
  226. data/ext/rbuv/timer.h +13 -0
  227. data/lib/rbuv/timer.rb +7 -0
  228. data/lib/rbuv/version.rb +1 -1
  229. data/lib/rbuv.rb +24 -2
  230. data/rbuv.gemspec +5 -1
  231. data/spec/spec_helper.rb +22 -0
  232. data/spec/timer_spec.rb +144 -0
  233. metadata +278 -9
@@ -0,0 +1,141 @@
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
+ #define WRITE_REQ_DATA "Hello, world."
29
+ #define NUM_WRITE_REQS (1000 * 1000)
30
+
31
+ typedef struct {
32
+ uv_write_t req;
33
+ uv_buf_t buf;
34
+ } write_req;
35
+
36
+
37
+ static write_req* write_reqs;
38
+ static uv_tcp_t tcp_client;
39
+ static uv_connect_t connect_req;
40
+ static uv_shutdown_t shutdown_req;
41
+
42
+ static int shutdown_cb_called = 0;
43
+ static int connect_cb_called = 0;
44
+ static int write_cb_called = 0;
45
+ static int close_cb_called = 0;
46
+
47
+ static void connect_cb(uv_connect_t* req, int status);
48
+ static void write_cb(uv_write_t* req, int status);
49
+ static void shutdown_cb(uv_shutdown_t* req, int status);
50
+ static void close_cb(uv_handle_t* handle);
51
+
52
+
53
+ static void connect_cb(uv_connect_t* req, int status) {
54
+ write_req* w;
55
+ int i;
56
+ int r;
57
+
58
+ ASSERT(req->handle == (uv_stream_t*)&tcp_client);
59
+
60
+ for (i = 0; i < NUM_WRITE_REQS; i++) {
61
+ w = &write_reqs[i];
62
+ r = uv_write(&w->req, req->handle, &w->buf, 1, write_cb);
63
+ ASSERT(r == 0);
64
+ }
65
+
66
+ r = uv_shutdown(&shutdown_req, req->handle, shutdown_cb);
67
+ ASSERT(r == 0);
68
+
69
+ connect_cb_called++;
70
+ }
71
+
72
+
73
+ static void write_cb(uv_write_t* req, int status) {
74
+ ASSERT(req != NULL);
75
+ ASSERT(status == 0);
76
+ write_cb_called++;
77
+ }
78
+
79
+
80
+ static void shutdown_cb(uv_shutdown_t* req, int status) {
81
+ ASSERT(req->handle == (uv_stream_t*)&tcp_client);
82
+ ASSERT(req->handle->write_queue_size == 0);
83
+
84
+ uv_close((uv_handle_t*)req->handle, close_cb);
85
+ free(write_reqs);
86
+
87
+ shutdown_cb_called++;
88
+ }
89
+
90
+
91
+ static void close_cb(uv_handle_t* handle) {
92
+ ASSERT(handle == (uv_handle_t*)&tcp_client);
93
+ close_cb_called++;
94
+ }
95
+
96
+
97
+ BENCHMARK_IMPL(tcp_write_batch) {
98
+ struct sockaddr_in addr;
99
+ uv_loop_t* loop;
100
+ uint64_t start;
101
+ uint64_t stop;
102
+ int i;
103
+ int r;
104
+
105
+ write_reqs = malloc(sizeof(*write_reqs) * NUM_WRITE_REQS);
106
+ ASSERT(write_reqs != NULL);
107
+
108
+ /* Prepare the data to write out. */
109
+ for (i = 0; i < NUM_WRITE_REQS; i++) {
110
+ write_reqs[i].buf = uv_buf_init(WRITE_REQ_DATA,
111
+ sizeof(WRITE_REQ_DATA) - 1);
112
+ }
113
+
114
+ loop = uv_default_loop();
115
+ addr = uv_ip4_addr("127.0.0.1", TEST_PORT);
116
+
117
+ r = uv_tcp_init(loop, &tcp_client);
118
+ ASSERT(r == 0);
119
+
120
+ r = uv_tcp_connect(&connect_req, &tcp_client, addr, connect_cb);
121
+ ASSERT(r == 0);
122
+
123
+ start = uv_hrtime();
124
+
125
+ r = uv_run(loop, UV_RUN_DEFAULT);
126
+ ASSERT(r == 0);
127
+
128
+ stop = uv_hrtime();
129
+
130
+ ASSERT(connect_cb_called == 1);
131
+ ASSERT(write_cb_called == NUM_WRITE_REQS);
132
+ ASSERT(shutdown_cb_called == 1);
133
+ ASSERT(close_cb_called == 1);
134
+
135
+ printf("%ld write requests in %.2fs.\n",
136
+ (long)NUM_WRITE_REQS,
137
+ (stop - start) / 1e9);
138
+
139
+ MAKE_VALGRIND_HAPPY();
140
+ return 0;
141
+ }
@@ -0,0 +1,64 @@
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
+ #define NUM_THREADS (20 * 1000)
29
+
30
+ static volatile int num_threads;
31
+
32
+
33
+ static void thread_entry(void* arg) {
34
+ ASSERT(arg == (void *) 42);
35
+ num_threads++;
36
+ /* FIXME write barrier? */
37
+ }
38
+
39
+
40
+ BENCHMARK_IMPL(thread_create) {
41
+ uint64_t start_time;
42
+ double duration;
43
+ uv_thread_t tid;
44
+ int i, r;
45
+
46
+ start_time = uv_hrtime();
47
+
48
+ for (i = 0; i < NUM_THREADS; i++) {
49
+ r = uv_thread_create(&tid, thread_entry, (void *) 42);
50
+ ASSERT(r == 0);
51
+
52
+ r = uv_thread_join(&tid);
53
+ ASSERT(r == 0);
54
+ }
55
+
56
+ duration = (uv_hrtime() - start_time) / 1e9;
57
+
58
+ ASSERT(num_threads == NUM_THREADS);
59
+
60
+ printf("%d threads created in %.2f seconds (%.0f/s)\n",
61
+ NUM_THREADS, duration, NUM_THREADS / duration);
62
+
63
+ return 0;
64
+ }
@@ -0,0 +1,238 @@
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 <stdio.h>
26
+ #include <stdlib.h>
27
+ #include <string.h>
28
+
29
+ #define EXPECTED "RANG TANG DING DONG I AM THE JAPANESE SANDMAN" /* "Take eight!" */
30
+
31
+ #define TEST_DURATION 5000 /* ms */
32
+
33
+ #define BASE_PORT 12345
34
+
35
+ struct sender_state {
36
+ struct sockaddr_in addr;
37
+ uv_udp_send_t send_req;
38
+ uv_udp_t udp_handle;
39
+ };
40
+
41
+ struct receiver_state {
42
+ struct sockaddr_in addr;
43
+ uv_udp_t udp_handle;
44
+ };
45
+
46
+ /* not used in timed mode */
47
+ static unsigned int packet_counter = (unsigned int) 1e6;
48
+
49
+ static int n_senders_;
50
+ static int n_receivers_;
51
+ static uv_buf_t bufs[5];
52
+ static struct sender_state senders[1024];
53
+ static struct receiver_state receivers[1024];
54
+
55
+ static unsigned int send_cb_called;
56
+ static unsigned int recv_cb_called;
57
+ static unsigned int close_cb_called;
58
+ static int timed;
59
+ static int exiting;
60
+
61
+
62
+ static uv_buf_t alloc_cb(uv_handle_t* handle, size_t suggested_size) {
63
+ static char slab[65536];
64
+ ASSERT(suggested_size <= sizeof slab);
65
+ return uv_buf_init(slab, sizeof slab);
66
+ }
67
+
68
+
69
+ static void send_cb(uv_udp_send_t* req, int status) {
70
+ struct sender_state* s;
71
+
72
+ ASSERT(req != NULL);
73
+
74
+ if (status != 0) {
75
+ ASSERT(status == -1);
76
+ ASSERT(uv_last_error(req->handle->loop).code == UV_ECANCELED);
77
+ return;
78
+ }
79
+
80
+ if (exiting)
81
+ return;
82
+
83
+ s = container_of(req, struct sender_state, send_req);
84
+ ASSERT(req->handle == &s->udp_handle);
85
+
86
+ if (timed)
87
+ goto send;
88
+
89
+ if (packet_counter == 0) {
90
+ uv_close((uv_handle_t*)&s->udp_handle, NULL);
91
+ return;
92
+ }
93
+
94
+ packet_counter--;
95
+
96
+ send:
97
+ ASSERT(0 == uv_udp_send(&s->send_req,
98
+ &s->udp_handle,
99
+ bufs,
100
+ ARRAY_SIZE(bufs),
101
+ s->addr,
102
+ send_cb));
103
+ send_cb_called++;
104
+ }
105
+
106
+
107
+ static void recv_cb(uv_udp_t* handle,
108
+ ssize_t nread,
109
+ uv_buf_t buf,
110
+ struct sockaddr* addr,
111
+ unsigned flags) {
112
+ if (nread == 0)
113
+ return;
114
+
115
+ if (nread == -1) {
116
+ ASSERT(uv_last_error(handle->loop).code == UV_ECANCELED);
117
+ return;
118
+ }
119
+
120
+ ASSERT(addr->sa_family == AF_INET);
121
+ ASSERT(!memcmp(buf.base, EXPECTED, nread));
122
+
123
+ recv_cb_called++;
124
+ }
125
+
126
+
127
+ static void close_cb(uv_handle_t* handle) {
128
+ ASSERT(handle != NULL);
129
+ close_cb_called++;
130
+ }
131
+
132
+
133
+ static void timeout_cb(uv_timer_t* timer, int status) {
134
+ int i;
135
+
136
+ exiting = 1;
137
+
138
+ for (i = 0; i < n_senders_; i++)
139
+ uv_close((uv_handle_t*)&senders[i].udp_handle, close_cb);
140
+
141
+ for (i = 0; i < n_receivers_; i++)
142
+ uv_close((uv_handle_t*)&receivers[i].udp_handle, close_cb);
143
+ }
144
+
145
+
146
+ static int pummel(unsigned int n_senders,
147
+ unsigned int n_receivers,
148
+ unsigned long timeout) {
149
+ uv_timer_t timer_handle;
150
+ uint64_t duration;
151
+ uv_loop_t* loop;
152
+ unsigned int i;
153
+
154
+ ASSERT(n_senders <= ARRAY_SIZE(senders));
155
+ ASSERT(n_receivers <= ARRAY_SIZE(receivers));
156
+
157
+ loop = uv_default_loop();
158
+
159
+ n_senders_ = n_senders;
160
+ n_receivers_ = n_receivers;
161
+
162
+ if (timeout) {
163
+ ASSERT(0 == uv_timer_init(loop, &timer_handle));
164
+ ASSERT(0 == uv_timer_start(&timer_handle, timeout_cb, timeout, 0));
165
+ /* Timer should not keep loop alive. */
166
+ uv_unref((uv_handle_t*)&timer_handle);
167
+ timed = 1;
168
+ }
169
+
170
+ for (i = 0; i < n_receivers; i++) {
171
+ struct receiver_state* s = receivers + i;
172
+ struct sockaddr_in addr = uv_ip4_addr("0.0.0.0", BASE_PORT + i);
173
+ ASSERT(0 == uv_udp_init(loop, &s->udp_handle));
174
+ ASSERT(0 == uv_udp_bind(&s->udp_handle, addr, 0));
175
+ ASSERT(0 == uv_udp_recv_start(&s->udp_handle, alloc_cb, recv_cb));
176
+ uv_unref((uv_handle_t*)&s->udp_handle);
177
+ }
178
+
179
+ bufs[0] = uv_buf_init(EXPECTED + 0, 10);
180
+ bufs[1] = uv_buf_init(EXPECTED + 10, 10);
181
+ bufs[2] = uv_buf_init(EXPECTED + 20, 10);
182
+ bufs[3] = uv_buf_init(EXPECTED + 30, 10);
183
+ bufs[4] = uv_buf_init(EXPECTED + 40, 5);
184
+
185
+ for (i = 0; i < n_senders; i++) {
186
+ struct sender_state* s = senders + i;
187
+ s->addr = uv_ip4_addr("127.0.0.1", BASE_PORT + (i % n_receivers));
188
+ ASSERT(0 == uv_udp_init(loop, &s->udp_handle));
189
+ ASSERT(0 == uv_udp_send(&s->send_req,
190
+ &s->udp_handle,
191
+ bufs,
192
+ ARRAY_SIZE(bufs),
193
+ s->addr,
194
+ send_cb));
195
+ }
196
+
197
+ duration = uv_hrtime();
198
+ ASSERT(0 == uv_run(loop, UV_RUN_DEFAULT));
199
+ duration = uv_hrtime() - duration;
200
+ /* convert from nanoseconds to milliseconds */
201
+ duration = duration / (uint64_t) 1e6;
202
+
203
+ printf("udp_pummel_%dv%d: %.0f/s received, %.0f/s sent. "
204
+ "%u received, %u sent in %.1f seconds.\n",
205
+ n_receivers,
206
+ n_senders,
207
+ recv_cb_called / (duration / 1000.0),
208
+ send_cb_called / (duration / 1000.0),
209
+ recv_cb_called,
210
+ send_cb_called,
211
+ duration / 1000.0);
212
+
213
+ MAKE_VALGRIND_HAPPY();
214
+ return 0;
215
+ }
216
+
217
+
218
+ #define X(a, b) \
219
+ BENCHMARK_IMPL(udp_pummel_##a##v##b) { \
220
+ return pummel(a, b, 0); \
221
+ } \
222
+ BENCHMARK_IMPL(udp_timed_pummel_##a##v##b) { \
223
+ return pummel(a, b, TEST_DURATION); \
224
+ }
225
+
226
+ X(1, 1)
227
+ X(1, 10)
228
+ X(1, 100)
229
+ X(1, 1000)
230
+ X(10, 10)
231
+ X(10, 100)
232
+ X(10, 1000)
233
+ X(100, 10)
234
+ X(100, 100)
235
+ X(100, 1000)
236
+ X(1000, 1000)
237
+
238
+ #undef X
@@ -0,0 +1,118 @@
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
+ }