r11032: r10342@SERNOX: metze | 2005-09-20 11:00:26 +0200

 - load our wins partners at start time:
 # this is a sample partner record:
 dn: name=SERNOX4-1,CN=PARTNERS
 objectClass: wreplPartner
 name: SERNOX4-9
 address: 172.31.1.1
 ourAddress: 172.31.9.1
 type: 0x3
 pullInterval: 3600

 - go through all winsdb records in the database and create the wins_owner table,
   but don't add ourself to it as out nbt task will update the db too, we refetch
   the local max_versiion, each time we need it, (that typicaly onces per replication cycle)

 metze
(This used to be commit 4490a2864e21f957547c30675e5f096a4409e5fb)

3 files changed, 265 insertions, 10 deletions

diff --git a/source4/include/structs.h b/source4/include/structs.h
index 86cd3e1de8..faa2637f6f 100644
--- a/source4/include/structs.h
+++ b/source4/include/structs.h
@@ -272,6 +272,7 @@ struct mutex_ops;
 
 struct ads_struct;
 
+struct wreplsrv_service;
 struct wreplsrv_in_connection;
 struct wreplsrv_in_call;
 
diff --git a/source4/wrepl_server/wrepl_server.c b/source4/wrepl_server/wrepl_server.c
index c0b5e3c08d..7486c31ed8 100644
--- a/source4/wrepl_server/wrepl_server.c
+++ b/source4/wrepl_server/wrepl_server.c
@@ -29,12 +29,27 @@
 #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"
 
 void wreplsrv_terminate_connection(struct wreplsrv_in_connection *wreplconn, const char *reason)
 {
 	stream_terminate_connection(wreplconn->conn, reason);
 }
 
+static struct wreplsrv_partner *wreplsrv_find_partner(struct wreplsrv_service *service, const char *peer_addr)
+{
+	struct wreplsrv_partner *cur;
+
+	for (cur = service->partners; cur; cur = cur->next) {
+		if (strcmp(cur->address, peer_addr) == 0) {
+			return cur;
+		}
+	}
+
+	return NULL;
+}
+
 /*
   called when we get a new connection
 */
@@ -42,6 +57,7 @@ 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) {
@@ -57,7 +73,13 @@ static void wreplsrv_accept(struct stream_connection *conn)
 		return;
 	}
 
-	/* TODO: find out if it's a partner */
+	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;
 
@@ -271,10 +293,201 @@ static NTSTATUS wreplsrv_open_winsdb(struct wreplsrv_service *service)
 }
 
 /*
+  load our replication partners
+*/
+static NTSTATUS wreplsrv_load_partners(struct wreplsrv_service *service)
+{
+	struct ldb_message **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 (res != NULL) {
+		talloc_steal(tmp_ctx, res);
+	}
+	if (ret < 0) goto failed;
+	if (ret == 0) goto done;
+
+	for (i=0; i < ret; i++) {
+		struct wreplsrv_partner *partner;
+
+		partner = talloc(service, struct wreplsrv_partner);
+		if (partner == NULL) goto failed;
+
+		partner->address	= ldb_msg_find_string(res[i], "address", NULL);
+		if (!partner->address) goto failed;
+		partner->name		= ldb_msg_find_string(res[i], "name", partner->address);
+		partner->type		= ldb_msg_find_int(res[i], "type", WINSREPL_PARTNER_BOTH);
+		partner->pull.interval	= ldb_msg_find_int(res[i], "pullInterval", WINSREPL_DEFAULT_PULL_INTERVAL);
+		partner->our_address	= ldb_msg_find_string(res[i], "ourAddress", NULL);
+
+		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;
+}
+
+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_message **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 (res != NULL) {
+		talloc_steal(tmp_ctx, res);
+	}
+	if (ret < 0) goto failed;
+	if (ret > 1) goto failed;
+
+	if (ret == 1) {
+		maxVersion = ldb_msg_find_uint64(res[0], "maxVersion", 0);
+	}
+
+failed:
+	talloc_free(tmp_ctx);
+	return maxVersion;
+}
+
+static struct wreplsrv_owner *wreplsrv_find_owner(struct wreplsrv_owner *table, const char *wins_owner)
+{
+	struct wreplsrv_owner *cur;
+
+	for (cur = table; cur; cur = cur->next) {
+		if (strcmp(cur->owner.address, wins_owner) == 0) {
+			return cur;
+		}
+	}
+
+	return NULL;
+}
+
+/*
+ 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 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_owner *table = *_table;
+	struct wreplsrv_owner *cur;
+
+	if (strcmp(WINSDB_OWNER_LOCAL, wins_owner) == 0) {
+		return NT_STATUS_OK;
+	}
+
+	cur = wreplsrv_find_owner(table, wins_owner);
+
+	/* if it doesn't exists yet, create one */
+	if (!cur) {
+		cur = talloc_zero(mem_ctx, struct wreplsrv_owner);
+		NT_STATUS_HAVE_NO_MEMORY(cur);
+
+		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 */
+
+		cur->partner		= wreplsrv_find_partner(service, wins_owner);
+
+		DLIST_ADD(table, cur);
+		*_table = table;
+	}
+
+	/* the min_version is always 0 here, and won't be updated */
+
+	/* 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;
+}
+
+/*
+  load the partner table
+*/
+static NTSTATUS wreplsrv_load_table(struct wreplsrv_service *service)
+{
+	struct ldb_message **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",
+		"version",
+		NULL
+	};
+
+	/* find the record in the WINS database */
+	ret = ldb_search(service->wins_db, NULL, LDB_SCOPE_SUBTREE,
+			 "(objectClass=wins)", attrs, &res);
+	if (res != NULL) {
+		talloc_steal(tmp_ctx, res);
+	}
+	status = NT_STATUS_INTERNAL_DB_CORRUPTION;
+	if (ret < 0) goto failed;
+	if (ret == 0) goto done;
+
+	for (i=0; i < ret; i++) {
+		wins_owner     = ldb_msg_find_string(res[i], "winsOwner", NULL);
+		version        = ldb_msg_find_uint64(res[i], "version", 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[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;
+}
+
+/*
   setup our replication partners
 */
 static NTSTATUS wreplsrv_setup_partners(struct wreplsrv_service *service)
 {
+	NTSTATUS status;
+
+	status = wreplsrv_load_partners(service);
+	NT_STATUS_NOT_OK_RETURN(status);
+
+	status = wreplsrv_load_table(service);
+	NT_STATUS_NOT_OK_RETURN(status);
+
 	return NT_STATUS_OK;
 }
 
@@ -309,7 +522,7 @@ static NTSTATUS wreplsrv_setup_sockets(struct wreplsrv_service *service)
 		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);
+						     "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)));
@@ -319,7 +532,7 @@ static NTSTATUS wreplsrv_setup_sockets(struct wreplsrv_service *service)
 	} else {
 		address = lp_socket_address();
 		status = stream_setup_socket(task->event_ctx, model_ops, &wreplsrv_stream_ops,
-					     "ipv4", address, &port, NULL);
+					     "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)));
diff --git a/source4/wrepl_server/wrepl_server.h b/source4/wrepl_server/wrepl_server.h
index ce694d159c..2448d78bf8 100644
--- a/source4/wrepl_server/wrepl_server.h
+++ b/source4/wrepl_server/wrepl_server.h
@@ -24,8 +24,6 @@ struct wreplsrv_service;
 struct wreplsrv_in_connection;
 struct wreplsrv_out_connection;
 struct wreplsrv_partner;
-struct wreplsrv_pull_partner_item;
-struct wreplsrv_push_partner_item;
 
 #define WREPLSRV_VALID_ASSOC_CTX	0x12345678
 #define WREPLSRV_INVALID_ASSOC_CTX	0x0000000a
@@ -97,6 +95,52 @@ struct wreplsrv_out_connection {
 	struct wreplsrv_partner *partner;
 };
 
+enum winsrepl_partner_type {
+	WINSREPL_PARTNER_PULL = 0x1,
+	WINSREPL_PARTNER_PUSH = 0x2,
+	WINSREPL_PARTNER_BOTH = (WINSREPL_PARTNER_PULL | WINSREPL_PARTNER_PUSH)
+};
+
+#define WINSREPL_DEFAULT_PULL_INTERVAL (30*60)
+
+/*
+ this represents one of our configured partners
+*/
+struct wreplsrv_partner {
+	struct wreplsrv_partner *prev,*next;
+
+	/* the netbios name of the partner, mostly just for debugging */
+	const char *name;
+
+	/* the ip-address of the partner */
+	const char *address;
+
+	/* 
+	 * as wins partners identified by ip-address, we need to use a specific source-ip
+	 *  when we want to connect to the partner
+	 */
+	const char *our_address;
+
+	/* the type of the partner, pull, push or both */
+	enum winsrepl_partner_type type;
+
+	/* pull specific options */
+	struct {
+		/* the interval between 2 pull replications to the partner */
+		uint32_t interval;
+	} pull;
+};
+
+struct wreplsrv_owner {
+	struct wreplsrv_owner *prev,*next;
+
+	/* this hold the owner_id (address), min_version, max_version and partner_type */
+	struct wrepl_wins_owner owner;
+
+	/* can be NULL if this owner isn't a configure partner */
+	struct wreplsrv_partner *partner; 
+};
+
 /*
   state of the whole wrepl service
 */
@@ -113,9 +157,6 @@ struct wreplsrv_service {
 	/* all partners (pull and push) */
 	struct wreplsrv_partner *partners;
 
-	/* all pull partners */
-	struct wreplsrv_pull_partner *pull_partners;
-
-	/* all push partners */
-	struct wreplsrv_push_partner *push_partners;
+	/* this is a list of each wins_owner we know about in our database */
+	struct wreplsrv_owner *table;
 };