struct p2p *p2p_init(const char *rpc_srv, const char *stun_srv)
{
char ip[64];
struct skt_addr tmpaddr;
static stun_addr _mapped;
struct p2p *p2p = CALLOC(1, struct p2p);
if (!p2p) {
loge("malloc failed: %d\n", errno);
return NULL;
}
p2p->rpc = rpc_create(rpc_srv, _rpc_port);
if (!p2p->rpc) {
loge("rpc_create failed\n");
return NULL;
}
RPC_REGISTER_MSG_MAP(BASIC_RPC_API_RESP);
rpc_set_cb(p2p->rpc, on_rpc_read, on_rpc_write, on_rpc_error, p2p);
skt_getaddr_by_fd(p2p->rpc->fd, &tmpaddr);
skt_addr_ntop(_local_ip, tmpaddr.ip);
//_local_port = tmpaddr.port;
//logi("_local_port = %d\n", _local_port);
stun_init(stun_srv);
p2p->nat.type = stun_nat_type();
p2p->nat.uuid = p2p->rpc->send_pkt.header.uuid_src;
p2p->nat.local.ip = skt_addr_pton(_local_ip);
_local_port = random_port();
p2p->nat.local.port = _local_port;
p2p->nat.fd = stun_socket(_local_ip, _local_port, &_mapped);
_mapped.addr = ntohl(_mapped.addr);
skt_addr_ntop(ip, _mapped.addr);
p2p->nat.reflect.ip = _mapped.addr;
p2p->nat.reflect.port = _mapped.port;
logi("get nat info from local\n");
logi("nat.type = %d\n", p2p->nat.type);
logi("nat.local_addr %s:%d\n", _local_ip, p2p->nat.local.port);
logi("nat.reflect_addr %s:%d\n", ip, p2p->nat.reflect.port);
p2p->rpc_state = P2P_RPC_INIT;
_p2p = p2p;
return p2p;
}
struct rpc *rpc_create(const char *host, uint16_t port)
{
char local_ip[INET_ADDRSTRLEN];
char remote_ip[INET_ADDRSTRLEN];
struct rpc *r = CALLOC(1, struct rpc);
if (!r) {
loge("malloc failed!\n");
return NULL;
}
memset(&r->recv_pkt, 0, sizeof(struct rpc_packet));
struct skt_connection *connect;
connect = skt_tcp_connect(host, port);
if (!connect) {
loge("connect failed!\n");
return NULL;
}
r->fd = connect->fd;
if (-1 == skt_set_block(r->fd)) {
loge("skt_set_block failed!\n");
}
r->evbase = gevent_base_create();
if (!r->evbase) {
loge("gevent_base_create failed!\n");
return NULL;
}
rpc_set_cb(r, on_read, on_write, on_error, r);
r->dispatch_thread = thread_create("rpc_dispatch", rpc_dispatch_thread, r);
r->state = rpc_inited;
if (thread_sem_wait(r->dispatch_thread, 2000) == -1) {
loge("wait response failed %d:%s\n", errno, strerror(errno));
return NULL;
}
skt_addr_ntop(local_ip, connect->local.ip);
skt_addr_ntop(remote_ip, connect->remote.ip);
logi("rpc[%08x] connect information:\n", r->send_pkt.header.uuid_src);
logi("local addr = %s:%d\n", local_ip, connect->local.port);
logi("remote addr = %s:%d\n", remote_ip, connect->remote.port);
return r;
}
