公共部分代码
/* common.h */ #ifndef COMMON_H #define COMMON_H #include <arpa/inet.h> #include <ctype.h> #include <dirent.h> #include <errno.h> #include <fcntl.h> #include <netdb.h> #include <netinet/in.h> #include <stdio.h> #include <stdlib.h> #include <string.h> #include <sys/wait.h> #include <sys/socket.h> #include <sys/types.h> #include <unistd.h> #define DEBUG 1 #define MAXSIZE 512 #define CLIENT_PORT_ID 30020 struct command { char arg[255]; char code[5]; }; int socket_create(int port); int socket_accept(int sock_listen); int socket_connect(int port, char *host); int recv_data(int sockfd, char* buf, int bufsize); int send_response(int sockfd, int rc); void trimstr(char *str, int n); void read_input(char* buffer, int size); #endif
/* common.c */ #include "common.h" /** * 创建监听套接字 * 错误返回 -1,正确返回套接字描述符 */ int socket_create(int port) { int sockfd; int yes = 1; struct sockaddr_in sock_addr; // 创建套接字 if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0) { perror("socket() error"); return -1; } // 设置本地套接字地址 sock_addr.sin_family = AF_INET; sock_addr.sin_port = htons(port); sock_addr.sin_addr.s_addr = htonl(INADDR_ANY); if (setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(int)) == -1) { close(sockfd); perror("setsockopt() error"); return -1; } // 绑定本地套接字地址到套接字 if (bind(sockfd, (struct sockaddr *) &sock_addr, sizeof(sock_addr)) < 0) { close(sockfd); perror("bind() error"); return -1; } // 将套接字设置为监听状态 if (listen(sockfd, 5) < 0) { close(sockfd); perror("listen() error"); return -1; } return sockfd; } /** * 套接字接受请求 * 错误返回 -1,正确返回新的连接套接字 */ int socket_accept(int sock_listen) { int sockfd; struct sockaddr_in client_addr; socklen_t len = sizeof(client_addr); sockfd = accept(sock_listen, (struct sockaddr *) &client_addr, &len); // 等待连接 if (sockfd < 0) { perror("accept() error"); return -1; } return sockfd; } /** * 连接远端主机 * 成功返回套接字描述符,失败返回 -1 */ int socket_connect(int port, char*host) { int sockfd; struct sockaddr_in dest_addr; /* 创建套接字 */ if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0) { perror("error creating socket"); return -1; } /* 设置协议地址 */ memset(&dest_addr, 0, sizeof(dest_addr)); dest_addr.sin_family = AF_INET; dest_addr.sin_port = htons(port); dest_addr.sin_addr.s_addr = inet_addr(host); /* 在套接字上创建连接 */ if(connect(sockfd, (struct sockaddr *)&dest_addr, sizeof(dest_addr)) < 0 ) { perror("error connecting to server"); return -1; } return sockfd; } /** * 接收数据 * 错误返回 -1,正确返回接收的字节数 */ int recv_data(int sockfd, char* buf, int bufsize) { size_t num_bytes; memset(buf, 0, bufsize); /* 调用 recv 函数读取套接字数据 */ num_bytes = recv(sockfd, buf, bufsize, 0); if (num_bytes < 0) return -1; return num_bytes; } /** * 去除字符串中的空格和换行符 */ void trimstr(char *str, int n) { int i; for (i = 0; i < n; i++) { if (isspace(str[i])) str[i] = 0; if (str[i] == '\n') str[i] = 0; } } /** * 发送响应码到 sockfd * 错误返回 -1,正确返回 0 */ int send_response(int sockfd, int rc) { int conv = htonl(rc); if (send(sockfd, &conv, sizeof conv, 0) < 0 ) { perror("error sending...\n"); return -1; } return 0; } /** * 从命令行中读取输入 */ void read_input(char* buffer, int size) { char *nl = NULL; memset(buffer, 0, size); if ( fgets(buffer, size, stdin) != NULL ) { nl = strchr(buffer, '\n'); if (nl) *nl = '\0'; // 出现换行符,则将该位置部位'\0'(字符串结尾) } }
客户端代码:
/* ftclient.h */ #ifndef FTCLIENT_H #define FTCLIENT_H #include "../common/common.h" int read_reply(); void print_reply(int rc); int ftclient_read_command(char* buf, int size, struct command *cstruct); int ftclient_get(int data_sock, int sock_control, char* arg); int ftclient_open_conn(int sock_con); int ftclient_list(int sock_data, int sock_con); int ftclient_send_cmd(struct command *cmd); void ftclient_login(); #endif
/* ftclient.c */ #include "ftclient.h" int sock_control; /** * 接收服务器响应 * 错误返回 -1,正确返回状态码 */ int read_reply() { int retcode = 0; if (recv(sock_control, &retcode, sizeof retcode, 0) < 0) { perror("client: error reading message from server\n"); return -1; } return ntohl(retcode); } /** * 打印响应信息 */ void print_reply(int rc) { switch (rc) { case 220: printf("220 Welcome, server ready.\n"); break; case 221: printf("221 Goodbye!\n"); break; case 226: printf("226 Closing data connection. Requested file action successful.\n"); break; case 550: printf("550 Requested action not taken. File unavailable.\n"); break; } } /** * 解析命令行到结构体 */ int ftclient_read_command(char* buf, int size, struct command *cstruct) { memset(cstruct->code, 0, sizeof(cstruct->code)); memset(cstruct->arg, 0, sizeof(cstruct->arg)); printf("ftclient> "); // 输入提示符 fflush(stdout); read_input(buf, size); // 等待用户输入命令 char *arg = NULL; arg = strtok (buf," "); arg = strtok (NULL, " "); if (arg != NULL) strncpy(cstruct->arg, arg, strlen(arg)); if (strcmp(buf, "list") == 0) strcpy(cstruct->code, "LIST"); else if (strcmp(buf, "get") == 0) strcpy(cstruct->code, "RETR"); else if (strcmp(buf, "quit") == 0) strcpy(cstruct->code, "QUIT"); else return -1; // 不合法 memset(buf, 0, 400); strcpy(buf, cstruct->code); // 存储命令到 buf 开始处 /* 如果命令带有参数,追加到 buf */ if (arg != NULL) { strcat(buf, " "); strncat(buf, cstruct->arg, strlen(cstruct->arg)); } return 0; } /** * 实现 get <filename> 命令行 */ int ftclient_get(int data_sock, int sock_control, char* arg) { char data[MAXSIZE]; int size; FILE* fd = fopen(arg, "w"); // 创建并打开名字为 arg 的文件 /* 将服务器传来的数据(文件内容)写入本地建立的文件 */ while ((size = recv(data_sock, data, MAXSIZE, 0)) > 0) fwrite(data, 1, size, fd); if (size < 0) perror("error\n"); fclose(fd); return 0; } /** * 打开数据连接 */ int ftclient_open_conn(int sock_con) { int sock_listen = socket_create(CLIENT_PORT_ID); /* 在控制连接上发起一个 ACK 确认 */ int ack = 1; if ((send(sock_con, (char*) &ack, sizeof(ack), 0)) < 0) { printf("client: ack write error :%d\n", errno); exit(1); } int sock_conn = socket_accept(sock_listen); close(sock_listen); return sock_conn; } /** * 实现 list 命令 */ int ftclient_list(int sock_data, int sock_con) { size_t num_recvd; char buf[MAXSIZE]; int tmp = 0; /* 等待服务器启动的信息 */ if (recv(sock_con, &tmp, sizeof tmp, 0) < 0) { perror("client: error reading message from server\n"); return -1; } memset(buf, 0, sizeof(buf)); /* 接收服务器传来的数据 */ while ((num_recvd = recv(sock_data, buf, MAXSIZE, 0)) > 0) { printf("%s", buf); memset(buf, 0, sizeof(buf)); } if (num_recvd < 0) perror("error"); /* 等待服务器完成的消息 */ if (recv(sock_con, &tmp, sizeof tmp, 0) < 0) { perror("client: error reading message from server\n"); return -1; } return 0; } /** * 输入含有命令(code)和参数(arg)的 command(cmd) 结构 * 连接 code + arg,并放进一个字符串,然后发送给服务器 */ int ftclient_send_cmd(struct command *cmd) { char buffer[MAXSIZE]; int rc; sprintf(buffer, "%s %s", cmd->code, cmd->arg); /* 发送命令字符串到服务器 */ rc = send(sock_control, buffer, (int)strlen(buffer), 0); if (rc < 0) { perror("Error sending command to server"); return -1; } return 0; } /** * 获取登录信息 * 发送到服务器认证 */ void ftclient_login() { struct command cmd; char user[256]; memset(user, 0, 256); /* 获取用户名 */ printf("Name: "); fflush(stdout); read_input(user, 256); /* 发送用户名到服务器 */ strcpy(cmd.code, "USER"); strcpy(cmd.arg, user); ftclient_send_cmd(&cmd); /* 等待应答码 331 */ int wait; recv(sock_control, &wait, sizeof wait, 0); /* 获得密码 */ fflush(stdout); char *pass = getpass("Password: "); /* 发送密码到服务器 */ strcpy(cmd.code, "PASS"); strcpy(cmd.arg, pass); ftclient_send_cmd(&cmd); /* 等待响应 */ int retcode = read_reply(); switch (retcode) { case 430: printf("Invalid username/password.\n"); exit(0); case 230: printf("Successful login.\n"); break; default: perror("error reading message from server"); exit(1); break; } } /* 主函数入口 */ int main(int argc, char* argv[]) { int data_sock, retcode, s; char buffer[MAXSIZE]; struct command cmd; struct addrinfo hints, *res, *rp; /* 命令行参数合法性检测 */ if (argc != 3) { printf("usage: ./ftclient hostname port\n"); exit(0); } char *host = argv[1]; //所要连接的服务器主机名 char *port = argv[2]; //所要链接到服务器程序端口号 /* 获得和服务器名匹配的地址 */ memset(&hints, 0, sizeof(struct addrinfo)); hints.ai_family = AF_UNSPEC; hints.ai_socktype = SOCK_STREAM; s = getaddrinfo(host, port, &hints, &res); if (s != 0) { printf("getaddrinfo() error %s", gai_strerror(s)); exit(1); } /* 找到对应的服务器地址并连接 */ for (rp = res; rp != NULL; rp = rp->ai_next) { sock_control = socket(rp->ai_family, rp->ai_socktype, rp->ai_protocol); // 创建控制套接字 if (sock_control < 0) continue; if(connect(sock_control, res->ai_addr, res->ai_addrlen)==0) // 和服务器连接 break; else { perror("connecting stream socket"); exit(1); } close(sock_control); } freeaddrinfo(rp); /* 连接成功,打印信息 */ printf("Connected to %s.\n", host); print_reply(read_reply()); /* 获取用户的名字和密码 */ ftclient_login(); while (1) { // 循环,直到用户输入 quit /* 得到用户输入的命令 */ if ( ftclient_read_command(buffer, sizeof buffer, &cmd) < 0) { printf("Invalid command\n"); continue; // 跳过本次循环,处理下一个命令 } /* 发送命令到服务器 */ if (send(sock_control, buffer, (int)strlen(buffer), 0) < 0 ) { close(sock_control); exit(1); } retcode = read_reply(); //读取服务器响应(服务器是否可以支持该命令?) if (retcode == 221) // 退出命令 { print_reply(221); break; } if (retcode == 502) printf("%d Invalid command.\n", retcode);// 不合法的输入,显示错误信息 else { // 命令合法 (RC = 200),处理命令 /* 打开数据连接 */ if ((data_sock = ftclient_open_conn(sock_control)) < 0) { perror("Error opening socket for data connection"); exit(1); } /* 执行命令 */ if (strcmp(cmd.code, "LIST") == 0) ftclient_list(data_sock, sock_control); else if (strcmp(cmd.code, "RETR") == 0) { if (read_reply() == 550) // 等待回复 { print_reply(550); close(data_sock); continue; } ftclient_get(data_sock, sock_control, cmd.arg); print_reply(read_reply()); } close(data_sock); } } // 循环得到更多的用户输入 close(sock_control); // 关闭套接字控制连接 return 0; }
makefile文件
CC := gcc CFLAGS := -Wall -g -Os SHDIR := ../common OBJS = ftclient.o $(SHDIR)/common.o all: ftclient ftclient: $(OBJS) @$(CC) -o ftclient $(CFLAGS) $(OBJS) $(OBJS) : %.o: %.c @$(CC) -c $(CFLAGS) $< -o $@ .PHONY: clean: @rm -f *.o ftclient @rm -f ../common/*.o @echo Done cleaning
服务端代码:
/* ftserve.h */ #ifndef FTSERVE_H #define FTSERVE_H #include "../common/common.h" void ftserve_retr(int sock_control, int sock_data, char* filename); int ftserve_list(int sock_data, int sock_control); int ftserve_start_data_conn(int sock_control); int ftserve_check_user(char*user, char*pass); int ftserve_login(int sock_control); int ftserve_recv_cmd(int sock_control, char*cmd, char*arg); void ftserve_process(int sock_control); #endif
/* ftserve.c */ #include "ftserve.h" /* 主函数入口 */ int main(int argc, char *argv[]) { int sock_listen, sock_control, port, pid; /* 命令行合法性检测 */ if (argc != 2) { printf("usage: ./ftserve port\n"); exit(0); } /* 将命令行传进来的服务器端口号(字符串)转换为整数 */ port = atoi(argv[1]); /* 创建监听套接字 */ if ((sock_listen = socket_create(port)) < 0 ) { perror("Error creating socket"); exit(1); } /* 循环接受不同的客户机请求 */ while(1) { /* 监听套接字接受连接请求,得到控制套接字,用于传递控制信息 */ if ((sock_control = socket_accept(sock_listen)) < 0 ) break; /* 创建子进程处理用户请求 */ if ((pid = fork()) < 0) perror("Error forking child process"); /* 子进程调用 ftserve_process 函数与客户端交互 */ else if (pid == 0) { close(sock_listen); // 子进程关闭父进程的监听套接字 ftserve_process(sock_control); close(sock_control); //用户请求处理完毕,关闭该套接字 exit(0); } close(sock_control); // 父进程关闭子进程的控制套接字 } close(sock_listen); return 0; } /** * 通过数据套接字发送特定的文件 * 控制信息交互通过控制套接字 * 处理无效的或者不存在的文件名 */ void ftserve_retr(int sock_control, int sock_data, char* filename) { FILE* fd = NULL; char data[MAXSIZE]; size_t num_read; fd = fopen(filename, "r"); // 打开文件 if (!fd) send_response(sock_control, 550); // 发送错误码 (550 Requested action not taken) else { send_response(sock_control, 150); // 发送 okay (150 File status okay) do { num_read = fread(data, 1, MAXSIZE, fd); // 读文件内容 if (num_read < 0) printf("error in fread()\n"); if (send(sock_data, data, num_read, 0) < 0) // 发送数据(文件内容) perror("error sending file\n"); } while (num_read > 0); send_response(sock_control, 226); // 发送消息:226: closing conn, file transfer successful fclose(fd); } } /** * 响应请求:发送当前所在目录的目录项列表 * 关闭数据连接 * 错误返回 -1,正确返回 0 */ int ftserve_list(int sock_data, int sock_control) { char data[MAXSIZE]; size_t num_read; FILE* fd; int rs = system("ls -l | tail -n+2 > tmp.txt"); //利用系统调用函数 system 执行命令,并重定向到 tmp.txt 文件 if ( rs < 0) { exit(1); } fd = fopen("tmp.txt", "r"); if (!fd) exit(1); /* 定位到文件的开始处 */ fseek(fd, SEEK_SET, 0); send_response(sock_control, 1); memset(data, 0, MAXSIZE); /* 通过数据连接,发送tmp.txt 文件的内容 */ while ((num_read = fread(data, 1, MAXSIZE, fd)) > 0) { if (send(sock_data, data, num_read, 0) < 0) perror("err"); memset(data, 0, MAXSIZE); } fclose(fd); send_response(sock_control, 226); // 发送应答码 226(关闭数据连接,请求的文件操作成功) return 0; } /** * 创建到客户机的一条数据连接 * 成功返回数据连接的套接字 * 失败返回 -1 */ int ftserve_start_data_conn(int sock_control) { char buf[1024]; int wait, sock_data; if (recv(sock_control, &wait, sizeof wait, 0) < 0 ) { perror("Error while waiting"); return -1; } struct sockaddr_in client_addr; socklen_t len = sizeof client_addr; getpeername(sock_control, (struct sockaddr*)&client_addr, &len); // 获得与控制套接字关联的外部地址(客户端地址) inet_ntop(AF_INET, &client_addr.sin_addr, buf, sizeof(buf)); /* 创建到客户机的数据连接 */ if ((sock_data = socket_connect(CLIENT_PORT_ID, buf)) < 0) return -1; return sock_data; } /** * 用户资格认证 * 认证成功返回 1,否则返回 0 */ int ftserve_check_user(char*user, char*pass) { char username[MAXSIZE]; char password[MAXSIZE]; char *pch; char buf[MAXSIZE]; char *line = NULL; size_t num_read; size_t len = 0; FILE* fd; int auth = 0; fd = fopen(".auth", "r"); //打开认证文件(记录用户名和密码) if (fd == NULL) { perror("file not found"); exit(1); } /* 读取".auth" 文件中的用户名和密码,验证用户身份的合法性 */ while ((num_read = getline(&line, &len, fd)) != -1) { memset(buf, 0, MAXSIZE); strcpy(buf, line); pch = strtok (buf," "); strcpy(username, pch); if (pch != NULL) { pch = strtok (NULL, " "); strcpy(password, pch); } /* 去除字符串中的空格和换行符 */ trimstr(password, (int)strlen(password)); if ((strcmp(user,username)==0) && (strcmp(pass,password)==0)) { auth = 1; // 匹配成功,标志变量 auth = 1,并返回 break; } } free(line); fclose(fd); return auth; } /* 用户登录*/ int ftserve_login(int sock_control) { char buf[MAXSIZE]; char user[MAXSIZE]; char pass[MAXSIZE]; memset(user, 0, MAXSIZE); memset(pass, 0, MAXSIZE); memset(buf, 0, MAXSIZE); /* 获得客户端传来的用户名 */ if ( (recv_data(sock_control, buf, sizeof(buf)) ) == -1) { perror("recv error\n"); exit(1); } int i = 5; int n = 0; while (buf[i] != 0) //buf[0-4]="USER" user[n++] = buf[i++]; /* 用户名正确,通知用户输入密码 */ send_response(sock_control, 331); /* 获得客户端传来的密码 */ memset(buf, 0, MAXSIZE); if ( (recv_data(sock_control, buf, sizeof(buf)) ) == -1) { perror("recv error\n"); exit(1); } i = 5; n = 0; while (buf[i] != 0) // buf[0 - 4] = "PASS" pass[n++] = buf[i++]; return (ftserve_check_user(user, pass)); // 用户名和密码验证,并返回 } /* 接收客户端的命令并响应,返回响应码 */ int ftserve_recv_cmd(int sock_control, char*cmd, char*arg) { int rc = 200; char buffer[MAXSIZE]; memset(buffer, 0, MAXSIZE); memset(cmd, 0, 5); memset(arg, 0, MAXSIZE); /* 接受客户端的命令 */ if ((recv_data(sock_control, buffer, sizeof(buffer)) ) == -1) { perror("recv error\n"); return -1; } /* 解析出用户的命令和参数 */ strncpy(cmd, buffer, 4); char *tmp = buffer + 5; strcpy(arg, tmp); if (strcmp(cmd, "QUIT")==0) rc = 221; else if ((strcmp(cmd, "USER") == 0) || (strcmp(cmd, "PASS") == 0) || (strcmp(cmd, "LIST") == 0) || (strcmp(cmd, "RETR") == 0)) rc = 200; else rc = 502; // 无效的命令 send_response(sock_control, rc); return rc; } /* 处理客户端请求 */ void ftserve_process(int sock_control) { int sock_data; char cmd[5]; char arg[MAXSIZE]; send_response(sock_control, 220); // 发送欢迎应答码 /* 用户认证 */ if (ftserve_login(sock_control) == 1) // 认证成功 send_response(sock_control, 230); else { send_response(sock_control, 430); // 认证失败 exit(0); } /* 处理用户的请求 */ while (1) { /* 接收命令,并解析,获得命令和参数 */ int rc = ftserve_recv_cmd(sock_control, cmd, arg); if ((rc < 0) || (rc == 221)) // 用户输入命令 "QUIT" break; if (rc == 200 ) { /* 创建和客户端的数据连接 */ if ((sock_data = ftserve_start_data_conn(sock_control)) < 0) { close(sock_control); exit(1); } /* 执行指令 */ if (strcmp(cmd, "LIST")==0) ftserve_list(sock_data, sock_control); else if (strcmp(cmd, "RETR")==0) ftserve_retr(sock_control, sock_data, arg); close(sock_data);// 关闭连接 } } }
里面还有一个保存密码账户文件.auth,这里就不列出。服务端的makefile和客户端makefile一样,只需要修改一下程序名及相关依赖名即可。