C++ KCP异步多线程封装

山有木兮啊

已于 2022-07-18 11:05:48 修改

阅读量1.4k

点赞数 1

分类专栏： c/c++ 文章标签： c++ KCP

于 2022-07-14 16:27:42 首次发布

本文链接：https://blog.csdn.net/DefiniteGoal/article/details/125786642

版权

c/c++ 专栏收录该内容

44 篇文章 3 订阅

订阅专栏

源代码

https://github.com/skywind3000/kcp
如果读源代码会发现其是线程不安全的，作者给出的答案是kcp只做算法上的设计，其他问题由使用者自行规划(https://github.com/skywind3000/kcp/issues/248)

技术

事件，定时器，多线程，协程，epoll，锁，原子操作，信号量，负载均衡(对每个线程绑定的KCP数量做均衡处理)

封装的源代码

https://github.com/TonyBeen/fiber_kcp

测试

客户端发送
在这里插入图片描述
服务端接收：

使用libhv库在windows下进行测试
服务端(ubuntu 18)：

windwos客户端：在这里插入图片描述

#include "hv/hloop.h"
#include "hv/hbase.h"
#include "hv/hsocket.h"
#include "hv/hthread.h"

#define RECV_BUFSIZE    8192
static char recvbuf[RECV_BUFSIZE];

hio_t* sockio = NULL;
hloop_t* loop = NULL;

static void on_recv(hio_t* io, void* buf, int readbytes)
{
    char localaddrstr[SOCKADDR_STRLEN] = { 0 };
    char peeraddrstr[SOCKADDR_STRLEN] = { 0 };
    printf("[%s] <=> [%s]: ",
        SOCKADDR_STR(hio_localaddr(io), localaddrstr),
        SOCKADDR_STR(hio_peeraddr(io), peeraddrstr));
    
    printf("%.*s\n", readbytes, (char*)buf);
    fflush(stdout);
}

DWORD thread_routine(void*)
{
    hio_read(sockio);
    hio_setcb_read(sockio, on_recv);
    hio_set_readbuf(sockio, recvbuf, RECV_BUFSIZE);
    hloop_run(loop);
    hloop_free(&loop);

    return 0;
}

int main()
{
    loop = hloop_new(0);
    const char* host = "49.***.***.220";
    uint16_t port = 12000;
    sockio = hloop_create_udp_client(loop, host, port);
    
    int fd = hio_fd(sockio);
    sockaddr_in addr;
    addr.sin_family = AF_INET;
    addr.sin_addr.s_addr = inet_addr(host);
    addr.sin_port = htons(port);

    sendto(fd, "CONNECT", 7, 0, (sockaddr*)&addr, sizeof(addr));
    uint16_t conv = 0;
    int nrc = recvfrom(fd, (char*)&conv, sizeof(conv), 0, NULL, NULL);
    if (nrc <= 0) {
        perror("recvfrom error");
        return 0;
    }

    printf("conv = 0x%x\n", conv);

    static kcp_setting_t s_kcp_setting;
    memset(&s_kcp_setting, 0, sizeof(kcp_setting_t));
    s_kcp_setting.conv = conv;
    s_kcp_setting.nodelay = 1;
    s_kcp_setting.interval = 10;
    s_kcp_setting.fastresend = 2;
    s_kcp_setting.nocwnd = 1;
    hio_set_kcp(sockio, &s_kcp_setting);

    hthread_t threadHandle = hthread_create(thread_routine, NULL);
    if (threadHandle == NULL) {
        printf("hthread_create error\n");
        return 0;
    }

    char buf[128] = { 0 };
    uint16_t times = 0;
    while (true) {
        snprintf(buf, sizeof(buf), "Hello (times: %d)", ++times);
        hio_write(sockio, buf, strlen(buf));
        printf("send -> %s\n", buf);
        Sleep(1000);
        if (times == 0xff) {
            break;
        }
    }

    return 0;
}