网络编程(一).TCP（1）

#include <stdio.h> //perror,printf 相关函数的声明包含在内
#include <netinet/in.h> //sockaddr_in,socket,AF_INET,SOCK_STREAM,htons,htonl,INADDR_ANY,setsockopt,SOL_SOCKET,SO_REUSEADDR,bind,listen,accept,recv,send 相关声明和定义包含在内
#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,cfd=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_STREAM,0))) //创建一个IPV4的TCP 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 == (listen(sfd,MAX_CONN)))  //在sfd上进行监听，最多允许同时有2个请求在队列中排队，此配置正是DDOS的攻击点，协议天然的缺陷在于，不论这个值设多设少，都不会是一个适合的值
  {
    perror("listen");
    return res;
  }
  else printf("Listening...\n");

  if(-1 == (cfd=accept(sfd,(struct sockaddr *)&client_sai,(socklen_t *)&addrlen)))   //接受连接，将返回的描述符赋给cfd
  {
    perror("accept");
    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)); //将缓存置零

  do
  {
    if(-1 == (recvbytes = recv(cfd,buf,sizeof(buf),0))) //从网络端接收数据，并且存到buf中
    {
      perror("recv");
      return res;
    }
    if(-1 == (writebytes = write(fa,buf,recvbytes))) //将buf中的数据写到文件中
    {
      printf("write error on:%s\n",filename);
      return res;
    }

  }while(recvbytes == sizeof(buf)); //如果读到的数据小于一整块了，就意味着数据已经读完，跳出循环

  
  if(-1  == (sendbytes = send(cfd,"OK",2,0))) //发送一个读完的信号给远端
  {
    perror("send");
    return res;
  }
  
  close(fa);
  close(sfd);
  close(cfd); //进行清理操作，关闭所有描述符
  res=0;
  return res;
}