summaryrefslogtreecommitdiff
path: root/code/core/timer.c
diff options
context:
space:
mode:
Diffstat (limited to 'code/core/timer.c')
-rw-r--r--code/core/timer.c224
1 files changed, 224 insertions, 0 deletions
diff --git a/code/core/timer.c b/code/core/timer.c
new file mode 100644
index 0000000..cf77d6e
--- /dev/null
+++ b/code/core/timer.c
@@ -0,0 +1,224 @@
+#include "core.h"
+
+#include <string.h>
+
+int ecp_timer_create(ECPTimer *timer) {
+ timer->head = -1;
+
+#ifdef ECP_WITH_PTHREAD
+ int rv = pthread_mutex_init(&timer->mutex, NULL);
+ if (rv) return ECP_ERR;
+#endif
+
+ return ECP_OK;
+}
+
+void ecp_timer_destroy(ECPTimer *timer) {
+#ifdef ECP_WITH_PTHREAD
+ pthread_mutex_destroy(&timer->mutex);
+#endif
+}
+
+int ecp_timer_item_init(ECPTimerItem *ti, ECPConnection *conn, unsigned char mtype, unsigned short cnt, unsigned int timeout) {
+ unsigned int abstime = conn->sock->ctx->tm.abstime_ms(timeout);
+
+ if (mtype >= ECP_MAX_MTYPE) return ECP_ERR_MAX_MTYPE;
+
+ ti->conn = conn;
+ ti->mtype = mtype;
+ ti->cnt = cnt-1;
+ ti->timeout = timeout;
+ ti->abstime = abstime;
+ ti->retry = NULL;
+ ti->pld = NULL;
+ ti->pld_size = 0;
+
+ return ECP_OK;
+}
+
+int ecp_timer_push(ECPTimerItem *ti) {
+ int i, is_reg, rv = ECP_OK;
+ ECPConnection *conn = ti->conn;
+ ECPTimer *timer = &conn->sock->timer;
+
+#ifdef ECP_WITH_PTHREAD
+ pthread_mutex_lock(&timer->mutex);
+ pthread_mutex_lock(&conn->mutex);
+#endif
+ is_reg = ecp_conn_is_reg(conn);
+ if (is_reg) conn->refcount++;
+#ifdef ECP_WITH_PTHREAD
+ pthread_mutex_unlock(&conn->mutex);
+#endif
+
+ if (timer->head == ECP_MAX_TIMER-1) rv = ECP_ERR_MAX_TIMER;
+ if (!rv && !is_reg) rv = ECP_ERR_CLOSED;
+
+ if (!rv) {
+ for (i=timer->head; i>=0; i--) {
+ if (timer->item[i].abstime >= ti->abstime) {
+ if (i != timer->head) memmove(timer->item+i+2, timer->item+i+1, sizeof(ECPTimerItem) * (timer->head-i));
+ timer->item[i+1] = *ti;
+ timer->head++;
+ break;
+ }
+ }
+ if (i == -1) {
+ if (timer->head != -1) memmove(timer->item+1, timer->item, sizeof(ECPTimerItem) * (timer->head+1));
+ timer->item[0] = *ti;
+ timer->head++;
+ }
+ }
+
+#ifdef ECP_WITH_PTHREAD
+ pthread_mutex_unlock(&timer->mutex);
+#endif
+
+ return rv;
+}
+
+void ecp_timer_pop(ECPConnection *conn, unsigned char mtype) {
+ int i;
+ ECPTimer *timer = &conn->sock->timer;
+
+#ifdef ECP_WITH_PTHREAD
+ pthread_mutex_lock(&timer->mutex);
+#endif
+
+ for (i=timer->head; i>=0; i--) {
+ ECPConnection *curr_conn = timer->item[i].conn;
+ if ((conn == curr_conn) && (mtype == timer->item[i].mtype)) {
+ if (i != timer->head) {
+ memmove(timer->item+i, timer->item+i+1, sizeof(ECPTimerItem) * (timer->head-i));
+ memset(timer->item+timer->head, 0, sizeof(ECPTimerItem));
+ } else {
+ memset(timer->item+i, 0, sizeof(ECPTimerItem));
+ }
+#ifdef ECP_WITH_PTHREAD
+ pthread_mutex_lock(&conn->mutex);
+#endif
+ conn->refcount--;
+#ifdef ECP_WITH_PTHREAD
+ pthread_mutex_unlock(&conn->mutex);
+#endif
+ timer->head--;
+ break;
+ }
+ }
+
+#ifdef ECP_WITH_PTHREAD
+ pthread_mutex_unlock(&timer->mutex);
+#endif
+
+}
+
+void ecp_timer_remove(ECPConnection *conn) {
+ int i;
+ ECPTimer *timer = &conn->sock->timer;
+
+#ifdef ECP_WITH_PTHREAD
+ pthread_mutex_lock(&timer->mutex);
+ pthread_mutex_lock(&conn->mutex);
+#endif
+
+ for (i=timer->head; i>=0; i--) {
+ ECPConnection *curr_conn = timer->item[i].conn;
+ if (conn == curr_conn) {
+ if (i != timer->head) {
+ memmove(timer->item+i, timer->item+i+1, sizeof(ECPTimerItem) * (timer->head-i));
+ memset(timer->item+timer->head, 0, sizeof(ECPTimerItem));
+ } else {
+ memset(timer->item+i, 0, sizeof(ECPTimerItem));
+ }
+ conn->refcount--;
+ timer->head--;
+ }
+ }
+
+#ifdef ECP_WITH_PTHREAD
+ pthread_mutex_unlock(&conn->mutex);
+ pthread_mutex_unlock(&timer->mutex);
+#endif
+
+}
+
+unsigned int ecp_timer_exe(ECPSocket *sock) {
+ int i;
+ unsigned int ret = 0;
+ ECPTimer *timer = &sock->timer;
+ ECPTimerItem to_exec[ECP_MAX_TIMER];
+ int to_exec_size = 0;
+ unsigned int now = sock->ctx->tm.abstime_ms(0);
+
+#ifdef ECP_WITH_PTHREAD
+ pthread_mutex_lock(&timer->mutex);
+#endif
+
+ for (i=timer->head; i>=0; i--) {
+ unsigned int abstime = timer->item[i].abstime;
+
+ if (abstime > now) {
+ ret = abstime - now;
+ break;
+ }
+ to_exec[to_exec_size] = timer->item[i];
+ to_exec_size++;
+ }
+ if (i != timer->head) {
+ memset(timer->item+i+1, 0, sizeof(ECPTimerItem) * (timer->head-i));
+ timer->head = i;
+ }
+
+#ifdef ECP_WITH_PTHREAD
+ pthread_mutex_unlock(&timer->mutex);
+#endif
+
+ for (i=to_exec_size-1; i>=0; i--) {
+ int rv = ECP_OK;
+ ECPConnection *conn = to_exec[i].conn;
+ ECPSocket *sock = conn->sock;
+ unsigned char mtype = to_exec[i].mtype;
+ unsigned char *pld = to_exec[i].pld;
+ unsigned char pld_size = to_exec[i].pld_size;
+ ecp_timer_retry_t *retry = to_exec[i].retry;
+ ecp_conn_handler_msg_t *handler = conn->sock->ctx->handler[conn->type] ? conn->sock->ctx->handler[conn->type]->msg[mtype] : NULL;
+
+ if (to_exec[i].cnt) {
+ to_exec[i].cnt--;
+ to_exec[i].abstime = now + to_exec[i].timeout;
+ if (retry) {
+ rv = retry(conn, to_exec+i);
+ } else {
+ ssize_t _rv = ecp_pld_send(conn, pld, pld_size);
+ if (_rv < 0) rv = _rv;
+ }
+ if (!rv) rv = ecp_timer_push(to_exec+i);
+ if (rv && (rv != ECP_ERR_CLOSED) && handler) handler(conn, mtype, NULL, rv);
+ } else if (handler) {
+ handler(conn, mtype, NULL, ECP_ERR_TIMEOUT);
+ }
+#ifdef ECP_WITH_PTHREAD
+ pthread_mutex_lock(&conn->mutex);
+#endif
+ conn->refcount--;
+#ifdef ECP_WITH_PTHREAD
+ pthread_mutex_unlock(&conn->mutex);
+#endif
+ }
+
+ return ret;
+}
+
+ssize_t ecp_timer_send(ECPConnection *conn, ecp_timer_retry_t *send_f, unsigned char mtype, unsigned short cnt, unsigned int timeout) {
+ int rv = ECP_OK;
+ ECPTimerItem ti;
+
+ rv = ecp_timer_item_init(&ti, conn, mtype, cnt, timeout);
+ if (rv) return rv;
+
+ ti.retry = send_f;
+ rv = ecp_timer_push(&ti);
+ if (rv) return rv;
+
+ return send_f(conn, NULL);
+}