Merge pull request #5 from CoolDesert/dev

fix seristring memory leak
cloudwu · Jul 29, 2016 · e591af4 · e591af4
2 parents 59c1cd3 + 47e0bdb
commit e591af4
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/serialize.c b/serialize.c
@@ -741,10 +741,14 @@ ldeserialize(lua_State *L) {
 
 static int
 seristring(lua_State *L) {
-	lpack(L);
-	lua_replace(L, 1);
-	lua_settop(L, 1);
+	struct write_block b;
+	wb_init(&b, NULL);
+	pack_from(L,&b,0);
+	struct block * ret = wb_close(&b);
+	lua_settop(L,0);
+	lua_pushlightuserdata(L,ret);
 	lserialize(L);
+	wb_free(&b);
 	void *buffer = lua_touserdata(L, -2);
 	int sz = lua_tointeger(L, -1);
 	lua_pushlstring(L, buffer, sz);