From: Björn Töpel <bjorn.to...@intel.com>

Introduce refcount_t, in favor of atomic_t.

Signed-off-by: Björn Töpel <bjorn.to...@intel.com>
---
 net/xdp/xdp_umem.c | 6 +++---
 net/xdp/xdp_umem.h | 2 +-
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/net/xdp/xdp_umem.c b/net/xdp/xdp_umem.c
index faa6ffbaf6ab..87998818116f 100644
--- a/net/xdp/xdp_umem.c
+++ b/net/xdp/xdp_umem.c
@@ -78,7 +78,7 @@ static void xdp_umem_release_deferred(struct work_struct 
*work)
 
 void xdp_get_umem(struct xdp_umem *umem)
 {
-       atomic_inc(&umem->users);
+       refcount_inc(&umem->users);
 }
 
 void xdp_put_umem(struct xdp_umem *umem)
@@ -86,7 +86,7 @@ void xdp_put_umem(struct xdp_umem *umem)
        if (!umem)
                return;
 
-       if (atomic_dec_and_test(&umem->users)) {
+       if (refcount_dec_and_test(&umem->users)) {
                INIT_WORK(&umem->work, xdp_umem_release_deferred);
                schedule_work(&umem->work);
        }
@@ -206,7 +206,7 @@ static int xdp_umem_reg(struct xdp_umem *umem, struct 
xdp_umem_reg *mr)
        umem->frame_size_log2 = ilog2(frame_size);
        umem->nfpp_mask = nfpp - 1;
        umem->nfpplog2 = ilog2(nfpp);
-       atomic_set(&umem->users, 1);
+       refcount_set(&umem->users, 1);
 
        err = xdp_umem_account_pages(umem);
        if (err)
diff --git a/net/xdp/xdp_umem.h b/net/xdp/xdp_umem.h
index 9802287ff19d..0881cf456230 100644
--- a/net/xdp/xdp_umem.h
+++ b/net/xdp/xdp_umem.h
@@ -27,7 +27,7 @@ struct xdp_umem {
        struct pid *pid;
        unsigned long address;
        size_t size;
-       atomic_t users;
+       refcount_t users;
        struct work_struct work;
 };
 
-- 
2.14.1

Reply via email to