One Level Up Top Level

src/os/win32/ngx_wsasend.c - nginx source code

Functions defined

Source code


  2. /*
  3. * Copyright (C) Igor Sysoev
  4. * Copyright (C) Nginx, Inc.
  5. */


  8. #include <ngx_config.h>
  9. #include <ngx_core.h>
  10. #include <ngx_event.h>


  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;

  22.     wev = c->write;

  24.     if (!wev->ready) {
  25.         return NGX_AGAIN;
  26.     }

  28.     /*
  29.      * WSABUF must be 4-byte aligned otherwise
  30.      * WSASend() will return undocumented WSAEINVAL error.
  31.      */

  33.     wsabuf.buf = (char *) buf;
  34.     wsabuf.len = size;

  36.     sent = 0;

  38.     n = WSASend(c->fd, &wsabuf, 1, &sent, 0, NULL, NULL);

  40.     ngx_log_debug4(NGX_LOG_DEBUG_EVENT, c->log, 0,
  41.                    "WSASend: fd:%d, %d, %ul of %uz", c->fd, n, sent, size);

  43.     if (n == 0) {
  44.         if (sent < size) {
  45.             wev->ready = 0;
  46.         }

  48.         c->sent += sent;

  50.         return sent;
  51.     }

  53.     err = ngx_socket_errno;

  55.     if (err == WSAEWOULDBLOCK) {
  56.         ngx_log_debug0(NGX_LOG_DEBUG_EVENT, c->log, err, "WSASend() not ready");
  57.         wev->ready = 0;
  58.         return NGX_AGAIN;
  59.     }

  61.     wev->error = 1;
  62.     ngx_connection_error(c, err, "WSASend() failed");

  64.     return NGX_ERROR;
  65. }


  68. ssize_t
  69. ‌ngx_overlapped_wsasend(ngx_connection_t *c, u_char *buf, size_t size)
  70. {
  71.     int               n;
  72.     u_long            sent;
  73.     ngx_err_t         err;
  74.     ngx_event_t      *wev;
  75.     LPWSAOVERLAPPED   ovlp;
  76.     WSABUF            wsabuf;

  78.     wev = c->write;

  80.     if (!wev->ready) {
  81.         return NGX_AGAIN;
  82.     }

  84.     ngx_log_debug1(NGX_LOG_DEBUG_EVENT, c->log, 0,
  85.                    "wev->complete: %d", wev->complete);

  87.     if (!wev->complete) {

  89.         /* post the overlapped WSASend() */

  91.         /*
  92.          * WSABUFs must be 4-byte aligned otherwise
  93.          * WSASend() will return undocumented WSAEINVAL error.
  94.          */

  96.         wsabuf.buf = (char *) buf;
  97.         wsabuf.len = size;

  99.         sent = 0;

  101.         ovlp = (LPWSAOVERLAPPED) &c->write->ovlp;
  102.         ngx_memzero(ovlp, sizeof(WSAOVERLAPPED));

  104.         n = WSASend(c->fd, &wsabuf, 1, &sent, 0, ovlp, NULL);

  106.         ngx_log_debug4(NGX_LOG_DEBUG_EVENT, c->log, 0,
  107.                        "WSASend: fd:%d, %d, %ul of %uz", c->fd, n, sent, size);

  109.         wev->complete = 0;

  111.         if (n == 0) {
  112.             if (ngx_event_flags & NGX_USE_IOCP_EVENT) {

  114.                 /*
  115.                  * if a socket was bound with I/O completion port then
  116.                  * GetQueuedCompletionStatus() would anyway return its status
  117.                  * despite that WSASend() was already complete
  118.                  */

  120.                 wev->active = 1;
  121.                 return NGX_AGAIN;
  122.             }

  124.             if (sent < size) {
  125.                 wev->ready = 0;
  126.             }

  128.             c->sent += sent;

  130.             return sent;
  131.         }

  133.         err = ngx_socket_errno;

  135.         if (err == WSA_IO_PENDING) {
  136.             ngx_log_debug0(NGX_LOG_DEBUG_EVENT, c->log, err,
  137.                            "WSASend() posted");
  138.             wev->active = 1;
  139.             return NGX_AGAIN;
  140.         }

  142.         wev->error = 1;
  143.         ngx_connection_error(c, err, "WSASend() failed");

  145.         return NGX_ERROR;
  146.     }

  148.     /* the overlapped WSASend() complete */

  150.     wev->complete = 0;
  151.     wev->active = 0;

  153.     if (ngx_event_flags & NGX_USE_IOCP_EVENT) {

  155.         if (wev->ovlp.error) {
  156.             ngx_connection_error(c, wev->ovlp.error, "WSASend() failed");
  157.             return NGX_ERROR;
  158.         }

  160.         sent = wev->available;

  162.     } else {
  163.         if (WSAGetOverlappedResult(c->fd, (LPWSAOVERLAPPED) &wev->ovlp,
  164.                                    &sent, 0, NULL)
  165.             == 0)
  166.         {
  167.             ngx_connection_error(c, ngx_socket_errno,
  168.                            "WSASend() or WSAGetOverlappedResult() failed");

  170.             return NGX_ERROR;
  171.         }
  172.     }

  174.     ngx_log_debug3(NGX_LOG_DEBUG_EVENT, c->log, 0,
  175.                    "WSAGetOverlappedResult: fd:%d, %ul of %uz",
  176.                    c->fd, sent, size);

  178.     if (sent < size) {
  179.         wev->ready = 0;
  180.     }

  182.     c->sent += sent;

  184.     return sent;
  185. }

One Level Up Top Level