• Home
  • Categories
  • WIKI
  • LANGUAGE: en
    • 自己动手写ftp服务器2(更进一步)
    社区开发 1816 views · 0 replies ·
    Tofloor
    poster avatar
    zhqh100
    deepin
    2014-04-30 08:29
    Author
    前面基本把框架搭建好，后面的逻辑处理就比较简单了，一步一步对功能进行完善。
    为了简化代码，对于一些合法性校验都是尽量简化，只处理核心功能。

    封装是编程的最基本思想，不过我这里还是尽量减少封装，因为较多的封装的话在看代码时会不得不停顿下来找调用到的地方，增加看代码的难度。

    开发语言：C语言
    操作系统：LinuxDeepin 2013

    增加了后面的一点交互，代码如下：
    2. #include
    3. #include
    4. #include
    5. #include
    6. #include
    7. #include
    8. #include
    10. #define bool unsigned short
    12. typedef struct cmd_opts {
    13.         bool daemonize;
    14.         bool listen_any;
    15.         char *listen_addr;
    16.         int port;
    17.         int max_conn;
    18.         int userid;
    19.         char *chrootdir;
    20. }cmd_opts;
    22. //判断字符是否是以某指令开始
    23. int strBeginWith(char *str,char *substr){
    24.         while(*str && *substr && *str==*substr){
    25.                 str++;
    26.                 substr++;
    27.         }
    28.         if(*substr=='\0')
    29.                 return 1;
    30.         else
    31.                 return 0;
    32. }
    34. //字符串转int
    35. int toint(char *str) {
    36.         if (strlen(str) < 1)
    37.         {
    38.                 return -1;
    39.         }
    40.         int retVal = str[0] - '0';
    41.         int i = 1;
    42.         while(str[i] >= '0' && str[i] <= '9')
    43.         {
    44.                 retVal = retVal * 10 + str[i] - '0';
    45.                 i++;
    46.         }
    47.         return retVal;
    48. }
    50. //向客户端发送消息
    51. int send_repl(int send_sock,char *msg) {
    52.         printf("Send Message:%s\n", msg);
    53.         if (send(send_sock, msg, strlen(msg),0) < 0) {
    54.                 perror("send");
    55.                 close(send_sock);
    56.                 return -1;
    57.         }
    58.         return 0;
    59. }
    62. //创建数据传输的连接，用于处理"LIST","STOR","RETR"等请求
    63. int make_client_connection(int client_port,const char* client_addr) {
    64.         int sock=-1;
    65.         struct sockaddr_in servaddr;
    66.         servaddr.sin_family = AF_INET;
    67.         servaddr.sin_addr.s_addr = inet_addr(client_addr);
    68.         servaddr.sin_port = htons (client_port);
    69.         if ((sock = socket (PF_INET, SOCK_STREAM, IPPROTO_TCP)) < 0) {
    70.                 send_repl(sock,"425 Can't open data connection.\r\n");
    71.                 perror("socket");
    72.                 return -1;
    73.         }
    74.         int status = connect (sock, (struct sockaddr *)&servaddr, sizeof (servaddr));
    75.         if(status!=0) {
    76.                 send_repl(sock,"425 Can't open data connection.\r\n");
    77.                 perror("connect");
    78.                 return -1;
    79.         }
    80.         return sock;
    81. }
    84. enum {
    85.         TRUE=1,
    86.         FALSE=0
    87. };
    89. int main(int argc,char *argv[])
    90. {
    91.         struct cmd_opts *opts= malloc(sizeof(struct cmd_opts));
    92.         //设置一些初始配置
    93.         opts->daemonize    = FALSE;
    94.         opts->listen_any   = TRUE;
    95.         opts->port         = 21;
    96.         opts->userid       = 0;
    97.         opts->chrootdir    = "./";
    98.         opts->max_conn     = 5;
    99.         opts->listen_addr  = NULL;
    100.         struct sockaddr_in servaddr;
    101.         memset((char *)&servaddr, 0, sizeof(servaddr));
    102.         //TCP协议
    103.         servaddr.sin_family = PF_INET;
    104.         servaddr.sin_addr.s_addr =  htonl(INADDR_ANY);
    105.         servaddr.sin_port = htons (opts->port);
    107.         int servaddr_len = sizeof(servaddr);
    108.         int sock;
    109.         if ((sock = socket (PF_INET, SOCK_STREAM, IPPROTO_TCP)) < 0) {
    110.                 perror("socket");
    111.                 return -1;
    112.         }
    113.         int flag = 1;
    114.         setsockopt(sock, SOL_SOCKET,SO_REUSEADDR,(char *) &flag, sizeof(int));
    115.        
    116.         setsockopt(sock, IPPROTO_TCP,TCP_NODELAY,(char *) &flag, sizeof(int));
    118.         //套接字绑定端口和地址
    119.         if(bind (sock, (struct sockaddr *)&servaddr, sizeof(servaddr))<0) {
    120.                 perror("bind");
    121.                 return -1;
    122.         }
    124.         if(listen(sock,opts->max_conn) <0) {
    125.                 perror("listen");
    126.                 return -1;
    127.         }
    128.         int connection;
    130.         //死循环等待客户端连接
    131.         while(1){
    132.                 printf("Begin to accept\n");
    133.                 if ((connection = accept(sock, (struct sockaddr *) &servaddr, &servaddr_len)) < 0) {
    134.                         perror("accept");
    135.                         return -1;
    136.                 }
    138.                 int pid = fork();
    139.                 char msg[]= "220 Service ready for new user.\r\n";
    140.                 char send_buff[1000];
    141.                 char read_buff[1000];
    142.                 char current_dir[1000];
    143.                 //客户端IP地址和端口号
    144.                 char client_addr[16];
    145.                 int client_port;
    146.                 //临时指针
    147.                 char *temp;
    148.                 int i;
    149.                 int client_fd;
    150.                 strcpy(current_dir,opts->chrootdir);
    152.                 //获取当前绝对路径
    153.                 getcwd(current_dir,sizeof(current_dir));
    155.                 int len = strlen(msg);
    156.                 if(pid==0) {
    157.                         //客户端连接后，服务器先主动发送200 Service ready
    158.                         if (send(connection, msg, len,0) < 0) {
    159.                                 perror("send");
    160.                                 close(connection);
    161.                                 return -1;
    162.                         }
    163.                         while(1){
    164.                                 //初始化读取buffer和发送buffer
    165.                                 memset(send_buff,0,sizeof(send_buff));
    166.                                 memset(read_buff,0,sizeof(read_buff));
    167.                                 //等待接收客户端指令
    168.                                 int recvbuff = recv(connection,read_buff,sizeof(read_buff),0);
    169.                                 if(recvbuff<1) {
    170.                                         return -1;
    171.                                 }
    172.                                 if(recvbuff==sizeof(read_buff)) {
    173.                                         return -2;
    174.                                 }
    175.                                 printf("Received:%s",read_buff);
    176.                                 if (strBeginWith(read_buff,"USER")){
    177.                                         //收到客户端发送的用户名，这里不处理用户名，均答复登陆成功
    178.                                         //注释代码也可用，只是这里修改后，发送的消息更精简了
    179.                                         //后面的人性化提示可以省略，但\r\n不可省略，经Filezilla和 CuteFTP客户端验证，必须把\r\n带上才可正常识别
    180.                                         // if (send_repl(connection, "331 Anonymous login okay, send your complete email as your password.\r\n") < 0)
    181.                                         if(send_repl(connection,"331 Anonymous\r\n") < 0)
    182.                                                 return -1;
    183.                                 }else if(strBeginWith(read_buff,"PASS")){
    184.                                         //可以看到ftp协议发送的密码均为明文发送
    185.                                         //这里不做密码校验，均答复已登录
    186.                                         // if (send_repl(connection, "230 User logged in, proceed.\r\n") < 0)
    187.                                         if (send_repl(connection, "230 User\r\n") < 0)
    188.                                                 return -1;
    189.                                 }else if (strBeginWith(read_buff,"SYST")){
    190.                                         //问操作系统类型
    191.                                         //下面UNIX的可替代选项参考 http://www.iana.org/assignments/operating-system-names/operating-system-names.xhtml
    192.                                         if (send_repl(connection, "215 UNIX\r\n") < 0)
    193.                                                 return -1;
    194.                                 }else if (strBeginWith(read_buff,"PWD")){
    195.                                         //当前路径
    196.                                         sprintf(send_buff,"257 %s\r\n",current_dir);
    197.                                         if (send_repl(connection, send_buff) < 0)
    198.                                                 return -1;
    199.                                 }else if (strBeginWith(read_buff,"PORT")){
    200.                                         //客户端发来的IP和端口号
    201.                                         char *temp = strtok(read_buff + 5,",");
    202.                                         strcpy(client_addr, temp);
    203.                                         for (i = 0; i < 3; ++i)
    204.                                         {
    205.                                                 strcat(client_addr, ".");
    206.                                                 temp = strtok(NULL, ",");
    207.                                                 strcat(client_addr, temp);
    208.                                         }
    209.                                         printf("client_addr = %s\n", client_addr);
    210.                                         temp = strtok(NULL, ",");
    211.                                         client_port = toint(temp);
    212.                                         temp = strtok(NULL, ",");
    213.                                         client_port = client_port * 256 + toint(temp);
    214.                                         printf("client_port = %d\n", client_port);
    215.                                         //回复传说中的200 OK
    216.                                         if (send_repl(connection, "200 \r\n") < 0)
    217.                                                 return -1;
    218.                                 }else if (strBeginWith(read_buff,"LIST")){
    219.                                         //当前路径
    220.                                         client_fd = make_client_connection(client_port, client_addr);
    221.                                         if (client_fd != -1){
    223.                                         }
    224.                                 }else{
    225.                                         //其他一律回复500,I don't know
    226.                                         if (send_repl(connection, "500 \r\n") < 0)
    227.                                                 return -1;
    228.                                 }
    229.                         }
    230.                 } else if(pid>0) {
    231.                         close(connection);
    232.                 }
    233.         }
    235.         return 0;
    236.        
    237. }
    Copy the Code

    用CuteFTP客户端登陆，打印信息如下：
    2.                 *** CuteFTP 9.0 - build Jun 25 2013 ***
    4. STATUS:>          [2014/4/30 0:28:13] Getting listing ""...
    5. STATUS:>          [2014/4/30 0:28:13] Connecting to FTP server... 192.168.11.29:21 (ip = 192.168.11.29)...
    6. STATUS:>          [2014/4/30 0:28:13] Socket connected. Waiting for welcome message...
    7.                 [2014/4/30 0:28:13] 220 Service ready for new user.
    8. STATUS:>          [2014/4/30 0:28:13] Connected. Authenticating...
    9. COMMAND:>        [2014/4/30 0:28:13] USER anonymous
    10.                 [2014/4/30 0:28:13] 331 Anonymous
    11. COMMAND:>        [2014/4/30 0:28:13] PASS *****
    12.                 [2014/4/30 0:28:13] 230 User
    13. STATUS:>          [2014/4/30 0:28:13] Login successful.
    14. COMMAND:>        [2014/4/30 0:28:13] SYST
    15.                 [2014/4/30 0:28:13] 215 UNIX
    16. STATUS:>          [2014/4/30 0:28:13] Host type detected: Unix.
    17. COMMAND:>        [2014/4/30 0:28:13] PWD
    18.                 [2014/4/30 0:28:13] 257 /home/taiji/桌面
    19. STATUS:>          [2014/4/30 0:28:13] Home directory: /home/taiji/桌面
    20. COMMAND:>        [2014/4/30 0:28:13] FEAT
    21.                 [2014/4/30 0:28:13] 500
    22. STATUS:>          [2014/4/30 0:28:13] This site doesn't support the 'features' command.
    23. STATUS:>          [2014/4/30 0:28:13] Setting up character encoding.
    24. COMMAND:>        [2014/4/30 0:28:13] OPTS UTF8 on
    25.                 [2014/4/30 0:28:13] 500
    26. STATUS:>          [2014/4/30 0:28:13] Using local encoding.
    27. COMMAND:>        [2014/4/30 0:28:13] REST 100
    28.                 [2014/4/30 0:28:13] 500
    29. STATUS:>          [2014/4/30 0:28:13] This site cannot resume broken downloads.
    30. COMMAND:>        [2014/4/30 0:28:13] PASV
    31.                 [2014/4/30 0:28:13] 500
    32. STATUS:>          [2014/4/30 0:28:13] PASV failed, trying PORT.
    33. COMMAND:>        [2014/4/30 0:28:13] PORT 192,168,11,16,209,127
    34.                 [2014/4/30 0:28:13] 200
    35. COMMAND:>        [2014/4/30 0:28:13] LIST
    Copy the Code
    Reply Favorite View the author
    All Replies

    No replies yet