test: Add new nlist() unit test

3 files changed, 81 insertions, 0 deletions

diff --git a/test/.gitignore b/test/.gitignore
index 2872d70..0ba3620 100644
--- a/test/.gitignore
+++ b/test/.gitignore
@@ -10,6 +10,7 @@ headers-overlay-gen.c
 headers-system-gen.c
 humanize
 md5
+nlist
 overlay
 proctitle-init
 proctitle
diff --git a/test/Makefile.am b/test/Makefile.am
index b7387c4..4399121 100644
--- a/test/Makefile.am
+++ b/test/Makefile.am
@@ -40,6 +40,7 @@ check_PROGRAMS = \
 	fparseln \
 	fpurge \
 	md5 \
+	nlist \
 	proctitle-init \
 	strl \
 	strmode \
diff --git a/test/nlist.c b/test/nlist.c
new file mode 100644
index 0000000..a0d4046
--- /dev/null
+++ b/test/nlist.c
@@ -0,0 +1,79 @@
+/*
+ * Copyright © 2017 Guillem Jover <guillem@hadrons.org>
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ * 3. The name of the author may not be used to endorse or promote products
+ *    derived from this software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
+ * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
+ * AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL
+ * THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
+ * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
+ * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
+ * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
+ * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
+ * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
+ * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#include <sys/types.h>
+#include <sys/stat.h>
+
+#include <stddef.h>
+#include <stdio.h>
+#include <assert.h>
+#include <fcntl.h>
+#include <nlist.h>
+
+static int data_prv_init = 50;
+extern int data_pub_init;
+extern int data_pub_uninit[2048];
+extern int *data_pub_ptr;
+
+int *data_pub_ptr = &data_prv_init;
+int data_pub_init = 10;
+int data_pub_uninit[2048];
+
+extern int
+func_pub(void);
+
+int
+func_pub(void)
+{
+	return 42;
+}
+
+int
+main(int argc, char **argv)
+{
+	struct nlist nl[] = {
+		{ .n_un.n_name = "main" },
+		{ .n_un.n_name = "func_pub" },
+		{ .n_un.n_name = "data_pub_uninit" },
+		{ .n_un.n_name = "data_pub_init" },
+		{ .n_un.n_name = "data_prv_init" },
+		{ .n_un.n_name = NULL },
+	}, *nlp;
+	int rc;
+
+	assert(*data_pub_ptr == 50);
+
+	rc = nlist(argv[0], nl);
+	assert(rc == 0);
+
+	assert(nl[0].n_type == (N_TEXT | N_EXT));
+	assert(nl[1].n_type == (N_TEXT | N_EXT));
+	assert(nl[2].n_type == (N_BSS | N_EXT));
+	assert(nl[3].n_type == (N_DATA | N_EXT));
+	assert(nl[4].n_type == (N_DATA));
+
+	return 0;
+}