nfsd4: define ->op_release for compound ops

Run a separate ->op_release function if necessary instead of depending
on the xdr encoder to do this.

Signed-off-by: J. Bruce Fields <bfields@redhat.com>

3 files changed, 36 insertions, 6 deletions

diff --git a/fs/nfsd/nfs4proc.c b/fs/nfsd/nfs4proc.c
index 01d7f2456f62..ac53ae2435c5 100644
--- a/fs/nfsd/nfs4proc.c
+++ b/fs/nfsd/nfs4proc.c
@@ -784,6 +784,14 @@ out:
         return status;
 }
 
+
+static void
+nfsd4_read_release(union nfsd4_op_u *u)
+{
+        if (u->read.rd_filp)
+                fput(u->read.rd_filp);
+}
+
 static __be32
 nfsd4_readdir(struct svc_rqst *rqstp, struct nfsd4_compound_state *cstate,
               union nfsd4_op_u *u)
@@ -912,6 +920,13 @@ nfsd4_secinfo_no_name(struct svc_rqst *rqstp, struct nfsd4_compound_state *cstat
         return nfs_ok;
 }
 
+static void
+nfsd4_secinfo_release(union nfsd4_op_u *u)
+{
+        if (u->secinfo.si_exp)
+                exp_put(u->secinfo.si_exp);
+}
+
 static __be32
 nfsd4_setattr(struct svc_rqst *rqstp, struct nfsd4_compound_state *cstate,
               union nfsd4_op_u *u)
@@ -1335,6 +1350,12 @@ out:
         return nfserr;
 }
 
+static void
+nfsd4_getdeviceinfo_release(union nfsd4_op_u *u)
+{
+        kfree(u->getdeviceinfo.gd_device);
+}
+
 static __be32
 nfsd4_layoutget(struct svc_rqst *rqstp,
                 struct nfsd4_compound_state *cstate, union nfsd4_op_u *u)
@@ -1415,6 +1436,12 @@ out:
         return nfserr;
 }
 
+static void
+nfsd4_layoutget_release(union nfsd4_op_u *u)
+{
+        kfree(u->layoutget.lg_content);
+}
+
 static __be32
 nfsd4_layoutcommit(struct svc_rqst *rqstp,
                 struct nfsd4_compound_state *cstate, union nfsd4_op_u *u)
@@ -2192,6 +2219,7 @@ static const struct nfsd4_operation nfsd4_ops[] = {
         },
         [OP_READ] = {
                 .op_func = nfsd4_read,
+                .op_release = nfsd4_read_release,
                 .op_name = "OP_READ",
                 .op_rsize_bop = nfsd4_read_rsize,
                 .op_get_currentstateid = nfsd4_get_readstateid,
@@ -2241,6 +2269,7 @@ static const struct nfsd4_operation nfsd4_ops[] = {
         },
         [OP_SECINFO] = {
                 .op_func = nfsd4_secinfo,
+                .op_release = nfsd4_secinfo_release,
                 .op_flags = OP_HANDLES_WRONGSEC,
                 .op_name = "OP_SECINFO",
                 .op_rsize_bop = nfsd4_secinfo_rsize,
@@ -2342,6 +2371,7 @@ static const struct nfsd4_operation nfsd4_ops[] = {
         },
         [OP_SECINFO_NO_NAME] = {
                 .op_func = nfsd4_secinfo_no_name,
+                .op_release = nfsd4_secinfo_release,
                 .op_flags = OP_HANDLES_WRONGSEC,
                 .op_name = "OP_SECINFO_NO_NAME",
                 .op_rsize_bop = nfsd4_secinfo_rsize,
@@ -2362,12 +2392,14 @@ static const struct nfsd4_operation nfsd4_ops[] = {
 #ifdef CONFIG_NFSD_PNFS
         [OP_GETDEVICEINFO] = {
                 .op_func = nfsd4_getdeviceinfo,
+                .op_release = nfsd4_getdeviceinfo_release,
                 .op_flags = ALLOWED_WITHOUT_FH,
                 .op_name = "OP_GETDEVICEINFO",
                 .op_rsize_bop = nfsd4_getdeviceinfo_rsize,
         },
         [OP_LAYOUTGET] = {
                 .op_func = nfsd4_layoutget,
+                .op_release = nfsd4_layoutget_release,
                 .op_flags = OP_MODIFIES_SOMETHING,
                 .op_name = "OP_LAYOUTGET",
                 .op_rsize_bop = nfsd4_layoutget_rsize,
diff --git a/fs/nfsd/nfs4xdr.c b/fs/nfsd/nfs4xdr.c
index 40ed23fda814..7d683e3aebf0 100644
--- a/fs/nfsd/nfs4xdr.c
+++ b/fs/nfsd/nfs4xdr.c
@@ -3593,8 +3593,6 @@ nfsd4_encode_read(struct nfsd4_compoundres *resp, __be32 nfserr,
                 xdr_truncate_encode(xdr, starting_len);
 
 out:
-        if (file)
-                fput(file);
         return nfserr;
 }
 
@@ -3838,8 +3836,6 @@ nfsd4_do_encode_secinfo(struct xdr_stream *xdr,
         *flavorsp = htonl(supported);
         nfserr = 0;
 out:
-        if (exp)
-                exp_put(exp);
         return nfserr;
 }
 
@@ -4172,7 +4168,6 @@ nfsd4_encode_getdeviceinfo(struct nfsd4_compoundres *resp, __be32 nfserr,
 
         nfserr = 0;
 out:
-        kfree(gdev->gd_device);
         dprintk("%s: done: %d\n", __func__, be32_to_cpu(nfserr));
         return nfserr;
 
@@ -4221,7 +4216,6 @@ nfsd4_encode_layoutget(struct nfsd4_compoundres *resp, __be32 nfserr,
         ops = nfsd4_layout_ops[lgp->lg_layout_type];
         nfserr = ops->encode_layoutget(xdr, lgp);
 out:
-        kfree(lgp->lg_content);
         return nfserr;
 }
 
@@ -4452,6 +4446,7 @@ nfsd4_encode_operation(struct nfsd4_compoundres *resp, struct nfsd4_op *op)
         struct xdr_stream *xdr = &resp->xdr;
         struct nfs4_stateowner *so = resp->cstate.replay_owner;
         struct svc_rqst *rqstp = resp->rqstp;
+        const struct nfsd4_operation *opdesc = op->opdesc;
         int post_err_offset;
         nfsd4_enc encoder;
         __be32 *p;
@@ -4470,6 +4465,8 @@ nfsd4_encode_operation(struct nfsd4_compoundres *resp, struct nfsd4_op *op)
                !nfsd4_enc_ops[op->opnum]);
         encoder = nfsd4_enc_ops[op->opnum];
         op->status = encoder(resp, op->status, &op->u);
+        if (opdesc && opdesc->op_release)
+                opdesc->op_release(&op->u);
         xdr_commit_encode(xdr);
 
         /* nfsd4_check_resp_size guarantees enough room for error status */
diff --git a/fs/nfsd/xdr4.h b/fs/nfsd/xdr4.h
index 90b928006bc7..1e6274e0e066 100644
--- a/fs/nfsd/xdr4.h
+++ b/fs/nfsd/xdr4.h
@@ -783,6 +783,7 @@ enum nfsd4_op_flags {
 struct nfsd4_operation {
         __be32 (*op_func)(struct svc_rqst *, struct nfsd4_compound_state *,
                         union nfsd4_op_u *);
+        void (*op_release)(union nfsd4_op_u *);
         u32 op_flags;
         char *op_name;
         /* Try to get response size before operation */