diff options
Diffstat (limited to 'source4/libcli/wins/winsrepl.c')
| -rw-r--r-- | source4/libcli/wins/winsrepl.c | 436 |
1 files changed, 436 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); +} |