X-Git-Url: https://git.saurik.com/apple/xnu.git/blobdiff_plain/060df5ea7c632b1ac8cc8aac1fb59758165c2084..6d2010ae8f7a6078e10b361c6962983bab233e0f:/libkern/kxld/tests/kxld_array_test.c

diff --git a/libkern/kxld/tests/kxld_array_test.c b/libkern/kxld/tests/kxld_array_test.c
new file mode 100644
index 000000000..4791712e1
--- /dev/null
+++ b/libkern/kxld/tests/kxld_array_test.c
@@ -0,0 +1,160 @@
+/*
+ * Copyright (c) 2009 Apple Inc. All rights reserved.
+ *
+ * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
+ * 
+ * This file contains Original Code and/or Modifications of Original Code
+ * as defined in and that are subject to the Apple Public Source License
+ * Version 2.0 (the 'License'). You may not use this file except in
+ * compliance with the License. The rights granted to you under the License
+ * may not be used to create, or enable the creation or redistribution of,
+ * unlawful or unlicensed copies of an Apple operating system, or to
+ * circumvent, violate, or enable the circumvention or violation of, any
+ * terms of an Apple operating system software license agreement.
+ * 
+ * Please obtain a copy of the License at
+ * http://www.opensource.apple.com/apsl/ and read it before using this file.
+ * 
+ * The Original Code and all software distributed under the License are
+ * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
+ * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
+ * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
+ * Please see the License for the specific language governing rights and
+ * limitations under the License.
+ * 
+ * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
+ */
+#include <assert.h>
+#include <limits.h>
+#include <stdlib.h>
+#include <strings.h>
+
+#include <mach/mach_init.h>
+
+#include "kxld_array.h"
+#include "kxld_test.h"
+#include "kxld_util.h"
+
+#define kNumStorageTestItems (u_int) (4 * PAGE_SIZE / sizeof(u_int))
+
+int
+main(int argc __unused, char *argv[] __unused)
+{
+    kern_return_t rval = KERN_FAILURE;
+    KXLDArray array;
+    u_int *item = 0;
+    u_int test_num = 0;
+    u_int idx = 0;
+    u_int titems = 0;
+    u_int storageTestItems[kNumStorageTestItems];
+    u_int i = 0;
+    
+    bzero(&array, sizeof(array));
+
+    kxld_set_logging_callback(kxld_test_log);
+    kxld_set_logging_callback_data("kxld_array_test", NULL);
+
+    kxld_log(0, 0, "%d: Initialize", ++test_num);
+
+    titems = PAGE_SIZE / sizeof(u_int);
+    rval = kxld_array_init(&array, sizeof(u_int), titems);
+    assert(rval == KERN_SUCCESS);
+    assert(array.nitems == titems);
+
+    kxld_log(0, 0, "%d: Get item", ++test_num);
+    idx = 0;
+    item = kxld_array_get_item(&array, idx);
+    assert(item);
+    assert(item == kxld_array_get_slot(&array, idx));
+
+    idx = titems - 1;
+    item = kxld_array_get_item(&array, idx);
+    assert(item);
+    assert(item == kxld_array_get_slot(&array, idx));
+
+    idx = titems;
+    item = kxld_array_get_item(&array, idx);
+    assert(!item);
+    /* We allocated the max number of items that could be stored in a page,
+     * so get_slot() and get_item() are equivalent.
+     */
+    assert(item == kxld_array_get_slot(&array, idx));
+
+    kxld_log(0, 0, "%d: Resize", ++test_num);
+
+    titems = 2 * PAGE_SIZE / sizeof(u_int) + 100;
+    rval = kxld_array_resize(&array, titems);
+    assert(rval == KERN_SUCCESS);
+    assert(array.nitems == titems);
+
+    kxld_log(0, 0, "%d: Get more items", ++test_num);
+    idx = 0;
+    item = kxld_array_get_item(&array, idx);
+    assert(item);
+    assert(item == kxld_array_get_slot(&array, idx));
+
+    idx = titems - 1;
+    item = kxld_array_get_item(&array, idx);
+    assert(item);
+    assert(item == kxld_array_get_slot(&array, idx));
+
+    idx = titems;
+    item = kxld_array_get_item(&array, idx);
+    assert(!item);
+    /* We allocated fewer items than could fit in a page, so get_slot() will
+     * return items even when get_item() does not.  See below for details.
+     */
+    assert(item != kxld_array_get_slot(&array, idx));
+
+    kxld_log(0, 0, "%d: Clear and attempt to get an item", ++test_num);
+    (void) kxld_array_clear(&array);
+    item = kxld_array_get_item(&array, 0);
+    assert(!item);
+
+    kxld_log(0, 0, "%d: Get slot", ++test_num);
+    /* The array allocates its internal storage in pages. Because get_slot()
+     * fetches items based on the allocated size, not the logical size, we
+     * calculate the max items get_slot() can retrieve based on page size.
+     */
+    titems = (u_int) (round_page(titems * sizeof(u_int)) / sizeof(u_int));
+    assert(!item);
+    item = kxld_array_get_slot(&array, 0);
+    assert(item);
+    item = kxld_array_get_slot(&array, titems - 1);
+    assert(item);
+    item = kxld_array_get_slot(&array, titems);
+    assert(!item);
+
+    kxld_log(0, 0, "%d: Reinitialize", ++test_num);
+
+    titems = kNumStorageTestItems;
+    rval = kxld_array_init(&array, sizeof(u_int), titems);
+    assert(rval == KERN_SUCCESS);
+    assert(array.nitems == titems);
+
+    kxld_log(0, 0, "%d: Storage test - %d insertions and finds", 
+        ++test_num, kNumStorageTestItems);
+    for (i = 0; i < titems; ++i) {
+        item = kxld_array_get_item(&array, i);
+        assert(item);
+
+        *item = (u_int) (random() % UINT_MAX);
+        storageTestItems[i] = *item;
+    }
+
+    for (i = 0; i < titems; ++i) {
+        item = kxld_array_get_item(&array, i);
+        assert(item);
+        assert(*item == storageTestItems[i]);
+    }
+
+    (void) kxld_array_deinit(&array);
+
+    kxld_log(0, 0, " ");
+    kxld_log(0, 0, "All tests passed!  Now check for memory leaks...");
+    
+    kxld_print_memory_report();
+
+    return 0;
+}