#include <linux/nfs_fs.h>
 #include <linux/nfs_page.h>
 #include <linux/module.h>
+#include <linux/sched/mm.h>
 
 #include <linux/sunrpc/metrics.h>
 
                struct nfs4_ff_layout_mirror *mirror;
                struct auth_cred acred = { .group_info = ff_zero_group };
                struct rpc_cred __rcu *cred;
+               struct cred *kcred;
                u32 ds_count, fh_count, id;
                int j;
 
 
                acred.gid = make_kgid(&init_user_ns, id);
 
+               if (gfp_flags & __GFP_FS)
+                       kcred = prepare_kernel_cred(NULL);
+               else {
+                       unsigned int nofs_flags = memalloc_nofs_save();
+                       kcred = prepare_kernel_cred(NULL);
+                       memalloc_nofs_restore(nofs_flags);
+               }
+               rc = -ENOMEM;
+               if (!kcred)
+                       goto out_err_free;
+               kcred->fsuid = acred.uid;
+               kcred->fsgid = acred.gid;
+               acred.cred = kcred;
+
                /* find the cred for it */
                rcu_assign_pointer(cred, rpc_lookup_generic_cred(&acred, 0, gfp_flags));
+               put_cred(kcred);
                if (IS_ERR(cred)) {
                        rc = PTR_ERR(cred);
                        goto out_err_free;
 
        } else {
                struct rpc_auth *auth = client->cl_auth;
                struct auth_cred acred = {};
+               struct cred *kcred;
+               struct rpc_cred *ret;
+
+               kcred = prepare_kernel_cred(NULL);
+               if (!kcred)
+                       return NULL;
 
                acred.uid = ses->se_cb_sec.uid;
                acred.gid = ses->se_cb_sec.gid;
-               return auth->au_ops->lookup_cred(client->cl_auth, &acred, 0);
+               kcred->uid = acred.uid;
+               kcred->gid = acred.gid;
+               acred.cred = kcred;
+               ret = auth->au_ops->lookup_cred(client->cl_auth, &acred, 0);
+               put_cred(kcred);
+               return ret;
        }
 }
 
 
 
 /* Work around the lack of a VFS credential */
 struct auth_cred {
+       const struct cred *cred;
        kuid_t  uid;
        kgid_t  gid;
        struct group_info *group_info;
        unsigned long           cr_expire;      /* when to gc */
        unsigned long           cr_flags;       /* various flags */
        refcount_t              cr_count;       /* ref count */
+       const struct cred       *cr_cred;
 
        kuid_t                  cr_uid;
 
 
        acred.uid = cred->fsuid;
        acred.gid = cred->fsgid;
        acred.group_info = cred->group_info;
+       acred.cred = cred;
        ret = auth->au_ops->lookup_cred(auth, &acred, flags);
        return ret;
 }
        cred->cr_auth = auth;
        cred->cr_ops = ops;
        cred->cr_expire = jiffies;
+       cred->cr_cred = get_cred(acred->cred);
        cred->cr_uid = acred->uid;
 }
 EXPORT_SYMBOL_GPL(rpcauth_init_cred);
        struct auth_cred acred = {
                .uid = GLOBAL_ROOT_UID,
                .gid = GLOBAL_ROOT_GID,
+               .cred = get_task_cred(&init_task),
        };
+       struct rpc_cred *ret;
 
        dprintk("RPC: %5u looking up %s cred\n",
                task->tk_pid, task->tk_client->cl_auth->au_ops->au_name);
-       return auth->au_ops->lookup_cred(auth, &acred, lookupflags);
+       ret = auth->au_ops->lookup_cred(auth, &acred, lookupflags);
+       put_cred(acred.cred);
+       return ret;
 }
 
 static struct rpc_cred *
 
                .gid = RPC_MACHINE_CRED_GROUPID,
                .principal = service_name,
                .machine_cred = 1,
+               .cred = get_task_cred(&init_task),
        };
+       struct rpc_cred *ret;
 
        dprintk("RPC:       looking up machine cred for service %s\n",
                        service_name);
-       return generic_auth.au_ops->lookup_cred(&generic_auth, &acred, 0);
+       ret = generic_auth.au_ops->lookup_cred(&generic_auth, &acred, 0);
+       put_cred(acred.cred);
+       return ret;
 }
 EXPORT_SYMBOL_GPL(rpc_lookup_machine_cred);
 
        gcred->acred.uid = acred->uid;
        gcred->acred.gid = acred->gid;
        gcred->acred.group_info = acred->group_info;
+       gcred->acred.cred = gcred->gc_base.cr_cred;
        gcred->acred.ac_flags = 0;
        if (gcred->acred.group_info != NULL)
                get_group_info(gcred->acred.group_info);
        dprintk("RPC:       generic_free_cred %p\n", gcred);
        if (gcred->acred.group_info != NULL)
                put_group_info(gcred->acred.group_info);
+       put_cred(cred->cr_cred);
        kfree(gcred);
 }
 
 
        struct gss_cl_ctx *ctx = rcu_dereference_protected(gss_cred->gc_ctx, 1);
 
        RCU_INIT_POINTER(gss_cred->gc_ctx, NULL);
+       put_cred(cred->cr_cred);
        call_rcu(&cred->cr_rcu, gss_free_cred_callback);
        if (ctx)
                gss_put_ctx(ctx);
        struct rpc_auth *auth = oldcred->cr_auth;
        struct auth_cred acred = {
                .uid = oldcred->cr_uid,
+               .cred = oldcred->cr_cred,
                .principal = gss_cred->gc_principal,
                .machine_cred = (gss_cred->gc_principal != NULL ? 1 : 0),
        };
 
 unx_free_cred(struct unx_cred *unx_cred)
 {
        dprintk("RPC:       unx_free_cred %p\n", unx_cred);
+       put_cred(unx_cred->uc_base.cr_cred);
        kfree(unx_cred);
 }