msvcrt: Skip small-block heap tests if _set_sbh_threshold is not available.

Wine-Bug: https://bugs.winehq.org/show_bug.cgi?id=49181
Signed-off-by: Piotr Caban <piotr@codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard@winehq.org>
This commit is contained in:
Piotr Caban 2021-11-19 13:46:12 +01:00 committed by Alexandre Julliard
parent c90d72588e
commit acc7907dc9
1 changed files with 26 additions and 15 deletions

View File

@ -23,11 +23,13 @@
#include <errno.h> #include <errno.h>
#include "wine/test.h" #include "wine/test.h"
static void (__cdecl *p_aligned_free)(void*) = NULL; static void (__cdecl *p_aligned_free)(void*);
static void * (__cdecl *p_aligned_malloc)(size_t,size_t) = NULL; static void * (__cdecl *p_aligned_malloc)(size_t,size_t);
static void * (__cdecl *p_aligned_offset_malloc)(size_t,size_t,size_t) = NULL; static void * (__cdecl *p_aligned_offset_malloc)(size_t,size_t,size_t);
static void * (__cdecl *p_aligned_realloc)(void*,size_t,size_t) = NULL; static void * (__cdecl *p_aligned_realloc)(void*,size_t,size_t);
static void * (__cdecl *p_aligned_offset_realloc)(void*,size_t,size_t,size_t) = NULL; static void * (__cdecl *p_aligned_offset_realloc)(void*,size_t,size_t,size_t);
static int (__cdecl *p__set_sbh_threshold)(size_t);
static size_t (__cdecl *p__get_sbh_threshold)(void);
static void test_aligned_malloc(unsigned int size, unsigned int alignment) static void test_aligned_malloc(unsigned int size, unsigned int alignment)
{ {
@ -415,28 +417,37 @@ static void test_aligned(void)
static void test_sbheap(void) static void test_sbheap(void)
{ {
HMODULE msvcrt = GetModuleHandleA("msvcrt.dll");
void *mem; void *mem;
int threshold; int threshold;
p__set_sbh_threshold = (void*)GetProcAddress(msvcrt, "_set_sbh_threshold");
p__get_sbh_threshold = (void*)GetProcAddress(msvcrt, "_get_sbh_threshold");
if (!p__set_sbh_threshold || !p__get_sbh_threshold)
{
win_skip("_set_sbh_threshold not available\n");
return;
}
if(sizeof(void*) == 8) { if(sizeof(void*) == 8) {
ok(!_set_sbh_threshold(0), "_set_sbh_threshold succeeded\n"); ok(!p__set_sbh_threshold(0), "_set_sbh_threshold succeeded\n");
ok(!_set_sbh_threshold(1000), "_set_sbh_threshold succeeded\n"); ok(!p__set_sbh_threshold(1000), "_set_sbh_threshold succeeded\n");
return; return;
} }
mem = malloc(1); mem = malloc(1);
ok(mem != NULL, "malloc failed\n"); ok(mem != NULL, "malloc failed\n");
ok(_set_sbh_threshold(1), "_set_sbh_threshold failed\n"); ok(p__set_sbh_threshold(1), "_set_sbh_threshold failed\n");
threshold = _get_sbh_threshold(); threshold = p__get_sbh_threshold();
ok(threshold == 16, "threshold = %d\n", threshold); ok(threshold == 16, "threshold = %d\n", threshold);
ok(_set_sbh_threshold(8), "_set_sbh_threshold failed\n"); ok(p__set_sbh_threshold(8), "_set_sbh_threshold failed\n");
threshold = _get_sbh_threshold(); threshold = p__get_sbh_threshold();
ok(threshold == 16, "threshold = %d\n", threshold); ok(threshold == 16, "threshold = %d\n", threshold);
ok(_set_sbh_threshold(1000), "_set_sbh_threshold failed\n"); ok(p__set_sbh_threshold(1000), "_set_sbh_threshold failed\n");
threshold = _get_sbh_threshold(); threshold = p__get_sbh_threshold();
ok(threshold == 1008, "threshold = %d\n", threshold); ok(threshold == 1008, "threshold = %d\n", threshold);
free(mem); free(mem);
@ -449,8 +460,8 @@ static void test_sbheap(void)
ok(mem != NULL, "realloc failed\n"); ok(mem != NULL, "realloc failed\n");
ok(!((UINT_PTR)mem & 0xf), "incorrect alignment (%p)\n", mem); ok(!((UINT_PTR)mem & 0xf), "incorrect alignment (%p)\n", mem);
ok(_set_sbh_threshold(0), "_set_sbh_threshold failed\n"); ok(p__set_sbh_threshold(0), "_set_sbh_threshold failed\n");
threshold = _get_sbh_threshold(); threshold = p__get_sbh_threshold();
ok(threshold == 0, "threshold = %d\n", threshold); ok(threshold == 0, "threshold = %d\n", threshold);
free(mem); free(mem);