lihao/skynet
1
0
Fork 0
mirror of https://github.com/cloudwu/skynet.git synced 2026-07-24 20:23:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

check dest type

Browse Source
This commit is contained in:
Cloud Wu
2014-08-18 16:44:19 +08:00
parent 0fb23f9ab8
commit f9438461eb
1 changed files with 11 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
lualib-src/lua-skynet.c
View File

@@ -126,6 +126,15 @@ _genid(lua_State *L) {
return 1; return 1;
} }
static const char *
get_dest_string(lua_State *L, int index) {
const char * dest_string = lua_tostring(L, index);
if (dest_string == NULL) {
luaL_error(L, "dest address type (%s) must be a string or number.", lua_typename(L, lua_type(L,index)));
}
return dest_string;
}
/* /*
unsigned address unsigned address
string address string address
@@ -141,7 +150,7 @@ _send(lua_State *L) {
uint32_t dest = lua_tounsigned(L, 1); uint32_t dest = lua_tounsigned(L, 1);
const char * dest_string = NULL; const char * dest_string = NULL;
if (dest == 0) { if (dest == 0) {
dest_string = lua_tostring(L, 1); dest_string = get_dest_string(L, 1);
} }
int type = luaL_checkinteger(L, 2); int type = luaL_checkinteger(L, 2);
@@ -195,7 +204,7 @@ _redirect(lua_State *L) {
uint32_t dest = lua_tounsigned(L,1); uint32_t dest = lua_tounsigned(L,1);
const char * dest_string = NULL; const char * dest_string = NULL;
if (dest == 0) { if (dest == 0) {
dest_string = lua_tostring(L,1); dest_string = get_dest_string(L, 1);
} }
uint32_t source = luaL_checkunsigned(L,2); uint32_t source = luaL_checkunsigned(L,2);
int type = luaL_checkinteger(L,3); int type = luaL_checkinteger(L,3);