1 files changed, 256 insertions, 0 deletions
diff --git a/source3/lib/packet.c b/source3/lib/packet.c
new file mode 100644
index 00000000000..2473a771e0b
--- /dev/null
+++ b/source3/lib/packet.c
@@ -0,0 +1,256 @@
+/* 
+   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 2 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, write to the Free Software
+   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+*/
+
+#include "includes.h"
+
+struct packet_context {
+	int fd;
+	struct 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;
+	}
+
+	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.data = in;
+	ctx->in.length += available;
+
+	return NT_STATUS_OK;
+}
+
+NTSTATUS packet_fd_read_sync(struct packet_context *ctx)
+{
+	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, NULL);
+
+	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 struct data_blob *data,
+				     size_t *length,
+				     void *private_data),
+		    NTSTATUS (*callback)(const struct data_blob *data,
+					 void *private_data),
+		    void *private_data,
+		    NTSTATUS *status)
+{
+	size_t length;
+	struct data_blob data;
+
+	if (!full_req(&ctx->in, &length, private_data)) {
+		return False;
+	}
+
+	SMB_ASSERT(length <= ctx->in.length);
+
+	data.data = ctx->in.data;
+	data.length = length;
+
+	*status = callback(&data, private_data);
+
+	memmove(ctx->in.data, ctx->in.data + length,
+		ctx->in.length - length);
+	ctx->in.length -= length;
+
+	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;
+		struct data_blob blob = va_arg(ap, struct 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++) {
+		struct data_blob blob = va_arg(ap, struct 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;
+}
+