port: tests: Add a simple unit test for shm operations

[lttng-ust.git] / tests / unit / libringbuffer / shm.c

diff --git a/tests/unit/libringbuffer/shm.c b/tests/unit/libringbuffer/shm.c
new file mode 100644 (file)
index 0000000..5ea1310
--- /dev/null
+++ b/tests/unit/libringbuffer/shm.c
@@ -0,0 +1,67 @@
+/* Copyright (C) 2020 Michael Jeanson <mjeanson@efficios.com>
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; only
+ * version 2.1 of the License.
+ *
+ * This library 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
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
+ */
+
+#include <assert.h>
+#include <stdio.h>
+#include <string.h>
+#include <sys/stat.h>
+#include <sys/types.h>
+#include <sys/mman.h>
+#include <fcntl.h>
+
+#include <lttng/align.h>
+#include "libringbuffer/shm.h"
+
+#include "tap.h"
+
+#define SHM_PATH "/ust-shm-test"
+
+int main()
+{
+       int shmfd;
+       size_t shmsize = LTTNG_UST_PAGE_SIZE * 10;
+       struct shm_object_table *table;
+       struct shm_object *shmobj;
+       struct shm_ref shm_ref;
+
+       plan_tests(5);
+
+       /* Open a zero byte shm fd */
+       shmfd = shm_open(SHM_PATH, O_RDWR | O_CREAT, S_IRUSR | S_IWUSR);
+       ok(shmfd > 0, "Open a POSIX shm fd");
+
+       /* Create a dummy shm object table to test the allocation function */
+       table = shm_object_table_create(1);
+       ok(table, "Create a shm object table");
+       assert(table);
+
+       /* This function sets the initial size of the shm with ftruncate and zeros it */
+       shmobj = shm_object_table_alloc(table, shmsize, SHM_OBJECT_SHM, shmfd, -1);
+       ok(shmobj, "Allocate the shm object table");
+       assert(shmobj);
+
+       shm_ref = zalloc_shm(shmobj, LTTNG_UST_PAGE_SIZE * 5);
+       ok(shm_ref.index != -1, "Allocate an object in the shm with sufficient space");
+
+       shm_ref = zalloc_shm(shmobj, LTTNG_UST_PAGE_SIZE * 6);
+       ok(shm_ref.index == -1, "Allocate an object in the shm with insufficient space");
+
+       /* Cleanup */
+       shm_object_table_destroy(table, 1);
+
+       return exit_status();
+}