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; }