add network
This commit is contained in:
King
316
reactor.c
Normal file
316
reactor.c
Normal file
@@ -0,0 +1,316 @@
|
||||
|
||||
|
||||
|
||||
#include <errno.h>
|
||||
#include <stdio.h>
|
||||
#include <sys/socket.h>
|
||||
#include <netinet/in.h>
|
||||
#include <string.h>
|
||||
#include <pthread.h>
|
||||
#include <unistd.h>
|
||||
#include <poll.h>
|
||||
#include <sys/epoll.h>
|
||||
#include <errno.h>
|
||||
#include <sys/time.h>
|
||||
|
||||
|
||||
#include "server.h"
|
||||
|
||||
|
||||
#define CONNECTION_SIZE 1024 // 1024 * 1024
|
||||
|
||||
#define MAX_PORTS 20
|
||||
|
||||
#define TIME_SUB_MS(tv1, tv2) ((tv1.tv_sec - tv2.tv_sec) * 1000 + (tv1.tv_usec - tv2.tv_usec) / 1000)
|
||||
|
||||
|
||||
|
||||
#if ENABLE_KVSTORE
|
||||
|
||||
typedef int (*msg_handler)(char *msg, int length, char *response);
|
||||
|
||||
static msg_handler kvs_handler;
|
||||
|
||||
int kvs_request(struct conn *c) {
|
||||
|
||||
printf("recv %d : %s\n", c->rlength, c->rbuffer);
|
||||
|
||||
c->wlength = kvs_handler(c->rbuffer, c->rlength, c->wbuffer);
|
||||
|
||||
}
|
||||
|
||||
int kvs_response(struct conn *c) {
|
||||
|
||||
|
||||
|
||||
}
|
||||
|
||||
|
||||
#endif
|
||||
|
||||
|
||||
|
||||
int accept_cb(int fd);
|
||||
int recv_cb(int fd);
|
||||
int send_cb(int fd);
|
||||
|
||||
|
||||
|
||||
int epfd = 0;
|
||||
struct timeval begin;
|
||||
|
||||
|
||||
|
||||
struct conn conn_list[CONNECTION_SIZE] = {0};
|
||||
// fd
|
||||
|
||||
|
||||
int set_event(int fd, int event, int flag) {
|
||||
|
||||
if (flag) { // non-zero add
|
||||
|
||||
struct epoll_event ev;
|
||||
ev.events = event;
|
||||
ev.data.fd = fd;
|
||||
epoll_ctl(epfd, EPOLL_CTL_ADD, fd, &ev);
|
||||
|
||||
} else { // zero mod
|
||||
|
||||
struct epoll_event ev;
|
||||
ev.events = event;
|
||||
ev.data.fd = fd;
|
||||
epoll_ctl(epfd, EPOLL_CTL_MOD, fd, &ev);
|
||||
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
|
||||
|
||||
int event_register(int fd, int event) {
|
||||
|
||||
if (fd < 0) return -1;
|
||||
|
||||
conn_list[fd].fd = fd;
|
||||
conn_list[fd].r_action.recv_callback = recv_cb;
|
||||
conn_list[fd].send_callback = send_cb;
|
||||
|
||||
memset(conn_list[fd].rbuffer, 0, BUFFER_LENGTH);
|
||||
conn_list[fd].rlength = 0;
|
||||
|
||||
memset(conn_list[fd].wbuffer, 0, BUFFER_LENGTH);
|
||||
conn_list[fd].wlength = 0;
|
||||
|
||||
set_event(fd, event, 1);
|
||||
}
|
||||
|
||||
|
||||
// listenfd(sockfd) --> EPOLLIN --> accept_cb
|
||||
int accept_cb(int fd) {
|
||||
|
||||
struct sockaddr_in clientaddr;
|
||||
socklen_t len = sizeof(clientaddr);
|
||||
|
||||
int clientfd = accept(fd, (struct sockaddr*)&clientaddr, &len);
|
||||
printf("accept finshed: %d, fd: %d\n", clientfd, fd);
|
||||
if (clientfd < 0) {
|
||||
printf("accept errno: %d --> %s\n", errno, strerror(errno));
|
||||
return -1;
|
||||
}
|
||||
|
||||
event_register(clientfd, EPOLLIN); // | EPOLLET
|
||||
|
||||
if ((clientfd % 1000) == 0) {
|
||||
|
||||
struct timeval current;
|
||||
gettimeofday(¤t, NULL);
|
||||
|
||||
int time_used = TIME_SUB_MS(current, begin);
|
||||
memcpy(&begin, ¤t, sizeof(struct timeval));
|
||||
|
||||
|
||||
printf("accept finshed: %d, time_used: %d\n", clientfd, time_used);
|
||||
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
int recv_cb(int fd) {
|
||||
|
||||
memset(conn_list[fd].rbuffer, 0, BUFFER_LENGTH );
|
||||
int count = recv(fd, conn_list[fd].rbuffer, BUFFER_LENGTH, 0);
|
||||
if (count == 0) { // disconnect
|
||||
printf("client disconnect: %d\n", fd);
|
||||
close(fd);
|
||||
|
||||
epoll_ctl(epfd, EPOLL_CTL_DEL, fd, NULL); // unfinished
|
||||
|
||||
return 0;
|
||||
} else if (count < 0) { //
|
||||
|
||||
printf("count: %d, errno: %d, %s\n", count, errno, strerror(errno));
|
||||
close(fd);
|
||||
epoll_ctl(epfd, EPOLL_CTL_DEL, fd, NULL);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
conn_list[fd].rlength = count;
|
||||
//printf("RECV: %s\n", conn_list[fd].rbuffer);
|
||||
|
||||
#if 0 // echo
|
||||
|
||||
conn_list[fd].wlength = conn_list[fd].rlength;
|
||||
memcpy(conn_list[fd].wbuffer, conn_list[fd].rbuffer, conn_list[fd].wlength);
|
||||
|
||||
printf("[%d]RECV: %s\n", conn_list[fd].rlength, conn_list[fd].rbuffer);
|
||||
|
||||
#elif ENABLE_HTTP
|
||||
|
||||
http_request(&conn_list[fd]);
|
||||
|
||||
#elif ENABLE_WEBSOCKET
|
||||
|
||||
ws_request(&conn_list[fd]);
|
||||
|
||||
#elif ENABLE_KVSTORE
|
||||
|
||||
kvs_request(&conn_list[fd]);
|
||||
|
||||
#endif
|
||||
|
||||
|
||||
set_event(fd, EPOLLOUT, 0);
|
||||
|
||||
return count;
|
||||
}
|
||||
|
||||
|
||||
int send_cb(int fd) {
|
||||
|
||||
#if ENABLE_HTTP
|
||||
|
||||
http_response(&conn_list[fd]);
|
||||
|
||||
#elif ENABLE_WEBSOCKET
|
||||
|
||||
ws_response(&conn_list[fd]);
|
||||
|
||||
#elif ENABLE_KVSTORE
|
||||
|
||||
kvs_response(&conn_list[fd]);
|
||||
|
||||
#endif
|
||||
|
||||
int count = 0;
|
||||
|
||||
#if 0
|
||||
if (conn_list[fd].status == 1) {
|
||||
//printf("SEND: %s\n", conn_list[fd].wbuffer);
|
||||
count = send(fd, conn_list[fd].wbuffer, conn_list[fd].wlength, 0);
|
||||
set_event(fd, EPOLLOUT, 0);
|
||||
} else if (conn_list[fd].status == 2) {
|
||||
set_event(fd, EPOLLOUT, 0);
|
||||
} else if (conn_list[fd].status == 0) {
|
||||
|
||||
if (conn_list[fd].wlength != 0) {
|
||||
count = send(fd, conn_list[fd].wbuffer, conn_list[fd].wlength, 0);
|
||||
}
|
||||
|
||||
set_event(fd, EPOLLIN, 0);
|
||||
}
|
||||
#else
|
||||
|
||||
if (conn_list[fd].wlength != 0) {
|
||||
count = send(fd, conn_list[fd].wbuffer, conn_list[fd].wlength, 0);
|
||||
}
|
||||
|
||||
set_event(fd, EPOLLIN, 0);
|
||||
|
||||
#endif
|
||||
//set_event(fd, EPOLLOUT, 0);
|
||||
|
||||
return count;
|
||||
}
|
||||
|
||||
|
||||
|
||||
int r_init_server(unsigned short port) {
|
||||
|
||||
int sockfd = socket(AF_INET, SOCK_STREAM, 0);
|
||||
|
||||
struct sockaddr_in servaddr;
|
||||
servaddr.sin_family = AF_INET;
|
||||
servaddr.sin_addr.s_addr = htonl(INADDR_ANY); // 0.0.0.0
|
||||
servaddr.sin_port = htons(port); // 0-1023,
|
||||
|
||||
if (-1 == bind(sockfd, (struct sockaddr*)&servaddr, sizeof(struct sockaddr))) {
|
||||
printf("bind failed: %s\n", strerror(errno));
|
||||
}
|
||||
|
||||
listen(sockfd, 10);
|
||||
//printf("listen finshed: %d\n", sockfd); // 3
|
||||
|
||||
return sockfd;
|
||||
|
||||
}
|
||||
|
||||
int reactor_start(unsigned short port, msg_handler handler) {
|
||||
|
||||
//unsigned short port = 2000;
|
||||
kvs_handler = handler;
|
||||
printf("reactor_entry: %d\n", port);
|
||||
|
||||
epfd = epoll_create(1);
|
||||
|
||||
int i = 0;
|
||||
|
||||
for (i = 0;i < MAX_PORTS;i ++) {
|
||||
|
||||
int sockfd = r_init_server(port + i);
|
||||
|
||||
conn_list[sockfd].fd = sockfd;
|
||||
conn_list[sockfd].r_action.recv_callback = accept_cb;
|
||||
|
||||
set_event(sockfd, EPOLLIN, 1);
|
||||
}
|
||||
|
||||
gettimeofday(&begin, NULL);
|
||||
|
||||
while (1) { // mainloop
|
||||
|
||||
struct epoll_event events[1024] = {0};
|
||||
int nready = epoll_wait(epfd, events, 1024, -1);
|
||||
|
||||
int i = 0;
|
||||
for (i = 0;i < nready;i ++) {
|
||||
|
||||
int connfd = events[i].data.fd;
|
||||
|
||||
#if 0
|
||||
if (events[i].events & EPOLLIN) {
|
||||
conn_list[connfd].r_action.recv_callback(connfd);
|
||||
} else if (events[i].events & EPOLLOUT) {
|
||||
conn_list[connfd].send_callback(connfd);
|
||||
}
|
||||
|
||||
#else
|
||||
if (events[i].events & EPOLLIN) {
|
||||
conn_list[connfd].r_action.recv_callback(connfd);
|
||||
}
|
||||
|
||||
if (events[i].events & EPOLLOUT) {
|
||||
conn_list[connfd].send_callback(connfd);
|
||||
}
|
||||
#endif
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user