msan_test.cc revision 02f4a942bff84f1266571740456dd9baa230d87b
10231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov//===-- msan_test.cc ------------------------------------------------------===// 20231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// 30231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// The LLVM Compiler Infrastructure 40231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// 50231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// This file is distributed under the University of Illinois Open Source 60231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// License. See LICENSE.TXT for details. 70231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// 80231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov//===----------------------------------------------------------------------===// 90231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// 100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// This file is a part of MemorySanitizer. 110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// 120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// MemorySanitizer unit tests. 130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov//===----------------------------------------------------------------------===// 140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include "sanitizer/msan_interface.h" 160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include "msandr_test_so.h" 170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include "gtest/gtest.h" 180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <stdlib.h> 200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <stdarg.h> 210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <stdio.h> 220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <assert.h> 230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <wchar.h> 240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 25e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov#include <dlfcn.h> 260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <unistd.h> 270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <limits.h> 280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <sys/time.h> 290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <sys/types.h> 300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <sys/stat.h> 310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <fcntl.h> 320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <sys/resource.h> 330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <sys/ioctl.h> 340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <sys/utsname.h> 350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <sys/mman.h> 360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <sys/vfs.h> 370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#if defined(__i386__) || defined(__x86_64__) 390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov# include <emmintrin.h> 400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov# define MSAN_HAS_M128 1 410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#else 420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov# define MSAN_HAS_M128 0 430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#endif 440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtypedef unsigned char U1; 460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtypedef unsigned short U2; // NOLINT 470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtypedef unsigned int U4; 480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtypedef unsigned long long U8; // NOLINT 490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtypedef signed char S1; 500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtypedef signed short S2; // NOLINT 510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtypedef signed int S4; 520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtypedef signed long long S8; // NOLINT 530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#define NOINLINE __attribute__((noinline)) 540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#define INLINE __attribute__((always_inline)) 550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovstatic bool TrackingOrigins() { 5711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov S8 x; 5811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov __msan_set_origin(&x, sizeof(x), 0x1234); 5911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov u32 origin = __msan_get_origin(&x); 6011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov __msan_set_origin(&x, sizeof(x), 0); 6111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov return origin == 0x1234; 6211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov} 630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 6411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov#define EXPECT_UMR(action) \ 650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov do { \ 660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_expect_umr(1); \ 670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov action; \ 680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_expect_umr(0); \ 690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } while (0) 700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 7111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov#define EXPECT_UMR_O(action, origin) \ 720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov do { \ 730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_expect_umr(1); \ 740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov action; \ 750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_expect_umr(0); \ 760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (TrackingOrigins()) \ 770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(origin, __msan_get_origin_tls()); \ 780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } while (0) 790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 8011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov#define EXPECT_UMR_S(action, stack_origin) \ 810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov do { \ 820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_expect_umr(1); \ 830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov action; \ 840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_expect_umr(0); \ 850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov u32 id = __msan_get_origin_tls(); \ 860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov const char *str = __msan_get_origin_descr_if_stack(id); \ 870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!str || strcmp(str, stack_origin)) { \ 880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov fprintf(stderr, "EXPECT_POISONED_S: id=%u %s, %s", \ 890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov id, stack_origin, str); \ 900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(1, 0); \ 910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } \ 920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } while (0) 930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov#define EXPECT_POISONED(x) ExpectPoisoned(x) 9511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 9611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovtemplate<typename T> 9711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovvoid ExpectPoisoned(const T& t) { 9811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NE(-1, __msan_test_shadow((void*)&t, sizeof(t))); 9911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov} 10011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 10111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov#define EXPECT_POISONED_O(x, origin) \ 10211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov ExpectPoisonedWithOrigin(x, origin) 10311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 10411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovtemplate<typename T> 10511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovvoid ExpectPoisonedWithOrigin(const T& t, unsigned origin) { 10611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NE(-1, __msan_test_shadow((void*)&t, sizeof(t))); 10711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov if (TrackingOrigins()) 10811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_EQ(origin, __msan_get_origin((void*)&t)); 10911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov} 11011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 11111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov#define EXPECT_POISONED_S(x, stack_origin) \ 11211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov ExpectPoisonedWithStackOrigin(x, stack_origin) 11311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 11411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovtemplate<typename T> 11511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovvoid ExpectPoisonedWithStackOrigin(const T& t, const char *stack_origin) { 11611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NE(-1, __msan_test_shadow((void*)&t, sizeof(t))); 11711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov u32 id = __msan_get_origin((void*)&t); 11811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov const char *str = __msan_get_origin_descr_if_stack(id); 11911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov if (!str || strcmp(str, stack_origin)) { 12011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov fprintf(stderr, "EXPECT_POISONED_S: id=%u %s, %s", 12111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov id, stack_origin, str); 12211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_EQ(1, 0); 12311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov } 12411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov} 12511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 12611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov#define EXPECT_NOT_POISONED(x) ExpectNotPoisoned(x) 12711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 12811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovtemplate<typename T> 12911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovvoid ExpectNotPoisoned(const T& t) { 13011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_EQ(-1, __msan_test_shadow((void*)&t, sizeof(t))); 13111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov} 1320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic U8 poisoned_array[100]; 1340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> 1350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovT *GetPoisoned(int i = 0, T val = 0) { 1360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *res = (T*)&poisoned_array[i]; 1370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *res = val; 1380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&poisoned_array[i], sizeof(T)); 1390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 1400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 1410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> 1430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovT *GetPoisonedO(int i, u32 origin, T val = 0) { 1440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *res = (T*)&poisoned_array[i]; 1450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *res = val; 1460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&poisoned_array[i], sizeof(T)); 1470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(&poisoned_array[i], sizeof(T), origin); 1480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 1490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 1500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// This function returns its parameter but in such a way that compiler 1520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// can not prove it. 1530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> 1540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE 1550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic T Ident(T t) { 1560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov volatile T ret = t; 1570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return ret; 1580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 1590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> NOINLINE T ReturnPoisoned() { return *GetPoisoned<T>(); } 1610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic volatile int g_one = 1; 1630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic volatile int g_zero = 0; 1640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic volatile int g_0 = 0; 1650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic volatile int g_1 = 1; 1660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovS4 a_s4[100]; 1680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovS8 a_s8[100]; 1690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, NegativeTest1) { 1710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 *x = GetPoisoned<S4>(); 1720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (g_one) 1730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *x = 0; 17411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*x); 1750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 1760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, PositiveTest1) { 1780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // Load to store. 17911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S1>()); 18011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S2>()); 18111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S4>()); 18211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S8>()); 1830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // S->S conversions. 18511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S1>()); 18611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S1>()); 18711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S1>()); 1880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 18911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S2>()); 19011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S2>()); 19111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S2>()); 1920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 19311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S4>()); 19411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S4>()); 19511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S4>()); 1960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 19711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S8>()); 19811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S8>()); 19911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S8>()); 2000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // ZExt 20211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<U1>()); 20311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<U1>()); 20411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<U1>()); 20511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<U2>()); 20611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<U2>()); 20711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<U4>()); 2080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // Unary ops. 21011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(- *GetPoisoned<S4>()); 2110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 21211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR(a_s4[g_zero] = 100 / *GetPoisoned<S4>(0, 1)); 2130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov a_s4[g_zero] = 1 - *GetPoisoned<S4>(); 2160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov a_s4[g_zero] = 1 + *GetPoisoned<S4>(); 2170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 2180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, Phi1) { 2200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 c; 2210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (g_one) { 2220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov c = *GetPoisoned<S4>(); 2230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } else { 2240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_break_optimization(0); 2250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov c = 0; 2260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } 22711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(c); 2280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 2290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, Phi2) { 2310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 i = *GetPoisoned<S4>(); 2320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 n = g_one; 23311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR(for (; i < g_one; i++);); 23411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(i); 2350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 2360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 23711929000ec2919192b3be457f5a44c71ed55215eEvgeniy StepanovNOINLINE void Arg1ExpectUMR(S4 a1) { EXPECT_POISONED(a1); } 23811929000ec2919192b3be457f5a44c71ed55215eEvgeniy StepanovNOINLINE void Arg2ExpectUMR(S4 a1, S4 a2) { EXPECT_POISONED(a2); } 23911929000ec2919192b3be457f5a44c71ed55215eEvgeniy StepanovNOINLINE void Arg3ExpectUMR(S1 a1, S4 a2, S8 a3) { EXPECT_POISONED(a3); } 2400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, ArgTest) { 2420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov Arg1ExpectUMR(*GetPoisoned<S4>()); 2430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov Arg2ExpectUMR(0, *GetPoisoned<S4>()); 2440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov Arg3ExpectUMR(0, 1, *GetPoisoned<S8>()); 2450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 2460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, CallAndRet) { 2490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) return; 2500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ReturnPoisoned<S1>(); 2510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ReturnPoisoned<S2>(); 2520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ReturnPoisoned<S4>(); 2530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ReturnPoisoned<S8>(); 2540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 25511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(ReturnPoisoned<S1>()); 25611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(ReturnPoisoned<S2>()); 25711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(ReturnPoisoned<S4>()); 25811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(ReturnPoisoned<S8>()); 2590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 2600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// malloc() in the following test may be optimized to produce a compile-time 2620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// undef value. Check that we trap on the volatile assignment anyway. 2630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, DISABLED_MallocNoIdent) { 2640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 *x = (int*)malloc(sizeof(S4)); 26511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*x); 2660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov free(x); 2670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 2680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, Malloc) { 2700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 *x = (int*)Ident(malloc(sizeof(S4))); 27111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*x); 2720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov free(x); 2730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 2740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, Realloc) { 2760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 *x = (int*)Ident(realloc(0, sizeof(S4))); 27711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[0]); 2780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[0] = 1; 2790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x = (int*)Ident(realloc(x, 2 * sizeof(S4))); 28011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[0]); // Ok, was inited before. 28111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[1]); 2820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x = (int*)Ident(realloc(x, 3 * sizeof(S4))); 28311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[0]); // Ok, was inited before. 28411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[2]); 28511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[1]); 2860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[2] = 1; // Init this here. Check that after realloc it is poisoned again. 2870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x = (int*)Ident(realloc(x, 2 * sizeof(S4))); 28811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[0]); // Ok, was inited before. 28911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[1]); 2900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x = (int*)Ident(realloc(x, 3 * sizeof(S4))); 29111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[1]); 29211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[2]); 2930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov free(x); 2940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 2950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, Calloc) { 2970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 *x = (int*)Ident(calloc(1, sizeof(S4))); 29811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*x); // Should not be poisoned. 2990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // EXPECT_EQ(0, *x); 3000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov free(x); 3010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 3020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 3030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, AndOr) { 3040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U4 *p = GetPoisoned<U4>(); 3050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // We poison two bytes in the midle of a 4-byte word to make the test 3060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // correct regardless of endianness. 3070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ((U1*)p)[1] = 0; 3080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ((U1*)p)[2] = 0xff; 30911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*p & 0x00ffff00); 31011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*p & 0x00ff0000); 31111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*p & 0x0000ff00); 31211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*p & 0xff000000); 31311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*p & 0x000000ff); 31411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*p & 0x0000ffff); 31511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*p & 0xffff0000); 3160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 31711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*p | 0xff0000ff); 31811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*p | 0xff00ffff); 31911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*p | 0xffff00ff); 32011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*p | 0xff000000); 32111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*p | 0x000000ff); 32211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*p | 0x0000ffff); 32311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*p | 0xffff0000); 3240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 32511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<bool>() & *GetPoisoned<bool>()); 3260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 3270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 3280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> 32911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovstatic bool applyNot(T value, T shadow) { 3300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_partial_poison(&value, &shadow, sizeof(T)); 33111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov return !value; 3320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 3330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 3340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, Not) { 33511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U4>(0x0, 0x0)); 33611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U4>(0xFFFFFFFF, 0x0)); 33711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(applyNot<U4>(0xFFFFFFFF, 0xFFFFFFFF)); 33811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U4>(0xFF000000, 0x0FFFFFFF)); 33911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U4>(0xFF000000, 0x00FFFFFF)); 34011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U4>(0xFF000000, 0x0000FFFF)); 34111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U4>(0xFF000000, 0x00000000)); 34211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(applyNot<U4>(0xFF000000, 0xFF000000)); 34311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U4>(0xFF800000, 0xFF000000)); 34411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(applyNot<U4>(0x00008000, 0x00008000)); 34511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 34611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U1>(0x0, 0x0)); 34711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U1>(0xFF, 0xFE)); 34811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U1>(0xFF, 0x0)); 34911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(applyNot<U1>(0xFF, 0xFF)); 35011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 35111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(applyNot<void*>((void*)0xFFFFFF, (void*)(-1))); 35211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<void*>((void*)0xFFFFFF, (void*)(-2))); 3530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 3540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 3550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, Shift) { 3560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U4 *up = GetPoisoned<U4>(); 3570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ((U1*)up)[0] = 0; 3580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ((U1*)up)[3] = 0xff; 35911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*up >> 30); 36011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*up >> 24); 36111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*up >> 23); 36211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*up >> 10); 3630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 36411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*up << 30); 36511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*up << 24); 36611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*up << 23); 36711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*up << 10); 3680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 3690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 *sp = (S4*)up; 37011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*sp >> 30); 37111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*sp >> 24); 37211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*sp >> 23); 37311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*sp >> 10); 3740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 3750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov sp = GetPoisoned<S4>(); 3760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ((S1*)sp)[1] = 0; 3770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ((S1*)sp)[2] = 0; 37811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*sp >> 31); 3790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 38011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(100 >> *GetPoisoned<S4>()); 38111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(100U >> *GetPoisoned<S4>()); 3820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 3830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 3840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE static int GetPoisonedZero() { 3850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int *zero = new int; 3860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *zero = 0; 3870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(zero, sizeof(*zero)); 3880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = *zero; 3890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete zero; 3900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 3910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 3920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 3930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, LoadFromDirtyAddress) { 3940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int *a = new int; 3950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *a = 0; 39611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR(__msan_break_optimization((void*)(U8)a[GetPoisonedZero()])); 3970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete a; 3980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 3990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, StoreToDirtyAddress) { 4010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int *a = new int; 40211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR(a[GetPoisonedZero()] = 0); 4030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_break_optimization(a); 4040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete a; 4050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE void StackTestFunc() { 4090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 p4; 4100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 ok4 = 1; 4110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S2 p2; 4120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S2 ok2 = 1; 4130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S1 p1; 4140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S1 ok1 = 1; 4150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_break_optimization(&p4); 4160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_break_optimization(&ok4); 4170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_break_optimization(&p2); 4180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_break_optimization(&ok2); 4190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_break_optimization(&p1); 4200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_break_optimization(&ok1); 4210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 42211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(p4); 42311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(p2); 42411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(p1); 42511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(ok1); 42611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(ok2); 42711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(ok4); 4280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, StackTest) { 4310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StackTestFunc(); 4320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE void StackStressFunc() { 4350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int foo[10000]; 4360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_break_optimization(foo); 4370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, DISABLED_StackStressTest) { 4400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov for (int i = 0; i < 1000000; i++) 4410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StackStressFunc(); 4420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> 4450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovvoid TestFloatingPoint() { 4460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov static volatile T v; 4470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov static T g[100]; 4480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_break_optimization(&g); 4490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *x = GetPoisoned<T>(); 4500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *y = GetPoisoned<T>(1); 45111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*x); 45211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED((long long)*x); 45311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED((int)*x); 4540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov g[0] = *x; 4550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov g[1] = *x + *y; 4560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov g[2] = *x - *y; 4570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov g[3] = *x * *y; 4580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, FloatingPointTest) { 4610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestFloatingPoint<float>(); 4620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestFloatingPoint<double>(); 4630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, DynMem) { 4660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 x = 0; 4670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 *y = GetPoisoned<S4>(); 4680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov memcpy(y, &x, g_one * sizeof(S4)); 46911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*y); 4700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic char *DynRetTestStr; 4730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, DynRet) { 4750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) return; 4760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ReturnPoisoned<S8>(); 47711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(clearenv()); 4780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, DynRet1) { 4820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) return; 4830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ReturnPoisoned<S8>(); 4840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct LargeStruct { 4870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 x[10]; 4880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 4890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE 4910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovLargeStruct LargeRetTest() { 4920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov LargeStruct res; 4930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[0] = *GetPoisoned<S4>(); 4940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[1] = *GetPoisoned<S4>(); 4950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[2] = *GetPoisoned<S4>(); 4960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[3] = *GetPoisoned<S4>(); 4970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[4] = *GetPoisoned<S4>(); 4980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[5] = *GetPoisoned<S4>(); 4990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[6] = *GetPoisoned<S4>(); 5000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[7] = *GetPoisoned<S4>(); 5010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[8] = *GetPoisoned<S4>(); 5020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[9] = *GetPoisoned<S4>(); 5030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 5040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, LargeRet) { 5070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov LargeStruct a = LargeRetTest(); 50811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(a.x[0]); 50911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(a.x[9]); 5100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, fread) { 5130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *x = new char[32]; 5140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov FILE *f = fopen("/proc/self/stat", "r"); 5150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(f); 5160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov fread(x, 1, 32, f); 51711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[0]); 51811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[16]); 51911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[31]); 5200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov fclose(f); 5210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete x; 5220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, read) { 5250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *x = new char[32]; 5260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int fd = open("/proc/self/stat", O_RDONLY); 5270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(fd > 0); 5280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int sz = read(fd, x, 32); 5290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(sz == 32); 53011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[0]); 53111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[16]); 53211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[31]); 5330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov close(fd); 5340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete x; 5350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, pread) { 5380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *x = new char[32]; 5390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int fd = open("/proc/self/stat", O_RDONLY); 5400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(fd > 0); 5410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int sz = pread(fd, x, 32, 0); 5420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(sz == 32); 54311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[0]); 54411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[16]); 54511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[31]); 5460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov close(fd); 5470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete x; 5480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// FIXME: fails now. 5510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, DISABLED_ioctl) { 5520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov struct winsize ws; 5530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(ioctl(2, TIOCGWINSZ, &ws), 0); 55411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(ws.ws_col); 5550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, readlink) { 5580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *x = new char[1000]; 5590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov readlink("/proc/self/exe", x, 1000); 56011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[0]); 5610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete [] x; 5620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, stat) { 5660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov struct stat* st = new struct stat; 5670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = stat("/proc/self/stat", st); 5680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(!res); 56911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(st->st_dev); 57011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(st->st_mode); 57111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(st->st_size); 5720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, statfs) { 5750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov struct statfs* st = new struct statfs; 5760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = statfs("/", st); 5770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(!res); 57811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(st->f_type); 57911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(st->f_bfree); 58011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(st->f_namelen); 5810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, pipe) { 5840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* pipefd = new int[2]; 5850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = pipe(pipefd); 5860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(!res); 58711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(pipefd[0]); 58811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(pipefd[1]); 5890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov close(pipefd[0]); 5900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov close(pipefd[1]); 5910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, getcwd) { 5940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char path[PATH_MAX + 1]; 5950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* res = getcwd(path, sizeof(path)); 5960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(res); 59711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(path[0]); 5980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 6000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, realpath) { 6010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov const char* relpath = "."; 6020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char path[PATH_MAX + 1]; 6030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* res = realpath(relpath, path); 6040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(res); 60511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(path[0]); 6060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 6070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 6080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, memcpy) { 6090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* x = new char[2]; 6100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* y = new char[2]; 6110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[0] = 1; 6120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[1] = *GetPoisoned<char>(); 6130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov memcpy(y, x, 2); 61411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(y[0]); 61511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(y[1]); 6160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 6170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 6180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, memmove) { 6190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* x = new char[2]; 6200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* y = new char[2]; 6210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[0] = 1; 6220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[1] = *GetPoisoned<char>(); 6230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov memmove(y, x, 2); 62411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(y[0]); 62511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(y[1]); 6260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 6270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 6280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, strdup) { 6290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *x = strdup("zzz"); 63011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*x); 6310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov free(x); 6320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 6330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 6340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T, int size> 6350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovvoid TestOverlapMemmove() { 6360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *x = new T[size]; 6370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(size >= 3); 6380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[2] = 0; 6390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov memmove(x, x + 1, (size - 1) * sizeof(T)); 64011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[1]); 6410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) { 6420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // FIXME: under DR we will lose this information 6430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // because accesses in memmove will unpoisin the shadow. 6440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // We need to use our own memove implementation instead of libc's. 64511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[0]); 64611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[2]); 6470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } 6480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete [] x; 6490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 6500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 6510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, overlap_memmove) { 6520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestOverlapMemmove<U1, 10>(); 6530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestOverlapMemmove<U1, 1000>(); 6540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestOverlapMemmove<U8, 4>(); 6550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestOverlapMemmove<U8, 1000>(); 6560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 6570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 6580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, strcpy) { // NOLINT 6590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* x = new char[3]; 6600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* y = new char[3]; 6610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[0] = 'a'; 6620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[1] = *GetPoisoned<char>(1, 1); 6630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[2] = 0; 6640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov strcpy(y, x); // NOLINT 66511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(y[0]); 66611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(y[1]); 66711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(y[2]); 6680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 6690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 6700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, strncpy) { // NOLINT 6710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* x = new char[3]; 6720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* y = new char[3]; 6730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[0] = 'a'; 6740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[1] = *GetPoisoned<char>(1, 1); 6750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[2] = 0; 6760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov strncpy(y, x, 2); // NOLINT 67711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(y[0]); 67811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(y[1]); 67911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(y[2]); 6800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 6810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 6820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, strtol) { 6830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *e; 6840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(1 == strtol("1", &e, 10)); 68511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((S8) e); 6860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 6870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 6880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, strtoll) { 6890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *e; 6900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(1 == strtoll("1", &e, 10)); 69111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((S8) e); 6920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 6930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 6940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, strtoul) { 6950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *e; 6960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(1 == strtoul("1", &e, 10)); 69711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((S8) e); 6980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 6990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 7000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, strtoull) { 7010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *e; 7020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(1 == strtoull("1", &e, 10)); 70311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((S8) e); 7040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 7050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 706e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy StepanovTEST(MemorySanitizer, strtod) { 707e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov char *e; 708e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov assert(0 != strtod("1.5", &e)); 70911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((S8) e); 710e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov} 711e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov 712e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy StepanovTEST(MemorySanitizer, strtof) { 713e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov char *e; 714e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov assert(0 != strtof("1.5", &e)); 71511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((S8) e); 716e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov} 717e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov 718e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy StepanovTEST(MemorySanitizer, strtold) { 719e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov char *e; 720e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov assert(0 != strtold("1.5", &e)); 72111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((S8) e); 722e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov} 723e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov 7240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, sprintf) { // NOLINT 7250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char buff[10]; 7260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_break_optimization(buff); 72711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(buff[0]); 7280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = sprintf(buff, "%d", 1234567); // NOLINT 7290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(res == 7); 7300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[0] == '1'); 7310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[1] == '2'); 7320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[2] == '3'); 7330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[6] == '7'); 7340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[7] == 0); 73511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(buff[8]); 7360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 7370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 7380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, snprintf) { 7390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char buff[10]; 7400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_break_optimization(buff); 74111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(buff[0]); 7420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = snprintf(buff, sizeof(buff), "%d", 1234567); 7430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(res == 7); 7440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[0] == '1'); 7450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[1] == '2'); 7460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[2] == '3'); 7470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[6] == '7'); 7480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[7] == 0); 74911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(buff[8]); 7500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 7510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 7520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, swprintf) { 7530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov wchar_t buff[10]; 7540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(sizeof(wchar_t) == 4); 7550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_break_optimization(buff); 75611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(buff[0]); 7570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = swprintf(buff, 9, L"%d", 1234567); 7580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(res == 7); 7590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[0] == '1'); 7600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[1] == '2'); 7610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[2] == '3'); 7620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[6] == '7'); 7630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[7] == 0); 76411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(buff[8]); 7650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 7660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 7670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, wcstombs) { 7680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov const wchar_t *x = L"abc"; 7690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char buff[10]; 7700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = wcstombs(buff, x, 4); 7710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(res, 3); 7720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(buff[0], 'a'); 7730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(buff[1], 'b'); 7740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(buff[2], 'c'); 7750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 7760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 7770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, gettimeofday) { 7780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov struct timeval tv; 7790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov struct timezone tz; 7800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_break_optimization(&tv); 7810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_break_optimization(&tz); 7820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(sizeof(tv) == 16); 7830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(sizeof(tz) == 8); 78411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(tv.tv_sec); 78511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(tv.tv_usec); 78611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(tz.tz_minuteswest); 78711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(tz.tz_dsttime); 7880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(0 == gettimeofday(&tv, &tz)); 78911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(tv.tv_sec); 79011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(tv.tv_usec); 79111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(tz.tz_minuteswest); 79211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(tz.tz_dsttime); 7930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 7940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 7950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, mmap) { 7960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov const int size = 4096; 7970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov void *p1, *p2; 7980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov p1 = mmap(0, size, PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANON, -1, 0); 7990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(p1, size); 8000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov munmap(p1, size); 8010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov for (int i = 0; i < 1000; i++) { 8020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov p2 = mmap(0, size, PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANON, -1, 0); 8030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (p2 == p1) 8040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov break; 8050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov else 8060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov munmap(p2, size); 8070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } 8080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (p1 == p2) { 80911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*(char*)p2); 8100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov munmap(p2, size); 8110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } 8120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 8130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 8140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// FIXME: enable and add ecvt. 8150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// FIXME: check why msandr does nt handle fcvt. 8160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, fcvt) { 8170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int a, b; 8180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_break_optimization(&a); 8190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_break_optimization(&b); 82011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(a); 82111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(b); 8220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *str = fcvt(12345.6789, 10, &a, &b); 82311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(a); 82411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(b); 8250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 8260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 8270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, LoadUnpoisoned) { 8280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S8 s = *GetPoisoned<S8>(); 82911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s); 8300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S8 safe = *GetPoisoned<S8>(); 8310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_load_unpoisoned(&s, sizeof(s), &safe); 83211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(safe); 8330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 8340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 8350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct StructWithDtor { 8360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ~StructWithDtor(); 8370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 8380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 8390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE StructWithDtor::~StructWithDtor() { 8400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_break_optimization(0); 8410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 8420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 8430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, Invoke) { 8440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructWithDtor s; // Will cause the calls to become invokes. 84511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(0); 84611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<int>()); 84711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(0); 84811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<int>()); 84911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(ReturnPoisoned<S4>()); 8500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 8510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 8520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, ptrtoint) { 8530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // Test that shadow is propagated through pointer-to-integer conversion. 8540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov void* p = (void*)0xABCD; 8550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(((char*)&p) + 1, sizeof(p)); 85611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((((uptr)p) & 0xFF) == 0); 8570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 8580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov void* q = (void*)0xABCD; 8590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&q, sizeof(q) - 1); 86011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED((((uptr)q) & 0xFF) == 0); 8610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 8620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 8630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn2(int guard, ...) { 8640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 8650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 86611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 86711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 86811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 86911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, double)); 8700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 8710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 8720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 8730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn(int guard, ...) { 8740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 8750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 87611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 87711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 8780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // The following call will overwrite __msan_param_tls. 8790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // Checks after it test that arg shadow was somehow saved across the call. 8800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn2(1, 2, 3, 4, *GetPoisoned<double>()); 88111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 88211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 8830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 8840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 8850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 8860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, VAArgTest) { 8870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* x = GetPoisoned<int>(); 8880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* y = GetPoisoned<int>(4); 8890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn(1, 13, *x, 42, *y); 8900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 8910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 8920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_many(int guard, ...) { 8930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 8940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 89511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 89611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 89711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 89811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 89911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 90011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 90111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 90211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 90311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 90411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 9050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 9060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, VAArgManyTest) { 9090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* x = GetPoisoned<int>(); 9100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* y = GetPoisoned<int>(4); 9110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_many(1, 2, *x, 3, 4, 5, 6, 7, 8, 9, *y); 9120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_pass2(va_list vl) { 91511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 91611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 91711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 9180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_pass(int guard, ...) { 9210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 9220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 92311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 9240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_pass2(vl); 9250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 9260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, VAArgPass) { 9290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* x = GetPoisoned<int>(); 9300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* y = GetPoisoned<int>(4); 9310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_pass(1, *x, 2, 3, *y); 9320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_copy2(va_list vl) { 93511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 93611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 9370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_copy(int guard, ...) { 9400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 9410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 94211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 94311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 9440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl2; 9450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_copy(vl2, vl); 9460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_copy2(vl2); 94711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 94811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 9490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 9500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, VAArgCopy) { 9530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* x = GetPoisoned<int>(); 9540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* y = GetPoisoned<int>(4); 9550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_copy(1, 2, *x, 3, *y); 9560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_ptr(int guard, ...) { 9590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 9600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 96111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int*)); 96211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int*)); 96311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int*)); 96411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, double*)); 9650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 9660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, VAArgPtr) { 9690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int** x = GetPoisoned<int*>(); 9700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov double** y = GetPoisoned<double*>(8); 9710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int z; 9720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_ptr(1, &z, *x, &z, *y); 9730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_overflow(int guard, ...) { 9760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 9770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 97811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 97911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 98011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 98111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 98211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 98311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 98411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 98511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, double)); 98611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, double)); 98711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, double)); 98811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, double)); 98911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, double)); 99011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int*)); 99111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, double)); 99211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, double)); 99311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 99411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 99511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, double)); 99611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int*)); 99711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 99811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 99911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, double)); 100011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int*)); 100111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 100211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 100311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, double)); 100411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int*)); 10050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 10060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 10070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 10080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 10090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, VAArgOverflow) { 10100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* x = GetPoisoned<int>(); 10110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov double* y = GetPoisoned<double>(8); 10120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int** p = GetPoisoned<int*>(16); 10130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int z; 10140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_overflow(1, 10150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1, 2, *x, 4, 5, 6, 10160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1.1, 2.2, 3.3, *y, 5.5, *p, 7.7, 8.8, 10170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // the following args will overflow for sure 10180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *x, *y, *p, 10190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 7, 9.9, &z, 10200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *x, *y, *p); 10210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 10220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 10230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_tlsoverwrite2(int guard, ...) { 10240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 10250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 102611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 10270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 10280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 10290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 10300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_tlsoverwrite(int guard, ...) { 10310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // This call will overwrite TLS contents unless it's backed up somewhere. 10320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_tlsoverwrite2(2, 42); 10330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 10340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 103511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 10360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 10370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 10380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 10390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, VAArgTLSOverwrite) { 10400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* x = GetPoisoned<int>(); 10410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_tlsoverwrite(1, *x); 10420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 10430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 10440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct StructByVal { 10450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int a, b, c, d, e, f; 10460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 10470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 10480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE void StructByValTestFunc(struct StructByVal s) { 104911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s.a); 105011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s.b); 105111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s.c); 105211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s.d); 105311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s.e); 105411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s.f); 10550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 10560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 10570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE void StructByValTestFunc1(struct StructByVal s) { 10580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructByValTestFunc(s); 10590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 10600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 10610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE void StructByValTestFunc2(int z, struct StructByVal s) { 10620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructByValTestFunc(s); 10630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 10640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 10650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, StructByVal) { 10660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // Large aggregates are passed as "byval" pointer argument in LLVM. 10670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov struct StructByVal s; 10680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov s.a = 1; 10690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov s.b = *GetPoisoned<int>(); 10700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov s.c = 2; 10710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov s.d = *GetPoisoned<int>(); 10720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov s.e = 3; 10730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov s.f = *GetPoisoned<int>(); 10740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructByValTestFunc(s); 10750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructByValTestFunc1(s); 10760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructByValTestFunc2(0, s); 10770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 10780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 10790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 10800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#if MSAN_HAS_M128 10810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE __m128i m128Eq(__m128i *a, __m128i *b) { return *a == *b; } 10820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE __m128i m128Lt(__m128i *a, __m128i *b) { return *a < *b; } 10830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, m128) { 10840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __m128i a = _mm_set1_epi16(0x1234); 10850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __m128i b = _mm_set1_epi16(0x7890); 108611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(m128Eq(&a, &b)); 108711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(m128Lt(&a, &b)); 10880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 10890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// FIXME: add more tests for __m128i. 10900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#endif // MSAN_HAS_M128 10910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 10920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// We should not complain when copying this poisoned hole. 10930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct StructWithHole { 10940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U4 a; 10950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // 4-byte hole. 10960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U8 b; 10970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 10980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 10990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE StructWithHole ReturnStructWithHole() { 11000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructWithHole res; 11010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&res, sizeof(res)); 11020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.a = 1; 11030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.b = 2; 11040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 11050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 11060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 11070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, StructWithHole) { 11080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructWithHole a = ReturnStructWithHole(); 11090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_break_optimization(&a); 11100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 11110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 11120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate <class T> 11130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE T ReturnStruct() { 11140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T res; 11150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&res, sizeof(res)); 11160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.a = 1; 11170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 11180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 11190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 11200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate <class T> 11210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE void TestReturnStruct() { 11220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T s1 = ReturnStruct<T>(); 112311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s1.a); 112411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s1.b); 11250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 11260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 11270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct SSS1 { 11280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int a, b, c; 11290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 11300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct SSS2 { 11310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int b, a, c; 11320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 11330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct SSS3 { 11340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int b, c, a; 11350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 11360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct SSS4 { 11370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int c, b, a; 11380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 11390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 11400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct SSS5 { 11410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int a; 11420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov float b; 11430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 11440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct SSS6 { 11450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int a; 11460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov double b; 11470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 11480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct SSS7 { 11490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S8 b; 11500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int a; 11510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 11520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct SSS8 { 11530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S2 b; 11540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S8 a; 11550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 11560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 11570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, IntStruct3) { 11580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestReturnStruct<SSS1>(); 11590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestReturnStruct<SSS2>(); 11600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestReturnStruct<SSS3>(); 11610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestReturnStruct<SSS4>(); 11620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestReturnStruct<SSS5>(); 11630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestReturnStruct<SSS6>(); 11640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestReturnStruct<SSS7>(); 11650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestReturnStruct<SSS8>(); 11660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 11670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 11680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct LongStruct { 11690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U1 a1, b1; 11700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U2 a2, b2; 11710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U4 a4, b4; 11720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U8 a8, b8; 11730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 11740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 11750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE LongStruct ReturnLongStruct1() { 11760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov LongStruct res; 11770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&res, sizeof(res)); 11780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.a1 = res.a2 = res.a4 = res.a8 = 111; 11790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // leaves b1, .., b8 poisoned. 11800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 11810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 11820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 11830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE LongStruct ReturnLongStruct2() { 11840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov LongStruct res; 11850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&res, sizeof(res)); 11860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.b1 = res.b2 = res.b4 = res.b8 = 111; 11870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // leaves a1, .., a8 poisoned. 11880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 11890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 11900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 11910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, LongStruct) { 11920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov LongStruct s1 = ReturnLongStruct1(); 11930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_print_shadow(&s1, sizeof(s1)); 119411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s1.a1); 119511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s1.a2); 119611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s1.a4); 119711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s1.a8); 11980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 119911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s1.b1); 120011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s1.b2); 120111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s1.b4); 120211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s1.b8); 12030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 12040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov LongStruct s2 = ReturnLongStruct2(); 12050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_print_shadow(&s2, sizeof(s2)); 120611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s2.b1); 120711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s2.b2); 120811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s2.b4); 120911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s2.b8); 12100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 121111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s2.a1); 121211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s2.a2); 121311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s2.a4); 121411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s2.a8); 12150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 12160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 12170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, getrlimit) { 12180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov struct rlimit limit; 12190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&limit, sizeof(limit)); 12200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int result = getrlimit(RLIMIT_DATA, &limit); 12210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(result == 0); 12220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov volatile rlim_t t; 12230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov t = limit.rlim_cur; 12240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov t = limit.rlim_max; 12250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 12260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1227e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy StepanovTEST(MemorySanitizer, getrusage) { 1228e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov struct rusage usage; 1229e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov __msan_poison(&usage, sizeof(usage)); 1230e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov int result = getrusage(RUSAGE_SELF, &usage); 1231e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov assert(result == 0); 1232e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov volatile struct timeval t; 123311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_utime.tv_sec); 123411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_utime.tv_usec); 123511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_stime.tv_sec); 123611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_stime.tv_usec); 123711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_maxrss); 123811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_minflt); 123911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_majflt); 124011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_inblock); 124111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_oublock); 124211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_nvcsw); 124311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_nivcsw); 1244e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov} 1245e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov 1246e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanovstatic void dladdr_testfn() {} 1247e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov 1248e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy StepanovTEST(MemorySanitizer, dladdr) { 1249e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov Dl_info info; 1250e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov __msan_poison(&info, sizeof(info)); 1251e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov int result = dladdr((const void*)dladdr_testfn, &info); 1252e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov assert(result != 0); 125311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((unsigned long)info.dli_fname); 1254e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov if (info.dli_fname) 125511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(info.dli_fname)); 125611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((unsigned long)info.dli_fbase); 125711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((unsigned long)info.dli_sname); 1258e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov if (info.dli_sname) 125911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(info.dli_sname)); 126011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((unsigned long)info.dli_saddr); 1261e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov} 1262e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov 1263996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy StepanovTEST(MemorySanitizer, scanf) { 1264996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov const char *input = "42 hello"; 1265996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov int* d = new int; 1266996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov char* s = new char[7]; 1267996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov int res = sscanf(input, "%d %5s", d, s); 1268996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov printf("res %d\n", res); 1269996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov assert(res == 2); 127011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*d); 127111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s[0]); 127211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s[1]); 127311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s[2]); 127411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s[3]); 127511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s[4]); 127611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s[5]); 127711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s[6]); 1278996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov delete s; 1279996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov delete d; 1280996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov} 1281996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov 12820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void* SimpleThread_threadfn(void* data) { 12830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return new int; 12840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 12850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 12860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, SimpleThread) { 12870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov pthread_t t; 12880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov void* p; 12890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = pthread_create(&t, NULL, SimpleThread_threadfn, NULL); 12900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(!res); 12910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res = pthread_join(t, &p); 12920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(!res); 12930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) // FIXME: intercept pthread_join (?). 12940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_unpoison(&p, sizeof(p)); 12950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete (int*)p; 12960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 12970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 12980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, uname) { 12990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov struct utsname u; 13000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = uname(&u); 13010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(!res); 130211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(u.sysname)); 130311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(u.nodename)); 130411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(u.release)); 130511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(u.version)); 130611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(u.machine)); 13070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> 131011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovstatic bool applySlt(T value, T shadow) { 13110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_partial_poison(&value, &shadow, sizeof(T)); 13120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov volatile bool zzz = true; 13130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // This "|| zzz" trick somehow makes LLVM emit "icmp slt" instead of 13140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // a shift-and-trunc to get at the highest bit. 131511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov volatile bool v = value < 0 || zzz; 131611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov return v; 13170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, SignedCompareWithZero) { 132011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applySlt<S4>(0xF, 0xF)); 132111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applySlt<S4>(0xF, 0xFF)); 132211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applySlt<S4>(0xF, 0xFFFFFF)); 132311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applySlt<S4>(0xF, 0x7FFFFFF)); 132411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR(applySlt<S4>(0xF, 0x80FFFFFF)); 132511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR(applySlt<S4>(0xF, 0xFFFFFFFF)); 13260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13282efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov// Volatile bitfield store is implemented as load-mask-store 13292efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov// Test that we don't warn on the store of (uninitialized) padding. 13302efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanovstruct VolatileBitfieldStruct { 13312efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov volatile unsigned x : 1; 13322efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov unsigned y : 1; 13332efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov}; 13342efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov 13352efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy StepanovTEST(MemorySanitizer, VolatileBitfield) { 13362efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov VolatileBitfieldStruct *S = new VolatileBitfieldStruct; 13372efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov S->x = 1; 133802f4a942bff84f1266571740456dd9baa230d87bEvgeniy Stepanov EXPECT_NOT_POISONED((unsigned)S->x); 133902f4a942bff84f1266571740456dd9baa230d87bEvgeniy Stepanov EXPECT_POISONED((unsigned)S->y); 13402efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov} 13412efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov 13420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerDr, StoreInDSOTest) { 13430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) return; 13440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* s = new char[10]; 13450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov dso_memfill(s, 9); 134611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s[5]); 134711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s[9]); 13480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovint return_poisoned_int() { 13510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return ReturnPoisoned<U8>(); 13520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerDr, ReturnFromDSOTest) { 13550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) return; 135611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(dso_callfn(return_poisoned_int)); 13570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE int TrashParamTLS(long long x, long long y, long long z) { //NOLINT 136011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x); 136111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(y); 136211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(z); 13630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return 0; 13640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic int CheckParamTLS(long long x, long long y, long long z) { //NOLINT 136711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x); 136811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(y); 136911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(z); 13700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return 0; 13710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerDr, CallFromDSOTest) { 13740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) return; 13750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S8* x = GetPoisoned<S8>(); 13760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S8* y = GetPoisoned<S8>(); 13770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S8* z = GetPoisoned<S8>(); 137811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(TrashParamTLS(*x, *y, *z)); 137911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(dso_callfn1(CheckParamTLS)); 13800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void StackStoreInDSOFn(int* x, int* y) { 138311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*x); 138411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*y); 13850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerDr, StackStoreInDSOTest) { 13880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) return; 13890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov dso_stack_store(StackStoreInDSOFn, 1); 13900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, SetGet) { 13930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(TrackingOrigins(), __msan_get_track_origins()); 13940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 13950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int x; 13960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(&x, sizeof(x), 1234); 13970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(1234, __msan_get_origin(&x)); 13980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(&x, sizeof(x), 5678); 13990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(5678, __msan_get_origin(&x)); 14000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(&x, sizeof(x), 0); 14010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(0, __msan_get_origin(&x)); 14020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovnamespace { 14050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct S { 14060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U4 dummy; 14070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U2 a; 14080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U2 b; 14090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 14100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// http://code.google.com/p/memory-sanitizer/issues/detail?id=6 14120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, DISABLED_InitializedStoreDoesNotChangeOrigin) { 14130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 14140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S s; 14160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov u32 origin = rand(); // NOLINT 14170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov s.a = *GetPoisonedO<U2>(0, origin); 14180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(origin, __msan_get_origin(&s.a)); 14190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(origin, __msan_get_origin(&s.b)); 14200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov s.b = 42; 14220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(origin, __msan_get_origin(&s.a)); 14230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(origin, __msan_get_origin(&s.b)); 14240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} // namespace 14260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T, class BinaryOp> 14280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovINLINE 14290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovvoid BinaryOpOriginTest(BinaryOp op) { 14300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov u32 ox = rand(); //NOLINT 14310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov u32 oy = rand(); //NOLINT 14320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *x = GetPoisonedO<T>(0, ox, 0); 14330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *y = GetPoisonedO<T>(1, oy, 0); 14340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *z = GetPoisonedO<T>(2, 0, 0); 14350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *z = op(*x, *y); 14370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov u32 origin = __msan_get_origin(z); 143811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*z, origin); 14390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(true, origin == ox || origin == oy); 14400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // y is poisoned, x is not. 14420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *x = 10101; 14430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *y = *GetPoisonedO<T>(1, oy); 14440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_break_optimization(x); 14450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(z, sizeof(*z), 0); 14460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *z = op(*x, *y); 144711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*z, oy); 14480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(__msan_get_origin(z), oy); 14490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // x is poisoned, y is not. 14510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *x = *GetPoisonedO<T>(0, ox); 14520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *y = 10101010; 14530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_break_optimization(y); 14540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(z, sizeof(*z), 0); 14550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *z = op(*x, *y); 145611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*z, ox); 14570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(__msan_get_origin(z), ox); 14580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> INLINE T XOR(const T &a, const T&b) { return a ^ b; } 14610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> INLINE T ADD(const T &a, const T&b) { return a + b; } 14620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> INLINE T SUB(const T &a, const T&b) { return a - b; } 14630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> INLINE T MUL(const T &a, const T&b) { return a * b; } 14640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> INLINE T AND(const T &a, const T&b) { return a & b; } 14650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> INLINE T OR (const T &a, const T&b) { return a | b; } 14660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, BinaryOp) { 14680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 14690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<S8>(XOR<S8>); 14700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<U8>(ADD<U8>); 14710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<S4>(SUB<S4>); 14720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<S4>(MUL<S4>); 14730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<U4>(OR<U4>); 14740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<U4>(AND<U4>); 14750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<double>(ADD<U4>); 14760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<float>(ADD<S4>); 14770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<double>(ADD<double>); 14780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<float>(ADD<double>); 14790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, Unary) { 14820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 148311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S8>(0, __LINE__), __LINE__); 148411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S8>(0, __LINE__), __LINE__); 148511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S8>(0, __LINE__), __LINE__); 148611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S8>(0, __LINE__), __LINE__); 14870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 148811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 148911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 149011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 149111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 14920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 149311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<U4>(0, __LINE__), __LINE__); 149411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<U4>(0, __LINE__), __LINE__); 149511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<U4>(0, __LINE__), __LINE__); 149611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<U4>(0, __LINE__), __LINE__); 14970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 149811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 149911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 150011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 150111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 15020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 150311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O((void*)*GetPoisonedO<S8>(0, __LINE__), __LINE__); 150411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O((U8)*GetPoisonedO<void*>(0, __LINE__), __LINE__); 15050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, EQ) { 15080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 150911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__) <= 11, __LINE__); 151011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__) == 11, __LINE__); 151111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<float>(0, __LINE__) == 1.1, __LINE__); 15120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, DIV) { 15150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 151611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<U8>(0, __LINE__) / 100, __LINE__); 151711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov unsigned o = __LINE__; 151811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR_O(volatile unsigned y = 100 / *GetPoisonedO<S4>(0, o, 1), o); 15190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, SHIFT) { 15220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 152311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<U8>(0, __LINE__) >> 10, __LINE__); 152411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S8>(0, __LINE__) >> 10, __LINE__); 152511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S8>(0, __LINE__) << 10, __LINE__); 152611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(10U << *GetPoisonedO<U8>(0, __LINE__), __LINE__); 152711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(-10 >> *GetPoisonedO<S8>(0, __LINE__), __LINE__); 152811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(-10 << *GetPoisonedO<S8>(0, __LINE__), __LINE__); 15290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T, int N> 15320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovvoid MemCpyTest() { 15330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int ox = __LINE__; 15340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *x = new T[N]; 15350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *y = new T[N]; 15360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *z = new T[N]; 15370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(x, N * sizeof(T)); 15380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(x, N * sizeof(T), ox); 15390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(y, N * sizeof(T), 777777); 15400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(z, N * sizeof(T), 888888); 154111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x); 154211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov memcpy(y, x, N * sizeof(T)); 154311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(y[0], ox); 154411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(y[N/2], ox); 154511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(y[N-1], ox); 154611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x); 154711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov memmove(z, x, N * sizeof(T)); 154811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(z[0], ox); 154911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(z[N/2], ox); 155011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(z[N-1], ox); 15510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, LargeMemCpy) { 15540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 15550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov MemCpyTest<U1, 10000>(); 15560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov MemCpyTest<U8, 10000>(); 15570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, SmallMemCpy) { 15600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 15610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov MemCpyTest<U8, 1>(); 15620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov MemCpyTest<U8, 2>(); 15630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov MemCpyTest<U8, 3>(); 15640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, Select) { 15670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 156811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(g_one ? 1 : *GetPoisonedO<S4>(0, __LINE__)); 156911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 15700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 x; 15710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_break_optimization(&x); 15720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x = g_1 ? *GetPoisonedO<S4>(0, __LINE__) : 0; 15730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 157411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(g_1 ? *GetPoisonedO<S4>(0, __LINE__) : 1, __LINE__); 157511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(g_0 ? 1 : *GetPoisonedO<S4>(0, __LINE__), __LINE__); 15760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovextern "C" 157911929000ec2919192b3be457f5a44c71ed55215eEvgeniy StepanovNOINLINE char AllocaTO() { 15800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int ar[100]; 15810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_break_optimization(ar); 158211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov return ar[10]; 15830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // fprintf(stderr, "Descr: %s\n", 15840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // __msan_get_origin_descr_if_stack(__msan_get_origin_tls())); 15850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, Alloca) { 15880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 158911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_S(AllocaTO(), "ar@AllocaTO"); 159011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_S(AllocaTO(), "ar@AllocaTO"); 159111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_S(AllocaTO(), "ar@AllocaTO"); 159211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_S(AllocaTO(), "ar@AllocaTO"); 15930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// FIXME: replace with a lit-like test. 15960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, DISABLED_AllocaDeath) { 15970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 159811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_DEATH(AllocaTO(), "ORIGIN: stack allocation: ar@AllocaTO"); 15990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 16000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 16010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE int RetvalOriginTest(u32 origin) { 16020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int *a = new int; 16030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_break_optimization(a); 16040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(a, sizeof(*a), origin); 16050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = *a; 16060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete a; 16070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 16080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 16090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 16100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, Retval) { 16110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 161211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(RetvalOriginTest(__LINE__), __LINE__); 16130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 16140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 16150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE void ParamOriginTest(int param, u32 origin) { 161611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(param, origin); 16170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 16180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 16190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, Param) { 16200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 16210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int *a = new int; 16220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov u32 origin = __LINE__; 16230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_break_optimization(a); 16240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(a, sizeof(*a), origin); 16250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ParamOriginTest(*a, origin); 16260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete a; 16270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 16280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 16290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, Invoke) { 16300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 16310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructWithDtor s; // Will cause the calls to become invokes. 163211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(RetvalOriginTest(__LINE__), __LINE__); 16330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 16340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 16350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, strlen) { 16360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S8 alignment; 16370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_break_optimization(&alignment); 16380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char x[4] = {'a', 'b', 0, 0}; 16390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&x[2], 1); 16400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov u32 origin = __LINE__; 16410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(x, sizeof(x), origin); 164211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR_O(volatile unsigned y = strlen(x), origin); 16430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 16440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 16450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, wcslen) { 16460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov wchar_t w[3] = {'a', 'b', 0}; 16470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov u32 origin = __LINE__; 16480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(w, sizeof(w), origin); 16490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&w[2], sizeof(wchar_t)); 165011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR_O(volatile unsigned y = wcslen(w), origin); 16510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 16520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 16530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#if MSAN_HAS_M128 16540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, StoreIntrinsic) { 16550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __m128 x, y; 16560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov u32 origin = __LINE__; 16570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(&x, sizeof(x), origin); 16580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&x, sizeof(x)); 16590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __builtin_ia32_storeups((float*)&y, x); 166011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(y, origin); 16610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 16620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#endif 16630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 16640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE void RecursiveMalloc(int depth) { 16650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov static int count; 16660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov count++; 16670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if ((count % (1024 * 1024)) == 0) 16680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov printf("RecursiveMalloc: %d\n", count); 16690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int *x1 = new int; 16700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int *x2 = new int; 16710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_break_optimization(x1); 16720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_break_optimization(x2); 16730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (depth > 0) { 16740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov RecursiveMalloc(depth-1); 16750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov RecursiveMalloc(depth-1); 16760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } 16770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete x1; 16780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete x2; 16790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 16800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 16810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerStress, DISABLED_MallocStackTrace) { 16820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov RecursiveMalloc(22); 16830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 16840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 16850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovint main(int argc, char **argv) { 16860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_poison_in_malloc(1); 16870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov testing::InitGoogleTest(&argc, argv); 16880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = RUN_ALL_TESTS(); 16890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 16900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 1691