r5414: - added libcli/wins/, a basic client library for WINS replication

- added a new IDL type "udlongr", which is like udlong, but with the
  two uint32 halves reversed

- modified the winsrepl.idl to cope with a wider range of packets
(This used to be commit bc8d60c918f2e268d591aac464fc6a78c38a4cf9)

2 files changed, 505 insertions, 0 deletions

diff --git a/source4/libcli/wins/winsrepl.c b/source4/libcli/wins/winsrepl.c
new file mode 100644
index 0000000000..6ee948202f
--- /dev/null
+++ b/source4/libcli/wins/winsrepl.c
@@ -0,0 +1,436 @@
+/* 
+   Unix SMB/CIFS implementation.
+
+   low level WINS replication client code
+
+   Copyright (C) Andrew Tridgell 2005
+   
+   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"
+#include "lib/events/events.h"
+#include "dlinklist.h"
+#include "lib/socket/socket.h"
+#include "libcli/wins/winsrepl.h"
+
+/*
+  mark all pending requests as dead - called when a socket error happens
+*/
+static void wrepl_socket_dead(struct wrepl_socket *wrepl_socket)
+{
+	event_set_fd_flags(wrepl_socket->fde, 0);
+
+	while (wrepl_socket->send_queue) {
+		struct wrepl_request *req = wrepl_socket->send_queue;
+		DLIST_REMOVE(wrepl_socket->send_queue, req);
+		req->state = WREPL_REQUEST_ERROR;
+		req->status = NT_STATUS_UNEXPECTED_NETWORK_ERROR;
+		if (req->async.fn) {
+			req->async.fn(req);
+		}
+	}
+	while (wrepl_socket->recv_queue) {
+		struct wrepl_request *req = wrepl_socket->recv_queue;
+		DLIST_REMOVE(wrepl_socket->recv_queue, req);
+		req->state = WREPL_REQUEST_ERROR;
+		req->status = NT_STATUS_UNEXPECTED_NETWORK_ERROR;
+		if (req->async.fn) {
+			req->async.fn(req);
+		}
+	}
+}
+
+/*
+  handle send events 
+*/
+static void wrepl_handler_send(struct wrepl_socket *wrepl_socket)
+{
+	while (wrepl_socket->send_queue) {
+		struct wrepl_request *req = wrepl_socket->send_queue;
+		size_t nsent;
+		NTSTATUS status;
+
+		status = socket_send(wrepl_socket->sock, &req->buffer, &nsent, 0);
+		if (NT_STATUS_IS_ERR(status)) {
+			wrepl_socket_dead(wrepl_socket);
+			return;
+		}
+		if (!NT_STATUS_IS_OK(status) || nsent == 0) return;
+
+		req->buffer.data   += nsent;
+		req->buffer.length -= nsent;
+		if (req->buffer.length != 0) {
+			return;
+		}
+
+		DLIST_REMOVE(wrepl_socket->send_queue, req);
+		DLIST_ADD_END(wrepl_socket->recv_queue, req, struct wrepl_request *);
+		req->state = WREPL_REQUEST_RECV;
+
+		EVENT_FD_READABLE(wrepl_socket->fde);
+	}
+
+	EVENT_FD_NOT_WRITEABLE(wrepl_socket->fde);
+}
+
+
+/*
+  handle recv events 
+*/
+static void wrepl_handler_recv(struct wrepl_socket *wrepl_socket)
+{
+	size_t nread;
+	struct wrepl_request *req = wrepl_socket->recv_queue;
+	DATA_BLOB blob;
+
+	if (req == NULL) {
+		EVENT_FD_NOT_READABLE(wrepl_socket->fde);
+		return;
+	}
+
+	if (req->buffer.length == 0) {
+		req->buffer = data_blob_talloc(req, NULL, 4);
+		if (req->buffer.data == NULL) {
+			req->status = NT_STATUS_NO_MEMORY;
+			goto failed;
+		}
+		req->num_read = 0;
+	}
+
+	/* read in the packet length */
+	if (req->num_read < 4) {
+		uint32_t req_length;
+
+		req->status = socket_recv(wrepl_socket->sock, 
+					  req->buffer.data + req->num_read,
+					  4 - req->num_read,
+					  &nread, 0);
+		if (NT_STATUS_IS_ERR(req->status)) goto failed;
+		if (!NT_STATUS_IS_OK(req->status)) return;
+
+		req->num_read += nread;
+		if (req->num_read != 4) return;
+
+		req_length = RIVAL(req->buffer.data, 0) + 4;
+
+		req->buffer.data = talloc_realloc(req, req->buffer.data, 
+						  uint8_t, req_length);
+		if (req->buffer.data == NULL) {
+			req->status = NT_STATUS_NO_MEMORY;
+			goto failed;
+		}
+		req->buffer.length = req_length;
+	}
+
+	/* read in the body */
+	req->status = socket_recv(wrepl_socket->sock, 
+				  req->buffer.data + req->num_read,
+				  req->buffer.length - req->num_read,
+				  &nread, 0);
+	if (NT_STATUS_IS_ERR(req->status)) goto failed;
+	if (!NT_STATUS_IS_OK(req->status)) return;
+
+	req->num_read += nread;
+	if (req->num_read != req->buffer.length) return;
+
+	req->packet = talloc(req, struct wrepl_packet);
+	if (req->packet == NULL) {
+		req->status = NT_STATUS_NO_MEMORY;
+		goto failed;
+	}
+
+	blob.data = req->buffer.data + 4;
+	blob.length = req->buffer.length - 4;
+	
+	/* we have a full request - parse it */
+	req->status = ndr_pull_struct_blob(&blob,
+					   req->packet, req->packet,
+					   (ndr_pull_flags_fn_t)ndr_pull_wrepl_packet);
+	if (!NT_STATUS_IS_OK(req->status)) {
+		DEBUG(2,("Failed to parse incoming WINS packet - %s\n",
+			 nt_errstr(req->status)));
+		DEBUG(10,("packet length %d\n", req->buffer.length));
+		NDR_PRINT_DEBUG(wrepl_packet, req->packet);
+		goto failed;
+	}
+
+	if (DEBUGLVL(10)) {
+		DEBUG(10,("Received WINS packet of length %d\n", req->buffer.length));
+		NDR_PRINT_DEBUG(wrepl_packet, req->packet);
+	}
+
+	DLIST_REMOVE(wrepl_socket->recv_queue, req);
+	req->state = WREPL_REQUEST_DONE;
+	if (req->async.fn) {
+		req->async.fn(req);
+	}
+	return;
+
+failed:
+	if (req->state == WREPL_REQUEST_RECV) {
+		DLIST_REMOVE(wrepl_socket->recv_queue, req);
+	}
+	req->state = WREPL_REQUEST_ERROR;
+	if (req->async.fn) {
+		req->async.fn(req);
+	}
+}
+
+
+/*
+  handler for winrepl events
+*/
+static void wrepl_handler(struct event_context *ev, struct fd_event *fde, 
+			  uint16_t flags, void *private)
+{
+	struct wrepl_socket *wrepl_socket = talloc_get_type(private, 
+							    struct wrepl_socket);
+	if (flags & EVENT_FD_WRITE) {
+		wrepl_handler_send(wrepl_socket);
+		return;
+	}
+	if (flags & EVENT_FD_READ) {
+		wrepl_handler_recv(wrepl_socket);
+	}
+}
+
+
+/*
+  handler for winrepl connection completion
+*/
+static void wrepl_connect_handler(struct event_context *ev, struct fd_event *fde, 
+				  uint16_t flags, void *private)
+{
+	struct wrepl_socket *wrepl_socket = talloc_get_type(private, 
+							    struct wrepl_socket);
+	struct wrepl_request *req = wrepl_socket->recv_queue;
+
+	talloc_free(fde);
+
+	if (req == NULL) return;
+
+	req->status = socket_connect_complete(wrepl_socket->sock, 0);
+	if (NT_STATUS_IS_ERR(req->status)) goto failed;
+
+	if (!NT_STATUS_IS_OK(req->status)) return;
+
+	wrepl_socket->fde = event_add_fd(wrepl_socket->event_ctx, wrepl_socket, 
+					 socket_get_fd(wrepl_socket->sock), 
+					 0,
+					 wrepl_handler, wrepl_socket);
+	if (wrepl_socket->fde == NULL) {
+		req->status = NT_STATUS_NO_MEMORY;
+	}
+
+
+failed:
+	DLIST_REMOVE(wrepl_socket->recv_queue, req);
+	if (!NT_STATUS_IS_OK(req->status)) {
+		req->state = WREPL_REQUEST_ERROR;
+	} else {
+		req->state = WREPL_REQUEST_DONE;
+	}
+	if (req->async.fn) {
+		req->async.fn(req);
+	}
+}
+
+
+/*
+  initialise a wrepl_socket. The event_ctx is optional, if provided then
+  operations will use that event context
+*/
+struct wrepl_socket *wrepl_socket_init(TALLOC_CTX *mem_ctx, 
+				       struct event_context *event_ctx)
+{
+	struct wrepl_socket *wrepl_socket;
+	NTSTATUS status;
+
+	wrepl_socket = talloc(mem_ctx, struct wrepl_socket);
+	if (wrepl_socket == NULL) goto failed;
+
+	if (event_ctx == NULL) {
+		wrepl_socket->event_ctx = event_context_init(wrepl_socket);
+	} else {
+		wrepl_socket->event_ctx = talloc_reference(wrepl_socket, event_ctx);
+	}
+	if (wrepl_socket->event_ctx == NULL) goto failed;
+
+	status = socket_create("ip", SOCKET_TYPE_STREAM, &wrepl_socket->sock, 0);
+	if (!NT_STATUS_IS_OK(status)) goto failed;
+
+	talloc_steal(wrepl_socket, wrepl_socket->sock);
+
+	wrepl_socket->send_queue = NULL;
+	wrepl_socket->recv_queue = NULL;
+
+	wrepl_socket->fde = event_add_fd(wrepl_socket->event_ctx, wrepl_socket, 
+					 socket_get_fd(wrepl_socket->sock), 
+					 EVENT_FD_WRITE,
+					 wrepl_connect_handler, wrepl_socket);
+
+	set_blocking(socket_get_fd(wrepl_socket->sock), False);
+	
+	return wrepl_socket;
+
+failed:
+	talloc_free(wrepl_socket);
+	return NULL;
+}
+
+
+/*
+  destroy a wrepl_request
+*/
+static int wrepl_request_destructor(void *ptr)
+{
+	struct wrepl_request *req = talloc_get_type(ptr, struct wrepl_request);
+	if (req->state == WREPL_REQUEST_SEND) {
+		DLIST_REMOVE(req->wrepl_socket->send_queue, req);
+	}
+	if (req->state == WREPL_REQUEST_RECV) {
+		DLIST_REMOVE(req->wrepl_socket->recv_queue, req);
+	}
+	req->state = WREPL_REQUEST_ERROR;
+	return 0;
+}
+
+/*
+  wait for a request to complete
+*/
+static NTSTATUS wrepl_request_wait(struct wrepl_request *req)
+{
+	NT_STATUS_HAVE_NO_MEMORY(req);
+	while (req->state < WREPL_REQUEST_DONE) {
+		event_loop_once(req->wrepl_socket->event_ctx);
+	}
+	return req->status;
+}
+
+
+/*
+  connect a wrepl_socket to a WINS server
+*/
+struct wrepl_request *wrepl_connect_send(struct wrepl_socket *wrepl_socket,
+					 const char *address)
+{
+	struct wrepl_request *req;
+	NTSTATUS status;
+
+	req = talloc_zero(wrepl_socket, struct wrepl_request);
+	if (req == NULL) goto failed;
+
+	req->wrepl_socket = wrepl_socket;
+	req->state        = WREPL_REQUEST_RECV;
+
+	DLIST_ADD(wrepl_socket->recv_queue, req);
+
+	talloc_set_destructor(req, wrepl_request_destructor);
+	
+	status = socket_connect(wrepl_socket->sock, NULL, 0, address, 
+				WINS_REPLICATION_PORT, 0);
+	if (!NT_STATUS_EQUAL(status, NT_STATUS_MORE_PROCESSING_REQUIRED)) goto failed;
+
+	return req;
+
+failed:
+	talloc_free(req);
+	return NULL;
+}
+
+/*
+  connect a wrepl_socket to a WINS server - recv side
+*/
+NTSTATUS wrepl_connect_recv(struct wrepl_request *req)
+{
+	return wrepl_request_wait(req);
+}
+
+
+/*
+  connect a wrepl_socket to a WINS server - sync API
+*/
+NTSTATUS wrepl_connect(struct wrepl_socket *wrepl_socket, const char *address)
+{
+	struct wrepl_request *req = wrepl_connect_send(wrepl_socket, address);
+	return wrepl_connect_recv(req);
+}
+
+
+/*
+  send a generic wins replication request
+*/
+struct wrepl_request *wrepl_request_send(struct wrepl_socket *wrepl_socket,
+					 struct wrepl_packet *packet)
+{
+	struct wrepl_request *req;
+	struct wrepl_wrap wrap;
+
+	req = talloc_zero(wrepl_socket, struct wrepl_request);
+	if (req == NULL) goto failed;
+
+	req->wrepl_socket = wrepl_socket;
+	req->state        = WREPL_REQUEST_SEND;
+
+	wrap.packet = *packet;
+	req->status = ndr_push_struct_blob(&req->buffer, req, &wrap,
+					   (ndr_push_flags_fn_t)ndr_push_wrepl_wrap);	
+	if (!NT_STATUS_IS_OK(req->status)) goto failed;
+
+	if (DEBUGLVL(10)) {
+		DEBUG(10,("Sending WINS packet of length %d\n", req->buffer.length));
+		NDR_PRINT_DEBUG(wrepl_packet, &wrap.packet);
+	}
+
+	DLIST_ADD(wrepl_socket->send_queue, req);
+
+	talloc_set_destructor(req, wrepl_request_destructor);
+
+	EVENT_FD_WRITEABLE(wrepl_socket->fde);
+	
+	return req;
+
+failed:
+	talloc_free(req);
+	return NULL;
+}
+
+/*
+  receive a generic WINS replication reply
+*/
+NTSTATUS wrepl_request_recv(struct wrepl_request *req,
+			    TALLOC_CTX *mem_ctx,
+			    struct wrepl_packet **packet)
+{
+	NTSTATUS status = wrepl_request_wait(req);
+	if (NT_STATUS_IS_OK(status)) {
+		*packet = talloc_steal(mem_ctx, req->packet);
+	}
+	talloc_free(req);
+	return status;
+}
+
+/*
+  a full WINS replication request/response
+*/
+NTSTATUS wrepl_request(struct wrepl_socket *wrepl_socket,
+		       TALLOC_CTX *mem_ctx,
+		       struct wrepl_packet *req_packet,
+		       struct wrepl_packet **reply_packet)
+{
+	struct wrepl_request *req = wrepl_request_send(wrepl_socket, req_packet);
+	return wrepl_request_recv(req, mem_ctx, reply_packet);
+}
diff --git a/source4/libcli/wins/winsrepl.h b/source4/libcli/wins/winsrepl.h
new file mode 100644
index 0000000000..3fd1e5406c
--- /dev/null
+++ b/source4/libcli/wins/winsrepl.h
@@ -0,0 +1,69 @@
+/*
+   Unix SMB/CIFS implementation.
+
+   structures for WINS replication client library
+
+   Copyright (C) Andrew Tridgell 2005
+
+   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 "librpc/gen_ndr/ndr_winsrepl.h"
+
+/*
+  main context structure for the wins replication client library
+*/
+struct wrepl_socket {
+	struct socket_context *sock;
+	struct event_context *event_ctx;
+
+	/* a queue of requests pending to be sent */
+	struct wrepl_request *send_queue;
+
+	/* a queue of replies waiting to be received */
+	struct wrepl_request *recv_queue;
+
+	/* the fd event */
+	struct fd_event *fde;
+};
+
+enum wrepl_request_state {
+	WREPL_REQUEST_SEND  = 0,
+	WREPL_REQUEST_RECV  = 1,
+	WREPL_REQUEST_DONE  = 2,
+	WREPL_REQUEST_ERROR = 3
+};
+
+/*
+  a WINS replication request
+*/
+struct wrepl_request {
+	struct wrepl_request *next, *prev;
+	struct wrepl_socket *wrepl_socket;
+
+	enum wrepl_request_state state;
+	NTSTATUS status;
+
+	DATA_BLOB buffer;
+
+	size_t num_read;
+
+	struct wrepl_packet *packet;
+
+	struct {
+		void (*fn)(struct wrepl_request *);
+		void *private;
+	} async;
+};