summaryrefslogtreecommitdiff
path: root/ecp/src/msgq.c
diff options
context:
space:
mode:
authorUros Majstorovic <majstor@majstor.org>2020-08-05 03:38:22 +0200
committerUros Majstorovic <majstor@majstor.org>2020-08-05 03:38:22 +0200
commit5cd610a07468137066ea4daa5176c3e7045113b0 (patch)
treea6a5b572572f8f37ec2cb87332fa46e9bcc53aa7 /ecp/src/msgq.c
parent2473a7d5c51806ab8651cd3c4e07a15b62084eb5 (diff)
ecp moved to root; fixed utils and tests
Diffstat (limited to 'ecp/src/msgq.c')
-rw-r--r--ecp/src/msgq.c136
1 files changed, 136 insertions, 0 deletions
diff --git a/ecp/src/msgq.c b/ecp/src/msgq.c
new file mode 100644
index 0000000..201af4a
--- /dev/null
+++ b/ecp/src/msgq.c
@@ -0,0 +1,136 @@
+#include "core.h"
+
+#ifdef ECP_WITH_MSGQ
+
+#include <sys/time.h>
+
+#define MIN(a,b) (((a)<(b))?(a):(b))
+#define MAX(a,b) (((a)>(b))?(a):(b))
+
+#define MSG_IDX_MASK(idx) ((idx) & ((ECP_MSGQ_MAX_MSG) - 1))
+
+static struct timespec *abstime_ts(struct timespec *ts, ecp_cts_t msec) {
+ struct timeval tv;
+ uint64_t us_start;
+
+ gettimeofday(&tv, NULL);
+ us_start = tv.tv_sec * (uint64_t) 1000000 + tv.tv_usec;
+ us_start += msec * 1000;
+ ts->tv_sec = us_start / 1000000;
+ ts->tv_nsec = (us_start % 1000000) * 1000;
+ return ts;
+}
+
+int ecp_conn_msgq_create(ECPConnMsgQ *msgq) {
+ int i;
+ int rv;
+
+ if (msgq == NULL) return ECP_ERR;
+
+ memset(msgq, 0, sizeof(ECPConnMsgQ));
+
+ rv = pthread_mutex_init(&msgq->mutex, NULL);
+ if (rv) return ECP_ERR;
+
+ for (i=0; i<ECP_MAX_MTYPE; i++) {
+ rv = pthread_cond_init(&msgq->cond[i], NULL);
+ if (rv) {
+ int j;
+
+ for (j=0; j<i; j++) {
+ pthread_cond_destroy(&msgq->cond[j]);
+ }
+ pthread_mutex_destroy(&msgq->mutex);
+ return ECP_ERR;
+ }
+ }
+
+ return ECP_OK;
+}
+
+void ecp_conn_msgq_destroy(ECPConnMsgQ *msgq) {
+ int i;
+
+ if (msgq == NULL) return;
+
+ for (i=0; i<ECP_MAX_MTYPE; i++) {
+ pthread_cond_destroy(&msgq->cond[i]);
+ }
+ pthread_mutex_destroy(&msgq->mutex);
+}
+
+int ecp_conn_msgq_start(ECPConnMsgQ *msgq, ecp_seq_t seq) {
+ if (msgq == NULL) return ECP_ERR;
+
+ msgq->seq_max = seq;
+ msgq->seq_start = seq + 1;
+
+ return ECP_OK;
+}
+
+int ecp_conn_msgq_push(ECPConnection *conn, ecp_seq_t seq, unsigned char mtype) {
+ ECPRBRecv *buf = conn->rbuf.recv;
+ ECPConnMsgQ *msgq = buf ? &buf->msgq : NULL;
+
+ mtype &= ECP_MTYPE_MASK;
+ if (msgq == NULL) return ECP_ERR;
+ if (mtype >= ECP_MAX_MTYPE) return ECP_ERR_MAX_MTYPE;
+
+ if ((unsigned short)(msgq->idx_w[mtype] - msgq->idx_r[mtype]) == ECP_MSGQ_MAX_MSG) return ECP_MSGQ_ERR_MAX_MSG;
+ if (msgq->idx_w[mtype] == msgq->idx_r[mtype]) pthread_cond_signal(&msgq->cond[mtype]);
+
+ msgq->seq_msg[mtype][MSG_IDX_MASK(msgq->idx_w[mtype])] = seq;
+ msgq->idx_w[mtype]++;
+
+ if (ECP_SEQ_LT(msgq->seq_max, seq)) msgq->seq_max = seq;
+
+ return ECP_OK;
+}
+
+ssize_t ecp_conn_msgq_pop(ECPConnection *conn, unsigned char mtype, unsigned char *msg, size_t msg_size, ecp_cts_t timeout) {
+ ECPRBRecv *buf = conn->rbuf.recv;
+ ECPConnMsgQ *msgq = buf ? &buf->msgq : NULL;
+ ssize_t rv = ECP_OK;
+ int idx;
+
+ if (msgq == NULL) return ECP_ERR;
+ if (mtype >= ECP_MAX_MTYPE) return ECP_ERR_MAX_MTYPE;
+
+ if (msgq->idx_r[mtype] == msgq->idx_w[mtype]) {
+ if (timeout == -1) {
+ pthread_cond_wait(&msgq->cond[mtype], &msgq->mutex);
+ } else {
+ struct timespec ts;
+ int _rv = pthread_cond_timedwait(&msgq->cond[mtype], &msgq->mutex, abstime_ts(&ts, timeout));
+ if (_rv) rv = ECP_ERR_TIMEOUT;
+ }
+ }
+ if (!rv) {
+ ecp_seq_t seq = msgq->seq_msg[mtype][MSG_IDX_MASK(msgq->idx_r[mtype])];
+ ecp_seq_t seq_offset = seq - buf->rbuf.seq_start;
+ unsigned int idx = ECP_RBUF_IDX_MASK(buf->rbuf.msg_start + seq_offset, buf->rbuf.msg_size);
+
+ msgq->idx_r[mtype]++;
+ buf->rbuf.msg[idx].flags &= ~ECP_RBUF_FLAG_IN_MSGQ;
+
+ if (msgq->seq_start == seq) {
+ int i, _idx = idx;
+ ecp_seq_t msg_cnt = msgq->seq_max - msgq->seq_start + 1;
+
+ for (i=0; i<msg_cnt; i++) {
+ if (buf->rbuf.msg[_idx].flags & ECP_RBUF_FLAG_IN_MSGQ) break;
+ _idx = ECP_RBUF_IDX_MASK(_idx + 1, buf->rbuf.msg_size);
+ }
+ msgq->seq_start += i;
+ }
+ rv = buf->rbuf.msg[idx].size - 1;
+ if (rv >= 0) {
+ rv = MIN(msg_size, rv);
+ memcpy(msg, buf->rbuf.msg[idx].msg + 1, rv);
+ }
+ }
+
+ return rv;
+}
+
+#endif /* ECP_WITH_MSGQ */ \ No newline at end of file