S/C based on TCP (2)

에코 클라이언트 문제 해결

에코 클라이언트 해결책

에코 클라이언트는 자신이 전송한 데이터의 크기를 알고있기 때문에, 이를 활용하여 정확히 전송한 바이트 크기만큼 데이터를 수신할 수 있어야 한다.
!=를 쓰지않고 <를 쓰는 이유는 예측하지 못한 상황의 발생으로 인한 무한루프의 형성을 막기 위해서이다.
echo_client2.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
void error_handling(char *message);

int main(int argc, char *argv[]){
	int sock;
    char message[BUF_SIZE];
    int str_len, recv_len, recv_cnt;
    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....");
        
    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));
        
        recv_len = 0;
        while(recv_len < str_len){
        	recv_cnt = read(sock, &message[recv_len], BUF_SIZE -1);
            if(recv_cnt == -1)
            	error_handling("read() error");
            recv_len += recv_cnt;
        }
        message[recv_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);
}

어플리케이션 프로토콜의 정의

수신할 데이터의 크기를 미리 파악하기 어려운 경우 어플리케이션 프로토콜(일종의 약속)이 필요하다.
데이터의 송수신 과정에서 데이터의 끝을 파악할 수 있는 약속을 별도로 정의해 데이터의 끝을 표현하거나, 송수신될 데이터의 크기를 미리 알려 대비가 가능하도록 해야한다.
그리고 이러한 약속들을 모아 어플리케이션 프로토콜이라고 한다.

계산기 서버, 클라이언트의 예

어플리케이션 프로토콜의 정의

• 클라이언트는 서버에 접속하자마자 피연산자의 개수 정보를 1바이트 정수형태로 전달한다.
• 클라이언트가 서버에 전달하는 정수 하나는 4바이트로 표현한다.
• 정수를 전달한 다음 연산의 종류를 전달한다. 연산정보를 1바이트로 전달한다.
• 문자 '+', '-', '*' 중 하나를 선택해 전달한다.
• 서버는 연산결과를 4바이트 정수의 형태로 클라이언트에게 전달한다.
• 연산결과를 얻은 클라이언트는 서버와의 연결을 종료한다.

op_server.c

#include <stdio.h>
#include <stdlib.h>
...
#define BUF_SIZE 1024
#define OPSZ 4
void error_handling(char *message);
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 sock_addr_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_addr, 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(i = 0; i < 5; i++){
    	opnt_cnt = 0;
        clnt_sock = accept(serv_sock, (struct sockaddr*)&clnt_adr, &clnt_adr_sc);
        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(i = 1; i < opnum; i++) result += opnds[i];
            break;
        case '-' :
        	for(i = 1; i < opnum; i++) result -= opnds[i];
            break;
        case '*' :
        	for(i = 1; i < opnum; i++) result *= opnds[i];
            break;
    }
    return result;
}

void error_handling(char *message)....

op_client.c

#include ...
#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(i = 0; i < opnd_cnt; i++){
    	printf("Operand %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 *message)...

• 데이터의 송수신을 위한 메모리 공간은 배열로 생성해야 누적 송수신하기 좋다.