使用c++實現一個FTP客戶端(二)

　　接上篇http://www.cnblogs.com/jzincnblogs/p/5213978.html，這篇主要記錄編程方面的重點。

　　客戶端使用了Windows Socket提供的API，支持上傳、下載、刪除、查看文件，斷點續傳，二進制/ASCII模式切換，被動模式切換，記錄操作日志等功能。

　　代碼包含的類如下：

　　　　①MySocket類，對SOCKET進行了簡單的封裝

//對winsock SOCKET的封裝
class MySocket
{
public:
    MySocket();
    //~MySocket();
    //重載向SOCKET類型轉換的運算符
    operator SOCKET() const; 
    //設置地址信息
    void SetAddrInfo(std::string host, int port);
    bool Connect();
    //bool Disconnect();
    bool Create(int af = AF_INET, int type = SOCK_STREAM, int protocol = IPPROTO_TCP);
    bool Close();
    //獲取主機ip
    std::string GetHostIP() const; 
    //獲取主機端口
    int GetPort() const; 
private:
    SOCKET sock;
    SOCKADDR_IN addr_in; //記錄連接的服務器的地址信息
    bool conn_flag; //判斷是否已連接
};

　　　　②Record類，存儲了客戶端與服務器的交互信息的數據結構

//枚舉類型,CMD代表命令信息,RES代表響應信息
enum log_type { CMD = 1, RES = 2 };

//與服務器的交互信息
class Record
{
    friend std::ostream & operator<<(std::ostream &os, const Record &rcd);
public:
    Record(log_type t, std::string m);
    Record(const Record &rcd);
    Record & operator=(const Record &rcd);
    //獲取信息內容
    std::string GetMsg() const;
private:
    log_type type; //信息類型
    std::string msg;
};

　　　　③Logger類，負責控制傳輸端口的發送命令，接收服務器響應，記錄、顯示操作日志等功能，包含一個Record類的vector，用於存儲此次程序運行的信息

class Logger
{
public:
    Logger(const std::string &host, int port);
    ~Logger();
    Logger(const Logger &logger) = delete;
    Logger & operator=(const Logger &logger) = delete;
    //發送命令
    void SendCmd(const std::string &cmd);
    //接收來自服務器的響應
    void RecvResponse();
    //記錄信息
    void Log(log_type type, const std::string &cmd);
    //獲取最后一條交互信息,用於驗證命令是否執行成功
    std::string GetLastLog() const;
    void DisplayLog() const;
private:
    MySocket sock_cmd; //發送接收命令的socket
    std::vector<Record> vec_rcd; //保存此次客戶端運行的交互信息
    //將信息記錄到文本文件中
    void WriteRecord();
};

　　　　④File類，用於存儲文件信息的數據結構

class File
{
    friend std::ostream & operator<<(std::ostream &os, const File &file);
public:
    //斜杠代表根目錄
    File(const std::string &n = "", const std::string &t = "", const int &s = 0, const std::string &p = "/");
    int GetSize() const;
private:
    std::string name;
    std::string path;
    std::string create_time;
    int size;
};

　　　　⑤FTPClient類，代碼的核心類

class FTPClient
{
public:
    FTPClient(const string &host, int port);
    bool Login(const string &usr, const string &pwd);
    //進入被動模式
    bool EnterPasvMode();
    //更新文件列表
    void UpdateFileList();
    //獲取指定文件信息
    File GetFileInfo(const string &f);
    void DisplayLog() const;
    //以二進制格式下載文件
    bool DownloadBinary(const string &f);
    //以ASCII格式下載文件
    bool DownloadASCII(const string &f);
    //上傳文件
    bool Upload(const string &f, bool binary);
    //刪除指定文件
    bool Delete(const string &f);
    //退出客戶端
    bool Quit();
private:
    Logger logger;
    MySocket sock_data; //用於傳輸數據的socket
    string host;
    int port;
    //
    void GetFileList();
    bool EnterASCIIMode(); //進入ASCII模式
    bool EnterBinaryMode(); //進入二進制模式
};