فهرست منبع

examples: add savl_test

ambrop7 13 سال پیش
والد
کامیت
0fc0ea39d8
3فایلهای تغییر یافته به همراه125 افزوده شده و 0 حذف شده
  1. 3 0
      examples/CMakeLists.txt
  2. 115 0
      examples/savl_test.c
  3. 7 0
      examples/savl_test_tree.h

+ 3 - 0
examples/CMakeLists.txt

@@ -23,6 +23,9 @@ if (BUILDING_SECURITY)
     add_executable(bavl_test bavl_test.c)
     target_link_libraries(bavl_test security)
 
+    add_executable(savl_test savl_test.c)
+    target_link_libraries(savl_test security)
+
     add_executable(bencryption_bench bencryption_bench.c)
     target_link_libraries(bencryption_bench system security)
 endif ()

+ 115 - 0
examples/savl_test.c

@@ -0,0 +1,115 @@
+/**
+ * @file bavl_test.c
+ * @author Ambroz Bizjak <ambrop7@gmail.com>
+ * 
+ * @section LICENSE
+ * 
+ * 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. Neither the name of the author nor the
+ *    names of its contributors may be used to endorse or promote products
+ *    derived from this software without specific prior written permission.
+ * 
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS 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 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 <stdlib.h>
+
+#include <misc/debug.h>
+#include <misc/balloc.h>
+#include <misc/compare.h>
+#include <structure/SAvl.h>
+#include <security/BRandom.h>
+
+struct mynode;
+
+#include "savl_test_tree.h"
+#include <structure/SAvl_decl.h>
+
+struct mynode {
+    int used;
+    int num;
+    MyTreeNode tree_node;
+};
+
+#include "savl_test_tree.h"
+#include <structure/SAvl_impl.h>
+
+static void verify (MyTree *tree)
+{
+    printf("Verifying...\n");
+    MyTree_Verify(tree, 0);
+}
+
+int main (int argc, char **argv)
+{
+    int num_nodes;
+    int num_random_delete;
+    
+    if (argc != 3 || (num_nodes = atoi(argv[1])) <= 0 || (num_random_delete = atoi(argv[2])) < 0) {
+        fprintf(stderr, "Usage: %s <num> <numrandomdelete>\n", (argc > 0 ? argv[0] : NULL));
+        return 1;
+    }
+    
+    struct mynode *nodes = BAllocArray(num_nodes, sizeof(*nodes));
+    ASSERT_FORCE(nodes)
+    
+    int *values_ins = BAllocArray(num_nodes, sizeof(int));
+    ASSERT_FORCE(values_ins)
+    
+    int *values = BAllocArray(num_random_delete, sizeof(int));
+    ASSERT_FORCE(values)
+    
+    MyTree tree;
+    MyTree_Init(&tree);
+    verify(&tree);
+    
+    printf("Inserting random values...\n");
+    BRandom_randomize((uint8_t *)values_ins, num_nodes * sizeof(int));
+    for (int i = 0; i < num_nodes; i++) {
+        nodes[i].num = values_ins[i];
+        if (MyTree_Insert(&tree, 0, &nodes[i], NULL)) {
+            nodes[i].used = 1;
+        } else {
+            nodes[i].used = 0;
+            printf("Insert collision!\n");
+        }
+    }
+    verify(&tree);
+    
+    printf("Removing random entries...\n");
+    int removed = 0;
+    BRandom_randomize((uint8_t *)values, num_random_delete * sizeof(int));
+    for (int i = 0; i < num_random_delete; i++) {
+        int index = (((unsigned int *)values)[i] % num_nodes);
+        struct mynode *node = nodes + index;
+        if (node->used) {
+            MyTree_Remove(&tree, 0, node);
+            node->used = 0;
+            removed++;
+        }
+    }
+    verify(&tree);
+    
+    printf("Removed %d entries\n", removed);
+    
+    BFree(nodes);
+    BFree(values_ins);
+    BFree(values);
+    
+    return 0;
+}

+ 7 - 0
examples/savl_test_tree.h

@@ -0,0 +1,7 @@
+#define SAVL_PARAM_NAME MyTree
+#define SAVL_PARAM_FEATURE_COUNTS 0
+#define SAVL_PARAM_FEATURE_NOKEYS 1
+#define SAVL_PARAM_TYPE_ENTRY struct mynode
+#define SAVL_PARAM_TYPE_ARG int
+#define SAVL_PARAM_FUN_COMPARE_ENTRIES(arg, entry1, entry2) B_COMPARE((entry1)->num, (entry2)->num)
+#define SAVL_PARAM_MEMBER_NODE tree_node