use correct allocation flag under different half contexts

luainkernel · Jul 19, 2018 · 978405b · 978405b
1 parent 59098f0
commit 978405b
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/lua/luaconf.h b/lua/luaconf.h
@@ -864,7 +864,12 @@ static inline int time(void *p)
 #endif
 
 #define free 			kfree
-#define realloc(a, b) 		krealloc(a, b, GFP_KERNEL)
+#include <linux/hardirq.h>
+static inline void *realloc(const void *p, size_t new_size)
+{
+	gfp_t flag = in_interrupt() ? GFP_ATOMIC : GFP_KERNEL;
+	return krealloc(p, new_size, flag);
+}
 
 /* signal.h */
 #define l_signalT	lu_byte