106.TCP传文件

• 客户端

  #define  _CRT_SECURE_NO_WARNINGS
  #include <stdio.h>
  #include <stdlib.h>
  #include <winsock.h>
  #include <string.h>
  #pragma  comment(lib,"ws2_32.lib")
  
  //定义端口
  #define PORT 6666
  //缓冲区大小
  #define BUF_SIZE 1024
  //文件名大小
  #define FILENAMESIZE  512
  //本地地址
  #define SIP "192.168.1.108"
  
  //主函数
  void main()
  {
      //用于初始化网络
      WSADATA wsadata;
      //定义版本
      WORD socketversion = MAKEWORD(2, 2);
      //初始化网络
      if (WSAStartup(socketversion,&wsadata)!=0)
      {
          printf("网络初始化失败");
          return;
      }
      //创建TCP通信
      SOCKET  client_socket = socket(AF_INET, SOCK_STREAM, 0);
      //判断是否创建成功
      if (client_socket==SOCKET_ERROR)
      {
          printf("网络创建通信socket出错");
          return;
      }
      //设定服务器信息
      SOCKADDR_IN sever_addr;
      sever_addr.sin_family = AF_INET;
      sever_addr.sin_addr.S_un.S_addr = inet_addr(SIP);
      //设定IP，端口
      sever_addr.sin_port = htons(PORT);
      //连接到服务器
      if (connect(client_socket, (SOCKADDR*)&sever_addr, sizeof(sever_addr)) == SOCKET_ERROR)
      {
          puts("连接错误");
          return;
      }
      char filename[512] = { 0 };
      //初始化
      scanf("%s", filename);
  
      char buffer[1024] = { 0 };
      memset(buffer, 0, 1024);
      strcpy(buffer, filename);
  
      //传递文件名，请求下载
      if (send(client_socket,buffer,1024,0) <0)//传输出错
      {
          puts("传输错误");
          return;
      }
  
      //写文件
      FILE *pf= fopen(filename, "wb");
      if (pf==NULL)
      {
          puts("文件打开失败");
          return;
  
  
      }
      else
      {
          memset(buffer, 0, 1024);
          int length = 0;
          printf("文件开始传输");
          //不断接收
          while ((length=recv(client_socket,buffer,1024,0))>0)
          {
              if (fwrite(buffer,sizeof(char),length,pf)<length)
              {
                  puts("文件传输错误");
              }
              memset(buffer, 0, 1024);
          }
          printf("文件传输成功");
          system("pause");
          system(filename);
      }
      //关闭文件网络，清理网络
      fclose(pf);
      closesocket(client_socket);
      WSACleanup();
  }

• 服务器端

  #define  _CRT_SECURE_NO_WARNINGS
  #include <stdio.h>
  #include <stdlib.h>
  #include <winsock.h>
  #include <string.h>
  #pragma  comment(lib,"ws2_32.lib")
  
  #define PORT 6666
  #define BUF_SIZE 1024
  #define FILENAMESIZE  512
  #define SIP "169.254.29.232"
  
  void main()
  {
      //初始化服务器ip地址与端口号并定义谁可以连接
      SOCKADDR_IN server_addr;
      server_addr.sin_family = AF_INET;
      server_addr.sin_port = htons(PORT);
      //本地ip地址
      server_addr.sin_addr.S_un.S_addr = INADDR_ANY;
      WSADATA wsadata;
      WORD socketversion = MAKEWORD(2, 2);//定义版本
      if (WSAStartup(socketversion, &wsadata) != 0)
      {
          printf("网络初始化失败");
          return;
      }
      //创建
      SOCKET server_socket = socket(AF_INET, SOCK_STREAM, 0);//创建
      if (server_socket == SOCKET_ERROR)
      {
          printf("网络创建通信socket出错");
          return;
      }
      //绑定
      if (SOCKET_ERROR == bind(server_socket,(SOCKADDR*) &server_addr, sizeof(server_addr)))
      {
          printf("bind 出错");
          return;
      }
      //监听
      if (SOCKET_ERROR == listen(server_socket,10))
      {
          printf("listen 出错");
          return;
      }
      //传文件
  
      while (1)
      {
          printf("进入listen，开始等待客户端");
          //客户端地址
          SOCKADDR_IN client_addr;
          int clientlength = sizeof(client_addr);
          //接收连接
          SOCKET new_socket = accept(server_socket, (SOCKADDR*)&client_addr, &clientlength);
          if (SOCKET_ERROR == new_socket)
          {
              printf("服务器已经满或者链接失败");
              break;
          }
          char buffer[1024] = { 0 };
          char filename[512] = { 0 };
          //接收信息
          if (recv(new_socket,buffer,1024,0)<0)
          {
              printf("收取文件名失败");
              break;
          }
          strcpy(filename, buffer);
          
          //以二进制方式读取
          FILE *pf = fopen(filename, "rb");
          if (pf == NULL)
          {
              printf("文件打开失败");
              break;
          } 
          else
          {
              memset(buffer, 0, 1024);//清零
              int recelength = 0;
              while ((recelength = fread(buffer, sizeof(char), 1024, pf)>0))
              {
                  if (send(new_socket,buffer,1024,0)<0)
                  {
                      printf("文件传输失败");
                      break;
                  }
                  memset(buffer, 0, 1024);//清零
              }
              fclose(pf);
              printf("传输完成");
  
          }
  
          closesocket(new_socket);
      }
      closesocket(server_socket);
      system("pause");
      WSACleanup();
  }