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

时间：2023-03-08 18:20:35
基于UDT connect连接通信以及文件传输--服务端

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

SERVER端

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

运行效果截图：

基于UDT connect连接通信以及文件传输--服务端
注：要想运行这两个程序直接把源代码复制下来，然后去运行肯定是不行的，这个是需要UDT其它类库的支持的，也就是说，需要你将UDT的源代码文件拷贝进你的项目文件中才行。源代码下载地址：http://blog.csdn.net/pingd/article/details/14519881
简单截图如下：

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