@@ -10,3 +10,8 @@ config ION_TEGRA
help
Choose this option if you wish to use ion on an nVidia Tegra.
+config ION_CMA
+ tristate "Ion CMA heap"
+ depends on ION && CMA
+ help
+ Choose this option to enable ION CMA heap.
@@ -1,2 +1,3 @@
obj-$(CONFIG_ION) += ion.o ion_heap.o ion_system_heap.o ion_carveout_heap.o
obj-$(CONFIG_ION_TEGRA) += tegra/
+obj-$(CONFIG_ION_CMA) += cma/
new file mode 100644
@@ -0,0 +1 @@
+obj-y += ion_cma_heap.o
new file mode 100644
@@ -0,0 +1,217 @@
+/*
+ * drivers/gpu/ion/ion_cma_heap.c
+ *
+ *
+ * This software is licensed under the terms of the GNU General Public
+ * License version 2, as published by the Free Software Foundation, and
+ * may be copied, distributed, and modified under those terms.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ */
+
+#include <linux/device.h>
+#include <linux/ion.h>
+#include <linux/platform_device.h>
+#include <linux/slab.h>
+#include <linux/errno.h>
+#include <linux/err.h>
+#include <linux/miscdevice.h>
+#include <linux/dma-mapping.h>
+
+/* for ion_heap_ops structure */
+#include "../ion_priv.h"
+
+#define ION_CMA_ALLOCATE_FAILED -1
+
+static u64 dma_mask = DMA_BIT_MASK(32);
+
+struct ion_heap **heaps;
+struct ion_device *ion_cma_device;
+
+struct ion_cma_buffer_info {
+ void *cpu_addr;
+ dma_addr_t handle;
+};
+
+/* ION CMA heap operations functions */
+static int ion_cma_allocate(struct ion_heap *heap, struct ion_buffer *buffer,
+ unsigned long len, unsigned long align,
+ unsigned long flags)
+{
+ struct ion_device *idev = heap->dev;
+ struct device *dev = idev->dev.parent;
+ struct ion_cma_buffer_info *info;
+
+ dev_dbg(dev, "Request buffer allocation len %ld\n", len);
+
+ info = kzalloc(sizeof(struct ion_cma_buffer_info), GFP_KERNEL);
+ if (!info) {
+ dev_err(dev, "Can't allocate buffer info\n");
+ return ION_CMA_ALLOCATE_FAILED;
+ }
+
+ info->cpu_addr = dma_alloc_coherent(dev, len, &(info->handle), 0);
+
+ if (!info->cpu_addr) {
+ dev_err(dev, "Fail to allocate buffer\n");
+ kfree(info);
+ return ION_CMA_ALLOCATE_FAILED;
+ }
+
+ /* keep this for memory release */
+ buffer->priv_virt = info;
+ dev_dbg(dev, "Allocate buffer %p\n", buffer);
+ return 0;
+}
+
+static void ion_cma_free(struct ion_buffer *buffer)
+{
+ struct ion_device *idev = buffer->dev;
+ struct device *dev = idev->dev.parent;
+ struct ion_cma_buffer_info *info = buffer->priv_virt;
+
+ dev_dbg(dev, "Release buffer %p\n", buffer);
+ dma_free_coherent(dev, buffer->size, info->cpu_addr, info->handle);
+ kfree(info);
+}
+
+/* return physical address in addr */
+static int ion_cma_phys(struct ion_heap *heap, struct ion_buffer *buffer,
+ ion_phys_addr_t *addr, size_t *len)
+{
+ struct ion_device *idev = buffer->dev;
+ struct device *dev = idev->dev.parent;
+ struct ion_cma_buffer_info *info = buffer->priv_virt;
+
+ dev_dbg(dev, "Return buffer %p physical address 0x%x\n", buffer,
+ virt_to_phys(info->cpu_addr));
+
+ *addr = virt_to_phys(info->cpu_addr);
+ *len = buffer->size;
+
+ return 0;
+}
+
+static int ion_cma_mmap(struct ion_heap *mapper, struct ion_buffer *buffer,
+ struct vm_area_struct *vma)
+{
+ struct ion_device *idev = buffer->dev;
+ struct device *dev = idev->dev.parent;
+ struct ion_cma_buffer_info *info = buffer->priv_virt;
+
+ return dma_mmap_coherent(dev, vma, info->cpu_addr, info->handle,
+ buffer->size);
+}
+
+static struct ion_heap_ops ion_cma_ops = {
+ .allocate = ion_cma_allocate,
+ .free = ion_cma_free,
+ .phys = ion_cma_phys,
+ .map_user = ion_cma_mmap,
+};
+
+struct ion_heap *ion_cma_heap_create(struct ion_platform_heap *data)
+{
+ struct ion_heap *heap;
+
+ heap = ion_heap_create(data);
+
+ if (heap)
+ heap->ops = &ion_cma_ops;
+
+ return heap;
+}
+
+/* ION CMA heap platform driver functions */
+int ion_cma_probe(struct platform_device *pdev)
+{
+ struct ion_platform_data *pdata = pdev->dev.platform_data;
+ int num_heaps = pdata->nr;
+ int i, err;
+
+ heaps = kzalloc(sizeof(struct ion_heap *) * num_heaps, GFP_KERNEL);
+
+ dev_info(&pdev->dev, "Create ION CMA device %p\n", &pdev->dev);
+ ion_cma_device = ion_device_create(NULL);
+ if (IS_ERR_OR_NULL(ion_cma_device)) {
+ kfree(heaps);
+ dev_err(&pdev->dev, "Can't create ION CMA device\n");
+ return PTR_ERR(ion_cma_device);
+ }
+
+ ion_cma_device->dev.parent = &pdev->dev;
+
+ /* set dma mask for this device */
+ pdev->dev.dma_mask = &dma_mask;
+ pdev->dev.coherent_dma_mask = DMA_BIT_MASK(32);
+
+ /* create all the heaps specified in board configuration file */
+ for (i = 0; i < num_heaps; i++) {
+ struct ion_platform_heap *heap_data = &pdata->heaps[i];
+
+ heaps[i] = ion_cma_heap_create(heap_data);
+ if (IS_ERR_OR_NULL(heaps[i])) {
+ dev_err(&pdev->dev, "Can't create heap %s\n",
+ heap_data->name);
+ err = PTR_ERR(heaps[i]);
+ goto err;
+ }
+ ion_device_add_heap(ion_cma_device, heaps[i]);
+ dev_dbg(&pdev->dev, "Add heap %s\n", heap_data->name);
+ }
+
+ platform_set_drvdata(pdev, ion_cma_device);
+ return 0;
+
+err:
+ for (i = 0; i < num_heaps; i++)
+ if (heaps[i])
+ ion_heap_destroy(heaps[i]);
+
+ kfree(heaps);
+ return err;
+}
+
+int ion_cma_remove(struct platform_device *pdev)
+{
+ struct ion_device *idev = platform_get_drvdata(pdev);
+ struct ion_platform_data *pdata = pdev->dev.platform_data;
+ int i, num_heaps;
+
+ num_heaps = pdata->nr;
+
+ /* remove ION device */
+ ion_device_destroy(idev);
+
+ /* remove all heaps create by the driver */
+ for (i = 0; i < num_heaps; i++)
+ if (heaps[i])
+ ion_heap_destroy(heaps[i]);
+
+ return 0;
+}
+
+static struct platform_driver ion_cma_driver = {
+ .probe = ion_cma_probe,
+ .remove = ion_cma_remove,
+ .driver = {
+ .name = "ion-cma"
+ }
+};
+
+static int __init ion_cma_init(void)
+{
+ return platform_driver_register(&ion_cma_driver);
+}
+
+static void __exit ion_cma_exit(void)
+{
+ platform_driver_unregister(&ion_cma_driver);
+}
+
+module_init(ion_cma_init);
+module_exit(ion_cma_exit);