drsuapi: make use of the 'more_data' field in DsGetNCChangesCtr[1|6]
[kamenim/samba.git] / source4 / libnet / libnet_vampire.c
1 /* 
2    Unix SMB/CIFS implementation.
3    
4    Extract the user/system database from a remote server
5
6    Copyright (C) Stefan Metzmacher      2004-2006
7    Copyright (C) Brad Henry 2005
8    Copyright (C) Andrew Bartlett <abartlet@samba.org> 2005-2008
9    
10    This program is free software; you can redistribute it and/or modify
11    it under the terms of the GNU General Public License as published by
12    the Free Software Foundation; either version 3 of the License, or
13    (at your option) any later version.
14    
15    This program is distributed in the hope that it will be useful,
16    but WITHOUT ANY WARRANTY; without even the implied warranty of
17    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18    GNU General Public License for more details.
19    
20    You should have received a copy of the GNU General Public License
21    along with this program.  If not, see <http://www.gnu.org/licenses/>.
22 */
23
24
25 #include "includes.h"
26 #include "libnet/libnet.h"
27 #include "lib/events/events.h"
28 #include "dsdb/samdb/samdb.h"
29 #include "lib/util/dlinklist.h"
30 #include "lib/ldb/include/ldb.h"
31 #include "lib/ldb/include/ldb_errors.h"
32 #include "librpc/ndr/libndr.h"
33 #include "librpc/gen_ndr/ndr_drsuapi.h"
34 #include "librpc/gen_ndr/ndr_drsblobs.h"
35 #include "librpc/gen_ndr/ndr_misc.h"
36 #include "system/time.h"
37 #include "lib/ldb_wrap.h"
38 #include "auth/auth.h"
39 #include "param/param.h"
40 #include "param/provision.h"
41
42 /* 
43 List of tasks vampire.py must perform:
44 - Domain Join
45  - but don't write the secrets.ldb
46  - results for this should be enough to handle the provision
47 - if vampire method is samsync 
48  - Provision using these results 
49   - do we still want to support this NT4 technology?
50 - Start samsync with libnet code
51  - provision in the callback 
52 - Write out the secrets database, using the code from libnet_Join
53
54 */
55 struct vampire_state {
56         const char *netbios_name;
57         struct libnet_JoinDomain *join;
58         struct cli_credentials *machine_account;
59         struct dsdb_schema *self_made_schema;
60         const struct dsdb_schema *schema;
61
62         struct ldb_context *ldb;
63
64         struct {
65                 uint32_t object_count;
66                 struct drsuapi_DsReplicaObjectListItemEx *first_object;
67                 struct drsuapi_DsReplicaObjectListItemEx *last_object;
68         } schema_part;
69
70         const char *targetdir;
71
72         struct loadparm_context *lp_ctx;
73         struct event_context *event_ctx;
74 };
75
76 static NTSTATUS vampire_prepare_db(void *private_data,
77                                               const struct libnet_BecomeDC_PrepareDB *p)
78 {
79         struct vampire_state *s = talloc_get_type(private_data, struct vampire_state);
80         struct provision_settings settings;
81         struct provision_result result;
82         NTSTATUS status;
83
84         settings.site_name = p->dest_dsa->site_name;
85         settings.root_dn_str = p->forest->root_dn_str;
86         settings.domain_dn_str = p->domain->dn_str;
87         settings.config_dn_str = p->forest->config_dn_str;
88         settings.schema_dn_str = p->forest->schema_dn_str;
89         settings.netbios_name = p->dest_dsa->netbios_name;
90         settings.realm = s->join->out.realm;
91         settings.domain = s->join->out.domain_name;
92         settings.server_dn_str = p->dest_dsa->server_dn_str;
93         settings.machine_password = generate_random_str(s, 16);
94         settings.targetdir = s->targetdir;
95
96         status = provision_bare(s, s->lp_ctx, &settings, &result);
97
98         if (!NT_STATUS_IS_OK(status)) {
99                 return status;
100         }
101
102         s->ldb = result.samdb;
103         s->lp_ctx = result.lp_ctx;
104
105         return NT_STATUS_OK;
106
107
108 }
109
110 static NTSTATUS vampire_check_options(void *private_data,
111                                              const struct libnet_BecomeDC_CheckOptions *o)
112 {
113         struct vampire_state *s = talloc_get_type(private_data, struct vampire_state);
114
115         DEBUG(0,("Become DC [%s] of Domain[%s]/[%s]\n",
116                 s->netbios_name,
117                 o->domain->netbios_name, o->domain->dns_name));
118
119         DEBUG(0,("Promotion Partner is Server[%s] from Site[%s]\n",
120                 o->source_dsa->dns_name, o->source_dsa->site_name));
121
122         DEBUG(0,("Options:crossRef behavior_version[%u]\n"
123                        "\tschema object_version[%u]\n"
124                        "\tdomain behavior_version[%u]\n"
125                        "\tdomain w2k3_update_revision[%u]\n", 
126                 o->forest->crossref_behavior_version,
127                 o->forest->schema_object_version,
128                 o->domain->behavior_version,
129                 o->domain->w2k3_update_revision));
130
131         return NT_STATUS_OK;
132 }
133
134 static NTSTATUS vampire_apply_schema(struct vampire_state *s,
135                                   const struct libnet_BecomeDC_StoreChunk *c)
136 {
137         WERROR status;
138         const struct drsuapi_DsReplicaOIDMapping_Ctr *mapping_ctr;
139         uint32_t object_count;
140         struct drsuapi_DsReplicaObjectListItemEx *first_object;
141         struct drsuapi_DsReplicaObjectListItemEx *cur;
142         uint32_t linked_attributes_count;
143         struct drsuapi_DsReplicaLinkedAttribute *linked_attributes;
144         const struct drsuapi_DsReplicaCursor2CtrEx *uptodateness_vector;
145         struct dsdb_extended_replicated_objects *objs;
146         struct repsFromTo1 *s_dsa;
147         char *tmp_dns_name;
148         struct ldb_message *msg;
149         struct ldb_val prefixMap_val;
150         struct ldb_message_element *prefixMap_el;
151         struct ldb_val schemaInfo_val;
152         uint32_t i;
153         int ret;
154         bool ok;
155
156         DEBUG(0,("Analyze and apply schema objects\n"));
157
158         s_dsa                   = talloc_zero(s, struct repsFromTo1);
159         NT_STATUS_HAVE_NO_MEMORY(s_dsa);
160         s_dsa->other_info       = talloc(s_dsa, struct repsFromTo1OtherInfo);
161         NT_STATUS_HAVE_NO_MEMORY(s_dsa->other_info);
162
163         switch (c->ctr_level) {
164         case 1:
165                 mapping_ctr                     = &c->ctr1->mapping_ctr;
166                 object_count                    = s->schema_part.object_count;
167                 first_object                    = s->schema_part.first_object;
168                 linked_attributes_count         = 0;
169                 linked_attributes               = NULL;
170                 s_dsa->highwatermark            = c->ctr1->new_highwatermark;
171                 s_dsa->source_dsa_obj_guid      = c->ctr1->source_dsa_guid;
172                 s_dsa->source_dsa_invocation_id = c->ctr1->source_dsa_invocation_id;
173                 uptodateness_vector             = NULL; /* TODO: map it */
174                 break;
175         case 6:
176                 mapping_ctr                     = &c->ctr6->mapping_ctr;
177                 object_count                    = s->schema_part.object_count;
178                 first_object                    = s->schema_part.first_object;
179                 linked_attributes_count         = 0; /* TODO: ! */
180                 linked_attributes               = NULL; /* TODO: ! */;
181                 s_dsa->highwatermark            = c->ctr6->new_highwatermark;
182                 s_dsa->source_dsa_obj_guid      = c->ctr6->source_dsa_guid;
183                 s_dsa->source_dsa_invocation_id = c->ctr6->source_dsa_invocation_id;
184                 uptodateness_vector             = c->ctr6->uptodateness_vector;
185                 break;
186         default:
187                 return NT_STATUS_INVALID_PARAMETER;
188         }
189
190         s_dsa->replica_flags            = DRSUAPI_DS_REPLICA_NEIGHBOUR_WRITEABLE
191                                         | DRSUAPI_DS_REPLICA_NEIGHBOUR_SYNC_ON_STARTUP
192                                         | DRSUAPI_DS_REPLICA_NEIGHBOUR_DO_SCHEDULED_SYNCS;
193         memset(s_dsa->schedule, 0x11, sizeof(s_dsa->schedule));
194
195         tmp_dns_name    = GUID_string(s_dsa->other_info, &s_dsa->source_dsa_obj_guid);
196         NT_STATUS_HAVE_NO_MEMORY(tmp_dns_name);
197         tmp_dns_name    = talloc_asprintf_append_buffer(tmp_dns_name, "._msdcs.%s", c->forest->dns_name);
198         NT_STATUS_HAVE_NO_MEMORY(tmp_dns_name);
199         s_dsa->other_info->dns_name = tmp_dns_name;
200
201         for (cur = first_object; cur; cur = cur->next_object) {
202                 bool is_attr = false;
203                 bool is_class = false;
204
205                 for (i=0; i < cur->object.attribute_ctr.num_attributes; i++) {
206                         struct drsuapi_DsReplicaAttribute *a;
207                         uint32_t j;
208                         const char *oid = NULL;
209
210                         a = &cur->object.attribute_ctr.attributes[i];
211                         status = dsdb_map_int2oid(s->self_made_schema, a->attid, s, &oid);
212                         if (!W_ERROR_IS_OK(status)) {
213                                 return werror_to_ntstatus(status);
214                         }
215
216                         switch (a->attid) {
217                         case DRSUAPI_ATTRIBUTE_objectClass:
218                                 for (j=0; j < a->value_ctr.num_values; j++) {
219                                         uint32_t val = 0xFFFFFFFF;
220
221                                         if (a->value_ctr.values[i].blob
222                                             && a->value_ctr.values[i].blob->length == 4) {
223                                                 val = IVAL(a->value_ctr.values[i].blob->data,0);
224                                         }
225
226                                         if (val == DRSUAPI_OBJECTCLASS_attributeSchema) {
227                                                 is_attr = true;
228                                         }
229                                         if (val == DRSUAPI_OBJECTCLASS_classSchema) {
230                                                 is_class = true;
231                                         }
232                                 }
233
234                                 break;
235                         default:
236                                 break;
237                         }
238                 }
239
240                 if (is_attr) {
241                         struct dsdb_attribute *sa;
242
243                         sa = talloc_zero(s->self_made_schema, struct dsdb_attribute);
244                         NT_STATUS_HAVE_NO_MEMORY(sa);
245
246                         status = dsdb_attribute_from_drsuapi(s->self_made_schema, &cur->object, s, sa);
247                         if (!W_ERROR_IS_OK(status)) {
248                                 return werror_to_ntstatus(status);
249                         }
250
251                         DLIST_ADD_END(s->self_made_schema->attributes, sa, struct dsdb_attribute *);
252                 }
253
254                 if (is_class) {
255                         struct dsdb_class *sc;
256
257                         sc = talloc_zero(s->self_made_schema, struct dsdb_class);
258                         NT_STATUS_HAVE_NO_MEMORY(sc);
259
260                         status = dsdb_class_from_drsuapi(s->self_made_schema, &cur->object, s, sc);
261                         if (!W_ERROR_IS_OK(status)) {
262                                 return werror_to_ntstatus(status);
263                         }
264
265                         DLIST_ADD_END(s->self_made_schema->classes, sc, struct dsdb_class *);
266                 }
267         }
268
269         /* attach the schema to the ldb */
270         ret = dsdb_set_schema(s->ldb, s->self_made_schema);
271         if (ret != LDB_SUCCESS) {
272                 return NT_STATUS_FOOBAR;
273         }
274         /* we don't want to access the self made schema anymore */
275         s->self_made_schema = NULL;
276         s->schema = dsdb_get_schema(s->ldb);
277
278         status = dsdb_extended_replicated_objects_commit(s->ldb,
279                                                          c->partition->nc.dn,
280                                                          mapping_ctr,
281                                                          object_count,
282                                                          first_object,
283                                                          linked_attributes_count,
284                                                          linked_attributes,
285                                                          s_dsa,
286                                                          uptodateness_vector,
287                                                          c->gensec_skey,
288                                                          s, &objs);
289         if (!W_ERROR_IS_OK(status)) {
290                 DEBUG(0,("Failed to commit objects: %s\n", win_errstr(status)));
291                 return werror_to_ntstatus(status);
292         }
293
294         if (lp_parm_bool(s->lp_ctx, NULL, "become dc", "dump objects", false)) {
295                 for (i=0; i < objs->num_objects; i++) {
296                         struct ldb_ldif ldif;
297                         fprintf(stdout, "#\n");
298                         ldif.changetype = LDB_CHANGETYPE_NONE;
299                         ldif.msg = objs->objects[i].msg;
300                         ldb_ldif_write_file(s->ldb, stdout, &ldif);
301                         NDR_PRINT_DEBUG(replPropertyMetaDataBlob, objs->objects[i].meta_data);
302                 }
303         }
304
305         msg = ldb_msg_new(objs);
306         NT_STATUS_HAVE_NO_MEMORY(msg);
307         msg->dn = objs->partition_dn;
308
309         status = dsdb_get_oid_mappings_ldb(s->schema, msg, &prefixMap_val, &schemaInfo_val);
310         if (!W_ERROR_IS_OK(status)) {
311                 DEBUG(0,("Failed dsdb_get_oid_mappings_ldb(%s)\n", win_errstr(status)));
312                 return werror_to_ntstatus(status);
313         }
314
315         /* we only add prefixMap here, because schemaInfo is a replicated attribute and already applied */
316         ret = ldb_msg_add_value(msg, "prefixMap", &prefixMap_val, &prefixMap_el);
317         if (ret != LDB_SUCCESS) {
318                 return NT_STATUS_FOOBAR;
319         }
320         prefixMap_el->flags = LDB_FLAG_MOD_REPLACE;
321
322         ret = ldb_modify(s->ldb, msg);
323         if (ret != LDB_SUCCESS) {
324                 DEBUG(0,("Failed to add prefixMap and schemaInfo %s\n", ldb_strerror(ret)));
325                 return NT_STATUS_FOOBAR;
326         }
327
328         talloc_free(s_dsa);
329         talloc_free(objs);
330
331         /* reopen the ldb */
332         talloc_free(s->ldb); /* this also free's the s->schema, because dsdb_set_schema() steals it */
333         s->schema = NULL;
334
335         DEBUG(0,("Reopen the SAM LDB with system credentials and a already stored schema\n"));
336         s->ldb = samdb_connect(s, s->event_ctx, s->lp_ctx, 
337                                system_session(s, s->lp_ctx));
338         if (!s->ldb) {
339                 DEBUG(0,("Failed to reopen sam.ldb\n"));
340                 return NT_STATUS_INTERNAL_DB_ERROR;
341         }
342
343         /* We must set these up to ensure the replMetaData is written correctly, before our NTDS Settings entry is replicated */
344         ok = samdb_set_ntds_invocation_id(s->ldb, &c->dest_dsa->invocation_id);
345         if (!ok) {
346                 DEBUG(0,("Failed to set cached ntds invocationId\n"));
347                 return NT_STATUS_FOOBAR;
348         }
349         ok = samdb_set_ntds_objectGUID(s->ldb, &c->dest_dsa->ntds_guid);
350         if (!ok) {
351                 DEBUG(0,("Failed to set cached ntds objectGUID\n"));
352                 return NT_STATUS_FOOBAR;
353         }
354
355         s->schema = dsdb_get_schema(s->ldb);
356         if (!s->schema) {
357                 DEBUG(0,("Failed to get loaded dsdb_schema\n"));
358                 return NT_STATUS_FOOBAR;
359         }
360
361         return NT_STATUS_OK;
362 }
363
364 static NTSTATUS vampire_schema_chunk(void *private_data,
365                                             const struct libnet_BecomeDC_StoreChunk *c)
366 {
367         struct vampire_state *s = talloc_get_type(private_data, struct vampire_state);
368         WERROR status;
369         const struct drsuapi_DsReplicaOIDMapping_Ctr *mapping_ctr;
370         uint32_t nc_object_count;
371         uint32_t object_count;
372         struct drsuapi_DsReplicaObjectListItemEx *first_object;
373         struct drsuapi_DsReplicaObjectListItemEx *cur;
374
375         switch (c->ctr_level) {
376         case 1:
377                 mapping_ctr             = &c->ctr1->mapping_ctr;
378                 nc_object_count         = c->ctr1->extended_ret; /* maybe w2k send this unexpected? */
379                 object_count            = c->ctr1->object_count;
380                 first_object            = c->ctr1->first_object;
381                 break;
382         case 6:
383                 mapping_ctr             = &c->ctr6->mapping_ctr;
384                 nc_object_count         = c->ctr6->nc_object_count;
385                 object_count            = c->ctr6->object_count;
386                 first_object            = c->ctr6->first_object;
387                 break;
388         default:
389                 return NT_STATUS_INVALID_PARAMETER;
390         }
391
392         if (nc_object_count) {
393                 DEBUG(0,("Schema-DN[%s] objects[%u/%u]\n",
394                         c->partition->nc.dn, object_count, nc_object_count));
395         } else {
396                 DEBUG(0,("Schema-DN[%s] objects[%u]\n",
397                 c->partition->nc.dn, object_count));
398         }
399
400         if (!s->schema) {
401                 s->self_made_schema = dsdb_new_schema(s, lp_iconv_convenience(s->lp_ctx));
402
403                 NT_STATUS_HAVE_NO_MEMORY(s->self_made_schema);
404
405                 status = dsdb_load_oid_mappings_drsuapi(s->self_made_schema, mapping_ctr);
406                 if (!W_ERROR_IS_OK(status)) {
407                         return werror_to_ntstatus(status);
408                 }
409
410                 s->schema = s->self_made_schema;
411         } else {
412                 status = dsdb_verify_oid_mappings_drsuapi(s->schema, mapping_ctr);
413                 if (!W_ERROR_IS_OK(status)) {
414                         return werror_to_ntstatus(status);
415                 }
416         }
417
418         if (!s->schema_part.first_object) {
419                 s->schema_part.object_count = object_count;
420                 s->schema_part.first_object = talloc_steal(s, first_object);
421         } else {
422                 s->schema_part.object_count             += object_count;
423                 s->schema_part.last_object->next_object = talloc_steal(s->schema_part.last_object,
424                                                                        first_object);
425         }
426         for (cur = first_object; cur->next_object; cur = cur->next_object) {}
427         s->schema_part.last_object = cur;
428
429         if (!c->partition->more_data) {
430                 return vampire_apply_schema(s, c);
431         }
432
433         return NT_STATUS_OK;
434 }
435
436 static NTSTATUS vampire_store_chunk(void *private_data,
437                                            const struct libnet_BecomeDC_StoreChunk *c)
438 {
439         struct vampire_state *s = talloc_get_type(private_data, struct vampire_state);
440         WERROR status;
441         const struct drsuapi_DsReplicaOIDMapping_Ctr *mapping_ctr;
442         uint32_t nc_object_count;
443         uint32_t object_count;
444         struct drsuapi_DsReplicaObjectListItemEx *first_object;
445         uint32_t linked_attributes_count;
446         struct drsuapi_DsReplicaLinkedAttribute *linked_attributes;
447         const struct drsuapi_DsReplicaCursor2CtrEx *uptodateness_vector;
448         struct dsdb_extended_replicated_objects *objs;
449         struct repsFromTo1 *s_dsa;
450         char *tmp_dns_name;
451         uint32_t i;
452
453         s_dsa                   = talloc_zero(s, struct repsFromTo1);
454         NT_STATUS_HAVE_NO_MEMORY(s_dsa);
455         s_dsa->other_info       = talloc(s_dsa, struct repsFromTo1OtherInfo);
456         NT_STATUS_HAVE_NO_MEMORY(s_dsa->other_info);
457
458         switch (c->ctr_level) {
459         case 1:
460                 mapping_ctr                     = &c->ctr1->mapping_ctr;
461                 nc_object_count                 = c->ctr1->extended_ret; /* maybe w2k send this unexpected? */
462                 object_count                    = c->ctr1->object_count;
463                 first_object                    = c->ctr1->first_object;
464                 linked_attributes_count         = 0;
465                 linked_attributes               = NULL;
466                 s_dsa->highwatermark            = c->ctr1->new_highwatermark;
467                 s_dsa->source_dsa_obj_guid      = c->ctr1->source_dsa_guid;
468                 s_dsa->source_dsa_invocation_id = c->ctr1->source_dsa_invocation_id;
469                 uptodateness_vector             = NULL; /* TODO: map it */
470                 break;
471         case 6:
472                 mapping_ctr                     = &c->ctr6->mapping_ctr;
473                 nc_object_count                 = c->ctr6->nc_object_count;
474                 object_count                    = c->ctr6->object_count;
475                 first_object                    = c->ctr6->first_object;
476                 linked_attributes_count         = c->ctr6->linked_attributes_count;
477                 linked_attributes               = c->ctr6->linked_attributes;
478                 s_dsa->highwatermark            = c->ctr6->new_highwatermark;
479                 s_dsa->source_dsa_obj_guid      = c->ctr6->source_dsa_guid;
480                 s_dsa->source_dsa_invocation_id = c->ctr6->source_dsa_invocation_id;
481                 uptodateness_vector             = c->ctr6->uptodateness_vector;
482                 break;
483         default:
484                 return NT_STATUS_INVALID_PARAMETER;
485         }
486
487         s_dsa->replica_flags            = DRSUAPI_DS_REPLICA_NEIGHBOUR_WRITEABLE
488                                         | DRSUAPI_DS_REPLICA_NEIGHBOUR_SYNC_ON_STARTUP
489                                         | DRSUAPI_DS_REPLICA_NEIGHBOUR_DO_SCHEDULED_SYNCS;
490         memset(s_dsa->schedule, 0x11, sizeof(s_dsa->schedule));
491
492         tmp_dns_name    = GUID_string(s_dsa->other_info, &s_dsa->source_dsa_obj_guid);
493         NT_STATUS_HAVE_NO_MEMORY(tmp_dns_name);
494         tmp_dns_name    = talloc_asprintf_append_buffer(tmp_dns_name, "._msdcs.%s", c->forest->dns_name);
495         NT_STATUS_HAVE_NO_MEMORY(tmp_dns_name);
496         s_dsa->other_info->dns_name = tmp_dns_name;
497
498         if (nc_object_count) {
499                 DEBUG(0,("Partition[%s] objects[%u/%u]\n",
500                         c->partition->nc.dn, object_count, nc_object_count));
501         } else {
502                 DEBUG(0,("Partition[%s] objects[%u]\n",
503                 c->partition->nc.dn, object_count));
504         }
505
506         status = dsdb_extended_replicated_objects_commit(s->ldb,
507                                                          c->partition->nc.dn,
508                                                          mapping_ctr,
509                                                          object_count,
510                                                          first_object,
511                                                          linked_attributes_count,
512                                                          linked_attributes,
513                                                          s_dsa,
514                                                          uptodateness_vector,
515                                                          c->gensec_skey,
516                                                          s, &objs);
517         if (!W_ERROR_IS_OK(status)) {
518                 DEBUG(0,("Failed to commit objects: %s\n", win_errstr(status)));
519                 return werror_to_ntstatus(status);
520         }
521
522         if (lp_parm_bool(s->lp_ctx, NULL, "become dc", "dump objects", false)) {
523                 for (i=0; i < objs->num_objects; i++) {
524                         struct ldb_ldif ldif;
525                         fprintf(stdout, "#\n");
526                         ldif.changetype = LDB_CHANGETYPE_NONE;
527                         ldif.msg = objs->objects[i].msg;
528                         ldb_ldif_write_file(s->ldb, stdout, &ldif);
529                         NDR_PRINT_DEBUG(replPropertyMetaDataBlob, objs->objects[i].meta_data);
530                 }
531         }
532         talloc_free(s_dsa);
533         talloc_free(objs);
534
535         for (i=0; i < linked_attributes_count; i++) {
536                 const struct dsdb_attribute *sa;
537
538                 if (!linked_attributes[i].identifier) {
539                         return NT_STATUS_FOOBAR;                
540                 }
541
542                 if (!linked_attributes[i].value.blob) {
543                         return NT_STATUS_FOOBAR;                
544                 }
545
546                 sa = dsdb_attribute_by_attributeID_id(s->schema,
547                                                       linked_attributes[i].attid);
548                 if (!sa) {
549                         return NT_STATUS_FOOBAR;
550                 }
551
552                 if (lp_parm_bool(s->lp_ctx, NULL, "become dc", "dump objects", false)) {
553                         DEBUG(0,("# %s\n", sa->lDAPDisplayName));
554                         NDR_PRINT_DEBUG(drsuapi_DsReplicaLinkedAttribute, &linked_attributes[i]);
555                         dump_data(0,
556                                 linked_attributes[i].value.blob->data,
557                                 linked_attributes[i].value.blob->length);
558                 }
559         }
560
561         return NT_STATUS_OK;
562 }
563
564 NTSTATUS libnet_Vampire(struct libnet_context *ctx, TALLOC_CTX *mem_ctx, 
565                         struct libnet_Vampire *r)
566 {
567         struct libnet_JoinDomain *join;
568         struct libnet_set_join_secrets *set_secrets;
569         struct libnet_BecomeDC b;
570         struct vampire_state *s;
571         struct ldb_message *msg;
572         int ldb_ret;
573         uint32_t i;
574         NTSTATUS status;
575
576         const char *account_name;
577         const char *netbios_name;
578         
579         r->out.error_string = NULL;
580
581         s = talloc_zero(mem_ctx, struct vampire_state);
582         if (!s) {
583                 return NT_STATUS_NO_MEMORY;
584         }
585
586         s->lp_ctx = ctx->lp_ctx;
587         s->event_ctx = ctx->event_ctx;
588
589         join = talloc_zero(s, struct libnet_JoinDomain);
590         if (!join) {
591                 return NT_STATUS_NO_MEMORY;
592         }
593                 
594         if (r->in.netbios_name != NULL) {
595                 netbios_name = r->in.netbios_name;
596         } else {
597                 netbios_name = talloc_reference(join, lp_netbios_name(ctx->lp_ctx));
598                 if (!netbios_name) {
599                         r->out.error_string = NULL;
600                         talloc_free(s);
601                         return NT_STATUS_NO_MEMORY;
602                 }
603         }
604
605         account_name = talloc_asprintf(join, "%s$", netbios_name);
606         if (!account_name) {
607                 r->out.error_string = NULL;
608                 talloc_free(s);
609                 return NT_STATUS_NO_MEMORY;
610         }
611         
612         join->in.domain_name    = r->in.domain_name;
613         join->in.account_name   = account_name;
614         join->in.netbios_name   = netbios_name;
615         join->in.level          = LIBNET_JOINDOMAIN_AUTOMATIC;
616         join->in.acct_type      = ACB_WSTRUST;
617         join->in.recreate_account = false;
618         status = libnet_JoinDomain(ctx, join, join);
619         if (!NT_STATUS_IS_OK(status)) {
620                 r->out.error_string = talloc_steal(mem_ctx, join->out.error_string);
621                 talloc_free(s);
622                 return status;
623         }
624         
625         s->join = join;
626
627         s->targetdir = r->in.targetdir;
628
629         ZERO_STRUCT(b);
630         b.in.domain_dns_name            = join->out.realm;
631         b.in.domain_netbios_name        = join->out.domain_name;
632         b.in.domain_sid                 = join->out.domain_sid;
633         b.in.source_dsa_address         = join->out.samr_binding->host;
634         b.in.dest_dsa_netbios_name      = netbios_name;
635
636         b.in.callbacks.private_data     = s;
637         b.in.callbacks.check_options    = vampire_check_options;
638         b.in.callbacks.prepare_db       = vampire_prepare_db;
639         b.in.callbacks.schema_chunk     = vampire_schema_chunk;
640         b.in.callbacks.config_chunk     = vampire_store_chunk;
641         b.in.callbacks.domain_chunk     = vampire_store_chunk;
642
643         status = libnet_BecomeDC(ctx, s, &b);
644         if (!NT_STATUS_IS_OK(status)) {
645                 printf("libnet_BecomeDC() failed - %s\n", nt_errstr(status));
646                 talloc_free(s);
647                 return status;
648         }
649
650         msg = ldb_msg_new(s);
651         if (!msg) {
652                 printf("ldb_msg_new() failed\n");
653                 talloc_free(s);
654                 return NT_STATUS_NO_MEMORY;
655         }
656         msg->dn = ldb_dn_new(msg, s->ldb, "@ROOTDSE");
657         if (!msg->dn) {
658                 printf("ldb_msg_new(@ROOTDSE) failed\n");
659                 talloc_free(s);
660                 return NT_STATUS_NO_MEMORY;
661         }
662
663         ldb_ret = ldb_msg_add_string(msg, "isSynchronized", "TRUE");
664         if (ldb_ret != LDB_SUCCESS) {
665                 printf("ldb_msg_add_string(msg, isSynchronized, TRUE) failed: %d\n", ldb_ret);
666                 talloc_free(s);
667                 return NT_STATUS_NO_MEMORY;
668         }
669
670         for (i=0; i < msg->num_elements; i++) {
671                 msg->elements[i].flags = LDB_FLAG_MOD_REPLACE;
672         }
673
674         printf("mark ROOTDSE with isSynchronized=TRUE\n");
675         ldb_ret = ldb_modify(s->ldb, msg);
676         if (ldb_ret != LDB_SUCCESS) {
677                 printf("ldb_modify() failed: %d\n", ldb_ret);
678                 talloc_free(s);
679                 return NT_STATUS_INTERNAL_DB_ERROR;
680         }
681
682         set_secrets = talloc_zero(s, struct libnet_set_join_secrets);
683         if (!set_secrets) {
684                 return NT_STATUS_NO_MEMORY;
685         }
686                 
687         set_secrets->in.domain_name = join->out.domain_name;
688         set_secrets->in.realm = join->out.realm;
689         set_secrets->in.account_name = account_name;
690         set_secrets->in.netbios_name = netbios_name;
691         set_secrets->in.join_type = SEC_CHAN_BDC;
692         set_secrets->in.join_password = join->out.join_password;
693         set_secrets->in.kvno = join->out.kvno;
694         set_secrets->in.domain_sid = join->out.domain_sid;
695         
696         status = libnet_set_join_secrets(ctx, set_secrets, set_secrets);
697         if (!NT_STATUS_IS_OK(status)) {
698                 r->out.error_string = talloc_steal(mem_ctx, set_secrets->out.error_string);
699                 talloc_free(s);
700                 return status;
701         }
702
703         r->out.domain_name = talloc_steal(r, join->out.domain_name);
704         r->out.domain_sid = talloc_steal(r, join->out.domain_sid);
705         talloc_free(s);
706         
707         return NT_STATUS_OK;
708
709 }