s3-build: only include memcache.h where needed.
[nivanova/samba-autobuild/.git] / source3 / lib / util_pw.c
index 0d7ffe09e9bc343a8b6358a9868d380e8c532a53..4d4041cd5fdb6f8d85eab66de7cf8dc3552d1914 100644 (file)
@@ -7,7 +7,7 @@
    
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
-   the Free Software Foundation; either version 2 of the License, or
+   the Free Software Foundation; either version 3 of the License, or
    (at your option) any later version.
    
    This program is distributed in the hope that it will be useful,
    GNU General Public License for more details.
    
    You should have received a copy of the GNU General Public License
-   along with this program; if not, write to the Free Software
-   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+   along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
 #include "includes.h"
+#include "memcache.h"
 
-static struct passwd *alloc_copy_passwd(const struct passwd *from) 
+struct passwd *tcopy_passwd(TALLOC_CTX *mem_ctx, const struct passwd *from) 
 {
-       struct passwd *ret = SMB_XMALLOC_P(struct passwd);
-       ZERO_STRUCTP(ret);
-       ret->pw_name = smb_xstrdup(from->pw_name);
-       ret->pw_passwd = smb_xstrdup(from->pw_passwd);
+       struct passwd *ret = TALLOC_P(mem_ctx, struct passwd);
+       if (!ret) {
+               return NULL;
+       }
+       ret->pw_name = talloc_strdup(ret, from->pw_name);
+       ret->pw_passwd = talloc_strdup(ret, from->pw_passwd);
        ret->pw_uid = from->pw_uid;
        ret->pw_gid = from->pw_gid;
-       ret->pw_gecos = smb_xstrdup(from->pw_gecos);
-       ret->pw_dir = smb_xstrdup(from->pw_dir);
-       ret->pw_shell = smb_xstrdup(from->pw_shell);
+       ret->pw_gecos = talloc_strdup(ret, from->pw_gecos);
+       ret->pw_dir = talloc_strdup(ret, from->pw_dir);
+       ret->pw_shell = talloc_strdup(ret, from->pw_shell);
        return ret;
 }
 
-void passwd_free (struct passwd **buf)
+void flush_pwnam_cache(void)
 {
-       if (!*buf) {
-               DEBUG(0, ("attempted double-free of allocated passwd\n"));
-               return;
-       }
-
-       SAFE_FREE((*buf)->pw_name);
-       SAFE_FREE((*buf)->pw_passwd);
-       SAFE_FREE((*buf)->pw_gecos);
-       SAFE_FREE((*buf)->pw_dir);
-       SAFE_FREE((*buf)->pw_shell);
-
-       SAFE_FREE(*buf);
+       memcache_flush(NULL, GETPWNAM_CACHE);
 }
 
-#define PWNAMCACHE_SIZE 4
-static struct passwd *pwnam_cache[PWNAMCACHE_SIZE];
-static BOOL pwnam_cache_initialized = False;
-
-static void init_pwnam_cache(void)
+struct passwd *getpwnam_alloc(TALLOC_CTX *mem_ctx, const char *name)
 {
-       int i;
-
-       if (pwnam_cache_initialized)
-               return;
-
-       for (i=0; i<PWNAMCACHE_SIZE; i++)
-               pwnam_cache[i] = NULL;
-
-       pwnam_cache_initialized = True;
-       return;
-}
-
-struct passwd *getpwnam_alloc(const char *name) 
-{
-       int i;
-
-       struct passwd *temp;
-
-       init_pwnam_cache();
+       struct passwd *pw, *for_cache;
 
-       for (i=0; i<PWNAMCACHE_SIZE; i++) {
-               if ((pwnam_cache[i] != NULL) && 
-                   (strcmp(name, pwnam_cache[i]->pw_name) == 0)) {
-                       DEBUG(10, ("Got %s from pwnam_cache\n", name));
-                       return alloc_copy_passwd(pwnam_cache[i]);
-               }
+       pw = (struct passwd *)memcache_lookup_talloc(
+               NULL, GETPWNAM_CACHE, data_blob_string_const_null(name));
+       if (pw != NULL) {
+               return tcopy_passwd(mem_ctx, pw);
        }
 
-       temp = sys_getpwnam(name);
-       
-       if (!temp) {
-#if 0
-               if (errno == ENOMEM) {
-                       /* what now? */
-               }
-#endif
+       pw = sys_getpwnam(name);
+       if (pw == NULL) {
                return NULL;
        }
 
-       for (i=0; i<PWNAMCACHE_SIZE; i++) {
-               if (pwnam_cache[i] == NULL)
-                       break;
+       for_cache = tcopy_passwd(talloc_autofree_context(), pw);
+       if (for_cache == NULL) {
+               return NULL;
        }
 
-       if (i == PWNAMCACHE_SIZE)
-               i = rand() % PWNAMCACHE_SIZE;
-
-       if (pwnam_cache[i] != NULL)
-               passwd_free(&pwnam_cache[i]);
-
-       pwnam_cache[i] = alloc_copy_passwd(temp);
+       memcache_add_talloc(NULL, GETPWNAM_CACHE,
+                           data_blob_string_const_null(name), &for_cache);
 
-       return alloc_copy_passwd(temp);
+       return tcopy_passwd(mem_ctx, pw);
 }
 
-struct passwd *getpwuid_alloc(uid_t uid) 
+struct passwd *getpwuid_alloc(TALLOC_CTX *mem_ctx, uid_t uid) 
 {
        struct passwd *temp;
 
@@ -128,5 +84,5 @@ struct passwd *getpwuid_alloc(uid_t uid)
                return NULL;
        }
 
-       return alloc_copy_passwd(temp);
+       return tcopy_passwd(mem_ctx, temp);
 }