C++ socket实现miniFTP

#pragma  #include<winsock.h> class ftpClient { private:   enum {     SERVER_PORT = 9999,     BUFFER_SIZE = 4096   };   sockaddr_in serverChannel;   char buffer[BUFFER_SIZE];   int serverSocket;   int clientSocket;   bool isConnect;   char name[50];    bool getFile();   bool putFile();   bool acknowledge();   bool sendRequest(char* instruction);   bool connect2Host(const char* hostName);   bool getWorkDir();  public:   ftpClient();   ~ftpClient();   void start(); }; 

#define _CRT_SECURE_NO_WARNINGS #include"ftpClient.h" #include<cstdio> #include<io.h> #include<cstring> #include<fstream>  ftpClient::ftpClient() {   WORD wVersionRequested;   WSADATA wsaData;   int ret;    //WinSock初始化：   wVersionRequested = MAKEWORD(2, 2);//希望使用的WinSock DLL的版本   ret = WSAStartup(wVersionRequested, &wsaData);   if (ret != 0)   {     printf("WSAStartup() failed!/n");   }   //确认WinSock DLL支持版本2.2：   if (LOBYTE(wsaData.wVersion) != 2 || HIBYTE(wsaData.wVersion) != 2)   {     WSACleanup();     printf("Invalid Winsock version!/n");   }   isConnect = false; }  void ftpClient::start() {   char c[100];   char d[100];   printf("这里是FTP客户端，您可以输入help查看操作方法,输入quit退出客户端/n");    while (1) {     scanf("%s", c);     if (strcmp(c, "help") == 0) {       printf("get [fileName]  --  下载文件/n"         "put [fileName]  --  上传文件/n"         "ftp [ip]     --  登录FTP/n"         "pwd        --  显示服务器当前工作文件夹/n"         "cd [dirName]   --  更改当前文件夹/n"         "close       --  关闭与当前ftp的连接/n"         "quit       --  退出客户端/n"         );     }     else if (strcmp(c, "get") == 0) {       scanf("%s", d);       strcat(c, " ");       strcat(c, d);       if (!isConnect) {         printf("you haven't connected to any server!/n");       }       else sendRequest(c);     }     else if (strcmp(c, "put") == 0) {       scanf("%s", d);       strcat(c, " ");       strcat(c, d);       if (!isConnect) {         printf("you haven't connected to any server!/n");       }       else sendRequest(c);     }     else if (strcmp(c, "ftp") == 0) {       scanf("%s", d);       if (!isConnect&&connect2Host(d)) {         isConnect = true;       }       else if(isConnect){         printf("you have already connected to server/n"           "please close the connection before connect to a new server/n");       }     }     else if (strcmp(c, "pwd") == 0) {       if (!isConnect) {         printf("you haven't connected to any server!/n");       }       else sendRequest(c);     }     else if (strcmp(c, "cd") == 0) {       scanf("%s", d);       strcat(c, " ");       strcat(c, d);       if (!isConnect) {         printf("you haven't connected to any server!/n");       }       else sendRequest(c);     }     else if (strcmp(c, "quit") == 0) {       if (isConnect) {         strcpy(c, "close");         isConnect = false;         send(clientSocket, c, strlen(c) + 1, 0);         closesocket(clientSocket);       }       break;     }     else if (strcmp(c, "close") == 0) {       if (isConnect) {         isConnect = false;         send(clientSocket, c, strlen(c) + 1, 0);         closesocket(clientSocket);       }     }     else {       printf("syntex error/n");     }   } }  bool ftpClient::connect2Host(const char* hostName) {   //创建socket   clientSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);    if (clientSocket < 0) {     printf("cannot create socket/n");     return false;   }   else printf("successfully create socket/n");   memset(&serverChannel, 0, sizeof(serverChannel));//初始化为0    serverChannel.sin_family = AF_INET;//channel协议家族AF_INET   serverChannel.sin_addr.S_un.S_addr = inet_addr(hostName);//地址   serverChannel.sin_port = htons(SERVER_PORT);//服务器端口                          //建立连接   serverSocket = connect(clientSocket, (sockaddr*)&serverChannel, sizeof(serverChannel));    if (serverSocket < 0) {     printf("cannot connect to the host/n");     return false;   }   else {     printf("successfully connect to the host/n");     return true;   } }  bool ftpClient::sendRequest(char* instruction) {   int r = send(clientSocket, instruction, strlen(instruction) + 1, 0);   if (r == SOCKET_ERROR) {     printf("request failed/n");     return false;   }   else {     printf("request success/n");     char opt[5];     int i = 0, j = 0;     while (instruction[i] != ' '&&instruction[i] != '/0') {       opt[i] = instruction[i];       i++;     }     opt[i] = '/0';     i++;     while (instruction[i] != '/0') {       name[j] = instruction[i];       i++, j++;     }     name[j] = '/0';     if (strcmp(opt, "get") == 0) {       if (getFile()) {         printf("successfully download/n");       }       else printf("download failed/n");     }     else if (strcmp(opt, "put") == 0) {       if (putFile()) {         printf("successfully upload/n");       }       else printf("upload failed/n");     }     else if (strcmp(opt, "pwd") == 0) {       if (!getWorkDir())         printf("get work directory failed/n");     }     else if (strcmp(opt, "cd") == 0) {       printf("operation finished/n");     }     else {       printf("syntex error/n");       return false;     }     return true;   } }  bool ftpClient::getFile() {   memset(buffer, 0, sizeof(buffer));   int ret;   char length[20];   ret = recv(clientSocket, length, sizeof(length), 0);   if (ret == SOCKET_ERROR) {     return false;   }   else if (strcmp(length, "NAK") == 0) {     return false;   }   int size = atoi(length);   std::ofstream out;    out.open(name, std::ios::binary);   if (!out) {     printf("cannot save the file/n");     return false;   }   while (size>0) {     ret = recv(clientSocket, buffer, BUFFER_SIZE, 0);     int s = size < BUFFER_SIZE ? size : BUFFER_SIZE;     if (ret == SOCKET_ERROR) {       out.close();       return false;     }     else if (strcmp(buffer, "NAK") == 0) {       out.close();       return false;     }     else {       out.write(buffer, s);     }     size -= BUFFER_SIZE;   }   out.close();   return acknowledge(); }  bool ftpClient::putFile() {   std::ifstream in;   //打开文件   in.open(name, std::ios::binary);   if (!in) {     printf("cannot open the file/n");     return false;   }   memset(buffer, 0, sizeof(buffer));   //得到文件的字节数   in.seekg(0, std::ios_base::end);   int sp = in.tellg();   int total_size = 0;   int r;   char length[20];   sprintf(length, "%d", sp);    //发送字节   r = send(clientSocket, length, sizeof(length), 0);   if (r == SOCKET_ERROR) {     return false;   }   while (sp > 0) {     in.clear();     in.seekg(total_size, std::ios_base::beg);     memset(buffer, 0, sizeof(buffer));     //读取文件     in.read(buffer, sizeof(buffer));     int size = sp < BUFFER_SIZE ? sp : BUFFER_SIZE;     total_size += size;     //发送文件     r = send(clientSocket, buffer, size, 0);      sp -= size;     if (r == SOCKET_ERROR) {       in.close();       return false;     }   }   in.close(); }  bool ftpClient::getWorkDir() {   printf("getWorkDir/n");   memset(buffer, 0, sizeof(buffer));   int ret;   char length[20];   ret = recv(clientSocket, length, sizeof(length), 0);   if (ret == SOCKET_ERROR) {     return false;   }   int size = atoi(length);   while (size>0) {     ret = recv(clientSocket, buffer, BUFFER_SIZE, 0);     if (ret == SOCKET_ERROR) {       return false;     }     else {       printf("%s", buffer);     }     size -= BUFFER_SIZE;   }   return true; }  bool ftpClient::acknowledge() {   int ret = recv(clientSocket, buffer, BUFFER_SIZE, 0);   if (ret > 0) {     if (strcmp(buffer, "NAK") == 0)return false;     else if (strcmp(buffer, "ACK") == 0)return true;   } }  ftpClient::~ftpClient() {   if (isConnect) {     isConnect = false;     char c[6];     strcpy(c, "close");     send(clientSocket, c, strlen(c) + 1, 0);     closesocket(clientSocket);   } } 

#define _CRT_SECURE_NO_WARNINGS #define _WINSOCK_DEPRECATED_NO_WARNINGS #pragma(lib,"ws2_32.lib") #include"ftpClient.h" #include<stdio.h>  int main() {   ftpClient a;   a.start();   return 0; } 

#pragma once #include<winsock.h>  class ftpServer { private:   enum {     SERVER_PORT = 9999,     BUFFER_SIZE = 4096,     QUEUE_SIZE = 10   };   char buffer[BUFFER_SIZE];   sockaddr_in serverChannel;   char name[50];   char workDir[100]; //store like C:/Users MARK:字符串末没有斜线！！   int serverSocket; //socket   int clientSocket;   bool sendFile();   bool receiveFile();   bool doPwd();   bool doCd();   bool isValidPath(char* path); public:   ftpServer();   bool start();//开启服务器 }; ftpServer.cpp

#define _CRT_SECURE_NO_WARNINGS  #include"ftpServer.h" #include<cstdio> #include<cstdlib> #include<fstream> #include<cstring>  ftpServer::ftpServer() {   WORD wVersionRequested;   WSADATA wsaData;   int ret;    //WinSock初始化：   wVersionRequested = MAKEWORD(2, 2);//希望使用的WinSock DLL的版本   ret = WSAStartup(wVersionRequested, &wsaData);   if (ret != 0)   {     printf("WSAStartup() failed!/n");   }   //确认WinSock DLL支持版本2.2：   if (LOBYTE(wsaData.wVersion) != 2 || HIBYTE(wsaData.wVersion) != 2)   {     WSACleanup();     printf("Invalid Winsock version!/n");   }   //workDir初始化为当前路径   system("cd > tempFile");   std::ifstream in("tempFile", std::ifstream::in);   in >> workDir;   in.close(); }  bool ftpServer::start() {   int on = 1;    //初始化服务器   memset(&serverChannel, 0, sizeof(serverChannel));   serverChannel.sin_family = AF_INET;   serverChannel.sin_addr.s_addr = htonl(INADDR_ANY);   serverChannel.sin_port = htons(SERVER_PORT);    //创建套接字   this->serverSocket = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);   if (serverSocket < 0) {     printf("cannot create socket/n");     return false;   }   else printf("successfully create socket/n");   setsockopt(serverSocket, SOL_SOCKET, SO_REUSEADDR,     (char*)&on, sizeof(on));    //绑定   int b = bind(serverSocket, (sockaddr*)&serverChannel,     sizeof(serverChannel));   if (b < 0) {     printf("bind error/n");     return false;   }   else printf("successfully bind/n");   //监听   int l = listen(serverSocket, QUEUE_SIZE);   if (l < 0) {     printf("listen failed/n");     return false;   }   else printf("successfully listen/n");   int len = sizeof(serverChannel);   //服务器等待连接   while (1) {     printf("waiting for connection.../n");     //接受一个连接     clientSocket = accept(serverSocket, (sockaddr*)&serverChannel,       &len);     if (clientSocket < 0) {       printf("accept failed/n");     }     else {       printf("successfully connect/n");       while (1) {         memset(buffer, 0, sizeof(buffer));         int ret;          ret = recv(clientSocket, buffer, BUFFER_SIZE, 0);          if (ret == SOCKET_ERROR) {           printf("receive failed/n");         }         else {           char opt[50];           printf("successfully receive/n");           int i = 0, j = 0;           printf("buffer = %s/n", buffer);           while (buffer[i] != ' '&&buffer[i] != '/0') {             opt[i] = buffer[i];             i++;           }           opt[i] = '/0';           if (buffer[i] != '/0') {             i++;           }           while (buffer[i] != '/0') {             name[j] = buffer[i];             i++, j++;           }           name[j] = '/0';            if (strcmp(opt, "get") == 0) {             char ret[4];             if (!sendFile()) {               strcpy(ret, "NAK");               send(clientSocket, ret, sizeof(ret), 0);             }             else {               strcpy(ret, "ACK");               send(clientSocket, ret, sizeof(ret), 0);             }           }           else if (strcmp(opt, "put") == 0) {             receiveFile();           }           else if (strcmp(opt, "pwd") == 0) {             doPwd();           }           else if (strcmp(opt, "cd") == 0) {             doCd();           }           else if (strcmp(opt, "close") == 0) {             break;           }           else {             printf("syntex error/n");           }         }       }     }   }   return true; }  bool ftpServer::sendFile() {   std::ifstream in;   char path[100];   strcpy(path, workDir);   strcat(path, "//");   strcat(path, name);    in.open(path, std::ios::binary);   if (!in) {     printf("cannot open the file/n");     return false;   }   memset(buffer, 0, sizeof(buffer));   in.seekg(0, std::ios_base::end);   int sp = in.tellg();   int total_size = 0;   int r;   char length[20];   sprintf(length, "%d", sp);      r = send(clientSocket, length, sizeof(length), 0);    if (r == SOCKET_ERROR) {     printf("send failed/n");     return false;   }   else {     printf("send success/n");   }    while (sp > 0) {     in.clear();     in.seekg(total_size, std::ios_base::beg);     memset(buffer, 0, sizeof(buffer));     in.read(buffer, sizeof(buffer));     int size = sp < BUFFER_SIZE ? sp : BUFFER_SIZE;     total_size += size;     r = send(clientSocket, buffer, size, 0);      sp -= size;     if (r == SOCKET_ERROR) {       printf("send failed/n");       return false;     }     else {       printf("send success/n");     }   }   in.close();   return true; }  bool ftpServer::receiveFile() {   char path[100];   strcpy(path, workDir);   strcat(path, "//");   strcat(path, name);   memset(buffer, 0, sizeof(buffer));   int ret;   char length[20];   ret = recv(clientSocket, length, sizeof(length), 0);   if (ret == SOCKET_ERROR) {     printf("receive failed/n");     return false;   }   else {     printf("successfully receive/n");   }   int size = atoi(length);   std::ofstream out;    out.open(path, std::ios::binary);   if (!out) {     printf("cannot save the file/n");     return false;   }   while (size>0) {     int s = size < BUFFER_SIZE ? size : BUFFER_SIZE;     ret = recv(clientSocket, buffer, BUFFER_SIZE, 0);     if (ret == SOCKET_ERROR) {       printf("receive failed/n");       break;     }     else {       printf("successfully receive/n");       out.write(buffer, s);     }     size -= BUFFER_SIZE;   }   out.close();   return true; }  bool ftpServer::doPwd() {   char temCMD[150];   memset(temCMD, 0, sizeof(temCMD));   strcat(temCMD, "echo ");   strcat(temCMD, workDir);   strcat(temCMD, " > tempFile");   system(temCMD);   memset(temCMD, 0, sizeof(temCMD));   strcat(temCMD, "dir /b ");   strcat(temCMD, workDir);   strcat(temCMD, " >> tempFile");   system(temCMD);    std::ifstream in("tempFile", std::fstream::in);   if (!in) {     printf("cannot open the file/n");     return false;   }   memset(buffer, 0, sizeof(buffer));   in.seekg(0, std::ios_base::end);   int sp = in.tellg();   int total_size = 0;   int r;   char length[20];   sprintf(length, "%d", sp);   r = send(clientSocket, length, sizeof(length), 0);    if (r == SOCKET_ERROR) {     printf("send failed/n");     return false;   }   else {     printf("send success/n");   }   while (sp > 0) {     in.clear();     in.seekg(total_size, std::ios_base::beg);     memset(buffer, 0, sizeof(buffer));     in.read(buffer, sizeof(buffer));     int size = sp < BUFFER_SIZE ? sp : BUFFER_SIZE;     total_size += size;     printf("transfer size = %d/n", total_size);     r = send(clientSocket, buffer, size, 0);      sp -= size;     if (r == SOCKET_ERROR) {       printf("send failed/n");       return false;     }     else {       printf("send success/n");     }   }   in.close();   return true; }  bool ftpServer::isValidPath(char* path) {   char temCMD[100];   memset(temCMD, 0, sizeof(temCMD));   strcat(temCMD, "cd ");   strcat(temCMD, path);   int res = system(temCMD);   return res == 0; }  bool ftpServer::doCd() {   for (int i = 0; name[i] != '/0'; ++i) {     if (name[i] == '/')       name[i] = '//';   }   if (name[0] == '.'&&name[1] == '.') {     char temDir[100];     strcpy(temDir, workDir);     for (int i = sizeof(temDir); i >= 0; --i) {       if (temDir[i] == '//') {         temDir[i] = '/0';         break;       }     }     strcat(temDir, name + 2);     if (isValidPath(temDir)) {       strcpy(workDir, temDir);     }     else {       return false;     }   }   else if (name[0] == '.'&&name[1] != '.') {     char temDir[100];     strcpy(temDir, workDir);     strcat(temDir, name + 1);     if (isValidPath(temDir)) {       strcpy(workDir, temDir);     }     else {       return false;     }   }   else if (name[1] == ':') {     if (isValidPath(name)) {       strcpy(workDir, name);     }     else {       return false;     }   }   else {     char temDir[100];     strcpy(temDir, workDir);     strcat(temDir, "//");     strcat(temDir, name);     if (isValidPath(temDir)) {       strcpy(workDir, temDir);     }     else {       return false;     }   }   return true; } 

#include"ftpServer.h" #pragma(lib,"ws2_32.lib") int main() {  ftpServer f;  f.start(); }