diff options
author | Stefan Metzmacher <metze@samba.org> | 2005-10-14 14:02:47 +0000 |
---|---|---|
committer | Gerald (Jerry) Carter <jerry@samba.org> | 2007-10-10 13:44:43 -0500 |
commit | cffd522b5c806508dfacfb10234e4c0a115c0a98 (patch) | |
tree | ec5271d813d2cb547757732815a434ba27a89582 /source4/nbt_server/wins | |
parent | d1e6c228692ff8b06d6eecd6be22fe0727e170ac (diff) | |
download | samba-cffd522b5c806508dfacfb10234e4c0a115c0a98.tar.gz samba-cffd522b5c806508dfacfb10234e4c0a115c0a98.tar.bz2 samba-cffd522b5c806508dfacfb10234e4c0a115c0a98.zip |
r11052: bring samba4 uptodate with the samba4-winsrepl branch,
before the bad merge
metze
(This used to be commit 471c0ca4abb17fb5f73c0efed195c67628c1c06e)
Diffstat (limited to 'source4/nbt_server/wins')
-rw-r--r-- | source4/nbt_server/wins/winsdb.c | 343 | ||||
-rw-r--r-- | source4/nbt_server/wins/winsdb.h | 12 | ||||
-rw-r--r-- | source4/nbt_server/wins/winsserver.c | 40 |
3 files changed, 83 insertions, 312 deletions
diff --git a/source4/nbt_server/wins/winsdb.c b/source4/nbt_server/wins/winsdb.c index 50baa50898..7ab8216837 100644 --- a/source4/nbt_server/wins/winsdb.c +++ b/source4/nbt_server/wins/winsdb.c @@ -33,6 +33,7 @@ */ static uint64_t winsdb_allocate_version(struct wins_server *winssrv) { + int trans; int ret; struct ldb_context *ldb = winssrv->wins_db; struct ldb_dn *dn; @@ -41,12 +42,21 @@ static uint64_t winsdb_allocate_version(struct wins_server *winssrv) TALLOC_CTX *tmp_ctx = talloc_new(winssrv); uint64_t maxVersion = 0; - dn = ldb_dn_explode(tmp_ctx, "CN=VERSION"); - if (!dn) goto failed; + trans = ldb_transaction_start(ldb); + if (trans != LDB_SUCCESS) goto failed; 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); } @@ -57,22 +67,27 @@ static uint64_t winsdb_allocate_version(struct wins_server *winssrv) msg->dn = dn; - ret = ldb_msg_add_empty(ldb, msg, "maxVersion", LDB_FLAG_MOD_REPLACE); + ret = ldb_msg_add_empty(msg, "objectClass", LDB_FLAG_MOD_REPLACE); + if (ret != 0) goto failed; + ret = ldb_msg_add_string(msg, "objectClass", "winsMaxVersion"); + if (ret != 0) goto failed; + ret = ldb_msg_add_empty(msg, "maxVersion", LDB_FLAG_MOD_REPLACE); if (ret != 0) goto failed; - ret = ldb_msg_add_fmt(ldb, msg, "maxVersion", "%llu", maxVersion); + ret = ldb_msg_add_fmt(msg, "maxVersion", "%llu", maxVersion); if (ret != 0) goto failed; ret = ldb_modify(ldb, msg); if (ret != 0) ret = ldb_add(ldb, msg); if (ret != 0) goto failed; - talloc_free(tmp_ctx); - return maxVersion; + trans = ldb_transaction_commit(ldb); + if (trans != LDB_SUCCESS) goto failed; talloc_free(tmp_ctx); return maxVersion; failed: + if (trans == LDB_SUCCESS) ldb_transaction_cancel(ldb); talloc_free(tmp_ctx); return 0; } @@ -172,177 +187,11 @@ static NTSTATUS winsdb_addr_decode(struct winsdb_record *rec, struct ldb_val *va if (!p) { /* support old entries, with only the address */ addr->address = talloc_steal(addr, val->data); - addr->wins_owner = rec->wins_owner; - addr->expire_time = rec->expire_time; - *_addr = addr; - return NT_STATUS_OK; - } - - *p = '\0';p++; - addr->address = talloc_strdup(addr, address); - if (!addr->address) { - status = NT_STATUS_NO_MEMORY; - goto failed; - } - - if (strncmp("winsOwner:", p, 10) != 0) { - status = NT_STATUS_INTERNAL_DB_CORRUPTION; - goto failed; - } - wins_owner = p + 10; - p = strchr(wins_owner, ';'); - if (!p) { - status = NT_STATUS_INTERNAL_DB_CORRUPTION; - goto failed; - } - - *p = '\0';p++; - addr->wins_owner = talloc_strdup(addr, wins_owner); - if (!addr->wins_owner) { - status = NT_STATUS_NO_MEMORY; - goto failed; - } - - if (strncmp("expireTime:", p, 11) != 0) { - status = NT_STATUS_INTERNAL_DB_CORRUPTION; - goto failed; - } - - expire_time = p + 11; - p = strchr(expire_time, ';'); - if (!p) { - status = NT_STATUS_INTERNAL_DB_CORRUPTION; - goto failed; - } - - *p = '\0';p++; - addr->expire_time = ldap_string_to_time(expire_time); - - *_addr = addr; - return NT_STATUS_OK; -failed: - talloc_free(addr); - return status; -} - -/* - encode the winsdb_addr("address") attribute like this: - "172.31.1.1;winsOwner:172.31.9.202;expireTime:20050923032330.0Z" -*/ -static int ldb_msg_add_winsdb_addr(struct ldb_context *ldb, struct ldb_message *msg, - const char *attr_name, struct winsdb_addr *addr) -{ - struct ldb_val val; - const char *str; - - dn = ldb_dn_string_compose(mem_ctx, NULL, "type=%02x", name->type); - - addresses[len]->address = talloc_strdup(addresses[len], address); - if (!addresses[len]->address) { - talloc_free(addresses); - return NULL; - } - - addresses[len]->wins_owner = talloc_strdup(addresses[len], wins_owner); - if (!addresses[len]->wins_owner) { - talloc_free(addresses); - return NULL; - } - - addresses[len]->expire_time = expire_time; - - addresses[len+1] = NULL; - - return addresses; -} - -void winsdb_addr_list_remove(struct winsdb_addr **addresses, const char *address) -{ - size_t i; - - for (i=0; addresses[i]; i++) { - if (strcmp(addresses[i]->address, address) == 0) { - break; - } - } - if (!addresses[i]) return; - - for (; addresses[i]; i++) { - addresses[i] = addresses[i+1]; - } - - return; -} - -struct winsdb_addr *winsdb_addr_list_check(struct winsdb_addr **addresses, const char *address) -{ - size_t i; - - for (i=0; addresses[i]; i++) { - if (strcmp(addresses[i]->address, address) == 0) { - return addresses[i]; - } - } - - return NULL; -} - -size_t winsdb_addr_list_length(struct winsdb_addr **addresses) -{ - size_t i; - for (i=0; addresses[i]; i++); - return i; -} - -const char **winsdb_addr_string_list(TALLOC_CTX *mem_ctx, struct winsdb_addr **addresses) -{ - size_t len = winsdb_addr_list_length(addresses); - const char **str_list; - size_t i; - - str_list = talloc_array(mem_ctx, const char *, len + 1); - if (!str_list) return NULL; - - for (i=0; i < len; i++) { - str_list[i] = talloc_strdup(str_list, addresses[i]->address); - if (!str_list[i]) { - talloc_free(str_list); - return NULL; + addr->wins_owner = talloc_reference(addr, rec->wins_owner); + if (!addr->wins_owner) { + status = NT_STATUS_NO_MEMORY; + goto failed; } - } - - str_list[len] = NULL; - return str_list; -} - -/* - decode the winsdb_addr("address") attribute: - "172.31.1.1" or - "172.31.1.1;winsOwner:172.31.9.202;expireTime:20050923032330.0Z" - are valid records -*/ -static struct winsdb_addr *winsdb_addr_decode(struct winsdb_record *rec, TALLOC_CTX *mem_ctx, struct ldb_val *val) -{ - NTSTATUS status; - struct winsdb_addr *addr; - char *address; - char *wins_owner; - char *expire_time; - char *p; - - addr = talloc(mem_ctx, struct winsdb_addr); - if (!addr) { - status = NT_STATUS_NO_MEMORY; - goto failed; - } - - address = (char *)val->data; - - p = strchr(address, ';'); - if (!p) { - /* support old entries, with only the address */ - addr->address = talloc_steal(addr, val->data); - addr->wins_owner = rec->wins_owner; addr->expire_time = rec->expire_time; *_addr = addr; return NT_STATUS_OK; @@ -386,7 +235,7 @@ static struct winsdb_addr *winsdb_addr_decode(struct winsdb_record *rec, TALLOC_ } *p = '\0';p++; - addr->expire_time = ldap_string_to_time(expire_time); + addr->expire_time = ldb_string_to_time(expire_time); *_addr = addr; return NT_STATUS_OK; @@ -399,7 +248,7 @@ failed: encode the winsdb_addr("address") attribute like this: "172.31.1.1;winsOwner:172.31.9.202;expireTime:20050923032330.0Z;" */ -static int ldb_msg_add_winsdb_addr(struct ldb_context *ldb, struct ldb_message *msg, +static int ldb_msg_add_winsdb_addr(struct ldb_message *msg, const char *attr_name, struct winsdb_addr *addr) { struct ldb_val val; @@ -407,13 +256,13 @@ static int ldb_msg_add_winsdb_addr(struct ldb_context *ldb, struct ldb_message * str = talloc_asprintf(msg, "%s;winsOwner:%s;expireTime:%s;", addr->address, addr->wins_owner, - ldap_timestring(msg, addr->expire_time)); + ldb_timestring(msg, addr->expire_time)); if (!str) return -1; val.data = discard_const_p(uint8_t, str); val.length = strlen(str); - return ldb_msg_add_value(ldb, msg, attr_name, &val); + return ldb_msg_add_value(msg, attr_name, &val); } struct winsdb_addr **winsdb_addr_list_make(TALLOC_CTX *mem_ctx) @@ -552,11 +401,11 @@ NTSTATUS winsdb_lookup(struct ldb_context *wins_db, if (!NT_STATUS_IS_OK(status)) goto failed; /* see if it has already expired */ - if (rec->state == WINS_REC_ACTIVE && + if (rec->state == WREPL_STATE_ACTIVE && rec->expire_time <= time(NULL)) { DEBUG(5,("WINS: expiring name %s (expired at %s)\n", nbt_name_string(tmp_ctx, rec->name), timestring(tmp_ctx, rec->expire_time))); - rec->state = WINS_REC_RELEASED; + rec->state = WREPL_STATE_RELEASED; } talloc_steal(mem_ctx, rec); @@ -587,69 +436,35 @@ NTSTATUS winsdb_record(struct ldb_message *msg, struct nbt_name *name, TALLOC_CT if (!NT_STATUS_IS_OK(status)) goto failed; } - /* parse it into a more convenient winsdb_record structure */ - rec->name = name; - rec->state = ldb_msg_find_int(res[0], "active", WINS_REC_RELEASED); - rec->nb_flags = ldb_msg_find_int(res[0], "nbFlags", 0); - rec->expire_time = ldb_string_to_time(ldb_msg_find_string(res[0], "expires", NULL)); - rec->registered_by = ldb_msg_find_string(res[0], "registeredBy", NULL); - rec->version = ldb_msg_find_uint64(res[0], "version", 0); - talloc_steal(rec, rec->wins_owner); - talloc_steal(rec, rec->registered_by); - - if (!rec->wins_owner) rec->wins_owner = WINSDB_OWNER_LOCAL; - - el = ldb_msg_find_element(msg, "address"); - if (el == NULL) { + if (strlen(name->name) > 15) { status = NT_STATUS_INTERNAL_DB_CORRUPTION; goto failed; } - - rec->addresses = talloc_array(rec, struct winsdb_addr *, el->num_values+1); - if (rec->addresses == NULL) { - status = NT_STATUS_NO_MEMORY; - goto failed; - } - - for (i=0;i<el->num_values;i++) { - status = winsdb_addr_decode(rec, &el->values[i], rec->addresses, &rec->addresses[i]); - if (!NT_STATUS_IS_OK(status)) goto failed; - } - rec->addresses[i] = NULL; - - *_rec = rec; - return NT_STATUS_OK; -failed: - if (NT_STATUS_EQUAL(NT_STATUS_INTERNAL_DB_CORRUPTION, status)) { - DEBUG(1,("winsdb_record: corrupted record: %s\n", ldb_dn_linearize(rec, msg->dn))); - } - talloc_free(rec); - return status; -} - - rec = talloc(mem_ctx, struct winsdb_record); - if (rec == NULL) { - status = NT_STATUS_NO_MEMORY; + if (name->scope && strlen(name->scope) > 238) { + status = NT_STATUS_INTERNAL_DB_CORRUPTION; goto failed; } - if (!name) { - status = winsdb_nbt_name(rec, msg->dn, &name); - if (!NT_STATUS_IS_OK(status)) goto failed; - } - /* parse it into a more convenient winsdb_record structure */ - rec->name = name; - rec->state = ldb_msg_find_int(msg, "state", WINS_REC_RELEASED); - rec->nb_flags = ldb_msg_find_int(msg, "nbFlags", 0); - rec->wins_owner = ldb_msg_find_string(msg, "winsOwner", NULL); - rec->expire_time = ldap_string_to_time(ldb_msg_find_string(msg, "expireTime", NULL)); - rec->registered_by = ldb_msg_find_string(msg, "registeredBy", NULL); - rec->version = ldb_msg_find_uint64(msg, "versionID", 0); + rec->name = name; + rec->type = ldb_msg_find_int(msg, "recordType", WREPL_TYPE_UNIQUE); + rec->state = ldb_msg_find_int(msg, "recordState", WREPL_STATE_RELEASED); + rec->node = ldb_msg_find_int(msg, "nodeType", WREPL_NODE_B); + rec->is_static = ldb_msg_find_int(msg, "isStatic", 0); + rec->expire_time = ldb_string_to_time(ldb_msg_find_string(msg, "expireTime", NULL)); + rec->version = ldb_msg_find_uint64(msg, "versionID", 0); + rec->wins_owner = ldb_msg_find_string(msg, "winsOwner", NULL); + rec->registered_by = ldb_msg_find_string(msg, "registeredBy", NULL); talloc_steal(rec, rec->wins_owner); talloc_steal(rec, rec->registered_by); - if (!rec->wins_owner) rec->wins_owner = WINSDB_OWNER_LOCAL; + if (!rec->wins_owner) { + rec->wins_owner = talloc_strdup(rec, WINSDB_OWNER_LOCAL); + if (rec->wins_owner == NULL) { + status = NT_STATUS_NO_MEMORY; + goto failed; + } + } el = ldb_msg_find_element(msg, "address"); if (el == NULL) { @@ -657,56 +472,11 @@ failed: goto failed; } - rec->addresses = talloc_array(rec, struct winsdb_addr *, el->num_values+1); - if (rec->addresses == NULL) { - status = NT_STATUS_NO_MEMORY; - goto failed; - } - - for (i=0;i<el->num_values;i++) { - status = winsdb_addr_decode(rec, &el->values[i], rec->addresses, &rec->addresses[i]); - if (!NT_STATUS_IS_OK(status)) goto failed; - } - rec->addresses[i] = NULL; - - *_rec = rec; - return NT_STATUS_OK; -failed: - if (NT_STATUS_EQUAL(NT_STATUS_INTERNAL_DB_CORRUPTION, status)) { - DEBUG(1,("winsdb_record: corrupted record: %s\n", ldb_dn_linearize(rec, msg->dn))); - } - talloc_free(rec); - return status; -} - - rec = talloc(mem_ctx, struct winsdb_record); - if (rec == NULL) { - status = NT_STATUS_NO_MEMORY; - goto failed; - } - - if (!name) { - status = winsdb_nbt_name(rec, msg->dn, &name); - if (!NT_STATUS_IS_OK(status)) goto failed; - } - - /* parse it into a more convenient winsdb_record structure */ - rec->name = name; - rec->state = ldb_msg_find_int(msg, "state", WINS_REC_RELEASED); - rec->nb_flags = ldb_msg_find_int(msg, "nbFlags", 0); - rec->wins_owner = ldb_msg_find_string(msg, "winsOwner", NULL); - rec->expire_time = ldap_string_to_time(ldb_msg_find_string(msg, "expireTime", NULL)); - rec->registered_by = ldb_msg_find_string(msg, "registeredBy", NULL); - rec->version = ldb_msg_find_uint64(msg, "versionID", 0); - talloc_steal(rec, rec->wins_owner); - talloc_steal(rec, rec->registered_by); - - if (!rec->wins_owner) rec->wins_owner = WINSDB_OWNER_LOCAL; - - el = ldb_msg_find_element(msg, "address"); - if (el == NULL) { - status = NT_STATUS_INTERNAL_DB_CORRUPTION; - goto failed; + if (rec->type == WREPL_TYPE_UNIQUE || rec->type == WREPL_TYPE_GROUP) { + if (el->num_values != 1) { + status = NT_STATUS_INTERNAL_DB_CORRUPTION; + goto failed; + } } rec->addresses = talloc_array(rec, struct winsdb_addr *, el->num_values+1); @@ -853,6 +623,9 @@ uint8_t winsdb_delete(struct wins_server *winssrv, struct winsdb_record *rec) int trans; int ret; + trans = ldb_transaction_start(ldb); + if (trans != LDB_SUCCESS) goto failed; + dn = winsdb_dn(tmp_ctx, rec->name); if (dn == NULL) goto failed; diff --git a/source4/nbt_server/wins/winsdb.h b/source4/nbt_server/wins/winsdb.h index 2ac1884063..d93d50bdf7 100644 --- a/source4/nbt_server/wins/winsdb.h +++ b/source4/nbt_server/wins/winsdb.h @@ -29,15 +29,6 @@ struct winsdb_addr { time_t expire_time; }; -#define WINSDB_OWNER_LOCAL "0.0.0.0" -#define WINSDB_GROUP_ADDRESS "255.255.255.255" - -struct winsdb_addr { - const char *address; - const char *wins_owner; - time_t expire_time; -}; - /* each record in the database contains the following information */ @@ -47,7 +38,6 @@ struct winsdb_record { enum wrepl_name_state state; enum wrepl_name_node node; BOOL is_static; - const char *wins_owner; time_t expire_time; uint64_t version; const char *wins_owner; @@ -55,8 +45,6 @@ struct winsdb_record { /* only needed for debugging problems */ const char *registered_by; - struct winsdb_addr **addresses; - uint64_t version; }; struct wins_server { diff --git a/source4/nbt_server/wins/winsserver.c b/source4/nbt_server/wins/winsserver.c index 0f6717f4a9..38540b87ad 100644 --- a/source4/nbt_server/wins/winsserver.c +++ b/source4/nbt_server/wins/winsserver.c @@ -70,13 +70,22 @@ static uint8_t wins_register_new(struct nbt_name_socket *nbtsock, enum wrepl_name_node node; BOOL mhomed = ((packet->operation & NBT_OPCODE) == NBT_OPCODE_MULTI_HOME_REG); - rec.name = name; - rec.nb_flags = nb_flags; - rec.state = WINS_REC_ACTIVE; - rec.wins_owner = WINSDB_OWNER_LOCAL; - rec.expire_time = time(NULL) + ttl; - rec.registered_by = src->addr; - rec.addresses = winsdb_addr_list_make(packet); +#define WREPL_NODE_NBT_FLAGS(nb_flags) \ + ((nb_flags & NBT_NM_OWNER_TYPE)>>13) + + type = wrepl_type(nb_flags, name, mhomed); + node = WREPL_NODE_NBT_FLAGS(nb_flags); + + rec.name = name; + rec.type = type; + rec.state = WREPL_STATE_ACTIVE; + rec.node = node; + rec.is_static = False; + rec.expire_time = time(NULL) + ttl; + rec.version = 0; /* will allocated later */ + rec.wins_owner = WINSDB_OWNER_LOCAL; + rec.registered_by = src->addr; + rec.addresses = winsdb_addr_list_make(packet); if (rec.addresses == NULL) return NBT_RCODE_SVR; rec.addresses = winsdb_addr_list_add(rec.addresses, @@ -148,7 +157,7 @@ static void nbtd_winsserver_register(struct nbt_name_socket *nbtsock, } else if (!NT_STATUS_IS_OK(status)) { rcode = NBT_RCODE_SVR; goto done; - } else if (rec->state != WINS_REC_ACTIVE) { + } else if (rec->state != WREPL_STATE_ACTIVE) { winsdb_delete(winssrv, rec); rcode = wins_register_new(nbtsock, packet, src); goto done; @@ -164,14 +173,14 @@ static void nbtd_winsserver_register(struct nbt_name_socket *nbtsock, /* if its an active unique name, and the registration is for a group, then see if the unique name owner still wants the name */ - if (!(rec->nb_flags & NBT_NM_GROUP) && (nb_flags & NBT_NM_GROUP)) { + if (!(rec->type == WREPL_TYPE_GROUP) && (nb_flags & NBT_NM_GROUP)) { wins_register_wack(nbtsock, packet, rec, src); return; } /* if the registration is for a group, then just update the expiry time and we are done */ - if (IS_GROUP_NAME(name, nb_flags)) { + if (nb_flags & NBT_NM_GROUP) { wins_update_ttl(nbtsock, packet, rec, src); goto done; } @@ -217,14 +226,15 @@ static void nbtd_winsserver_query(struct nbt_name_socket *nbtsock, struct nbt_name *name = &packet->questions[0].name; struct winsdb_record *rec; const char **addresses; + uint16_t nb_flags = 0; /* TODO: ... */ status = winsdb_lookup(winssrv->wins_db, name, packet, &rec); - if (!NT_STATUS_IS_OK(status) || rec->state != WINS_REC_ACTIVE) { + if (!NT_STATUS_IS_OK(status) || rec->state != WREPL_STATE_ACTIVE) { nbtd_negative_name_query_reply(nbtsock, packet, src); return; } - if (IS_GROUP_NAME(name, rec->nb_flags)) { + if (rec->type == WREPL_TYPE_GROUP) { addresses = talloc_array(packet, const char *, 2); if (addresses == NULL) { nbtd_negative_name_query_reply(nbtsock, packet, src); @@ -241,7 +251,7 @@ static void nbtd_winsserver_query(struct nbt_name_socket *nbtsock, } nbtd_name_query_reply(nbtsock, packet, src, name, - 0, rec->nb_flags, addresses); + 0, nb_flags, addresses); } /* @@ -260,8 +270,8 @@ static void nbtd_winsserver_release(struct nbt_name_socket *nbtsock, status = winsdb_lookup(winssrv->wins_db, name, packet, &rec); if (!NT_STATUS_IS_OK(status) || - rec->state != WINS_REC_ACTIVE || - IS_GROUP_NAME(name, rec->nb_flags)) { + rec->state != WREPL_STATE_ACTIVE || + rec->type == WREPL_TYPE_GROUP) { goto done; } |