r4088: Get medieval on our ass about malloc.... :-). Take control of all our allocation
[jra/samba/.git] / source3 / auth / auth_util.c
index e6cc0fe5b30d05c878ec5ddc4b68f0fbcd667ef1..d985c0a54fcadb4af27427a49d28a8fa4f97c280 100644 (file)
@@ -132,7 +132,7 @@ static NTSTATUS make_user_info(auth_usersupplied_info **user_info,
 
        DEBUG(5,("attempting to make a user_info for %s (%s)\n", internal_username, smb_name));
 
-       *user_info = malloc(sizeof(**user_info));
+       *user_info = SMB_MALLOC_P(auth_usersupplied_info);
        if (!user_info) {
                DEBUG(0,("malloc failed for user_info (size %lu)\n", (unsigned long)sizeof(*user_info)));
                return NT_STATUS_NO_MEMORY;
@@ -142,7 +142,7 @@ static NTSTATUS make_user_info(auth_usersupplied_info **user_info,
 
        DEBUG(5,("making strings for %s's user_info struct\n", internal_username));
 
-       (*user_info)->smb_name.str = strdup(smb_name);
+       (*user_info)->smb_name.str = SMB_STRDUP(smb_name);
        if ((*user_info)->smb_name.str) { 
                (*user_info)->smb_name.len = strlen(smb_name);
        } else {
@@ -150,7 +150,7 @@ static NTSTATUS make_user_info(auth_usersupplied_info **user_info,
                return NT_STATUS_NO_MEMORY;
        }
        
-       (*user_info)->internal_username.str = strdup(internal_username);
+       (*user_info)->internal_username.str = SMB_STRDUP(internal_username);
        if ((*user_info)->internal_username.str) { 
                (*user_info)->internal_username.len = strlen(internal_username);
        } else {
@@ -158,7 +158,7 @@ static NTSTATUS make_user_info(auth_usersupplied_info **user_info,
                return NT_STATUS_NO_MEMORY;
        }
 
-       (*user_info)->domain.str = strdup(domain);
+       (*user_info)->domain.str = SMB_STRDUP(domain);
        if ((*user_info)->domain.str) { 
                (*user_info)->domain.len = strlen(domain);
        } else {
@@ -166,7 +166,7 @@ static NTSTATUS make_user_info(auth_usersupplied_info **user_info,
                return NT_STATUS_NO_MEMORY;
        }
 
-       (*user_info)->client_domain.str = strdup(client_domain);
+       (*user_info)->client_domain.str = SMB_STRDUP(client_domain);
        if ((*user_info)->client_domain.str) { 
                (*user_info)->client_domain.len = strlen(client_domain);
        } else {
@@ -174,7 +174,7 @@ static NTSTATUS make_user_info(auth_usersupplied_info **user_info,
                return NT_STATUS_NO_MEMORY;
        }
 
-       (*user_info)->wksta_name.str = strdup(wksta_name);
+       (*user_info)->wksta_name.str = SMB_STRDUP(wksta_name);
        if ((*user_info)->wksta_name.str) { 
                (*user_info)->wksta_name.len = strlen(wksta_name);
        } else {
@@ -523,7 +523,7 @@ static NTSTATUS create_nt_user_token(const DOM_SID *user_sid, const DOM_SID *gro
        int i;
        int sid_ndx;
        
-       if ((ptoken = malloc( sizeof(NT_USER_TOKEN) ) ) == NULL) {
+       if ((ptoken = SMB_MALLOC_P(NT_USER_TOKEN)) == NULL) {
                DEBUG(0, ("create_nt_token: Out of memory allocating token\n"));
                nt_status = NT_STATUS_NO_MEMORY;
                return nt_status;
@@ -533,7 +533,7 @@ static NTSTATUS create_nt_user_token(const DOM_SID *user_sid, const DOM_SID *gro
 
        ptoken->num_sids = n_groupSIDs + 5;
 
-       if ((ptoken->user_sids = (DOM_SID *)malloc( sizeof(DOM_SID) * ptoken->num_sids )) == NULL) {
+       if ((ptoken->user_sids = SMB_MALLOC_ARRAY( DOM_SID, ptoken->num_sids )) == NULL) {
                DEBUG(0, ("create_nt_token: Out of memory allocating SIDs\n"));
                nt_status = NT_STATUS_NO_MEMORY;
                return nt_status;
@@ -610,7 +610,7 @@ NT_USER_TOKEN *create_nt_token(uid_t uid, gid_t gid, int ngroups, gid_t *groups,
                return NULL;
        }
 
-       group_sids = malloc(sizeof(DOM_SID) * ngroups);
+       group_sids = SMB_MALLOC_ARRAY(DOM_SID, ngroups);
        if (!group_sids) {
                DEBUG(0, ("create_nt_token: malloc() failed for DOM_SID list!\n"));
                return NULL;
@@ -657,47 +657,27 @@ static NTSTATUS get_user_groups(const char *username, uid_t uid, gid_t gid,
 
        *n_groups = 0;
        *groups   = NULL;
-       
-       /* Try winbind first */
 
-       if ( strchr(username, *lp_winbind_separator()) ) {
-               n_unix_groups = winbind_getgroups( username, unix_groups );
+       if (strchr(username, *lp_winbind_separator()) == NULL) {
+               NTSTATUS result;
 
-               DEBUG(10,("get_user_groups: winbind_getgroups(%s): result = %s\n", username, 
-                         n_unix_groups == -1 ? "FAIL" : "SUCCESS"));
-                         
-               if ( n_unix_groups == -1 )
-                       return NT_STATUS_NO_SUCH_USER; /* what should this return value be? */  
+               become_root();
+               result = pdb_enum_group_memberships(username, gid, groups,
+                                                   unix_groups, n_groups);
+               unbecome_root();
+               return result;
        }
-       else {
-               /* fallback to getgrouplist() */
-               
-               n_unix_groups = groups_max();
-               
-               if ((*unix_groups = malloc( sizeof(gid_t) * n_unix_groups ) ) == NULL) {
-                       DEBUG(0, ("get_user_groups: Out of memory allocating unix group list\n"));
-                       return NT_STATUS_NO_MEMORY;
-               }
+
+       /* We have the separator, this must be winbind */
        
-               if (sys_getgrouplist(username, gid, *unix_groups, &n_unix_groups) == -1) {
-               
-                       gid_t *groups_tmp;
-                       
-                       groups_tmp = Realloc(*unix_groups, sizeof(gid_t) * n_unix_groups);
-                       
-                       if (!groups_tmp) {
-                               SAFE_FREE(*unix_groups);
-                               return NT_STATUS_NO_MEMORY;
-                       }
-                       *unix_groups = groups_tmp;
+       n_unix_groups = winbind_getgroups( username, unix_groups );
 
-                       if (sys_getgrouplist(username, gid, *unix_groups, &n_unix_groups) == -1) {
-                               DEBUG(0, ("get_user_groups: failed to get the unix group list\n"));
-                               SAFE_FREE(*unix_groups);
-                               return NT_STATUS_NO_SUCH_USER; /* what should this return value be? */
-                       }
-               }
-       }
+       DEBUG(10,("get_user_groups: winbind_getgroups(%s): result = %s\n",
+                 username,  n_unix_groups == -1 ? "FAIL" : "SUCCESS"));
+                         
+       if ( n_unix_groups == -1 )
+               return NT_STATUS_NO_SUCH_USER; /* what should this return
+                                               * value be? */  
 
        debug_unix_user_token(DBGC_CLASS, 5, uid, gid, n_unix_groups, *unix_groups);
        
@@ -705,7 +685,7 @@ static NTSTATUS get_user_groups(const char *username, uid_t uid, gid_t gid,
        
        if (n_unix_groups > 0) {
        
-               *groups   = malloc(sizeof(DOM_SID) * n_unix_groups);
+               *groups   = SMB_MALLOC_ARRAY(DOM_SID, n_unix_groups);
                
                if (!*groups) {
                        DEBUG(0, ("get_user_group: malloc() failed for DOM_SID list!\n"));
@@ -735,7 +715,7 @@ static NTSTATUS get_user_groups(const char *username, uid_t uid, gid_t gid,
 
 static NTSTATUS make_server_info(auth_serversupplied_info **server_info)
 {
-       *server_info = malloc(sizeof(**server_info));
+       *server_info = SMB_MALLOC_P(auth_serversupplied_info);
        if (!*server_info) {
                DEBUG(0,("make_server_info: malloc failed!\n"));
                return NT_STATUS_NO_MEMORY;
@@ -884,7 +864,7 @@ NTSTATUS make_server_info_pw(auth_serversupplied_info **server_info,
  Make (and fill) a user_info struct for a guest login.
 ***************************************************************************/
 
-NTSTATUS make_server_info_guest(auth_serversupplied_info **server_info)
+static NTSTATUS make_new_server_info_guest(auth_serversupplied_info **server_info)
 {
        NTSTATUS nt_status;
        SAM_ACCOUNT *sampass = NULL;
@@ -919,6 +899,49 @@ NTSTATUS make_server_info_guest(auth_serversupplied_info **server_info)
        return nt_status;
 }
 
+static auth_serversupplied_info *copy_serverinfo(auth_serversupplied_info *src)
+{
+       auth_serversupplied_info *dst;
+
+       if (!NT_STATUS_IS_OK(make_server_info(&dst)))
+               return NULL;
+
+       dst->guest = src->guest;
+       dst->uid = src->uid;
+       dst->gid = src->gid;
+       dst->n_groups = src->n_groups;
+       if (src->n_groups != 0)
+               dst->groups = memdup(src->groups, sizeof(gid_t)*dst->n_groups);
+       else
+               dst->groups = NULL;
+       dst->ptok = dup_nt_token(src->ptok);
+       dst->user_session_key = data_blob(src->user_session_key.data,
+                                         src->user_session_key.length);
+       dst->lm_session_key = data_blob(src->lm_session_key.data,
+                                         src->lm_session_key.length);
+       pdb_copy_sam_account(src->sam_account, &dst->sam_account);
+       dst->pam_handle = NULL;
+       dst->unix_name = smb_xstrdup(src->unix_name);
+
+       return dst;
+}
+
+static auth_serversupplied_info *guest_info = NULL;
+
+BOOL init_guest_info(void)
+{
+       if (guest_info != NULL)
+               return True;
+
+       return NT_STATUS_IS_OK(make_new_server_info_guest(&guest_info));
+}
+
+NTSTATUS make_server_info_guest(auth_serversupplied_info **server_info)
+{
+       *server_info = copy_serverinfo(guest_info);
+       return (*server_info != NULL) ? NT_STATUS_OK : NT_STATUS_NO_MEMORY;
+}
+
 /***************************************************************************
  Purely internal function for make_server_info_info3
  Fill the sam account from getpwnam
@@ -930,18 +953,23 @@ static NTSTATUS fill_sam_account(TALLOC_CTX *mem_ctx,
                                 uid_t *uid, gid_t *gid,
                                 SAM_ACCOUNT **sam_account)
 {
-       fstring dom_user;
+       fstring dom_user, lower_username;
        fstring real_username;
        struct passwd *passwd;
 
-       fstr_sprintf(dom_user, "%s%s%s", domain, lp_winbind_separator(), 
-               username);
+       fstrcpy( lower_username, username );
+       strlower_m( lower_username );
+
+       fstr_sprintf(dom_user, "%s%c%s", domain, *lp_winbind_separator(), 
+               lower_username);
 
        /* get the passwd struct but don't create the user if he/she 
           does not exist.  We were explicitly called from a following
           a winbindd authentication request so we should assume that 
           nss_winbindd is working */
 
+       map_username( dom_user );
+
        if ( !(passwd = smb_getpwnam( dom_user, real_username, True )) )
                return NT_STATUS_NO_SUCH_USER;
 
@@ -972,25 +1000,25 @@ struct passwd *smb_getpwnam( char *domuser, fstring save_username, BOOL create )
 {
        struct passwd *pw = NULL;
        char *p;
-       fstring mapped_username;
-       fstring strip_username;
+       fstring username;
        
        /* we only save a copy of the username it has been mangled 
           by winbindd use default domain */
           
        save_username[0] = '\0';
-       
-       /* save a local copy of the username and run it through the 
-          username map */
           
-       fstrcpy( mapped_username, domuser );
-       map_username( mapped_username );        
+       /* don't call map_username() here since it has to be done higher 
+          up the stack so we don't call it mutliple times */
+
+       fstrcpy( username, domuser );
        
-       p = strchr_m( mapped_username, *lp_winbind_separator() );
+       p = strchr_m( username, *lp_winbind_separator() );
        
        /* code for a DOMAIN\user string */
        
        if ( p ) {
+               fstring strip_username;
+
                pw = Get_Pwnam( domuser );
                if ( pw ) {     
                        /* make sure we get the case of the username correct */
@@ -999,8 +1027,10 @@ struct passwd *smb_getpwnam( char *domuser, fstring save_username, BOOL create )
                        if ( !strchr_m( pw->pw_name, *lp_winbind_separator() ) ) {
                                char *domain;
                                
-                               domain = mapped_username;
+                               /* split the domain and username into 2 strings */
                                *p = '\0';
+                               domain = username;
+
                                fstr_sprintf(save_username, "%s%c%s", domain, *lp_winbind_separator(), pw->pw_name);
                        }
                        else
@@ -1011,26 +1041,26 @@ struct passwd *smb_getpwnam( char *domuser, fstring save_username, BOOL create )
                }
 
                /* setup for lookup of just the username */
-               /* remember that p and mapped_username are overlapping memory */
+               /* remember that p and username are overlapping memory */
 
                p++;
                fstrcpy( strip_username, p );
-               fstrcpy( mapped_username, strip_username );
+               fstrcpy( username, strip_username );
        }
        
        /* just lookup a plain username */
        
-       pw = Get_Pwnam(mapped_username);
+       pw = Get_Pwnam(username);
                
        /* Create local user if requested. */
        
        if ( !pw && create ) {
                /* Don't add a machine account. */
-               if (mapped_username[strlen(mapped_username)-1] == '$')
+               if (username[strlen(username)-1] == '$')
                        return NULL;
 
-               auth_add_user_script(NULL, mapped_username);
-               pw = Get_Pwnam(mapped_username);
+               auth_add_user_script(NULL, username);
+               pw = Get_Pwnam(username);
        }
        
        /* one last check for a valid passwd struct */
@@ -1102,18 +1132,25 @@ NTSTATUS make_server_info_info3(TALLOC_CTX *mem_ctx,
        }
        
        /* try to fill the SAM account..  If getpwnam() fails, then try the 
-          add user script (2.2.x behavior) */
+          add user script (2.2.x behavior).
+
+          We use the _unmapped_ username here in an attempt to provide
+          consistent username mapping behavior between kerberos and NTLM[SSP]
+          authentication in domain mode security.  I.E. Username mapping should
+          be applied to the fully qualified username (e.g. DOMAIN\user) and
+          no just the login name.  Yes this mean swe called map_username()
+          unnecessarily in make_user_info_map() but that is how the current
+          code is designed.  Making the change here is the least disruptive 
+          place.    -- jerry */
           
-       nt_status = fill_sam_account(mem_ctx, nt_domain, internal_username,
+       nt_status = fill_sam_account(mem_ctx, nt_domain, sent_nt_username,
                &found_username, &uid, &gid, &sam_account);
 
        if (NT_STATUS_EQUAL(nt_status, NT_STATUS_NO_SUCH_USER)) {
-               DEBUG(3,("User %s does not exist, trying to add it\n", 
-                       internal_username));
-               auth_add_user_script(nt_domain, internal_username);
-               nt_status = fill_sam_account(mem_ctx, nt_domain, 
-                       internal_username, &found_username,
-                       &uid, &gid, &sam_account);
+               DEBUG(3,("User %s does not exist, trying to add it\n", internal_username));
+               auth_add_user_script( nt_domain, sent_nt_username );
+               nt_status = fill_sam_account( mem_ctx, nt_domain, sent_nt_username, 
+                       &found_username, &uid, &gid, &sam_account );
        }
        
        if (!NT_STATUS_IS_OK(nt_status)) {
@@ -1207,7 +1244,7 @@ NTSTATUS make_server_info_info3(TALLOC_CTX *mem_ctx,
        
        /* Create a 'combined' list of all SIDs we might want in the SD */
        
-       all_group_SIDs = malloc(sizeof(DOM_SID) * (info3->num_groups2 +info3->num_other_sids));
+       all_group_SIDs = SMB_MALLOC_ARRAY(DOM_SID,info3->num_groups2 + info3->num_other_sids + n_lgroupSIDs);
        
        if (!all_group_SIDs) {
                DEBUG(0, ("malloc() failed for DOM_SID list!\n"));
@@ -1216,12 +1253,6 @@ NTSTATUS make_server_info_info3(TALLOC_CTX *mem_ctx,
                return NT_STATUS_NO_MEMORY;
        }
 
-#if 0  /* JERRY -- no such thing as local groups in current code */
-       /* Copy the 'local' sids */
-       memcpy(all_group_SIDs, lgroupSIDs, sizeof(DOM_SID) * n_lgroupSIDs);
-       SAFE_FREE(lgroupSIDs);
-#endif
-
        /* and create (by appending rids) the 'domain' sids */
        
        for (i = 0; i < info3->num_groups2; i++) {
@@ -1254,13 +1285,22 @@ NTSTATUS make_server_info_info3(TALLOC_CTX *mem_ctx,
                sid_copy(&all_group_SIDs[info3->num_groups2 + i],
                         &info3->other_sids[i].sid);
        }
+
+
+       /* add local alias sids */ 
+
+       for (i = 0; i < n_lgroupSIDs; i++) {
+               sid_copy(&all_group_SIDs[info3->num_groups2 +
+                                        info3->num_other_sids + i],
+                        &lgroupSIDs[i]);
+       }
        
        /* Where are the 'global' sids... */
 
        /* can the user be guest? if yes, where is it stored? */
        
        nt_status = create_nt_user_token(&user_sid, &group_sid,
-               info3->num_groups2 + info3->num_other_sids,
+               info3->num_groups2 + info3->num_other_sids + n_lgroupSIDs,
                all_group_SIDs, False, &token);
                
        if ( !NT_STATUS_IS_OK(nt_status) ) {
@@ -1353,7 +1393,7 @@ BOOL make_auth_methods(struct auth_context *auth_context, auth_methods **auth_me
                smb_panic("make_auth_methods: pointer to auth_method pointer is NULL!\n");
        }
 
-       *auth_method = talloc(auth_context->mem_ctx, sizeof(**auth_method));
+       *auth_method = TALLOC_P(auth_context->mem_ctx, auth_methods);
        if (!*auth_method) {
                DEBUG(0,("make_auth_method: malloc failed!\n"));
                return False;
@@ -1388,7 +1428,7 @@ NT_USER_TOKEN *dup_nt_token(NT_USER_TOKEN *ptoken)
        if (!ptoken)
                return NULL;
 
-    if ((token = (NT_USER_TOKEN *)malloc( sizeof(NT_USER_TOKEN) ) ) == NULL)
+    if ((token = SMB_MALLOC_P(NT_USER_TOKEN)) == NULL)
         return NULL;
 
     ZERO_STRUCTP(token);