2003-06-11 23:28:34 +08:00
|
|
|
|
|
|
|
#include <ngx_config.h>
|
|
|
|
#include <ngx_core.h>
|
|
|
|
#include <ngx_event.h>
|
|
|
|
|
|
|
|
|
2003-11-14 15:20:34 +08:00
|
|
|
ngx_chain_t *ngx_wsasend_chain(ngx_connection_t *c, ngx_chain_t *in)
|
|
|
|
{
|
|
|
|
int rc;
|
|
|
|
char *prev;
|
|
|
|
size_t size, sent;
|
|
|
|
LPWSABUF wsabuf;
|
|
|
|
ngx_err_t err;
|
|
|
|
ngx_event_t *wev;
|
|
|
|
ngx_array_t wsabufs;
|
|
|
|
ngx_chain_t *cl;
|
|
|
|
|
|
|
|
wev = c->write;
|
|
|
|
|
|
|
|
if (!wev->ready) {
|
|
|
|
return in;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* WSABUFs must be 4-byte aligned otherwise
|
|
|
|
* WSASend() will return undocumented WSAEINVAL error.
|
|
|
|
*/
|
|
|
|
|
|
|
|
ngx_init_array(wsabufs, c->pool, 10, sizeof(WSABUF), NGX_CHAIN_ERROR);
|
|
|
|
|
|
|
|
prev = NULL;
|
|
|
|
wsabuf = NULL;
|
|
|
|
|
|
|
|
/* create the WSABUF and coalesce the neighbouring bufs */
|
|
|
|
|
|
|
|
for (cl = in; cl; cl = cl->next) {
|
|
|
|
|
|
|
|
if (prev == cl->hunk->pos) {
|
|
|
|
wsabuf->len += cl->hunk->last - cl->hunk->pos;
|
|
|
|
prev = cl->hunk->last;
|
|
|
|
|
|
|
|
} else {
|
|
|
|
ngx_test_null(wsabuf, ngx_push_array(&wsabufs), NGX_CHAIN_ERROR);
|
|
|
|
wsabuf->buf = cl->hunk->pos;
|
|
|
|
wsabuf->len = cl->hunk->last - cl->hunk->pos;
|
|
|
|
prev = cl->hunk->last;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
rc = WSASend(c->fd, wsabufs.elts, wsabufs.nelts, &sent, 0, NULL, NULL);
|
|
|
|
|
|
|
|
if (rc == -1) {
|
|
|
|
err = ngx_errno;
|
|
|
|
|
|
|
|
if (err == WSAEWOULDBLOCK) {
|
|
|
|
ngx_log_error(NGX_LOG_INFO, c->log, err, "WSASend() EAGAIN");
|
|
|
|
wev->ready = 0;
|
|
|
|
return in;
|
|
|
|
|
|
|
|
} else {
|
|
|
|
wev->error = 1;
|
|
|
|
ngx_log_error(NGX_LOG_CRIT, c->log, err, "WSASend() failed");
|
|
|
|
return NGX_CHAIN_ERROR;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2004-02-12 01:08:49 +08:00
|
|
|
ngx_log_debug1(NGX_LOG_DEBUG_EVENT, c->log, 0, "WSASend(): %d", sent);
|
2003-11-14 15:20:34 +08:00
|
|
|
|
|
|
|
c->sent += sent;
|
|
|
|
|
|
|
|
for (cl = in; cl && sent > 0; cl = cl->next) {
|
|
|
|
|
|
|
|
size = cl->hunk->last - cl->hunk->pos;
|
|
|
|
|
|
|
|
if (sent >= size) {
|
|
|
|
sent -= size;
|
|
|
|
|
|
|
|
if (cl->hunk->type & NGX_HUNK_IN_MEMORY) {
|
|
|
|
cl->hunk->pos = cl->hunk->last;
|
|
|
|
}
|
|
|
|
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (cl->hunk->type & NGX_HUNK_IN_MEMORY) {
|
|
|
|
cl->hunk->pos += sent;
|
|
|
|
}
|
|
|
|
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (cl) {
|
|
|
|
wev->ready = 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
return cl;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
ngx_chain_t *ngx_overlapped_wsasend_chain(ngx_connection_t *c, ngx_chain_t *in)
|
|
|
|
{
|
|
|
|
int rc;
|
|
|
|
char *prev;
|
|
|
|
size_t size, sent;
|
|
|
|
LPWSABUF wsabuf;
|
|
|
|
ngx_err_t err;
|
|
|
|
ngx_event_t *wev;
|
|
|
|
ngx_array_t wsabufs;
|
|
|
|
ngx_chain_t *cl;
|
|
|
|
LPWSAOVERLAPPED ovlp;
|
|
|
|
|
|
|
|
wev = c->write;
|
|
|
|
|
|
|
|
if (!wev->ready) {
|
|
|
|
return in;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!wev->complete) {
|
|
|
|
|
|
|
|
/* post the overlapped WSASend() */
|
|
|
|
|
|
|
|
/*
|
|
|
|
* WSABUFs must be 4-byte aligned otherwise
|
|
|
|
* WSASend() will return undocumented WSAEINVAL error.
|
|
|
|
*/
|
|
|
|
|
|
|
|
ngx_init_array(wsabufs, c->pool, 10, sizeof(WSABUF), NGX_CHAIN_ERROR);
|
|
|
|
|
|
|
|
prev = NULL;
|
|
|
|
wsabuf = NULL;
|
|
|
|
|
|
|
|
/* create the WSABUF and coalesce the neighbouring bufs */
|
|
|
|
|
|
|
|
for (cl = in; cl; cl = cl->next) {
|
|
|
|
|
|
|
|
if (prev == cl->hunk->pos) {
|
|
|
|
wsabuf->len += cl->hunk->last - cl->hunk->pos;
|
|
|
|
prev = cl->hunk->last;
|
|
|
|
|
|
|
|
} else {
|
|
|
|
ngx_test_null(wsabuf, ngx_push_array(&wsabufs),
|
|
|
|
NGX_CHAIN_ERROR);
|
|
|
|
wsabuf->buf = cl->hunk->pos;
|
|
|
|
wsabuf->len = cl->hunk->last - cl->hunk->pos;
|
|
|
|
prev = cl->hunk->last;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
ovlp = (LPWSAOVERLAPPED) &c->write->ovlp;
|
|
|
|
ngx_memzero(ovlp, sizeof(WSAOVERLAPPED));
|
|
|
|
|
|
|
|
rc = WSASend(c->fd, wsabufs.elts, wsabufs.nelts, &sent, 0, ovlp, NULL);
|
|
|
|
|
|
|
|
wev->complete = 0;
|
|
|
|
|
|
|
|
if (rc == -1) {
|
|
|
|
err = ngx_errno;
|
|
|
|
|
|
|
|
if (err == WSA_IO_PENDING) {
|
|
|
|
wev->active = 1;
|
|
|
|
return in;
|
|
|
|
|
|
|
|
} else {
|
|
|
|
wev->error = 1;
|
|
|
|
ngx_log_error(NGX_LOG_CRIT, c->log, err, "WSASend() failed");
|
|
|
|
return NGX_CHAIN_ERROR;
|
|
|
|
}
|
|
|
|
|
|
|
|
} else if (ngx_event_flags & NGX_USE_IOCP_EVENT) {
|
|
|
|
|
|
|
|
/*
|
|
|
|
* if a socket was bound with I/O completion port then
|
|
|
|
* GetQueuedCompletionStatus() would anyway return its status
|
|
|
|
* despite that WSASend() was already complete
|
|
|
|
*/
|
|
|
|
|
|
|
|
wev->active = 1;
|
|
|
|
return in;
|
|
|
|
}
|
|
|
|
|
|
|
|
} else {
|
|
|
|
|
|
|
|
/* the overlapped WSASend() complete */
|
|
|
|
|
|
|
|
wev->complete = 0;
|
|
|
|
wev->active = 0;
|
|
|
|
|
|
|
|
if (ngx_event_flags & NGX_USE_IOCP_EVENT) {
|
|
|
|
if (wev->ovlp.error) {
|
|
|
|
ngx_log_error(NGX_LOG_ERR, c->log, wev->ovlp.error,
|
|
|
|
"WSASend() failed");
|
|
|
|
return NGX_CHAIN_ERROR;
|
|
|
|
}
|
|
|
|
|
|
|
|
sent = wev->available;
|
|
|
|
|
|
|
|
} else {
|
|
|
|
if (WSAGetOverlappedResult(c->fd, (LPWSAOVERLAPPED) &wev->ovlp,
|
|
|
|
&sent, 0, NULL) == 0) {
|
|
|
|
ngx_log_error(NGX_LOG_CRIT, c->log, ngx_socket_errno,
|
|
|
|
"WSASend() or WSAGetOverlappedResult() failed");
|
|
|
|
|
|
|
|
return NGX_CHAIN_ERROR;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2004-02-12 01:08:49 +08:00
|
|
|
ngx_log_debug1(NGX_LOG_DEBUG_EVENT, c->log, 0, "WSASend(): %d", sent);
|
2003-11-14 15:20:34 +08:00
|
|
|
|
|
|
|
c->sent += sent;
|
|
|
|
|
|
|
|
for (cl = in; cl && sent > 0; cl = cl->next) {
|
|
|
|
|
|
|
|
size = cl->hunk->last - cl->hunk->pos;
|
|
|
|
|
|
|
|
if (sent >= size) {
|
|
|
|
sent -= size;
|
|
|
|
|
|
|
|
if (cl->hunk->type & NGX_HUNK_IN_MEMORY) {
|
|
|
|
cl->hunk->pos = cl->hunk->last;
|
|
|
|
}
|
|
|
|
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (cl->hunk->type & NGX_HUNK_IN_MEMORY) {
|
|
|
|
cl->hunk->pos += sent;
|
|
|
|
}
|
|
|
|
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (cl) {
|
|
|
|
wev->ready = 0;
|
|
|
|
|
|
|
|
} else {
|
|
|
|
wev->ready = 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
return cl;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
#if 0
|
|
|
|
|
2003-06-11 23:28:34 +08:00
|
|
|
ngx_chain_t *ngx_wsasend_chain(ngx_connection_t *c, ngx_chain_t *in)
|
|
|
|
{
|
|
|
|
int rc;
|
|
|
|
char *prev;
|
|
|
|
size_t size, sent;
|
|
|
|
LPWSABUF wsabuf;
|
|
|
|
ngx_err_t err;
|
|
|
|
ngx_event_t *wev;
|
|
|
|
ngx_array_t wsabufs;
|
|
|
|
ngx_chain_t *ce;
|
|
|
|
LPWSAOVERLAPPED ovlp;
|
|
|
|
|
|
|
|
#if 0
|
|
|
|
|
|
|
|
iocp:
|
|
|
|
if ready
|
|
|
|
get result
|
|
|
|
update chain
|
|
|
|
return if done;
|
|
|
|
wsasend
|
|
|
|
|
|
|
|
non-block
|
|
|
|
for ( ;; ) {
|
|
|
|
wsasend
|
|
|
|
if no again
|
|
|
|
update chain
|
|
|
|
return if done;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
for ( ;; ) {
|
|
|
|
|
|
|
|
make buffers and limit data for both ovlp and nonblocked,
|
|
|
|
configured in events module
|
|
|
|
|
|
|
|
if (iocp && ready) {
|
|
|
|
get result
|
|
|
|
|
|
|
|
} else {
|
|
|
|
if (file)
|
|
|
|
transmitfile
|
|
|
|
else
|
|
|
|
wsasend
|
|
|
|
|
|
|
|
if (iocp)
|
|
|
|
return chain
|
|
|
|
return chain if again
|
|
|
|
here is result
|
|
|
|
}
|
|
|
|
|
|
|
|
if (result)
|
|
|
|
update chain;
|
|
|
|
return chain if done
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
#endif
|
|
|
|
|
|
|
|
wev = c->write;
|
|
|
|
|
2003-10-13 00:49:16 +08:00
|
|
|
if (((ngx_event_flags & NGX_USE_AIO_EVENT) && !wev->ready)
|
|
|
|
|| ((ngx_event_flags & NGX_USE_AIO_EVENT) == 0))
|
2003-06-11 23:28:34 +08:00
|
|
|
{
|
|
|
|
/*
|
|
|
|
* WSABUFs must be 4-byte aligned otherwise
|
|
|
|
* WSASend() will return undocumented WSAEINVAL error.
|
|
|
|
*/
|
|
|
|
|
|
|
|
ngx_init_array(wsabufs, c->pool, 10, sizeof(WSABUF), NGX_CHAIN_ERROR);
|
|
|
|
|
|
|
|
prev = NULL;
|
|
|
|
wsabuf = NULL;
|
|
|
|
|
|
|
|
/* create the WSABUF and coalesce the neighbouring chain entries */
|
|
|
|
for (ce = in; ce; ce = ce->next) {
|
|
|
|
|
|
|
|
if (prev == ce->hunk->pos) {
|
|
|
|
wsabuf->len += ce->hunk->last - ce->hunk->pos;
|
|
|
|
prev = ce->hunk->last;
|
|
|
|
|
|
|
|
} else {
|
|
|
|
ngx_test_null(wsabuf, ngx_push_array(&wsabufs),
|
|
|
|
NGX_CHAIN_ERROR);
|
|
|
|
wsabuf->buf = ce->hunk->pos;
|
|
|
|
wsabuf->len = ce->hunk->last - ce->hunk->pos;
|
|
|
|
prev = ce->hunk->last;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2003-10-13 00:49:16 +08:00
|
|
|
if (ngx_event_flags & NGX_USE_AIO_EVENT) {
|
2003-06-11 23:28:34 +08:00
|
|
|
ovlp = (LPWSAOVERLAPPED) &c->write->ovlp;
|
|
|
|
ngx_memzero(ovlp, sizeof(WSAOVERLAPPED));
|
|
|
|
|
|
|
|
} else {
|
|
|
|
ovlp = NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
rc = WSASend(c->fd, wsabufs.elts, wsabufs.nelts, &sent, 0, ovlp, NULL);
|
|
|
|
|
|
|
|
if (rc == -1) {
|
|
|
|
err = ngx_errno;
|
|
|
|
if (err == WSA_IO_PENDING) {
|
|
|
|
sent = 0;
|
|
|
|
|
|
|
|
} else if (err == WSAEWOULDBLOCK) {
|
|
|
|
sent = 0;
|
|
|
|
ngx_log_error(NGX_LOG_INFO, c->log, err, "WSASend() EAGAIN");
|
|
|
|
|
|
|
|
} else {
|
|
|
|
ngx_log_error(NGX_LOG_CRIT, c->log, err, "WSASend() failed");
|
|
|
|
return NGX_CHAIN_ERROR;
|
|
|
|
}
|
|
|
|
|
|
|
|
} else {
|
|
|
|
|
2003-10-13 00:49:16 +08:00
|
|
|
if (ngx_event_flags & NGX_USE_IOCP_EVENT) {
|
2003-06-11 23:28:34 +08:00
|
|
|
|
|
|
|
/*
|
|
|
|
* If a socket was bound with I/O completion port then
|
|
|
|
* GetQueuedCompletionStatus() would anyway return its status
|
|
|
|
* despite that WSASend() was already completed.
|
|
|
|
*/
|
|
|
|
|
|
|
|
sent = 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
} else {
|
2003-10-13 00:49:16 +08:00
|
|
|
if (ngx_event_flags & NGX_USE_IOCP_EVENT) {
|
2003-06-11 23:28:34 +08:00
|
|
|
wev->ready = 0;
|
|
|
|
|
|
|
|
/* the overlapped WSASend() completed */
|
|
|
|
|
|
|
|
if (wev->ovlp.error) {
|
|
|
|
ngx_log_error(NGX_LOG_ERR, c->log, wev->ovlp.error,
|
|
|
|
"WSASend() failed");
|
|
|
|
return NGX_CHAIN_ERROR;
|
|
|
|
}
|
|
|
|
|
|
|
|
sent = wev->available;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
ngx_log_debug(c->log, "WSASend(): %d" _ sent);
|
|
|
|
|
|
|
|
c->sent += sent;
|
|
|
|
|
|
|
|
for (ce = in; ce && sent > 0; ce = ce->next) {
|
|
|
|
|
|
|
|
size = ce->hunk->last - ce->hunk->pos;
|
|
|
|
|
|
|
|
if (sent >= size) {
|
|
|
|
sent -= size;
|
|
|
|
|
|
|
|
if (ce->hunk->type & NGX_HUNK_IN_MEMORY) {
|
|
|
|
ce->hunk->pos = ce->hunk->last;
|
|
|
|
}
|
|
|
|
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (ce->hunk->type & NGX_HUNK_IN_MEMORY) {
|
|
|
|
ce->hunk->pos += sent;
|
|
|
|
}
|
|
|
|
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
return ce;
|
|
|
|
}
|
2003-11-14 15:20:34 +08:00
|
|
|
|
|
|
|
#endif
|