Remove "idmap alloc config : range" parameter
[kai/samba.git] / source3 / winbindd / idmap_ldap.c
1 /*
2    Unix SMB/CIFS implementation.
3
4    idmap LDAP backend
5
6    Copyright (C) Tim Potter             2000
7    Copyright (C) Jim McDonough <jmcd@us.ibm.com>        2003
8    Copyright (C) Gerald Carter          2003
9    Copyright (C) Simo Sorce             2003-2007
10
11    This program is free software; you can redistribute it and/or modify
12    it under the terms of the GNU General Public License as published by
13    the Free Software Foundation; either version 3 of the License, or
14    (at your option) any later version.
15
16    This program is distributed in the hope that it will be useful,
17    but WITHOUT ANY WARRANTY; without even the implied warranty of
18    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
19    GNU General Public License for more details.
20
21    You should have received a copy of the GNU General Public License
22    along with this program.  If not, see <http://www.gnu.org/licenses/>.
23 */
24
25 #include "includes.h"
26 #include "winbindd.h"
27
28 #undef DBGC_CLASS
29 #define DBGC_CLASS DBGC_IDMAP
30
31 #include <lber.h>
32 #include <ldap.h>
33
34 #include "smbldap.h"
35
36 static char *idmap_fetch_secret(const char *backend, bool alloc,
37                                 const char *domain, const char *identity)
38 {
39         char *tmp, *ret;
40         int r;
41
42         if (alloc) {
43                 r = asprintf(&tmp, "IDMAP_ALLOC_%s", backend);
44         } else {
45                 r = asprintf(&tmp, "IDMAP_%s_%s", backend, domain);
46         }
47
48         if (r < 0)
49                 return NULL;
50
51         strupper_m(tmp); /* make sure the key is case insensitive */
52         ret = secrets_fetch_generic(tmp, identity);
53
54         SAFE_FREE(tmp);
55
56         return ret;
57 }
58
59 struct idmap_ldap_context {
60         struct smbldap_state *smbldap_state;
61         char *url;
62         char *suffix;
63         char *user_dn;
64         uint32_t filter_low_id, filter_high_id;         /* Filter range */
65         bool anon;
66 };
67
68 struct idmap_ldap_alloc_context {
69         struct smbldap_state *smbldap_state;
70         char *url;
71         char *suffix;
72         char *user_dn;
73         uid_t low_uid, high_uid;      /* Range of uids */
74         gid_t low_gid, high_gid;      /* Range of gids */
75
76 };
77
78 #define CHECK_ALLOC_DONE(mem) do { \
79         if (!mem) { \
80                 DEBUG(0, ("Out of memory!\n")); \
81                 ret = NT_STATUS_NO_MEMORY; \
82                 goto done; \
83         } } while (0)
84
85 /**********************************************************************
86  IDMAP ALLOC TDB BACKEND
87 **********************************************************************/
88
89 static struct idmap_ldap_alloc_context *idmap_alloc_ldap;
90
91 /*********************************************************************
92  ********************************************************************/
93
94 static NTSTATUS get_credentials( TALLOC_CTX *mem_ctx,
95                                  struct smbldap_state *ldap_state,
96                                  const char *config_option,
97                                  struct idmap_domain *dom,
98                                  char **dn )
99 {
100         NTSTATUS ret = NT_STATUS_UNSUCCESSFUL;
101         char *secret = NULL;
102         const char *tmp = NULL;
103         char *user_dn = NULL;
104         bool anon = False;
105
106         /* assume anonymous if we don't have a specified user */
107
108         tmp = lp_parm_const_string(-1, config_option, "ldap_user_dn", NULL);
109
110         if ( tmp ) {
111                 if (!dom) {
112                         /* only the alloc backend can pass in a NULL dom */
113                         secret = idmap_fetch_secret("ldap", True,
114                                                     NULL, tmp);
115                 } else {
116                         secret = idmap_fetch_secret("ldap", False,
117                                                     dom->name, tmp);
118                 }
119
120                 if (!secret) {
121                         DEBUG(0, ("get_credentials: Unable to fetch "
122                                   "auth credentials for %s in %s\n",
123                                   tmp, (dom==NULL)?"ALLOC":dom->name));
124                         ret = NT_STATUS_ACCESS_DENIED;
125                         goto done;
126                 }
127                 *dn = talloc_strdup(mem_ctx, tmp);
128                 CHECK_ALLOC_DONE(*dn);
129         } else {
130                 if (!fetch_ldap_pw(&user_dn, &secret)) {
131                         DEBUG(2, ("get_credentials: Failed to lookup ldap "
132                                   "bind creds. Using anonymous connection.\n"));
133                         anon = True;
134                 } else {
135                         *dn = talloc_strdup(mem_ctx, user_dn);
136                         SAFE_FREE( user_dn );
137                         CHECK_ALLOC_DONE(*dn);
138                 }
139         }
140
141         smbldap_set_creds(ldap_state, anon, *dn, secret);
142         ret = NT_STATUS_OK;
143
144 done:
145         SAFE_FREE(secret);
146
147         return ret;
148 }
149
150
151 /**********************************************************************
152  Verify the sambaUnixIdPool entry in the directory.
153 **********************************************************************/
154
155 static NTSTATUS verify_idpool(void)
156 {
157         NTSTATUS ret;
158         TALLOC_CTX *ctx;
159         LDAPMessage *result = NULL;
160         LDAPMod **mods = NULL;
161         const char **attr_list;
162         char *filter;
163         int count;
164         int rc;
165
166         if ( ! idmap_alloc_ldap) {
167                 return NT_STATUS_UNSUCCESSFUL;
168         }
169
170         ctx = talloc_new(idmap_alloc_ldap);
171         if ( ! ctx) {
172                 DEBUG(0, ("Out of memory!\n"));
173                 return NT_STATUS_NO_MEMORY;
174         }
175
176         filter = talloc_asprintf(ctx, "(objectclass=%s)", LDAP_OBJ_IDPOOL);
177         CHECK_ALLOC_DONE(filter);
178
179         attr_list = get_attr_list(ctx, idpool_attr_list);
180         CHECK_ALLOC_DONE(attr_list);
181
182         rc = smbldap_search(idmap_alloc_ldap->smbldap_state,
183                                 idmap_alloc_ldap->suffix,
184                                 LDAP_SCOPE_SUBTREE,
185                                 filter,
186                                 attr_list,
187                                 0,
188                                 &result);
189
190         if (rc != LDAP_SUCCESS) {
191                 DEBUG(1, ("Unable to verify the idpool, "
192                           "cannot continue initialization!\n"));
193                 return NT_STATUS_UNSUCCESSFUL;
194         }
195
196         count = ldap_count_entries(idmap_alloc_ldap->smbldap_state->ldap_struct,
197                                    result);
198
199         ldap_msgfree(result);
200
201         if ( count > 1 ) {
202                 DEBUG(0,("Multiple entries returned from %s (base == %s)\n",
203                         filter, idmap_alloc_ldap->suffix));
204                 ret = NT_STATUS_UNSUCCESSFUL;
205                 goto done;
206         }
207         else if (count == 0) {
208                 char *uid_str, *gid_str;
209
210                 uid_str = talloc_asprintf(ctx, "%lu",
211                                 (unsigned long)idmap_alloc_ldap->low_uid);
212                 gid_str = talloc_asprintf(ctx, "%lu",
213                                 (unsigned long)idmap_alloc_ldap->low_gid);
214
215                 smbldap_set_mod(&mods, LDAP_MOD_ADD,
216                                 "objectClass", LDAP_OBJ_IDPOOL);
217                 smbldap_set_mod(&mods, LDAP_MOD_ADD,
218                                 get_attr_key2string(idpool_attr_list,
219                                                     LDAP_ATTR_UIDNUMBER),
220                                 uid_str);
221                 smbldap_set_mod(&mods, LDAP_MOD_ADD,
222                                 get_attr_key2string(idpool_attr_list,
223                                                     LDAP_ATTR_GIDNUMBER),
224                                 gid_str);
225                 if (mods) {
226                         rc = smbldap_modify(idmap_alloc_ldap->smbldap_state,
227                                                 idmap_alloc_ldap->suffix,
228                                                 mods);
229                         ldap_mods_free(mods, True);
230                 } else {
231                         ret = NT_STATUS_UNSUCCESSFUL;
232                         goto done;
233                 }
234         }
235
236         ret = (rc == LDAP_SUCCESS)?NT_STATUS_OK:NT_STATUS_UNSUCCESSFUL;
237 done:
238         talloc_free(ctx);
239         return ret;
240 }
241
242 /*****************************************************************************
243  Initialise idmap database.
244 *****************************************************************************/
245
246 static NTSTATUS idmap_ldap_alloc_init(const char *params)
247 {
248         NTSTATUS ret = NT_STATUS_UNSUCCESSFUL;
249         const char *range;
250         const char *tmp;
251         uid_t low_uid = 0;
252         uid_t high_uid = 0;
253         gid_t low_gid = 0;
254         gid_t high_gid = 0;
255
256         /* Only do init if we are online */
257         if (idmap_is_offline()) {
258                 return NT_STATUS_FILE_IS_OFFLINE;
259         }
260
261         idmap_alloc_ldap = TALLOC_ZERO_P(NULL, struct idmap_ldap_alloc_context);
262         CHECK_ALLOC_DONE( idmap_alloc_ldap );
263
264         /* load ranges */
265
266         if (!lp_idmap_uid(&low_uid, &high_uid)
267             || !lp_idmap_gid(&low_gid, &high_gid)) {
268                 DEBUG(1, ("idmap uid or idmap gid missing\n"));
269                 ret = NT_STATUS_UNSUCCESSFUL;
270                 goto done;
271         }
272
273         idmap_alloc_ldap->low_uid = low_uid;
274         idmap_alloc_ldap->high_uid = high_uid;
275         idmap_alloc_ldap->low_gid = low_gid;
276         idmap_alloc_ldap->high_gid= high_gid;
277
278         if (idmap_alloc_ldap->high_uid <= idmap_alloc_ldap->low_uid) {
279                 DEBUG(1, ("idmap uid range invalid\n"));
280                 DEBUGADD(1, ("idmap will be unable to map foreign SIDs\n"));
281                 ret = NT_STATUS_UNSUCCESSFUL;
282                 goto done;
283         }
284
285         if (idmap_alloc_ldap->high_gid <= idmap_alloc_ldap->low_gid) {
286                 DEBUG(1, ("idmap gid range invalid\n"));
287                 DEBUGADD(1, ("idmap will be unable to map foreign SIDs\n"));
288                 ret = NT_STATUS_UNSUCCESSFUL;
289                 goto done;
290         }
291
292         if (params && *params) {
293                 /* assume location is the only parameter */
294                 idmap_alloc_ldap->url = talloc_strdup(idmap_alloc_ldap, params);
295         } else {
296                 tmp = lp_parm_const_string(-1, "idmap alloc config",
297                                            "ldap_url", NULL);
298
299                 if ( ! tmp) {
300                         DEBUG(1, ("ERROR: missing idmap ldap url\n"));
301                         ret = NT_STATUS_UNSUCCESSFUL;
302                         goto done;
303                 }
304
305                 idmap_alloc_ldap->url = talloc_strdup(idmap_alloc_ldap, tmp);
306         }
307         CHECK_ALLOC_DONE( idmap_alloc_ldap->url );
308
309         tmp = lp_parm_const_string(-1, "idmap alloc config",
310                                    "ldap_base_dn", NULL);
311         if ( ! tmp || ! *tmp) {
312                 tmp = lp_ldap_idmap_suffix();
313                 if ( ! tmp) {
314                         DEBUG(1, ("ERROR: missing idmap ldap suffix\n"));
315                         ret = NT_STATUS_UNSUCCESSFUL;
316                         goto done;
317                 }
318         }
319
320         idmap_alloc_ldap->suffix = talloc_strdup(idmap_alloc_ldap, tmp);
321         CHECK_ALLOC_DONE( idmap_alloc_ldap->suffix );
322
323         ret = smbldap_init(idmap_alloc_ldap, winbind_event_context(),
324                            idmap_alloc_ldap->url,
325                            &idmap_alloc_ldap->smbldap_state);
326         if (!NT_STATUS_IS_OK(ret)) {
327                 DEBUG(1, ("ERROR: smbldap_init (%s) failed!\n",
328                           idmap_alloc_ldap->url));
329                 goto done;
330         }
331
332         ret = get_credentials( idmap_alloc_ldap,
333                                idmap_alloc_ldap->smbldap_state,
334                                "idmap alloc config", NULL,
335                                &idmap_alloc_ldap->user_dn );
336         if ( !NT_STATUS_IS_OK(ret) ) {
337                 DEBUG(1,("idmap_ldap_alloc_init: Failed to get connection "
338                          "credentials (%s)\n", nt_errstr(ret)));
339                 goto done;
340         }
341
342         /* see if the idmap suffix and sub entries exists */
343
344         ret = verify_idpool();
345
346  done:
347         if ( !NT_STATUS_IS_OK( ret ) )
348                 TALLOC_FREE( idmap_alloc_ldap );
349
350         return ret;
351 }
352
353 /********************************
354  Allocate a new uid or gid
355 ********************************/
356
357 static NTSTATUS idmap_ldap_allocate_id(struct unixid *xid)
358 {
359         TALLOC_CTX *ctx;
360         NTSTATUS ret = NT_STATUS_UNSUCCESSFUL;
361         int rc = LDAP_SERVER_DOWN;
362         int count = 0;
363         LDAPMessage *result = NULL;
364         LDAPMessage *entry = NULL;
365         LDAPMod **mods = NULL;
366         char *id_str;
367         char *new_id_str;
368         char *filter = NULL;
369         const char *dn = NULL;
370         const char **attr_list;
371         const char *type;
372
373         /* Only do query if we are online */
374         if (idmap_is_offline()) {
375                 return NT_STATUS_FILE_IS_OFFLINE;
376         }
377
378         if ( ! idmap_alloc_ldap) {
379                 return NT_STATUS_UNSUCCESSFUL;
380         }
381
382         ctx = talloc_new(idmap_alloc_ldap);
383         if ( ! ctx) {
384                 DEBUG(0, ("Out of memory!\n"));
385                 return NT_STATUS_NO_MEMORY;
386         }
387
388         /* get type */
389         switch (xid->type) {
390
391         case ID_TYPE_UID:
392                 type = get_attr_key2string(idpool_attr_list,
393                                            LDAP_ATTR_UIDNUMBER);
394                 break;
395
396         case ID_TYPE_GID:
397                 type = get_attr_key2string(idpool_attr_list,
398                                            LDAP_ATTR_GIDNUMBER);
399                 break;
400
401         default:
402                 DEBUG(2, ("Invalid ID type (0x%x)\n", xid->type));
403                 return NT_STATUS_INVALID_PARAMETER;
404         }
405
406         filter = talloc_asprintf(ctx, "(objectClass=%s)", LDAP_OBJ_IDPOOL);
407         CHECK_ALLOC_DONE(filter);
408
409         attr_list = get_attr_list(ctx, idpool_attr_list);
410         CHECK_ALLOC_DONE(attr_list);
411
412         DEBUG(10, ("Search of the id pool (filter: %s)\n", filter));
413
414         rc = smbldap_search(idmap_alloc_ldap->smbldap_state,
415                                 idmap_alloc_ldap->suffix,
416                                LDAP_SCOPE_SUBTREE, filter,
417                                attr_list, 0, &result);
418
419         if (rc != LDAP_SUCCESS) {
420                 DEBUG(0,("%s object not found\n", LDAP_OBJ_IDPOOL));
421                 goto done;
422         }
423
424         talloc_autofree_ldapmsg(ctx, result);
425
426         count = ldap_count_entries(idmap_alloc_ldap->smbldap_state->ldap_struct,
427                                    result);
428         if (count != 1) {
429                 DEBUG(0,("Single %s object not found\n", LDAP_OBJ_IDPOOL));
430                 goto done;
431         }
432
433         entry = ldap_first_entry(idmap_alloc_ldap->smbldap_state->ldap_struct,
434                                  result);
435
436         dn = smbldap_talloc_dn(ctx,
437                                idmap_alloc_ldap->smbldap_state->ldap_struct,
438                                entry);
439         if ( ! dn) {
440                 goto done;
441         }
442
443         if ( ! (id_str = smbldap_talloc_single_attribute(idmap_alloc_ldap->smbldap_state->ldap_struct,
444                                 entry, type, ctx))) {
445                 DEBUG(0,("%s attribute not found\n", type));
446                 goto done;
447         }
448         if ( ! id_str) {
449                 DEBUG(0,("Out of memory\n"));
450                 ret = NT_STATUS_NO_MEMORY;
451                 goto done;
452         }
453
454         xid->id = strtoul(id_str, NULL, 10);
455
456         /* make sure we still have room to grow */
457
458         switch (xid->type) {
459         case ID_TYPE_UID:
460                 if (xid->id > idmap_alloc_ldap->high_uid) {
461                         DEBUG(0,("Cannot allocate uid above %lu!\n",
462                                  (unsigned long)idmap_alloc_ldap->high_uid));
463                         goto done;
464                 }
465                 break;
466
467         case ID_TYPE_GID:
468                 if (xid->id > idmap_alloc_ldap->high_gid) {
469                         DEBUG(0,("Cannot allocate gid above %lu!\n",
470                                  (unsigned long)idmap_alloc_ldap->high_uid));
471                         goto done;
472                 }
473                 break;
474
475         default:
476                 /* impossible */
477                 goto done;
478         }
479
480         new_id_str = talloc_asprintf(ctx, "%lu", (unsigned long)xid->id + 1);
481         if ( ! new_id_str) {
482                 DEBUG(0,("Out of memory\n"));
483                 ret = NT_STATUS_NO_MEMORY;
484                 goto done;
485         }
486
487         smbldap_set_mod(&mods, LDAP_MOD_DELETE, type, id_str);
488         smbldap_set_mod(&mods, LDAP_MOD_ADD, type, new_id_str);
489
490         if (mods == NULL) {
491                 DEBUG(0,("smbldap_set_mod() failed.\n"));
492                 goto done;
493         }
494
495         DEBUG(10, ("Try to atomically increment the id (%s -> %s)\n",
496                    id_str, new_id_str));
497
498         rc = smbldap_modify(idmap_alloc_ldap->smbldap_state, dn, mods);
499
500         ldap_mods_free(mods, True);
501
502         if (rc != LDAP_SUCCESS) {
503                 DEBUG(1,("Failed to allocate new %s. "
504                          "smbldap_modify() failed.\n", type));
505                 goto done;
506         }
507
508         ret = NT_STATUS_OK;
509
510 done:
511         talloc_free(ctx);
512         return ret;
513 }
514
515 /**********************************
516  Get current highest id.
517 **********************************/
518
519 static NTSTATUS idmap_ldap_get_hwm(struct unixid *xid)
520 {
521         TALLOC_CTX *memctx;
522         NTSTATUS ret = NT_STATUS_UNSUCCESSFUL;
523         int rc = LDAP_SERVER_DOWN;
524         int count = 0;
525         LDAPMessage *result = NULL;
526         LDAPMessage *entry = NULL;
527         char *id_str;
528         char *filter = NULL;
529         const char **attr_list;
530         const char *type;
531
532         /* Only do query if we are online */
533         if (idmap_is_offline()) {
534                 return NT_STATUS_FILE_IS_OFFLINE;
535         }
536
537         if ( ! idmap_alloc_ldap) {
538                 return NT_STATUS_UNSUCCESSFUL;
539         }
540
541         memctx = talloc_new(idmap_alloc_ldap);
542         if ( ! memctx) {
543                 DEBUG(0, ("Out of memory!\n"));
544                 return NT_STATUS_NO_MEMORY;
545         }
546
547         /* get type */
548         switch (xid->type) {
549
550         case ID_TYPE_UID:
551                 type = get_attr_key2string(idpool_attr_list,
552                                            LDAP_ATTR_UIDNUMBER);
553                 break;
554
555         case ID_TYPE_GID:
556                 type = get_attr_key2string(idpool_attr_list,
557                                            LDAP_ATTR_GIDNUMBER);
558                 break;
559
560         default:
561                 DEBUG(2, ("Invalid ID type (0x%x)\n", xid->type));
562                 return NT_STATUS_INVALID_PARAMETER;
563         }
564
565         filter = talloc_asprintf(memctx, "(objectClass=%s)", LDAP_OBJ_IDPOOL);
566         CHECK_ALLOC_DONE(filter);
567
568         attr_list = get_attr_list(memctx, idpool_attr_list);
569         CHECK_ALLOC_DONE(attr_list);
570
571         rc = smbldap_search(idmap_alloc_ldap->smbldap_state,
572                                 idmap_alloc_ldap->suffix,
573                                LDAP_SCOPE_SUBTREE, filter,
574                                attr_list, 0, &result);
575
576         if (rc != LDAP_SUCCESS) {
577                 DEBUG(0,("%s object not found\n", LDAP_OBJ_IDPOOL));
578                 goto done;
579         }
580
581         talloc_autofree_ldapmsg(memctx, result);
582
583         count = ldap_count_entries(idmap_alloc_ldap->smbldap_state->ldap_struct,
584                                    result);
585         if (count != 1) {
586                 DEBUG(0,("Single %s object not found\n", LDAP_OBJ_IDPOOL));
587                 goto done;
588         }
589
590         entry = ldap_first_entry(idmap_alloc_ldap->smbldap_state->ldap_struct,
591                                  result);
592
593         id_str = smbldap_talloc_single_attribute(idmap_alloc_ldap->smbldap_state->ldap_struct,
594                         entry, type, memctx);
595         if ( ! id_str) {
596                 DEBUG(0,("%s attribute not found\n", type));
597                 goto done;
598         }
599         if ( ! id_str) {
600                 DEBUG(0,("Out of memory\n"));
601                 ret = NT_STATUS_NO_MEMORY;
602                 goto done;
603         }
604
605         xid->id = strtoul(id_str, NULL, 10);
606
607         ret = NT_STATUS_OK;
608 done:
609         talloc_free(memctx);
610         return ret;
611 }
612 /**********************************
613  Set highest id.
614 **********************************/
615
616 static NTSTATUS idmap_ldap_set_hwm(struct unixid *xid)
617 {
618         TALLOC_CTX *ctx;
619         NTSTATUS ret = NT_STATUS_UNSUCCESSFUL;
620         int rc = LDAP_SERVER_DOWN;
621         int count = 0;
622         LDAPMessage *result = NULL;
623         LDAPMessage *entry = NULL;
624         LDAPMod **mods = NULL;
625         char *new_id_str;
626         char *filter = NULL;
627         const char *dn = NULL;
628         const char **attr_list;
629         const char *type;
630
631         /* Only do query if we are online */
632         if (idmap_is_offline()) {
633                 return NT_STATUS_FILE_IS_OFFLINE;
634         }
635
636         if ( ! idmap_alloc_ldap) {
637                 return NT_STATUS_UNSUCCESSFUL;
638         }
639
640         ctx = talloc_new(idmap_alloc_ldap);
641         if ( ! ctx) {
642                 DEBUG(0, ("Out of memory!\n"));
643                 return NT_STATUS_NO_MEMORY;
644         }
645
646         /* get type */
647         switch (xid->type) {
648
649         case ID_TYPE_UID:
650                 type = get_attr_key2string(idpool_attr_list,
651                                            LDAP_ATTR_UIDNUMBER);
652                 break;
653
654         case ID_TYPE_GID:
655                 type = get_attr_key2string(idpool_attr_list,
656                                            LDAP_ATTR_GIDNUMBER);
657                 break;
658
659         default:
660                 DEBUG(2, ("Invalid ID type (0x%x)\n", xid->type));
661                 return NT_STATUS_INVALID_PARAMETER;
662         }
663
664         filter = talloc_asprintf(ctx, "(objectClass=%s)", LDAP_OBJ_IDPOOL);
665         CHECK_ALLOC_DONE(filter);
666
667         attr_list = get_attr_list(ctx, idpool_attr_list);
668         CHECK_ALLOC_DONE(attr_list);
669
670         rc = smbldap_search(idmap_alloc_ldap->smbldap_state,
671                                 idmap_alloc_ldap->suffix,
672                                LDAP_SCOPE_SUBTREE, filter,
673                                attr_list, 0, &result);
674
675         if (rc != LDAP_SUCCESS) {
676                 DEBUG(0,("%s object not found\n", LDAP_OBJ_IDPOOL));
677                 goto done;
678         }
679
680         talloc_autofree_ldapmsg(ctx, result);
681
682         count = ldap_count_entries(idmap_alloc_ldap->smbldap_state->ldap_struct,
683                                    result);
684         if (count != 1) {
685                 DEBUG(0,("Single %s object not found\n", LDAP_OBJ_IDPOOL));
686                 goto done;
687         }
688
689         entry = ldap_first_entry(idmap_alloc_ldap->smbldap_state->ldap_struct,
690                                  result);
691
692         dn = smbldap_talloc_dn(ctx,
693                                 idmap_alloc_ldap->smbldap_state->ldap_struct,
694                                 entry);
695         if ( ! dn) {
696                 goto done;
697         }
698
699         new_id_str = talloc_asprintf(ctx, "%lu", (unsigned long)xid->id);
700         if ( ! new_id_str) {
701                 DEBUG(0,("Out of memory\n"));
702                 ret = NT_STATUS_NO_MEMORY;
703                 goto done;
704         }
705
706         smbldap_set_mod(&mods, LDAP_MOD_REPLACE, type, new_id_str);
707
708         if (mods == NULL) {
709                 DEBUG(0,("smbldap_set_mod() failed.\n"));
710                 goto done;
711         }
712
713         rc = smbldap_modify(idmap_alloc_ldap->smbldap_state, dn, mods);
714
715         ldap_mods_free(mods, True);
716
717         if (rc != LDAP_SUCCESS) {
718                 DEBUG(1,("Failed to allocate new %s. "
719                          "smbldap_modify() failed.\n", type));
720                 goto done;
721         }
722
723         ret = NT_STATUS_OK;
724
725 done:
726         talloc_free(ctx);
727         return ret;
728 }
729
730 /**********************************
731  Close idmap ldap alloc
732 **********************************/
733
734 static NTSTATUS idmap_ldap_alloc_close(void)
735 {
736         if (idmap_alloc_ldap) {
737                 smbldap_free_struct(&idmap_alloc_ldap->smbldap_state);
738                 DEBUG(5,("The connection to the LDAP server was closed\n"));
739                 /* maybe free the results here --metze */
740                 TALLOC_FREE(idmap_alloc_ldap);
741         }
742         return NT_STATUS_OK;
743 }
744
745
746 /**********************************************************************
747  IDMAP MAPPING LDAP BACKEND
748 **********************************************************************/
749
750 static int idmap_ldap_close_destructor(struct idmap_ldap_context *ctx)
751 {
752         smbldap_free_struct(&ctx->smbldap_state);
753         DEBUG(5,("The connection to the LDAP server was closed\n"));
754         /* maybe free the results here --metze */
755
756         return 0;
757 }
758
759 /********************************
760  Initialise idmap database.
761 ********************************/
762
763 static NTSTATUS idmap_ldap_db_init(struct idmap_domain *dom,
764                                    const char *params)
765 {
766         NTSTATUS ret;
767         struct idmap_ldap_context *ctx = NULL;
768         char *config_option = NULL;
769         const char *range = NULL;
770         const char *tmp = NULL;
771
772         /* Only do init if we are online */
773         if (idmap_is_offline()) {
774                 return NT_STATUS_FILE_IS_OFFLINE;
775         }
776
777         ctx = TALLOC_ZERO_P(dom, struct idmap_ldap_context);
778         if ( ! ctx) {
779                 DEBUG(0, ("Out of memory!\n"));
780                 return NT_STATUS_NO_MEMORY;
781         }
782
783         config_option = talloc_asprintf(ctx, "idmap config %s", dom->name);
784         if ( ! config_option) {
785                 DEBUG(0, ("Out of memory!\n"));
786                 ret = NT_STATUS_NO_MEMORY;
787                 goto done;
788         }
789
790         /* load ranges */
791         range = lp_parm_const_string(-1, config_option, "range", NULL);
792         if (range && range[0]) {
793                 if ((sscanf(range, "%u - %u", &ctx->filter_low_id,
794                                                 &ctx->filter_high_id) != 2) ||
795                     (ctx->filter_low_id > ctx->filter_high_id)) {
796                         DEBUG(1, ("ERROR: invalid filter range [%s]", range));
797                         ctx->filter_low_id = 0;
798                         ctx->filter_high_id = 0;
799                 }
800         }
801
802         if (params != NULL) {
803                 /* assume location is the only parameter */
804                 ctx->url = talloc_strdup(ctx, params);
805         } else {
806                 tmp = lp_parm_const_string(-1, config_option, "ldap_url", NULL);
807
808                 if ( ! tmp) {
809                         DEBUG(1, ("ERROR: missing idmap ldap url\n"));
810                         ret = NT_STATUS_UNSUCCESSFUL;
811                         goto done;
812                 }
813
814                 ctx->url = talloc_strdup(ctx, tmp);
815         }
816         CHECK_ALLOC_DONE(ctx->url);
817
818         tmp = lp_parm_const_string(-1, config_option, "ldap_base_dn", NULL);
819         if ( ! tmp || ! *tmp) {
820                 tmp = lp_ldap_idmap_suffix();
821                 if ( ! tmp) {
822                         DEBUG(1, ("ERROR: missing idmap ldap suffix\n"));
823                         ret = NT_STATUS_UNSUCCESSFUL;
824                         goto done;
825                 }
826         }
827
828         ctx->suffix = talloc_strdup(ctx, tmp);
829         CHECK_ALLOC_DONE(ctx->suffix);
830
831         ret = smbldap_init(ctx, winbind_event_context(), ctx->url,
832                            &ctx->smbldap_state);
833         if (!NT_STATUS_IS_OK(ret)) {
834                 DEBUG(1, ("ERROR: smbldap_init (%s) failed!\n", ctx->url));
835                 goto done;
836         }
837
838         ret = get_credentials( ctx, ctx->smbldap_state, config_option,
839                                dom, &ctx->user_dn );
840         if ( !NT_STATUS_IS_OK(ret) ) {
841                 DEBUG(1,("idmap_ldap_db_init: Failed to get connection "
842                          "credentials (%s)\n", nt_errstr(ret)));
843                 goto done;
844         }
845
846         /* set the destructor on the context, so that resource are properly
847            freed if the contexts is released */
848
849         talloc_set_destructor(ctx, idmap_ldap_close_destructor);
850
851         dom->private_data = ctx;
852
853         talloc_free(config_option);
854         return NT_STATUS_OK;
855
856 /*failed */
857 done:
858         talloc_free(ctx);
859         return ret;
860 }
861
862 /* max number of ids requested per batch query */
863 #define IDMAP_LDAP_MAX_IDS 30
864
865 /**********************************
866  lookup a set of unix ids.
867 **********************************/
868
869 /* this function searches up to IDMAP_LDAP_MAX_IDS entries
870  * in maps for a match */
871 static struct id_map *find_map_by_id(struct id_map **maps,
872                                      enum id_type type,
873                                      uint32_t id)
874 {
875         int i;
876
877         for (i = 0; i < IDMAP_LDAP_MAX_IDS; i++) {
878                 if (maps[i] == NULL) { /* end of the run */
879                         return NULL;
880                 }
881                 if ((maps[i]->xid.type == type) && (maps[i]->xid.id == id)) {
882                         return maps[i];
883                 }
884         }
885
886         return NULL;
887 }
888
889 static NTSTATUS idmap_ldap_unixids_to_sids(struct idmap_domain *dom,
890                                            struct id_map **ids)
891 {
892         NTSTATUS ret;
893         TALLOC_CTX *memctx;
894         struct idmap_ldap_context *ctx;
895         LDAPMessage *result = NULL;
896         const char *uidNumber;
897         const char *gidNumber;
898         const char **attr_list;
899         char *filter = NULL;
900         bool multi = False;
901         int idx = 0;
902         int bidx = 0;
903         int count;
904         int rc;
905         int i;
906
907         /* Only do query if we are online */
908         if (idmap_is_offline()) {
909                 return NT_STATUS_FILE_IS_OFFLINE;
910         }
911
912         ctx = talloc_get_type(dom->private_data, struct idmap_ldap_context);
913
914         memctx = talloc_new(ctx);
915         if ( ! memctx) {
916                 DEBUG(0, ("Out of memory!\n"));
917                 return NT_STATUS_NO_MEMORY;
918         }
919
920         uidNumber = get_attr_key2string(idpool_attr_list, LDAP_ATTR_UIDNUMBER);
921         gidNumber = get_attr_key2string(idpool_attr_list, LDAP_ATTR_GIDNUMBER);
922
923         attr_list = get_attr_list(memctx, sidmap_attr_list);
924
925         if ( ! ids[1]) {
926                 /* if we are requested just one mapping use the simple filter */
927
928                 filter = talloc_asprintf(memctx, "(&(objectClass=%s)(%s=%lu))",
929                                 LDAP_OBJ_IDMAP_ENTRY,
930                                 (ids[0]->xid.type==ID_TYPE_UID)?uidNumber:gidNumber,
931                                 (unsigned long)ids[0]->xid.id);
932                 CHECK_ALLOC_DONE(filter);
933                 DEBUG(10, ("Filter: [%s]\n", filter));
934         } else {
935                 /* multiple mappings */
936                 multi = True;
937         }
938
939 again:
940         if (multi) {
941
942                 talloc_free(filter);
943                 filter = talloc_asprintf(memctx,
944                                          "(&(objectClass=%s)(|",
945                                          LDAP_OBJ_IDMAP_ENTRY);
946                 CHECK_ALLOC_DONE(filter);
947
948                 bidx = idx;
949                 for (i = 0; (i < IDMAP_LDAP_MAX_IDS) && ids[idx]; i++, idx++) {
950                         filter = talloc_asprintf_append_buffer(filter, "(%s=%lu)",
951                                         (ids[idx]->xid.type==ID_TYPE_UID)?uidNumber:gidNumber,
952                                         (unsigned long)ids[idx]->xid.id);
953                         CHECK_ALLOC_DONE(filter);
954                 }
955                 filter = talloc_asprintf_append_buffer(filter, "))");
956                 CHECK_ALLOC_DONE(filter);
957                 DEBUG(10, ("Filter: [%s]\n", filter));
958         } else {
959                 bidx = 0;
960                 idx = 1;
961         }
962
963         rc = smbldap_search(ctx->smbldap_state, ctx->suffix, LDAP_SCOPE_SUBTREE,
964                 filter, attr_list, 0, &result);
965
966         if (rc != LDAP_SUCCESS) {
967                 DEBUG(3,("Failure looking up ids (%s)\n", ldap_err2string(rc)));
968                 ret = NT_STATUS_UNSUCCESSFUL;
969                 goto done;
970         }
971
972         count = ldap_count_entries(ctx->smbldap_state->ldap_struct, result);
973
974         if (count == 0) {
975                 DEBUG(10, ("NO SIDs found\n"));
976         }
977
978         for (i = 0; i < count; i++) {
979                 LDAPMessage *entry = NULL;
980                 char *sidstr = NULL;
981                 char *tmp = NULL;
982                 enum id_type type;
983                 struct id_map *map;
984                 uint32_t id;
985
986                 if (i == 0) { /* first entry */
987                         entry = ldap_first_entry(ctx->smbldap_state->ldap_struct,
988                                                  result);
989                 } else { /* following ones */
990                         entry = ldap_next_entry(ctx->smbldap_state->ldap_struct,
991                                                 entry);
992                 }
993                 if ( ! entry) {
994                         DEBUG(2, ("ERROR: Unable to fetch ldap entries "
995                                   "from results\n"));
996                         break;
997                 }
998
999                 /* first check if the SID is present */
1000                 sidstr = smbldap_talloc_single_attribute(
1001                                 ctx->smbldap_state->ldap_struct,
1002                                 entry, LDAP_ATTRIBUTE_SID, memctx);
1003                 if ( ! sidstr) { /* no sid, skip entry */
1004                         DEBUG(2, ("WARNING SID not found on entry\n"));
1005                         continue;
1006                 }
1007
1008                 /* now try to see if it is a uid, if not try with a gid
1009                  * (gid is more common, but in case both uidNumber and
1010                  * gidNumber are returned the SID is mapped to the uid
1011                  *not the gid) */
1012                 type = ID_TYPE_UID;
1013                 tmp = smbldap_talloc_single_attribute(
1014                                 ctx->smbldap_state->ldap_struct,
1015                                 entry, uidNumber, memctx);
1016                 if ( ! tmp) {
1017                         type = ID_TYPE_GID;
1018                         tmp = smbldap_talloc_single_attribute(
1019                                         ctx->smbldap_state->ldap_struct,
1020                                         entry, gidNumber, memctx);
1021                 }
1022                 if ( ! tmp) { /* wow very strange entry, how did it match ? */
1023                         DEBUG(5, ("Unprobable match on (%s), no uidNumber, "
1024                                   "nor gidNumber returned\n", sidstr));
1025                         TALLOC_FREE(sidstr);
1026                         continue;
1027                 }
1028
1029                 id = strtoul(tmp, NULL, 10);
1030                 if ((id == 0) ||
1031                     (ctx->filter_low_id && (id < ctx->filter_low_id)) ||
1032                     (ctx->filter_high_id && (id > ctx->filter_high_id))) {
1033                         DEBUG(5, ("Requested id (%u) out of range (%u - %u). "
1034                                   "Filtered!\n", id,
1035                                   ctx->filter_low_id, ctx->filter_high_id));
1036                         TALLOC_FREE(sidstr);
1037                         TALLOC_FREE(tmp);
1038                         continue;
1039                 }
1040                 TALLOC_FREE(tmp);
1041
1042                 map = find_map_by_id(&ids[bidx], type, id);
1043                 if (!map) {
1044                         DEBUG(2, ("WARNING: couldn't match sid (%s) "
1045                                   "with requested ids\n", sidstr));
1046                         TALLOC_FREE(sidstr);
1047                         continue;
1048                 }
1049
1050                 if ( ! string_to_sid(map->sid, sidstr)) {
1051                         DEBUG(2, ("ERROR: Invalid SID on entry\n"));
1052                         TALLOC_FREE(sidstr);
1053                         continue;
1054                 }
1055                 TALLOC_FREE(sidstr);
1056
1057                 /* mapped */
1058                 map->status = ID_MAPPED;
1059
1060                 DEBUG(10, ("Mapped %s -> %lu (%d)\n", sid_string_dbg(map->sid),
1061                            (unsigned long)map->xid.id, map->xid.type));
1062         }
1063
1064         /* free the ldap results */
1065         if (result) {
1066                 ldap_msgfree(result);
1067                 result = NULL;
1068         }
1069
1070         if (multi && ids[idx]) { /* still some values to map */
1071                 goto again;
1072         }
1073
1074         ret = NT_STATUS_OK;
1075
1076         /* mark all unknwon/expired ones as unmapped */
1077         for (i = 0; ids[i]; i++) {
1078                 if (ids[i]->status != ID_MAPPED)
1079                         ids[i]->status = ID_UNMAPPED;
1080         }
1081
1082 done:
1083         talloc_free(memctx);
1084         return ret;
1085 }
1086
1087 /**********************************
1088  lookup a set of sids.
1089 **********************************/
1090
1091 /* this function searches up to IDMAP_LDAP_MAX_IDS entries
1092  * in maps for a match */
1093 static struct id_map *find_map_by_sid(struct id_map **maps, DOM_SID *sid)
1094 {
1095         int i;
1096
1097         for (i = 0; i < IDMAP_LDAP_MAX_IDS; i++) {
1098                 if (maps[i] == NULL) { /* end of the run */
1099                         return NULL;
1100                 }
1101                 if (sid_equal(maps[i]->sid, sid)) {
1102                         return maps[i];
1103                 }
1104         }
1105
1106         return NULL;
1107 }
1108
1109 static NTSTATUS idmap_ldap_sids_to_unixids(struct idmap_domain *dom,
1110                                            struct id_map **ids)
1111 {
1112         LDAPMessage *entry = NULL;
1113         NTSTATUS ret;
1114         TALLOC_CTX *memctx;
1115         struct idmap_ldap_context *ctx;
1116         LDAPMessage *result = NULL;
1117         const char *uidNumber;
1118         const char *gidNumber;
1119         const char **attr_list;
1120         char *filter = NULL;
1121         bool multi = False;
1122         int idx = 0;
1123         int bidx = 0;
1124         int count;
1125         int rc;
1126         int i;
1127
1128         /* Only do query if we are online */
1129         if (idmap_is_offline()) {
1130                 return NT_STATUS_FILE_IS_OFFLINE;
1131         }
1132
1133         ctx = talloc_get_type(dom->private_data, struct idmap_ldap_context);
1134
1135         memctx = talloc_new(ctx);
1136         if ( ! memctx) {
1137                 DEBUG(0, ("Out of memory!\n"));
1138                 return NT_STATUS_NO_MEMORY;
1139         }
1140
1141         uidNumber = get_attr_key2string(idpool_attr_list, LDAP_ATTR_UIDNUMBER);
1142         gidNumber = get_attr_key2string(idpool_attr_list, LDAP_ATTR_GIDNUMBER);
1143
1144         attr_list = get_attr_list(memctx, sidmap_attr_list);
1145
1146         if ( ! ids[1]) {
1147                 /* if we are requested just one mapping use the simple filter */
1148
1149                 filter = talloc_asprintf(memctx, "(&(objectClass=%s)(%s=%s))",
1150                                 LDAP_OBJ_IDMAP_ENTRY,
1151                                 LDAP_ATTRIBUTE_SID,
1152                                 sid_string_talloc(memctx, ids[0]->sid));
1153                 CHECK_ALLOC_DONE(filter);
1154                 DEBUG(10, ("Filter: [%s]\n", filter));
1155         } else {
1156                 /* multiple mappings */
1157                 multi = True;
1158         }
1159
1160 again:
1161         if (multi) {
1162
1163                 TALLOC_FREE(filter);
1164                 filter = talloc_asprintf(memctx,
1165                                          "(&(objectClass=%s)(|",
1166                                          LDAP_OBJ_IDMAP_ENTRY);
1167                 CHECK_ALLOC_DONE(filter);
1168
1169                 bidx = idx;
1170                 for (i = 0; (i < IDMAP_LDAP_MAX_IDS) && ids[idx]; i++, idx++) {
1171                         filter = talloc_asprintf_append_buffer(filter, "(%s=%s)",
1172                                         LDAP_ATTRIBUTE_SID,
1173                                         sid_string_talloc(memctx,
1174                                                           ids[idx]->sid));
1175                         CHECK_ALLOC_DONE(filter);
1176                 }
1177                 filter = talloc_asprintf_append_buffer(filter, "))");
1178                 CHECK_ALLOC_DONE(filter);
1179                 DEBUG(10, ("Filter: [%s]", filter));
1180         } else {
1181                 bidx = 0;
1182                 idx = 1;
1183         }
1184
1185         rc = smbldap_search(ctx->smbldap_state, ctx->suffix, LDAP_SCOPE_SUBTREE,
1186                 filter, attr_list, 0, &result);
1187
1188         if (rc != LDAP_SUCCESS) {
1189                 DEBUG(3,("Failure looking up sids (%s)\n",
1190                          ldap_err2string(rc)));
1191                 ret = NT_STATUS_UNSUCCESSFUL;
1192                 goto done;
1193         }
1194
1195         count = ldap_count_entries(ctx->smbldap_state->ldap_struct, result);
1196
1197         if (count == 0) {
1198                 DEBUG(10, ("NO SIDs found\n"));
1199         }
1200
1201         for (i = 0; i < count; i++) {
1202                 char *sidstr = NULL;
1203                 char *tmp = NULL;
1204                 enum id_type type;
1205                 struct id_map *map;
1206                 DOM_SID sid;
1207                 uint32_t id;
1208
1209                 if (i == 0) { /* first entry */
1210                         entry = ldap_first_entry(ctx->smbldap_state->ldap_struct,
1211                                                  result);
1212                 } else { /* following ones */
1213                         entry = ldap_next_entry(ctx->smbldap_state->ldap_struct,
1214                                                 entry);
1215                 }
1216                 if ( ! entry) {
1217                         DEBUG(2, ("ERROR: Unable to fetch ldap entries "
1218                                   "from results\n"));
1219                         break;
1220                 }
1221
1222                 /* first check if the SID is present */
1223                 sidstr = smbldap_talloc_single_attribute(
1224                                 ctx->smbldap_state->ldap_struct,
1225                                 entry, LDAP_ATTRIBUTE_SID, memctx);
1226                 if ( ! sidstr) { /* no sid ??, skip entry */
1227                         DEBUG(2, ("WARNING SID not found on entry\n"));
1228                         continue;
1229                 }
1230
1231                 if ( ! string_to_sid(&sid, sidstr)) {
1232                         DEBUG(2, ("ERROR: Invalid SID on entry\n"));
1233                         TALLOC_FREE(sidstr);
1234                         continue;
1235                 }
1236
1237                 map = find_map_by_sid(&ids[bidx], &sid);
1238                 if (!map) {
1239                         DEBUG(2, ("WARNING: couldn't find entry sid (%s) "
1240                                   "in ids", sidstr));
1241                         TALLOC_FREE(sidstr);
1242                         continue;
1243                 }
1244
1245                 TALLOC_FREE(sidstr);
1246
1247                 /* now try to see if it is a uid, if not try with a gid
1248                  * (gid is more common, but in case both uidNumber and
1249                  * gidNumber are returned the SID is mapped to the uid
1250                  * not the gid) */
1251                 type = ID_TYPE_UID;
1252                 tmp = smbldap_talloc_single_attribute(
1253                                 ctx->smbldap_state->ldap_struct,
1254                                 entry, uidNumber, memctx);
1255                 if ( ! tmp) {
1256                         type = ID_TYPE_GID;
1257                         tmp = smbldap_talloc_single_attribute(
1258                                         ctx->smbldap_state->ldap_struct,
1259                                         entry, gidNumber, memctx);
1260                 }
1261                 if ( ! tmp) { /* no ids ?? */
1262                         DEBUG(5, ("no uidNumber, "
1263                                   "nor gidNumber attributes found\n"));
1264                         continue;
1265                 }
1266
1267                 id = strtoul(tmp, NULL, 10);
1268                 if ((id == 0) ||
1269                     (ctx->filter_low_id && (id < ctx->filter_low_id)) ||
1270                     (ctx->filter_high_id && (id > ctx->filter_high_id))) {
1271                         DEBUG(5, ("Requested id (%u) out of range (%u - %u). "
1272                                   "Filtered!\n", id,
1273                                   ctx->filter_low_id, ctx->filter_high_id));
1274                         TALLOC_FREE(tmp);
1275                         continue;
1276                 }
1277                 TALLOC_FREE(tmp);
1278
1279                 /* mapped */
1280                 map->xid.type = type;
1281                 map->xid.id = id;
1282                 map->status = ID_MAPPED;
1283
1284                 DEBUG(10, ("Mapped %s -> %lu (%d)\n", sid_string_dbg(map->sid),
1285                            (unsigned long)map->xid.id, map->xid.type));
1286         }
1287
1288         /* free the ldap results */
1289         if (result) {
1290                 ldap_msgfree(result);
1291                 result = NULL;
1292         }
1293
1294         if (multi && ids[idx]) { /* still some values to map */
1295                 goto again;
1296         }
1297
1298         ret = NT_STATUS_OK;
1299
1300         /* mark all unknwon/expired ones as unmapped */
1301         for (i = 0; ids[i]; i++) {
1302                 if (ids[i]->status != ID_MAPPED)
1303                         ids[i]->status = ID_UNMAPPED;
1304         }
1305
1306 done:
1307         talloc_free(memctx);
1308         return ret;
1309 }
1310
1311 /**********************************
1312  set a mapping.
1313 **********************************/
1314
1315 /* TODO: change this:  This function cannot be called to modify a mapping,
1316  * only set a new one */
1317
1318 static NTSTATUS idmap_ldap_set_mapping(struct idmap_domain *dom,
1319                                        const struct id_map *map)
1320 {
1321         NTSTATUS ret;
1322         TALLOC_CTX *memctx;
1323         struct idmap_ldap_context *ctx;
1324         LDAPMessage *entry = NULL;
1325         LDAPMod **mods = NULL;
1326         const char *type;
1327         char *id_str;
1328         char *sid;
1329         char *dn;
1330         int rc = -1;
1331
1332         /* Only do query if we are online */
1333         if (idmap_is_offline()) {
1334                 return NT_STATUS_FILE_IS_OFFLINE;
1335         }
1336
1337         ctx = talloc_get_type(dom->private_data, struct idmap_ldap_context);
1338
1339         switch(map->xid.type) {
1340         case ID_TYPE_UID:
1341                 type = get_attr_key2string(sidmap_attr_list,
1342                                            LDAP_ATTR_UIDNUMBER);
1343                 break;
1344
1345         case ID_TYPE_GID:
1346                 type = get_attr_key2string(sidmap_attr_list,
1347                                            LDAP_ATTR_GIDNUMBER);
1348                 break;
1349
1350         default:
1351                 return NT_STATUS_INVALID_PARAMETER;
1352         }
1353
1354         memctx = talloc_new(ctx);
1355         if ( ! memctx) {
1356                 DEBUG(0, ("Out of memory!\n"));
1357                 return NT_STATUS_NO_MEMORY;
1358         }
1359
1360         id_str = talloc_asprintf(memctx, "%lu", (unsigned long)map->xid.id);
1361         CHECK_ALLOC_DONE(id_str);
1362
1363         sid = talloc_strdup(memctx, sid_string_talloc(memctx, map->sid));
1364         CHECK_ALLOC_DONE(sid);
1365
1366         dn = talloc_asprintf(memctx, "%s=%s,%s",
1367                         get_attr_key2string(sidmap_attr_list, LDAP_ATTR_SID),
1368                         sid,
1369                         ctx->suffix);
1370         CHECK_ALLOC_DONE(dn);
1371
1372         smbldap_set_mod(&mods, LDAP_MOD_ADD,
1373                         "objectClass", LDAP_OBJ_IDMAP_ENTRY);
1374
1375         smbldap_make_mod(ctx->smbldap_state->ldap_struct,
1376                          entry, &mods, type, id_str);
1377
1378         smbldap_make_mod(ctx->smbldap_state->ldap_struct, entry, &mods,
1379                          get_attr_key2string(sidmap_attr_list, LDAP_ATTR_SID),
1380                          sid);
1381
1382         if ( ! mods) {
1383                 DEBUG(2, ("ERROR: No mods?\n"));
1384                 ret = NT_STATUS_UNSUCCESSFUL;
1385                 goto done;
1386         }
1387
1388         /* TODO: remove conflicting mappings! */
1389
1390         smbldap_set_mod(&mods, LDAP_MOD_ADD, "objectClass", LDAP_OBJ_SID_ENTRY);
1391
1392         DEBUG(10, ("Set DN %s (%s -> %s)\n", dn, sid, id_str));
1393
1394         rc = smbldap_add(ctx->smbldap_state, dn, mods);
1395         ldap_mods_free(mods, True);
1396
1397         if (rc != LDAP_SUCCESS) {
1398                 char *ld_error = NULL;
1399                 ldap_get_option(ctx->smbldap_state->ldap_struct,
1400                                 LDAP_OPT_ERROR_STRING, &ld_error);
1401                 DEBUG(0,("ldap_set_mapping_internals: Failed to add %s to %lu "
1402                          "mapping [%s]\n", sid,
1403                          (unsigned long)map->xid.id, type));
1404                 DEBUG(0, ("ldap_set_mapping_internals: Error was: %s (%s)\n",
1405                         ld_error ? ld_error : "(NULL)", ldap_err2string (rc)));
1406                 if (ld_error) {
1407                         ldap_memfree(ld_error);
1408                 }
1409                 ret = NT_STATUS_UNSUCCESSFUL;
1410                 goto done;
1411         }
1412
1413         DEBUG(10,("ldap_set_mapping: Successfully created mapping from %s to "
1414                   "%lu [%s]\n", sid, (unsigned long)map->xid.id, type));
1415
1416         ret = NT_STATUS_OK;
1417
1418 done:
1419         talloc_free(memctx);
1420         return ret;
1421 }
1422
1423 /**********************************
1424  Close the idmap ldap instance
1425 **********************************/
1426
1427 static NTSTATUS idmap_ldap_close(struct idmap_domain *dom)
1428 {
1429         struct idmap_ldap_context *ctx;
1430
1431         if (dom->private_data) {
1432                 ctx = talloc_get_type(dom->private_data,
1433                                       struct idmap_ldap_context);
1434
1435                 talloc_free(ctx);
1436                 dom->private_data = NULL;
1437         }
1438
1439         return NT_STATUS_OK;
1440 }
1441
1442 static struct idmap_methods idmap_ldap_methods = {
1443
1444         .init = idmap_ldap_db_init,
1445         .unixids_to_sids = idmap_ldap_unixids_to_sids,
1446         .sids_to_unixids = idmap_ldap_sids_to_unixids,
1447         .set_mapping = idmap_ldap_set_mapping,
1448         .close_fn = idmap_ldap_close
1449 };
1450
1451 static struct idmap_alloc_methods idmap_ldap_alloc_methods = {
1452
1453         .init = idmap_ldap_alloc_init,
1454         .allocate_id = idmap_ldap_allocate_id,
1455         .get_id_hwm = idmap_ldap_get_hwm,
1456         .set_id_hwm = idmap_ldap_set_hwm,
1457         .close_fn = idmap_ldap_alloc_close,
1458         /* .dump_data = TODO */
1459 };
1460
1461 NTSTATUS idmap_alloc_ldap_init(void)
1462 {
1463         return smb_register_idmap_alloc(SMB_IDMAP_INTERFACE_VERSION, "ldap",
1464                                         &idmap_ldap_alloc_methods);
1465 }
1466
1467 NTSTATUS idmap_ldap_init(void)
1468 {
1469         NTSTATUS ret;
1470
1471         /* FIXME: bad hack to actually register also the alloc_ldap module
1472          * without changining configure.in */
1473         ret = idmap_alloc_ldap_init();
1474         if (! NT_STATUS_IS_OK(ret)) {
1475                 return ret;
1476         }
1477         return smb_register_idmap(SMB_IDMAP_INTERFACE_VERSION, "ldap",
1478                                   &idmap_ldap_methods);
1479 }
1480