UDT源码剖析（二）：UDT自带例程recvfile注释

#ifndef WIN32
   #include <arpa/inet.h>
   #include <netdb.h>
#else
   #include <winsock2.h>
   #include <ws2tcpip.h>
#endif
#include <fstream>
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <udt.h>


using namespace std;


int main(int argc, char* argv[])
{
   if ((argc != 5) || (0 == atoi(argv[2])))
   {
      cout << "usage: recvfile server_ip server_port remote_filename local_filename" << endl;
      return -1;
   }


   // use this function to initialize the UDT library
   // 初始化UDT库
   UDT::startup();


   // 地址信息提示、目标的地址信息指针
   struct addrinfo hints, *peer;


   // 内存初始化，全部置为0，设置同sendfile
   memset(&hints, 0, sizeof(struct addrinfo));
   hints.ai_flags = AI_PASSIVE;
   hints.ai_family = AF_INET;
   hints.ai_socktype = SOCK_STREAM;


   // 创建一个本地Socket
   UDTSOCKET fhandle = UDT::socket(hints.ai_family, hints.ai_socktype, hints.ai_protocol);


   // 根据传入参数和地址信息提示创建目标地址信息
   if (0 != getaddrinfo(argv[1], argv[2], &hints, &peer))
   {
      cout << "incorrect server/peer address. " << argv[1] << ":" << argv[2] << endl;
      return -1;
   }


   // connect to the server, implict bind
   // 连接到服务器，第一个参数为Socket，第二个为连接地址，第三个为地址长度（底层用来判断是否符合IPv4、IPv6标准的长度）
   if (UDT::ERROR == UDT::connect(fhandle, peer->ai_addr, peer->ai_addrlen))
   {
      cout << "connect: " << UDT::getlasterror().getErrorMessage() << endl;
      return -1;
   }


   // 释放通过getaddrinfo分配的地址信息
   freeaddrinfo(peer);


   // send name information of the requested file
   // 发送请求的文件名称
   int len = strlen(argv[3]);


   // 向目标发送文件名的长度
   // 第一个参数为Socket，第二个参数为数据地址，第三个参数为数据长度，第四个参数会被底层忽略，没有意义
   if (UDT::ERROR == UDT::send(fhandle, (char*)&len, sizeof(int), 0))
   {
      cout << "send: " << UDT::getlasterror().getErrorMessage() << endl;
      return -1;
   }


   // 向目标发送文件名
   // 第一个参数为Socket，第二个参数为数据地址，第三个参数为数据长度，第四个参数会被底层忽略，没有意义
   if (UDT::ERROR == UDT::send(fhandle, argv[3], len, 0))
   {
      cout << "send: " << UDT::getlasterror().getErrorMessage() << endl;
      return -1;
   }


   // get size information
   // 接收文件大小
   int64_t size;


   // 接收一个文件大小数据，内容为一个int64_t值，长度为int64_t的长度，数值保存在size中
   // 第一个参数为Socket，第二个参数为数据保存地址，第三个参数为数据长度，第四个参数底层会忽略，没有意义
   if (UDT::ERROR == UDT::recv(fhandle, (char*)&size, sizeof(int64_t), 0))
   {
      cout << "send: " << UDT::getlasterror().getErrorMessage() << endl;
      return -1;
   }


   // 负值表示没有该文件
   if (size < 0)
   {
      cout << "no such file " << argv[3] << " on the server\n";
      return -1;
   }


   // receive the file
   // 开始接收文件
   // 通过二进制方式写入文件，如果文件已存在则覆盖
   fstream ofs(argv[4], ios::out | ios::binary | ios::trunc);


   // 接收到的数据大小以及偏移量
   int64_t recvsize;
   int64_t offset = 0;


   // 从0位置开始接收文件数据
   // 第一个参数为Socket，第二个参数为文件流，第三个参数为偏移量，第四个参数为文件大小
   if (UDT::ERROR == (recvsize = UDT::recvfile(fhandle, ofs, offset, size)))
   {
      cout << "recvfile: " << UDT::getlasterror().getErrorMessage() << endl;
      return -1;
   }


   // 关闭连接Socket
   UDT::close(fhandle);


   // 关闭文件流
   ofs.close();


   // use this function to release the UDT library
   // 释放UDT库
   UDT::cleanup();


   return 0;
}