Linux网络编程综合运用之MiniFtp实现(八)

上节中实现了"USER"和"PASS"命令，以下：

事实上FTP是有不少命令组成的，若是就采用上面的这种方法来实现的话，就会有不少if...else if语句，代码显得很臃肿，因此有必要想办法来避免这种写法，因此一个新的方式既将诞生------命令映射，实际上在以前读取配置文件变量时就已经接触到了，下面则看具体的作法：

下面这个表是FTP命令的声明：

函数	说明
static void do_user(session_t *sess);
static void do_pass(session_t *sess);
static void do_cwd(session_t *sess);
static void do_cdup(session_t *sess);
static void do_quit(session_t *sess);
static void do_port(session_t *sess);
static void do_pasv(session_t *sess);
static void do_type(session_t *sess);
static void do_stru(session_t *sess);
static void do_mode(session_t *sess);
static void do_retr(session_t *sess);
static void do_stor(session_t *sess);
static void do_appe(session_t *sess);
static void do_list(session_t *sess);
static void do_nlst(session_t *sess);
static void do_rest(session_t *sess);
static void do_abor(session_t *sess);
static void do_pwd(session_t *sess);
static void do_mkd(session_t *sess);
static void do_rmd(session_t *sess);
static void do_dele(session_t *sess);
static void do_rnfr(session_t *sess);
static void do_rnto(session_t *sess);
static void do_site(session_t *sess);
static void do_syst(session_t *sess);
static void do_feat(session_t *sess);
static void do_size(session_t *sess);
static void do_stat(session_t *sess);
static void do_noop(session_t *sess);
static void do_help(session_t *sess);

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

因此在程序中先声明：

记得当时实现配置模块时，一个配置项与配置项变量相对应，而这边则应该是一个命令字符串与一个命令处理函数相对应，于是也能定义一个结构体来配置这种对应关系，以下：

FTP命令与命令处理函数对应表

typedef struct ftpcmd {        const char *cmd;//命令字符串        void (*cmd_handler)(session_t *sess);//函数指针 } ftpcmd_t;   static ftpcmd_t ctrl_cmds[] = {        /* 访问控制命令 */        {"USER",      do_user   },//若是是USER命令，则执行do_user方法        {"PASS",       do_pass   },        {"CWD",              do_cwd   },        {"XCWD",     do_cwd   },        {"CDUP",      do_cdup  },        {"XCUP",      do_cdup  },        {"QUIT",       do_quit   },        {"ACCT",      NULL     },        {"SMNT",      NULL     },        {"REIN",       NULL     },//这种命令是没有执行函数的        /* 传输参数命令 */        {"PORT",       do_port   },        {"PASV",       do_pasv   },        {"TYPE",       do_type   },        {"STRU",       do_stru    },        {"MODE",     do_mode },          /* 服务命令 */        {"RETR",       do_retr    },        {"STOR",       do_stor    },        {"APPE",       do_appe  },        {"LIST", do_list     },        {"NLST",       do_nlst    },        {"REST",       do_rest    },        {"ABOR",      do_abor   },        {"\377\364\377\362ABOR", do_abor},        {"PWD",        do_pwd   },        {"XPWD",     do_pwd   },        {"MKD",              do_mkd   },        {"XMKD",     do_mkd   },        {"RMD",        do_rmd   },        {"XRMD",     do_rmd   },        {"DELE",      do_dele   },        {"RNFR",      do_rnfr   },        {"RNTO",      do_rnto   },        {"SITE", do_site    },        {"SYST",       do_syst    },        {"FEAT",       do_feat },        {"SIZE", do_size    },        {"STAT", do_stat    },        {"NOOP",      do_noop  },        {"HELP",       do_help   },        {"STOU",      NULL     },        {"ALLO",      NULL     } };

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

另外发现，以前配置模块的最后是NULL结尾的，而这里并无，因此此次是用另一种遍历方法来判断是否遍历完，下面先将上面的映射关系代码拷至代码中：

ftpproto.c:

#include "ftpproto.h"
#include "sysutil.h"
#include "str.h"
#include "ftpcodes.h"

static void ftp_reply(session_t *sess, int status, const char *text);
static void do_user(session_t *sess);
static void do_pass(session_t *sess);
static void do_cwd(session_t *sess);
static void do_cdup(session_t *sess);
static void do_quit(session_t *sess);
static void do_port(session_t *sess);
static void do_pasv(session_t *sess);
static void do_type(session_t *sess);
static void do_stru(session_t *sess);
static void do_mode(session_t *sess);
static void do_retr(session_t *sess);
static void do_stor(session_t *sess);
static void do_appe(session_t *sess);
static void do_list(session_t *sess);
static void do_nlst(session_t *sess);
static void do_rest(session_t *sess);
static void do_abor(session_t *sess);
static void do_pwd(session_t *sess);
static void do_mkd(session_t *sess);
static void do_rmd(session_t *sess);
static void do_dele(session_t *sess);
static void do_rnfr(session_t *sess);
static void do_rnto(session_t *sess);
static void do_site(session_t *sess);
static void do_syst(session_t *sess);
static void do_feat(session_t *sess);
static void do_size(session_t *sess);
static void do_stat(session_t *sess);
static void do_noop(session_t *sess);
static void do_help(session_t *sess);

typedef struct ftpcmd { const char *cmd; void (*cmd_handler)(session_t *sess); } ftpcmd_t; static ftpcmd_t ctrl_cmds[] = { /* 访问控制命令 */ {"USER", do_user }, {"PASS", do_pass }, {"CWD", do_cwd }, {"XCWD", do_cwd }, {"CDUP", do_cdup }, {"XCUP", do_cdup }, {"QUIT", do_quit }, {"ACCT", NULL }, {"SMNT", NULL }, {"REIN", NULL }, /* 传输参数命令 */ {"PORT", do_port }, {"PASV", do_pasv }, {"TYPE", do_type }, {"STRU", do_stru }, {"MODE", do_mode }, /* 服务命令 */ {"RETR", do_retr }, {"STOR", do_stor }, {"APPE", do_appe }, {"LIST", do_list }, {"NLST", do_nlst }, {"REST", do_rest }, {"ABOR", do_abor }, {"\377\364\377\362ABOR", do_abor}, {"PWD", do_pwd }, {"XPWD", do_pwd }, {"MKD", do_mkd }, {"XMKD", do_mkd }, {"RMD", do_rmd }, {"XRMD", do_rmd }, {"DELE", do_dele }, {"RNFR", do_rnfr }, {"RNTO", do_rnto }, {"SITE", do_site }, {"SYST", do_syst }, {"FEAT", do_feat }, {"SIZE", do_size }, {"STAT", do_stat }, {"NOOP", do_noop }, {"HELP", do_help }, {"STOU", NULL }, {"ALLO", NULL } }; void handle_child(session_t *sess)
{
    writen(sess->ctrl_fd, "220 (miniftpd 0.1)\r\n", strlen("220 (miniftpd 0.1)\r\n"));
    int ret;
    while (1)
    {
        memset(sess->cmdline, 0, sizeof(sess->cmdline));
        memset(sess->cmd, 0, sizeof(sess->cmd));
        memset(sess->arg, 0, sizeof(sess->arg));
        ret = readline(sess->ctrl_fd, sess->cmdline, MAX_COMMAND_LINE);
        if (ret == -1)
            ERR_EXIT("readline");
        else if (ret == 0)
            exit(EXIT_SUCCESS);

        printf("cmdline=[%s]\n", sess->cmdline);
        // 去除\r\n
        str_trim_crlf(sess->cmdline);
        printf("cmdline=[%s]\n", sess->cmdline);
        // 解析FTP命令与参数
        str_split(sess->cmdline, sess->cmd, sess->arg, ' ');
        printf("cmd=[%s] arg=[%s]\n", sess->cmd, sess->arg);
        // 将命令转换为大写
        str_upper(sess->cmd);
        // 处理FTP命令
        if (strcmp("USER", sess->cmd) == 0)
        {
            do_user(sess);
        }
        else if (strcmp("PASS", sess->cmd) == 0)
        {
            do_pass(sess);
        }
    }
}

static void do_user(session_t *sess)
{
    //USER jjl
    struct passwd *pw = getpwnam(sess->arg);
    if (pw == NULL)
    {
        // 用户不存在
        ftp_reply(sess, FTP_LOGINERR, "Login incorrect.");
        return;
    }
    sess->uid = pw->pw_uid;
    ftp_reply(sess, FTP_GIVEPWORD, "Please specify the password.");
}

static void do_pass(session_t *sess)
{
    // PASS 123456
    struct passwd *pw = getpwuid(sess->uid);
    if (pw == NULL)
    {
        // 用户不存在
        ftp_reply(sess, FTP_LOGINERR, "Login incorrect.");
        return;
    }

    printf("name=[%s]\n", pw->pw_name);
    struct spwd *sp = getspnam(pw->pw_name);
    if (sp == NULL)
    { // 没有找到影子文件，则表明登陆也是失败的
        ftp_reply(sess, FTP_LOGINERR, "Login incorrect.");
        return;
    }
    
    // 将明文进行加密
    char *encrypted_pass = crypt(sess->arg, sp->sp_pwdp);
    // 验证密码
    if (strcmp(encrypted_pass, sp->sp_pwdp) != 0)
    {
        ftp_reply(sess, FTP_LOGINERR, "Login incorrect.");
        return;
    }

    setegid(pw->pw_gid);
    seteuid(pw->pw_uid);
    chdir(pw->pw_dir);

    ftp_reply(sess, FTP_LOGINOK, "Login successful.");
}

static void ftp_reply(session_t *sess, int status, const char *text)
{
    char buf[1024] = {0};
    sprintf(buf, "%d %s\r\n", status, text);
    writen(sess->ctrl_fd, buf, strlen(buf));
}

而像以前的配置映射的方式代码写法以下：

可是因为此次并无以NULL结尾，因此说得换一种新的方式，以下：

编译一下：

因此须要实现这些函数：

#include "ftpproto.h"
#include "sysutil.h"
#include "str.h"
#include "ftpcodes.h"

static void ftp_reply(session_t *sess, int status, const char *text);
static void do_user(session_t *sess);
static void do_pass(session_t *sess);
static void do_cwd(session_t *sess);
static void do_cdup(session_t *sess);
static void do_quit(session_t *sess);
static void do_port(session_t *sess);
static void do_pasv(session_t *sess);
static void do_type(session_t *sess);
static void do_stru(session_t *sess);
static void do_mode(session_t *sess);
static void do_retr(session_t *sess);
static void do_stor(session_t *sess);
static void do_appe(session_t *sess);
static void do_list(session_t *sess);
static void do_nlst(session_t *sess);
static void do_rest(session_t *sess);
static void do_abor(session_t *sess);
static void do_pwd(session_t *sess);
static void do_mkd(session_t *sess);
static void do_rmd(session_t *sess);
static void do_dele(session_t *sess);
static void do_rnfr(session_t *sess);
static void do_rnto(session_t *sess);
static void do_site(session_t *sess);
static void do_syst(session_t *sess);
static void do_feat(session_t *sess);
static void do_size(session_t *sess);
static void do_stat(session_t *sess);
static void do_noop(session_t *sess);
static void do_help(session_t *sess);

typedef struct ftpcmd
{
    const char *cmd;
    void (*cmd_handler)(session_t *sess);
} ftpcmd_t;


static ftpcmd_t ctrl_cmds[] = {
    /* 访问控制命令 */
    {"USER",    do_user    },
    {"PASS",    do_pass    },
    {"CWD",        do_cwd    },
    {"XCWD",    do_cwd    },
    {"CDUP",    do_cdup    },
    {"XCUP",    do_cdup    },
    {"QUIT",    do_quit    },
    {"ACCT",    NULL    },
    {"SMNT",    NULL    },
    {"REIN",    NULL    },
    /* 传输参数命令 */
    {"PORT",    do_port    },
    {"PASV",    do_pasv    },
    {"TYPE",    do_type    },
    {"STRU",    do_stru    },
    {"MODE",    do_mode    },

    /* 服务命令 */
    {"RETR",    do_retr    },
    {"STOR",    do_stor    },
    {"APPE",    do_appe    },
    {"LIST",    do_list    },
    {"NLST",    do_nlst    },
    {"REST",    do_rest    },
    {"ABOR",    do_abor    },
    {"\377\364\377\362ABOR", do_abor},
    {"PWD",        do_pwd    },
    {"XPWD",    do_pwd    },
    {"MKD",        do_mkd    },
    {"XMKD",    do_mkd    },
    {"RMD",        do_rmd    },
    {"XRMD",    do_rmd    },
    {"DELE",    do_dele    },
    {"RNFR",    do_rnfr    },
    {"RNTO",    do_rnto    },
    {"SITE",    do_site    },
    {"SYST",    do_syst    },
    {"FEAT",    do_feat },
    {"SIZE",    do_size    },
    {"STAT",    do_stat    },
    {"NOOP",    do_noop    },
    {"HELP",    do_help    },
    {"STOU",    NULL    },
    {"ALLO",    NULL    }
};

void handle_child(session_t *sess)
{
    writen(sess->ctrl_fd, "220 (miniftpd 0.1)\r\n", strlen("220 (miniftpd 0.1)\r\n"));
    int ret;
    while (1)
    {
        memset(sess->cmdline, 0, sizeof(sess->cmdline));
        memset(sess->cmd, 0, sizeof(sess->cmd));
        memset(sess->arg, 0, sizeof(sess->arg));
        ret = readline(sess->ctrl_fd, sess->cmdline, MAX_COMMAND_LINE);
        if (ret == -1)
            ERR_EXIT("readline");
        else if (ret == 0)
            exit(EXIT_SUCCESS);

        printf("cmdline=[%s]\n", sess->cmdline);
        // 去除\r\n
        str_trim_crlf(sess->cmdline);
        printf("cmdline=[%s]\n", sess->cmdline);
        // 解析FTP命令与参数
        str_split(sess->cmdline, sess->cmd, sess->arg, ' ');
        printf("cmd=[%s] arg=[%s]\n", sess->cmd, sess->arg);
        // 将命令转换为大写
        str_upper(sess->cmd);
        // 处理FTP命令
        /*
        if (strcmp("USER", sess->cmd) == 0)
        {
            do_user(sess);
        }
        else if (strcmp("PASS", sess->cmd) == 0)
        {
            do_pass(sess);
        }*/
        int i;
        int size = sizeof(ctrl_cmds) / sizeof(ctrl_cmds[0]);
        for (i=0; i<size; i++)
        {
            if (strcmp(ctrl_cmds[i].cmd, sess->cmd) == 0)
            {
                if (ctrl_cmds[i].cmd_handler != NULL)
                {
                    ctrl_cmds[i].cmd_handler(sess);
                }
                else
                {
                    ftp_reply(sess, FTP_COMMANDNOTIMPL, "Unimplement command.");
                }
                
                break;
            }
        }

        if (i == size)
        {
            ftp_reply(sess, FTP_BADCMD, "Unknown command.");
        }
    }
}

static void ftp_reply(session_t *sess, int status, const char *text)
{
    char buf[1024] = {0};
    sprintf(buf, "%d %s\r\n", status, text);
    writen(sess->ctrl_fd, buf, strlen(buf));
}

static void do_user(session_t *sess)
{
    //USER jjl
    struct passwd *pw = getpwnam(sess->arg);
    if (pw == NULL)
    {
        // 用户不存在
        ftp_reply(sess, FTP_LOGINERR, "Login incorrect.");
        return;
    }
    sess->uid = pw->pw_uid;
    ftp_reply(sess, FTP_GIVEPWORD, "Please specify the password.");
}

static void do_pass(session_t *sess)
{
    // PASS 123456
    struct passwd *pw = getpwuid(sess->uid);
    if (pw == NULL)
    {
        // 用户不存在
        ftp_reply(sess, FTP_LOGINERR, "Login incorrect.");
        return;
    }

    printf("name=[%s]\n", pw->pw_name);
    struct spwd *sp = getspnam(pw->pw_name);
    if (sp == NULL)
    { // 没有找到影子文件，则表明登陆也是失败的
        ftp_reply(sess, FTP_LOGINERR, "Login incorrect.");
        return;
    }
    
    // 将明文进行加密
    char *encrypted_pass = crypt(sess->arg, sp->sp_pwdp);
    // 验证密码
    if (strcmp(encrypted_pass, sp->sp_pwdp) != 0)
    {
        ftp_reply(sess, FTP_LOGINERR, "Login incorrect.");
        return;
    }

    setegid(pw->pw_gid);
    seteuid(pw->pw_uid);
    chdir(pw->pw_dir);

    ftp_reply(sess, FTP_LOGINOK, "Login successful.");
}

static void do_cwd(session_t *sess)
{
}

static void do_cdup(session_t *sess)
{
}

static void do_quit(session_t *sess)
{
}

static void do_port(session_t *sess)
{
}

static void do_pasv(session_t *sess)
{
}

static void do_type(session_t *sess)
{
}

static void do_stru(session_t *sess)
{
}

static void do_mode(session_t *sess)
{
}

static void do_retr(session_t *sess)
{
}

static void do_stor(session_t *sess)
{
}

static void do_appe(session_t *sess)
{
}

static void do_list(session_t *sess)
{
}

static void do_nlst(session_t *sess)
{
}

static void do_rest(session_t *sess)
{
}

static void do_abor(session_t *sess)
{
}

static void do_pwd(session_t *sess)
{
}

static void do_mkd(session_t *sess)
{
}

static void do_rmd(session_t *sess)
{
}

static void do_dele(session_t *sess)
{
}

static void do_rnfr(session_t *sess)
{
}

static void do_rnto(session_t *sess)
{
}

static void do_site(session_t *sess)
{
}

static void do_syst(session_t *sess)
{
}

static void do_feat(session_t *sess)
{
}

static void do_size(session_t *sess)
{
}

static void do_stat(session_t *sess)
{
}

static void do_noop(session_t *sess)
{
}

static void do_help(session_t *sess)
{
}

再次编译运行:

可见，命令解析一切OK，接下来就一个个命令来实现，首先是"SYST"命令，对照着vsftpd来作：

因此对应的函数中给出以下输出：

编译运行：

其中"FEAT"表示Feature，表示服务端的特性，实际上这条命令不实现也没有关系，若是暂且不想实现它，则能够给它对应的处理函数配置成NULL，以下：

编译运行：

紧着着发送了CLNT命令，这个命令没有，因此就提示无效的命令：

【注意】：当无效的命令时，咱们也必须给它响应，不然客户端就会阻塞。

当响应了CLNT命令以后，最后发送了REST 100命令，这表示断点续传，对比着vsftpd输出结果来看：

因此，咱们也来先实现FEAT命令，实现完以后，看是否最后还会发送RESET 100命令，将FEAT的命令注释还原：

而后处理它对应的函数：

而后接vsftpd的响应格式来输出：

再次编译运行：

接着来实现PWD命令，先来看下vsftpd这个命令是如何响应的：

下面来实现PWD命令：

【说明】:

编译运行：

先看一下vsftpd中的"TYPE A"的响应：

若是是输入其它的TYPE命令呢？能够按以下步骤输出：

而若是输入"TYPE I"呢？

因此则照着这几种状况实现TYPE命令：

另外这个状态须要记录在session当中，以后会用到，因此须要增长一个变量至session结构体中：

修改它的初始化：

在do_type函数中来进行记录：

【说明】:Ascii和二进制协议进行传输的区别就在因而否要处理"\r\n"，这个在以前FTP协议时有说明过。

编译运行：

接下来就要进行列表的传输了，在传输时须要建立数据链接通道，在建立通道以前是须要协商使用PORT模式仍是PASV模式，因此这里先发送PASV模式出来了，这个下次来实现，先学到这~