asan_test.cc revision 7dd282c0c7d3ca32ee94092289bd802e16fe9867
1b0ddf227b25388ad836e560c00f0aec84026b0e5Alexey Samsonov//===-- asan_test.cc ------------------------------------------------------===// 21e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// 31e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// The LLVM Compiler Infrastructure 41e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// 51e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// This file is distributed under the University of Illinois Open Source 61e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// License. See LICENSE.TXT for details. 71e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// 81e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//===----------------------------------------------------------------------===// 91e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// 101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// This file is a part of AddressSanitizer, an address sanity checker. 111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// 121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany//===----------------------------------------------------------------------===// 131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include "asan_test_utils.h" 141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 15938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void *malloc_fff(size_t size) { 161e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *res = malloc/**/(size); break_optimization(0); return res;} 17938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void *malloc_eee(size_t size) { 181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *res = malloc_fff(size); break_optimization(0); return res;} 19938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void *malloc_ddd(size_t size) { 201e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *res = malloc_eee(size); break_optimization(0); return res;} 21938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void *malloc_ccc(size_t size) { 221e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *res = malloc_ddd(size); break_optimization(0); return res;} 23938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void *malloc_bbb(size_t size) { 241e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *res = malloc_ccc(size); break_optimization(0); return res;} 25938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void *malloc_aaa(size_t size) { 261e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *res = malloc_bbb(size); break_optimization(0); return res;} 271e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 281e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#ifndef __APPLE__ 29938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void *memalign_fff(size_t alignment, size_t size) { 301e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *res = memalign/**/(alignment, size); break_optimization(0); return res;} 31938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void *memalign_eee(size_t alignment, size_t size) { 321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *res = memalign_fff(alignment, size); break_optimization(0); return res;} 33938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void *memalign_ddd(size_t alignment, size_t size) { 341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *res = memalign_eee(alignment, size); break_optimization(0); return res;} 35938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void *memalign_ccc(size_t alignment, size_t size) { 361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *res = memalign_ddd(alignment, size); break_optimization(0); return res;} 37938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void *memalign_bbb(size_t alignment, size_t size) { 381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *res = memalign_ccc(alignment, size); break_optimization(0); return res;} 39938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void *memalign_aaa(size_t alignment, size_t size) { 401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *res = memalign_bbb(alignment, size); break_optimization(0); return res;} 411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#endif // __APPLE__ 421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 431e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 44938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void free_ccc(void *p) { free(p); break_optimization(0);} 45938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void free_bbb(void *p) { free_ccc(p); break_optimization(0);} 46938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void free_aaa(void *p) { free_bbb(p); break_optimization(0);} 471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 492dcf44991a8ec1ca7c8051eb27c5ff158530bdc3Timur Iskhodzhanovtemplate<typename T> 50938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void uaf_test(int size, int off) { 511e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *p = (char *)malloc_aaa(size); 521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free_aaa(p); 531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (int i = 1; i < 100; i++) 541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free_aaa(malloc_aaa(i)); 551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany fprintf(stderr, "writing %ld byte(s) at %p with offset %d\n", 561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany (long)sizeof(T), p, off); 571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_write((T*)(p + off)); 581e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 591e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 6013ebae606b526399771e9cca1d6a9d24458ad0f1Kostya SerebryanyTEST(AddressSanitizer, HasFeatureAddressSanitizerTest) { 6113ebae606b526399771e9cca1d6a9d24458ad0f1Kostya Serebryany#if defined(__has_feature) && __has_feature(address_sanitizer) 6213ebae606b526399771e9cca1d6a9d24458ad0f1Kostya Serebryany bool asan = 1; 63badab16503e025326a46a45da8faa6fd20e5fc3dKostya Serebryany#elif defined(__SANITIZE_ADDRESS__) 64badab16503e025326a46a45da8faa6fd20e5fc3dKostya Serebryany bool asan = 1; 6513ebae606b526399771e9cca1d6a9d24458ad0f1Kostya Serebryany#else 6613ebae606b526399771e9cca1d6a9d24458ad0f1Kostya Serebryany bool asan = 0; 6713ebae606b526399771e9cca1d6a9d24458ad0f1Kostya Serebryany#endif 6813ebae606b526399771e9cca1d6a9d24458ad0f1Kostya Serebryany EXPECT_EQ(true, asan); 691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 701e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 711e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, SimpleDeathTest) { 721e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(exit(1), ""); 731e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 741e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 751e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, VariousMallocsTest) { 761e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *a = (int*)malloc(100 * sizeof(int)); 771e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany a[50] = 0; 781e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(a); 791e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 801e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *r = (int*)malloc(10); 811e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany r = (int*)realloc(r, 2000 * sizeof(int)); 821e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany r[1000] = 0; 831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(r); 841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *b = new int[100]; 861e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany b[50] = 0; 871e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany delete [] b; 881e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *c = new int; 901e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany *c = 0; 911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany delete c; 921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 93e0e9eea99d7a81a15c4bb15ff2e5ebd92b50817bEvgeniy Stepanov#if !defined(__APPLE__) && !defined(ANDROID) && !defined(__ANDROID__) 941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *pm; 951e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int pm_res = posix_memalign((void**)&pm, kPageSize, kPageSize); 961e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(0, pm_res); 971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(pm); 987274eafceb832e1088f30b71652992787ba8c456Evgeniy Stepanov#endif 991e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1004dd8ba8238b1b698953628affe6e5b2edf3b3e3fAlexey Samsonov#if !defined(__APPLE__) 1011e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *ma = (int*)memalign(kPageSize, kPageSize); 10276e842845c927e92c2c21c41ac51e364fa17528cAlexey Samsonov EXPECT_EQ(0U, (uintptr_t)ma % kPageSize); 1031e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany ma[123] = 0; 1041e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(ma); 1051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#endif // __APPLE__ 1061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 1071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1081e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, CallocTest) { 1091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *a = (int*)calloc(100, sizeof(int)); 1101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(0, a[10]); 1111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(a); 1121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 1131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1141e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, VallocTest) { 1151e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *a = valloc(100); 11676e842845c927e92c2c21c41ac51e364fa17528cAlexey Samsonov EXPECT_EQ(0U, (uintptr_t)a % kPageSize); 1171e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(a); 1181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 1191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1201e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#ifndef __APPLE__ 1211e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, PvallocTest) { 1221e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *a = (char*)pvalloc(kPageSize + 100); 12376e842845c927e92c2c21c41ac51e364fa17528cAlexey Samsonov EXPECT_EQ(0U, (uintptr_t)a % kPageSize); 1241e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany a[kPageSize + 101] = 1; // we should not report an error here. 1251e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(a); 1261e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1271e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany a = (char*)pvalloc(0); // pvalloc(0) should allocate at least one page. 12876e842845c927e92c2c21c41ac51e364fa17528cAlexey Samsonov EXPECT_EQ(0U, (uintptr_t)a % kPageSize); 1291e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany a[101] = 1; // we should not report an error here. 1301e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(a); 1311e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 1321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#endif // __APPLE__ 1331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid *TSDWorker(void *test_key) { 1351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (test_key) { 1361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany pthread_setspecific(*(pthread_key_t*)test_key, (void*)0xfeedface); 1371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 1381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return NULL; 1391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 1401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid TSDDestructor(void *tsd) { 1421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Spawning a thread will check that the current thread id is not -1. 1431e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany pthread_t th; 1442697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_CREATE(&th, NULL, TSDWorker, NULL); 1452697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_JOIN(th, NULL); 1461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 1471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// This tests triggers the thread-specific data destruction fiasco which occurs 1491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// if we don't manage the TSD destructors ourselves. We create a new pthread 1501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// key with a non-NULL destructor which is likely to be put after the destructor 1511e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// of AsanThread in the list of destructors. 1521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// In this case the TSD for AsanThread will be destroyed before TSDDestructor 1531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// is called for the child thread, and a CHECK will fail when we call 1541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// pthread_create() to spawn the grandchild. 1551e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_TSDTest) { 1561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany pthread_t th; 1571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany pthread_key_t test_key; 1581e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany pthread_key_create(&test_key, TSDDestructor); 1592697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_CREATE(&th, NULL, TSDWorker, &test_key); 1602697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_JOIN(th, NULL); 1611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany pthread_key_delete(test_key); 1621e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 1631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1641e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, UAF_char) { 165ca9b5dd9922fb7f6e0ca06dbba7db7398b8997f1Kostya Serebryany const char *uaf_string = "AddressSanitizer:.*heap-use-after-free"; 1661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(uaf_test<U1>(1, 0), uaf_string); 1671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(uaf_test<U1>(10, 0), uaf_string); 1681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(uaf_test<U1>(10, 10), uaf_string); 1691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(uaf_test<U1>(kLargeMalloc, 0), uaf_string); 1701e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(uaf_test<U1>(kLargeMalloc, kLargeMalloc / 2), uaf_string); 1711e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 1721e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 173366984e3aa286f7b4fb45f5c9e703f2768c407edKostya SerebryanyTEST(AddressSanitizer, UAF_long_double) { 1747f4df1a0e1e1b2ace94d4d78b03c222887dbb00cKostya Serebryany if (sizeof(long double) == sizeof(double)) return; 175366984e3aa286f7b4fb45f5c9e703f2768c407edKostya Serebryany long double *p = Ident(new long double[10]); 176366984e3aa286f7b4fb45f5c9e703f2768c407edKostya Serebryany EXPECT_DEATH(Ident(p)[12] = 0, "WRITE of size 10"); 177366984e3aa286f7b4fb45f5c9e703f2768c407edKostya Serebryany EXPECT_DEATH(Ident(p)[0] = Ident(p)[12], "READ of size 10"); 178366984e3aa286f7b4fb45f5c9e703f2768c407edKostya Serebryany delete [] Ident(p); 179366984e3aa286f7b4fb45f5c9e703f2768c407edKostya Serebryany} 180366984e3aa286f7b4fb45f5c9e703f2768c407edKostya Serebryany 181366984e3aa286f7b4fb45f5c9e703f2768c407edKostya Serebryanystruct Packed5 { 182366984e3aa286f7b4fb45f5c9e703f2768c407edKostya Serebryany int x; 183366984e3aa286f7b4fb45f5c9e703f2768c407edKostya Serebryany char c; 184366984e3aa286f7b4fb45f5c9e703f2768c407edKostya Serebryany} __attribute__((packed)); 185366984e3aa286f7b4fb45f5c9e703f2768c407edKostya Serebryany 186366984e3aa286f7b4fb45f5c9e703f2768c407edKostya Serebryany 187366984e3aa286f7b4fb45f5c9e703f2768c407edKostya SerebryanyTEST(AddressSanitizer, UAF_Packed5) { 188366984e3aa286f7b4fb45f5c9e703f2768c407edKostya Serebryany Packed5 *p = Ident(new Packed5[2]); 189366984e3aa286f7b4fb45f5c9e703f2768c407edKostya Serebryany EXPECT_DEATH(p[0] = p[3], "READ of size 5"); 190366984e3aa286f7b4fb45f5c9e703f2768c407edKostya Serebryany EXPECT_DEATH(p[3] = p[0], "WRITE of size 5"); 191366984e3aa286f7b4fb45f5c9e703f2768c407edKostya Serebryany delete [] Ident(p); 192366984e3aa286f7b4fb45f5c9e703f2768c407edKostya Serebryany} 193366984e3aa286f7b4fb45f5c9e703f2768c407edKostya Serebryany 1941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#if ASAN_HAS_BLACKLIST 1951e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, IgnoreTest) { 1961e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *x = Ident(new int); 1971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany delete Ident(x); 1981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany *x = 0; 1991e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 2001e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#endif // ASAN_HAS_BLACKLIST 2011e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2021e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanystruct StructWithBitField { 2031e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int bf1:1; 2041e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int bf2:1; 2051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int bf3:1; 2061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int bf4:29; 2071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}; 2081e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2091e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, BitFieldPositiveTest) { 2101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany StructWithBitField *x = new StructWithBitField; 2111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany delete Ident(x); 2121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(x->bf1 = 0, "use-after-free"); 2131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(x->bf2 = 0, "use-after-free"); 2141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(x->bf3 = 0, "use-after-free"); 2151e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(x->bf4 = 0, "use-after-free"); 21676e842845c927e92c2c21c41ac51e364fa17528cAlexey Samsonov} 2171e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanystruct StructWithBitFields_8_24 { 2191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int a:8; 2201e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int b:24; 2211e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}; 2221e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2231e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, BitFieldNegativeTest) { 2241e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany StructWithBitFields_8_24 *x = Ident(new StructWithBitFields_8_24); 2251e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x->a = 0; 2261e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x->b = 0; 2271e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany delete Ident(x); 2281e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 2291e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2301e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, OutOfMemoryTest) { 2315af39e50366f1aacbebc284f572f08ad1ad07357Kostya Serebryany size_t size = SANITIZER_WORDSIZE == 64 ? (size_t)(1ULL << 48) : (0xf0000000); 2321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(0, realloc(0, size)); 2331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(0, realloc(0, ~Ident(0))); 2341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(0, malloc(size)); 2351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(0, malloc(~Ident(0))); 2361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(0, calloc(1, size)); 2371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(0, calloc(1, ~Ident(0))); 2381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 2391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#if ASAN_NEEDS_SEGV 24158b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenkonamespace { 24258b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko 243ca9b5dd9922fb7f6e0ca06dbba7db7398b8997f1Kostya Serebryanyconst char kUnknownCrash[] = "AddressSanitizer: SEGV on unknown address"; 24458b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenkoconst char kOverriddenHandler[] = "ASan signal handler has been overridden\n"; 24558b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko 2461e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, WildAddressTest) { 2471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *c = (char*)0x123; 24858b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko EXPECT_DEATH(*c = 0, kUnknownCrash); 24958b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko} 25058b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko 25158b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenkovoid my_sigaction_sighandler(int, siginfo_t*, void*) { 25258b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko fprintf(stderr, kOverriddenHandler); 25358b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko exit(1); 25458b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko} 25558b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko 256b0ddf227b25388ad836e560c00f0aec84026b0e5Alexey Samsonovvoid my_signal_sighandler(int signum) { 25758b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko fprintf(stderr, kOverriddenHandler); 25858b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko exit(1); 25958b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko} 26058b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko 26158b017bb286a8f4413a9348d2c87c52de38cff6dAlexander PotapenkoTEST(AddressSanitizer, SignalTest) { 26258b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko struct sigaction sigact; 26358b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko memset(&sigact, 0, sizeof(sigact)); 26458b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko sigact.sa_sigaction = my_sigaction_sighandler; 26558b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko sigact.sa_flags = SA_SIGINFO; 26658b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko // ASan should silently ignore sigaction()... 26758b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko EXPECT_EQ(0, sigaction(SIGSEGV, &sigact, 0)); 26858b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko#ifdef __APPLE__ 26958b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko EXPECT_EQ(0, sigaction(SIGBUS, &sigact, 0)); 27058b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko#endif 27158b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko char *c = (char*)0x123; 27258b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko EXPECT_DEATH(*c = 0, kUnknownCrash); 27358b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko // ... and signal(). 27458b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko EXPECT_EQ(0, signal(SIGSEGV, my_signal_sighandler)); 27558b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko EXPECT_DEATH(*c = 0, kUnknownCrash); 27658b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko} 277b0ddf227b25388ad836e560c00f0aec84026b0e5Alexey Samsonov} // namespace 2781e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#endif 2791e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2801e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanystatic void TestLargeMalloc(size_t size) { 2811e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char buff[1024]; 2821e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany sprintf(buff, "is located 1 bytes to the left of %lu-byte", (long)size); 2831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(Ident((char*)malloc(size))[-1] = 0, buff); 2841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 2851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2861e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, LargeMallocTest) { 28700545a3625aaa765eef2da9b3e197a0e00e94550Evgeniy Stepanov const int max_size = (SANITIZER_WORDSIZE == 32) ? 1 << 26 : 1 << 28; 28800545a3625aaa765eef2da9b3e197a0e00e94550Evgeniy Stepanov for (int i = 113; i < max_size; i = i * 2 + 13) { 2891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany TestLargeMalloc(i); 2901e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 2911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 2921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2931e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, HugeMallocTest) { 294d39a34ee33aae73fdde065f784bdc19f67b91ae2Kostya Serebryany if (SANITIZER_WORDSIZE != 64) return; 295d39a34ee33aae73fdde065f784bdc19f67b91ae2Kostya Serebryany size_t n_megs = 4100; 2961e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany TestLargeMalloc(n_megs << 20); 2971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 2981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 299c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryany#ifndef __APPLE__ 300c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryanyvoid MemalignRun(size_t align, size_t size, int idx) { 301c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryany char *p = (char *)memalign(align, size); 302c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryany Ident(p)[idx] = 0; 303c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryany free(p); 304c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryany} 305c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryany 306c311105fd64219bb96b6467c3707ebda355be87bKostya SerebryanyTEST(AddressSanitizer, memalign) { 307c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryany for (int align = 16; align <= (1 << 23); align *= 2) { 308c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryany size_t size = align * 5; 309c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryany EXPECT_DEATH(MemalignRun(align, size, -1), 310c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryany "is located 1 bytes to the left"); 311c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryany EXPECT_DEATH(MemalignRun(align, size, size + 1), 312c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryany "is located 1 bytes to the right"); 313c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryany } 314c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryany} 315c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryany#endif 316c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryany 3171e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid *ManyThreadsWorker(void *a) { 3181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (int iter = 0; iter < 100; iter++) { 3191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (size_t size = 100; size < 2000; size *= 2) { 3201e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(Ident(malloc(size))); 3211e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 3221e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 3231e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return 0; 3241e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 3251e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 3261e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, ManyThreadsTest) { 3272697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany const size_t kNumThreads = 3282697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany (SANITIZER_WORDSIZE == 32 || ASAN_AVOID_EXPENSIVE_TESTS) ? 30 : 1000; 3291e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany pthread_t t[kNumThreads]; 3301e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (size_t i = 0; i < kNumThreads; i++) { 3312697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_CREATE(&t[i], 0, ManyThreadsWorker, (void*)i); 3321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 3331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (size_t i = 0; i < kNumThreads; i++) { 3342697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_JOIN(t[i], 0); 3351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 3361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 3371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 3381e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, ReallocTest) { 3391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany const int kMinElem = 5; 3401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *ptr = (int*)malloc(sizeof(int) * kMinElem); 3411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany ptr[3] = 3; 3421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (int i = 0; i < 10000; i++) { 3431e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany ptr = (int*)realloc(ptr, 34448ddbef1d051875b2d0b204e8d78300b58d80a85Evgeniy Stepanov (my_rand() % 1000 + kMinElem) * sizeof(int)); 3451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(3, ptr[3]); 3461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 347a3ab1a7da70a5ef111257ba8887920c1fdcd7be5Alexey Samsonov free(ptr); 348a3ab1a7da70a5ef111257ba8887920c1fdcd7be5Alexey Samsonov // Realloc pointer returned by malloc(0). 349a3ab1a7da70a5ef111257ba8887920c1fdcd7be5Alexey Samsonov int *ptr2 = Ident((int*)malloc(0)); 350a3ab1a7da70a5ef111257ba8887920c1fdcd7be5Alexey Samsonov ptr2 = Ident((int*)realloc(ptr2, sizeof(*ptr2))); 351a3ab1a7da70a5ef111257ba8887920c1fdcd7be5Alexey Samsonov *ptr2 = 42; 352a3ab1a7da70a5ef111257ba8887920c1fdcd7be5Alexey Samsonov EXPECT_EQ(42, *ptr2); 353a3ab1a7da70a5ef111257ba8887920c1fdcd7be5Alexey Samsonov free(ptr2); 3541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 3551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 356d91699339fd7e022aef0eaa8f401af0f8942068bAlexey SamsonovTEST(AddressSanitizer, ZeroSizeMallocTest) { 357d91699339fd7e022aef0eaa8f401af0f8942068bAlexey Samsonov // Test that malloc(0) and similar functions don't return NULL. 358d91699339fd7e022aef0eaa8f401af0f8942068bAlexey Samsonov void *ptr = Ident(malloc(0)); 359a180078ee0120c049f9fb495f930053f80f105aaAlexey Samsonov EXPECT_TRUE(NULL != ptr); 360d91699339fd7e022aef0eaa8f401af0f8942068bAlexey Samsonov free(ptr); 361d91699339fd7e022aef0eaa8f401af0f8942068bAlexey Samsonov#if !defined(__APPLE__) && !defined(ANDROID) && !defined(__ANDROID__) 362d91699339fd7e022aef0eaa8f401af0f8942068bAlexey Samsonov int pm_res = posix_memalign(&ptr, 1<<20, 0); 363d91699339fd7e022aef0eaa8f401af0f8942068bAlexey Samsonov EXPECT_EQ(0, pm_res); 364a180078ee0120c049f9fb495f930053f80f105aaAlexey Samsonov EXPECT_TRUE(NULL != ptr); 365d91699339fd7e022aef0eaa8f401af0f8942068bAlexey Samsonov free(ptr); 366d91699339fd7e022aef0eaa8f401af0f8942068bAlexey Samsonov#endif 367a180078ee0120c049f9fb495f930053f80f105aaAlexey Samsonov int *int_ptr = new int[0]; 368d91699339fd7e022aef0eaa8f401af0f8942068bAlexey Samsonov int *int_ptr2 = new int[0]; 369a180078ee0120c049f9fb495f930053f80f105aaAlexey Samsonov EXPECT_TRUE(NULL != int_ptr); 370a180078ee0120c049f9fb495f930053f80f105aaAlexey Samsonov EXPECT_TRUE(NULL != int_ptr2); 371a180078ee0120c049f9fb495f930053f80f105aaAlexey Samsonov EXPECT_NE(int_ptr, int_ptr2); 372d91699339fd7e022aef0eaa8f401af0f8942068bAlexey Samsonov delete[] int_ptr; 373d91699339fd7e022aef0eaa8f401af0f8942068bAlexey Samsonov delete[] int_ptr2; 374d91699339fd7e022aef0eaa8f401af0f8942068bAlexey Samsonov} 375d91699339fd7e022aef0eaa8f401af0f8942068bAlexey Samsonov 3764fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey Samsonov#ifndef __APPLE__ 3774fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey Samsonovstatic const char *kMallocUsableSizeErrorMsg = 378ca9b5dd9922fb7f6e0ca06dbba7db7398b8997f1Kostya Serebryany "AddressSanitizer: attempting to call malloc_usable_size()"; 3794fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey Samsonov 3804fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey SamsonovTEST(AddressSanitizer, MallocUsableSizeTest) { 3814fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey Samsonov const size_t kArraySize = 100; 3824fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey Samsonov char *array = Ident((char*)malloc(kArraySize)); 3834fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey Samsonov int *int_ptr = Ident(new int); 38476e842845c927e92c2c21c41ac51e364fa17528cAlexey Samsonov EXPECT_EQ(0U, malloc_usable_size(NULL)); 3854fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey Samsonov EXPECT_EQ(kArraySize, malloc_usable_size(array)); 3864fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey Samsonov EXPECT_EQ(sizeof(int), malloc_usable_size(int_ptr)); 3874fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey Samsonov EXPECT_DEATH(malloc_usable_size((void*)0x123), kMallocUsableSizeErrorMsg); 3884fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey Samsonov EXPECT_DEATH(malloc_usable_size(array + kArraySize / 2), 3894fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey Samsonov kMallocUsableSizeErrorMsg); 3904fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey Samsonov free(array); 3914fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey Samsonov EXPECT_DEATH(malloc_usable_size(array), kMallocUsableSizeErrorMsg); 3924fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey Samsonov} 3934fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey Samsonov#endif 3944fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey Samsonov 3951e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid WrongFree() { 3961e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *x = (int*)malloc(100 * sizeof(int)); 3971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Use the allocated memory, otherwise Clang will optimize it out. 3981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(x); 3991e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(x + 1); 4001e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 4011e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 4021e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, WrongFreeTest) { 4037dd282c0c7d3ca32ee94092289bd802e16fe9867Alexey Samsonov EXPECT_DEATH(WrongFree(), ASAN_PCRE_DOTALL 4047dd282c0c7d3ca32ee94092289bd802e16fe9867Alexey Samsonov "ERROR: AddressSanitizer: attempting free.*not malloc" 4057dd282c0c7d3ca32ee94092289bd802e16fe9867Alexey Samsonov ".*is located 4 bytes inside of 400-byte region" 4067dd282c0c7d3ca32ee94092289bd802e16fe9867Alexey Samsonov ".*allocated by thread"); 4071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 4081e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 4091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid DoubleFree() { 4101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *x = (int*)malloc(100 * sizeof(int)); 4111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany fprintf(stderr, "DoubleFree: x=%p\n", x); 4121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(x); 4131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(x); 4141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany fprintf(stderr, "should have failed in the second free(%p)\n", x); 4151e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany abort(); 4161e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 4171e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 4181e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DoubleFreeTest) { 41927f4932a8a0845303b33d69ddce096c445002e60Kostya Serebryany EXPECT_DEATH(DoubleFree(), ASAN_PCRE_DOTALL 420ca9b5dd9922fb7f6e0ca06dbba7db7398b8997f1Kostya Serebryany "ERROR: AddressSanitizer: attempting double-free" 42127f4932a8a0845303b33d69ddce096c445002e60Kostya Serebryany ".*is located 0 bytes inside of 400-byte region" 42227f4932a8a0845303b33d69ddce096c445002e60Kostya Serebryany ".*freed by thread T0 here" 42327f4932a8a0845303b33d69ddce096c445002e60Kostya Serebryany ".*previously allocated by thread T0 here"); 4241e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 4251e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 4261e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanytemplate<int kSize> 427938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void SizedStackTest() { 4281e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char a[kSize]; 4291e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *A = Ident((char*)&a); 4301e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (size_t i = 0; i < kSize; i++) 4311e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany A[i] = i; 4321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(A[-1] = 0, ""); 4331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(A[-20] = 0, ""); 4341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(A[-31] = 0, ""); 4351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(A[kSize] = 0, ""); 4361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(A[kSize + 1] = 0, ""); 4371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(A[kSize + 10] = 0, ""); 4381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(A[kSize + 31] = 0, ""); 4391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 4401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 4411e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, SimpleStackTest) { 4421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SizedStackTest<1>(); 4431e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SizedStackTest<2>(); 4441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SizedStackTest<3>(); 4451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SizedStackTest<4>(); 4461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SizedStackTest<5>(); 4471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SizedStackTest<6>(); 4481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SizedStackTest<7>(); 4491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SizedStackTest<16>(); 4501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SizedStackTest<25>(); 4511e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SizedStackTest<34>(); 4521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SizedStackTest<43>(); 4531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SizedStackTest<51>(); 4541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SizedStackTest<62>(); 4551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SizedStackTest<64>(); 4561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SizedStackTest<128>(); 4571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 4581e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 4591e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, ManyStackObjectsTest) { 4601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char XXX[10]; 4611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char YYY[20]; 4621e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char ZZZ[30]; 4631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(XXX); 4641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(YYY); 4651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(Ident(ZZZ)[-1] = 0, ASAN_PCRE_DOTALL "XXX.*YYY.*ZZZ"); 4661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 4671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 468938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE static void Frame0(int frame, char *a, char *b, char *c) { 4691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char d[4] = {0}; 4701e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *D = Ident(d); 4711e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany switch (frame) { 4721e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany case 3: a[5]++; break; 4731e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany case 2: b[5]++; break; 4741e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany case 1: c[5]++; break; 4751e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany case 0: D[5]++; break; 4761e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 4771e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 478938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE static void Frame1(int frame, char *a, char *b) { 4791e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char c[4] = {0}; Frame0(frame, a, b, c); 4801e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany break_optimization(0); 4811e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 482938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE static void Frame2(int frame, char *a) { 4831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char b[4] = {0}; Frame1(frame, a, b); 4841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany break_optimization(0); 4851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 486938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE static void Frame3(int frame) { 4871e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char a[4] = {0}; Frame2(frame, a); 4881e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany break_optimization(0); 4891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 4901e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 4911e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, GuiltyStackFrame0Test) { 4921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(Frame3(0), "located .*in frame <.*Frame0"); 4931e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 4941e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, GuiltyStackFrame1Test) { 4951e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(Frame3(1), "located .*in frame <.*Frame1"); 4961e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 4971e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, GuiltyStackFrame2Test) { 4981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(Frame3(2), "located .*in frame <.*Frame2"); 4991e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 5001e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, GuiltyStackFrame3Test) { 5011e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(Frame3(3), "located .*in frame <.*Frame3"); 5021e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 5031e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 504938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void LongJmpFunc1(jmp_buf buf) { 5051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // create three red zones for these two stack objects. 5061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int a; 5071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int b; 5081e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 5091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *A = Ident(&a); 5101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *B = Ident(&b); 5111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany *A = *B; 5121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany longjmp(buf, 1); 5131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 5141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 515d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya SerebryanyNOINLINE void BuiltinLongJmpFunc1(jmp_buf buf) { 516d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany // create three red zones for these two stack objects. 517d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany int a; 518d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany int b; 519d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany 520d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany int *A = Ident(&a); 521d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany int *B = Ident(&b); 522d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany *A = *B; 523d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany __builtin_longjmp((void**)buf, 1); 524d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany} 525d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany 526938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void UnderscopeLongJmpFunc1(jmp_buf buf) { 5271e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // create three red zones for these two stack objects. 5281e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int a; 5291e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int b; 5301e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 5311e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *A = Ident(&a); 5321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *B = Ident(&b); 5331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany *A = *B; 5341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany _longjmp(buf, 1); 5351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 5361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 537938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void SigLongJmpFunc1(sigjmp_buf buf) { 5381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // create three red zones for these two stack objects. 5391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int a; 5401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int b; 5411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 5421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *A = Ident(&a); 5431e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *B = Ident(&b); 5441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany *A = *B; 5451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany siglongjmp(buf, 1); 5461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 5471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 5481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 549938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void TouchStackFunc() { 5501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int a[100]; // long array will intersect with redzones from LongJmpFunc1. 5511e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *A = Ident(a); 5521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (int i = 0; i < 100; i++) 5531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany A[i] = i*i; 5541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 5551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 5561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Test that we handle longjmp and do not report fals positives on stack. 5571e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, LongJmpTest) { 5581e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany static jmp_buf buf; 5591e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (!setjmp(buf)) { 5601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany LongJmpFunc1(buf); 5611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } else { 5621e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany TouchStackFunc(); 5631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 5641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 5651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 566b989143d0be56496e8d5fcf75969af35a058792aKostya Serebryany#if not defined(__ANDROID__) 567c3a5c173f228cbb15e332e6bbc17c76ebd55d7a8Kostya SerebryanyTEST(AddressSanitizer, BuiltinLongJmpTest) { 568d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany static jmp_buf buf; 569c3a5c173f228cbb15e332e6bbc17c76ebd55d7a8Kostya Serebryany if (!__builtin_setjmp((void**)buf)) { 570d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany BuiltinLongJmpFunc1(buf); 571d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany } else { 572d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany TouchStackFunc(); 573d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany } 574d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany} 575b989143d0be56496e8d5fcf75969af35a058792aKostya Serebryany#endif // not defined(__ANDROID__) 576d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany 5771e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, UnderscopeLongJmpTest) { 5781e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany static jmp_buf buf; 5791e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (!_setjmp(buf)) { 5801e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany UnderscopeLongJmpFunc1(buf); 5811e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } else { 5821e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany TouchStackFunc(); 5831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 5841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 5851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 5861e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, SigLongJmpTest) { 5871e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany static sigjmp_buf buf; 5881e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (!sigsetjmp(buf, 1)) { 5891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SigLongJmpFunc1(buf); 5901e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } else { 5911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany TouchStackFunc(); 5921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 5931e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 5941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 5951e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#ifdef __EXCEPTIONS 596938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void ThrowFunc() { 5971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // create three red zones for these two stack objects. 5981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int a; 5991e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int b; 6001e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 6011e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *A = Ident(&a); 6021e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *B = Ident(&b); 6031e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany *A = *B; 6041e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany ASAN_THROW(1); 6051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 6061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 6071e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, CxxExceptionTest) { 6081e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (ASAN_UAR) return; 6091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // TODO(kcc): this test crashes on 32-bit for some reason... 6105af39e50366f1aacbebc284f572f08ad1ad07357Kostya Serebryany if (SANITIZER_WORDSIZE == 32) return; 6111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany try { 6121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany ThrowFunc(); 6131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } catch(...) {} 6141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany TouchStackFunc(); 6151e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 6161e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#endif 6171e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 6181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid *ThreadStackReuseFunc1(void *unused) { 6191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // create three red zones for these two stack objects. 6201e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int a; 6211e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int b; 6221e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 6231e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *A = Ident(&a); 6241e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *B = Ident(&b); 6251e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany *A = *B; 6261e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany pthread_exit(0); 6271e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return 0; 6281e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 6291e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 6301e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid *ThreadStackReuseFunc2(void *unused) { 6311e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany TouchStackFunc(); 6321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return 0; 6331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 6341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 6351e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, ThreadStackReuseTest) { 6361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany pthread_t t; 6372697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_CREATE(&t, 0, ThreadStackReuseFunc1, 0); 6382697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_JOIN(t, 0); 6392697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_CREATE(&t, 0, ThreadStackReuseFunc2, 0); 6402697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_JOIN(t, 0); 6411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 6421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 643f6607d097833cd53460735c1a5a2fd52b97613ebTimur Iskhodzhanov#if defined(__i386__) || defined(__x86_64__) 6441e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, Store128Test) { 6451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *a = Ident((char*)malloc(Ident(12))); 6461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *p = a; 6471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (((uintptr_t)a % 16) != 0) 6481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany p = a + 8; 6491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany assert(((uintptr_t)p % 16) == 0); 6501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany __m128i value_wide = _mm_set1_epi16(0x1234); 6511e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(_mm_store_si128((__m128i*)p, value_wide), 652ca9b5dd9922fb7f6e0ca06dbba7db7398b8997f1Kostya Serebryany "AddressSanitizer: heap-buffer-overflow"); 653acd5c617b219e0f059620c2a3928d2cc821d4534Kostya Serebryany EXPECT_DEATH(_mm_store_si128((__m128i*)p, value_wide), 6541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "WRITE of size 16"); 6551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(_mm_store_si128((__m128i*)p, value_wide), 6561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "located 0 bytes to the right of 12-byte"); 6571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(a); 6581e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 6591e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#endif 6601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 6616cbfae439b81221d2250ffe0331958f130259755Kostya Serebryanystring RightOOBErrorMessage(int oob_distance, bool is_write) { 6621e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany assert(oob_distance >= 0); 6631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char expected_str[100]; 6641b057b20db71ec6cc2bac460c8b2848f0889d47dKostya Serebryany sprintf(expected_str, ASAN_PCRE_DOTALL 6651b057b20db71ec6cc2bac460c8b2848f0889d47dKostya Serebryany "buffer-overflow.*%s.*located %d bytes to the right", 666ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko is_write ? "WRITE" : "READ", oob_distance); 6671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return string(expected_str); 6681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 6691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 6706cbfae439b81221d2250ffe0331958f130259755Kostya Serebryanystring RightOOBWriteMessage(int oob_distance) { 671ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko return RightOOBErrorMessage(oob_distance, /*is_write*/true); 672ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko} 673ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko 6746cbfae439b81221d2250ffe0331958f130259755Kostya Serebryanystring RightOOBReadMessage(int oob_distance) { 675ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko return RightOOBErrorMessage(oob_distance, /*is_write*/false); 676ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko} 677ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko 6786cbfae439b81221d2250ffe0331958f130259755Kostya Serebryanystring LeftOOBErrorMessage(int oob_distance, bool is_write) { 6791e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany assert(oob_distance > 0); 6801e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char expected_str[100]; 6810b770b48e4ae053a7d0610e1c682405afba58fdaKostya Serebryany sprintf(expected_str, ASAN_PCRE_DOTALL "%s.*located %d bytes to the left", 682ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko is_write ? "WRITE" : "READ", oob_distance); 6831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return string(expected_str); 6841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 6851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 6866cbfae439b81221d2250ffe0331958f130259755Kostya Serebryanystring LeftOOBWriteMessage(int oob_distance) { 687ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko return LeftOOBErrorMessage(oob_distance, /*is_write*/true); 688ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko} 689ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko 6906cbfae439b81221d2250ffe0331958f130259755Kostya Serebryanystring LeftOOBReadMessage(int oob_distance) { 691ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko return LeftOOBErrorMessage(oob_distance, /*is_write*/false); 692ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko} 693ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko 6946cbfae439b81221d2250ffe0331958f130259755Kostya Serebryanystring LeftOOBAccessMessage(int oob_distance) { 695ada9ba14325df9e206f454c50bd1b55ea9ecdbb9Alexander Potapenko assert(oob_distance > 0); 696ada9ba14325df9e206f454c50bd1b55ea9ecdbb9Alexander Potapenko char expected_str[100]; 697ada9ba14325df9e206f454c50bd1b55ea9ecdbb9Alexander Potapenko sprintf(expected_str, "located %d bytes to the left", oob_distance); 698ada9ba14325df9e206f454c50bd1b55ea9ecdbb9Alexander Potapenko return string(expected_str); 699ada9ba14325df9e206f454c50bd1b55ea9ecdbb9Alexander Potapenko} 700ada9ba14325df9e206f454c50bd1b55ea9ecdbb9Alexander Potapenko 7016cbfae439b81221d2250ffe0331958f130259755Kostya Serebryanychar* MallocAndMemsetString(size_t size, char ch) { 70244997c354c7cdb5257d9a41c837b12a6133f2d6bKostya Serebryany char *s = Ident((char*)malloc(size)); 70344997c354c7cdb5257d9a41c837b12a6133f2d6bKostya Serebryany memset(s, ch, size); 70444997c354c7cdb5257d9a41c837b12a6133f2d6bKostya Serebryany return s; 70544997c354c7cdb5257d9a41c837b12a6133f2d6bKostya Serebryany} 7066cbfae439b81221d2250ffe0331958f130259755Kostya Serebryany 7076cbfae439b81221d2250ffe0331958f130259755Kostya Serebryanychar* MallocAndMemsetString(size_t size) { 70844997c354c7cdb5257d9a41c837b12a6133f2d6bKostya Serebryany return MallocAndMemsetString(size, 'z'); 70944997c354c7cdb5257d9a41c837b12a6133f2d6bKostya Serebryany} 71044997c354c7cdb5257d9a41c837b12a6133f2d6bKostya Serebryany 711ca2849c2819b5c7a8771a1e8bc449cf8f5ef6527Alexey Samsonov#if defined(__linux__) && !defined(ANDROID) && !defined(__ANDROID__) 712c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany#define READ_TEST(READ_N_BYTES) \ 713c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany char *x = new char[10]; \ 714c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany int fd = open("/proc/self/stat", O_RDONLY); \ 715c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany ASSERT_GT(fd, 0); \ 716c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany EXPECT_DEATH(READ_N_BYTES, \ 717c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany ASAN_PCRE_DOTALL \ 718c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany "AddressSanitizer: heap-buffer-overflow" \ 719c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany ".* is located 0 bytes to the right of 10-byte region"); \ 720c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany close(fd); \ 721c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany delete [] x; \ 722c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany 7238530e2b953f0b34ecd267a6aba5f155d5c08c5c8Kostya SerebryanyTEST(AddressSanitizer, pread) { 724c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany READ_TEST(pread(fd, x, 15, 0)); 7258530e2b953f0b34ecd267a6aba5f155d5c08c5c8Kostya Serebryany} 7268530e2b953f0b34ecd267a6aba5f155d5c08c5c8Kostya Serebryany 7278530e2b953f0b34ecd267a6aba5f155d5c08c5c8Kostya SerebryanyTEST(AddressSanitizer, pread64) { 728c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany READ_TEST(pread64(fd, x, 15, 0)); 7298530e2b953f0b34ecd267a6aba5f155d5c08c5c8Kostya Serebryany} 7308530e2b953f0b34ecd267a6aba5f155d5c08c5c8Kostya Serebryany 7318530e2b953f0b34ecd267a6aba5f155d5c08c5c8Kostya SerebryanyTEST(AddressSanitizer, read) { 732c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany READ_TEST(read(fd, x, 15)); 733c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany} 734ca2849c2819b5c7a8771a1e8bc449cf8f5ef6527Alexey Samsonov#endif // defined(__linux__) && !defined(ANDROID) && !defined(__ANDROID__) 7358530e2b953f0b34ecd267a6aba5f155d5c08c5c8Kostya Serebryany 7361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// This test case fails 7371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Clang optimizes memcpy/memset calls which lead to unaligned access 7381e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_MemIntrinsicUnalignedAccessTest) { 7391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int size = Ident(4096); 7401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *s = Ident((char*)malloc(size)); 741ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(memset(s + size - 1, 0, 2), RightOOBWriteMessage(0)); 7421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(s); 7431e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 7441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 7451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// TODO(samsonov): Add a test with malloc(0) 7461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// TODO(samsonov): Add tests for str* and mem* functions. 7471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 748938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE static int LargeFunction(bool do_bad_access) { 7491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *x = new int[100]; 7501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[0]++; 7511e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[1]++; 7521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[2]++; 7531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[3]++; 7541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[4]++; 7551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[5]++; 7561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[6]++; 7571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[7]++; 7581e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[8]++; 7591e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[9]++; 7601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 7611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[do_bad_access ? 100 : 0]++; int res = __LINE__; 7621e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 7631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[10]++; 7641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[11]++; 7651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[12]++; 7661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[13]++; 7671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[14]++; 7681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[15]++; 7691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[16]++; 7701e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[17]++; 7711e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[18]++; 7721e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[19]++; 7731e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 7741e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany delete x; 7751e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return res; 7761e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 7771e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 7781e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Test the we have correct debug info for the failing instruction. 7791e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// This test requires the in-process symbolizer to be enabled by default. 7801e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_LargeFunctionSymbolizeTest) { 7811e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int failing_line = LargeFunction(false); 7821e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char expected_warning[128]; 7835a15541fc2bbe46a5ce0cd802b5c1055f2ffb14eKostya Serebryany sprintf(expected_warning, "LargeFunction.*asan_test.*:%d", failing_line); 7841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(LargeFunction(true), expected_warning); 7851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 7861e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 7871e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Check that we unwind and symbolize correctly. 7881e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_MallocFreeUnwindAndSymbolizeTest) { 7891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *a = (int*)malloc_aaa(sizeof(int)); 7901e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany *a = 1; 7911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free_aaa(a); 7921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(*a = 1, "free_ccc.*free_bbb.*free_aaa.*" 7931e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "malloc_fff.*malloc_eee.*malloc_ddd"); 7941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 7951e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 796b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryanystatic bool TryToSetThreadName(const char *name) { 797b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany#if defined(__linux__) && defined(PR_SET_NAME) 798b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany return 0 == prctl(PR_SET_NAME, (unsigned long)name, 0, 0, 0); 799b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany#else 800b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany return false; 801716e2f25123bf9b20fbc6b582803a3929b78b96dKostya Serebryany#endif 802716e2f25123bf9b20fbc6b582803a3929b78b96dKostya Serebryany} 803716e2f25123bf9b20fbc6b582803a3929b78b96dKostya Serebryany 8041e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid *ThreadedTestAlloc(void *a) { 805b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany EXPECT_EQ(true, TryToSetThreadName("AllocThr")); 8061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int **p = (int**)a; 8071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany *p = new int; 8081e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return 0; 8091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 8101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 8111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid *ThreadedTestFree(void *a) { 812b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany EXPECT_EQ(true, TryToSetThreadName("FreeThr")); 8131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int **p = (int**)a; 8141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany delete *p; 8151e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return 0; 8161e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 8171e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 8181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid *ThreadedTestUse(void *a) { 819b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany EXPECT_EQ(true, TryToSetThreadName("UseThr")); 8201e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int **p = (int**)a; 8211e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany **p = 1; 8221e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return 0; 8231e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 8241e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 8251e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid ThreadedTestSpawn() { 8261e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany pthread_t t; 8271e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *x; 8282697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_CREATE(&t, 0, ThreadedTestAlloc, &x); 8292697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_JOIN(t, 0); 8302697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_CREATE(&t, 0, ThreadedTestFree, &x); 8312697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_JOIN(t, 0); 8322697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_CREATE(&t, 0, ThreadedTestUse, &x); 8332697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_JOIN(t, 0); 8341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 8351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 8361e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, ThreadedTest) { 8371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(ThreadedTestSpawn(), 8381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany ASAN_PCRE_DOTALL 8391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "Thread T.*created" 8401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany ".*Thread T.*created" 8411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany ".*Thread T.*created"); 8421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 8431e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 844b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryanyvoid *ThreadedTestFunc(void *unused) { 845b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany // Check if prctl(PR_SET_NAME) is supported. Return if not. 846b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany if (!TryToSetThreadName("TestFunc")) 847b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany return 0; 848716e2f25123bf9b20fbc6b582803a3929b78b96dKostya Serebryany EXPECT_DEATH(ThreadedTestSpawn(), 849716e2f25123bf9b20fbc6b582803a3929b78b96dKostya Serebryany ASAN_PCRE_DOTALL 850a390ece58317a25ef26866cef6753df4b06a3e65Kostya Serebryany "WRITE .*thread T. .UseThr." 851a390ece58317a25ef26866cef6753df4b06a3e65Kostya Serebryany ".*freed by thread T. .FreeThr. here:" 852a390ece58317a25ef26866cef6753df4b06a3e65Kostya Serebryany ".*previously allocated by thread T. .AllocThr. here:" 853b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany ".*Thread T. .UseThr. created by T.*TestFunc" 854b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany ".*Thread T. .FreeThr. created by T" 855b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany ".*Thread T. .AllocThr. created by T" 856716e2f25123bf9b20fbc6b582803a3929b78b96dKostya Serebryany ""); 857b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany return 0; 858b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany} 859b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany 860b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya SerebryanyTEST(AddressSanitizer, ThreadNamesTest) { 861b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany // Run ThreadedTestFunc in a separate thread because it tries to set a 862b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany // thread name and we don't want to change the main thread's name. 863b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany pthread_t t; 864b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany PTHREAD_CREATE(&t, 0, ThreadedTestFunc, 0); 865b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany PTHREAD_JOIN(t, 0); 866716e2f25123bf9b20fbc6b582803a3929b78b96dKostya Serebryany} 867716e2f25123bf9b20fbc6b582803a3929b78b96dKostya Serebryany 8681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#if ASAN_NEEDS_SEGV 8691e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, ShadowGapTest) { 8705af39e50366f1aacbebc284f572f08ad1ad07357Kostya Serebryany#if SANITIZER_WORDSIZE == 32 8711e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *addr = (char*)0x22000000; 8721e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#else 8731e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *addr = (char*)0x0000100000080000; 8741e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#endif 875ca9b5dd9922fb7f6e0ca06dbba7db7398b8997f1Kostya Serebryany EXPECT_DEATH(*addr = 1, "AddressSanitizer: SEGV on unknown"); 8761e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 8771e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#endif // ASAN_NEEDS_SEGV 8781e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 8791e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C" { 880938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE static void UseThenFreeThenUse() { 8811e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *x = Ident((char*)malloc(8)); 8821e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany *x = 1; 8831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free_aaa(x); 8841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany *x = 2; 8851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 8861e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 8871e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 8881e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, UseThenFreeThenUseTest) { 8891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(UseThenFreeThenUse(), "freed by thread"); 8901e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 8911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 8921e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, StrDupTest) { 8931e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(strdup(Ident("123"))); 8941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 8951e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 8961e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Currently we create and poison redzone at right of global variables. 8971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanystatic char static110[110]; 8981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyconst char ConstGlob[7] = {1, 2, 3, 4, 5, 6, 7}; 8991e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanystatic const char StaticConstGlob[3] = {9, 8, 7}; 9001e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 9011e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, GlobalTest) { 9021e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany static char func_static15[15]; 9031e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 9041e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany static char fs1[10]; 9051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany static char fs2[10]; 9061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany static char fs3[10]; 9071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 9081e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany glob5[Ident(0)] = 0; 9091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany glob5[Ident(1)] = 0; 9101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany glob5[Ident(2)] = 0; 9111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany glob5[Ident(3)] = 0; 9121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany glob5[Ident(4)] = 0; 9131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 9141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(glob5[Ident(5)] = 0, 9151e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "0 bytes to the right of global variable.*glob5.* size 5"); 9161e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(glob5[Ident(5+6)] = 0, 9171e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "6 bytes to the right of global variable.*glob5.* size 5"); 9181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(static110); // avoid optimizations 9191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany static110[Ident(0)] = 0; 9201e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany static110[Ident(109)] = 0; 9211e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(static110[Ident(110)] = 0, 9221e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "0 bytes to the right of global variable"); 9231e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(static110[Ident(110+7)] = 0, 9241e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "7 bytes to the right of global variable"); 9251e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 9261e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(func_static15); // avoid optimizations 9271e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany func_static15[Ident(0)] = 0; 9281e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(func_static15[Ident(15)] = 0, 9291e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "0 bytes to the right of global variable"); 9301e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(func_static15[Ident(15 + 9)] = 0, 9311e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "9 bytes to the right of global variable"); 9321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 9331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(fs1); 9341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(fs2); 9351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(fs3); 9361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 9371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // We don't create left redzones, so this is not 100% guaranteed to fail. 9381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // But most likely will. 9391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(fs2[Ident(-1)] = 0, "is located.*of global variable"); 9401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 9411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(Ident(Ident(ConstGlob)[8]), 9421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "is located 1 bytes to the right of .*ConstGlob"); 9431e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(Ident(Ident(StaticConstGlob)[5]), 9441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "is located 2 bytes to the right of .*StaticConstGlob"); 9451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 9461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // call stuff from another file. 9471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany GlobalsTest(0); 9481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 9491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 9501e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, GlobalStringConstTest) { 9511e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany static const char *zoo = "FOOBAR123"; 9521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany const char *p = Ident(zoo); 9531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(Ident(p[15]), "is ascii string 'FOOBAR123'"); 9541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 9551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 956c37ca572876a585b6a910f14a85cc7ba7fc22f20Kostya SerebryanyTEST(AddressSanitizer, FileNameInGlobalReportTest) { 957c37ca572876a585b6a910f14a85cc7ba7fc22f20Kostya Serebryany static char zoo[10]; 958c37ca572876a585b6a910f14a85cc7ba7fc22f20Kostya Serebryany const char *p = Ident(zoo); 959c37ca572876a585b6a910f14a85cc7ba7fc22f20Kostya Serebryany // The file name should be present in the report. 9605a15541fc2bbe46a5ce0cd802b5c1055f2ffb14eKostya Serebryany EXPECT_DEATH(Ident(p[15]), "zoo.*asan_test."); 961c37ca572876a585b6a910f14a85cc7ba7fc22f20Kostya Serebryany} 962c37ca572876a585b6a910f14a85cc7ba7fc22f20Kostya Serebryany 9631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyint *ReturnsPointerToALocalObject() { 9641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int a = 0; 9651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return Ident(&a); 9661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 9671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 968918b18a849cea679bff31a4e0dfb3745302c240aKostya Serebryany#if ASAN_UAR == 1 9691e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, LocalReferenceReturnTest) { 9701e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *(*f)() = Ident(ReturnsPointerToALocalObject); 971918b18a849cea679bff31a4e0dfb3745302c240aKostya Serebryany int *p = f(); 972918b18a849cea679bff31a4e0dfb3745302c240aKostya Serebryany // Call 'f' a few more times, 'p' should still be poisoned. 973918b18a849cea679bff31a4e0dfb3745302c240aKostya Serebryany for (int i = 0; i < 32; i++) 974918b18a849cea679bff31a4e0dfb3745302c240aKostya Serebryany f(); 975ca9b5dd9922fb7f6e0ca06dbba7db7398b8997f1Kostya Serebryany EXPECT_DEATH(*p = 1, "AddressSanitizer: stack-use-after-return"); 976918b18a849cea679bff31a4e0dfb3745302c240aKostya Serebryany EXPECT_DEATH(*p = 1, "is located.*in frame .*ReturnsPointerToALocal"); 9771e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 978918b18a849cea679bff31a4e0dfb3745302c240aKostya Serebryany#endif 9791e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 9801e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanytemplate <int kSize> 981938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE static void FuncWithStack() { 9821e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char x[kSize]; 9831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(x)[0] = 0; 9841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(x)[kSize-1] = 0; 9851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 9861e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 9871e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanystatic void LotsOfStackReuse() { 9881e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int LargeStack[10000]; 9891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(LargeStack)[0] = 0; 9901e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (int i = 0; i < 10000; i++) { 9911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany FuncWithStack<128 * 1>(); 9921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany FuncWithStack<128 * 2>(); 9931e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany FuncWithStack<128 * 4>(); 9941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany FuncWithStack<128 * 8>(); 9951e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany FuncWithStack<128 * 16>(); 9961e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany FuncWithStack<128 * 32>(); 9971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany FuncWithStack<128 * 64>(); 9981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany FuncWithStack<128 * 128>(); 9991e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany FuncWithStack<128 * 256>(); 10001e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany FuncWithStack<128 * 512>(); 10011e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(LargeStack)[0] = 0; 10021e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 10031e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 10041e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 10051e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, StressStackReuseTest) { 10061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany LotsOfStackReuse(); 10071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 10081e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 10091e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, ThreadedStressStackReuseTest) { 10101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany const int kNumThreads = 20; 10111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany pthread_t t[kNumThreads]; 10121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (int i = 0; i < kNumThreads; i++) { 10132697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_CREATE(&t[i], 0, (void* (*)(void *x))LotsOfStackReuse, 0); 10141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 10151e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (int i = 0; i < kNumThreads; i++) { 10162697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_JOIN(t[i], 0); 10171e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 10181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 10191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1020f58f998066db0231e521169d2f50af439ceecb49Kostya Serebryanystatic void *PthreadExit(void *a) { 1021f58f998066db0231e521169d2f50af439ceecb49Kostya Serebryany pthread_exit(0); 10227b7b55e99166511ce1e66bf6b79c46ae4aa60d17Evgeniy Stepanov return 0; 1023f58f998066db0231e521169d2f50af439ceecb49Kostya Serebryany} 1024f58f998066db0231e521169d2f50af439ceecb49Kostya Serebryany 1025f58f998066db0231e521169d2f50af439ceecb49Kostya SerebryanyTEST(AddressSanitizer, PthreadExitTest) { 1026f58f998066db0231e521169d2f50af439ceecb49Kostya Serebryany pthread_t t; 1027f58f998066db0231e521169d2f50af439ceecb49Kostya Serebryany for (int i = 0; i < 1000; i++) { 10282697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_CREATE(&t, 0, PthreadExit, 0); 10292697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_JOIN(t, 0); 1030f58f998066db0231e521169d2f50af439ceecb49Kostya Serebryany } 1031f58f998066db0231e521169d2f50af439ceecb49Kostya Serebryany} 1032f58f998066db0231e521169d2f50af439ceecb49Kostya Serebryany 10331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#ifdef __EXCEPTIONS 1034938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE static void StackReuseAndException() { 10351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int large_stack[1000]; 10361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(large_stack); 10371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany ASAN_THROW(1); 10381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 10391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 10401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// TODO(kcc): support exceptions with use-after-return. 10411e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_StressStackReuseAndExceptionsTest) { 10421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (int i = 0; i < 10000; i++) { 10431e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany try { 10441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany StackReuseAndException(); 10451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } catch(...) { 10461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 10471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 10481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 10491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#endif 10501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 10511e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, MlockTest) { 10521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(0, mlockall(MCL_CURRENT)); 10531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(0, mlock((void*)0x12345, 0x5678)); 10541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(0, munlockall()); 10551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(0, munlock((void*)0x987, 0x654)); 10561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 10571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1058c655cfa8dc0fdf5b521e565f6cbdbf7981fba1b2Kostya Serebryanystruct LargeStruct { 1059c655cfa8dc0fdf5b521e565f6cbdbf7981fba1b2Kostya Serebryany int foo[100]; 1060c655cfa8dc0fdf5b521e565f6cbdbf7981fba1b2Kostya Serebryany}; 1061c655cfa8dc0fdf5b521e565f6cbdbf7981fba1b2Kostya Serebryany 1062c655cfa8dc0fdf5b521e565f6cbdbf7981fba1b2Kostya Serebryany// Test for bug http://llvm.org/bugs/show_bug.cgi?id=11763. 1063c655cfa8dc0fdf5b521e565f6cbdbf7981fba1b2Kostya Serebryany// Struct copy should not cause asan warning even if lhs == rhs. 1064c655cfa8dc0fdf5b521e565f6cbdbf7981fba1b2Kostya SerebryanyTEST(AddressSanitizer, LargeStructCopyTest) { 1065c655cfa8dc0fdf5b521e565f6cbdbf7981fba1b2Kostya Serebryany LargeStruct a; 1066c655cfa8dc0fdf5b521e565f6cbdbf7981fba1b2Kostya Serebryany *Ident(&a) = *Ident(&a); 1067c655cfa8dc0fdf5b521e565f6cbdbf7981fba1b2Kostya Serebryany} 1068c655cfa8dc0fdf5b521e565f6cbdbf7981fba1b2Kostya Serebryany 10691cffef3e39cad4eb33606afbebac78d6637361acAlexey SamsonovATTRIBUTE_NO_ADDRESS_SAFETY_ANALYSIS 10703be19f4fa3519e358fd41a5feda82c39a22ef107Kostya Serebryanystatic void NoAddressSafety() { 10713be19f4fa3519e358fd41a5feda82c39a22ef107Kostya Serebryany char *foo = new char[10]; 10723be19f4fa3519e358fd41a5feda82c39a22ef107Kostya Serebryany Ident(foo)[10] = 0; 10733be19f4fa3519e358fd41a5feda82c39a22ef107Kostya Serebryany delete [] foo; 10743be19f4fa3519e358fd41a5feda82c39a22ef107Kostya Serebryany} 10753be19f4fa3519e358fd41a5feda82c39a22ef107Kostya Serebryany 10763be19f4fa3519e358fd41a5feda82c39a22ef107Kostya SerebryanyTEST(AddressSanitizer, AttributeNoAddressSafetyTest) { 10773be19f4fa3519e358fd41a5feda82c39a22ef107Kostya Serebryany Ident(NoAddressSafety)(); 10783be19f4fa3519e358fd41a5feda82c39a22ef107Kostya Serebryany} 10793be19f4fa3519e358fd41a5feda82c39a22ef107Kostya Serebryany 108029e09223b9d4f53f1be528d6a572b66034c26611Alexey Samsonov// It doesn't work on Android, as calls to new/delete go through malloc/free. 10815d48617936d6415f2037a5e1b9d57f9077ee92efAlexander Potapenko#if !defined(ANDROID) && !defined(__ANDROID__) 1082fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryanystatic string MismatchStr(const string &str) { 1083fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany return string("AddressSanitizer: alloc-dealloc-mismatch \\(") + str; 1084fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany} 1085fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany 10868f0e311767b783db4a15139dddfe7a924fa3724eAlexey SamsonovTEST(AddressSanitizer, AllocDeallocMismatch) { 1087fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany EXPECT_DEATH(free(Ident(new int)), 1088fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany MismatchStr("operator new vs free")); 1089fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany EXPECT_DEATH(free(Ident(new int[2])), 1090fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany MismatchStr("operator new \\[\\] vs free")); 1091fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany EXPECT_DEATH(delete (Ident(new int[2])), 1092fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany MismatchStr("operator new \\[\\] vs operator delete")); 1093fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany EXPECT_DEATH(delete (Ident((int*)malloc(2 * sizeof(int)))), 1094fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany MismatchStr("malloc vs operator delete")); 1095fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany EXPECT_DEATH(delete [] (Ident(new int)), 1096fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany MismatchStr("operator new vs operator delete \\[\\]")); 1097fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany EXPECT_DEATH(delete [] (Ident((int*)malloc(2 * sizeof(int)))), 1098fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany MismatchStr("malloc vs operator delete \\[\\]")); 1099fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany} 110029e09223b9d4f53f1be528d6a572b66034c26611Alexey Samsonov#endif 1101fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany 11021e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// ------------------ demo tests; run each one-by-one ------------- 11031e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// e.g. --gtest_filter=*DemoOOBLeftHigh --gtest_also_run_disabled_tests 11041e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_DemoThreadedTest) { 11051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany ThreadedTestSpawn(); 11061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 11071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 11081e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid *SimpleBugOnSTack(void *x = 0) { 11091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char a[20]; 11101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(a)[20] = 0; 11111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return 0; 11121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 11131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 11141e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_DemoStackTest) { 11151e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SimpleBugOnSTack(); 11161e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 11171e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 11181e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_DemoThreadStackTest) { 11191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany pthread_t t; 11202697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_CREATE(&t, 0, SimpleBugOnSTack, 0); 11212697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_JOIN(t, 0); 11221e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 11231e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 11241e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_DemoUAFLowIn) { 11251e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany uaf_test<U1>(10, 0); 11261e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 11271e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_DemoUAFLowLeft) { 11281e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany uaf_test<U1>(10, -2); 11291e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 11301e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_DemoUAFLowRight) { 11311e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany uaf_test<U1>(10, 10); 11321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 11331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 11341e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_DemoUAFHigh) { 11351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany uaf_test<U1>(kLargeMalloc, 0); 11361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 11371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 11381e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_DemoOOM) { 11395af39e50366f1aacbebc284f572f08ad1ad07357Kostya Serebryany size_t size = SANITIZER_WORDSIZE == 64 ? (size_t)(1ULL << 40) : (0xf0000000); 11401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany printf("%p\n", malloc(size)); 11411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 11421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 11431e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_DemoDoubleFreeTest) { 11441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany DoubleFree(); 11451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 11461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 11471e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_DemoNullDerefTest) { 11481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *a = 0; 11491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(a)[10] = 0; 11501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 11511e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 11521e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_DemoFunctionStaticTest) { 11531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany static char a[100]; 11541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany static char b[100]; 11551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany static char c[100]; 11561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(a); 11571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(b); 11581e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(c); 11591e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(a)[5] = 0; 11601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(b)[105] = 0; 11611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(a)[5] = 0; 11621e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 11631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 11641e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_DemoTooMuchMemoryTest) { 11651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany const size_t kAllocSize = (1 << 28) - 1024; 11661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany size_t total_size = 0; 11671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany while (true) { 11681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *x = (char*)malloc(kAllocSize); 11691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany memset(x, 0, kAllocSize); 11701e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany total_size += kAllocSize; 11710aa04b3f432bc55e7c116d5280685ee3f243be8eKostya Serebryany fprintf(stderr, "total: %ldM %p\n", (long)total_size >> 20, x); 11721e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 11731e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 11741e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 11754eaa17848ed02cae465f10aa5fe095e42dbf0d61Kostya Serebryany// http://code.google.com/p/address-sanitizer/issues/detail?id=66 11760796393f2088e3b75c4110934451160eeb037fabKostya SerebryanyTEST(AddressSanitizer, BufferOverflowAfterManyFrees) { 11774eaa17848ed02cae465f10aa5fe095e42dbf0d61Kostya Serebryany for (int i = 0; i < 1000000; i++) { 11784eaa17848ed02cae465f10aa5fe095e42dbf0d61Kostya Serebryany delete [] (Ident(new char [8644])); 11794eaa17848ed02cae465f10aa5fe095e42dbf0d61Kostya Serebryany } 11804eaa17848ed02cae465f10aa5fe095e42dbf0d61Kostya Serebryany char *x = new char[8192]; 1181ca9b5dd9922fb7f6e0ca06dbba7db7398b8997f1Kostya Serebryany EXPECT_DEATH(x[Ident(8192)] = 0, "AddressSanitizer: heap-buffer-overflow"); 11824eaa17848ed02cae465f10aa5fe095e42dbf0d61Kostya Serebryany delete [] Ident(x); 11834eaa17848ed02cae465f10aa5fe095e42dbf0d61Kostya Serebryany} 11844eaa17848ed02cae465f10aa5fe095e42dbf0d61Kostya Serebryany 11851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 11865b6eab9dc5572a66e3af54ab087255ffa4dd5185Evgeniy Stepanov// Test that instrumentation of stack allocations takes into account 11875b6eab9dc5572a66e3af54ab087255ffa4dd5185Evgeniy Stepanov// AllocSize of a type, and not its StoreSize (16 vs 10 bytes for long double). 11885b6eab9dc5572a66e3af54ab087255ffa4dd5185Evgeniy Stepanov// See http://llvm.org/bugs/show_bug.cgi?id=12047 for more details. 11895b6eab9dc5572a66e3af54ab087255ffa4dd5185Evgeniy StepanovTEST(AddressSanitizer, LongDoubleNegativeTest) { 11905b6eab9dc5572a66e3af54ab087255ffa4dd5185Evgeniy Stepanov long double a, b; 11919b90e95d4dc59c6d97ed73b2470f5e21e7b346d5Kostya Serebryany static long double c; 11925b6eab9dc5572a66e3af54ab087255ffa4dd5185Evgeniy Stepanov memcpy(Ident(&a), Ident(&b), sizeof(long double)); 11939b90e95d4dc59c6d97ed73b2470f5e21e7b346d5Kostya Serebryany memcpy(Ident(&c), Ident(&b), sizeof(long double)); 119476e842845c927e92c2c21c41ac51e364fa17528cAlexey Samsonov} 1195