实验楼楼赛第4期C语言项目挑战题 -远程 Shell

题目链接:click me
这道题实际上考察的是 Linux 上的 C 语言网络编程。并且只给出了客户端的源代码和服务端的二进制文件。需要根据客户端代码编译和运行的状态来找 bug，必要的时候需要进行调试跟踪。

在客户端代码文件中包含了多个 socket 编程中容易犯的错误：

错误1：TCP 连接

服务器端使用 TCP 连接，代码中创建 socket 的时候用的是 UDP 连接方式，需要使用：

int sock = socket(AF_INET, SOCK_STREAM, 0);

错误2：端口号使用网络字节序

字节序使用错误也是常见的 socket 错误，需要使用 htons 函数将主机字节序的端口号改为网络字节序：

connection.sin_port = htons(PORT);

错误3：强制类型转化

connect 函数中第二个参数需要使用 struct sockaddr 类型指针，但 connection 的类型不匹配，这个错误在编译的提示中可以很容易看出来。修改的代码为：

if (connect(sock, (struct sockaddr*)&connection, sizeof(connection)) != 0)

错误4：if 中的 == 误用

pid 变量没有初始化直接在 if 中用作 == 比较，此处应该是 =，pid 为 fork 创建后返回的进程 ID：

if(pid = fork())

最终修改后的代码：

#include "share.h"

void send_cmd(int sock, int pid) {
	char str[MAX_MSG_LENGTH] = {0};
	printf("> ");
	while (fgets(str, MAX_MSG_LENGTH, stdin) == str) {
		if(strncmp(str, END_STRING, strlen(END_STRING)) == 0) break;
		if(send(sock, str, strlen(str)+1, 0) < 0) perro("send");
	}
	kill(pid, SIGKILL);
	printf("Goodbye.\n");
}

void receive(int sock) {
	char buf[MAX_MSG_LENGTH] = {0};
	int filled = 0;	
	
    while(filled = recv(sock, buf, MAX_MSG_LENGTH-1, 0)){
		buf[filled] = '\0';
		printf("%s", buf);
		fflush(stdout);		
	}	
	printf("Server disconnected.\n");
}

int main(int argc, char **argv) {
	if(argc != 2) perro("args");

	int sock = socket(AF_INET, SOCK_STREAM, 0);
	if(sock == -1) perro("socket");

	struct in_addr server_addr;
	if(!inet_aton(argv[1], &server_addr)) perro("inet_aton");

	struct sockaddr_in connection;
	connection.sin_family = AF_INET;
	memcpy(&connection.sin_addr, &server_addr, sizeof(server_addr));
	connection.sin_port = htons(PORT);
	if (connect(sock, (struct sockaddr*)&connection, sizeof(connection)) != 0) perro("connect");
	
	int pid;	
	if(pid = fork()) send_cmd(sock, pid);
	else receive(sock);
	
	return 0;
}