[sanitizer] Extract check_mem_is_good into header

[vitalybuka]

make sure check works even with NDEBUG

compiler-rt/test/sanitizer_common/TestCases/Linux/open_memstream.cpp

 // RUN: %clangxx -m64 -O0 -g -xc++ %s -o %t && %run %t
 // RUN: %clangxx -m64 -O3 -g -xc++ %s -o %t && %run %t
 // REQUIRES: x86_64-target-arch
 
 #include <assert.h>
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 
-#ifndef __has_feature
-#define __has_feature(x) 0
-#endif
-
-#if __has_feature(memory_sanitizer)
-#include <sanitizer/msan_interface.h>
-static void check_mem_is_good(void *p, size_t s) {
-  __msan_check_mem_is_initialized(p, s);
-}
-#elif __has_feature(address_sanitizer)
-#include <sanitizer/asan_interface.h>
-static void check_mem_is_good(void *p, size_t s) {
-  assert(__asan_region_is_poisoned(p, s) == 0);
-}
-#else
-static void check_mem_is_good(void *p, size_t s) {}
-#endif
+#include "sanitizer_common/sanitizer_specific.h"
 
 static void run(bool flush) {
   char *buf;
   size_t buf_len;
   fprintf(stderr, " &buf %p, &buf_len %p\n", &buf, &buf_len);
   FILE *fp = open_memstream(&buf, &buf_len);
   fprintf(fp, "hello");
   if (flush) {

compiler-rt/test/sanitizer_common/lit.common.cfg.py

if default_tool_options_str:
   default_tool_options_str += ':'
 
 extra_link_flags = []
 
 if config.host_os in ['Linux']:
   extra_link_flags += ["-ldl"]
 
 clang_cflags = config.debug_info_flags + tool_cflags + [config.target_cflags]
+clang_cflags += ["-I%s" % os.path.dirname(os.path.dirname(__file__))]
 clang_cflags += extra_link_flags
 clang_cxxflags = config.cxx_mode_flags + clang_cflags
 
 def build_invocation(compile_flags):
   return " " + " ".join([config.clang] + compile_flags) + " "
 
 config.substitutions.append( ("%clang ", build_invocation(clang_cflags)) )
 config.substitutions.append( ("%clangxx ", build_invocation(clang_cxxflags)) )

compiler-rt/test/sanitizer_common/sanitizer_specific.h

+#ifndef __SANITIZER_COMMON_SANITIZER_SPECIFIC_H__
+#define __SANITIZER_COMMON_SANITIZER_SPECIFIC_H__
+
+#ifndef __has_feature
+#  define __has_feature(x) 0
+#endif
+
+#if __has_feature(memory_sanitizer)
+#  include <sanitizer/msan_interface.h>
+static void check_mem_is_good(void *p, size_t s) {
+  __msan_check_mem_is_initialized(p, s);
+}
+#elif __has_feature(address_sanitizer)
+#  include <sanitizer/asan_interface.h>
+#  include <stdlib.h>
+static void check_mem_is_good(void *p, size_t s) {
+  if (__asan_region_is_poisoned(p, s))
+    abort();
+}
+#else
+static void check_mem_is_good(void *p, size_t s) {}
+#endif
+
+#endif // __SANITIZER_COMMON_SANITIZER_SPECIFIC_H__
 No newline at end of file