怎样实现UDP可靠传输

相对于TCP,由于UDP是面向无连接、不可靠的传输协议。
如果我们要用UDP去实现可靠的传输,则需要解决两个问题:丢包和后发先至(包的顺序)。

解决方法:
1)给数据包编号,按照包的顺序接收并存储;
2)接收端接收到数据包后发送确认信息给发送端,发送端接收确认数据以后再继续发送下一个包,如果接收端收到的数据包的编号不是期望的编号,则要求发送端重新发送。

下面是一个示例程序:
该程序定义一个包的结构体,其中包含数据和包头,包头里包含有包的编号和数据大小,经过测试后,该程序可以成功传输一个视频文件。
具体实现代码如下:

1.server端代码如下:

#include<sys/types.h> 
#include<sys/socket.h> 
#include<unistd.h> 
#include<netinet/in.h> 
#include<arpa/inet.h> 
#include<stdio.h> 
#include<stdlib.h> 
#include<errno.h> 
#include<netdb.h> 
#include<stdarg.h> 
#include<string.h> 

#define SERVER_PORT 8000 
#define BUFFER_SIZE 1024 
#define FILE_NAME_MAX_SIZE 512 

/* 包头 */
typedef struct
{ 
  int id; 
  int buf_size; 
}PackInfo; 

/* 接收包 */
struct SendPack 
{ 
  PackInfo head; 
  char buf[BUFFER_SIZE]; 
} data; 

int main() 
{ 
  /* 发送id */
  int send_id = 0; 

  /* 接收id */
  int receive_id = 0; 

  /* 创建UDP套接口 */
  struct sockaddr_in server_addr; 
  bzero(&server_addr, sizeof(server_addr)); 
  server_addr.sin_family = AF_INET; 
  server_addr.sin_addr.s_addr = htonl(INADDR_ANY); 
  server_addr.sin_port = htons(SERVER_PORT); 

  /* 创建socket */
  int server_socket_fd = socket(AF_INET, SOCK_DGRAM, 0); 
  if(server_socket_fd == -1) 
  { 
    perror("Create Socket Failed:"); 
    exit(1); 
  } 

  /* 绑定套接口 */
  if(-1 == (bind(server_socket_fd,(struct sockaddr*)&server_addr,sizeof(server_addr)))) 
  { 
    perror("Server Bind Failed:"); 
    exit(1); 
  } 

  /* 数据传输 */
  while(1) 
  {   
    /* 定义一个地址,用于捕获客户端地址 */
    struct sockaddr_in client_addr; 
    socklen_t client_addr_length = sizeof(client_addr); 

    /* 接收数据 */
    char buffer[BUFFER_SIZE]; 
    bzero(buffer, BUFFER_SIZE); 
    if(recvfrom(server_socket_fd, buffer, BUFFER_SIZE,0,(struct sockaddr*)&client_addr, &client_addr_length) == -1) 
    { 
      perror("Receive Data Failed:"); 
      exit(1); 
    } 

    /* 从buffer中拷贝出file_name */
    char file_name[FILE_NAME_MAX_SIZE+1]; 
    bzero(file_name,FILE_NAME_MAX_SIZE+1); 
    strncpy(file_name, buffer, strlen(buffer)>FILE_NAME_MAX_SIZE?FILE_NAME_MAX_SIZE:strlen(buffer)); 
    printf("%s\n", file_name); 

    /* 打开文件 */
    FILE *fp = fopen(file_name, "r"); 
    if(NULL == fp) 
    { 
      printf("File:%s Not Found.\n", file_name); 
    } 
    else
    { 
      int len = 0; 
      /* 每读取一段数据,便将其发给客户端 */
      while(1) 
      { 
        PackInfo pack_info; 

        if(receive_id == send_id) 
        { 
          ++send_id; 
          if((len = fread(data.buf, sizeof(char), BUFFER_SIZE, fp)) > 0) 
          { 
            data.head.id = send_id; /* 发送id放进包头,用于标记顺序 */
            data.head.buf_size = len; /* 记录数据长度 */
            if(sendto(server_socket_fd, (char*)&data, sizeof(data), 0, (struct sockaddr*)&client_addr, client_addr_length) < 0) 
            { 
              perror("Send File Failed:"); 
              break; 
            } 
            /* 接收确认消息 */
            recvfrom(server_socket_fd, (char*)&pack_info, sizeof(pack_info), 0, (struct sockaddr*)&client_addr, &client_addr_length); 
            receive_id = pack_info.id;  
          } 
          else
          { 
            break; 
          } 
        } 
        else
        { 
          /* 如果接收的id和发送的id不相同,重新发送 */
          if(sendto(server_socket_fd, (char*)&data, sizeof(data), 0, (struct sockaddr*)&client_addr, client_addr_length) < 0) 
          { 
            perror("Send File Failed:"); 
            break; 
          } 
          /* 接收确认消息 */
          recvfrom(server_socket_fd, (char*)&pack_info, sizeof(pack_info), 0, (struct sockaddr*)&client_addr, &client_addr_length); 
          receive_id = pack_info.id;  
        } 
      } 
      /* 关闭文件 */
      fclose(fp); 
      printf("File:%s Transfer Successful!\n", file_name); 
    } 
  } 
  close(server_socket_fd); 
  return 0; 
}

2.client端代码如下:

#include<sys/types.h> 
#include<sys/socket.h> 
#include<unistd.h> 
#include<netinet/in.h> 
#include<arpa/inet.h> 
#include<stdio.h> 
#include<stdlib.h> 
#include<errno.h> 
#include<netdb.h> 
#include<stdarg.h> 
#include<string.h> 

#define SERVER_PORT 8000 
#define BUFFER_SIZE 1024 
#define FILE_NAME_MAX_SIZE 512 

/* 包头 */
typedef struct 
{ 
  int id; 
  int buf_size; 
}PackInfo; 

/* 接收包 */
struct RecvPack 
{ 
  PackInfo head; 
  char buf[BUFFER_SIZE]; 
} data; 


int main() 
{ 
  int id = 1; 

  /* 服务端地址 */
  struct sockaddr_in server_addr; 
  bzero(&server_addr, sizeof(server_addr)); 
  server_addr.sin_family = AF_INET; 
  server_addr.sin_addr.s_addr = inet_addr("127.0.0.1"); 
  server_addr.sin_port = htons(SERVER_PORT); 
  socklen_t server_addr_length = sizeof(server_addr); 

  /* 创建socket */
  int client_socket_fd = socket(AF_INET, SOCK_DGRAM, 0); 
  if(client_socket_fd < 0) 
  { 
    perror("Create Socket Failed:"); 
    exit(1); 
  } 

  /* 输入文件名到缓冲区 */
  char file_name[FILE_NAME_MAX_SIZE+1]; 
  bzero(file_name, FILE_NAME_MAX_SIZE+1); 
  printf("Please Input File Name On Server: "); 
  scanf("%s", file_name); 

  char buffer[BUFFER_SIZE]; 
  bzero(buffer, BUFFER_SIZE); 
  strncpy(buffer, file_name, strlen(file_name)>BUFFER_SIZE?BUFFER_SIZE:strlen(file_name)); 

  /* 发送文件名 */
  if(sendto(client_socket_fd,buffer,BUFFER_SIZE,0,(struct sockaddr*)&server_addr,server_addr_length) < 0) 
  { 
    perror("Send File Name Failed:"); 
    exit(1); 
  } 

  /* 打开文件,准备写入 */
  FILE *fp = fopen(file_name, "w"); 
  if(NULL == fp) 
  { 
    printf("File:\t%s Can Not Open To Write\n", file_name);  
    exit(1); 
  } 

  /* 从服务器接收数据,并写入文件 */
  int len = 0; 
  while(1) 
  { 
    PackInfo pack_info; 

    if((len = recvfrom(client_socket_fd, (char*)&data, sizeof(data), 0, (struct sockaddr*)&server_addr,&server_addr_length)) > 0) 
    { 
      if(data.head.id == id) 
      { 
        pack_info.id = data.head.id; 
        pack_info.buf_size = data.head.buf_size; 
        ++id; 
        /* 发送数据包确认信息 */
        if(sendto(client_socket_fd, (char*)&pack_info, sizeof(pack_info), 0, (struct sockaddr*)&server_addr, server_addr_length) < 0) 
        { 
          printf("Send confirm information failed!"); 
        } 
        /* 写入文件 */
        if(fwrite(data.buf, sizeof(char), data.head.buf_size, fp) < data.head.buf_size) 
        { 
          printf("File:\t%s Write Failed\n", file_name); 
          break; 
        } 
      } 
      else if(data.head.id < id) /* 如果是重发的包 */
      { 
        pack_info.id = data.head.id; 
        pack_info.buf_size = data.head.buf_size; 
        /* 重发数据包确认信息 */
        if(sendto(client_socket_fd, (char*)&pack_info, sizeof(pack_info), 0, (struct sockaddr*)&server_addr, server_addr_length) < 0) 
        { 
          printf("Send confirm information failed!"); 
        } 
      } 
      else
      {} 
    } 
    else
    { 
      break; 
    } 
  } 

  printf("Receive File:\t%s From Server IP Successful!\n", file_name); 
  fclose(fp); 
  close(client_socket_fd); 
  return 0; 
}
  • 7
    点赞
  • 55
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值