static void new_connection(void *ud, zn_State *S) {
zn_BufferPoolNode *data = zn_getbuffer(&pool);
zn_recvonheader(&data->recv, on_header, data);
zn_recvonpacket(&data->recv, on_client_packet, data);
data->tcp = zn_newtcp(S);
if (zn_connect(data->tcp, addr, port, on_connect, data) != ZN_OK)
zn_post(S, new_connection, NULL);
}
static BOOL WINAPI on_interrupted(DWORD dwCtrlEvent) {
if (!deinited) {
deinited = 1;
/* windows ctrl handler is running at another thread */
zn_post(S, (zn_PostHandler*)cleanup, NULL);
}
return TRUE;
}
int main(int argc, const char **argv) {
int i, client_count = 1;
if (argc == 2 && strcmp(argv[1], "-h") == 0) {
printf("usage: %s [(client/server) [ip [port]]]\n", argv[0]);
exit(0);
}
if (argc > 1) {
size_t n = strlen(argv[1]);
if (n >= 6 && memcmp(argv[1], "client", 6) == 0) {
is_client = 1;
if (n > 6) client_count = atoi(argv[1] + 6);
}
}
if (argc > 2) {
strncpy(addr, argv[2], ZN_MAX_ADDRLEN-1);
}
if (argc > 3) {
unsigned p = atoi(argv[3]);
if (p != 0) port = p;
}
zn_initialize();
printf("znet engine: %s\n", zn_engine());
if ((S = zn_newstate()) == NULL) return 2;
zn_initbuffpool(&pool);
{
unsigned short dlen = htons(BLOCK_SIZE-2);
memcpy(send_data, &dlen, 2);
}
if (is_client) {
printf("client count: %d\n", client_count);
printf("connecting to %s:%d ...\n", addr, port);
for (i = 0; i < client_count; ++i)
zn_post(S, new_connection, NULL);
}
else {
zn_Accept *accept;
if ((accept = zn_newaccept(S)) == NULL) return 2;
zn_listen(accept, addr, port);
zn_accept(accept, on_accept, NULL);
printf("listening at %s:%d ...\n", addr, port);
}
zn_starttimer(zn_newtimer(S, on_summary, NULL), 1000);
register_interrupted();
return zn_run(S, ZN_RUN_LOOP);
}
static void client_error(zn_Tcp *tcp) {
zn_deltcp(tcp);
zn_post(S, on_error, NULL);
}
static void on_error(void *ud, zn_State *S) {
zn_Tcp *tcp = zn_newtcp(S);
if (zn_connect(tcp, addr, port, on_connect, &client) != ZN_OK)
zn_post(S, on_error, NULL);
}
inline void EventLoop::post(OnPostHandler&& h)
{
auto newh = new OnPostHandler(h);
zn_post(S, post_cb, newh);
}
static void client_error(zn_Tcp *tcp, zn_BufferPoolNode *data) {
zn_putbuffer(&pool, data);
zn_deltcp(tcp);
zn_post(S, new_connection, NULL);
}
