xxq250
/
homework-jianmu
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

how to do a pressure-test upon eok: refer to comments in tests/examples/c/epoll.c

This commit is contained in:
freemine 2021-01-26 10:58:10 +08:00
parent 7960dbe108
commit 4a669b9b31
1 changed files with 69 additions and 88 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

157
tests/examples/c/epoll.c
View File

@ -13,6 +13,12 @@
* along with this program. If not, see <http://www.gnu.org/licenses/>. * along with this program. If not, see <http://www.gnu.org/licenses/>.
*/ */
// how to use to do a pressure-test upon eok
// tester: cat /dev/urandom | nc -c <ip> <port>
// testee: ./debug/build/bin/epoll -l <port> > /dev/null
// compare against: nc -l <port> > /dev/null
// monitor and compare : glances
#ifdef __APPLE__ #ifdef __APPLE__
#include "eok.h" #include "eok.h"
#else // __APPLE__ #else // __APPLE__
@ -68,55 +74,62 @@ static int ep_dummy = 0;
static ep_t* ep_create(void); static ep_t* ep_create(void);
static void ep_destroy(ep_t *ep); static void ep_destroy(ep_t *ep);
static void* routine(void* arg); static void* routine(void* arg);
static int open_connect(unsigned short port);
static int open_listen(unsigned short port); static int open_listen(unsigned short port);
typedef struct client_s client_t; typedef struct fde_s fde_t;
struct client_s { struct fde_s {
int skt; int skt;
void (*on_event)(ep_t *ep, struct epoll_event *events, client_t *client); void (*on_event)(ep_t *ep, struct epoll_event *events, fde_t *client);
volatile unsigned int state; // 1: listenning; 2: connected
}; };
static void echo_event(ep_t *ep, struct epoll_event *ev, client_t *client); static void listen_event(ep_t *ep, struct epoll_event *ev, fde_t *client);
static void null_event(ep_t *ep, struct epoll_event *ev, fde_t *client);
#define usage(arg0, fmt, ...) do { \
if (fmt[0]) { \
fprintf(stderr, "" fmt "\n", ##__VA_ARGS__); \
} \
fprintf(stderr, "usage:\n"); \
fprintf(stderr, " %s -l <port> : specify listenning port\n", arg0); \
} while (0)
int main(int argc, char *argv[]) { int main(int argc, char *argv[]) {
char *prg = basename(argv[0]);
if (argc==1) {
usage(prg, "");
return 0;
}
ep_t* ep = ep_create(); ep_t* ep = ep_create();
A(ep, "failed"); A(ep, "failed");
int skt = open_connect(6789); for (int i=1; i<argc; ++i) {
if (skt!=-1) { const char *arg = argv[i];
client_t *client = (client_t*)calloc(1, sizeof(*client)); if (0==strcmp(arg, "-l")) {
if (client) { ++i;
if (i>=argc) {
usage(prg, "expecting <port> after -l, but got nothing");
return 1; // confirmed potential leakage
}
arg = argv[i];
int port = atoi(arg);
int skt = open_listen(port);
if (skt==-1) continue;
fde_t *client = (fde_t*)calloc(1, sizeof(*client));
if (!client) {
E("out of memory");
close(skt);
continue;
}
client->skt = skt; client->skt = skt;
client->on_event = echo_event; client->on_event = listen_event;
client->state = 2;
struct epoll_event ev = {0}; struct epoll_event ev = {0};
ev.events = EPOLLIN | EPOLLERR | EPOLLHUP | EPOLLRDHUP; ev.events = EPOLLIN | EPOLLERR | EPOLLHUP | EPOLLRDHUP;
ev.data.ptr = client; ev.data.ptr = client;
A(0==epoll_ctl(ep->ep, EPOLL_CTL_ADD, skt, &ev), ""); A(0==epoll_ctl(ep->ep, EPOLL_CTL_ADD, skt, &ev), "");
continue;
} }
usage(prg, "unknown argument: [%s]", arg);
return 1;
} }
skt = open_listen(0);
if (skt!=-1) {
client_t *client = (client_t*)calloc(1, sizeof(*client));
if (client) {
client->skt = skt;
client->on_event = echo_event;
client->state = 1;
struct epoll_event ev = {0};
ev.events = EPOLLIN | EPOLLERR | EPOLLHUP | EPOLLRDHUP;
ev.data.ptr = client;
A(0==epoll_ctl(ep->ep, EPOLL_CTL_ADD, skt, &ev), "");
}
}
// char c = '\0';
// while ((c=getchar())!=EOF) {
// switch (c) {
// case 'q': break;
// default: continue;
// }
// }
// getchar();
char *line = NULL; char *line = NULL;
size_t linecap = 0; size_t linecap = 0;
ssize_t linelen; ssize_t linelen;
@ -205,7 +218,7 @@ static void* routine(void* arg) {
continue; continue;
} }
A(ev->data.ptr, "internal logic error"); A(ev->data.ptr, "internal logic error");
client_t *client = (client_t*)ev->data.ptr; fde_t *client = (fde_t*)ev->data.ptr;
client->on_event(ep, ev, client); client->on_event(ep, ev, client);
continue; continue;
} }
@ -223,7 +236,7 @@ static int open_listen(unsigned short port) {
do { do {
struct sockaddr_in si = {0}; struct sockaddr_in si = {0};
si.sin_family = AF_INET; si.sin_family = AF_INET;
si.sin_addr.s_addr = inet_addr("127.0.0.1"); si.sin_addr.s_addr = inet_addr("0.0.0.0");
si.sin_port = htons(port); si.sin_port = htons(port);
r = bind(skt, (struct sockaddr*)&si, sizeof(si)); r = bind(skt, (struct sockaddr*)&si, sizeof(si));
if (r) { if (r) {
@ -249,63 +262,31 @@ static int open_listen(unsigned short port) {
return -1; return -1;
} }
static int open_connect(unsigned short port) { static void listen_event(ep_t *ep, struct epoll_event *ev, fde_t *client) {
int r = 0; A(ev->events & EPOLLIN, "internal logic error");
int skt = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
if (skt==-1) {
E("socket() failed");
return -1;
}
do {
struct sockaddr_in si = {0};
si.sin_family = AF_INET;
si.sin_addr.s_addr = inet_addr("127.0.0.1");
si.sin_port = htons(port);
r = connect(skt, (struct sockaddr*)&si, sizeof(si));
if (r) {
E("connect(%u) failed", port);
break;
}
memset(&si, 0, sizeof(si));
socklen_t len = sizeof(si);
r = getsockname(skt, (struct sockaddr *)&si, &len);
if (r) {
E("getsockname() failed");
}
A(len==sizeof(si), "internal logic error");
D("connected: %d", ntohs(si.sin_port));
return skt;
} while (0);
close(skt);
return -1;
}
static void echo_event(ep_t *ep, struct epoll_event *ev, client_t *client) {
if (ev->events & EPOLLIN) {
if (client->state==1) {
struct sockaddr_in si = {0}; struct sockaddr_in si = {0};
socklen_t silen = sizeof(si); socklen_t silen = sizeof(si);
int skt = accept(client->skt, (struct sockaddr*)&si, &silen); int skt = accept(client->skt, (struct sockaddr*)&si, &silen);
if (skt!=-1) { A(skt!=-1, "internal logic error");
client_t *server = (client_t*)calloc(1, sizeof(*server)); fde_t *server = (fde_t*)calloc(1, sizeof(*server));
if (server) { if (!server) {
close(skt);
return;
}
server->skt = skt; server->skt = skt;
server->on_event = echo_event; server->on_event = null_event;
server->state = 2; struct epoll_event ee = {0};
struct epoll_event ev = {0}; ee.events = EPOLLIN | EPOLLERR | EPOLLHUP | EPOLLRDHUP;
ev.events = EPOLLIN | EPOLLERR | EPOLLHUP | EPOLLRDHUP; ee.data.ptr = server;
ev.data.ptr = server; A(0==epoll_ctl(ep->ep, EPOLL_CTL_ADD, skt, &ee), "");
A(0==epoll_ctl(ep->ep, EPOLL_CTL_ADD, skt, &ev), ""); }
}
} static void null_event(ep_t *ep, struct epoll_event *ev, fde_t *client) {
} if (ev->events & EPOLLIN) {
if (client->state==2) { char buf[8192];
char buf[4]; int n = recv(client->skt, buf, sizeof(buf), 0);
int n = recv(client->skt, buf, sizeof(buf)-1, 0); A(n>=0 && n<=sizeof(buf), "internal logic error:[%d]", n);
A(n>=0 && n<sizeof(buf), "internal logic error:[%d]", n); A(n==fwrite(buf, 1, n, stdout), "internal logic error");
buf[n] = '\0';
fprintf(stderr, "events[%x]:%s\n", ev->events, buf);
}
} }
if (ev->events & (EPOLLERR | EPOLLHUP | EPOLLRDHUP)) { if (ev->events & (EPOLLERR | EPOLLHUP | EPOLLRDHUP)) {
A(0==pthread_mutex_lock(&ep->lock), ""); A(0==pthread_mutex_lock(&ep->lock), "");