#include "lib/messaging/irpc.h"
#include "librpc/gen_ndr/ndr_winsrepl.h"
#include "wrepl_server/wrepl_server.h"
-
-void wreplsrv_terminate_connection(struct wreplsrv_in_connection *wreplconn, const char *reason)
-{
- stream_terminate_connection(wreplconn->conn, reason);
-}
+#include "nbt_server/wins/winsdb.h"
+#include "ldb/include/ldb.h"
+#include "ldb/include/ldb_errors.h"
/*
- called when we get a new connection
+ open winsdb
*/
-static void wreplsrv_accept(struct stream_connection *conn)
+static NTSTATUS wreplsrv_open_winsdb(struct wreplsrv_service *service)
{
- struct wreplsrv_service *service = talloc_get_type(conn->private, struct wreplsrv_service);
- struct wreplsrv_in_connection *wreplconn;
-
- wreplconn = talloc_zero(conn, struct wreplsrv_in_connection);
- if (!wreplconn) {
- stream_terminate_connection(conn, "wreplsrv_accept: out of memory");
- return;
+ service->wins_db = winsdb_connect(service);
+ if (!service->wins_db) {
+ return NT_STATUS_INTERNAL_DB_ERROR;
}
- 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;
- }
+ /* the default renew interval is 6 days */
+ service->config.renew_interval = lp_parm_int(-1,"wreplsrv","renew_interval", 6*24*60*60);
+
+ /* the default tombstone (extinction) interval is 6 days */
+ service->config.tombstone_interval= lp_parm_int(-1,"wreplsrv","tombstone_interval", 6*24*60*60);
- /* TODO: find out if it's a partner */
+ /* the default tombstone (extinction) timeout is 1 day */
+ service->config.tombstone_timeout = lp_parm_int(-1,"wreplsrv","tombstone_timeout", 1*24*60*60);
- conn->private = wreplconn;
+ /* the default verify interval is 24 days */
+ service->config.verify_interval = lp_parm_int(-1,"wreplsrv","verify_interval", 24*24*60*60);
- irpc_add_name(conn->msg_ctx, "wreplsrv_connection");
+ /* the maximun interval to the next periodic processing event */
+ service->config.periodic_interval = lp_parm_int(-1,"wreplsrv","periodic_interval", 60);
+
+ return NT_STATUS_OK;
}
-/*
- receive some data on a WREPL connection
-*/
-static void wreplsrv_recv(struct stream_connection *conn, uint16_t flags)
+struct wreplsrv_partner *wreplsrv_find_partner(struct wreplsrv_service *service, const char *peer_addr)
{
- 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;
- }
+ struct wreplsrv_partner *cur;
- 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;
+ for (cur = service->partners; cur; cur = cur->next) {
+ if (strcmp(cur->address, peer_addr) == 0) {
+ return cur;
}
- 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;
+ return NULL;
+}
- wreplconn->partial_read += nread;
- if (wreplconn->partial_read != 4) return;
+/*
+ load our replication partners
+*/
+static NTSTATUS wreplsrv_load_partners(struct wreplsrv_service *service)
+{
+ struct ldb_result *res = NULL;
+ int ret;
+ TALLOC_CTX *tmp_ctx = talloc_new(service);
+ int i;
+
+ /* find the record in the WINS database */
+ ret = ldb_search(service->wins_db, ldb_dn_explode(tmp_ctx, "CN=PARTNERS"), LDB_SCOPE_ONELEVEL,
+ "(objectClass=wreplPartner)", NULL, &res);
+ if (ret != LDB_SUCCESS) goto failed;
+ talloc_steal(tmp_ctx, res);
+ if (res->count == 0) goto done;
+
+ for (i=0; i < res->count; i++) {
+ struct wreplsrv_partner *partner;
+
+ partner = talloc_zero(service, struct wreplsrv_partner);
+ if (partner == NULL) goto failed;
+
+ partner->service = service;
+ partner->address = ldb_msg_find_string(res->msgs[i], "address", NULL);
+ if (!partner->address) goto failed;
+ partner->name = ldb_msg_find_string(res->msgs[i], "name", partner->address);
+ partner->type = ldb_msg_find_uint(res->msgs[i], "type", WINSREPL_PARTNER_BOTH);
+ partner->pull.interval = ldb_msg_find_uint(res->msgs[i], "pullInterval",
+ WINSREPL_DEFAULT_PULL_INTERVAL);
+ partner->pull.retry_interval = ldb_msg_find_uint(res->msgs[i], "pullRetryInterval",
+ WINSREPL_DEFAULT_PULL_RETRY_INTERVAL);
+ partner->our_address = ldb_msg_find_string(res->msgs[i], "ourAddress", NULL);
+ partner->push.change_count = ldb_msg_find_uint(res->msgs[i], "pushChangeCount",
+ WINSREPL_DEFAULT_PUSH_CHANGE_COUNT);
+ partner->push.use_inform = ldb_msg_find_uint(res->msgs[i], "pushUseInform", False);
+
+ talloc_steal(partner, partner->address);
+ talloc_steal(partner, partner->name);
+ talloc_steal(partner, partner->our_address);
+
+ DLIST_ADD(service->partners, partner);
+ }
+done:
+ talloc_free(tmp_ctx);
+ return NT_STATUS_OK;
+failed:
+ talloc_free(tmp_ctx);
+ return NT_STATUS_FOOBAR;
+}
- packet_length = RIVAL(wreplconn->partial.data, 0) + 4;
+BOOL wreplsrv_is_our_address(struct wreplsrv_service *service, const char *address)
+{
+ const char *our_address;
- 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;
+ if (lp_interfaces() && lp_bind_interfaces_only()) {
+ int num_interfaces = iface_count();
+ int i;
+ for(i = 0; i < num_interfaces; i++) {
+ our_address = iface_n_ip(i);
+ if (strcasecmp(our_address, address) == 0) {
+ return True;
+ }
+ }
+ } else {
+ our_address = lp_socket_address();
+ if (strcasecmp(our_address, address) == 0) {
+ return True;
}
- 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;
+ return False;
+}
- call = talloc_zero(wreplconn, struct wreplsrv_in_call);
- if (!call) {
- status = NT_STATUS_NO_MEMORY;
- goto failed;
+uint64_t wreplsrv_local_max_version(struct wreplsrv_service *service)
+{
+ int ret;
+ struct ldb_context *ldb = service->wins_db;
+ struct ldb_dn *dn;
+ struct ldb_result *res = NULL;
+ TALLOC_CTX *tmp_ctx = talloc_new(service);
+ uint64_t maxVersion = 0;
+
+ dn = ldb_dn_explode(tmp_ctx, "CN=VERSION");
+ if (!dn) goto failed;
+
+ /* find the record in the WINS database */
+ ret = ldb_search(ldb, dn, LDB_SCOPE_BASE,
+ NULL, NULL, &res);
+ if (ret != LDB_SUCCESS) goto failed;
+ talloc_steal(tmp_ctx, res);
+ if (res->count > 1) goto failed;
+
+ if (res->count == 1) {
+ maxVersion = ldb_msg_find_uint64(res->msgs[0], "maxVersion", 0);
}
- 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;
- }
+failed:
+ talloc_free(tmp_ctx);
+ return maxVersion;
+}
- /*
- * 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;
+NTSTATUS wreplsrv_fill_wrepl_table(struct wreplsrv_service *service,
+ TALLOC_CTX *mem_ctx,
+ struct wrepl_table *table_out,
+ const char *our_ip,
+ const char *initiator,
+ BOOL full_table)
+{
+ struct wreplsrv_owner *cur;
+ uint64_t local_max_version;
+ uint32_t i = 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);
+ table_out->partner_count = 0;
+ table_out->partners = NULL;
+ table_out->initiator = initiator;
+
+ local_max_version = wreplsrv_local_max_version(service);
+ if (local_max_version > 0) {
+ table_out->partner_count++;
}
- /* 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;
+ for (cur = service->table; full_table && cur; cur = cur->next) {
+ table_out->partner_count++;
}
- /* 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;
+ table_out->partners = talloc_array(mem_ctx, struct wrepl_wins_owner, table_out->partner_count);
+ NT_STATUS_HAVE_NO_MEMORY(table_out->partners);
- 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);
+ if (local_max_version > 0) {
+ table_out->partners[i].address = our_ip;
+ table_out->partners[i].min_version = 0;
+ table_out->partners[i].max_version = local_max_version;
+ table_out->partners[i].type = 1;
+ i++;
}
- rep = talloc(wreplconn, struct data_blob_list_item);
- if (!rep) {
- status = NT_STATUS_NO_MEMORY;
- goto failed;
+ for (cur = service->table; full_table && cur; cur = cur->next) {
+ table_out->partners[i] = cur->owner;
+ i++;
}
- rep->blob = packet_out_blob;
- talloc_steal(rep, packet_out_blob.data);
- /* we don't need the call anymore */
- talloc_free(call);
+ return NT_STATUS_OK;
+}
- if (!wreplconn->send_queue) {
- EVENT_FD_WRITEABLE(conn->event.fde);
- }
- DLIST_ADD_END(wreplconn->send_queue, rep, struct data_blob_list_item *);
+struct wreplsrv_owner *wreplsrv_find_owner(struct wreplsrv_owner *table, const char *wins_owner)
+{
+ struct wreplsrv_owner *cur;
- if (wreplconn->terminate) {
- EVENT_FD_NOT_READABLE(conn->event.fde);
- } else {
- EVENT_FD_READABLE(conn->event.fde);
+ for (cur = table; cur; cur = cur->next) {
+ if (strcmp(cur->owner.address, wins_owner) == 0) {
+ return cur;
+ }
}
- return;
-failed:
- wreplsrv_terminate_connection(wreplconn, nt_errstr(status));
+ return NULL;
}
/*
- called when we can write to a connection
+ update the wins_owner_table max_version, if the given version is the highest version
+ if no entry for the wins_owner exists yet, create one
*/
-static void wreplsrv_send(struct stream_connection *conn, uint16_t flags)
+NTSTATUS wreplsrv_add_table(struct wreplsrv_service *service,
+ TALLOC_CTX *mem_ctx, struct wreplsrv_owner **_table,
+ const char *wins_owner, uint64_t version)
{
- struct wreplsrv_in_connection *wreplconn = talloc_get_type(conn->private, struct wreplsrv_in_connection);
- NTSTATUS status;
+ struct wreplsrv_owner *table = *_table;
+ struct wreplsrv_owner *cur;
- while (wreplconn->send_queue) {
- struct data_blob_list_item *rep = wreplconn->send_queue;
- size_t sendlen;
+ if (strcmp(WINSDB_OWNER_LOCAL, wins_owner) == 0) {
+ return NT_STATUS_OK;
+ }
- status = socket_send(conn->socket, &rep->blob, &sendlen, 0);
- if (NT_STATUS_IS_ERR(status)) goto failed;
- if (!NT_STATUS_IS_OK(status)) return;
+ cur = wreplsrv_find_owner(table, wins_owner);
- rep->blob.length -= sendlen;
- rep->blob.data += sendlen;
+ /* if it doesn't exists yet, create one */
+ if (!cur) {
+ cur = talloc_zero(mem_ctx, struct wreplsrv_owner);
+ NT_STATUS_HAVE_NO_MEMORY(cur);
- if (rep->blob.length == 0) {
- DLIST_REMOVE(wreplconn->send_queue, rep);
- talloc_free(rep);
- }
- }
+ cur->owner.address = talloc_strdup(cur, wins_owner);
+ NT_STATUS_HAVE_NO_MEMORY(cur->owner.address);
+ cur->owner.min_version = 0;
+ cur->owner.max_version = 0;
+ cur->owner.type = 1; /* don't know why this is always 1 */
- if (wreplconn->terminate) {
- wreplsrv_terminate_connection(wreplconn, "connection terminated after all pending packets are send");
- return;
- }
+ cur->partner = wreplsrv_find_partner(service, wins_owner);
- EVENT_FD_NOT_WRITEABLE(conn->event.fde);
- return;
-
-failed:
- wreplsrv_terminate_connection(wreplconn, nt_errstr(status));
-}
+ DLIST_ADD(table, cur);
+ *_table = table;
+ }
-static const struct stream_server_ops wreplsrv_stream_ops = {
- .name = "wreplsrv",
- .accept_connection = wreplsrv_accept,
- .recv_handler = wreplsrv_recv,
- .send_handler = wreplsrv_send,
-};
+ /* the min_version is always 0 here, and won't be updated */
-/*
- 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;
+ /* if the given version is higher the then current nax_version, update */
+ if (cur->owner.max_version < version) {
+ cur->owner.max_version = version;
}
return NT_STATUS_OK;
}
/*
- setup our replication partners
+ load the partner table
*/
-static NTSTATUS wreplsrv_setup_partners(struct wreplsrv_service *service)
+static NTSTATUS wreplsrv_load_table(struct wreplsrv_service *service)
{
+ struct ldb_result *res = NULL;
+ int ret;
+ NTSTATUS status;
+ TALLOC_CTX *tmp_ctx = talloc_new(service);
+ int i;
+ const char *wins_owner;
+ uint64_t version;
+ const char * const attrs[] = {
+ "winsOwner",
+ "versionID",
+ NULL
+ };
+
+ /* find the record in the WINS database */
+ ret = ldb_search(service->wins_db, NULL, LDB_SCOPE_SUBTREE,
+ "(objectClass=winsRecord)", attrs, &res);
+ status = NT_STATUS_INTERNAL_DB_CORRUPTION;
+ if (ret != LDB_SUCCESS) goto failed;
+ talloc_steal(tmp_ctx, res);
+ if (res->count == 0) goto done;
+
+ for (i=0; i < res->count; i++) {
+ wins_owner = ldb_msg_find_string(res->msgs[i], "winsOwner", NULL);
+ version = ldb_msg_find_uint64(res->msgs[i], "versionID", 0);
+
+ if (wins_owner) {
+ status = wreplsrv_add_table(service,
+ service, &service->table,
+ wins_owner, version);
+ if (!NT_STATUS_IS_OK(status)) goto failed;
+ }
+ talloc_free(res->msgs[i]);
+
+ /* TODO: what's abut the per address owners? */
+ }
+done:
+ talloc_free(tmp_ctx);
return NT_STATUS_OK;
+failed:
+ talloc_free(tmp_ctx);
+ return status;
}
/*
- startup the wrepl port 42 server sockets
+ setup our replication partners
*/
-static NTSTATUS wreplsrv_setup_sockets(struct wreplsrv_service *service)
+static NTSTATUS wreplsrv_setup_partners(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;
+ status = wreplsrv_load_partners(service);
+ NT_STATUS_NOT_OK_RETURN(status);
- /* 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, NULL);
- 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, NULL);
- 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;
- }
- }
+ status = wreplsrv_load_table(service);
+ NT_STATUS_NOT_OK_RETURN(status);
return NT_STATUS_OK;
}
return;
}
+ status = wreplsrv_setup_periodic(service);
+ if (!NT_STATUS_IS_OK(status)) {
+ task_server_terminate(task, "wreplsrv_task_init: wreplsrv_setup_periodic() failed");
+ return;
+ }
+
irpc_add_name(task->msg_ctx, "wrepl_server");
}
*/
static NTSTATUS wreplsrv_init(struct event_context *event_ctx, const struct model_ops *model_ops)
{
+ if (!lp_wins_support()) {
+ return NT_STATUS_OK;
+ }
+
return task_server_startup(event_ctx, model_ops, wreplsrv_task_init);
}