1 files changed, 415 insertions, 0 deletions
diff --git a/test/libelf/tset/elf_rand/rand.m4 b/test/libelf/tset/elf_rand/rand.m4
new file mode 100644
index 000000000000..bdcf436456b5
--- /dev/null
+++ b/test/libelf/tset/elf_rand/rand.m4
@@ -0,0 +1,415 @@
+/*-
+ * Copyright (c) 2019 Joseph Koshy
+ * All rights reserved.
+ *
+ * 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.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``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 OR CONTRIBUTORS 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.
+ *
+ * $Id$
+ */
+
+#include <sys/types.h>
+#include <sys/stat.h>
+
+#include <ar.h>
+#include <libelf.h>
+#include <limits.h>
+#include <unistd.h>
+
+#include "elfts.h"
+#include "tet_api.h"
+
+IC_REQUIRES_VERSION_INIT();
+
+include(`elfts.m4')
+
+/*
+ * The following definitions should match those in `./Makefile'.
+ */
+define(`TP_ARFILE',`"a.ar"')
+define(`TP_NONARCHIVE', `"s1"')
+
+/*
+ * The use of an offset less than SARMAG should fail.
+ */
+void
+tcSeekBelowSarmag(void)
+{
+	Elf *ar;
+	off_t offset;
+	int error, fd, result;
+
+	fd = -1;
+	ar = NULL;
+	result = TET_UNRESOLVED;
+
+	TP_CHECK_INITIALIZATION();
+	TP_ANNOUNCE("elf_rand() fails for an offset less than SARMAG");
+
+	TS_OPEN_FILE(ar, TP_ARFILE, ELF_C_READ, fd);
+
+	result = TET_PASS;
+
+	if ((offset = elf_rand(ar, 1)) != 0) {
+		TP_FAIL("elf_rand() succeeded with offset=%lld",
+		    (unsigned long long) offset);
+	} else if ((error = elf_errno()) != ELF_E_ARGUMENT) {
+		TP_FAIL("unexpected error=%d \"%s\"", error,
+		    elf_errmsg(error));
+	}
+
+	(void) elf_end(ar);
+	(void) close(fd);
+
+	tet_result(result);
+}
+
+/*
+ * The use of an offset greater than the largest valid file offset
+ * should fail.
+ */
+void
+tcSeekMoreThanFileSize(void)
+{
+	Elf *ar;
+	off_t offset;
+	struct stat sb;
+	int error, fd, result;
+
+	result = TET_UNRESOLVED;
+	ar = NULL;
+	fd = -1;
+
+	TP_CHECK_INITIALIZATION();
+	TP_ANNOUNCE("elf_rand() fails with a too-large offset");
+
+	TS_OPEN_FILE(ar, TP_ARFILE, ELF_C_READ, fd);
+
+	/* Get the file size of the archive. */
+	if (fstat(fd, &sb) < 0) {
+		TP_UNRESOLVED("cannot determine the size of \"%s\"",
+		    TP_ARFILE);
+		goto done;
+	}
+
+	result = TET_PASS;
+
+	if ((offset = elf_rand(ar, sb.st_size)) != 0) {
+		TP_FAIL("elf_rand() succeeded with offset=%lld",
+		    (unsigned long long) offset);
+	} else if ((error = elf_errno()) != ELF_E_ARGUMENT) {
+		TP_FAIL("unexpected error=%d \"%s\"", error,
+		    elf_errmsg(error));
+	}
+
+done:
+	if (ar)
+		(void) elf_end(ar);
+	if (fd != -1)
+		(void) close(fd);
+
+	tet_result(result);
+}
+
+/*
+ * An offset with value SARMAG is accepted.
+ */
+void
+tcOffsetEqualsSARMAG(void)
+{
+	Elf *ar;
+	off_t offset;
+	int fd, result;
+
+	fd = -1;
+	ar = NULL;
+	result = TET_UNRESOLVED;
+
+	TP_CHECK_INITIALIZATION();
+	TP_ANNOUNCE("elf_rand(SARMAG) succeeds.");
+
+	TS_OPEN_FILE(ar, TP_ARFILE, ELF_C_READ, fd);
+
+	if ((offset = elf_rand(ar, SARMAG)) != SARMAG) {
+		TP_FAIL("unexpected offset: %lld",
+		    (long long) offset);
+		goto done;
+	}
+
+	result = TET_PASS;
+
+done:
+	if (ar)
+		(void) elf_end(ar);
+	if (fd != -1)
+		(void) close(fd);
+
+	tet_result(result);
+}
+
+/*
+ * Invoking elf_rand() on a non-archive should fail.
+ */
+void
+tcOnNonArchive(void)
+{
+	Elf *e;
+	off_t offset;
+	int error, fd, result;
+
+	fd = -1;
+	e = NULL;
+	result = TET_UNRESOLVED;
+
+	TP_CHECK_INITIALIZATION();
+	TP_ANNOUNCE("elf_rand(non-archive) fails.");
+
+	TS_OPEN_FILE(e, TP_NONARCHIVE, ELF_C_READ, fd);
+
+	if ((offset = elf_rand(e, SARMAG)) != 0 ||
+	    (error = elf_errno()) != ELF_E_ARGUMENT) {
+		TP_FAIL("unexpected offset=%lld",
+			(long long) offset);
+		goto done;
+	}
+
+	result = TET_PASS;
+
+done:
+	if (e)
+		(void) elf_end(e);
+	if (fd != -1)
+		(void) close(fd);
+
+	tet_result(result);
+}
+
+/*
+ * Use an offset value that could cause an overflow.
+ */
+void
+tcOffsetOverflow(void)
+{
+	Elf *ar;
+	off_t offset;
+	uint64_t max_offset;
+	int error, fd, result;
+
+	fd = -1;
+	ar = NULL;
+	result = TET_UNRESOLVED;
+
+	/* A even offset that is close to overflowing. */
+	max_offset = (1ULL << (sizeof(off_t) * CHAR_BIT - 1)) - 2;
+
+	TP_CHECK_INITIALIZATION();
+	TP_ANNOUNCE("offset close to overflowing an off_t");
+
+	TS_OPEN_FILE(ar, TP_ARFILE, ELF_C_READ, fd);
+
+	if ((offset = elf_rand(ar, (off_t) max_offset)) != 0) {
+		TP_FAIL("unexpected success, offset=%lld",
+		    (long long) offset);
+		goto done;
+	}
+
+	result = TET_PASS;
+
+done:
+	if (ar)
+		(void) elf_end(ar);
+	if (fd != -1)
+		(void) close(fd);
+
+	tet_result(result);
+}
+
+/*
+ * Setting the offset to a value that does not correspond to an ar header
+ * should fail.
+ */
+void
+tcOffsetNotCorrespondingToAnArchiveHeader(void)
+{
+	Elf *ar;
+	off_t offset;
+	int error, fd, result;
+
+	fd = -1;
+	ar = NULL;
+	result = TET_UNRESOLVED;
+
+	TP_CHECK_INITIALIZATION();
+	TP_ANNOUNCE("elf_rand(non-header-offset) should fail.");
+
+	TS_OPEN_FILE(ar, TP_ARFILE, ELF_C_READ, fd);
+
+	if ((offset = elf_rand(ar, SARMAG+2)) != 0) {
+		TP_FAIL("unexpected success, offset=%lld",
+		    (long long) offset);
+		goto done;
+	} else if ((error = elf_errno()) != ELF_E_ARCHIVE) {
+		TP_FAIL("unexpected error=%d \"%s\"", error,
+		    elf_errmsg(error));
+		goto done;
+	}
+
+	result = TET_PASS;
+
+done:
+	if (ar)
+		(void) elf_end(ar);
+	if (fd != -1)
+		(void) close(fd);
+
+	tet_result(result);
+}
+
+/*
+ * Odd values of offsets are not legal.
+ */
+void
+tcOddOffset(void)
+{
+	Elf *ar;
+	off_t offset;
+	int error, fd, result;
+
+	fd = -1;
+	ar = NULL;
+	result = TET_UNRESOLVED;
+
+	TP_CHECK_INITIALIZATION();
+	TP_ANNOUNCE("elf_rand(odd-offset-value) should fail.");
+
+	TS_OPEN_FILE(ar, TP_ARFILE, ELF_C_READ, fd);
+
+	if ((offset = elf_rand(ar, SARMAG+1)) != 0) {
+		TP_FAIL("unexpected success, offset=%lld",
+		    (long long) offset);
+		goto done;
+	} else if ((error = elf_errno()) != ELF_E_ARGUMENT) {
+		TP_FAIL("unexpected error=%d \"%s\"", error,
+		    elf_errmsg(error));
+		goto done;
+	}
+
+	result = TET_PASS;
+
+done:
+	if (ar)
+		(void) elf_end(ar);
+	if (fd != -1)
+		(void) close(fd);
+
+	tet_result(result);
+}
+
+/*
+ * Negative offset values are not legal.
+ */
+void
+tcNegativeOffset(void)
+{
+	Elf *ar;
+	off_t offset;
+	int error, fd, result;
+
+	fd = -1;
+	ar = NULL;
+	result = TET_UNRESOLVED;
+
+	TP_CHECK_INITIALIZATION();
+	TP_ANNOUNCE("elf_rand(odd-offset-value) should fail.");
+
+	TS_OPEN_FILE(ar, TP_ARFILE, ELF_C_READ, fd);
+
+	if ((offset = elf_rand(ar, -SARMAG)) != 0) {
+		TP_FAIL("unexpected success, offset=%lld",
+		    (long long) offset);
+		goto done;
+	} else if ((error = elf_errno()) != ELF_E_ARGUMENT) {
+		TP_FAIL("unexpected error=%d \"%s\"", error,
+		    elf_errmsg(error));
+		goto done;
+	}
+
+	result = TET_PASS;
+
+done:
+	if (ar)
+		(void) elf_end(ar);
+	if (fd != -1)
+		(void) close(fd);
+
+	tet_result(result);
+}
+
+
+/* These offsets correspond to archive TP_ARFILE. */
+static off_t valid_offsets[] = {
+	SARMAG,	/* File 's1'. */
+	80	/* File 's2'. */
+};
+
+static const int number_of_offsets =
+    sizeof(valid_offsets) / sizeof(valid_offsets[0]);
+
+/*
+ * Valid offsets should be usable.
+ */
+void
+tcValidOffsets(void)
+{
+	Elf *ar;
+	off_t offset;
+	int i, error, fd, result;
+
+	fd = -1;
+	ar = NULL;
+	result = TET_UNRESOLVED;
+
+	TP_CHECK_INITIALIZATION();
+	TP_ANNOUNCE("elf_rand(valid-offsets) succeeds.");
+
+	TS_OPEN_FILE(ar, TP_ARFILE, ELF_C_READ, fd);
+
+	for (i = 0; i < number_of_offsets; i++) {
+		if ((offset = elf_rand(ar, valid_offsets[i])) !=
+		    valid_offsets[i]) {
+			error = elf_errno();
+			TP_FAIL("failed to seek to offset %lld, error=%d "
+			    "\"%s\"", (long long) offset, error,
+			    elf_errmsg(error));
+			goto done;
+		}
+	}
+
+	result = TET_PASS;
+
+done:
+	if (ar)
+		(void) elf_end(ar);
+	if (fd != -1)
+		(void) close(fd);
+
+	tet_result(result);
+}