r20680: prepare the helper functions for applying replicated objects
[samba.git] / source4 / dsdb / repl / replicated_objects.c
1 /* 
2    Unix SMB/CIFS mplementation.
3    Helper functions for applying replicated objects
4    
5    Copyright (C) Stefan Metzmacher 2007
6     
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 2 of the License, or
10    (at your option) any later version.
11    
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.
16    
17    You should have received a copy of the GNU General Public License
18    along with this program; if not, write to the Free Software
19    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
20    
21 */
22
23 #include "includes.h"
24 #include "dsdb/samdb/samdb.h"
25 #include "lib/ldb/include/ldb_errors.h"
26 #include "lib/util/dlinklist.h"
27 #include "librpc/gen_ndr/ndr_misc.h"
28 #include "librpc/gen_ndr/ndr_drsuapi.h"
29 #include "librpc/gen_ndr/ndr_drsblobs.h"
30
31 static WERROR dsdb_convert_object(struct ldb_context *ldb,
32                                   const struct dsdb_schema *schema,
33                                   struct dsdb_extended_replicated_objects *ctr,
34                                   const struct drsuapi_DsReplicaObjectListItemEx *in,
35                                   TALLOC_CTX *mem_ctx,
36                                   struct dsdb_extended_replicated_object *out)
37 {
38         NTSTATUS nt_status;
39         WERROR status;
40         uint32_t i;
41         struct ldb_message *msg;
42         struct replPropertyMetaDataBlob *md;
43         struct ldb_val guid_value;
44         NTTIME whenChanged = 0;
45         time_t whenChanged_t;
46         const char *whenChanged_s;
47         const char *rdn_name;
48         const struct ldb_val *rdn_value;
49         const struct dsdb_attribute *rdn_attr;
50         uint32_t rdn_attid;
51         struct drsuapi_DsReplicaAttribute *name_a;
52         struct drsuapi_DsReplicaMetaData *name_d;
53         struct replPropertyMetaData1 *rdn_m;
54         int ret;
55
56         if (!in->object.identifier) {
57                 return WERR_FOOBAR;
58         }
59
60         if (!in->object.identifier->dn || !in->object.identifier->dn[0]) {
61                 return WERR_FOOBAR;
62         }
63
64         msg = ldb_msg_new(mem_ctx);
65         W_ERROR_HAVE_NO_MEMORY(msg);
66
67         msg->dn                 = ldb_dn_new(msg, ldb, in->object.identifier->dn);
68         W_ERROR_HAVE_NO_MEMORY(msg->dn);
69
70         rdn_name        = ldb_dn_get_rdn_name(msg->dn);
71         rdn_attr        = dsdb_attribute_by_lDAPDisplayName(schema, rdn_name);
72         if (!rdn_attr) {
73                 return WERR_FOOBAR;
74         }
75         rdn_attid       = rdn_attr->attributeID_id;
76         rdn_value       = ldb_dn_get_rdn_val(msg->dn);
77
78         msg->num_elements       = in->object.attribute_ctr.num_attributes;
79         msg->elements           = talloc_array(msg, struct ldb_message_element,
80                                                msg->num_elements);
81         W_ERROR_HAVE_NO_MEMORY(msg->elements);
82
83         for (i=0; i < msg->num_elements; i++) {
84                 status = dsdb_attribute_drsuapi_to_ldb(schema,
85                                                        &in->object.attribute_ctr.attributes[i],
86                                                        msg->elements, &msg->elements[i]);
87                 W_ERROR_NOT_OK_RETURN(status);
88         }
89
90         if (in->object.attribute_ctr.num_attributes != 0 && !in->meta_data_ctr) {
91                 return WERR_FOOBAR;
92         }
93
94         if (in->object.attribute_ctr.num_attributes != in->meta_data_ctr->count) {
95                 return WERR_FOOBAR;
96         }
97
98         md = talloc(mem_ctx, struct replPropertyMetaDataBlob);
99         W_ERROR_HAVE_NO_MEMORY(md);
100
101         md->version             = 1;
102         md->reserved            = 0;
103         md->ctr.ctr1.count      = in->meta_data_ctr->count;
104         md->ctr.ctr1.reserved   = 0;
105         md->ctr.ctr1.array      = talloc_array(mem_ctx,
106                                                struct replPropertyMetaData1,
107                                                md->ctr.ctr1.count + 1); /* +1 because of the RDN attribute */
108         W_ERROR_HAVE_NO_MEMORY(md->ctr.ctr1.array);
109
110         for (i=0; i < in->meta_data_ctr->count; i++) {
111                 struct drsuapi_DsReplicaAttribute *a;
112                 struct drsuapi_DsReplicaMetaData *d;
113                 struct replPropertyMetaData1 *m;
114
115                 a = &in->object.attribute_ctr.attributes[i];
116                 d = &in->meta_data_ctr->meta_data[i];
117                 m = &md->ctr.ctr1.array[i];
118
119                 m->attid                        = a->attid;
120                 m->version                      = d->version;
121                 m->orginating_time              = d->orginating_time;
122                 m->orginating_invocation_id     = d->orginating_invocation_id;
123                 m->orginating_usn               = d->orginating_usn;
124                 m->local_usn                    = 0;
125
126                 if (d->orginating_time > whenChanged) {
127                         whenChanged = d->orginating_time;
128                 }
129
130                 if (a->attid == DRSUAPI_ATTRIBUTE_name) {
131                         name_a = a;
132                         name_d = d;
133                         rdn_m = &md->ctr.ctr1.array[md->ctr.ctr1.count];
134                 }
135         }
136
137         if (!name_d) {
138                 return WERR_FOOBAR;
139         }
140
141         ret = ldb_msg_add_value(msg, rdn_attr->lDAPDisplayName, rdn_value, NULL);
142         if (ret != LDB_SUCCESS) {
143                 return WERR_FOOBAR;
144         }
145
146         nt_status = ndr_push_struct_blob(&guid_value, msg, &in->object.identifier->guid,
147                                          (ndr_push_flags_fn_t)ndr_push_GUID);
148         if (!NT_STATUS_IS_OK(nt_status)) {
149                 return ntstatus_to_werror(nt_status);
150         }
151         ret = ldb_msg_add_value(msg, "objectGUID", &guid_value, NULL);
152         if (ret != LDB_SUCCESS) {
153                 return WERR_FOOBAR;
154         }
155
156         whenChanged_t = nt_time_to_unix(whenChanged);
157         whenChanged_s = ldb_timestring(msg, whenChanged_t);
158         W_ERROR_HAVE_NO_MEMORY(whenChanged_s);
159         ret = ldb_msg_add_string(msg, "whenChanged", whenChanged_s);
160         if (ret != LDB_SUCCESS) {
161                 return WERR_FOOBAR;
162         }
163
164         rdn_m->attid                            = rdn_attid;
165         rdn_m->version                          = name_d->version;
166         rdn_m->orginating_time                  = name_d->orginating_time;
167         rdn_m->orginating_invocation_id         = name_d->orginating_invocation_id;
168         rdn_m->orginating_usn                   = name_d->orginating_usn;
169         rdn_m->local_usn                        = 0;
170         md->ctr.ctr1.count++;
171
172         out->msg        = msg;
173         out->guid_value = guid_value;
174         out->meta_data  = md;
175         return WERR_OK;
176 }
177
178 WERROR dsdb_extended_replicated_objects_commit(struct ldb_context *ldb,
179                                                const char *partition_dn,
180                                                const struct dsdb_schema *schema,
181                                                uint32_t ctr_level,
182                                                const struct drsuapi_DsGetNCChangesCtr1 *ctr1,
183                                                const struct drsuapi_DsGetNCChangesCtr6 *ctr6,
184                                                TALLOC_CTX *mem_ctx,
185                                                struct dsdb_extended_replicated_objects **_out)
186 {
187         WERROR status;
188         struct dsdb_extended_replicated_objects *out;
189         struct ldb_result *ext_res;
190         const struct drsuapi_DsReplicaOIDMapping_Ctr *mapping_ctr;
191         uint32_t object_count;
192         struct drsuapi_DsReplicaObjectListItemEx *first_object;
193         struct drsuapi_DsReplicaObjectListItemEx *cur;
194         uint32_t i;
195         int ret;
196
197         switch (ctr_level) {
198         case 1:
199                 mapping_ctr             = &ctr1->mapping_ctr;
200                 object_count            = ctr1->object_count;
201                 first_object            = ctr1->first_object;
202                 break;
203         case 6:
204                 mapping_ctr             = &ctr6->mapping_ctr;
205                 object_count            = ctr6->object_count;
206                 first_object            = ctr6->first_object;
207                 break;
208         default:
209                 return WERR_INVALID_PARAM;
210         }
211
212         status = dsdb_verify_oid_mappings(schema, mapping_ctr);
213         W_ERROR_NOT_OK_RETURN(status);
214
215         out = talloc_zero(mem_ctx, struct dsdb_extended_replicated_objects);
216         W_ERROR_HAVE_NO_MEMORY(out);
217
218         out->partition_dn = ldb_dn_new(out, ldb, partition_dn);
219         W_ERROR_HAVE_NO_MEMORY(out->partition_dn);
220
221         out->num_objects        = object_count;
222         out->objects            = talloc_array(out,
223                                                struct dsdb_extended_replicated_object,
224                                                out->num_objects);
225         W_ERROR_HAVE_NO_MEMORY(out->objects);
226
227         for (i=0, cur = first_object; cur; cur = cur->next_object, i++) {
228                 if (i == out->num_objects) {
229                         return WERR_FOOBAR;
230                 }
231
232                 status = dsdb_convert_object(ldb, schema, out, cur, out->objects, &out->objects[i]);
233                 W_ERROR_NOT_OK_RETURN(status);
234         }
235         if (i != out->num_objects) {
236                 return WERR_FOOBAR;
237         }
238
239         /* TODO: handle linked attributes */
240
241         ret = ldb_extended(ldb, DSDB_EXTENDED_REPLICATED_OBJECTS_OID, out, &ext_res);
242         if (ret != LDB_SUCCESS) {
243                 DEBUG(0,("Failed to apply records: %d\n",
244                         ret));
245                 talloc_free(out);
246                 return WERR_FOOBAR;
247         }
248         talloc_free(ext_res);
249
250         if (_out) {
251                 *_out = out;
252         } else {
253                 talloc_free(out);
254         }
255
256         return WERR_OK;
257 }