source: trunk/server/source3/lib/packet.c@ 533

/* 
   Unix SMB/CIFS implementation.
   Packet handling
   Copyright (C) Volker Lendecke 2007
   
   This program is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation; either version 3 of the License, or
   (at your option) any later version.
   
   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.
   
   You should have received a copy of the GNU General Public License
   along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

#include "includes.h"

struct packet_context {
        int fd;
        DATA_BLOB in, out;
};

/*
 * Close the underlying fd
 */
static int packet_context_destructor(struct packet_context *ctx)
{
        return close(ctx->fd);
}

/*
 * Initialize a packet context. The fd is given to the packet context, meaning
 * that it is automatically closed when the packet context is freed.
 */
struct packet_context *packet_init(TALLOC_CTX *mem_ctx, int fd)
{
        struct packet_context *result;

        if (!(result = TALLOC_ZERO_P(mem_ctx, struct packet_context))) {
                return NULL;
        }

        result->fd = fd;
        talloc_set_destructor(result, packet_context_destructor);
        return result;
}

/*
 * Pull data from the fd
 */
NTSTATUS packet_fd_read(struct packet_context *ctx)
{
        int res, available;
        size_t new_size;
        uint8 *in;

        res = ioctl(ctx->fd, FIONREAD, &available);

        if (res == -1) {
                DEBUG(10, ("ioctl(FIONREAD) failed: %s\n", strerror(errno)));
                return map_nt_error_from_unix(errno);
        }

        SMB_ASSERT(available >= 0);

        if (available == 0) {
                return NT_STATUS_END_OF_FILE;
        }

        new_size = ctx->in.length + available;

        if (new_size < ctx->in.length) {
                DEBUG(0, ("integer wrap\n"));
                return NT_STATUS_NO_MEMORY;
        }

        if (!(in = TALLOC_REALLOC_ARRAY(ctx, ctx->in.data, uint8, new_size))) {
                DEBUG(10, ("talloc failed\n"));
                return NT_STATUS_NO_MEMORY;
        }

        ctx->in.data = in;

        res = recv(ctx->fd, in + ctx->in.length, available, 0);

        if (res < 0) {
                DEBUG(10, ("recv failed: %s\n", strerror(errno)));
                return map_nt_error_from_unix(errno);
        }

        if (res == 0) {
                return NT_STATUS_END_OF_FILE;
        }

        ctx->in.length += res;

        return NT_STATUS_OK;
}

NTSTATUS packet_fd_read_sync(struct packet_context *ctx,
                             struct timeval *timeout)
{
        int res;
        fd_set r_fds;

        FD_ZERO(&r_fds);
        FD_SET(ctx->fd, &r_fds);

        res = sys_select(ctx->fd+1, &r_fds, NULL, NULL, timeout);

        if (res == 0) {
                DEBUG(10, ("select timed out\n"));
                return NT_STATUS_IO_TIMEOUT;
        }

        if (res == -1) {
                DEBUG(10, ("select returned %s\n", strerror(errno)));
                return map_nt_error_from_unix(errno);
        }

        return packet_fd_read(ctx);
}

bool packet_handler(struct packet_context *ctx,
                    bool (*full_req)(const uint8_t *buf,
                                     size_t available,
                                     size_t *length,
                                     void *priv),
                    NTSTATUS (*callback)(uint8_t *buf, size_t length,
                                         void *priv),
                    void *priv, NTSTATUS *status)
{
        size_t length;
        uint8_t *buf;

        if (!full_req(ctx->in.data, ctx->in.length, &length, priv)) {
                return False;
        }

        if (length > ctx->in.length) {
                *status = NT_STATUS_INTERNAL_ERROR;
                return true;
        }

        if (length == ctx->in.length) {
                buf = ctx->in.data;
                ctx->in.data = NULL;
                ctx->in.length = 0;
        } else {
                buf = (uint8_t *)TALLOC_MEMDUP(ctx, ctx->in.data, length);
                if (buf == NULL) {
                        *status = NT_STATUS_NO_MEMORY;
                        return true;
                }

                memmove(ctx->in.data, ctx->in.data + length,
                        ctx->in.length - length);
                ctx->in.length -= length;
        }

        *status = callback(buf, length, priv);
        return True;
}

/*
 * How many bytes of outgoing data do we have pending?
 */
size_t packet_outgoing_bytes(struct packet_context *ctx)
{
        return ctx->out.length;
}

/*
 * Push data to the fd
 */
NTSTATUS packet_fd_write(struct packet_context *ctx)
{
        ssize_t sent;

        sent = send(ctx->fd, ctx->out.data, ctx->out.length, 0);

        if (sent == -1) {
                DEBUG(0, ("send failed: %s\n", strerror(errno)));
                return map_nt_error_from_unix(errno);
        }

        memmove(ctx->out.data, ctx->out.data + sent,
                ctx->out.length - sent);
        ctx->out.length -= sent;

        return NT_STATUS_OK;
}

/*
 * Sync flush all outgoing bytes
 */
NTSTATUS packet_flush(struct packet_context *ctx)
{
        while (ctx->out.length != 0) {
                NTSTATUS status = packet_fd_write(ctx);
                if (!NT_STATUS_IS_OK(status)) {
                        return status;
                }
        }
        return NT_STATUS_OK;
}

/*
 * Send a list of DATA_BLOBs
 *
 * Example:  packet_send(ctx, 2, data_blob_const(&size, sizeof(size)),
 *                       data_blob_const(buf, size));
 */
NTSTATUS packet_send(struct packet_context *ctx, int num_blobs, ...)
{
        va_list ap;
        int i;
        size_t len;
        uint8 *out;

        len = ctx->out.length;

        va_start(ap, num_blobs);
        for (i=0; i<num_blobs; i++) {
                size_t tmp;
                DATA_BLOB blob = va_arg(ap, DATA_BLOB);

                tmp = len + blob.length;
                if (tmp < len) {
                        DEBUG(0, ("integer overflow\n"));
                        va_end(ap);
                        return NT_STATUS_NO_MEMORY;
                }
                len = tmp;
        }
        va_end(ap);

        if (len == 0) {
                return NT_STATUS_OK;
        }

        if (!(out = TALLOC_REALLOC_ARRAY(ctx, ctx->out.data, uint8, len))) {
                DEBUG(0, ("talloc failed\n"));
                return NT_STATUS_NO_MEMORY;
        }

        ctx->out.data = out;

        va_start(ap, num_blobs);
        for (i=0; i<num_blobs; i++) {
                DATA_BLOB blob = va_arg(ap, DATA_BLOB);

                memcpy(ctx->out.data+ctx->out.length, blob.data, blob.length);
                ctx->out.length += blob.length;
        }
        va_end(ap);

        SMB_ASSERT(ctx->out.length == len);
        return NT_STATUS_OK;
}

/*
 * Get the packet context's file descriptor
 */
int packet_get_fd(struct packet_context *ctx)
{
        return ctx->fd;
}