diff options
Diffstat (limited to 'include/linux/vmalloc.h')
| -rw-r--r-- | include/linux/vmalloc.h | 37 |
1 files changed, 35 insertions, 2 deletions
diff --git a/include/linux/vmalloc.h b/include/linux/vmalloc.h index eb6284d7..debdcedb 100644 --- a/include/linux/vmalloc.h +++ b/include/linux/vmalloc.h @@ -1,8 +1,41 @@ #ifndef __TOOLS_LINUX_VMALLOC_H #define __TOOLS_LINUX_VMALLOC_H -#define vmalloc(size) malloc(size) -#define __vmalloc(size, flags, prot) malloc(size) +#include <stdlib.h> +#include <sys/mman.h> + +#include "tools-util.h" + +#define PAGE_KERNEL 0 +#define PAGE_KERNEL_EXEC 1 + #define vfree(p) free(p) +static inline void *__vmalloc(unsigned long size, gfp_t gfp_mask, unsigned prot) +{ + void *p = aligned_alloc(PAGE_SIZE, size); + + if (p && prot == PAGE_KERNEL_EXEC) { + if (mprotect(p, size, PROT_READ|PROT_WRITE|PROT_EXEC)) { + vfree(p); + p = NULL; + } + } + + if (p && (gfp_mask & __GFP_ZERO)) + memset(p, 0, size); + + return p; +} + +static inline void *vmalloc(unsigned long size) +{ + return __vmalloc(size, GFP_KERNEL, PAGE_KERNEL); +} + +static inline void *vzalloc(unsigned long size) +{ + return __vmalloc(size, GFP_KERNEL|__GFP_ZERO, PAGE_KERNEL); +} + #endif /* __TOOLS_LINUX_VMALLOC_H */ |