time_wait 상태 재현하기

jinwook han·2022년 6월 5일

서버 클라이언트 모델에서, TIME_WAIT TCP 상태를 재현한다.
서버는 daytime 서버를 활용한다.
코드는 모두를 위한 리눅스 책 코드를 사용했다.
아래 repository의 daytime.c, daytimed.c 파일에서 코드를 볼 수 있다.

https://github.com/Jpub/Linux_for_Everyone
(위 코드에 약간의 변형을 함)

클라이언트

daytime.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netdb.h>

static int open_connection(char *host, char *service);

int main(int argc, char *argv[]) {
    int sock;
    FILE *f;
    char buf[1024];

    sock = open_connection((argc > 1 ? argv[1] : "localhost"), "daytime"); /* 1 */
    f = fdopen(sock, "r");
    if (!f) {
        perror("fdopen(3)");
	exit(1);
    }
    fgets(buf, sizeof buf, f); /* 5 */
    fclose(f); /* 6 */
    fputs(buf, stdout); /* 7 */
    exit(0);

}

static int open_connection(char *host, char *service) {
    int sock;
    struct addrinfo hints, *res, *ai;
    int err;

    memset(&hints, 0, sizeof(struct addrinfo));
    hints.ai_family = AF_UNSPEC;
    hints.ai_socktype = SOCK_STREAM;
    if ((err = getaddrinfo(host, service, &hints, &res)) != 0) { /* 2 */
        fprintf(stderr, "getaddrinfo(3): %s\n", gai_strerror(err));
	exit(1);
    }
    
    for (ai = res; ai; ai = ai -> ai_next) {
        sock = socket(ai->ai_family, ai->ai_socktype, ai->ai_protocol); /* 3 */
	if (sock < 0) {
	    continue;
	}
	if (connect(sock, ai -> ai_addr, ai -> ai_addrlen) < 0) { /* 4 */
	    close(sock);
	    continue;
	}
	freeaddrinfo(res);
	return sock;
    }
    fprintf(stderr, "socket(2)/connect(2) failed");
    freeaddrinfo(res);
    exit(1);
}

인자에 따라 localhost 또는 외부 ip를 넘긴다. 서비스명은 daytime으로 설정해서, port 13번에 대해서 연결할 수 있도록 한다.
host name으로 ip를 조회한다.
2에서 얻은 정보로 socket 구조체를 생성한다.
서버와 연결한다.
서버에서 받은 입력(time 정보)를 버퍼로 옮긴다.
서버와 커넥션을 끊는다.
버퍼의 내용을 출력한다.

서버

daytimed.c

#if defined(__digital__) && defined(__unix__)
# ifdef _XOPEN_SOURCE
#  undef _XOPEN_SOURCE
# endif
# define _XOPEN_SOURCE 500
# define _OSF_SOURCE
#endif
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netdb.h>
#include <signal.h>

#define DEFAULT_PORT 13

static int listen_socket(int port);

int main(int argc, char *argv[]) {
    struct sockaddr_storage addr;
    socklen_t addrlen = sizeof addr;
    int sock, server;
    time_t t;
    struct tm *tm;
    char *timestr;

    server = listen_socket(argc > 1? atoi(argv[1]) : DEFAULT_PORT); /* 1 */

    for(;;) {
        sock = accept(server, (struct sockaddr*)&addr, &addrlen); /* 6 */
    
        if (sock < 0) {
            perror("accept(2)");
	    exit(1);
        }

        time(&t);
        tm = localtime(&t);
        timestr = asctime(tm);
        write(sock, timestr, strlen(timestr)); /* 7 */
        close(sock); /* 8 */
    }
    close(server);
    exit(0);
}

static int listen_socket(int port) {
    struct addrinfo hints, *res, *ai;
    int err;
    char service[16];
    
    memset(&hints, 0, sizeof(struct addrinfo));
    hints.ai_family = AF_UNSPEC;
    hints.ai_socktype = SOCK_STREAM;
    hints.ai_flags = AI_PASSIVE;
    snprintf(service, sizeof service, "%d", port);

    if ((err = getaddrinfo(NULL, service, &hints, &res))!= 0) { /* 2 */
	fprintf(stderr, "%s\n", gai_strerror(err));
	exit(1);
    }

    for (ai = res; ai; ai = ai->ai_next) {
	int sock;

	sock = socket(ai->ai_family, ai->ai_socktype, ai->ai_protocol); /* 3 */
	if (sock < 0) continue;
	if (bind(sock, ai->ai_addr, ai->ai_addrlen) < 0) { /* 4 */
	    close(sock);
	    continue;
	}
	if (listen(sock, 5) < 0) { /* 5 */
	    close(sock);
	    continue;
	}
	freeaddrinfo(res);
	fprintf(stderr, "listening on port %d...\n", port);
	return sock;
    }
    fprintf(stderr, "cannot listen socket\n");
    exit(1);
}