第五章-基于TCP的服务器端/客户端（2）

5.1 回声客户端的完美实现

问题：回声客户端存在一个问题，由于基于TCP的数据传输，数据是不存在边界的，客户端通过多次调用write传送数据，可能是一次性传递给服务器端，但是服务器端通过调用一次write函数传输数据时，如果数据过大，操作系统可能把数据分成多个数据包发送给客户端。，那么在此过程中，客户端可能尚未收到全部数据包就调用read函数。这些问题都源自TCP的数据传输特性。

//回声客户端echo_client.c
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<unistd.h>
#include<arpa/inet.h>
#include<sys/socket.h>
void error_handling(char* message);
#define BUF_SIZE 1024
int main(int argc, char* argv[]){
	int sock;
	struct sockaddr_in serv_addr;
	char message[BUF_SIZE];
	int str_len=0, recv_len=0,recv_cnt;
	int idx=0,read_len=0;

	if(argc!=3){
		printf("Usage : %s <IP> <port>\n", argv[0]);
		exit(1);
	}

	sock=socket(PF_INET, SOCK_STREAM,0);
	if(sock==-1)
		error_handling("socket() error");

	memset(&serv_addr, 0, sizeof(serv_addr));
	serv_addr.sin_family=AF_INET;
	serv_addr.sin_addr.s_addr=inet_addr(argv[1]);
	serv_addr.sin_port=htons(atoi(argv[2]));
	
	if(connect(sock, (struct sockaddr*)&serv_addr,sizeof(serv_addr))==-1)
		error_handling("connect() error!");
	else
		puts("Connected....");
	while(1){
		fputs("Input message(Q to quit): ",stdout);
		fgets(message, BUF_SIZE, stdin);
		if(!strcmp(message,"q\n") || !strcmp(message, "Q\n"))
			break;
		str_len = write(sock, message, strlen(message));
		//回声问题解决方案，就是提前确定接受数据大小
		while(recv_len < str_len) {
			recv_cnt= read(sock, message, BUF_SIZE-1);
			if(recv_cnt == -1)
				error_handling("read() error!");
			recv_len += recv_cnt;
		}
		message[str_len] = 0;
		printf("Message from server: %s", message);
	}
	close(sock);
	return 0;

}
void error_handling(char* message){
	fputs(message,stderr);
	fputc('\n',stderr);
	exit(1);
}

如果问题不在于回声客户端：定义应用程序协议（规则）

之前的回声客户端可以提前知道接受数据的长度，但若无法预知接受数据长度，就需要定义应用层协议。之前回声服务器端/客户端就定义了如下协议：“收到Q就立即终止连接”。

计算器服务端/客户端示例

// op_client.c
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<unistd.h>
#include<arpa/inet.h>
#include<sys/socket.h>
#define BUF_SIZE 1024
#define RLT_SIZE 4
#define OPSZ 4
void error_handling(char* message);

int main(int argc, char* argv[]) {
    int sock;
    char opmsg[BUF_SIZE];
    int result, opnd_cnt, i;
    struct sockaddr_in serv_adr;
    if (argc != 3) {
        printf("Usage : %s <IP> <port>\n", argv[0]);
        exit(1);
    }

    sock = socket(PF_INET, SOCK_STREAM, 0);

    if (sock == -1) {
        error_handling("socket() error");
    }

    memset(&serv_adr, 0, sizeof(serv_adr));
    serv_adr.sin_family = AF_INET;
    serv_adr.sin_addr.s_addr = inet_addr(argv[1]);
    serv_adr.sin_port = htons(atoi(argv[2]));

    if (connect(sock, (struct sockaddr*)&serv_adr, sizeof(serv_adr)) == -1) {
        error_handling("connect() error!");
    } else {
        puts("Connected.....");
    }

    fputs("Operand count: ", stdout);

    scanf("%d", &opnd_cnt);
    opmsg[0] = (char)opnd_cnt;

    for (int i = 0; i < opnd_cnt; i++) {
        printf("Opnd_cnt %d: ", i + 1);
        scanf("%d", (int*)&opmsg[i * OPSZ + 1]);
    }

    fgetc(stdin);
    fputs("Operator: ", stdout);
    scanf("%c", &opmsg[opnd_cnt*OPSZ + 1]);
    write(sock, opmsg, opnd_cnt*OPSZ + 2);
    read(sock, &result, RLT_SIZE);

    printf("Operation result: %d \n", result);
    close(sock);
    return 0;
}

void error_handling(char* messgae) {
	fputs(message, stderr);
	fputc('\n', stderr);
	exit(1);
}
//
// op_server.c
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<unistd.h>
#include<arpa/inet.h>
#include<sys/socket.h>
#define BUF_SIZE 1024
#define OPSZ 4
void error_handling(cahr* messgae);

int calculate(int opnum, int opnds[], char operator);

int main(int argc, char* argv[]) {
    int serv_sock, clnt_sock;
    char opinfo[BUF_SIZE];
    int result, opnd_cnt, i;
    int recv_cnt, recv_len;
    struct sockaddr_in serv_adr, clnt_adr;
    socklen_t clnt_adr_sz;
    if (argc != 2) {
        printf("Usage : %s <port>\n", argv[0]);
        exit(1);
    }

    serv_sock = socket(PF_INET, SOCK_STREAM, 0);
    if (serv_sock == -1) {
        error_handling("socket() error");
    }

    memset(&serv_adr, 0, sizeof(serv_adr));
    serv_adr.sin_family = AF_INET;
    serv_adr.sin_addr.s_addr = htonl(INADDR_ANY);
    serv_adr.sin_port = htons(atoi(argv[1]));

    if (bind(serv_sock, (struct sockaddr*)&serv_adr, sizeof(serv_adr)) == -1) {
        error_handling("bind() error");
    }
    if (listen(serv_sock, 5) == -1) {
        error_handling("listen() error");
    }
    clnt_adr_sz = sizeof(clnt_adr);
    for (int i = 0; i < 5; i++) {
        opnd_cnt = 0;
        clnt_sock = accept(serv_sock, (struct sockaddr*)&clnt_adr, &clnt_adr_sz);
        read(clnt_sock, &opnd_cnt, 1);

        recv_len = 0;
        while ((opnd_cnt * OPSZ + 1) > recv_len) {
            recv_cnt = read(clnt_sock, &opinfo[recv_len], BUF_SIZE - 1);
            recv_len += recv_cnt;
        }
        result = calculate(opnd_cnt, (int*)opinfo, opinfo[recv_len - 1]);
        write(clnt_sock, (char*)&result, sizeof(result));
        close(clnt_sock);
    }
    close(serv_sock);
    return 0;
}

int calculate(int opnum, int opnds[], char op) {
    int result = opnds[0], i;
    switch(op) {
        case '+':
            for (int i = 1; i < opnum; i++) result += opnds[i];
            break;
        case '-':
            for (int i = 1; i < opnum; i++) result -= opnds[i];
            break;
        case '*':
            for (int i = 1; i < opnum; i++) result *= opnds[i];
            break;
    }
    return result;
}

void error_handling(char* message) {
    fputs(messgae, stderr);
    fputc('\n', stderr);
    exit(1);
}

5.2 TCP原理

TCP套接字中的 I/O缓冲

TCP套接字的数据收发没有边界。write函数调用后并非立即传输数据，而是现将数据移至输出缓冲。read函数调用瞬间，是从输入缓冲读取数据，数据事先先到达输入缓冲。这就是I/O缓冲。

这些I/O缓冲有如下特性：

1. I/O缓冲在每个套接字中单独存在
2. I/O缓冲在创建套接字时自动生成
3. 即使关闭套接字也会继续传输输出缓冲区中遗留的数据
4. 关闭套接字将丢失输入缓冲中的数据

为了解决客户端缓冲和服务端缓冲大小不一致的问题，也就是服务端传过来的数据大于客户端缓冲的大小。这时可以使用TCP中的滑动窗口协议，来控制数据流。使其不会发生超过输入缓冲大小的数据传输。