bittorrent 学习(三) MSG

msg.c中

int转化 char[4] char[4]转化int的函数如下(有多种方案)

 int int_to_char(int i, unsigned char c[])

 {

     c[] = i % ;

     c[] = (i - c[]) /  % ;

     c[] = (i - c[] - c[] * ) /  /  % ;

     c[] = (i - c[] - c[] *  - c[] *  * ) /  /  /  % ;

     return ;

 }

 int IntToChar(int j, unsigned char c[]) {

     int i = ;

     for (i = ; i < ; i++) {

         c[i] = (j >> ( - i - ) * ) & 0xff;

     }

     return ;

 }

 int CharToInt1(unsigned char c[]) {

     int i = ;

     unsigned char* p = (unsigned char*)&i;

     for (int j = ; j >  ; j--) {

         memcpy(p, &c[j], sizeof(c[j]));

         p++;

     }

     memcpy(p, &c[], sizeof(c[]));

     return i;

 }

 int CharToInt(unsigned char c[]) {

     int i = ;

     for (int j = ; j < ; j++) {

         i += c[j];

         i <<= ;

     }

     i += c[];

     return i;

 }

 int char_to_int(unsigned char c[])

 {

     int i;

     i = c[] *  *  *  + c[] *  *  + c[] *  + c[];

     return i;

 }

创建握手信息的函数 create_handshake_msg()

握手消息

客户端与一个peer建立TCP连接后，首先向peer发送握手消息，peer收到握手消息后回应一个握手消息。握手消息是一个长度固定为68字节的消息，格式如下：
<pstrlen><pstr><reserved><info_hash><peer_id>

握手消息参数
参数含义
pstrlen pstr的长度，固定为19
pstr BitTorrent协议的关键字，即”BitTorrent protocol”
reserved 占8字节，用于扩展BT协议。可忽略
info_hash 与发往Tracker的GET请求中的info_hash为同一个值，固定为20字节
peer_id 与发往Tracker的GET请求中的peer_id为同一个值，固定为20字节

info_hash就是种子文件中 4:info 后的种子信息计算的哈希值

 int create_handshake_msg(char *info_hash,char *peer_id,Peer *peer)

 {

     int            i;

     unsigned char  keyword[] = "BitTorrent protocol", c = 0x00;

     unsigned char  *buffer = peer->out_msg + peer->msg_len;

     int            len = MSG_SIZE - peer->msg_len;

     if(len < )  return -;  // 68为握手消息的固定长度

     buffer[] = ;

     for(i = ; i < ; i++)  buffer[i+]  = keyword[i];

     for(i = ; i < ;  i++)  buffer[i+] = c;

     for(i = ; i < ; i++)  buffer[i+] = info_hash[i];

     for(i = ; i < ; i++)  buffer[i+] = peer_id[i];

     peer->msg_len += ;

     return ;

 }

心跳函数表明I自己在线 create_chock_interested_msg()

发送4个0

 int create_keep_alive_msg(Peer *peer)

 {

     unsigned char  *buffer = peer->out_msg + peer->msg_len;

     int            len = MSG_SIZE - peer->msg_len;

     if(len < )  return -;  // 4为keep_alive消息的固定长度

     memset(buffer,,);

     peer->msg_len += ;

     return ;

 }

chock 函数 create_chock_interested_msg() 表明对方由于某种原因阻塞比如贡献度不够？或者是否感兴趣

结构都是一致的 5字节第四个为1 第五个表示消息类型(choke、unchoke、interested、uninterested)

 int create_chock_interested_msg(int type,Peer *peer)

 {

     unsigned char  *buffer = peer->out_msg + peer->msg_len;

     int            len = MSG_SIZE - peer->msg_len;

     // 5为choke、unchoke、interested、uninterested消息的固定长度

     if(len < )  return -;

     memset(buffer,,);

     buffer[] = ;

     buffer[] = type;

     peer->msg_len += ;

     return ;

 }

表示是否拥有该index的piece函数 create_have_msg()

9个字节最后4个字节是index的int to char 计算的结果。第4 5个字节填写 5 4 。这个似乎在协议说明里没看到

 int create_have_msg(int index,Peer *peer)

 {

     unsigned char  *buffer = peer->out_msg + peer->msg_len;

     int            len = MSG_SIZE - peer->msg_len;

     unsigned char  c[];

     if(len < )  return -;  // 9为have消息的固定长度

     memset(buffer,,);

     buffer[] = ;

     buffer[] = ;

     int_to_char(index,c);

     buffer[] = c[];

     buffer[] = c[];

     buffer[] = c[];

     buffer[] = c[];

     peer->msg_len += ;

     return ;

 }

创建位图消息 create_bitfield_msg() 传递本地位图文件消息

长度不定位图长度+5 前4位是位图长度 int to char 的数组第五位标记为5 后面是位图

 int create_bitfield_msg(char *bitfield,int bitfield_len,Peer *peer)

 {

     int            i;

     unsigned char  c[];

     unsigned char  *buffer = peer->out_msg + peer->msg_len;

     int            len = MSG_SIZE - peer->msg_len;

     if( len < bitfield_len+ )  {  // bitfield消息的长度为bitfield_len+5

         printf("%s:%d buffer too small\n",__FILE__,__LINE__);

         return -;

     }

     int_to_char(bitfield_len+,c);

     for(i = ; i < ; i++)  buffer[i] = c[i];

     buffer[] = ;

     for(i = ; i < bitfield_len; i++) buffer[i+] = bitfield[i];

     peer->msg_len += bitfield_len+;

     return ;

 }

请求文件的内容函数 create_request_msg()

前5字节固定第4 5字节填写13 6.后面每4个字节分别填写 index begin end的int to char。

因为peer之间交换数据时以slice（长度为16KB的块）为单位的，因此request消息中length的值一般为16K。最大不超过128K

 int create_request_msg(int index,int begin,int length,Peer *peer)

 {

     int            i;

     unsigned char  c[];

     unsigned char  *buffer = peer->out_msg + peer->msg_len;

     int            len = MSG_SIZE - peer->msg_len;

     if(len < )  return -;  // 17为request消息的固定长度

     memset(buffer,,);

     buffer[] = ;

     buffer[] = ;

     int_to_char(index,c);

     for(i = ; i < ; i++)  buffer[i+]  = c[i];

     int_to_char(begin,c);

     for(i = ; i < ; i++)  buffer[i+]  = c[i];

     int_to_char(length,c);

     for(i = ; i < ; i++)  buffer[i+] = c[i];

     peer->msg_len += ;

     return ;

 }

创建piece信息 create_piece_msg()

长度为piece长度加13。

头4个字节是总长度 piece_length+13经过 int to char转化。接下来是一个标记7的字节。然后是index begin两个长度int to char.接下来是传输piece

 int create_piece_msg(int index,int begin,char *block,int b_len,Peer *peer)

 {

     int            i;

     unsigned char  c[];

     unsigned char  *buffer = peer->out_msg + peer->msg_len;

     int            len = MSG_SIZE - peer->msg_len;

     if( len < b_len+ ) {  // piece消息的长度为b_len+13

         printf("IP:%s len:%d\n",peer->ip,len);

         printf("%s:%d buffer too small\n",__FILE__,__LINE__);

         return -;

     }

     int_to_char(b_len+,c);

     for(i = ; i < ; i++)      buffer[i]    = c[i];

     buffer[] = ;

     int_to_char(index,c);

     for(i = ; i < ; i++)      buffer[i+]  = c[i];

     int_to_char(begin,c);

     for(i = ; i < ; i++)      buffer[i+]  = c[i];

     for(i = ; i < b_len; i++)  buffer[i+] = block[i];

     peer->msg_len += b_len+;

     return ;

 }

取消信息用于取消对某个piece的请求 create_cancel_msg()

固定长度17

第4 5个字节赋值 13 8. 后面12个字节分别是 index begin length的int to char 数组

 int create_cancel_msg(int index,int begin,int length,Peer *peer)

 {

     int            i;

     unsigned char  c[];

     unsigned char  *buffer = peer->out_msg + peer->msg_len;

     int            len = MSG_SIZE - peer->msg_len;

     if(len < )  return -;  // 17为cancel消息的固定长度

     memset(buffer,,);

     buffer[] = ;

     buffer[] = ;

     int_to_char(index,c);

     for(i = ; i < ; i++)  buffer[i+]  = c[i];

     int_to_char(begin,c);

     for(i = ; i < ; i++)  buffer[i+]  = c[i];

     int_to_char(length,c);

     for(i = ; i < ; i++)  buffer[i+] = c[i];

     peer->msg_len += ;

     return ;

 }

传递端口信息固定7个字节第四五字节为3 9 第六七个字节为端口 int to char的后两个字节

 int create_port_msg(int port,Peer *peer)

 {

     unsigned char  c[];

     unsigned char  *buffer = peer->out_msg + peer->msg_len;

     int            len = MSG_SIZE - peer->msg_len;

     if( len < )  return ;  // 7为port消息的固定长度

     memset(buffer,,);

     buffer[] = ;

     buffer[] = ;

     int_to_char(port,c);

     buffer[] = c[];

     buffer[] = c[];

     peer->msg_len += ;

     return ;

 }

print_msg_buffer() 打印buffer内容

 // 以十六进制的形式打印消息的内容,用于调试

 int print_msg_buffer(unsigned char *buffer, int len)

 {

     int i;

     for(i = ; i < len; i++) {

         printf("%.2x ",buffer[i]);

         if( (i+) %  ==  )  printf("\n");

     }

     printf("\n");

     return ;

 }

is_complete_message()检测是否一条完整信息存放在BUF中

首先比对固定头信息的集中类型比如握手 choke interested have request cancel port 然后检测 bitfield 和piece不定长协议。最后根据头文件信息指定的长度是否吻合判断是否接收到未知定义还是未结束的消息

 int is_complete_message(unsigned char *buff,unsigned int len,int *ok_len)

 {

     unsigned int   i;

     char           btkeyword[];

     unsigned char  keep_alive[]   = { 0x0, 0x0, 0x0, 0x0 };

     unsigned char  chocke[]       = { 0x0, 0x0, 0x0, 0x1, 0x0};

     unsigned char  unchocke[]     = { 0x0, 0x0, 0x0, 0x1, 0x1};

     unsigned char  interested[]   = { 0x0, 0x0, 0x0, 0x1, 0x2};

     unsigned char  uninterested[] = { 0x0, 0x0, 0x0, 0x1, 0x3};

     unsigned char  have[]         = { 0x0, 0x0, 0x0, 0x5, 0x4};

     unsigned char  request[]      = { 0x0, 0x0, 0x0, 0xd, 0x6};

     unsigned char  cancel[]       = { 0x0, 0x0, 0x0, 0xd, 0x8};

     unsigned char  port[]         = { 0x0, 0x0, 0x0, 0x3, 0x9};

     if(buff==NULL || len<= || ok_len==NULL)  return -;

     *ok_len = ;

     btkeyword[] = ;

     memcpy(&btkeyword[],"BitTorrent protocol",);  // BitTorrent协议关键字

     unsigned char  c[];

     unsigned int   length;

     for(i = ; i < len; ) {

         // 握手、chocke、have等消息的长度是固定的

         if( i+<=len && memcmp(&buff[i],btkeyword,)== )         i += ;

         else if( i+ <=len && memcmp(&buff[i],keep_alive,)== )    i += ;

         else if( i+ <=len && memcmp(&buff[i],chocke,)== )        i += ;

         else if( i+ <=len && memcmp(&buff[i],unchocke,)== )      i += ;

         else if( i+ <=len && memcmp(&buff[i],interested,)== )    i += ;

         else if( i+ <=len && memcmp(&buff[i],uninterested,)== )  i += ;

         else if( i+ <=len && memcmp(&buff[i],have,)== )          i += ;

         else if( i+<=len && memcmp(&buff[i],request,)== )       i += ;

         else if( i+<=len && memcmp(&buff[i],cancel,)== )        i += ;

         else if( i+ <=len && memcmp(&buff[i],port,)== )          i += ;

         // bitfield消息的长度是变化的

         else if( i+ <=len && buff[i+]== )  {

             c[] = buff[i];   c[] = buff[i+];

             c[] = buff[i+]; c[] = buff[i+];

             length = char_to_int(c);

             // 消息长度占4字节,消息本身占length个字节

             if( i++length <= len )  i += +length;

             else { *ok_len = i; return -; }

         }

         // piece消息的长度也是变化的

         else if( i+ <=len && buff[i+]== )  {

             c[] = buff[i];   c[] = buff[i+];

             c[] = buff[i+]; c[] = buff[i+];

             length = char_to_int(c);

             // 消息长度占4字节,消息本身占length个字节

             if( i++length <= len )  i += +length;

             else { *ok_len = i; return -; }

         }

         else {

             // 处理未知类型的消息

             if(i+ <= len) {

                 c[] = buff[i];   c[] = buff[i+];

                 c[] = buff[i+]; c[] = buff[i+];

                 length = char_to_int(c);

                 // 消息长度占4字节,消息本身占length个字节

                 if(i++length <= len)  { i += +length; continue; }

                 else { *ok_len = i; return -; }

             }

             // 如果也不是未知消息类型,则认为目前接收的数据还不是一个完整的消息

             *ok_len = i;

             return -;

         }

     }

     *ok_len = i;

     return ;

 }

process_handshake_msg()处理握手信息

比对info_hash 判断是否需要处理还是丢弃然后从initial进入到 handshaked状态

 int process_handshake_msg(Peer *peer,unsigned char *buff,int len)

 {

     if(peer==NULL || buff==NULL)  return -;

     if(memcmp(info_hash,buff+,) != ) {

         peer->state = CLOSING;

         // 丢弃发送缓冲区中的数据

         discard_send_buffer(peer);

         clear_btcache_before_peer_close(peer);

         close(peer->socket);

         return -;

     }

     memcpy(peer->id,buff+,);

     (peer->id)[] = '\0';

     if(peer->state == INITIAL) {

         peer->state = HANDSHAKED;

         create_handshake_msg(info_hash,peer_id,peer);

     }

     if(peer->state == HALFSHAKED)  peer->state = HANDSHAKED;

     peer->start_timestamp = time(NULL);

     return ;

 }

process_keep_alive_msg() 心跳处理更新下peer的时间不做其他处理

 int process_keep_alive_msg(Peer *peer,unsigned char *buff,int len)

 {

     if(peer==NULL || buff==NULL)  return -;

     peer->start_timestamp = time(NULL);

     return ;

 }

process_choke_msg处理函数如果peer未关闭且处于未堵塞状态更改为堵塞时间重新计数

 int process_choke_msg(Peer *peer,unsigned char *buff,int len)

 {

     if(peer==NULL || buff==NULL)  return -;

     if( peer->state!=CLOSING && peer->peer_choking== ) {

         peer->peer_choking = ;

         peer->last_down_timestamp = ;

         peer->down_count          = ;

         peer->down_rate           = ;

     }

     peer->start_timestamp = time(NULL);

     return ;

 }

process_unchoke_msg 解开堵塞状态创建interested的pieces

 int process_unchoke_msg(Peer *peer,unsigned char *buff,int len)

 {

     if(peer==NULL || buff==NULL)  return -;

     if( peer->state!=CLOSING && peer->peer_choking== ) {

         peer->peer_choking = ;

         if(peer->am_interested == )  create_req_slice_msg(peer);

         else {

             peer->am_interested = is_interested(&(peer->bitmap), bitmap);

             if(peer->am_interested == ) create_req_slice_msg(peer);

             else printf("Received unchoke but Not interested to IP:%s \n",peer->ip);

         }

         peer->last_down_timestamp = ;

         peer->down_count          = ;

         peer->down_rate           = ;

     }

     peer->start_timestamp = time(NULL);

     return ;

 }

process_interested_msg更新时间根据位图更新感兴趣的状态

 int process_interested_msg(Peer *peer,unsigned char *buff,int len)

 {

     if(peer==NULL || buff==NULL)  return -;

     if( peer->state!=CLOSING && peer->state==DATA ) {

         peer->peer_interested = is_interested(bitmap, &(peer->bitmap));

         if(peer->peer_interested == )  return -;

         if(peer->am_choking == ) create_chock_interested_msg(,peer);

     }

     peer->start_timestamp = time(NULL);

     return ;

 }

process_uninterested_msg 更新时间从要求名单中删除peer相关的数据

 int process_uninterested_msg(Peer *peer,unsigned char *buff,int len)

 {

     if(peer==NULL || buff==NULL)  return -;

     if( peer->state!=CLOSING && peer->state==DATA ) {

         peer->peer_interested = ;

         cancel_requested_list(peer);

     }

     peer->start_timestamp = time(NULL);

     return ;

 }

process_have_msg 收到其他peer的have信息更新该peermap 再次检测是否感兴趣。无兴趣的点也有1/3的概率进行回复

 int process_have_msg(Peer *peer,unsigned char *buff,int len)

 {

     int           rand_num;

     unsigned char c[];

     if(peer==NULL || buff==NULL)  return -;

     srand(time(NULL));

     rand_num = rand() % ;

     if( peer->state!=CLOSING && peer->state==DATA ) {

         c[] = buff[]; c[] = buff[];

         c[] = buff[]; c[] = buff[];

         if(peer->bitmap.bitfield != NULL)

             set_bit_value(&(peer->bitmap),char_to_int(c),);

         if(peer->am_interested == ) {

             peer->am_interested = is_interested(&(peer->bitmap), bitmap);

             // 由原来的对peer不感兴趣变为感兴趣时,发interested消息

             if(peer->am_interested == ) create_chock_interested_msg(,peer);

         } else {  // 收到三个have则发一个interested消息

             if(rand_num == ) create_chock_interested_msg(,peer);

         }

     }

     peer->start_timestamp = time(NULL);

     return ;

 }

process_cancel_msg 放弃请求的处理

遍历记录链表删除

 int process_cancel_msg(Peer *peer,unsigned char *buff,int len)

 {

     unsigned char c[];

     int           index, begin, length;

     if(peer==NULL || buff==NULL)  return -;

     c[] = buff[];  c[] = buff[];

     c[] = buff[];  c[] = buff[];

     index = char_to_int(c);

     c[] = buff[];  c[] = buff[];

     c[] = buff[]; c[] = buff[];

     begin = char_to_int(c);

     c[] = buff[]; c[] = buff[];

     c[] = buff[]; c[] = buff[];

     length = char_to_int(c);

     Request_piece *p, *q;

     p = q = peer->Requested_piece_head;

     while(p != NULL) {

         if( p->index==index && p->begin==begin && p->length==length ) {

             if(p == peer->Requested_piece_head)

                 peer->Requested_piece_head = p->next;

             else

                 q->next = p->next;

             free(p);

             break;

         }

         q = p;

         p = p->next;

     }    

     peer->start_timestamp = time(NULL);

     return ;

 }

后面的函数基本都是函数名自解释看看代码即可

参考

《linux c编程实战》第十三章节btcorrent 及代码

http://www.cnblogs.com/UnGeek/p/6052776.html

https://blog.csdn.net/str999_cn/article/details/78880189

bittorrent 学习(三) MSG的更多相关文章

DjangoRestFramework学习三之认证组件、权限组件、频率组件、url注册器、响应器、分页组件
DjangoRestFramework学习三之认证组件.权限组件.频率组件.url注册器.响应器.分页组件本节目录一认证组件二权限组件三频率组件四 URL注册器五响应器六分 ...
day91 DjangoRestFramework学习三之认证组件、权限组件、频率组件、url注册器、响应器、分页组件
DjangoRestFramework学习三之认证组件.权限组件.频率组件.url注册器.响应器.分页组件本节目录一认证组件二权限组件三频率组件四 URL注册器五响应器六分 ...
day 89 DjangoRestFramework学习三之认证组件、权限组件、频率组件、url注册器、响应器、分页组件
DjangoRestFramework学习三之认证组件.权限组件.频率组件.url注册器.响应器.分页组件本节目录一认证组件二权限组件三频率组件四 URL注册器五响应器六分 ...
day 82 Vue学习三之vue组件
Vue学习三之vue组件本节目录一什么是组件二 v-model双向数据绑定三组件基础四父子组件传值五平行组件传值六 xxx 七 xxx 八 xxx 一什么是组件首先给 ...
HTTP学习三：HTTPS
HTTP学习三:HTTPS 1 HTTP安全问题 HTTP1.0/1.1在网络中是明文传输的,因此会被黑客进行攻击. 1.1 窃取数据因为HTTP1.0/1.1是明文的,黑客很容易获得用户的重要数据 ...
TweenMax动画库学习(三)
目录 TweenMax动画库学习(一) TweenMax动画库学习(二) TweenMax动画库学习(三) ...
Struts2框架学习(三) 数据处理
Struts2框架学习(三) 数据处理 Struts2框架框架使用OGNL语言和值栈技术实现数据的流转处理. 值栈就相当于一个容器,用来存放数据,而OGNL是一种快速查询数据的语言. 值栈:Value ...
4.机器学习——统计学习三要素与最大似然估计、最大后验概率估计及L1、L2正则化
1.前言之前我一直对于“最大似然估计”犯迷糊,今天在看了陶轻松.忆臻.nebulaf91等人的博客以及李航老师的<统计学习方法>后,豁然开朗,于是在此记下一些心得体会. “最大似然估计” ...
[ZZ] 深度学习三巨头之一来清华演讲了，你只需要知道这7点
深度学习三巨头之一来清华演讲了,你只需要知道这7点 http://wemedia.ifeng.com/10939074/wemedia.shtml Yann LeCun还提到了一项FAIR开发的,用于 ...

随机推荐

IntelliJ IDEA 破解Jrebel6.3.0安装
首先下载所必要的两个文件(jrebel3.6.0和cracked文件) 密码:pvsd 注意:如果不是该版本的Jrebel该破解文件可能无效. 步骤1:安装解压文件得出两个压缩包在idea中选择 ...
分布式基本理论 CAP
谈及分布式,必然谈到 CAP, CAP 已经是被说烂了的一个话题, 绕不开, 逃不掉. 而且, 理解起来会有些吃力. 分布式的CAP理论告诉我们 “任何一个分布式系统都无法同时满足一致性(Co ...
URL 传递问题
工作日记: ----更正:如下做改成%26是不行的.正在寻求解决方法在K2邮件发送正文中我拼接了URL如:http://shisupportqa:8090/WorkflowPages/SendFil ...
for循环，while循环，do while循环
for循环: for循环格式: for(初始化语句;判断条件语句;控制条件语句) { 循环体语句; } 例子:取五位数各个位数的练习 public static void main(String[] ...
Maven项目强制更新，解决Failed to read artifact descriptor for xxx.jar问题
导入的maven项目pom.xml现红叉分析原因:在maven本地仓库中找不到相应的jar包. 解决方案:让maven强制更新依赖. 项目右击菜单,Maven -> Update Projec ...
mysql安装，以及从csv插入数据
1.mysql安装用管理员身份打开cmd命令行工具,cd到解压文件的bin目录: ・MySql的Windows服务安装: mysqld install 回车・生成无密码的root用户: my ...
VirtualBox安装Ubuntu16.04过程
1. 软件版本 Windows: Win7/Win10 VirtualBox: VirtualBox-6.0.24-108355-Win Ubuntu: ubuntu-16.04-desktop-am ...
SpringCloud-day03-服务注册与发现组件Eureka
5.服务注册与发现组件Eureka 5.1Eureka简介: Eureka是Netflix开发的服务发现框架,本身是一个基于REST的服务,主要用于定位运行在AWS域中的中间层服务,以达到负载均衡和中 ...
fiddler抓取https请求
Fiddler抓取https设置详解(图文):https://www.cnblogs.com/joshua317/p/8670923.html Fiddler要抓取到https请求我们还需要Fiddl ...
探究编译后，try-with-resources括号中的object是否关闭，以及两种写法编译后的对比
源码(@TargetApi(Build.VERSION_CODES.KITKAT)) public List<T> test1() { String sql = "selxe x ...

bittorrent 学习(三) MSG

bittorrent 学习(三) MSG的更多相关文章

随机推荐

热门专题