[Linux]Centos 리눅스에서 epoll 로 에코서버를 돌려보자. [2/2]

컴퓨터 팁/OS 2009. 5. 16. 13:17 posted by 야매코더

정말 간만이다..
지난 한달간 힘든일은 한번에 온다는 이야기를 몸소 체험했다.
이 한마디는 꼭 하고 싶다..
대한민국의 여성들은 다이어트 중지하고 운동해서 튼튼한 어머니가 되어 주시요.
나도 이쁘면 모두 되는줄 알았는데.. 날씬하기만 하고 약한 여성은 어머니로서는 힘들다는것을 반드시 기억하시고 운동하시요. plz~~

자..이번엔 저번에 하다만 이풀부터 해보자..
요즘 epoll에 대한 관심이 많아졌는지..검색하면 많이 나온다..
초간단 에코서버를 해보려한다..

1. 소스를 작성한다..

소스를 입력하는 방법은 여러가지이지만 가장 전통적(?)인 vi를 이용해서 넣는 화면이다. 물론 xwindows 라서 색상도 이쁘게 잘 나온다. 이외에 다양한 편집기가 있다고 하는데 우선 vi로 하고 담에 다른 방법으로 해보겠다.

소스는...

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <sys/epoll.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <netinet/tcp.h>
#include <fcntl.h>
#include <sys/ioctl.h>
#include <errno.h>

#define EPOLL_SIZE 1000
#define EPOLL_EVENT_SIZE 100
#define OPENPORT 7011

int main(int argc, char **argv) {
struct epoll_event *events;
struct epoll_event ev;
struct sockaddr_in addr, clientaddr;
int clilen;
int sfd, efd, cfd;
int i, j;
int max_got_events;
int result;
int readn;
int sendflags = 0;
char buf_in[256] = { '\0' };
int count = 0;
// epoll 소켓 생성
if ((efd = epoll_create(EPOLL_EVENT_SIZE)) < 0) {
  perror("epoll_create (1) error");
  return -1;
}
// init pool
events = (struct epoll_event *) malloc(sizeof(*events) * EPOLL_SIZE);
if (NULL == events) {
  perror("epoll_create error");
  return -1;
}
// 서버 소켓 생성
clilen = sizeof(clientaddr);
sfd = socket(AF_INET, SOCK_STREAM, 0);
if (sfd == -1) {
  perror("socket error :");
  close(sfd);
  return -1;
}
// 소켓 정보 넣고 bind, listen 처리
addr.sin_family = AF_INET;
addr.sin_port = htons(OPENPORT);
addr.sin_addr.s_addr = htonl(INADDR_ANY);
if (bind(sfd, (struct sockaddr *) &addr, sizeof(addr)) == -1) {
  close(sfd);
  return -2;
}
listen(sfd, 5);
if (sfd < 0) {
  perror("init_sock error");
  return 1;
}
printf("Running Epoll Server port %d\n", OPENPORT);
ev.events = EPOLLIN;
ev.data.fd = sfd;
result = epoll_ctl(efd, EPOLL_CTL_ADD, sfd, &ev);
if (result < 0) {
  perror("epoll error");
  return 1;
}
// 서버 실제 루트
while (1) {
  max_got_events = epoll_wait(efd, events, EPOLL_SIZE, -1);
  for (i = 0; i < max_got_events; i++) {
   if (events[i].data.fd == sfd) {
    printf("접속 : %s", buf_in);
    cfd = accept(sfd, (struct sockaddr *) &clientaddr, &clilen);
    if (cfd < 0) {
     perror("Accept error");
     return -1;
    }
    ev.events = EPOLLIN;
    ev.data.fd = cfd;
    epoll_ctl(efd, EPOLL_CTL_ADD, cfd, &ev);
    printf("Accepted socket : %d\n", cfd);
    count++;
   } else {
    cfd = events[i].data.fd;
    memset(buf_in, 0x00, 256);
    readn = read(cfd, buf_in, 255);
    if (readn <= 0) {
     epoll_ctl(efd, EPOLL_CTL_DEL, cfd, &ev);
     close(cfd);
     printf("Close fd %d\n", cfd);
     count--;
    } else {
     printf("%s", buf_in);
     for (j = 5; j < count + 5; j++) {
      if (cfd != j) {
       send(j, buf_in, strlen(buf_in), sendflags);
      }
     }
    }
   }
  }
}
return 1;
}