你好,今天是期中大作业讲解课。诚如一位同学所言,这次的大作业不是在考察网络编程的细节,而是在考如何使用系统API完成cd、pwd、ls等功能。不过呢,网络编程的框架总归还是要掌握的。
我研读了大部分同学的代码,基本上是做得不错的,美中不足的是能动手完成代码编写和调试的同学偏少。我还是秉持一贯的看法,计算机程序设计是一门实战性很强的学科,如果只是单纯地听讲解,没有自己动手这一环,对知识的掌握总归还是差那么点意思。
代码我已经push到这里,你可以点进链接看一下。
废话少说,我贴下我的客户端程序:
#include "lib/common.h"
#define MAXLINE 1024
int main(int argc, char **argv) {
if (argc != 3) {
error(1, 0, "usage: tcp_client <IPaddress> <port>");
}
int port = atoi(argv[2]);
int socket_fd = tcp_client(argv[1], port);
char recv_line[MAXLINE], send_line[MAXLINE];
int n;
fd_set readmask;
fd_set allreads;
FD_ZERO(&allreads);
FD_SET(0, &allreads);
FD_SET(socket_fd, &allreads);
for (;;) {
readmask = allreads;
int rc = select(socket_fd + 1, &readmask, NULL, NULL, NULL);
if (rc <= 0) {
error(1, errno, "select failed");
}
if (FD_ISSET(socket_fd, &readmask)) {
n = read(socket_fd, recv_line, MAXLINE);
if (n < 0) {
error(1, errno, "read error");
} else if (n == 0) {
printf("server closed \n");
break;
}
recv_line[n] = 0;
fputs(recv_line, stdout);
fputs("\n", stdout);
}
if (FD_ISSET(STDIN_FILENO, &readmask)) {
if (fgets(send_line, MAXLINE, stdin) != NULL) {
int i = strlen(send_line);
if (send_line[i - 1] == '\n') {
send_line[i - 1] = 0;
}
if (strncmp(send_line, "quit", strlen(send_line)) == 0) {
if (shutdown(socket_fd, 1)) {
error(1, errno, "shutdown failed");
}
}
size_t rt = write(socket_fd, send_line, strlen(send_line));
if (rt < 0) {
error(1, errno, "write failed ");
}
}
}
}
exit(0);
}
客户端的代码主要考虑的是使用select同时处理标准输入和套接字,我看到有同学使用fgets来循环等待用户输入,然后再把输入的命令通过套接字发送出去,当然也是可以正常工作的,只不过不能及时响应来自服务端的命令结果,所以,我还是推荐使用select来同时处理标准输入和套接字。
这里select如果发现标准输入有事件,读出标准输入的字符,就会通过调用write方法发送出去。如果发现输入的是quit,则调用shutdown方法关闭连接的一端。
如果select发现套接字流有可读事件,则从套接字中读出数据,并把数据打印到标准输出上;如果读到了EOF,表示该客户端需要退出,直接退出循环,通过调用exit来完成进程的退出。
下面是我写的服务器端程序:
#include "lib/common.h"
static int count;
static void sig_int(int signo) {
printf("\nreceived %d datagrams\n", count);
exit(0);
}
char *run_cmd(char *cmd) {
char *data = malloc(16384);
bzero(data, sizeof(data));
FILE *fdp;
const int max_buffer = 256;
char buffer[max_buffer];
fdp = popen(cmd, "r");
char *data_index = data;
if (fdp) {
while (!feof(fdp)) {
if (fgets(buffer, max_buffer, fdp) != NULL) {
int len = strlen(buffer);
memcpy(data_index, buffer, len);
data_index += len;
}
}
pclose(fdp);
}
return data;
}
int main(int argc, char **argv) {
int listenfd;
listenfd = socket(AF_INET, SOCK_STREAM, 0);
struct sockaddr_in server_addr;
bzero(&server_addr, sizeof(server_addr));
server_addr.sin_family = AF_INET;
server_addr.sin_addr.s_addr = htonl(INADDR_ANY);
server_addr.sin_port = htons(SERV_PORT);
int on = 1;
setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on));
int rt1 = bind(listenfd, (struct sockaddr *) &server_addr, sizeof(server_addr));
if (rt1 < 0) {
error(1, errno, "bind failed ");
}
int rt2 = listen(listenfd, LISTENQ);
if (rt2 < 0) {
error(1, errno, "listen failed ");
}
signal(SIGPIPE, SIG_IGN);
int connfd;
struct sockaddr_in client_addr;
socklen_t client_len = sizeof(client_addr);
char buf[256];
count = 0;
while (1) {
if ((connfd = accept(listenfd, (struct sockaddr *) &client_addr, &client_len)) < 0) {
error(1, errno, "bind failed ");
}
while (1) {
bzero(buf, sizeof(buf));
int n = read(connfd, buf, sizeof(buf));
if (n < 0) {
error(1, errno, "error read message");
} else if (n == 0) {
printf("client closed \n");
close(connfd);
break;
}
count++;
buf[n] = 0;
if (strncmp(buf, "ls", n) == 0) {
char *result = run_cmd("ls");
if (send(connfd, result, strlen(result), 0) < 0)
return 1;
} else if (strncmp(buf, "pwd", n) == 0) {
char buf[256];
char *result = getcwd(buf, 256);
if (send(connfd, result, strlen(result), 0) < 0){
return 1;
}
free(result);
} else if (strncmp(buf, "cd ", 3) == 0) {
char target[256];
bzero(target, sizeof(target));
memcpy(target, buf + 3, strlen(buf) - 3);
if (chdir(target) == -1) {
printf("change dir failed, %s\n", target);
}
} else {
char *error = "error: unknown input type";
if (send(connfd, error, strlen(error), 0) < 0)
return 1;
}
}
}
exit(0);
}
服务器端程序需要两层循环,第一层循环控制多个客户端连接,当然咱们这里没有考虑使用并发,这在第三个模块中会讲到。严格来说,现在的服务器端程序每次只能服务一个客户连接。
第二层循环控制和单个连接的数据交互,因为我们不止完成一次命令交互的过程,所以这一层循环也是必须的。
大部分同学都完成了这个两层循环的设计,我觉得非常棒。
在第一层循环里通过accept完成了连接的建立,获得连接套接字。
在第二层循环里,先通过调用read函数从套接字获取字节流。我这里处理的方式是反复使用了buf缓冲,每次使用之前记得都要调用bzero完成初始化,以便重复利用。
如果读取数据为0,则说明客户端尝试关闭连接,这种情况下,需要跳出第二层循环,进入accept阻塞调用,等待新的客户连接到来。我看到有同学使用了goto来完成跳转,其实使用break跳出就可以了,也有同学忘记跳转了,这里需要再仔细看一下。
在读出客户端的命令之后,就进入处理环节。通过字符串比较命令,进入不同的处理分支。C语言的strcmp或者strncmp可以帮助我们进行字符串比较,这个比较类似于Java语言的String equalsIgnoreCase方法。当然,如果命令的格式有错,需要我们把错误信息通过套接字传给客户端。
对于“pwd”命令,我是通过调用getcwd来完成的,getcwd是一个C语言的API,可以获得当前的路径。
对于“cd”命令,我是通过调用chdir来完成的,cd是一个C语言的API,可以将当前目录切换到指定的路径。有的同学在这里还判断支持了“cd ~”,回到了当前用户的HOME路径,这个非常棒,我就没有考虑这种情况了。
对于“ls”命令,我看到有同学是调用了scandir方法,获得当前路径下的所有文件列表,再根据每个文件类型,进行了格式化的输出。这个方法非常棒,是一个标准实现。我这里呢,为了显得稍微不一样,通过了popen的方法,执行了ls的bash命令,把bash命令的结果通过文件字节流的方式读出,再将该字节流通过套接字传给客户端。我看到有的同学在自己的程序里也是这么做的。
这次的期中大作业,主要考察了客户端-服务器编程的基础知识。
客户端程序考察使用select多路复用,一方面从标准输入接收字节流,另一方面通过套接字读写,以及使用shutdown关闭半连接的能力。
服务器端程序则考察套接字读写的能力,以及对端连接关闭情况下的异常处理等能力。
不过,服务器端程序目前只能一次服务一个客户端连接,不具备并发服务的能力。如何编写一个具备高并发服务能力的服务器端程序,将是我们接下来课程的重点。我们将会重点讲述基于I/O多路复用的事件驱动模型,并以此为基础设计一个高并发网络编程框架,通过这个框架,实现一个HTTP服务器。挑战和难度越来越高,你准备好了吗?