// driver.unpack(msg, sz)
static int
lunpack(lua_State *L) {
struct skynet_socket_message *message = lua_touserdata(L,1);
int size = luaL_checkinteger(L,2);
// 把消息的属性压栈作为lua返回值
lua_pushinteger(L, message->type);
lua_pushinteger(L, message->id);
lua_pushinteger(L, message->ud);
if (message->buffer == NULL) {
// padding的数据
// SOCKET_OPEN SOCKET_ACCEPT SOCKET_ERROR 没有把data数据写入缓冲区，而是放入之后的内存空间
// 把padding的数据压栈作为lua返回值
lua_pushlstring(L, (char *)(message+1),size - sizeof(*message));
} else {
// 把消息缓冲区指针压栈作为lua返回值
lua_pushlightuserdata(L, message->buffer);
}
if (message->type == SKYNET_SOCKET_TYPE_UDP) {
// 如果消息是udp类型，把UDP地址压栈作为lua返回值
int addrsz = 0;
const char * addrstring = skynet_socket_udp_address(message, &addrsz);
if (addrstring) {
lua_pushlstring(L, addrstring, addrsz);
return 5;
}
}
return 4;
}
/*
lightuserdata msg
integer size
return type n1 n2 ptr_or_string
*/
static int
lunpack(lua_State *L) {
struct skynet_socket_message *message = lua_touserdata(L,1);
int size = luaL_checkinteger(L,2);
lua_pushinteger(L, message->type);
lua_pushinteger(L, message->id);
lua_pushinteger(L, message->ud);
if (message->buffer == NULL) {
lua_pushlstring(L, (char *)(message+1),size - sizeof(*message));
} else {
lua_pushlightuserdata(L, message->buffer);
}
if (message->type == SKYNET_SOCKET_TYPE_UDP) {
int addrsz = 0;
const char * addrstring = skynet_socket_udp_address(message, &addrsz);
if (addrstring) {
lua_pushlstring(L, addrstring, addrsz);
return 5;
}
}
return 4;
}
static void
dispatch_socket_msg(struct otu *u, const struct skynet_socket_message * message, int sz) {
struct skynet_context * ctx = u->ctx;
switch(message->type) {
case SKYNET_SOCKET_TYPE_CONNECT: {
assert (message->id == u->service_id);
break;
}
case SKYNET_SOCKET_TYPE_UDP: {
char* pkt = message->buffer;
int pkt_sz = message->ud;
int addr_sz = 0;
const char * addr = skynet_socket_udp_address(message, &addr_sz);
char* last_err = NULL;
//Ver len check
if(pkt_sz < sizeof(ot_pkt_hdr_t)){
last_err = "1.ShortHdr.";
goto exit_drop;
}
ot_pkt_hdr_t *hdr = (ot_pkt_hdr_t *)pkt;
u16_t hdr_len = ntohs(hdr->len);
if(hdr_len > pkt_sz || hdr_len < sizeof(ot_pkt_hdr_t) || hdr_len > OT_PACKET_SIZE_MAX){ //len check
last_err = "2.HdrLenErr.";
goto exit_drop;
}
if(hdr->proto_ver != OT_PROTO_VER_V3A){
last_err = "3.HdrVerErr.";
goto exit_drop;
}
#define HDR_ONLY (hdr_len == sizeof(ot_pkt_hdr_t))
#define DID_IS_BROADCAST (IS_ALL_FF(hdr->did.byte, sizeof(hdr->did.byte)))
if(HDR_ONLY){
//时间戳请求处理
if(DID_IS_BROADCAST){
hdr->ts = htonl(skynet_now()/100 + skynet_starttime());
//int err =
skynet_socket_udp_send(ctx, message->id, addr, pkt, pkt_sz);
}
//保活
else {
uint64_t did64 = ntohll(hdr->did.u64);
int idx = hashid64_lookup(&u->hash, did64);
if (idx >= 0) {
struct udp_peer *p = &u->peer[idx];
(void)p;
//dispatch_msg(u, p, message->id, message->buffer, message->ud);
goto exit_drop;
} else {
//todo 提交DID未知请求
goto exit_drop;
}
}
}
break;
//get key, decrypt packet.
//dispatch_msg
exit_drop:
skynet_free(message->buffer);
break;
}
default:
skynet_error(ctx, "OTU: unknown type(%d)", message->type);
break;
}
}
