summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat
-rw-r--r--include/ouroboros/CMakeLists.txt1
-rw-r--r--include/ouroboros/shm_du_map.h74
-rw-r--r--src/lib/shm_du_map.c432
-rw-r--r--src/lib/tests/CMakeLists.txt15
-rw-r--r--src/lib/tests/shm_du_map_test_create.c51
-rw-r--r--src/lib/tests/shm_du_map_test_du_buff_create.c95
-rw-r--r--src/lib/tests/shm_du_map_test_prod_cons.c153
7 files changed, 820 insertions, 1 deletions
diff --git a/include/ouroboros/CMakeLists.txt b/include/ouroboros/CMakeLists.txt
index 167cf0e2..a196140b 100644
--- a/include/ouroboros/CMakeLists.txt
+++ b/include/ouroboros/CMakeLists.txt
@@ -14,6 +14,7 @@ set(HEADER_FILES
list.h
logs.h
rina_name.h
+ shm_du_map.h
sockets.h
utils.h
)
diff --git a/include/ouroboros/shm_du_map.h b/include/ouroboros/shm_du_map.h
new file mode 100644
index 00000000..b3533fa3
--- /dev/null
+++ b/include/ouroboros/shm_du_map.h
@@ -0,0 +1,74 @@
+/*
+ * Ouroboros - Copyright (C) 2016
+ *
+ * Shared memory map for data units
+ *
+ * Dimitri Staessens <[email protected]>
+ * Sander Vrijders <[email protected]>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+ */
+
+#ifndef OUROBOROS_SHM_DU_MAP_H
+#define OUROBOROS_SHM_DU_MAP_H
+
+#ifndef _POSIX_C_SOURCE
+#define _POSIX_C_SOURCE 199506L
+#endif
+
+#define OUROBOROS_PREFIX "shm_du_map"
+
+#ifndef SHM_DU_BUFF_BLOCK_SIZE
+#define SHM_DU_BUFF_BLOCK_SIZE sysconf(_SC_PAGESIZE)
+#endif
+
+#ifndef SHM_DU_MAP_FILENAME
+#define SHM_DU_MAP_FILENAME "ouroboros_du_map"
+#endif
+
+#ifndef SHM_DU_MAP_SIZE
+#define SHM_DU_MAP_SIZE 10
+#endif
+
+#include "common.h"
+#include "logs.h"
+
+struct shm_du_buff;
+
+struct shm_du_map;
+
+struct shm_du_map * shm_du_map_create();
+struct shm_du_map * shm_du_map_open();
+void shm_du_map_close(struct shm_du_map * dum);
+
+struct shm_du_buff * shm_create_du_buff(struct shm_du_map * dum,
+ size_t size,
+ size_t headspace,
+ uint8_t * data,
+ size_t len);
+int shm_release_du_buff(struct shm_du_map * dum);
+
+uint8_t * shm_du_buff_head_alloc(struct shm_du_map * dum,
+ struct shm_du_buff * sdb,
+ size_t size);
+uint8_t * shm_du_buff_tail_alloc(struct shm_du_map * dum,
+ struct shm_du_buff * sdb,
+ size_t size);
+int shm_du_buff_head_release(struct shm_du_buff * sdb,
+ size_t size);
+int shm_du_buff_tail_release(struct shm_du_buff * sdb,
+ size_t size);
+
+#endif /* OUROBOROS_SHM_DU_MAP_H */
diff --git a/src/lib/shm_du_map.c b/src/lib/shm_du_map.c
new file mode 100644
index 00000000..d071b0c3
--- /dev/null
+++ b/src/lib/shm_du_map.c
@@ -0,0 +1,432 @@
+/*
+ * Ouroboros - Copyright (C) 2016
+ *
+ * Shared memory map for data units
+ *
+ * Dimitri Staessens <[email protected]>
+ * Sander Vrijders <[email protected]>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+ */
+
+#ifndef SHM_DU_MAP_C
+#define SHM_DU_MAP_C
+
+#include <ouroboros/shm_du_map.h>
+
+#include <pthread.h>
+#include <sys/mman.h>
+#include <fcntl.h>
+#include <malloc.h>
+#include <string.h>
+
+
+#define SHM_DU_BLOCK_DATA_SIZE (SHM_DU_BUFF_BLOCK_SIZE - \
+ sizeof(struct shm_block))
+#define SHM_BLOCKS_IN_MAP (1 << SHM_DU_MAP_SIZE)
+#define SHM_BLOCKS_SIZE (SHM_DU_BUFF_BLOCK_SIZE * SHM_BLOCKS_IN_MAP)
+#define SHM_BUFFS_SIZE (SHM_BLOCKS_IN_MAP * sizeof (struct shm_du_buff))
+#define SHM_FILE_SIZE (SHM_BLOCKS_IN_MAP * (SHM_DU_BUFF_BLOCK_SIZE \
+ + sizeof(struct shm_du_buff) \
+ + sizeof(uint8_t)) \
+ + 2 * sizeof (size_t) \
+ + sizeof(pthread_mutex_t))
+
+#define idx_to_block_ptr(dum, i) ((struct shm_block *) \
+ (dum->shm_base + i * SHM_DU_BUFF_BLOCK_SIZE))
+#define idx_to_du_buff_ptr(dum, i) (dum->ptr_du_buff + i)
+#define du_buff_ptr_to_idx(dum, sdb) ((sdb - dum->ptr_du_buff) / sizeof *sdb)
+#define block_ptr_to_idx(dum, sdb) (((uint8_t *)sdb - dum->shm_base) \
+ / SHM_DU_BUFF_BLOCK_SIZE)
+
+#define shm_map_used(dum) ((*(dum->ptr_head) + SHM_BLOCKS_IN_MAP - \
+ *(dum->ptr_tail)) & (SHM_BLOCKS_IN_MAP - 1))
+
+#define shm_map_free(dum, i)(shm_map_used(dum) + i + 1 < SHM_BLOCKS_IN_MAP)
+
+struct shm_block {
+ size_t size;
+ long next;
+ long prev;
+};
+
+struct shm_du_buff {
+ size_t size;
+ size_t du_head;
+ size_t du_tail;
+};
+
+struct shm_du_map {
+ uint8_t * shm_base; /* start of blocks */
+ struct shm_du_buff * ptr_du_buff; /* start of du_buff structs */
+ size_t * ptr_head; /* start of ringbuffer head */
+ size_t * ptr_tail; /* start of ringbuffer tail */
+ pthread_mutex_t * shm_mutex; /* lock all free space in shm */
+ int fd;
+};
+
+struct shm_du_map * shm_du_map_create()
+{
+ struct shm_du_map * dum;
+ int shm_fd;
+ uint8_t * shm_base;
+ pthread_mutexattr_t attr;
+
+ dum = malloc(sizeof *dum);
+ if (dum == NULL) {
+ LOG_ERR("Could not allocate struct.");
+ return NULL;
+ }
+
+ shm_fd = shm_open(SHM_DU_MAP_FILENAME, O_CREAT | O_EXCL | O_RDWR, 0666);
+ if (shm_fd == -1) {
+ LOG_ERR("Failed creating shared memory map.");
+ free(dum);
+ return NULL;
+ }
+
+ if (lseek (shm_fd,SHM_FILE_SIZE - 1, SEEK_SET) < 0) {
+ LOG_ERR("Failed to extend shared memory map.");
+ free(dum);
+ return NULL;
+ }
+
+ if (write (shm_fd, "", 1) != 1) {
+ LOG_ERR("Failed to finalise extension of shared memory map.");
+ free(dum);
+ return NULL;
+ }
+
+ shm_base = mmap(NULL,
+ SHM_FILE_SIZE,
+ PROT_READ | PROT_WRITE,
+ MAP_SHARED,
+ shm_fd,
+ 0);
+
+ if (shm_base == MAP_FAILED) {
+ LOG_ERR("Failed to map shared memory.");
+
+ if (shm_unlink(SHM_DU_MAP_FILENAME) == -1)
+ LOG_ERR("Failed to remove invalid shm.");
+
+ free(dum);
+ return NULL;
+ }
+#ifdef CONFIG_OUROBOROS_DEBUG
+ memset(shm_base, 0, SHM_FILE_SIZE);
+#endif
+ dum->shm_base = shm_base;
+ dum->ptr_du_buff = (struct shm_du_buff *)
+ ((uint8_t *) dum->shm_base + SHM_BLOCKS_SIZE);
+ dum->ptr_head = (size_t *)
+ ((uint8_t *) dum->ptr_du_buff + SHM_BUFFS_SIZE);
+ dum->ptr_tail = (size_t *)
+ ((uint8_t *) dum->ptr_head + sizeof(size_t));
+ dum->shm_mutex = (pthread_mutex_t *)
+ ((uint8_t *) dum->ptr_tail + sizeof(size_t));
+
+ pthread_mutexattr_init(&attr);
+ pthread_mutexattr_setpshared(&attr, PTHREAD_PROCESS_SHARED);
+ pthread_mutex_init(dum->shm_mutex, &attr);
+
+ *dum->ptr_head = 0;
+ *dum->ptr_tail = 0;
+
+ dum->fd = shm_fd;
+
+ return dum;
+}
+
+struct shm_du_map * shm_du_map_open()
+{
+ struct shm_du_map * dum;
+ int shm_fd;
+ uint8_t * shm_base;
+
+ shm_fd = shm_open(SHM_DU_MAP_FILENAME, O_RDWR, 0666);
+ if (shm_fd == -1) {
+ LOG_ERR("Failed opening shared memory for du_buff.");
+ return NULL;
+ }
+
+ shm_base = mmap(NULL,
+ SHM_FILE_SIZE,
+ PROT_READ | PROT_WRITE,
+ MAP_SHARED,
+ shm_fd,
+ 0);
+ if (shm_base == MAP_FAILED) {
+ LOG_ERR("Failed to map shared memory.");
+
+ if (shm_unlink(SHM_DU_MAP_FILENAME) == -1)
+ LOG_ERR("Failed to unlink invalid shm.");
+
+ return NULL;
+ }
+
+ dum = malloc(sizeof *dum);
+ if (dum == NULL) {
+ LOG_ERR("Could not allocate struct.");
+ return NULL;
+ }
+
+ dum->shm_base = shm_base;
+ dum->ptr_du_buff = (struct shm_du_buff *)
+ ((uint8_t *) dum->shm_base + SHM_BLOCKS_SIZE);
+ dum->ptr_head = (size_t *)
+ ((uint8_t *) dum->ptr_du_buff + SHM_BUFFS_SIZE);
+ dum->ptr_tail = (size_t *)
+ ((uint8_t *) dum->ptr_head + sizeof(size_t));
+ dum->shm_mutex = (pthread_mutex_t *)
+ ((uint8_t *) dum->ptr_tail + sizeof(size_t));
+
+ return dum;
+}
+
+void shm_du_map_close(struct shm_du_map * dum)
+{
+ if (dum == NULL) {
+ LOG_DBGF("Bogus input. Bugging out.");
+ return;
+ }
+
+ if (munmap(dum->shm_base, SHM_FILE_SIZE) == -1)
+ LOG_DBGF("Couldn't unmap shared memory.");
+
+ if (shm_unlink(SHM_DU_MAP_FILENAME) == -1)
+ LOG_DBGF("Failed to unlink shm.");
+
+ free(dum);
+}
+
+struct shm_du_buff * shm_create_du_buff(struct shm_du_map * dum,
+ size_t size,
+ size_t headspace,
+ uint8_t * data,
+ size_t len)
+{
+ struct shm_du_buff * sdb;
+ long prev_index = -1;
+ size_t remaining = size;
+ size_t ts = size - (headspace + len);
+ uint8_t * read_pos = data;
+ size_t blocks = 0;
+ int sz = size;
+
+ if (dum == NULL || data == NULL) {
+ LOG_DBGF("Bogus input, bugging out.");
+ return NULL;
+ }
+
+ if (headspace >= size) {
+ LOG_DBGF("Index out of bounds.");
+ return NULL;
+ }
+
+ if (headspace + len > size) {
+ LOG_DBGF("Buffer too small for data.");
+ return NULL;
+ }
+
+ if (headspace > SHM_DU_BLOCK_DATA_SIZE || ts > SHM_DU_BLOCK_DATA_SIZE) {
+ LOG_ERR("Illegal du_buff: Cannot fit PCI in DU_BUFF_BLOCK.");
+ return NULL;
+ }
+
+ pthread_mutex_lock(dum->shm_mutex);
+
+ while (sz > 0) {
+ sz -= SHM_DU_BLOCK_DATA_SIZE;
+ blocks++;
+ }
+
+ if (!shm_map_free(dum, blocks)) {
+ pthread_mutex_unlock(dum->shm_mutex);
+ LOG_DBGF("Allocation failed, Out of Memory.");
+ return NULL;
+ }
+
+ sdb = dum->ptr_du_buff + *dum->ptr_head;
+
+ sdb->size = size;
+ sdb->du_head = headspace;
+ sdb->du_tail = sdb->du_head + len;
+
+ while (remaining > 0) {
+ struct shm_block * shm_buf;
+ long bytes_to_copy = len;
+ uint8_t * write_pos;
+
+ shm_buf = idx_to_block_ptr(dum, *(dum->ptr_head));
+
+ write_pos = (uint8_t *) shm_buf + sizeof *shm_buf;
+
+ shm_buf->size = remaining < SHM_DU_BLOCK_DATA_SIZE ?
+ remaining : SHM_DU_BLOCK_DATA_SIZE;
+
+ bytes_to_copy = shm_buf->size;
+
+ if (remaining <= SHM_DU_BLOCK_DATA_SIZE)
+ bytes_to_copy -= ts;
+ else if (remaining - ts <= SHM_DU_BLOCK_DATA_SIZE)
+ shm_buf->size = remaining - ts;
+
+ remaining -= shm_buf->size;
+
+ if (prev_index == -1) {
+#ifdef CONFIG_OUROBOROS_DEBUG
+ memset(write_pos, 0, sdb->du_head);
+#endif
+ write_pos += sdb->du_head;
+ bytes_to_copy -= sdb->du_head;
+ }
+
+ if (prev_index != -1)
+ idx_to_block_ptr(dum, prev_index)->next =
+ *(dum->ptr_head);
+
+ if (len > 0) {
+ memcpy(write_pos, read_pos, bytes_to_copy);
+ }
+ read_pos += bytes_to_copy;
+#ifdef CONFIG_OUROBOROS_DEBUG
+ if (remaining == 0) {
+ write_pos + = bytes_to_copy;
+ memset(write_pos, 0, ts);
+ }
+#endif
+ shm_buf->next = -1;
+ shm_buf->prev = prev_index;
+
+ prev_index = *dum->ptr_head;
+
+ *(dum->ptr_head) = (*dum->ptr_head + 1)
+ & (SHM_BLOCKS_IN_MAP - 1);
+ }
+
+ pthread_mutex_unlock(dum->shm_mutex);
+
+ return sdb;
+}
+
+int shm_release_du_buff(struct shm_du_map * dum)
+{
+ int released = 0;
+
+ pthread_mutex_lock(dum->shm_mutex);
+
+ if (*dum->ptr_head == *dum->ptr_tail) {
+ LOG_DBGF("Attempt to free empty ringbuffer. Nothing to do.");
+ pthread_mutex_unlock(dum->shm_mutex);
+ return -1;
+ }
+
+ while (idx_to_block_ptr(dum, *dum->ptr_tail)->next != -1) {
+ *(dum->ptr_tail) = (*dum->ptr_tail + 1)
+ & (SHM_BLOCKS_IN_MAP -1);
+ released++;
+ }
+
+ *(dum->ptr_tail) = (*dum->ptr_tail + 1) & (SHM_BLOCKS_IN_MAP - 1);
+
+ pthread_mutex_unlock(dum->shm_mutex);
+
+ return 0;
+}
+
+uint8_t * shm_du_buff_head_alloc(struct shm_du_map * dum,
+ struct shm_du_buff * sdb,
+ size_t size)
+{
+ uint8_t * ret;
+
+ if (sdb == NULL) {
+ LOG_DBGF("Bogus input, bugging out.");
+ return NULL;
+ }
+
+ if ((long) (sdb->du_head - size) < 0) {
+ LOG_DBGF("Failed to allocate PCI headspace.");
+ return NULL;
+ }
+
+ sdb->du_head -= size;
+
+ ret = (uint8_t *) idx_to_block_ptr(dum, du_buff_ptr_to_idx(dum,sdb));
+
+ return ret + sizeof(struct shm_block) + sdb->du_head;
+}
+
+uint8_t * shm_du_buff_tail_alloc(struct shm_du_map * dum,
+ struct shm_du_buff * sdb,
+ size_t size)
+{
+ uint8_t * ret;
+
+ if (sdb == NULL) {
+ LOG_DBGF("Bogus input, bugging out.");
+ return NULL;
+ }
+
+ if (sdb->du_tail + size >= sdb->size) {
+ LOG_DBGF("Failed to allocate PCI tailspace.");
+ return NULL;
+ }
+
+ sdb->du_tail += size;
+
+ ret = (uint8_t *) idx_to_block_ptr(dum, du_buff_ptr_to_idx(dum,sdb));
+
+ return ret + sizeof(struct shm_block) + sdb->du_tail;
+}
+
+int shm_du_buff_head_release(struct shm_du_buff * sdb,
+ size_t size)
+{
+ if (sdb == NULL) {
+ LOG_DBGF("Bogus input, bugging out.");
+ return -EINVAL;
+ }
+
+ if (size > sdb->du_tail - sdb->du_head) {
+ LOG_DBGF("Tried to release beyond sdu boundary.");
+ return -EOVERFLOW;
+ }
+
+ sdb->du_head += size;
+
+ return sdb->du_head;
+}
+
+int shm_du_buff_tail_release(struct shm_du_buff * sdb,
+ size_t size)
+{
+ if (sdb == NULL) {
+ LOG_DBGF("Bogus input, bugging out.");
+ return -EINVAL;
+ }
+
+ if (size > sdb->du_tail - sdb->du_head) {
+ LOG_DBGF("Tried to release beyond sdu boundary.");
+ return -EOVERFLOW;
+ }
+
+ sdb->du_tail -= size;
+
+ return sdb->du_tail;
+}
+
+#endif
diff --git a/src/lib/tests/CMakeLists.txt b/src/lib/tests/CMakeLists.txt
index 99df7232..d4bcb4e4 100644
--- a/src/lib/tests/CMakeLists.txt
+++ b/src/lib/tests/CMakeLists.txt
@@ -1,14 +1,27 @@
get_filename_component(tmp ".." ABSOLUTE)
get_filename_component(src_folder "${tmp}" NAME)
+find_library(LIBRT_LIBRARIES rt)
+if(NOT LIBRT_LIBRARIES)
+ message(FATAL_ERROR "librt not found")
+endif()
+
+find_library(LIBPTHREAD_LIBRARIES pthread)
+if(NOT LIBPTHREAD_LIBRARIES)
+ message(FATAL_ERROR "libpthread not found")
+endif()
+
create_test_sourcelist(${src_folder}_tests test_suite.c
# Add new tests here
bitmap_test.c
du_buff_test.c
+ shm_du_map_test_create.c
+ shm_du_map_test_du_buff_create.c
+ shm_du_map_test_prod_cons.c
)
add_executable(${src_folder}_test EXCLUDE_FROM_ALL ${${src_folder}_tests})
-target_link_libraries(${src_folder}_test ouroboros)
+target_link_libraries(${src_folder}_test ouroboros rt pthread)
add_dependencies(check ${src_folder}_test)
diff --git a/src/lib/tests/shm_du_map_test_create.c b/src/lib/tests/shm_du_map_test_create.c
new file mode 100644
index 00000000..dcf3bd45
--- /dev/null
+++ b/src/lib/tests/shm_du_map_test_create.c
@@ -0,0 +1,51 @@
+/*
+ * Ouroboros - Copyright (C) 2016
+ *
+ * Test of the Shared Memory Map
+ *
+ * Dimitri Staessens <[email protected]>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+ */
+
+#include <ouroboros/shm_du_map.h>
+#include <ouroboros/common.h>
+#include <sys/mman.h>
+
+int shm_du_map_test_create(int argc, char ** argv)
+{
+ struct shm_du_map * dum;
+ struct shm_du_map * dum2;
+
+ shm_unlink(SHM_DU_MAP_FILENAME);
+
+ dum = shm_du_map_create();
+
+ if (dum == NULL)
+ return -1;
+
+ dum2 = shm_du_map_open();
+
+ if (dum2 == NULL) {
+ shm_du_map_close(dum);
+ return 1;
+ }
+
+ shm_du_map_close(dum2);
+
+ shm_du_map_close(dum);
+
+ return 0; /* tests succeeded */
+}
diff --git a/src/lib/tests/shm_du_map_test_du_buff_create.c b/src/lib/tests/shm_du_map_test_du_buff_create.c
new file mode 100644
index 00000000..6787a1bc
--- /dev/null
+++ b/src/lib/tests/shm_du_map_test_du_buff_create.c
@@ -0,0 +1,95 @@
+/*
+ * Ouroboros - Copyright (C) 2016
+ *
+ * Test of the Shared Memory Map
+ *
+ * Dimitri Staessens <[email protected]>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+ */
+
+#include <ouroboros/shm_du_map.h>
+#include <sys/mman.h>
+
+
+#define SHM_DU_BLOCK_DATA_SIZE (SHM_DU_BUFF_BLOCK_SIZE - 3 * sizeof(long))
+#define TEST_BUFF_SIZE (16 * SHM_DU_BLOCK_DATA_SIZE)
+
+#define MAX(a,b) (a > b ? a : b)
+#define MIN(a,b) (a < b ? a : b)
+
+int shm_du_map_test_du_buff_create(int argc, char ** argv)
+{
+ struct shm_du_map * dum;
+ struct shm_du_map * dum2;
+
+ int i, j, k;
+ int i_inc, j_inc, k_inc;
+
+ uint8_t bits[TEST_BUFF_SIZE];
+
+ shm_unlink(SHM_DU_MAP_FILENAME);
+
+ dum = shm_du_map_create();
+
+ if (dum == NULL)
+ return -1;
+
+ for (i = 0; i < TEST_BUFF_SIZE; i++)
+ bits[i] = 0;
+
+ i_inc = MAX(1, SHM_DU_BLOCK_DATA_SIZE / 4);
+ j_inc = MAX(1, SHM_DU_BLOCK_DATA_SIZE / 8);
+ k_inc = MAX(1, SHM_DU_BLOCK_DATA_SIZE / 16);
+
+ for (i = SHM_DU_BLOCK_DATA_SIZE / 4;
+ i <= TEST_BUFF_SIZE;
+ i += i_inc) {
+ for (j = 0; j < i; j += j_inc) {
+ for (k = 0; k < i - j; k += k_inc) {
+ if (k > SHM_DU_BLOCK_DATA_SIZE)
+ continue;
+
+ if (i - (j + k) > SHM_DU_BLOCK_DATA_SIZE)
+ continue;
+
+ struct shm_du_buff * dub = shm_create_du_buff(
+ dum,
+ i,
+ k,
+ bits,
+ j);
+ if (dub == NULL) {
+ shm_du_map_close(dum);
+ return -1;
+ }
+ shm_release_du_buff(dum);
+ }
+ }
+ }
+
+ dum2 = shm_du_map_open();
+
+ if (dum2 == NULL) {
+ shm_du_map_close(dum);
+ return 1;
+ }
+
+ shm_du_map_close(dum2);
+
+ shm_du_map_close(dum);
+
+ return 0; /* tests succeeded */
+}
diff --git a/src/lib/tests/shm_du_map_test_prod_cons.c b/src/lib/tests/shm_du_map_test_prod_cons.c
new file mode 100644
index 00000000..ff9dfb0a
--- /dev/null
+++ b/src/lib/tests/shm_du_map_test_prod_cons.c
@@ -0,0 +1,153 @@
+/*
+ * Ouroboros - Copyright (C) 2016
+ *
+ * Test of the Shared Memory Map
+ *
+ * Dimitri Staessens <[email protected]>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
+ */
+
+#include <ouroboros/shm_du_map.h>
+#include <sys/types.h>
+#include <stdlib.h>
+#include <time.h>
+#include <pthread.h>
+#include "shm_du_map.c"
+
+#define TEST_BUFF_SIZE (SHM_DU_BLOCK_DATA_SIZE)
+
+#define MAX(a,b) (a > b ? a : b)
+#define MIN(a,b) (a < b ? a : b)
+
+int sync;
+
+void * produce()
+{
+ struct shm_du_map * dum;
+ long test_buf_size = 0;
+ uint8_t * test_values;
+ int headspace;
+ int tailspace;
+ long i;
+ long bytes_written = 0;
+ struct timespec starttime;
+ struct timespec stoptime;
+ double elapsed;
+
+ dum = shm_du_map_open();
+ if (dum == NULL)
+ return (void *)-1;
+
+ srand(time(NULL));
+
+ test_values = malloc (sizeof *test_values * TEST_BUFF_SIZE);
+ for (i = 0; i < TEST_BUFF_SIZE; i++)
+ test_values[i] = 170;
+
+ clock_gettime(CLOCK_MONOTONIC, &starttime);
+ for (i = 0; i < SHM_BLOCKS_IN_MAP; i++) {
+ struct shm_du_buff * sdb;
+ size_t len;
+
+ test_buf_size = TEST_BUFF_SIZE;
+
+ headspace = 32;
+ tailspace = 8;
+
+ len = test_buf_size - (headspace + tailspace);
+
+ sdb = shm_create_du_buff(dum,
+ test_buf_size,
+ headspace,
+ test_values,
+ len);
+
+ if (sdb != NULL) {
+ bytes_written += len;
+ }
+ else {
+ sync = -2;
+ break;
+ }
+ }
+
+ clock_gettime(CLOCK_MONOTONIC, &stoptime);
+ elapsed =(stoptime.tv_sec + stoptime.tv_nsec / 1000000000.0) -
+ (starttime.tv_sec + starttime.tv_nsec / 1000000000.0);
+ LOG_INFO("%ld bytes written in %.1lf ms = %lf Gb/s",
+ bytes_written,
+ elapsed * 1000.0,
+ bytes_written * 8 / (elapsed * 1000000000));
+
+ free(test_values);
+ shm_du_map_close(dum);
+
+ sync = -1;
+
+ return 0;
+}
+
+void * consume()
+{
+ struct shm_du_map * dum;
+
+ struct timespec ts;
+
+ ts.tv_sec = 0;
+ ts.tv_nsec = 1000;
+
+ dum = shm_du_map_open();
+
+ if (dum == NULL)
+ pthread_exit((void *) -1);
+
+ while (!sync) {
+ while (!shm_release_du_buff(dum));
+ nanosleep(&ts, NULL);
+ }
+
+ shm_du_map_close(dum);
+
+ return 0;
+}
+
+int shm_du_map_test_prod_cons(int argc, char ** argv)
+{
+ struct shm_du_map * dum;
+
+ int res1;
+
+ pthread_t producer;
+ pthread_t consumer;
+ shm_unlink(SHM_DU_MAP_FILENAME);
+
+ dum = shm_du_map_create();
+
+ if (dum == NULL)
+ return -1;
+
+ sync = 0;
+
+ res1 = (int) pthread_create(&producer, NULL, produce, NULL);
+ pthread_create(&consumer, NULL, consume, NULL);
+
+ pthread_join(producer, NULL);
+ pthread_join(consumer, NULL);
+
+ shm_du_map_close(dum);
+
+ return res1;
+}
generated by cgit v1.2.3 (git 2.39.1) at 2024-11-28 22:29:34 +0000