lib: modules: Change XXX_init interface from XXX_init(void) to XXX_init(TALLOC_CTX *)
[amitay/samba.git] / source3 / groupdb / mapping.c
index 70c96f186ce45af3cb80275a9887fdffaa58b4aa..ac70fe68c482c34966a6c66c581910e4d9e99154 100644 (file)
@@ -26,7 +26,8 @@
 #include "groupdb/mapping.h"
 #include "../libcli/security/security.h"
 #include "lib/winbind_util.h"
-#include "tdb_compat.h"
+#include <tdb.h>
+#include "groupdb/mapping_tdb.h"
 
 static const struct mapping_backend *backend;
 
@@ -50,24 +51,48 @@ initialise first time the mapping list
 ****************************************************************************/
 NTSTATUS add_initial_entry(gid_t gid, const char *sid, enum lsa_SidType sid_name_use, const char *nt_name, const char *comment)
 {
-       GROUP_MAP map;
+       NTSTATUS status;
+       GROUP_MAP *map;
 
        if(!init_group_mapping()) {
                DEBUG(0,("failed to initialize group mapping\n"));
                return NT_STATUS_UNSUCCESSFUL;
        }
 
-       map.gid=gid;
-       if (!string_to_sid(&map.sid, sid)) {
+       map = talloc_zero(NULL, GROUP_MAP);
+       if (!map) {
+               return NT_STATUS_NO_MEMORY;
+       }
+
+       map->gid=gid;
+       if (!string_to_sid(&map->sid, sid)) {
                DEBUG(0, ("string_to_sid failed: %s", sid));
-               return NT_STATUS_UNSUCCESSFUL;
+               status = NT_STATUS_UNSUCCESSFUL;
+               goto done;
+       }
+
+       map->sid_name_use=sid_name_use;
+       map->nt_name = talloc_strdup(map, nt_name);
+       if (!map->nt_name) {
+               status = NT_STATUS_NO_MEMORY;
+               goto done;
+       }
+
+       if (comment) {
+               map->comment = talloc_strdup(map, comment);
+       } else {
+               map->comment = talloc_strdup(map, "");
+       }
+       if (!map->comment) {
+               status = NT_STATUS_NO_MEMORY;
+               goto done;
        }
 
-       map.sid_name_use=sid_name_use;
-       fstrcpy(map.nt_name, nt_name);
-       fstrcpy(map.comment, comment);
+       status = pdb_add_group_mapping_entry(map);
 
-       return pdb_add_group_mapping_entry(&map);
+done:
+       TALLOC_FREE(map);
+       return status;
 }
 
 static NTSTATUS alias_memberships(const struct dom_sid *members, size_t num_members,
@@ -127,13 +152,19 @@ bool get_domain_group_from_sid(struct dom_sid sid, GROUP_MAP *map)
        /* special case check for rid 513 */
 
        if ( !ret ) {
-               uint32 rid;
+               uint32_t rid;
 
                sid_peek_rid( &sid, &rid );
 
                if ( rid == DOMAIN_RID_USERS ) {
-                       fstrcpy( map->nt_name, "None" );
-                       fstrcpy( map->comment, "Ordinary Users" );
+                       map->nt_name = talloc_strdup(map, "None");
+                       if (!map->nt_name) {
+                               return false;
+                       }
+                       map->comment = talloc_strdup(map, "Ordinary Users");
+                       if (!map->comment) {
+                               return false;
+                       }
                        sid_copy( &map->sid, &sid );
                        map->sid_name_use = SID_NAME_DOM_GRP;
                        map->gid = (gid_t)-1;
@@ -182,11 +213,11 @@ int smb_create_group(const char *unix_group, gid_t *new_gid)
 
        /* defer to scripts */
 
-       if ( *lp_addgroup_script() ) {
+       if ( *lp_add_group_script(talloc_tos()) ) {
                TALLOC_CTX *ctx = talloc_tos();
 
                add_script = talloc_strdup(ctx,
-                                       lp_addgroup_script());
+                                       lp_add_group_script(ctx));
                if (!add_script) {
                        return -1;
                }
@@ -196,7 +227,7 @@ int smb_create_group(const char *unix_group, gid_t *new_gid)
                        return -1;
                }
 
-               ret = smbrun(add_script, &fd);
+               ret = smbrun(add_script, &fd, NULL);
                DEBUG(ret ? 0 : 3,("smb_create_group: Running the command `%s' gave %d\n",add_script,ret));
                if (ret == 0) {
                        smb_nscd_flush_group_cache();
@@ -238,11 +269,11 @@ int smb_delete_group(const char *unix_group)
 
        /* defer to scripts */
 
-       if ( *lp_delgroup_script() ) {
+       if ( *lp_delete_group_script(talloc_tos()) ) {
                TALLOC_CTX *ctx = talloc_tos();
 
                del_script = talloc_strdup(ctx,
-                               lp_delgroup_script());
+                               lp_delete_group_script(ctx));
                if (!del_script) {
                        return -1;
                }
@@ -251,7 +282,7 @@ int smb_delete_group(const char *unix_group)
                if (!del_script) {
                        return -1;
                }
-               ret = smbrun(del_script,NULL);
+               ret = smbrun(del_script, NULL, NULL);
                DEBUG(ret ? 0 : 3,("smb_delete_group: Running the command `%s' gave %d\n",del_script,ret));
                if (ret == 0) {
                        smb_nscd_flush_group_cache();
@@ -273,11 +304,11 @@ int smb_set_primary_group(const char *unix_group, const char* unix_user)
 
        /* defer to scripts */
 
-       if ( *lp_setprimarygroup_script() ) {
+       if ( *lp_set_primary_group_script(talloc_tos()) ) {
                TALLOC_CTX *ctx = talloc_tos();
 
                add_script = talloc_strdup(ctx,
-                               lp_setprimarygroup_script());
+                               lp_set_primary_group_script(ctx));
                if (!add_script) {
                        return -1;
                }
@@ -291,7 +322,7 @@ int smb_set_primary_group(const char *unix_group, const char* unix_user)
                if (!add_script) {
                        return -1;
                }
-               ret = smbrun(add_script,NULL);
+               ret = smbrun(add_script, NULL, NULL);
                flush_pwnam_cache();
                DEBUG(ret ? 0 : 3,("smb_set_primary_group: "
                         "Running the command `%s' gave %d\n",add_script,ret));
@@ -315,11 +346,11 @@ int smb_add_user_group(const char *unix_group, const char *unix_user)
 
        /* defer to scripts */
 
-       if ( *lp_addusertogroup_script() ) {
+       if ( *lp_add_user_to_group_script(talloc_tos()) ) {
                TALLOC_CTX *ctx = talloc_tos();
 
                add_script = talloc_strdup(ctx,
-                               lp_addusertogroup_script());
+                               lp_add_user_to_group_script(ctx));
                if (!add_script) {
                        return -1;
                }
@@ -333,7 +364,7 @@ int smb_add_user_group(const char *unix_group, const char *unix_user)
                if (!add_script) {
                        return -1;
                }
-               ret = smbrun(add_script,NULL);
+               ret = smbrun(add_script, NULL, NULL);
                DEBUG(ret ? 0 : 3,("smb_add_user_group: Running the command `%s' gave %d\n",add_script,ret));
                if (ret == 0) {
                        smb_nscd_flush_group_cache();
@@ -355,11 +386,11 @@ int smb_delete_user_group(const char *unix_group, const char *unix_user)
 
        /* defer to scripts */
 
-       if ( *lp_deluserfromgroup_script() ) {
+       if ( *lp_delete_user_from_group_script(talloc_tos()) ) {
                TALLOC_CTX *ctx = talloc_tos();
 
                del_script = talloc_strdup(ctx,
-                               lp_deluserfromgroup_script());
+                               lp_delete_user_from_group_script(ctx));
                if (!del_script) {
                        return -1;
                }
@@ -373,7 +404,7 @@ int smb_delete_user_group(const char *unix_group, const char *unix_user)
                if (!del_script) {
                        return -1;
                }
-               ret = smbrun(del_script,NULL);
+               ret = smbrun(del_script, NULL, NULL);
                DEBUG(ret ? 0 : 3,("smb_delete_user_group: Running the command `%s' gave %d\n",del_script,ret));
                if (ret == 0) {
                        smb_nscd_flush_group_cache();
@@ -452,9 +483,11 @@ NTSTATUS pdb_default_delete_group_mapping_entry(struct pdb_methods *methods,
 }
 
 NTSTATUS pdb_default_enum_group_mapping(struct pdb_methods *methods,
-                                          const struct dom_sid *sid, enum lsa_SidType sid_name_use,
-                                          GROUP_MAP **pp_rmap, size_t *p_num_entries,
-                                          bool unix_only)
+                                       const struct dom_sid *sid,
+                                       enum lsa_SidType sid_name_use,
+                                       GROUP_MAP ***pp_rmap,
+                                       size_t *p_num_entries,
+                                       bool unix_only)
 {
        if (!init_group_mapping()) {
                DEBUG(0,("failed to initialize group mapping\n"));
@@ -465,14 +498,14 @@ NTSTATUS pdb_default_enum_group_mapping(struct pdb_methods *methods,
 }
 
 NTSTATUS pdb_default_create_alias(struct pdb_methods *methods,
-                                 const char *name, uint32 *rid)
+                                 const char *name, uint32_t *rid)
 {
        struct dom_sid sid;
        enum lsa_SidType type;
-       uint32 new_rid;
+       uint32_t new_rid;
        gid_t gid;
        bool exists;
-       GROUP_MAP map;
+       GROUP_MAP *map;
        TALLOC_CTX *mem_ctx;
        NTSTATUS status;
 
@@ -485,15 +518,16 @@ NTSTATUS pdb_default_create_alias(struct pdb_methods *methods,
 
        exists = lookup_name(mem_ctx, name, LOOKUP_NAME_LOCAL,
                             NULL, NULL, &sid, &type);
-       TALLOC_FREE(mem_ctx);
 
        if (exists) {
-               return NT_STATUS_ALIAS_EXISTS;
+               status = NT_STATUS_ALIAS_EXISTS;
+               goto done;
        }
 
        if (!pdb_new_rid(&new_rid)) {
                DEBUG(0, ("Could not allocate a RID.\n"));
-               return NT_STATUS_ACCESS_DENIED;
+               status = NT_STATUS_ACCESS_DENIED;
+               goto done;
        }
 
        sid_compose(&sid, get_global_sam_sid(), new_rid);
@@ -501,29 +535,46 @@ NTSTATUS pdb_default_create_alias(struct pdb_methods *methods,
        if (!winbind_allocate_gid(&gid)) {
                DEBUG(3, ("Could not get a gid out of winbind - "
                          "wasted a rid :-(\n"));
-               return NT_STATUS_ACCESS_DENIED;
+               status = NT_STATUS_ACCESS_DENIED;
+               goto done;
        }
 
        DEBUG(10, ("Creating alias %s with gid %u and rid %u\n",
                   name, (unsigned int)gid, (unsigned int)new_rid));
 
-       map.gid = gid;
-       sid_copy(&map.sid, &sid);
-       map.sid_name_use = SID_NAME_ALIAS;
-       fstrcpy(map.nt_name, name);
-       fstrcpy(map.comment, "");
+       map = talloc_zero(mem_ctx, GROUP_MAP);
+       if (!map) {
+               status = NT_STATUS_NO_MEMORY;
+               goto done;
+       }
+
+       map->gid = gid;
+       sid_copy(&map->sid, &sid);
+       map->sid_name_use = SID_NAME_ALIAS;
+       map->nt_name = talloc_strdup(map, name);
+       if (!map->nt_name) {
+               status = NT_STATUS_NO_MEMORY;
+               goto done;
+       }
+       map->comment = talloc_strdup(map, "");
+       if (!map->comment) {
+               status = NT_STATUS_NO_MEMORY;
+               goto done;
+       }
 
-       status = pdb_add_group_mapping_entry(&map);
+       status = pdb_add_group_mapping_entry(map);
 
        if (!NT_STATUS_IS_OK(status)) {
                DEBUG(0, ("Could not add group mapping entry for alias %s "
                          "(%s)\n", name, nt_errstr(status)));
-               return status;
+               goto done;
        }
 
        *rid = new_rid;
 
-       return NT_STATUS_OK;
+done:
+       TALLOC_FREE(mem_ctx);
+       return status;
 }
 
 NTSTATUS pdb_default_delete_alias(struct pdb_methods *methods,
@@ -536,38 +587,78 @@ NTSTATUS pdb_default_get_aliasinfo(struct pdb_methods *methods,
                                   const struct dom_sid *sid,
                                   struct acct_info *info)
 {
-       GROUP_MAP map;
+       NTSTATUS status = NT_STATUS_OK;
+       GROUP_MAP *map;
 
-       if (!pdb_getgrsid(&map, *sid))
-               return NT_STATUS_NO_SUCH_ALIAS;
+       map = talloc_zero(NULL, GROUP_MAP);
+       if (!map) {
+               return NT_STATUS_NO_MEMORY;
+       }
+
+       if (!pdb_getgrsid(map, *sid)) {
+               status = NT_STATUS_NO_SUCH_ALIAS;
+               goto done;
+       }
 
-       if ((map.sid_name_use != SID_NAME_ALIAS) &&
-           (map.sid_name_use != SID_NAME_WKN_GRP)) {
+       if ((map->sid_name_use != SID_NAME_ALIAS) &&
+           (map->sid_name_use != SID_NAME_WKN_GRP)) {
                DEBUG(2, ("%s is a %s, expected an alias\n",
                          sid_string_dbg(sid),
-                         sid_type_lookup(map.sid_name_use)));
-               return NT_STATUS_NO_SUCH_ALIAS;
+                         sid_type_lookup(map->sid_name_use)));
+               status = NT_STATUS_NO_SUCH_ALIAS;
+               goto done;
        }
 
-       fstrcpy(info->acct_name, map.nt_name);
-       fstrcpy(info->acct_desc, map.comment);
-       sid_peek_rid(&map.sid, &info->rid);
-       return NT_STATUS_OK;
+       info->acct_name = talloc_move(info, &map->nt_name);
+       if (!info->acct_name) {
+               status = NT_STATUS_NO_MEMORY;
+               goto done;
+       }
+       info->acct_desc = talloc_move(info, &map->comment);
+       if (!info->acct_desc) {
+               status = NT_STATUS_NO_MEMORY;
+               goto done;
+       }
+       sid_peek_rid(&map->sid, &info->rid);
+
+done:
+       TALLOC_FREE(map);
+       return status;
 }
 
 NTSTATUS pdb_default_set_aliasinfo(struct pdb_methods *methods,
                                   const struct dom_sid *sid,
                                   struct acct_info *info)
 {
-       GROUP_MAP map;
+       NTSTATUS status = NT_STATUS_OK;
+       GROUP_MAP *map;
 
-       if (!pdb_getgrsid(&map, *sid))
-               return NT_STATUS_NO_SUCH_ALIAS;
+       map = talloc_zero(NULL, GROUP_MAP);
+       if (!map) {
+               return NT_STATUS_NO_MEMORY;
+       }
 
-       fstrcpy(map.nt_name, info->acct_name);
-       fstrcpy(map.comment, info->acct_desc);
+       if (!pdb_getgrsid(map, *sid)) {
+               status = NT_STATUS_NO_SUCH_ALIAS;
+               goto done;
+       }
+
+       map->nt_name = talloc_strdup(map, info->acct_name);
+       if (!map->nt_name) {
+               status = NT_STATUS_NO_MEMORY;
+               goto done;
+       }
+       map->comment = talloc_strdup(map, info->acct_desc);
+       if (!map->comment) {
+               status = NT_STATUS_NO_MEMORY;
+               goto done;
+       }
+
+       status = pdb_update_group_mapping_entry(map);
 
-       return pdb_update_group_mapping_entry(&map);
+done:
+       TALLOC_FREE(map);
+       return status;
 }
 
 NTSTATUS pdb_default_add_aliasmem(struct pdb_methods *methods,
@@ -607,7 +698,7 @@ NTSTATUS pdb_default_alias_memberships(struct pdb_methods *methods,
                                       const struct dom_sid *domain_sid,
                                       const struct dom_sid *members,
                                       size_t num_members,
-                                      uint32 **pp_alias_rids,
+                                      uint32_t **pp_alias_rids,
                                       size_t *p_num_alias_rids)
 {
        struct dom_sid *alias_sids;
@@ -635,7 +726,7 @@ NTSTATUS pdb_default_alias_memberships(struct pdb_methods *methods,
                return NT_STATUS_OK;
        }
 
-       *pp_alias_rids = talloc_array(mem_ctx, uint32, num_alias_sids);
+       *pp_alias_rids = talloc_array(mem_ctx, uint32_t, num_alias_sids);
        if (*pp_alias_rids == NULL)
                return NT_STATUS_NO_MEMORY;
 
@@ -699,54 +790,22 @@ NTSTATUS pdb_nop_enum_group_mapping(struct pdb_methods *methods,
        return NT_STATUS_UNSUCCESSFUL;
 }
 
-/****************************************************************************
- These need to be redirected through pdb_interface.c
-****************************************************************************/
-bool pdb_get_dom_grp_info(const struct dom_sid *sid, struct acct_info *info)
-{
-       GROUP_MAP map;
-       bool res;
-
-       become_root();
-       res = get_domain_group_from_sid(*sid, &map);
-       unbecome_root();
-
-       if (!res)
-               return False;
-
-       fstrcpy(info->acct_name, map.nt_name);
-       fstrcpy(info->acct_desc, map.comment);
-       sid_peek_rid(sid, &info->rid);
-       return True;
-}
-
-bool pdb_set_dom_grp_info(const struct dom_sid *sid, const struct acct_info *info)
-{
-       GROUP_MAP map;
-
-       if (!get_domain_group_from_sid(*sid, &map))
-               return False;
-
-       fstrcpy(map.nt_name, info->acct_name);
-       fstrcpy(map.comment, info->acct_desc);
-
-       return NT_STATUS_IS_OK(pdb_update_group_mapping_entry(&map));
-}
-
-/********************************************************************
- Really just intended to be called by smbd
-********************************************************************/
-
-NTSTATUS pdb_create_builtin_alias(uint32 rid)
+/**
+* @brief Add a new group mapping
+*
+* @param[in] gid gid to use to store the mapping. If gid is 0,
+*                new gid will be allocated from winbind
+*
+* @return Normal NTSTATUS return
+*/
+NTSTATUS pdb_create_builtin_alias(uint32_t rid, gid_t gid)
 {
        struct dom_sid sid;
        enum lsa_SidType type;
-       gid_t gid;
-       GROUP_MAP map;
-       TALLOC_CTX *mem_ctx;
+       gid_t gidformap;
+       GROUP_MAP *map;
        NTSTATUS status;
        const char *name = NULL;
-       fstring groupname;
 
        DEBUG(10, ("Trying to create builtin alias %d\n", rid));
 
@@ -754,40 +813,54 @@ NTSTATUS pdb_create_builtin_alias(uint32 rid)
                return NT_STATUS_NO_SUCH_ALIAS;
        }
 
-       if ( (mem_ctx = talloc_new(NULL)) == NULL ) {
+       /* use map as overall temp mem context */
+       map = talloc_zero(NULL, GROUP_MAP);
+       if (!map) {
                return NT_STATUS_NO_MEMORY;
        }
 
-       if ( !lookup_sid(mem_ctx, &sid, NULL, &name, &type) ) {
-               TALLOC_FREE( mem_ctx );
-               return NT_STATUS_NO_SUCH_ALIAS;
+       if (!lookup_sid(map, &sid, NULL, &name, &type)) {
+               status = NT_STATUS_NO_SUCH_ALIAS;
+               goto done;
        }
 
-       /* validate RID so copy the name and move on */
-
-       fstrcpy( groupname, name );
-       TALLOC_FREE( mem_ctx );
-
-       if (!winbind_allocate_gid(&gid)) {
-               DEBUG(3, ("pdb_create_builtin_alias: Could not get a gid out of winbind\n"));
-               return NT_STATUS_ACCESS_DENIED;
+       if (gid == 0) {
+               if (!winbind_allocate_gid(&gidformap)) {
+                       DEBUG(3, ("pdb_create_builtin_alias: Could not get a "
+                                 "gid out of winbind\n"));
+                       status = NT_STATUS_ACCESS_DENIED;
+                       goto done;
+               }
+       } else {
+               gidformap = gid;
        }
 
-       DEBUG(10,("Creating alias %s with gid %u\n", groupname, (unsigned int)gid));
+       DEBUG(10, ("Creating alias %s with gid %u\n", name,
+                  (unsigned) gidformap));
 
-       map.gid = gid;
-       sid_copy(&map.sid, &sid);
-       map.sid_name_use = SID_NAME_ALIAS;
-       strlcpy(map.nt_name, groupname, sizeof(map.nt_name));
-       strlcpy(map.comment, "", sizeof(map.comment));
+       map->gid = gidformap;
+       sid_copy(&map->sid, &sid);
+       map->sid_name_use = SID_NAME_ALIAS;
+       map->nt_name = talloc_strdup(map, name);
+       if (!map->nt_name) {
+               status = NT_STATUS_NO_MEMORY;
+               goto done;
+       }
+       map->comment = talloc_strdup(map, "");
+       if (!map->comment) {
+               status = NT_STATUS_NO_MEMORY;
+               goto done;
+       }
 
-       status = pdb_add_group_mapping_entry(&map);
+       status = pdb_add_group_mapping_entry(map);
 
        if (!NT_STATUS_IS_OK(status)) {
                DEBUG(0, ("pdb_create_builtin_alias: Could not add group mapping entry for alias %d "
                          "(%s)\n", rid, nt_errstr(status)));
        }
 
+done:
+       TALLOC_FREE(map);
        return status;
 }