1 files changed, 13 insertions, 0 deletions
diff --git a/ion_physical_heap.h b/ion_physical_heap.h
index fa66671..f48ccd8 100644
--- a/ion_physical_heap.h
+++ b/ion_physical_heap.h
@@ -4,6 +4,7 @@
  */
 
 #include <linux/samsung-dma-heap.h>
+#include <linux/genalloc.h>
 
 typedef void(ion_physical_heap_allocate_callback)(struct samsung_dma_buffer *buffer,
 						  void *ctx);
@@ -15,3 +16,15 @@ struct dma_heap *ion_physical_heap_create(phys_addr_t base, size_t size,
 					  ion_physical_heap_allocate_callback alloc_cb,
 					  ion_physical_heap_free_callback free_cb,
 					  void *ctx);
+
+struct ion_physical_heap {
+	struct gen_pool *pool;
+	phys_addr_t base;
+	size_t size;
+
+	ion_physical_heap_allocate_callback *allocate_cb;
+	void *allocate_ctx;
+
+	ion_physical_heap_free_callback *free_cb;
+	void *free_ctx;
+};