nginx/src/os/win32/ngx_wsasend_chain.c

419 lines
9.7 KiB
C
Raw Normal View History

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