linux 2.6.16.38 w/ vs2.0.3-rc1
[linux-2.6.git] / net / sunrpc / svcauth_unix.c
index 082684d..3e6c694 100644 (file)
@@ -1,5 +1,6 @@
 #include <linux/types.h>
 #include <linux/sched.h>
+#include <linux/module.h>
 #include <linux/sunrpc/types.h>
 #include <linux/sunrpc/xdr.h>
 #include <linux/sunrpc/svcsock.h>
@@ -7,6 +8,7 @@
 #include <linux/err.h>
 #include <linux/seq_file.h>
 #include <linux/hash.h>
+#include <linux/string.h>
 
 #define RPCDBG_FACILITY        RPCDBG_AUTH
 
  */
 
 
-static char *strdup(char *s)
-{
-       char *rv = kmalloc(strlen(s)+1, GFP_KERNEL);
-       if (rv)
-               strcpy(rv, s);
-       return rv;
-}
-
 struct unix_domain {
        struct auth_domain      h;
        int     addr_changes;
@@ -54,12 +48,10 @@ struct auth_domain *unix_domain_find(char *name)
        if (new == NULL)
                return NULL;
        cache_init(&new->h.h);
-       atomic_inc(&new->h.h.refcnt);
-       new->h.name = strdup(name);
+       new->h.name = kstrdup(name, GFP_KERNEL);
        new->h.flavour = RPC_AUTH_UNIX;
        new->addr_changes = 0;
        new->h.h.expiry_time = NEVER;
-       new->h.h.flags = 0;
 
        rv = auth_domain_lookup(&new->h, 2);
        if (rv == &new->h) {
@@ -91,14 +83,14 @@ static void svcauth_unix_domain_release(struct auth_domain *dom)
 
 struct ip_map {
        struct cache_head       h;
-       char                    *m_class; /* e.g. "nfsd" */
+       char                    m_class[8]; /* e.g. "nfsd" */
        struct in_addr          m_addr;
        struct unix_domain      *m_client;
        int                     m_add_change;
 };
 static struct cache_head       *ip_table[IP_HASHMAX];
 
-void ip_map_put(struct cache_head *item, struct cache_detail *cd)
+static void ip_map_put(struct cache_head *item, struct cache_detail *cd)
 {
        struct ip_map *im = container_of(item, struct ip_map,h);
        if (cache_put(item, cd)) {
@@ -109,10 +101,22 @@ void ip_map_put(struct cache_head *item, struct cache_detail *cd)
        }
 }
 
+#if IP_HASHBITS == 8
+/* hash_long on a 64 bit machine is currently REALLY BAD for
+ * IP addresses in reverse-endian (i.e. on a little-endian machine).
+ * So use a trivial but reliable hash instead
+ */
+static inline int hash_ip(unsigned long ip)
+{
+       int hash = ip ^ (ip>>16);
+       return (hash ^ (hash>>8)) & 0xff;
+}
+#endif
+
 static inline int ip_map_hash(struct ip_map *item)
 {
        return hash_str(item->m_class, IP_HASHBITS) ^ 
-               hash_long((unsigned long)item->m_addr.s_addr, IP_HASHBITS);
+               hash_ip((unsigned long)item->m_addr.s_addr);
 }
 static inline int ip_map_match(struct ip_map *item, struct ip_map *tmp)
 {
@@ -121,8 +125,7 @@ static inline int ip_map_match(struct ip_map *item, struct ip_map *tmp)
 }
 static inline void ip_map_init(struct ip_map *new, struct ip_map *item)
 {
-       new->m_class = item->m_class;
-       item->m_class = NULL;
+       strcpy(new->m_class, item->m_class);
        new->m_addr.s_addr = item->m_addr.s_addr;
 }
 static inline void ip_map_update(struct ip_map *new, struct ip_map *item)
@@ -152,11 +155,14 @@ static void ip_map_request(struct cache_detail *cd,
 }
 
 static struct ip_map *ip_map_lookup(struct ip_map *, int);
+
 static int ip_map_parse(struct cache_detail *cd,
                          char *mesg, int mlen)
 {
        /* class ipaddress [domainname] */
-       char class[50], buf[50];
+       /* should be safe just to use the start of the input buffer
+        * for scratch: */
+       char *buf = mesg;
        int len;
        int b1,b2,b3,b4;
        char c;
@@ -169,11 +175,11 @@ static int ip_map_parse(struct cache_detail *cd,
        mesg[mlen-1] = 0;
 
        /* class */
-       len = qword_get(&mesg, class, 50);
+       len = qword_get(&mesg, ipm.m_class, sizeof(ipm.m_class));
        if (len <= 0) return -EINVAL;
 
        /* ip address */
-       len = qword_get(&mesg, buf, 50);
+       len = qword_get(&mesg, buf, mlen);
        if (len <= 0) return -EINVAL;
 
        if (sscanf(buf, "%u.%u.%u.%u%c", &b1, &b2, &b3, &b4, &c) != 4)
@@ -184,7 +190,7 @@ static int ip_map_parse(struct cache_detail *cd,
                return -EINVAL;
 
        /* domainname, or empty for NEGATIVE */
-       len = qword_get(&mesg, buf, 50);
+       len = qword_get(&mesg, buf, mlen);
        if (len < 0) return -EINVAL;
 
        if (len) {
@@ -194,9 +200,6 @@ static int ip_map_parse(struct cache_detail *cd,
        } else
                dom = NULL;
 
-       ipm.m_class = strdup(class);
-       if (ipm.m_class == NULL)
-               return -ENOMEM;
        ipm.m_addr.s_addr =
                htonl((((((b1<<8)|b2)<<8)|b3)<<8)|b4);
        ipm.h.flags = 0;
@@ -212,7 +215,6 @@ static int ip_map_parse(struct cache_detail *cd,
                ip_map_put(&ipmp->h, &ip_map_cache);
        if (dom)
                auth_domain_put(dom);
-       if (ipm.m_class) kfree(ipm.m_class);
        if (!ipmp)
                return -ENOMEM;
        cache_flush();
@@ -252,6 +254,7 @@ static int ip_map_show(struct seq_file *m,
        
 
 struct cache_detail ip_map_cache = {
+       .owner          = THIS_MODULE,
        .hash_size      = IP_HASHMAX,
        .hash_table     = ip_table,
        .name           = "auth.unix.ip",
@@ -272,9 +275,7 @@ int auth_unix_add_addr(struct in_addr addr, struct auth_domain *dom)
        if (dom->flavour != RPC_AUTH_UNIX)
                return -EINVAL;
        udom = container_of(dom, struct unix_domain, h);
-       ip.m_class = strdup("nfsd");
-       if (!ip.m_class)
-               return -ENOMEM;
+       strcpy(ip.m_class, "nfsd");
        ip.m_addr = addr;
        ip.m_client = udom;
        ip.m_add_change = udom->addr_changes+1;
@@ -282,7 +283,7 @@ int auth_unix_add_addr(struct in_addr addr, struct auth_domain *dom)
        ip.h.expiry_time = NEVER;
        
        ipmp = ip_map_lookup(&ip, 1);
-       if (ip.m_class) kfree(ip.m_class);
+
        if (ipmp) {
                ip_map_put(&ipmp->h, &ip_map_cache);
                return 0;
@@ -306,7 +307,7 @@ struct auth_domain *auth_unix_lookup(struct in_addr addr)
        struct ip_map key, *ipm;
        struct auth_domain *rv;
 
-       key.m_class = "nfsd";
+       strcpy(key.m_class, "nfsd");
        key.m_addr = addr;
 
        ipm = ip_map_lookup(&key, 0);
@@ -317,7 +318,8 @@ struct auth_domain *auth_unix_lookup(struct in_addr addr)
                return NULL;
 
        if ((ipm->m_client->addr_changes - ipm->m_add_change) >0) {
-               set_bit(CACHE_NEGATIVE, &ipm->h.flags);
+               if (test_and_set_bit(CACHE_NEGATIVE, &ipm->h.flags) == 0)
+                       auth_domain_put(&ipm->m_client->h);
                rv = NULL;
        } else {
                rv = &ipm->m_client->h;
@@ -333,14 +335,48 @@ void svcauth_unix_purge(void)
        cache_purge(&auth_domain_cache);
 }
 
+static int
+svcauth_unix_set_client(struct svc_rqst *rqstp)
+{
+       struct ip_map key, *ipm;
+
+       rqstp->rq_client = NULL;
+       if (rqstp->rq_proc == 0)
+               return SVC_OK;
+
+       strcpy(key.m_class, rqstp->rq_server->sv_program->pg_class);
+       key.m_addr = rqstp->rq_addr.sin_addr;
+
+       ipm = ip_map_lookup(&key, 0);
+
+       if (ipm == NULL)
+               return SVC_DENIED;
+
+       switch (cache_check(&ip_map_cache, &ipm->h, &rqstp->rq_chandle)) {
+               default:
+                       BUG();
+               case -EAGAIN:
+                       return SVC_DROP;
+               case -ENOENT:
+                       return SVC_DENIED;
+               case 0:
+                       rqstp->rq_client = &ipm->m_client->h;
+                       cache_get(&rqstp->rq_client->h);
+                       ip_map_put(&ipm->h, &ip_map_cache);
+                       break;
+       }
+       return SVC_OK;
+}
 
 static int
 svcauth_null_accept(struct svc_rqst *rqstp, u32 *authp)
 {
-       struct iovec    *argv = &rqstp->rq_arg.head[0];
-       struct iovec    *resv = &rqstp->rq_res.head[0];
-       int             rv=0;
-       struct ip_map key, *ipm;
+       struct kvec     *argv = &rqstp->rq_arg.head[0];
+       struct kvec     *resv = &rqstp->rq_res.head[0];
+       struct svc_cred *cred = &rqstp->rq_cred;
+
+       cred->cr_group_info = NULL;
+       rqstp->rq_client = NULL;
 
        if (argv->iov_len < 3*4)
                return SVC_GARBAGE;
@@ -357,45 +393,17 @@ svcauth_null_accept(struct svc_rqst *rqstp, u32 *authp)
        }
 
        /* Signal that mapping to nobody uid/gid is required */
-       rqstp->rq_cred.cr_uid = (uid_t) -1;
-       rqstp->rq_cred.cr_gid = (gid_t) -1;
-       rqstp->rq_cred.cr_group_info = groups_alloc(0);
-       if (rqstp->rq_cred.cr_group_info == NULL)
+       cred->cr_uid = (uid_t) -1;
+       cred->cr_gid = (gid_t) -1;
+       cred->cr_group_info = groups_alloc(0);
+       if (cred->cr_group_info == NULL)
                return SVC_DROP; /* kmalloc failure - client must retry */
 
        /* Put NULL verifier */
        svc_putu32(resv, RPC_AUTH_NULL);
        svc_putu32(resv, 0);
 
-       key.m_class = rqstp->rq_server->sv_program->pg_class;
-       key.m_addr = rqstp->rq_addr.sin_addr;
-
-       ipm = ip_map_lookup(&key, 0);
-
-       rqstp->rq_client = NULL;
-
-       if (ipm)
-               switch (cache_check(&ip_map_cache, &ipm->h, &rqstp->rq_chandle)) {
-               case -EAGAIN:
-                       rv = SVC_DROP;
-                       break;
-               case -ENOENT:
-                       rv = SVC_OK; /* rq_client is NULL */
-                       break;
-               case 0:
-                       rqstp->rq_client = &ipm->m_client->h;
-                       cache_get(&rqstp->rq_client->h);
-                       ip_map_put(&ipm->h, &ip_map_cache);
-                       rv = SVC_OK;
-                       break;
-               default: BUG();
-               }
-       else rv = SVC_DROP;
-
-       if (rqstp->rq_client == NULL && rqstp->rq_proc != 0)
-               *authp = rpc_autherr_badcred;
-
-       return rv;
+       return SVC_OK;
 }
 
 static int
@@ -404,6 +412,9 @@ svcauth_null_release(struct svc_rqst *rqstp)
        if (rqstp->rq_client)
                auth_domain_put(rqstp->rq_client);
        rqstp->rq_client = NULL;
+       if (rqstp->rq_cred.cr_group_info)
+               put_group_info(rqstp->rq_cred.cr_group_info);
+       rqstp->rq_cred.cr_group_info = NULL;
 
        return 0; /* don't drop */
 }
@@ -411,22 +422,22 @@ svcauth_null_release(struct svc_rqst *rqstp)
 
 struct auth_ops svcauth_null = {
        .name           = "null",
+       .owner          = THIS_MODULE,
        .flavour        = RPC_AUTH_NULL,
        .accept         = svcauth_null_accept,
        .release        = svcauth_null_release,
+       .set_client     = svcauth_unix_set_client,
 };
 
 
-int
+static int
 svcauth_unix_accept(struct svc_rqst *rqstp, u32 *authp)
 {
-       struct iovec    *argv = &rqstp->rq_arg.head[0];
-       struct iovec    *resv = &rqstp->rq_res.head[0];
+       struct kvec     *argv = &rqstp->rq_arg.head[0];
+       struct kvec     *resv = &rqstp->rq_res.head[0];
        struct svc_cred *cred = &rqstp->rq_cred;
        u32             slen, i;
        int             len   = argv->iov_len;
-       int             rv=0;
-       struct ip_map key, *ipm;
 
        cred->cr_group_info = NULL;
        rqstp->rq_client = NULL;
@@ -458,46 +469,18 @@ svcauth_unix_accept(struct svc_rqst *rqstp, u32 *authp)
                return SVC_DENIED;
        }
 
-
-       key.m_class = rqstp->rq_server->sv_program->pg_class;
-       key.m_addr = rqstp->rq_addr.sin_addr;
-
-
-       ipm = ip_map_lookup(&key, 0);
-
-       if (ipm)
-               switch (cache_check(&ip_map_cache, &ipm->h, &rqstp->rq_chandle)) {
-               case -EAGAIN:
-                       rv = SVC_DROP;
-                       break;
-               case -ENOENT:
-                       rv = SVC_OK; /* rq_client is NULL */
-                       break;
-               case 0:
-                       rqstp->rq_client = &ipm->m_client->h;
-                       cache_get(&rqstp->rq_client->h);
-                       ip_map_put(&ipm->h, &ip_map_cache);
-                       rv = SVC_OK;
-                       break;
-               default: BUG();
-               }
-       else rv = SVC_DROP;
-
-       if (rv  == SVC_OK && rqstp->rq_client == NULL && rqstp->rq_proc != 0)
-               goto badcred;
-
        /* Put NULL verifier */
        svc_putu32(resv, RPC_AUTH_NULL);
        svc_putu32(resv, 0);
 
-       return rv;
+       return SVC_OK;
 
 badcred:
        *authp = rpc_autherr_badcred;
        return SVC_DENIED;
 }
 
-int
+static int
 svcauth_unix_release(struct svc_rqst *rqstp)
 {
        /* Verifier (such as it is) is already in place.
@@ -515,9 +498,11 @@ svcauth_unix_release(struct svc_rqst *rqstp)
 
 struct auth_ops svcauth_unix = {
        .name           = "unix",
+       .owner          = THIS_MODULE,
        .flavour        = RPC_AUTH_UNIX,
        .accept         = svcauth_unix_accept,
        .release        = svcauth_unix_release,
        .domain_release = svcauth_unix_domain_release,
+       .set_client     = svcauth_unix_set_client,
 };