summaryrefslogtreecommitdiff
path: root/source4/wrepl_server
diff options
context:
space:
mode:
Diffstat (limited to 'source4/wrepl_server')
-rw-r--r--source4/wrepl_server/wrepl_in_connection.c319
-rw-r--r--source4/wrepl_server/wrepl_server.c308
2 files changed, 330 insertions, 297 deletions
diff --git a/source4/wrepl_server/wrepl_in_connection.c b/source4/wrepl_server/wrepl_in_connection.c
new file mode 100644
index 0000000000..ed9c9998d7
--- /dev/null
+++ b/source4/wrepl_server/wrepl_in_connection.c
@@ -0,0 +1,319 @@
+/*
+ Unix SMB/CIFS implementation.
+
+ WINS Replication server
+
+ Copyright (C) Stefan Metzmacher 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 "dlinklist.h"
+#include "lib/events/events.h"
+#include "lib/socket/socket.h"
+#include "smbd/service_task.h"
+#include "smbd/service_stream.h"
+#include "lib/messaging/irpc.h"
+#include "librpc/gen_ndr/ndr_winsrepl.h"
+#include "wrepl_server/wrepl_server.h"
+#include "nbt_server/wins/winsdb.h"
+#include "ldb/include/ldb.h"
+
+static void wreplsrv_terminate_in_connection(struct wreplsrv_in_connection *wreplconn, const char *reason)
+{
+ stream_terminate_connection(wreplconn->conn, reason);
+}
+
+/*
+ called when we get a new connection
+*/
+static void wreplsrv_accept(struct stream_connection *conn)
+{
+ struct wreplsrv_service *service = talloc_get_type(conn->private, struct wreplsrv_service);
+ struct wreplsrv_in_connection *wreplconn;
+ const char *peer_ip;
+
+ wreplconn = talloc_zero(conn, struct wreplsrv_in_connection);
+ if (!wreplconn) {
+ stream_terminate_connection(conn, "wreplsrv_accept: out of memory");
+ return;
+ }
+
+ wreplconn->conn = conn;
+ wreplconn->service = service;
+ wreplconn->our_ip = socket_get_my_addr(conn->socket, wreplconn);
+ if (!wreplconn->our_ip) {
+ wreplsrv_terminate_in_connection(wreplconn, "wreplsrv_accept: out of memory");
+ return;
+ }
+
+ peer_ip = socket_get_peer_addr(conn->socket, wreplconn);
+ if (!peer_ip) {
+ wreplsrv_terminate_in_connection(wreplconn, "wreplsrv_accept: out of memory");
+ return;
+ }
+
+ wreplconn->partner = wreplsrv_find_partner(service, peer_ip);
+
+ conn->private = wreplconn;
+
+ irpc_add_name(conn->msg_ctx, "wreplsrv_connection");
+}
+
+/*
+ receive some data on a WREPL connection
+*/
+static void wreplsrv_recv(struct stream_connection *conn, uint16_t flags)
+{
+ struct wreplsrv_in_connection *wreplconn = talloc_get_type(conn->private, struct wreplsrv_in_connection);
+ struct wreplsrv_in_call *call;
+ DATA_BLOB packet_in_blob;
+ DATA_BLOB packet_out_blob;
+ struct wrepl_wrap packet_out_wrap;
+ struct data_blob_list_item *rep;
+ NTSTATUS status = NT_STATUS_UNSUCCESSFUL;
+ size_t nread;
+
+ /* avoid recursion, because of half async code */
+ if (wreplconn->processing) {
+ EVENT_FD_NOT_READABLE(conn->event.fde);
+ return;
+ }
+
+ if (wreplconn->partial.length == 0) {
+ wreplconn->partial = data_blob_talloc(wreplconn, NULL, 4);
+ if (wreplconn->partial.data == NULL) {
+ status = NT_STATUS_NO_MEMORY;
+ goto failed;
+ }
+ wreplconn->partial_read = 0;
+ }
+
+ /* read in the packet length */
+ if (wreplconn->partial_read < 4) {
+ uint32_t packet_length;
+
+ status = socket_recv(conn->socket,
+ wreplconn->partial.data + wreplconn->partial_read,
+ 4 - wreplconn->partial_read,
+ &nread, 0);
+ if (NT_STATUS_IS_ERR(status)) goto failed;
+ if (!NT_STATUS_IS_OK(status)) return;
+
+ wreplconn->partial_read += nread;
+ if (wreplconn->partial_read != 4) return;
+
+ packet_length = RIVAL(wreplconn->partial.data, 0) + 4;
+
+ wreplconn->partial.data = talloc_realloc(wreplconn, wreplconn->partial.data,
+ uint8_t, packet_length);
+ if (wreplconn->partial.data == NULL) {
+ status = NT_STATUS_NO_MEMORY;
+ goto failed;
+ }
+ wreplconn->partial.length = packet_length;
+ }
+
+ /* read in the body */
+ status = socket_recv(conn->socket,
+ wreplconn->partial.data + wreplconn->partial_read,
+ wreplconn->partial.length - wreplconn->partial_read,
+ &nread, 0);
+ if (NT_STATUS_IS_ERR(status)) goto failed;
+ if (!NT_STATUS_IS_OK(status)) return;
+
+ wreplconn->partial_read += nread;
+ if (wreplconn->partial_read != wreplconn->partial.length) return;
+
+ packet_in_blob.data = wreplconn->partial.data + 4;
+ packet_in_blob.length = wreplconn->partial.length - 4;
+
+ call = talloc_zero(wreplconn, struct wreplsrv_in_call);
+ if (!call) {
+ status = NT_STATUS_NO_MEMORY;
+ goto failed;
+ }
+ call->wreplconn = wreplconn;
+
+ /* we have a full request - parse it */
+ status = ndr_pull_struct_blob(&packet_in_blob,
+ call, &call->req_packet,
+ (ndr_pull_flags_fn_t)ndr_pull_wrepl_packet);
+ if (!NT_STATUS_IS_OK(status)) {
+ DEBUG(2,("Failed to parse incoming WINS-Replication packet - %s\n",
+ nt_errstr(status)));
+ DEBUG(10,("packet length %u\n", wreplconn->partial.length));
+ NDR_PRINT_DEBUG(wrepl_packet, &call->req_packet);
+ goto failed;
+ }
+
+ /*
+ * we have parsed the request, so we can reset the wreplconn->partial_read,
+ * maybe we could also free wreplconn->partial, but for now we keep it,
+ * and overwrite it the next time
+ */
+ wreplconn->partial_read = 0;
+
+ if (DEBUGLVL(10)) {
+ DEBUG(10,("Received WINS-Replication packet of length %u\n", wreplconn->partial.length));
+ NDR_PRINT_DEBUG(wrepl_packet, &call->req_packet);
+ }
+
+ /* actually process the request */
+ wreplconn->processing = True;
+ status = wreplsrv_in_call(call);
+ wreplconn->processing = False;
+ if (NT_STATUS_IS_ERR(status)) goto failed;
+ if (!NT_STATUS_IS_OK(status)) {
+ /* w2k just ignores invalid packets, so we do */
+ DEBUG(10,("Received WINS-Replication packet was invalid, we just ignore it\n"));
+ talloc_free(call);
+ return;
+ }
+
+ /* and now encode the reply */
+ packet_out_wrap.packet = call->rep_packet;
+ status = ndr_push_struct_blob(&packet_out_blob, call, &packet_out_wrap,
+ (ndr_push_flags_fn_t)ndr_push_wrepl_wrap);
+ if (!NT_STATUS_IS_OK(status)) goto failed;
+
+ if (DEBUGLVL(10)) {
+ DEBUG(10,("Sending WINS-Replication packet of length %d\n", (int)packet_out_blob.length));
+ NDR_PRINT_DEBUG(wrepl_packet, &call->rep_packet);
+ }
+
+ rep = talloc(wreplconn, struct data_blob_list_item);
+ if (!rep) {
+ status = NT_STATUS_NO_MEMORY;
+ goto failed;
+ }
+
+ rep->blob = packet_out_blob;
+ talloc_steal(rep, packet_out_blob.data);
+ /* we don't need the call anymore */
+ talloc_free(call);
+
+ if (!wreplconn->send_queue) {
+ EVENT_FD_WRITEABLE(conn->event.fde);
+ }
+ DLIST_ADD_END(wreplconn->send_queue, rep, struct data_blob_list_item *);
+
+ if (wreplconn->terminate) {
+ EVENT_FD_NOT_READABLE(conn->event.fde);
+ } else {
+ EVENT_FD_READABLE(conn->event.fde);
+ }
+ return;
+
+failed:
+ wreplsrv_terminate_in_connection(wreplconn, nt_errstr(status));
+}
+
+/*
+ called when we can write to a connection
+*/
+static void wreplsrv_send(struct stream_connection *conn, uint16_t flags)
+{
+ struct wreplsrv_in_connection *wreplconn = talloc_get_type(conn->private, struct wreplsrv_in_connection);
+ NTSTATUS status;
+
+ while (wreplconn->send_queue) {
+ struct data_blob_list_item *rep = wreplconn->send_queue;
+ size_t sendlen;
+
+ status = socket_send(conn->socket, &rep->blob, &sendlen, 0);
+ if (NT_STATUS_IS_ERR(status)) goto failed;
+ if (!NT_STATUS_IS_OK(status)) return;
+
+ rep->blob.length -= sendlen;
+ rep->blob.data += sendlen;
+
+ if (rep->blob.length == 0) {
+ DLIST_REMOVE(wreplconn->send_queue, rep);
+ talloc_free(rep);
+ }
+ }
+
+ if (wreplconn->terminate) {
+ wreplsrv_terminate_in_connection(wreplconn, "connection terminated after all pending packets are send");
+ return;
+ }
+
+ EVENT_FD_NOT_WRITEABLE(conn->event.fde);
+ return;
+
+failed:
+ wreplsrv_terminate_in_connection(wreplconn, nt_errstr(status));
+}
+
+static const struct stream_server_ops wreplsrv_stream_ops = {
+ .name = "wreplsrv",
+ .accept_connection = wreplsrv_accept,
+ .recv_handler = wreplsrv_recv,
+ .send_handler = wreplsrv_send,
+};
+
+/*
+ startup the wrepl port 42 server sockets
+*/
+NTSTATUS wreplsrv_setup_sockets(struct wreplsrv_service *service)
+{
+ NTSTATUS status;
+ struct task_server *task = service->task;
+ const struct model_ops *model_ops;
+ const char *address;
+ uint16_t port = WINS_REPLICATION_PORT;
+
+ /* within the wrepl task we want to be a single process, so
+ ask for the single process model ops and pass these to the
+ stream_setup_socket() call. */
+ model_ops = process_model_byname("single");
+ if (!model_ops) {
+ DEBUG(0,("Can't find 'single' process model_ops"));
+ return NT_STATUS_INTERNAL_ERROR;
+ }
+
+ if (lp_interfaces() && lp_bind_interfaces_only()) {
+ int num_interfaces = iface_count();
+ int i;
+
+ /* We have been given an interfaces line, and been
+ told to only bind to those interfaces. Create a
+ socket per interface and bind to only these.
+ */
+ for(i = 0; i < num_interfaces; i++) {
+ address = iface_n_ip(i);
+ status = stream_setup_socket(task->event_ctx, model_ops, &wreplsrv_stream_ops,
+ "ipv4", address, &port, service);
+ if (!NT_STATUS_IS_OK(status)) {
+ DEBUG(0,("stream_setup_socket(address=%s,port=%u) failed - %s\n",
+ address, port, nt_errstr(status)));
+ return status;
+ }
+ }
+ } else {
+ address = lp_socket_address();
+ status = stream_setup_socket(task->event_ctx, model_ops, &wreplsrv_stream_ops,
+ "ipv4", address, &port, service);
+ if (!NT_STATUS_IS_OK(status)) {
+ DEBUG(0,("stream_setup_socket(address=%s,port=%u) failed - %s\n",
+ address, port, nt_errstr(status)));
+ return status;
+ }
+ }
+
+ return NT_STATUS_OK;
+}
diff --git a/source4/wrepl_server/wrepl_server.c b/source4/wrepl_server/wrepl_server.c
index e2da68d2b4..866090f400 100644
--- a/source4/wrepl_server/wrepl_server.c
+++ b/source4/wrepl_server/wrepl_server.c
@@ -32,12 +32,20 @@
#include "nbt_server/wins/winsdb.h"
#include "ldb/include/ldb.h"
-void wreplsrv_terminate_connection(struct wreplsrv_in_connection *wreplconn, const char *reason)
+/*
+ open winsdb
+*/
+static NTSTATUS wreplsrv_open_winsdb(struct wreplsrv_service *service)
{
- stream_terminate_connection(wreplconn->conn, reason);
+ service->wins_db = winsdb_connect(service);
+ if (!service->wins_db) {
+ return NT_STATUS_INTERNAL_DB_ERROR;
+ }
+
+ return NT_STATUS_OK;
}
-static struct wreplsrv_partner *wreplsrv_find_partner(struct wreplsrv_service *service, const char *peer_addr)
+struct wreplsrv_partner *wreplsrv_find_partner(struct wreplsrv_service *service, const char *peer_addr)
{
struct wreplsrv_partner *cur;
@@ -51,248 +59,6 @@ static struct wreplsrv_partner *wreplsrv_find_partner(struct wreplsrv_service *s
}
/*
- called when we get a new connection
-*/
-static void wreplsrv_accept(struct stream_connection *conn)
-{
- struct wreplsrv_service *service = talloc_get_type(conn->private, struct wreplsrv_service);
- struct wreplsrv_in_connection *wreplconn;
- const char *peer_ip;
-
- wreplconn = talloc_zero(conn, struct wreplsrv_in_connection);
- if (!wreplconn) {
- stream_terminate_connection(conn, "wreplsrv_accept: out of memory");
- return;
- }
-
- wreplconn->conn = conn;
- wreplconn->service = service;
- wreplconn->our_ip = socket_get_my_addr(conn->socket, wreplconn);
- if (!wreplconn->our_ip) {
- wreplsrv_terminate_connection(wreplconn, "wreplsrv_accept: out of memory");
- return;
- }
-
- peer_ip = socket_get_peer_addr(conn->socket, wreplconn);
- if (!peer_ip) {
- wreplsrv_terminate_connection(wreplconn, "wreplsrv_accept: out of memory");
- return;
- }
-
- wreplconn->partner = wreplsrv_find_partner(service, peer_ip);
-
- conn->private = wreplconn;
-
- irpc_add_name(conn->msg_ctx, "wreplsrv_connection");
-}
-
-/*
- receive some data on a WREPL connection
-*/
-static void wreplsrv_recv(struct stream_connection *conn, uint16_t flags)
-{
- struct wreplsrv_in_connection *wreplconn = talloc_get_type(conn->private, struct wreplsrv_in_connection);
- struct wreplsrv_in_call *call;
- DATA_BLOB packet_in_blob;
- DATA_BLOB packet_out_blob;
- struct wrepl_wrap packet_out_wrap;
- struct data_blob_list_item *rep;
- NTSTATUS status = NT_STATUS_UNSUCCESSFUL;
- size_t nread;
-
- /* avoid recursion, because of half async code */
- if (wreplconn->processing) {
- EVENT_FD_NOT_READABLE(conn->event.fde);
- return;
- }
-
- if (wreplconn->partial.length == 0) {
- wreplconn->partial = data_blob_talloc(wreplconn, NULL, 4);
- if (wreplconn->partial.data == NULL) {
- status = NT_STATUS_NO_MEMORY;
- goto failed;
- }
- wreplconn->partial_read = 0;
- }
-
- /* read in the packet length */
- if (wreplconn->partial_read < 4) {
- uint32_t packet_length;
-
- status = socket_recv(conn->socket,
- wreplconn->partial.data + wreplconn->partial_read,
- 4 - wreplconn->partial_read,
- &nread, 0);
- if (NT_STATUS_IS_ERR(status)) goto failed;
- if (!NT_STATUS_IS_OK(status)) return;
-
- wreplconn->partial_read += nread;
- if (wreplconn->partial_read != 4) return;
-
- packet_length = RIVAL(wreplconn->partial.data, 0) + 4;
-
- wreplconn->partial.data = talloc_realloc(wreplconn, wreplconn->partial.data,
- uint8_t, packet_length);
- if (wreplconn->partial.data == NULL) {
- status = NT_STATUS_NO_MEMORY;
- goto failed;
- }
- wreplconn->partial.length = packet_length;
- }
-
- /* read in the body */
- status = socket_recv(conn->socket,
- wreplconn->partial.data + wreplconn->partial_read,
- wreplconn->partial.length - wreplconn->partial_read,
- &nread, 0);
- if (NT_STATUS_IS_ERR(status)) goto failed;
- if (!NT_STATUS_IS_OK(status)) return;
-
- wreplconn->partial_read += nread;
- if (wreplconn->partial_read != wreplconn->partial.length) return;
-
- packet_in_blob.data = wreplconn->partial.data + 4;
- packet_in_blob.length = wreplconn->partial.length - 4;
-
- call = talloc_zero(wreplconn, struct wreplsrv_in_call);
- if (!call) {
- status = NT_STATUS_NO_MEMORY;
- goto failed;
- }
- call->wreplconn = wreplconn;
-
- /* we have a full request - parse it */
- status = ndr_pull_struct_blob(&packet_in_blob,
- call, &call->req_packet,
- (ndr_pull_flags_fn_t)ndr_pull_wrepl_packet);
- if (!NT_STATUS_IS_OK(status)) {
- DEBUG(2,("Failed to parse incoming WINS-Replication packet - %s\n",
- nt_errstr(status)));
- DEBUG(10,("packet length %u\n", wreplconn->partial.length));
- NDR_PRINT_DEBUG(wrepl_packet, &call->req_packet);
- goto failed;
- }
-
- /*
- * we have parsed the request, so we can reset the wreplconn->partial_read,
- * maybe we could also free wreplconn->partial, but for now we keep it,
- * and overwrite it the next time
- */
- wreplconn->partial_read = 0;
-
- if (DEBUGLVL(10)) {
- DEBUG(10,("Received WINS-Replication packet of length %u\n", wreplconn->partial.length));
- NDR_PRINT_DEBUG(wrepl_packet, &call->req_packet);
- }
-
- /* actually process the request */
- wreplconn->processing = True;
- status = wreplsrv_in_call(call);
- wreplconn->processing = False;
- if (NT_STATUS_IS_ERR(status)) goto failed;
- if (!NT_STATUS_IS_OK(status)) {
- /* w2k just ignores invalid packets, so we do */
- DEBUG(10,("Received WINS-Replication packet was invalid, we just ignore it\n"));
- talloc_free(call);
- return;
- }
-
- /* and now encode the reply */
- packet_out_wrap.packet = call->rep_packet;
- status = ndr_push_struct_blob(&packet_out_blob, call, &packet_out_wrap,
- (ndr_push_flags_fn_t)ndr_push_wrepl_wrap);
- if (!NT_STATUS_IS_OK(status)) goto failed;
-
- if (DEBUGLVL(10)) {
- DEBUG(10,("Sending WINS-Replication packet of length %d\n", (int)packet_out_blob.length));
- NDR_PRINT_DEBUG(wrepl_packet, &call->rep_packet);
- }
-
- rep = talloc(wreplconn, struct data_blob_list_item);
- if (!rep) {
- status = NT_STATUS_NO_MEMORY;
- goto failed;
- }
-
- rep->blob = packet_out_blob;
- talloc_steal(rep, packet_out_blob.data);
- /* we don't need the call anymore */
- talloc_free(call);
-
- if (!wreplconn->send_queue) {
- EVENT_FD_WRITEABLE(conn->event.fde);
- }
- DLIST_ADD_END(wreplconn->send_queue, rep, struct data_blob_list_item *);
-
- if (wreplconn->terminate) {
- EVENT_FD_NOT_READABLE(conn->event.fde);
- } else {
- EVENT_FD_READABLE(conn->event.fde);
- }
- return;
-
-failed:
- wreplsrv_terminate_connection(wreplconn, nt_errstr(status));
-}
-
-/*
- called when we can write to a connection
-*/
-static void wreplsrv_send(struct stream_connection *conn, uint16_t flags)
-{
- struct wreplsrv_in_connection *wreplconn = talloc_get_type(conn->private, struct wreplsrv_in_connection);
- NTSTATUS status;
-
- while (wreplconn->send_queue) {
- struct data_blob_list_item *rep = wreplconn->send_queue;
- size_t sendlen;
-
- status = socket_send(conn->socket, &rep->blob, &sendlen, 0);
- if (NT_STATUS_IS_ERR(status)) goto failed;
- if (!NT_STATUS_IS_OK(status)) return;
-
- rep->blob.length -= sendlen;
- rep->blob.data += sendlen;
-
- if (rep->blob.length == 0) {
- DLIST_REMOVE(wreplconn->send_queue, rep);
- talloc_free(rep);
- }
- }
-
- if (wreplconn->terminate) {
- wreplsrv_terminate_connection(wreplconn, "connection terminated after all pending packets are send");
- return;
- }
-
- EVENT_FD_NOT_WRITEABLE(conn->event.fde);
- return;
-
-failed:
- wreplsrv_terminate_connection(wreplconn, nt_errstr(status));
-}
-
-static const struct stream_server_ops wreplsrv_stream_ops = {
- .name = "wreplsrv",
- .accept_connection = wreplsrv_accept,
- .recv_handler = wreplsrv_recv,
- .send_handler = wreplsrv_send,
-};
-
-/*
- open winsdb
-*/
-static NTSTATUS wreplsrv_open_winsdb(struct wreplsrv_service *service)
-{
- service->wins_db = winsdb_connect(service);
- if (!service->wins_db) {
- return NT_STATUS_INTERNAL_DB_ERROR;
- }
-
- return NT_STATUS_OK;
-}
-
-/*
load our replication partners
*/
static NTSTATUS wreplsrv_load_partners(struct wreplsrv_service *service)
@@ -492,58 +258,6 @@ static NTSTATUS wreplsrv_setup_partners(struct wreplsrv_service *service)
}
/*
- startup the wrepl port 42 server sockets
-*/
-static NTSTATUS wreplsrv_setup_sockets(struct wreplsrv_service *service)
-{
- NTSTATUS status;
- struct task_server *task = service->task;
- const struct model_ops *model_ops;
- const char *address;
- uint16_t port = WINS_REPLICATION_PORT;
-
- /* within the wrepl task we want to be a single process, so
- ask for the single process model ops and pass these to the
- stream_setup_socket() call. */
- model_ops = process_model_byname("single");
- if (!model_ops) {
- DEBUG(0,("Can't find 'single' process model_ops"));
- return NT_STATUS_INTERNAL_ERROR;
- }
-
- if (lp_interfaces() && lp_bind_interfaces_only()) {
- int num_interfaces = iface_count();
- int i;
-
- /* We have been given an interfaces line, and been
- told to only bind to those interfaces. Create a
- socket per interface and bind to only these.
- */
- for(i = 0; i < num_interfaces; i++) {
- address = iface_n_ip(i);
- status = stream_setup_socket(task->event_ctx, model_ops, &wreplsrv_stream_ops,
- "ipv4", address, &port, service);
- if (!NT_STATUS_IS_OK(status)) {
- DEBUG(0,("stream_setup_socket(address=%s,port=%u) failed - %s\n",
- address, port, nt_errstr(status)));
- return status;
- }
- }
- } else {
- address = lp_socket_address();
- status = stream_setup_socket(task->event_ctx, model_ops, &wreplsrv_stream_ops,
- "ipv4", address, &port, service);
- if (!NT_STATUS_IS_OK(status)) {
- DEBUG(0,("stream_setup_socket(address=%s,port=%u) failed - %s\n",
- address, port, nt_errstr(status)));
- return status;
- }
- }
-
- return NT_STATUS_OK;
-}
-
-/*
startup the wrepl task
*/
static void wreplsrv_task_init(struct task_server *task)