msan_test.cc revision 9d1525ec52430d0b8ffd6d0893b7f5529105b321
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 156e5ff89e8980b7c2ca0a39811433fb7ac2c74372Evgeniy Stepanov#ifndef MSAN_EXTERNAL_TEST_CONFIG 166e5ff89e8980b7c2ca0a39811433fb7ac2c74372Evgeniy Stepanov#include "msan_test_config.h" 176e5ff89e8980b7c2ca0a39811433fb7ac2c74372Evgeniy Stepanov#endif // MSAN_EXTERNAL_TEST_CONFIG 186e5ff89e8980b7c2ca0a39811433fb7ac2c74372Evgeniy Stepanov 190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include "sanitizer/msan_interface.h" 200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include "msandr_test_so.h" 210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <stdlib.h> 230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <stdarg.h> 240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <stdio.h> 250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <assert.h> 260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <wchar.h> 277cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov#include <math.h> 280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 299530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov#include <arpa/inet.h> 30e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov#include <dlfcn.h> 31103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov#include <grp.h> 320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <unistd.h> 332bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov#include <link.h> 340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <limits.h> 350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <sys/time.h> 360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <sys/types.h> 370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <sys/stat.h> 380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <fcntl.h> 390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <sys/resource.h> 400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <sys/ioctl.h> 410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <sys/utsname.h> 420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <sys/mman.h> 430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <sys/vfs.h> 44d97a15a931ae6696a1071e1471c9a019e821904bEvgeniy Stepanov#include <dirent.h> 45e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov#include <pwd.h> 46134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov#include <sys/socket.h> 470a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov#include <netdb.h> 480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#if defined(__i386__) || defined(__x86_64__) 500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov# include <emmintrin.h> 510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov# define MSAN_HAS_M128 1 520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#else 530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov# define MSAN_HAS_M128 0 540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#endif 550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtypedef unsigned char U1; 570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtypedef unsigned short U2; // NOLINT 580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtypedef unsigned int U4; 590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtypedef unsigned long long U8; // NOLINT 600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtypedef signed char S1; 610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtypedef signed short S2; // NOLINT 620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtypedef signed int S4; 630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtypedef signed long long S8; // NOLINT 640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#define NOINLINE __attribute__((noinline)) 650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#define INLINE __attribute__((always_inline)) 660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 6711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovstatic bool TrackingOrigins() { 6811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov S8 x; 6911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov __msan_set_origin(&x, sizeof(x), 0x1234); 70250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 origin = __msan_get_origin(&x); 7111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov __msan_set_origin(&x, sizeof(x), 0); 7211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov return origin == 0x1234; 7311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov} 740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 7511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov#define EXPECT_UMR(action) \ 760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov do { \ 770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_expect_umr(1); \ 780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov action; \ 790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_expect_umr(0); \ 800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } while (0) 810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 8211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov#define EXPECT_UMR_O(action, origin) \ 830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov do { \ 840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_expect_umr(1); \ 850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov action; \ 860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_expect_umr(0); \ 870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (TrackingOrigins()) \ 8812c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov EXPECT_EQ(origin, __msan_get_umr_origin()); \ 890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } while (0) 900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov#define EXPECT_UMR_S(action, stack_origin) \ 920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov do { \ 930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_expect_umr(1); \ 940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov action; \ 950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_expect_umr(0); \ 96250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 id = __msan_get_umr_origin(); \ 970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov const char *str = __msan_get_origin_descr_if_stack(id); \ 980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!str || strcmp(str, stack_origin)) { \ 990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov fprintf(stderr, "EXPECT_POISONED_S: id=%u %s, %s", \ 1000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov id, stack_origin, str); \ 1010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(1, 0); \ 1020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } \ 1030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } while (0) 1040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 10511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov#define EXPECT_POISONED(x) ExpectPoisoned(x) 10611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 10711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovtemplate<typename T> 10811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovvoid ExpectPoisoned(const T& t) { 10911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NE(-1, __msan_test_shadow((void*)&t, sizeof(t))); 11011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov} 11111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 11211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov#define EXPECT_POISONED_O(x, origin) \ 11311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov ExpectPoisonedWithOrigin(x, origin) 11411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 11511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovtemplate<typename T> 11611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovvoid ExpectPoisonedWithOrigin(const T& t, unsigned origin) { 11711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NE(-1, __msan_test_shadow((void*)&t, sizeof(t))); 11811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov if (TrackingOrigins()) 11911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_EQ(origin, __msan_get_origin((void*)&t)); 12011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov} 12111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 12211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov#define EXPECT_POISONED_S(x, stack_origin) \ 12311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov ExpectPoisonedWithStackOrigin(x, stack_origin) 12411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 12511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovtemplate<typename T> 12611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovvoid ExpectPoisonedWithStackOrigin(const T& t, const char *stack_origin) { 12711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NE(-1, __msan_test_shadow((void*)&t, sizeof(t))); 128250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 id = __msan_get_origin((void*)&t); 12911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov const char *str = __msan_get_origin_descr_if_stack(id); 13011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov if (!str || strcmp(str, stack_origin)) { 13111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov fprintf(stderr, "EXPECT_POISONED_S: id=%u %s, %s", 13211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov id, stack_origin, str); 13311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_EQ(1, 0); 13411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov } 13511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov} 13611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 13711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov#define EXPECT_NOT_POISONED(x) ExpectNotPoisoned(x) 13811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 13911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovtemplate<typename T> 14011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovvoid ExpectNotPoisoned(const T& t) { 14111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_EQ(-1, __msan_test_shadow((void*)&t, sizeof(t))); 14211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov} 1430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic U8 poisoned_array[100]; 1450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> 1460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovT *GetPoisoned(int i = 0, T val = 0) { 1470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *res = (T*)&poisoned_array[i]; 1480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *res = val; 1490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&poisoned_array[i], sizeof(T)); 1500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 1510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 1520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> 154250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy StepanovT *GetPoisonedO(int i, U4 origin, T val = 0) { 1550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *res = (T*)&poisoned_array[i]; 1560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *res = val; 1570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&poisoned_array[i], sizeof(T)); 1580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(&poisoned_array[i], sizeof(T), origin); 1590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 1600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 1610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// This function returns its parameter but in such a way that compiler 1630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// can not prove it. 1640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> 1650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE 1660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic T Ident(T t) { 1670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov volatile T ret = t; 1680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return ret; 1690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 1700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> NOINLINE T ReturnPoisoned() { return *GetPoisoned<T>(); } 1720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic volatile int g_one = 1; 1740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic volatile int g_zero = 0; 1750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic volatile int g_0 = 0; 1760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic volatile int g_1 = 1; 1770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovS4 a_s4[100]; 1790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovS8 a_s8[100]; 1800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 18112c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov// Check that malloc poisons memory. 18212c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov// A lot of tests below depend on this. 18312c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy StepanovTEST(MemorySanitizerSanity, PoisonInMalloc) { 18412c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov int *x = (int*)malloc(sizeof(int)); 18512c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov EXPECT_POISONED(*x); 18612c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov free(x); 18712c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov} 18812c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov 1890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, NegativeTest1) { 1900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 *x = GetPoisoned<S4>(); 1910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (g_one) 1920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *x = 0; 19311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*x); 1940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 1950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, PositiveTest1) { 1970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // Load to store. 19811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S1>()); 19911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S2>()); 20011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S4>()); 20111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S8>()); 2020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // S->S conversions. 20411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S1>()); 20511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S1>()); 20611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S1>()); 2070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 20811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S2>()); 20911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S2>()); 21011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S2>()); 2110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 21211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S4>()); 21311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S4>()); 21411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S4>()); 2150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 21611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S8>()); 21711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S8>()); 21811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S8>()); 2190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // ZExt 22111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<U1>()); 22211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<U1>()); 22311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<U1>()); 22411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<U2>()); 22511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<U2>()); 22611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<U4>()); 2270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // Unary ops. 22911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(- *GetPoisoned<S4>()); 2300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 23111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR(a_s4[g_zero] = 100 / *GetPoisoned<S4>(0, 1)); 2320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov a_s4[g_zero] = 1 - *GetPoisoned<S4>(); 2350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov a_s4[g_zero] = 1 + *GetPoisoned<S4>(); 2360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 2370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, Phi1) { 2390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 c; 2400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (g_one) { 2410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov c = *GetPoisoned<S4>(); 2420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } else { 24312c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(0); 2440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov c = 0; 2450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } 24611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(c); 2470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 2480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, Phi2) { 2500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 i = *GetPoisoned<S4>(); 2510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 n = g_one; 25211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR(for (; i < g_one; i++);); 25311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(i); 2540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 2550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 25611929000ec2919192b3be457f5a44c71ed55215eEvgeniy StepanovNOINLINE void Arg1ExpectUMR(S4 a1) { EXPECT_POISONED(a1); } 25711929000ec2919192b3be457f5a44c71ed55215eEvgeniy StepanovNOINLINE void Arg2ExpectUMR(S4 a1, S4 a2) { EXPECT_POISONED(a2); } 25811929000ec2919192b3be457f5a44c71ed55215eEvgeniy StepanovNOINLINE void Arg3ExpectUMR(S1 a1, S4 a2, S8 a3) { EXPECT_POISONED(a3); } 2590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, ArgTest) { 2610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov Arg1ExpectUMR(*GetPoisoned<S4>()); 2620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov Arg2ExpectUMR(0, *GetPoisoned<S4>()); 2630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov Arg3ExpectUMR(0, 1, *GetPoisoned<S8>()); 2640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 2650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, CallAndRet) { 2680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) return; 2690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ReturnPoisoned<S1>(); 2700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ReturnPoisoned<S2>(); 2710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ReturnPoisoned<S4>(); 2720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ReturnPoisoned<S8>(); 2730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 27411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(ReturnPoisoned<S1>()); 27511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(ReturnPoisoned<S2>()); 27611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(ReturnPoisoned<S4>()); 27711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(ReturnPoisoned<S8>()); 2780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 2790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// malloc() in the following test may be optimized to produce a compile-time 2810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// undef value. Check that we trap on the volatile assignment anyway. 2820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, DISABLED_MallocNoIdent) { 2830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 *x = (int*)malloc(sizeof(S4)); 28411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*x); 2850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov free(x); 2860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 2870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, Malloc) { 2890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 *x = (int*)Ident(malloc(sizeof(S4))); 29011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*x); 2910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov free(x); 2920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 2930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, Realloc) { 2950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 *x = (int*)Ident(realloc(0, sizeof(S4))); 29611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[0]); 2970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[0] = 1; 2980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x = (int*)Ident(realloc(x, 2 * sizeof(S4))); 29911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[0]); // Ok, was inited before. 30011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[1]); 3010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x = (int*)Ident(realloc(x, 3 * sizeof(S4))); 30211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[0]); // Ok, was inited before. 30311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[2]); 30411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[1]); 3050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[2] = 1; // Init this here. Check that after realloc it is poisoned again. 3060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x = (int*)Ident(realloc(x, 2 * sizeof(S4))); 30711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[0]); // Ok, was inited before. 30811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[1]); 3090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x = (int*)Ident(realloc(x, 3 * sizeof(S4))); 31011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[1]); 31111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[2]); 3120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov free(x); 3130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 3140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 3150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, Calloc) { 3160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 *x = (int*)Ident(calloc(1, sizeof(S4))); 31711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*x); // Should not be poisoned. 3180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // EXPECT_EQ(0, *x); 3190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov free(x); 3200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 3210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 3220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, AndOr) { 3230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U4 *p = GetPoisoned<U4>(); 3240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // We poison two bytes in the midle of a 4-byte word to make the test 3250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // correct regardless of endianness. 3260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ((U1*)p)[1] = 0; 3270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ((U1*)p)[2] = 0xff; 32811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*p & 0x00ffff00); 32911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*p & 0x00ff0000); 33011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*p & 0x0000ff00); 33111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*p & 0xff000000); 33211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*p & 0x000000ff); 33311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*p & 0x0000ffff); 33411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*p & 0xffff0000); 3350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 33611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*p | 0xff0000ff); 33711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*p | 0xff00ffff); 33811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*p | 0xffff00ff); 33911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*p | 0xff000000); 34011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*p | 0x000000ff); 34111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*p | 0x0000ffff); 34211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*p | 0xffff0000); 3430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 34411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<bool>() & *GetPoisoned<bool>()); 3450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 3460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 3470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> 34811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovstatic bool applyNot(T value, T shadow) { 3490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_partial_poison(&value, &shadow, sizeof(T)); 35011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov return !value; 3510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 3520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 3530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, Not) { 35411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U4>(0x0, 0x0)); 35511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U4>(0xFFFFFFFF, 0x0)); 35611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(applyNot<U4>(0xFFFFFFFF, 0xFFFFFFFF)); 35711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U4>(0xFF000000, 0x0FFFFFFF)); 35811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U4>(0xFF000000, 0x00FFFFFF)); 35911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U4>(0xFF000000, 0x0000FFFF)); 36011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U4>(0xFF000000, 0x00000000)); 36111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(applyNot<U4>(0xFF000000, 0xFF000000)); 36211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U4>(0xFF800000, 0xFF000000)); 36311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(applyNot<U4>(0x00008000, 0x00008000)); 36411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 36511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U1>(0x0, 0x0)); 36611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U1>(0xFF, 0xFE)); 36711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U1>(0xFF, 0x0)); 36811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(applyNot<U1>(0xFF, 0xFF)); 36911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 37011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(applyNot<void*>((void*)0xFFFFFF, (void*)(-1))); 37111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<void*>((void*)0xFFFFFF, (void*)(-2))); 3720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 3730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 3740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, Shift) { 3750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U4 *up = GetPoisoned<U4>(); 3760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ((U1*)up)[0] = 0; 3770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ((U1*)up)[3] = 0xff; 37811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*up >> 30); 37911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*up >> 24); 38011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*up >> 23); 38111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*up >> 10); 3820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 38311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*up << 30); 38411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*up << 24); 38511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*up << 23); 38611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*up << 10); 3870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 3880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 *sp = (S4*)up; 38911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*sp >> 30); 39011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*sp >> 24); 39111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*sp >> 23); 39211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*sp >> 10); 3930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 3940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov sp = GetPoisoned<S4>(); 3950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ((S1*)sp)[1] = 0; 3960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ((S1*)sp)[2] = 0; 39711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*sp >> 31); 3980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 39911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(100 >> *GetPoisoned<S4>()); 40011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(100U >> *GetPoisoned<S4>()); 4010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE static int GetPoisonedZero() { 4040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int *zero = new int; 4050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *zero = 0; 4060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(zero, sizeof(*zero)); 4070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = *zero; 4080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete zero; 4090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 4100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, LoadFromDirtyAddress) { 4130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int *a = new int; 4140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *a = 0; 41512c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov EXPECT_UMR(break_optimization((void*)(U8)a[GetPoisonedZero()])); 4160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete a; 4170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, StoreToDirtyAddress) { 4200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int *a = new int; 42111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR(a[GetPoisonedZero()] = 0); 42212c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(a); 4230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete a; 4240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE void StackTestFunc() { 4280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 p4; 4290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 ok4 = 1; 4300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S2 p2; 4310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S2 ok2 = 1; 4320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S1 p1; 4330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S1 ok1 = 1; 43412c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&p4); 43512c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&ok4); 43612c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&p2); 43712c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&ok2); 43812c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&p1); 43912c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&ok1); 4400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 44111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(p4); 44211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(p2); 44311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(p1); 44411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(ok1); 44511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(ok2); 44611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(ok4); 4470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, StackTest) { 4500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StackTestFunc(); 4510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE void StackStressFunc() { 4540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int foo[10000]; 45512c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(foo); 4560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, DISABLED_StackStressTest) { 4590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov for (int i = 0; i < 1000000; i++) 4600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StackStressFunc(); 4610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> 4640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovvoid TestFloatingPoint() { 4650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov static volatile T v; 4660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov static T g[100]; 46712c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&g); 4680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *x = GetPoisoned<T>(); 4690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *y = GetPoisoned<T>(1); 47011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*x); 47111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED((long long)*x); 47211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED((int)*x); 4730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov g[0] = *x; 4740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov g[1] = *x + *y; 4750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov g[2] = *x - *y; 4760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov g[3] = *x * *y; 4770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, FloatingPointTest) { 4800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestFloatingPoint<float>(); 4810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestFloatingPoint<double>(); 4820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, DynMem) { 4850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 x = 0; 4860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 *y = GetPoisoned<S4>(); 4870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov memcpy(y, &x, g_one * sizeof(S4)); 48811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*y); 4890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic char *DynRetTestStr; 4920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, DynRet) { 4940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) return; 4950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ReturnPoisoned<S8>(); 49611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(clearenv()); 4970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, DynRet1) { 5010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) return; 5020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ReturnPoisoned<S8>(); 5030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct LargeStruct { 5060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 x[10]; 5070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 5080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE 5100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovLargeStruct LargeRetTest() { 5110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov LargeStruct res; 5120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[0] = *GetPoisoned<S4>(); 5130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[1] = *GetPoisoned<S4>(); 5140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[2] = *GetPoisoned<S4>(); 5150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[3] = *GetPoisoned<S4>(); 5160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[4] = *GetPoisoned<S4>(); 5170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[5] = *GetPoisoned<S4>(); 5180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[6] = *GetPoisoned<S4>(); 5190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[7] = *GetPoisoned<S4>(); 5200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[8] = *GetPoisoned<S4>(); 5210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[9] = *GetPoisoned<S4>(); 5220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 5230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, LargeRet) { 5260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov LargeStruct a = LargeRetTest(); 52711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(a.x[0]); 52811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(a.x[9]); 5290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, fread) { 5320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *x = new char[32]; 5330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov FILE *f = fopen("/proc/self/stat", "r"); 5340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(f); 5350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov fread(x, 1, 32, f); 53611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[0]); 53711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[16]); 53811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[31]); 5390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov fclose(f); 5400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete x; 5410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, read) { 5440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *x = new char[32]; 5450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int fd = open("/proc/self/stat", O_RDONLY); 5460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(fd > 0); 5470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int sz = read(fd, x, 32); 5480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(sz == 32); 54911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[0]); 55011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[16]); 55111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[31]); 5520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov close(fd); 5530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete x; 5540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, pread) { 5570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *x = new char[32]; 5580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int fd = open("/proc/self/stat", O_RDONLY); 5590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(fd > 0); 5600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int sz = pread(fd, x, 32, 0); 5610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(sz == 32); 56211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[0]); 56311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[16]); 56411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[31]); 5650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov close(fd); 5660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete x; 5670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// FIXME: fails now. 5700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, DISABLED_ioctl) { 5710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov struct winsize ws; 5720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(ioctl(2, TIOCGWINSZ, &ws), 0); 57311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(ws.ws_col); 5740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, readlink) { 5770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *x = new char[1000]; 5780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov readlink("/proc/self/exe", x, 1000); 57911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[0]); 5800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete [] x; 5810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, stat) { 5850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov struct stat* st = new struct stat; 5860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = stat("/proc/self/stat", st); 5870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(!res); 58811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(st->st_dev); 58911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(st->st_mode); 59011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(st->st_size); 5910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, statfs) { 5940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov struct statfs* st = new struct statfs; 5950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = statfs("/", st); 5960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(!res); 59711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(st->f_type); 59811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(st->f_bfree); 59911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(st->f_namelen); 6000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 6010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 6020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, pipe) { 6030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* pipefd = new int[2]; 6040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = pipe(pipefd); 6050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(!res); 60611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(pipefd[0]); 60711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(pipefd[1]); 6080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov close(pipefd[0]); 6090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov close(pipefd[1]); 6100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 6110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 612134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy StepanovTEST(MemorySanitizer, pipe2) { 613134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov int* pipefd = new int[2]; 614134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov int res = pipe2(pipefd, O_NONBLOCK); 615134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov assert(!res); 616134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov EXPECT_NOT_POISONED(pipefd[0]); 617134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov EXPECT_NOT_POISONED(pipefd[1]); 618134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov close(pipefd[0]); 619134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov close(pipefd[1]); 620134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov} 621134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov 622134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy StepanovTEST(MemorySanitizer, socketpair) { 623134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov int sv[2]; 624134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov int res = socketpair(AF_UNIX, SOCK_STREAM, 0, sv); 625134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov assert(!res); 626134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov EXPECT_NOT_POISONED(sv[0]); 627134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov EXPECT_NOT_POISONED(sv[1]); 628134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov close(sv[0]); 629134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov close(sv[1]); 630134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov} 631134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov 6329f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy StepanovTEST(MemorySanitizer, bind_getsockname) { 6339f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov int sock = socket(AF_UNIX, SOCK_STREAM, 0); 6349f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov 6359f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov struct sockaddr_in sai; 6369f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov memset(&sai, 0, sizeof(sai)); 6379f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov sai.sin_family = AF_UNIX; 6389f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov int res = bind(sock, (struct sockaddr *)&sai, sizeof(sai)); 6399f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov 6409f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov assert(!res); 6419f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov char buf[200]; 6429f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov socklen_t addrlen; 6439f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov EXPECT_UMR(getsockname(sock, (struct sockaddr *)&buf, &addrlen)); 6449f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov 6459f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov addrlen = sizeof(buf); 6469f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov res = getsockname(sock, (struct sockaddr *)&buf, &addrlen); 6479f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov EXPECT_NOT_POISONED(addrlen); 6489f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov EXPECT_NOT_POISONED(buf[0]); 6499f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov EXPECT_NOT_POISONED(buf[addrlen - 1]); 6509f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov EXPECT_POISONED(buf[addrlen]); 6519f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov close(sock); 6529f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov} 6539f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov 6549d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy StepanovTEST(MemorySanitizer, accept) { 6559d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov int listen_socket = socket(AF_INET, SOCK_STREAM, 0); 6569d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_LT(0, listen_socket); 6579d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov 6589d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov struct sockaddr_in sai; 6599d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov sai.sin_family = AF_INET; 6609d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov sai.sin_port = 0; 6619d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov sai.sin_addr.s_addr = htonl(INADDR_LOOPBACK); 6629d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov int res = bind(listen_socket, (struct sockaddr *)&sai, sizeof(sai)); 6639d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_EQ(0, res); 6649d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov 6659d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov res = listen(listen_socket, 1); 6669d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_EQ(0, res); 6679d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov 6689d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov socklen_t sz = sizeof(sai); 6699d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov res = getsockname(listen_socket, (struct sockaddr *)&sai, &sz); 6709d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_EQ(0, res); 6719d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_EQ(sizeof(sai), sz); 6729d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov 6739d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov int connect_socket = socket(AF_INET, SOCK_STREAM, 0); 6749d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_LT(0, connect_socket); 6759d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov res = fcntl(connect_socket, F_SETFL, O_NONBLOCK); 6769d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_EQ(0, res); 6779d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov res = connect(connect_socket, (struct sockaddr *)&sai, sizeof(sai)); 6789d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_EQ(-1, res); 6799d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_EQ(EINPROGRESS, errno); 6809d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov 6819d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov __msan_poison(&sai, sizeof(sai)); 6829d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov int new_sock = accept(listen_socket,(struct sockaddr *)&sai, &sz); 6839d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_LT(0, new_sock); 6849d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_EQ(sizeof(sai), sz); 6859d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov EXPECT_NOT_POISONED(sai); 6869d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov 6879d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov close(new_sock); 6889d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov close(connect_socket); 6899d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov close(listen_socket); 6909d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov} 6919d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov 6920a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov#define EXPECT_HOSTENT_NOT_POISONED(he) \ 6930a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov do { \ 6940a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(*(he)); \ 6950a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *) 0, (he)->h_name); \ 6960a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *) 0, (he)->h_aliases); \ 6970a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *) 0, (he)->h_addr_list); \ 6980a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(strlen((he)->h_name)); \ 6990a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov char **p = (he)->h_aliases; \ 7000a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov while (*p) { \ 7010a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(*p)); \ 7020a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ++p; \ 7030a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov } \ 7040a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov char **q = (he)->h_addr_list; \ 7050a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov while (*q) { \ 7060a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(*q[0]); \ 7070a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ++q; \ 7080a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov } \ 7090a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(*q); \ 7100a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov } while (0) 7110a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov 7120a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy StepanovTEST(MemorySanitizer, gethostent) { 7130a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent *he = gethostent(); 7140a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *)NULL, he); 7150a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_HOSTENT_NOT_POISONED(he); 7160a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov} 7170a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov 7180a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy StepanovTEST(MemorySanitizer, gethostbyname) { 7190a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent *he = gethostbyname("localhost"); 7200a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *)NULL, he); 7210a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_HOSTENT_NOT_POISONED(he); 7220a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov} 7230a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov 7240a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy StepanovTEST(MemorySanitizer, gethostbyname2) { 7250a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent *he = gethostbyname2("localhost", AF_INET); 7260a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *)NULL, he); 7270a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_HOSTENT_NOT_POISONED(he); 7280a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov} 7290a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov 7300a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy StepanovTEST(MemorySanitizer, gethostbyaddr) { 7310a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov in_addr_t addr = inet_addr("127.0.0.1"); 7320a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(addr); 7330a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent *he = gethostbyaddr(&addr, sizeof(addr), AF_INET); 7340a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *)NULL, he); 7350a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_HOSTENT_NOT_POISONED(he); 7360a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov} 7370a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov 7380a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy StepanovTEST(MemorySanitizer, gethostent_r) { 7390a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov char buf[2000]; 7400a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent he; 7410a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent *result; 7420a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov int err; 7430a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov int res = gethostent_r(&he, buf, sizeof(buf), &result, &err); 7440a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_EQ(0, res); 7450a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(result); 7460a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *)NULL, result); 7470a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_HOSTENT_NOT_POISONED(result); 7480a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(err); 7490a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov} 7500a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov 7510a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy StepanovTEST(MemorySanitizer, gethostbyname_r) { 7520a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov char buf[2000]; 7530a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent he; 7540a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent *result; 7550a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov int err; 7560a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov int res = gethostbyname_r("localhost", &he, buf, sizeof(buf), &result, &err); 7570a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_EQ(0, res); 7580a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(result); 7590a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *)NULL, result); 7600a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_HOSTENT_NOT_POISONED(result); 7610a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(err); 7620a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov} 7630a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov 7640a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy StepanovTEST(MemorySanitizer, gethostbyname2_r) { 7650a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov char buf[2000]; 7660a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent he; 7670a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent *result; 7680a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov int err; 7690a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov int res = gethostbyname2_r("localhost", AF_INET, &he, buf, sizeof(buf), 7700a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov &result, &err); 7710a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_EQ(0, res); 7720a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(result); 7730a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *)NULL, result); 7740a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_HOSTENT_NOT_POISONED(result); 7750a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(err); 7760a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov} 7770a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov 7780a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy StepanovTEST(MemorySanitizer, gethostbyaddr_r) { 7790a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov char buf[2000]; 7800a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent he; 7810a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent *result; 7820a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov int err; 7830a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov in_addr_t addr = inet_addr("127.0.0.1"); 7840a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(addr); 7850a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov int res = gethostbyaddr_r(&addr, sizeof(addr), AF_INET, &he, buf, sizeof(buf), 7860a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov &result, &err); 7870a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_EQ(0, res); 7880a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(result); 7890a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *)NULL, result); 7900a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_HOSTENT_NOT_POISONED(result); 7910a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(err); 7920a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov} 7930a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov 794f32be42523a199674ea665a499db131591e64e08Evgeniy StepanovTEST(MemorySanitizer, getsockopt) { 795f32be42523a199674ea665a499db131591e64e08Evgeniy Stepanov int sock = socket(AF_UNIX, SOCK_STREAM, 0); 796f32be42523a199674ea665a499db131591e64e08Evgeniy Stepanov struct linger l[2]; 797f32be42523a199674ea665a499db131591e64e08Evgeniy Stepanov socklen_t sz = sizeof(l[0]); 798f32be42523a199674ea665a499db131591e64e08Evgeniy Stepanov int res = getsockopt(sock, SOL_SOCKET, SO_LINGER, &l[0], &sz); 799f32be42523a199674ea665a499db131591e64e08Evgeniy Stepanov ASSERT_EQ(0, res); 800f32be42523a199674ea665a499db131591e64e08Evgeniy Stepanov ASSERT_EQ(sizeof(l[0]), sz); 801f32be42523a199674ea665a499db131591e64e08Evgeniy Stepanov EXPECT_NOT_POISONED(l[0]); 802f32be42523a199674ea665a499db131591e64e08Evgeniy Stepanov EXPECT_POISONED(*(char *)(l + 1)); 803f32be42523a199674ea665a499db131591e64e08Evgeniy Stepanov} 804f32be42523a199674ea665a499db131591e64e08Evgeniy Stepanov 8050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, getcwd) { 8060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char path[PATH_MAX + 1]; 8070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* res = getcwd(path, sizeof(path)); 8080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(res); 80911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(path[0]); 8100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 8110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 8127eed04c4dce69ad1e485edbf6dd963e176b52e0dEvgeniy StepanovTEST(MemorySanitizer, getcwd_gnu) { 8137eed04c4dce69ad1e485edbf6dd963e176b52e0dEvgeniy Stepanov char* res = getcwd(NULL, 0); 8147eed04c4dce69ad1e485edbf6dd963e176b52e0dEvgeniy Stepanov assert(res); 8157eed04c4dce69ad1e485edbf6dd963e176b52e0dEvgeniy Stepanov EXPECT_NOT_POISONED(res[0]); 8167eed04c4dce69ad1e485edbf6dd963e176b52e0dEvgeniy Stepanov free(res); 8177eed04c4dce69ad1e485edbf6dd963e176b52e0dEvgeniy Stepanov} 8187eed04c4dce69ad1e485edbf6dd963e176b52e0dEvgeniy Stepanov 819d97a15a931ae6696a1071e1471c9a019e821904bEvgeniy StepanovTEST(MemorySanitizer, readdir) { 820d97a15a931ae6696a1071e1471c9a019e821904bEvgeniy Stepanov DIR *dir = opendir("."); 821d97a15a931ae6696a1071e1471c9a019e821904bEvgeniy Stepanov struct dirent *d = readdir(dir); 822d97a15a931ae6696a1071e1471c9a019e821904bEvgeniy Stepanov assert(d); 823d97a15a931ae6696a1071e1471c9a019e821904bEvgeniy Stepanov EXPECT_NOT_POISONED(d->d_name[0]); 824d97a15a931ae6696a1071e1471c9a019e821904bEvgeniy Stepanov closedir(dir); 825d97a15a931ae6696a1071e1471c9a019e821904bEvgeniy Stepanov} 826d97a15a931ae6696a1071e1471c9a019e821904bEvgeniy Stepanov 8270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, realpath) { 8280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov const char* relpath = "."; 8290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char path[PATH_MAX + 1]; 8300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* res = realpath(relpath, path); 8310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(res); 83211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(path[0]); 8330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 8340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 8350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, memcpy) { 8360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* x = new char[2]; 8370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* y = new char[2]; 8380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[0] = 1; 8390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[1] = *GetPoisoned<char>(); 8400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov memcpy(y, x, 2); 84111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(y[0]); 84211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(y[1]); 8430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 8440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 8450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, memmove) { 8460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* x = new char[2]; 8470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* y = new char[2]; 8480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[0] = 1; 8490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[1] = *GetPoisoned<char>(); 8500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov memmove(y, x, 2); 85111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(y[0]); 85211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(y[1]); 8530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 8540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 8550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, strdup) { 8568aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov char buf[4] = "abc"; 8578aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov __msan_poison(buf + 2, sizeof(*buf)); 8588aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov char *x = strdup(buf); 8598aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov EXPECT_NOT_POISONED(x[0]); 8608aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov EXPECT_NOT_POISONED(x[1]); 8618aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov EXPECT_POISONED(x[2]); 8628aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov EXPECT_NOT_POISONED(x[3]); 8638aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov free(x); 8648aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov} 8658aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov 8668aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy StepanovTEST(MemorySanitizer, strndup) { 8678aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov char buf[4] = "abc"; 8688aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov __msan_poison(buf + 2, sizeof(*buf)); 8698aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov char *x = strndup(buf, 3); 8708aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov EXPECT_NOT_POISONED(x[0]); 8718aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov EXPECT_NOT_POISONED(x[1]); 8728aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov EXPECT_POISONED(x[2]); 8738aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov EXPECT_NOT_POISONED(x[3]); 8740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov free(x); 8750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 8760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 8778aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy StepanovTEST(MemorySanitizer, strndup_short) { 8788aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov char buf[4] = "abc"; 8798aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov __msan_poison(buf + 1, sizeof(*buf)); 8808aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov __msan_poison(buf + 2, sizeof(*buf)); 8818aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov char *x = strndup(buf, 2); 8828aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov EXPECT_NOT_POISONED(x[0]); 8838aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov EXPECT_POISONED(x[1]); 8841d21bd159c3830dd7a6c05854408442d91a336adEvgeniy Stepanov EXPECT_NOT_POISONED(x[2]); 8858aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov free(x); 8868aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov} 8878aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov 8888aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov 8890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T, int size> 8900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovvoid TestOverlapMemmove() { 8910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *x = new T[size]; 8920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(size >= 3); 8930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[2] = 0; 8940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov memmove(x, x + 1, (size - 1) * sizeof(T)); 89511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[1]); 8960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) { 8970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // FIXME: under DR we will lose this information 8980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // because accesses in memmove will unpoisin the shadow. 8990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // We need to use our own memove implementation instead of libc's. 90011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[0]); 90111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[2]); 9020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } 9030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete [] x; 9040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, overlap_memmove) { 9070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestOverlapMemmove<U1, 10>(); 9080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestOverlapMemmove<U1, 1000>(); 9090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestOverlapMemmove<U8, 4>(); 9100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestOverlapMemmove<U8, 1000>(); 9110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, strcpy) { // NOLINT 9140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* x = new char[3]; 9150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* y = new char[3]; 9160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[0] = 'a'; 9170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[1] = *GetPoisoned<char>(1, 1); 9180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[2] = 0; 9190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov strcpy(y, x); // NOLINT 92011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(y[0]); 92111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(y[1]); 92211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(y[2]); 9230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, strncpy) { // NOLINT 9260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* x = new char[3]; 9270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* y = new char[3]; 9280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[0] = 'a'; 9290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[1] = *GetPoisoned<char>(1, 1); 9300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[2] = 0; 9310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov strncpy(y, x, 2); // NOLINT 93211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(y[0]); 93311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(y[1]); 93411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(y[2]); 9350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, strtol) { 9380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *e; 9390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(1 == strtol("1", &e, 10)); 94011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((S8) e); 9410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, strtoll) { 9440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *e; 9450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(1 == strtoll("1", &e, 10)); 94611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((S8) e); 9470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, strtoul) { 9500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *e; 9510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(1 == strtoul("1", &e, 10)); 95211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((S8) e); 9530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, strtoull) { 9560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *e; 9570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(1 == strtoull("1", &e, 10)); 95811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((S8) e); 9590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 961e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy StepanovTEST(MemorySanitizer, strtod) { 962e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov char *e; 963e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov assert(0 != strtod("1.5", &e)); 96411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((S8) e); 965e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov} 966e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov 967e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy StepanovTEST(MemorySanitizer, strtof) { 968e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov char *e; 969e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov assert(0 != strtof("1.5", &e)); 97011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((S8) e); 971e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov} 972e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov 973e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy StepanovTEST(MemorySanitizer, strtold) { 974e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov char *e; 975e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov assert(0 != strtold("1.5", &e)); 97611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((S8) e); 977e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov} 978e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov 9790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, sprintf) { // NOLINT 9800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char buff[10]; 98112c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(buff); 98211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(buff[0]); 9830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = sprintf(buff, "%d", 1234567); // NOLINT 9840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(res == 7); 9850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[0] == '1'); 9860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[1] == '2'); 9870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[2] == '3'); 9880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[6] == '7'); 9890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[7] == 0); 99011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(buff[8]); 9910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, snprintf) { 9940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char buff[10]; 99512c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(buff); 99611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(buff[0]); 9970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = snprintf(buff, sizeof(buff), "%d", 1234567); 9980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(res == 7); 9990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[0] == '1'); 10000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[1] == '2'); 10010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[2] == '3'); 10020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[6] == '7'); 10030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[7] == 0); 100411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(buff[8]); 10050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 10060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 10070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, swprintf) { 10080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov wchar_t buff[10]; 10090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(sizeof(wchar_t) == 4); 101012c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(buff); 101111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(buff[0]); 10120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = swprintf(buff, 9, L"%d", 1234567); 10130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(res == 7); 10140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[0] == '1'); 10150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[1] == '2'); 10160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[2] == '3'); 10170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[6] == '7'); 10180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[7] == 0); 101911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(buff[8]); 10200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 10210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 10220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, wcstombs) { 10230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov const wchar_t *x = L"abc"; 10240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char buff[10]; 10250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = wcstombs(buff, x, 4); 10260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(res, 3); 10270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(buff[0], 'a'); 10280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(buff[1], 'b'); 10290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(buff[2], 'c'); 10300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 10310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 10320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, gettimeofday) { 10330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov struct timeval tv; 10340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov struct timezone tz; 103512c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&tv); 103612c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&tz); 10370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(sizeof(tv) == 16); 10380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(sizeof(tz) == 8); 103911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(tv.tv_sec); 104011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(tv.tv_usec); 104111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(tz.tz_minuteswest); 104211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(tz.tz_dsttime); 10430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(0 == gettimeofday(&tv, &tz)); 104411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(tv.tv_sec); 104511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(tv.tv_usec); 104611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(tz.tz_minuteswest); 104711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(tz.tz_dsttime); 10480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 10490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1050e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy StepanovTEST(MemorySanitizer, clock_gettime) { 1051e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov struct timespec tp; 1052e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_POISONED(tp.tv_sec); 1053e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_POISONED(tp.tv_nsec); 1054e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov assert(0 == clock_gettime(CLOCK_REALTIME, &tp)); 1055e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(tp.tv_sec); 1056e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(tp.tv_nsec); 1057e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov} 1058e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov 10597cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy StepanovTEST(MemorySanitizer, clock_getres) { 10607cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov struct timespec tp; 10617cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov EXPECT_POISONED(tp.tv_sec); 10627cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov EXPECT_POISONED(tp.tv_nsec); 10637cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov assert(0 == clock_getres(CLOCK_REALTIME, 0)); 10647cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov EXPECT_POISONED(tp.tv_sec); 10657cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov EXPECT_POISONED(tp.tv_nsec); 10667cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov assert(0 == clock_getres(CLOCK_REALTIME, &tp)); 10677cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov EXPECT_NOT_POISONED(tp.tv_sec); 10687cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov EXPECT_NOT_POISONED(tp.tv_nsec); 10697cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov} 10707cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov 1071e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy StepanovTEST(MemorySanitizer, getitimer) { 1072e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov struct itimerval it1, it2; 1073e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov int res; 1074e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_POISONED(it1.it_interval.tv_sec); 1075e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_POISONED(it1.it_interval.tv_usec); 1076e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_POISONED(it1.it_value.tv_sec); 1077e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_POISONED(it1.it_value.tv_usec); 1078e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov res = getitimer(ITIMER_VIRTUAL, &it1); 1079e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov assert(!res); 1080e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(it1.it_interval.tv_sec); 1081e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(it1.it_interval.tv_usec); 1082e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(it1.it_value.tv_sec); 1083e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(it1.it_value.tv_usec); 1084e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov 1085e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov it1.it_interval.tv_sec = it1.it_value.tv_sec = 10000; 1086e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov it1.it_interval.tv_usec = it1.it_value.tv_usec = 0; 1087e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov 1088e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov res = setitimer(ITIMER_VIRTUAL, &it1, &it2); 1089e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov assert(!res); 1090e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(it2.it_interval.tv_sec); 1091e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(it2.it_interval.tv_usec); 1092e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(it2.it_value.tv_sec); 1093e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(it2.it_value.tv_usec); 1094e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov 1095e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov // Check that old_value can be 0, and disable the timer. 1096e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov memset(&it1, 0, sizeof(it1)); 1097e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov res = setitimer(ITIMER_VIRTUAL, &it1, 0); 1098e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov assert(!res); 1099e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov} 1100e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov 1101fef660506e9e5703fedfee01d614abd4b741c738Evgeniy StepanovTEST(MemorySanitizer, time) { 1102fef660506e9e5703fedfee01d614abd4b741c738Evgeniy Stepanov time_t t; 1103fef660506e9e5703fedfee01d614abd4b741c738Evgeniy Stepanov EXPECT_POISONED(t); 1104fef660506e9e5703fedfee01d614abd4b741c738Evgeniy Stepanov time_t t2 = time(&t); 1105fef660506e9e5703fedfee01d614abd4b741c738Evgeniy Stepanov assert(t2 != (time_t)-1); 1106fef660506e9e5703fedfee01d614abd4b741c738Evgeniy Stepanov EXPECT_NOT_POISONED(t); 1107fef660506e9e5703fedfee01d614abd4b741c738Evgeniy Stepanov} 1108fef660506e9e5703fedfee01d614abd4b741c738Evgeniy Stepanov 11099358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy StepanovTEST(MemorySanitizer, localtime) { 11109358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov time_t t = 123; 11119358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov struct tm *time = localtime(&t); 11129358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov assert(time != 0); 11139358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov EXPECT_NOT_POISONED(time->tm_sec); 11149358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov EXPECT_NOT_POISONED(time->tm_hour); 11159358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov EXPECT_NOT_POISONED(time->tm_year); 11169358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov EXPECT_NOT_POISONED(time->tm_isdst); 11179358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov} 11189358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov 11199358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy StepanovTEST(MemorySanitizer, localtime_r) { 11209358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov time_t t = 123; 11219358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov struct tm time; 11229358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov struct tm *res = localtime_r(&t, &time); 11239358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov assert(res != 0); 11249358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov EXPECT_NOT_POISONED(time.tm_sec); 11259358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov EXPECT_NOT_POISONED(time.tm_hour); 11269358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov EXPECT_NOT_POISONED(time.tm_year); 11279358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov EXPECT_NOT_POISONED(time.tm_isdst); 11289358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov} 11299358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov 11300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, mmap) { 11310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov const int size = 4096; 11320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov void *p1, *p2; 11330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov p1 = mmap(0, size, PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANON, -1, 0); 11340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(p1, size); 11350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov munmap(p1, size); 11360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov for (int i = 0; i < 1000; i++) { 11370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov p2 = mmap(0, size, PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANON, -1, 0); 11380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (p2 == p1) 11390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov break; 11400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov else 11410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov munmap(p2, size); 11420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } 11430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (p1 == p2) { 114411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*(char*)p2); 11450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov munmap(p2, size); 11460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } 11470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 11480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 11490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// FIXME: enable and add ecvt. 11500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// FIXME: check why msandr does nt handle fcvt. 11510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, fcvt) { 11520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int a, b; 115312c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&a); 115412c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&b); 115511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(a); 115611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(b); 11570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *str = fcvt(12345.6789, 10, &a, &b); 115811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(a); 115911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(b); 11600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 11610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 11627cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy StepanovTEST(MemorySanitizer, frexp) { 11637cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov int x; 11647cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov x = *GetPoisoned<int>(); 11657cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov double r = frexp(1.1, &x); 11667cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov EXPECT_NOT_POISONED(r); 11677cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov EXPECT_NOT_POISONED(x); 11687cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov 11697cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov x = *GetPoisoned<int>(); 11707cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov float rf = frexpf(1.1, &x); 11717cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov EXPECT_NOT_POISONED(rf); 11727cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov EXPECT_NOT_POISONED(x); 11737cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov 11747cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov x = *GetPoisoned<int>(); 11757cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov double rl = frexpl(1.1, &x); 11767cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov EXPECT_NOT_POISONED(rl); 11777cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov EXPECT_NOT_POISONED(x); 11787cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov} 11797cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov 118006658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanovnamespace { 118106658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov 118206658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanovstatic int cnt; 118306658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov 118406658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanovvoid SigactionHandler(int signo, siginfo_t* si, void* uc) { 118506658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov assert(signo == SIGPROF); 118606658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov assert(si); 118706658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov EXPECT_NOT_POISONED(si->si_errno); 118806658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov EXPECT_NOT_POISONED(si->si_pid); 118906658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov#if __linux__ 119006658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov# if defined(__x86_64__) 119106658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov EXPECT_NOT_POISONED(((ucontext_t*)uc)->uc_mcontext.gregs[REG_RIP]); 119206658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov# elif defined(__i386__) 119306658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov EXPECT_NOT_POISONED(((ucontext_t*)uc)->uc_mcontext.gregs[REG_EIP]); 119406658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov# endif 119506658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov#endif 119606658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov ++cnt; 119706658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov} 119806658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov 119906658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy StepanovTEST(MemorySanitizer, sigaction) { 120006658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov struct sigaction act = {}; 1201cd3049da150124156502b1a8c05e4c4887786cc5Evgeniy Stepanov struct sigaction oldact = {}; 12026d0b7f6c41bed64419dd74a8daf83de5e9f54de9Evgeniy Stepanov struct sigaction origact = {}; 12036d0b7f6c41bed64419dd74a8daf83de5e9f54de9Evgeniy Stepanov 12046d0b7f6c41bed64419dd74a8daf83de5e9f54de9Evgeniy Stepanov sigaction(SIGPROF, 0, &origact); 12056d0b7f6c41bed64419dd74a8daf83de5e9f54de9Evgeniy Stepanov 120606658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov act.sa_flags |= SA_SIGINFO; 120706658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov act.sa_sigaction = &SigactionHandler; 120806658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov sigaction(SIGPROF, &act, 0); 120906658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov 121006658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov kill(getpid(), SIGPROF); 121106658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov 121206658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov act.sa_flags &= ~SA_SIGINFO; 121306658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov act.sa_handler = SIG_DFL; 121406658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov sigaction(SIGPROF, &act, 0); 121506658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov 121606658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov act.sa_flags &= ~SA_SIGINFO; 121706658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov act.sa_handler = SIG_IGN; 1218cd3049da150124156502b1a8c05e4c4887786cc5Evgeniy Stepanov sigaction(SIGPROF, &act, &oldact); 1219cd3049da150124156502b1a8c05e4c4887786cc5Evgeniy Stepanov EXPECT_FALSE(oldact.sa_flags & SA_SIGINFO); 1220cd3049da150124156502b1a8c05e4c4887786cc5Evgeniy Stepanov EXPECT_EQ(SIG_DFL, oldact.sa_handler); 122106658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov kill(getpid(), SIGPROF); 122206658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov 122306658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov act.sa_flags |= SA_SIGINFO; 122406658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov act.sa_sigaction = &SigactionHandler; 1225cd3049da150124156502b1a8c05e4c4887786cc5Evgeniy Stepanov sigaction(SIGPROF, &act, &oldact); 1226cd3049da150124156502b1a8c05e4c4887786cc5Evgeniy Stepanov EXPECT_FALSE(oldact.sa_flags & SA_SIGINFO); 1227cd3049da150124156502b1a8c05e4c4887786cc5Evgeniy Stepanov EXPECT_EQ(SIG_IGN, oldact.sa_handler); 122806658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov kill(getpid(), SIGPROF); 122906658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov 123006658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov act.sa_flags &= ~SA_SIGINFO; 123106658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov act.sa_handler = SIG_DFL; 1232cd3049da150124156502b1a8c05e4c4887786cc5Evgeniy Stepanov sigaction(SIGPROF, &act, &oldact); 1233cd3049da150124156502b1a8c05e4c4887786cc5Evgeniy Stepanov EXPECT_TRUE(oldact.sa_flags & SA_SIGINFO); 1234cd3049da150124156502b1a8c05e4c4887786cc5Evgeniy Stepanov EXPECT_EQ(&SigactionHandler, oldact.sa_sigaction); 123506658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov EXPECT_EQ(2, cnt); 12366d0b7f6c41bed64419dd74a8daf83de5e9f54de9Evgeniy Stepanov 12376d0b7f6c41bed64419dd74a8daf83de5e9f54de9Evgeniy Stepanov sigaction(SIGPROF, &origact, 0); 123806658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov} 123906658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov 124006658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov} // namespace 124106658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov 12420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct StructWithDtor { 12430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ~StructWithDtor(); 12440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 12450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 12460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE StructWithDtor::~StructWithDtor() { 124712c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(0); 12480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 12490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 12500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, Invoke) { 12510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructWithDtor s; // Will cause the calls to become invokes. 125211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(0); 125311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<int>()); 125411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(0); 125511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<int>()); 125611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(ReturnPoisoned<S4>()); 12570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 12580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 12590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, ptrtoint) { 12600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // Test that shadow is propagated through pointer-to-integer conversion. 12610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov void* p = (void*)0xABCD; 12620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(((char*)&p) + 1, sizeof(p)); 1263250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov EXPECT_NOT_POISONED((((uintptr_t)p) & 0xFF) == 0); 12640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 12650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov void* q = (void*)0xABCD; 12660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&q, sizeof(q) - 1); 1267250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov EXPECT_POISONED((((uintptr_t)q) & 0xFF) == 0); 12680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 12690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 12700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn2(int guard, ...) { 12710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 12720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 127311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 127411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 127511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 127611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, double)); 12770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 12780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 12790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 12800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn(int guard, ...) { 12810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 12820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 128311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 128411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 12850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // The following call will overwrite __msan_param_tls. 12860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // Checks after it test that arg shadow was somehow saved across the call. 12870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn2(1, 2, 3, 4, *GetPoisoned<double>()); 128811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 128911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 12900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 12910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 12920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 12930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, VAArgTest) { 12940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* x = GetPoisoned<int>(); 12950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* y = GetPoisoned<int>(4); 12960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn(1, 13, *x, 42, *y); 12970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 12980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 12990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_many(int guard, ...) { 13000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 13010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 130211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 130311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 130411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 130511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 130611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 130711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 130811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 130911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 131011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 131111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 13120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 13130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, VAArgManyTest) { 13160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* x = GetPoisoned<int>(); 13170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* y = GetPoisoned<int>(4); 13180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_many(1, 2, *x, 3, 4, 5, 6, 7, 8, 9, *y); 13190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_pass2(va_list vl) { 132211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 132311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 132411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 13250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_pass(int guard, ...) { 13280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 13290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 133011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 13310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_pass2(vl); 13320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 13330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, VAArgPass) { 13360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* x = GetPoisoned<int>(); 13370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* y = GetPoisoned<int>(4); 13380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_pass(1, *x, 2, 3, *y); 13390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_copy2(va_list vl) { 134211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 134311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 13440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_copy(int guard, ...) { 13470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 13480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 134911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 135011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 13510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl2; 13520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_copy(vl2, vl); 13530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_copy2(vl2); 135411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 135511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 13560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 13570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, VAArgCopy) { 13600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* x = GetPoisoned<int>(); 13610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* y = GetPoisoned<int>(4); 13620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_copy(1, 2, *x, 3, *y); 13630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_ptr(int guard, ...) { 13660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 13670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 136811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int*)); 136911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int*)); 137011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int*)); 137111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, double*)); 13720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 13730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, VAArgPtr) { 13760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int** x = GetPoisoned<int*>(); 13770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov double** y = GetPoisoned<double*>(8); 13780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int z; 13790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_ptr(1, &z, *x, &z, *y); 13800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_overflow(int guard, ...) { 13830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 13840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 138511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 138611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 138711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 138811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 138911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 139011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 139111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 139211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, double)); 139311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, double)); 139411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, double)); 139511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, double)); 139611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, double)); 139711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int*)); 139811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, double)); 139911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, double)); 140011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 140111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 140211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, double)); 140311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int*)); 140411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 140511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 140611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, double)); 140711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int*)); 140811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 140911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 141011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, double)); 141111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int*)); 14120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 14140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, VAArgOverflow) { 14170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* x = GetPoisoned<int>(); 14180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov double* y = GetPoisoned<double>(8); 14190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int** p = GetPoisoned<int*>(16); 14200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int z; 14210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_overflow(1, 14220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1, 2, *x, 4, 5, 6, 14230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1.1, 2.2, 3.3, *y, 5.5, *p, 7.7, 8.8, 14240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // the following args will overflow for sure 14250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *x, *y, *p, 14260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 7, 9.9, &z, 14270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *x, *y, *p); 14280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_tlsoverwrite2(int guard, ...) { 14310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 14320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 143311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 14340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 14350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_tlsoverwrite(int guard, ...) { 14380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // This call will overwrite TLS contents unless it's backed up somewhere. 14390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_tlsoverwrite2(2, 42); 14400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 14410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 144211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 14430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 14440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, VAArgTLSOverwrite) { 14470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* x = GetPoisoned<int>(); 14480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_tlsoverwrite(1, *x); 14490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct StructByVal { 14520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int a, b, c, d, e, f; 14530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 14540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE void StructByValTestFunc(struct StructByVal s) { 145611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s.a); 145711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s.b); 145811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s.c); 145911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s.d); 146011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s.e); 146111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s.f); 14620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE void StructByValTestFunc1(struct StructByVal s) { 14650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructByValTestFunc(s); 14660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE void StructByValTestFunc2(int z, struct StructByVal s) { 14690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructByValTestFunc(s); 14700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, StructByVal) { 14730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // Large aggregates are passed as "byval" pointer argument in LLVM. 14740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov struct StructByVal s; 14750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov s.a = 1; 14760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov s.b = *GetPoisoned<int>(); 14770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov s.c = 2; 14780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov s.d = *GetPoisoned<int>(); 14790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov s.e = 3; 14800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov s.f = *GetPoisoned<int>(); 14810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructByValTestFunc(s); 14820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructByValTestFunc1(s); 14830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructByValTestFunc2(0, s); 14840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#if MSAN_HAS_M128 1488b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy StepanovNOINLINE __m128i m128Eq(__m128i *a, __m128i *b) { return _mm_cmpeq_epi16(*a, *b); } 1489b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy StepanovNOINLINE __m128i m128Lt(__m128i *a, __m128i *b) { return _mm_cmplt_epi16(*a, *b); } 14900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, m128) { 14910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __m128i a = _mm_set1_epi16(0x1234); 14920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __m128i b = _mm_set1_epi16(0x7890); 149311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(m128Eq(&a, &b)); 149411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(m128Lt(&a, &b)); 14950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// FIXME: add more tests for __m128i. 14970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#endif // MSAN_HAS_M128 14980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// We should not complain when copying this poisoned hole. 15000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct StructWithHole { 15010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U4 a; 15020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // 4-byte hole. 15030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U8 b; 15040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 15050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE StructWithHole ReturnStructWithHole() { 15070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructWithHole res; 15080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&res, sizeof(res)); 15090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.a = 1; 15100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.b = 2; 15110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 15120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, StructWithHole) { 15150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructWithHole a = ReturnStructWithHole(); 151612c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&a); 15170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate <class T> 15200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE T ReturnStruct() { 15210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T res; 15220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&res, sizeof(res)); 15230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.a = 1; 15240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 15250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate <class T> 15280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE void TestReturnStruct() { 15290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T s1 = ReturnStruct<T>(); 153011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s1.a); 153111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s1.b); 15320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct SSS1 { 15350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int a, b, c; 15360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 15370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct SSS2 { 15380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int b, a, c; 15390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 15400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct SSS3 { 15410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int b, c, a; 15420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 15430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct SSS4 { 15440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int c, b, a; 15450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 15460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct SSS5 { 15480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int a; 15490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov float b; 15500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 15510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct SSS6 { 15520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int a; 15530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov double b; 15540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 15550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct SSS7 { 15560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S8 b; 15570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int a; 15580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 15590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct SSS8 { 15600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S2 b; 15610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S8 a; 15620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 15630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, IntStruct3) { 15650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestReturnStruct<SSS1>(); 15660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestReturnStruct<SSS2>(); 15670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestReturnStruct<SSS3>(); 15680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestReturnStruct<SSS4>(); 15690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestReturnStruct<SSS5>(); 15700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestReturnStruct<SSS6>(); 15710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestReturnStruct<SSS7>(); 15720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestReturnStruct<SSS8>(); 15730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct LongStruct { 15760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U1 a1, b1; 15770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U2 a2, b2; 15780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U4 a4, b4; 15790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U8 a8, b8; 15800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 15810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE LongStruct ReturnLongStruct1() { 15830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov LongStruct res; 15840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&res, sizeof(res)); 15850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.a1 = res.a2 = res.a4 = res.a8 = 111; 15860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // leaves b1, .., b8 poisoned. 15870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 15880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE LongStruct ReturnLongStruct2() { 15910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov LongStruct res; 15920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&res, sizeof(res)); 15930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.b1 = res.b2 = res.b4 = res.b8 = 111; 15940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // leaves a1, .., a8 poisoned. 15950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 15960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, LongStruct) { 15990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov LongStruct s1 = ReturnLongStruct1(); 16000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_print_shadow(&s1, sizeof(s1)); 160111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s1.a1); 160211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s1.a2); 160311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s1.a4); 160411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s1.a8); 16050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 160611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s1.b1); 160711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s1.b2); 160811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s1.b4); 160911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s1.b8); 16100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 16110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov LongStruct s2 = ReturnLongStruct2(); 16120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_print_shadow(&s2, sizeof(s2)); 161311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s2.b1); 161411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s2.b2); 161511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s2.b4); 161611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s2.b8); 16170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 161811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s2.a1); 161911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s2.a2); 162011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s2.a4); 162111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s2.a8); 16220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 16230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 16240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, getrlimit) { 16250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov struct rlimit limit; 16260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&limit, sizeof(limit)); 16270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int result = getrlimit(RLIMIT_DATA, &limit); 16280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(result == 0); 16292bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov EXPECT_NOT_POISONED(limit.rlim_cur); 16302bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov EXPECT_NOT_POISONED(limit.rlim_max); 16310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 16320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1633e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy StepanovTEST(MemorySanitizer, getrusage) { 1634e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov struct rusage usage; 1635e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov __msan_poison(&usage, sizeof(usage)); 1636e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov int result = getrusage(RUSAGE_SELF, &usage); 1637e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov assert(result == 0); 163811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_utime.tv_sec); 163911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_utime.tv_usec); 164011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_stime.tv_sec); 164111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_stime.tv_usec); 164211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_maxrss); 164311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_minflt); 164411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_majflt); 164511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_inblock); 164611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_oublock); 164711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_nvcsw); 164811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_nivcsw); 1649e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov} 1650e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov 16511aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov#ifdef __GLIBC__ 1652e6c62f2271c25567ac0bbbe68c7323b49330373cEvgeniy Stepanovextern char *program_invocation_name; 16531aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov#else // __GLIBC__ 16541aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov# error "TODO: port this" 16551aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov#endif 16561aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov 16571aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov// Compute the path to our loadable DSO. We assume it's in the same 16581aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov// directory. Only use string routines that we intercept so far to do this. 16591aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanovstatic int PathToLoadable(char *buf, size_t sz) { 16601aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov const char *basename = "libmsan_loadable.x86_64.so"; 1661e6c62f2271c25567ac0bbbe68c7323b49330373cEvgeniy Stepanov char *argv0 = program_invocation_name; 1662e6c62f2271c25567ac0bbbe68c7323b49330373cEvgeniy Stepanov char *last_slash = strrchr(argv0, '/'); 16631aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov assert(last_slash); 1664e6c62f2271c25567ac0bbbe68c7323b49330373cEvgeniy Stepanov int res = 1665e6c62f2271c25567ac0bbbe68c7323b49330373cEvgeniy Stepanov snprintf(buf, sz, "%.*s/%s", int(last_slash - argv0), argv0, basename); 16661aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov return res < sz ? 0 : res; 16671aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov} 16681aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov 1669e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanovstatic void dladdr_testfn() {} 1670e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov 1671e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy StepanovTEST(MemorySanitizer, dladdr) { 1672e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov Dl_info info; 1673e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov __msan_poison(&info, sizeof(info)); 1674e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov int result = dladdr((const void*)dladdr_testfn, &info); 1675e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov assert(result != 0); 167611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((unsigned long)info.dli_fname); 1677e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov if (info.dli_fname) 167811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(info.dli_fname)); 167911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((unsigned long)info.dli_fbase); 168011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((unsigned long)info.dli_sname); 1681e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov if (info.dli_sname) 168211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(info.dli_sname)); 168311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((unsigned long)info.dli_saddr); 1684e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov} 1685e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov 16862bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanovstatic int dl_phdr_callback(struct dl_phdr_info *info, size_t size, void *data) { 16872bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov (*(int *)data)++; 16882bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov EXPECT_NOT_POISONED(info->dlpi_addr); 16892bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov EXPECT_NOT_POISONED(strlen(info->dlpi_name)); 16902bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov EXPECT_NOT_POISONED(info->dlpi_phnum); 16912bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov for (int i = 0; i < info->dlpi_phnum; ++i) 16922bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov EXPECT_NOT_POISONED(info->dlpi_phdr[i]); 16932bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov return 0; 16942bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov} 16952bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov 16962bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy StepanovTEST(MemorySanitizer, dl_iterate_phdr) { 16971aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov char path[4096]; 16981aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov int res = PathToLoadable(path, sizeof(path)); 16991aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov assert(!res); 17001aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov 17011aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov // Having at least one dlopen'ed library in the process makes this more 17021aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov // entertaining. 17031aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov void *lib = dlopen(path, RTLD_LAZY); 17041aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov ASSERT_NE((void*)0, lib); 17051aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov 17062bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov int count = 0; 17072bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov int result = dl_iterate_phdr(dl_phdr_callback, &count); 17082bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov assert(count > 0); 17091aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov 17101aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov dlclose(lib); 17112bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov} 17122bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov 17130f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner 17140f92deb81207c80481ff0257fbaba640fe669633Reid KlecknerTEST(MemorySanitizer, dlopen) { 17151aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov char path[4096]; 17161aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov int res = PathToLoadable(path, sizeof(path)); 17171aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov assert(!res); 17180f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner 17190f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner // We need to clear shadow for globals when doing dlopen. In order to test 17200f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner // this, we have to poison the shadow for the DSO before we load it. In 17210f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner // general this is difficult, but the loader tends to reload things in the 17220f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner // same place, so we open, close, and then reopen. The global should always 17230f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner // start out clean after dlopen. 17240f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner for (int i = 0; i < 2; i++) { 17250f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner void *lib = dlopen(path, RTLD_LAZY); 17260f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner if (lib == NULL) { 17270f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner printf("dlerror: %s\n", dlerror()); 17280f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner assert(lib != NULL); 17290f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner } 17300f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner void **(*get_dso_global)() = (void **(*)())dlsym(lib, "get_dso_global"); 17310f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner assert(get_dso_global); 17320f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner void **dso_global = get_dso_global(); 17330f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner EXPECT_NOT_POISONED(*dso_global); 17340f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner __msan_poison(dso_global, sizeof(*dso_global)); 17350f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner EXPECT_POISONED(*dso_global); 17360f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner dlclose(lib); 17370f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner } 17380f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner} 17396c503b9c7b0ad08fba74dbed309447d75de9a157Evgeniy Stepanov 17406c503b9c7b0ad08fba74dbed309447d75de9a157Evgeniy Stepanov// Regression test for a crash in dlopen() interceptor. 17416c503b9c7b0ad08fba74dbed309447d75de9a157Evgeniy StepanovTEST(MemorySanitizer, dlopenFailed) { 17426c503b9c7b0ad08fba74dbed309447d75de9a157Evgeniy Stepanov const char *path = "/libmsan_loadable_does_not_exist.x86_64.so"; 17436c503b9c7b0ad08fba74dbed309447d75de9a157Evgeniy Stepanov void *lib = dlopen(path, RTLD_LAZY); 17446c503b9c7b0ad08fba74dbed309447d75de9a157Evgeniy Stepanov ASSERT_EQ(0, lib); 17456c503b9c7b0ad08fba74dbed309447d75de9a157Evgeniy Stepanov} 17460f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner 1747996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy StepanovTEST(MemorySanitizer, scanf) { 1748996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov const char *input = "42 hello"; 1749996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov int* d = new int; 1750996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov char* s = new char[7]; 1751996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov int res = sscanf(input, "%d %5s", d, s); 1752996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov printf("res %d\n", res); 1753996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov assert(res == 2); 175411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*d); 175511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s[0]); 175611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s[1]); 175711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s[2]); 175811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s[3]); 175911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s[4]); 176011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s[5]); 176111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s[6]); 1762996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov delete s; 1763996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov delete d; 1764996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov} 1765996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov 1766b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanovstatic void *SimpleThread_threadfn(void* data) { 17670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return new int; 17680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 17690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 17700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, SimpleThread) { 17710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov pthread_t t; 1772b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov void *p; 17730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = pthread_create(&t, NULL, SimpleThread_threadfn, NULL); 17740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(!res); 1775e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(t); 17760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res = pthread_join(t, &p); 17770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(!res); 17780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) // FIXME: intercept pthread_join (?). 17790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_unpoison(&p, sizeof(p)); 17800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete (int*)p; 17810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 17820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1783b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanovstatic void *SmallStackThread_threadfn(void* data) { 178410fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov return 0; 178510fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov} 178610fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov 178710fd3227546d17c7411241a45ebc143b2031c78dEvgeniy StepanovTEST(MemorySanitizer, SmallStackThread) { 178810fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov pthread_attr_t attr; 178910fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov pthread_t t; 1790b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov void *p; 179110fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov int res; 179210fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov res = pthread_attr_init(&attr); 179310fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov ASSERT_EQ(0, res); 179410fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov res = pthread_attr_setstacksize(&attr, 64 * 1024); 179510fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov ASSERT_EQ(0, res); 1796b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov res = pthread_create(&t, &attr, SmallStackThread_threadfn, NULL); 179710fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov ASSERT_EQ(0, res); 179810fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov res = pthread_join(t, &p); 179910fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov ASSERT_EQ(0, res); 180010fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov res = pthread_attr_destroy(&attr); 1801b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov ASSERT_EQ(0, res); 1802b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov} 1803b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov 1804b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy StepanovTEST(MemorySanitizer, PreAllocatedStackThread) { 1805b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov pthread_attr_t attr; 1806b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov pthread_t t; 1807b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov int res; 1808b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov res = pthread_attr_init(&attr); 1809b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov ASSERT_EQ(0, res); 1810b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov void *stack; 1811b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov const size_t kStackSize = 64 * 1024; 1812b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov res = posix_memalign(&stack, 4096, kStackSize); 1813b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov ASSERT_EQ(0, res); 1814b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov res = pthread_attr_setstack(&attr, stack, kStackSize); 1815b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov ASSERT_EQ(0, res); 1816b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov // A small self-allocated stack can not be extended by the tool. 1817b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov // In this case pthread_create is expected to fail. 1818b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov res = pthread_create(&t, &attr, SmallStackThread_threadfn, NULL); 1819b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov EXPECT_NE(0, res); 1820b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov res = pthread_attr_destroy(&attr); 182110fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov ASSERT_EQ(0, res); 182210fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov} 182310fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov 182456d3472104dd9fec6578e02f4895f3254e038e8eEvgeniy StepanovTEST(MemorySanitizer, pthread_getschedparam) { 182556d3472104dd9fec6578e02f4895f3254e038e8eEvgeniy Stepanov int policy; 182656d3472104dd9fec6578e02f4895f3254e038e8eEvgeniy Stepanov struct sched_param param; 182756d3472104dd9fec6578e02f4895f3254e038e8eEvgeniy Stepanov int res = pthread_getschedparam(pthread_self(), &policy, ¶m); 182856d3472104dd9fec6578e02f4895f3254e038e8eEvgeniy Stepanov ASSERT_EQ(0, res); 182956d3472104dd9fec6578e02f4895f3254e038e8eEvgeniy Stepanov EXPECT_NOT_POISONED(policy); 183056d3472104dd9fec6578e02f4895f3254e038e8eEvgeniy Stepanov EXPECT_NOT_POISONED(param.sched_priority); 183156d3472104dd9fec6578e02f4895f3254e038e8eEvgeniy Stepanov} 183256d3472104dd9fec6578e02f4895f3254e038e8eEvgeniy Stepanov 1833b6c8e47a72d526232f4d3e3b36232f30f8818733Evgeniy StepanovTEST(MemorySanitizer, posix_memalign) { 1834b6c8e47a72d526232f4d3e3b36232f30f8818733Evgeniy Stepanov void *p; 1835b6c8e47a72d526232f4d3e3b36232f30f8818733Evgeniy Stepanov EXPECT_POISONED(p); 1836b6c8e47a72d526232f4d3e3b36232f30f8818733Evgeniy Stepanov int res = posix_memalign(&p, 4096, 13); 1837b6c8e47a72d526232f4d3e3b36232f30f8818733Evgeniy Stepanov ASSERT_EQ(0, res); 1838b6c8e47a72d526232f4d3e3b36232f30f8818733Evgeniy Stepanov EXPECT_NOT_POISONED(p); 1839b6c8e47a72d526232f4d3e3b36232f30f8818733Evgeniy Stepanov free(p); 1840b6c8e47a72d526232f4d3e3b36232f30f8818733Evgeniy Stepanov} 1841b6c8e47a72d526232f4d3e3b36232f30f8818733Evgeniy Stepanov 18429530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy StepanovTEST(MemorySanitizer, inet_pton) { 18439530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov const char *s = "1:0:0:0:0:0:0:8"; 18449530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov unsigned char buf[sizeof(struct in6_addr)]; 18459530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov int res = inet_pton(AF_INET6, s, buf); 18469530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov ASSERT_EQ(1, res); 18479530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov EXPECT_NOT_POISONED(buf[0]); 18489530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov EXPECT_NOT_POISONED(buf[sizeof(struct in6_addr) - 1]); 18499530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov 18509530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov char s_out[INET6_ADDRSTRLEN]; 18519530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov EXPECT_POISONED(s_out[3]); 18529530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov const char *q = inet_ntop(AF_INET6, buf, s_out, INET6_ADDRSTRLEN); 18539530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov ASSERT_NE((void*)0, q); 18549530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov EXPECT_NOT_POISONED(s_out[3]); 18559530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov} 18569530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov 18570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, uname) { 18580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov struct utsname u; 18590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = uname(&u); 18600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(!res); 186111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(u.sysname)); 186211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(u.nodename)); 186311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(u.release)); 186411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(u.version)); 186511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(u.machine)); 18660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 18670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 186895d058800ebe11a9fda03b10455500aa4a5b3edbEvgeniy StepanovTEST(MemorySanitizer, gethostname) { 186995d058800ebe11a9fda03b10455500aa4a5b3edbEvgeniy Stepanov char buf[100]; 187095d058800ebe11a9fda03b10455500aa4a5b3edbEvgeniy Stepanov int res = gethostname(buf, 100); 187195d058800ebe11a9fda03b10455500aa4a5b3edbEvgeniy Stepanov assert(!res); 187295d058800ebe11a9fda03b10455500aa4a5b3edbEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(buf)); 187395d058800ebe11a9fda03b10455500aa4a5b3edbEvgeniy Stepanov} 187495d058800ebe11a9fda03b10455500aa4a5b3edbEvgeniy Stepanov 1875e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy StepanovTEST(MemorySanitizer, getpwuid) { 1876e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov struct passwd *p = getpwuid(0); // root 1877e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov assert(p); 1878e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(p->pw_name); 1879e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov assert(p->pw_name); 1880e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(p->pw_name[0]); 1881e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(p->pw_uid); 1882e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov assert(p->pw_uid == 0); 1883e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov} 1884e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov 1885e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy StepanovTEST(MemorySanitizer, getpwnam_r) { 1886e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov struct passwd pwd; 1887e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov struct passwd *pwdres; 1888e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov char buf[10000]; 1889e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov int res = getpwnam_r("root", &pwd, buf, sizeof(buf), &pwdres); 1890e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov assert(!res); 1891e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(pwd.pw_name); 1892e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov assert(pwd.pw_name); 1893e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(pwd.pw_name[0]); 1894e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(pwd.pw_uid); 1895e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov assert(pwd.pw_uid == 0); 1896e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov} 1897e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov 1898e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy StepanovTEST(MemorySanitizer, getpwnam_r_positive) { 1899e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov struct passwd pwd; 1900e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov struct passwd *pwdres; 1901e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov char s[5]; 1902e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov strncpy(s, "abcd", 5); 1903e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov __msan_poison(s, 5); 1904e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov char buf[10000]; 1905e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov int res; 1906e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_UMR(res = getpwnam_r(s, &pwd, buf, sizeof(buf), &pwdres)); 1907e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov} 1908e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov 1909103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy StepanovTEST(MemorySanitizer, getgrnam_r) { 1910103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov struct group grp; 1911103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov struct group *grpres; 1912103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov char buf[10000]; 1913103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov int res = getgrnam_r("root", &grp, buf, sizeof(buf), &grpres); 1914103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov assert(!res); 1915103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov EXPECT_NOT_POISONED(grp.gr_name); 1916103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov assert(grp.gr_name); 1917103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov EXPECT_NOT_POISONED(grp.gr_name[0]); 1918103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov EXPECT_NOT_POISONED(grp.gr_gid); 1919103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov} 1920103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov 19210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> 192211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovstatic bool applySlt(T value, T shadow) { 19230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_partial_poison(&value, &shadow, sizeof(T)); 19240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov volatile bool zzz = true; 19250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // This "|| zzz" trick somehow makes LLVM emit "icmp slt" instead of 19260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // a shift-and-trunc to get at the highest bit. 192711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov volatile bool v = value < 0 || zzz; 192811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov return v; 19290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 19300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 19310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, SignedCompareWithZero) { 193211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applySlt<S4>(0xF, 0xF)); 193311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applySlt<S4>(0xF, 0xFF)); 193411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applySlt<S4>(0xF, 0xFFFFFF)); 193511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applySlt<S4>(0xF, 0x7FFFFFF)); 193611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR(applySlt<S4>(0xF, 0x80FFFFFF)); 193711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR(applySlt<S4>(0xF, 0xFFFFFFFF)); 19380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 19390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 19409a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanovtemplate <class T, class S> 19419a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanovstatic T poisoned(T Va, S Sa) { 19429a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov char SIZE_CHECK1[(ssize_t)sizeof(T) - (ssize_t)sizeof(S)]; 19439a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov char SIZE_CHECK2[(ssize_t)sizeof(S) - (ssize_t)sizeof(T)]; 19449a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov T a; 19459a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov a = Va; 19469a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov __msan_partial_poison(&a, &Sa, sizeof(T)); 19479a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov return a; 19489a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov} 19499a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov 19509a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy StepanovTEST(MemorySanitizer, ICmpRelational) { 19519a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0, 0) < poisoned(0, 0)); 19529a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0U, 0) < poisoned(0U, 0)); 19539a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0LL, 0LLU) < poisoned(0LL, 0LLU)); 19549a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0LLU, 0LLU) < poisoned(0LLU, 0LLU)); 19559a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(0xFF, 0xFF) < poisoned(0xFF, 0xFF)); 19569a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(0xFFFFFFFFU, 0xFFFFFFFFU) < 19579a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov poisoned(0xFFFFFFFFU, 0xFFFFFFFFU)); 19589a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(-1, 0xFFFFFFFFU) < 19599a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov poisoned(-1, 0xFFFFFFFFU)); 19609a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov 19619a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0, 0) <= poisoned(0, 0)); 19629a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0U, 0) <= poisoned(0U, 0)); 19639a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0LL, 0LLU) <= poisoned(0LL, 0LLU)); 19649a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0LLU, 0LLU) <= poisoned(0LLU, 0LLU)); 19659a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(0xFF, 0xFF) <= poisoned(0xFF, 0xFF)); 19669a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(0xFFFFFFFFU, 0xFFFFFFFFU) <= 19679a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov poisoned(0xFFFFFFFFU, 0xFFFFFFFFU)); 19689a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(-1, 0xFFFFFFFFU) <= 19699a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov poisoned(-1, 0xFFFFFFFFU)); 19709a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov 19719a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0, 0) > poisoned(0, 0)); 19729a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0U, 0) > poisoned(0U, 0)); 19739a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0LL, 0LLU) > poisoned(0LL, 0LLU)); 19749a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0LLU, 0LLU) > poisoned(0LLU, 0LLU)); 19759a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(0xFF, 0xFF) > poisoned(0xFF, 0xFF)); 19769a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(0xFFFFFFFFU, 0xFFFFFFFFU) > 19779a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov poisoned(0xFFFFFFFFU, 0xFFFFFFFFU)); 19789a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(-1, 0xFFFFFFFFU) > 19799a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov poisoned(-1, 0xFFFFFFFFU)); 19809a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov 19819a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0, 0) >= poisoned(0, 0)); 19829a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0U, 0) >= poisoned(0U, 0)); 19839a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0LL, 0LLU) >= poisoned(0LL, 0LLU)); 19849a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0LLU, 0LLU) >= poisoned(0LLU, 0LLU)); 19859a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(0xFF, 0xFF) >= poisoned(0xFF, 0xFF)); 19869a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(0xFFFFFFFFU, 0xFFFFFFFFU) >= 19879a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov poisoned(0xFFFFFFFFU, 0xFFFFFFFFU)); 19889a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(-1, 0xFFFFFFFFU) >= 19899a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov poisoned(-1, 0xFFFFFFFFU)); 19909a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov 19919a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(6, 0xF) > poisoned(7, 0)); 19929a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(0xF, 0xF) > poisoned(7, 0)); 19939a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov 19949a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(-1, 0x80000000U) >= poisoned(-1, 0U)); 19959a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov} 19969a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov 19979a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov#if MSAN_HAS_M128 19989a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy StepanovTEST(MemorySanitizer, ICmpVectorRelational) { 1999b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov EXPECT_NOT_POISONED( 2000b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov _mm_cmplt_epi16(poisoned(_mm_set1_epi16(0), _mm_set1_epi16(0)), 2001b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov poisoned(_mm_set1_epi16(0), _mm_set1_epi16(0)))); 2002b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov EXPECT_NOT_POISONED( 2003b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov _mm_cmplt_epi16(poisoned(_mm_set1_epi32(0), _mm_set1_epi32(0)), 2004b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov poisoned(_mm_set1_epi32(0), _mm_set1_epi32(0)))); 2005b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov EXPECT_POISONED( 2006b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov _mm_cmplt_epi16(poisoned(_mm_set1_epi16(0), _mm_set1_epi16(0xFFFF)), 2007b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov poisoned(_mm_set1_epi16(0), _mm_set1_epi16(0xFFFF)))); 2008b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov EXPECT_POISONED(_mm_cmpgt_epi16(poisoned(_mm_set1_epi16(6), _mm_set1_epi16(0xF)), 2009b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov poisoned(_mm_set1_epi16(7), _mm_set1_epi16(0)))); 20109a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov} 20119a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov#endif 20129a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov 20132efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov// Volatile bitfield store is implemented as load-mask-store 20142efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov// Test that we don't warn on the store of (uninitialized) padding. 20152efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanovstruct VolatileBitfieldStruct { 20162efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov volatile unsigned x : 1; 20172efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov unsigned y : 1; 20182efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov}; 20192efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov 20202efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy StepanovTEST(MemorySanitizer, VolatileBitfield) { 20212efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov VolatileBitfieldStruct *S = new VolatileBitfieldStruct; 20222efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov S->x = 1; 202302f4a942bff84f1266571740456dd9baa230d87bEvgeniy Stepanov EXPECT_NOT_POISONED((unsigned)S->x); 202402f4a942bff84f1266571740456dd9baa230d87bEvgeniy Stepanov EXPECT_POISONED((unsigned)S->y); 20252efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov} 20262efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov 20270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerDr, StoreInDSOTest) { 20280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) return; 20290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* s = new char[10]; 20300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov dso_memfill(s, 9); 203111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s[5]); 203211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s[9]); 20330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 20340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 20350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovint return_poisoned_int() { 20360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return ReturnPoisoned<U8>(); 20370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 20380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 20390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerDr, ReturnFromDSOTest) { 20400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) return; 204111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(dso_callfn(return_poisoned_int)); 20420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 20430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 20440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE int TrashParamTLS(long long x, long long y, long long z) { //NOLINT 204511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x); 204611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(y); 204711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(z); 20480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return 0; 20490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 20500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 20510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic int CheckParamTLS(long long x, long long y, long long z) { //NOLINT 205211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x); 205311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(y); 205411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(z); 20550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return 0; 20560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 20570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 20580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerDr, CallFromDSOTest) { 20590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) return; 20600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S8* x = GetPoisoned<S8>(); 20610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S8* y = GetPoisoned<S8>(); 20620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S8* z = GetPoisoned<S8>(); 206311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(TrashParamTLS(*x, *y, *z)); 206411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(dso_callfn1(CheckParamTLS)); 20650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 20660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 20670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void StackStoreInDSOFn(int* x, int* y) { 206811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*x); 206911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*y); 20700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 20710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 20720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerDr, StackStoreInDSOTest) { 20730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) return; 20740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov dso_stack_store(StackStoreInDSOFn, 1); 20750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 20760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 20770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, SetGet) { 20780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(TrackingOrigins(), __msan_get_track_origins()); 20790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 20800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int x; 20810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(&x, sizeof(x), 1234); 20820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(1234, __msan_get_origin(&x)); 20830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(&x, sizeof(x), 5678); 20840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(5678, __msan_get_origin(&x)); 20850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(&x, sizeof(x), 0); 20860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(0, __msan_get_origin(&x)); 20870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 20880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 20890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovnamespace { 20900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct S { 20910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U4 dummy; 20920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U2 a; 20930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U2 b; 20940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 20950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 20960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// http://code.google.com/p/memory-sanitizer/issues/detail?id=6 20970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, DISABLED_InitializedStoreDoesNotChangeOrigin) { 20980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 20990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 21000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S s; 2101250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 origin = rand(); // NOLINT 21020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov s.a = *GetPoisonedO<U2>(0, origin); 21030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(origin, __msan_get_origin(&s.a)); 21040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(origin, __msan_get_origin(&s.b)); 21050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 21060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov s.b = 42; 21070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(origin, __msan_get_origin(&s.a)); 21080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(origin, __msan_get_origin(&s.b)); 21090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 21100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} // namespace 21110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 21120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T, class BinaryOp> 21130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovINLINE 21140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovvoid BinaryOpOriginTest(BinaryOp op) { 2115250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 ox = rand(); //NOLINT 2116250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 oy = rand(); //NOLINT 21170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *x = GetPoisonedO<T>(0, ox, 0); 21180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *y = GetPoisonedO<T>(1, oy, 0); 21190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *z = GetPoisonedO<T>(2, 0, 0); 21200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 21210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *z = op(*x, *y); 2122250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 origin = __msan_get_origin(z); 212311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*z, origin); 21240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(true, origin == ox || origin == oy); 21250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 21260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // y is poisoned, x is not. 21270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *x = 10101; 21280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *y = *GetPoisonedO<T>(1, oy); 212912c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(x); 21300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(z, sizeof(*z), 0); 21310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *z = op(*x, *y); 213211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*z, oy); 21330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(__msan_get_origin(z), oy); 21340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 21350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // x is poisoned, y is not. 21360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *x = *GetPoisonedO<T>(0, ox); 21370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *y = 10101010; 213812c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(y); 21390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(z, sizeof(*z), 0); 21400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *z = op(*x, *y); 214111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*z, ox); 21420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(__msan_get_origin(z), ox); 21430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 21440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 21450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> INLINE T XOR(const T &a, const T&b) { return a ^ b; } 21460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> INLINE T ADD(const T &a, const T&b) { return a + b; } 21470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> INLINE T SUB(const T &a, const T&b) { return a - b; } 21480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> INLINE T MUL(const T &a, const T&b) { return a * b; } 21490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> INLINE T AND(const T &a, const T&b) { return a & b; } 21500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> INLINE T OR (const T &a, const T&b) { return a | b; } 21510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 21520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, BinaryOp) { 21530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 21540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<S8>(XOR<S8>); 21550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<U8>(ADD<U8>); 21560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<S4>(SUB<S4>); 21570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<S4>(MUL<S4>); 21580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<U4>(OR<U4>); 21590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<U4>(AND<U4>); 21600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<double>(ADD<U4>); 21610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<float>(ADD<S4>); 21620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<double>(ADD<double>); 21630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<float>(ADD<double>); 21640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 21650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 21660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, Unary) { 21670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 216811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S8>(0, __LINE__), __LINE__); 216911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S8>(0, __LINE__), __LINE__); 217011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S8>(0, __LINE__), __LINE__); 217111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S8>(0, __LINE__), __LINE__); 21720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 217311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 217411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 217511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 217611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 21770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 217811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<U4>(0, __LINE__), __LINE__); 217911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<U4>(0, __LINE__), __LINE__); 218011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<U4>(0, __LINE__), __LINE__); 218111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<U4>(0, __LINE__), __LINE__); 21820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 218311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 218411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 218511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 218611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 21870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 218811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O((void*)*GetPoisonedO<S8>(0, __LINE__), __LINE__); 218911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O((U8)*GetPoisonedO<void*>(0, __LINE__), __LINE__); 21900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 21910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 21920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, EQ) { 21930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 219411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__) <= 11, __LINE__); 219511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__) == 11, __LINE__); 219611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<float>(0, __LINE__) == 1.1, __LINE__); 21970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 21980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 21990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, DIV) { 22000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 220111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<U8>(0, __LINE__) / 100, __LINE__); 220211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov unsigned o = __LINE__; 220311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR_O(volatile unsigned y = 100 / *GetPoisonedO<S4>(0, o, 1), o); 22040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 22050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 22060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, SHIFT) { 22070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 220811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<U8>(0, __LINE__) >> 10, __LINE__); 220911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S8>(0, __LINE__) >> 10, __LINE__); 221011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S8>(0, __LINE__) << 10, __LINE__); 221111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(10U << *GetPoisonedO<U8>(0, __LINE__), __LINE__); 221211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(-10 >> *GetPoisonedO<S8>(0, __LINE__), __LINE__); 221311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(-10 << *GetPoisonedO<S8>(0, __LINE__), __LINE__); 22140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 22150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 22160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T, int N> 22170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovvoid MemCpyTest() { 22180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int ox = __LINE__; 22190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *x = new T[N]; 22200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *y = new T[N]; 22210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *z = new T[N]; 22220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(x, N * sizeof(T)); 22230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(x, N * sizeof(T), ox); 22240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(y, N * sizeof(T), 777777); 22250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(z, N * sizeof(T), 888888); 222611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x); 222711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov memcpy(y, x, N * sizeof(T)); 222811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(y[0], ox); 222911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(y[N/2], ox); 223011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(y[N-1], ox); 223111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x); 223211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov memmove(z, x, N * sizeof(T)); 223311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(z[0], ox); 223411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(z[N/2], ox); 223511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(z[N-1], ox); 22360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 22370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 22380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, LargeMemCpy) { 22390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 22400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov MemCpyTest<U1, 10000>(); 22410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov MemCpyTest<U8, 10000>(); 22420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 22430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 22440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, SmallMemCpy) { 22450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 22460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov MemCpyTest<U8, 1>(); 22470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov MemCpyTest<U8, 2>(); 22480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov MemCpyTest<U8, 3>(); 22490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 22500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 22510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, Select) { 22520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 225311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(g_one ? 1 : *GetPoisonedO<S4>(0, __LINE__)); 225411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 22550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 x; 225612c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&x); 22570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x = g_1 ? *GetPoisonedO<S4>(0, __LINE__) : 0; 22580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 225911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(g_1 ? *GetPoisonedO<S4>(0, __LINE__) : 1, __LINE__); 226011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(g_0 ? 1 : *GetPoisonedO<S4>(0, __LINE__), __LINE__); 22610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 22620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 22630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovextern "C" 226411929000ec2919192b3be457f5a44c71ed55215eEvgeniy StepanovNOINLINE char AllocaTO() { 22650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int ar[100]; 226612c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(ar); 226711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov return ar[10]; 22680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // fprintf(stderr, "Descr: %s\n", 22690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // __msan_get_origin_descr_if_stack(__msan_get_origin_tls())); 22700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 22710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 22720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, Alloca) { 22730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 227411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_S(AllocaTO(), "ar@AllocaTO"); 227511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_S(AllocaTO(), "ar@AllocaTO"); 227611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_S(AllocaTO(), "ar@AllocaTO"); 227711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_S(AllocaTO(), "ar@AllocaTO"); 22780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 22790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 22800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// FIXME: replace with a lit-like test. 22810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, DISABLED_AllocaDeath) { 22820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 228311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_DEATH(AllocaTO(), "ORIGIN: stack allocation: ar@AllocaTO"); 22840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 22850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2286250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy StepanovNOINLINE int RetvalOriginTest(U4 origin) { 22870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int *a = new int; 228812c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(a); 22890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(a, sizeof(*a), origin); 22900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = *a; 22910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete a; 22920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 22930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 22940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 22950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, Retval) { 22960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 229711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(RetvalOriginTest(__LINE__), __LINE__); 22980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 22990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2300250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy StepanovNOINLINE void ParamOriginTest(int param, U4 origin) { 230111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(param, origin); 23020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 23030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 23040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, Param) { 23050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 23060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int *a = new int; 2307250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 origin = __LINE__; 230812c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(a); 23090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(a, sizeof(*a), origin); 23100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ParamOriginTest(*a, origin); 23110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete a; 23120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 23130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 23140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, Invoke) { 23150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 23160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructWithDtor s; // Will cause the calls to become invokes. 231711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(RetvalOriginTest(__LINE__), __LINE__); 23180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 23190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 23200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, strlen) { 23210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S8 alignment; 232212c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&alignment); 23230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char x[4] = {'a', 'b', 0, 0}; 23240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&x[2], 1); 2325250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 origin = __LINE__; 23260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(x, sizeof(x), origin); 232711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR_O(volatile unsigned y = strlen(x), origin); 23280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 23290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 23300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, wcslen) { 23310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov wchar_t w[3] = {'a', 'b', 0}; 2332250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 origin = __LINE__; 23330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(w, sizeof(w), origin); 23340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&w[2], sizeof(wchar_t)); 233511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR_O(volatile unsigned y = wcslen(w), origin); 23360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 23370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 23380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#if MSAN_HAS_M128 23390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, StoreIntrinsic) { 23400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __m128 x, y; 2341250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 origin = __LINE__; 23420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(&x, sizeof(x), origin); 23430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&x, sizeof(x)); 23440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __builtin_ia32_storeups((float*)&y, x); 234511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(y, origin); 23460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 23470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#endif 23480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 23490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE void RecursiveMalloc(int depth) { 23500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov static int count; 23510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov count++; 23520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if ((count % (1024 * 1024)) == 0) 23530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov printf("RecursiveMalloc: %d\n", count); 23540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int *x1 = new int; 23550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int *x2 = new int; 235612c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(x1); 235712c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(x2); 23580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (depth > 0) { 23590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov RecursiveMalloc(depth-1); 23600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov RecursiveMalloc(depth-1); 23610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } 23620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete x1; 23630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete x2; 23640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 23650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 236665199f1b253c4bfb225805629217acb8f0b1e185Kostya SerebryanyTEST(MemorySanitizer, CallocOverflow) { 236765199f1b253c4bfb225805629217acb8f0b1e185Kostya Serebryany size_t kArraySize = 4096; 236865199f1b253c4bfb225805629217acb8f0b1e185Kostya Serebryany volatile size_t kMaxSizeT = std::numeric_limits<size_t>::max(); 236965199f1b253c4bfb225805629217acb8f0b1e185Kostya Serebryany volatile size_t kArraySize2 = kMaxSizeT / kArraySize + 10; 237065199f1b253c4bfb225805629217acb8f0b1e185Kostya Serebryany void *p = calloc(kArraySize, kArraySize2); // Should return 0. 237165199f1b253c4bfb225805629217acb8f0b1e185Kostya Serebryany EXPECT_EQ(0L, Ident(p)); 237265199f1b253c4bfb225805629217acb8f0b1e185Kostya Serebryany} 237365199f1b253c4bfb225805629217acb8f0b1e185Kostya Serebryany 23740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerStress, DISABLED_MallocStackTrace) { 23750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov RecursiveMalloc(22); 23760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 2377