| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125 |
- From d33bde3c487c722541ad359e1d22090a78df0c77 Mon Sep 17 00:00:00 2001
- From: Zhao Qiang <[email protected]>
- Date: Tue, 11 Jul 2017 16:47:18 +0800
- Subject: [PATCH] add devm_alloc_percpu support
- Signed-off-by: Zhao Qiang <[email protected]>
- ---
- drivers/base/devres.c | 66 ++++++++++++++++++++++++++++++++++++++++++++++++++
- include/linux/device.h | 19 +++++++++++++++
- 2 files changed, 85 insertions(+)
- diff --git a/drivers/base/devres.c b/drivers/base/devres.c
- index 8fc654f0807b..71d577025285 100644
- --- a/drivers/base/devres.c
- +++ b/drivers/base/devres.c
- @@ -10,6 +10,7 @@
- #include <linux/device.h>
- #include <linux/module.h>
- #include <linux/slab.h>
- +#include <linux/percpu.h>
-
- #include "base.h"
-
- @@ -985,3 +986,68 @@ void devm_free_pages(struct device *dev, unsigned long addr)
- &devres));
- }
- EXPORT_SYMBOL_GPL(devm_free_pages);
- +
- +static void devm_percpu_release(struct device *dev, void *pdata)
- +{
- + void __percpu *p;
- +
- + p = *(void __percpu **)pdata;
- + free_percpu(p);
- +}
- +
- +static int devm_percpu_match(struct device *dev, void *data, void *p)
- +{
- + struct devres *devr = container_of(data, struct devres, data);
- +
- + return *(void **)devr->data == p;
- +}
- +
- +/**
- + * __devm_alloc_percpu - Resource-managed alloc_percpu
- + * @dev: Device to allocate per-cpu memory for
- + * @size: Size of per-cpu memory to allocate
- + * @align: Alignment of per-cpu memory to allocate
- + *
- + * Managed alloc_percpu. Per-cpu memory allocated with this function is
- + * automatically freed on driver detach.
- + *
- + * RETURNS:
- + * Pointer to allocated memory on success, NULL on failure.
- + */
- +void __percpu *__devm_alloc_percpu(struct device *dev, size_t size,
- + size_t align)
- +{
- + void *p;
- + void __percpu *pcpu;
- +
- + pcpu = __alloc_percpu(size, align);
- + if (!pcpu)
- + return NULL;
- +
- + p = devres_alloc(devm_percpu_release, sizeof(void *), GFP_KERNEL);
- + if (!p) {
- + free_percpu(pcpu);
- + return NULL;
- + }
- +
- + *(void __percpu **)p = pcpu;
- +
- + devres_add(dev, p);
- +
- + return pcpu;
- +}
- +EXPORT_SYMBOL_GPL(__devm_alloc_percpu);
- +
- +/**
- + * devm_free_percpu - Resource-managed free_percpu
- + * @dev: Device this memory belongs to
- + * @pdata: Per-cpu memory to free
- + *
- + * Free memory allocated with devm_alloc_percpu().
- + */
- +void devm_free_percpu(struct device *dev, void __percpu *pdata)
- +{
- + WARN_ON(devres_destroy(dev, devm_percpu_release, devm_percpu_match,
- + (void *)pdata));
- +}
- +EXPORT_SYMBOL_GPL(devm_free_percpu);
- diff --git a/include/linux/device.h b/include/linux/device.h
- index bc41e87a969b..0a2135cbddc9 100644
- --- a/include/linux/device.h
- +++ b/include/linux/device.h
- @@ -686,6 +686,25 @@ void __iomem *devm_ioremap_resource(struct device *dev, struct resource *res);
- int devm_add_action(struct device *dev, void (*action)(void *), void *data);
- void devm_remove_action(struct device *dev, void (*action)(void *), void *data);
-
- +/**
- + * devm_alloc_percpu - Resource-managed alloc_percpu
- + * @dev: Device to allocate per-cpu memory for
- + * @type: Type to allocate per-cpu memory for
- + *
- + * Managed alloc_percpu. Per-cpu memory allocated with this function is
- + * automatically freed on driver detach.
- + *
- + * RETURNS:
- + * Pointer to allocated memory on success, NULL on failure.
- + */
- +#define devm_alloc_percpu(dev, type) \
- + ((typeof(type) __percpu *)__devm_alloc_percpu((dev), sizeof(type), \
- + __alignof__(type)))
- +
- +void __percpu *__devm_alloc_percpu(struct device *dev, size_t size,
- + size_t align);
- +void devm_free_percpu(struct device *dev, void __percpu *pdata);
- +
- static inline int devm_add_action_or_reset(struct device *dev,
- void (*action)(void *), void *data)
- {
- --
- 2.11.1
|
