初始化内容

third/mimalloc/test/testhelper.h

@@ -0,0 +1,98 @@
+/* ----------------------------------------------------------------------------
+Copyright (c) 2018-2020, Microsoft Research, Daan Leijen
+This is free software; you can redistribute it and/or modify it under the
+terms of the MIT license. A copy of the license can be found in the file
+"LICENSE" at the root of this distribution.
+-----------------------------------------------------------------------------*/
+#ifndef TESTHELPER_H_
+#define TESTHELPER_H_
+
+#include <stdbool.h>
+#include <stdio.h>
+#include <errno.h>
+
+// ---------------------------------------------------------------------------
+// Test macros: CHECK(name,predicate) and CHECK_BODY(name,body)
+// ---------------------------------------------------------------------------
+static int ok = 0;
+static int failed = 0;
+
+static bool check_result(bool result, const char* testname, const char* fname, long lineno) {
+  if (!(result)) {
+    failed++;
+    fprintf(stderr,"\n  FAILED: %s: %s:%ld\n", testname, fname, lineno);
+    /* exit(1); */
+  }
+  else {
+    ok++;
+    fprintf(stderr, "ok.\n");
+  }
+  return true;
+}
+
+#define CHECK_BODY(name) \
+  fprintf(stderr,"test: %s...  ", name ); \
+  errno = 0; \
+  for(bool done = false, result = true; !done; done = check_result(result,name,__FILE__,__LINE__))
+
+#define CHECK(name,expr)      CHECK_BODY(name){ result = (expr); }
+
+// Print summary of test. Return value can be directly use as a return value for main().
+static inline int print_test_summary(void)
+{
+  fprintf(stderr,"\n\n---------------------------------------------\n"
+                 "succeeded: %i\n"
+                 "failed   : %i\n\n", ok, failed);
+  return failed;
+}
+
+#endif // TESTHELPER_H_
+
+// ------------------------------------------------------
+// helper to run on threads
+// ------------------------------------------------------
+typedef bool (*mi_thread_fun_t)(void);
+
+bool mi_run_on_thread(mi_thread_fun_t fun);
+
+typedef struct mi_thread_fun_args_s {
+  mi_thread_fun_t fun;
+  bool result;
+} mi_thread_fun_args_t;
+
+#ifdef _WIN32
+#ifndef WIN32_LEAN_AND_MEAN
+#define WIN32_LEAN_AND_MEAN
+#endif
+#ifndef NOMINMAX
+#define NOMINMAX
+#endif
+#include <windows.h>
+static DWORD WINAPI mi_win_thread_entry(LPVOID varg) {
+  mi_thread_fun_args_t* arg = (mi_thread_fun_args_t*)varg;
+  arg->result = arg->fun(); 
+  return 0;
+}
+bool mi_run_on_thread(mi_thread_fun_t fun) {
+  mi_thread_fun_args_t arg = { fun, false };
+  HANDLE thread = CreateThread(NULL, 0, &mi_win_thread_entry, &arg, 0, NULL);
+  if (thread == NULL) return false;
+  WaitForSingleObject(thread, INFINITE);
+  CloseHandle(thread);
+  return arg.result;
+}
+#else
+#include <pthread.h>
+static void* mi_pthread_entry(void* varg) {
+  mi_thread_fun_args_t* arg = (mi_thread_fun_args_t*)varg;
+  arg->result = arg->fun(); 
+  return NULL;
+}
+bool mi_run_on_thread(mi_thread_fun_t fun) {
+  mi_thread_fun_args_t arg = { fun, false };
+  pthread_t thread;
+  if (pthread_create(&thread, NULL, &mi_pthread_entry, &arg) != 0) return false;
+  pthread_join(thread, NULL);
+  return arg.result;
+}
+#endif