p2p sdk接收文件服务端多线程模型源码

#  KKP2P SDK介绍
kkp2p sdk是库快科技（kkuai.com）研发的支持p2p通信的中间件，是一套与业务无关的通用的p2p sdk库， 可以免费下载试用版本使用。
一句话概括其特点：支持面向登录账号进行网络编程。
即只要传入对端的账号id，应用层就能得到一个socket句柄fd，应用层通过读写该socket fd句柄和对端进行通信。

优秀特性	说明
跨平台	kkp2p的sdk库是由c语言开发，在linux、windows、android、ios等平台编译出了静态库，以及其他一些嵌入式平台也编译出了静态库，大家可以直接下载进行使用。云端服务是由golang语言开发，也支持在各种平台下编译出直接可以运行的程序，配置也比较简单，大家下载之后就可以按照官网文档说明自行进行部署。
体积小	kkp2p不依赖于任何第三方库，编译出来的库只有500KB左右大小。
性能强	在服务器上测试，P2P方式通信的速度可以超过10MB每秒；中转方式通信的速度取决于您云端服务器的带宽
易使用	提供了类似于socket编程接口的kkp2p_connect、kkp2p_listen、kkp2p_accept、kkp2p_read、kkp2p_write几个核心函数,使用起来非常简单方便。您只要指定对端的登录账号通过kkp2p_connect函数就能和对端创建一个虚拟的传输管道，然后通过kkp2p_read和kkp2p_write函数来读写数据和对端进行通信。您还可以通过参数指定是使用P2P方式通信还是使用中转（relay）方式通信，完全不用关心底层传输通道的创建和管理细节，一切由kkp2p的sdk库帮您解决。
高安全	支持加密通道传输，您只需要在kkp2p_connect的参数中指定需要加密数据即可。sdk会自动创建一个加密的虚拟通信管道出来；您写入明文，sdk会自动加密成密文传输；sdk收到密文，会自动解密成明文返回给您。通信双方的共同密钥是双方的sdk通过DH算法自动协商而成，外界无法获取；并且每次sdk的启动都会自动协商生成一个新的动态密钥，严格保障您的通信数据的安全。如果您为了提升数据传输的性能，不想对数据进行加解密，只需要在创建连接的函数kkp2p_connect参数中指定不需要加密数据即可。sdk的数据加解密功能只有在商业版本中才有，在个人试用版本中没有该功能。
通用性	kkp2p是一套适用于各种场景的通用的通信中间件，会完全透传用户的数据，您可以灵活的自定义通信双方的协议，kkp2p不会解析您的业务数据。kkp2p的P2P通信是基于udp实现的，kkp2p会自动帮您解决丢包、乱序、重传问题，也会根据您的实际带宽做自适应的带宽流控，您使用起来具有tcp传输的效果，相当于是用udp模拟实现了tcp。中转（relay）通信是基于tcp原生实现的。

#  接收文件服务端源码

接收文件过程：首先接收4字节的文件大小（网络字节序，暂支持4G左右大小，如果要超过4G，则协议需要使用8字节大小），然后再接收文件内容；接收完成后，发送响应，最后在退出。

 服务端可以并行接收多个客户端发送过来的文件，虽然是个单线程，但是通过epoll实现了并发接收，读者需要有epoll背景知识，该代码在linux平台编译运行验证通过。

我们可以看到，p2p sdk的用法非常类似原生的网络编程用法，非常简单易用。

#  源码讲解

#include <stdio.h>
#include <stdint.h>
#include <errno.h>
#include <string.h>
#include <arpa/inet.h>
#include <sys/types.h>
#include <fcntl.h>
#include <unistd.h>
#include <map>

// 去kkuai.com获取
// 得包含kkp2p sdk的头文件，以及链接libkkp2p.a
#include "kkp2p_sdk.h"

// 全局变量
kkp2p_engine_t* p2pEngine = NULL;


// 简单封装一下发送指定长度数据接口
int SendData(int fd, char* buff, int len) {
    int sended = 0 ;
    while (sended < len) {
        // 1秒超时
        int wl = kkp2p_write(fd, buff + sended, len - sended, 1000);
        if (wl < 0) {
            printf("SendData error,fd:%d,ret:%d,len:%d,errno:%d,desc:%s.\n",fd,wl, len, errno, strerror(errno));
            return -1;
        }
        sended += wl;
    }
    return len;
}

// 简单封装一下读取指定长度数据接口
int RecvData(int fd, char* buff, int len) {
    int recved = 0 ;
    while (recved < len) {
        // 1秒超时
        int wl = kkp2p_read(fd, buff + recved, len - recved, 1000);
        if (wl < 0) {
            printf("RecvData error,fd:%d,ret:%d,len:%d,errno:%d,desc:%s.\n",fd,wl, len,errno, strerror(errno));
            return -1;
        }
            recved += wl;
    }
    return len;
}

// 接收线程，每个文件对应一个
void* ReadThread(void* arg)
{
    kkp2p_channel_t* channel = (kkp2p_channel_t*)arg;

    // 首先接收4个字节文件长度
    uint32_t fileSize = 0;
    int recved = RecvData(channel->fd, (char*)&fileSize, sizeof(fileSize));
    if (recved < 0) {
        printf("kkp2p_read file size error,channel id:%u,exit thread.\n", channel->channel_id);
        kkp2p_close_fd(channel->fd);
        kkp2p_close_channel(p2pEngine, channel->channel_id);
        free(channel);
        return NULL;
    }
    fileSize = ntohl(fileSize);

    // 创建一个新文件，准备接收文件内容
    char fileName[1024];
    memset(fileName, 0, sizeof(fileName));
    sprintf(fileName,"%d_%d.txt", time(NULL), channel->fd);
    FILE* pWrite = fopen(fileName, "w+");
    if (pWrite == NULL) {
        printf("fopen %s error,channel id:%u,exit thread.\n", fileName, channel->channel_id);
        kkp2p_close_fd(channel->fd);
        kkp2p_close_channel(p2pEngine, channel->channel_id);
        free(channel);
        return NULL;
    }

    // 循转接收文件内容
    uint32_t totalRecved  = 0 ;
    char szBuff[1024];
    while (totalRecved < fileSize) {
        int recved = kkp2p_read(channel->fd, szBuff, sizeof(szBuff),1000);
        if (recved < 0) {
            // 接收出错
            printf("kkp2p_read error,channel id:%u,exit thread.\n", channel->channel_id);
            break;
        } else if (recved == 0) {
            // 超时
            // timeout
        } else {
            // 写文件，接收完成发送响应
            totalRecved += recved;
            fwrite(szBuff, 1, recved, pWrite);
            if (totalRecved >= fileSize) {
                char ch = '1';
                SendData(channel->fd, &ch, 1);
                printf("recv file success,file size:%u,fd:%d,channel id:%u.\n",fileSize, channel->fd, channel->channel_id);
            }
        }
    }

    // 接收失败发送响应
    if (totalRecved < fileSize) {
         char ch = '0';
        SendData(channel->fd, &ch, 1);
        printf("recv file error,file size:%u,recved:%u,fd:%d,channel id:%u.\n",fileSize, totalRecved,channel->fd, channel->channel_id);
    }
    fflush(pWrite);
    fclose(pWrite);
    
    // 关闭连接的代理fd句柄，以及连接
    kkp2p_close_fd(channel->fd);
    kkp2p_close_channel(p2pEngine, channel->channel_id);
    free(channel);
}

// 输入两个参数，第一个是peerId，即登录账号，第二个是账号对应的密钥，云端禁止非法账号登录
int main(int argc, char** argv)
{
    if (argc < 3) {
        printf("usage:%s peerId peerKey\n",argv[0]);
        return -1;
    }

    char* peerId = argv[1];
    char* peerKey = argv[2];

    // 初始化 kkp2p sdk 
    // 包括云端登录域名端口，局域网端口，日志配置
    kkp2p_engine_conf_t kkp2p_conf;
    kkp2p_conf.login_domain = "125.72.218.199";
    kkp2p_conf.login_port = 3080;
    kkp2p_conf.lan_search_port = 3549;
    kkp2p_conf.max_log_size = 1024*1024*10;
    kkp2p_conf.log_path = NULL;
    p2pEngine = kkp2p_engine_init(&kkp2p_conf, 5000);
    if (p2pEngine == NULL) {
        printf("init kkp2p engine error.\n");
        return -1;
    }
    
    // 切换日志级别到debug模式
    kkp2p_switch_log_level(p2pEngine, 4);

     // 将peerId加入到云端和本地局域网
    kkp2p_join_lan(p2pEngine, peerId);
    kkp2p_join_net(p2pEngine, peerId, peerKey);

    printf("init success,begin loop.\n");

    // 主线程循环接收远端连接，p2p传输连接或者中转传输连接
    kkp2p_channel_t* channel = (kkp2p_channel_t*)calloc(1, sizeof(kkp2p_channel_t));
    while(1) {
        int ret = kkp2p_accept(p2pEngine, 1000, channel);
        if (ret < 0) {
            // 出错
            printf("kkp2p_accept error,exit\n");
            free(channel);
            break;
        } else if (ret == 0) {
            // 超时
             continue;
        } else {
             // 有新连接过来，启动新线程处理
             // transmit_mode是连接类型，1是p2p连接，2是中转连接
             pthread_t ThreadId;
             printf("accept new connection,fd:%d, mode is %d,channel id:%u.\n",channel->fd, channel->transmit_mode, channel->channel_id);
             pthread_create(&ThreadId, NULL, ReadThread,(void*)channel);
             channel = (kkp2p_channel_t*)calloc(1, sizeof(kkp2p_channel_t));
        }
    }
    
    // 退出
    kkp2p_engine_destroy(p2pEngine);
    return 0;
}