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(, );
       //初始化网络
       if (WSAStartup(socketversion,&wsadata)!=)
       {
           printf("网络初始化失败");
           return;
       }
       //创建TCP通信
       SOCKET  client_socket = socket(AF_INET, SOCK_STREAM, );
       //判断是否创建成功
       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[] = {  };
       //初始化
       scanf("%s", filename);
  
       char buffer[] = {  };
       memset(buffer, , );
       strcpy(buffer, filename);
  
       //传递文件名，请求下载
       if (send(client_socket,buffer,,) <)//传输出错
       {
           puts("传输错误");
           return;
       }
  
       //写文件
       FILE *pf= fopen(filename, "wb");
       if (pf==NULL)
       {
           puts("文件打开失败");
           return;
  
       }
       else
       {
           memset(buffer, , );
           int length = ;
           printf("文件开始传输");
           //不断接收
           while ((length=recv(client_socket,buffer,,))>)
           {
               if (fwrite(buffer,sizeof(char),length,pf)<length)
               {
                   puts("文件传输错误");
               }
               memset(buffer, , );
           }
           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(, );//定义版本
       if (WSAStartup(socketversion, &wsadata) != )
       {
           printf("网络初始化失败");
           return;
       }
       //创建
       SOCKET server_socket = socket(AF_INET, SOCK_STREAM, );//创建
       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,))
       {
           printf("listen 出错");
           return;
       }
       //传文件
  
       while ()
       {
           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[] = {  };
           char filename[] = {  };
           //接收信息
           if (recv(new_socket,buffer,,)<)
           {
               printf("收取文件名失败");
               break;
           }
           strcpy(filename, buffer);
  
           //以二进制方式读取
           FILE *pf = fopen(filename, "rb");
           if (pf == NULL)
           {
               printf("文件打开失败");
               break;
           }
           else
           {
               memset(buffer, , );//清零
               int recelength = ;
               while ((recelength = fread(buffer, sizeof(char), , pf)>))
               {
                   if (send(new_socket,buffer,,)<)
                   {
                       printf("文件传输失败");
                       break;
                   }
                   memset(buffer, , );//清零
               }
               fclose(pf);
               printf("传输完成");
  
           }
  
           closesocket(new_socket);
       }
       closesocket(server_socket);
       system("pause");
       WSACleanup();
   }