use skynet_malloc in skynet_realloc

wushangzetom · Apr 22, 2014 · a99befd · a99befd
1 parent 61c8c76
commit a99befd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/skynet-src/malloc_hook.c b/skynet-src/malloc_hook.c
@@ -158,7 +158,7 @@ skynet_malloc(size_t size) {
 
 void *
 skynet_realloc(void *ptr, size_t size) {
-    if (ptr == NULL) return malloc(size);
+    if (ptr == NULL) return skynet_malloc(size);
 
     void* rawptr = clean_prefix(ptr);
     void *newptr = je_realloc(rawptr, size+PREFIX_SIZE);

diff --git a/skynet-src/skynet_main.c b/skynet-src/skynet_main.c
@@ -93,7 +93,7 @@ main(int argc, char *argv[]) {
 
 	struct skynet_config config;
 
-	struct lua_State *L = luaL_newstate();
+	struct lua_State *L = lua_newstate(skynet_lalloc, NULL);
 	luaL_openlibs(L);	// link lua lib
 	lua_close(L);