#include "lib_acl.h" #include #include #include #include #include "fiber/libfiber.h" #define STACK_SIZE 160000 static int __rw_timeout = 0; static int http_client(ACL_VSTREAM *cstream, const char* res, size_t len) { char buf[8192]; int ret; cstream->rw_timeout = __rw_timeout; while (1) { ret = acl_vstream_gets(cstream, buf, sizeof(buf) - 1); if (ret == ACL_VSTREAM_EOF) { printf("gets error: %s\r\n", acl_last_serror()); return -1; } buf[ret] = 0; if (strcmp(buf, "\r\n") == 0 || strcmp(buf, "\n") == 0) break; } if (acl_vstream_writen(cstream, res, len) == ACL_VSTREAM_EOF) { printf("write error\r\n"); return -1; } return 0; } static void echo_client(ACL_FIBER *fiber acl_unused, void *ctx) { ACL_VSTREAM *cstream = (ACL_VSTREAM *) ctx; const char* res = "HTTP/1.1 200 OK\r\n" "Date: Tue, 31 May 2016 14:20:28 GMT\r\n" "Server: acl\r\n" "Content-Type: text/html\r\n" "Content-Length: 12\r\n" "Connection: Keep-Alive\r\n" "\r\n" "hello world!"; size_t len = strlen(res); while (1) { if (http_client(cstream, res, len) < 0) break; } acl_vstream_close(cstream); } static void fiber_accept(ACL_FIBER *fiber acl_unused, void *ctx) { ACL_VSTREAM *sstream = (ACL_VSTREAM *) ctx; int fd; for (;;) { ACL_VSTREAM *cstream = acl_vstream_accept(sstream, NULL, 0); if (cstream == NULL) { printf("acl_vstream_accept error %s\r\n", acl_last_serror()); break; } fd = ACL_VSTREAM_SOCK(cstream); acl_fiber_create(echo_client, cstream, STACK_SIZE); printf("accept one over: %d\r\n", fd); } acl_vstream_close(sstream); } static void usage(const char *procname) { printf("usage: %s -h [help] -s listen_addr -r rw_timeout\r\n", procname); } static void fiber_dummy(ACL_FIBER *fiber, void *ctx acl_unused) { printf(">>>curr fiber: %d\r\n", acl_fiber_id(fiber)); } int main(int argc, char *argv[]) { char addr[64]; ACL_VSTREAM *sstream; int ch; snprintf(addr, sizeof(addr), "%s", "127.0.0.1:9001"); while ((ch = getopt(argc, argv, "hs:r:")) > 0) { switch (ch) { case 'h': usage(argv[0]); return 0; case 's': snprintf(addr, sizeof(addr), "%s", optarg); break; case 'r': __rw_timeout = atoi(optarg); break; default: break; } } sstream = acl_vstream_listen(addr, 128); if (sstream == NULL) { printf("acl_vstream_listen error %s\r\n", acl_last_serror()); return 1; } acl_fiber_create(fiber_dummy, NULL, 1024000); acl_fiber_create(fiber_dummy, NULL, 1024000); printf("listen %s ok\r\n", addr); printf("%s: call fiber_creater\r\n", __FUNCTION__); acl_fiber_create(fiber_accept, sstream, STACK_SIZE); printf("call fiber_schedule\r\n"); acl_fiber_schedule(); return 0; }