2725
|
1
|
|
2 /*
|
|
3 * Copyright (C) Igor Sysoev
|
4412
|
4 * Copyright (C) Nginx, Inc.
|
2725
|
5 */
|
|
6
|
|
7
|
|
8 #include <ngx_config.h>
|
|
9 #include <ngx_core.h>
|
|
10 #include <ngx_event.h>
|
|
11
|
|
12
|
|
13 ssize_t
|
|
14 ngx_wsasend(ngx_connection_t *c, u_char *buf, size_t size)
|
|
15 {
|
|
16 int n;
|
|
17 u_long sent;
|
|
18 ngx_err_t err;
|
|
19 ngx_event_t *wev;
|
|
20 WSABUF wsabuf;
|
|
21
|
|
22 wev = c->write;
|
|
23
|
|
24 if (!wev->ready) {
|
|
25 return NGX_AGAIN;
|
|
26 }
|
|
27
|
|
28 /*
|
|
29 * WSABUF must be 4-byte aligned otherwise
|
|
30 * WSASend() will return undocumented WSAEINVAL error.
|
|
31 */
|
|
32
|
|
33 wsabuf.buf = (char *) buf;
|
|
34 wsabuf.len = size;
|
|
35
|
|
36 sent = 0;
|
|
37
|
|
38 n = WSASend(c->fd, &wsabuf, 1, &sent, 0, NULL, NULL);
|
|
39
|
|
40 ngx_log_debug4(NGX_LOG_DEBUG_EVENT, c->log, 0,
|
|
41 "WSASend: fd:%d, %d, %ul of %uz", c->fd, n, sent, size);
|
|
42
|
|
43 if (n == 0) {
|
|
44 if (sent < size) {
|
|
45 wev->ready = 0;
|
|
46 }
|
|
47
|
|
48 c->sent += sent;
|
|
49
|
|
50 return sent;
|
|
51 }
|
|
52
|
|
53 err = ngx_socket_errno;
|
|
54
|
|
55 if (err == WSAEWOULDBLOCK) {
|
|
56 ngx_log_debug0(NGX_LOG_DEBUG_EVENT, c->log, err, "WSASend() not ready");
|
|
57 return NGX_AGAIN;
|
|
58 }
|
|
59
|
|
60 wev->error = 1;
|
|
61 ngx_connection_error(c, err, "WSASend() failed");
|
|
62
|
|
63 return NGX_ERROR;
|
|
64 }
|
|
65
|
|
66
|
|
67 ssize_t
|
|
68 ngx_overlapped_wsasend(ngx_connection_t *c, u_char *buf, size_t size)
|
|
69 {
|
|
70 int n;
|
|
71 u_long sent;
|
|
72 ngx_err_t err;
|
|
73 ngx_event_t *wev;
|
|
74 LPWSAOVERLAPPED ovlp;
|
|
75 WSABUF wsabuf;
|
|
76
|
|
77 wev = c->write;
|
|
78
|
|
79 if (!wev->ready) {
|
|
80 return NGX_AGAIN;
|
|
81 }
|
|
82
|
|
83 ngx_log_debug1(NGX_LOG_DEBUG_EVENT, c->log, 0,
|
|
84 "wev->complete: %d", wev->complete);
|
|
85
|
|
86 if (!wev->complete) {
|
|
87
|
|
88 /* post the overlapped WSASend() */
|
|
89
|
|
90 /*
|
|
91 * WSABUFs must be 4-byte aligned otherwise
|
|
92 * WSASend() will return undocumented WSAEINVAL error.
|
|
93 */
|
|
94
|
|
95 wsabuf.buf = (char *) buf;
|
|
96 wsabuf.len = size;
|
|
97
|
|
98 sent = 0;
|
|
99
|
|
100 ovlp = (LPWSAOVERLAPPED) &c->write->ovlp;
|
|
101 ngx_memzero(ovlp, sizeof(WSAOVERLAPPED));
|
|
102
|
|
103 n = WSASend(c->fd, &wsabuf, 1, &sent, 0, ovlp, NULL);
|
|
104
|
|
105 ngx_log_debug4(NGX_LOG_DEBUG_EVENT, c->log, 0,
|
|
106 "WSASend: fd:%d, %d, %ul of %uz", c->fd, n, sent, size);
|
|
107
|
|
108 wev->complete = 0;
|
|
109
|
|
110 if (n == 0) {
|
|
111 if (ngx_event_flags & NGX_USE_IOCP_EVENT) {
|
|
112
|
|
113 /*
|
|
114 * if a socket was bound with I/O completion port then
|
|
115 * GetQueuedCompletionStatus() would anyway return its status
|
|
116 * despite that WSASend() was already complete
|
|
117 */
|
|
118
|
|
119 wev->active = 1;
|
|
120 return NGX_AGAIN;
|
|
121 }
|
|
122
|
|
123 if (sent < size) {
|
|
124 wev->ready = 0;
|
|
125 }
|
|
126
|
|
127 c->sent += sent;
|
|
128
|
|
129 return sent;
|
|
130 }
|
|
131
|
|
132 err = ngx_socket_errno;
|
|
133
|
|
134 if (err == WSA_IO_PENDING) {
|
|
135 ngx_log_debug0(NGX_LOG_DEBUG_EVENT, c->log, err,
|
|
136 "WSASend() posted");
|
|
137 wev->active = 1;
|
|
138 return NGX_AGAIN;
|
|
139 }
|
|
140
|
|
141 wev->error = 1;
|
|
142 ngx_connection_error(c, err, "WSASend() failed");
|
|
143
|
|
144 return NGX_ERROR;
|
|
145 }
|
|
146
|
|
147 /* the overlapped WSASend() complete */
|
|
148
|
|
149 wev->complete = 0;
|
|
150 wev->active = 0;
|
|
151
|
|
152 if (ngx_event_flags & NGX_USE_IOCP_EVENT) {
|
|
153
|
|
154 if (wev->ovlp.error) {
|
|
155 ngx_connection_error(c, wev->ovlp.error, "WSASend() failed");
|
|
156 return NGX_ERROR;
|
|
157 }
|
|
158
|
|
159 sent = wev->available;
|
|
160
|
|
161 } else {
|
|
162 if (WSAGetOverlappedResult(c->fd, (LPWSAOVERLAPPED) &wev->ovlp,
|
|
163 &sent, 0, NULL)
|
|
164 == 0)
|
|
165 {
|
|
166 ngx_connection_error(c, ngx_socket_errno,
|
|
167 "WSASend() or WSAGetOverlappedResult() failed");
|
|
168
|
|
169 return NGX_ERROR;
|
|
170 }
|
|
171 }
|
|
172
|
|
173 ngx_log_debug3(NGX_LOG_DEBUG_EVENT, c->log, 0,
|
|
174 "WSAGetOverlappedResult: fd:%d, %ul of %uz",
|
|
175 c->fd, sent, size);
|
|
176
|
|
177 if (sent < size) {
|
|
178 wev->ready = 0;
|
|
179 }
|
|
180
|
|
181 c->sent += sent;
|
|
182
|
|
183 return sent;
|
|
184 }
|