Mercurial > hg > nginx
annotate src/os/win32/ngx_wsasend.c @ 4450:4919fb357a5d stable-1.0
Merge of r4406, r4413: copyrights updated.
| author | Maxim Dounin <mdounin@mdounin.ru> |
|---|---|
| date | Sun, 05 Feb 2012 20:02:59 +0000 |
| parents | d43d73277c5c |
| children | 53eb1e67e432 |
| rev | line source |
|---|---|
| 2725 | 1 |
| 2 /* | |
| 3 * Copyright (C) Igor Sysoev | |
|
4450
4919fb357a5d
Merge of r4406, r4413: copyrights updated.
Maxim Dounin <mdounin@mdounin.ru>
parents:
2725
diff
changeset
|
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 } |