unit/src/nxt_recvbuf.c

/*
 * Copyright (C) Igor Sysoev
 * Copyright (C) NGINX, Inc.
 */

#include <nxt_main.h>


nxt_uint_t
nxt_recvbuf_mem_coalesce(nxt_recvbuf_coalesce_t *rb)
{
    u_char     *last;
    size_t     size, total;
    nxt_int_t  n;
    nxt_buf_t  *b;

    total = 0;
    last = NULL;
    n = -1;

    for (b = rb->buf; b != NULL; b = b->next) {

        nxt_prefetch(b->next);

        size = b->mem.end - b->mem.free;

        if (b->mem.free != last) {

            if (++n >= rb->nmax) {
                goto done;
            }

            nxt_iobuf_set(&rb->iobuf[n], b->mem.free, size);

        } else {
            nxt_iobuf_add(&rb->iobuf[n], size);
        }

        nxt_thread_log_debug("recvbuf: %ui, %p, %uz", n,
                             nxt_iobuf_data(&rb->iobuf[n]),
                             nxt_iobuf_size(&rb->iobuf[n]));

        total += size;
        last = b->mem.end;
    }

    n++;

done:

    rb->size = total;

    return n;
}


void
nxt_recvbuf_update(nxt_buf_t *b, size_t sent)
{
    size_t  size;

    while (b != NULL && sent != 0) {

        nxt_prefetch(b->next);

        if (!nxt_buf_is_sync(b)) {

            size = b->mem.end - b->mem.free;

            if (sent < size) {
                b->mem.free += sent;
                return;
            }

            b->mem.free = b->mem.end;
            sent -= size;
        }

        b = b->next;
    }
}