• Home
  • Categories
  • WIKI
  • LANGUAGE: en
    • 自己动手写ftp服务器3(像模像样)
    社区开发 1778 views · 2 replies ·
    Tofloor
    poster avatar
    zhqh100
    deepin
    2014-05-01 06:43
    Author
    接下来实现List功能，之后就可以看到文件夹下的列表了，也可以证明，这确实是一个像ftp服务器的一堆代码了

    开发语言：C语言
    操作系统：LinuxDeepin 2013
    2. #include
    3. #include
    4. #include
    5. #include
    6. #include
    7. #include
    8. #include
    9. #include
    10. #include
    11. #include
    12. #include
    13. #include
    14. #define bool unsigned short
    16. typedef struct cmd_opts {
    17.         bool daemonize;
    18.         bool listen_any;
    19.         char *listen_addr;
    20.         int port;
    21.         int max_conn;
    22.         int userid;
    23.         char *chrootdir;
    24. }cmd_opts;
    26. //格式化字符串line
    27. static int write_file(char *line,const char *mode,int num,const char *user,const char * group,int size,const char *date,const char*fl_name) {
    28.         sprintf(line,"%s %3d %-4s %-4s %8d %12s %s\r\n",mode,num,user,group,size,date,fl_name);
    29.         return 0;
    30. }
    32. //判断字符是否是以某指令开始
    33. int strBeginWith(char *str,char *substr){
    34.         while(*str && *substr && *str==*substr){
    35.                 str++;
    36.                 substr++;
    37.         }
    38.         if(*substr=='\0')
    39.                 return -1;
    40.         else
    41.                 return 0;
    42. }
    44. //字符串转int
    45. int toint(char *str) {
    46.         if (strlen(str) < 1)
    47.         {
    48.                 return -1;
    49.         }
    50.         int retVal = str[0] - '0';
    52.         int i = 1;
    53.         while(str[i] >= '0' && str[i] <= '9')
    54.         {
    55.                 retVal = retVal * 10 + str[i] - '0';
    56.                 i++;
    57.         }
    58.         return retVal;
    59. }
    61. //向客户端发送消息
    62. int send_repl(int send_sock,char *msg) {
    63.         printf("Send Message:%s\n", msg);
    64.         if (send(send_sock, msg, strlen(msg),0) < 0) {
    65.                 perror("send");
    66.                 close(send_sock);
    67.                 return -1;
    68.         }
    69.         return 0;
    70. }
    72. //创建数据传输的连接，用于处理"LIST","STOR","RETR"等请求
    73. int make_client_connection(int client_port,const char* client_addr) {
    74.         int sock=-1;
    76.         struct sockaddr_in servaddr;
    77.         servaddr.sin_family = AF_INET;
    78.         servaddr.sin_addr.s_addr = inet_addr(client_addr);
    79.         servaddr.sin_port = htons (client_port);
    81.         if ((sock = socket (PF_INET, SOCK_STREAM, IPPROTO_TCP)) < 0) {
    82.                 send_repl(sock,"425 Can't open data connection.\r\n");
    83.                 perror("socket");
    84.                 return -1;
    85.         }
    86.         int status = connect (sock, (struct sockaddr *)&servaddr, sizeof (servaddr));
    88.         if(status!=0) {
    89.                 send_repl(sock,"425 Can't open data connection.\r\n");
    90.                 perror("connect");
    91.                 return -1;
    92.         }
    93.         return sock;
    94. }
    96. //zhqh100@gmail.com
    97. enum {
    98.         TRUE=1,
    99.         FALSE=0
    100. };
    102. int main(int argc,char *argv[])
    103. {
    104.         struct cmd_opts *opts= malloc(sizeof(struct cmd_opts));
    106.         //设置一些初始配置
    107.         opts->daemonize    = FALSE;
    108.         opts->listen_any   = TRUE;
    109.         opts->port         = 21;
    110.         opts->userid       = 0;
    111.         opts->chrootdir    = "./";
    112.         opts->max_conn     = 5;
    113.         opts->listen_addr  = NULL;
    114.         struct sockaddr_in servaddr;
    115.         memset((char *)&servaddr, 0, sizeof(servaddr));
    117.         //TCP协议
    118.         servaddr.sin_family = PF_INET;
    119.         servaddr.sin_addr.s_addr =  htonl(INADDR_ANY);
    120.         servaddr.sin_port = htons (opts->port);
    122.         int servaddr_len = sizeof(servaddr);
    123.         int sock;
    125.         if ((sock = socket (PF_INET, SOCK_STREAM, IPPROTO_TCP)) < 0) {
    126.                 perror("socket");
    127.                 return -1;
    128.         }
    130.         int flag = 1;
    131.         setsockopt(sock, SOL_SOCKET,SO_REUSEADDR,(char *) &flag, sizeof(int));
    132.         setsockopt(sock, IPPROTO_TCP,TCP_NODELAY,(char *) &flag, sizeof(int));
    134.         //套接字绑定端口和地址
    135.         if(bind (sock, (struct sockaddr *)&servaddr, sizeof(servaddr))<0) {
    136.                 perror("bind");
    137.                 return -1;
    138.         }
    140.         if(listen(sock,opts->max_conn) <0) {
    141.                 perror("listen");
    142.                 return -1;
    143.         }
    145.         int connection;
    146.         struct stat *s_buff = malloc(sizeof(struct stat));
    147.         char mode[11]        = "----------";
    148.         char date[16];
    149.         struct passwd * pass_info = NULL;
    150.         struct group * group_info;
    151.         struct dirent *d_next;
    152.         char line[300];
    153.         DIR *dir;
    155.         //死循环等待客户端连接
    156.         while(1){
    157.                 printf("Begin to accept\n");
    158.                 if ((connection = accept(sock, (struct sockaddr *) &servaddr, &servaddr_len)) < 0) {
    159.                         perror("accept");
    160.                         return -1;
    161.                 }
    163.                 int pid = fork();
    165.                 char msg[]= "220 Service ready for new user.\r\n";
    166.                 char send_buff[1000];
    167.                 char read_buff[1000];
    168.                 char current_dir[1000];
    170.                 //客户端IP地址和端口号
    171.                 char client_addr[16];
    172.                 int client_port;
    174.                 //临时指针
    175.                 char *temp;
    176.                 int i;
    177.                 int client_fd;
    178.                 strcpy(current_dir,opts->chrootdir);
    180.                 //获取当前绝对路径
    181.                 getcwd(current_dir,sizeof(current_dir));
    183.                 int len = strlen(msg);
    184.                 if(pid==0) {
    185.                         //客户端连接后，服务器先主动发送200 Service ready
    186.                         if (send(connection, msg, len,0) < 0) {
    187.                                 perror("send");
    188.                                 close(connection);
    189.                                 return -1;
    190.                         }
    191.                         while(1){
    192.                                 //初始化读取buffer和发送buffer
    193.                                 memset(send_buff,0,sizeof(send_buff));
    194.                                 memset(read_buff,0,sizeof(read_buff));
    195.                                 //等待接收客户端指令
    196.                                 int recvbuff = recv(connection,read_buff,sizeof(read_buff),0);
    197.                                 if(recvbuff<1) {
    198.                                         return -1;
    199.                                 }
    200.                                 if(recvbuff==sizeof(read_buff)) {
    201.                                         return -2;
    202.                                 }
    203.                                 printf("Received:%s",read_buff);
    204.                                 if (strBeginWith(read_buff,"USER")){
    205.                                         //收到客户端发送的用户名，这里不处理用户名，均答复登陆成功
    206.                                         //注释代码也可用，只是这里修改后，发送的消息更精简了
    207.                                         //后面的人性化提示可以省略，但\r\n不可省略，经Filezilla和 CuteFTP客户端验证，必须把\r\n带上才可正常识别
    208.                                         // if (send_repl(connection, "331 Anonymous login okay, send your complete email as your password.\r\n") < 0)
    209.                                         if(send_repl(connection,"331 Anonymous\r\n") < 0)
    210.                                                 return -1;
    211.                                 }else if(strBeginWith(read_buff,"PASS")){
    212.                                         //可以看到ftp协议发送的密码均为明文发送
    213.                                         //这里不做密码校验，均答复已登录
    214.                                         // if (send_repl(connection, "230 User logged in, proceed.\r\n") < 0)
    215.                                         if (send_repl(connection, "230 User\r\n") < 0)
    216.                                                 return -1;
    217.                                 }else if (strBeginWith(read_buff,"SYST")){
    218.                                         //问操作系统类型
    219.                                         //下面UNIX的可替代选项参考 http://www.iana.org/assignments/operating-system-names/operating-system-names.xhtml
    220.                                         if (send_repl(connection, "215 UNIX\r\n") < 0)
    221.                                                 return -1;
    222.                                 }else if (strBeginWith(read_buff,"PWD")){
    223.                                         //当前路径
    224.                                         sprintf(send_buff,"257 %s\r\n",current_dir);
    225.                                         if (send_repl(connection, send_buff) < 0)
    226.                                                 return -1;
    227.                                 }else if (strBeginWith(read_buff,"PORT")){
    228.                                         //客户端发来的IP和端口号
    229.                                         char *temp = strtok(read_buff + 5,",");
    230.                                         strcpy(client_addr, temp);
    231.                                         for (i = 0; i < 3; ++i)
    232.                                         {
    233.                                                 strcat(client_addr, ".");
    234.                                                 temp = strtok(NULL, ",");
    235.                                                 strcat(client_addr, temp);
    236.                                         }
    237.                                         printf("client_addr = %s\n", client_addr);
    238.                                         temp = strtok(NULL, ",");
    239.                                         client_port = toint(temp);
    240.                                         temp = strtok(NULL, ",");
    241.                                         client_port = client_port * 256 + toint(temp);
    242.                                         printf("client_port = %d\n", client_port);
    243.                                         //回复传说中的200 OK
    244.                                         if (send_repl(connection, "200 \r\n") < 0)
    245.                                                 return -1;
    246.                                 }else if (strBeginWith(read_buff,"LIST")){
    247.                                         //当前路径
    248.                                         client_fd = make_client_connection(client_port, client_addr);
    249.                                         if (client_fd != -1){
    250.                                                 dir = opendir(current_dir);
    251.                                                 send_repl(connection,"150 File status okay; about to open data connection.\r\n");
    253.                                                 while(1) {
    254.                                                         d_next = readdir(dir);
    255.                                                         if(d_next==NULL)
    256.                                                                 break;
    257.                                                         line[0]='\0';
    258.                                                         stat(d_next->d_name,s_buff);
    259.                                                         int b_mask = s_buff->st_mode & S_IFMT;
    261.                                                         //判断是文件还是文件夹
    262.                                                         if(b_mask == S_IFDIR) {
    263.                                                                 mode[0]='d';
    264.                                                         } else if(b_mask == S_IFREG){
    265.                                                                 mode[0]='-';
    266.                                                         } else {
    267.                                                                 return FALSE;
    268.                                                         }
    270.                                                         mode[1] = (s_buff->st_mode & S_IRUSR)?'r':'-';
    271.                                                         mode[2] = (s_buff->st_mode & S_IWUSR)?'w':'-';
    272.                                                         mode[3] = (s_buff->st_mode & S_IXUSR)?'x':'-';
    273.                                                         mode[4] = (s_buff->st_mode & S_IRGRP)?'r':'-';
    274.                                                         mode[5] = (s_buff->st_mode & S_IWGRP)?'w':'-';
    275.                                                         mode[6] = (s_buff->st_mode & S_IXGRP)?'x':'-';
    276.                                                         mode[7] = (s_buff->st_mode & S_IROTH)?'r':'-';
    277.                                                         mode[8] = (s_buff->st_mode & S_IWOTH)?'w':'-';
    278.                                                         mode[9] = (s_buff->st_mode & S_IXOTH)?'x':'-';
    280.                                                         strftime(date,13,"%b %d %H:%M",localtime(&(s_buff->st_mtime)));
    281.                                                         pass_info = getpwuid(s_buff->st_uid);
    282.                                                         group_info = getgrgid(s_buff->st_gid);
    283.                                                         write_file(line,mode,s_buff->st_nlink,pass_info->pw_name,group_info->gr_name,s_buff->st_size,date,d_next->d_name);
    284.                                                         printf("line = %s\n", line);
    285.                                                         send(client_fd,line,strlen(line),0);
    286.                                                 }
    287.                                                 close(client_fd);
    288.                                                 send_repl(connection,"226 Closing data connection.\r\n");
    289.                                                 closedir(dir);
    290.                                         }
    291.                                 }else{
    292.                                         //其他一律回复500,I don't know
    293.                                         if (send_repl(connection, "500 \r\n") < 0)
    294.                                                 return -1;
    295.                                 }
    296.                         }
    297.                 } else if(pid>0) {
    298.                         close(connection);
    299.                 }
    300.         }
    301.         return 0;
    302. }
    Copy the Code

    CuteFtp登陆看到如下信息：
    QQ图片20140430222730.jpg
    Reply Favorite View the author
    All Replies
    cxbii
    deepin
    2014-05-02 04:24
    #1
    支持，楼主是否考虑出个专题？
    Reply View the author
    zhqh100
    deepin
    2014-05-02 06:34
    #2
    支持，楼主是否考虑出个专题？

    专题怎么出啊？
    我这个再有两篇就很快完结了，不会很深入，接下来计划是《自己动手写http服务器》
    Reply View the author