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,716 @@
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 <assert.h>
23
+
24
+ #include "uv.h"
25
+ #include "internal.h"
26
+ #include "handle-inl.h"
27
+ #include "stream-inl.h"
28
+ #include "req-inl.h"
29
+
30
+
31
+ /*
32
+ * Threshold of active udp streams for which to preallocate udp read buffers.
33
+ */
34
+ const unsigned int uv_active_udp_streams_threshold = 0;
35
+
36
+ /* A zero-size buffer for use by uv_udp_read */
37
+ static char uv_zero_[] = "";
38
+
39
+ int uv_udp_getsockname(uv_udp_t* handle, struct sockaddr* name,
40
+ int* namelen) {
41
+ uv_loop_t* loop = handle->loop;
42
+ int result;
43
+
44
+ if (!(handle->flags & UV_HANDLE_BOUND)) {
45
+ uv__set_sys_error(loop, WSAEINVAL);
46
+ return -1;
47
+ }
48
+
49
+ result = getsockname(handle->socket, name, namelen);
50
+ if (result != 0) {
51
+ uv__set_sys_error(loop, WSAGetLastError());
52
+ return -1;
53
+ }
54
+
55
+ return 0;
56
+ }
57
+
58
+
59
+ static int uv_udp_set_socket(uv_loop_t* loop, uv_udp_t* handle,
60
+ SOCKET socket) {
61
+ DWORD yes = 1;
62
+ WSAPROTOCOL_INFOW info;
63
+ int opt_len;
64
+
65
+ assert(handle->socket == INVALID_SOCKET);
66
+
67
+ /* Set the socket to nonblocking mode */
68
+ if (ioctlsocket(socket, FIONBIO, &yes) == SOCKET_ERROR) {
69
+ uv__set_sys_error(loop, WSAGetLastError());
70
+ return -1;
71
+ }
72
+
73
+ /* Make the socket non-inheritable */
74
+ if (!SetHandleInformation((HANDLE)socket, HANDLE_FLAG_INHERIT, 0)) {
75
+ uv__set_sys_error(loop, GetLastError());
76
+ return -1;
77
+ }
78
+
79
+ /* Associate it with the I/O completion port. */
80
+ /* Use uv_handle_t pointer as completion key. */
81
+ if (CreateIoCompletionPort((HANDLE)socket,
82
+ loop->iocp,
83
+ (ULONG_PTR)socket,
84
+ 0) == NULL) {
85
+ uv__set_sys_error(loop, GetLastError());
86
+ return -1;
87
+ }
88
+
89
+ if (pSetFileCompletionNotificationModes) {
90
+ /* All know windowses that support SetFileCompletionNotificationModes */
91
+ /* have a bug that makes it impossible to use this function in */
92
+ /* conjunction with datagram sockets. We can work around that but only */
93
+ /* if the user is using the default UDP driver (AFD) and has no other */
94
+ /* LSPs stacked on top. Here we check whether that is the case. */
95
+ opt_len = (int) sizeof info;
96
+ if (!getsockopt(socket,
97
+ SOL_SOCKET,
98
+ SO_PROTOCOL_INFOW,
99
+ (char*) &info,
100
+ &opt_len) == SOCKET_ERROR) {
101
+ uv__set_sys_error(loop, GetLastError());
102
+ return -1;
103
+ }
104
+
105
+ if (info.ProtocolChain.ChainLen == 1) {
106
+ if (pSetFileCompletionNotificationModes((HANDLE)socket,
107
+ FILE_SKIP_SET_EVENT_ON_HANDLE |
108
+ FILE_SKIP_COMPLETION_PORT_ON_SUCCESS)) {
109
+ handle->flags |= UV_HANDLE_SYNC_BYPASS_IOCP;
110
+ handle->func_wsarecv = uv_wsarecv_workaround;
111
+ handle->func_wsarecvfrom = uv_wsarecvfrom_workaround;
112
+ } else if (GetLastError() != ERROR_INVALID_FUNCTION) {
113
+ uv__set_sys_error(loop, GetLastError());
114
+ return -1;
115
+ }
116
+ }
117
+ }
118
+
119
+ handle->socket = socket;
120
+
121
+ return 0;
122
+ }
123
+
124
+
125
+ int uv_udp_init(uv_loop_t* loop, uv_udp_t* handle) {
126
+ uv__handle_init(loop, (uv_handle_t*) handle, UV_UDP);
127
+
128
+ handle->socket = INVALID_SOCKET;
129
+ handle->reqs_pending = 0;
130
+ handle->activecnt = 0;
131
+ handle->func_wsarecv = WSARecv;
132
+ handle->func_wsarecvfrom = WSARecvFrom;
133
+
134
+ uv_req_init(loop, (uv_req_t*) &(handle->recv_req));
135
+ handle->recv_req.type = UV_UDP_RECV;
136
+ handle->recv_req.data = handle;
137
+
138
+ return 0;
139
+ }
140
+
141
+
142
+ void uv_udp_close(uv_loop_t* loop, uv_udp_t* handle) {
143
+ uv_udp_recv_stop(handle);
144
+ closesocket(handle->socket);
145
+
146
+ uv__handle_start(handle);
147
+
148
+ if (handle->reqs_pending == 0) {
149
+ uv_want_endgame(loop, (uv_handle_t*) handle);
150
+ }
151
+ }
152
+
153
+
154
+ void uv_udp_endgame(uv_loop_t* loop, uv_udp_t* handle) {
155
+ if (handle->flags & UV_HANDLE_CLOSING &&
156
+ handle->reqs_pending == 0) {
157
+ assert(!(handle->flags & UV_HANDLE_CLOSED));
158
+ uv__handle_stop(handle);
159
+ uv__handle_close(handle);
160
+ }
161
+ }
162
+
163
+
164
+ static int uv__bind(uv_udp_t* handle,
165
+ int domain,
166
+ struct sockaddr* addr,
167
+ int addrsize,
168
+ unsigned int flags) {
169
+ int r;
170
+ DWORD no = 0, yes = 1;
171
+
172
+ if ((flags & UV_UDP_IPV6ONLY) && domain != AF_INET6) {
173
+ /* UV_UDP_IPV6ONLY is supported only for IPV6 sockets */
174
+ uv__set_artificial_error(handle->loop, UV_EINVAL);
175
+ return -1;
176
+ }
177
+
178
+ if (handle->socket == INVALID_SOCKET) {
179
+ SOCKET sock = socket(domain, SOCK_DGRAM, 0);
180
+ if (sock == INVALID_SOCKET) {
181
+ uv__set_sys_error(handle->loop, WSAGetLastError());
182
+ return -1;
183
+ }
184
+
185
+ if (uv_udp_set_socket(handle->loop, handle, sock) == -1) {
186
+ closesocket(sock);
187
+ return -1;
188
+ }
189
+ }
190
+
191
+ if (domain == AF_INET6 && !(flags & UV_UDP_IPV6ONLY)) {
192
+ /* On windows IPV6ONLY is on by default. */
193
+ /* If the user doesn't specify it libuv turns it off. */
194
+
195
+ /* TODO: how to handle errors? This may fail if there is no ipv4 stack */
196
+ /* available, or when run on XP/2003 which have no support for dualstack */
197
+ /* sockets. For now we're silently ignoring the error. */
198
+ setsockopt(handle->socket,
199
+ IPPROTO_IPV6,
200
+ IPV6_V6ONLY,
201
+ (char*) &no,
202
+ sizeof no);
203
+ }
204
+
205
+ r = setsockopt(handle->socket,
206
+ SOL_SOCKET,
207
+ SO_REUSEADDR,
208
+ (char*) &yes,
209
+ sizeof yes);
210
+ if (r == SOCKET_ERROR) {
211
+ uv__set_sys_error(handle->loop, WSAGetLastError());
212
+ return -1;
213
+ }
214
+
215
+ r = bind(handle->socket, addr, addrsize);
216
+ if (r == SOCKET_ERROR) {
217
+ uv__set_sys_error(handle->loop, WSAGetLastError());
218
+ return -1;
219
+ }
220
+
221
+ handle->flags |= UV_HANDLE_BOUND;
222
+
223
+ return 0;
224
+ }
225
+
226
+
227
+ int uv__udp_bind(uv_udp_t* handle, struct sockaddr_in addr,
228
+ unsigned int flags) {
229
+ return uv__bind(handle,
230
+ AF_INET,
231
+ (struct sockaddr*) &addr,
232
+ sizeof(struct sockaddr_in),
233
+ flags);
234
+ }
235
+
236
+
237
+ int uv__udp_bind6(uv_udp_t* handle, struct sockaddr_in6 addr,
238
+ unsigned int flags) {
239
+ if (uv_allow_ipv6) {
240
+ handle->flags |= UV_HANDLE_IPV6;
241
+ return uv__bind(handle,
242
+ AF_INET6,
243
+ (struct sockaddr*) &addr,
244
+ sizeof(struct sockaddr_in6),
245
+ flags);
246
+ } else {
247
+ uv__set_sys_error(handle->loop, WSAEAFNOSUPPORT);
248
+ return -1;
249
+ }
250
+ }
251
+
252
+
253
+ static void uv_udp_queue_recv(uv_loop_t* loop, uv_udp_t* handle) {
254
+ uv_req_t* req;
255
+ uv_buf_t buf;
256
+ DWORD bytes, flags;
257
+ int result;
258
+
259
+ assert(handle->flags & UV_HANDLE_READING);
260
+ assert(!(handle->flags & UV_HANDLE_READ_PENDING));
261
+
262
+ req = &handle->recv_req;
263
+ memset(&req->overlapped, 0, sizeof(req->overlapped));
264
+
265
+ /*
266
+ * Preallocate a read buffer if the number of active streams is below
267
+ * the threshold.
268
+ */
269
+ if (loop->active_udp_streams < uv_active_udp_streams_threshold) {
270
+ handle->flags &= ~UV_HANDLE_ZERO_READ;
271
+
272
+ handle->recv_buffer = handle->alloc_cb((uv_handle_t*) handle, 65536);
273
+ assert(handle->recv_buffer.len > 0);
274
+
275
+ buf = handle->recv_buffer;
276
+ memset(&handle->recv_from, 0, sizeof handle->recv_from);
277
+ handle->recv_from_len = sizeof handle->recv_from;
278
+ flags = 0;
279
+
280
+ result = handle->func_wsarecvfrom(handle->socket,
281
+ (WSABUF*) &buf,
282
+ 1,
283
+ &bytes,
284
+ &flags,
285
+ (struct sockaddr*) &handle->recv_from,
286
+ &handle->recv_from_len,
287
+ &req->overlapped,
288
+ NULL);
289
+
290
+ if (UV_SUCCEEDED_WITHOUT_IOCP(result == 0)) {
291
+ /* Process the req without IOCP. */
292
+ handle->flags |= UV_HANDLE_READ_PENDING;
293
+ req->overlapped.InternalHigh = bytes;
294
+ handle->reqs_pending++;
295
+ uv_insert_pending_req(loop, req);
296
+ } else if (UV_SUCCEEDED_WITH_IOCP(result == 0)) {
297
+ /* The req will be processed with IOCP. */
298
+ handle->flags |= UV_HANDLE_READ_PENDING;
299
+ handle->reqs_pending++;
300
+ } else {
301
+ /* Make this req pending reporting an error. */
302
+ SET_REQ_ERROR(req, WSAGetLastError());
303
+ uv_insert_pending_req(loop, req);
304
+ handle->reqs_pending++;
305
+ }
306
+
307
+ } else {
308
+ handle->flags |= UV_HANDLE_ZERO_READ;
309
+
310
+ buf.base = (char*) uv_zero_;
311
+ buf.len = 0;
312
+ flags = MSG_PEEK;
313
+
314
+ result = handle->func_wsarecv(handle->socket,
315
+ (WSABUF*) &buf,
316
+ 1,
317
+ &bytes,
318
+ &flags,
319
+ &req->overlapped,
320
+ NULL);
321
+
322
+ if (UV_SUCCEEDED_WITHOUT_IOCP(result == 0)) {
323
+ /* Process the req without IOCP. */
324
+ handle->flags |= UV_HANDLE_READ_PENDING;
325
+ req->overlapped.InternalHigh = bytes;
326
+ handle->reqs_pending++;
327
+ uv_insert_pending_req(loop, req);
328
+ } else if (UV_SUCCEEDED_WITH_IOCP(result == 0)) {
329
+ /* The req will be processed with IOCP. */
330
+ handle->flags |= UV_HANDLE_READ_PENDING;
331
+ handle->reqs_pending++;
332
+ } else {
333
+ /* Make this req pending reporting an error. */
334
+ SET_REQ_ERROR(req, WSAGetLastError());
335
+ uv_insert_pending_req(loop, req);
336
+ handle->reqs_pending++;
337
+ }
338
+ }
339
+ }
340
+
341
+
342
+ int uv_udp_recv_start(uv_udp_t* handle, uv_alloc_cb alloc_cb,
343
+ uv_udp_recv_cb recv_cb) {
344
+ uv_loop_t* loop = handle->loop;
345
+
346
+ if (handle->flags & UV_HANDLE_READING) {
347
+ uv__set_sys_error(loop, WSAEALREADY);
348
+ return -1;
349
+ }
350
+
351
+ if (!(handle->flags & UV_HANDLE_BOUND) &&
352
+ uv_udp_bind(handle, uv_addr_ip4_any_, 0) < 0) {
353
+ return -1;
354
+ }
355
+
356
+ handle->flags |= UV_HANDLE_READING;
357
+ INCREASE_ACTIVE_COUNT(loop, handle);
358
+ loop->active_udp_streams++;
359
+
360
+ handle->recv_cb = recv_cb;
361
+ handle->alloc_cb = alloc_cb;
362
+
363
+ /* If reading was stopped and then started again, there could still be a */
364
+ /* recv request pending. */
365
+ if (!(handle->flags & UV_HANDLE_READ_PENDING))
366
+ uv_udp_queue_recv(loop, handle);
367
+
368
+ return 0;
369
+ }
370
+
371
+
372
+ int uv_udp_recv_stop(uv_udp_t* handle) {
373
+ if (handle->flags & UV_HANDLE_READING) {
374
+ handle->flags &= ~UV_HANDLE_READING;
375
+ handle->loop->active_udp_streams--;
376
+ DECREASE_ACTIVE_COUNT(loop, handle);
377
+ }
378
+
379
+ return 0;
380
+ }
381
+
382
+
383
+ static int uv__udp_send(uv_udp_send_t* req, uv_udp_t* handle, uv_buf_t bufs[],
384
+ int bufcnt, struct sockaddr* addr, int addr_len, uv_udp_send_cb cb) {
385
+ uv_loop_t* loop = handle->loop;
386
+ DWORD result, bytes;
387
+
388
+ uv_req_init(loop, (uv_req_t*) req);
389
+ req->type = UV_UDP_SEND;
390
+ req->handle = handle;
391
+ req->cb = cb;
392
+ memset(&req->overlapped, 0, sizeof(req->overlapped));
393
+
394
+ result = WSASendTo(handle->socket,
395
+ (WSABUF*)bufs,
396
+ bufcnt,
397
+ &bytes,
398
+ 0,
399
+ addr,
400
+ addr_len,
401
+ &req->overlapped,
402
+ NULL);
403
+
404
+ if (UV_SUCCEEDED_WITHOUT_IOCP(result == 0)) {
405
+ /* Request completed immediately. */
406
+ req->queued_bytes = 0;
407
+ handle->reqs_pending++;
408
+ REGISTER_HANDLE_REQ(loop, handle, req);
409
+ uv_insert_pending_req(loop, (uv_req_t*)req);
410
+ } else if (UV_SUCCEEDED_WITH_IOCP(result == 0)) {
411
+ /* Request queued by the kernel. */
412
+ req->queued_bytes = uv_count_bufs(bufs, bufcnt);
413
+ handle->reqs_pending++;
414
+ REGISTER_HANDLE_REQ(loop, handle, req);
415
+ } else {
416
+ /* Send failed due to an error. */
417
+ uv__set_sys_error(loop, WSAGetLastError());
418
+ return -1;
419
+ }
420
+
421
+ return 0;
422
+ }
423
+
424
+
425
+ int uv_udp_send(uv_udp_send_t* req, uv_udp_t* handle, uv_buf_t bufs[],
426
+ int bufcnt, struct sockaddr_in addr, uv_udp_send_cb cb) {
427
+
428
+ if (!(handle->flags & UV_HANDLE_BOUND) &&
429
+ uv_udp_bind(handle, uv_addr_ip4_any_, 0) < 0) {
430
+ return -1;
431
+ }
432
+
433
+ return uv__udp_send(req,
434
+ handle,
435
+ bufs,
436
+ bufcnt,
437
+ (struct sockaddr*) &addr,
438
+ sizeof addr,
439
+ cb);
440
+ }
441
+
442
+
443
+ int uv_udp_send6(uv_udp_send_t* req, uv_udp_t* handle, uv_buf_t bufs[],
444
+ int bufcnt, struct sockaddr_in6 addr, uv_udp_send_cb cb) {
445
+
446
+ if (!(handle->flags & UV_HANDLE_BOUND) &&
447
+ uv_udp_bind6(handle, uv_addr_ip6_any_, 0) < 0) {
448
+ return -1;
449
+ }
450
+
451
+ return uv__udp_send(req,
452
+ handle,
453
+ bufs,
454
+ bufcnt,
455
+ (struct sockaddr*) &addr,
456
+ sizeof addr,
457
+ cb);
458
+ }
459
+
460
+
461
+ void uv_process_udp_recv_req(uv_loop_t* loop, uv_udp_t* handle,
462
+ uv_req_t* req) {
463
+ uv_buf_t buf;
464
+ int partial;
465
+
466
+ assert(handle->type == UV_UDP);
467
+
468
+ handle->flags &= ~UV_HANDLE_READ_PENDING;
469
+
470
+ if (!REQ_SUCCESS(req)) {
471
+ DWORD err = GET_REQ_SOCK_ERROR(req);
472
+ if (err == WSAEMSGSIZE) {
473
+ /* Not a real error, it just indicates that the received packet */
474
+ /* was bigger than the receive buffer. */
475
+ } else if (err == WSAECONNRESET || err == WSAENETRESET) {
476
+ /* A previous sendto operation failed; ignore this error. If */
477
+ /* zero-reading we need to call WSARecv/WSARecvFrom _without_ the */
478
+ /* MSG_PEEK flag to clear out the error queue. For nonzero reads, */
479
+ /* immediately queue a new receive. */
480
+ if (!(handle->flags & UV_HANDLE_ZERO_READ)) {
481
+ goto done;
482
+ }
483
+ } else {
484
+ /* A real error occurred. Report the error to the user only if we're */
485
+ /* currently reading. */
486
+ if (handle->flags & UV_HANDLE_READING) {
487
+ uv__set_sys_error(loop, err);
488
+ uv_udp_recv_stop(handle);
489
+ buf = (handle->flags & UV_HANDLE_ZERO_READ) ?
490
+ uv_buf_init(NULL, 0) : handle->recv_buffer;
491
+ handle->recv_cb(handle, -1, buf, NULL, 0);
492
+ }
493
+ goto done;
494
+ }
495
+ }
496
+
497
+ if (!(handle->flags & UV_HANDLE_ZERO_READ)) {
498
+ /* Successful read */
499
+ partial = !REQ_SUCCESS(req);
500
+ handle->recv_cb(handle,
501
+ req->overlapped.InternalHigh,
502
+ handle->recv_buffer,
503
+ (struct sockaddr*) &handle->recv_from,
504
+ partial ? UV_UDP_PARTIAL : 0);
505
+ } else if (handle->flags & UV_HANDLE_READING) {
506
+ DWORD bytes, err, flags;
507
+ struct sockaddr_storage from;
508
+ int from_len;
509
+
510
+ /* Do a nonblocking receive */
511
+ /* TODO: try to read multiple datagrams at once. FIONREAD maybe? */
512
+ buf = handle->alloc_cb((uv_handle_t*) handle, 65536);
513
+ assert(buf.len > 0);
514
+
515
+ memset(&from, 0, sizeof from);
516
+ from_len = sizeof from;
517
+
518
+ flags = 0;
519
+
520
+ if (WSARecvFrom(handle->socket,
521
+ (WSABUF*)&buf,
522
+ 1,
523
+ &bytes,
524
+ &flags,
525
+ (struct sockaddr*) &from,
526
+ &from_len,
527
+ NULL,
528
+ NULL) != SOCKET_ERROR) {
529
+
530
+ /* Message received */
531
+ handle->recv_cb(handle, bytes, buf, (struct sockaddr*) &from, 0);
532
+ } else {
533
+ err = WSAGetLastError();
534
+ if (err == WSAEMSGSIZE) {
535
+ /* Message truncated */
536
+ handle->recv_cb(handle,
537
+ bytes,
538
+ buf,
539
+ (struct sockaddr*) &from,
540
+ UV_UDP_PARTIAL);
541
+ } if (err == WSAEWOULDBLOCK) {
542
+ /* Kernel buffer empty */
543
+ uv__set_sys_error(loop, WSAEWOULDBLOCK);
544
+ handle->recv_cb(handle, 0, buf, NULL, 0);
545
+ } else if (err != WSAECONNRESET && err != WSAENETRESET) {
546
+ /* Serious error. WSAECONNRESET/WSANETRESET is ignored because this */
547
+ /* just indicates that a previous sendto operation failed. */
548
+ uv_udp_recv_stop(handle);
549
+ uv__set_sys_error(loop, err);
550
+ handle->recv_cb(handle, -1, buf, NULL, 0);
551
+ }
552
+ }
553
+ }
554
+
555
+ done:
556
+ /* Post another read if still reading and not closing. */
557
+ if ((handle->flags & UV_HANDLE_READING) &&
558
+ !(handle->flags & UV_HANDLE_READ_PENDING)) {
559
+ uv_udp_queue_recv(loop, handle);
560
+ }
561
+
562
+ DECREASE_PENDING_REQ_COUNT(handle);
563
+ }
564
+
565
+
566
+ void uv_process_udp_send_req(uv_loop_t* loop, uv_udp_t* handle,
567
+ uv_udp_send_t* req) {
568
+ assert(handle->type == UV_UDP);
569
+
570
+ UNREGISTER_HANDLE_REQ(loop, handle, req);
571
+
572
+ if (req->cb) {
573
+ if (REQ_SUCCESS(req)) {
574
+ req->cb(req, 0);
575
+ } else {
576
+ uv__set_sys_error(loop, GET_REQ_SOCK_ERROR(req));
577
+ req->cb(req, -1);
578
+ }
579
+ }
580
+
581
+ DECREASE_PENDING_REQ_COUNT(handle);
582
+ }
583
+
584
+
585
+ int uv_udp_set_membership(uv_udp_t* handle, const char* multicast_addr,
586
+ const char* interface_addr, uv_membership membership) {
587
+ int optname;
588
+ struct ip_mreq mreq;
589
+
590
+ /* If the socket is unbound, bind to inaddr_any. */
591
+ if (!(handle->flags & UV_HANDLE_BOUND) &&
592
+ uv_udp_bind(handle, uv_addr_ip4_any_, 0) < 0) {
593
+ return -1;
594
+ }
595
+
596
+ if (handle->flags & UV_HANDLE_IPV6) {
597
+ uv__set_artificial_error(handle->loop, UV_ENOSYS);
598
+ return -1;
599
+ }
600
+
601
+ memset(&mreq, 0, sizeof mreq);
602
+
603
+ if (interface_addr) {
604
+ mreq.imr_interface.s_addr = inet_addr(interface_addr);
605
+ } else {
606
+ mreq.imr_interface.s_addr = htonl(INADDR_ANY);
607
+ }
608
+
609
+ mreq.imr_multiaddr.s_addr = inet_addr(multicast_addr);
610
+
611
+ switch (membership) {
612
+ case UV_JOIN_GROUP:
613
+ optname = IP_ADD_MEMBERSHIP;
614
+ break;
615
+ case UV_LEAVE_GROUP:
616
+ optname = IP_DROP_MEMBERSHIP;
617
+ break;
618
+ default:
619
+ uv__set_artificial_error(handle->loop, UV_EFAULT);
620
+ return -1;
621
+ }
622
+
623
+ if (setsockopt(handle->socket,
624
+ IPPROTO_IP,
625
+ optname,
626
+ (char*) &mreq,
627
+ sizeof mreq) == SOCKET_ERROR) {
628
+ uv__set_sys_error(handle->loop, WSAGetLastError());
629
+ return -1;
630
+ }
631
+
632
+ return 0;
633
+ }
634
+
635
+
636
+ int uv_udp_set_broadcast(uv_udp_t* handle, int value) {
637
+ BOOL optval = (BOOL) value;
638
+
639
+ /* If the socket is unbound, bind to inaddr_any. */
640
+ if (!(handle->flags & UV_HANDLE_BOUND) &&
641
+ uv_udp_bind(handle, uv_addr_ip4_any_, 0) < 0) {
642
+ return -1;
643
+ }
644
+
645
+ if (setsockopt(handle->socket,
646
+ SOL_SOCKET,
647
+ SO_BROADCAST,
648
+ (char*) &optval,
649
+ sizeof optval)) {
650
+ uv__set_sys_error(handle->loop, WSAGetLastError());
651
+ return -1;
652
+ }
653
+ return 0;
654
+ }
655
+
656
+
657
+ #define SOCKOPT_SETTER(name, option4, option6, validate) \
658
+ int uv_udp_set_##name(uv_udp_t* handle, int value) { \
659
+ DWORD optval = (DWORD) value; \
660
+ \
661
+ if (!(validate(value))) { \
662
+ uv__set_artificial_error(handle->loop, UV_EINVAL); \
663
+ return -1; \
664
+ } \
665
+ \
666
+ /* If the socket is unbound, bind to inaddr_any. */ \
667
+ if (!(handle->flags & UV_HANDLE_BOUND) && \
668
+ uv_udp_bind(handle, uv_addr_ip4_any_, 0) < 0) { \
669
+ return -1; \
670
+ } \
671
+ \
672
+ if (!(handle->flags & UV_HANDLE_IPV6)) { \
673
+ /* Set IPv4 socket option */ \
674
+ if (setsockopt(handle->socket, \
675
+ IPPROTO_IP, \
676
+ option4, \
677
+ (char*) &optval, \
678
+ sizeof optval)) { \
679
+ uv__set_sys_error(handle->loop, WSAGetLastError()); \
680
+ return -1; \
681
+ } \
682
+ } else { \
683
+ /* Set IPv6 socket option */ \
684
+ if (setsockopt(handle->socket, \
685
+ IPPROTO_IPV6, \
686
+ option6, \
687
+ (char*) &optval, \
688
+ sizeof optval)) { \
689
+ uv__set_sys_error(handle->loop, WSAGetLastError()); \
690
+ return -1; \
691
+ } \
692
+ } \
693
+ return 0; \
694
+ }
695
+
696
+ #define VALIDATE_TTL(value) ((value) >= 1 && (value) <= 255)
697
+ #define VALIDATE_MULTICAST_TTL(value) ((value) >= -1 && (value) <= 255)
698
+ #define VALIDATE_MULTICAST_LOOP(value) (1)
699
+
700
+ SOCKOPT_SETTER(ttl,
701
+ IP_TTL,
702
+ IPV6_HOPLIMIT,
703
+ VALIDATE_TTL)
704
+ SOCKOPT_SETTER(multicast_ttl,
705
+ IP_MULTICAST_TTL,
706
+ IPV6_MULTICAST_HOPS,
707
+ VALIDATE_MULTICAST_TTL)
708
+ SOCKOPT_SETTER(multicast_loop,
709
+ IP_MULTICAST_LOOP,
710
+ IPV6_MULTICAST_LOOP,
711
+ VALIDATE_MULTICAST_LOOP)
712
+
713
+ #undef SOCKOPT_SETTER
714
+ #undef VALIDATE_TTL
715
+ #undef VALIDATE_MULTICAST_TTL
716
+ #undef VALIDATE_MULTICAST_LOOP