网络编程(二).UDP

#include <stdio.h> //perror,printf  相关函数的声明包含在内
#include <netinet/in.h> //sockaddr_in,socket,AF_INET,SOCK_DGRAM,htons,htonl,INADDR_ANY,setsockopt,SOL_SOCKET,SO_REUSEADDR,bind,recvfrom,sendto 相关声明和定义包含在内
#include <string.h> //memset 相关函数的声明包含在内
#include <unistd.h> //write,close 相关函数的声明包含在内
#include <fcntl.h> //open,O_RDWR,O_CREAT,O_TRUNC 相关函数的声明和宏定义包含在内

#define MAX_CONN 2
#define BUF_SIZE 1024
#define PORT 9000 

int main()
{
  struct sockaddr_in server_sai,client_sai;
  int sfd=0,res=-1,on=1,recvbytes=0,sendbytes=0,writebytes=0,fa=0;
  int addrlen=sizeof(struct sockaddr);
  char buf[BUF_SIZE];
  char *filename="/tmp/x.download"; //进行各种变量的定义和初始化

  if(-1 == (sfd=socket(AF_INET,SOCK_DGRAM,0))) //创建一个IPV4的UDP socket
  {
    perror("socket");
    return res;
  }
  
  server_sai.sin_family=AF_INET; //IPV4 协议族
  server_sai.sin_port=htons(PORT); //9000端口
  server_sai.sin_addr.s_addr=htonl(INADDR_ANY); //0.0.0.0 的通配监听
  
  memset(&(server_sai.sin_zero),0,sizeof(server_sai.sin_zero)); //将剩余部分填零

  setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&on,sizeof(on)); //closesocket（一般不会立即关闭而经历TIME_WAIT的过程）后想继续重用该socket

  if(-1 == bind(sfd,(struct sockaddr *)&server_sai,sizeof(struct sockaddr))) //将 sfd 和 socket 地址进行绑定
  {
    perror("bind");
    return res;
  }
  
  if (-1==(fa=open(filename,O_RDWR|O_CREAT|O_TRUNC,0644))) //以写的方式打开目标文件，也就是服务端数据的存放处
  {
    printf("cannot open file:%s\n",filename);
    return res;
  }

  memset(buf,0,sizeof(buf)); //将缓存置零

  int i=0; //设定一个变量来追踪分包数，可选项

  do
  {
    i++; //每分一包，加一次，可选项
    if(-1 == (recvbytes = recvfrom(sfd,buf,sizeof(buf),0,(struct sockaddr *)&client_sai,(socklen_t *)&addrlen))) //从远端获取数据，放到buf中
    {
      perror("recvfrom");
      return res;
    }
    if(-1 == (writebytes = write(fa,buf,recvbytes))) //将buf中的数据写到文件中
    {
      printf("write error on:%s\n",filename);
      return res;
    }
    if(-1  == (sendbytes = sendto(sfd,"DONE",4,0,(struct sockaddr *)&client_sai,sizeof(struct sockaddr)))) //发送一个确认信息给远端，来同步节奏
    {
      perror("sendto");
      return res;
    }
  }while(recvbytes == sizeof(buf)); //如果读到的数据小于一整块，就意味着数据已经读完，跳出循环

  printf("i:%d\nrecvbytes:%d\n",i,recvbytes); //将分包的数目和最后一包数据的大小显示出来，可选项
  
  if(-1  == (sendbytes = sendto(sfd,"DONE",4,0,(struct sockaddr *)&client_sai,sizeof(struct sockaddr)))) //发送一个确认信息给远端
  {
    perror("sendto");
    return res;
  }
  
  close(fa);
  close(sfd); //进行清理操作，关闭所有描述符

  res=0;
  return res;
}