r9698: Fix a bit of memory management
authorJelmer Vernooij <jelmer@samba.org>
Sat, 27 Aug 2005 23:47:17 +0000 (23:47 +0000)
committerGerald (Jerry) Carter <jerry@samba.org>
Wed, 10 Oct 2007 18:34:42 +0000 (13:34 -0500)
More minor bugfixes
Support mapping objectclasses and do mapping on 'dn' field as well (not just msg->dn)

source/include/smb.h
source/lib/ldb/ldb_map/ldb_map.c
source/lib/ldb/ldb_map/ldb_map.h
source/lib/ldb/samba/samba3sam.c
source/lib/ldb/tests/test-samba3sam.sh
source/scripting/ejs/mprutil.c

index 864ad1ba023ca0f17e4bfcd038ac1d225b03947c..3c03c7d7f197c171dfbe15dace3c45d181a46a91 100644 (file)
@@ -187,14 +187,14 @@ enum smb_signing_state {SMB_SIGNING_OFF, SMB_SIGNING_SUPPORTED,
 #define NTCREATEX_ACTION_UNKNOWN 5
 
 /* used to hold an arbitrary blob of data */
-typedef struct data_blob {
+typedef struct datablob {
        uint8_t *data;
        size_t length;
 } DATA_BLOB;
 
 /* by making struct ldb_val and DATA_BLOB the same, we can simplify
    a fair bit of code */
-#define ldb_val data_blob
+#define ldb_val datablob
 
 /* 64 bit time (100 nanosec) 1601 - cifs6.txt, section 3.5, page 30, 4 byte aligned */
 typedef uint64_t NTTIME;
index d2bbea7c7ade5c68d655b93ddbc391b214615763..5ae4a435d86a29f4b6dcf7590d2e6a3941471ba9 100644 (file)
  *  - objectclass hint in ldb_map_attribute 
  *     for use when multiple remote attributes (independant of each other)
  *     map to one local attribute. E.g.: (uid, gidNumber) -> unixName
+ *     (use MAP_GENERATE instead ?) 
  */
 
+static const struct ldb_map_attribute builtin_attribute_maps[];
+
 struct map_private {
-       const struct ldb_map_attribute *attribute_maps;
-       const struct ldb_map_objectclass *objectclass_maps;
+       struct ldb_map_context context;
        const char *last_err_string;
 };
 
+
 /* find an attribute by the local name */
-static const struct ldb_map_attribute *map_find_attr_local(struct ldb_module *module, const char *attr)
+static const struct ldb_map_attribute *map_find_attr_local(struct ldb_map_context *privdat, const char *attr)
 {
-       struct map_private *privdat = module->private_data;
        int i;
+
        for (i = 0; privdat->attribute_maps[i].local_name; i++) {
-               if (!strcmp(privdat->attribute_maps[i].local_name, attr)) 
+               if (!ldb_attr_cmp(privdat->attribute_maps[i].local_name, attr)) 
                        return &privdat->attribute_maps[i];
        }
 
@@ -53,17 +56,26 @@ static const struct ldb_map_attribute *map_find_attr_local(struct ldb_module *mo
 }
 
 /* find an attribute by the remote name */
-static const struct ldb_map_attribute *map_find_attr_remote(struct ldb_module *module, const char *attr)
+static const struct ldb_map_attribute *map_find_attr_remote(struct ldb_map_context *privdat, const char *attr)
 {
-       struct map_private *privdat = module->private_data;
        int i;
+
        for (i = 0; privdat->attribute_maps[i].local_name; i++) {
-               if (privdat->attribute_maps[i].type != MAP_RENAME &&
-                       privdat->attribute_maps[i].type != MAP_CONVERT) 
+               if (privdat->attribute_maps[i].type == MAP_IGNORE)
+                       continue;
+
+               if (privdat->attribute_maps[i].type == MAP_GENERATE)
                        continue;
 
-               if (!strcmp(privdat->attribute_maps[i].u.rename.remote_name, attr)) 
+               if (privdat->attribute_maps[i].type == MAP_KEEP &&
+                       ldb_attr_cmp(privdat->attribute_maps[i].local_name, attr) == 0)
                        return &privdat->attribute_maps[i];
+
+               if ((privdat->attribute_maps[i].type == MAP_RENAME ||
+                       privdat->attribute_maps[i].type == MAP_CONVERT) &&
+                       ldb_attr_cmp(privdat->attribute_maps[i].u.rename.remote_name, attr) == 0) 
+                       return &privdat->attribute_maps[i];
+
        }
 
        return NULL;
@@ -74,11 +86,13 @@ static struct ldb_parse_tree *ldb_map_parse_tree(struct ldb_module *module, TALL
        int i;
        const struct ldb_map_attribute *attr;
        struct ldb_parse_tree *new_tree;
+       enum ldb_map_attr_type map_type;
+       struct ldb_val value, newvalue;
+       struct ldb_map_context *privdat = &((struct map_private *)module->private_data)->context;
 
        if (tree == NULL)
                return NULL;
        
-       new_tree = talloc_memdup(ctx, tree, sizeof(*tree));
 
        /* Find attr in question and:
         *  - if it has a convert_operator function, run that
@@ -86,6 +100,8 @@ static struct ldb_parse_tree *ldb_map_parse_tree(struct ldb_module *module, TALL
 
        if (tree->operation == LDB_OP_AND || 
                tree->operation == LDB_OP_OR) {
+               
+               new_tree = talloc_memdup(ctx, tree, sizeof(*tree));
                new_tree->u.list.elements = talloc_array(new_tree, struct ldb_parse_tree *, tree->u.list.num_elements);
                for (i = 0; i < new_tree->u.list.num_elements; i++) {
                        new_tree->u.list.elements[i] = ldb_map_parse_tree(module, new_tree, tree->u.list.elements[i]);
@@ -95,6 +111,7 @@ static struct ldb_parse_tree *ldb_map_parse_tree(struct ldb_module *module, TALL
        }
                
        if (tree->operation == LDB_OP_NOT) {
+               new_tree = talloc_memdup(ctx, tree, sizeof(*tree));
                new_tree->u.isnot.child = ldb_map_parse_tree(module, new_tree, tree->u.isnot.child);
                return new_tree;
        }
@@ -105,47 +122,74 @@ static struct ldb_parse_tree *ldb_map_parse_tree(struct ldb_module *module, TALL
         * (all have attr as the first element)
         */
 
-       if (new_tree->operation == LDB_OP_EQUALITY) {
-               new_tree->u.equality.value = ldb_val_dup(new_tree, &tree->u.equality.value);
-       } else if (new_tree->operation == LDB_OP_SUBSTRING) {
-               new_tree->u.substring.chunks = NULL; /* FIXME! */
-       } else if (new_tree->operation == LDB_OP_LESS || 
-                          new_tree->operation == LDB_OP_GREATER ||
-                          new_tree->operation == LDB_OP_APPROX) {
-               new_tree->u.comparison.value = ldb_val_dup(new_tree, &tree->u.comparison.value);
-       } else if (new_tree->operation == LDB_OP_EXTENDED) {
-               new_tree->u.extended.value = ldb_val_dup(new_tree, &tree->u.extended.value);
-               new_tree->u.extended.rule_id = talloc_strdup(new_tree, tree->u.extended.rule_id);
-       }
-
-       attr = map_find_attr_local(module, tree->u.equality.attr);
+       attr = map_find_attr_local(privdat, tree->u.equality.attr);
 
-       if (!attr || attr->type == MAP_KEEP) {
+       if (!attr) {
                DEBUG(0, ("Unable to find local attribute '%s', leaving as is\n", tree->u.equality.attr));
-               new_tree->u.equality.attr = talloc_strdup(new_tree, tree->u.equality.attr);
-               return new_tree;
+               map_type = MAP_KEEP;
+       } else {
+               map_type = attr->type;
        }
 
-       if (attr->type == MAP_IGNORE) {
-               talloc_free(new_tree);
+       if (attr && attr->convert_operator) {
+               /* Run convert_operator */
+               return attr->convert_operator(privdat, module, tree);
+       }
+
+       if (map_type == MAP_IGNORE)
+               return NULL;
+
+       if (map_type == MAP_GENERATE) {
+               DEBUG(0, ("Can't do conversion for MAP_GENERATE in map_parse_tree without convert_operator for '%s'\n", tree->u.equality.attr));
                return NULL;
        }
 
-       if (attr->convert_operator) {
-               /* Run convert_operator */
-               talloc_free(new_tree);
-               new_tree = attr->convert_operator(module, tree);
+       if (tree->operation == LDB_OP_EQUALITY) {
+               value = tree->u.equality.value;
+       } else if (tree->operation == LDB_OP_LESS || tree->operation == LDB_OP_GREATER ||
+                          tree->operation == LDB_OP_APPROX) {
+               value = tree->u.comparison.value;
+       } else if (tree->operation == LDB_OP_EXTENDED) {
+               value = tree->u.extended.value;
+       }
+       
+       new_tree = talloc_memdup(ctx, tree, sizeof(*tree));
 
+       if (map_type == MAP_KEEP) {
+               new_tree->u.equality.attr = talloc_strdup(new_tree, tree->u.equality.attr);
+       } else { /* MAP_RENAME / MAP_CONVERT */
+               new_tree->u.equality.attr = talloc_strdup(new_tree, attr->u.rename.remote_name);
+       }
+
+       if (new_tree->operation == LDB_OP_PRESENT) 
+               return new_tree;
+               
+       if (new_tree->operation == LDB_OP_SUBSTRING) {
+               new_tree->u.substring.chunks = NULL; /* FIXME! */
                return new_tree;
+       }
+
+       if (map_type == MAP_CONVERT) {
+               newvalue = attr->u.convert.convert_local(privdat, new_tree, &value);
        } else {
-               new_tree->u.equality.attr = talloc_strdup(new_tree, attr->u.rename.remote_name);
+               newvalue = ldb_val_dup(new_tree, &value);
+       }
+
+       if (new_tree->operation == LDB_OP_EQUALITY) {
+               new_tree->u.equality.value = newvalue;
+       } else if (new_tree->operation == LDB_OP_LESS || new_tree->operation == LDB_OP_GREATER ||
+                          new_tree->operation == LDB_OP_APPROX) {
+               new_tree->u.comparison.value = newvalue;
+       } else if (new_tree->operation == LDB_OP_EXTENDED) {
+               new_tree->u.extended.value = newvalue;
+               new_tree->u.extended.rule_id = talloc_strdup(new_tree, tree->u.extended.rule_id);
        }
        
        return new_tree;
 }
 
 /* Remote DN -> Local DN */
-static struct ldb_dn *map_remote_dn(struct ldb_module *module, const struct ldb_dn *dn)
+static struct ldb_dn *map_remote_dn(struct ldb_map_context *privdat, TALLOC_CTX *ctx, const struct ldb_dn *dn)
 {
        struct ldb_dn *newdn;
        int i;
@@ -153,7 +197,7 @@ static struct ldb_dn *map_remote_dn(struct ldb_module *module, const struct ldb_
        if (dn == NULL)
                return NULL;
 
-       newdn = talloc_memdup(module, dn, sizeof(*dn));
+       newdn = talloc_memdup(ctx, dn, sizeof(*dn));
        if (!newdn) 
                return NULL;
 
@@ -166,49 +210,49 @@ static struct ldb_dn *map_remote_dn(struct ldb_module *module, const struct ldb_
         * complete rdn */
        
        for (i = 0; i < dn->comp_num; i++) {
-               const struct ldb_map_attribute *attr = map_find_attr_remote(module, dn->components[i].name);
+               const struct ldb_map_attribute *attr = map_find_attr_remote(privdat, dn->components[i].name);
+               enum ldb_map_attr_type map_type;
 
                /* Unknown attribute - leave this dn as is and hope the best... */
-               if (!attr || attr->type == MAP_KEEP) {
-                       newdn->components[i].name = talloc_strdup(newdn->components, dn->components[i].name);
-                       newdn->components[i].value = ldb_val_dup(newdn->components, &dn->components[i].value);
-               } else if (attr->type == MAP_IGNORE || attr->type == MAP_GENERATE) {
+               if (!attr) map_type = MAP_KEEP;
+               else map_type = attr->type;
+                       
+               switch (map_type) { 
+               case MAP_IGNORE:
+               case MAP_GENERATE:
                        DEBUG(0, ("Local MAP_IGNORE or MAP_GENERATE attribute '%s' used in DN!", dn->components[i].name));
                        talloc_free(newdn);
                        return NULL;
-               } else if (attr->type == MAP_CONVERT) {
-                       struct ldb_message_element elm, *newelm;
-                       struct ldb_val vals[1] = { dn->components[i].value };
-                       
-                       elm.flags = 0;
-                       elm.name = attr->u.convert.remote_name;
-                       elm.num_values = 1;
-                       elm.values = vals;
-
-                       newelm = attr->u.convert.convert_remote(module, attr->local_name, &elm);
-
-                       newdn->components[i].name = talloc_strdup(newdn->components, newelm->name);
-                       newdn->components[i].value = ldb_val_dup(newdn->components, &newelm->values[0]);
 
-                       talloc_free(newelm);
-               } else if (attr->type == MAP_RENAME) {
+               case MAP_KEEP:
+                       newdn->components[i].name = talloc_strdup(newdn->components, dn->components[i].name);
+                       newdn->components[i].value = ldb_val_dup(newdn->components, &dn->components[i].value);
+                       break;
+                       
+               case MAP_CONVERT:
+                       newdn->components[i].name = talloc_strdup(newdn->components, attr->local_name);
+                       newdn->components[i].value = attr->u.convert.convert_remote(privdat, ctx, &dn->components[i].value);
+                       break;
+                       
+               case MAP_RENAME:
                        newdn->components[i].name = talloc_strdup(newdn->components, attr->local_name);
                        newdn->components[i].value = ldb_val_dup(newdn->components, &dn->components[i].value);
+                       break;
                }
        }
        return newdn;
 }
 
 /* Local DN -> Remote DN */
-static struct ldb_dn *map_local_dn(struct ldb_module *module, const struct ldb_dn *dn)
-{      struct ldb_dn *newdn;
+static struct ldb_dn *map_local_dn(struct ldb_map_context *privdat, TALLOC_CTX *ctx, const struct ldb_dn *dn)
+{      
+       struct ldb_dn *newdn;
        int i;
-       struct ldb_parse_tree eqtree, *new_eqtree;
 
        if (dn == NULL)
                return NULL;
 
-       newdn = talloc_memdup(module, dn, sizeof(*dn));
+       newdn = talloc_memdup(ctx, dn, sizeof(*dn));
        if (!newdn) 
                return NULL;
 
@@ -221,55 +265,34 @@ static struct ldb_dn *map_local_dn(struct ldb_module *module, const struct ldb_d
         * complete rdn using an equality convert_operator call */
        
        for (i = 0; i < dn->comp_num; i++) {
-               const struct ldb_map_attribute *attr = map_find_attr_local(module, dn->components[i].name);
+               const struct ldb_map_attribute *attr = map_find_attr_local(privdat, dn->components[i].name);
+               enum ldb_map_attr_type map_type;
 
                /* Unknown attribute - leave this dn as is and hope the best... */
-               if (!attr || attr->type == MAP_KEEP) {
-                       newdn->components[i].name = talloc_strdup(newdn->components, dn->components[i].name);
-                       newdn->components[i].value = ldb_val_dup(newdn->components, &dn->components[i].value);
-                       continue;
-               } else if (attr->type == MAP_IGNORE || attr->type == MAP_GENERATE) {
+               if (!attr) map_type = MAP_KEEP; else map_type = attr->type;
+               
+               switch (map_type) 
+               {
+                       case MAP_IGNORE: 
+                       case MAP_GENERATE:
                        DEBUG(0, ("Local MAP_IGNORE/MAP_GENERATE attribute '%s' used in DN!", dn->components[i].name));
                        talloc_free(newdn);
                        return NULL;
-               }
-
-               /* Simple rename/convert only */
-               if (attr->convert_operator) {
-                       /* Fancy stuff */
-                       eqtree.operation = LDB_OP_EQUALITY;
-                       eqtree.u.equality.attr = dn->components[i].name;
-                       eqtree.u.equality.value = dn->components[i].value;
 
-                       new_eqtree = ldb_map_parse_tree(module, newdn, &eqtree);
-
-                       /* Silently continue for now */
-                       if (!new_eqtree) {
-                               DEBUG(0, ("Unable to convert RDN for attribute %s\n", dn->components[i].name));
-                               continue;
-                       }
-
-                       newdn->components[i].name = talloc_strdup(newdn->components, new_eqtree->u.equality.attr);
-                       newdn->components[i].value = ldb_val_dup(newdn->components, &new_eqtree->u.equality.value);
-
-                       talloc_free(new_eqtree);
-               } else if (attr->type == MAP_CONVERT) {
-                       struct ldb_message_element elm, *newelm;
-                       struct ldb_val vals[1] = { dn->components[i].value };
-                       elm.flags = 0;
-                       elm.name = attr->local_name;
-                       elm.num_values = 1;
-                       elm.values = vals;
-
-                       newelm = attr->u.convert.convert_local(module, attr->u.convert.remote_name, &elm);
-
-                       newdn->components[i].name = talloc_strdup(newdn->components, newelm->name);
-                       newdn->components[i].value = ldb_val_dup(newdn->components, &newelm->values[0]);
+                       case MAP_CONVERT: 
+                               newdn->components[i].name = talloc_strdup(newdn->components, attr->u.convert.remote_name);
+                               newdn->components[i].value = attr->u.convert.convert_local(privdat, newdn->components, &dn->components[i].value);
+                       break;
+                       
+                       case MAP_RENAME:
+                               newdn->components[i].name = talloc_strdup(newdn->components, attr->u.rename.remote_name);
+                               newdn->components[i].value = ldb_val_dup(newdn->components, &dn->components[i].value);
+                       break;
 
-                       talloc_free(newelm);
-               } else if (attr->type == MAP_RENAME) {
-                       newdn->components[i].name = talloc_strdup(newdn->components, attr->u.rename.remote_name);
-                       newdn->components[i].value = ldb_val_dup(newdn->components, &dn->components[i].value);
+                       case MAP_KEEP:
+                               newdn->components[i].name = talloc_strdup(newdn->components, dn->components[i].name);
+                               newdn->components[i].value = ldb_val_dup(newdn->components, &dn->components[i].value);
+                       continue;
                }
        }
 
@@ -282,6 +305,7 @@ static const char **ldb_map_attrs(struct ldb_module *module, const char *const a
        int i;
        const char **ret;
        int ar_size = 0, last_element = 0;
+       struct ldb_map_context *privdat = &((struct map_private *)module->private_data)->context;
 
        if (attrs == NULL) 
                return NULL;
@@ -294,14 +318,15 @@ static const char **ldb_map_attrs(struct ldb_module *module, const char *const a
 
        for (i = 0; attrs[i]; i++) {
                int j;
-               const struct ldb_map_attribute *attr = map_find_attr_local(module, attrs[i]);
+               const struct ldb_map_attribute *attr = map_find_attr_local(privdat, attrs[i]);
+               enum ldb_map_attr_type map_type;
 
                if (!attr) {
                        DEBUG(0, ("Local attribute '%s' does not have a definition!\n", attrs[i]));
-                       continue;
-               }
+                       map_type = MAP_IGNORE;
+               } else map_type = attr->type;
 
-               switch (attr->type)
+               switch (map_type)
                { 
                        case MAP_IGNORE: break;
                        case MAP_KEEP: 
@@ -337,22 +362,29 @@ static const char **ldb_map_attrs(struct ldb_module *module, const char *const a
                                break;
                } 
        }
+       
+       if (last_element >= ar_size) {
+               ret = talloc_realloc(module, ret, const char *, ar_size+1);
+               ar_size++;
+       }
 
-       return NULL;
+       ret[last_element] = NULL;
+
+       return ret;
 }
 
 static const char **available_local_attributes(struct ldb_module *module, const struct ldb_message *msg)
 {
-       struct map_private *map = module->private_data;
+       struct ldb_map_context *privdat = &((struct map_private *)module->private_data)->context;
        int i, j;
        int count = 0;
        const char **ret = talloc_array(module, const char *, 1);
 
        ret[0] = NULL;
 
-       for (i = 0; map->attribute_maps[i].local_name; i++) {
+       for (i = 0; privdat->attribute_maps[i].local_name; i++) {
                BOOL avail = False;
-               const struct ldb_map_attribute *attr = &map->attribute_maps[i];
+               const struct ldb_map_attribute *attr = &privdat->attribute_maps[i];
 
                /* If all remote attributes for this attribute are present, add the 
                 * local one to the list */
@@ -391,12 +423,13 @@ static const char **available_local_attributes(struct ldb_module *module, const
 /* Used for search */
 static struct ldb_message *ldb_map_message_incoming(struct ldb_module *module, const char * const*attrs, const struct ldb_message *mi)
 {
-       int i;
+       int i, j;
        struct ldb_message *msg = talloc_zero(module, struct ldb_message);
        struct ldb_message_element *elm, *oldelm;
+       struct ldb_map_context *privdat = &((struct map_private *)module->private_data)->context;
        const char **newattrs = NULL;
 
-       msg->dn = map_remote_dn(module, mi->dn);
+       msg->dn = map_remote_dn(privdat, module, mi->dn);
 
        /* Loop over attrs, find in ldb_map_attribute array and 
         * run generate() */
@@ -409,14 +442,15 @@ static struct ldb_message *ldb_map_message_incoming(struct ldb_module *module, c
        }
 
        for (i = 0; attrs[i]; i++) {
-               const struct ldb_map_attribute *attr = map_find_attr_local(module, attrs[i]);
+               const struct ldb_map_attribute *attr = map_find_attr_local(privdat, attrs[i]);
+               enum ldb_map_attr_type map_type;
 
                if (!attr) {
                        DEBUG(0, ("Unable to find local attribute '%s' when generating incoming message\n", attrs[i]));
-                       continue;
-               }
+                       map_type = MAP_IGNORE;
+               } else map_type = attr->type;
 
-               switch (attr->type) {
+               switch (map_type) {
                        case MAP_IGNORE:break;
                        case MAP_RENAME:
                                oldelm = ldb_msg_find_element(mi, attr->u.rename.remote_name);
@@ -434,10 +468,15 @@ static struct ldb_message *ldb_map_message_incoming(struct ldb_module *module, c
                                oldelm = ldb_msg_find_element(mi, attr->u.rename.remote_name);
                                if (!oldelm) continue;
 
-                               elm = attr->u.convert.convert_remote(msg, attr->local_name, oldelm);
-                               if (!elm) continue;
+                               elm = talloc(msg, struct ldb_message_element);
+                               elm->name = talloc_strdup(elm, attr->local_name);
+                               elm->num_values = oldelm->num_values;
+                               elm->values = talloc_array(elm, struct ldb_val, elm->num_values);
 
-                               ldb_msg_add(module->ldb, msg, elm, elm->flags);
+                               for (j = 0; j < oldelm->num_values; j++)
+                                       elm->values[j] = attr->u.convert.convert_remote(privdat, elm, &oldelm->values[j]);
+
+                               ldb_msg_add(module->ldb, msg, elm, oldelm->flags);
                                break;
 
                        case MAP_KEEP:
@@ -454,7 +493,7 @@ static struct ldb_message *ldb_map_message_incoming(struct ldb_module *module, c
                                break;
 
                        case MAP_GENERATE:
-                               elm = attr->u.generate.generate_local(msg, attr->local_name, mi);
+                               elm = attr->u.generate.generate_local(privdat, msg, attr->local_name, mi);
                                if (!elm) continue;
 
                                ldb_msg_add(module->ldb, msg, elm, elm->flags);
@@ -473,24 +512,27 @@ static struct ldb_message *ldb_map_message_incoming(struct ldb_module *module, c
 /* Used for add, modify */
 static struct ldb_message *ldb_map_message_outgoing(struct ldb_module *module, const struct ldb_message *mo)
 {
+       struct ldb_map_context *privdat = &((struct map_private *)module->private_data)->context;
        struct ldb_message *msg = talloc_zero(module, struct ldb_message);
        struct ldb_message_element *elm;
-       int i;
+       int i,j;
        
        msg->private_data = mo->private_data;
        
-       msg->dn = map_local_dn(module, mo->dn);
+       msg->dn = map_local_dn(privdat, module, mo->dn);
 
        /* Loop over mi and call generate_remote for each attribute */
        for (i = 0; i < mo->num_elements; i++) {
-               const struct ldb_map_attribute *attr = map_find_attr_local(module, mo->elements[i].name);
+               const struct ldb_map_attribute *attr = map_find_attr_local(privdat, mo->elements[i].name);
+               enum ldb_map_attr_type map_type;
 
                if (!attr) {
                        DEBUG(0, ("Undefined local attribute '%s', ignoring\n", mo->elements[i].name));
+                       map_type = MAP_IGNORE;
                        continue;
-               }
+               } else map_type = attr->type;
 
-               switch (attr->type) {
+               switch (map_type) {
                case MAP_IGNORE: break;
                case MAP_RENAME:
                        elm = talloc(msg, struct ldb_message_element);
@@ -503,8 +545,17 @@ static struct ldb_message *ldb_map_message_outgoing(struct ldb_module *module, c
                        break;
 
                case MAP_CONVERT:
-                       elm = attr->u.convert.convert_local(msg, attr->u.convert.remote_name, &mo->elements[i]);
-                       ldb_msg_add(module->ldb, msg, elm, elm->flags);
+                       elm = talloc(msg, struct ldb_message_element);
+
+                       elm->name = talloc_strdup(elm, attr->u.rename.remote_name);
+                       elm->num_values = mo->elements[i].num_values;
+                       elm->values = talloc_array(elm, struct ldb_val, elm->num_values);
+                       
+                       for (j = 0; j < elm->num_values; j++) {
+                               elm->values[j] = attr->u.convert.convert_local(privdat, msg, &mo->elements[i].values[j]);
+                       }
+
+                       ldb_msg_add(module->ldb, msg, elm, mo->elements[i].flags);
                        break;
 
                case MAP_KEEP:
@@ -518,7 +569,7 @@ static struct ldb_message *ldb_map_message_outgoing(struct ldb_module *module, c
                        break;
 
                case MAP_GENERATE:
-                       attr->u.generate.generate_remote(attr->local_name, mo, msg);
+                       attr->u.generate.generate_remote(privdat, attr->local_name, mo, msg);
                        break;
                } 
        }
@@ -531,11 +582,12 @@ static struct ldb_message *ldb_map_message_outgoing(struct ldb_module *module, c
 */
 static int map_rename(struct ldb_module *module, const struct ldb_dn *olddn, const struct ldb_dn *newdn)
 {
+       struct ldb_map_context *privdat = &((struct map_private *)module->private_data)->context;
        struct ldb_dn *n_olddn, *n_newdn;
        int ret;
        
-       n_olddn = map_local_dn(module, olddn);
-       n_newdn = map_local_dn(module, newdn);
+       n_olddn = map_local_dn(privdat, module, olddn);
+       n_newdn = map_local_dn(privdat, module, newdn);
 
        ret = ldb_next_rename_record(module, n_olddn, n_newdn);
 
@@ -550,10 +602,11 @@ static int map_rename(struct ldb_module *module, const struct ldb_dn *olddn, con
 */
 static int map_delete(struct ldb_module *module, const struct ldb_dn *dn)
 {
+       struct ldb_map_context *privdat = &((struct map_private *)module->private_data)->context;
        struct ldb_dn *newdn;
        int ret;
 
-       newdn = map_local_dn(module, dn);
+       newdn = map_local_dn(privdat, module, dn);
 
        ret = ldb_next_delete_record(module, newdn);
 
@@ -574,11 +627,12 @@ static int map_search_bytree(struct ldb_module *module, const struct ldb_dn *bas
        struct ldb_parse_tree *new_tree;
        struct ldb_dn *new_base;
        struct ldb_message **newres;
+       struct ldb_map_context *privdat = &((struct map_private *)module->private_data)->context;
        int i;
 
        new_tree = ldb_map_parse_tree(module, module, tree);
        newattrs = ldb_map_attrs(module, attrs); 
-       new_base = map_local_dn(module, base);
+       new_base = map_local_dn(privdat, module, base);
 
        ret = ldb_next_search_bytree(module, new_base, scope, new_tree, newattrs, &newres);
 
@@ -633,8 +687,6 @@ static int map_add(struct ldb_module *module, const struct ldb_message *msg)
 }
 
 
-
-
 /*
   modify a record
 */
@@ -689,6 +741,7 @@ static const struct ldb_module_ops map_ops = {
 /* the init function */
 struct ldb_module *ldb_map_init(struct ldb_context *ldb, const struct ldb_map_attribute *attrs, const struct ldb_map_objectclass *ocls, const char *options[])
 {
+       int i, j;
        struct ldb_module *ctx;
        struct map_private *data;
 
@@ -704,8 +757,26 @@ struct ldb_module *ldb_map_init(struct ldb_context *ldb, const struct ldb_map_at
 
        data->last_err_string = NULL;
 
-       data->attribute_maps = attrs;
-       data->objectclass_maps = ocls;
+       /* Get list of attribute maps */
+       j = 0;
+       data->context.attribute_maps = NULL;
+
+       for (i = 0; attrs[i].local_name; i++) {
+               data->context.attribute_maps = talloc_realloc(data, data->context.attribute_maps, struct ldb_map_attribute, j+1);
+               data->context.attribute_maps[j] = attrs[i];
+               j++;
+       }
+
+       for (i = 0; builtin_attribute_maps[i].local_name; i++) {
+               data->context.attribute_maps = talloc_realloc(data, data->context.attribute_maps, struct ldb_map_attribute, j+1);
+               data->context.attribute_maps[j] = builtin_attribute_maps[i];
+               j++;
+       }
+
+       data->context.attribute_maps = talloc_realloc(data, data->context.attribute_maps, struct ldb_map_attribute, j+1);
+       ZERO_STRUCT(data->context.attribute_maps[j].local_name);
+
+       data->context.objectclass_maps = ocls;
        ctx->private_data = data;
        ctx->ldb = ldb;
        ctx->prev = ctx->next = NULL;
@@ -713,3 +784,100 @@ struct ldb_module *ldb_map_init(struct ldb_context *ldb, const struct ldb_map_at
 
        return ctx;
 }
+
+static struct ldb_val map_convert_local_dn(struct ldb_map_context *map, TALLOC_CTX *ctx, const struct ldb_val *val)
+{
+       struct ldb_dn *dn, *newdn;;
+       struct ldb_val *newval;
+
+       dn = ldb_dn_explode(ctx, (char *)val->data);
+
+       newdn = map_local_dn(map, ctx, dn);
+
+       talloc_free(dn);
+
+       newval = talloc(ctx, struct ldb_val);
+       newval->data = (uint8_t *)ldb_dn_linearize(ctx, newdn);
+       newval->length = strlen((char *)newval->data);
+
+       talloc_free(newdn);
+
+       return *newval;
+}
+
+static struct ldb_val map_convert_remote_dn(struct ldb_map_context *map, TALLOC_CTX *ctx, const struct ldb_val *val)
+{
+       struct ldb_dn *dn, *newdn;;
+       struct ldb_val *newval;
+
+       dn = ldb_dn_explode(ctx, (char *)val->data);
+
+       newdn = map_remote_dn(map, ctx, dn);
+
+       talloc_free(dn);
+
+       newval = talloc(ctx, struct ldb_val);
+       newval->data = (uint8_t *)ldb_dn_linearize(ctx, newdn);
+       newval->length = strlen((char *)newval->data);
+
+       talloc_free(newdn);
+
+       return *newval;
+}
+
+static struct ldb_val map_convert_local_objectclass(struct ldb_map_context *map, TALLOC_CTX *ctx, const struct ldb_val *val)
+{
+       int i;
+
+       for (i = 0; map->objectclass_maps[i].local_name; i++) {
+               if (!strcmp(map->objectclass_maps[i].local_name, (char *)val->data)) {
+                       struct ldb_val newval;
+                       newval.data = (uint8_t*)talloc_strdup(ctx, map->objectclass_maps[i].remote_name);
+                       newval.length = strlen((char *)newval.data);
+
+                       return ldb_val_dup(ctx, &newval);
+               }
+       }
+
+       DEBUG(1, ("Unable to map local object class '%s'\n", (char *)val->data));
+       return ldb_val_dup(ctx, val); 
+}
+
+static struct ldb_val map_convert_remote_objectclass(struct ldb_map_context *map, TALLOC_CTX *ctx, const struct ldb_val *val)
+{
+       int i;
+
+       for (i = 0; map->objectclass_maps[i].remote_name; i++) {
+               if (!strcmp(map->objectclass_maps[i].remote_name, (char *)val->data)) {
+                       struct ldb_val newval;
+                       newval.data = (uint8_t*)talloc_strdup(ctx, map->objectclass_maps[i].local_name);
+                       newval.length = strlen((char *)newval.data);
+
+                       return ldb_val_dup(ctx, &newval);
+               }
+       }
+
+       DEBUG(1, ("Unable to map remote object class '%s'\n", (char *)val->data));
+       return ldb_val_dup(ctx, val); 
+}
+
+static const struct ldb_map_attribute builtin_attribute_maps[] = {
+       {
+               .local_name = "dn",
+               .type = MAP_CONVERT,
+               .u.convert.remote_name = "dn",
+               .u.convert.convert_local = map_convert_local_dn,
+               .u.convert.convert_remote = map_convert_remote_dn,
+       },
+       {
+               .local_name = "objectclass",
+               .type = MAP_CONVERT,
+               .u.convert.remote_name = "objectclass",
+               .u.convert.convert_local = map_convert_local_objectclass,
+               .u.convert.convert_remote = map_convert_remote_objectclass,
+       },
+       {
+               .local_name = NULL,
+       }
+};
+
index bdeafc827a829e3df5d3a9a888b8074cffed28c6..60e09975b7b99228489ca167a4dc136a44780906 100644 (file)
  * returning too much attributes in ldb_search() doesn't)
  */
 
+struct ldb_map_context;
+
 struct ldb_map_attribute 
 {
        const char *local_name; /* local name */
 
-       enum { 
+       enum ldb_map_attr_type 
                MAP_IGNORE, /* Ignore this local attribute. Doesn't exist remotely.  */
-               MAP_KEEP,   /* Keep as is */
+               MAP_KEEP,   /* Keep as is. Same name locally and remotely. */
                MAP_RENAME, /* Simply rename the attribute. Name changes, data is the same */
                MAP_CONVERT, /* Rename + convert data */
                MAP_GENERATE /* Use generate function for generating new name/data. 
@@ -53,7 +55,7 @@ struct ldb_map_attribute
        } type;
        
        /* if set, will be called for expressions that contain this attribute */
-       struct ldb_parse_tree *(*convert_operator) (TALLOC_CTX *ctx, const struct ldb_parse_tree *);    
+       struct ldb_parse_tree *(*convert_operator) (struct ldb_map_context *, TALLOC_CTX *ctx, const struct ldb_parse_tree *);  
 
        union { 
                struct {
@@ -62,27 +64,21 @@ struct ldb_map_attribute
                
                struct {
                        const char *remote_name;
-
-                       struct ldb_message_element *(*convert_local) (
-                               TALLOC_CTX *ctx, 
-                               const char *remote_attr,
-                               const struct ldb_message_element *);
-
-                       struct ldb_message_element *(*convert_remote) (
-                               TALLOC_CTX *ctx,
-                               const char *local_attr,
-                               const struct ldb_message_element *);
+                       struct ldb_val (*convert_local) (struct ldb_map_context *, TALLOC_CTX *, const struct ldb_val *);
+                       struct ldb_val (*convert_remote) (struct ldb_map_context *, TALLOC_CTX *, const struct ldb_val *);
                } convert;
        
                struct {
                        /* Generate the local attribute from remote message */
                        struct ldb_message_element *(*generate_local) (
+                                       struct ldb_map_context *, 
                                        TALLOC_CTX *ctx, 
-                                       const char *attr, 
+                                       const char *attr,
                                        const struct ldb_message *remote);
 
                        /* Update remote message with information from local message */
                        void (*generate_remote) (
+                                       struct ldb_map_context *, 
                                        const char *local_attr,
                                        const struct ldb_message *local, 
                                        struct ldb_message *remote);
@@ -102,4 +98,10 @@ struct ldb_map_objectclass
        const char *remote_name;
 };
 
+struct ldb_map_context
+{
+       struct ldb_map_attribute *attribute_maps;
+       const struct ldb_map_objectclass *objectclass_maps;
+};
+
 #endif /* __LDB_MAP_H__ */
index c1ac691e92c9174cd0ce1149e66f688d6c49e041..ceb52b8a16230bb892eec0b9035c26213e2ec211 100644 (file)
  * sambaMungedDial
  * sambaLogonHours */
 
-static struct ldb_message_element *convert_sid_rid(TALLOC_CTX *ctx, const char *remote_attr, const struct ldb_message_element *el)
+static struct ldb_val convert_sid_rid(struct ldb_map_context *map, TALLOC_CTX *ctx, const struct ldb_val *val)
 {
-       struct ldb_message_element *ret = talloc(ctx, struct ldb_message_element);
-       int i;
-
        printf("Converting SID TO RID *\n");
 
-       ret->flags = el->flags;
-       ret->name = talloc_strdup(ret, remote_attr);
-       ret->num_values = el->num_values;
-       ret->values = talloc_array(ret, struct ldb_val, ret->num_values);
-
-       for (i = 0; i < ret->num_values; i++) {
-               ret->values[i] = ldb_val_dup(ret->values, &el->values[i]);
-       }
-
-       return ret;
+       return ldb_val_dup(ctx, val);
 }
 
-static struct ldb_message_element *convert_rid_sid(TALLOC_CTX *ctx, const char *remote_attr, const struct ldb_message_element *el)
+static struct ldb_val convert_rid_sid(struct ldb_map_context *map, TALLOC_CTX *ctx, const struct ldb_val *val)
 {
-       struct ldb_message_element *ret = talloc(ctx, struct ldb_message_element);
-       int i;
-
        printf("Converting RID TO SID *\n");
 
-       ret->flags = el->flags;
-       ret->name = talloc_strdup(ret, remote_attr);
-       ret->num_values = el->num_values;
-       ret->values = talloc_array(ret, struct ldb_val, ret->num_values);
-
-       for (i = 0; i < ret->num_values; i++) {
-               ret->values[i] = ldb_val_dup(ret->values, &el->values[i]);
-       }
-
-       return ret;
+       return ldb_val_dup(ctx, val);
 }
 
-static struct ldb_message_element *convert_unix_id2name(TALLOC_CTX *ctx, const char *remote_attr, const struct ldb_message_element *el)
+static struct ldb_val convert_unix_id2name(struct ldb_map_context *map, TALLOC_CTX *ctx, const struct ldb_val *val)
 {
-       int i;
-       struct ldb_message_element *ret = talloc(ctx, struct ldb_message_element);
-
        printf("Converting UNIX ID to name\n");
 
-       ret->flags = el->flags;
-       ret->name = talloc_strdup(ret, remote_attr);
-       ret->num_values = el->num_values;
-       ret->values = talloc_array(ret, struct ldb_val, ret->num_values);
-
-       for (i = 0; i < ret->num_values; i++) {
-               ret->values[i] = ldb_val_dup(ret->values, &el->values[i]);
-       }
-
-       return ret;
+       return ldb_val_dup(ctx, val);
 }
 
-static struct ldb_message_element *convert_unix_name2id(TALLOC_CTX *ctx, const char *remote_attr, const struct ldb_message_element *el)
+static struct ldb_val convert_unix_name2id(struct ldb_map_context *map, TALLOC_CTX *ctx, const struct ldb_val *val)
 {
-       struct ldb_message_element *ret = talloc(ctx, struct ldb_message_element);
-       int i;
-
        printf("Converting UNIX name to ID\n");
 
-       ret->flags = el->flags;
-       ret->name = talloc_strdup(ret, remote_attr);
-       ret->num_values = el->num_values;
-       ret->values = talloc_array(ret, struct ldb_val, ret->num_values);
-
-       for (i = 0; i < ret->num_values; i++) {
-               ret->values[i] = ldb_val_dup(ret->values, &el->values[i]);
-       }
-
-       return ret;
+       return ldb_val_dup(ctx, val);
 }
 
 const struct ldb_map_objectclass samba3_objectclasses[] = {
        { "group", "sambaGroupMapping" },
        { "user", "sambaSAMAccount" },
        { "domain", "sambaDomain" },
+       { NULL, NULL }
 };
 
 const struct ldb_map_attribute samba3_attributes[] = 
@@ -253,12 +206,21 @@ const struct ldb_map_attribute samba3_attributes[] =
                .u.rename.remote_name = "sambaSID", 
        },
 
-       /* sambaPwdLastSet -> pwdLastSet*/
+       /* sambaPwdLastSet -> pwdLastSet */
        {
                .local_name = "pwdLastSet",
                .type = MAP_RENAME,
                .u.rename.remote_name = "sambaPwdLastSet",
        },      
+       
+       /* cn -> cn */
+       {
+               .local_name = "cn",
+               .type = MAP_KEEP,
+       },
+       {
+               .local_name = NULL,
+       }
 };
 
        /* the init function */
index 73610a2be2261639854e456324213544349f9e61..988806d8956645305770520b86a38148e314ba08 100755 (executable)
@@ -30,5 +30,18 @@ echo "Checking for existance of record (mapped)"
 $VALGRIND ldbsearch $OPT "(cn=Foo)" unixName lastLogon cn || exit 1
 
 echo "Checking for existance of record (non-mapped)"
-$VALGRIND ldbsearch $LOC "(cn=Foo)" uid sambaLogonTime cn || exit 1
+$VALGRIND ldbsearch $LOC"(cn=foo)" uid sambaLogonTime cn || exit 1
 
+echo "Adding record with mapped attribute in dn"
+$VALGRIND ldbadd $OPT <<EOF
+dn: unixName=nobody,dc=idealx,dc=org
+unixName: nobody 
+cn: Niemand
+
+EOF
+
+echo "Checking for existance of record (mapped)"
+$VALGRIND ldbsearch $OPT "(unixName=nobody)" unixName cn dn || exit 1
+
+echo "Checking for existance of record (non-mapped)"
+$VALGRIND ldbsearch $OPT "(uid=nobody)" unixName cn dn || exit 1
index 38cfd2c4934fd90569b372c5062898d70b344bd2..0e23738b3032b8471d1a7f82195bcb673099e26b 100644 (file)
@@ -323,7 +323,7 @@ struct MprVar mprNTSTATUS(NTSTATUS status)
 struct MprVar mprDataBlob(DATA_BLOB blob)
 {
        struct MprVar res;
-       struct data_blob *pblob = talloc(mprMemCtx(), struct data_blob);
+       DATA_BLOB *pblob = talloc(mprMemCtx(), DATA_BLOB);
        *pblob = data_blob_talloc(pblob, blob.data, blob.length);
 
        res = mprObject("DATA_BLOB");