| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779 |
- #include "hloop.h"
- #include "hevent.h"
- #include "iowatcher.h"
- #include "hdef.h"
- #include "hbase.h"
- #include "hlog.h"
- #include "hmath.h"
- #include "htime.h"
- #include "hsocket.h"
- #define PAUSE_TIME 10 // ms
- #define MAX_BLOCK_TIME 1000 // ms
- #define IO_ARRAY_INIT_SIZE 1024
- #define CUSTOM_EVENT_QUEUE_INIT_SIZE 16
- static void hio_init(hio_t* io);
- static void hio_ready(hio_t* io);
- static void hio_done(hio_t* io);
- static void hio_free(hio_t* io);
- static void __hidle_del(hidle_t* idle);
- static void __htimer_del(htimer_t* timer);
- static int timers_compare(const struct heap_node* lhs, const struct heap_node* rhs) {
- return TIMER_ENTRY(lhs)->next_timeout < TIMER_ENTRY(rhs)->next_timeout;
- }
- static int hloop_process_idles(hloop_t* loop) {
- int nidles = 0;
- struct list_node* node = loop->idles.next;
- hidle_t* idle = NULL;
- while (node != &loop->idles) {
- idle = IDLE_ENTRY(node);
- node = node->next;
- if (idle->repeat != INFINITE) {
- --idle->repeat;
- }
- if (idle->repeat == 0) {
- __hidle_del(idle);
- }
- EVENT_PENDING(idle);
- ++nidles;
- }
- return nidles;
- }
- static int hloop_process_timers(hloop_t* loop) {
- int ntimers = 0;
- htimer_t* timer = NULL;
- uint64_t now_hrtime = hloop_now_hrtime(loop);
- while (loop->timers.root) {
- timer = TIMER_ENTRY(loop->timers.root);
- if (timer->next_timeout > now_hrtime) {
- break;
- }
- if (timer->repeat != INFINITE) {
- --timer->repeat;
- }
- if (timer->repeat == 0) {
- __htimer_del(timer);
- }
- else {
- heap_dequeue(&loop->timers);
- if (timer->event_type == HEVENT_TYPE_TIMEOUT) {
- while (timer->next_timeout <= now_hrtime) {
- timer->next_timeout += ((htimeout_t*)timer)->timeout*1000;
- }
- }
- else if (timer->event_type == HEVENT_TYPE_PERIOD) {
- hperiod_t* period = (hperiod_t*)timer;
- timer->next_timeout = calc_next_timeout(period->minute, period->hour, period->day,
- period->week, period->month) * 1e6;
- }
- heap_insert(&loop->timers, &timer->node);
- }
- EVENT_PENDING(timer);
- ++ntimers;
- }
- return ntimers;
- }
- static int hloop_process_ios(hloop_t* loop, int timeout) {
- int nevents = iowatcher_poll_events(loop, timeout);
- if (nevents < 0) {
- hloge("poll_events error=%d", -nevents);
- }
- return nevents < 0 ? 0 : nevents;
- }
- static int hloop_process_pendings(hloop_t* loop) {
- if (loop->npendings == 0) return 0;
- hevent_t* cur = NULL;
- hevent_t* next = NULL;
- int ncbs = 0;
- for (int i = HEVENT_PRIORITY_SIZE-1; i >= 0; --i) {
- cur = loop->pendings[i];
- while (cur) {
- next = cur->pending_next;
- if (cur->pending) {
- if (cur->active && cur->cb) {
- cur->cb(cur);
- ++ncbs;
- }
- cur->pending = 0;
- if (cur->destroy) {
- EVENT_DEL(cur);
- }
- }
- cur = next;
- }
- loop->pendings[i] = NULL;
- }
- loop->npendings = 0;
- return ncbs;
- }
- static int hloop_process_events(hloop_t* loop) {
- // ios -> timers -> idles
- int nios, ntimers, nidles;
- nios = ntimers = nidles = 0;
- // calc blocktime
- int32_t blocktime = MAX_BLOCK_TIME;
- hloop_update_time(loop);
- if (loop->timers.root) {
- uint64_t next_min_timeout = TIMER_ENTRY(loop->timers.root)->next_timeout;
- int64_t blocktime_us = next_min_timeout - hloop_now_hrtime(loop);
- if (blocktime_us <= 0) goto process_timers;
- blocktime = blocktime_us / 1000;
- ++blocktime;
- blocktime = MIN(blocktime, MAX_BLOCK_TIME);
- }
- if (loop->nios) {
- nios = hloop_process_ios(loop, blocktime);
- }
- else {
- msleep(blocktime);
- }
- hloop_update_time(loop);
- process_timers:
- if (loop->ntimers) {
- ntimers = hloop_process_timers(loop);
- }
- int npendings = loop->npendings;
- if (npendings == 0) {
- if (loop->nidles) {
- nidles= hloop_process_idles(loop);
- }
- }
- int ncbs = hloop_process_pendings(loop);
- //printd("blocktime=%d nios=%d/%u ntimers=%d/%u nidles=%d/%u nactives=%d npendings=%d ncbs=%d\n",
- //blocktime, nios, loop->nios, ntimers, loop->ntimers, nidles, loop->nidles,
- //loop->nactives, npendings, ncbs);
- return ncbs;
- }
- static void hloop_init(hloop_t* loop) {
- loop->status = HLOOP_STATUS_STOP;
- // idles
- list_init(&loop->idles);
- // timers
- heap_init(&loop->timers, timers_compare);
- // ios: init when hio_add
- //io_array_init(&loop->ios, IO_ARRAY_INIT_SIZE);
- // iowatcher: init when iowatcher_add_event
- //iowatcher_init(loop);
- // custom_events: init when hloop_post_event
- //event_queue_init(&loop->custom_events, 4);
- loop->sockpair[0] = loop->sockpair[1] = -1;
- hmutex_init(&loop->custom_events_mutex);
- // NOTE: init start_time here, because htimer_add use it.
- loop->start_ms = timestamp_ms();
- loop->start_hrtime = loop->cur_hrtime = gethrtime();
- }
- static void hloop_cleanup(hloop_t* loop) {
- // pendings
- printd("cleanup pendings...\n");
- for (int i = 0; i < HEVENT_PRIORITY_SIZE; ++i) {
- loop->pendings[i] = NULL;
- }
- // idles
- printd("cleanup idles...\n");
- struct list_node* node = loop->idles.next;
- hidle_t* idle;
- while (node != &loop->idles) {
- idle = IDLE_ENTRY(node);
- node = node->next;
- SAFE_FREE(idle);
- }
- list_init(&loop->idles);
- // timers
- printd("cleanup timers...\n");
- htimer_t* timer;
- while (loop->timers.root) {
- timer = TIMER_ENTRY(loop->timers.root);
- heap_dequeue(&loop->timers);
- SAFE_FREE(timer);
- }
- heap_init(&loop->timers, NULL);
- // ios
- printd("cleanup ios...\n");
- for (int i = 0; i < loop->ios.maxsize; ++i) {
- hio_t* io = loop->ios.ptr[i];
- if (io) {
- if ((!(io->io_type&HIO_TYPE_STDIO)) && io->active) {
- hio_close(io);
- }
- hio_free(io);
- }
- }
- io_array_cleanup(&loop->ios);
- // readbuf
- if (loop->readbuf.base && loop->readbuf.len) {
- free(loop->readbuf.base);
- loop->readbuf.base = NULL;
- loop->readbuf.len = 0;
- }
- // iowatcher
- iowatcher_cleanup(loop);
- // custom_events
- hmutex_lock(&loop->custom_events_mutex);
- if (loop->sockpair[0] != -1 && loop->sockpair[1] != -1) {
- closesocket(loop->sockpair[0]);
- closesocket(loop->sockpair[1]);
- loop->sockpair[0] = loop->sockpair[1] = -1;
- }
- event_queue_cleanup(&loop->custom_events);
- hmutex_unlock(&loop->custom_events_mutex);
- hmutex_destroy(&loop->custom_events_mutex);
- }
- hloop_t* hloop_new(int flags) {
- hloop_t* loop;
- SAFE_ALLOC_SIZEOF(loop);
- memset(loop, 0, sizeof(hloop_t));
- hloop_init(loop);
- loop->flags |= flags;
- return loop;
- }
- void hloop_free(hloop_t** pp) {
- if (pp && *pp) {
- hloop_cleanup(*pp);
- SAFE_FREE(*pp);
- *pp = NULL;
- }
- }
- int hloop_run(hloop_t* loop) {
- loop->status = HLOOP_STATUS_RUNNING;
- while (loop->status != HLOOP_STATUS_STOP) {
- if (loop->status == HLOOP_STATUS_PAUSE) {
- msleep(PAUSE_TIME);
- hloop_update_time(loop);
- continue;
- }
- ++loop->loop_cnt;
- if (loop->nactives == 0) break;
- hloop_process_events(loop);
- if (loop->flags & HLOOP_FLAG_RUN_ONCE) {
- break;
- }
- }
- loop->status = HLOOP_STATUS_STOP;
- loop->end_hrtime = gethrtime();
- if (loop->flags & HLOOP_FLAG_AUTO_FREE) {
- hloop_cleanup(loop);
- SAFE_FREE(loop);
- }
- return 0;
- }
- int hloop_stop(hloop_t* loop) {
- loop->status = HLOOP_STATUS_STOP;
- return 0;
- }
- int hloop_pause(hloop_t* loop) {
- if (loop->status == HLOOP_STATUS_RUNNING) {
- loop->status = HLOOP_STATUS_PAUSE;
- }
- return 0;
- }
- int hloop_resume(hloop_t* loop) {
- if (loop->status == HLOOP_STATUS_PAUSE) {
- loop->status = HLOOP_STATUS_RUNNING;
- }
- return 0;
- }
- void hloop_update_time(hloop_t* loop) {
- loop->cur_hrtime = gethrtime();
- if (ABS((int64_t)hloop_now(loop) - (int64_t)time(NULL)) > 1) {
- // systemtime changed, we adjust start_ms
- loop->start_ms = timestamp_ms() - (loop->cur_hrtime - loop->start_hrtime) / 1000;
- }
- }
- uint64_t hloop_now(hloop_t* loop) {
- return loop->start_ms / 1000 + (loop->cur_hrtime - loop->start_hrtime) / 1000000;
- }
- uint64_t hloop_now_ms(hloop_t* loop) {
- return loop->start_ms + (loop->cur_hrtime - loop->start_hrtime) / 1000;
- }
- uint64_t hloop_now_hrtime(hloop_t* loop) {
- return loop->start_ms * 1000 + (loop->cur_hrtime - loop->start_hrtime);
- }
- void hloop_set_userdata(hloop_t* loop, void* userdata) {
- loop->userdata = userdata;
- }
- void* hloop_userdata(hloop_t* loop) {
- return loop->userdata;
- }
- hidle_t* hidle_add(hloop_t* loop, hidle_cb cb, uint32_t repeat) {
- hidle_t* idle;
- SAFE_ALLOC_SIZEOF(idle);
- idle->event_type = HEVENT_TYPE_IDLE;
- idle->priority = HEVENT_LOWEST_PRIORITY;
- idle->repeat = repeat;
- list_add(&idle->node, &loop->idles);
- EVENT_ADD(loop, idle, cb);
- loop->nidles++;
- return idle;
- }
- static void __hidle_del(hidle_t* idle) {
- if (idle->destroy) return;
- idle->destroy = 1;
- list_del(&idle->node);
- idle->loop->nidles--;
- }
- void hidle_del(hidle_t* idle) {
- if (!idle->active) return;
- EVENT_DEL(idle);
- __hidle_del(idle);
- }
- htimer_t* htimer_add(hloop_t* loop, htimer_cb cb, uint64_t timeout, uint32_t repeat) {
- if (timeout == 0) return NULL;
- htimeout_t* timer;
- SAFE_ALLOC_SIZEOF(timer);
- timer->event_type = HEVENT_TYPE_TIMEOUT;
- timer->priority = HEVENT_HIGHEST_PRIORITY;
- timer->repeat = repeat;
- timer->timeout = timeout;
- hloop_update_time(loop);
- timer->next_timeout = hloop_now_hrtime(loop) + timeout*1000;
- heap_insert(&loop->timers, &timer->node);
- EVENT_ADD(loop, timer, cb);
- loop->ntimers++;
- return (htimer_t*)timer;
- }
- void htimer_reset(htimer_t* timer) {
- if (timer->event_type != HEVENT_TYPE_TIMEOUT) {
- return;
- }
- hloop_t* loop = timer->loop;
- htimeout_t* timeout = (htimeout_t*)timer;
- if (timer->pending) {
- if (timer->repeat == 0) {
- timer->repeat = 1;
- }
- }
- else {
- heap_remove(&loop->timers, &timer->node);
- }
- timer->next_timeout = hloop_now_hrtime(loop) + timeout->timeout*1000;
- heap_insert(&loop->timers, &timer->node);
- EVENT_RESET(timer);
- }
- htimer_t* htimer_add_period(hloop_t* loop, htimer_cb cb,
- int8_t minute, int8_t hour, int8_t day,
- int8_t week, int8_t month, uint32_t repeat) {
- if (minute > 59 || hour > 23 || day > 31 || week > 6 || month > 12) {
- return NULL;
- }
- hperiod_t* timer;
- SAFE_ALLOC_SIZEOF(timer);
- timer->event_type = HEVENT_TYPE_PERIOD;
- timer->priority = HEVENT_HIGH_PRIORITY;
- timer->repeat = repeat;
- timer->minute = minute;
- timer->hour = hour;
- timer->day = day;
- timer->month = month;
- timer->week = week;
- timer->next_timeout = calc_next_timeout(minute, hour, day, week, month) * 1e6;
- heap_insert(&loop->timers, &timer->node);
- EVENT_ADD(loop, timer, cb);
- loop->ntimers++;
- return (htimer_t*)timer;
- }
- static void __htimer_del(htimer_t* timer) {
- if (timer->destroy) return;
- heap_remove(&timer->loop->timers, &timer->node);
- timer->loop->ntimers--;
- timer->destroy = 1;
- }
- void htimer_del(htimer_t* timer) {
- if (!timer->active) return;
- __htimer_del(timer);
- EVENT_DEL(timer);
- }
- void hio_init(hio_t* io) {
- memset(io, 0, sizeof(hio_t));
- io->event_type = HEVENT_TYPE_IO;
- io->event_index[0] = io->event_index[1] = -1;
- // write_queue init when hwrite try_write failed
- //write_queue_init(&io->write_queue, 4);;
- }
- static void fill_io_type(hio_t* io) {
- int type = 0;
- socklen_t optlen = sizeof(int);
- int ret = getsockopt(io->fd, SOL_SOCKET, SO_TYPE, (char*)&type, &optlen);
- printd("getsockopt SO_TYPE fd=%d ret=%d type=%d errno=%d\n", io->fd, ret, type, socket_errno());
- if (ret == 0) {
- switch (type) {
- case SOCK_STREAM: io->io_type = HIO_TYPE_TCP; break;
- case SOCK_DGRAM: io->io_type = HIO_TYPE_UDP; break;
- case SOCK_RAW: io->io_type = HIO_TYPE_IP; break;
- default: io->io_type = HIO_TYPE_SOCKET; break;
- }
- }
- else if (socket_errno() == ENOTSOCK) {
- switch (io->fd) {
- case 0: io->io_type = HIO_TYPE_STDIN; break;
- case 1: io->io_type = HIO_TYPE_STDOUT; break;
- case 2: io->io_type = HIO_TYPE_STDERR; break;
- default: io->io_type = HIO_TYPE_FILE; break;
- }
- }
- }
- static void hio_socket_init(hio_t* io) {
- // nonblocking
- nonblocking(io->fd);
- // fill io->localaddr io->peeraddr
- if (io->localaddr == NULL) {
- SAFE_ALLOC(io->localaddr, sizeof(sockaddr_un));
- }
- if (io->peeraddr == NULL) {
- SAFE_ALLOC(io->peeraddr, sizeof(sockaddr_un));
- }
- socklen_t addrlen = sizeof(sockaddr_un);
- int ret = getsockname(io->fd, io->localaddr, &addrlen);
- printd("getsockname fd=%d ret=%d errno=%d\n", io->fd, ret, socket_errno());
- // NOTE:
- // tcp_server peeraddr set by accept
- // udp_server peeraddr set by recvfrom
- // tcp_client/udp_client peeraddr set by hio_setpeeraddr
- if (io->io_type == HIO_TYPE_TCP || io->io_type == HIO_TYPE_SSL) {
- // tcp acceptfd
- addrlen = sizeof(sockaddr_un);
- ret = getpeername(io->fd, io->peeraddr, &addrlen);
- printd("getpeername fd=%d ret=%d errno=%d\n", io->fd, ret, socket_errno());
- }
- }
- void hio_ready(hio_t* io) {
- if (io->ready) return;
- io->ready = 1;
- io->closed = 0;
- io->accept = io->connect = io->connectex = 0;
- io->recv = io->send = 0;
- io->recvfrom = io->sendto = 0;
- io->io_type = HIO_TYPE_UNKNOWN;
- io->error = 0;
- io->events = io->revents = 0;
- io->read_cb = NULL;
- io->write_cb = NULL;
- io->close_cb = 0;
- io->accept_cb = 0;
- io->connect_cb = 0;
- io->event_index[0] = io->event_index[1] = -1;
- io->hovlp = NULL;
- fill_io_type(io);
- if (io->io_type & HIO_TYPE_SOCKET) {
- hio_socket_init(io);
- }
- }
- void hio_done(hio_t* io) {
- io->ready = 0;
- offset_buf_t* pbuf = NULL;
- while (!write_queue_empty(&io->write_queue)) {
- pbuf = write_queue_front(&io->write_queue);
- SAFE_FREE(pbuf->base);
- write_queue_pop_front(&io->write_queue);
- }
- write_queue_cleanup(&io->write_queue);
- }
- void hio_free(hio_t* io) {
- if (io == NULL) return;
- hio_done(io);
- SAFE_FREE(io->localaddr);
- SAFE_FREE(io->peeraddr);
- SAFE_FREE(io);
- }
- hio_t* hio_get(hloop_t* loop, int fd) {
- if (loop->ios.maxsize == 0) {
- io_array_init(&loop->ios, IO_ARRAY_INIT_SIZE);
- }
- if (fd >= loop->ios.maxsize) {
- int newsize = ceil2e(fd);
- io_array_resize(&loop->ios, newsize > fd ? newsize : 2*fd);
- }
- hio_t* io = loop->ios.ptr[fd];
- if (io == NULL) {
- SAFE_ALLOC_SIZEOF(io);
- hio_init(io);
- io->loop = loop;
- io->fd = fd;
- loop->ios.ptr[fd] = io;
- }
- if (!io->ready) {
- hio_ready(io);
- }
- return io;
- }
- int hio_add(hio_t* io, hio_cb cb, int events) {
- printd("hio_add fd=%d events=%d\n", io->fd, events);
- hloop_t* loop = io->loop;
- if (!io->ready) {
- hio_ready(io);
- }
- if (!io->active) {
- EVENT_ADD(loop, io, cb);
- loop->nios++;
- }
- if (cb) {
- io->cb = (hevent_cb)cb;
- }
- iowatcher_add_event(loop, io->fd, events);
- io->events |= events;
- return 0;
- }
- int hio_del(hio_t* io, int events) {
- printd("hio_del fd=%d io->events=%d events=%d\n", io->fd, io->events, events);
- if (!io->active) return 0;
- iowatcher_del_event(io->loop, io->fd, events);
- io->events &= ~events;
- if (io->events == 0) {
- io->loop->nios--;
- // NOTE: not EVENT_DEL, avoid free
- EVENT_INACTIVE(io);
- hio_done(io);
- }
- return 0;
- }
- hio_t* hread(hloop_t* loop, int fd, void* buf, size_t len, hread_cb read_cb) {
- hio_t* io = hio_get(loop, fd);
- if (io == NULL) return NULL;
- io->readbuf.base = (char*)buf;
- io->readbuf.len = len;
- if (read_cb) {
- io->read_cb = read_cb;
- }
- hio_read(io);
- return io;
- }
- hio_t* hwrite(hloop_t* loop, int fd, const void* buf, size_t len, hwrite_cb write_cb) {
- hio_t* io = hio_get(loop, fd);
- if (io == NULL) return NULL;
- if (write_cb) {
- io->write_cb = write_cb;
- }
- hio_write(io, buf, len);
- return io;
- }
- hio_t* haccept(hloop_t* loop, int listenfd, haccept_cb accept_cb) {
- hio_t* io = hio_get(loop, listenfd);
- if (io == NULL) return NULL;
- io->accept = 1;
- if (accept_cb) {
- io->accept_cb = accept_cb;
- }
- hio_accept(io);
- return io;
- }
- hio_t* hconnect (hloop_t* loop, int connfd, hconnect_cb connect_cb) {
- hio_t* io = hio_get(loop, connfd);
- if (io == NULL) return NULL;
- io->connect = 1;
- if (connect_cb) {
- io->connect_cb = connect_cb;
- }
- hio_connect(io);
- return io;
- }
- hio_t* hrecv (hloop_t* loop, int connfd, void* buf, size_t len, hread_cb read_cb) {
- //hio_t* io = hio_get(loop, connfd);
- //if (io == NULL) return NULL;
- //io->recv = 1;
- //if (io->io_type != HIO_TYPE_SSL) {
- //io->io_type = HIO_TYPE_TCP;
- //}
- return hread(loop, connfd, buf, len, read_cb);
- }
- hio_t* hsend (hloop_t* loop, int connfd, const void* buf, size_t len, hwrite_cb write_cb) {
- //hio_t* io = hio_get(loop, connfd);
- //if (io == NULL) return NULL;
- //io->send = 1;
- //if (io->io_type != HIO_TYPE_SSL) {
- //io->io_type = HIO_TYPE_TCP;
- //}
- return hwrite(loop, connfd, buf, len, write_cb);
- }
- hio_t* hrecvfrom (hloop_t* loop, int sockfd, void* buf, size_t len, hread_cb read_cb) {
- //hio_t* io = hio_get(loop, sockfd);
- //if (io == NULL) return NULL;
- //io->recvfrom = 1;
- //io->io_type = HIO_TYPE_UDP;
- return hread(loop, sockfd, buf, len, read_cb);
- }
- hio_t* hsendto (hloop_t* loop, int sockfd, const void* buf, size_t len, hwrite_cb write_cb) {
- //hio_t* io = hio_get(loop, sockfd);
- //if (io == NULL) return NULL;
- //io->sendto = 1;
- //io->io_type = HIO_TYPE_UDP;
- return hwrite(loop, sockfd, buf, len, write_cb);
- }
- hio_t* create_tcp_server (hloop_t* loop, const char* host, int port, haccept_cb accept_cb) {
- int listenfd = Listen(port, host);
- if (listenfd < 0) {
- return NULL;
- }
- hio_t* io = haccept(loop, listenfd, accept_cb);
- if (io == NULL) {
- closesocket(listenfd);
- }
- return io;
- }
- hio_t* create_tcp_client (hloop_t* loop, const char* host, int port, hconnect_cb connect_cb) {
- sockaddr_un peeraddr;
- socklen_t addrlen = sizeof(peeraddr);
- memset(&peeraddr, 0, addrlen);
- int ret = sockaddr_assign(&peeraddr, host, port);
- if (ret != 0) {
- //printf("unknown host: %s\n", host);
- return NULL;
- }
- int connfd = socket(peeraddr.sa.sa_family, SOCK_STREAM, 0);
- if (connfd < 0) {
- perror("socket");
- return NULL;
- }
- hio_t* io = hio_get(loop, connfd);
- if (io == NULL) return NULL;
- hio_set_peeraddr(io, &peeraddr.sa, sockaddrlen(&peeraddr));
- hconnect(loop, connfd, connect_cb);
- return io;
- }
- // @server: socket -> bind -> hrecvfrom
- hio_t* create_udp_server(hloop_t* loop, const char* host, int port) {
- int bindfd = Bind(port, host, SOCK_DGRAM);
- if (bindfd < 0) {
- return NULL;
- }
- return hio_get(loop, bindfd);
- }
- // @client: Resolver -> socket -> hio_get -> hio_set_peeraddr
- hio_t* create_udp_client(hloop_t* loop, const char* host, int port) {
- sockaddr_un peeraddr;
- socklen_t addrlen = sizeof(peeraddr);
- memset(&peeraddr, 0, addrlen);
- int ret = sockaddr_assign(&peeraddr, host, port);
- if (ret != 0) {
- //printf("unknown host: %s\n", host);
- return NULL;
- }
- int sockfd = socket(peeraddr.sa.sa_family, SOCK_DGRAM, 0);
- if (sockfd < 0) {
- perror("socket");
- return NULL;
- }
- hio_t* io = hio_get(loop, sockfd);
- if (io == NULL) return NULL;
- hio_set_peeraddr(io, &peeraddr.sa, sockaddrlen(&peeraddr));
- return io;
- }
- static void sockpair_read_cb(hio_t* io, void* buf, int readbytes) {
- hloop_t* loop = io->loop;
- hevent_t* pev = NULL;
- hevent_t ev;
- for (int i = 0; i < readbytes; ++i) {
- hmutex_lock(&loop->custom_events_mutex);
- if (event_queue_empty(&loop->custom_events)) {
- goto unlock;
- }
- pev = event_queue_front(&loop->custom_events);
- if (pev == NULL) {
- goto unlock;
- }
- ev = *pev;
- event_queue_pop_front(&loop->custom_events);
- // NOTE: unlock before cb, avoid deadlock if hloop_post_event called in cb.
- hmutex_unlock(&loop->custom_events_mutex);
- if (ev.cb) {
- ev.cb(&ev);
- }
- }
- return;
- unlock:
- hmutex_unlock(&loop->custom_events_mutex);
- }
- void hloop_post_event(hloop_t* loop, hevent_t* ev) {
- char buf = '1';
- hmutex_lock(&loop->custom_events_mutex);
- if (loop->sockpair[0] <= 0 && loop->sockpair[1] <= 0) {
- if (Socketpair(AF_INET, SOCK_STREAM, 0, loop->sockpair) != 0) {
- hloge("socketpair error");
- goto unlock;
- }
- hread(loop, loop->sockpair[1], loop->readbuf.base, loop->readbuf.len, sockpair_read_cb);
- }
- if (loop->custom_events.maxsize == 0) {
- event_queue_init(&loop->custom_events, CUSTOM_EVENT_QUEUE_INIT_SIZE);
- }
- if (ev->loop == NULL) {
- ev->loop = loop;
- }
- if (ev->event_type == 0) {
- ev->event_type = HEVENT_TYPE_CUSTOM;
- }
- if (ev->event_id == 0) {
- ev->event_id = ++loop->event_counter;
- }
- event_queue_push_back(&loop->custom_events, ev);
- hwrite(loop, loop->sockpair[0], &buf, 1, NULL);
- unlock:
- hmutex_unlock(&loop->custom_events_mutex);
- }
|