From a8792202aee8f34e5d24d59ab075ef3e0dac2aff Mon Sep 17 00:00:00 2001 From: Dimitri Staessens Date: Sat, 26 Mar 2022 12:12:39 +0100 Subject: lib: Refactor writing to rbuff Writing valid packets to the rbuff (add crc check, encrypt) is now extracted into a function. Signed-off-by: Dimitri Staessens Signed-off-by: Sander Vrijders --- src/lib/dev.c | 126 ++++++++++++++++++++++++++-------------------------------- 1 file changed, 57 insertions(+), 69 deletions(-) diff --git a/src/lib/dev.c b/src/lib/dev.c index a96a41aa..d7c98366 100644 --- a/src/lib/dev.c +++ b/src/lib/dev.c @@ -1082,6 +1082,60 @@ static int flow_keepalive(int fd) return 0; } +static int flow_tx_sdb(struct flow * flow, + struct shm_du_buff * sdb, + bool block, + struct timespec * abstime) +{ + struct timespec now; + ssize_t idx; + int ret; + + clock_gettime(PTHREAD_COND_CLOCK, &now); + + pthread_rwlock_wrlock(&ai.lock); + + flow->snd_act = now; + + pthread_rwlock_unlock(&ai.lock); + + idx = shm_du_buff_get_idx(sdb); + + pthread_rwlock_rdlock(&ai.lock); + + if (shm_du_buff_len(sdb) > 0) { + if (frcti_snd(flow->frcti, sdb) < 0) + goto enomem; + + if (flow->qs.cypher_s > 0 && crypt_encrypt(flow, sdb) < 0) + goto enomem; + + if (flow->qs.ber == 0 && add_crc(sdb) != 0) + goto enomem; + } + + pthread_cleanup_push(__cleanup_rwlock_unlock, &ai.lock); + + if (!block) + ret = shm_rbuff_write(flow->tx_rb, idx); + else + ret = shm_rbuff_write_b(flow->tx_rb, idx, abstime); + + if (ret < 0) + shm_rdrbuff_remove(ai.rdrb, idx); + else + shm_flow_set_notify(flow->set, flow->flow_id, FLOW_PKT); + + pthread_cleanup_pop(true); + + return 0; + +enomem: + pthread_rwlock_unlock(&ai.lock); + shm_rdrbuff_remove(ai.rdrb, idx); + return -ENOMEM; +} + ssize_t flow_write(int fd, const void * buf, size_t count) @@ -1153,50 +1207,12 @@ ssize_t flow_write(int fd, if (idx < 0) return idx; - clock_gettime(PTHREAD_COND_CLOCK, &abs); - - pthread_rwlock_wrlock(&ai.lock); - - flow->snd_act = abs; - - pthread_rwlock_unlock(&ai.lock); - if (count > 0) memcpy(ptr, buf, count); - pthread_rwlock_rdlock(&ai.lock); - - if (count > 0) { - if (frcti_snd(flow->frcti, sdb) < 0) - goto enomem; - - if (flow->qs.cypher_s > 0 && crypt_encrypt(flow, sdb) < 0) - goto enomem; - - if (flow->qs.ber == 0 && add_crc(sdb) != 0) - goto enomem; - } - - pthread_cleanup_push(__cleanup_rwlock_unlock, &ai.lock); - - if (flags & FLOWFWNOBLOCK) - ret = shm_rbuff_write(flow->tx_rb, idx); - else - ret = shm_rbuff_write_b(flow->tx_rb, idx, abstime); - - if (ret < 0) - shm_rdrbuff_remove(ai.rdrb, idx); - else - shm_flow_set_notify(flow->set, flow->flow_id, FLOW_PKT); - - pthread_cleanup_pop(true); + ret = flow_tx_sdb(flow, sdb, flags & FLOWFWNOBLOCK, abstime); return ret < 0 ? (ssize_t) ret : (ssize_t) count; - - enomem: - pthread_rwlock_unlock(&ai.lock); - shm_rdrbuff_remove(ai.rdrb, idx); - return -ENOMEM; } static bool invalid_pkt(struct flow * flow, @@ -1917,16 +1933,12 @@ int ipcp_flow_read(int fd, int ipcp_flow_write(int fd, struct shm_du_buff * sdb) { - struct timespec now; struct flow * flow; int ret; - ssize_t idx; assert(fd >= 0 && fd < SYS_MAX_FLOWS); assert(sdb); - clock_gettime(PTHREAD_COND_CLOCK, &now); - flow = &ai.flows[fd]; pthread_rwlock_wrlock(&ai.lock); @@ -1941,32 +1953,7 @@ int ipcp_flow_write(int fd, return -EPERM; } - assert(flow->tx_rb); - - idx = shm_du_buff_get_idx(sdb); - - if (frcti_snd(flow->frcti, sdb) < 0) { - pthread_rwlock_unlock(&ai.lock); - return -ENOMEM; - } - - if (flow->qs.ber == 0 && add_crc(sdb) != 0) { - pthread_rwlock_unlock(&ai.lock); - shm_rdrbuff_remove(ai.rdrb, idx); - return -ENOMEM; - } - - ret = shm_rbuff_write_b(flow->tx_rb, idx, NULL); - if (ret == 0) - shm_flow_set_notify(flow->set, flow->flow_id, FLOW_PKT); - else - shm_rdrbuff_remove(ai.rdrb, idx); - - flow->snd_act = now; - - pthread_rwlock_unlock(&ai.lock); - - assert(ret <= 0); + ret = flow_tx_sdb(flow, sdb, false, NULL); return ret; } @@ -2075,6 +2062,7 @@ int local_flow_write(int fd, pthread_rwlock_unlock(&ai.lock); return -ENOTALLOC; } + ret = shm_rbuff_write_b(flow->tx_rb, idx, NULL); if (ret == 0) shm_flow_set_notify(flow->set, flow->flow_id, FLOW_PKT); -- cgit v1.2.3