使用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(); //进入二进制模式
};