2 Unix SMB/CIFS mplementation.
3 Helper functions for applying replicated objects
5 Copyright (C) Stefan Metzmacher <metze@samba.org> 2007
7 This program is free software; you can redistribute it and/or modify
8 it under the terms of the GNU General Public License as published by
9 the Free Software Foundation; either version 3 of the License, or
10 (at your option) any later version.
12 This program is distributed in the hope that it will be useful,
13 but WITHOUT ANY WARRANTY; without even the implied warranty of
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 GNU General Public License for more details.
17 You should have received a copy of the GNU General Public License
18 along with this program. If not, see <http://www.gnu.org/licenses/>.
23 #include "dsdb/samdb/samdb.h"
24 #include "lib/ldb/include/ldb_errors.h"
25 #include "../lib/util/dlinklist.h"
26 #include "librpc/gen_ndr/ndr_misc.h"
27 #include "librpc/gen_ndr/ndr_drsuapi.h"
28 #include "librpc/gen_ndr/ndr_drsblobs.h"
29 #include "../lib/crypto/crypto.h"
30 #include "../libcli/drsuapi/drsuapi.h"
31 #include "libcli/auth/libcli_auth.h"
32 #include "param/param.h"
34 static WERROR
dsdb_convert_object_ex(struct ldb_context
*ldb
,
35 const struct dsdb_schema
*schema
,
36 const struct drsuapi_DsReplicaObjectListItemEx
*in
,
37 const DATA_BLOB
*gensec_skey
,
39 struct dsdb_extended_replicated_object
*out
)
42 enum ndr_err_code ndr_err
;
45 struct ldb_message
*msg
;
46 struct replPropertyMetaDataBlob
*md
;
47 struct ldb_val guid_value
;
48 NTTIME whenChanged
= 0;
50 const char *whenChanged_s
;
51 const char *rdn_name
= NULL
;
52 const struct ldb_val
*rdn_value
= NULL
;
53 const struct dsdb_attribute
*rdn_attr
= NULL
;
55 struct drsuapi_DsReplicaAttribute
*name_a
= NULL
;
56 struct drsuapi_DsReplicaMetaData
*name_d
= NULL
;
57 struct replPropertyMetaData1
*rdn_m
= NULL
;
58 struct dom_sid
*sid
= NULL
;
62 if (!in
->object
.identifier
) {
66 if (!in
->object
.identifier
->dn
|| !in
->object
.identifier
->dn
[0]) {
70 if (in
->object
.attribute_ctr
.num_attributes
!= 0 && !in
->meta_data_ctr
) {
74 if (in
->object
.attribute_ctr
.num_attributes
!= in
->meta_data_ctr
->count
) {
78 sid
= &in
->object
.identifier
->sid
;
79 if (sid
->num_auths
> 0) {
80 rid
= sid
->sub_auths
[sid
->num_auths
- 1];
83 msg
= ldb_msg_new(mem_ctx
);
84 W_ERROR_HAVE_NO_MEMORY(msg
);
86 msg
->dn
= ldb_dn_new(msg
, ldb
, in
->object
.identifier
->dn
);
87 W_ERROR_HAVE_NO_MEMORY(msg
->dn
);
89 rdn_name
= ldb_dn_get_rdn_name(msg
->dn
);
90 rdn_attr
= dsdb_attribute_by_lDAPDisplayName(schema
, rdn_name
);
94 rdn_attid
= rdn_attr
->attributeID_id
;
95 rdn_value
= ldb_dn_get_rdn_val(msg
->dn
);
97 msg
->num_elements
= in
->object
.attribute_ctr
.num_attributes
;
98 msg
->elements
= talloc_array(msg
, struct ldb_message_element
,
100 W_ERROR_HAVE_NO_MEMORY(msg
->elements
);
102 md
= talloc(mem_ctx
, struct replPropertyMetaDataBlob
);
103 W_ERROR_HAVE_NO_MEMORY(md
);
107 md
->ctr
.ctr1
.count
= in
->meta_data_ctr
->count
;
108 md
->ctr
.ctr1
.reserved
= 0;
109 md
->ctr
.ctr1
.array
= talloc_array(mem_ctx
,
110 struct replPropertyMetaData1
,
111 md
->ctr
.ctr1
.count
+ 1); /* +1 because of the RDN attribute */
112 W_ERROR_HAVE_NO_MEMORY(md
->ctr
.ctr1
.array
);
114 for (i
=0; i
< in
->meta_data_ctr
->count
; i
++) {
115 struct drsuapi_DsReplicaAttribute
*a
;
116 struct drsuapi_DsReplicaMetaData
*d
;
117 struct replPropertyMetaData1
*m
;
118 struct ldb_message_element
*e
;
121 a
= &in
->object
.attribute_ctr
.attributes
[i
];
122 d
= &in
->meta_data_ctr
->meta_data
[i
];
123 m
= &md
->ctr
.ctr1
.array
[i
];
124 e
= &msg
->elements
[i
];
126 for (j
=0; j
<a
->value_ctr
.num_values
; j
++) {
127 status
= drsuapi_decrypt_attribute(a
->value_ctr
.values
[j
].blob
, gensec_skey
, rid
, a
);
128 W_ERROR_NOT_OK_RETURN(status
);
131 status
= dsdb_attribute_drsuapi_to_ldb(ldb
, schema
, a
, msg
->elements
, e
);
132 W_ERROR_NOT_OK_RETURN(status
);
135 m
->version
= d
->version
;
136 m
->originating_change_time
= d
->originating_change_time
;
137 m
->originating_invocation_id
= d
->originating_invocation_id
;
138 m
->originating_usn
= d
->originating_usn
;
141 if (d
->originating_change_time
> whenChanged
) {
142 whenChanged
= d
->originating_change_time
;
145 if (a
->attid
== DRSUAPI_ATTRIBUTE_name
) {
148 rdn_m
= &md
->ctr
.ctr1
.array
[md
->ctr
.ctr1
.count
];
153 struct ldb_message_element
*el
;
154 el
= ldb_msg_find_element(msg
, rdn_attr
->lDAPDisplayName
);
156 ret
= ldb_msg_add_value(msg
, rdn_attr
->lDAPDisplayName
, rdn_value
, NULL
);
157 if (ret
!= LDB_SUCCESS
) {
161 if (el
->num_values
!= 1) {
162 DEBUG(0,(__location__
": Unexpected num_values=%u\n",
166 if (!ldb_val_equal_exact(&el
->values
[0], rdn_value
)) {
167 DEBUG(0,(__location__
": RDN value changed? '%*.*s' '%*.*s'\n",
168 (int)el
->values
[0].length
, (int)el
->values
[0].length
, el
->values
[0].data
,
169 (int)rdn_value
->length
, (int)rdn_value
->length
, rdn_value
->data
));
174 rdn_m
->attid
= rdn_attid
;
175 rdn_m
->version
= name_d
->version
;
176 rdn_m
->originating_change_time
= name_d
->originating_change_time
;
177 rdn_m
->originating_invocation_id
= name_d
->originating_invocation_id
;
178 rdn_m
->originating_usn
= name_d
->originating_usn
;
179 rdn_m
->local_usn
= 0;
180 md
->ctr
.ctr1
.count
++;
184 whenChanged_t
= nt_time_to_unix(whenChanged
);
185 whenChanged_s
= ldb_timestring(msg
, whenChanged_t
);
186 W_ERROR_HAVE_NO_MEMORY(whenChanged_s
);
188 ndr_err
= ndr_push_struct_blob(&guid_value
, msg
,
189 lp_iconv_convenience(ldb_get_opaque(ldb
, "loadparm")),
190 &in
->object
.identifier
->guid
,
191 (ndr_push_flags_fn_t
)ndr_push_GUID
);
192 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err
)) {
193 nt_status
= ndr_map_error2ntstatus(ndr_err
);
194 return ntstatus_to_werror(nt_status
);
198 out
->guid_value
= guid_value
;
199 out
->when_changed
= whenChanged_s
;
204 WERROR
dsdb_extended_replicated_objects_convert(struct ldb_context
*ldb
,
205 const char *partition_dn
,
206 const struct drsuapi_DsReplicaOIDMapping_Ctr
*mapping_ctr
,
207 uint32_t object_count
,
208 const struct drsuapi_DsReplicaObjectListItemEx
*first_object
,
209 uint32_t linked_attributes_count
,
210 const struct drsuapi_DsReplicaLinkedAttribute
*linked_attributes
,
211 const struct repsFromTo1
*source_dsa
,
212 const struct drsuapi_DsReplicaCursor2CtrEx
*uptodateness_vector
,
213 const DATA_BLOB
*gensec_skey
,
215 struct dsdb_extended_replicated_objects
**objects
)
218 const struct dsdb_schema
*schema
;
219 struct dsdb_extended_replicated_objects
*out
;
220 const struct drsuapi_DsReplicaObjectListItemEx
*cur
;
223 schema
= dsdb_get_schema(ldb
);
225 return WERR_DS_SCHEMA_NOT_LOADED
;
228 status
= dsdb_schema_pfm_contains_drsuapi_pfm(schema
->prefixmap
, mapping_ctr
);
229 W_ERROR_NOT_OK_RETURN(status
);
231 out
= talloc_zero(mem_ctx
, struct dsdb_extended_replicated_objects
);
232 W_ERROR_HAVE_NO_MEMORY(out
);
233 out
->version
= DSDB_EXTENDED_REPLICATED_OBJECTS_VERSION
;
235 out
->partition_dn
= ldb_dn_new(out
, ldb
, partition_dn
);
236 W_ERROR_HAVE_NO_MEMORY(out
->partition_dn
);
238 out
->source_dsa
= source_dsa
;
239 out
->uptodateness_vector
= uptodateness_vector
;
241 out
->num_objects
= object_count
;
242 out
->objects
= talloc_array(out
,
243 struct dsdb_extended_replicated_object
,
245 W_ERROR_HAVE_NO_MEMORY(out
->objects
);
247 /* pass the linked attributes down to the repl_meta_data
249 out
->linked_attributes_count
= linked_attributes_count
;
250 out
->linked_attributes
= linked_attributes
;
252 for (i
=0, cur
= first_object
; cur
; cur
= cur
->next_object
, i
++) {
253 if (i
== out
->num_objects
) {
257 status
= dsdb_convert_object_ex(ldb
, schema
,
259 out
->objects
, &out
->objects
[i
]);
260 if (!W_ERROR_IS_OK(status
)) {
261 DEBUG(0,("Failed to convert object %s\n", cur
->object
.identifier
->dn
));
265 if (i
!= out
->num_objects
) {
273 WERROR
dsdb_extended_replicated_objects_commit(struct ldb_context
*ldb
,
274 struct dsdb_extended_replicated_objects
*objects
,
275 uint64_t *notify_uSN
)
277 struct ldb_result
*ext_res
;
279 uint64_t seq_num1
, seq_num2
;
281 /* TODO: handle linked attributes */
283 /* wrap the extended operation in a transaction
284 See [MS-DRSR] 3.3.2 Transactions
286 ret
= ldb_transaction_start(ldb
);
287 if (ret
!= LDB_SUCCESS
) {
288 DEBUG(0,(__location__
" Failed to start transaction\n"));
292 ret
= dsdb_load_partition_usn(ldb
, objects
->partition_dn
, &seq_num1
);
293 if (ret
!= LDB_SUCCESS
) {
294 DEBUG(0,(__location__
" Failed to load partition uSN\n"));
295 ldb_transaction_cancel(ldb
);
299 ret
= ldb_extended(ldb
, DSDB_EXTENDED_REPLICATED_OBJECTS_OID
, objects
, &ext_res
);
300 if (ret
!= LDB_SUCCESS
) {
301 DEBUG(0,("Failed to apply records: %s: %s\n",
302 ldb_errstring(ldb
), ldb_strerror(ret
)));
303 ldb_transaction_cancel(ldb
);
306 talloc_free(ext_res
);
308 ret
= ldb_transaction_prepare_commit(ldb
);
309 if (ret
!= LDB_SUCCESS
) {
310 DEBUG(0,(__location__
" Failed to prepare commit of transaction\n"));
314 ret
= dsdb_load_partition_usn(ldb
, objects
->partition_dn
, &seq_num2
);
315 if (ret
!= LDB_SUCCESS
) {
316 DEBUG(0,(__location__
" Failed to load partition uSN\n"));
317 ldb_transaction_cancel(ldb
);
321 /* if this replication partner didn't need to be notified
322 before this transaction then it still doesn't need to be
323 notified, as the changes came from this server */
324 if (seq_num2
> seq_num1
&& seq_num1
<= *notify_uSN
) {
325 *notify_uSN
= seq_num2
;
328 ret
= ldb_transaction_commit(ldb
);
329 if (ret
!= LDB_SUCCESS
) {
330 DEBUG(0,(__location__
" Failed to commit transaction\n"));
335 DEBUG(2,("Replicated %u objects (%u linked attributes) for %s\n",
336 objects
->num_objects
, objects
->linked_attributes_count
,
337 ldb_dn_get_linearized(objects
->partition_dn
)));
342 static WERROR
dsdb_convert_object(struct ldb_context
*ldb
,
343 const struct dsdb_schema
*schema
,
344 const struct drsuapi_DsReplicaObjectListItem
*in
,
346 struct ldb_message
**_msg
)
350 struct ldb_message
*msg
;
352 if (!in
->object
.identifier
) {
356 if (!in
->object
.identifier
->dn
|| !in
->object
.identifier
->dn
[0]) {
360 msg
= ldb_msg_new(mem_ctx
);
361 W_ERROR_HAVE_NO_MEMORY(msg
);
363 msg
->dn
= ldb_dn_new(msg
, ldb
, in
->object
.identifier
->dn
);
364 W_ERROR_HAVE_NO_MEMORY(msg
->dn
);
366 msg
->num_elements
= in
->object
.attribute_ctr
.num_attributes
;
367 msg
->elements
= talloc_array(msg
, struct ldb_message_element
,
369 W_ERROR_HAVE_NO_MEMORY(msg
->elements
);
371 for (i
=0; i
< msg
->num_elements
; i
++) {
372 struct drsuapi_DsReplicaAttribute
*a
;
373 struct ldb_message_element
*e
;
375 a
= &in
->object
.attribute_ctr
.attributes
[i
];
376 e
= &msg
->elements
[i
];
378 status
= dsdb_attribute_drsuapi_to_ldb(ldb
, schema
, a
, msg
->elements
, e
);
379 W_ERROR_NOT_OK_RETURN(status
);
388 WERROR
dsdb_origin_objects_commit(struct ldb_context
*ldb
,
390 const struct drsuapi_DsReplicaObjectListItem
*first_object
,
392 struct drsuapi_DsReplicaObjectIdentifier2
**_ids
)
395 const struct dsdb_schema
*schema
;
396 const struct drsuapi_DsReplicaObjectListItem
*cur
;
397 struct ldb_message
**objects
;
398 struct drsuapi_DsReplicaObjectIdentifier2
*ids
;
400 uint32_t num_objects
= 0;
401 const char * const attrs
[] = {
406 struct ldb_result
*res
;
409 schema
= dsdb_get_schema(ldb
);
411 return WERR_DS_SCHEMA_NOT_LOADED
;
414 for (cur
= first_object
; cur
; cur
= cur
->next_object
) {
418 if (num_objects
== 0) {
422 ret
= ldb_transaction_start(ldb
);
423 if (ret
!= LDB_SUCCESS
) {
424 return WERR_DS_INTERNAL_FAILURE
;
427 objects
= talloc_array(mem_ctx
, struct ldb_message
*,
429 if (objects
== NULL
) {
434 for (i
=0, cur
= first_object
; cur
; cur
= cur
->next_object
, i
++) {
435 status
= dsdb_convert_object(ldb
, schema
,
436 cur
, objects
, &objects
[i
]);
437 if (!W_ERROR_IS_OK(status
)) {
442 ids
= talloc_array(mem_ctx
,
443 struct drsuapi_DsReplicaObjectIdentifier2
,
450 for (i
=0; i
< num_objects
; i
++) {
451 struct dom_sid
*sid
= NULL
;
452 struct ldb_request
*add_req
;
454 DEBUG(6,(__location__
": adding %s\n",
455 ldb_dn_get_linearized(objects
[i
]->dn
)));
457 ret
= ldb_build_add_req(&add_req
,
463 ldb_op_default_callback
,
465 if (ret
!= LDB_SUCCESS
) {
466 status
= WERR_DS_INTERNAL_FAILURE
;
470 ret
= ldb_request_add_control(add_req
, LDB_CONTROL_RELAX_OID
, true, NULL
);
471 if (ret
!= LDB_SUCCESS
) {
472 status
= WERR_DS_INTERNAL_FAILURE
;
476 ret
= ldb_request(ldb
, add_req
);
477 if (ret
== LDB_SUCCESS
) {
478 ret
= ldb_wait(add_req
->handle
, LDB_WAIT_ALL
);
480 if (ret
!= LDB_SUCCESS
) {
481 DEBUG(0,(__location__
": Failed add of %s - %s\n",
482 ldb_dn_get_linearized(objects
[i
]->dn
), ldb_errstring(ldb
)));
483 status
= WERR_DS_INTERNAL_FAILURE
;
487 talloc_free(add_req
);
489 ret
= ldb_search(ldb
, objects
, &res
, objects
[i
]->dn
,
490 LDB_SCOPE_BASE
, attrs
,
492 if (ret
!= LDB_SUCCESS
) {
493 status
= WERR_DS_INTERNAL_FAILURE
;
496 ids
[i
].guid
= samdb_result_guid(res
->msgs
[0], "objectGUID");
497 sid
= samdb_result_dom_sid(objects
, res
->msgs
[0], "objectSid");
501 ZERO_STRUCT(ids
[i
].sid
);
505 ret
= ldb_transaction_commit(ldb
);
506 if (ret
!= LDB_SUCCESS
) {
507 return WERR_DS_INTERNAL_FAILURE
;
510 talloc_free(objects
);
517 talloc_free(objects
);
518 ldb_transaction_cancel(ldb
);