From 51baffab5fffa2b50f540ffd0053d957172dfcb0 Mon Sep 17 00:00:00 2001 From: Andrew Tridgell Date: Wed, 9 Sep 2009 21:00:01 +1000 Subject: s4:drs split addentry and getncchanges into separate files These will get quite complex eventually, I think we are better separating them so the code is a bit easier to follow --- source4/rpc_server/config.mk | 2 + source4/rpc_server/drsuapi/addentry.c | 81 ++++++++ source4/rpc_server/drsuapi/dcesrv_drsuapi.c | 292 +--------------------------- source4/rpc_server/drsuapi/dcesrv_drsuapi.h | 12 ++ source4/rpc_server/drsuapi/drsutil.c | 63 +++++- source4/rpc_server/drsuapi/getncchanges.c | 215 ++++++++++++++++++++ source4/rpc_server/drsuapi/updaterefs.c | 2 +- 7 files changed, 375 insertions(+), 292 deletions(-) create mode 100644 source4/rpc_server/drsuapi/addentry.c create mode 100644 source4/rpc_server/drsuapi/getncchanges.c diff --git a/source4/rpc_server/config.mk b/source4/rpc_server/config.mk index d5aff844c1..f60f83327b 100644 --- a/source4/rpc_server/config.mk +++ b/source4/rpc_server/config.mk @@ -185,6 +185,8 @@ PRIVATE_DEPENDENCIES = \ dcerpc_drsuapi_OBJ_FILES = $(rpc_serversrcdir)/drsuapi/dcesrv_drsuapi.o \ $(rpc_serversrcdir)/drsuapi/updaterefs.o \ + $(rpc_serversrcdir)/drsuapi/getncchanges.o \ + $(rpc_serversrcdir)/drsuapi/addentry.o \ $(rpc_serversrcdir)/drsuapi/drsutil.o ################################################ diff --git a/source4/rpc_server/drsuapi/addentry.c b/source4/rpc_server/drsuapi/addentry.c new file mode 100644 index 0000000000..2667922f3f --- /dev/null +++ b/source4/rpc_server/drsuapi/addentry.c @@ -0,0 +1,81 @@ +/* + Unix SMB/CIFS implementation. + + implement the DsAddEntry call + + Copyright (C) Stefan Metzmacher 2009 + Copyright (C) Andrew Tridgell 2009 + + 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 3 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, see . +*/ + +#include "includes.h" +#include "librpc/gen_ndr/ndr_drsuapi.h" +#include "rpc_server/dcerpc_server.h" +#include "rpc_server/common/common.h" +#include "dsdb/samdb/samdb.h" +#include "lib/ldb/include/ldb_errors.h" +#include "param/param.h" +#include "librpc/gen_ndr/ndr_drsblobs.h" +#include "auth/auth.h" +#include "rpc_server/drsuapi/dcesrv_drsuapi.h" + +/* + drsuapi_DsAddEntry +*/ +WERROR dcesrv_drsuapi_DsAddEntry(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, + struct drsuapi_DsAddEntry *r) +{ + WERROR status; + struct drsuapi_bind_state *b_state; + struct dcesrv_handle *h; + uint32_t num = 0; + struct drsuapi_DsReplicaObjectIdentifier2 *ids = NULL; + + if (DEBUGLVL(4)) { + NDR_PRINT_FUNCTION_DEBUG(drsuapi_DsAddEntry, NDR_IN, r); + } + + /* TODO: check which out level the client supports */ + + ZERO_STRUCTP(r->out.ctr); + r->out.level_out = 3; + r->out.ctr->ctr3.level = 1; + r->out.ctr->ctr3.error = talloc_zero(mem_ctx, union drsuapi_DsAddEntryError); + + DCESRV_PULL_HANDLE_WERR(h, r->in.bind_handle, DRSUAPI_BIND_HANDLE); + b_state = h->data; + + switch (r->in.level) { + case 2: + status = dsdb_origin_objects_commit(b_state->sam_ctx, + mem_ctx, + &r->in.req->req2.first_object, + &num, + &ids); + if (!W_ERROR_IS_OK(status)) { + r->out.ctr->ctr3.error->info1.status = status; + W_ERROR_NOT_OK_RETURN(status); + } + + r->out.ctr->ctr3.count = num; + r->out.ctr->ctr3.objects = ids; + + return WERR_OK; + default: + break; + } + + return WERR_FOOBAR; +} diff --git a/source4/rpc_server/drsuapi/dcesrv_drsuapi.c b/source4/rpc_server/drsuapi/dcesrv_drsuapi.c index 73cc8cb1ef..8ea7defe3b 100644 --- a/source4/rpc_server/drsuapi/dcesrv_drsuapi.c +++ b/source4/rpc_server/drsuapi/dcesrv_drsuapi.c @@ -24,12 +24,12 @@ #include "librpc/gen_ndr/ndr_drsuapi.h" #include "rpc_server/dcerpc_server.h" #include "rpc_server/common/common.h" -#include "rpc_server/drsuapi/dcesrv_drsuapi.h" #include "dsdb/samdb/samdb.h" #include "lib/ldb/include/ldb_errors.h" #include "param/param.h" #include "librpc/gen_ndr/ndr_drsblobs.h" #include "messaging/irpc.h" +#include "rpc_server/drsuapi/dcesrv_drsuapi.h" /* drsuapi_DsBind @@ -258,250 +258,6 @@ static WERROR dcesrv_drsuapi_DsReplicaSync(struct dcesrv_call_state *dce_call, T return WERR_OK; } -int drsuapi_search_with_extended_dn(struct ldb_context *ldb, - TALLOC_CTX *mem_ctx, - struct ldb_result **_res, - struct ldb_dn *basedn, - enum ldb_scope scope, - const char * const *attrs, - const char *format, ...) -{ - va_list ap; - int ret; - struct ldb_request *req; - char *filter; - TALLOC_CTX *tmp_ctx; - struct ldb_result *res; - - tmp_ctx = talloc_new(mem_ctx); - - res = talloc_zero(tmp_ctx, struct ldb_result); - if (!res) { - return LDB_ERR_OPERATIONS_ERROR; - } - - va_start(ap, format); - filter = talloc_vasprintf(tmp_ctx, format, ap); - va_end(ap); - - if (filter == NULL) { - talloc_free(tmp_ctx); - return LDB_ERR_OPERATIONS_ERROR; - } - - ret = ldb_build_search_req(&req, ldb, tmp_ctx, - basedn, - scope, - filter, - attrs, - NULL, - res, - ldb_search_default_callback, - NULL); - if (ret != LDB_SUCCESS) { - talloc_free(tmp_ctx); - return ret; - } - - ret = ldb_request_add_control(req, LDB_CONTROL_EXTENDED_DN_OID, true, NULL); - if (ret != LDB_SUCCESS) { - return ret; - } - - ret = ldb_request(ldb, req); - if (ret == LDB_SUCCESS) { - ret = ldb_wait(req->handle, LDB_WAIT_ALL); - } - - talloc_free(req); - *_res = res; - return ret; -} - - -/* - drsuapi_DsGetNCChanges -*/ -static WERROR dcesrv_drsuapi_DsGetNCChanges(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, - struct drsuapi_DsGetNCChanges *r) -{ - struct ldb_result *site_res; - struct drsuapi_DsReplicaObjectIdentifier *ncRoot; - struct drsuapi_bind_state *b_state; - struct ldb_dn *ncRoot_dn; - int ret; - int i; - int j; - int uSN; - struct dsdb_schema *schema; - struct drsuapi_DsReplicaOIDMapping_Ctr *ctr; - time_t t = time(NULL); - NTTIME now; - struct drsuapi_DsReplicaObjectListItemEx *currentObject; - struct dom_sid *zero_sid; - struct ldb_dn *obj_dn; - enum ndr_err_code ndr_err; - const struct ldb_val *md_value; - struct replPropertyMetaDataBlob md; - ZERO_STRUCT(md); - md.version = 1; - b_state = talloc_zero(mem_ctx, struct drsuapi_bind_state); - W_ERROR_HAVE_NO_MEMORY(b_state); - zero_sid = talloc_zero(mem_ctx, struct dom_sid); - /* - * connect to the samdb - */ - b_state->sam_ctx = samdb_connect(b_state, dce_call->event_ctx, dce_call->conn->dce_ctx->lp_ctx, dce_call->conn->auth_state.session_info); - if (!b_state->sam_ctx) { - return WERR_FOOBAR; - } - - /* Check request revision. */ - if (r->in.level != 8) { - return WERR_REVISION_MISMATCH; - } - - /* Perform access checks. */ - if (r->in.req->req8.naming_context == NULL) { - return WERR_DS_DRA_INVALID_PARAMETER; - } - - ncRoot = r->in.req->req8.naming_context; - if (ncRoot == NULL) { - return WERR_DS_DRA_BAD_NC; - } - - DEBUG(4,("DsGetNSChanges with uSHChanged >= %llu\n", - (unsigned long long)r->in.req->req8.highwatermark.highest_usn)); - - /* Construct response. */ - ncRoot_dn = ldb_dn_new(mem_ctx, b_state->sam_ctx, ncRoot->dn); - ret = drsuapi_search_with_extended_dn(b_state->sam_ctx, mem_ctx, &site_res, - ncRoot_dn, LDB_SCOPE_SUBTREE, NULL, - "(&(uSNChanged>=%lld)(objectClass=*))", r->in.req->req8.highwatermark.highest_usn); - if (ret != LDB_SUCCESS) { - return WERR_DS_DRA_INTERNAL_ERROR; - } - - *r->out.level_out = 6; - r->out.ctr->ctr6.naming_context = talloc(mem_ctx, struct drsuapi_DsReplicaObjectIdentifier); - *r->out.ctr->ctr6.naming_context = *ncRoot; - /* TODO: linked attributes*/ - r->out.ctr->ctr6.linked_attributes_count = 0; - r->out.ctr->ctr6.linked_attributes = NULL; - - r->out.ctr->ctr6.object_count = 0; - r->out.ctr->ctr6.more_data = false; - r->out.ctr->ctr6.uptodateness_vector = NULL; - - /* Prefix mapping */ - schema = dsdb_get_schema(b_state->sam_ctx); - if (!schema) { - DEBUG(0,("No schema in b_state->sam_ctx")); - } - - dsdb_get_oid_mappings_drsuapi(schema, true, mem_ctx, &ctr); - r->out.ctr->ctr6.mapping_ctr = *ctr; - - r->out.ctr->ctr6.source_dsa_guid = *(samdb_ntds_objectGUID(b_state->sam_ctx)); - r->out.ctr->ctr6.source_dsa_invocation_id = *(samdb_ntds_invocation_id(b_state->sam_ctx)); - - r->out.ctr->ctr6.old_highwatermark = r->in.req->req8.highwatermark; - r->out.ctr->ctr6.new_highwatermark = r->in.req->req8.highwatermark; - - r->out.ctr->ctr6.first_object = talloc(mem_ctx, struct drsuapi_DsReplicaObjectListItemEx); - currentObject = r->out.ctr->ctr6.first_object; - - for(i=0; icount; i++) { - uSN = ldb_msg_find_attr_as_int(site_res->msgs[i],"uSNChanged", -1); - r->out.ctr->ctr6.object_count++; - if (uSN > r->out.ctr->ctr6.new_highwatermark.highest_usn) { - r->out.ctr->ctr6.new_highwatermark.highest_usn = uSN; - } - - if (ldb_dn_compare(ncRoot_dn, site_res->msgs[i]->dn) == 0) { - currentObject->is_nc_prefix = true; - currentObject->parent_object_guid = NULL; - } else { - currentObject->is_nc_prefix = false; - currentObject->parent_object_guid = talloc(mem_ctx, struct GUID); - *currentObject->parent_object_guid = samdb_result_guid(site_res->msgs[i], "parentGUID"); - } - currentObject->next_object = NULL; - - currentObject->meta_data_ctr = talloc(mem_ctx, struct drsuapi_DsReplicaMetaDataCtr); - md_value = ldb_msg_find_ldb_val(site_res->msgs[i], "replPropertyMetaData"); - if (md_value) { - ndr_err = ndr_pull_struct_blob(md_value, mem_ctx, - lp_iconv_convenience(ldb_get_opaque(b_state->sam_ctx, "loadparm")), &md, - (ndr_pull_flags_fn_t)ndr_pull_replPropertyMetaDataBlob); - if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) { - return WERR_DS_DRA_INTERNAL_ERROR; - } - - if (md.version != 1) { - return WERR_DS_DRA_INTERNAL_ERROR; - } - - currentObject->meta_data_ctr->count = md.ctr.ctr1.count; - currentObject->meta_data_ctr->meta_data = talloc_array(mem_ctx, struct drsuapi_DsReplicaMetaData, md.ctr.ctr1.count); - for (j=0; jmeta_data_ctr->meta_data[j].originating_change_time = md.ctr.ctr1.array[j].originating_change_time; - currentObject->meta_data_ctr->meta_data[j].version = md.ctr.ctr1.array[j].version; - currentObject->meta_data_ctr->meta_data[j].originating_invocation_id = md.ctr.ctr1.array[j].originating_invocation_id; - currentObject->meta_data_ctr->meta_data[j].originating_usn = md.ctr.ctr1.array[j].originating_usn; - } - } else { - currentObject->meta_data_ctr->meta_data = talloc(mem_ctx, struct drsuapi_DsReplicaMetaData); - currentObject->meta_data_ctr->count = 0; - } - currentObject->object.identifier = talloc(mem_ctx, struct drsuapi_DsReplicaObjectIdentifier); - obj_dn = ldb_msg_find_attr_as_dn(b_state->sam_ctx, mem_ctx, site_res->msgs[i], "distinguishedName"); - currentObject->object.identifier->dn = ldb_dn_get_linearized(obj_dn); - currentObject->object.identifier->guid = GUID_zero(); - currentObject->object.identifier->sid = *zero_sid; - - currentObject->object.attribute_ctr.num_attributes = site_res->msgs[i]->num_elements; - /* Exclude non-replicate attributes from the responce.*/ - for (j=0; jmsgs[i]->num_elements; j++) { - const struct dsdb_attribute *sa; - sa = dsdb_attribute_by_lDAPDisplayName(schema, site_res->msgs[i]->elements[j].name); - if (sa && sa->systemFlags & 0x00000001) { - ldb_msg_remove_attr(site_res->msgs[i], site_res->msgs[i]->elements[j].name); - currentObject->object.attribute_ctr.num_attributes--; - } - } - currentObject->object.attribute_ctr.attributes = talloc_array(mem_ctx, struct drsuapi_DsReplicaAttribute, - currentObject->object.attribute_ctr.num_attributes); - for (j=0; jobject.attribute_ctr.num_attributes; j++) { - dsdb_attribute_ldb_to_drsuapi(b_state->sam_ctx, schema,&site_res->msgs[i]->elements[j], mem_ctx, - ¤tObject->object.attribute_ctr.attributes[j]); - } - - if (i == (site_res->count-1)) { - break; - } - currentObject->next_object = talloc(mem_ctx, struct drsuapi_DsReplicaObjectListItemEx); - currentObject = currentObject->next_object; - } - - r->out.ctr->ctr6.uptodateness_vector = talloc(mem_ctx, struct drsuapi_DsReplicaCursor2CtrEx); - - r->out.ctr->ctr6.uptodateness_vector->version = 2; - r->out.ctr->ctr6.uptodateness_vector->count = 1; - r->out.ctr->ctr6.uptodateness_vector->reserved1 = 0; - r->out.ctr->ctr6.uptodateness_vector->reserved2 = 0; - r->out.ctr->ctr6.uptodateness_vector->cursors = talloc(mem_ctx, struct drsuapi_DsReplicaCursor2); - - r->out.ctr->ctr6.uptodateness_vector->cursors[0].source_dsa_invocation_id = *(samdb_ntds_invocation_id(b_state->sam_ctx)); - r->out.ctr->ctr6.uptodateness_vector->cursors[0].highest_usn = r->out.ctr->ctr6.new_highwatermark.highest_usn; - unix_to_nt_time(&now, t); - r->out.ctr->ctr6.uptodateness_vector->cursors[0].last_sync_success = now; - - return WERR_OK; -} - - /* DRSUAPI_REPLICA_ADD */ @@ -716,7 +472,7 @@ static WERROR dcesrv_drsuapi_DsRemoveDSServer(struct dcesrv_call_state *dce_call bool ok; ZERO_STRUCT(r->out.res); - r->out.level_out = 1; + *r->out.level_out = 1; DCESRV_PULL_HANDLE_WERR(h, r->in.bind_handle, DRSUAPI_BIND_HANDLE); b_state = h->data; @@ -1028,50 +784,6 @@ static WERROR dcesrv_drsuapi_DsGetDomainControllerInfo(struct dcesrv_call_state return WERR_UNKNOWN_LEVEL; } -/* - drsuapi_DsAddEntry -*/ -static WERROR dcesrv_drsuapi_DsAddEntry(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, - struct drsuapi_DsAddEntry *r) -{ - WERROR status; - struct drsuapi_bind_state *b_state; - struct dcesrv_handle *h; - uint32_t num = 0; - struct drsuapi_DsReplicaObjectIdentifier2 *ids = NULL; - - /* TODO: check which out level the client supports */ - - ZERO_STRUCTP(r->out.ctr); - r->out.level_out = 3; - r->out.ctr->ctr3.level = 1; - r->out.ctr->ctr3.error = talloc_zero(mem_ctx, union drsuapi_DsAddEntryError); - - DCESRV_PULL_HANDLE_WERR(h, r->in.bind_handle, DRSUAPI_BIND_HANDLE); - b_state = h->data; - - switch (r->in.level) { - case 2: - status = dsdb_origin_objects_commit(b_state->sam_ctx, - mem_ctx, - &r->in.req->req2.first_object, - &num, - &ids); - if (!W_ERROR_IS_OK(status)) { - r->out.ctr->ctr3.error->info1.status = status; - W_ERROR_NOT_OK_RETURN(status); - } - - r->out.ctr->ctr3.count = num; - r->out.ctr->ctr3.objects = ids; - - return WERR_OK; - default: - break; - } - - return WERR_FOOBAR; -} /* diff --git a/source4/rpc_server/drsuapi/dcesrv_drsuapi.h b/source4/rpc_server/drsuapi/dcesrv_drsuapi.h index a53c47f409..3f69a3fb52 100644 --- a/source4/rpc_server/drsuapi/dcesrv_drsuapi.h +++ b/source4/rpc_server/drsuapi/dcesrv_drsuapi.h @@ -40,7 +40,19 @@ struct drsuapi_bind_state { /* prototypes of internal functions */ WERROR dcesrv_drsuapi_DsReplicaUpdateRefs(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, struct drsuapi_DsReplicaUpdateRefs *r); +WERROR dcesrv_drsuapi_DsGetNCChanges(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, + struct drsuapi_DsGetNCChanges *r); +WERROR dcesrv_drsuapi_DsAddEntry(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, + struct drsuapi_DsAddEntry *r); char *drs_ObjectIdentifier_to_string(TALLOC_CTX *mem_ctx, struct drsuapi_DsReplicaObjectIdentifier *nc); +int drsuapi_search_with_extended_dn(struct ldb_context *ldb, + TALLOC_CTX *mem_ctx, + struct ldb_result **_res, + struct ldb_dn *basedn, + enum ldb_scope scope, + const char * const *attrs, + const char *format, ...) PRINTF_ATTRIBUTE(7,8); + diff --git a/source4/rpc_server/drsuapi/drsutil.c b/source4/rpc_server/drsuapi/drsutil.c index f14c072c59..f8736141d5 100644 --- a/source4/rpc_server/drsuapi/drsutil.c +++ b/source4/rpc_server/drsuapi/drsutil.c @@ -23,12 +23,12 @@ #include "librpc/gen_ndr/ndr_drsuapi.h" #include "rpc_server/dcerpc_server.h" #include "rpc_server/common/common.h" -#include "rpc_server/drsuapi/dcesrv_drsuapi.h" #include "dsdb/samdb/samdb.h" #include "lib/ldb/include/ldb_errors.h" #include "param/param.h" #include "librpc/gen_ndr/ndr_drsblobs.h" #include "libcli/security/dom_sid.h" +#include "rpc_server/drsuapi/dcesrv_drsuapi.h" /* format a drsuapi_DsReplicaObjectIdentifier naming context as a string @@ -45,3 +45,64 @@ char *drs_ObjectIdentifier_to_string(TALLOC_CTX *mem_ctx, talloc_free(sid); return ret; } + +int drsuapi_search_with_extended_dn(struct ldb_context *ldb, + TALLOC_CTX *mem_ctx, + struct ldb_result **_res, + struct ldb_dn *basedn, + enum ldb_scope scope, + const char * const *attrs, + const char *format, ...) +{ + va_list ap; + int ret; + struct ldb_request *req; + char *filter; + TALLOC_CTX *tmp_ctx; + struct ldb_result *res; + + tmp_ctx = talloc_new(mem_ctx); + + res = talloc_zero(tmp_ctx, struct ldb_result); + if (!res) { + return LDB_ERR_OPERATIONS_ERROR; + } + + va_start(ap, format); + filter = talloc_vasprintf(tmp_ctx, format, ap); + va_end(ap); + + if (filter == NULL) { + talloc_free(tmp_ctx); + return LDB_ERR_OPERATIONS_ERROR; + } + + ret = ldb_build_search_req(&req, ldb, tmp_ctx, + basedn, + scope, + filter, + attrs, + NULL, + res, + ldb_search_default_callback, + NULL); + if (ret != LDB_SUCCESS) { + talloc_free(tmp_ctx); + return ret; + } + + ret = ldb_request_add_control(req, LDB_CONTROL_EXTENDED_DN_OID, true, NULL); + if (ret != LDB_SUCCESS) { + return ret; + } + + ret = ldb_request(ldb, req); + if (ret == LDB_SUCCESS) { + ret = ldb_wait(req->handle, LDB_WAIT_ALL); + } + + talloc_free(req); + *_res = res; + return ret; +} + diff --git a/source4/rpc_server/drsuapi/getncchanges.c b/source4/rpc_server/drsuapi/getncchanges.c new file mode 100644 index 0000000000..25221d41b5 --- /dev/null +++ b/source4/rpc_server/drsuapi/getncchanges.c @@ -0,0 +1,215 @@ +/* + Unix SMB/CIFS implementation. + + implement the DRSUpdateRefs call + + Copyright (C) Anatoliy Atanasov 2009 + Copyright (C) Andrew Tridgell 2009 + + 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 3 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, see . +*/ + +#include "includes.h" +#include "librpc/gen_ndr/ndr_drsuapi.h" +#include "rpc_server/dcerpc_server.h" +#include "rpc_server/common/common.h" +#include "dsdb/samdb/samdb.h" +#include "lib/ldb/include/ldb_errors.h" +#include "param/param.h" +#include "librpc/gen_ndr/ndr_drsblobs.h" +#include "auth/auth.h" +#include "rpc_server/drsuapi/dcesrv_drsuapi.h" + +/* + drsuapi_DsGetNCChanges +*/ +WERROR dcesrv_drsuapi_DsGetNCChanges(struct dcesrv_call_state *dce_call, TALLOC_CTX *mem_ctx, + struct drsuapi_DsGetNCChanges *r) +{ + struct ldb_result *site_res; + struct drsuapi_DsReplicaObjectIdentifier *ncRoot; + struct drsuapi_bind_state *b_state; + struct ldb_dn *ncRoot_dn; + int ret; + int i; + int j; + int uSN; + struct dsdb_schema *schema; + struct drsuapi_DsReplicaOIDMapping_Ctr *ctr; + time_t t = time(NULL); + NTTIME now; + struct drsuapi_DsReplicaObjectListItemEx *currentObject; + struct dom_sid *zero_sid; + struct ldb_dn *obj_dn; + enum ndr_err_code ndr_err; + const struct ldb_val *md_value; + struct replPropertyMetaDataBlob md; + ZERO_STRUCT(md); + md.version = 1; + b_state = talloc_zero(mem_ctx, struct drsuapi_bind_state); + W_ERROR_HAVE_NO_MEMORY(b_state); + zero_sid = talloc_zero(mem_ctx, struct dom_sid); + /* + * connect to the samdb + */ + b_state->sam_ctx = samdb_connect(b_state, dce_call->event_ctx, dce_call->conn->dce_ctx->lp_ctx, dce_call->conn->auth_state.session_info); + if (!b_state->sam_ctx) { + return WERR_FOOBAR; + } + + /* Check request revision. */ + if (r->in.level != 8) { + return WERR_REVISION_MISMATCH; + } + + /* Perform access checks. */ + if (r->in.req->req8.naming_context == NULL) { + return WERR_DS_DRA_INVALID_PARAMETER; + } + + ncRoot = r->in.req->req8.naming_context; + if (ncRoot == NULL) { + return WERR_DS_DRA_BAD_NC; + } + + DEBUG(4,("DsGetNSChanges with uSHChanged >= %llu\n", + (unsigned long long)r->in.req->req8.highwatermark.highest_usn)); + + /* Construct response. */ + ncRoot_dn = ldb_dn_new(mem_ctx, b_state->sam_ctx, ncRoot->dn); + ret = drsuapi_search_with_extended_dn(b_state->sam_ctx, mem_ctx, &site_res, + ncRoot_dn, LDB_SCOPE_SUBTREE, NULL, + "(&(uSNChanged>=%llu)(objectClass=*))", + (unsigned long long)r->in.req->req8.highwatermark.highest_usn); + if (ret != LDB_SUCCESS) { + return WERR_DS_DRA_INTERNAL_ERROR; + } + + *r->out.level_out = 6; + r->out.ctr->ctr6.naming_context = talloc(mem_ctx, struct drsuapi_DsReplicaObjectIdentifier); + *r->out.ctr->ctr6.naming_context = *ncRoot; + /* TODO: linked attributes*/ + r->out.ctr->ctr6.linked_attributes_count = 0; + r->out.ctr->ctr6.linked_attributes = NULL; + + r->out.ctr->ctr6.object_count = 0; + r->out.ctr->ctr6.more_data = false; + r->out.ctr->ctr6.uptodateness_vector = NULL; + + /* Prefix mapping */ + schema = dsdb_get_schema(b_state->sam_ctx); + if (!schema) { + DEBUG(0,("No schema in b_state->sam_ctx")); + } + + dsdb_get_oid_mappings_drsuapi(schema, true, mem_ctx, &ctr); + r->out.ctr->ctr6.mapping_ctr = *ctr; + + r->out.ctr->ctr6.source_dsa_guid = *(samdb_ntds_objectGUID(b_state->sam_ctx)); + r->out.ctr->ctr6.source_dsa_invocation_id = *(samdb_ntds_invocation_id(b_state->sam_ctx)); + + r->out.ctr->ctr6.old_highwatermark = r->in.req->req8.highwatermark; + r->out.ctr->ctr6.new_highwatermark = r->in.req->req8.highwatermark; + + r->out.ctr->ctr6.first_object = talloc(mem_ctx, struct drsuapi_DsReplicaObjectListItemEx); + currentObject = r->out.ctr->ctr6.first_object; + + for(i=0; icount; i++) { + uSN = ldb_msg_find_attr_as_int(site_res->msgs[i],"uSNChanged", -1); + r->out.ctr->ctr6.object_count++; + if (uSN > r->out.ctr->ctr6.new_highwatermark.highest_usn) { + r->out.ctr->ctr6.new_highwatermark.highest_usn = uSN; + } + + if (ldb_dn_compare(ncRoot_dn, site_res->msgs[i]->dn) == 0) { + currentObject->is_nc_prefix = true; + currentObject->parent_object_guid = NULL; + } else { + currentObject->is_nc_prefix = false; + currentObject->parent_object_guid = talloc(mem_ctx, struct GUID); + *currentObject->parent_object_guid = samdb_result_guid(site_res->msgs[i], "parentGUID"); + } + currentObject->next_object = NULL; + + currentObject->meta_data_ctr = talloc(mem_ctx, struct drsuapi_DsReplicaMetaDataCtr); + md_value = ldb_msg_find_ldb_val(site_res->msgs[i], "replPropertyMetaData"); + if (md_value) { + ndr_err = ndr_pull_struct_blob(md_value, mem_ctx, + lp_iconv_convenience(ldb_get_opaque(b_state->sam_ctx, "loadparm")), &md, + (ndr_pull_flags_fn_t)ndr_pull_replPropertyMetaDataBlob); + if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) { + return WERR_DS_DRA_INTERNAL_ERROR; + } + + if (md.version != 1) { + return WERR_DS_DRA_INTERNAL_ERROR; + } + + currentObject->meta_data_ctr->count = md.ctr.ctr1.count; + currentObject->meta_data_ctr->meta_data = talloc_array(mem_ctx, struct drsuapi_DsReplicaMetaData, md.ctr.ctr1.count); + for (j=0; jmeta_data_ctr->meta_data[j].originating_change_time = md.ctr.ctr1.array[j].originating_change_time; + currentObject->meta_data_ctr->meta_data[j].version = md.ctr.ctr1.array[j].version; + currentObject->meta_data_ctr->meta_data[j].originating_invocation_id = md.ctr.ctr1.array[j].originating_invocation_id; + currentObject->meta_data_ctr->meta_data[j].originating_usn = md.ctr.ctr1.array[j].originating_usn; + } + } else { + currentObject->meta_data_ctr->meta_data = talloc(mem_ctx, struct drsuapi_DsReplicaMetaData); + currentObject->meta_data_ctr->count = 0; + } + currentObject->object.identifier = talloc(mem_ctx, struct drsuapi_DsReplicaObjectIdentifier); + obj_dn = ldb_msg_find_attr_as_dn(b_state->sam_ctx, mem_ctx, site_res->msgs[i], "distinguishedName"); + currentObject->object.identifier->dn = ldb_dn_get_linearized(obj_dn); + currentObject->object.identifier->guid = GUID_zero(); + currentObject->object.identifier->sid = *zero_sid; + + currentObject->object.attribute_ctr.num_attributes = site_res->msgs[i]->num_elements; + /* Exclude non-replicate attributes from the responce.*/ + for (j=0; jmsgs[i]->num_elements; j++) { + const struct dsdb_attribute *sa; + sa = dsdb_attribute_by_lDAPDisplayName(schema, site_res->msgs[i]->elements[j].name); + if (sa && sa->systemFlags & SYSTEM_FLAG_CR_NTDS_NC) { + ldb_msg_remove_attr(site_res->msgs[i], site_res->msgs[i]->elements[j].name); + currentObject->object.attribute_ctr.num_attributes--; + } + } + currentObject->object.attribute_ctr.attributes = talloc_array(mem_ctx, struct drsuapi_DsReplicaAttribute, + currentObject->object.attribute_ctr.num_attributes); + for (j=0; jobject.attribute_ctr.num_attributes; j++) { + dsdb_attribute_ldb_to_drsuapi(b_state->sam_ctx, schema,&site_res->msgs[i]->elements[j], mem_ctx, + ¤tObject->object.attribute_ctr.attributes[j]); + } + + if (i == (site_res->count-1)) { + break; + } + currentObject->next_object = talloc(mem_ctx, struct drsuapi_DsReplicaObjectListItemEx); + currentObject = currentObject->next_object; + } + + r->out.ctr->ctr6.uptodateness_vector = talloc(mem_ctx, struct drsuapi_DsReplicaCursor2CtrEx); + + r->out.ctr->ctr6.uptodateness_vector->version = 2; + r->out.ctr->ctr6.uptodateness_vector->count = 1; + r->out.ctr->ctr6.uptodateness_vector->reserved1 = 0; + r->out.ctr->ctr6.uptodateness_vector->reserved2 = 0; + r->out.ctr->ctr6.uptodateness_vector->cursors = talloc(mem_ctx, struct drsuapi_DsReplicaCursor2); + + r->out.ctr->ctr6.uptodateness_vector->cursors[0].source_dsa_invocation_id = *(samdb_ntds_invocation_id(b_state->sam_ctx)); + r->out.ctr->ctr6.uptodateness_vector->cursors[0].highest_usn = r->out.ctr->ctr6.new_highwatermark.highest_usn; + unix_to_nt_time(&now, t); + r->out.ctr->ctr6.uptodateness_vector->cursors[0].last_sync_success = now; + + return WERR_OK; +} diff --git a/source4/rpc_server/drsuapi/updaterefs.c b/source4/rpc_server/drsuapi/updaterefs.c index 4a888099c0..2090dd5dd5 100644 --- a/source4/rpc_server/drsuapi/updaterefs.c +++ b/source4/rpc_server/drsuapi/updaterefs.c @@ -23,12 +23,12 @@ #include "librpc/gen_ndr/ndr_drsuapi.h" #include "rpc_server/dcerpc_server.h" #include "rpc_server/common/common.h" -#include "rpc_server/drsuapi/dcesrv_drsuapi.h" #include "dsdb/samdb/samdb.h" #include "lib/ldb/include/ldb_errors.h" #include "param/param.h" #include "librpc/gen_ndr/ndr_drsblobs.h" #include "auth/auth.h" +#include "rpc_server/drsuapi/dcesrv_drsuapi.h" struct repsTo { uint32_t count; -- cgit