msvcrt/tests: add _aligned_malloc tests

Robert Reif reif at earthlink.net
Fri Jul 27 19:04:25 CDT 2007


Add _aligned_malloc tests.
-------------- next part --------------
diff --git a/dlls/msvcrt/tests/heap.c b/dlls/msvcrt/tests/heap.c
index c46d939..fc3523a 100644
--- a/dlls/msvcrt/tests/heap.c
+++ b/dlls/msvcrt/tests/heap.c
@@ -19,8 +19,321 @@
  */
 
 #include <stdlib.h>
+#include <malloc.h>
+#include <errno.h>
 #include "wine/test.h"
 
+static void (*p_aligned_free)(void*) = NULL;
+static void * (*p_aligned_malloc)(size_t,size_t) = NULL;
+static void * (*p_aligned_offset_malloc)(size_t,size_t,size_t) = NULL;
+static void * (*p_aligned_realloc)(void*,size_t,size_t) = NULL;
+static void * (*p_aligned_offset_realloc)(void*,size_t,size_t,size_t) = NULL;
+
+static void test_aligned_malloc(size_t size, size_t alignment)
+{
+    void *mem;
+
+    mem = p_aligned_malloc(size, alignment);
+
+    if ((alignment & (alignment - 1)) == 0)
+        ok(mem != NULL, "_aligned_malloc(%d, %d) failed\n", size, alignment);
+    else
+        ok(mem == NULL, "_aligned_malloc(%d, %d) should have failed\n", size, alignment);
+
+    if (mem)
+    {
+        ok(((DWORD_PTR)mem & (alignment ? alignment - 1 : 0)) == 0,
+           "_aligned_malloc(%d, %d) not aligned: %p\n", size, alignment, mem);
+        if (winetest_debug > 1)
+        {
+            void *saved;
+            saved = *(void **)((DWORD_PTR)((char *)mem - sizeof(void *)) & ~(sizeof(void *) - 1));
+            trace("_aligned_malloc(%3d, %3d) returns %p, saved = %p, diff = %d\n",
+                  size, alignment, mem, saved, (char *)saved - (char *)mem);
+        }
+        p_aligned_free(mem);
+    }
+    else
+        ok(errno == EINVAL, "_aligned_malloc(%d, %d) errno: %d != %d\n", size, alignment, errno, EINVAL);
+}
+
+static void test_aligned_offset_malloc(size_t size, size_t alignment, size_t offset)
+{
+    void *mem;
+
+    mem = p_aligned_offset_malloc(size, alignment, offset);
+
+    if ((alignment & (alignment - 1)) == 0)
+        if (offset < size)
+            ok(mem != NULL, "_aligned_offset_malloc(%d, %d, %d) failed\n", size, alignment, offset);
+        else
+            ok(errno == EINVAL, "_aligned_offset_malloc(%d, %d, %d) errno: %d != %d\n", size, alignment, offset, errno, EINVAL);
+    else
+        ok(mem == NULL, "_aligned_offset_malloc(%d, %d, %d) should have failed\n", size, alignment, offset);
+
+    if (mem)
+    {
+        ok(((DWORD_PTR)((char *)mem + offset) & (alignment ? alignment - 1 : 0)) == 0,
+           "_aligned_offset_malloc(%d, %d, %d) not aligned: %p\n", size, alignment, offset, mem);
+        if (winetest_debug > 1)
+        {
+            void *saved;
+            saved = *(void **)((DWORD_PTR)((char *)mem - sizeof(void *)) & ~(sizeof(void *) - 1));
+            trace("_aligned_offset_malloc(%3d, %3d, %3d) returns %p, saved = %p, diff = %d\n",
+                  size, alignment, offset, mem, saved, (char *)saved - (char *)mem);
+        }
+        p_aligned_free(mem);
+    }
+    else
+        ok(errno == EINVAL, "_aligned_offset_malloc(%d, %d, %d) errno: %d != %d\n", size, alignment, offset, errno, EINVAL);
+}
+
+static void test_aligned_realloc(size_t size1, size_t size2, size_t alignment)
+{
+    void *mem, *mem1, *mem2;
+
+    mem = p_aligned_malloc(size1, alignment);
+
+    if ((alignment & (alignment - 1)) == 0)
+        ok(mem != NULL, "_aligned_malloc(%d, %d) failed\n", size1, alignment);
+    else
+        ok(mem == NULL, "_aligned_malloc(%d, %d) should have failed\n", size1, alignment);
+
+    if (mem)
+    {
+	mem1 = malloc(size1);
+        if (mem1)
+        {
+            int i;
+            for (i = 0; i < size1; i++)
+                ((char *)mem)[i] = i + 1;
+            memcpy(mem1, mem, size1);
+        }
+
+        ok(((DWORD_PTR)mem & (alignment ? alignment - 1 : 0)) == 0,
+           "_aligned_malloc(%d, %d) not aligned: %p\n", size1, alignment, mem);
+        if (winetest_debug > 1)
+        {
+            void *saved;
+            saved = *(void **)((DWORD_PTR)((char *)mem - sizeof(void *)) & ~(sizeof(void *) - 1));
+            trace("_aligned_malloc(%3d, %3d) returns %p, saved = %p, diff = %d\n",
+                  size1, alignment, mem, saved, (char *)saved - (char *)mem);
+        }
+
+        mem2 = p_aligned_realloc(mem, size2, alignment);
+       
+        ok(mem2 != NULL, "_aligned_realloc(%p, %d, %d) failed\n", mem, size2, alignment);
+
+        if (mem2)
+        {
+            ok(((DWORD_PTR)mem2 & (alignment ? alignment - 1 : 0)) == 0,
+               "_aligned_realloc(%p, %d, %d) not aligned: %p\n", mem, size2, alignment, mem2);
+            if (winetest_debug > 1)
+            {
+                void *saved;
+                saved = *(void **)((DWORD_PTR)((char *)mem2 - sizeof(void *)) & ~(sizeof(void *) - 1));
+                trace("_aligned_realloc(%p, %3d, %3d) returns %p, saved = %p, diff = %d\n",
+                      mem, size2, alignment, mem2, saved, (char *)saved - (char *)mem2);
+            }
+            if (mem1)
+            {
+                ok(memcmp(mem2, mem1, min(size1, size2))==0, "_aligned_realloc(%p, %d, %d) has different data\n", mem, size2, alignment);
+                if (memcmp(mem2, mem1, min(size1, size2)) && winetest_debug > 1)
+                {
+                    int i;
+                    for (i = 0; i < min(size1, size2); i++)
+                    {
+                        if (((char *)mem2)[i] != ((char *)mem1)[i])    
+                            trace("%d: %02x != %02x\n", i, ((char *)mem2)[i] & 0xff, ((char *)mem1)[i] & 0xff);
+                    }
+                }
+            }
+        }
+        else
+            ok(errno == EINVAL, "_aligned_realloc(%p, %d, %d) errno: %d != %d\n", mem, size2, alignment, errno, EINVAL);
+
+        p_aligned_free(mem);
+        free(mem1);
+    }
+    else
+        ok(errno == EINVAL, "_aligned_malloc(%d, %d) errno: %d != %d\n", size1, alignment, errno, EINVAL);
+}
+
+static void test_aligned_offset_realloc(size_t size1, size_t size2, size_t alignment, size_t offset)
+{
+    void *mem, *mem1, *mem2;
+
+    mem = p_aligned_offset_malloc(size1, alignment, offset);
+
+    if ((alignment & (alignment - 1)) == 0)
+        ok(mem != NULL, "_aligned_offset_malloc(%d, %d, %d) failed\n", size1, alignment, offset);
+    else
+        ok(mem == NULL, "_aligned_offset_malloc(%d, %d, %d) should have failed\n", size1, alignment, offset);
+
+    if (mem)
+    {
+	mem1 = malloc(size1);
+        if (mem1)
+        {
+            int i;
+            for (i = 0; i < size1; i++)
+                ((char *)mem)[i] = i + 1;
+            memcpy(mem1, mem, size1);
+        }
+
+        ok(((DWORD_PTR)((char *)mem + offset) & (alignment ? alignment - 1 : 0)) == 0,
+           "_aligned_offset_malloc(%d, %d, %d) not aligned: %p\n", size1, alignment, offset, mem);
+        if (winetest_debug > 1)
+        {
+            void *saved;
+            saved = *(void **)((DWORD_PTR)((char *)mem - sizeof(void *)) & ~(sizeof(void *) - 1));
+            trace("_aligned_offset_malloc(%3d, %3d, %3d) returns %p, saved = %p, diff = %d\n",
+                  size1, alignment, offset, mem, saved, (char *)saved - (char *)mem);
+        }
+
+        mem2 = p_aligned_offset_realloc(mem, size2, alignment, offset);
+       
+        ok(mem2 != NULL, "_aligned_offset_realloc(%p, %d, %d, %d) failed\n", mem, size2, alignment, offset);
+
+        if (mem2)
+        {
+            ok(((DWORD_PTR)((char *)mem + offset) & (alignment ? alignment - 1 : 0)) == 0,
+               "_aligned_offset_realloc(%p, %d, %d, %d) not aligned: %p\n", mem, size2, alignment, offset, mem2);
+            if (winetest_debug > 1)
+            {
+                void *saved;
+                saved = *(void **)((DWORD_PTR)((char *)mem2 - sizeof(void *)) & ~(sizeof(void *) - 1));
+                trace("_aligned_offset_realloc(%p, %3d, %3d, %3d) returns %p, saved = %p, diff = %d\n",
+                      mem, size2, alignment, offset, mem2, saved, (char *)saved - (char *)mem2);
+            }
+            if (mem1)
+            {
+                ok(memcmp(mem2, mem1, min(size1, size2))==0, "_aligned_offset_realloc(%p, %d, %d, %d) has different data\n", mem, size2, alignment, offset);
+                if (memcmp(mem2, mem1, min(size1, size2)) && winetest_debug > 1)
+                {
+                    int i;
+                    for (i = 0; i < min(size1, size2); i++)
+                    {
+                        if (((char *)mem2)[i] != ((char *)mem1)[i])    
+                            trace("%d: %02x != %02x\n", i, ((char *)mem2)[i] & 0xff, ((char *)mem1)[i] & 0xff);
+                    }
+                }
+            }
+        }
+        else
+            ok(errno == EINVAL, "_aligned_offset_realloc(%p, %d, %d, %d) errno: %d != %d\n", mem, size2, alignment, offset, errno, EINVAL);
+
+        p_aligned_free(mem);
+        free(mem1);
+    }
+    else
+        ok(errno == EINVAL, "_aligned_offset_malloc(%d, %d) errno: %d != %d\n", size1, alignment, errno, EINVAL);
+}
+
+static void test_aligned(void)
+{
+    HMODULE msvcrt = GetModuleHandle("msvcrt.dll");
+
+    if (msvcrt == NULL)
+        return;
+
+    p_aligned_free = (void*)GetProcAddress(msvcrt, "_aligned_free");
+    p_aligned_malloc = (void*)GetProcAddress(msvcrt, "_aligned_malloc");
+    p_aligned_offset_malloc = (void*)GetProcAddress(msvcrt, "_aligned_offset_malloc");
+    p_aligned_realloc = (void*)GetProcAddress(msvcrt, "_aligned_realloc");
+    p_aligned_offset_realloc = (void*)GetProcAddress(msvcrt, "_aligned_offset_realloc");
+
+    if (!p_aligned_free || !p_aligned_malloc || !p_aligned_offset_malloc || !p_aligned_realloc || !p_aligned_offset_realloc)
+    {
+        skip("aligned memory tests skipped\n"); 
+        return;
+    }
+    
+    test_aligned_malloc(256, 0);
+    test_aligned_malloc(256, 1);
+    test_aligned_malloc(256, 2);
+    test_aligned_malloc(256, 3);
+    test_aligned_malloc(256, 4);
+    test_aligned_malloc(256, 8);
+    test_aligned_malloc(256, 16);
+    test_aligned_malloc(256, 32);
+    test_aligned_malloc(256, 64);
+    test_aligned_malloc(256, 127);
+    test_aligned_malloc(256, 128);
+
+    test_aligned_offset_malloc(256, 0, 0);
+    test_aligned_offset_malloc(256, 1, 0);
+    test_aligned_offset_malloc(256, 2, 0);
+    test_aligned_offset_malloc(256, 3, 0);
+    test_aligned_offset_malloc(256, 4, 0);
+    test_aligned_offset_malloc(256, 8, 0);
+    test_aligned_offset_malloc(256, 16, 0);
+    test_aligned_offset_malloc(256, 32, 0);
+    test_aligned_offset_malloc(256, 64, 0);
+    test_aligned_offset_malloc(256, 127, 0);
+    test_aligned_offset_malloc(256, 128, 0);
+
+    test_aligned_offset_malloc(256, 0, 4);
+    test_aligned_offset_malloc(256, 1, 4);
+    test_aligned_offset_malloc(256, 2, 4);
+    test_aligned_offset_malloc(256, 3, 4);
+    test_aligned_offset_malloc(256, 4, 4);
+    test_aligned_offset_malloc(256, 8, 4);
+    test_aligned_offset_malloc(256, 16, 4);
+    test_aligned_offset_malloc(256, 32, 4);
+    test_aligned_offset_malloc(256, 64, 4);
+    test_aligned_offset_malloc(256, 127, 4);
+    test_aligned_offset_malloc(256, 128, 4);
+
+    test_aligned_offset_malloc(256, 8, 7);
+    test_aligned_offset_malloc(256, 8, 9);
+    test_aligned_offset_malloc(256, 8, 16);
+    test_aligned_offset_malloc(256, 8, 17);
+    test_aligned_offset_malloc(256, 8, 254);
+    test_aligned_offset_malloc(256, 8, 255);
+    test_aligned_offset_malloc(256, 8, 256);
+    test_aligned_offset_malloc(256, 8, 512);
+
+    test_aligned_realloc(256, 512, 0);
+    test_aligned_realloc(256, 128, 0);
+    test_aligned_realloc(256, 512, 4);
+    test_aligned_realloc(256, 128, 4);
+    test_aligned_realloc(256, 512, 8);
+    test_aligned_realloc(256, 128, 8);
+    test_aligned_realloc(256, 512, 16);
+    test_aligned_realloc(256, 128, 16);
+    test_aligned_realloc(256, 512, 32);
+    test_aligned_realloc(256, 128, 32);
+    test_aligned_realloc(256, 512, 64);
+    test_aligned_realloc(256, 128, 64);
+
+    test_aligned_offset_realloc(256, 512, 0, 0);
+    test_aligned_offset_realloc(256, 128, 0, 0);
+    test_aligned_offset_realloc(256, 512, 4, 0);
+    test_aligned_offset_realloc(256, 128, 4, 0);
+    test_aligned_offset_realloc(256, 512, 8, 0);
+    test_aligned_offset_realloc(256, 128, 8, 0);
+    test_aligned_offset_realloc(256, 512, 16, 0);
+    test_aligned_offset_realloc(256, 128, 16, 0);
+    test_aligned_offset_realloc(256, 512, 32, 0);
+    test_aligned_offset_realloc(256, 128, 32, 0);
+    test_aligned_offset_realloc(256, 512, 64, 0);
+    test_aligned_offset_realloc(256, 128, 64, 0);
+
+    test_aligned_offset_realloc(256, 512, 0, 4);
+    test_aligned_offset_realloc(256, 128, 0, 4);
+    test_aligned_offset_realloc(256, 512, 4, 4);
+    test_aligned_offset_realloc(256, 128, 4, 4);
+    test_aligned_offset_realloc(256, 512, 8, 4);
+    test_aligned_offset_realloc(256, 128, 8, 4);
+    test_aligned_offset_realloc(256, 512, 16, 4);
+    test_aligned_offset_realloc(256, 128, 16, 4);
+    test_aligned_offset_realloc(256, 512, 32, 4);
+    test_aligned_offset_realloc(256, 128, 32, 4);
+    test_aligned_offset_realloc(256, 512, 64, 4);
+    test_aligned_offset_realloc(256, 128, 64, 4);
+}
+
 START_TEST(heap)
 {
     void *mem;
@@ -39,4 +352,6 @@ START_TEST(heap)
     
     mem = realloc(NULL, 0);
     ok(mem != NULL, "memory not (re)allocated for size 0\n");
+
+    test_aligned();
 }


More information about the wine-patches mailing list