From: Ben Widawsky <ben.widawsky@intel.com>
To: Xiyu Yang <xiyuyang19@fudan.edu.cn>
Cc: Andrew Morton <akpm@linux-foundation.org>,
Feng Tang <feng.tang@intel.com>,
Mike Kravetz <mike.kravetz@oracle.com>,
Muchun Song <songmuchun@bytedance.com>,
Yanfei Xu <yanfei.xu@windriver.com>,
linux-kernel@vger.kernel.org, linux-mm@kvack.org,
yuanxzhang@fudan.edu.cn, Xin Tan <tanxin.ctf@gmail.com>
Subject: Re: [PATCH] mm/mempolicy: Convert from atomic_t to refcount_t on mempolicy->refcnt
Date: Tue, 20 Jul 2021 08:37:00 -0700 [thread overview]
Message-ID: <20210720153700.tqq4d7fronda42ro@intel.com> (raw)
In-Reply-To: <1626683671-64407-1-git-send-email-xiyuyang19@fudan.edu.cn>
On 21-07-19 16:34:28, Xiyu Yang wrote:
> refcount_t type and corresponding API can protect refcounters from
> accidental underflow and overflow and further use-after-free situations.
>
> Signed-off-by: Xiyu Yang <xiyuyang19@fudan.edu.cn>
> Signed-off-by: Xin Tan <tanxin.ctf@gmail.com>
lgtm
Acked-by: Ben Widawsky <ben.widawsky@intel.com>
> ---
> include/linux/mempolicy.h | 5 +++--
> mm/mempolicy.c | 8 ++++----
> 2 files changed, 7 insertions(+), 6 deletions(-)
>
> diff --git a/include/linux/mempolicy.h b/include/linux/mempolicy.h
> index 0aaf91b496e2..faec94994eb7 100644
> --- a/include/linux/mempolicy.h
> +++ b/include/linux/mempolicy.h
> @@ -6,6 +6,7 @@
> #ifndef _LINUX_MEMPOLICY_H
> #define _LINUX_MEMPOLICY_H 1
>
> +#include <linux/refcount.h>
> #include <linux/sched.h>
> #include <linux/mmzone.h>
> #include <linux/dax.h>
> @@ -43,7 +44,7 @@ struct mm_struct;
> * to 1, representing the caller of mpol_dup().
> */
> struct mempolicy {
> - atomic_t refcnt;
> + refcount_t refcnt;
> unsigned short mode; /* See MPOL_* above */
> unsigned short flags; /* See set_mempolicy() MPOL_F_* above */
> nodemask_t nodes; /* interleave/bind/perfer */
> @@ -94,7 +95,7 @@ static inline struct mempolicy *mpol_dup(struct mempolicy *pol)
> static inline void mpol_get(struct mempolicy *pol)
> {
> if (pol)
> - atomic_inc(&pol->refcnt);
> + refcount_inc(&pol->refcnt);
> }
>
> extern bool __mpol_equal(struct mempolicy *a, struct mempolicy *b);
> diff --git a/mm/mempolicy.c b/mm/mempolicy.c
> index e32360e90274..829a14f34b43 100644
> --- a/mm/mempolicy.c
> +++ b/mm/mempolicy.c
> @@ -298,7 +298,7 @@ static struct mempolicy *mpol_new(unsigned short mode, unsigned short flags,
> policy = kmem_cache_alloc(policy_cache, GFP_KERNEL);
> if (!policy)
> return ERR_PTR(-ENOMEM);
> - atomic_set(&policy->refcnt, 1);
> + refcount_set(&policy->refcnt, 1);
> policy->mode = mode;
> policy->flags = flags;
>
> @@ -308,7 +308,7 @@ static struct mempolicy *mpol_new(unsigned short mode, unsigned short flags,
> /* Slow path of a mpol destructor. */
> void __mpol_put(struct mempolicy *p)
> {
> - if (!atomic_dec_and_test(&p->refcnt))
> + if (!refcount_dec_and_test(&p->refcnt))
> return;
> kmem_cache_free(policy_cache, p);
> }
> @@ -2290,7 +2290,7 @@ struct mempolicy *__mpol_dup(struct mempolicy *old)
> nodemask_t mems = cpuset_mems_allowed(current);
> mpol_rebind_policy(new, &mems);
> }
> - atomic_set(&new->refcnt, 1);
> + refcount_set(&new->refcnt, 1);
> return new;
> }
>
> @@ -2581,7 +2581,7 @@ static int shared_policy_replace(struct shared_policy *sp, unsigned long start,
> goto alloc_new;
>
> *mpol_new = *n->policy;
> - atomic_set(&mpol_new->refcnt, 1);
> + refcount_set(&mpol_new->refcnt, 1);
> sp_node_init(n_new, end, n->end, mpol_new);
> n->end = start;
> sp_insert(sp, n_new);
> --
> 2.7.4
>
next prev parent reply other threads:[~2021-07-20 15:37 UTC|newest]
Thread overview: 3+ messages / expand[flat|nested] mbox.gz Atom feed top
2021-07-19 8:34 Xiyu Yang
2021-07-20 15:37 ` Ben Widawsky [this message]
2021-07-21 6:42 ` Muchun Song
Reply instructions:
You may reply publicly to this message via plain-text email
using any one of the following methods:
* Save the following mbox file, import it into your mail client,
and reply-to-all from there: mbox
Avoid top-posting and favor interleaved quoting:
https://en.wikipedia.org/wiki/Posting_style#Interleaved_style
* Reply using the --to, --cc, and --in-reply-to
switches of git-send-email(1):
git send-email \
--in-reply-to=20210720153700.tqq4d7fronda42ro@intel.com \
--to=ben.widawsky@intel.com \
--cc=akpm@linux-foundation.org \
--cc=feng.tang@intel.com \
--cc=linux-kernel@vger.kernel.org \
--cc=linux-mm@kvack.org \
--cc=mike.kravetz@oracle.com \
--cc=songmuchun@bytedance.com \
--cc=tanxin.ctf@gmail.com \
--cc=xiyuyang19@fudan.edu.cn \
--cc=yanfei.xu@windriver.com \
--cc=yuanxzhang@fudan.edu.cn \
/path/to/YOUR_REPLY
https://kernel.org/pub/software/scm/git/docs/git-send-email.html
* If your mail client supports setting the In-Reply-To header
via mailto: links, try the mailto: link
Be sure your reply has a Subject: header at the top and a blank line
before the message body.
This is a public inbox, see mirroring instructions
for how to clone and mirror all data and code used for this inbox