nginx-quic/src/event/ngx_event_recv.c

#include <ngx_config.h>
#include <ngx_core.h>
#include <ngx_errno.h>
#include <ngx_log.h>
#include <ngx_recv.h>
#include <ngx_connection.h>

ssize_t ngx_event_recv_core(ngx_connection_t *c, char *buf, size_t size)
{
    ssize_t       n;
    ngx_err_t     err;
    ngx_event_t  *ev;

    ev = c->read;

/* DEBUG */
#if (HAVE_KQUEUE)
    if (ngx_event_flags & NGX_HAVE_KQUEUE_EVENT) {
        ngx_log_debug(c->log, "ngx_event_recv: eof:%d, avail:%d, err:%d" _
                      ev->eof _ ev->available _ ev->error);
    }
#endif

#if (USE_KQUEUE)

    if (ev->eof && ev->available == 0) {

        if (ev->error == 0) {
            return 0;
        }

        ngx_set_socket_errno(ev->error);
        err = ev->error;
        n = -1;

    } else {
        n = ngx_recv(c->fd, buf, size, 0);

        if (n == -1) {
            err = ngx_socket_errno;
        }
    }

    if (n == -1) {
        ev->ready = 0;

        if (err == NGX_ECONNRESET && ev->ignore_econnreset) {
            return 0;
        }

        if (err == NGX_EAGAIN) {
            ngx_log_error(NGX_LOG_INFO, c->log, err, "recv() returned EAGAIN");
            return NGX_AGAIN;
        }

        ngx_log_error(NGX_LOG_ERR, c->log, err, "recv() failed");
        return NGX_ERROR;
    }

    ev->available -= n;
    if (ev->available == 0) {
        ev->ready = 0;
    }

    return n;

#elif (HAVE_KQUEUE)

    if ((ngx_event_flags & NGX_HAVE_KQUEUE_EVENT)
        && ev->eof && ev->available == 0) {

        if (ev->error == 0) {
            return 0;
        }

        ngx_set_socket_errno(ev->error);
        err = ev->error;
        n = -1;

    } else {
        n = ngx_recv(c->fd, buf, size, 0);
ngx_log_debug(c->log, "ngx_event_recv: read:%d:%d" _ n _ size);

        if (n == -1) {
            err = ngx_socket_errno;
        }
    }

    if (n == -1) {
        ev->ready = 0;

        if (err == NGX_ECONNRESET && ev->ignore_econnreset) {
            return 0;
        }

        if (err == NGX_EAGAIN) {
            ngx_log_error(NGX_LOG_INFO, c->log, err, "recv() returned EAGAIN");
            return NGX_AGAIN;
        }

        ngx_log_error(NGX_LOG_ERR, c->log, err, "recv() failed");
        return NGX_ERROR;
    }

    if (ngx_event_flags & NGX_HAVE_KQUEUE_EVENT) {
        ev->available -= n;
        if (ev->available == 0) {
            ev->ready = 0;
        }

    } else if ((size_t) n < size) {
        ev->ready = 0;
    }

    return n;

#else /* not kqueue */

    n = ngx_recv(c->fd, buf, size, 0);

    if (n == -1) {
        err = ngx_socket_errno;

        ev->ready = 0;

        if (err == NGX_ECONNRESET && ev->ignore_econnreset) {
            return 0;
        }

        if (err == NGX_EAGAIN) {
            ngx_log_error(NGX_LOG_INFO, c->log, err, "recv() returned EAGAIN");
            return NGX_AGAIN;
        }

        ngx_log_error(NGX_LOG_ERR, c->log, err, "recv() failed");
        return NGX_ERROR;
    }

    if ((size_t) n < size) {
        ev->ready = 0;
    }

    return n;

#endif
}