码迷，mamicode.com
首页 > 其他好文 > 详细

基于UDT connect连接通信以及文件传输--服务端

时间：2017-05-12 18:55:23      阅读：287      评论：0      收藏：0      [点我收藏+]

标签：namespace   地址   strcpy   程序   pre   set   ddr   ref   csdn   

网上与UDT相关的资料不多，与UDT相关的源码例子更少。最近在接触UDT，也是因为缺少相关的资料，导致学习起来甚感痛苦。下面将我自己这两天弄出来的代码贴出来，希望对在寻找相关资料的童鞋有一定的帮助。与服务端相对应的客户端在另一篇博文中

[cpp] view plain copy
 
  1.   

        SERVER端

[cpp] view plain copy
 
  1. #include <iostream>  
  2. #include "udt.h"  
  3. #include <io.h>  
  4.   
  5. #pragma comment(lib,"ws2_32.lib")  
  6.   
  7. using namespace std;  
  8.   
  9. #define MAXLEN 4096  
  10.   
  11. int main(int argc,char *argv[])  
  12.  {     
  13.      if ((argc != 4))  
  14.      {  
  15.         cout<<"Use: appserver.exe server_port client_ip client_port"<<endl;  
  16.         return 0;  
  17.      }  
  18.   
  19.      //startup  
  20.     //这里是对UDT的启动记性初始化操作  
  21.      if (UDT::ERROR == UDT::startup())  
  22.      {  
  23.          cout<<"startup: "<<UDT::getlasterror().getErrorMessage()<<endl;  
  24.      }else{  
  25.         cout<<"startup suc..."<<endl;  
  26.      }  
  27.   
  28.      //socket  
  29.     //像声明一个普通的socket一样声明一个UDTSOCKET  
  30.      UDTSOCKET serv = UDT::socket(AF_INET, SOCK_DGRAM, 0);  
  31.      if (UDT::ERROR == serv)  
  32.     {  
  33.         cout<<"socket: "<<UDT::getlasterror().getErrorMessage()<<endl;  
  34.     }else{  
  35.         cout<<"client suc..."<<endl;  
  36.     }  
  37.   
  38.      //声明udp socket，这里是udp的哈，不是udt  
  39.      int sersocket = socket(AF_INET,SOCK_DGRAM,0);  
  40.      if (SOCKET_ERROR == sersocket)  
  41.     {  
  42.         cout<<"udp socket error!"<<endl;  
  43.     }else{  
  44.         cout<<"clientsocket suc..."<<endl;  
  45.     }  
  46.   
  47.     //为了能够在局域网中直接进行处理，先默认设置两个  
  48.      sockaddr_in my_addr,client_addr;  
  49.      my_addr.sin_family = AF_INET;  
  50.      my_addr.sin_port = htons(atoi(argv[1]));  
  51.      my_addr.sin_addr.s_addr = INADDR_ANY;  
  52.      memset(&(my_addr.sin_zero), ‘\0‘, 8);  
  53.      bind(sersocket,(struct sockaddr*)&my_addr,sizeof(my_addr));  
  54.   
  55.      client_addr.sin_family = AF_INET;  
  56.      client_addr.sin_port = htons(atoi(argv[3]));  
  57.      client_addr.sin_addr.s_addr = inet_addr(argv[2]);  
  58.      //client_addr.sin_addr.s_addr = inet_addr("127.0.0.1");  
  59.      memset(&(client_addr.sin_zero), ‘\0‘, 8);  
  60.   
  61.      int mss = 1052;//最大传输单位  
  62.      //设置收发缓冲区大小 接收限时  和地址重用  
  63.     if(   !( UDT::ERROR != (UDT::setsockopt(serv, 0, UDT_SNDBUF, new int(32000), sizeof(int)))  
  64.         && UDT::ERROR != (UDT::setsockopt(serv, 0, UDP_RCVBUF, new int(32000), sizeof(int)))  
  65.         && UDT::ERROR != (UDT::setsockopt(serv,0,UDT_REUSEADDR,new int(1),sizeof(int)))  
  66.         && UDT::ERROR != (UDT::setsockopt(serv, 0, UDT_RENDEZVOUS, new bool(true), sizeof(bool))))  
  67.         && UDT::ERROR != (UDT::setsockopt(serv, 0, UDT_MSS, &mss, sizeof(int)) ))  
  68.     {  
  69.         cout<<"udt socket: "<<UDT::getlasterror().getErrorMessage()<<endl;  
  70.         UDT::close(serv);  
  71.         return 0;  
  72.     }  
  73.     //这里是直接将udp的接口绑定在udt的接口之上，如果不这样做的话是没法使用UDT中的SOCK_DGRAM的  
  74.      if (UDT::ERROR == UDT::bind2(serv,sersocket))  
  75.      {  
  76.         cout<<"udt bind2:"<<UDT::getlasterror().getErrorMessage()<<endl;  
  77.         return 0;  
  78.      }else{  
  79.         cout<<"bind2 suc"<<endl;  
  80.      }  
  81.     //这里也是关键部分，与client端对应的connect操作，就是相互之间的打洞处理  
  82.      if (UDT::ERROR == UDT::connect(serv, (sockaddr*)&client_addr, sizeof(client_addr)))  
  83.      {  
  84.        cout << "connect: " << UDT::getlasterror().getErrorMessage();  
  85.        UDT::close(serv);  
  86.        return 0;  
  87.      }else{  
  88.         cout<<"connetc suc"<<endl;  
  89.      }  
  90.     //这里已经可以正常接收了，接收从client发过来的filename，目的是用于本地的文件创建  
  91.      char filename[100];  
  92.      if (UDT::ERROR == UDT::recvmsg(serv, filename, 100))  
  93.      {  
  94.        cout << "recv:" << UDT::getlasterror().getErrorMessage() << endl;  
  95.        return 0;  
  96.      }  
  97.      cout <<"filename: "<< filename <<endl;  
  98.   
  99.     //使用FILE进行文件操作，关于文件的相关操作这里不详述了，实在不懂的可以留言  
  100.      FILE *fp;  
  101.      char localfile[100];  
  102.      memset(localfile,0,sizeof(localfile));  
  103.      strcpy(localfile,"d:\\");  
  104.      strcat(localfile,filename);  
  105.      if((fp = fopen(localfile,"w+"))==NULL)  
  106.      {  
  107.         cout<<filename<<" open failure!"<<endl;  
  108.         return 0;  
  109.      }  
  110.      fclose(fp);  
  111.      fp = fopen(localfile,"wb");  
  112.   
  113.      char data[MAXLEN];  
  114.      int len=0,package=0,filelen=0;  
  115.      UDT::TRACEINFO trace;  
  116.      UDT::perfmon(serv,&trace);    
  117.      while(1)  
  118.      {  
  119.         //前面部分打开文件后，这里就是循环接收文件并保存  
  120.         memset(data,0,sizeof(data));  
  121.         len = UDT::recvmsg(serv, data, MAXLEN);  
  122.         filelen += len;  
  123.         //cout<<"filelen = "<<filelen<<endl;  
  124.         if (strncmp("quit",data,4)==0)  
  125.         {  
  126.             cout<<data<<endl;  
  127.             fclose(fp);  
  128.             break;  
  129.         }else  
  130.         {  
  131.             package ++;//record recv all the packages  
  132.         }  
  133.         fwrite(data,len,1,fp);        
  134.      }  
  135.      fclose(fp);  
  136.      fp = fopen(localfile,"rb");  
  137.      fseek(fp,0,SEEK_END);//move to the end  
  138.      filelen = ftell(fp)  
  139.      fseek(fp,0,SEEK_SET);  
  140.      cout<<"filesize = "<<filelen<<endl;  
  141.      UDT::perfmon(serv,&trace);  
  142.      cout << "speed = " << trace.mbpsRecvRate << "Mbits/sec" << endl;  
  143.      cout<<"recv all the packages: "<<package<<endl;  
  144.   
  145.      fclose(fp);  
  146.      UDT::close(serv);  
  147.      UDT::cleanup();  
  148.   
  149.      return 1;  
  150.  }   


 
 
运行效果截图：
技术分享
 
注：要想运行这两个程序直接把源代码复制下来，然后去运行肯定是不行的，这个是需要UDT其它类库的支持的，也就是说，需要你将UDT的源代码文件拷贝进你的项目文件中才行。源代码下载地址：http://blog.csdn.net/pingd/article/details/14519881
简单截图如下：
技术分享

基于UDT connect连接通信以及文件传输--服务端

标签：namespace   地址   strcpy   程序   pre   set   ddr   ref   csdn   

原文地址：http://www.cnblogs.com/lidabo/p/6846647.html
踩
(0)
赞
(0)
   
举报
评论 一句话评论（0）
登录后才能评论！
分享档案
更多>
2021年07月29日 (22)
2021年07月28日 (40)
2021年07月27日 (32)
2021年07月26日 (79)
2021年07月23日 (29)
2021年07月22日 (30)
2021年07月21日 (42)
2021年07月20日 (16)
2021年07月19日 (90)
2021年07月16日 (35)
周排行
mamicode.com排行更多图片
更多
友情链接
兰亭集智   国之画   百度统计   站长统计   阿里云   chrome插件   新版天听网
关于我们 - 联系我们 - 留言反馈
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码！