util: MemoryMap test

Change-Id: I54680685326a85cfd723a47e8aef8d71662c9aeb

src/core/util/CMakeLists.txt

@@ -74,3 +74,7 @@ if (BUILD_LLVM_DISASSEMBLER)
   add_subdirectory(llvmdisassembler)
 endif (BUILD_LLVM_DISASSEMBLER)
 
+### Tests
+add_executable(memorymap-test testing/memorymap-test.cc)
+target_link_libraries(memorymap-test fail-util)
+add_test(NAME memorymap-test WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}/testing COMMAND memorymap-test)

src/core/util/testing/memorymap-test.cc

@@ -0,0 +1,128 @@
+#include <iostream>
+#include <sstream>
+#include <set>
+#include "util/MemoryMap.hpp"
+
+using namespace fail;
+using std::cerr;
+using std::endl;
+
+#define LEN(arr) (sizeof(arr)/sizeof(*arr))
+
+// start, size, start, size, ...
+uint32_t inside[] = { 10, 5, 11, 1, 12, 1, 15, 1, 17, 1, 19, 1, 18, 2, 23, 2, 22, 1, 21, 1 };
+uint32_t outside[] = { 0, 10, 16, 1, 20, 1, 25, 10 };
+
+void test_failed(std::string msg)
+{
+	cerr << "MemoryMap test failed (" << msg << ")!" << endl;
+}
+
+// pass by value intentional
+void test(MemoryMap mm)
+{
+	std::stringstream ss;
+	for (unsigned i = 0; i < LEN(inside); i += 2) {
+		uint32_t start = inside[i], size = inside[i+1];
+		uint32_t end = start + size;
+
+		if (!mm.isMatching(start, size)) {
+			ss << "1 " << start << " " << size;
+			test_failed(ss.str());
+			ss.str("");
+		}
+
+		for (; start < end; ++start) {
+			if (!mm.isMatching(start)) {
+				ss << "2 " << start;
+				test_failed(ss.str());
+				ss.str("");
+			}
+		}
+	}
+	for (unsigned i = 0; i < LEN(outside); i += 2) {
+		uint32_t start = outside[i], size = outside[i+1];
+		uint32_t end = start + size;
+
+		if (mm.isMatching(start, size)) {
+			ss << "3 " << start << " " << size;
+			test_failed(ss.str());
+			ss.str("");
+		}
+
+		for (; start < end; ++start) {
+			if (mm.isMatching(start)) {
+				ss << "4 " << start;
+				test_failed(ss.str());
+				ss.str("");
+			}
+		}
+	}
+	for (MemoryMap::iterator it = mm.begin(); it != mm.end(); ++it) {
+		cerr << *it << " ";
+		bool found = false;
+		for (unsigned i = 0; i < LEN(inside); i += 2) {
+			uint32_t start = inside[i], size = inside[i+1];
+			uint32_t end = start + size;
+			if (*it >= start && *it < end) {
+				found = true;
+				break;
+			}
+		}
+		if (!found) {
+			ss << "5 " << *it;
+			test_failed(ss.str());
+			ss.str("");
+		}
+		found = false;
+		for (unsigned i = 0; i < LEN(outside); i += 2) {
+			uint32_t start = outside[i], size = outside[i+1];
+			uint32_t end = start + size;
+			if (*it >= start && *it < end) {
+				found = true;
+				break;
+			}
+		}
+		if (found) {
+			ss << "6 " << *it;
+			test_failed(ss.str());
+			ss.str("");
+		}
+	}
+	cerr << "Test finished" << endl;
+}
+
+int main()
+{
+	MemoryMap mm;
+	char const *filename_tmp = "tmp.memorymap";
+	char const *filename_test1 = "test1.memorymap";
+	char const *filename_test2 = "test2.memorymap";
+
+	for (unsigned i = 0; i < LEN(inside); i += 2) {
+		mm.add(inside[i], inside[i+1]);
+	}
+	mm.dump(cerr);
+	
+	test(mm);
+
+	mm.writeToFile(filename_tmp);
+	mm.clear();
+	mm.readFromFile(filename_tmp);
+
+	mm.dump(cerr);
+
+	test(mm);
+
+	// intentionally omitting mm.clear() here
+	mm.readFromFile(filename_test1);
+	test(mm);
+
+	mm.clear();
+	mm.readFromFile(filename_test1);
+	test(mm);
+
+	mm.clear();
+	mm.readFromFile(filename_test2);
+	test(mm);
+}

src/core/util/testing/test1.memorymap

@@ -0,0 +1,13 @@
+10	1
+11	1
+12	1
+13	1
+14	1
+15	1
+17	1
+18	1
+19	1
+21	1
+22	1
+23	1
+24	1

src/core/util/testing/test2.memorymap

@@ -0,0 +1,3 @@
+10 6
+17 3
+21 4