#include <sys/time.h> 
#include <sys/types.h> 
#include <unistd.h> 
 
#include <sys/socket.h> 
#include <sys/types.h> 
 
#include <netinet/in.h> 
#include <arpa/inet.h> 
 
#include <stdio.h> 
#include <string.h> 
 
#define MAXLINE 1024  
#define PORTNUM 3600 
#define SOCK_SETSIZE 1021 
 
int main(int argc, char **argv) 
{ 
    int listen_fd, client_fd; 
    socklen_t addrlen; 
    int fd_num; 
    int maxfd = 0; 
    int sockfd; 
    int readn; 
    int i= 0; 
    char buf[MAXLINE]; 
    fd_set readfds, allfds; 
 
    struct sockaddr_in server_addr, client_addr; 
 
    if((listen_fd = socket(AF_INET, SOCK_STREAM, 0)) == -1) 
    { 
        perror("socket error"); 
        return 1; 
    }    
    memset((void *)&server_addr, 0x00, sizeof(server_addr)); 
    server_addr.sin_family = AF_INET; 
    server_addr.sin_addr.s_addr = htonl(INADDR_ANY); 
    server_addr.sin_port = htons(PORTNUM); 
     
    if(bind(listen_fd, (struct sockaddr *)&server_addr, sizeof(server_addr)) == -1) 
    { 
        perror("bind error"); 
        return 1; 
    }    
    if(listen(listen_fd, 5) == -1) 
    { 
        perror("listen error"); 
        return 1; 
    }    
     
    FD_ZERO(&readfds); 
    FD_SET(listen_fd, &readfds); 
 
    maxfd = listen_fd; 
    while(1) 
    { 
        allfds = readfds; 
        printf("Select Wait %d\n", maxfd); 
        fd_num = select(maxfd + 1 , &allfds, (fd_set *)0, 
                      (fd_set *)0, NULL); 
 
        if (FD_ISSET(listen_fd, &allfds)) 
        { 
            addrlen = sizeof(client_addr); 
            client_fd = accept(listen_fd, 
                    (struct sockaddr *)&client_addr, &addrlen); 
 
            FD_SET(client_fd,&readfds); 
 
            if (client_fd > maxfd) 
                maxfd = client_fd; 
            printf("Accept OK\n"); 
            continue; 
        } 
 
        for (i = 0; i <= maxfd; i++) 
        { 
            sockfd = i; 
            if (FD_ISSET(sockfd, &allfds)) 
            { 
                if( (readn = read(sockfd, buf, MAXLINE-1)) == 0) 
                { 
                    printf("close\n"); 
                    close(sockfd); 
                    FD_CLR(sockfd, &readfds); 
                } 
                else 
                { 
                    buf[readn] = '\0'; 
                    write(sockfd, buf, strlen(buf)); 
                } 
                if (--fd_num <= 0) 
                    break; 
            } 
        } 
    } 
} 
 

2 모아서 처리하기

select는 소켓 버퍼에 데이터가 남아 있으면 fd_set을 1로 둔다. 그러므로 read 함수로 소켓 버퍼를 비우지 못하면, 여러 번 select를 호출하게 된다. MAXLINE의 크기를 작게 한 다음 테스트 해보기 바란다. select를 여러번 호출하는 것은 그다지 좋은 방법은 아니다. 차라리 버퍼에 있는 내용을 다 가져올 때까지 read를 호출하는게 더 나을 것이다.

while( (readn = read(sockfd, buf, MAXLINE-1)) > 0) 
{ 
    write(sockfd, buf, readn); 
} 
 

그러나 이 방법은 버퍼를 비우고 나서, read에서 봉쇄되어버린다는 문제가 있다. 이 문제를 해결하기 위해서는 소켓을 비봉쇄로 해야 한다.

3 비봉쇄 입출력 적용

모아서 처리하기는 좋은 방법이지만, 제대로 활용하기 위해서는 비봉쇄 소켓으로 만들어야 한다. 다음은 매개 변수로 넘어온 파일을 비봉쇄로 만드는 함수다. accept함수가 가져오는 연결 소켓을 아래 함수를 이용해서 비 봉쇄로 만들면 된다.

void nonblock(int sockfd)  
{  
    int opts;  
    opts = fcntl(sockfd, F_GETFL);  
    if(opts < 0)  
    {  
        perror("fcntl(F_GETFL)\n");  
        exit(1);  
    }  
    opts = (opts | O_NONBLOCK);  
    if(fcntl(sockfd, F_SETFL, opts) < 0)  
    {  
        perror("fcntl(F_SETFL)\n");  
        exit(1);  
    }  
 

비봉쇄 소켓은 읽을 데이터가 없으면 즉시 반환하는데 이 때 값이 -1이다. 그로므로 에러의 종류를 확인하기 위해서 errno값을 검사해야 한다. 더 이상 읽을 데이터가 없어서 -1을 반환할 경우 errno가 EAGAIN로 설정된다. 그러므로 다음과 같이 코드를 변경하면 된다.

while( (readn = read(sockfd, buf, MAXLINE-1)) > 0) 
{ 
   write(...); 
} 
if(readn == -1) 
{ 
    if (errno != EAGAIN) 
    { 
        close(sockfd); 
        FD_CLR(sockfd, &readfds); 
    } 
} 
 

category_network programing

태그 : 입출력다중화 네트워크프로그래밍 시스템프로그래밍 C fork select