devres: add devm_alloc_percpu()

[mirror_ubuntu-artful-kernel.git] / drivers / base / devres.c

diff --git a/drivers/base/devres.c b/drivers/base/devres.c
index 8fc654f0807bff66c2473b88010e67fe5ab0cb36..71d577025285bb816a46973877099fea7747e5d4 100644 (file)
--- 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);