UDPSession *
UDPSession::createSession(int sockfd) {
int flags = fcntl(sockfd, F_GETFL, 0);
if (flags < 0) {
return nullptr;
}
if (fcntl(sockfd, F_SETFL, flags | O_NONBLOCK) < 0) {
return nullptr;
}
UDPSession *sess = new(UDPSession);
sess->m_sockfd = sockfd;
sess->m_kcp = ikcp_create(IUINT32(rand()), sess);
sess->m_buf = (char *) malloc(UDPSession::mtuLimit);
sess->m_streambuf = (char *) malloc(UDPSession::streamBufferLimit);
sess->m_kcp->output = sess->out_wrapper;
if (sess->m_kcp == nullptr || sess->m_buf == nullptr || sess->m_streambuf == nullptr) {
if (nullptr != sess->m_kcp) { ikcp_release(sess->m_kcp); }
if (nullptr != sess->m_buf) { free(sess->m_buf); }
if (nullptr != sess->m_streambuf) { free(sess->m_streambuf); }
return nullptr;
}
return sess;
}
void
UDPSession::Destroy(UDPSession *sess) {
if (nullptr == sess) return;
if (0 != sess->m_sockfd) { close(sess->m_sockfd); }
if (nullptr != sess->m_kcp) { ikcp_release(sess->m_kcp); }
if (nullptr != sess->m_buf) { free(sess->m_buf); }
if (nullptr != sess->m_streambuf) { free(sess->m_streambuf); }
delete sess;
}
static int
ldestroy(lua_State *l) {
struct kcpctx *ctx = (struct kcpctx *)luaL_checkudata(l, 1, LKCP_MT);
if (!ctx)
return luaL_argerror(l, 1, "parameter self invalid.");
assert(ctx->kcp);
luaL_unref(l, LUA_REGISTRYINDEX, ctx->ref);
ikcp_release(ctx->kcp);
ctx->kcp = NULL;
ctx->ud = NULL;
ctx->ref = 0;
ctx->conv = 0;
ctx->timeout = 0;
return 0;
}
int sg_etp_session_close(sg_etp_session_t * session)
{
LOG_D("ikcp_peeksize: %d, ikcp_waitsnd: %d",
ikcp_peeksize(session->kcp), ikcp_waitsnd(session->kcp));
if (ikcp_waitsnd(session->kcp) > 0 || ikcp_peeksize(session->kcp) > 0)
{
session->to_close = true; /* mark for close later */
return OK;
}
LOG_D("close session");
ikcp_release(session->kcp);
uv_timer_stop(&(session->timer));
if (session->udp == &(session->udp_hdl)) /* self-contained udp, should close here */
{
uv_close((uv_handle_t*)session->udp, on_uv_close_done);
}
if (session->loop == &(session->loop_hdl)) /* self-contained loop, should close here */
{
uv_loop_close(session->loop);
}
SG_CALLBACK(session->on_close, session, OK, "ok");
if (NULL != session->recv_buf)
{
free(session->recv_buf);
}
free(session);
return OK;
}
connection::~connection(void)
{
ikcp_release(p_kcp_);
p_kcp_ = NULL;
conv_ = 0;
}
void Conn::shutdown() {
ikcp_release(_kcp);
}
// 测试用例
void test(int mode)
{
// 创建模拟网络：丢包率10%，Rtt 60ms~125ms
vnet = new LatencySimulator(10, 60, 125);
// 创建两个端点的 kcp对象，第一个参数 conv是会话编号，同一个会话需要相同
// 最后一个是 user参数，用来传递标识
ikcpcb *kcp1 = ikcp_create(0x11223344, (void*)0);
ikcpcb *kcp2 = ikcp_create(0x11223344, (void*)1);
// 设置kcp的下层输出，这里为 udp_output，模拟udp网络输出函数
kcp1->output = udp_output;
kcp2->output = udp_output;
IUINT32 current = iclock();
IUINT32 slap = current + 20;
IUINT32 index = 0;
IUINT32 next = 0;
IINT64 sumrtt = 0;
int count = 0;
int maxrtt = 0;
// 配置窗口大小：平均延迟200ms，每20ms发送一个包，
// 而考虑到丢包重发，设置最大收发窗口为128
ikcp_wndsize(kcp1, 128, 128);
ikcp_wndsize(kcp2, 128, 128);
// 判断测试用例的模式
if (mode == 0) {
// 默认模式
ikcp_nodelay(kcp1, 0, 10, 0, 0);
ikcp_nodelay(kcp2, 0, 10, 0, 0);
}
else if (mode == 1) {
// 普通模式，关闭流控等
ikcp_nodelay(kcp1, 0, 10, 0, 1);
ikcp_nodelay(kcp2, 0, 10, 0, 1);
} else {
// 启动快速模式
// 第二个参数 nodelay-启用以后若干常规加速将启动
// 第三个参数 interval为内部处理时钟，默认设置为 10ms
// 第四个参数 resend为快速重传指标，设置为2
// 第五个参数 为是否禁用常规流控，这里禁止
ikcp_nodelay(kcp1, 1, 10, 2, 1);
ikcp_nodelay(kcp2, 1, 10, 2, 1);
kcp1->rx_minrto = 10;
kcp1->fastresend = 1;
}
char buffer[2000];
int hr;
IUINT32 ts1 = iclock();
while (1) {
isleep(1);
current = iclock();
ikcp_update(kcp1, iclock());
ikcp_update(kcp2, iclock());
// 每隔 20ms，kcp1发送数据
for (; current >= slap; slap += 20) {
((IUINT32*)buffer)[0] = index++;
((IUINT32*)buffer)[1] = current;
// 发送上层协议包
ikcp_send(kcp1, buffer, 8);
}
// 处理虚拟网络：检测是否有udp包从p1->p2
while (1) {
hr = vnet->recv(1, buffer, 2000);
if (hr < 0) break;
// 如果 p2收到udp，则作为下层协议输入到kcp2
ikcp_input(kcp2, buffer, hr);
}
// 处理虚拟网络：检测是否有udp包从p2->p1
while (1) {
hr = vnet->recv(0, buffer, 2000);
if (hr < 0) break;
// 如果 p1收到udp，则作为下层协议输入到kcp1
ikcp_input(kcp1, buffer, hr);
}
// kcp2接收到任何包都返回回去
while (1) {
hr = ikcp_recv(kcp2, buffer, 10);
// 没有收到包就退出
if (hr < 0) break;
// 如果收到包就回射
ikcp_send(kcp2, buffer, hr);
}
// kcp1收到kcp2的回射数据
while (1) {
hr = ikcp_recv(kcp1, buffer, 10);
// 没有收到包就退出
if (hr < 0) break;
IUINT32 sn = *(IUINT32*)(buffer + 0);
IUINT32 ts = *(IUINT32*)(buffer + 4);
IUINT32 rtt = current - ts;
if (sn != next) {
// 如果收到的包不连续
printf("ERROR sn %d<->%d\n", (int)count, (int)next);
return;
}
next++;
sumrtt += rtt;
count++;
if (rtt > (IUINT32)maxrtt) maxrtt = rtt;
printf("[RECV] mode=%d sn=%d rtt=%d\n", mode, (int)sn, (int)rtt);
}
if (next > 1000) break;
}
ts1 = iclock() - ts1;
ikcp_release(kcp1);
ikcp_release(kcp2);
const char *names[3] = { "default", "normal", "fast" };
printf("%s mode result (%dms):\n", names[mode], (int)ts1);
printf("avgrtt=%d maxrtt=%d tx=%d\n", (int)(sumrtt / count), (int)maxrtt, (int)vnet->tx1);
printf("press enter to next ...\n");
char ch; scanf("%c", &ch);
}
/*@ckcpuv*/
void conn_shutdown(conn_t * thiz) {
ikcp_release(thiz->_kcp);
}
