#ifndef SERVER_H
#define SERVER_H
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <assert.h>
#include <sys/epoll.h>
#include <signal.h>
#include <fcntl.h>
#include "bussiness.h"
#define LISTENTQ 5
#define DEF_PORT 6000
#define MAX_EPOLL_SIZE 10
#define DEF_PROCESS_NUM 5
int create_tcpsvr(char *ip, int port);
void fill_sockaddr(struct sockaddr_in *addr,char *ip, int port);
void comm_to_client(int sockfd);
void epoll_business(int epollfd, int listenfd, struct epoll_event ev);
int init_epoll(int listenfd);
void create_process(int sockfd, int i);
#endif

#include "server.h"
/*
 * Create a TCP Server
 * Return socketfd
 */
int create_tcpsvr(char *ip,int port)
{
    int sockfd;
    struct sockaddr_in addr;

    if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == EOF){
        perror("create_tcpsvr(),socket()");
        exit(EXIT_FAILURE);
    }

    fill_sockaddr(&addr,ip,port);

    if (bind(sockfd, (struct sockaddr *)&addr, sizeof(addr)) == EOF){
        perror("create_tcpsvr(),bind()");
        exit(EXIT_FAILURE);
    }

    if (listen(sockfd,LISTENTQ) == EOF) {
        perror("create_tcpsvr(),bind()");
        exit(EXIT_FAILURE);
    }
    return sockfd;
}

/**
 * Set TCP server's address
 */
void fill_sockaddr(struct sockaddr_in *addr, char *ip, int port)
{
    assert(addr);

    addr -> sin_family = AF_INET;
    addr -> sin_port = htons(port?port:DEF_PORT);
    if (ip == NULL) {
        addr -> sin_addr.s_addr = htonl(INADDR_ANY);
    } else if((addr -> sin_addr.s_addr = inet_addr(ip)) == EOF) {
       perror("fill_sockaddr(),inet_addr()");
       exit(EXIT_FAILURE);
    }
}
/*
 * Communicate to client
 */
void comm_to_clt(int listenfd)
{   
    printf("TCP SERVER IS WAITING!\n");

    struct epoll_event events[MAX_EPOLL_SIZE];
    int fd_num;

    int epollfd = init_epoll(listenfd);

    while (1) {

        fd_num = epoll_wait(epollfd,events,MAX_EPOLL_SIZE,-1);
        if (fd_num == EOF) {
            perror("comm_to_clt(),epoll_wait()");
            continue;
        }
        while (--fd_num >= 0) {
            epoll_business(epollfd, listenfd, events[fd_num]);
        }
    }   
}
/*
 * Initlization epoll
 */
int init_epoll(int listenfd)
{
    struct epoll_event ev;
    int epollfd;

    if((epollfd = epoll_create(MAX_EPOLL_SIZE)) == EOF) {
        perror("init_epoll(),epoll_create()");
        exit(EXIT_FAILURE);
    }

    ev.events = EPOLLIN;
    ev.data.fd = listenfd;
    if (epoll_ctl(epollfd, EPOLL_CTL_ADD,listenfd, &ev)) {
        perror("init_epoll(),epoll_ctl()");
        exit(EXIT_FAILURE);
    }
    return epollfd;
}

/*
 * Respond for io change
 */
void epoll_business(int epollfd, int listenfd, struct epoll_event ev)
{
    struct epoll_event ev_n;
    if (ev.data.fd == listenfd) {
        int clt_fd = accept(listenfd, 0, 0);
        if (clt_fd == EOF) {
            perror("epoll_business(), accept()");
            return;
        }

        fcntl(clt_fd, F_SETFL, O_NONBLOCK);
        ev_n.events = EPOLLIN|EPOLLET;
        ev_n.data.fd = clt_fd;
        if (epoll_ctl(epollfd,EPOLL_CTL_ADD, clt_fd, &ev_n) == EOF) {
            perror("epoll_business(), epoll_ctl()");
            return;
        }
    }else {
        main_service(ev.data.fd);
        if (epoll_ctl(epollfd,EPOLL_CTL_DEL, ev.data.fd, &ev) == EOF) {
            perror("epoll_business(), epoll_ctl()");
        }
        close(ev.data.fd);
    }
}
/*
 * Create some process
 */

void create_process(int sockfd, int i)
{
    /*
     * Avoid zombie process
     */
    signal(SIGCHLD,SIG_IGN);

    while (i--) {
        if (fork() == 0) {
            comm_to_clt(sockfd);
        }
    }
    while(1){
        sleep(100);
    }
}
int main(int argc, char *argv[])
{
    int sockfd = create_tcpsvr(0, 0);
    create_process(sockfd,DEF_PROCESS_NUM);
    exit(EXIT_SUCCESS);
}

#ifndef BUSSINESS_H
#define BUSSINESS_H
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <assert.h>
#define BUFF_SIZE 4096
void main_service(int sockfd);
int write_sock(int sockfd, char *buff, size_t length);
int read_sock(int sockfd, char *buff, size_t llength);
#endif

#include "bussiness.h"

void main_service(int sockfd)
{
    char buff[BUFF_SIZE];
    size_t buff_len;
    if ((buff_len = read_sock(sockfd, buff, BUFF_SIZE)) == EOF) {
        perror("main_service(),read_sock()");
        return;
    } else {
  //業(yè)務邏輯從這里開始
        printf("%s", buff);
    }

}
/*
 * Recive messages
 * Return the length
 */
int read_sock(int sockfd, char *buff, size_t length)
{
    assert(buff);
    return read(sockfd, buff, length);
}
/*
 * Send messages
 * Return the length
 */
int write_sock(int sockfd, char *buff,size_t length)
{
    assert(buff);
    return write(sockfd, buff, length);
}