asan_test.cc revision 44997c354c7cdb5257d9a41c837b12a6133f2d6b
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 <stdio.h> 141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include <signal.h> 151e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include <stdlib.h> 161e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include <string.h> 17af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany#include <strings.h> 181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include <pthread.h> 191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include <stdint.h> 201e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include <setjmp.h> 211e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include <assert.h> 22eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany#include <algorithm> 231e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 24716e2f25123bf9b20fbc6b582803a3929b78b96dKostya Serebryany#ifdef __linux__ 25716e2f25123bf9b20fbc6b582803a3929b78b96dKostya Serebryany# include <sys/prctl.h> 268530e2b953f0b34ecd267a6aba5f155d5c08c5c8Kostya Serebryany# include <sys/types.h> 278530e2b953f0b34ecd267a6aba5f155d5c08c5c8Kostya Serebryany# include <sys/stat.h> 288530e2b953f0b34ecd267a6aba5f155d5c08c5c8Kostya Serebryany# include <fcntl.h> 297125bb35fb2351040534a194c9c1aa8cb71cfb19Kostya Serebryany#include <unistd.h> 30716e2f25123bf9b20fbc6b582803a3929b78b96dKostya Serebryany#endif 31716e2f25123bf9b20fbc6b582803a3929b78b96dKostya Serebryany 32f6607d097833cd53460735c1a5a2fd52b97613ebTimur Iskhodzhanov#if defined(__i386__) || defined(__x86_64__) 331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include <emmintrin.h> 341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#endif 351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include "asan_test_utils.h" 371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#ifndef __APPLE__ 391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include <malloc.h> 408c745fc58d8aca66db1170b1cdbbc0a7743c20a0Alexander Potapenko#endif 411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#if ASAN_HAS_EXCEPTIONS 431e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany# define ASAN_THROW(x) throw (x) 441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#else 451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany# define ASAN_THROW(x) 461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#endif 471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#include <sys/mman.h> 491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanytypedef uint8_t U1; 511e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanytypedef uint16_t U2; 521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanytypedef uint32_t U4; 531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanytypedef uint64_t U8; 541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanystatic const int kPageSize = 4096; 561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyconst size_t kLargeMalloc = 1 << 24; 581e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 592dcf44991a8ec1ca7c8051eb27c5ff158530bdc3Timur Iskhodzhanovtemplate<typename T> 60938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void asan_write(T *a) { 611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany *a = 0; 621e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 64938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void asan_write_sized_aligned(uint8_t *p, size_t size) { 6576e842845c927e92c2c21c41ac51e364fa17528cAlexey Samsonov EXPECT_EQ(0U, ((uintptr_t)p % size)); 661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (size == 1) asan_write((uint8_t*)p); 671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany else if (size == 2) asan_write((uint16_t*)p); 681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany else if (size == 4) asan_write((uint32_t*)p); 691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany else if (size == 8) asan_write((uint64_t*)p); 701e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 711e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 72938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void *malloc_fff(size_t size) { 731e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *res = malloc/**/(size); break_optimization(0); return res;} 74938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void *malloc_eee(size_t size) { 751e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *res = malloc_fff(size); break_optimization(0); return res;} 76938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void *malloc_ddd(size_t size) { 771e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *res = malloc_eee(size); break_optimization(0); return res;} 78938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void *malloc_ccc(size_t size) { 791e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *res = malloc_ddd(size); break_optimization(0); return res;} 80938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void *malloc_bbb(size_t size) { 811e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *res = malloc_ccc(size); break_optimization(0); return res;} 82938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void *malloc_aaa(size_t size) { 831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *res = malloc_bbb(size); break_optimization(0); return res;} 841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#ifndef __APPLE__ 86938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void *memalign_fff(size_t alignment, size_t size) { 871e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *res = memalign/**/(alignment, size); break_optimization(0); return res;} 88938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void *memalign_eee(size_t alignment, size_t size) { 891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *res = memalign_fff(alignment, size); break_optimization(0); return res;} 90938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void *memalign_ddd(size_t alignment, size_t size) { 911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *res = memalign_eee(alignment, size); break_optimization(0); return res;} 92938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void *memalign_ccc(size_t alignment, size_t size) { 931e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *res = memalign_ddd(alignment, size); break_optimization(0); return res;} 94938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void *memalign_bbb(size_t alignment, size_t size) { 951e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *res = memalign_ccc(alignment, size); break_optimization(0); return res;} 96938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void *memalign_aaa(size_t alignment, size_t size) { 971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *res = memalign_bbb(alignment, size); break_optimization(0); return res;} 981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#endif // __APPLE__ 991e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1001e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 101938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void free_ccc(void *p) { free(p); break_optimization(0);} 102938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void free_bbb(void *p) { free_ccc(p); break_optimization(0);} 103938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void free_aaa(void *p) { free_bbb(p); break_optimization(0);} 1041e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1052dcf44991a8ec1ca7c8051eb27c5ff158530bdc3Timur Iskhodzhanovtemplate<typename T> 106938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void oob_test(int size, int off) { 1071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *p = (char*)malloc_aaa(size); 1081e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // fprintf(stderr, "writing %d byte(s) into [%p,%p) with offset %d\n", 1091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // sizeof(T), p, p + size, off); 1101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_write((T*)(p + off)); 1111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free_aaa(p); 1121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 1131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1152dcf44991a8ec1ca7c8051eb27c5ff158530bdc3Timur Iskhodzhanovtemplate<typename T> 116938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void uaf_test(int size, int off) { 1171e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *p = (char *)malloc_aaa(size); 1181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free_aaa(p); 1191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (int i = 1; i < 100; i++) 1201e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free_aaa(malloc_aaa(i)); 1211e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany fprintf(stderr, "writing %ld byte(s) at %p with offset %d\n", 1221e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany (long)sizeof(T), p, off); 1231e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_write((T*)(p + off)); 1241e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 1251e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 12613ebae606b526399771e9cca1d6a9d24458ad0f1Kostya SerebryanyTEST(AddressSanitizer, HasFeatureAddressSanitizerTest) { 12713ebae606b526399771e9cca1d6a9d24458ad0f1Kostya Serebryany#if defined(__has_feature) && __has_feature(address_sanitizer) 12813ebae606b526399771e9cca1d6a9d24458ad0f1Kostya Serebryany bool asan = 1; 129badab16503e025326a46a45da8faa6fd20e5fc3dKostya Serebryany#elif defined(__SANITIZE_ADDRESS__) 130badab16503e025326a46a45da8faa6fd20e5fc3dKostya Serebryany bool asan = 1; 13113ebae606b526399771e9cca1d6a9d24458ad0f1Kostya Serebryany#else 13213ebae606b526399771e9cca1d6a9d24458ad0f1Kostya Serebryany bool asan = 0; 13313ebae606b526399771e9cca1d6a9d24458ad0f1Kostya Serebryany#endif 13413ebae606b526399771e9cca1d6a9d24458ad0f1Kostya Serebryany EXPECT_EQ(true, asan); 1351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 1361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1371e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, SimpleDeathTest) { 1381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(exit(1), ""); 1391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 1401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1411e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, VariousMallocsTest) { 1421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *a = (int*)malloc(100 * sizeof(int)); 1431e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany a[50] = 0; 1441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(a); 1451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *r = (int*)malloc(10); 1471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany r = (int*)realloc(r, 2000 * sizeof(int)); 1481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany r[1000] = 0; 1491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(r); 1501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1511e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *b = new int[100]; 1521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany b[50] = 0; 1531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany delete [] b; 1541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *c = new int; 1561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany *c = 0; 1571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany delete c; 1581e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 159e0e9eea99d7a81a15c4bb15ff2e5ebd92b50817bEvgeniy Stepanov#if !defined(__APPLE__) && !defined(ANDROID) && !defined(__ANDROID__) 1601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *pm; 1611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int pm_res = posix_memalign((void**)&pm, kPageSize, kPageSize); 1621e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(0, pm_res); 1631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(pm); 1647274eafceb832e1088f30b71652992787ba8c456Evgeniy Stepanov#endif 1651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1664dd8ba8238b1b698953628affe6e5b2edf3b3e3fAlexey Samsonov#if !defined(__APPLE__) 1671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *ma = (int*)memalign(kPageSize, kPageSize); 16876e842845c927e92c2c21c41ac51e364fa17528cAlexey Samsonov EXPECT_EQ(0U, (uintptr_t)ma % kPageSize); 1691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany ma[123] = 0; 1701e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(ma); 1711e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#endif // __APPLE__ 1721e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 1731e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1741e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, CallocTest) { 1751e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *a = (int*)calloc(100, sizeof(int)); 1761e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(0, a[10]); 1771e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(a); 1781e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 1791e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1801e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, VallocTest) { 1811e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *a = valloc(100); 18276e842845c927e92c2c21c41ac51e364fa17528cAlexey Samsonov EXPECT_EQ(0U, (uintptr_t)a % kPageSize); 1831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(a); 1841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 1851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1861e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#ifndef __APPLE__ 1871e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, PvallocTest) { 1881e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *a = (char*)pvalloc(kPageSize + 100); 18976e842845c927e92c2c21c41ac51e364fa17528cAlexey Samsonov EXPECT_EQ(0U, (uintptr_t)a % kPageSize); 1901e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany a[kPageSize + 101] = 1; // we should not report an error here. 1911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(a); 1921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1931e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany a = (char*)pvalloc(0); // pvalloc(0) should allocate at least one page. 19476e842845c927e92c2c21c41ac51e364fa17528cAlexey Samsonov EXPECT_EQ(0U, (uintptr_t)a % kPageSize); 1951e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany a[101] = 1; // we should not report an error here. 1961e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(a); 1971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 1981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#endif // __APPLE__ 1991e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2001e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid *TSDWorker(void *test_key) { 2011e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (test_key) { 2021e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany pthread_setspecific(*(pthread_key_t*)test_key, (void*)0xfeedface); 2031e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 2041e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return NULL; 2051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 2061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid TSDDestructor(void *tsd) { 2081e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Spawning a thread will check that the current thread id is not -1. 2091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany pthread_t th; 2102697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_CREATE(&th, NULL, TSDWorker, NULL); 2112697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_JOIN(th, NULL); 2121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 2131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// This tests triggers the thread-specific data destruction fiasco which occurs 2151e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// if we don't manage the TSD destructors ourselves. We create a new pthread 2161e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// key with a non-NULL destructor which is likely to be put after the destructor 2171e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// of AsanThread in the list of destructors. 2181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// In this case the TSD for AsanThread will be destroyed before TSDDestructor 2191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// is called for the child thread, and a CHECK will fail when we call 2201e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// pthread_create() to spawn the grandchild. 2211e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_TSDTest) { 2221e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany pthread_t th; 2231e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany pthread_key_t test_key; 2241e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany pthread_key_create(&test_key, TSDDestructor); 2252697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_CREATE(&th, NULL, TSDWorker, &test_key); 2262697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_JOIN(th, NULL); 2271e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany pthread_key_delete(test_key); 2281e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 2291e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2302dcf44991a8ec1ca7c8051eb27c5ff158530bdc3Timur Iskhodzhanovtemplate<typename T> 2311e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid OOBTest() { 2321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char expected_str[100]; 2331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (int size = sizeof(T); size < 20; size += 5) { 2341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (int i = -5; i < 0; i++) { 2351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany const char *str = 2361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "is located.*%d byte.*to the left"; 2371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany sprintf(expected_str, str, abs(i)); 2381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(oob_test<T>(size, i), expected_str); 2391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 2401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 24176e842845c927e92c2c21c41ac51e364fa17528cAlexey Samsonov for (int i = 0; i < (int)(size - sizeof(T) + 1); i++) 2421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany oob_test<T>(size, i); 2431e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 244709a33e1cf20eb7f00653fe32dc07714b3f2c633Kostya Serebryany for (int i = size - sizeof(T) + 1; i <= (int)(size + 2 * sizeof(T)); i++) { 2451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany const char *str = 2461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "is located.*%d byte.*to the right"; 2471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int off = i >= size ? (i - size) : 0; 2481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // we don't catch unaligned partially OOB accesses. 2491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (i % sizeof(T)) continue; 2501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany sprintf(expected_str, str, off); 2511e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(oob_test<T>(size, i), expected_str); 2521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 2531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 2541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(oob_test<T>(kLargeMalloc, -1), 2561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "is located.*1 byte.*to the left"); 2571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(oob_test<T>(kLargeMalloc, kLargeMalloc), 2581e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "is located.*0 byte.*to the right"); 2591e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 2601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// TODO(glider): the following tests are EXTREMELY slow on Darwin: 2621e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// AddressSanitizer.OOB_char (125503 ms) 2631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// AddressSanitizer.OOB_int (126890 ms) 2641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// AddressSanitizer.OOBRightTest (315605 ms) 2651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// AddressSanitizer.SimpleStackTest (366559 ms) 2661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2671e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, OOB_char) { 2681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany OOBTest<U1>(); 2691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 2701e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2711e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, OOB_int) { 2721e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany OOBTest<U4>(); 2731e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 2741e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2751e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, OOBRightTest) { 2761e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (size_t access_size = 1; access_size <= 8; access_size *= 2) { 2771e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (size_t alloc_size = 1; alloc_size <= 8; alloc_size++) { 2781e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (size_t offset = 0; offset <= 8; offset += access_size) { 2791e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *p = malloc(alloc_size); 2801e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // allocated: [p, p + alloc_size) 2811e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // accessed: [p + offset, p + offset + access_size) 2821e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany uint8_t *addr = (uint8_t*)p + offset; 2831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (offset + access_size <= alloc_size) { 2841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany asan_write_sized_aligned(addr, access_size); 2851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } else { 2861e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int outside_bytes = offset > alloc_size ? (offset - alloc_size) : 0; 2871e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany const char *str = 2881e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "is located.%d *byte.*to the right"; 2891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char expected_str[100]; 2901e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany sprintf(expected_str, str, outside_bytes); 2911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(asan_write_sized_aligned(addr, access_size), 2921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany expected_str); 2931e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 2941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(p); 2951e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 2961e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 2971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 2981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 2991e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 300f1877cf0a314f407ac535ab1606fdac4f9b05026Kostya Serebryany#if ASAN_ALLOCATOR_VERSION == 2 // Broken with the asan_allocator1 301f1877cf0a314f407ac535ab1606fdac4f9b05026Kostya SerebryanyTEST(AddressSanitizer, LargeOOBRightTest) { 302f1877cf0a314f407ac535ab1606fdac4f9b05026Kostya Serebryany size_t large_power_of_two = 1 << 19; 303f1877cf0a314f407ac535ab1606fdac4f9b05026Kostya Serebryany for (size_t i = 16; i <= 256; i *= 2) { 304f1877cf0a314f407ac535ab1606fdac4f9b05026Kostya Serebryany size_t size = large_power_of_two - i; 305f1877cf0a314f407ac535ab1606fdac4f9b05026Kostya Serebryany char *p = Ident(new char[size]); 306f1877cf0a314f407ac535ab1606fdac4f9b05026Kostya Serebryany EXPECT_DEATH(p[size] = 0, "is located 0 bytes to the right"); 307f1877cf0a314f407ac535ab1606fdac4f9b05026Kostya Serebryany delete [] p; 308f1877cf0a314f407ac535ab1606fdac4f9b05026Kostya Serebryany } 309f1877cf0a314f407ac535ab1606fdac4f9b05026Kostya Serebryany} 310f1877cf0a314f407ac535ab1606fdac4f9b05026Kostya Serebryany#endif // ASAN_ALLOCATOR_VERSION == 2 311f1877cf0a314f407ac535ab1606fdac4f9b05026Kostya Serebryany 3121e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, UAF_char) { 313ca9b5dd9922fb7f6e0ca06dbba7db7398b8997f1Kostya Serebryany const char *uaf_string = "AddressSanitizer:.*heap-use-after-free"; 3141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(uaf_test<U1>(1, 0), uaf_string); 3151e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(uaf_test<U1>(10, 0), uaf_string); 3161e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(uaf_test<U1>(10, 10), uaf_string); 3171e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(uaf_test<U1>(kLargeMalloc, 0), uaf_string); 3181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(uaf_test<U1>(kLargeMalloc, kLargeMalloc / 2), uaf_string); 3191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 3201e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 3211e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#if ASAN_HAS_BLACKLIST 3221e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, IgnoreTest) { 3231e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *x = Ident(new int); 3241e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany delete Ident(x); 3251e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany *x = 0; 3261e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 3271e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#endif // ASAN_HAS_BLACKLIST 3281e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 3291e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanystruct StructWithBitField { 3301e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int bf1:1; 3311e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int bf2:1; 3321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int bf3:1; 3331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int bf4:29; 3341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}; 3351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 3361e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, BitFieldPositiveTest) { 3371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany StructWithBitField *x = new StructWithBitField; 3381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany delete Ident(x); 3391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(x->bf1 = 0, "use-after-free"); 3401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(x->bf2 = 0, "use-after-free"); 3411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(x->bf3 = 0, "use-after-free"); 3421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(x->bf4 = 0, "use-after-free"); 34376e842845c927e92c2c21c41ac51e364fa17528cAlexey Samsonov} 3441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 3451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanystruct StructWithBitFields_8_24 { 3461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int a:8; 3471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int b:24; 3481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}; 3491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 3501e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, BitFieldNegativeTest) { 3511e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany StructWithBitFields_8_24 *x = Ident(new StructWithBitFields_8_24); 3521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x->a = 0; 3531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x->b = 0; 3541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany delete Ident(x); 3551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 3561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 3571e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, OutOfMemoryTest) { 3585af39e50366f1aacbebc284f572f08ad1ad07357Kostya Serebryany size_t size = SANITIZER_WORDSIZE == 64 ? (size_t)(1ULL << 48) : (0xf0000000); 3591e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(0, realloc(0, size)); 3601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(0, realloc(0, ~Ident(0))); 3611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(0, malloc(size)); 3621e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(0, malloc(~Ident(0))); 3631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(0, calloc(1, size)); 3641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(0, calloc(1, ~Ident(0))); 3651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 3661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 3671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#if ASAN_NEEDS_SEGV 36858b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenkonamespace { 36958b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko 370ca9b5dd9922fb7f6e0ca06dbba7db7398b8997f1Kostya Serebryanyconst char kUnknownCrash[] = "AddressSanitizer: SEGV on unknown address"; 37158b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenkoconst char kOverriddenHandler[] = "ASan signal handler has been overridden\n"; 37258b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko 3731e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, WildAddressTest) { 3741e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *c = (char*)0x123; 37558b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko EXPECT_DEATH(*c = 0, kUnknownCrash); 37658b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko} 37758b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko 37858b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenkovoid my_sigaction_sighandler(int, siginfo_t*, void*) { 37958b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko fprintf(stderr, kOverriddenHandler); 38058b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko exit(1); 38158b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko} 38258b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko 383b0ddf227b25388ad836e560c00f0aec84026b0e5Alexey Samsonovvoid my_signal_sighandler(int signum) { 38458b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko fprintf(stderr, kOverriddenHandler); 38558b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko exit(1); 38658b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko} 38758b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko 38858b017bb286a8f4413a9348d2c87c52de38cff6dAlexander PotapenkoTEST(AddressSanitizer, SignalTest) { 38958b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko struct sigaction sigact; 39058b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko memset(&sigact, 0, sizeof(sigact)); 39158b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko sigact.sa_sigaction = my_sigaction_sighandler; 39258b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko sigact.sa_flags = SA_SIGINFO; 39358b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko // ASan should silently ignore sigaction()... 39458b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko EXPECT_EQ(0, sigaction(SIGSEGV, &sigact, 0)); 39558b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko#ifdef __APPLE__ 39658b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko EXPECT_EQ(0, sigaction(SIGBUS, &sigact, 0)); 39758b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko#endif 39858b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko char *c = (char*)0x123; 39958b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko EXPECT_DEATH(*c = 0, kUnknownCrash); 40058b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko // ... and signal(). 40158b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko EXPECT_EQ(0, signal(SIGSEGV, my_signal_sighandler)); 40258b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko EXPECT_DEATH(*c = 0, kUnknownCrash); 40358b017bb286a8f4413a9348d2c87c52de38cff6dAlexander Potapenko} 404b0ddf227b25388ad836e560c00f0aec84026b0e5Alexey Samsonov} // namespace 4051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#endif 4061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 4071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanystatic void MallocStress(size_t n) { 40848ddbef1d051875b2d0b204e8d78300b58d80a85Evgeniy Stepanov uint32_t seed = my_rand(); 4091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (size_t iter = 0; iter < 10; iter++) { 4101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany vector<void *> vec; 4111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (size_t i = 0; i < n; i++) { 4121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if ((i % 3) == 0) { 4131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (vec.empty()) continue; 41448ddbef1d051875b2d0b204e8d78300b58d80a85Evgeniy Stepanov size_t idx = my_rand_r(&seed) % vec.size(); 4151e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany void *ptr = vec[idx]; 4161e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany vec[idx] = vec.back(); 4171e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany vec.pop_back(); 4181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free_aaa(ptr); 4191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } else { 42048ddbef1d051875b2d0b204e8d78300b58d80a85Evgeniy Stepanov size_t size = my_rand_r(&seed) % 1000 + 1; 4211e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#ifndef __APPLE__ 42248ddbef1d051875b2d0b204e8d78300b58d80a85Evgeniy Stepanov size_t alignment = 1 << (my_rand_r(&seed) % 7 + 3); 4231e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *ptr = (char*)memalign_aaa(alignment, size); 4241e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#else 4251e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *ptr = (char*) malloc_aaa(size); 4261e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#endif 4271e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany vec.push_back(ptr); 4281e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany ptr[0] = 0; 4291e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany ptr[size-1] = 0; 4301e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany ptr[size/2] = 0; 4311e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 4321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 4331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (size_t i = 0; i < vec.size(); i++) 4341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free_aaa(vec[i]); 4351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 4361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 4371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 4381e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, MallocStressTest) { 4398ae44ac678644265ba82a1b5696a402c9301a85cEvgeniy Stepanov MallocStress((ASAN_LOW_MEMORY) ? 20000 : 200000); 4401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 4411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 4421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanystatic void TestLargeMalloc(size_t size) { 4431e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char buff[1024]; 4441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany sprintf(buff, "is located 1 bytes to the left of %lu-byte", (long)size); 4451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(Ident((char*)malloc(size))[-1] = 0, buff); 4461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 4471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 4481e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, LargeMallocTest) { 4491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (int i = 113; i < (1 << 28); i = i * 2 + 13) { 4501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany TestLargeMalloc(i); 4511e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 4521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 4531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 454788e1d704436065beb514fdde86070f61d0394edEvgeniy Stepanov#if ASAN_LOW_MEMORY != 1 4551e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, HugeMallocTest) { 4561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#ifdef __APPLE__ 4571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // It was empirically found out that 1215 megabytes is the maximum amount of 458e4d141dbbaf5185f332a185c22308340657d243eKostya Serebryany // memory available to the process under AddressSanitizer on 32-bit Mac 10.6. 459e4d141dbbaf5185f332a185c22308340657d243eKostya Serebryany // 32-bit Mac 10.7 gives even less (< 1G). 4601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // (the libSystem malloc() allows allocating up to 2300 megabytes without 4611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // ASan). 4625af39e50366f1aacbebc284f572f08ad1ad07357Kostya Serebryany size_t n_megs = SANITIZER_WORDSIZE == 32 ? 500 : 4100; 4631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#else 4645af39e50366f1aacbebc284f572f08ad1ad07357Kostya Serebryany size_t n_megs = SANITIZER_WORDSIZE == 32 ? 2600 : 4100; 4651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#endif 4661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany TestLargeMalloc(n_megs << 20); 4671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 468788e1d704436065beb514fdde86070f61d0394edEvgeniy Stepanov#endif 4691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 470c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryany#ifndef __APPLE__ 471c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryanyvoid MemalignRun(size_t align, size_t size, int idx) { 472c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryany char *p = (char *)memalign(align, size); 473c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryany Ident(p)[idx] = 0; 474c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryany free(p); 475c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryany} 476c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryany 477c311105fd64219bb96b6467c3707ebda355be87bKostya SerebryanyTEST(AddressSanitizer, memalign) { 478c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryany for (int align = 16; align <= (1 << 23); align *= 2) { 479c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryany size_t size = align * 5; 480c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryany EXPECT_DEATH(MemalignRun(align, size, -1), 481c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryany "is located 1 bytes to the left"); 482c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryany EXPECT_DEATH(MemalignRun(align, size, size + 1), 483c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryany "is located 1 bytes to the right"); 484c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryany } 485c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryany} 486c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryany#endif 487c311105fd64219bb96b6467c3707ebda355be87bKostya Serebryany 4881e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, ThreadedMallocStressTest) { 4891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany const int kNumThreads = 4; 4908ae44ac678644265ba82a1b5696a402c9301a85cEvgeniy Stepanov const int kNumIterations = (ASAN_LOW_MEMORY) ? 10000 : 100000; 4911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany pthread_t t[kNumThreads]; 4921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (int i = 0; i < kNumThreads; i++) { 4932697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_CREATE(&t[i], 0, (void* (*)(void *x))MallocStress, 4948ae44ac678644265ba82a1b5696a402c9301a85cEvgeniy Stepanov (void*)kNumIterations); 4951e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 4961e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (int i = 0; i < kNumThreads; i++) { 4972697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_JOIN(t[i], 0); 4981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 4991e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 5001e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 5011e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid *ManyThreadsWorker(void *a) { 5021e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (int iter = 0; iter < 100; iter++) { 5031e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (size_t size = 100; size < 2000; size *= 2) { 5041e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(Ident(malloc(size))); 5051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 5061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 5071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return 0; 5081e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 5091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 5101e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, ManyThreadsTest) { 5112697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany const size_t kNumThreads = 5122697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany (SANITIZER_WORDSIZE == 32 || ASAN_AVOID_EXPENSIVE_TESTS) ? 30 : 1000; 5131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany pthread_t t[kNumThreads]; 5141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (size_t i = 0; i < kNumThreads; i++) { 5152697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_CREATE(&t[i], 0, ManyThreadsWorker, (void*)i); 5161e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 5171e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (size_t i = 0; i < kNumThreads; i++) { 5182697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_JOIN(t[i], 0); 5191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 5201e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 5211e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 5221e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, ReallocTest) { 5231e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany const int kMinElem = 5; 5241e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *ptr = (int*)malloc(sizeof(int) * kMinElem); 5251e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany ptr[3] = 3; 5261e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (int i = 0; i < 10000; i++) { 5271e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany ptr = (int*)realloc(ptr, 52848ddbef1d051875b2d0b204e8d78300b58d80a85Evgeniy Stepanov (my_rand() % 1000 + kMinElem) * sizeof(int)); 5291e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(3, ptr[3]); 5301e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 5311e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 5321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 5334fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey Samsonov#ifndef __APPLE__ 5344fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey Samsonovstatic const char *kMallocUsableSizeErrorMsg = 535ca9b5dd9922fb7f6e0ca06dbba7db7398b8997f1Kostya Serebryany "AddressSanitizer: attempting to call malloc_usable_size()"; 5364fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey Samsonov 5374fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey SamsonovTEST(AddressSanitizer, MallocUsableSizeTest) { 5384fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey Samsonov const size_t kArraySize = 100; 5394fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey Samsonov char *array = Ident((char*)malloc(kArraySize)); 5404fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey Samsonov int *int_ptr = Ident(new int); 54176e842845c927e92c2c21c41ac51e364fa17528cAlexey Samsonov EXPECT_EQ(0U, malloc_usable_size(NULL)); 5424fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey Samsonov EXPECT_EQ(kArraySize, malloc_usable_size(array)); 5434fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey Samsonov EXPECT_EQ(sizeof(int), malloc_usable_size(int_ptr)); 5444fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey Samsonov EXPECT_DEATH(malloc_usable_size((void*)0x123), kMallocUsableSizeErrorMsg); 5454fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey Samsonov EXPECT_DEATH(malloc_usable_size(array + kArraySize / 2), 5464fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey Samsonov kMallocUsableSizeErrorMsg); 5474fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey Samsonov free(array); 5484fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey Samsonov EXPECT_DEATH(malloc_usable_size(array), kMallocUsableSizeErrorMsg); 5494fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey Samsonov} 5504fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey Samsonov#endif 5514fd95f141f78906570c15a8a3b4cf0a7b50a201dAlexey Samsonov 5521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid WrongFree() { 5531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *x = (int*)malloc(100 * sizeof(int)); 5541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Use the allocated memory, otherwise Clang will optimize it out. 5551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(x); 5561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(x + 1); 5571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 5581e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 5591e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, WrongFreeTest) { 560c16201337a4bb974093ff75aa45682739a331d05Kostya Serebryany EXPECT_DEATH(WrongFree(), 561ca9b5dd9922fb7f6e0ca06dbba7db7398b8997f1Kostya Serebryany "ERROR: AddressSanitizer: attempting free.*not malloc"); 5621e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 5631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 5641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid DoubleFree() { 5651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *x = (int*)malloc(100 * sizeof(int)); 5661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany fprintf(stderr, "DoubleFree: x=%p\n", x); 5671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(x); 5681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(x); 5691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany fprintf(stderr, "should have failed in the second free(%p)\n", x); 5701e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany abort(); 5711e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 5721e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 5731e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DoubleFreeTest) { 57427f4932a8a0845303b33d69ddce096c445002e60Kostya Serebryany EXPECT_DEATH(DoubleFree(), ASAN_PCRE_DOTALL 575ca9b5dd9922fb7f6e0ca06dbba7db7398b8997f1Kostya Serebryany "ERROR: AddressSanitizer: attempting double-free" 57627f4932a8a0845303b33d69ddce096c445002e60Kostya Serebryany ".*is located 0 bytes inside of 400-byte region" 57727f4932a8a0845303b33d69ddce096c445002e60Kostya Serebryany ".*freed by thread T0 here" 57827f4932a8a0845303b33d69ddce096c445002e60Kostya Serebryany ".*previously allocated by thread T0 here"); 5791e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 5801e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 5811e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanytemplate<int kSize> 582938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void SizedStackTest() { 5831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char a[kSize]; 5841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *A = Ident((char*)&a); 5851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (size_t i = 0; i < kSize; i++) 5861e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany A[i] = i; 5871e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(A[-1] = 0, ""); 5881e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(A[-20] = 0, ""); 5891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(A[-31] = 0, ""); 5901e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(A[kSize] = 0, ""); 5911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(A[kSize + 1] = 0, ""); 5921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(A[kSize + 10] = 0, ""); 5931e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(A[kSize + 31] = 0, ""); 5941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 5951e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 5961e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, SimpleStackTest) { 5971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SizedStackTest<1>(); 5981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SizedStackTest<2>(); 5991e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SizedStackTest<3>(); 6001e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SizedStackTest<4>(); 6011e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SizedStackTest<5>(); 6021e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SizedStackTest<6>(); 6031e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SizedStackTest<7>(); 6041e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SizedStackTest<16>(); 6051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SizedStackTest<25>(); 6061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SizedStackTest<34>(); 6071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SizedStackTest<43>(); 6081e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SizedStackTest<51>(); 6091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SizedStackTest<62>(); 6101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SizedStackTest<64>(); 6111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SizedStackTest<128>(); 6121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 6131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 6141e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, ManyStackObjectsTest) { 6151e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char XXX[10]; 6161e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char YYY[20]; 6171e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char ZZZ[30]; 6181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(XXX); 6191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(YYY); 6201e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(Ident(ZZZ)[-1] = 0, ASAN_PCRE_DOTALL "XXX.*YYY.*ZZZ"); 6211e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 6221e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 623938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE static void Frame0(int frame, char *a, char *b, char *c) { 6241e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char d[4] = {0}; 6251e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *D = Ident(d); 6261e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany switch (frame) { 6271e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany case 3: a[5]++; break; 6281e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany case 2: b[5]++; break; 6291e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany case 1: c[5]++; break; 6301e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany case 0: D[5]++; break; 6311e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 6321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 633938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE static void Frame1(int frame, char *a, char *b) { 6341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char c[4] = {0}; Frame0(frame, a, b, c); 6351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany break_optimization(0); 6361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 637938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE static void Frame2(int frame, char *a) { 6381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char b[4] = {0}; Frame1(frame, a, b); 6391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany break_optimization(0); 6401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 641938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE static void Frame3(int frame) { 6421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char a[4] = {0}; Frame2(frame, a); 6431e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany break_optimization(0); 6441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 6451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 6461e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, GuiltyStackFrame0Test) { 6471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(Frame3(0), "located .*in frame <.*Frame0"); 6481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 6491e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, GuiltyStackFrame1Test) { 6501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(Frame3(1), "located .*in frame <.*Frame1"); 6511e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 6521e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, GuiltyStackFrame2Test) { 6531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(Frame3(2), "located .*in frame <.*Frame2"); 6541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 6551e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, GuiltyStackFrame3Test) { 6561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(Frame3(3), "located .*in frame <.*Frame3"); 6571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 6581e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 659938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void LongJmpFunc1(jmp_buf buf) { 6601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // create three red zones for these two stack objects. 6611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int a; 6621e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int b; 6631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 6641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *A = Ident(&a); 6651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *B = Ident(&b); 6661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany *A = *B; 6671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany longjmp(buf, 1); 6681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 6691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 670d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya SerebryanyNOINLINE void BuiltinLongJmpFunc1(jmp_buf buf) { 671d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany // create three red zones for these two stack objects. 672d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany int a; 673d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany int b; 674d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany 675d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany int *A = Ident(&a); 676d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany int *B = Ident(&b); 677d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany *A = *B; 678d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany __builtin_longjmp((void**)buf, 1); 679d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany} 680d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany 681938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void UnderscopeLongJmpFunc1(jmp_buf buf) { 6821e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // create three red zones for these two stack objects. 6831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int a; 6841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int b; 6851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 6861e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *A = Ident(&a); 6871e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *B = Ident(&b); 6881e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany *A = *B; 6891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany _longjmp(buf, 1); 6901e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 6911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 692938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void SigLongJmpFunc1(sigjmp_buf buf) { 6931e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // create three red zones for these two stack objects. 6941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int a; 6951e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int b; 6961e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 6971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *A = Ident(&a); 6981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *B = Ident(&b); 6991e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany *A = *B; 7001e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany siglongjmp(buf, 1); 7011e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 7021e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 7031e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 704938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void TouchStackFunc() { 7051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int a[100]; // long array will intersect with redzones from LongJmpFunc1. 7061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *A = Ident(a); 7071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (int i = 0; i < 100; i++) 7081e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany A[i] = i*i; 7091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 7101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 7111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Test that we handle longjmp and do not report fals positives on stack. 7121e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, LongJmpTest) { 7131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany static jmp_buf buf; 7141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (!setjmp(buf)) { 7151e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany LongJmpFunc1(buf); 7161e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } else { 7171e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany TouchStackFunc(); 7181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 7191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 7201e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 721b989143d0be56496e8d5fcf75969af35a058792aKostya Serebryany#if not defined(__ANDROID__) 722c3a5c173f228cbb15e332e6bbc17c76ebd55d7a8Kostya SerebryanyTEST(AddressSanitizer, BuiltinLongJmpTest) { 723d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany static jmp_buf buf; 724c3a5c173f228cbb15e332e6bbc17c76ebd55d7a8Kostya Serebryany if (!__builtin_setjmp((void**)buf)) { 725d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany BuiltinLongJmpFunc1(buf); 726d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany } else { 727d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany TouchStackFunc(); 728d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany } 729d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany} 730b989143d0be56496e8d5fcf75969af35a058792aKostya Serebryany#endif // not defined(__ANDROID__) 731d3ca78fdb296d2a51fc224da673ebfff508acf44Kostya Serebryany 7321e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, UnderscopeLongJmpTest) { 7331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany static jmp_buf buf; 7341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (!_setjmp(buf)) { 7351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany UnderscopeLongJmpFunc1(buf); 7361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } else { 7371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany TouchStackFunc(); 7381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 7391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 7401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 7411e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, SigLongJmpTest) { 7421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany static sigjmp_buf buf; 7431e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (!sigsetjmp(buf, 1)) { 7441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SigLongJmpFunc1(buf); 7451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } else { 7461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany TouchStackFunc(); 7471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 7481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 7491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 7501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#ifdef __EXCEPTIONS 751938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE void ThrowFunc() { 7521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // create three red zones for these two stack objects. 7531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int a; 7541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int b; 7551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 7561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *A = Ident(&a); 7571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *B = Ident(&b); 7581e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany *A = *B; 7591e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany ASAN_THROW(1); 7601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 7611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 7621e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, CxxExceptionTest) { 7631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (ASAN_UAR) return; 7641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // TODO(kcc): this test crashes on 32-bit for some reason... 7655af39e50366f1aacbebc284f572f08ad1ad07357Kostya Serebryany if (SANITIZER_WORDSIZE == 32) return; 7661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany try { 7671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany ThrowFunc(); 7681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } catch(...) {} 7691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany TouchStackFunc(); 7701e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 7711e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#endif 7721e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 7731e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid *ThreadStackReuseFunc1(void *unused) { 7741e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // create three red zones for these two stack objects. 7751e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int a; 7761e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int b; 7771e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 7781e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *A = Ident(&a); 7791e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *B = Ident(&b); 7801e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany *A = *B; 7811e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany pthread_exit(0); 7821e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return 0; 7831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 7841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 7851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid *ThreadStackReuseFunc2(void *unused) { 7861e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany TouchStackFunc(); 7871e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return 0; 7881e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 7891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 7901e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, ThreadStackReuseTest) { 7911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany pthread_t t; 7922697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_CREATE(&t, 0, ThreadStackReuseFunc1, 0); 7932697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_JOIN(t, 0); 7942697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_CREATE(&t, 0, ThreadStackReuseFunc2, 0); 7952697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_JOIN(t, 0); 7961e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 7971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 798f6607d097833cd53460735c1a5a2fd52b97613ebTimur Iskhodzhanov#if defined(__i386__) || defined(__x86_64__) 7991e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, Store128Test) { 8001e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *a = Ident((char*)malloc(Ident(12))); 8011e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *p = a; 8021e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (((uintptr_t)a % 16) != 0) 8031e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany p = a + 8; 8041e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany assert(((uintptr_t)p % 16) == 0); 8051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany __m128i value_wide = _mm_set1_epi16(0x1234); 8061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(_mm_store_si128((__m128i*)p, value_wide), 807ca9b5dd9922fb7f6e0ca06dbba7db7398b8997f1Kostya Serebryany "AddressSanitizer: heap-buffer-overflow"); 808acd5c617b219e0f059620c2a3928d2cc821d4534Kostya Serebryany EXPECT_DEATH(_mm_store_si128((__m128i*)p, value_wide), 8091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "WRITE of size 16"); 8101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(_mm_store_si128((__m128i*)p, value_wide), 8111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "located 0 bytes to the right of 12-byte"); 8121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(a); 8131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 8141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#endif 8151e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 816ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenkostatic string RightOOBErrorMessage(int oob_distance, bool is_write) { 8171e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany assert(oob_distance >= 0); 8181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char expected_str[100]; 8190b770b48e4ae053a7d0610e1c682405afba58fdaKostya Serebryany sprintf(expected_str, ASAN_PCRE_DOTALL "%s.*located %d bytes to the right", 820ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko is_write ? "WRITE" : "READ", oob_distance); 8211e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return string(expected_str); 8221e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 8231e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 824ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenkostatic string RightOOBWriteMessage(int oob_distance) { 825ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko return RightOOBErrorMessage(oob_distance, /*is_write*/true); 826ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko} 827ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko 828ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenkostatic string RightOOBReadMessage(int oob_distance) { 829ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko return RightOOBErrorMessage(oob_distance, /*is_write*/false); 830ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko} 831ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko 832ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenkostatic string LeftOOBErrorMessage(int oob_distance, bool is_write) { 8331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany assert(oob_distance > 0); 8341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char expected_str[100]; 8350b770b48e4ae053a7d0610e1c682405afba58fdaKostya Serebryany sprintf(expected_str, ASAN_PCRE_DOTALL "%s.*located %d bytes to the left", 836ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko is_write ? "WRITE" : "READ", oob_distance); 8371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return string(expected_str); 8381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 8391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 840ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenkostatic string LeftOOBWriteMessage(int oob_distance) { 841ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko return LeftOOBErrorMessage(oob_distance, /*is_write*/true); 842ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko} 843ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko 844ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenkostatic string LeftOOBReadMessage(int oob_distance) { 845ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko return LeftOOBErrorMessage(oob_distance, /*is_write*/false); 846ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko} 847ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko 848ada9ba14325df9e206f454c50bd1b55ea9ecdbb9Alexander Potapenkostatic string LeftOOBAccessMessage(int oob_distance) { 849ada9ba14325df9e206f454c50bd1b55ea9ecdbb9Alexander Potapenko assert(oob_distance > 0); 850ada9ba14325df9e206f454c50bd1b55ea9ecdbb9Alexander Potapenko char expected_str[100]; 851ada9ba14325df9e206f454c50bd1b55ea9ecdbb9Alexander Potapenko sprintf(expected_str, "located %d bytes to the left", oob_distance); 852ada9ba14325df9e206f454c50bd1b55ea9ecdbb9Alexander Potapenko return string(expected_str); 853ada9ba14325df9e206f454c50bd1b55ea9ecdbb9Alexander Potapenko} 854ada9ba14325df9e206f454c50bd1b55ea9ecdbb9Alexander Potapenko 8552dcf44991a8ec1ca7c8051eb27c5ff158530bdc3Timur Iskhodzhanovtemplate<typename T> 8561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid MemSetOOBTestTemplate(size_t length) { 8571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (length == 0) return; 8581e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany size_t size = Ident(sizeof(T) * length); 8591e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany T *array = Ident((T*)malloc(size)); 8601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int element = Ident(42); 8611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int zero = Ident(0); 8627125bb35fb2351040534a194c9c1aa8cb71cfb19Kostya Serebryany void *(*MEMSET)(void *s, int c, size_t n) = Ident(memset); 8631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // memset interval inside array 8647125bb35fb2351040534a194c9c1aa8cb71cfb19Kostya Serebryany MEMSET(array, element, size); 8657125bb35fb2351040534a194c9c1aa8cb71cfb19Kostya Serebryany MEMSET(array, element, size - 1); 8667125bb35fb2351040534a194c9c1aa8cb71cfb19Kostya Serebryany MEMSET(array + length - 1, element, sizeof(T)); 8677125bb35fb2351040534a194c9c1aa8cb71cfb19Kostya Serebryany MEMSET(array, element, 1); 8681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 8691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // memset 0 bytes 8707125bb35fb2351040534a194c9c1aa8cb71cfb19Kostya Serebryany MEMSET(array - 10, element, zero); 8717125bb35fb2351040534a194c9c1aa8cb71cfb19Kostya Serebryany MEMSET(array - 1, element, zero); 8727125bb35fb2351040534a194c9c1aa8cb71cfb19Kostya Serebryany MEMSET(array, element, zero); 8737125bb35fb2351040534a194c9c1aa8cb71cfb19Kostya Serebryany MEMSET(array + length, 0, zero); 8747125bb35fb2351040534a194c9c1aa8cb71cfb19Kostya Serebryany MEMSET(array + length + 1, 0, zero); 8751e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 8761e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // try to memset bytes to the right of array 8777125bb35fb2351040534a194c9c1aa8cb71cfb19Kostya Serebryany EXPECT_DEATH(MEMSET(array, 0, size + 1), 8787125bb35fb2351040534a194c9c1aa8cb71cfb19Kostya Serebryany RightOOBWriteMessage(0)); 8797125bb35fb2351040534a194c9c1aa8cb71cfb19Kostya Serebryany EXPECT_DEATH(MEMSET((char*)(array + length) - 1, element, 6), 8807125bb35fb2351040534a194c9c1aa8cb71cfb19Kostya Serebryany RightOOBWriteMessage(0)); 8817125bb35fb2351040534a194c9c1aa8cb71cfb19Kostya Serebryany EXPECT_DEATH(MEMSET(array + 1, element, size + sizeof(T)), 882ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko RightOOBWriteMessage(0)); 8831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // whole interval is to the right 8847125bb35fb2351040534a194c9c1aa8cb71cfb19Kostya Serebryany EXPECT_DEATH(MEMSET(array + length + 1, 0, 10), 885ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko RightOOBWriteMessage(sizeof(T))); 8861e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 8871e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // try to memset bytes to the left of array 8887125bb35fb2351040534a194c9c1aa8cb71cfb19Kostya Serebryany EXPECT_DEATH(MEMSET((char*)array - 1, element, size), 889ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko LeftOOBWriteMessage(1)); 8907125bb35fb2351040534a194c9c1aa8cb71cfb19Kostya Serebryany EXPECT_DEATH(MEMSET((char*)array - 5, 0, 6), 891ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko LeftOOBWriteMessage(5)); 892c35314a119aba6d5e7fa0a06024f9cb1c49e6f5fKostya Serebryany if (length >= 100) { 893c35314a119aba6d5e7fa0a06024f9cb1c49e6f5fKostya Serebryany // Large OOB, we find it only if the redzone is large enough. 894c35314a119aba6d5e7fa0a06024f9cb1c49e6f5fKostya Serebryany EXPECT_DEATH(memset(array - 5, element, size + 5 * sizeof(T)), 895c35314a119aba6d5e7fa0a06024f9cb1c49e6f5fKostya Serebryany LeftOOBWriteMessage(5 * sizeof(T))); 896c35314a119aba6d5e7fa0a06024f9cb1c49e6f5fKostya Serebryany } 8971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // whole interval is to the left 8987125bb35fb2351040534a194c9c1aa8cb71cfb19Kostya Serebryany EXPECT_DEATH(MEMSET(array - 2, 0, sizeof(T)), 899ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko LeftOOBWriteMessage(2 * sizeof(T))); 9001e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 9011e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // try to memset bytes both to the left & to the right 9027125bb35fb2351040534a194c9c1aa8cb71cfb19Kostya Serebryany EXPECT_DEATH(MEMSET((char*)array - 2, element, size + 4), 903ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko LeftOOBWriteMessage(2)); 9041e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 9051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(array); 9061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 9071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 9081e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, MemSetOOBTest) { 9091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany MemSetOOBTestTemplate<char>(100); 9101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany MemSetOOBTestTemplate<int>(5); 9111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany MemSetOOBTestTemplate<double>(256); 9121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // We can test arrays of structres/classes here, but what for? 9131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 9141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 915eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany// Try to allocate two arrays of 'size' bytes that are near each other. 916eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany// Strictly speaking we are not guaranteed to find such two pointers, 917eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany// but given the structure of asan's allocator we will. 91813ddcf695e10bed51faceff49469e1c83d64b51cKostya Serebryanystatic bool AllocateTwoAdjacentArrays(char **x1, char **x2, size_t size) { 919eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany vector<char *> v; 920eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany bool res = false; 921eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany for (size_t i = 0; i < 1000U && !res; i++) { 922eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany v.push_back(new char[size]); 923eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany if (i == 0) continue; 924eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany sort(v.begin(), v.end()); 925eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany for (size_t j = 1; j < v.size(); j++) { 926eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany assert(v[j] > v[j-1]); 927c0d37b35439897aea821319a669832b16848cb65Kostya Serebryany if ((size_t)(v[j] - v[j-1]) < size * 2) { 928eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany *x2 = v[j]; 929eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany *x1 = v[j-1]; 930eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany res = true; 931eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany break; 932eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany } 933eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany } 934eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany } 935eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany 936eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany for (size_t i = 0; i < v.size(); i++) { 937eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany if (res && v[i] == *x1) continue; 938eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany if (res && v[i] == *x2) continue; 939eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany delete [] v[i]; 940eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany } 941eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany return res; 942eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany} 943eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany 944eb2809311c94b73c269ccef8d68ae368642e5754Kostya SerebryanyTEST(AddressSanitizer, LargeOOBInMemset) { 945eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany for (size_t size = 200; size < 100000; size += size / 2) { 946eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany char *x1, *x2; 94713ddcf695e10bed51faceff49469e1c83d64b51cKostya Serebryany if (!Ident(AllocateTwoAdjacentArrays)(&x1, &x2, size)) 948eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany continue; 949eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany // fprintf(stderr, " large oob memset: %p %p %zd\n", x1, x2, size); 950eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany // Do a memset on x1 with huge out-of-bound access that will end up in x2. 95113ddcf695e10bed51faceff49469e1c83d64b51cKostya Serebryany EXPECT_DEATH(Ident(memset)(x1, 0, size * 2), 95213ddcf695e10bed51faceff49469e1c83d64b51cKostya Serebryany "is located 0 bytes to the right"); 953eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany delete [] x1; 954eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany delete [] x2; 955eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany return; 956eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany } 957eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany assert(0 && "Did not find two adjacent malloc-ed pointers"); 958eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany} 959eb2809311c94b73c269ccef8d68ae368642e5754Kostya Serebryany 9601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Same test for memcpy and memmove functions 9612dcf44991a8ec1ca7c8051eb27c5ff158530bdc3Timur Iskhodzhanovtemplate <typename T, class M> 9621e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid MemTransferOOBTestTemplate(size_t length) { 9631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (length == 0) return; 9641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany size_t size = Ident(sizeof(T) * length); 9651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany T *src = Ident((T*)malloc(size)); 9661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany T *dest = Ident((T*)malloc(size)); 9671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int zero = Ident(0); 9681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 9691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // valid transfer of bytes between arrays 9701e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany M::transfer(dest, src, size); 9711e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany M::transfer(dest + 1, src, size - sizeof(T)); 9721e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany M::transfer(dest, src + length - 1, sizeof(T)); 9731e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany M::transfer(dest, src, 1); 9741e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 9751e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // transfer zero bytes 9761e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany M::transfer(dest - 1, src, 0); 9771e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany M::transfer(dest + length, src, zero); 9781e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany M::transfer(dest, src - 1, zero); 9791e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany M::transfer(dest, src, zero); 9801e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 9811e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // try to change mem to the right of dest 9821e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(M::transfer(dest + 1, src, size), 9837125bb35fb2351040534a194c9c1aa8cb71cfb19Kostya Serebryany RightOOBWriteMessage(0)); 9841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(M::transfer((char*)(dest + length) - 1, src, 5), 9857125bb35fb2351040534a194c9c1aa8cb71cfb19Kostya Serebryany RightOOBWriteMessage(0)); 9861e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 9871e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // try to change mem to the left of dest 9881e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(M::transfer(dest - 2, src, size), 989ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko LeftOOBWriteMessage(2 * sizeof(T))); 9901e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(M::transfer((char*)dest - 3, src, 4), 991ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko LeftOOBWriteMessage(3)); 9921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 9931e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // try to access mem to the right of src 9941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(M::transfer(dest, src + 2, size), 9957125bb35fb2351040534a194c9c1aa8cb71cfb19Kostya Serebryany RightOOBReadMessage(0)); 9961e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(M::transfer(dest, (char*)(src + length) - 3, 6), 9977125bb35fb2351040534a194c9c1aa8cb71cfb19Kostya Serebryany RightOOBReadMessage(0)); 9981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 9991e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // try to access mem to the left of src 10001e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(M::transfer(dest, src - 1, size), 1001ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko LeftOOBReadMessage(sizeof(T))); 10021e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(M::transfer(dest, (char*)src - 6, 7), 1003ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko LeftOOBReadMessage(6)); 10041e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 10051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Generally we don't need to test cases where both accessing src and writing 10061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // to dest address to poisoned memory. 10071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 10081e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany T *big_src = Ident((T*)malloc(size * 2)); 10091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany T *big_dest = Ident((T*)malloc(size * 2)); 10101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // try to change mem to both sides of dest 10111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(M::transfer(dest - 1, big_src, size * 2), 1012ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko LeftOOBWriteMessage(sizeof(T))); 10131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // try to access mem to both sides of src 10141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(M::transfer(big_dest, src - 2, size * 2), 1015ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko LeftOOBReadMessage(2 * sizeof(T))); 10161e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 10171e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(src); 10181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(dest); 10191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(big_src); 10201e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(big_dest); 10211e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 10221e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 10231e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyclass MemCpyWrapper { 10241e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany public: 10251e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany static void* transfer(void *to, const void *from, size_t size) { 10267125bb35fb2351040534a194c9c1aa8cb71cfb19Kostya Serebryany return Ident(memcpy)(to, from, size); 10271e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 10281e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}; 10291e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, MemCpyOOBTest) { 10301e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany MemTransferOOBTestTemplate<char, MemCpyWrapper>(100); 10311e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany MemTransferOOBTestTemplate<int, MemCpyWrapper>(1024); 10321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 10331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 10341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyclass MemMoveWrapper { 10351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany public: 10361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany static void* transfer(void *to, const void *from, size_t size) { 10377125bb35fb2351040534a194c9c1aa8cb71cfb19Kostya Serebryany return Ident(memmove)(to, from, size); 10381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 10391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany}; 10401e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, MemMoveOOBTest) { 10411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany MemTransferOOBTestTemplate<char, MemMoveWrapper>(100); 10421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany MemTransferOOBTestTemplate<int, MemMoveWrapper>(1024); 10431e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 10441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 10451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Tests for string functions 10461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 10471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Used for string functions tests 10481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanystatic char global_string[] = "global"; 10491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanystatic size_t global_string_length = 6; 10501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 10511e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Input to a test is a zero-terminated string str with given length 10521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Accesses to the bytes to the left and to the right of str 10531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// are presumed to produce OOB errors 10541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid StrLenOOBTestTemplate(char *str, size_t length, bool is_global) { 10551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Normal strlen calls 10561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(strlen(str), length); 10571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (length > 0) { 105876e842845c927e92c2c21c41ac51e364fa17528cAlexey Samsonov EXPECT_EQ(length - 1, strlen(str + 1)); 105976e842845c927e92c2c21c41ac51e364fa17528cAlexey Samsonov EXPECT_EQ(0U, strlen(str + length)); 10601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 10611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Arg of strlen is not malloced, OOB access 10621e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany if (!is_global) { 10631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // We don't insert RedZones to the left of global variables 1064ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(strlen(str - 1)), LeftOOBReadMessage(1)); 1065ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(strlen(str - 5)), LeftOOBReadMessage(5)); 10661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 1067ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(strlen(str + length + 1)), RightOOBReadMessage(0)); 10681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Overwrite terminator 10691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany str[length] = 'a'; 10701e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // String is not zero-terminated, strlen will lead to OOB access 1071ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(strlen(str)), RightOOBReadMessage(0)); 1072ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(strlen(str + length)), RightOOBReadMessage(0)); 10731e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Restore terminator 10741e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany str[length] = 0; 10751e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 10761e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, StrLenOOBTest) { 10771e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Check heap-allocated string 10781e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany size_t length = Ident(10); 10791e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *heap_string = Ident((char*)malloc(length + 1)); 10801e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char stack_string[10 + 1]; 1081badab16503e025326a46a45da8faa6fd20e5fc3dKostya Serebryany break_optimization(&stack_string); 108276e842845c927e92c2c21c41ac51e364fa17528cAlexey Samsonov for (size_t i = 0; i < length; i++) { 10831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany heap_string[i] = 'a'; 10841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany stack_string[i] = 'b'; 10851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 10861e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany heap_string[length] = 0; 10871e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany stack_string[length] = 0; 10881e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany StrLenOOBTestTemplate(heap_string, length, false); 10891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // TODO(samsonov): Fix expected messages in StrLenOOBTestTemplate to 10901e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // make test for stack_string work. Or move it to output tests. 10911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // StrLenOOBTestTemplate(stack_string, length, false); 10921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany StrLenOOBTestTemplate(global_string, global_string_length, true); 10931e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(heap_string); 10941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 10951e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 109644997c354c7cdb5257d9a41c837b12a6133f2d6bKostya Serebryanystatic inline char* MallocAndMemsetString(size_t size, char ch) { 109744997c354c7cdb5257d9a41c837b12a6133f2d6bKostya Serebryany char *s = Ident((char*)malloc(size)); 109844997c354c7cdb5257d9a41c837b12a6133f2d6bKostya Serebryany memset(s, ch, size); 109944997c354c7cdb5257d9a41c837b12a6133f2d6bKostya Serebryany return s; 110044997c354c7cdb5257d9a41c837b12a6133f2d6bKostya Serebryany} 110144997c354c7cdb5257d9a41c837b12a6133f2d6bKostya Serebryanystatic inline char* MallocAndMemsetString(size_t size) { 110244997c354c7cdb5257d9a41c837b12a6133f2d6bKostya Serebryany return MallocAndMemsetString(size, 'z'); 110344997c354c7cdb5257d9a41c837b12a6133f2d6bKostya Serebryany} 110444997c354c7cdb5257d9a41c837b12a6133f2d6bKostya Serebryany 11051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#ifndef __APPLE__ 11061e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, StrNLenOOBTest) { 11071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany size_t size = Ident(123); 1108af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany char *str = MallocAndMemsetString(size); 11091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Normal strnlen calls. 11101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(strnlen(str - 1, 0)); 11111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(strnlen(str, size)); 11121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(strnlen(str + size - 1, 1)); 11131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany str[size - 1] = '\0'; 11141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(strnlen(str, 2 * size)); 11151e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Argument points to not allocated memory. 1116ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(strnlen(str - 1, 1)), LeftOOBReadMessage(1)); 1117ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(strnlen(str + size, 1)), RightOOBReadMessage(0)); 11181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Overwrite the terminating '\0' and hit unallocated memory. 11191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany str[size - 1] = 'z'; 1120ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(strnlen(str, size + 1)), RightOOBReadMessage(0)); 11211e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(str); 11221e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 11231e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#endif 11241e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 11251e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, StrDupOOBTest) { 11261e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany size_t size = Ident(42); 1127af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany char *str = MallocAndMemsetString(size); 11281e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *new_str; 11291e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Normal strdup calls. 11301e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany str[size - 1] = '\0'; 11311e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany new_str = strdup(str); 11321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(new_str); 11331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany new_str = strdup(str + size - 1); 11341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(new_str); 11351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Argument points to not allocated memory. 1136ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(strdup(str - 1)), LeftOOBReadMessage(1)); 1137ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(strdup(str + size)), RightOOBReadMessage(0)); 11381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Overwrite the terminating '\0' and hit unallocated memory. 11391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany str[size - 1] = 'z'; 1140ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(strdup(str)), RightOOBReadMessage(0)); 11411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(str); 11421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 11431e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 11441e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, StrCpyOOBTest) { 11451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany size_t to_size = Ident(30); 11461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany size_t from_size = Ident(6); // less than to_size 11471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *to = Ident((char*)malloc(to_size)); 11481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *from = Ident((char*)malloc(from_size)); 11491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Normal strcpy calls. 11501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany strcpy(from, "hello"); 11511e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany strcpy(to, from); 11521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany strcpy(to + to_size - from_size, from); 11531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Length of "from" is too small. 1154ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(strcpy(from, "hello2")), RightOOBWriteMessage(0)); 11551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // "to" or "from" points to not allocated memory. 1156ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(strcpy(to - 1, from)), LeftOOBWriteMessage(1)); 1157ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(strcpy(to, from - 1)), LeftOOBReadMessage(1)); 1158ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(strcpy(to, from + from_size)), RightOOBReadMessage(0)); 1159ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(strcpy(to + to_size, from)), RightOOBWriteMessage(0)); 11601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Overwrite the terminating '\0' character and hit unallocated memory. 11611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany from[from_size - 1] = '!'; 1162ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(strcpy(to, from)), RightOOBReadMessage(0)); 11631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(to); 11641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(from); 11651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 11661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 11671e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, StrNCpyOOBTest) { 11681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany size_t to_size = Ident(20); 11691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany size_t from_size = Ident(6); // less than to_size 11701e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *to = Ident((char*)malloc(to_size)); 11711e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // From is a zero-terminated string "hello\0" of length 6 11721e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *from = Ident((char*)malloc(from_size)); 11731e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany strcpy(from, "hello"); 11741e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // copy 0 bytes 11751e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany strncpy(to, from, 0); 11761e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany strncpy(to - 1, from - 1, 0); 11771e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // normal strncpy calls 11781e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany strncpy(to, from, from_size); 11791e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany strncpy(to, from, to_size); 11801e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany strncpy(to, from + from_size - 1, to_size); 11811e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany strncpy(to + to_size - 1, from, 1); 11821e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // One of {to, from} points to not allocated memory 11831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(Ident(strncpy(to, from - 1, from_size)), 1184ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko LeftOOBReadMessage(1)); 11851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(Ident(strncpy(to - 1, from, from_size)), 1186ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko LeftOOBWriteMessage(1)); 11871e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(Ident(strncpy(to, from + from_size, 1)), 1188ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko RightOOBReadMessage(0)); 11891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(Ident(strncpy(to + to_size, from, 1)), 1190ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko RightOOBWriteMessage(0)); 11911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Length of "to" is too small 11921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(Ident(strncpy(to + to_size - from_size + 1, from, from_size)), 1193ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko RightOOBWriteMessage(0)); 11941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(Ident(strncpy(to + 1, from, to_size)), 1195ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko RightOOBWriteMessage(0)); 11961e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Overwrite terminator in from 11971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany from[from_size - 1] = '!'; 11981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // normal strncpy call 11991e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany strncpy(to, from, from_size); 12001e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Length of "from" is too small 12011e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(Ident(strncpy(to, from, to_size)), 1202ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko RightOOBReadMessage(0)); 12031e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(to); 12041e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(from); 12051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 12061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1207fa7e95da75c30b7111224e68cb4405357bdc0f4bAlexey Samsonov// Users may have different definitions of "strchr" and "index", so provide 1208fa7e95da75c30b7111224e68cb4405357bdc0f4bAlexey Samsonov// function pointer typedefs and overload RunStrChrTest implementation. 1209fa7e95da75c30b7111224e68cb4405357bdc0f4bAlexey Samsonov// We can't use macro for RunStrChrTest body here, as this macro would 1210fa7e95da75c30b7111224e68cb4405357bdc0f4bAlexey Samsonov// confuse EXPECT_DEATH gtest macro. 1211fa7e95da75c30b7111224e68cb4405357bdc0f4bAlexey Samsonovtypedef char*(*PointerToStrChr1)(const char*, int); 1212fa7e95da75c30b7111224e68cb4405357bdc0f4bAlexey Samsonovtypedef char*(*PointerToStrChr2)(char*, int); 1213fa7e95da75c30b7111224e68cb4405357bdc0f4bAlexey Samsonov 1214fa7e95da75c30b7111224e68cb4405357bdc0f4bAlexey SamsonovUSED static void RunStrChrTest(PointerToStrChr1 StrChr) { 12151e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany size_t size = Ident(100); 1216af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany char *str = MallocAndMemsetString(size); 12171e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany str[10] = 'q'; 12181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany str[11] = '\0'; 12191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(str, StrChr(str, 'z')); 12201e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(str + 10, StrChr(str, 'q')); 12211e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(NULL, StrChr(str, 'a')); 12221e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // StrChr argument points to not allocated memory. 1223ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(StrChr(str - 1, 'z')), LeftOOBReadMessage(1)); 1224ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(StrChr(str + size, 'z')), RightOOBReadMessage(0)); 12251e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Overwrite the terminator and hit not allocated memory. 12261e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany str[11] = 'z'; 1227ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(StrChr(str, 'a')), RightOOBReadMessage(0)); 12281e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(str); 12291e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 1230fa7e95da75c30b7111224e68cb4405357bdc0f4bAlexey SamsonovUSED static void RunStrChrTest(PointerToStrChr2 StrChr) { 1231fa7e95da75c30b7111224e68cb4405357bdc0f4bAlexey Samsonov size_t size = Ident(100); 1232fa7e95da75c30b7111224e68cb4405357bdc0f4bAlexey Samsonov char *str = MallocAndMemsetString(size); 1233fa7e95da75c30b7111224e68cb4405357bdc0f4bAlexey Samsonov str[10] = 'q'; 1234fa7e95da75c30b7111224e68cb4405357bdc0f4bAlexey Samsonov str[11] = '\0'; 1235fa7e95da75c30b7111224e68cb4405357bdc0f4bAlexey Samsonov EXPECT_EQ(str, StrChr(str, 'z')); 1236fa7e95da75c30b7111224e68cb4405357bdc0f4bAlexey Samsonov EXPECT_EQ(str + 10, StrChr(str, 'q')); 1237fa7e95da75c30b7111224e68cb4405357bdc0f4bAlexey Samsonov EXPECT_EQ(NULL, StrChr(str, 'a')); 1238fa7e95da75c30b7111224e68cb4405357bdc0f4bAlexey Samsonov // StrChr argument points to not allocated memory. 1239ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(StrChr(str - 1, 'z')), LeftOOBReadMessage(1)); 1240ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(StrChr(str + size, 'z')), RightOOBReadMessage(0)); 1241fa7e95da75c30b7111224e68cb4405357bdc0f4bAlexey Samsonov // Overwrite the terminator and hit not allocated memory. 1242fa7e95da75c30b7111224e68cb4405357bdc0f4bAlexey Samsonov str[11] = 'z'; 1243ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(StrChr(str, 'a')), RightOOBReadMessage(0)); 1244fa7e95da75c30b7111224e68cb4405357bdc0f4bAlexey Samsonov free(str); 1245fa7e95da75c30b7111224e68cb4405357bdc0f4bAlexey Samsonov} 1246fa7e95da75c30b7111224e68cb4405357bdc0f4bAlexey Samsonov 12471e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, StrChrAndIndexOOBTest) { 12481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany RunStrChrTest(&strchr); 12491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany RunStrChrTest(&index); 12501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 12511e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 12521e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, StrCmpAndFriendsLogicTest) { 12531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // strcmp 12541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(0, strcmp("", "")); 12551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(0, strcmp("abcd", "abcd")); 1256af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany EXPECT_GT(0, strcmp("ab", "ac")); 1257af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany EXPECT_GT(0, strcmp("abc", "abcd")); 1258af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany EXPECT_LT(0, strcmp("acc", "abc")); 1259af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany EXPECT_LT(0, strcmp("abcd", "abc")); 12601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 12611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // strncmp 12621e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(0, strncmp("a", "b", 0)); 12631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(0, strncmp("abcd", "abcd", 10)); 12641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(0, strncmp("abcd", "abcef", 3)); 1265af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany EXPECT_GT(0, strncmp("abcde", "abcfa", 4)); 1266af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany EXPECT_GT(0, strncmp("a", "b", 5)); 1267af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany EXPECT_GT(0, strncmp("bc", "bcde", 4)); 1268af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany EXPECT_LT(0, strncmp("xyz", "xyy", 10)); 1269af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany EXPECT_LT(0, strncmp("baa", "aaa", 1)); 1270af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany EXPECT_LT(0, strncmp("zyx", "", 2)); 1271af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany 1272af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany // strcasecmp 1273af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany EXPECT_EQ(0, strcasecmp("", "")); 1274af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany EXPECT_EQ(0, strcasecmp("zzz", "zzz")); 1275af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany EXPECT_EQ(0, strcasecmp("abCD", "ABcd")); 1276af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany EXPECT_GT(0, strcasecmp("aB", "Ac")); 1277af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany EXPECT_GT(0, strcasecmp("ABC", "ABCd")); 1278af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany EXPECT_LT(0, strcasecmp("acc", "abc")); 1279af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany EXPECT_LT(0, strcasecmp("ABCd", "abc")); 1280af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany 1281af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany // strncasecmp 1282af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany EXPECT_EQ(0, strncasecmp("a", "b", 0)); 1283af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany EXPECT_EQ(0, strncasecmp("abCD", "ABcd", 10)); 1284af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany EXPECT_EQ(0, strncasecmp("abCd", "ABcef", 3)); 1285af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany EXPECT_GT(0, strncasecmp("abcde", "ABCfa", 4)); 1286af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany EXPECT_GT(0, strncasecmp("a", "B", 5)); 1287af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany EXPECT_GT(0, strncasecmp("bc", "BCde", 4)); 1288af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany EXPECT_LT(0, strncasecmp("xyz", "xyy", 10)); 1289af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany EXPECT_LT(0, strncasecmp("Baa", "aaa", 1)); 1290af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany EXPECT_LT(0, strncasecmp("zyx", "", 2)); 129152fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany 129252fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany // memcmp 129352fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany EXPECT_EQ(0, memcmp("a", "b", 0)); 129452fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany EXPECT_EQ(0, memcmp("ab\0c", "ab\0c", 4)); 129552fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany EXPECT_GT(0, memcmp("\0ab", "\0ac", 3)); 129652fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany EXPECT_GT(0, memcmp("abb\0", "abba", 4)); 129752fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany EXPECT_LT(0, memcmp("ab\0cd", "ab\0c\0", 5)); 129852fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany EXPECT_LT(0, memcmp("zza", "zyx", 3)); 1299af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany} 1300af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany 1301af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryanytypedef int(*PointerToStrCmp)(const char*, const char*); 1302af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryanyvoid RunStrCmpTest(PointerToStrCmp StrCmp) { 13031e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany size_t size = Ident(100); 1304b3ef8a4521954d1ca5e728446c389e1f8cc01fc4Kostya Serebryany int fill = 'o'; 1305b3ef8a4521954d1ca5e728446c389e1f8cc01fc4Kostya Serebryany char *s1 = MallocAndMemsetString(size, fill); 1306b3ef8a4521954d1ca5e728446c389e1f8cc01fc4Kostya Serebryany char *s2 = MallocAndMemsetString(size, fill); 13071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany s1[size - 1] = '\0'; 13081e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany s2[size - 1] = '\0'; 1309af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany // Normal StrCmp calls 1310af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany Ident(StrCmp(s1, s2)); 1311af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany Ident(StrCmp(s1, s2 + size - 1)); 1312af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany Ident(StrCmp(s1 + size - 1, s2 + size - 1)); 13131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany s1[size - 1] = 'z'; 13141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany s2[size - 1] = 'x'; 1315af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany Ident(StrCmp(s1, s2)); 13161e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // One of arguments points to not allocated memory. 1317ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(StrCmp)(s1 - 1, s2), LeftOOBReadMessage(1)); 1318ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(StrCmp)(s1, s2 - 1), LeftOOBReadMessage(1)); 1319ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(StrCmp)(s1 + size, s2), RightOOBReadMessage(0)); 1320ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(StrCmp)(s1, s2 + size), RightOOBReadMessage(0)); 13211e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Hit unallocated memory and die. 1322b3ef8a4521954d1ca5e728446c389e1f8cc01fc4Kostya Serebryany s1[size - 1] = fill; 1323ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(StrCmp)(s1, s1), RightOOBReadMessage(0)); 1324ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(StrCmp)(s1 + size - 1, s2), RightOOBReadMessage(0)); 13251e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(s1); 13261e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(s2); 13271e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 13281e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1329af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya SerebryanyTEST(AddressSanitizer, StrCmpOOBTest) { 1330af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany RunStrCmpTest(&strcmp); 1331af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany} 1332af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany 1333af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya SerebryanyTEST(AddressSanitizer, StrCaseCmpOOBTest) { 1334af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany RunStrCmpTest(&strcasecmp); 1335af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany} 1336af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany 1337af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryanytypedef int(*PointerToStrNCmp)(const char*, const char*, size_t); 1338af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryanyvoid RunStrNCmpTest(PointerToStrNCmp StrNCmp) { 13391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany size_t size = Ident(100); 13401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *s1 = MallocAndMemsetString(size); 13411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *s2 = MallocAndMemsetString(size); 13421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany s1[size - 1] = '\0'; 13431e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany s2[size - 1] = '\0'; 1344af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany // Normal StrNCmp calls 1345af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany Ident(StrNCmp(s1, s2, size + 2)); 13461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany s1[size - 1] = 'z'; 13471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany s2[size - 1] = 'x'; 1348af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany Ident(StrNCmp(s1 + size - 2, s2 + size - 2, size)); 13491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany s2[size - 1] = 'z'; 1350af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany Ident(StrNCmp(s1 - 1, s2 - 1, 0)); 1351af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany Ident(StrNCmp(s1 + size - 1, s2 + size - 1, 1)); 13521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // One of arguments points to not allocated memory. 1353ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(StrNCmp)(s1 - 1, s2, 1), LeftOOBReadMessage(1)); 1354ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(StrNCmp)(s1, s2 - 1, 1), LeftOOBReadMessage(1)); 1355ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(StrNCmp)(s1 + size, s2, 1), RightOOBReadMessage(0)); 1356ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(StrNCmp)(s1, s2 + size, 1), RightOOBReadMessage(0)); 13571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Hit unallocated memory and die. 1358ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(StrNCmp)(s1 + 1, s2 + 1, size), RightOOBReadMessage(0)); 1359ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(StrNCmp)(s1 + size - 1, s2, 2), RightOOBReadMessage(0)); 13601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(s1); 13611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(s2); 13621e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 13631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1364af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya SerebryanyTEST(AddressSanitizer, StrNCmpOOBTest) { 1365af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany RunStrNCmpTest(&strncmp); 1366af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany} 1367af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany 1368af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya SerebryanyTEST(AddressSanitizer, StrNCaseCmpOOBTest) { 1369af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany RunStrNCmpTest(&strncasecmp); 1370af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany} 1371af0f01d77c2a495f023ffbf6cce85b33bbd2306dKostya Serebryany 137252fb238ccc45781e4e1d097ae1ee748c898b5825Kostya SerebryanyTEST(AddressSanitizer, MemCmpOOBTest) { 137352fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany size_t size = Ident(100); 137452fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany char *s1 = MallocAndMemsetString(size); 137552fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany char *s2 = MallocAndMemsetString(size); 137652fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany // Normal memcmp calls. 137752fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany Ident(memcmp(s1, s2, size)); 137852fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany Ident(memcmp(s1 + size - 1, s2 + size - 1, 1)); 137952fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany Ident(memcmp(s1 - 1, s2 - 1, 0)); 138052fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany // One of arguments points to not allocated memory. 1381ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(memcmp)(s1 - 1, s2, 1), LeftOOBReadMessage(1)); 1382ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(memcmp)(s1, s2 - 1, 1), LeftOOBReadMessage(1)); 1383ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(memcmp)(s1 + size, s2, 1), RightOOBReadMessage(0)); 1384ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(memcmp)(s1, s2 + size, 1), RightOOBReadMessage(0)); 138552fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany // Hit unallocated memory and die. 1386ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(memcmp)(s1 + 1, s2 + 1, size), RightOOBReadMessage(0)); 1387ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(memcmp)(s1 + size - 1, s2, 2), RightOOBReadMessage(0)); 138852fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany // Zero bytes are not terminators and don't prevent from OOB. 138952fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany s1[size - 1] = '\0'; 139052fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany s2[size - 1] = '\0'; 1391ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Ident(memcmp)(s1, s2, size + 1), RightOOBReadMessage(0)); 139252fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany free(s1); 139352fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany free(s2); 139452fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany} 139552fb238ccc45781e4e1d097ae1ee748c898b5825Kostya Serebryany 13960985ca240812ac5519168a6aecbccf4c513ae243Kostya SerebryanyTEST(AddressSanitizer, StrCatOOBTest) { 1397ada9ba14325df9e206f454c50bd1b55ea9ecdbb9Alexander Potapenko // strcat() reads strlen(to) bytes from |to| before concatenating. 13980985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany size_t to_size = Ident(100); 13990985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany char *to = MallocAndMemsetString(to_size); 14000985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany to[0] = '\0'; 14010985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany size_t from_size = Ident(20); 14020985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany char *from = MallocAndMemsetString(from_size); 14030985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany from[from_size - 1] = '\0'; 14040985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany // Normal strcat calls. 14050985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany strcat(to, from); 14060985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany strcat(to, from); 14070985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany strcat(to + from_size, from + from_size - 2); 140837b3fcd6fdec5740fe51fc1315c5d4d54313de98Alexander Potapenko // Passing an invalid pointer is an error even when concatenating an empty 140937b3fcd6fdec5740fe51fc1315c5d4d54313de98Alexander Potapenko // string. 1410ada9ba14325df9e206f454c50bd1b55ea9ecdbb9Alexander Potapenko EXPECT_DEATH(strcat(to - 1, from + from_size - 1), LeftOOBAccessMessage(1)); 14110985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany // One of arguments points to not allocated memory. 1412ada9ba14325df9e206f454c50bd1b55ea9ecdbb9Alexander Potapenko EXPECT_DEATH(strcat(to - 1, from), LeftOOBAccessMessage(1)); 1413ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(strcat(to, from - 1), LeftOOBReadMessage(1)); 1414ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(strcat(to + to_size, from), RightOOBWriteMessage(0)); 1415ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(strcat(to, from + from_size), RightOOBReadMessage(0)); 14160985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany 14170985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany // "from" is not zero-terminated. 14180985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany from[from_size - 1] = 'z'; 1419ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(strcat(to, from), RightOOBReadMessage(0)); 14200985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany from[from_size - 1] = '\0'; 14210985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany // "to" is not zero-terminated. 14220985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany memset(to, 'z', to_size); 1423ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(strcat(to, from), RightOOBWriteMessage(0)); 14240985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany // "to" is too short to fit "from". 14250985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany to[to_size - from_size + 1] = '\0'; 1426ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(strcat(to, from), RightOOBWriteMessage(0)); 14270985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany // length of "to" is just enough. 14280985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany strcat(to, from + 1); 1429c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov 1430c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov free(to); 1431c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov free(from); 1432c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov} 1433c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov 1434c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey SamsonovTEST(AddressSanitizer, StrNCatOOBTest) { 1435ada9ba14325df9e206f454c50bd1b55ea9ecdbb9Alexander Potapenko // strncat() reads strlen(to) bytes from |to| before concatenating. 1436c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov size_t to_size = Ident(100); 1437c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov char *to = MallocAndMemsetString(to_size); 1438c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov to[0] = '\0'; 1439c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov size_t from_size = Ident(20); 1440c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov char *from = MallocAndMemsetString(from_size); 1441c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov // Normal strncat calls. 1442c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov strncat(to, from, 0); 1443c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov strncat(to, from, from_size); 1444c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov from[from_size - 1] = '\0'; 1445c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov strncat(to, from, 2 * from_size); 144637b3fcd6fdec5740fe51fc1315c5d4d54313de98Alexander Potapenko // Catenating empty string with an invalid string is still an error. 1447ada9ba14325df9e206f454c50bd1b55ea9ecdbb9Alexander Potapenko EXPECT_DEATH(strncat(to - 1, from, 0), LeftOOBAccessMessage(1)); 1448c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov strncat(to, from + from_size - 1, 10); 1449c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov // One of arguments points to not allocated memory. 1450ada9ba14325df9e206f454c50bd1b55ea9ecdbb9Alexander Potapenko EXPECT_DEATH(strncat(to - 1, from, 2), LeftOOBAccessMessage(1)); 1451ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(strncat(to, from - 1, 2), LeftOOBReadMessage(1)); 1452ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(strncat(to + to_size, from, 2), RightOOBWriteMessage(0)); 1453ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(strncat(to, from + from_size, 2), RightOOBReadMessage(0)); 1454c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov 1455c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov memset(from, 'z', from_size); 1456c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov memset(to, 'z', to_size); 1457c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov to[0] = '\0'; 1458c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov // "from" is too short. 1459ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(strncat(to, from, from_size + 1), RightOOBReadMessage(0)); 1460c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov // "to" is not zero-terminated. 1461ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(strncat(to + 1, from, 1), RightOOBWriteMessage(0)); 1462c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov // "to" is too short to fit "from". 1463c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov to[0] = 'z'; 1464c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov to[to_size - from_size + 1] = '\0'; 1465ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(strncat(to, from, from_size - 1), RightOOBWriteMessage(0)); 1466c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov // "to" is just enough. 1467c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov strncat(to, from, from_size - 2); 1468c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov 1469c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov free(to); 1470c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov free(from); 14710985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany} 14720985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany 1473c5e72a3b7c60f1b2d9d9be3a07d397d5b5f872beKostya Serebryanystatic string OverlapErrorMessage(const string &func) { 1474c5e72a3b7c60f1b2d9d9be3a07d397d5b5f872beKostya Serebryany return func + "-param-overlap"; 1475c5e72a3b7c60f1b2d9d9be3a07d397d5b5f872beKostya Serebryany} 14761e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 14771e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, StrArgsOverlapTest) { 14781e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany size_t size = Ident(100); 14791e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *str = Ident((char*)malloc(size)); 14801e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 14810503ce8824393174c17acb26d62551f18a0fd82aAlexander Potapenko// Do not check memcpy() on OS X 10.7 and later, where it actually aliases 14820503ce8824393174c17acb26d62551f18a0fd82aAlexander Potapenko// memmove(). 14830503ce8824393174c17acb26d62551f18a0fd82aAlexander Potapenko#if !defined(__APPLE__) || !defined(MAC_OS_X_VERSION_10_7) || \ 14840503ce8824393174c17acb26d62551f18a0fd82aAlexander Potapenko (MAC_OS_X_VERSION_MAX_ALLOWED < MAC_OS_X_VERSION_10_7) 14851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Check "memcpy". Use Ident() to avoid inlining. 14861e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany memset(str, 'z', size); 14871e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(memcpy)(str + 1, str + 11, 10); 14881e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(memcpy)(str, str, 0); 1489c5e72a3b7c60f1b2d9d9be3a07d397d5b5f872beKostya Serebryany EXPECT_DEATH(Ident(memcpy)(str, str + 14, 15), OverlapErrorMessage("memcpy")); 1490c5e72a3b7c60f1b2d9d9be3a07d397d5b5f872beKostya Serebryany EXPECT_DEATH(Ident(memcpy)(str + 14, str, 15), OverlapErrorMessage("memcpy")); 14910503ce8824393174c17acb26d62551f18a0fd82aAlexander Potapenko#endif 1492c655cfa8dc0fdf5b521e565f6cbdbf7981fba1b2Kostya Serebryany 1493c655cfa8dc0fdf5b521e565f6cbdbf7981fba1b2Kostya Serebryany // We do not treat memcpy with to==from as a bug. 1494c655cfa8dc0fdf5b521e565f6cbdbf7981fba1b2Kostya Serebryany // See http://llvm.org/bugs/show_bug.cgi?id=11763. 1495c655cfa8dc0fdf5b521e565f6cbdbf7981fba1b2Kostya Serebryany // EXPECT_DEATH(Ident(memcpy)(str + 20, str + 20, 1), 1496c655cfa8dc0fdf5b521e565f6cbdbf7981fba1b2Kostya Serebryany // OverlapErrorMessage("memcpy")); 14971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 14981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Check "strcpy". 14991e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany memset(str, 'z', size); 15001e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany str[9] = '\0'; 15011e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany strcpy(str + 10, str); 1502c5e72a3b7c60f1b2d9d9be3a07d397d5b5f872beKostya Serebryany EXPECT_DEATH(strcpy(str + 9, str), OverlapErrorMessage("strcpy")); 1503c5e72a3b7c60f1b2d9d9be3a07d397d5b5f872beKostya Serebryany EXPECT_DEATH(strcpy(str, str + 4), OverlapErrorMessage("strcpy")); 15041e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany strcpy(str, str + 5); 15051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 15061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // Check "strncpy". 15071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany memset(str, 'z', size); 15081e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany strncpy(str, str + 10, 10); 1509c5e72a3b7c60f1b2d9d9be3a07d397d5b5f872beKostya Serebryany EXPECT_DEATH(strncpy(str, str + 9, 10), OverlapErrorMessage("strncpy")); 1510c5e72a3b7c60f1b2d9d9be3a07d397d5b5f872beKostya Serebryany EXPECT_DEATH(strncpy(str + 9, str, 10), OverlapErrorMessage("strncpy")); 15111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany str[10] = '\0'; 15121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany strncpy(str + 11, str, 20); 1513c5e72a3b7c60f1b2d9d9be3a07d397d5b5f872beKostya Serebryany EXPECT_DEATH(strncpy(str + 10, str, 20), OverlapErrorMessage("strncpy")); 15141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 15150985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany // Check "strcat". 15160985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany memset(str, 'z', size); 15170985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany str[10] = '\0'; 15180985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany str[20] = '\0'; 15190985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany strcat(str, str + 10); 152037b3fcd6fdec5740fe51fc1315c5d4d54313de98Alexander Potapenko EXPECT_DEATH(strcat(str, str + 11), OverlapErrorMessage("strcat")); 15210985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany str[10] = '\0'; 15220985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany strcat(str + 11, str); 1523c5e72a3b7c60f1b2d9d9be3a07d397d5b5f872beKostya Serebryany EXPECT_DEATH(strcat(str, str + 9), OverlapErrorMessage("strcat")); 1524c5e72a3b7c60f1b2d9d9be3a07d397d5b5f872beKostya Serebryany EXPECT_DEATH(strcat(str + 9, str), OverlapErrorMessage("strcat")); 1525c5e72a3b7c60f1b2d9d9be3a07d397d5b5f872beKostya Serebryany EXPECT_DEATH(strcat(str + 10, str), OverlapErrorMessage("strcat")); 15260985ca240812ac5519168a6aecbccf4c513ae243Kostya Serebryany 1527c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov // Check "strncat". 1528c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov memset(str, 'z', size); 1529c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov str[10] = '\0'; 1530c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov strncat(str, str + 10, 10); // from is empty 153137b3fcd6fdec5740fe51fc1315c5d4d54313de98Alexander Potapenko EXPECT_DEATH(strncat(str, str + 11, 10), OverlapErrorMessage("strncat")); 1532c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov str[10] = '\0'; 1533c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov str[20] = '\0'; 1534c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov strncat(str + 5, str, 5); 1535c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov str[10] = '\0'; 1536c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov EXPECT_DEATH(strncat(str + 5, str, 6), OverlapErrorMessage("strncat")); 1537c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov EXPECT_DEATH(strncat(str, str + 9, 10), OverlapErrorMessage("strncat")); 1538c1bdd5adca453f2fae238a4c53ade35ae60b85daAlexey Samsonov 15391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(str); 15401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 15411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1542847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonovvoid CallAtoi(const char *nptr) { 1543847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov Ident(atoi(nptr)); 1544847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov} 1545847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonovvoid CallAtol(const char *nptr) { 1546847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov Ident(atol(nptr)); 1547847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov} 1548847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonovvoid CallAtoll(const char *nptr) { 1549847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov Ident(atoll(nptr)); 1550847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov} 1551847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonovtypedef void(*PointerToCallAtoi)(const char*); 1552847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov 1553847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonovvoid RunAtoiOOBTest(PointerToCallAtoi Atoi) { 1554847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov char *array = MallocAndMemsetString(10, '1'); 1555847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov // Invalid pointer to the string. 1556ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Atoi(array + 11), RightOOBReadMessage(1)); 1557ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Atoi(array - 1), LeftOOBReadMessage(1)); 1558847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov // Die if a buffer doesn't have terminating NULL. 1559ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Atoi(array), RightOOBReadMessage(0)); 1560847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov // Make last symbol a terminating NULL or other non-digit. 1561847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov array[9] = '\0'; 1562847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov Atoi(array); 1563847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov array[9] = 'a'; 1564847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov Atoi(array); 1565847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov Atoi(array + 9); 1566847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov // Sometimes we need to detect overflow if no digits are found. 1567847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov memset(array, ' ', 10); 1568ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Atoi(array), RightOOBReadMessage(0)); 1569847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov array[9] = '-'; 1570ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Atoi(array), RightOOBReadMessage(0)); 1571ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Atoi(array + 9), RightOOBReadMessage(0)); 1572847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov array[8] = '-'; 1573847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov Atoi(array); 1574fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany free(array); 1575847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov} 1576847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov 1577847f932ab0405757946433b81d3b2952b306b0bcAlexey SamsonovTEST(AddressSanitizer, AtoiAndFriendsOOBTest) { 1578847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov RunAtoiOOBTest(&CallAtoi); 1579847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov RunAtoiOOBTest(&CallAtol); 1580847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov RunAtoiOOBTest(&CallAtoll); 1581847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov} 1582847f932ab0405757946433b81d3b2952b306b0bcAlexey Samsonov 15838f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonovvoid CallStrtol(const char *nptr, char **endptr, int base) { 15848f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov Ident(strtol(nptr, endptr, base)); 15858f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov} 15868f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonovvoid CallStrtoll(const char *nptr, char **endptr, int base) { 15878f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov Ident(strtoll(nptr, endptr, base)); 15888f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov} 15898f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonovtypedef void(*PointerToCallStrtol)(const char*, char**, int); 15908f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov 15918f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonovvoid RunStrtolOOBTest(PointerToCallStrtol Strtol) { 159284ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov char *array = MallocAndMemsetString(3); 159384ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov char *endptr = NULL; 159484ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov array[0] = '1'; 159584ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov array[1] = '2'; 159684ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov array[2] = '3'; 159784ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov // Invalid pointer to the string. 1598ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Strtol(array + 3, NULL, 0), RightOOBReadMessage(0)); 1599ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Strtol(array - 1, NULL, 0), LeftOOBReadMessage(1)); 160084ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov // Buffer overflow if there is no terminating null (depends on base). 16018f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov Strtol(array, &endptr, 3); 160284ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov EXPECT_EQ(array + 2, endptr); 1603ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Strtol(array, NULL, 0), RightOOBReadMessage(0)); 160484ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov array[2] = 'z'; 16058f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov Strtol(array, &endptr, 35); 160684ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov EXPECT_EQ(array + 2, endptr); 1607ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Strtol(array, NULL, 36), RightOOBReadMessage(0)); 160884ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov // Add terminating zero to get rid of overflow. 160984ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov array[2] = '\0'; 16108f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov Strtol(array, NULL, 36); 161184ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov // Don't check for overflow if base is invalid. 16128f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov Strtol(array - 1, NULL, -1); 16138f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov Strtol(array + 3, NULL, 1); 161484ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov // Sometimes we need to detect overflow if no digits are found. 161584ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov array[0] = array[1] = array[2] = ' '; 1616ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Strtol(array, NULL, 0), RightOOBReadMessage(0)); 161784ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov array[2] = '+'; 1618ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Strtol(array, NULL, 0), RightOOBReadMessage(0)); 161984ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov array[2] = '-'; 1620ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(Strtol(array, NULL, 0), RightOOBReadMessage(0)); 162184ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov array[1] = '+'; 16228f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov Strtol(array, NULL, 0); 162384ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov array[1] = array[2] = 'z'; 16248f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov Strtol(array, &endptr, 0); 162584ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov EXPECT_EQ(array, endptr); 16268f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov Strtol(array + 2, NULL, 0); 162784ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov EXPECT_EQ(array, endptr); 1628fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany free(array); 162984ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov} 163084ba324b94d22a69acc823e845f9718fd2984e44Alexey Samsonov 16318f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey SamsonovTEST(AddressSanitizer, StrtollOOBTest) { 16328f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov RunStrtolOOBTest(&CallStrtoll); 16338f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov} 16348f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey SamsonovTEST(AddressSanitizer, StrtolOOBTest) { 16358f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov RunStrtolOOBTest(&CallStrtol); 16368f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov} 16378f6a77f3f59ebc5f40f235e51836f468f30b9110Alexey Samsonov 16381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// At the moment we instrument memcpy/memove/memset calls at compile time so we 16391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// can't handle OOB error if these functions are called by pointer, see disabled 16401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// MemIntrinsicCallByPointerTest below 16411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanytypedef void*(*PointerToMemTransfer)(void*, const void*, size_t); 16421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanytypedef void*(*PointerToMemSet)(void*, int, size_t); 16431e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 16441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid CallMemSetByPointer(PointerToMemSet MemSet) { 16451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany size_t size = Ident(100); 16461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *array = Ident((char*)malloc(size)); 1647ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(MemSet(array, 0, 101), RightOOBWriteMessage(0)); 16481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(array); 16491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 16501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 16511e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid CallMemTransferByPointer(PointerToMemTransfer MemTransfer) { 16521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany size_t size = Ident(100); 16531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *src = Ident((char*)malloc(size)); 16541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *dst = Ident((char*)malloc(size)); 1655ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(MemTransfer(dst, src, 101), RightOOBWriteMessage(0)); 16561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(src); 16571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(dst); 16581e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 16591e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 16601e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_MemIntrinsicCallByPointerTest) { 16611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany CallMemSetByPointer(&memset); 16621e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany CallMemTransferByPointer(&memcpy); 16631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany CallMemTransferByPointer(&memmove); 16641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 16651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1666ca2849c2819b5c7a8771a1e8bc449cf8f5ef6527Alexey Samsonov#if defined(__linux__) && !defined(ANDROID) && !defined(__ANDROID__) 1667c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany#define READ_TEST(READ_N_BYTES) \ 1668c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany char *x = new char[10]; \ 1669c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany int fd = open("/proc/self/stat", O_RDONLY); \ 1670c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany ASSERT_GT(fd, 0); \ 1671c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany EXPECT_DEATH(READ_N_BYTES, \ 1672c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany ASAN_PCRE_DOTALL \ 1673c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany "AddressSanitizer: heap-buffer-overflow" \ 1674c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany ".* is located 0 bytes to the right of 10-byte region"); \ 1675c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany close(fd); \ 1676c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany delete [] x; \ 1677c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany 16788530e2b953f0b34ecd267a6aba5f155d5c08c5c8Kostya SerebryanyTEST(AddressSanitizer, pread) { 1679c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany READ_TEST(pread(fd, x, 15, 0)); 16808530e2b953f0b34ecd267a6aba5f155d5c08c5c8Kostya Serebryany} 16818530e2b953f0b34ecd267a6aba5f155d5c08c5c8Kostya Serebryany 16828530e2b953f0b34ecd267a6aba5f155d5c08c5c8Kostya SerebryanyTEST(AddressSanitizer, pread64) { 1683c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany READ_TEST(pread64(fd, x, 15, 0)); 16848530e2b953f0b34ecd267a6aba5f155d5c08c5c8Kostya Serebryany} 16858530e2b953f0b34ecd267a6aba5f155d5c08c5c8Kostya Serebryany 16868530e2b953f0b34ecd267a6aba5f155d5c08c5c8Kostya SerebryanyTEST(AddressSanitizer, read) { 1687c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany READ_TEST(read(fd, x, 15)); 1688c20b321d49f0eff60f1394d56e623d8ca94f24d7Kostya Serebryany} 1689ca2849c2819b5c7a8771a1e8bc449cf8f5ef6527Alexey Samsonov#endif // defined(__linux__) && !defined(ANDROID) && !defined(__ANDROID__) 16908530e2b953f0b34ecd267a6aba5f155d5c08c5c8Kostya Serebryany 16911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// This test case fails 16921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Clang optimizes memcpy/memset calls which lead to unaligned access 16931e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_MemIntrinsicUnalignedAccessTest) { 16941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int size = Ident(4096); 16951e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *s = Ident((char*)malloc(size)); 1696ef8dfd8dffbf4385baf24e5aaea6c2cc7c6c936aAlexander Potapenko EXPECT_DEATH(memset(s + size - 1, 0, 2), RightOOBWriteMessage(0)); 16971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(s); 16981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 16991e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 17001e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// TODO(samsonov): Add a test with malloc(0) 17011e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// TODO(samsonov): Add tests for str* and mem* functions. 17021e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1703938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE static int LargeFunction(bool do_bad_access) { 17041e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *x = new int[100]; 17051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[0]++; 17061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[1]++; 17071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[2]++; 17081e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[3]++; 17091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[4]++; 17101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[5]++; 17111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[6]++; 17121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[7]++; 17131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[8]++; 17141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[9]++; 17151e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 17161e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[do_bad_access ? 100 : 0]++; int res = __LINE__; 17171e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 17181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[10]++; 17191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[11]++; 17201e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[12]++; 17211e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[13]++; 17221e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[14]++; 17231e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[15]++; 17241e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[16]++; 17251e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[17]++; 17261e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[18]++; 17271e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany x[19]++; 17281e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 17291e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany delete x; 17301e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return res; 17311e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 17321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 17331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Test the we have correct debug info for the failing instruction. 17341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// This test requires the in-process symbolizer to be enabled by default. 17351e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_LargeFunctionSymbolizeTest) { 17361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int failing_line = LargeFunction(false); 17371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char expected_warning[128]; 17385a15541fc2bbe46a5ce0cd802b5c1055f2ffb14eKostya Serebryany sprintf(expected_warning, "LargeFunction.*asan_test.*:%d", failing_line); 17391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(LargeFunction(true), expected_warning); 17401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 17411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 17421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Check that we unwind and symbolize correctly. 17431e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_MallocFreeUnwindAndSymbolizeTest) { 17441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *a = (int*)malloc_aaa(sizeof(int)); 17451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany *a = 1; 17461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free_aaa(a); 17471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(*a = 1, "free_ccc.*free_bbb.*free_aaa.*" 17481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "malloc_fff.*malloc_eee.*malloc_ddd"); 17491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 17501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1751b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryanystatic bool TryToSetThreadName(const char *name) { 1752b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany#if defined(__linux__) && defined(PR_SET_NAME) 1753b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany return 0 == prctl(PR_SET_NAME, (unsigned long)name, 0, 0, 0); 1754b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany#else 1755b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany return false; 1756716e2f25123bf9b20fbc6b582803a3929b78b96dKostya Serebryany#endif 1757716e2f25123bf9b20fbc6b582803a3929b78b96dKostya Serebryany} 1758716e2f25123bf9b20fbc6b582803a3929b78b96dKostya Serebryany 17591e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid *ThreadedTestAlloc(void *a) { 1760b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany EXPECT_EQ(true, TryToSetThreadName("AllocThr")); 17611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int **p = (int**)a; 17621e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany *p = new int; 17631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return 0; 17641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 17651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 17661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid *ThreadedTestFree(void *a) { 1767b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany EXPECT_EQ(true, TryToSetThreadName("FreeThr")); 17681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int **p = (int**)a; 17691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany delete *p; 17701e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return 0; 17711e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 17721e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 17731e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid *ThreadedTestUse(void *a) { 1774b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany EXPECT_EQ(true, TryToSetThreadName("UseThr")); 17751e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int **p = (int**)a; 17761e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany **p = 1; 17771e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return 0; 17781e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 17791e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 17801e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid ThreadedTestSpawn() { 17811e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany pthread_t t; 17821e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *x; 17832697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_CREATE(&t, 0, ThreadedTestAlloc, &x); 17842697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_JOIN(t, 0); 17852697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_CREATE(&t, 0, ThreadedTestFree, &x); 17862697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_JOIN(t, 0); 17872697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_CREATE(&t, 0, ThreadedTestUse, &x); 17882697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_JOIN(t, 0); 17891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 17901e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 17911e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, ThreadedTest) { 17921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(ThreadedTestSpawn(), 17931e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany ASAN_PCRE_DOTALL 17941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "Thread T.*created" 17951e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany ".*Thread T.*created" 17961e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany ".*Thread T.*created"); 17971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 17981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1799b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryanyvoid *ThreadedTestFunc(void *unused) { 1800b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany // Check if prctl(PR_SET_NAME) is supported. Return if not. 1801b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany if (!TryToSetThreadName("TestFunc")) 1802b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany return 0; 1803716e2f25123bf9b20fbc6b582803a3929b78b96dKostya Serebryany EXPECT_DEATH(ThreadedTestSpawn(), 1804716e2f25123bf9b20fbc6b582803a3929b78b96dKostya Serebryany ASAN_PCRE_DOTALL 1805a390ece58317a25ef26866cef6753df4b06a3e65Kostya Serebryany "WRITE .*thread T. .UseThr." 1806a390ece58317a25ef26866cef6753df4b06a3e65Kostya Serebryany ".*freed by thread T. .FreeThr. here:" 1807a390ece58317a25ef26866cef6753df4b06a3e65Kostya Serebryany ".*previously allocated by thread T. .AllocThr. here:" 1808b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany ".*Thread T. .UseThr. created by T.*TestFunc" 1809b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany ".*Thread T. .FreeThr. created by T" 1810b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany ".*Thread T. .AllocThr. created by T" 1811716e2f25123bf9b20fbc6b582803a3929b78b96dKostya Serebryany ""); 1812b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany return 0; 1813b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany} 1814b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany 1815b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya SerebryanyTEST(AddressSanitizer, ThreadNamesTest) { 1816b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany // Run ThreadedTestFunc in a separate thread because it tries to set a 1817b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany // thread name and we don't want to change the main thread's name. 1818b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany pthread_t t; 1819b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany PTHREAD_CREATE(&t, 0, ThreadedTestFunc, 0); 1820b8a59a0e030d0aacf4df1fe8f7fb4e6f82bab654Kostya Serebryany PTHREAD_JOIN(t, 0); 1821716e2f25123bf9b20fbc6b582803a3929b78b96dKostya Serebryany} 1822716e2f25123bf9b20fbc6b582803a3929b78b96dKostya Serebryany 18231e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#if ASAN_NEEDS_SEGV 18241e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, ShadowGapTest) { 18255af39e50366f1aacbebc284f572f08ad1ad07357Kostya Serebryany#if SANITIZER_WORDSIZE == 32 18261e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *addr = (char*)0x22000000; 18271e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#else 18281e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *addr = (char*)0x0000100000080000; 18291e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#endif 1830ca9b5dd9922fb7f6e0ca06dbba7db7398b8997f1Kostya Serebryany EXPECT_DEATH(*addr = 1, "AddressSanitizer: SEGV on unknown"); 18311e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 18321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#endif // ASAN_NEEDS_SEGV 18331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 18341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern "C" { 1835938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE static void UseThenFreeThenUse() { 18361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *x = Ident((char*)malloc(8)); 18371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany *x = 1; 18381e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free_aaa(x); 18391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany *x = 2; 18401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 18411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 18421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 18431e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, UseThenFreeThenUseTest) { 18441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(UseThenFreeThenUse(), "freed by thread"); 18451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 18461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 18471e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, StrDupTest) { 18481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany free(strdup(Ident("123"))); 18491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 18501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 18511e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// Currently we create and poison redzone at right of global variables. 18521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanychar glob5[5]; 18531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanystatic char static110[110]; 18541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyconst char ConstGlob[7] = {1, 2, 3, 4, 5, 6, 7}; 18551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanystatic const char StaticConstGlob[3] = {9, 8, 7}; 18561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyextern int GlobalsTest(int x); 18571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 18581e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, GlobalTest) { 18591e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany static char func_static15[15]; 18601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 18611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany static char fs1[10]; 18621e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany static char fs2[10]; 18631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany static char fs3[10]; 18641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 18651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany glob5[Ident(0)] = 0; 18661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany glob5[Ident(1)] = 0; 18671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany glob5[Ident(2)] = 0; 18681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany glob5[Ident(3)] = 0; 18691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany glob5[Ident(4)] = 0; 18701e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 18711e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(glob5[Ident(5)] = 0, 18721e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "0 bytes to the right of global variable.*glob5.* size 5"); 18731e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(glob5[Ident(5+6)] = 0, 18741e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "6 bytes to the right of global variable.*glob5.* size 5"); 18751e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(static110); // avoid optimizations 18761e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany static110[Ident(0)] = 0; 18771e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany static110[Ident(109)] = 0; 18781e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(static110[Ident(110)] = 0, 18791e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "0 bytes to the right of global variable"); 18801e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(static110[Ident(110+7)] = 0, 18811e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "7 bytes to the right of global variable"); 18821e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 18831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(func_static15); // avoid optimizations 18841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany func_static15[Ident(0)] = 0; 18851e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(func_static15[Ident(15)] = 0, 18861e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "0 bytes to the right of global variable"); 18871e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(func_static15[Ident(15 + 9)] = 0, 18881e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "9 bytes to the right of global variable"); 18891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 18901e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(fs1); 18911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(fs2); 18921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(fs3); 18931e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 18941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // We don't create left redzones, so this is not 100% guaranteed to fail. 18951e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // But most likely will. 18961e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(fs2[Ident(-1)] = 0, "is located.*of global variable"); 18971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 18981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(Ident(Ident(ConstGlob)[8]), 18991e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "is located 1 bytes to the right of .*ConstGlob"); 19001e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(Ident(Ident(StaticConstGlob)[5]), 19011e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany "is located 2 bytes to the right of .*StaticConstGlob"); 19021e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 19031e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany // call stuff from another file. 19041e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany GlobalsTest(0); 19051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 19061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 19071e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, GlobalStringConstTest) { 19081e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany static const char *zoo = "FOOBAR123"; 19091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany const char *p = Ident(zoo); 19101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_DEATH(Ident(p[15]), "is ascii string 'FOOBAR123'"); 19111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 19121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1913c37ca572876a585b6a910f14a85cc7ba7fc22f20Kostya SerebryanyTEST(AddressSanitizer, FileNameInGlobalReportTest) { 1914c37ca572876a585b6a910f14a85cc7ba7fc22f20Kostya Serebryany static char zoo[10]; 1915c37ca572876a585b6a910f14a85cc7ba7fc22f20Kostya Serebryany const char *p = Ident(zoo); 1916c37ca572876a585b6a910f14a85cc7ba7fc22f20Kostya Serebryany // The file name should be present in the report. 19175a15541fc2bbe46a5ce0cd802b5c1055f2ffb14eKostya Serebryany EXPECT_DEATH(Ident(p[15]), "zoo.*asan_test."); 1918c37ca572876a585b6a910f14a85cc7ba7fc22f20Kostya Serebryany} 1919c37ca572876a585b6a910f14a85cc7ba7fc22f20Kostya Serebryany 19201e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyint *ReturnsPointerToALocalObject() { 19211e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int a = 0; 19221e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return Ident(&a); 19231e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 19241e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1925918b18a849cea679bff31a4e0dfb3745302c240aKostya Serebryany#if ASAN_UAR == 1 19261e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, LocalReferenceReturnTest) { 19271e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *(*f)() = Ident(ReturnsPointerToALocalObject); 1928918b18a849cea679bff31a4e0dfb3745302c240aKostya Serebryany int *p = f(); 1929918b18a849cea679bff31a4e0dfb3745302c240aKostya Serebryany // Call 'f' a few more times, 'p' should still be poisoned. 1930918b18a849cea679bff31a4e0dfb3745302c240aKostya Serebryany for (int i = 0; i < 32; i++) 1931918b18a849cea679bff31a4e0dfb3745302c240aKostya Serebryany f(); 1932ca9b5dd9922fb7f6e0ca06dbba7db7398b8997f1Kostya Serebryany EXPECT_DEATH(*p = 1, "AddressSanitizer: stack-use-after-return"); 1933918b18a849cea679bff31a4e0dfb3745302c240aKostya Serebryany EXPECT_DEATH(*p = 1, "is located.*in frame .*ReturnsPointerToALocal"); 19341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 1935918b18a849cea679bff31a4e0dfb3745302c240aKostya Serebryany#endif 19361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 19371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanytemplate <int kSize> 1938938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE static void FuncWithStack() { 19391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char x[kSize]; 19401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(x)[0] = 0; 19411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(x)[kSize-1] = 0; 19421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 19431e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 19441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanystatic void LotsOfStackReuse() { 19451e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int LargeStack[10000]; 19461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(LargeStack)[0] = 0; 19471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (int i = 0; i < 10000; i++) { 19481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany FuncWithStack<128 * 1>(); 19491e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany FuncWithStack<128 * 2>(); 19501e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany FuncWithStack<128 * 4>(); 19511e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany FuncWithStack<128 * 8>(); 19521e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany FuncWithStack<128 * 16>(); 19531e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany FuncWithStack<128 * 32>(); 19541e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany FuncWithStack<128 * 64>(); 19551e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany FuncWithStack<128 * 128>(); 19561e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany FuncWithStack<128 * 256>(); 19571e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany FuncWithStack<128 * 512>(); 19581e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(LargeStack)[0] = 0; 19591e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 19601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 19611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 19621e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, StressStackReuseTest) { 19631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany LotsOfStackReuse(); 19641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 19651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 19661e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, ThreadedStressStackReuseTest) { 19671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany const int kNumThreads = 20; 19681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany pthread_t t[kNumThreads]; 19691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (int i = 0; i < kNumThreads; i++) { 19702697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_CREATE(&t[i], 0, (void* (*)(void *x))LotsOfStackReuse, 0); 19711e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 19721e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (int i = 0; i < kNumThreads; i++) { 19732697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_JOIN(t[i], 0); 19741e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 19751e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 19761e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 1977f58f998066db0231e521169d2f50af439ceecb49Kostya Serebryanystatic void *PthreadExit(void *a) { 1978f58f998066db0231e521169d2f50af439ceecb49Kostya Serebryany pthread_exit(0); 19797b7b55e99166511ce1e66bf6b79c46ae4aa60d17Evgeniy Stepanov return 0; 1980f58f998066db0231e521169d2f50af439ceecb49Kostya Serebryany} 1981f58f998066db0231e521169d2f50af439ceecb49Kostya Serebryany 1982f58f998066db0231e521169d2f50af439ceecb49Kostya SerebryanyTEST(AddressSanitizer, PthreadExitTest) { 1983f58f998066db0231e521169d2f50af439ceecb49Kostya Serebryany pthread_t t; 1984f58f998066db0231e521169d2f50af439ceecb49Kostya Serebryany for (int i = 0; i < 1000; i++) { 19852697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_CREATE(&t, 0, PthreadExit, 0); 19862697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_JOIN(t, 0); 1987f58f998066db0231e521169d2f50af439ceecb49Kostya Serebryany } 1988f58f998066db0231e521169d2f50af439ceecb49Kostya Serebryany} 1989f58f998066db0231e521169d2f50af439ceecb49Kostya Serebryany 19901e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#ifdef __EXCEPTIONS 1991938106753364d96cedc11cbdc35a15030fb44d0cTimur IskhodzhanovNOINLINE static void StackReuseAndException() { 19921e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int large_stack[1000]; 19931e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(large_stack); 19941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany ASAN_THROW(1); 19951e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 19961e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 19971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// TODO(kcc): support exceptions with use-after-return. 19981e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_StressStackReuseAndExceptionsTest) { 19991e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany for (int i = 0; i < 10000; i++) { 20001e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany try { 20011e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany StackReuseAndException(); 20021e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } catch(...) { 20031e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 20041e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 20051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 20061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany#endif 20071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 20081e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, MlockTest) { 20091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(0, mlockall(MCL_CURRENT)); 20101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(0, mlock((void*)0x12345, 0x5678)); 20111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(0, munlockall()); 20121e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany EXPECT_EQ(0, munlock((void*)0x987, 0x654)); 20131e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 20141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 2015c655cfa8dc0fdf5b521e565f6cbdbf7981fba1b2Kostya Serebryanystruct LargeStruct { 2016c655cfa8dc0fdf5b521e565f6cbdbf7981fba1b2Kostya Serebryany int foo[100]; 2017c655cfa8dc0fdf5b521e565f6cbdbf7981fba1b2Kostya Serebryany}; 2018c655cfa8dc0fdf5b521e565f6cbdbf7981fba1b2Kostya Serebryany 2019c655cfa8dc0fdf5b521e565f6cbdbf7981fba1b2Kostya Serebryany// Test for bug http://llvm.org/bugs/show_bug.cgi?id=11763. 2020c655cfa8dc0fdf5b521e565f6cbdbf7981fba1b2Kostya Serebryany// Struct copy should not cause asan warning even if lhs == rhs. 2021c655cfa8dc0fdf5b521e565f6cbdbf7981fba1b2Kostya SerebryanyTEST(AddressSanitizer, LargeStructCopyTest) { 2022c655cfa8dc0fdf5b521e565f6cbdbf7981fba1b2Kostya Serebryany LargeStruct a; 2023c655cfa8dc0fdf5b521e565f6cbdbf7981fba1b2Kostya Serebryany *Ident(&a) = *Ident(&a); 2024c655cfa8dc0fdf5b521e565f6cbdbf7981fba1b2Kostya Serebryany} 2025c655cfa8dc0fdf5b521e565f6cbdbf7981fba1b2Kostya Serebryany 20261cffef3e39cad4eb33606afbebac78d6637361acAlexey SamsonovATTRIBUTE_NO_ADDRESS_SAFETY_ANALYSIS 20273be19f4fa3519e358fd41a5feda82c39a22ef107Kostya Serebryanystatic void NoAddressSafety() { 20283be19f4fa3519e358fd41a5feda82c39a22ef107Kostya Serebryany char *foo = new char[10]; 20293be19f4fa3519e358fd41a5feda82c39a22ef107Kostya Serebryany Ident(foo)[10] = 0; 20303be19f4fa3519e358fd41a5feda82c39a22ef107Kostya Serebryany delete [] foo; 20313be19f4fa3519e358fd41a5feda82c39a22ef107Kostya Serebryany} 20323be19f4fa3519e358fd41a5feda82c39a22ef107Kostya Serebryany 20333be19f4fa3519e358fd41a5feda82c39a22ef107Kostya SerebryanyTEST(AddressSanitizer, AttributeNoAddressSafetyTest) { 20343be19f4fa3519e358fd41a5feda82c39a22ef107Kostya Serebryany Ident(NoAddressSafety)(); 20353be19f4fa3519e358fd41a5feda82c39a22ef107Kostya Serebryany} 20363be19f4fa3519e358fd41a5feda82c39a22ef107Kostya Serebryany 203729e09223b9d4f53f1be528d6a572b66034c26611Alexey Samsonov// TODO(glider): Enable this test on Mac. 203829e09223b9d4f53f1be528d6a572b66034c26611Alexey Samsonov// It doesn't work on Android, as calls to new/delete go through malloc/free. 203929e09223b9d4f53f1be528d6a572b66034c26611Alexey Samsonov#if !defined(__APPLE__) && !defined(ANDROID) && !defined(__ANDROID__) 2040fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryanystatic string MismatchStr(const string &str) { 2041fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany return string("AddressSanitizer: alloc-dealloc-mismatch \\(") + str; 2042fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany} 2043fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany 20448f0e311767b783db4a15139dddfe7a924fa3724eAlexey SamsonovTEST(AddressSanitizer, AllocDeallocMismatch) { 2045fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany EXPECT_DEATH(free(Ident(new int)), 2046fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany MismatchStr("operator new vs free")); 2047fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany EXPECT_DEATH(free(Ident(new int[2])), 2048fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany MismatchStr("operator new \\[\\] vs free")); 2049fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany EXPECT_DEATH(delete (Ident(new int[2])), 2050fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany MismatchStr("operator new \\[\\] vs operator delete")); 2051fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany EXPECT_DEATH(delete (Ident((int*)malloc(2 * sizeof(int)))), 2052fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany MismatchStr("malloc vs operator delete")); 2053fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany EXPECT_DEATH(delete [] (Ident(new int)), 2054fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany MismatchStr("operator new vs operator delete \\[\\]")); 2055fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany EXPECT_DEATH(delete [] (Ident((int*)malloc(2 * sizeof(int)))), 2056fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany MismatchStr("malloc vs operator delete \\[\\]")); 2057fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany} 205829e09223b9d4f53f1be528d6a572b66034c26611Alexey Samsonov#endif 2059fe6d91684bcda766593800f6307233f1a33d31f6Kostya Serebryany 20601e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// ------------------ demo tests; run each one-by-one ------------- 20611e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany// e.g. --gtest_filter=*DemoOOBLeftHigh --gtest_also_run_disabled_tests 20621e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_DemoThreadedTest) { 20631e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany ThreadedTestSpawn(); 20641e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 20651e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 20661e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryanyvoid *SimpleBugOnSTack(void *x = 0) { 20671e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char a[20]; 20681e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(a)[20] = 0; 20691e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany return 0; 20701e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 20711e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 20721e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_DemoStackTest) { 20731e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany SimpleBugOnSTack(); 20741e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 20751e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 20761e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_DemoThreadStackTest) { 20771e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany pthread_t t; 20782697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_CREATE(&t, 0, SimpleBugOnSTack, 0); 20792697687059e64a4f1319dc23a0a3ca59982d53f6Kostya Serebryany PTHREAD_JOIN(t, 0); 20801e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 20811e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 20821e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_DemoUAFLowIn) { 20831e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany uaf_test<U1>(10, 0); 20841e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 20851e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_DemoUAFLowLeft) { 20861e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany uaf_test<U1>(10, -2); 20871e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 20881e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_DemoUAFLowRight) { 20891e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany uaf_test<U1>(10, 10); 20901e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 20911e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 20921e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_DemoUAFHigh) { 20931e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany uaf_test<U1>(kLargeMalloc, 0); 20941e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 20951e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 20961e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_DemoOOBLeftLow) { 20971e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany oob_test<U1>(10, -1); 20981e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 20991e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 21001e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_DemoOOBLeftHigh) { 21011e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany oob_test<U1>(kLargeMalloc, -1); 21021e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 21031e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 21041e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_DemoOOBRightLow) { 21051e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany oob_test<U1>(10, 10); 21061e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 21071e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 21081e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_DemoOOBRightHigh) { 21091e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany oob_test<U1>(kLargeMalloc, kLargeMalloc); 21101e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 21111e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 21121e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_DemoOOM) { 21135af39e50366f1aacbebc284f572f08ad1ad07357Kostya Serebryany size_t size = SANITIZER_WORDSIZE == 64 ? (size_t)(1ULL << 40) : (0xf0000000); 21141e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany printf("%p\n", malloc(size)); 21151e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 21161e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 21171e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_DemoDoubleFreeTest) { 21181e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany DoubleFree(); 21191e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 21201e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 21211e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_DemoNullDerefTest) { 21221e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany int *a = 0; 21231e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(a)[10] = 0; 21241e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 21251e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 21261e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_DemoFunctionStaticTest) { 21271e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany static char a[100]; 21281e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany static char b[100]; 21291e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany static char c[100]; 21301e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(a); 21311e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(b); 21321e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(c); 21331e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(a)[5] = 0; 21341e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(b)[105] = 0; 21351e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany Ident(a)[5] = 0; 21361e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 21371e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 21381e172b4bdec57329bf904f063a29f99cddf2d85fKostya SerebryanyTEST(AddressSanitizer, DISABLED_DemoTooMuchMemoryTest) { 21391e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany const size_t kAllocSize = (1 << 28) - 1024; 21401e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany size_t total_size = 0; 21411e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany while (true) { 21421e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany char *x = (char*)malloc(kAllocSize); 21431e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany memset(x, 0, kAllocSize); 21441e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany total_size += kAllocSize; 21450aa04b3f432bc55e7c116d5280685ee3f243be8eKostya Serebryany fprintf(stderr, "total: %ldM %p\n", (long)total_size >> 20, x); 21461e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany } 21471e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany} 21481e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 21494eaa17848ed02cae465f10aa5fe095e42dbf0d61Kostya Serebryany// http://code.google.com/p/address-sanitizer/issues/detail?id=66 21500796393f2088e3b75c4110934451160eeb037fabKostya SerebryanyTEST(AddressSanitizer, BufferOverflowAfterManyFrees) { 21514eaa17848ed02cae465f10aa5fe095e42dbf0d61Kostya Serebryany for (int i = 0; i < 1000000; i++) { 21524eaa17848ed02cae465f10aa5fe095e42dbf0d61Kostya Serebryany delete [] (Ident(new char [8644])); 21534eaa17848ed02cae465f10aa5fe095e42dbf0d61Kostya Serebryany } 21544eaa17848ed02cae465f10aa5fe095e42dbf0d61Kostya Serebryany char *x = new char[8192]; 2155ca9b5dd9922fb7f6e0ca06dbba7db7398b8997f1Kostya Serebryany EXPECT_DEATH(x[Ident(8192)] = 0, "AddressSanitizer: heap-buffer-overflow"); 21564eaa17848ed02cae465f10aa5fe095e42dbf0d61Kostya Serebryany delete [] Ident(x); 21574eaa17848ed02cae465f10aa5fe095e42dbf0d61Kostya Serebryany} 21584eaa17848ed02cae465f10aa5fe095e42dbf0d61Kostya Serebryany 21591e172b4bdec57329bf904f063a29f99cddf2d85fKostya Serebryany 21605b6eab9dc5572a66e3af54ab087255ffa4dd5185Evgeniy Stepanov// Test that instrumentation of stack allocations takes into account 21615b6eab9dc5572a66e3af54ab087255ffa4dd5185Evgeniy Stepanov// AllocSize of a type, and not its StoreSize (16 vs 10 bytes for long double). 21625b6eab9dc5572a66e3af54ab087255ffa4dd5185Evgeniy Stepanov// See http://llvm.org/bugs/show_bug.cgi?id=12047 for more details. 21635b6eab9dc5572a66e3af54ab087255ffa4dd5185Evgeniy StepanovTEST(AddressSanitizer, LongDoubleNegativeTest) { 21645b6eab9dc5572a66e3af54ab087255ffa4dd5185Evgeniy Stepanov long double a, b; 21659b90e95d4dc59c6d97ed73b2470f5e21e7b346d5Kostya Serebryany static long double c; 21665b6eab9dc5572a66e3af54ab087255ffa4dd5185Evgeniy Stepanov memcpy(Ident(&a), Ident(&b), sizeof(long double)); 21679b90e95d4dc59c6d97ed73b2470f5e21e7b346d5Kostya Serebryany memcpy(Ident(&c), Ident(&b), sizeof(long double)); 216876e842845c927e92c2c21c41ac51e364fa17528cAlexey Samsonov} 2169