Api refactoring (#115)

API refactoring

1 files changed, 2 insertions, 1 deletions

diff --git a/lib/dma.h b/lib/dma.h
index 3e2aaf2..32140d8 100644
--- a/lib/dma.h
+++ b/lib/dma.h
@@ -73,6 +73,7 @@
 #include <sys/mman.h>
 #include <stdint.h>
 #include <stdlib.h>
+#include <limits.h>
 
 #include "muser.h"
 #include "common.h"
@@ -119,7 +120,7 @@ dma_controller_add_region(dma_controller_t *dma,
 int
 dma_controller_remove_region(dma_controller_t *dma,
                              dma_addr_t dma_addr, size_t size,
-                             int (*unmap_dma) (void*, uint64_t), void *data);
+                             lm_unmap_dma_cb_t *unmap_dma, void *data);
 
 // Helper for dma_addr_to_sg() slow path.
 int