5178253ae1734f766e26f139a93db22bd85d504e
[ira/wip.git] / source4 / dsdb / common / util.c
1 /* 
2    Unix SMB/CIFS implementation.
3    Samba utility functions
4
5    Copyright (C) Andrew Tridgell 2004
6    Copyright (C) Volker Lendecke 2004
7    Copyright (C) Andrew Bartlett <abartlet@samba.org> 2006
8    Copyright (C) Jelmer Vernooij <jelmer@samba.org> 2007
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 #include "includes.h"
25 #include "events/events.h"
26 #include "ldb.h"
27 #include "ldb_errors.h"
28 #include "../lib/util/util_ldb.h"
29 #include "../lib/crypto/crypto.h"
30 #include "dsdb/samdb/samdb.h"
31 #include "libcli/security/security.h"
32 #include "librpc/gen_ndr/ndr_security.h"
33 #include "librpc/gen_ndr/ndr_misc.h"
34 #include "../libds/common/flags.h"
35 #include "dsdb/common/proto.h"
36 #include "libcli/ldap/ldap_ndr.h"
37 #include "param/param.h"
38 #include "libcli/auth/libcli_auth.h"
39 #include "librpc/gen_ndr/ndr_drsblobs.h"
40 #include "system/locale.h"
41 #include "lib/util/tsort.h"
42 #include "dsdb/common/util.h"
43
44 /*
45   search the sam for the specified attributes in a specific domain, filter on
46   objectSid being in domain_sid.
47 */
48 int samdb_search_domain(struct ldb_context *sam_ldb,
49                         TALLOC_CTX *mem_ctx, 
50                         struct ldb_dn *basedn,
51                         struct ldb_message ***res,
52                         const char * const *attrs,
53                         const struct dom_sid *domain_sid,
54                         const char *format, ...)  _PRINTF_ATTRIBUTE(7,8)
55 {
56         va_list ap;
57         int i, count;
58
59         va_start(ap, format);
60         count = gendb_search_v(sam_ldb, mem_ctx, basedn,
61                                res, attrs, format, ap);
62         va_end(ap);
63
64         i=0;
65
66         while (i<count) {
67                 struct dom_sid *entry_sid;
68
69                 entry_sid = samdb_result_dom_sid(mem_ctx, (*res)[i], "objectSid");
70
71                 if ((entry_sid == NULL) ||
72                     (!dom_sid_in_domain(domain_sid, entry_sid))) {
73                         /* Delete that entry from the result set */
74                         (*res)[i] = (*res)[count-1];
75                         count -= 1;
76                         talloc_free(entry_sid);
77                         continue;
78                 }
79                 talloc_free(entry_sid);
80                 i += 1;
81         }
82
83         return count;
84 }
85
86 /*
87   search the sam for a single string attribute in exactly 1 record
88 */
89 const char *samdb_search_string_v(struct ldb_context *sam_ldb,
90                                   TALLOC_CTX *mem_ctx,
91                                   struct ldb_dn *basedn,
92                                   const char *attr_name,
93                                   const char *format, va_list ap) _PRINTF_ATTRIBUTE(5,0)
94 {
95         int count;
96         const char *attrs[2] = { NULL, NULL };
97         struct ldb_message **res = NULL;
98
99         attrs[0] = attr_name;
100
101         count = gendb_search_v(sam_ldb, mem_ctx, basedn, &res, attrs, format, ap);
102         if (count > 1) {                
103                 DEBUG(1,("samdb: search for %s %s not single valued (count=%d)\n", 
104                          attr_name, format, count));
105         }
106         if (count != 1) {
107                 talloc_free(res);
108                 return NULL;
109         }
110
111         return samdb_result_string(res[0], attr_name, NULL);
112 }
113
114 /*
115   search the sam for a single string attribute in exactly 1 record
116 */
117 const char *samdb_search_string(struct ldb_context *sam_ldb,
118                                 TALLOC_CTX *mem_ctx,
119                                 struct ldb_dn *basedn,
120                                 const char *attr_name,
121                                 const char *format, ...) _PRINTF_ATTRIBUTE(5,6)
122 {
123         va_list ap;
124         const char *str;
125
126         va_start(ap, format);
127         str = samdb_search_string_v(sam_ldb, mem_ctx, basedn, attr_name, format, ap);
128         va_end(ap);
129
130         return str;
131 }
132
133 struct ldb_dn *samdb_search_dn(struct ldb_context *sam_ldb,
134                                TALLOC_CTX *mem_ctx,
135                                struct ldb_dn *basedn,
136                                const char *format, ...) _PRINTF_ATTRIBUTE(4,5)
137 {
138         va_list ap;
139         struct ldb_dn *ret;
140         struct ldb_message **res = NULL;
141         int count;
142
143         va_start(ap, format);
144         count = gendb_search_v(sam_ldb, mem_ctx, basedn, &res, NULL, format, ap);
145         va_end(ap);
146
147         if (count != 1) return NULL;
148
149         ret = talloc_steal(mem_ctx, res[0]->dn);
150         talloc_free(res);
151
152         return ret;
153 }
154
155 /*
156   search the sam for a dom_sid attribute in exactly 1 record
157 */
158 struct dom_sid *samdb_search_dom_sid(struct ldb_context *sam_ldb,
159                                      TALLOC_CTX *mem_ctx,
160                                      struct ldb_dn *basedn,
161                                      const char *attr_name,
162                                      const char *format, ...) _PRINTF_ATTRIBUTE(5,6)
163 {
164         va_list ap;
165         int count;
166         struct ldb_message **res;
167         const char *attrs[2] = { NULL, NULL };
168         struct dom_sid *sid;
169
170         attrs[0] = attr_name;
171
172         va_start(ap, format);
173         count = gendb_search_v(sam_ldb, mem_ctx, basedn, &res, attrs, format, ap);
174         va_end(ap);
175         if (count > 1) {                
176                 DEBUG(1,("samdb: search for %s %s not single valued (count=%d)\n", 
177                          attr_name, format, count));
178         }
179         if (count != 1) {
180                 talloc_free(res);
181                 return NULL;
182         }
183         sid = samdb_result_dom_sid(mem_ctx, res[0], attr_name);
184         talloc_free(res);
185         return sid;     
186 }
187
188 /*
189   return the count of the number of records in the sam matching the query
190 */
191 int samdb_search_count(struct ldb_context *sam_ldb,
192                        struct ldb_dn *basedn,
193                        const char *format, ...) _PRINTF_ATTRIBUTE(3,4)
194 {
195         va_list ap;
196         struct ldb_message **res;
197         const char *attrs[] = { NULL };
198         int ret;
199         TALLOC_CTX *tmp_ctx = talloc_new(sam_ldb);
200
201         va_start(ap, format);
202         ret = gendb_search_v(sam_ldb, tmp_ctx, basedn, &res, attrs, format, ap);
203         va_end(ap);
204         talloc_free(tmp_ctx);
205
206         return ret;
207 }
208
209
210 /*
211   search the sam for a single integer attribute in exactly 1 record
212 */
213 unsigned int samdb_search_uint(struct ldb_context *sam_ldb,
214                          TALLOC_CTX *mem_ctx,
215                          unsigned int default_value,
216                          struct ldb_dn *basedn,
217                          const char *attr_name,
218                          const char *format, ...) _PRINTF_ATTRIBUTE(6,7)
219 {
220         va_list ap;
221         int count;
222         struct ldb_message **res;
223         const char *attrs[2] = { NULL, NULL };
224
225         attrs[0] = attr_name;
226
227         va_start(ap, format);
228         count = gendb_search_v(sam_ldb, mem_ctx, basedn, &res, attrs, format, ap);
229         va_end(ap);
230
231         if (count != 1) {
232                 return default_value;
233         }
234
235         return samdb_result_uint(res[0], attr_name, default_value);
236 }
237
238 /*
239   search the sam for a single signed 64 bit integer attribute in exactly 1 record
240 */
241 int64_t samdb_search_int64(struct ldb_context *sam_ldb,
242                            TALLOC_CTX *mem_ctx,
243                            int64_t default_value,
244                            struct ldb_dn *basedn,
245                            const char *attr_name,
246                            const char *format, ...) _PRINTF_ATTRIBUTE(6,7)
247 {
248         va_list ap;
249         int count;
250         struct ldb_message **res;
251         const char *attrs[2] = { NULL, NULL };
252
253         attrs[0] = attr_name;
254
255         va_start(ap, format);
256         count = gendb_search_v(sam_ldb, mem_ctx, basedn, &res, attrs, format, ap);
257         va_end(ap);
258
259         if (count != 1) {
260                 return default_value;
261         }
262
263         return samdb_result_int64(res[0], attr_name, default_value);
264 }
265
266 /*
267   search the sam for multipe records each giving a single string attribute
268   return the number of matches, or -1 on error
269 */
270 int samdb_search_string_multiple(struct ldb_context *sam_ldb,
271                                  TALLOC_CTX *mem_ctx,
272                                  struct ldb_dn *basedn,
273                                  const char ***strs,
274                                  const char *attr_name,
275                                  const char *format, ...) _PRINTF_ATTRIBUTE(6,7)
276 {
277         va_list ap;
278         int count, i;
279         const char *attrs[2] = { NULL, NULL };
280         struct ldb_message **res = NULL;
281
282         attrs[0] = attr_name;
283
284         va_start(ap, format);
285         count = gendb_search_v(sam_ldb, mem_ctx, basedn, &res, attrs, format, ap);
286         va_end(ap);
287
288         if (count <= 0) {
289                 return count;
290         }
291
292         /* make sure its single valued */
293         for (i=0;i<count;i++) {
294                 if (res[i]->num_elements != 1) {
295                         DEBUG(1,("samdb: search for %s %s not single valued\n", 
296                                  attr_name, format));
297                         talloc_free(res);
298                         return -1;
299                 }
300         }
301
302         *strs = talloc_array(mem_ctx, const char *, count+1);
303         if (! *strs) {
304                 talloc_free(res);
305                 return -1;
306         }
307
308         for (i=0;i<count;i++) {
309                 (*strs)[i] = samdb_result_string(res[i], attr_name, NULL);
310         }
311         (*strs)[count] = NULL;
312
313         return count;
314 }
315
316 /*
317   pull a uint from a result set. 
318 */
319 unsigned int samdb_result_uint(const struct ldb_message *msg, const char *attr, unsigned int default_value)
320 {
321         return ldb_msg_find_attr_as_uint(msg, attr, default_value);
322 }
323
324 /*
325   pull a (signed) int64 from a result set. 
326 */
327 int64_t samdb_result_int64(const struct ldb_message *msg, const char *attr, int64_t default_value)
328 {
329         return ldb_msg_find_attr_as_int64(msg, attr, default_value);
330 }
331
332 /*
333   pull a string from a result set. 
334 */
335 const char *samdb_result_string(const struct ldb_message *msg, const char *attr, 
336                                 const char *default_value)
337 {
338         return ldb_msg_find_attr_as_string(msg, attr, default_value);
339 }
340
341 struct ldb_dn *samdb_result_dn(struct ldb_context *ldb, TALLOC_CTX *mem_ctx, const struct ldb_message *msg,
342                                const char *attr, struct ldb_dn *default_value)
343 {
344         struct ldb_dn *ret_dn = ldb_msg_find_attr_as_dn(ldb, mem_ctx, msg, attr);
345         if (!ret_dn) {
346                 return default_value;
347         }
348         return ret_dn;
349 }
350
351 /*
352   pull a rid from a objectSid in a result set. 
353 */
354 uint32_t samdb_result_rid_from_sid(TALLOC_CTX *mem_ctx, const struct ldb_message *msg, 
355                                    const char *attr, uint32_t default_value)
356 {
357         struct dom_sid *sid;
358         uint32_t rid;
359
360         sid = samdb_result_dom_sid(mem_ctx, msg, attr);
361         if (sid == NULL) {
362                 return default_value;
363         }
364         rid = sid->sub_auths[sid->num_auths-1];
365         talloc_free(sid);
366         return rid;
367 }
368
369 /*
370   pull a dom_sid structure from a objectSid in a result set. 
371 */
372 struct dom_sid *samdb_result_dom_sid(TALLOC_CTX *mem_ctx, const struct ldb_message *msg, 
373                                      const char *attr)
374 {
375         const struct ldb_val *v;
376         struct dom_sid *sid;
377         enum ndr_err_code ndr_err;
378         v = ldb_msg_find_ldb_val(msg, attr);
379         if (v == NULL) {
380                 return NULL;
381         }
382         sid = talloc(mem_ctx, struct dom_sid);
383         if (sid == NULL) {
384                 return NULL;
385         }
386         ndr_err = ndr_pull_struct_blob(v, sid, NULL, sid,
387                                        (ndr_pull_flags_fn_t)ndr_pull_dom_sid);
388         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
389                 talloc_free(sid);
390                 return NULL;
391         }
392         return sid;
393 }
394
395 /*
396   pull a guid structure from a objectGUID in a result set. 
397 */
398 struct GUID samdb_result_guid(const struct ldb_message *msg, const char *attr)
399 {
400         const struct ldb_val *v;
401         struct GUID guid;
402         NTSTATUS status;
403
404         v = ldb_msg_find_ldb_val(msg, attr);
405         if (!v) return GUID_zero();
406
407         status = GUID_from_ndr_blob(v, &guid);
408         if (!NT_STATUS_IS_OK(status)) {
409                 return GUID_zero();
410         }
411
412         return guid;
413 }
414
415 /*
416   pull a sid prefix from a objectSid in a result set. 
417   this is used to find the domain sid for a user
418 */
419 struct dom_sid *samdb_result_sid_prefix(TALLOC_CTX *mem_ctx, const struct ldb_message *msg, 
420                                         const char *attr)
421 {
422         struct dom_sid *sid = samdb_result_dom_sid(mem_ctx, msg, attr);
423         if (!sid || sid->num_auths < 1) return NULL;
424         sid->num_auths--;
425         return sid;
426 }
427
428 /*
429   pull a NTTIME in a result set. 
430 */
431 NTTIME samdb_result_nttime(const struct ldb_message *msg, const char *attr,
432                            NTTIME default_value)
433 {
434         return ldb_msg_find_attr_as_uint64(msg, attr, default_value);
435 }
436
437 /*
438  * Windows stores 0 for lastLogoff.
439  * But when a MS DC return the lastLogoff (as Logoff Time)
440  * it returns 0x7FFFFFFFFFFFFFFF, not returning this value in this case
441  * cause windows 2008 and newer version to fail for SMB requests
442  */
443 NTTIME samdb_result_last_logoff(const struct ldb_message *msg)
444 {
445         NTTIME ret = ldb_msg_find_attr_as_uint64(msg, "lastLogoff",0);
446
447         if (ret == 0)
448                 ret = 0x7FFFFFFFFFFFFFFFULL;
449
450         return ret;
451 }
452
453 /*
454  * Windows uses both 0 and 9223372036854775807 (0x7FFFFFFFFFFFFFFFULL) to
455  * indicate an account doesn't expire.
456  *
457  * When Windows initially creates an account, it sets
458  * accountExpires = 9223372036854775807 (0x7FFFFFFFFFFFFFFF).  However,
459  * when changing from an account having a specific expiration date to
460  * that account never expiring, it sets accountExpires = 0.
461  *
462  * Consolidate that logic here to allow clearer logic for account expiry in
463  * the rest of the code.
464  */
465 NTTIME samdb_result_account_expires(const struct ldb_message *msg)
466 {
467         NTTIME ret = ldb_msg_find_attr_as_uint64(msg, "accountExpires",
468                                                  0);
469
470         if (ret == 0)
471                 ret = 0x7FFFFFFFFFFFFFFFULL;
472
473         return ret;
474 }
475
476 /*
477   pull a uint64_t from a result set. 
478 */
479 uint64_t samdb_result_uint64(const struct ldb_message *msg, const char *attr,
480                              uint64_t default_value)
481 {
482         return ldb_msg_find_attr_as_uint64(msg, attr, default_value);
483 }
484
485
486 /*
487   construct the allow_password_change field from the PwdLastSet attribute and the 
488   domain password settings
489 */
490 NTTIME samdb_result_allow_password_change(struct ldb_context *sam_ldb, 
491                                           TALLOC_CTX *mem_ctx, 
492                                           struct ldb_dn *domain_dn, 
493                                           struct ldb_message *msg, 
494                                           const char *attr)
495 {
496         uint64_t attr_time = samdb_result_uint64(msg, attr, 0);
497         int64_t minPwdAge;
498
499         if (attr_time == 0) {
500                 return 0;
501         }
502
503         minPwdAge = samdb_search_int64(sam_ldb, mem_ctx, 0, domain_dn, "minPwdAge", NULL);
504
505         /* yes, this is a -= not a += as minPwdAge is stored as the negative
506            of the number of 100-nano-seconds */
507         attr_time -= minPwdAge;
508
509         return attr_time;
510 }
511
512 /*
513   construct the force_password_change field from the PwdLastSet
514   attribute, the userAccountControl and the domain password settings
515 */
516 NTTIME samdb_result_force_password_change(struct ldb_context *sam_ldb, 
517                                           TALLOC_CTX *mem_ctx, 
518                                           struct ldb_dn *domain_dn, 
519                                           struct ldb_message *msg)
520 {
521         uint64_t attr_time = samdb_result_uint64(msg, "pwdLastSet", 0);
522         uint32_t userAccountControl = samdb_result_uint64(msg, "userAccountControl", 0);
523         int64_t maxPwdAge;
524
525         /* Machine accounts don't expire, and there is a flag for 'no expiry' */
526         if (!(userAccountControl & UF_NORMAL_ACCOUNT)
527             || (userAccountControl & UF_DONT_EXPIRE_PASSWD)) {
528                 return 0x7FFFFFFFFFFFFFFFULL;
529         }
530
531         if (attr_time == 0) {
532                 return 0;
533         }
534
535         maxPwdAge = samdb_search_int64(sam_ldb, mem_ctx, 0, domain_dn, "maxPwdAge", NULL);
536         if (maxPwdAge == 0) {
537                 return 0x7FFFFFFFFFFFFFFFULL;
538         } else {
539                 attr_time -= maxPwdAge;
540         }
541
542         return attr_time;
543 }
544
545 /*
546   pull a samr_Password structutre from a result set. 
547 */
548 struct samr_Password *samdb_result_hash(TALLOC_CTX *mem_ctx, const struct ldb_message *msg, const char *attr)
549 {
550         struct samr_Password *hash = NULL;
551         const struct ldb_val *val = ldb_msg_find_ldb_val(msg, attr);
552         if (val && (val->length >= sizeof(hash->hash))) {
553                 hash = talloc(mem_ctx, struct samr_Password);
554                 memcpy(hash->hash, val->data, MIN(val->length, sizeof(hash->hash)));
555         }
556         return hash;
557 }
558
559 /*
560   pull an array of samr_Password structutres from a result set. 
561 */
562 unsigned int samdb_result_hashes(TALLOC_CTX *mem_ctx, const struct ldb_message *msg,
563                            const char *attr, struct samr_Password **hashes)
564 {
565         unsigned int count, i;
566         const struct ldb_val *val = ldb_msg_find_ldb_val(msg, attr);
567
568         *hashes = NULL;
569         if (!val) {
570                 return 0;
571         }
572         count = val->length / 16;
573         if (count == 0) {
574                 return 0;
575         }
576
577         *hashes = talloc_array(mem_ctx, struct samr_Password, count);
578         if (! *hashes) {
579                 return 0;
580         }
581
582         for (i=0;i<count;i++) {
583                 memcpy((*hashes)[i].hash, (i*16)+(char *)val->data, 16);
584         }
585
586         return count;
587 }
588
589 NTSTATUS samdb_result_passwords(TALLOC_CTX *mem_ctx, struct loadparm_context *lp_ctx, struct ldb_message *msg, 
590                                 struct samr_Password **lm_pwd, struct samr_Password **nt_pwd) 
591 {
592         struct samr_Password *lmPwdHash, *ntPwdHash;
593         if (nt_pwd) {
594                 int num_nt;
595                 num_nt = samdb_result_hashes(mem_ctx, msg, "unicodePwd", &ntPwdHash);
596                 if (num_nt == 0) {
597                         *nt_pwd = NULL;
598                 } else if (num_nt > 1) {
599                         return NT_STATUS_INTERNAL_DB_CORRUPTION;
600                 } else {
601                         *nt_pwd = &ntPwdHash[0];
602                 }
603         }
604         if (lm_pwd) {
605                 /* Ensure that if we have turned off LM
606                  * authentication, that we never use the LM hash, even
607                  * if we store it */
608                 if (lp_lanman_auth(lp_ctx)) {
609                         int num_lm;
610                         num_lm = samdb_result_hashes(mem_ctx, msg, "dBCSPwd", &lmPwdHash);
611                         if (num_lm == 0) {
612                                 *lm_pwd = NULL;
613                         } else if (num_lm > 1) {
614                                 return NT_STATUS_INTERNAL_DB_CORRUPTION;
615                         } else {
616                                 *lm_pwd = &lmPwdHash[0];
617                         }
618                 } else {
619                         *lm_pwd = NULL;
620                 }
621         }
622         return NT_STATUS_OK;
623 }
624
625 /*
626   pull a samr_LogonHours structutre from a result set. 
627 */
628 struct samr_LogonHours samdb_result_logon_hours(TALLOC_CTX *mem_ctx, struct ldb_message *msg, const char *attr)
629 {
630         struct samr_LogonHours hours;
631         const int units_per_week = 168;
632         const struct ldb_val *val = ldb_msg_find_ldb_val(msg, attr);
633         ZERO_STRUCT(hours);
634         hours.bits = talloc_array(mem_ctx, uint8_t, units_per_week);
635         if (!hours.bits) {
636                 return hours;
637         }
638         hours.units_per_week = units_per_week;
639         memset(hours.bits, 0xFF, units_per_week);
640         if (val) {
641                 memcpy(hours.bits, val->data, MIN(val->length, units_per_week));
642         }
643         return hours;
644 }
645
646 /*
647   pull a set of account_flags from a result set. 
648
649   This requires that the attributes: 
650    pwdLastSet
651    userAccountControl
652   be included in 'msg'
653 */
654 uint32_t samdb_result_acct_flags(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx, 
655                                  struct ldb_message *msg, struct ldb_dn *domain_dn)
656 {
657         uint32_t userAccountControl = ldb_msg_find_attr_as_uint(msg, "userAccountControl", 0);
658         uint32_t acct_flags = ds_uf2acb(userAccountControl);
659         NTTIME must_change_time;
660         NTTIME now;
661
662         must_change_time = samdb_result_force_password_change(sam_ctx, mem_ctx, 
663                                                               domain_dn, msg);
664
665         /* Test account expire time */
666         unix_to_nt_time(&now, time(NULL));
667         /* check for expired password */
668         if (must_change_time < now) {
669                 acct_flags |= ACB_PW_EXPIRED;
670         }
671         return acct_flags;
672 }
673
674 struct lsa_BinaryString samdb_result_parameters(TALLOC_CTX *mem_ctx,
675                                                 struct ldb_message *msg,
676                                                 const char *attr)
677 {
678         struct lsa_BinaryString s;
679         const struct ldb_val *val = ldb_msg_find_ldb_val(msg, attr);
680
681         ZERO_STRUCT(s);
682
683         if (!val) {
684                 return s;
685         }
686
687         s.array = talloc_array(mem_ctx, uint16_t, val->length/2);
688         if (!s.array) {
689                 return s;
690         }
691         s.length = s.size = val->length;
692         memcpy(s.array, val->data, val->length);
693
694         return s;
695 }
696
697 /* Find an attribute, with a particular value */
698
699 /* The current callers of this function expect a very specific
700  * behaviour: In particular, objectClass subclass equivilance is not
701  * wanted.  This means that we should not lookup the schema for the
702  * comparison function */
703 struct ldb_message_element *samdb_find_attribute(struct ldb_context *ldb, 
704                                                  const struct ldb_message *msg, 
705                                                  const char *name, const char *value)
706 {
707         int i;
708         struct ldb_message_element *el = ldb_msg_find_element(msg, name);
709
710         if (!el) {
711                 return NULL;
712         }
713
714         for (i=0;i<el->num_values;i++) {
715                 if (ldb_attr_cmp(value, (char *)el->values[i].data) == 0) {
716                         return el;
717                 }
718         }
719
720         return NULL;
721 }
722
723 int samdb_find_or_add_value(struct ldb_context *ldb, struct ldb_message *msg, const char *name, const char *set_value)
724 {
725         if (samdb_find_attribute(ldb, msg, name, set_value) == NULL) {
726                 return samdb_msg_add_string(ldb, msg, msg, name, set_value);
727         }
728         return LDB_SUCCESS;
729 }
730
731 int samdb_find_or_add_attribute(struct ldb_context *ldb, struct ldb_message *msg, const char *name, const char *set_value)
732 {
733         struct ldb_message_element *el;
734
735         el = ldb_msg_find_element(msg, name);
736         if (el) {
737                 return LDB_SUCCESS;
738         }
739
740         return samdb_msg_add_string(ldb, msg, msg, name, set_value);
741 }
742
743
744
745 /*
746   add a string element to a message
747 */
748 int samdb_msg_add_string(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
749                          const char *attr_name, const char *str)
750 {
751         char *s = talloc_strdup(mem_ctx, str);
752         char *a = talloc_strdup(mem_ctx, attr_name);
753         if (s == NULL || a == NULL) {
754                 return LDB_ERR_OPERATIONS_ERROR;
755         }
756         return ldb_msg_add_string(msg, a, s);
757 }
758
759 /*
760   add a dom_sid element to a message
761 */
762 int samdb_msg_add_dom_sid(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
763                          const char *attr_name, struct dom_sid *sid)
764 {
765         struct ldb_val v;
766         enum ndr_err_code ndr_err;
767
768         ndr_err = ndr_push_struct_blob(&v, mem_ctx, 
769                                        lp_iconv_convenience(ldb_get_opaque(sam_ldb, "loadparm")),
770                                        sid,
771                                        (ndr_push_flags_fn_t)ndr_push_dom_sid);
772         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
773                 return -1;
774         }
775         return ldb_msg_add_value(msg, attr_name, &v, NULL);
776 }
777
778
779 /*
780   add a delete element operation to a message
781 */
782 int samdb_msg_add_delete(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
783                          const char *attr_name)
784 {
785         /* we use an empty replace rather than a delete, as it allows for 
786            dsdb_replace() to be used everywhere */
787         return ldb_msg_add_empty(msg, attr_name, LDB_FLAG_MOD_REPLACE, NULL);
788 }
789
790 /*
791   add a add attribute value to a message
792 */
793 int samdb_msg_add_addval(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
794                          const char *attr_name, const char *value)
795 {
796         struct ldb_message_element *el;
797         char *a, *v;
798         int ret;
799         a = talloc_strdup(mem_ctx, attr_name);
800         if (a == NULL)
801                 return -1;
802         v = talloc_strdup(mem_ctx, value);
803         if (v == NULL)
804                 return -1;
805         ret = ldb_msg_add_string(msg, a, v);
806         if (ret != 0)
807                 return ret;
808         el = ldb_msg_find_element(msg, a);
809         if (el == NULL)
810                 return -1;
811         el->flags = LDB_FLAG_MOD_ADD;
812         return 0;
813 }
814
815 /*
816   add a delete attribute value to a message
817 */
818 int samdb_msg_add_delval(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
819                          const char *attr_name, const char *value)
820 {
821         struct ldb_message_element *el;
822         char *a, *v;
823         int ret;
824         a = talloc_strdup(mem_ctx, attr_name);
825         if (a == NULL)
826                 return -1;
827         v = talloc_strdup(mem_ctx, value);
828         if (v == NULL)
829                 return -1;
830         ret = ldb_msg_add_string(msg, a, v);
831         if (ret != 0)
832                 return ret;
833         el = ldb_msg_find_element(msg, a);
834         if (el == NULL)
835                 return -1;
836         el->flags = LDB_FLAG_MOD_DELETE;
837         return 0;
838 }
839
840 /*
841   add a int element to a message
842 */
843 int samdb_msg_add_int(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
844                        const char *attr_name, int v)
845 {
846         const char *s = talloc_asprintf(mem_ctx, "%d", v);
847         return samdb_msg_add_string(sam_ldb, mem_ctx, msg, attr_name, s);
848 }
849
850 /*
851   add a unsigned int element to a message
852 */
853 int samdb_msg_add_uint(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
854                        const char *attr_name, unsigned int v)
855 {
856         return samdb_msg_add_int(sam_ldb, mem_ctx, msg, attr_name, (int)v);
857 }
858
859 /*
860   add a (signed) int64_t element to a message
861 */
862 int samdb_msg_add_int64(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
863                         const char *attr_name, int64_t v)
864 {
865         const char *s = talloc_asprintf(mem_ctx, "%lld", (long long)v);
866         return samdb_msg_add_string(sam_ldb, mem_ctx, msg, attr_name, s);
867 }
868
869 /*
870   add a uint64_t element to a message
871 */
872 int samdb_msg_add_uint64(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
873                         const char *attr_name, uint64_t v)
874 {
875         return samdb_msg_add_int64(sam_ldb, mem_ctx, msg, attr_name, (int64_t)v);
876 }
877
878 /*
879   add a samr_Password element to a message
880 */
881 int samdb_msg_add_hash(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
882                        const char *attr_name, struct samr_Password *hash)
883 {
884         struct ldb_val val;
885         val.data = talloc_memdup(mem_ctx, hash->hash, 16);
886         if (!val.data) {
887                 return -1;
888         }
889         val.length = 16;
890         return ldb_msg_add_value(msg, attr_name, &val, NULL);
891 }
892
893 /*
894   add a samr_Password array to a message
895 */
896 int samdb_msg_add_hashes(TALLOC_CTX *mem_ctx, struct ldb_message *msg,
897                          const char *attr_name, struct samr_Password *hashes, unsigned int count)
898 {
899         struct ldb_val val;
900         int i;
901         val.data = talloc_array_size(mem_ctx, 16, count);
902         val.length = count*16;
903         if (!val.data) {
904                 return -1;
905         }
906         for (i=0;i<count;i++) {
907                 memcpy(i*16 + (char *)val.data, hashes[i].hash, 16);
908         }
909         return ldb_msg_add_value(msg, attr_name, &val, NULL);
910 }
911
912 /*
913   add a acct_flags element to a message
914 */
915 int samdb_msg_add_acct_flags(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
916                              const char *attr_name, uint32_t v)
917 {
918         return samdb_msg_add_uint(sam_ldb, mem_ctx, msg, attr_name, ds_acb2uf(v));
919 }
920
921 /*
922   add a logon_hours element to a message
923 */
924 int samdb_msg_add_logon_hours(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
925                               const char *attr_name, struct samr_LogonHours *hours)
926 {
927         struct ldb_val val;
928         val.length = hours->units_per_week / 8;
929         val.data = hours->bits;
930         return ldb_msg_add_value(msg, attr_name, &val, NULL);
931 }
932
933 /*
934   add a parameters element to a message
935 */
936 int samdb_msg_add_parameters(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
937                              const char *attr_name, struct lsa_BinaryString *parameters)
938 {
939         struct ldb_val val;
940         val.length = parameters->length;
941         val.data = (uint8_t *)parameters->array;
942         return ldb_msg_add_value(msg, attr_name, &val, NULL);
943 }
944 /*
945   add a general value element to a message
946 */
947 int samdb_msg_add_value(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
948                               const char *attr_name, const struct ldb_val *val)
949 {
950         return ldb_msg_add_value(msg, attr_name, val, NULL);
951 }
952
953 /*
954   sets a general value element to a message
955 */
956 int samdb_msg_set_value(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
957                         const char *attr_name, const struct ldb_val *val)
958 {
959         struct ldb_message_element *el;
960
961         el = ldb_msg_find_element(msg, attr_name);
962         if (el) {
963                 el->num_values = 0;
964         }
965         return ldb_msg_add_value(msg, attr_name, val, NULL);
966 }
967
968 /*
969   set a string element in a message
970 */
971 int samdb_msg_set_string(struct ldb_context *sam_ldb, TALLOC_CTX *mem_ctx, struct ldb_message *msg,
972                          const char *attr_name, const char *str)
973 {
974         struct ldb_message_element *el;
975
976         el = ldb_msg_find_element(msg, attr_name);
977         if (el) {
978                 el->num_values = 0;
979         }
980         return samdb_msg_add_string(sam_ldb, mem_ctx, msg, attr_name, str);
981 }
982
983 /*
984  * Handle ldb_request in transaction
985  */
986 static int dsdb_autotransaction_request(struct ldb_context *sam_ldb,
987                                         struct ldb_request *req)
988 {
989         int ret;
990
991         ret = ldb_transaction_start(sam_ldb);
992         if (ret != LDB_SUCCESS) {
993                 return ret;
994         }
995
996         ret = ldb_request(sam_ldb, req);
997         if (ret == LDB_SUCCESS) {
998                 ret = ldb_wait(req->handle, LDB_WAIT_ALL);
999         }
1000
1001         if (ret == LDB_SUCCESS) {
1002                 return ldb_transaction_commit(sam_ldb);
1003         }
1004         ldb_transaction_cancel(sam_ldb);
1005
1006         return ret;
1007 }
1008
1009 /*
1010   return a default security descriptor
1011 */
1012 struct security_descriptor *samdb_default_security_descriptor(TALLOC_CTX *mem_ctx)
1013 {
1014         struct security_descriptor *sd;
1015
1016         sd = security_descriptor_initialise(mem_ctx);
1017
1018         return sd;
1019 }
1020
1021 struct ldb_dn *samdb_base_dn(struct ldb_context *sam_ctx) 
1022 {
1023         return ldb_get_default_basedn(sam_ctx);
1024 }
1025
1026 struct ldb_dn *samdb_config_dn(struct ldb_context *sam_ctx) 
1027 {
1028         return ldb_get_config_basedn(sam_ctx);
1029 }
1030
1031 struct ldb_dn *samdb_schema_dn(struct ldb_context *sam_ctx) 
1032 {
1033         return ldb_get_schema_basedn(sam_ctx);
1034 }
1035
1036 struct ldb_dn *samdb_aggregate_schema_dn(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx) 
1037 {
1038         struct ldb_dn *schema_dn = ldb_get_schema_basedn(sam_ctx);
1039         struct ldb_dn *aggregate_dn;
1040         if (!schema_dn) {
1041                 return NULL;
1042         }
1043
1044         aggregate_dn = ldb_dn_copy(mem_ctx, schema_dn);
1045         if (!aggregate_dn) {
1046                 return NULL;
1047         }
1048         if (!ldb_dn_add_child_fmt(aggregate_dn, "CN=Aggregate")) {
1049                 return NULL;
1050         }
1051         return aggregate_dn;
1052 }
1053
1054 struct ldb_dn *samdb_root_dn(struct ldb_context *sam_ctx) 
1055 {
1056         return ldb_get_root_basedn(sam_ctx);
1057 }
1058
1059 struct ldb_dn *samdb_partitions_dn(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx)
1060 {
1061         struct ldb_dn *new_dn;
1062
1063         new_dn = ldb_dn_copy(mem_ctx, samdb_config_dn(sam_ctx));
1064         if ( ! ldb_dn_add_child_fmt(new_dn, "CN=Partitions")) {
1065                 talloc_free(new_dn);
1066                 return NULL;
1067         }
1068         return new_dn;
1069 }
1070
1071 struct ldb_dn *samdb_sites_dn(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx)
1072 {
1073         struct ldb_dn *new_dn;
1074
1075         new_dn = ldb_dn_copy(mem_ctx, samdb_config_dn(sam_ctx));
1076         if ( ! ldb_dn_add_child_fmt(new_dn, "CN=Sites")) {
1077                 talloc_free(new_dn);
1078                 return NULL;
1079         }
1080         return new_dn;
1081 }
1082
1083 /*
1084   work out the domain sid for the current open ldb
1085 */
1086 const struct dom_sid *samdb_domain_sid(struct ldb_context *ldb)
1087 {
1088         TALLOC_CTX *tmp_ctx;
1089         const struct dom_sid *domain_sid;
1090         const char *attrs[] = {
1091                 "objectSid",
1092                 NULL
1093         };
1094         struct ldb_result *res;
1095         int ret;
1096
1097         /* see if we have a cached copy */
1098         domain_sid = (struct dom_sid *)ldb_get_opaque(ldb, "cache.domain_sid");
1099         if (domain_sid) {
1100                 return domain_sid;
1101         }
1102
1103         tmp_ctx = talloc_new(ldb);
1104         if (tmp_ctx == NULL) {
1105                 goto failed;
1106         }
1107
1108         ret = ldb_search(ldb, tmp_ctx, &res, ldb_get_default_basedn(ldb), LDB_SCOPE_BASE, attrs, "objectSid=*");
1109
1110         if (ret != LDB_SUCCESS) {
1111                 goto failed;
1112         }
1113
1114         if (res->count != 1) {
1115                 goto failed;
1116         }
1117
1118         domain_sid = samdb_result_dom_sid(tmp_ctx, res->msgs[0], "objectSid");
1119         if (domain_sid == NULL) {
1120                 goto failed;
1121         }
1122
1123         /* cache the domain_sid in the ldb */
1124         if (ldb_set_opaque(ldb, "cache.domain_sid", discard_const_p(struct dom_sid, domain_sid)) != LDB_SUCCESS) {
1125                 goto failed;
1126         }
1127
1128         talloc_steal(ldb, domain_sid);
1129         talloc_free(tmp_ctx);
1130
1131         return domain_sid;
1132
1133 failed:
1134         talloc_free(tmp_ctx);
1135         return NULL;
1136 }
1137
1138 /*
1139   get domain sid from cache
1140 */
1141 const struct dom_sid *samdb_domain_sid_cache_only(struct ldb_context *ldb)
1142 {
1143         return (struct dom_sid *)ldb_get_opaque(ldb, "cache.domain_sid");
1144 }
1145
1146 bool samdb_set_domain_sid(struct ldb_context *ldb, const struct dom_sid *dom_sid_in)
1147 {
1148         TALLOC_CTX *tmp_ctx;
1149         struct dom_sid *dom_sid_new;
1150         struct dom_sid *dom_sid_old;
1151
1152         /* see if we have a cached copy */
1153         dom_sid_old = talloc_get_type(ldb_get_opaque(ldb, 
1154                                                      "cache.domain_sid"), struct dom_sid);
1155
1156         tmp_ctx = talloc_new(ldb);
1157         if (tmp_ctx == NULL) {
1158                 goto failed;
1159         }
1160
1161         dom_sid_new = dom_sid_dup(tmp_ctx, dom_sid_in);
1162         if (!dom_sid_new) {
1163                 goto failed;
1164         }
1165
1166         /* cache the domain_sid in the ldb */
1167         if (ldb_set_opaque(ldb, "cache.domain_sid", dom_sid_new) != LDB_SUCCESS) {
1168                 goto failed;
1169         }
1170
1171         talloc_steal(ldb, dom_sid_new);
1172         talloc_free(tmp_ctx);
1173         talloc_free(dom_sid_old);
1174
1175         return true;
1176
1177 failed:
1178         DEBUG(1,("Failed to set our own cached domain SID in the ldb!\n"));
1179         talloc_free(tmp_ctx);
1180         return false;
1181 }
1182
1183 /* Obtain the short name of the flexible single master operator
1184  * (FSMO), such as the PDC Emulator */
1185 const char *samdb_result_fsmo_name(struct ldb_context *ldb, TALLOC_CTX *mem_ctx, const struct ldb_message *msg, 
1186                              const char *attr)
1187 {
1188         /* Format is cn=NTDS Settings,cn=<NETBIOS name of FSMO>,.... */
1189         struct ldb_dn *fsmo_dn = ldb_msg_find_attr_as_dn(ldb, mem_ctx, msg, attr);
1190         const struct ldb_val *val = ldb_dn_get_component_val(fsmo_dn, 1);
1191         const char *name = ldb_dn_get_component_name(fsmo_dn, 1);
1192
1193         if (!name || (ldb_attr_cmp(name, "cn") != 0)) {
1194                 /* Ensure this matches the format.  This gives us a
1195                  * bit more confidence that a 'cn' value will be a
1196                  * ascii string */
1197                 return NULL;
1198         }
1199         if (val) {
1200                 return (char *)val->data;
1201         }
1202         return NULL;
1203 }
1204
1205 /*
1206   work out the ntds settings dn for the current open ldb
1207 */
1208 struct ldb_dn *samdb_ntds_settings_dn(struct ldb_context *ldb)
1209 {
1210         TALLOC_CTX *tmp_ctx;
1211         const char *root_attrs[] = { "dsServiceName", NULL };
1212         int ret;
1213         struct ldb_result *root_res;
1214         struct ldb_dn *settings_dn;
1215
1216         /* see if we have a cached copy */
1217         settings_dn = (struct ldb_dn *)ldb_get_opaque(ldb, "cache.settings_dn");
1218         if (settings_dn) {
1219                 return settings_dn;
1220         }
1221
1222         tmp_ctx = talloc_new(ldb);
1223         if (tmp_ctx == NULL) {
1224                 goto failed;
1225         }
1226
1227         ret = ldb_search(ldb, tmp_ctx, &root_res, ldb_dn_new(tmp_ctx, ldb, ""), LDB_SCOPE_BASE, root_attrs, NULL);
1228         if (ret) {
1229                 DEBUG(1,("Searching for dsServiceName in rootDSE failed: %s\n", 
1230                          ldb_errstring(ldb)));
1231                 goto failed;
1232         }
1233
1234         if (root_res->count != 1) {
1235                 goto failed;
1236         }
1237
1238         settings_dn = ldb_msg_find_attr_as_dn(ldb, tmp_ctx, root_res->msgs[0], "dsServiceName");
1239
1240         /* cache the domain_sid in the ldb */
1241         if (ldb_set_opaque(ldb, "cache.settings_dn", settings_dn) != LDB_SUCCESS) {
1242                 goto failed;
1243         }
1244
1245         talloc_steal(ldb, settings_dn);
1246         talloc_free(tmp_ctx);
1247
1248         return settings_dn;
1249
1250 failed:
1251         DEBUG(1,("Failed to find our own NTDS Settings DN in the ldb!\n"));
1252         talloc_free(tmp_ctx);
1253         return NULL;
1254 }
1255
1256 /*
1257   work out the ntds settings invocationId for the current open ldb
1258 */
1259 const struct GUID *samdb_ntds_invocation_id(struct ldb_context *ldb)
1260 {
1261         TALLOC_CTX *tmp_ctx;
1262         const char *attrs[] = { "invocationId", NULL };
1263         int ret;
1264         struct ldb_result *res;
1265         struct GUID *invocation_id;
1266
1267         /* see if we have a cached copy */
1268         invocation_id = (struct GUID *)ldb_get_opaque(ldb, "cache.invocation_id");
1269         if (invocation_id) {
1270                 return invocation_id;
1271         }
1272
1273         tmp_ctx = talloc_new(ldb);
1274         if (tmp_ctx == NULL) {
1275                 goto failed;
1276         }
1277
1278         ret = ldb_search(ldb, tmp_ctx, &res, samdb_ntds_settings_dn(ldb), LDB_SCOPE_BASE, attrs, NULL);
1279         if (ret) {
1280                 goto failed;
1281         }
1282
1283         if (res->count != 1) {
1284                 goto failed;
1285         }
1286
1287         invocation_id = talloc(tmp_ctx, struct GUID);
1288         if (!invocation_id) {
1289                 goto failed;
1290         }
1291
1292         *invocation_id = samdb_result_guid(res->msgs[0], "invocationId");
1293
1294         /* cache the domain_sid in the ldb */
1295         if (ldb_set_opaque(ldb, "cache.invocation_id", invocation_id) != LDB_SUCCESS) {
1296                 goto failed;
1297         }
1298
1299         talloc_steal(ldb, invocation_id);
1300         talloc_free(tmp_ctx);
1301
1302         return invocation_id;
1303
1304 failed:
1305         DEBUG(1,("Failed to find our own NTDS Settings invocationId in the ldb!\n"));
1306         talloc_free(tmp_ctx);
1307         return NULL;
1308 }
1309
1310 bool samdb_set_ntds_invocation_id(struct ldb_context *ldb, const struct GUID *invocation_id_in)
1311 {
1312         TALLOC_CTX *tmp_ctx;
1313         struct GUID *invocation_id_new;
1314         struct GUID *invocation_id_old;
1315
1316         /* see if we have a cached copy */
1317         invocation_id_old = (struct GUID *)ldb_get_opaque(ldb, 
1318                                                          "cache.invocation_id");
1319
1320         tmp_ctx = talloc_new(ldb);
1321         if (tmp_ctx == NULL) {
1322                 goto failed;
1323         }
1324
1325         invocation_id_new = talloc(tmp_ctx, struct GUID);
1326         if (!invocation_id_new) {
1327                 goto failed;
1328         }
1329
1330         *invocation_id_new = *invocation_id_in;
1331
1332         /* cache the domain_sid in the ldb */
1333         if (ldb_set_opaque(ldb, "cache.invocation_id", invocation_id_new) != LDB_SUCCESS) {
1334                 goto failed;
1335         }
1336
1337         talloc_steal(ldb, invocation_id_new);
1338         talloc_free(tmp_ctx);
1339         talloc_free(invocation_id_old);
1340
1341         return true;
1342
1343 failed:
1344         DEBUG(1,("Failed to set our own cached invocationId in the ldb!\n"));
1345         talloc_free(tmp_ctx);
1346         return false;
1347 }
1348
1349 /*
1350   work out the ntds settings objectGUID for the current open ldb
1351 */
1352 const struct GUID *samdb_ntds_objectGUID(struct ldb_context *ldb)
1353 {
1354         TALLOC_CTX *tmp_ctx;
1355         const char *attrs[] = { "objectGUID", NULL };
1356         int ret;
1357         struct ldb_result *res;
1358         struct GUID *ntds_guid;
1359
1360         /* see if we have a cached copy */
1361         ntds_guid = (struct GUID *)ldb_get_opaque(ldb, "cache.ntds_guid");
1362         if (ntds_guid) {
1363                 return ntds_guid;
1364         }
1365
1366         tmp_ctx = talloc_new(ldb);
1367         if (tmp_ctx == NULL) {
1368                 goto failed;
1369         }
1370
1371         ret = ldb_search(ldb, tmp_ctx, &res, samdb_ntds_settings_dn(ldb), LDB_SCOPE_BASE, attrs, NULL);
1372         if (ret) {
1373                 goto failed;
1374         }
1375
1376         if (res->count != 1) {
1377                 goto failed;
1378         }
1379
1380         ntds_guid = talloc(tmp_ctx, struct GUID);
1381         if (!ntds_guid) {
1382                 goto failed;
1383         }
1384
1385         *ntds_guid = samdb_result_guid(res->msgs[0], "objectGUID");
1386
1387         /* cache the domain_sid in the ldb */
1388         if (ldb_set_opaque(ldb, "cache.ntds_guid", ntds_guid) != LDB_SUCCESS) {
1389                 goto failed;
1390         }
1391
1392         talloc_steal(ldb, ntds_guid);
1393         talloc_free(tmp_ctx);
1394
1395         return ntds_guid;
1396
1397 failed:
1398         DEBUG(1,("Failed to find our own NTDS Settings objectGUID in the ldb!\n"));
1399         talloc_free(tmp_ctx);
1400         return NULL;
1401 }
1402
1403 bool samdb_set_ntds_objectGUID(struct ldb_context *ldb, const struct GUID *ntds_guid_in)
1404 {
1405         TALLOC_CTX *tmp_ctx;
1406         struct GUID *ntds_guid_new;
1407         struct GUID *ntds_guid_old;
1408
1409         /* see if we have a cached copy */
1410         ntds_guid_old = (struct GUID *)ldb_get_opaque(ldb, "cache.ntds_guid");
1411
1412         tmp_ctx = talloc_new(ldb);
1413         if (tmp_ctx == NULL) {
1414                 goto failed;
1415         }
1416
1417         ntds_guid_new = talloc(tmp_ctx, struct GUID);
1418         if (!ntds_guid_new) {
1419                 goto failed;
1420         }
1421
1422         *ntds_guid_new = *ntds_guid_in;
1423
1424         /* cache the domain_sid in the ldb */
1425         if (ldb_set_opaque(ldb, "cache.ntds_guid", ntds_guid_new) != LDB_SUCCESS) {
1426                 goto failed;
1427         }
1428
1429         talloc_steal(ldb, ntds_guid_new);
1430         talloc_free(tmp_ctx);
1431         talloc_free(ntds_guid_old);
1432
1433         return true;
1434
1435 failed:
1436         DEBUG(1,("Failed to set our own cached invocationId in the ldb!\n"));
1437         talloc_free(tmp_ctx);
1438         return false;
1439 }
1440
1441 /*
1442   work out the server dn for the current open ldb
1443 */
1444 struct ldb_dn *samdb_server_dn(struct ldb_context *ldb, TALLOC_CTX *mem_ctx)
1445 {
1446         return ldb_dn_get_parent(mem_ctx, samdb_ntds_settings_dn(ldb));
1447 }
1448
1449 /*
1450   work out the server dn for the current open ldb
1451 */
1452 struct ldb_dn *samdb_server_site_dn(struct ldb_context *ldb, TALLOC_CTX *mem_ctx)
1453 {
1454         struct ldb_dn *server_dn;
1455         struct ldb_dn *server_site_dn;
1456
1457         server_dn = samdb_server_dn(ldb, mem_ctx);
1458         if (!server_dn) return NULL;
1459
1460         server_site_dn = ldb_dn_get_parent(mem_ctx, server_dn);
1461
1462         talloc_free(server_dn);
1463         return server_site_dn;
1464 }
1465
1466 /*
1467   find a 'reference' DN that points at another object
1468   (eg. serverReference, rIDManagerReference etc)
1469  */
1470 int samdb_reference_dn(struct ldb_context *ldb, TALLOC_CTX *mem_ctx, struct ldb_dn *base,
1471                        const char *attribute, struct ldb_dn **dn)
1472 {
1473         const char *attrs[2];
1474         struct ldb_result *res;
1475         int ret;
1476
1477         attrs[0] = attribute;
1478         attrs[1] = NULL;
1479
1480         ret = ldb_search(ldb, mem_ctx, &res, base, LDB_SCOPE_BASE, attrs, NULL);
1481         if (ret != LDB_SUCCESS) {
1482                 return ret;
1483         }
1484         if (res->count != 1) {
1485                 talloc_free(res);
1486                 return LDB_ERR_NO_SUCH_OBJECT;
1487         }
1488
1489         *dn = ldb_msg_find_attr_as_dn(ldb, mem_ctx, res->msgs[0], attribute);
1490         if (!*dn) {
1491                 talloc_free(res);
1492                 return LDB_ERR_NO_SUCH_ATTRIBUTE;
1493         }
1494
1495         talloc_free(res);
1496         return LDB_SUCCESS;
1497 }
1498
1499 /*
1500   find our machine account via the serverReference attribute in the
1501   server DN
1502  */
1503 int samdb_server_reference_dn(struct ldb_context *ldb, TALLOC_CTX *mem_ctx, struct ldb_dn **dn)
1504 {
1505         struct ldb_dn *server_dn;
1506         int ret;
1507
1508         server_dn = samdb_server_dn(ldb, mem_ctx);
1509         if (server_dn == NULL) {
1510                 return LDB_ERR_NO_SUCH_OBJECT;
1511         }
1512
1513         ret = samdb_reference_dn(ldb, mem_ctx, server_dn, "serverReference", dn);
1514         talloc_free(server_dn);
1515
1516         return ret;
1517 }
1518
1519 /*
1520   find the RID Manager$ DN via the rIDManagerReference attribute in the
1521   base DN
1522  */
1523 int samdb_rid_manager_dn(struct ldb_context *ldb, TALLOC_CTX *mem_ctx, struct ldb_dn **dn)
1524 {
1525         return samdb_reference_dn(ldb, mem_ctx, samdb_base_dn(ldb), "rIDManagerReference", dn);
1526 }
1527
1528 /*
1529   find the RID Set DN via the rIDSetReferences attribute in our
1530   machine account DN
1531  */
1532 int samdb_rid_set_dn(struct ldb_context *ldb, TALLOC_CTX *mem_ctx, struct ldb_dn **dn)
1533 {
1534         struct ldb_dn *server_ref_dn;
1535         int ret;
1536
1537         ret = samdb_server_reference_dn(ldb, mem_ctx, &server_ref_dn);
1538         if (ret != LDB_SUCCESS) {
1539                 return ret;
1540         }
1541         ret = samdb_reference_dn(ldb, mem_ctx, server_ref_dn, "rIDSetReferences", dn);
1542         talloc_free(server_ref_dn);
1543         return ret;
1544 }
1545
1546 const char *samdb_server_site_name(struct ldb_context *ldb, TALLOC_CTX *mem_ctx)
1547 {
1548         const struct ldb_val *val = ldb_dn_get_rdn_val(samdb_server_site_dn(ldb, mem_ctx));
1549
1550         if (val != NULL)
1551                 return (const char *) val->data;
1552         else
1553                 return NULL;
1554 }
1555
1556 /*
1557   work out if we are the PDC for the domain of the current open ldb
1558 */
1559 bool samdb_is_pdc(struct ldb_context *ldb)
1560 {
1561         const char *dom_attrs[] = { "fSMORoleOwner", NULL };
1562         int ret;
1563         struct ldb_result *dom_res;
1564         TALLOC_CTX *tmp_ctx;
1565         bool is_pdc;
1566         struct ldb_dn *pdc;
1567
1568         tmp_ctx = talloc_new(ldb);
1569         if (tmp_ctx == NULL) {
1570                 DEBUG(1, ("talloc_new failed in samdb_is_pdc"));
1571                 return false;
1572         }
1573
1574         ret = ldb_search(ldb, tmp_ctx, &dom_res, ldb_get_default_basedn(ldb), LDB_SCOPE_BASE, dom_attrs, NULL);
1575         if (ret) {
1576                 DEBUG(1,("Searching for fSMORoleOwner in %s failed: %s\n", 
1577                          ldb_dn_get_linearized(ldb_get_default_basedn(ldb)), 
1578                          ldb_errstring(ldb)));
1579                 goto failed;
1580         }
1581         if (dom_res->count != 1) {
1582                 goto failed;
1583         }
1584
1585         pdc = ldb_msg_find_attr_as_dn(ldb, tmp_ctx, dom_res->msgs[0], "fSMORoleOwner");
1586
1587         if (ldb_dn_compare(samdb_ntds_settings_dn(ldb), pdc) == 0) {
1588                 is_pdc = true;
1589         } else {
1590                 is_pdc = false;
1591         }
1592
1593         talloc_free(tmp_ctx);
1594
1595         return is_pdc;
1596
1597 failed:
1598         DEBUG(1,("Failed to find if we are the PDC for this ldb\n"));
1599         talloc_free(tmp_ctx);
1600         return false;
1601 }
1602
1603 /*
1604   work out if we are a Global Catalog server for the domain of the current open ldb
1605 */
1606 bool samdb_is_gc(struct ldb_context *ldb)
1607 {
1608         const char *attrs[] = { "options", NULL };
1609         int ret, options;
1610         struct ldb_result *res;
1611         TALLOC_CTX *tmp_ctx;
1612
1613         tmp_ctx = talloc_new(ldb);
1614         if (tmp_ctx == NULL) {
1615                 DEBUG(1, ("talloc_new failed in samdb_is_pdc"));
1616                 return false;
1617         }
1618
1619         /* Query cn=ntds settings,.... */
1620         ret = ldb_search(ldb, tmp_ctx, &res, samdb_ntds_settings_dn(ldb), LDB_SCOPE_BASE, attrs, NULL);
1621         if (ret) {
1622                 talloc_free(tmp_ctx);
1623                 return false;
1624         }
1625         if (res->count != 1) {
1626                 talloc_free(tmp_ctx);
1627                 return false;
1628         }
1629
1630         options = ldb_msg_find_attr_as_int(res->msgs[0], "options", 0);
1631         talloc_free(tmp_ctx);
1632
1633         /* if options attribute has the 0x00000001 flag set, then enable the global catlog */
1634         if (options & 0x000000001) {
1635                 return true;
1636         }
1637         return false;
1638 }
1639
1640 /* Find a domain object in the parents of a particular DN.  */
1641 int samdb_search_for_parent_domain(struct ldb_context *ldb, TALLOC_CTX *mem_ctx, struct ldb_dn *dn,
1642                                    struct ldb_dn **parent_dn, const char **errstring)
1643 {
1644         TALLOC_CTX *local_ctx;
1645         struct ldb_dn *sdn = dn;
1646         struct ldb_result *res = NULL;
1647         int ret = 0;
1648         const char *attrs[] = { NULL };
1649
1650         local_ctx = talloc_new(mem_ctx);
1651         if (local_ctx == NULL) return LDB_ERR_OPERATIONS_ERROR;
1652
1653         while ((sdn = ldb_dn_get_parent(local_ctx, sdn))) {
1654                 ret = ldb_search(ldb, local_ctx, &res, sdn, LDB_SCOPE_BASE, attrs,
1655                                  "(|(objectClass=domain)(objectClass=builtinDomain))");
1656                 if (ret == LDB_SUCCESS) {
1657                         if (res->count == 1) {
1658                                 break;
1659                         }
1660                 } else {
1661                         break;
1662                 }
1663         }
1664
1665         if (ret != LDB_SUCCESS) {
1666                 *errstring = talloc_asprintf(mem_ctx, "Error searching for parent domain of %s, failed searching for %s: %s",
1667                                              ldb_dn_get_linearized(dn),
1668                                              ldb_dn_get_linearized(sdn),
1669                                              ldb_errstring(ldb));
1670                 talloc_free(local_ctx);
1671                 return ret;
1672         }
1673         if (res->count != 1) {
1674                 *errstring = talloc_asprintf(mem_ctx, "Invalid dn (%s), not child of a domain object",
1675                                              ldb_dn_get_linearized(dn));
1676                 DEBUG(0,(__location__ ": %s\n", *errstring));
1677                 talloc_free(local_ctx);
1678                 return LDB_ERR_CONSTRAINT_VIOLATION;
1679         }
1680
1681         *parent_dn = talloc_steal(mem_ctx, res->msgs[0]->dn);
1682         talloc_free(local_ctx);
1683         return ret;
1684 }
1685
1686
1687 /*
1688  * Performs checks on a user password (plaintext UNIX format - attribute
1689  * "password"). The remaining parameters have to be extracted from the domain
1690  * object in the AD.
1691  *
1692  * Result codes from "enum samr_ValidationStatus" (consider "samr.idl")
1693  */
1694 enum samr_ValidationStatus samdb_check_password(const DATA_BLOB *password,
1695                                                 const uint32_t pwdProperties,
1696                                                 const uint32_t minPwdLength)
1697 {
1698         /* checks if the "minPwdLength" property is satisfied */
1699         if (minPwdLength > password->length)
1700                 return SAMR_VALIDATION_STATUS_PWD_TOO_SHORT;
1701
1702         /* checks the password complexity */
1703         if (((pwdProperties & DOMAIN_PASSWORD_COMPLEX) != 0)
1704                         && (password->data != NULL)
1705                         && (!check_password_quality((const char *) password->data)))
1706                 return SAMR_VALIDATION_STATUS_NOT_COMPLEX_ENOUGH;
1707
1708         return SAMR_VALIDATION_STATUS_SUCCESS;
1709 }
1710
1711 /*
1712  * Sets the user password using plaintext UTF16 (attribute "new_password") or
1713  * LM (attribute "lmNewHash") or NT (attribute "ntNewHash") hash. Also pass
1714  * as parameter if it's a user change or not ("userChange"). The "rejectReason"
1715  * gives some more informations if the changed failed.
1716  *
1717  * The caller should have a LDB transaction wrapping this.
1718  *
1719  * Results: NT_STATUS_OK, NT_STATUS_INTERNAL_DB_CORRUPTION,
1720  *   NT_STATUS_INVALID_PARAMETER, NT_STATUS_UNSUCCESSFUL,
1721  *   NT_STATUS_PASSWORD_RESTRICTION
1722  */
1723 NTSTATUS samdb_set_password(struct ldb_context *ctx, TALLOC_CTX *mem_ctx,
1724                             struct ldb_dn *user_dn, struct ldb_dn *domain_dn,
1725                             struct ldb_message *mod,
1726                             const DATA_BLOB *new_password,
1727                             struct samr_Password *param_lmNewHash,
1728                             struct samr_Password *param_ntNewHash,
1729                             bool user_change,
1730                             enum samPwdChangeReason *reject_reason,
1731                             struct samr_DomInfo1 **_dominfo)
1732 {
1733         const char * const user_attrs[] = { "userAccountControl",
1734                                             "lmPwdHistory",
1735                                             "ntPwdHistory", 
1736                                             "dBCSPwd", "unicodePwd", 
1737                                             "objectSid", 
1738                                             "pwdLastSet", NULL };
1739         const char * const domain_attrs[] = { "minPwdLength", "pwdProperties",
1740                                               "pwdHistoryLength",
1741                                               "maxPwdAge", "minPwdAge", NULL };
1742         NTTIME pwdLastSet;
1743         uint32_t minPwdLength, pwdProperties, pwdHistoryLength;
1744         int64_t maxPwdAge, minPwdAge;
1745         uint32_t userAccountControl;
1746         struct samr_Password *sambaLMPwdHistory, *sambaNTPwdHistory,
1747                 *lmPwdHash, *ntPwdHash, *lmNewHash, *ntNewHash;
1748         struct samr_Password local_lmNewHash, local_ntNewHash;
1749         int sambaLMPwdHistory_len, sambaNTPwdHistory_len;
1750         struct dom_sid *domain_sid;
1751         struct ldb_message **res;
1752         bool restrictions;
1753         int count;
1754         time_t now = time(NULL);
1755         NTTIME now_nt;
1756         int i;
1757
1758         /* we need to know the time to compute password age */
1759         unix_to_nt_time(&now_nt, now);
1760
1761         /* pull all the user parameters */
1762         count = gendb_search_dn(ctx, mem_ctx, user_dn, &res, user_attrs);
1763         if (count != 1) {
1764                 return NT_STATUS_INTERNAL_DB_CORRUPTION;
1765         }
1766         userAccountControl = samdb_result_uint(res[0], "userAccountControl", 0);
1767         sambaLMPwdHistory_len = samdb_result_hashes(mem_ctx, res[0],
1768                                          "lmPwdHistory", &sambaLMPwdHistory);
1769         sambaNTPwdHistory_len = samdb_result_hashes(mem_ctx, res[0],
1770                                          "ntPwdHistory", &sambaNTPwdHistory);
1771         lmPwdHash = samdb_result_hash(mem_ctx, res[0], "dBCSPwd");
1772         ntPwdHash = samdb_result_hash(mem_ctx, res[0], "unicodePwd");
1773         pwdLastSet = samdb_result_uint64(res[0], "pwdLastSet", 0);
1774
1775         /* Copy parameters */
1776         lmNewHash = param_lmNewHash;
1777         ntNewHash = param_ntNewHash;
1778
1779         /* Only non-trust accounts have restrictions (possibly this
1780          * test is the wrong way around, but I like to be restrictive
1781          * if possible */
1782         restrictions = !(userAccountControl & (UF_INTERDOMAIN_TRUST_ACCOUNT
1783                                                |UF_WORKSTATION_TRUST_ACCOUNT
1784                                                |UF_SERVER_TRUST_ACCOUNT)); 
1785
1786         if (domain_dn != NULL) {
1787                 /* pull the domain parameters */
1788                 count = gendb_search_dn(ctx, mem_ctx, domain_dn, &res,
1789                                                                 domain_attrs);
1790                 if (count != 1) {
1791                         DEBUG(2, ("samdb_set_password: Domain DN %s is invalid, for user %s\n", 
1792                                   ldb_dn_get_linearized(domain_dn),
1793                                   ldb_dn_get_linearized(user_dn)));
1794                         return NT_STATUS_NO_SUCH_DOMAIN;
1795                 }
1796         } else {
1797                 /* work out the domain sid, and pull the domain from there */
1798                 domain_sid = samdb_result_sid_prefix(mem_ctx, res[0],
1799                                                                 "objectSid");
1800                 if (domain_sid == NULL) {
1801                         return NT_STATUS_INTERNAL_DB_CORRUPTION;
1802                 }
1803
1804                 count = gendb_search(ctx, mem_ctx, NULL, &res, domain_attrs, 
1805                                 "(objectSid=%s)",
1806                                 ldap_encode_ndr_dom_sid(mem_ctx, domain_sid));
1807                 if (count != 1) {
1808                         DEBUG(2, ("samdb_set_password: Could not find domain to match SID: %s, for user %s\n", 
1809                                   dom_sid_string(mem_ctx, domain_sid),
1810                                   ldb_dn_get_linearized(user_dn)));
1811                         return NT_STATUS_NO_SUCH_DOMAIN;
1812                 }
1813         }
1814
1815         minPwdLength = samdb_result_uint(res[0], "minPwdLength", 0);
1816         pwdProperties = samdb_result_uint(res[0], "pwdProperties", 0);
1817         pwdHistoryLength = samdb_result_uint(res[0], "pwdHistoryLength", 0);
1818         maxPwdAge = samdb_result_int64(res[0], "maxPwdAge", 0);
1819         minPwdAge = samdb_result_int64(res[0], "minPwdAge", 0);
1820
1821         if ((userAccountControl & UF_PASSWD_NOTREQD) != 0) {
1822                 /* see [MS-ADTS] 2.2.15 */
1823                 minPwdLength = 0;
1824         }
1825
1826         if (_dominfo != NULL) {
1827                 struct samr_DomInfo1 *dominfo;
1828                 /* on failure we need to fill in the reject reasons */
1829                 dominfo = talloc(mem_ctx, struct samr_DomInfo1);
1830                 if (dominfo == NULL) {
1831                         return NT_STATUS_NO_MEMORY;
1832                 }
1833                 dominfo->min_password_length     = minPwdLength;
1834                 dominfo->password_properties     = pwdProperties;
1835                 dominfo->password_history_length = pwdHistoryLength;
1836                 dominfo->max_password_age        = maxPwdAge;
1837                 dominfo->min_password_age        = minPwdAge;
1838                 *_dominfo = dominfo;
1839         }
1840
1841         if ((restrictions != 0) && (new_password != 0)) {
1842                 char *new_pass;
1843
1844                 /* checks if the "minPwdLength" property is satisfied */
1845                 if ((restrictions != 0)
1846                         && (minPwdLength > utf16_len_n(
1847                                 new_password->data, new_password->length)/2)) {
1848                         if (reject_reason) {
1849                                 *reject_reason = SAM_PWD_CHANGE_PASSWORD_TOO_SHORT;
1850                         }
1851                         return NT_STATUS_PASSWORD_RESTRICTION;
1852                 }
1853
1854                 /* Create the NT hash */
1855                 mdfour(local_ntNewHash.hash, new_password->data,
1856                                                         new_password->length);
1857
1858                 ntNewHash = &local_ntNewHash;
1859
1860                 /* Only check complexity if we can convert it at all.  Assuming unconvertable passwords are 'strong' */
1861                 if (convert_string_talloc_convenience(mem_ctx,
1862                           lp_iconv_convenience(ldb_get_opaque(ctx, "loadparm")),
1863                           CH_UTF16, CH_UNIX,
1864                           new_password->data, new_password->length,
1865                           (void **)&new_pass, NULL, false)) {
1866
1867                         /* checks the password complexity */
1868                         if ((restrictions != 0)
1869                                 && ((pwdProperties
1870                                         & DOMAIN_PASSWORD_COMPLEX) != 0)
1871                                 && (!check_password_quality(new_pass))) {
1872                                 if (reject_reason) {
1873                                         *reject_reason = SAM_PWD_CHANGE_NOT_COMPLEX;
1874                                 }
1875                                 return NT_STATUS_PASSWORD_RESTRICTION;
1876                         }
1877
1878                         /* compute the new lm hashes (for checking history - case insenitivly!) */
1879                         if (E_deshash(new_pass, local_lmNewHash.hash)) {
1880                                 lmNewHash = &local_lmNewHash;
1881                         }
1882                 }
1883         }
1884
1885         if ((restrictions != 0) && user_change) {
1886                 /* are all password changes disallowed? */
1887                 if ((pwdProperties & DOMAIN_REFUSE_PASSWORD_CHANGE) != 0) {
1888                         if (reject_reason) {
1889                                 *reject_reason = SAM_PWD_CHANGE_NO_ERROR;
1890                         }
1891                         return NT_STATUS_PASSWORD_RESTRICTION;
1892                 }
1893
1894                 /* can this user change the password? */
1895                 if ((userAccountControl & UF_PASSWD_CANT_CHANGE) != 0) {
1896                         if (reject_reason) {
1897                                 *reject_reason = SAM_PWD_CHANGE_NO_ERROR;
1898                         }
1899                         return NT_STATUS_PASSWORD_RESTRICTION;
1900                 }
1901
1902                 /* Password minimum age: yes, this is a minus. The ages are in negative 100nsec units! */
1903                 if (pwdLastSet - minPwdAge > now_nt) {
1904                         if (reject_reason) {
1905                                 *reject_reason = SAM_PWD_CHANGE_NO_ERROR;
1906                         }
1907                         return NT_STATUS_PASSWORD_RESTRICTION;
1908                 }
1909
1910                 /* check the immediately past password */
1911                 if (pwdHistoryLength > 0) {
1912                         if (lmNewHash && lmPwdHash && memcmp(lmNewHash->hash,
1913                                         lmPwdHash->hash, 16) == 0) {
1914                                 if (reject_reason) {
1915                                         *reject_reason = SAM_PWD_CHANGE_PWD_IN_HISTORY;
1916                                 }
1917                                 return NT_STATUS_PASSWORD_RESTRICTION;
1918                         }
1919                         if (ntNewHash && ntPwdHash && memcmp(ntNewHash->hash,
1920                                         ntPwdHash->hash, 16) == 0) {
1921                                 if (reject_reason) {
1922                                         *reject_reason = SAM_PWD_CHANGE_PWD_IN_HISTORY;
1923                                 }
1924                                 return NT_STATUS_PASSWORD_RESTRICTION;
1925                         }
1926                 }
1927
1928                 /* check the password history */
1929                 sambaLMPwdHistory_len = MIN(sambaLMPwdHistory_len,
1930                                                         pwdHistoryLength);
1931                 sambaNTPwdHistory_len = MIN(sambaNTPwdHistory_len,
1932                                                         pwdHistoryLength);
1933
1934                 for (i=0; lmNewHash && i<sambaLMPwdHistory_len;i++) {
1935                         if (memcmp(lmNewHash->hash, sambaLMPwdHistory[i].hash,
1936                                         16) == 0) {
1937                                 if (reject_reason) {
1938                                         *reject_reason = SAM_PWD_CHANGE_PWD_IN_HISTORY;
1939                                 }
1940                                 return NT_STATUS_PASSWORD_RESTRICTION;
1941                         }
1942                 }
1943                 for (i=0; ntNewHash && i<sambaNTPwdHistory_len;i++) {
1944                         if (memcmp(ntNewHash->hash, sambaNTPwdHistory[i].hash,
1945                                          16) == 0) {
1946                                 if (reject_reason) {
1947                                         *reject_reason = SAM_PWD_CHANGE_PWD_IN_HISTORY;
1948                                 }
1949                                 return NT_STATUS_PASSWORD_RESTRICTION;
1950                         }
1951                 }
1952         }
1953
1954 #define CHECK_RET(x) do { if (x != 0) return NT_STATUS_NO_MEMORY; } while(0)
1955
1956         /* the password is acceptable. Start forming the new fields */
1957         if (new_password != NULL) {
1958                 /* if we know the cleartext UTF16 password, then set it.
1959                  * Modules in ldb will set all the appropriate
1960                  * hashes */
1961                 CHECK_RET(ldb_msg_add_value(mod, "clearTextPassword", new_password, NULL));
1962         } else {
1963                 /* we don't have the cleartext, so set what we have of the
1964                  * hashes */
1965
1966                 if (lmNewHash) {
1967                         CHECK_RET(samdb_msg_add_hash(ctx, mem_ctx, mod, "dBCSPwd", lmNewHash));
1968                 }
1969
1970                 if (ntNewHash) {
1971                         CHECK_RET(samdb_msg_add_hash(ctx, mem_ctx, mod, "unicodePwd", ntNewHash));
1972                 }
1973         }
1974
1975         if (reject_reason) {
1976                 *reject_reason = SAM_PWD_CHANGE_NO_ERROR;
1977         }
1978         return NT_STATUS_OK;
1979 }
1980
1981
1982 /*
1983  * Sets the user password using plaintext UTF16 (attribute "new_password") or
1984  * LM (attribute "lmNewHash") or NT (attribute "ntNewHash") hash. Also pass
1985  * as parameter if it's a user change or not ("userChange"). The "rejectReason"
1986  * gives some more informations if the changed failed.
1987  *
1988  * This wrapper function for "samdb_set_password" takes a SID as input rather
1989  * than a user DN.
1990  *
1991  * This call encapsulates a new LDB transaction for changing the password;
1992  * therefore the user hasn't to start a new one.
1993  *
1994  * Results: NT_STATUS_OK, NT_STATUS_INTERNAL_DB_CORRUPTION,
1995  *   NT_STATUS_INVALID_PARAMETER, NT_STATUS_UNSUCCESSFUL,
1996  *   NT_STATUS_PASSWORD_RESTRICTION
1997  */
1998 NTSTATUS samdb_set_password_sid(struct ldb_context *ldb, TALLOC_CTX *mem_ctx,
1999                                 const struct dom_sid *user_sid,
2000                                 const DATA_BLOB *new_password,
2001                                 struct samr_Password *lmNewHash, 
2002                                 struct samr_Password *ntNewHash,
2003                                 bool user_change,
2004                                 enum samPwdChangeReason *reject_reason,
2005                                 struct samr_DomInfo1 **_dominfo) 
2006 {
2007         NTSTATUS nt_status;
2008         struct ldb_dn *user_dn;
2009         struct ldb_message *msg;
2010         int ret;
2011
2012         ret = ldb_transaction_start(ldb);
2013         if (ret != LDB_SUCCESS) {
2014                 DEBUG(1, ("Failed to start transaction: %s\n", ldb_errstring(ldb)));
2015                 return NT_STATUS_TRANSACTION_ABORTED;
2016         }
2017
2018         user_dn = samdb_search_dn(ldb, mem_ctx, NULL,
2019                                   "(&(objectSid=%s)(objectClass=user))", 
2020                                   ldap_encode_ndr_dom_sid(mem_ctx, user_sid));
2021         if (!user_dn) {
2022                 ldb_transaction_cancel(ldb);
2023                 DEBUG(3, ("samdb_set_password_sid: SID %s not found in samdb, returning NO_SUCH_USER\n",
2024                           dom_sid_string(mem_ctx, user_sid)));
2025                 return NT_STATUS_NO_SUCH_USER;
2026         }
2027
2028         msg = ldb_msg_new(mem_ctx);
2029         if (msg == NULL) {
2030                 ldb_transaction_cancel(ldb);
2031                 talloc_free(user_dn);
2032                 return NT_STATUS_NO_MEMORY;
2033         }
2034
2035         msg->dn = ldb_dn_copy(msg, user_dn);
2036         if (!msg->dn) {
2037                 ldb_transaction_cancel(ldb);
2038                 talloc_free(user_dn);
2039                 talloc_free(msg);
2040                 return NT_STATUS_NO_MEMORY;
2041         }
2042
2043         nt_status = samdb_set_password(ldb, mem_ctx,
2044                                        user_dn, NULL,
2045                                        msg, new_password,
2046                                        lmNewHash, ntNewHash,
2047                                        user_change,
2048                                        reject_reason, _dominfo);
2049         if (!NT_STATUS_IS_OK(nt_status)) {
2050                 ldb_transaction_cancel(ldb);
2051                 talloc_free(user_dn);
2052                 talloc_free(msg);
2053                 return nt_status;
2054         }
2055
2056         /* modify the samdb record */
2057         ret = dsdb_replace(ldb, msg, 0);
2058         if (ret != LDB_SUCCESS) {
2059                 ldb_transaction_cancel(ldb);
2060                 talloc_free(user_dn);
2061                 talloc_free(msg);
2062                 return NT_STATUS_ACCESS_DENIED;
2063         }
2064
2065         talloc_free(msg);
2066
2067         ret = ldb_transaction_commit(ldb);
2068         if (ret != LDB_SUCCESS) {
2069                 DEBUG(0,("Failed to commit transaction to change password on %s: %s\n",
2070                          ldb_dn_get_linearized(user_dn),
2071                          ldb_errstring(ldb)));
2072                 talloc_free(user_dn);
2073                 return NT_STATUS_TRANSACTION_ABORTED;
2074         }
2075
2076         talloc_free(user_dn);
2077         return NT_STATUS_OK;
2078 }
2079
2080
2081 NTSTATUS samdb_create_foreign_security_principal(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx, 
2082                                                  struct dom_sid *sid, struct ldb_dn **ret_dn) 
2083 {
2084         struct ldb_message *msg;
2085         struct ldb_dn *basedn;
2086         char *sidstr;
2087         int ret;
2088
2089         sidstr = dom_sid_string(mem_ctx, sid);
2090         NT_STATUS_HAVE_NO_MEMORY(sidstr);
2091
2092         /* We might have to create a ForeignSecurityPrincipal, even if this user
2093          * is in our own domain */
2094
2095         msg = ldb_msg_new(sidstr);
2096         if (msg == NULL) {
2097                 talloc_free(sidstr);
2098                 return NT_STATUS_NO_MEMORY;
2099         }
2100
2101         ret = dsdb_wellknown_dn(sam_ctx, sidstr, samdb_base_dn(sam_ctx),
2102                                 DS_GUID_FOREIGNSECURITYPRINCIPALS_CONTAINER,
2103                                 &basedn);
2104         if (ret != LDB_SUCCESS) {
2105                 DEBUG(0, ("Failed to find DN for "
2106                           "ForeignSecurityPrincipal container - %s\n", ldb_errstring(sam_ctx)));
2107                 talloc_free(sidstr);
2108                 return NT_STATUS_INTERNAL_DB_CORRUPTION;
2109         }
2110
2111         /* add core elements to the ldb_message for the alias */
2112         msg->dn = basedn;
2113         if ( ! ldb_dn_add_child_fmt(msg->dn, "CN=%s", sidstr)) {
2114                 talloc_free(sidstr);
2115                 return NT_STATUS_NO_MEMORY;
2116         }
2117
2118         samdb_msg_add_string(sam_ctx, msg, msg,
2119                              "objectClass",
2120                              "foreignSecurityPrincipal");
2121
2122         /* create the alias */
2123         ret = ldb_add(sam_ctx, msg);
2124         if (ret != LDB_SUCCESS) {
2125                 DEBUG(0,("Failed to create foreignSecurityPrincipal "
2126                          "record %s: %s\n", 
2127                          ldb_dn_get_linearized(msg->dn),
2128                          ldb_errstring(sam_ctx)));
2129                 talloc_free(sidstr);
2130                 return NT_STATUS_INTERNAL_DB_CORRUPTION;
2131         }
2132
2133         *ret_dn = talloc_steal(mem_ctx, msg->dn);
2134         talloc_free(sidstr);
2135
2136         return NT_STATUS_OK;
2137 }
2138
2139
2140 /*
2141   Find the DN of a domain, assuming it to be a dotted.dns name
2142 */
2143
2144 struct ldb_dn *samdb_dns_domain_to_dn(struct ldb_context *ldb, TALLOC_CTX *mem_ctx, const char *dns_domain) 
2145 {
2146         int i;
2147         TALLOC_CTX *tmp_ctx = talloc_new(mem_ctx);
2148         const char *binary_encoded;
2149         const char **split_realm;
2150         struct ldb_dn *dn;
2151
2152         if (!tmp_ctx) {
2153                 return NULL;
2154         }
2155
2156         split_realm = (const char **)str_list_make(tmp_ctx, dns_domain, ".");
2157         if (!split_realm) {
2158                 talloc_free(tmp_ctx);
2159                 return NULL;
2160         }
2161         dn = ldb_dn_new(mem_ctx, ldb, NULL);
2162         for (i=0; split_realm[i]; i++) {
2163                 binary_encoded = ldb_binary_encode_string(tmp_ctx, split_realm[i]);
2164                 if (!ldb_dn_add_base_fmt(dn, "dc=%s", binary_encoded)) {
2165                         DEBUG(2, ("Failed to add dc=%s element to DN %s\n",
2166                                   binary_encoded, ldb_dn_get_linearized(dn)));
2167                         talloc_free(tmp_ctx);
2168                         return NULL;
2169                 }
2170         }
2171         if (!ldb_dn_validate(dn)) {
2172                 DEBUG(2, ("Failed to validated DN %s\n",
2173                           ldb_dn_get_linearized(dn)));
2174                 talloc_free(tmp_ctx);
2175                 return NULL;
2176         }
2177         talloc_free(tmp_ctx);
2178         return dn;
2179 }
2180
2181 /*
2182   Find the DN of a domain, be it the netbios or DNS name 
2183 */
2184 struct ldb_dn *samdb_domain_to_dn(struct ldb_context *ldb, TALLOC_CTX *mem_ctx, 
2185                                   const char *domain_name) 
2186 {
2187         const char * const domain_ref_attrs[] = {
2188                 "ncName", NULL
2189         };
2190         const char * const domain_ref2_attrs[] = {
2191                 NULL
2192         };
2193         struct ldb_result *res_domain_ref;
2194         char *escaped_domain = ldb_binary_encode_string(mem_ctx, domain_name);
2195         /* find the domain's DN */
2196         int ret_domain = ldb_search(ldb, mem_ctx,
2197                                             &res_domain_ref, 
2198                                             samdb_partitions_dn(ldb, mem_ctx), 
2199                                             LDB_SCOPE_ONELEVEL, 
2200                                             domain_ref_attrs,
2201                                             "(&(nETBIOSName=%s)(objectclass=crossRef))", 
2202                                             escaped_domain);
2203         if (ret_domain != 0) {
2204                 return NULL;
2205         }
2206
2207         if (res_domain_ref->count == 0) {
2208                 ret_domain = ldb_search(ldb, mem_ctx,
2209                                                 &res_domain_ref, 
2210                                                 samdb_dns_domain_to_dn(ldb, mem_ctx, domain_name),
2211                                                 LDB_SCOPE_BASE,
2212                                                 domain_ref2_attrs,
2213                                                 "(objectclass=domain)");
2214                 if (ret_domain != 0) {
2215                         return NULL;
2216                 }
2217
2218                 if (res_domain_ref->count == 1) {
2219                         return res_domain_ref->msgs[0]->dn;
2220                 }
2221                 return NULL;
2222         }
2223
2224         if (res_domain_ref->count > 1) {
2225                 DEBUG(0,("Found %d records matching domain [%s]\n", 
2226                          ret_domain, domain_name));
2227                 return NULL;
2228         }
2229
2230         return samdb_result_dn(ldb, mem_ctx, res_domain_ref->msgs[0], "nCName", NULL);
2231
2232 }
2233
2234
2235 /*
2236   use a GUID to find a DN
2237  */
2238 int dsdb_find_dn_by_guid(struct ldb_context *ldb, 
2239                          TALLOC_CTX *mem_ctx,
2240                          const char *guid_str, struct ldb_dn **dn)
2241 {
2242         int ret;
2243         struct ldb_result *res;
2244         const char *attrs[] = { NULL };
2245         struct ldb_request *search_req;
2246         char *expression;
2247         struct ldb_search_options_control *options;
2248
2249         expression = talloc_asprintf(mem_ctx, "objectGUID=%s", guid_str);
2250         if (!expression) {
2251                 DEBUG(0, (__location__ ": out of memory\n"));
2252                 return LDB_ERR_OPERATIONS_ERROR;
2253         }
2254
2255         res = talloc_zero(expression, struct ldb_result);
2256         if (!res) {
2257                 DEBUG(0, (__location__ ": out of memory\n"));
2258                 talloc_free(expression);
2259                 return LDB_ERR_OPERATIONS_ERROR;
2260         }
2261
2262         ret = ldb_build_search_req(&search_req, ldb, expression,
2263                                    ldb_get_default_basedn(ldb),
2264                                    LDB_SCOPE_SUBTREE,
2265                                    expression, attrs,
2266                                    NULL,
2267                                    res, ldb_search_default_callback,
2268                                    NULL);
2269         if (ret != LDB_SUCCESS) {
2270                 talloc_free(expression);
2271                 return ret;
2272         }
2273
2274         /* we need to cope with cross-partition links, so search for
2275            the GUID over all partitions */
2276         options = talloc(search_req, struct ldb_search_options_control);
2277         if (options == NULL) {
2278                 DEBUG(0, (__location__ ": out of memory\n"));
2279                 talloc_free(expression);
2280                 return LDB_ERR_OPERATIONS_ERROR;
2281         }
2282         options->search_options = LDB_SEARCH_OPTION_PHANTOM_ROOT;
2283
2284         ret = ldb_request_add_control(search_req, LDB_CONTROL_EXTENDED_DN_OID, true, NULL);
2285         if (ret != LDB_SUCCESS) {
2286                 talloc_free(expression);
2287                 return ret;
2288         }
2289
2290         ret = ldb_request_add_control(search_req,
2291                                       LDB_CONTROL_SEARCH_OPTIONS_OID,
2292                                       true, options);
2293         if (ret != LDB_SUCCESS) {
2294                 talloc_free(expression);
2295                 return ret;
2296         }
2297
2298         ret = ldb_request(ldb, search_req);
2299         if (ret != LDB_SUCCESS) {
2300                 talloc_free(expression);
2301                 return ret;
2302         }
2303
2304         ret = ldb_wait(search_req->handle, LDB_WAIT_ALL);
2305         if (ret != LDB_SUCCESS) {
2306                 talloc_free(expression);
2307                 return ret;
2308         }
2309
2310         /* this really should be exactly 1, but there is a bug in the
2311            partitions module that can return two here with the
2312            search_options control set */
2313         if (res->count < 1) {
2314                 talloc_free(expression);
2315                 return LDB_ERR_NO_SUCH_OBJECT;
2316         }
2317
2318         *dn = talloc_steal(mem_ctx, res->msgs[0]->dn);
2319         talloc_free(expression);
2320
2321         return LDB_SUCCESS;
2322 }
2323
2324 /*
2325   search for attrs on one DN, allowing for deleted objects
2326  */
2327 int dsdb_search_dn_with_deleted(struct ldb_context *ldb,
2328                                 TALLOC_CTX *mem_ctx,
2329                                 struct ldb_result **_res,
2330                                 struct ldb_dn *basedn,
2331                                 const char * const *attrs)
2332 {
2333         int ret;
2334         struct ldb_request *req;
2335         TALLOC_CTX *tmp_ctx;
2336         struct ldb_result *res;
2337
2338         tmp_ctx = talloc_new(mem_ctx);
2339
2340         res = talloc_zero(tmp_ctx, struct ldb_result);
2341         if (!res) {
2342                 talloc_free(tmp_ctx);
2343                 return LDB_ERR_OPERATIONS_ERROR;
2344         }
2345
2346         ret = ldb_build_search_req(&req, ldb, tmp_ctx,
2347                                    basedn,
2348                                    LDB_SCOPE_BASE,
2349                                    NULL,
2350                                    attrs,
2351                                    NULL,
2352                                    res,
2353                                    ldb_search_default_callback,
2354                                    NULL);
2355         if (ret != LDB_SUCCESS) {
2356                 talloc_free(tmp_ctx);
2357                 return ret;
2358         }
2359
2360         ret = ldb_request_add_control(req, LDB_CONTROL_SHOW_DELETED_OID, true, NULL);
2361         if (ret != LDB_SUCCESS) {
2362                 talloc_free(tmp_ctx);
2363                 return ret;
2364         }
2365
2366         ret = ldb_request(ldb, req);
2367         if (ret == LDB_SUCCESS) {
2368                 ret = ldb_wait(req->handle, LDB_WAIT_ALL);
2369         }
2370
2371         *_res = talloc_steal(mem_ctx, res);
2372         talloc_free(tmp_ctx);
2373         return ret;
2374 }
2375
2376
2377 /*
2378   use a DN to find a GUID with a given attribute name
2379  */
2380 int dsdb_find_guid_attr_by_dn(struct ldb_context *ldb,
2381                               struct ldb_dn *dn, const char *attribute,
2382                               struct GUID *guid)
2383 {
2384         int ret;
2385         struct ldb_result *res;
2386         const char *attrs[2];
2387         TALLOC_CTX *tmp_ctx = talloc_new(ldb);
2388
2389         attrs[0] = attribute;
2390         attrs[1] = NULL;
2391
2392         ret = dsdb_search_dn_with_deleted(ldb, tmp_ctx, &res, dn, attrs);
2393         if (ret != LDB_SUCCESS) {
2394                 talloc_free(tmp_ctx);
2395                 return ret;
2396         }
2397         if (res->count < 1) {
2398                 talloc_free(tmp_ctx);
2399                 return LDB_ERR_NO_SUCH_OBJECT;
2400         }
2401         *guid = samdb_result_guid(res->msgs[0], attribute);
2402         talloc_free(tmp_ctx);
2403         return LDB_SUCCESS;
2404 }
2405
2406 /*
2407   use a DN to find a GUID
2408  */
2409 int dsdb_find_guid_by_dn(struct ldb_context *ldb,
2410                          struct ldb_dn *dn, struct GUID *guid)
2411 {
2412         return dsdb_find_guid_attr_by_dn(ldb, dn, "objectGUID", guid);
2413 }
2414
2415
2416
2417 /*
2418  adds the given GUID to the given ldb_message. This value is added
2419  for the given attr_name (may be either "objectGUID" or "parentGUID").
2420  */
2421 int dsdb_msg_add_guid(struct ldb_message *msg,
2422                 struct GUID *guid,
2423                 const char *attr_name)
2424 {
2425         int ret;
2426         struct ldb_val v;
2427         NTSTATUS status;
2428         TALLOC_CTX *tmp_ctx =  talloc_init("dsdb_msg_add_guid");
2429
2430         status = GUID_to_ndr_blob(guid, tmp_ctx, &v);
2431         if (!NT_STATUS_IS_OK(status)) {
2432                 ret = LDB_ERR_OPERATIONS_ERROR;
2433                 goto done;
2434         }
2435
2436         ret = ldb_msg_add_steal_value(msg, attr_name, &v);
2437         if (ret != LDB_SUCCESS) {
2438                 DEBUG(4,(__location__ ": Failed to add %s to the message\n",
2439                                          attr_name));
2440                 goto done;
2441         }
2442
2443         ret = LDB_SUCCESS;
2444
2445 done:
2446         talloc_free(tmp_ctx);
2447         return ret;
2448
2449 }
2450
2451
2452 /*
2453   use a DN to find a SID
2454  */
2455 int dsdb_find_sid_by_dn(struct ldb_context *ldb, 
2456                         struct ldb_dn *dn, struct dom_sid *sid)
2457 {
2458         int ret;
2459         struct ldb_result *res;
2460         const char *attrs[] = { "objectSID", NULL };
2461         TALLOC_CTX *tmp_ctx = talloc_new(ldb);
2462         struct dom_sid *s;
2463
2464         ZERO_STRUCTP(sid);
2465
2466         ret = dsdb_search_dn_with_deleted(ldb, tmp_ctx, &res, dn, attrs);
2467         if (ret != LDB_SUCCESS) {
2468                 talloc_free(tmp_ctx);
2469                 return ret;
2470         }
2471         if (res->count < 1) {
2472                 talloc_free(tmp_ctx);
2473                 return LDB_ERR_NO_SUCH_OBJECT;
2474         }
2475         s = samdb_result_dom_sid(tmp_ctx, res->msgs[0], "objectSID");
2476         if (s == NULL) {
2477                 talloc_free(tmp_ctx);
2478                 return LDB_ERR_NO_SUCH_OBJECT;
2479         }
2480         *sid = *s;
2481         talloc_free(tmp_ctx);
2482         return LDB_SUCCESS;
2483 }
2484
2485
2486
2487 /*
2488   load a repsFromTo blob list for a given partition GUID
2489   attr must be "repsFrom" or "repsTo"
2490  */
2491 WERROR dsdb_loadreps(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx, struct ldb_dn *dn,
2492                      const char *attr, struct repsFromToBlob **r, uint32_t *count)
2493 {
2494         const char *attrs[] = { attr, NULL };
2495         struct ldb_result *res = NULL;
2496         TALLOC_CTX *tmp_ctx = talloc_new(mem_ctx);
2497         int i;
2498         struct ldb_message_element *el;
2499
2500         *r = NULL;
2501         *count = 0;
2502
2503         if (ldb_search(sam_ctx, tmp_ctx, &res, dn, LDB_SCOPE_BASE, attrs, NULL) != LDB_SUCCESS ||
2504             res->count < 1) {
2505                 DEBUG(0,("dsdb_loadreps: failed to read partition object\n"));
2506                 talloc_free(tmp_ctx);
2507                 return WERR_DS_DRA_INTERNAL_ERROR;
2508         }
2509
2510         el = ldb_msg_find_element(res->msgs[0], attr);
2511         if (el == NULL) {
2512                 /* it's OK to be empty */
2513                 talloc_free(tmp_ctx);
2514                 return WERR_OK;
2515         }
2516
2517         *count = el->num_values;
2518         *r = talloc_array(mem_ctx, struct repsFromToBlob, *count);
2519         if (*r == NULL) {
2520                 talloc_free(tmp_ctx);
2521                 return WERR_DS_DRA_INTERNAL_ERROR;
2522         }
2523
2524         for (i=0; i<(*count); i++) {
2525                 enum ndr_err_code ndr_err;
2526                 ndr_err = ndr_pull_struct_blob(&el->values[i], 
2527                                                mem_ctx, lp_iconv_convenience(ldb_get_opaque(sam_ctx, "loadparm")),
2528                                                &(*r)[i], 
2529                                                (ndr_pull_flags_fn_t)ndr_pull_repsFromToBlob);
2530                 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
2531                         talloc_free(tmp_ctx);
2532                         return WERR_DS_DRA_INTERNAL_ERROR;
2533                 }
2534         }
2535
2536         talloc_free(tmp_ctx);
2537         
2538         return WERR_OK;
2539 }
2540
2541 /*
2542   save the repsFromTo blob list for a given partition GUID
2543   attr must be "repsFrom" or "repsTo"
2544  */
2545 WERROR dsdb_savereps(struct ldb_context *sam_ctx, TALLOC_CTX *mem_ctx, struct ldb_dn *dn,
2546                      const char *attr, struct repsFromToBlob *r, uint32_t count)
2547 {
2548         TALLOC_CTX *tmp_ctx = talloc_new(mem_ctx);
2549         struct ldb_message *msg;
2550         struct ldb_message_element *el;
2551         int i;
2552
2553         msg = ldb_msg_new(tmp_ctx);
2554         msg->dn = dn;
2555         if (ldb_msg_add_empty(msg, attr, LDB_FLAG_MOD_REPLACE, &el) != LDB_SUCCESS) {
2556                 goto failed;
2557         }
2558
2559         el->values = talloc_array(msg, struct ldb_val, count);
2560         if (!el->values) {
2561                 goto failed;
2562         }
2563
2564         for (i=0; i<count; i++) {
2565                 struct ldb_val v;
2566                 enum ndr_err_code ndr_err;
2567
2568                 ndr_err = ndr_push_struct_blob(&v, tmp_ctx, lp_iconv_convenience(ldb_get_opaque(sam_ctx, "loadparm")),
2569                                                &r[i], 
2570                                                (ndr_push_flags_fn_t)ndr_push_repsFromToBlob);
2571                 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
2572                         goto failed;
2573                 }
2574
2575                 el->num_values++;
2576                 el->values[i] = v;
2577         }
2578
2579         if (ldb_modify(sam_ctx, msg) != LDB_SUCCESS) {
2580                 DEBUG(0,("Failed to store %s - %s\n", attr, ldb_errstring(sam_ctx)));
2581                 goto failed;
2582         }
2583
2584         talloc_free(tmp_ctx);
2585         
2586         return WERR_OK;
2587
2588 failed:
2589         talloc_free(tmp_ctx);
2590         return WERR_DS_DRA_INTERNAL_ERROR;
2591 }
2592
2593
2594 /*
2595   load the uSNHighest and the uSNUrgent attributes from the @REPLCHANGED
2596   object for a partition
2597  */
2598 int dsdb_load_partition_usn(struct ldb_context *ldb, struct ldb_dn *dn,
2599                                 uint64_t *uSN, uint64_t *urgent_uSN)
2600 {
2601         struct ldb_request *req;
2602         int ret;
2603         TALLOC_CTX *tmp_ctx = talloc_new(ldb);
2604         struct dsdb_control_current_partition *p_ctrl;
2605         struct ldb_result *res;
2606
2607         res = talloc_zero(tmp_ctx, struct ldb_result);
2608         if (!res) {
2609                 talloc_free(tmp_ctx);
2610                 return LDB_ERR_OPERATIONS_ERROR;
2611         }
2612
2613         ret = ldb_build_search_req(&req, ldb, tmp_ctx,
2614                                    ldb_dn_new(tmp_ctx, ldb, "@REPLCHANGED"),
2615                                    LDB_SCOPE_BASE,
2616                                    NULL, NULL,
2617                                    NULL,
2618                                    res, ldb_search_default_callback,
2619                                    NULL);
2620         if (ret != LDB_SUCCESS) {
2621                 talloc_free(tmp_ctx);
2622                 return ret;
2623         }
2624
2625         p_ctrl = talloc(req, struct dsdb_control_current_partition);
2626         if (p_ctrl == NULL) {
2627                 talloc_free(res);
2628                 return LDB_ERR_OPERATIONS_ERROR;
2629         }
2630         p_ctrl->version = DSDB_CONTROL_CURRENT_PARTITION_VERSION;
2631         p_ctrl->dn = dn;
2632         
2633
2634         ret = ldb_request_add_control(req,
2635                                       DSDB_CONTROL_CURRENT_PARTITION_OID,
2636                                       false, p_ctrl);
2637         if (ret != LDB_SUCCESS) {
2638                 talloc_free(tmp_ctx);
2639                 return ret;
2640         }
2641         
2642         /* Run the new request */
2643         ret = ldb_request(ldb, req);
2644         
2645         if (ret == LDB_SUCCESS) {
2646                 ret = ldb_wait(req->handle, LDB_WAIT_ALL);
2647         }
2648
2649         if (ret == LDB_ERR_NO_SUCH_OBJECT) {
2650                 /* it hasn't been created yet, which means
2651                    an implicit value of zero */
2652                 *uSN = 0;
2653                 talloc_free(tmp_ctx);
2654                 return LDB_SUCCESS;
2655         }
2656
2657         if (ret != LDB_SUCCESS) {
2658                 talloc_free(tmp_ctx);
2659                 return ret;
2660         }
2661
2662         if (res->count < 1) {
2663                 *uSN = 0;
2664                 if (urgent_uSN) {
2665                         *urgent_uSN = 0;
2666                 }
2667         } else {
2668                 *uSN = ldb_msg_find_attr_as_uint64(res->msgs[0], "uSNHighest", 0);
2669                 if (urgent_uSN) {
2670                         *urgent_uSN = ldb_msg_find_attr_as_uint64(res->msgs[0], "uSNUrgent", 0);
2671                 }
2672         }
2673
2674         talloc_free(tmp_ctx);
2675
2676         return LDB_SUCCESS;     
2677 }
2678
2679 /*
2680   save uSNHighest and uSNUrgent attributes in the @REPLCHANGED object for a
2681   partition
2682  */
2683 int dsdb_save_partition_usn(struct ldb_context *ldb, struct ldb_dn *dn,
2684                                 uint64_t uSN, uint64_t urgent_uSN)
2685 {
2686         struct ldb_request *req;
2687         struct ldb_message *msg;
2688         struct dsdb_control_current_partition *p_ctrl;
2689         int ret;
2690
2691         msg = ldb_msg_new(ldb);
2692         if (msg == NULL) {
2693                 return LDB_ERR_OPERATIONS_ERROR;
2694         }
2695
2696         msg->dn = ldb_dn_new(msg, ldb, "@REPLCHANGED");
2697         if (msg->dn == NULL) {
2698                 talloc_free(msg);
2699                 return LDB_ERR_OPERATIONS_ERROR;
2700         }
2701         
2702         ret = ldb_msg_add_fmt(msg, "uSNHighest", "%llu", (unsigned long long)uSN);
2703         if (ret != LDB_SUCCESS) {
2704                 talloc_free(msg);
2705                 return ret;
2706         }
2707         msg->elements[0].flags = LDB_FLAG_MOD_REPLACE;
2708         
2709         /* urgent_uSN is optional so may not be stored */
2710         if (urgent_uSN) {
2711                 ret = ldb_msg_add_fmt(msg, "uSNUrgent", "%llu", (unsigned long long)urgent_uSN);
2712                 if (ret != LDB_SUCCESS) {
2713                         talloc_free(msg);
2714                         return ret;
2715                 }
2716                 msg->elements[1].flags = LDB_FLAG_MOD_REPLACE;
2717         }
2718
2719
2720         p_ctrl = talloc(msg, struct dsdb_control_current_partition);
2721         if (p_ctrl == NULL) {
2722                 talloc_free(msg);
2723                 return LDB_ERR_OPERATIONS_ERROR;
2724         }
2725         p_ctrl->version = DSDB_CONTROL_CURRENT_PARTITION_VERSION;
2726         p_ctrl->dn = dn;
2727
2728         ret = ldb_build_mod_req(&req, ldb, msg,
2729                                 msg,
2730                                 NULL,
2731                                 NULL, ldb_op_default_callback,
2732                                 NULL);
2733 again:
2734         if (ret != LDB_SUCCESS) {
2735                 talloc_free(msg);
2736                 return ret;
2737         }
2738         
2739         ret = ldb_request_add_control(req,
2740                                       DSDB_CONTROL_CURRENT_PARTITION_OID,
2741                                       false, p_ctrl);
2742         if (ret != LDB_SUCCESS) {
2743                 talloc_free(msg);
2744                 return ret;
2745         }
2746         
2747         /* Run the new request */
2748         ret = ldb_request(ldb, req);
2749         
2750         if (ret == LDB_SUCCESS) {
2751                 ret = ldb_wait(req->handle, LDB_WAIT_ALL);
2752         }
2753         if (ret == LDB_ERR_NO_SUCH_OBJECT) {
2754                 ret = ldb_build_add_req(&req, ldb, msg,
2755                                         msg,
2756                                         NULL,
2757                                         NULL, ldb_op_default_callback,
2758                                         NULL);
2759                 goto again;
2760         }
2761         
2762         talloc_free(msg);
2763         
2764         return ret;
2765 }
2766
2767 int drsuapi_DsReplicaCursor2_compare(const struct drsuapi_DsReplicaCursor2 *c1,
2768                                                    const struct drsuapi_DsReplicaCursor2 *c2)
2769 {
2770         return GUID_compare(&c1->source_dsa_invocation_id, &c2->source_dsa_invocation_id);
2771 }
2772
2773 int drsuapi_DsReplicaCursor_compare(const struct drsuapi_DsReplicaCursor *c1,
2774                                     const struct drsuapi_DsReplicaCursor *c2)
2775 {
2776         return GUID_compare(&c1->source_dsa_invocation_id, &c2->source_dsa_invocation_id);
2777 }
2778
2779 /*
2780   see if we are a RODC
2781
2782   TODO: This should take a sam_ctx, and lookup the right object (with
2783   a cache)
2784 */
2785 bool samdb_rodc(struct loadparm_context *lp_ctx)
2786 {
2787         return lp_parm_bool(lp_ctx, NULL, "repl", "RODC", false);
2788 }
2789
2790
2791 /*
2792   return NTDS options flags. See MS-ADTS 7.1.1.2.2.1.2.1.1 
2793
2794   flags are DS_NTDS_OPTION_*
2795 */
2796 int samdb_ntds_options(struct ldb_context *ldb, uint32_t *options)
2797 {
2798         TALLOC_CTX *tmp_ctx;
2799         const char *attrs[] = { "options", NULL };
2800         int ret;
2801         struct ldb_result *res;
2802
2803         tmp_ctx = talloc_new(ldb);
2804         if (tmp_ctx == NULL) {
2805                 goto failed;
2806         }
2807
2808         ret = ldb_search(ldb, tmp_ctx, &res, samdb_ntds_settings_dn(ldb), LDB_SCOPE_BASE, attrs, NULL);
2809         if (ret) {
2810                 goto failed;
2811         }
2812
2813         if (res->count != 1) {
2814                 goto failed;
2815         }
2816
2817         *options = samdb_result_uint(res->msgs[0], "options", 0);
2818
2819         talloc_free(tmp_ctx);
2820
2821         return LDB_SUCCESS;
2822
2823 failed:
2824         DEBUG(1,("Failed to find our own NTDS Settings objectGUID in the ldb!\n"));
2825         talloc_free(tmp_ctx);
2826         return LDB_ERR_NO_SUCH_OBJECT;
2827 }
2828
2829 /*
2830  * Function which generates a "lDAPDisplayName" attribute from a "CN" one.
2831  * Algorithm implemented according to MS-ADTS 3.1.1.2.3.4
2832  */
2833 const char *samdb_cn_to_lDAPDisplayName(TALLOC_CTX *mem_ctx, const char *cn)
2834 {
2835         char **tokens, *ret;
2836         size_t i;
2837
2838         tokens = str_list_make(mem_ctx, cn, " -_");
2839         if (tokens == NULL)
2840                 return NULL;
2841
2842         /* "tolower()" and "toupper()" should also work properly on 0x00 */
2843         tokens[0][0] = tolower(tokens[0][0]);
2844         for (i = 1; i < str_list_length((const char **)tokens); i++)
2845                 tokens[i][0] = toupper(tokens[i][0]);
2846
2847         ret = talloc_strdup(mem_ctx, tokens[0]);
2848         for (i = 1; i < str_list_length((const char **)tokens); i++)
2849                 ret = talloc_asprintf_append_buffer(ret, "%s", tokens[i]);
2850
2851         talloc_free(tokens);
2852
2853         return ret;
2854 }
2855
2856 /*
2857   return domain functional level
2858   returns DS_DOMAIN_FUNCTION_*
2859  */
2860 int dsdb_functional_level(struct ldb_context *ldb)
2861 {
2862         int *domainFunctionality =
2863                 talloc_get_type(ldb_get_opaque(ldb, "domainFunctionality"), int);
2864         if (!domainFunctionality) {
2865                 DEBUG(0,(__location__ ": WARNING: domainFunctionality not setup\n"));
2866                 return DS_DOMAIN_FUNCTION_2000;
2867         }
2868         return *domainFunctionality;
2869 }
2870
2871 /*
2872   set a GUID in an extended DN structure
2873  */
2874 int dsdb_set_extended_dn_guid(struct ldb_dn *dn, const struct GUID *guid, const char *component_name)
2875 {
2876         struct ldb_val v;
2877         NTSTATUS status;
2878         int ret;
2879
2880         status = GUID_to_ndr_blob(guid, dn, &v);
2881         if (!NT_STATUS_IS_OK(status)) {
2882                 return LDB_ERR_INVALID_ATTRIBUTE_SYNTAX;
2883         }
2884
2885         ret = ldb_dn_set_extended_component(dn, component_name, &v);
2886         data_blob_free(&v);
2887         return ret;
2888 }
2889
2890 /*
2891   return a GUID from a extended DN structure
2892  */
2893 NTSTATUS dsdb_get_extended_dn_guid(struct ldb_dn *dn, struct GUID *guid, const char *component_name)
2894 {
2895         const struct ldb_val *v;
2896
2897         v = ldb_dn_get_extended_component(dn, component_name);
2898         if (v == NULL) {
2899                 return NT_STATUS_OBJECT_NAME_NOT_FOUND;
2900         }
2901
2902         return GUID_from_ndr_blob(v, guid);
2903 }
2904
2905 /*
2906   return a uint64_t from a extended DN structure
2907  */
2908 NTSTATUS dsdb_get_extended_dn_uint64(struct ldb_dn *dn, uint64_t *val, const char *component_name)
2909 {
2910         const struct ldb_val *v;
2911         char *s;
2912
2913         v = ldb_dn_get_extended_component(dn, component_name);
2914         if (v == NULL) {
2915                 return NT_STATUS_OBJECT_NAME_NOT_FOUND;
2916         }
2917         s = talloc_strndup(dn, (const char *)v->data, v->length);
2918         NT_STATUS_HAVE_NO_MEMORY(s);
2919
2920         *val = strtoull(s, NULL, 0);
2921
2922         talloc_free(s);
2923         return NT_STATUS_OK;
2924 }
2925
2926 /*
2927   return a NTTIME from a extended DN structure
2928  */
2929 NTSTATUS dsdb_get_extended_dn_nttime(struct ldb_dn *dn, NTTIME *nttime, const char *component_name)
2930 {
2931         return dsdb_get_extended_dn_uint64(dn, nttime, component_name);
2932 }
2933
2934 /*
2935   return a uint32_t from a extended DN structure
2936  */
2937 NTSTATUS dsdb_get_extended_dn_uint32(struct ldb_dn *dn, uint32_t *val, const char *component_name)
2938 {
2939         const struct ldb_val *v;
2940         char *s;
2941
2942         v = ldb_dn_get_extended_component(dn, component_name);
2943         if (v == NULL) {
2944                 return NT_STATUS_OBJECT_NAME_NOT_FOUND;
2945         }
2946
2947         s = talloc_strndup(dn, (const char *)v->data, v->length);
2948         NT_STATUS_HAVE_NO_MEMORY(s);
2949
2950         *val = strtoul(s, NULL, 0);
2951
2952         talloc_free(s);
2953         return NT_STATUS_OK;
2954 }
2955
2956 /*
2957   return RMD_FLAGS directly from a ldb_dn
2958   returns 0 if not found
2959  */
2960 uint32_t dsdb_dn_rmd_flags(struct ldb_dn *dn)
2961 {
2962         const struct ldb_val *v;
2963         char buf[32];
2964         v = ldb_dn_get_extended_component(dn, "RMD_FLAGS");
2965         if (!v || v->length > sizeof(buf)-1) return 0;
2966         strncpy(buf, (const char *)v->data, v->length);
2967         buf[v->length] = 0;
2968         return strtoul(buf, NULL, 10);
2969 }
2970
2971 /*
2972   return RMD_FLAGS directly from a ldb_val for a DN
2973   returns 0 if RMD_FLAGS is not found
2974  */
2975 uint32_t dsdb_dn_val_rmd_flags(struct ldb_val *val)
2976 {
2977         const char *p;
2978         uint32_t flags;
2979         char *end;
2980
2981         if (val->length < 13) {
2982                 return 0;
2983         }
2984         p = memmem(val->data, val->length-2, "<RMD_FLAGS=", 11);
2985         if (!p) {
2986                 return 0;
2987         }
2988         flags = strtoul(p+11, &end, 10);
2989         if (!end || *end != '>') {
2990                 /* it must end in a > */
2991                 return 0;
2992         }
2993         return flags;
2994 }
2995
2996 /*
2997   return true if a ldb_val containing a DN in storage form is deleted
2998  */
2999 bool dsdb_dn_is_deleted_val(struct ldb_val *val)
3000 {
3001         return (dsdb_dn_val_rmd_flags(val) & DSDB_RMD_FLAG_DELETED) != 0;
3002 }
3003
3004 /*
3005   return true if a ldb_val containing a DN in storage form is
3006   in the upgraded w2k3 linked attribute format
3007  */
3008 bool dsdb_dn_is_upgraded_link_val(struct ldb_val *val)
3009 {
3010         return memmem(val->data, val->length, "<RMD_ADDTIME=", 13) != NULL;
3011 }
3012
3013 /*
3014   return a DN for a wellknown GUID
3015  */
3016 int dsdb_wellknown_dn(struct ldb_context *samdb, TALLOC_CTX *mem_ctx,
3017                       struct ldb_dn *nc_root, const char *wk_guid,
3018                       struct ldb_dn **wkguid_dn)
3019 {
3020         TALLOC_CTX *tmp_ctx = talloc_new(mem_ctx);
3021         const char *attrs[] = { NULL };
3022         int ret;
3023         struct ldb_dn *dn;
3024         struct ldb_result *res;
3025
3026         /* construct the magic WKGUID DN */
3027         dn = ldb_dn_new_fmt(tmp_ctx, samdb, "<WKGUID=%s,%s>",
3028                             wk_guid, ldb_dn_get_linearized(nc_root));
3029         if (!wkguid_dn) {
3030                 talloc_free(tmp_ctx);
3031                 return LDB_ERR_OPERATIONS_ERROR;
3032         }
3033
3034         ret = dsdb_search_dn_with_deleted(samdb, tmp_ctx, &res, dn, attrs);
3035         if (ret != LDB_SUCCESS) {
3036                 talloc_free(tmp_ctx);
3037                 return ret;
3038         }
3039
3040         (*wkguid_dn) = talloc_steal(mem_ctx, res->msgs[0]->dn);
3041         talloc_free(tmp_ctx);
3042         return LDB_SUCCESS;
3043 }
3044
3045
3046 static int dsdb_dn_compare_ptrs(struct ldb_dn **dn1, struct ldb_dn **dn2)
3047 {
3048         return ldb_dn_compare(*dn1, *dn2);
3049 }
3050
3051 /*
3052   find a NC root given a DN within the NC
3053  */
3054 int dsdb_find_nc_root(struct ldb_context *samdb, TALLOC_CTX *mem_ctx, struct ldb_dn *dn,
3055                       struct ldb_dn **nc_root)
3056 {
3057         const char *root_attrs[] = { "namingContexts", NULL };
3058         TALLOC_CTX *tmp_ctx;
3059         int ret;
3060         struct ldb_message_element *el;
3061         struct ldb_result *root_res;
3062         int i;
3063         struct ldb_dn **nc_dns;
3064
3065         tmp_ctx = talloc_new(samdb);
3066         if (tmp_ctx == NULL) {
3067                 return LDB_ERR_OPERATIONS_ERROR;
3068         }
3069
3070         ret = ldb_search(samdb, tmp_ctx, &root_res,
3071                          ldb_dn_new(tmp_ctx, samdb, ""), LDB_SCOPE_BASE, root_attrs, NULL);
3072         if (ret) {
3073                 DEBUG(1,("Searching for namingContexts in rootDSE failed: %s\n", ldb_errstring(samdb)));
3074                 talloc_free(tmp_ctx);
3075                 return ret;
3076        }
3077
3078        el = ldb_msg_find_element(root_res->msgs[0], "namingContexts");
3079        if (!el) {
3080                DEBUG(1,("Finding namingContexts element in root_res failed: %s\n",
3081                         ldb_errstring(samdb)));
3082                talloc_free(tmp_ctx);
3083                return LDB_ERR_NO_SUCH_ATTRIBUTE;
3084        }
3085
3086        nc_dns = talloc_array(tmp_ctx, struct ldb_dn *, el->num_values);
3087        if (!nc_dns) {
3088                talloc_free(tmp_ctx);
3089                return LDB_ERR_OPERATIONS_ERROR;
3090        }
3091
3092        for (i=0; i<el->num_values; i++) {
3093                nc_dns[i] = ldb_dn_from_ldb_val(nc_dns, samdb, &el->values[i]);
3094                if (nc_dns[i] == NULL) {
3095                        talloc_free(tmp_ctx);
3096                        return LDB_ERR_OPERATIONS_ERROR;
3097                }
3098        }
3099
3100        TYPESAFE_QSORT(nc_dns, el->num_values, dsdb_dn_compare_ptrs);
3101
3102        for (i=0; i<el->num_values; i++) {
3103                if (ldb_dn_compare_base(nc_dns[i], dn) == 0) {
3104                        (*nc_root) = talloc_steal(mem_ctx, nc_dns[i]);
3105                        talloc_free(tmp_ctx);
3106                        return LDB_SUCCESS;
3107                }
3108        }
3109
3110        talloc_free(tmp_ctx);
3111        return LDB_ERR_NO_SUCH_OBJECT;
3112 }
3113
3114
3115 /*
3116   find the deleted objects DN for any object, by looking for the NC
3117   root, then looking up the wellknown GUID
3118  */
3119 int dsdb_get_deleted_objects_dn(struct ldb_context *ldb,
3120                                 TALLOC_CTX *mem_ctx, struct ldb_dn *obj_dn,
3121                                 struct ldb_dn **do_dn)
3122 {
3123         struct ldb_dn *nc_root;
3124         int ret;
3125
3126         ret = dsdb_find_nc_root(ldb, mem_ctx, obj_dn, &nc_root);
3127         if (ret != LDB_SUCCESS) {
3128                 return ret;
3129         }
3130
3131         ret = dsdb_wellknown_dn(ldb, mem_ctx, nc_root, DS_GUID_DELETED_OBJECTS_CONTAINER, do_dn);
3132         talloc_free(nc_root);
3133         return ret;
3134 }
3135
3136 /*
3137   return the tombstoneLifetime, in days
3138  */
3139 int dsdb_tombstone_lifetime(struct ldb_context *ldb, uint32_t *lifetime)
3140 {
3141         struct ldb_dn *dn;
3142         dn = samdb_config_dn(ldb);
3143         if (!dn) {
3144                 return LDB_ERR_NO_SUCH_OBJECT;
3145         }
3146         dn = ldb_dn_copy(ldb, dn);
3147         if (!dn) {
3148                 return LDB_ERR_OPERATIONS_ERROR;
3149         }
3150         /* see MS-ADTS section 7.1.1.2.4.1.1. There doesn't appear to
3151          be a wellknown GUID for this */
3152         if (!ldb_dn_add_child_fmt(dn, "CN=Directory Service,CN=Windows NT,CN=Services")) {
3153                 talloc_free(dn);
3154                 return LDB_ERR_OPERATIONS_ERROR;
3155         }
3156
3157         *lifetime = samdb_search_uint(ldb, dn, 180, dn, "tombstoneLifetime", "objectClass=nTDSService");
3158         talloc_free(dn);
3159         return LDB_SUCCESS;
3160 }
3161
3162 /*
3163   compare a ldb_val to a string case insensitively
3164  */
3165 int samdb_ldb_val_case_cmp(const char *s, struct ldb_val *v)
3166 {
3167         size_t len = strlen(s);
3168         int ret;
3169         if (len > v->length) return 1;
3170         ret = strncasecmp(s, (const char *)v->data, v->length);
3171         if (ret != 0) return ret;
3172         if (v->length > len && v->data[len] != 0) {
3173                 return -1;
3174         }
3175         return 0;
3176 }
3177
3178
3179 /*
3180   load the UDV for a partition in v2 format
3181   The list is returned sorted, and with our local cursor added
3182  */
3183 int dsdb_load_udv_v2(struct ldb_context *samdb, struct ldb_dn *dn, TALLOC_CTX *mem_ctx,
3184                      struct drsuapi_DsReplicaCursor2 **cursors, uint32_t *count)
3185 {
3186         static const char *attrs[] = { "replUpToDateVector", NULL };
3187         struct ldb_result *r;
3188         const struct ldb_val *ouv_value;
3189         int ret, i;
3190         uint64_t highest_usn;
3191         const struct GUID *our_invocation_id;
3192         struct timeval now = timeval_current();
3193
3194         ret = ldb_search(samdb, mem_ctx, &r, dn, LDB_SCOPE_BASE, attrs, NULL);
3195         if (ret != LDB_SUCCESS) {
3196                 return ret;
3197         }
3198
3199         ouv_value = ldb_msg_find_ldb_val(r->msgs[0], "replUpToDateVector");
3200         if (ouv_value) {
3201                 enum ndr_err_code ndr_err;
3202                 struct replUpToDateVectorBlob ouv;
3203
3204                 ndr_err = ndr_pull_struct_blob(ouv_value, r,
3205                                                lp_iconv_convenience(ldb_get_opaque(samdb, "loadparm")), &ouv,
3206                                                (ndr_pull_flags_fn_t)ndr_pull_replUpToDateVectorBlob);
3207                 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
3208                         talloc_free(r);
3209                         return LDB_ERR_INVALID_ATTRIBUTE_SYNTAX;
3210                 }
3211                 if (ouv.version != 2) {
3212                         /* we always store as version 2, and
3213                          * replUpToDateVector is not replicated
3214                          */
3215                         return LDB_ERR_INVALID_ATTRIBUTE_SYNTAX;
3216                 }
3217
3218                 *count = ouv.ctr.ctr2.count;
3219                 *cursors = talloc_steal(mem_ctx, ouv.ctr.ctr2.cursors);
3220         } else {
3221                 *count = 0;
3222                 *cursors = NULL;
3223         }
3224
3225         talloc_free(r);
3226
3227         our_invocation_id = samdb_ntds_invocation_id(samdb);
3228         if (!our_invocation_id) {
3229                 DEBUG(0,(__location__ ": No invocationID on samdb - %s\n", ldb_errstring(samdb)));
3230                 talloc_free(*cursors);
3231                 return LDB_ERR_OPERATIONS_ERROR;
3232         }
3233
3234         ret = dsdb_load_partition_usn(samdb, dn, &highest_usn, NULL);
3235         if (ret != LDB_SUCCESS) {
3236                 /* nothing to add - this can happen after a vampire */
3237                 TYPESAFE_QSORT(*cursors, *count, drsuapi_DsReplicaCursor2_compare);
3238                 return LDB_SUCCESS;
3239         }
3240
3241         for (i=0; i<*count; i++) {
3242                 if (GUID_equal(our_invocation_id, &(*cursors)[i].source_dsa_invocation_id)) {
3243                         (*cursors)[i].highest_usn = highest_usn;
3244                         (*cursors)[i].last_sync_success = timeval_to_nttime(&now);
3245                         TYPESAFE_QSORT(*cursors, *count, drsuapi_DsReplicaCursor2_compare);
3246                         return LDB_SUCCESS;
3247                 }
3248         }
3249
3250         (*cursors) = talloc_realloc(mem_ctx, *cursors, struct drsuapi_DsReplicaCursor2, (*count)+1);
3251         if (! *cursors) {
3252                 return LDB_ERR_OPERATIONS_ERROR;
3253         }
3254
3255         (*cursors)[*count].source_dsa_invocation_id = *our_invocation_id;
3256         (*cursors)[*count].highest_usn = highest_usn;
3257         (*cursors)[*count].last_sync_success = timeval_to_nttime(&now);
3258         (*count)++;
3259
3260         TYPESAFE_QSORT(*cursors, *count, drsuapi_DsReplicaCursor2_compare);
3261
3262         return LDB_SUCCESS;
3263 }
3264
3265 /*
3266   load the UDV for a partition in version 1 format
3267   The list is returned sorted, and with our local cursor added
3268  */
3269 int dsdb_load_udv_v1(struct ldb_context *samdb, struct ldb_dn *dn, TALLOC_CTX *mem_ctx,
3270                      struct drsuapi_DsReplicaCursor **cursors, uint32_t *count)
3271 {
3272         struct drsuapi_DsReplicaCursor2 *v2;
3273         int ret, i;
3274
3275         ret = dsdb_load_udv_v2(samdb, dn, mem_ctx, &v2, count);
3276         if (ret != LDB_SUCCESS) {
3277                 return ret;
3278         }
3279
3280         if (*count == 0) {
3281                 talloc_free(v2);
3282                 *cursors = NULL;
3283                 return LDB_SUCCESS;
3284         }
3285
3286         *cursors = talloc_array(mem_ctx, struct drsuapi_DsReplicaCursor, *count);
3287         if (*cursors == NULL) {
3288                 talloc_free(v2);
3289                 return LDB_ERR_OPERATIONS_ERROR;
3290         }
3291
3292         for (i=0; i<*count; i++) {
3293                 (*cursors)[i].source_dsa_invocation_id = v2[i].source_dsa_invocation_id;
3294                 (*cursors)[i].highest_usn = v2[i].highest_usn;
3295         }
3296         talloc_free(v2);
3297         return LDB_SUCCESS;
3298 }
3299
3300 /*
3301   add a set of controls to a ldb_request structure based on a set of
3302   flags. See util.h for a list of available flags
3303  */
3304 int dsdb_request_add_controls(struct ldb_request *req, uint32_t dsdb_flags)
3305 {
3306         int ret;
3307         if (dsdb_flags & DSDB_SEARCH_SEARCH_ALL_PARTITIONS) {
3308                 struct ldb_search_options_control *options;
3309                 /* Using the phantom root control allows us to search all partitions */
3310                 options = talloc(req, struct ldb_search_options_control);
3311                 if (options == NULL) {
3312                         return LDB_ERR_OPERATIONS_ERROR;
3313                 }
3314                 options->search_options = LDB_SEARCH_OPTION_PHANTOM_ROOT;
3315
3316                 ret = ldb_request_add_control(req,
3317                                               LDB_CONTROL_SEARCH_OPTIONS_OID,
3318                                               true, options);
3319                 if (ret != LDB_SUCCESS) {
3320                         return ret;
3321                 }
3322         }
3323
3324         if (dsdb_flags & DSDB_SEARCH_SHOW_DELETED) {
3325                 ret = ldb_request_add_control(req, LDB_CONTROL_SHOW_DELETED_OID, true, NULL);
3326                 if (ret != LDB_SUCCESS) {
3327                         return ret;
3328                 }
3329         }
3330
3331         if (dsdb_flags & DSDB_SEARCH_SHOW_DN_IN_STORAGE_FORMAT) {
3332                 ret = ldb_request_add_control(req, DSDB_CONTROL_DN_STORAGE_FORMAT_OID, true, NULL);
3333                 if (ret != LDB_SUCCESS) {
3334                         return ret;
3335                 }
3336         }
3337
3338         if (dsdb_flags & DSDB_SEARCH_SHOW_EXTENDED_DN) {
3339                 struct ldb_extended_dn_control *extended_ctrl = talloc(req, struct ldb_extended_dn_control);
3340                 if (!extended_ctrl) {
3341                         return LDB_ERR_OPERATIONS_ERROR;
3342                 }
3343                 extended_ctrl->type = 1;
3344
3345                 ret = ldb_request_add_control(req, LDB_CONTROL_EXTENDED_DN_OID, true, extended_ctrl);
3346                 if (ret != LDB_SUCCESS) {
3347                         return ret;
3348                 }
3349         }
3350
3351         if (dsdb_flags & DSDB_SEARCH_REVEAL_INTERNALS) {
3352                 ret = ldb_request_add_control(req, LDB_CONTROL_REVEAL_INTERNALS, false, NULL);
3353                 if (ret != LDB_SUCCESS) {
3354                         return ret;
3355                 }
3356         }
3357
3358         if (dsdb_flags & DSDB_MODIFY_RELAX) {
3359                 ret = ldb_request_add_control(req, LDB_CONTROL_RELAX_OID, false, NULL);
3360                 if (ret != LDB_SUCCESS) {
3361                         return ret;
3362                 }
3363         }
3364
3365         if (dsdb_flags & DSDB_MODIFY_PERMISSIVE) {
3366                 ret = ldb_request_add_control(req, LDB_CONTROL_PERMISSIVE_MODIFY_OID, false, NULL);
3367                 if (ret != LDB_SUCCESS) {
3368                         return ret;
3369                 }
3370         }
3371
3372         if (dsdb_flags & DSDB_FLAG_AS_SYSTEM) {
3373                 ret = ldb_request_add_control(req, LDB_CONTROL_AS_SYSTEM_OID, false, NULL);
3374                 if (ret != LDB_SUCCESS) {
3375                         return ret;
3376                 }
3377         }
3378
3379         return LDB_SUCCESS;
3380 }
3381
3382 /*
3383   a modify with a set of controls
3384 */
3385 int dsdb_modify(struct ldb_context *ldb, const struct ldb_message *message,
3386                 uint32_t dsdb_flags)
3387 {
3388         struct ldb_request *req;
3389         int ret;
3390
3391         ret = ldb_build_mod_req(&req, ldb, ldb,
3392                                 message,
3393                                 NULL,
3394                                 NULL,
3395                                 ldb_op_default_callback,
3396                                 NULL);
3397
3398         if (ret != LDB_SUCCESS) return ret;
3399
3400         ret = dsdb_request_add_controls(req, dsdb_flags);
3401         if (ret != LDB_SUCCESS) {
3402                 talloc_free(req);
3403                 return ret;
3404         }
3405
3406         ret = dsdb_autotransaction_request(ldb, req);
3407
3408         talloc_free(req);
3409         return ret;
3410 }
3411
3412 /*
3413   like dsdb_modify() but set all the element flags to
3414   LDB_FLAG_MOD_REPLACE
3415  */
3416 int dsdb_replace(struct ldb_context *ldb, struct ldb_message *msg, uint32_t dsdb_flags)
3417 {
3418         int i;
3419
3420         /* mark all the message elements as LDB_FLAG_MOD_REPLACE */
3421         for (i=0;i<msg->num_elements;i++) {
3422                 msg->elements[i].flags = LDB_FLAG_MOD_REPLACE;
3423         }
3424
3425         return dsdb_modify(ldb, msg, dsdb_flags);
3426 }