From f542c5ebb8068ccd4f9176684eb62183afbe7e5c Mon Sep 17 00:00:00 2001
From: "arseny.kapoulkine"
 <arseny.kapoulkine@99668b35-9821-0410-8761-19e4c4f06640>
Date: Thu, 6 May 2010 20:28:36 +0000
Subject: Integrated changes from unicode branch to trunk

git-svn-id: http://pugixml.googlecode.com/svn/trunk@383 99668b35-9821-0410-8761-19e4c4f06640
---
 tests/allocator.cpp | 89 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 89 insertions(+)
 create mode 100644 tests/allocator.cpp

(limited to 'tests/allocator.cpp')

diff --git a/tests/allocator.cpp b/tests/allocator.cpp
new file mode 100644
index 0000000..7075194
--- /dev/null
+++ b/tests/allocator.cpp
@@ -0,0 +1,89 @@
+#include "allocator.hpp"
+
+#include <string.h>
+
+// Low-level allocation functions
+#if defined(_WIN32) || defined(_WIN64)
+#	ifdef __MWERKS__
+#		pragma ANSI_strict off // disable ANSI strictness to include windows.h
+#		pragma cpp_extensions on // enable some extensions to include windows.h
+#	endif
+
+#	include <windows.h>
+
+namespace
+{
+	const size_t PAGE_SIZE = 4096;
+
+	void* allocate(size_t size)
+	{
+		size_t aligned_size = (size + PAGE_SIZE - 1) & ~(PAGE_SIZE - 1);
+
+		void* ptr = VirtualAlloc(0, aligned_size + PAGE_SIZE, MEM_COMMIT | MEM_RESERVE, PAGE_READWRITE);
+		if (!ptr) return 0;
+
+		void* end = (char*)ptr + aligned_size;
+
+		DWORD old_flags;
+		VirtualProtect(end, PAGE_SIZE, PAGE_NOACCESS, &old_flags);
+
+		return (char*)end - size;
+	}
+
+	void deallocate(void* ptr, size_t size)
+	{
+		size_t aligned_size = (size + PAGE_SIZE - 1) & ~(PAGE_SIZE - 1);
+
+		void* rptr = (char*)ptr + size - aligned_size;
+
+		DWORD old_flags;
+		VirtualProtect(rptr, aligned_size + PAGE_SIZE, PAGE_NOACCESS, &old_flags);
+	}
+}
+#else
+#	include <stdlib.h>
+
+namespace
+{
+	void* allocate(size_t size)
+	{
+		return malloc(size);
+	}
+
+	void deallocate(void* ptr, size_t size)
+	{
+		(void)size;
+
+		free(ptr);
+	}
+}
+#endif
+
+// High-level allocation functions
+void* memory_allocate(size_t size)
+{
+	void* result = allocate(size + sizeof(size_t));
+	if (!result) return 0;
+
+	memcpy(result, &size, sizeof(size_t));
+
+	return (size_t*)result + 1;
+}
+
+size_t memory_size(void* ptr)
+{
+	size_t result;
+	memcpy(&result, (size_t*)ptr - 1, sizeof(size_t));
+
+	return result;
+}
+
+void memory_deallocate(void* ptr)
+{
+	if (!ptr) return;
+
+	size_t size = memory_size(ptr);
+
+	deallocate((size_t*)ptr - 1, size + sizeof(size_t));
+}
+
-- 
cgit v1.2.3