首页 > 代码库 > Web服务器文件传输程序客户端程序实现

Web服务器文件传输程序客户端程序实现

 

1、 客户端程序--主函数

客户端主程序的流程图如下:

技术分享 

主程序主要是分析输入的命令,根据不同命令调用不同的函数处理或者进行出错处理,函数代码如下:

#include "common.h"

int main(void)

{

char cline[COMMAND_LINE];             /* 缓冲区,存储用户输入的命令 */

struct command_line command; /* 命令结构,存储分解后的命令 */

int sock_fd;

struct sockaddr_in sin;                /* 服务器端的地址结构 */

printf("myftp$: ");                       /* 打印提示符 */

fflush(stdout);                             /* 保证提示符显示 */

while(fgets(cline, MAX_LINE, stdin) != NULL){ /* 得到一行命令 */

        if(split(command, cline) == -1) /* 将命令拆分为命令名和命令参数的形式*/

               exit(1);

        /* 根据命令名作命令处理,比较命令名和每一个合法命令,匹配则处理 */

        if(strcasecmp(command.name, "get") == 0){ /* get命令 */

               if(do_get(command.argv[1], command.argv[2], sock_fd) == -1)

                      exit(1); /* 命令处理出错则退出 */

        }else if(strcasecmp(command.name, "put") == 0){ /* put命令 */

               if(do_put(command.argv[1], command.argv[2], sock_fd) == -1)

                      exit(1);

        }else if(strcasecmp(command.name, "cd") == 0){ /* cd命令 */

               if(do_cd(command.argv[1]) == -1)

                      exit(1);

        }else if(strcasecmp(command.name, "!cd") == 0){ /* !cd命令 */

               if(do_serv_cd(command.argv[1], sock_fd) == -1)

                      exit(1);

        }else if(strcasecmp(command.name, "ls") == 0){ /* ls命令 */

               if(do_ls(command.argv[1]) == -1)

                      exit(1);

        }else if(strcasecmp(command.name, "!ls") == 0){ /* !ls命令 */

               if(do_serv_ls(command.argv[1], sock_fd) == -1)

                      exit(1);

        }else if(strcasecmp(command.name, "connect") == 0){ /* connect               命令 */

               if(do_connect(command.argv[1], &sin, &sock_fd) == -1)

                      exit(1);

        }else if(strcasecmp(command.name, "bye") == 0){ /* bye命令 */

               if(do_bye(sock_fd) == -1)

                      exit(1);

               break; /* 向服务器端发送退出信息后,则跳出循环,退出程序 */

        }else{ /* 错误命令,打印程序的用法 */

               printf("wrong command\n");

               printf("usage : command arg1, arg2, ... argn\n");

        }

        printf("myftp$: "); /* 再次打印提示符,准备接受新的命令 */

        fflush(stdout);

}

if(close(sock_fd) == -1){ /* 执行bye命令后执行到这里,关闭通信用套接字 */

        perror("fail to close");

        exit(1);

}

return 0; /* 程序正常退出 */

}

2、客户端程序--命令拆分

命令拆分程序可以根据之前的程序设计来进行编写,先判断是什么命令,再根据命令类型的不同调用不同的函数,将命令中的参数分离出来作为。主程序中调用该程序,根据不同的命令调用不同的函数,并且将参数传递给处理函数,流程图如下:

技术分享 

程序代码如下:

#include "common.h"

/*下面的宏将cline串中从pos所表示的位置开始,跳过连续的空格和制表符 */

#define del_blank(pos, cline); { 
while(cline[pos] != \0 && (cline[pos] ==   || cline[pos] == \t))
{ 
        pos++; 
} 
}

/* 下面的宏得到一个命令参数。

* 将cline串中从pos所表示的位置的内容复制到arg缓冲区中,直到遇到空格、制表符或者结束符为止

*/

#define get_arg(arg, pos, cline); { 
int i = 0; 
while(cline[pos] != \0 && cline[pos] !=  && cline[pos] != /t){ 
        arg[i++] = com[pos++]; 
} 
}

/* 将一个命令字符串分割为命令参数并存储在command_struct结构中

 * 成功则返回拆分后的命令参数的个数,失败返回-1

* command : 存储命令和命令参数的结构体

* cline : 命令字符串

*/

int split(struct command_line * command, char cline[ ])

{

int i;

int pos = 0;

clien[strlen(cline) - 1] = \0; /* 得到命令字符串的长度,将最后一个‘\n’                                                               替换为‘\0’ */

del_blank(pos, cline);                    /* 过滤空格,直到遇到第一个参数 */

i = 0;

while(cline[pos] != \0){ /* 处理整个命令字符串 */

        /* 为存储命令参数的数组分配空间 */

        if((command->argv[i] = (char *)malloc(MAX_LENGTH)) == NULL){

               perror("fail to malloc");

               return -1;

        }

        /* 得到一个参数,将两个空格之间的内容复制到存储参数的数组 */

        get_arg(command->argv[i], pos, cline);

        i++;                                       /* 下一个参数 */

        del_blank(pos, cline);     /* 过滤掉空格 */

}

command->argv[i] = NULL;          /* 命令参数数组以NULL结尾 */

comand->name = command->.argv[0]; /* 命令名和第一个命令参数实际上指向同一块内存区域 */

return i;

}

3、客户端程序—命令模块处理

命令处理模块是这个程序的核心模块,客户端的命令处理程序有7个。

(1)do_connect()函数

do_connect()函数完成了创建套接字并且连接的工作,这个函数返回说明客户端已经和服务器成功连接。do_connect()函数处理connect命令,成功返回0,失败返回-1。connect命令的形式为:connect arg1,参数 arg1是服务器的IP地址。

参数说明:

ip : 字符指针,指向服务器的地址,是一个十进制点分字符串;

sin : 地质结构指针,指向服务器地址结构,在connect函数中填充;

sock_fd : 整型指针,指向通信用套接字描述符,该套接字在connct函数中设置,将描述符传回。

int do_connect(char *ip,struct sockaddr_in *sin, int *sock_fd)

{

int sfd; /* 临时的套接字描述符 */

bzero(&sin, sizeof(struct sockaddr_in)); /* 将地址结构清空 */

sin.sin_family = AF_INET;                             /* 使用IPv4地址族 */

/* 将点分十进制形式的IP地址转换成为二进制形式的地址,并且存储在地址结构中 */

if(inet_pton(AF_INET, ip, &sin.sin_addr) == -1){

        perror("wrong format of ip address"); /* 如果地址格式为非法则出错 */

        return -1;

}

sin.sin_port = htons(PORT); /* 将端口号转换成为网络字节序存储在地址结构中 */

if(sfd = socket(AF_INET, SOCK_STREAM, 0)) == -1){ /* 创建套接字 */

        perror("fail to creat socket");

        return -1;      

}

/* 使用该套接字,和填充好的地址结构进行连接 */

if(connect(sfd, (struct sockaddr *)sin, sizeof(struct sockaddr_in)) ==     -1){

        perror("fail to connect");

        return -1;

}

*sock_fd = sfd; /* 将sfd保存的套接字的描述符赋值给sock_fd所指向区域返回 */

return 0;

}

(2)do_get函数处理get命令,成功返回0,失败返回-1。get命令的形式为:get arg1 arg2。该命令从服务器端取得文件,arg1表示源文件,arg2表示目的路径,如果该文件已存在则覆盖。

参数说明:

src : 源文件,是一个绝对路径

dst : 目的目录,是一个绝对的路径

sock_fd : 通信用的套接字描述符

int do_get(const char *src, const char *dst, int sock_fd)

{

char *dst_file;        /* 目的路径 */

char *p;

struct stat statbuf;/* 文件状态缓冲区 */

int n, fd;

char buf[MAX_LINE];/* 命令缓冲区,存储发送给服务器端的命令 */

int len;

int res = -1;          /* 返回值默认的值为-1 */

if(src =http://www.mamicode.com/= NULL || dst == NULL){/* 检查源文件和目的地址是不是空串 */

        printf(wrong command\n);  /* 是空串则报错 */

        return -1;

}

/* 如果源文件路径的最后一个字符是‘/’,则说明源文件不是一个普通文件,而是一个目录   */

  if(src[strlen(src)-1]==/){

        printf("source file should be a regular file\n");

        return -1;

}

/* 为最终的目的文件路径分配内存空间

* 文件路径由目标目录和源文件的文件名组成

* 所以该缓冲区的大小最大是两个文件路径的长

*/

  if( (dst_file = (char *)malloc(strlen(dst) + strlen(src))) == NULL){

        perror("fail to malloc");

        return -1;                       /* 如果内存分配不成功则返回-1 */

  }

  strcpy(dst_file, dst);         /* 将目标目录复制到dst_file缓冲区中 */

  /* 如果目标目录的结尾不是‘/’,则添加‘/’,例如/home/admin—>/home/admin/ */

  if(dst_file[strlen(dst_file) - 1] != /)

        strcat(dst_file, "/");

  p = rindex(src, /);              /* 取源文件路径中最后一个‘/’,其后面是文件名 */

  strcat(dst_file, p + 1);    /* 跳过‘/’,得到源文件文件名 */

/* 如果目标文件不存在则创建该文件,使用权限字“0644”

* 该权限表示所有者拥有读写权限而组用户和其他用户只有读权限

* 如果该文件存在,则将文件截短为0打开,覆盖原文件

*/

  if((fd = open(dst_file, O_WRONLY | O_CREAT | O_TRUNC, 0644)) == -1){

        perror("fail to open dst-file");

        goto end2; /* 打开文件失败,退出时只需要释放dst_file的内存,而不需要关闭            文件 */

  }

  if(fstat(fd, &statbuf) == -1){ /* 取目标文件的文件状态 */

        perror("fail to stat dst-file");

        goto end1;

  }

/* 如果目标文件已经存在,且不是一个普通文件,则无法传输

* 因为这样会造成将已经存在的目录等其他特殊文件覆盖

*/

  if(!S_ISREG(statbuf.st_mode)){

        printf("dst-file should be a regular file");

        goto end1;

  }

    sprintf(buf, "GET %s", src); /* 准备向服务器端发送GET命令 */

  if(my_write(sock_fd, buf, strlen(buf)+1) == -1) /* 发出命令 */

        goto end1;                                                               /* 出错则退出 */

  if( (n = my_read(sock_fd, buf, MAX_LINE)) <= 0){ /* 接收服务器端发回的确                                                                                       认信息 */

        goto end1;

  }

  if(buf[0] == E){ /* 如果收到的信息是ERR,表示出错 */

        write(STDOUT_FILENO, buf, n); /* 向屏幕输出服务器发来的出错信息 */

        res = 0;

        goto end1;

  }

/* 如果对没有出错,服务器反馈的信息格式为“OK 请求文件的文件大小”

* 跳过字符串“OK”和一个空格,从第三个字符开始,取得文件的长度

*/

  len = atoi(&buf[3]);

/* 向服务器发出准备好的信息,服务器接收到此信息后开始传送文件的内容 */

  if(my_write(sock_fd, "RDY", 3) == -1)

        goto end1;

  while(1){ /* 循环读,直到读完所有的文件内容 */

        n = my_read(sock_fd, buf, MAX_LINE); /* 读服务器传送来的内容 */

        if(n > 0){ /* 读到的字节数大于0,说明正常 */

               write(fd, buf, n); /* 将读到的内容写到打开的目标文件中去 */

               len -= n; /* 文件总长度减少 */

        }else if(len == 0){ /* 读到的字节数等于0,说明通信已经结束 */

               /* 如果请求的文件已经读完,则正常结束此次命令的执行 */

               printf("OK\n");

               break;

        }else /* 读到的字节数小于0,出错 */

               goto end1;

}

  res = 0; /* 运行到此,则函数正常返回,返回值为0 */

end1:

  free(dst_file); /* 释放动态分配的内存空间 */

end2:

  close(fd); /* 关闭文件 */

return res;

}

 

Web服务器文件传输程序客户端程序实现