--- /dev/null
+/*
+ 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);
+}
[
uuid("0-1-2-3-4"),
version(0.0),
- pointer_default(unique),
- depends(nbt)
+ pointer_default(unique)
]
interface wrepl
{
const int WINS_REPLICATION_PORT = 42;
- declare enum nbt_name_type;
-
typedef [flag(NDR_BIG_ENDIAN)] struct {
ipv4address owner;
ipv4address ip;
} wrepl_ip;
typedef [flag(NDR_LITTLE_ENDIAN)] struct {
- uint32 num_ips;
- wrepl_ip ips[num_ips];
+ uint32 num_ips;
+ wrepl_ip ips[num_ips];
ipv4address unknown;
} wrepl_address_list;
typedef [nodiscriminant] union {
- [case(0)] wrepl_ip address;
+ [case(0)] wrepl_ip address;
[case(2)] wrepl_address_list addresses;
} wrepl_addresses;
typedef struct {
- uint32 id_high;
- uint32 id_low;
- } wrepl_id;
-
- typedef struct {
- uint32 name_len;
- astring15 name;
- nbt_name_type type;
- uint32 unknown;
- uint32 flags;
+ uint32 name_len;
+ uint8 name[name_len];
+ uint32 flags;
[flag(NDR_LITTLE_ENDIAN)] uint32 group_flag;
- wrepl_id id;
+ udlongr id;
[switch_is(flags & 2)] wrepl_addresses addresses;
} wrepl_wins_name;
typedef struct {
- uint32 num_names;
+ uint32 num_names;
wrepl_wins_name names[num_names];
} wrepl_send_reply;
typedef struct {
ipv4address address;
- wrepl_id max_version;
- wrepl_id min_version;
- uint32 type;
+ udlongr max_version;
+ udlongr min_version;
+ uint32 type;
} wrepl_wins_owner;
typedef struct {
- uint32 partner_count;
+ uint32 partner_count;
wrepl_wins_owner partners[partner_count];
- ipv4address initiator;
+ ipv4address initiator;
} wrepl_table;
typedef [v1_enum] enum {
typedef [nodiscriminant] union {
[case(WREPL_REPL_TABLE_QUERY)] ;
- [case(WREPL_REPL_TABLE_REPLY)] wrepl_table table;
+ [case(WREPL_REPL_TABLE_REPLY)] wrepl_table table;
[case(WREPL_REPL_SEND_REQUEST)] wrepl_wins_owner owner;
- [case(WREPL_REPL_SEND_REPLY)] wrepl_send_reply reply;
+ [case(WREPL_REPL_SEND_REPLY)] wrepl_send_reply reply;
[case(WREPL_REPL_UPDATE)] ;
- [case(WREPL_REPL_INFORM)] wrepl_table table;
+ [case(WREPL_REPL_INFORM)] wrepl_table table;
} wrepl_replication_info;
typedef struct {
} wrepl_mess_type;
typedef [nodiscriminant] union {
- [case(WREPL_START_ASSOCIATION)] wrepl_start start;
+ [case(WREPL_START_ASSOCIATION)] wrepl_start start;
[case(WREPL_START_ASSOCIATION_REPLY)] wrepl_start start_reply;
- [case(WREPL_STOP_ASSOCIATION)] wrepl_stop stop;
- [case(WREPL_REPLICATION)] wrepl_replication replication;
+ [case(WREPL_STOP_ASSOCIATION)] wrepl_stop stop;
+ [case(WREPL_REPLICATION)] wrepl_replication replication;
} wrepl_message;
+ /*
+ the opcode appears to be a bitfield, but as far as I can tell
+ you must always set the following bits. Additional bits don't
+ seem to matter. Very strange.
+ */
+ const int WREPL_OPCODE_BITS = 0x7800;
+
- typedef [gensize,flag(NDR_NOALIGN|NDR_BIG_ENDIAN|NDR_PAHEX),public] struct {
- uint32 opcode;
- uint32 assoc_ctx;
- wrepl_mess_type mess_type;
+ typedef [gensize,flag(NDR_BIG_ENDIAN|NDR_PAHEX),public] struct {
+ uint32 opcode;
+ uint32 assoc_ctx;
+ wrepl_mess_type mess_type;
[switch_is(mess_type)] wrepl_message message;
[flag(NDR_REMAINING)] DATA_BLOB padding;
} wrepl_packet;
- typedef [flag(NDR_NOALIGN|NDR_BIG_ENDIAN|NDR_PAHEX),public] struct {
- [value(ndr_size_wrepl_packet(&r->packet, ndr->flags))] uint32 size;
+ typedef [flag(NDR_BIG_ENDIAN|NDR_PAHEX),public] struct {
+ [value(ndr_size_wrepl_packet(&r->packet, ndr->flags))] uint32 size;
wrepl_packet packet;
} wrepl_wrap;
git.samba.org / bbaumbach / samba-autobuild / .git / commitdiff