fedora core 6 1.2949 + vserver 2.2.0
[linux-2.6.git] / net / sunrpc / svcauth_unix.c
index 3e6c694..a568f9f 100644 (file)
@@ -9,6 +9,7 @@
 #include <linux/seq_file.h>
 #include <linux/hash.h>
 #include <linux/string.h>
+#include <net/sock.h>
 
 #define RPCDBG_FACILITY        RPCDBG_AUTH
 
@@ -27,41 +28,39 @@ struct unix_domain {
        /* other stuff later */
 };
 
+extern struct auth_ops svcauth_unix;
+
 struct auth_domain *unix_domain_find(char *name)
 {
-       struct auth_domain *rv, ud;
-       struct unix_domain *new;
-
-       ud.name = name;
-       
-       rv = auth_domain_lookup(&ud, 0);
-
- foundit:
-       if (rv && rv->flavour != RPC_AUTH_UNIX) {
-               auth_domain_put(rv);
-               return NULL;
-       }
-       if (rv)
-               return rv;
-
-       new = kmalloc(sizeof(*new), GFP_KERNEL);
-       if (new == NULL)
-               return NULL;
-       cache_init(&new->h.h);
-       new->h.name = kstrdup(name, GFP_KERNEL);
-       new->h.flavour = RPC_AUTH_UNIX;
-       new->addr_changes = 0;
-       new->h.h.expiry_time = NEVER;
-
-       rv = auth_domain_lookup(&new->h, 2);
-       if (rv == &new->h) {
-               if (atomic_dec_and_test(&new->h.h.refcnt)) BUG();
-       } else {
-               auth_domain_put(&new->h);
-               goto foundit;
+       struct auth_domain *rv;
+       struct unix_domain *new = NULL;
+
+       rv = auth_domain_lookup(name, NULL);
+       while(1) {
+               if (rv) {
+                       if (new && rv != &new->h)
+                               auth_domain_put(&new->h);
+
+                       if (rv->flavour != &svcauth_unix) {
+                               auth_domain_put(rv);
+                               return NULL;
+                       }
+                       return rv;
+               }
+
+               new = kmalloc(sizeof(*new), GFP_KERNEL);
+               if (new == NULL)
+                       return NULL;
+               kref_init(&new->h.ref);
+               new->h.name = kstrdup(name, GFP_KERNEL);
+               if (new->h.name == NULL) {
+                       kfree(new);
+                       return NULL;
+               }
+               new->h.flavour = &svcauth_unix;
+               new->addr_changes = 0;
+               rv = auth_domain_lookup(name, &new->h);
        }
-
-       return rv;
 }
 
 static void svcauth_unix_domain_release(struct auth_domain *dom)
@@ -90,15 +89,15 @@ struct ip_map {
 };
 static struct cache_head       *ip_table[IP_HASHMAX];
 
-static void ip_map_put(struct cache_head *item, struct cache_detail *cd)
+static void ip_map_put(struct kref *kref)
 {
+       struct cache_head *item = container_of(kref, struct cache_head, ref);
        struct ip_map *im = container_of(item, struct ip_map,h);
-       if (cache_put(item, cd)) {
-               if (test_bit(CACHE_VALID, &item->flags) &&
-                   !test_bit(CACHE_NEGATIVE, &item->flags))
-                       auth_domain_put(&im->m_client->h);
-               kfree(im);
-       }
+
+       if (test_bit(CACHE_VALID, &item->flags) &&
+           !test_bit(CACHE_NEGATIVE, &item->flags))
+               auth_domain_put(&im->m_client->h);
+       kfree(im);
 }
 
 #if IP_HASHBITS == 8
@@ -106,34 +105,44 @@ static void ip_map_put(struct cache_head *item, struct cache_detail *cd)
  * 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)
+static inline int hash_ip(__be32 ip)
 {
-       int hash = ip ^ (ip>>16);
+       int hash = (__force u32)ip ^ ((__force u32)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_ip((unsigned long)item->m_addr.s_addr);
-}
-static inline int ip_map_match(struct ip_map *item, struct ip_map *tmp)
+static int ip_map_match(struct cache_head *corig, struct cache_head *cnew)
 {
-       return strcmp(tmp->m_class, item->m_class) == 0
-               && tmp->m_addr.s_addr == item->m_addr.s_addr;
+       struct ip_map *orig = container_of(corig, struct ip_map, h);
+       struct ip_map *new = container_of(cnew, struct ip_map, h);
+       return strcmp(orig->m_class, new->m_class) == 0
+               && orig->m_addr.s_addr == new->m_addr.s_addr;
 }
-static inline void ip_map_init(struct ip_map *new, struct ip_map *item)
+static void ip_map_init(struct cache_head *cnew, struct cache_head *citem)
 {
+       struct ip_map *new = container_of(cnew, struct ip_map, h);
+       struct ip_map *item = container_of(citem, struct ip_map, h);
+
        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)
+static void update(struct cache_head *cnew, struct cache_head *citem)
 {
-       cache_get(&item->m_client->h.h);
+       struct ip_map *new = container_of(cnew, struct ip_map, h);
+       struct ip_map *item = container_of(citem, struct ip_map, h);
+
+       kref_get(&item->m_client->h.ref);
        new->m_client = item->m_client;
        new->m_add_change = item->m_add_change;
 }
+static struct cache_head *ip_map_alloc(void)
+{
+       struct ip_map *i = kmalloc(sizeof(*i), GFP_KERNEL);
+       if (i)
+               return &i->h;
+       else
+               return NULL;
+}
 
 static void ip_map_request(struct cache_detail *cd,
                                  struct cache_head *h,
@@ -141,7 +150,7 @@ static void ip_map_request(struct cache_detail *cd,
 {
        char text_addr[20];
        struct ip_map *im = container_of(h, struct ip_map, h);
-       __u32 addr = im->m_addr.s_addr;
+       __be32 addr = im->m_addr.s_addr;
        
        snprintf(text_addr, 20, "%u.%u.%u.%u",
                 ntohl(addr) >> 24 & 0xff,
@@ -154,7 +163,8 @@ static void ip_map_request(struct cache_detail *cd,
        (*bpp)[-1] = '\n';
 }
 
-static struct ip_map *ip_map_lookup(struct ip_map *, int);
+static struct ip_map *ip_map_lookup(char *class, struct in_addr addr);
+static int ip_map_update(struct ip_map *ipm, struct unix_domain *udom, time_t expiry);
 
 static int ip_map_parse(struct cache_detail *cd,
                          char *mesg, int mlen)
@@ -166,7 +176,11 @@ static int ip_map_parse(struct cache_detail *cd,
        int len;
        int b1,b2,b3,b4;
        char c;
-       struct ip_map ipm, *ipmp;
+       char class[8];
+       struct in_addr addr;
+       int err;
+
+       struct ip_map *ipmp;
        struct auth_domain *dom;
        time_t expiry;
 
@@ -175,7 +189,7 @@ static int ip_map_parse(struct cache_detail *cd,
        mesg[mlen-1] = 0;
 
        /* class */
-       len = qword_get(&mesg, ipm.m_class, sizeof(ipm.m_class));
+       len = qword_get(&mesg, class, sizeof(class));
        if (len <= 0) return -EINVAL;
 
        /* ip address */
@@ -200,25 +214,22 @@ static int ip_map_parse(struct cache_detail *cd,
        } else
                dom = NULL;
 
-       ipm.m_addr.s_addr =
+       addr.s_addr =
                htonl((((((b1<<8)|b2)<<8)|b3)<<8)|b4);
-       ipm.h.flags = 0;
-       if (dom) {
-               ipm.m_client = container_of(dom, struct unix_domain, h);
-               ipm.m_add_change = ipm.m_client->addr_changes;
+
+       ipmp = ip_map_lookup(class,addr);
+       if (ipmp) {
+               err = ip_map_update(ipmp,
+                            container_of(dom, struct unix_domain, h),
+                            expiry);
        } else
-               set_bit(CACHE_NEGATIVE, &ipm.h.flags);
-       ipm.h.expiry_time = expiry;
+               err = -ENOMEM;
 
-       ipmp = ip_map_lookup(&ipm, 1);
-       if (ipmp)
-               ip_map_put(&ipmp->h, &ip_map_cache);
        if (dom)
                auth_domain_put(dom);
-       if (!ipmp)
-               return -ENOMEM;
+
        cache_flush();
-       return 0;
+       return err;
 }
 
 static int ip_map_show(struct seq_file *m,
@@ -243,10 +254,10 @@ static int ip_map_show(struct seq_file *m,
 
        seq_printf(m, "%s %d.%d.%d.%d %s\n",
                   im->m_class,
-                  htonl(addr.s_addr) >> 24 & 0xff,
-                  htonl(addr.s_addr) >> 16 & 0xff,
-                  htonl(addr.s_addr) >>  8 & 0xff,
-                  htonl(addr.s_addr) >>  0 & 0xff,
+                  ntohl(addr.s_addr) >> 24 & 0xff,
+                  ntohl(addr.s_addr) >> 16 & 0xff,
+                  ntohl(addr.s_addr) >>  8 & 0xff,
+                  ntohl(addr.s_addr) >>  0 & 0xff,
                   dom
                   );
        return 0;
@@ -262,32 +273,70 @@ struct cache_detail ip_map_cache = {
        .cache_request  = ip_map_request,
        .cache_parse    = ip_map_parse,
        .cache_show     = ip_map_show,
+       .match          = ip_map_match,
+       .init           = ip_map_init,
+       .update         = update,
+       .alloc          = ip_map_alloc,
 };
 
-static DefineSimpleCacheLookup(ip_map, 0)
+static struct ip_map *ip_map_lookup(char *class, struct in_addr addr)
+{
+       struct ip_map ip;
+       struct cache_head *ch;
+
+       strcpy(ip.m_class, class);
+       ip.m_addr = addr;
+       ch = sunrpc_cache_lookup(&ip_map_cache, &ip.h,
+                                hash_str(class, IP_HASHBITS) ^
+                                hash_ip(addr.s_addr));
+
+       if (ch)
+               return container_of(ch, struct ip_map, h);
+       else
+               return NULL;
+}
 
+static int ip_map_update(struct ip_map *ipm, struct unix_domain *udom, time_t expiry)
+{
+       struct ip_map ip;
+       struct cache_head *ch;
+
+       ip.m_client = udom;
+       ip.h.flags = 0;
+       if (!udom)
+               set_bit(CACHE_NEGATIVE, &ip.h.flags);
+       else {
+               ip.m_add_change = udom->addr_changes;
+               /* if this is from the legacy set_client system call,
+                * we need m_add_change to be one higher
+                */
+               if (expiry == NEVER)
+                       ip.m_add_change++;
+       }
+       ip.h.expiry_time = expiry;
+       ch = sunrpc_cache_update(&ip_map_cache,
+                                &ip.h, &ipm->h,
+                                hash_str(ipm->m_class, IP_HASHBITS) ^
+                                hash_ip(ipm->m_addr.s_addr));
+       if (!ch)
+               return -ENOMEM;
+       cache_put(ch, &ip_map_cache);
+       return 0;
+}
 
 int auth_unix_add_addr(struct in_addr addr, struct auth_domain *dom)
 {
        struct unix_domain *udom;
-       struct ip_map ip, *ipmp;
+       struct ip_map *ipmp;
 
-       if (dom->flavour != RPC_AUTH_UNIX)
+       if (dom->flavour != &svcauth_unix)
                return -EINVAL;
        udom = container_of(dom, struct unix_domain, h);
-       strcpy(ip.m_class, "nfsd");
-       ip.m_addr = addr;
-       ip.m_client = udom;
-       ip.m_add_change = udom->addr_changes+1;
-       ip.h.flags = 0;
-       ip.h.expiry_time = NEVER;
-       
-       ipmp = ip_map_lookup(&ip, 1);
+       ipmp = ip_map_lookup("nfsd", addr);
 
-       if (ipmp) {
-               ip_map_put(&ipmp->h, &ip_map_cache);
-               return 0;
-       } else
+       if (ipmp)
+               return ip_map_update(ipmp, udom, NEVER);
+       else
                return -ENOMEM;
 }
 
@@ -295,7 +344,7 @@ int auth_unix_forget_old(struct auth_domain *dom)
 {
        struct unix_domain *udom;
        
-       if (dom->flavour != RPC_AUTH_UNIX)
+       if (dom->flavour != &svcauth_unix)
                return -EINVAL;
        udom = container_of(dom, struct unix_domain, h);
        udom->addr_changes++;
@@ -304,13 +353,10 @@ int auth_unix_forget_old(struct auth_domain *dom)
 
 struct auth_domain *auth_unix_lookup(struct in_addr addr)
 {
-       struct ip_map key, *ipm;
+       struct ip_map *ipm;
        struct auth_domain *rv;
 
-       strcpy(key.m_class, "nfsd");
-       key.m_addr = addr;
-
-       ipm = ip_map_lookup(&key, 0);
+       ipm = ip_map_lookup("nfsd", addr);
 
        if (!ipm)
                return NULL;
@@ -323,31 +369,79 @@ struct auth_domain *auth_unix_lookup(struct in_addr addr)
                rv = NULL;
        } else {
                rv = &ipm->m_client->h;
-               cache_get(&rv->h);
+               kref_get(&rv->ref);
        }
-       ip_map_put(&ipm->h, &ip_map_cache);
+       cache_put(&ipm->h, &ip_map_cache);
        return rv;
 }
 
 void svcauth_unix_purge(void)
 {
        cache_purge(&ip_map_cache);
-       cache_purge(&auth_domain_cache);
+}
+
+static inline struct ip_map *
+ip_map_cached_get(struct svc_rqst *rqstp)
+{
+       struct ip_map *ipm;
+       struct svc_sock *svsk = rqstp->rq_sock;
+       spin_lock_bh(&svsk->sk_defer_lock);
+       ipm = svsk->sk_info_authunix;
+       if (ipm != NULL) {
+               if (!cache_valid(&ipm->h)) {
+                       /*
+                        * The entry has been invalidated since it was
+                        * remembered, e.g. by a second mount from the
+                        * same IP address.
+                        */
+                       svsk->sk_info_authunix = NULL;
+                       spin_unlock_bh(&svsk->sk_defer_lock);
+                       cache_put(&ipm->h, &ip_map_cache);
+                       return NULL;
+               }
+               cache_get(&ipm->h);
+       }
+       spin_unlock_bh(&svsk->sk_defer_lock);
+       return ipm;
+}
+
+static inline void
+ip_map_cached_put(struct svc_rqst *rqstp, struct ip_map *ipm)
+{
+       struct svc_sock *svsk = rqstp->rq_sock;
+
+       spin_lock_bh(&svsk->sk_defer_lock);
+       if (svsk->sk_sock->type == SOCK_STREAM &&
+           svsk->sk_info_authunix == NULL) {
+               /* newly cached, keep the reference */
+               svsk->sk_info_authunix = ipm;
+               ipm = NULL;
+       }
+       spin_unlock_bh(&svsk->sk_defer_lock);
+       if (ipm)
+               cache_put(&ipm->h, &ip_map_cache);
+}
+
+void
+svcauth_unix_info_release(void *info)
+{
+       struct ip_map *ipm = info;
+       cache_put(&ipm->h, &ip_map_cache);
 }
 
 static int
 svcauth_unix_set_client(struct svc_rqst *rqstp)
 {
-       struct ip_map key, *ipm;
+       struct ip_map *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);
+       ipm = ip_map_cached_get(rqstp);
+       if (ipm == NULL)
+               ipm = ip_map_lookup(rqstp->rq_server->sv_program->pg_class,
+                                   rqstp->rq_addr.sin_addr);
 
        if (ipm == NULL)
                return SVC_DENIED;
@@ -356,20 +450,21 @@ svcauth_unix_set_client(struct svc_rqst *rqstp)
                default:
                        BUG();
                case -EAGAIN:
+               case -ETIMEDOUT:
                        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);
+                       kref_get(&rqstp->rq_client->ref);
+                       ip_map_cached_put(rqstp, ipm);
                        break;
        }
        return SVC_OK;
 }
 
 static int
-svcauth_null_accept(struct svc_rqst *rqstp, u32 *authp)
+svcauth_null_accept(struct svc_rqst *rqstp, __be32 *authp)
 {
        struct kvec     *argv = &rqstp->rq_arg.head[0];
        struct kvec     *resv = &rqstp->rq_res.head[0];
@@ -386,7 +481,7 @@ svcauth_null_accept(struct svc_rqst *rqstp, u32 *authp)
                *authp = rpc_autherr_badcred;
                return SVC_DENIED;
        }
-       if (svc_getu32(argv) != RPC_AUTH_NULL || svc_getu32(argv) != 0) {
+       if (svc_getu32(argv) != htonl(RPC_AUTH_NULL) || svc_getu32(argv) != 0) {
                dprintk("svc: bad null verf\n");
                *authp = rpc_autherr_badverf;
                return SVC_DENIED;
@@ -400,8 +495,8 @@ svcauth_null_accept(struct svc_rqst *rqstp, u32 *authp)
                return SVC_DROP; /* kmalloc failure - client must retry */
 
        /* Put NULL verifier */
-       svc_putu32(resv, RPC_AUTH_NULL);
-       svc_putu32(resv, 0);
+       svc_putnl(resv, RPC_AUTH_NULL);
+       svc_putnl(resv, 0);
 
        return SVC_OK;
 }
@@ -431,7 +526,7 @@ struct auth_ops svcauth_null = {
 
 
 static int
-svcauth_unix_accept(struct svc_rqst *rqstp, u32 *authp)
+svcauth_unix_accept(struct svc_rqst *rqstp, __be32 *authp)
 {
        struct kvec     *argv = &rqstp->rq_arg.head[0];
        struct kvec     *resv = &rqstp->rq_res.head[0];
@@ -447,31 +542,31 @@ svcauth_unix_accept(struct svc_rqst *rqstp, u32 *authp)
 
        svc_getu32(argv);                       /* length */
        svc_getu32(argv);                       /* time stamp */
-       slen = XDR_QUADLEN(ntohl(svc_getu32(argv)));    /* machname length */
+       slen = XDR_QUADLEN(svc_getnl(argv));    /* machname length */
        if (slen > 64 || (len -= (slen + 3)*4) < 0)
                goto badcred;
-       argv->iov_base = (void*)((u32*)argv->iov_base + slen);  /* skip machname */
+       argv->iov_base = (void*)((__be32*)argv->iov_base + slen);       /* skip machname */
        argv->iov_len -= slen*4;
 
-       cred->cr_uid = ntohl(svc_getu32(argv));         /* uid */
-       cred->cr_gid = ntohl(svc_getu32(argv));         /* gid */
-       slen = ntohl(svc_getu32(argv));                 /* gids length */
+       cred->cr_uid = svc_getnl(argv);         /* uid */
+       cred->cr_gid = svc_getnl(argv);         /* gid */
+       slen = svc_getnl(argv);                 /* gids length */
        if (slen > 16 || (len -= (slen + 2)*4) < 0)
                goto badcred;
        cred->cr_group_info = groups_alloc(slen);
        if (cred->cr_group_info == NULL)
                return SVC_DROP;
        for (i = 0; i < slen; i++)
-               GROUP_AT(cred->cr_group_info, i) = ntohl(svc_getu32(argv));
+               GROUP_AT(cred->cr_group_info, i) = svc_getnl(argv);
 
-       if (svc_getu32(argv) != RPC_AUTH_NULL || svc_getu32(argv) != 0) {
+       if (svc_getu32(argv) != htonl(RPC_AUTH_NULL) || svc_getu32(argv) != 0) {
                *authp = rpc_autherr_badverf;
                return SVC_DENIED;
        }
 
        /* Put NULL verifier */
-       svc_putu32(resv, RPC_AUTH_NULL);
-       svc_putu32(resv, 0);
+       svc_putnl(resv, RPC_AUTH_NULL);
+       svc_putnl(resv, 0);
 
        return SVC_OK;