msan_test.cc revision 263800ba59720f9be69735b4dad625fca92430bc
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> 41359d7fc7daf099ab8ef9a860564542d2581544aaEvgeniy Stepanov#include <sys/sysinfo.h> 420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <sys/utsname.h> 430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <sys/mman.h> 440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <sys/vfs.h> 45d97a15a931ae6696a1071e1471c9a019e821904bEvgeniy Stepanov#include <dirent.h> 46e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov#include <pwd.h> 47134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov#include <sys/socket.h> 480a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov#include <netdb.h> 490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#if defined(__i386__) || defined(__x86_64__) 510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov# include <emmintrin.h> 520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov# define MSAN_HAS_M128 1 530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#else 540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov# define MSAN_HAS_M128 0 550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#endif 560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtypedef unsigned char U1; 580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtypedef unsigned short U2; // NOLINT 590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtypedef unsigned int U4; 600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtypedef unsigned long long U8; // NOLINT 610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtypedef signed char S1; 620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtypedef signed short S2; // NOLINT 630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtypedef signed int S4; 640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtypedef signed long long S8; // NOLINT 650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#define NOINLINE __attribute__((noinline)) 660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#define INLINE __attribute__((always_inline)) 670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 6811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovstatic bool TrackingOrigins() { 6911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov S8 x; 7011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov __msan_set_origin(&x, sizeof(x), 0x1234); 71250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 origin = __msan_get_origin(&x); 7211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov __msan_set_origin(&x, sizeof(x), 0); 7311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov return origin == 0x1234; 7411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov} 750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 7611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov#define EXPECT_UMR(action) \ 770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov do { \ 780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_expect_umr(1); \ 790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov action; \ 800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_expect_umr(0); \ 810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } while (0) 820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 8311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov#define EXPECT_UMR_O(action, origin) \ 840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov do { \ 850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_expect_umr(1); \ 860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov action; \ 870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_expect_umr(0); \ 880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (TrackingOrigins()) \ 8912c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov EXPECT_EQ(origin, __msan_get_umr_origin()); \ 900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } while (0) 910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov#define EXPECT_UMR_S(action, stack_origin) \ 930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov do { \ 940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_expect_umr(1); \ 950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov action; \ 960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_expect_umr(0); \ 97250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 id = __msan_get_umr_origin(); \ 980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov const char *str = __msan_get_origin_descr_if_stack(id); \ 990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!str || strcmp(str, stack_origin)) { \ 1000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov fprintf(stderr, "EXPECT_POISONED_S: id=%u %s, %s", \ 1010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov id, stack_origin, str); \ 1020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(1, 0); \ 1030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } \ 1040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } while (0) 1050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 10611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov#define EXPECT_POISONED(x) ExpectPoisoned(x) 10711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 10811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovtemplate<typename T> 10911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovvoid ExpectPoisoned(const T& t) { 11011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NE(-1, __msan_test_shadow((void*)&t, sizeof(t))); 11111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov} 11211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 11311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov#define EXPECT_POISONED_O(x, origin) \ 11411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov ExpectPoisonedWithOrigin(x, origin) 11511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 11611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovtemplate<typename T> 11711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovvoid ExpectPoisonedWithOrigin(const T& t, unsigned origin) { 11811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NE(-1, __msan_test_shadow((void*)&t, sizeof(t))); 11911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov if (TrackingOrigins()) 12011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_EQ(origin, __msan_get_origin((void*)&t)); 12111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov} 12211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 12311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov#define EXPECT_POISONED_S(x, stack_origin) \ 12411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov ExpectPoisonedWithStackOrigin(x, stack_origin) 12511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 12611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovtemplate<typename T> 12711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovvoid ExpectPoisonedWithStackOrigin(const T& t, const char *stack_origin) { 12811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NE(-1, __msan_test_shadow((void*)&t, sizeof(t))); 129250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 id = __msan_get_origin((void*)&t); 13011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov const char *str = __msan_get_origin_descr_if_stack(id); 13111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov if (!str || strcmp(str, stack_origin)) { 13211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov fprintf(stderr, "EXPECT_POISONED_S: id=%u %s, %s", 13311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov id, stack_origin, str); 13411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_EQ(1, 0); 13511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov } 13611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov} 13711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 13811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov#define EXPECT_NOT_POISONED(x) ExpectNotPoisoned(x) 13911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 14011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovtemplate<typename T> 14111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovvoid ExpectNotPoisoned(const T& t) { 14211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_EQ(-1, __msan_test_shadow((void*)&t, sizeof(t))); 14311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov} 1440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic U8 poisoned_array[100]; 1460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> 1470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovT *GetPoisoned(int i = 0, T val = 0) { 1480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *res = (T*)&poisoned_array[i]; 1490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *res = val; 1500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&poisoned_array[i], sizeof(T)); 1510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 1520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 1530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> 155250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy StepanovT *GetPoisonedO(int i, U4 origin, T val = 0) { 1560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *res = (T*)&poisoned_array[i]; 1570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *res = val; 1580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&poisoned_array[i], sizeof(T)); 1590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(&poisoned_array[i], sizeof(T), origin); 1600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 1610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 1620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// This function returns its parameter but in such a way that compiler 1640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// can not prove it. 1650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> 1660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE 1670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic T Ident(T t) { 1680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov volatile T ret = t; 1690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return ret; 1700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 1710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> NOINLINE T ReturnPoisoned() { return *GetPoisoned<T>(); } 1730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic volatile int g_one = 1; 1750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic volatile int g_zero = 0; 1760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic volatile int g_0 = 0; 1770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic volatile int g_1 = 1; 1780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovS4 a_s4[100]; 1800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovS8 a_s8[100]; 1810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 18212c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov// Check that malloc poisons memory. 18312c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov// A lot of tests below depend on this. 18412c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy StepanovTEST(MemorySanitizerSanity, PoisonInMalloc) { 18512c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov int *x = (int*)malloc(sizeof(int)); 18612c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov EXPECT_POISONED(*x); 18712c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov free(x); 18812c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov} 18912c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov 1900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, NegativeTest1) { 1910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 *x = GetPoisoned<S4>(); 1920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (g_one) 1930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *x = 0; 19411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*x); 1950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 1960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, PositiveTest1) { 1980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // Load to store. 19911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S1>()); 20011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S2>()); 20111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S4>()); 20211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S8>()); 2030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // S->S conversions. 20511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S1>()); 20611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S1>()); 20711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S1>()); 2080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 20911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S2>()); 21011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S2>()); 21111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S2>()); 2120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 21311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S4>()); 21411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S4>()); 21511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S4>()); 2160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 21711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S8>()); 21811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S8>()); 21911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S8>()); 2200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // ZExt 22211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<U1>()); 22311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<U1>()); 22411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<U1>()); 22511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<U2>()); 22611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<U2>()); 22711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<U4>()); 2280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // Unary ops. 23011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(- *GetPoisoned<S4>()); 2310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 23211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR(a_s4[g_zero] = 100 / *GetPoisoned<S4>(0, 1)); 2330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov a_s4[g_zero] = 1 - *GetPoisoned<S4>(); 2360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov a_s4[g_zero] = 1 + *GetPoisoned<S4>(); 2370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 2380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, Phi1) { 2400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 c; 2410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (g_one) { 2420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov c = *GetPoisoned<S4>(); 2430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } else { 24412c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(0); 2450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov c = 0; 2460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } 24711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(c); 2480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 2490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, Phi2) { 2510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 i = *GetPoisoned<S4>(); 2520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 n = g_one; 25311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR(for (; i < g_one; i++);); 25411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(i); 2550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 2560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 25711929000ec2919192b3be457f5a44c71ed55215eEvgeniy StepanovNOINLINE void Arg1ExpectUMR(S4 a1) { EXPECT_POISONED(a1); } 25811929000ec2919192b3be457f5a44c71ed55215eEvgeniy StepanovNOINLINE void Arg2ExpectUMR(S4 a1, S4 a2) { EXPECT_POISONED(a2); } 25911929000ec2919192b3be457f5a44c71ed55215eEvgeniy StepanovNOINLINE void Arg3ExpectUMR(S1 a1, S4 a2, S8 a3) { EXPECT_POISONED(a3); } 2600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, ArgTest) { 2620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov Arg1ExpectUMR(*GetPoisoned<S4>()); 2630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov Arg2ExpectUMR(0, *GetPoisoned<S4>()); 2640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov Arg3ExpectUMR(0, 1, *GetPoisoned<S8>()); 2650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 2660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, CallAndRet) { 2690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) return; 2700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ReturnPoisoned<S1>(); 2710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ReturnPoisoned<S2>(); 2720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ReturnPoisoned<S4>(); 2730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ReturnPoisoned<S8>(); 2740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 27511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(ReturnPoisoned<S1>()); 27611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(ReturnPoisoned<S2>()); 27711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(ReturnPoisoned<S4>()); 27811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(ReturnPoisoned<S8>()); 2790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 2800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// malloc() in the following test may be optimized to produce a compile-time 2820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// undef value. Check that we trap on the volatile assignment anyway. 2830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, DISABLED_MallocNoIdent) { 2840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 *x = (int*)malloc(sizeof(S4)); 28511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*x); 2860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov free(x); 2870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 2880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, Malloc) { 2900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 *x = (int*)Ident(malloc(sizeof(S4))); 29111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*x); 2920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov free(x); 2930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 2940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, Realloc) { 2960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 *x = (int*)Ident(realloc(0, sizeof(S4))); 29711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[0]); 2980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[0] = 1; 2990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x = (int*)Ident(realloc(x, 2 * sizeof(S4))); 30011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[0]); // Ok, was inited before. 30111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[1]); 3020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x = (int*)Ident(realloc(x, 3 * sizeof(S4))); 30311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[0]); // Ok, was inited before. 30411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[2]); 30511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[1]); 3060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[2] = 1; // Init this here. Check that after realloc it is poisoned again. 3070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x = (int*)Ident(realloc(x, 2 * sizeof(S4))); 30811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[0]); // Ok, was inited before. 30911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[1]); 3100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x = (int*)Ident(realloc(x, 3 * sizeof(S4))); 31111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[1]); 31211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[2]); 3130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov free(x); 3140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 3150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 3160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, Calloc) { 3170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 *x = (int*)Ident(calloc(1, sizeof(S4))); 31811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*x); // Should not be poisoned. 3190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // EXPECT_EQ(0, *x); 3200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov free(x); 3210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 3220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 3230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, AndOr) { 3240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U4 *p = GetPoisoned<U4>(); 3250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // We poison two bytes in the midle of a 4-byte word to make the test 3260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // correct regardless of endianness. 3270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ((U1*)p)[1] = 0; 3280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ((U1*)p)[2] = 0xff; 32911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*p & 0x00ffff00); 33011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*p & 0x00ff0000); 33111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*p & 0x0000ff00); 33211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*p & 0xff000000); 33311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*p & 0x000000ff); 33411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*p & 0x0000ffff); 33511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*p & 0xffff0000); 3360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 33711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*p | 0xff0000ff); 33811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*p | 0xff00ffff); 33911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*p | 0xffff00ff); 34011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*p | 0xff000000); 34111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*p | 0x000000ff); 34211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*p | 0x0000ffff); 34311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*p | 0xffff0000); 3440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 34511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<bool>() & *GetPoisoned<bool>()); 3460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 3470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 3480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> 34911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovstatic bool applyNot(T value, T shadow) { 3500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_partial_poison(&value, &shadow, sizeof(T)); 35111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov return !value; 3520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 3530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 3540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, Not) { 35511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U4>(0x0, 0x0)); 35611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U4>(0xFFFFFFFF, 0x0)); 35711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(applyNot<U4>(0xFFFFFFFF, 0xFFFFFFFF)); 35811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U4>(0xFF000000, 0x0FFFFFFF)); 35911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U4>(0xFF000000, 0x00FFFFFF)); 36011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U4>(0xFF000000, 0x0000FFFF)); 36111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U4>(0xFF000000, 0x00000000)); 36211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(applyNot<U4>(0xFF000000, 0xFF000000)); 36311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U4>(0xFF800000, 0xFF000000)); 36411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(applyNot<U4>(0x00008000, 0x00008000)); 36511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 36611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U1>(0x0, 0x0)); 36711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U1>(0xFF, 0xFE)); 36811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U1>(0xFF, 0x0)); 36911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(applyNot<U1>(0xFF, 0xFF)); 37011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 37111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(applyNot<void*>((void*)0xFFFFFF, (void*)(-1))); 37211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<void*>((void*)0xFFFFFF, (void*)(-2))); 3730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 3740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 3750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, Shift) { 3760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U4 *up = GetPoisoned<U4>(); 3770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ((U1*)up)[0] = 0; 3780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ((U1*)up)[3] = 0xff; 37911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*up >> 30); 38011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*up >> 24); 38111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*up >> 23); 38211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*up >> 10); 3830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 38411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*up << 30); 38511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*up << 24); 38611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*up << 23); 38711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*up << 10); 3880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 3890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 *sp = (S4*)up; 39011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*sp >> 30); 39111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*sp >> 24); 39211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*sp >> 23); 39311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*sp >> 10); 3940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 3950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov sp = GetPoisoned<S4>(); 3960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ((S1*)sp)[1] = 0; 3970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ((S1*)sp)[2] = 0; 39811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*sp >> 31); 3990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 40011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(100 >> *GetPoisoned<S4>()); 40111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(100U >> *GetPoisoned<S4>()); 4020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE static int GetPoisonedZero() { 4050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int *zero = new int; 4060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *zero = 0; 4070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(zero, sizeof(*zero)); 4080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = *zero; 4090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete zero; 4100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 4110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, LoadFromDirtyAddress) { 4140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int *a = new int; 4150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *a = 0; 41612c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov EXPECT_UMR(break_optimization((void*)(U8)a[GetPoisonedZero()])); 4170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete a; 4180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, StoreToDirtyAddress) { 4210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int *a = new int; 42211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR(a[GetPoisonedZero()] = 0); 42312c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(a); 4240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete a; 4250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE void StackTestFunc() { 4290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 p4; 4300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 ok4 = 1; 4310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S2 p2; 4320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S2 ok2 = 1; 4330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S1 p1; 4340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S1 ok1 = 1; 43512c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&p4); 43612c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&ok4); 43712c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&p2); 43812c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&ok2); 43912c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&p1); 44012c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&ok1); 4410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 44211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(p4); 44311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(p2); 44411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(p1); 44511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(ok1); 44611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(ok2); 44711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(ok4); 4480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, StackTest) { 4510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StackTestFunc(); 4520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE void StackStressFunc() { 4550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int foo[10000]; 45612c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(foo); 4570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, DISABLED_StackStressTest) { 4600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov for (int i = 0; i < 1000000; i++) 4610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StackStressFunc(); 4620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> 4650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovvoid TestFloatingPoint() { 4660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov static volatile T v; 4670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov static T g[100]; 46812c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&g); 4690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *x = GetPoisoned<T>(); 4700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *y = GetPoisoned<T>(1); 47111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*x); 47211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED((long long)*x); 47311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED((int)*x); 4740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov g[0] = *x; 4750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov g[1] = *x + *y; 4760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov g[2] = *x - *y; 4770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov g[3] = *x * *y; 4780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, FloatingPointTest) { 4810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestFloatingPoint<float>(); 4820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestFloatingPoint<double>(); 4830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, DynMem) { 4860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 x = 0; 4870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 *y = GetPoisoned<S4>(); 4880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov memcpy(y, &x, g_one * sizeof(S4)); 48911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*y); 4900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic char *DynRetTestStr; 4930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, DynRet) { 4950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) return; 4960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ReturnPoisoned<S8>(); 49711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(clearenv()); 4980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, DynRet1) { 5020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) return; 5030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ReturnPoisoned<S8>(); 5040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct LargeStruct { 5070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 x[10]; 5080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 5090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE 5110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovLargeStruct LargeRetTest() { 5120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov LargeStruct res; 5130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[0] = *GetPoisoned<S4>(); 5140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[1] = *GetPoisoned<S4>(); 5150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[2] = *GetPoisoned<S4>(); 5160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[3] = *GetPoisoned<S4>(); 5170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[4] = *GetPoisoned<S4>(); 5180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[5] = *GetPoisoned<S4>(); 5190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[6] = *GetPoisoned<S4>(); 5200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[7] = *GetPoisoned<S4>(); 5210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[8] = *GetPoisoned<S4>(); 5220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[9] = *GetPoisoned<S4>(); 5230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 5240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, LargeRet) { 5270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov LargeStruct a = LargeRetTest(); 52811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(a.x[0]); 52911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(a.x[9]); 5300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, fread) { 5330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *x = new char[32]; 5340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov FILE *f = fopen("/proc/self/stat", "r"); 5350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(f); 5360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov fread(x, 1, 32, f); 53711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[0]); 53811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[16]); 53911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[31]); 5400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov fclose(f); 5410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete x; 5420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, read) { 5450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *x = new char[32]; 5460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int fd = open("/proc/self/stat", O_RDONLY); 5470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(fd > 0); 5480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int sz = read(fd, x, 32); 5490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(sz == 32); 55011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[0]); 55111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[16]); 55211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[31]); 5530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov close(fd); 5540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete x; 5550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, pread) { 5580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *x = new char[32]; 5590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int fd = open("/proc/self/stat", O_RDONLY); 5600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(fd > 0); 5610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int sz = pread(fd, x, 32, 0); 5620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(sz == 32); 56311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[0]); 56411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[16]); 56511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[31]); 5660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov close(fd); 5670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete x; 5680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 570b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy StepanovTEST(MemorySanitizer, readv) { 571b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov char buf[2011]; 572b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov struct iovec iov[2]; 573b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov iov[0].iov_base = buf + 1; 574b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov iov[0].iov_len = 5; 575b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov iov[1].iov_base = buf + 10; 576b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov iov[1].iov_len = 2000; 577b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov int fd = open("/proc/self/stat", O_RDONLY); 578b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov assert(fd > 0); 579b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov int sz = readv(fd, iov, 2); 580b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov ASSERT_LT(sz, 5 + 2000); 581b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov ASSERT_GT(sz, iov[0].iov_len); 582b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov EXPECT_POISONED(buf[0]); 583b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov EXPECT_NOT_POISONED(buf[1]); 584b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov EXPECT_NOT_POISONED(buf[5]); 585b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov EXPECT_POISONED(buf[6]); 586b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov EXPECT_POISONED(buf[9]); 587b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov EXPECT_NOT_POISONED(buf[10]); 588b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov EXPECT_NOT_POISONED(buf[10 + (sz - 1) - 5]); 589b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov EXPECT_POISONED(buf[11 + (sz - 1) - 5]); 590b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov close(fd); 591b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov} 592b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov 593b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy StepanovTEST(MemorySanitizer, preadv) { 594b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov char buf[2011]; 595b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov struct iovec iov[2]; 596b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov iov[0].iov_base = buf + 1; 597b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov iov[0].iov_len = 5; 598b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov iov[1].iov_base = buf + 10; 599b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov iov[1].iov_len = 2000; 600b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov int fd = open("/proc/self/stat", O_RDONLY); 601b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov assert(fd > 0); 602b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov int sz = preadv(fd, iov, 2, 3); 603b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov ASSERT_LT(sz, 5 + 2000); 604b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov ASSERT_GT(sz, iov[0].iov_len); 605b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov EXPECT_POISONED(buf[0]); 606b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov EXPECT_NOT_POISONED(buf[1]); 607b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov EXPECT_NOT_POISONED(buf[5]); 608b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov EXPECT_POISONED(buf[6]); 609b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov EXPECT_POISONED(buf[9]); 610b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov EXPECT_NOT_POISONED(buf[10]); 611b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov EXPECT_NOT_POISONED(buf[10 + (sz - 1) - 5]); 612b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov EXPECT_POISONED(buf[11 + (sz - 1) - 5]); 613b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov close(fd); 614b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov} 615b916e6a9b3efa4907b70a2dcd418c76b044171e7Evgeniy Stepanov 6160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// FIXME: fails now. 6170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, DISABLED_ioctl) { 6180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov struct winsize ws; 6190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(ioctl(2, TIOCGWINSZ, &ws), 0); 62011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(ws.ws_col); 6210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 6220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 6230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, readlink) { 6240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *x = new char[1000]; 6250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov readlink("/proc/self/exe", x, 1000); 62611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[0]); 6270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete [] x; 6280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 6290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 6300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 6310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, stat) { 6320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov struct stat* st = new struct stat; 6330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = stat("/proc/self/stat", st); 6340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(!res); 63511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(st->st_dev); 63611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(st->st_mode); 63711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(st->st_size); 6380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 6390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 6400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, statfs) { 6410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov struct statfs* st = new struct statfs; 6420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = statfs("/", st); 6430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(!res); 64411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(st->f_type); 64511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(st->f_bfree); 64611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(st->f_namelen); 6470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 6480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 6490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, pipe) { 6500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* pipefd = new int[2]; 6510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = pipe(pipefd); 6520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(!res); 65311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(pipefd[0]); 65411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(pipefd[1]); 6550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov close(pipefd[0]); 6560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov close(pipefd[1]); 6570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 6580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 659134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy StepanovTEST(MemorySanitizer, pipe2) { 660134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov int* pipefd = new int[2]; 661134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov int res = pipe2(pipefd, O_NONBLOCK); 662134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov assert(!res); 663134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov EXPECT_NOT_POISONED(pipefd[0]); 664134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov EXPECT_NOT_POISONED(pipefd[1]); 665134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov close(pipefd[0]); 666134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov close(pipefd[1]); 667134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov} 668134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov 669134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy StepanovTEST(MemorySanitizer, socketpair) { 670134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov int sv[2]; 671134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov int res = socketpair(AF_UNIX, SOCK_STREAM, 0, sv); 672134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov assert(!res); 673134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov EXPECT_NOT_POISONED(sv[0]); 674134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov EXPECT_NOT_POISONED(sv[1]); 675134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov close(sv[0]); 676134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov close(sv[1]); 677134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov} 678134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov 6799f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy StepanovTEST(MemorySanitizer, bind_getsockname) { 6809f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov int sock = socket(AF_UNIX, SOCK_STREAM, 0); 6819f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov 6829f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov struct sockaddr_in sai; 6839f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov memset(&sai, 0, sizeof(sai)); 6849f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov sai.sin_family = AF_UNIX; 6859f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov int res = bind(sock, (struct sockaddr *)&sai, sizeof(sai)); 6869f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov 6879f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov assert(!res); 6889f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov char buf[200]; 6899f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov socklen_t addrlen; 6909f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov EXPECT_UMR(getsockname(sock, (struct sockaddr *)&buf, &addrlen)); 6919f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov 6929f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov addrlen = sizeof(buf); 6939f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov res = getsockname(sock, (struct sockaddr *)&buf, &addrlen); 6949f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov EXPECT_NOT_POISONED(addrlen); 6959f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov EXPECT_NOT_POISONED(buf[0]); 6969f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov EXPECT_NOT_POISONED(buf[addrlen - 1]); 6979f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov EXPECT_POISONED(buf[addrlen]); 6989f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov close(sock); 6999f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov} 7009f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov 7019d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy StepanovTEST(MemorySanitizer, accept) { 7029d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov int listen_socket = socket(AF_INET, SOCK_STREAM, 0); 7039d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_LT(0, listen_socket); 7049d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov 7059d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov struct sockaddr_in sai; 7069d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov sai.sin_family = AF_INET; 7079d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov sai.sin_port = 0; 7089d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov sai.sin_addr.s_addr = htonl(INADDR_LOOPBACK); 7099d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov int res = bind(listen_socket, (struct sockaddr *)&sai, sizeof(sai)); 7109d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_EQ(0, res); 7119d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov 7129d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov res = listen(listen_socket, 1); 7139d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_EQ(0, res); 7149d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov 7159d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov socklen_t sz = sizeof(sai); 7169d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov res = getsockname(listen_socket, (struct sockaddr *)&sai, &sz); 7179d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_EQ(0, res); 7189d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_EQ(sizeof(sai), sz); 719bc33e138d82759074f8333239f96506027731e20Evgeniy Stepanov 7209d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov int connect_socket = socket(AF_INET, SOCK_STREAM, 0); 7219d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_LT(0, connect_socket); 7229d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov res = fcntl(connect_socket, F_SETFL, O_NONBLOCK); 7239d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_EQ(0, res); 7249d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov res = connect(connect_socket, (struct sockaddr *)&sai, sizeof(sai)); 7259d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_EQ(-1, res); 7269d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_EQ(EINPROGRESS, errno); 7279d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov 7289d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov __msan_poison(&sai, sizeof(sai)); 729bc33e138d82759074f8333239f96506027731e20Evgeniy Stepanov int new_sock = accept(listen_socket, (struct sockaddr *)&sai, &sz); 7309d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_LT(0, new_sock); 7319d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_EQ(sizeof(sai), sz); 7329d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov EXPECT_NOT_POISONED(sai); 7339d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov 734bc33e138d82759074f8333239f96506027731e20Evgeniy Stepanov __msan_poison(&sai, sizeof(sai)); 735bc33e138d82759074f8333239f96506027731e20Evgeniy Stepanov res = getpeername(new_sock, (struct sockaddr *)&sai, &sz); 736bc33e138d82759074f8333239f96506027731e20Evgeniy Stepanov ASSERT_EQ(0, res); 737bc33e138d82759074f8333239f96506027731e20Evgeniy Stepanov ASSERT_EQ(sizeof(sai), sz); 738bc33e138d82759074f8333239f96506027731e20Evgeniy Stepanov EXPECT_NOT_POISONED(sai); 739bc33e138d82759074f8333239f96506027731e20Evgeniy Stepanov 7409d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov close(new_sock); 7419d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov close(connect_socket); 7429d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov close(listen_socket); 7439d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov} 7449d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov 745512c616cacf70ca029a2bf719a482b902f3687cdEvgeniy StepanovTEST(MemorySanitizer, getaddrinfo) { 746512c616cacf70ca029a2bf719a482b902f3687cdEvgeniy Stepanov struct addrinfo *ai; 747512c616cacf70ca029a2bf719a482b902f3687cdEvgeniy Stepanov struct addrinfo hints; 748512c616cacf70ca029a2bf719a482b902f3687cdEvgeniy Stepanov memset(&hints, 0, sizeof(hints)); 749512c616cacf70ca029a2bf719a482b902f3687cdEvgeniy Stepanov hints.ai_family = AF_INET; 750512c616cacf70ca029a2bf719a482b902f3687cdEvgeniy Stepanov int res = getaddrinfo("localhost", NULL, &hints, &ai); 751512c616cacf70ca029a2bf719a482b902f3687cdEvgeniy Stepanov ASSERT_EQ(0, res); 752512c616cacf70ca029a2bf719a482b902f3687cdEvgeniy Stepanov EXPECT_NOT_POISONED(*ai); 753512c616cacf70ca029a2bf719a482b902f3687cdEvgeniy Stepanov ASSERT_EQ(sizeof(sockaddr_in), ai->ai_addrlen); 754512c616cacf70ca029a2bf719a482b902f3687cdEvgeniy Stepanov EXPECT_NOT_POISONED(*(sockaddr_in*)ai->ai_addr); 755512c616cacf70ca029a2bf719a482b902f3687cdEvgeniy Stepanov} 756512c616cacf70ca029a2bf719a482b902f3687cdEvgeniy Stepanov 7570a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov#define EXPECT_HOSTENT_NOT_POISONED(he) \ 7580a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov do { \ 7590a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(*(he)); \ 7600a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *) 0, (he)->h_name); \ 7610a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *) 0, (he)->h_aliases); \ 7620a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *) 0, (he)->h_addr_list); \ 7630a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(strlen((he)->h_name)); \ 7640a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov char **p = (he)->h_aliases; \ 7650a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov while (*p) { \ 7660a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(*p)); \ 7670a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ++p; \ 7680a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov } \ 7690a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov char **q = (he)->h_addr_list; \ 7700a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov while (*q) { \ 7710a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(*q[0]); \ 7720a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ++q; \ 7730a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov } \ 7740a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(*q); \ 7750a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov } while (0) 7760a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov 7770a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy StepanovTEST(MemorySanitizer, gethostent) { 7780a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent *he = gethostent(); 7790a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *)NULL, he); 7800a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_HOSTENT_NOT_POISONED(he); 7810a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov} 7820a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov 783263800ba59720f9be69735b4dad625fca92430bcEvgeniy Stepanov#ifndef MSAN_TEST_DISABLE_GETHOSTBYNAME 784263800ba59720f9be69735b4dad625fca92430bcEvgeniy Stepanov 7850a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy StepanovTEST(MemorySanitizer, gethostbyname) { 7860a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent *he = gethostbyname("localhost"); 7870a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *)NULL, he); 7880a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_HOSTENT_NOT_POISONED(he); 7890a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov} 7900a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov 791263800ba59720f9be69735b4dad625fca92430bcEvgeniy Stepanov#endif // MSAN_TEST_DISABLE_GETHOSTBYNAME 792263800ba59720f9be69735b4dad625fca92430bcEvgeniy Stepanov 7930a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy StepanovTEST(MemorySanitizer, gethostbyname2) { 7940a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent *he = gethostbyname2("localhost", AF_INET); 7950a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *)NULL, he); 7960a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_HOSTENT_NOT_POISONED(he); 7970a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov} 7980a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov 7990a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy StepanovTEST(MemorySanitizer, gethostbyaddr) { 8000a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov in_addr_t addr = inet_addr("127.0.0.1"); 8010a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(addr); 8020a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent *he = gethostbyaddr(&addr, sizeof(addr), AF_INET); 8030a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *)NULL, he); 8040a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_HOSTENT_NOT_POISONED(he); 8050a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov} 8060a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov 8070a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy StepanovTEST(MemorySanitizer, gethostent_r) { 8080a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov char buf[2000]; 8090a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent he; 8100a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent *result; 8110a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov int err; 8120a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov int res = gethostent_r(&he, buf, sizeof(buf), &result, &err); 8130a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_EQ(0, res); 8140a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(result); 8150a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *)NULL, result); 8160a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_HOSTENT_NOT_POISONED(result); 8170a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(err); 8180a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov} 8190a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov 8200a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy StepanovTEST(MemorySanitizer, gethostbyname_r) { 8210a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov char buf[2000]; 8220a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent he; 8230a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent *result; 8240a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov int err; 8250a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov int res = gethostbyname_r("localhost", &he, buf, sizeof(buf), &result, &err); 8260a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_EQ(0, res); 8270a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(result); 8280a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *)NULL, result); 8290a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_HOSTENT_NOT_POISONED(result); 8300a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(err); 8310a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov} 8320a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov 8330a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy StepanovTEST(MemorySanitizer, gethostbyname2_r) { 8340a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov char buf[2000]; 8350a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent he; 8360a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent *result; 8370a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov int err; 8380a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov int res = gethostbyname2_r("localhost", AF_INET, &he, buf, sizeof(buf), 8390a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov &result, &err); 8400a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_EQ(0, res); 8410a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(result); 8420a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *)NULL, result); 8430a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_HOSTENT_NOT_POISONED(result); 8440a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(err); 8450a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov} 8460a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov 8470a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy StepanovTEST(MemorySanitizer, gethostbyaddr_r) { 8480a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov char buf[2000]; 8490a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent he; 8500a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent *result; 8510a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov int err; 8520a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov in_addr_t addr = inet_addr("127.0.0.1"); 8530a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(addr); 8540a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov int res = gethostbyaddr_r(&addr, sizeof(addr), AF_INET, &he, buf, sizeof(buf), 8550a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov &result, &err); 8560a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_EQ(0, res); 8570a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(result); 8580a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *)NULL, result); 8590a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_HOSTENT_NOT_POISONED(result); 8600a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(err); 8610a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov} 8620a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov 863f32be42523a199674ea665a499db131591e64e08Evgeniy StepanovTEST(MemorySanitizer, getsockopt) { 864f32be42523a199674ea665a499db131591e64e08Evgeniy Stepanov int sock = socket(AF_UNIX, SOCK_STREAM, 0); 865f32be42523a199674ea665a499db131591e64e08Evgeniy Stepanov struct linger l[2]; 866f32be42523a199674ea665a499db131591e64e08Evgeniy Stepanov socklen_t sz = sizeof(l[0]); 867f32be42523a199674ea665a499db131591e64e08Evgeniy Stepanov int res = getsockopt(sock, SOL_SOCKET, SO_LINGER, &l[0], &sz); 868f32be42523a199674ea665a499db131591e64e08Evgeniy Stepanov ASSERT_EQ(0, res); 869f32be42523a199674ea665a499db131591e64e08Evgeniy Stepanov ASSERT_EQ(sizeof(l[0]), sz); 870f32be42523a199674ea665a499db131591e64e08Evgeniy Stepanov EXPECT_NOT_POISONED(l[0]); 871f32be42523a199674ea665a499db131591e64e08Evgeniy Stepanov EXPECT_POISONED(*(char *)(l + 1)); 872f32be42523a199674ea665a499db131591e64e08Evgeniy Stepanov} 873f32be42523a199674ea665a499db131591e64e08Evgeniy Stepanov 8740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, getcwd) { 8750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char path[PATH_MAX + 1]; 8760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* res = getcwd(path, sizeof(path)); 8770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(res); 87811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(path[0]); 8790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 8800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 8817eed04c4dce69ad1e485edbf6dd963e176b52e0dEvgeniy StepanovTEST(MemorySanitizer, getcwd_gnu) { 8827eed04c4dce69ad1e485edbf6dd963e176b52e0dEvgeniy Stepanov char* res = getcwd(NULL, 0); 8837eed04c4dce69ad1e485edbf6dd963e176b52e0dEvgeniy Stepanov assert(res); 8847eed04c4dce69ad1e485edbf6dd963e176b52e0dEvgeniy Stepanov EXPECT_NOT_POISONED(res[0]); 8857eed04c4dce69ad1e485edbf6dd963e176b52e0dEvgeniy Stepanov free(res); 8867eed04c4dce69ad1e485edbf6dd963e176b52e0dEvgeniy Stepanov} 8877eed04c4dce69ad1e485edbf6dd963e176b52e0dEvgeniy Stepanov 888d97a15a931ae6696a1071e1471c9a019e821904bEvgeniy StepanovTEST(MemorySanitizer, readdir) { 889d97a15a931ae6696a1071e1471c9a019e821904bEvgeniy Stepanov DIR *dir = opendir("."); 890d97a15a931ae6696a1071e1471c9a019e821904bEvgeniy Stepanov struct dirent *d = readdir(dir); 891d97a15a931ae6696a1071e1471c9a019e821904bEvgeniy Stepanov assert(d); 892d97a15a931ae6696a1071e1471c9a019e821904bEvgeniy Stepanov EXPECT_NOT_POISONED(d->d_name[0]); 893d97a15a931ae6696a1071e1471c9a019e821904bEvgeniy Stepanov closedir(dir); 894d97a15a931ae6696a1071e1471c9a019e821904bEvgeniy Stepanov} 895d97a15a931ae6696a1071e1471c9a019e821904bEvgeniy Stepanov 896b5cf98f76fd62236f2945bff17b3cdb4e8a5c2f4Evgeniy StepanovTEST(MemorySanitizer, readdir_r) { 897b5cf98f76fd62236f2945bff17b3cdb4e8a5c2f4Evgeniy Stepanov DIR *dir = opendir("."); 898b5cf98f76fd62236f2945bff17b3cdb4e8a5c2f4Evgeniy Stepanov struct dirent d; 899b5cf98f76fd62236f2945bff17b3cdb4e8a5c2f4Evgeniy Stepanov struct dirent *pd; 900b5cf98f76fd62236f2945bff17b3cdb4e8a5c2f4Evgeniy Stepanov int res = readdir_r(dir, &d, &pd); 901b5cf98f76fd62236f2945bff17b3cdb4e8a5c2f4Evgeniy Stepanov assert(!res); 902b5cf98f76fd62236f2945bff17b3cdb4e8a5c2f4Evgeniy Stepanov EXPECT_NOT_POISONED(pd); 903b5cf98f76fd62236f2945bff17b3cdb4e8a5c2f4Evgeniy Stepanov EXPECT_NOT_POISONED(d.d_name[0]); 904b5cf98f76fd62236f2945bff17b3cdb4e8a5c2f4Evgeniy Stepanov closedir(dir); 905b5cf98f76fd62236f2945bff17b3cdb4e8a5c2f4Evgeniy Stepanov} 906b5cf98f76fd62236f2945bff17b3cdb4e8a5c2f4Evgeniy Stepanov 9070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, realpath) { 9080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov const char* relpath = "."; 9090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char path[PATH_MAX + 1]; 9100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* res = realpath(relpath, path); 9110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(res); 91211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(path[0]); 9130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, memcpy) { 9160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* x = new char[2]; 9170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* y = new char[2]; 9180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[0] = 1; 9190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[1] = *GetPoisoned<char>(); 9200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov memcpy(y, x, 2); 92111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(y[0]); 92211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(y[1]); 9230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, memmove) { 9260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* x = new char[2]; 9270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* y = new char[2]; 9280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[0] = 1; 9290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[1] = *GetPoisoned<char>(); 9300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov memmove(y, x, 2); 93111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(y[0]); 93211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(y[1]); 9330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, strdup) { 9368aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov char buf[4] = "abc"; 9378aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov __msan_poison(buf + 2, sizeof(*buf)); 9388aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov char *x = strdup(buf); 9398aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov EXPECT_NOT_POISONED(x[0]); 9408aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov EXPECT_NOT_POISONED(x[1]); 9418aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov EXPECT_POISONED(x[2]); 9428aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov EXPECT_NOT_POISONED(x[3]); 9438aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov free(x); 9448aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov} 9458aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov 9468aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy StepanovTEST(MemorySanitizer, strndup) { 9478aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov char buf[4] = "abc"; 9488aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov __msan_poison(buf + 2, sizeof(*buf)); 9498aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov char *x = strndup(buf, 3); 9508aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov EXPECT_NOT_POISONED(x[0]); 9518aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov EXPECT_NOT_POISONED(x[1]); 9528aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov EXPECT_POISONED(x[2]); 9538aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov EXPECT_NOT_POISONED(x[3]); 9540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov free(x); 9550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9578aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy StepanovTEST(MemorySanitizer, strndup_short) { 9588aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov char buf[4] = "abc"; 9598aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov __msan_poison(buf + 1, sizeof(*buf)); 9608aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov __msan_poison(buf + 2, sizeof(*buf)); 9618aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov char *x = strndup(buf, 2); 9628aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov EXPECT_NOT_POISONED(x[0]); 9638aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov EXPECT_POISONED(x[1]); 9641d21bd159c3830dd7a6c05854408442d91a336adEvgeniy Stepanov EXPECT_NOT_POISONED(x[2]); 9658aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov free(x); 9668aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov} 9678aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov 9688aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov 9690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T, int size> 9700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovvoid TestOverlapMemmove() { 9710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *x = new T[size]; 9720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(size >= 3); 9730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[2] = 0; 9740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov memmove(x, x + 1, (size - 1) * sizeof(T)); 97511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[1]); 9760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) { 9770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // FIXME: under DR we will lose this information 9780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // because accesses in memmove will unpoisin the shadow. 9790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // We need to use our own memove implementation instead of libc's. 98011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[0]); 98111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[2]); 9820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } 9830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete [] x; 9840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, overlap_memmove) { 9870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestOverlapMemmove<U1, 10>(); 9880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestOverlapMemmove<U1, 1000>(); 9890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestOverlapMemmove<U8, 4>(); 9900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestOverlapMemmove<U8, 1000>(); 9910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, strcpy) { // NOLINT 9940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* x = new char[3]; 9950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* y = new char[3]; 9960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[0] = 'a'; 9970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[1] = *GetPoisoned<char>(1, 1); 9980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[2] = 0; 9990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov strcpy(y, x); // NOLINT 100011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(y[0]); 100111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(y[1]); 100211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(y[2]); 10030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 10040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 10050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, strncpy) { // NOLINT 10060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* x = new char[3]; 10070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* y = new char[3]; 10080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[0] = 'a'; 10090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[1] = *GetPoisoned<char>(1, 1); 10100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[2] = 0; 10110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov strncpy(y, x, 2); // NOLINT 101211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(y[0]); 101311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(y[1]); 101411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(y[2]); 10150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 10160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 10170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, strtol) { 10180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *e; 10190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(1 == strtol("1", &e, 10)); 102011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((S8) e); 10210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 10220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 10230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, strtoll) { 10240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *e; 10250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(1 == strtoll("1", &e, 10)); 102611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((S8) e); 10270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 10280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 10290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, strtoul) { 10300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *e; 10310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(1 == strtoul("1", &e, 10)); 103211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((S8) e); 10330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 10340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 10350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, strtoull) { 10360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *e; 10370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(1 == strtoull("1", &e, 10)); 103811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((S8) e); 10390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 10400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1041e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy StepanovTEST(MemorySanitizer, strtod) { 1042e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov char *e; 1043e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov assert(0 != strtod("1.5", &e)); 104411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((S8) e); 1045e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov} 1046e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov 1047e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy StepanovTEST(MemorySanitizer, strtof) { 1048e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov char *e; 1049e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov assert(0 != strtof("1.5", &e)); 105011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((S8) e); 1051e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov} 1052e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov 1053e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy StepanovTEST(MemorySanitizer, strtold) { 1054e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov char *e; 1055e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov assert(0 != strtold("1.5", &e)); 105611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((S8) e); 1057e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov} 1058e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov 1059c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy StepanovTEST(MemorySanitizer, modf) { 1060c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy Stepanov double x, y; 1061c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy Stepanov x = modf(2.1, &y); 1062c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy Stepanov EXPECT_NOT_POISONED(y); 1063c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy Stepanov} 1064c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy Stepanov 1065c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy StepanovTEST(MemorySanitizer, modff) { 1066c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy Stepanov float x, y; 1067c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy Stepanov x = modff(2.1, &y); 1068c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy Stepanov EXPECT_NOT_POISONED(y); 1069c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy Stepanov} 1070c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy Stepanov 1071c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy StepanovTEST(MemorySanitizer, modfl) { 1072c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy Stepanov long double x, y; 1073c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy Stepanov x = modfl(2.1, &y); 1074c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy Stepanov EXPECT_NOT_POISONED(y); 1075c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy Stepanov} 1076c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy Stepanov 10770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, sprintf) { // NOLINT 10780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char buff[10]; 107912c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(buff); 108011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(buff[0]); 10810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = sprintf(buff, "%d", 1234567); // NOLINT 10820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(res == 7); 10830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[0] == '1'); 10840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[1] == '2'); 10850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[2] == '3'); 10860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[6] == '7'); 10870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[7] == 0); 108811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(buff[8]); 10890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 10900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 10910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, snprintf) { 10920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char buff[10]; 109312c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(buff); 109411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(buff[0]); 10950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = snprintf(buff, sizeof(buff), "%d", 1234567); 10960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(res == 7); 10970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[0] == '1'); 10980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[1] == '2'); 10990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[2] == '3'); 11000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[6] == '7'); 11010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[7] == 0); 110211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(buff[8]); 11030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 11040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 11050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, swprintf) { 11060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov wchar_t buff[10]; 11070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(sizeof(wchar_t) == 4); 110812c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(buff); 110911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(buff[0]); 11100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = swprintf(buff, 9, L"%d", 1234567); 11110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(res == 7); 11120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[0] == '1'); 11130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[1] == '2'); 11140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[2] == '3'); 11150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[6] == '7'); 11160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[7] == 0); 111711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(buff[8]); 11180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 11190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 11200797ed4bfc4adaa1436dbd153b276ed9c917f1b7Evgeniy StepanovTEST(MemorySanitizer, asprintf) { // NOLINT 11210797ed4bfc4adaa1436dbd153b276ed9c917f1b7Evgeniy Stepanov char *pbuf; 11220797ed4bfc4adaa1436dbd153b276ed9c917f1b7Evgeniy Stepanov EXPECT_POISONED(pbuf); 11230797ed4bfc4adaa1436dbd153b276ed9c917f1b7Evgeniy Stepanov int res = asprintf(&pbuf, "%d", 1234567); // NOLINT 11240797ed4bfc4adaa1436dbd153b276ed9c917f1b7Evgeniy Stepanov assert(res == 7); 11250797ed4bfc4adaa1436dbd153b276ed9c917f1b7Evgeniy Stepanov EXPECT_NOT_POISONED(pbuf); 11260797ed4bfc4adaa1436dbd153b276ed9c917f1b7Evgeniy Stepanov assert(pbuf[0] == '1'); 11270797ed4bfc4adaa1436dbd153b276ed9c917f1b7Evgeniy Stepanov assert(pbuf[1] == '2'); 11280797ed4bfc4adaa1436dbd153b276ed9c917f1b7Evgeniy Stepanov assert(pbuf[2] == '3'); 11290797ed4bfc4adaa1436dbd153b276ed9c917f1b7Evgeniy Stepanov assert(pbuf[6] == '7'); 11300797ed4bfc4adaa1436dbd153b276ed9c917f1b7Evgeniy Stepanov assert(pbuf[7] == 0); 11310797ed4bfc4adaa1436dbd153b276ed9c917f1b7Evgeniy Stepanov free(pbuf); 11320797ed4bfc4adaa1436dbd153b276ed9c917f1b7Evgeniy Stepanov} 11330797ed4bfc4adaa1436dbd153b276ed9c917f1b7Evgeniy Stepanov 11340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, wcstombs) { 11350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov const wchar_t *x = L"abc"; 11360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char buff[10]; 11370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = wcstombs(buff, x, 4); 11380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(res, 3); 11390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(buff[0], 'a'); 11400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(buff[1], 'b'); 11410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(buff[2], 'c'); 11420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 11430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 11440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, gettimeofday) { 11450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov struct timeval tv; 11460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov struct timezone tz; 114712c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&tv); 114812c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&tz); 11490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(sizeof(tv) == 16); 11500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(sizeof(tz) == 8); 115111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(tv.tv_sec); 115211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(tv.tv_usec); 115311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(tz.tz_minuteswest); 115411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(tz.tz_dsttime); 11550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(0 == gettimeofday(&tv, &tz)); 115611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(tv.tv_sec); 115711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(tv.tv_usec); 115811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(tz.tz_minuteswest); 115911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(tz.tz_dsttime); 11600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 11610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1162e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy StepanovTEST(MemorySanitizer, clock_gettime) { 1163e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov struct timespec tp; 1164e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_POISONED(tp.tv_sec); 1165e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_POISONED(tp.tv_nsec); 1166e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov assert(0 == clock_gettime(CLOCK_REALTIME, &tp)); 1167e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(tp.tv_sec); 1168e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(tp.tv_nsec); 1169e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov} 1170e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov 11717cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy StepanovTEST(MemorySanitizer, clock_getres) { 11727cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov struct timespec tp; 11737cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov EXPECT_POISONED(tp.tv_sec); 11747cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov EXPECT_POISONED(tp.tv_nsec); 11757cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov assert(0 == clock_getres(CLOCK_REALTIME, 0)); 11767cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov EXPECT_POISONED(tp.tv_sec); 11777cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov EXPECT_POISONED(tp.tv_nsec); 11787cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov assert(0 == clock_getres(CLOCK_REALTIME, &tp)); 11797cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov EXPECT_NOT_POISONED(tp.tv_sec); 11807cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov EXPECT_NOT_POISONED(tp.tv_nsec); 11817cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov} 11827cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov 1183e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy StepanovTEST(MemorySanitizer, getitimer) { 1184e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov struct itimerval it1, it2; 1185e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov int res; 1186e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_POISONED(it1.it_interval.tv_sec); 1187e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_POISONED(it1.it_interval.tv_usec); 1188e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_POISONED(it1.it_value.tv_sec); 1189e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_POISONED(it1.it_value.tv_usec); 1190e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov res = getitimer(ITIMER_VIRTUAL, &it1); 1191e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov assert(!res); 1192e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(it1.it_interval.tv_sec); 1193e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(it1.it_interval.tv_usec); 1194e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(it1.it_value.tv_sec); 1195e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(it1.it_value.tv_usec); 1196e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov 1197e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov it1.it_interval.tv_sec = it1.it_value.tv_sec = 10000; 1198e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov it1.it_interval.tv_usec = it1.it_value.tv_usec = 0; 1199e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov 1200e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov res = setitimer(ITIMER_VIRTUAL, &it1, &it2); 1201e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov assert(!res); 1202e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(it2.it_interval.tv_sec); 1203e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(it2.it_interval.tv_usec); 1204e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(it2.it_value.tv_sec); 1205e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(it2.it_value.tv_usec); 1206e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov 1207e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov // Check that old_value can be 0, and disable the timer. 1208e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov memset(&it1, 0, sizeof(it1)); 1209e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov res = setitimer(ITIMER_VIRTUAL, &it1, 0); 1210e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov assert(!res); 1211e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov} 1212e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov 1213fef660506e9e5703fedfee01d614abd4b741c738Evgeniy StepanovTEST(MemorySanitizer, time) { 1214fef660506e9e5703fedfee01d614abd4b741c738Evgeniy Stepanov time_t t; 1215fef660506e9e5703fedfee01d614abd4b741c738Evgeniy Stepanov EXPECT_POISONED(t); 1216fef660506e9e5703fedfee01d614abd4b741c738Evgeniy Stepanov time_t t2 = time(&t); 1217fef660506e9e5703fedfee01d614abd4b741c738Evgeniy Stepanov assert(t2 != (time_t)-1); 1218fef660506e9e5703fedfee01d614abd4b741c738Evgeniy Stepanov EXPECT_NOT_POISONED(t); 1219fef660506e9e5703fedfee01d614abd4b741c738Evgeniy Stepanov} 1220fef660506e9e5703fedfee01d614abd4b741c738Evgeniy Stepanov 12219358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy StepanovTEST(MemorySanitizer, localtime) { 12229358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov time_t t = 123; 12239358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov struct tm *time = localtime(&t); 12249358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov assert(time != 0); 12259358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov EXPECT_NOT_POISONED(time->tm_sec); 12269358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov EXPECT_NOT_POISONED(time->tm_hour); 12279358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov EXPECT_NOT_POISONED(time->tm_year); 12289358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov EXPECT_NOT_POISONED(time->tm_isdst); 12299358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov} 12309358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov 12319358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy StepanovTEST(MemorySanitizer, localtime_r) { 12329358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov time_t t = 123; 12339358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov struct tm time; 12349358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov struct tm *res = localtime_r(&t, &time); 12359358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov assert(res != 0); 12369358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov EXPECT_NOT_POISONED(time.tm_sec); 12379358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov EXPECT_NOT_POISONED(time.tm_hour); 12389358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov EXPECT_NOT_POISONED(time.tm_year); 12399358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov EXPECT_NOT_POISONED(time.tm_isdst); 12409358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov} 12419358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov 12420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, mmap) { 12430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov const int size = 4096; 12440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov void *p1, *p2; 12450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov p1 = mmap(0, size, PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANON, -1, 0); 12460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(p1, size); 12470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov munmap(p1, size); 12480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov for (int i = 0; i < 1000; i++) { 12490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov p2 = mmap(0, size, PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANON, -1, 0); 12500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (p2 == p1) 12510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov break; 12520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov else 12530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov munmap(p2, size); 12540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } 12550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (p1 == p2) { 125611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*(char*)p2); 12570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov munmap(p2, size); 12580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } 12590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 12600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 12610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// FIXME: enable and add ecvt. 12620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// FIXME: check why msandr does nt handle fcvt. 12630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, fcvt) { 12640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int a, b; 126512c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&a); 126612c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&b); 126711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(a); 126811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(b); 12690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *str = fcvt(12345.6789, 10, &a, &b); 127011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(a); 127111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(b); 12720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 12730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 12747cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy StepanovTEST(MemorySanitizer, frexp) { 12757cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov int x; 12767cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov x = *GetPoisoned<int>(); 12777cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov double r = frexp(1.1, &x); 12787cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov EXPECT_NOT_POISONED(r); 12797cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov EXPECT_NOT_POISONED(x); 12807cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov 12817cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov x = *GetPoisoned<int>(); 12827cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov float rf = frexpf(1.1, &x); 12837cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov EXPECT_NOT_POISONED(rf); 12847cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov EXPECT_NOT_POISONED(x); 12857cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov 12867cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov x = *GetPoisoned<int>(); 12877cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov double rl = frexpl(1.1, &x); 12887cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov EXPECT_NOT_POISONED(rl); 12897cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov EXPECT_NOT_POISONED(x); 12907cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov} 12917cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov 129206658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanovnamespace { 129306658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov 129406658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanovstatic int cnt; 129506658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov 129606658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanovvoid SigactionHandler(int signo, siginfo_t* si, void* uc) { 129706658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov assert(signo == SIGPROF); 129806658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov assert(si); 129906658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov EXPECT_NOT_POISONED(si->si_errno); 130006658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov EXPECT_NOT_POISONED(si->si_pid); 130106658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov#if __linux__ 130206658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov# if defined(__x86_64__) 130306658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov EXPECT_NOT_POISONED(((ucontext_t*)uc)->uc_mcontext.gregs[REG_RIP]); 130406658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov# elif defined(__i386__) 130506658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov EXPECT_NOT_POISONED(((ucontext_t*)uc)->uc_mcontext.gregs[REG_EIP]); 130606658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov# endif 130706658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov#endif 130806658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov ++cnt; 130906658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov} 131006658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov 131106658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy StepanovTEST(MemorySanitizer, sigaction) { 131206658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov struct sigaction act = {}; 1313cd3049da150124156502b1a8c05e4c4887786cc5Evgeniy Stepanov struct sigaction oldact = {}; 13146d0b7f6c41bed64419dd74a8daf83de5e9f54de9Evgeniy Stepanov struct sigaction origact = {}; 13156d0b7f6c41bed64419dd74a8daf83de5e9f54de9Evgeniy Stepanov 13166d0b7f6c41bed64419dd74a8daf83de5e9f54de9Evgeniy Stepanov sigaction(SIGPROF, 0, &origact); 13176d0b7f6c41bed64419dd74a8daf83de5e9f54de9Evgeniy Stepanov 131806658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov act.sa_flags |= SA_SIGINFO; 131906658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov act.sa_sigaction = &SigactionHandler; 132006658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov sigaction(SIGPROF, &act, 0); 132106658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov 132206658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov kill(getpid(), SIGPROF); 132306658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov 132406658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov act.sa_flags &= ~SA_SIGINFO; 132506658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov act.sa_handler = SIG_DFL; 132606658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov sigaction(SIGPROF, &act, 0); 132706658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov 132806658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov act.sa_flags &= ~SA_SIGINFO; 132906658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov act.sa_handler = SIG_IGN; 1330cd3049da150124156502b1a8c05e4c4887786cc5Evgeniy Stepanov sigaction(SIGPROF, &act, &oldact); 1331cd3049da150124156502b1a8c05e4c4887786cc5Evgeniy Stepanov EXPECT_FALSE(oldact.sa_flags & SA_SIGINFO); 1332cd3049da150124156502b1a8c05e4c4887786cc5Evgeniy Stepanov EXPECT_EQ(SIG_DFL, oldact.sa_handler); 133306658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov kill(getpid(), SIGPROF); 133406658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov 133506658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov act.sa_flags |= SA_SIGINFO; 133606658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov act.sa_sigaction = &SigactionHandler; 1337cd3049da150124156502b1a8c05e4c4887786cc5Evgeniy Stepanov sigaction(SIGPROF, &act, &oldact); 1338cd3049da150124156502b1a8c05e4c4887786cc5Evgeniy Stepanov EXPECT_FALSE(oldact.sa_flags & SA_SIGINFO); 1339cd3049da150124156502b1a8c05e4c4887786cc5Evgeniy Stepanov EXPECT_EQ(SIG_IGN, oldact.sa_handler); 134006658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov kill(getpid(), SIGPROF); 134106658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov 134206658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov act.sa_flags &= ~SA_SIGINFO; 134306658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov act.sa_handler = SIG_DFL; 1344cd3049da150124156502b1a8c05e4c4887786cc5Evgeniy Stepanov sigaction(SIGPROF, &act, &oldact); 1345cd3049da150124156502b1a8c05e4c4887786cc5Evgeniy Stepanov EXPECT_TRUE(oldact.sa_flags & SA_SIGINFO); 1346cd3049da150124156502b1a8c05e4c4887786cc5Evgeniy Stepanov EXPECT_EQ(&SigactionHandler, oldact.sa_sigaction); 134706658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov EXPECT_EQ(2, cnt); 13486d0b7f6c41bed64419dd74a8daf83de5e9f54de9Evgeniy Stepanov 13496d0b7f6c41bed64419dd74a8daf83de5e9f54de9Evgeniy Stepanov sigaction(SIGPROF, &origact, 0); 135006658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov} 135106658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov 135206658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov} // namespace 135306658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov 13540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct StructWithDtor { 13550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ~StructWithDtor(); 13560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 13570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE StructWithDtor::~StructWithDtor() { 135912c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(0); 13600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, Invoke) { 13630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructWithDtor s; // Will cause the calls to become invokes. 136411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(0); 136511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<int>()); 136611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(0); 136711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<int>()); 136811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(ReturnPoisoned<S4>()); 13690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, ptrtoint) { 13720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // Test that shadow is propagated through pointer-to-integer conversion. 13730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov void* p = (void*)0xABCD; 13740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(((char*)&p) + 1, sizeof(p)); 1375250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov EXPECT_NOT_POISONED((((uintptr_t)p) & 0xFF) == 0); 13760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov void* q = (void*)0xABCD; 13780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&q, sizeof(q) - 1); 1379250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov EXPECT_POISONED((((uintptr_t)q) & 0xFF) == 0); 13800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn2(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_NOT_POISONED(va_arg(vl, int)); 138811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, double)); 13890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 13900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn(int guard, ...) { 13930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 13940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 139511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 139611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 13970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // The following call will overwrite __msan_param_tls. 13980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // Checks after it test that arg shadow was somehow saved across the call. 13990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn2(1, 2, 3, 4, *GetPoisoned<double>()); 140011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 140111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 14020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 14030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, VAArgTest) { 14060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* x = GetPoisoned<int>(); 14070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* y = GetPoisoned<int>(4); 14080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn(1, 13, *x, 42, *y); 14090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_many(int guard, ...) { 14120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 14130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 141411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 141511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 141611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 141711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 141811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 141911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 142011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 142111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 142211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 142311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 14240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 14250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, VAArgManyTest) { 14280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* x = GetPoisoned<int>(); 14290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* y = GetPoisoned<int>(4); 14300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_many(1, 2, *x, 3, 4, 5, 6, 7, 8, 9, *y); 14310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_pass2(va_list vl) { 143411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 143511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 143611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 14370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_pass(int guard, ...) { 14400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 14410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 144211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 14430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_pass2(vl); 14440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 14450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, VAArgPass) { 14480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* x = GetPoisoned<int>(); 14490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* y = GetPoisoned<int>(4); 14500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_pass(1, *x, 2, 3, *y); 14510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_copy2(va_list vl) { 145411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 145511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 14560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_copy(int guard, ...) { 14590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 14600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 146111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 146211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 14630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl2; 14640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_copy(vl2, vl); 14650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_copy2(vl2); 146611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 146711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 14680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 14690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, VAArgCopy) { 14720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* x = GetPoisoned<int>(); 14730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* y = GetPoisoned<int>(4); 14740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_copy(1, 2, *x, 3, *y); 14750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_ptr(int guard, ...) { 14780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 14790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 148011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int*)); 148111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int*)); 148211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int*)); 148311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, double*)); 14840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 14850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, VAArgPtr) { 14880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int** x = GetPoisoned<int*>(); 14890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov double** y = GetPoisoned<double*>(8); 14900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int z; 14910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_ptr(1, &z, *x, &z, *y); 14920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_overflow(int guard, ...) { 14950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 14960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 149711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 149811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 149911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 150011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 150111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 150211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 150311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 150411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, double)); 150511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, double)); 150611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, double)); 150711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, double)); 150811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, double)); 150911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int*)); 151011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, double)); 151111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, double)); 151211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 151311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 151411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, double)); 151511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int*)); 151611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 151711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 151811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, double)); 151911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int*)); 152011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 152111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 152211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, double)); 152311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int*)); 15240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 15260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, VAArgOverflow) { 15290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* x = GetPoisoned<int>(); 15300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov double* y = GetPoisoned<double>(8); 15310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int** p = GetPoisoned<int*>(16); 15320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int z; 15330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_overflow(1, 15340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1, 2, *x, 4, 5, 6, 15350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1.1, 2.2, 3.3, *y, 5.5, *p, 7.7, 8.8, 15360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // the following args will overflow for sure 15370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *x, *y, *p, 15380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 7, 9.9, &z, 15390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *x, *y, *p); 15400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_tlsoverwrite2(int guard, ...) { 15430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 15440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 154511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 15460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 15470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_tlsoverwrite(int guard, ...) { 15500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // This call will overwrite TLS contents unless it's backed up somewhere. 15510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_tlsoverwrite2(2, 42); 15520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 15530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 155411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 15550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 15560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, VAArgTLSOverwrite) { 15590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* x = GetPoisoned<int>(); 15600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_tlsoverwrite(1, *x); 15610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct StructByVal { 15640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int a, b, c, d, e, f; 15650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 15660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE void StructByValTestFunc(struct StructByVal s) { 156811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s.a); 156911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s.b); 157011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s.c); 157111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s.d); 157211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s.e); 157311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s.f); 15740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE void StructByValTestFunc1(struct StructByVal s) { 15770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructByValTestFunc(s); 15780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE void StructByValTestFunc2(int z, struct StructByVal s) { 15810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructByValTestFunc(s); 15820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, StructByVal) { 15850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // Large aggregates are passed as "byval" pointer argument in LLVM. 15860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov struct StructByVal s; 15870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov s.a = 1; 15880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov s.b = *GetPoisoned<int>(); 15890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov s.c = 2; 15900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov s.d = *GetPoisoned<int>(); 15910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov s.e = 3; 15920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov s.f = *GetPoisoned<int>(); 15930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructByValTestFunc(s); 15940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructByValTestFunc1(s); 15950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructByValTestFunc2(0, s); 15960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#if MSAN_HAS_M128 1600b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy StepanovNOINLINE __m128i m128Eq(__m128i *a, __m128i *b) { return _mm_cmpeq_epi16(*a, *b); } 1601b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy StepanovNOINLINE __m128i m128Lt(__m128i *a, __m128i *b) { return _mm_cmplt_epi16(*a, *b); } 16020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, m128) { 16030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __m128i a = _mm_set1_epi16(0x1234); 16040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __m128i b = _mm_set1_epi16(0x7890); 160511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(m128Eq(&a, &b)); 160611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(m128Lt(&a, &b)); 16070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 16080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// FIXME: add more tests for __m128i. 16090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#endif // MSAN_HAS_M128 16100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 16110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// We should not complain when copying this poisoned hole. 16120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct StructWithHole { 16130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U4 a; 16140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // 4-byte hole. 16150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U8 b; 16160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 16170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 16180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE StructWithHole ReturnStructWithHole() { 16190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructWithHole res; 16200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&res, sizeof(res)); 16210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.a = 1; 16220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.b = 2; 16230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 16240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 16250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 16260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, StructWithHole) { 16270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructWithHole a = ReturnStructWithHole(); 162812c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&a); 16290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 16300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 16310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate <class T> 16320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE T ReturnStruct() { 16330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T res; 16340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&res, sizeof(res)); 16350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.a = 1; 16360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 16370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 16380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 16390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate <class T> 16400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE void TestReturnStruct() { 16410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T s1 = ReturnStruct<T>(); 164211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s1.a); 164311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s1.b); 16440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 16450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 16460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct SSS1 { 16470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int a, b, c; 16480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 16490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct SSS2 { 16500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int b, a, c; 16510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 16520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct SSS3 { 16530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int b, c, a; 16540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 16550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct SSS4 { 16560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int c, b, a; 16570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 16580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 16590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct SSS5 { 16600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int a; 16610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov float b; 16620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 16630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct SSS6 { 16640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int a; 16650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov double b; 16660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 16670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct SSS7 { 16680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S8 b; 16690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int a; 16700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 16710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct SSS8 { 16720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S2 b; 16730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S8 a; 16740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 16750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 16760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, IntStruct3) { 16770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestReturnStruct<SSS1>(); 16780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestReturnStruct<SSS2>(); 16790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestReturnStruct<SSS3>(); 16800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestReturnStruct<SSS4>(); 16810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestReturnStruct<SSS5>(); 16820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestReturnStruct<SSS6>(); 16830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestReturnStruct<SSS7>(); 16840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestReturnStruct<SSS8>(); 16850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 16860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 16870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct LongStruct { 16880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U1 a1, b1; 16890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U2 a2, b2; 16900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U4 a4, b4; 16910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U8 a8, b8; 16920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 16930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 16940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE LongStruct ReturnLongStruct1() { 16950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov LongStruct res; 16960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&res, sizeof(res)); 16970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.a1 = res.a2 = res.a4 = res.a8 = 111; 16980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // leaves b1, .., b8 poisoned. 16990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 17000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 17010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 17020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE LongStruct ReturnLongStruct2() { 17030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov LongStruct res; 17040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&res, sizeof(res)); 17050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.b1 = res.b2 = res.b4 = res.b8 = 111; 17060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // leaves a1, .., a8 poisoned. 17070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 17080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 17090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 17100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, LongStruct) { 17110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov LongStruct s1 = ReturnLongStruct1(); 17120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_print_shadow(&s1, sizeof(s1)); 171311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s1.a1); 171411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s1.a2); 171511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s1.a4); 171611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s1.a8); 17170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 171811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s1.b1); 171911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s1.b2); 172011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s1.b4); 172111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s1.b8); 17220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 17230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov LongStruct s2 = ReturnLongStruct2(); 17240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_print_shadow(&s2, sizeof(s2)); 172511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s2.b1); 172611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s2.b2); 172711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s2.b4); 172811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s2.b8); 17290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 173011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s2.a1); 173111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s2.a2); 173211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s2.a4); 173311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s2.a8); 17340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 17350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 17360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, getrlimit) { 17370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov struct rlimit limit; 17380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&limit, sizeof(limit)); 17390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int result = getrlimit(RLIMIT_DATA, &limit); 17400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(result == 0); 17412bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov EXPECT_NOT_POISONED(limit.rlim_cur); 17422bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov EXPECT_NOT_POISONED(limit.rlim_max); 17430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 17440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1745e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy StepanovTEST(MemorySanitizer, getrusage) { 1746e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov struct rusage usage; 1747e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov __msan_poison(&usage, sizeof(usage)); 1748e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov int result = getrusage(RUSAGE_SELF, &usage); 1749e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov assert(result == 0); 175011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_utime.tv_sec); 175111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_utime.tv_usec); 175211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_stime.tv_sec); 175311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_stime.tv_usec); 175411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_maxrss); 175511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_minflt); 175611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_majflt); 175711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_inblock); 175811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_oublock); 175911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_nvcsw); 176011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_nivcsw); 1761e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov} 1762e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov 17631aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov#ifdef __GLIBC__ 1764e6c62f2271c25567ac0bbbe68c7323b49330373cEvgeniy Stepanovextern char *program_invocation_name; 17651aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov#else // __GLIBC__ 17661aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov# error "TODO: port this" 17671aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov#endif 17681aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov 1769e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanovstatic void dladdr_testfn() {} 1770e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov 1771e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy StepanovTEST(MemorySanitizer, dladdr) { 1772e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov Dl_info info; 1773e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov __msan_poison(&info, sizeof(info)); 1774e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov int result = dladdr((const void*)dladdr_testfn, &info); 1775e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov assert(result != 0); 177611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((unsigned long)info.dli_fname); 1777e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov if (info.dli_fname) 177811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(info.dli_fname)); 177911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((unsigned long)info.dli_fbase); 178011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((unsigned long)info.dli_sname); 1781e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov if (info.dli_sname) 178211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(info.dli_sname)); 178311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((unsigned long)info.dli_saddr); 1784e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov} 1785e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov 17862bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanovstatic int dl_phdr_callback(struct dl_phdr_info *info, size_t size, void *data) { 17872bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov (*(int *)data)++; 17882bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov EXPECT_NOT_POISONED(info->dlpi_addr); 17892bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov EXPECT_NOT_POISONED(strlen(info->dlpi_name)); 17902bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov EXPECT_NOT_POISONED(info->dlpi_phnum); 17912bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov for (int i = 0; i < info->dlpi_phnum; ++i) 17922bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov EXPECT_NOT_POISONED(info->dlpi_phdr[i]); 17932bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov return 0; 17942bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov} 17952bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov 1796263800ba59720f9be69735b4dad625fca92430bcEvgeniy Stepanov#ifndef MSAN_TEST_DISABLE_DLOPEN 1797263800ba59720f9be69735b4dad625fca92430bcEvgeniy Stepanov 1798263800ba59720f9be69735b4dad625fca92430bcEvgeniy Stepanov// Compute the path to our loadable DSO. We assume it's in the same 1799263800ba59720f9be69735b4dad625fca92430bcEvgeniy Stepanov// directory. Only use string routines that we intercept so far to do this. 1800263800ba59720f9be69735b4dad625fca92430bcEvgeniy Stepanovstatic int PathToLoadable(char *buf, size_t sz) { 1801263800ba59720f9be69735b4dad625fca92430bcEvgeniy Stepanov const char *basename = "libmsan_loadable.x86_64.so"; 1802263800ba59720f9be69735b4dad625fca92430bcEvgeniy Stepanov char *argv0 = program_invocation_name; 1803263800ba59720f9be69735b4dad625fca92430bcEvgeniy Stepanov char *last_slash = strrchr(argv0, '/'); 1804263800ba59720f9be69735b4dad625fca92430bcEvgeniy Stepanov assert(last_slash); 1805263800ba59720f9be69735b4dad625fca92430bcEvgeniy Stepanov int res = 1806263800ba59720f9be69735b4dad625fca92430bcEvgeniy Stepanov snprintf(buf, sz, "%.*s/%s", int(last_slash - argv0), argv0, basename); 1807263800ba59720f9be69735b4dad625fca92430bcEvgeniy Stepanov return res < sz ? 0 : res; 1808263800ba59720f9be69735b4dad625fca92430bcEvgeniy Stepanov} 1809263800ba59720f9be69735b4dad625fca92430bcEvgeniy Stepanov 18102bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy StepanovTEST(MemorySanitizer, dl_iterate_phdr) { 18111aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov char path[4096]; 18121aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov int res = PathToLoadable(path, sizeof(path)); 18131aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov assert(!res); 18141aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov 18151aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov // Having at least one dlopen'ed library in the process makes this more 18161aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov // entertaining. 18171aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov void *lib = dlopen(path, RTLD_LAZY); 18181aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov ASSERT_NE((void*)0, lib); 18191aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov 18202bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov int count = 0; 18212bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov int result = dl_iterate_phdr(dl_phdr_callback, &count); 18222bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov assert(count > 0); 18231aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov 18241aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov dlclose(lib); 18252bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov} 18262bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov 18270f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner 18280f92deb81207c80481ff0257fbaba640fe669633Reid KlecknerTEST(MemorySanitizer, dlopen) { 18291aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov char path[4096]; 18301aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov int res = PathToLoadable(path, sizeof(path)); 18311aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov assert(!res); 18320f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner 18330f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner // We need to clear shadow for globals when doing dlopen. In order to test 18340f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner // this, we have to poison the shadow for the DSO before we load it. In 18350f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner // general this is difficult, but the loader tends to reload things in the 18360f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner // same place, so we open, close, and then reopen. The global should always 18370f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner // start out clean after dlopen. 18380f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner for (int i = 0; i < 2; i++) { 18390f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner void *lib = dlopen(path, RTLD_LAZY); 18400f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner if (lib == NULL) { 18410f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner printf("dlerror: %s\n", dlerror()); 18420f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner assert(lib != NULL); 18430f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner } 18440f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner void **(*get_dso_global)() = (void **(*)())dlsym(lib, "get_dso_global"); 18450f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner assert(get_dso_global); 18460f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner void **dso_global = get_dso_global(); 18470f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner EXPECT_NOT_POISONED(*dso_global); 18480f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner __msan_poison(dso_global, sizeof(*dso_global)); 18490f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner EXPECT_POISONED(*dso_global); 18500f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner dlclose(lib); 18510f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner } 18520f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner} 18536c503b9c7b0ad08fba74dbed309447d75de9a157Evgeniy Stepanov 18546c503b9c7b0ad08fba74dbed309447d75de9a157Evgeniy Stepanov// Regression test for a crash in dlopen() interceptor. 18556c503b9c7b0ad08fba74dbed309447d75de9a157Evgeniy StepanovTEST(MemorySanitizer, dlopenFailed) { 18566c503b9c7b0ad08fba74dbed309447d75de9a157Evgeniy Stepanov const char *path = "/libmsan_loadable_does_not_exist.x86_64.so"; 18576c503b9c7b0ad08fba74dbed309447d75de9a157Evgeniy Stepanov void *lib = dlopen(path, RTLD_LAZY); 18586c503b9c7b0ad08fba74dbed309447d75de9a157Evgeniy Stepanov ASSERT_EQ(0, lib); 18596c503b9c7b0ad08fba74dbed309447d75de9a157Evgeniy Stepanov} 18600f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner 1861263800ba59720f9be69735b4dad625fca92430bcEvgeniy Stepanov#endif // MSAN_TEST_DISABLE_DLOPEN 1862263800ba59720f9be69735b4dad625fca92430bcEvgeniy Stepanov 1863996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy StepanovTEST(MemorySanitizer, scanf) { 1864996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov const char *input = "42 hello"; 1865996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov int* d = new int; 1866996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov char* s = new char[7]; 1867996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov int res = sscanf(input, "%d %5s", d, s); 1868996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov printf("res %d\n", res); 1869996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov assert(res == 2); 187011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*d); 187111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s[0]); 187211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s[1]); 187311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s[2]); 187411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s[3]); 187511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s[4]); 187611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s[5]); 187711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s[6]); 1878996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov delete s; 1879996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov delete d; 1880996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov} 1881996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov 1882b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanovstatic void *SimpleThread_threadfn(void* data) { 18830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return new int; 18840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 18850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 18860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, SimpleThread) { 18870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov pthread_t t; 1888b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov void *p; 18890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = pthread_create(&t, NULL, SimpleThread_threadfn, NULL); 18900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(!res); 1891e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(t); 18920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res = pthread_join(t, &p); 18930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(!res); 18940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) // FIXME: intercept pthread_join (?). 18950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_unpoison(&p, sizeof(p)); 18960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete (int*)p; 18970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 18980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1899b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanovstatic void *SmallStackThread_threadfn(void* data) { 190010fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov return 0; 190110fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov} 190210fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov 190310fd3227546d17c7411241a45ebc143b2031c78dEvgeniy StepanovTEST(MemorySanitizer, SmallStackThread) { 190410fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov pthread_attr_t attr; 190510fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov pthread_t t; 1906b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov void *p; 190710fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov int res; 190810fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov res = pthread_attr_init(&attr); 190910fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov ASSERT_EQ(0, res); 191010fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov res = pthread_attr_setstacksize(&attr, 64 * 1024); 191110fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov ASSERT_EQ(0, res); 1912b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov res = pthread_create(&t, &attr, SmallStackThread_threadfn, NULL); 191310fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov ASSERT_EQ(0, res); 191410fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov res = pthread_join(t, &p); 191510fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov ASSERT_EQ(0, res); 191610fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov res = pthread_attr_destroy(&attr); 1917b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov ASSERT_EQ(0, res); 1918b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov} 1919b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov 1920b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy StepanovTEST(MemorySanitizer, PreAllocatedStackThread) { 1921b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov pthread_attr_t attr; 1922b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov pthread_t t; 1923b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov int res; 1924b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov res = pthread_attr_init(&attr); 1925b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov ASSERT_EQ(0, res); 1926b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov void *stack; 1927b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov const size_t kStackSize = 64 * 1024; 1928b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov res = posix_memalign(&stack, 4096, kStackSize); 1929b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov ASSERT_EQ(0, res); 1930b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov res = pthread_attr_setstack(&attr, stack, kStackSize); 1931b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov ASSERT_EQ(0, res); 1932b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov // A small self-allocated stack can not be extended by the tool. 1933b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov // In this case pthread_create is expected to fail. 1934b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov res = pthread_create(&t, &attr, SmallStackThread_threadfn, NULL); 1935b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov EXPECT_NE(0, res); 1936b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov res = pthread_attr_destroy(&attr); 193710fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov ASSERT_EQ(0, res); 193810fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov} 193910fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov 194056d3472104dd9fec6578e02f4895f3254e038e8eEvgeniy StepanovTEST(MemorySanitizer, pthread_getschedparam) { 194156d3472104dd9fec6578e02f4895f3254e038e8eEvgeniy Stepanov int policy; 194256d3472104dd9fec6578e02f4895f3254e038e8eEvgeniy Stepanov struct sched_param param; 194356d3472104dd9fec6578e02f4895f3254e038e8eEvgeniy Stepanov int res = pthread_getschedparam(pthread_self(), &policy, ¶m); 194456d3472104dd9fec6578e02f4895f3254e038e8eEvgeniy Stepanov ASSERT_EQ(0, res); 194556d3472104dd9fec6578e02f4895f3254e038e8eEvgeniy Stepanov EXPECT_NOT_POISONED(policy); 194656d3472104dd9fec6578e02f4895f3254e038e8eEvgeniy Stepanov EXPECT_NOT_POISONED(param.sched_priority); 194756d3472104dd9fec6578e02f4895f3254e038e8eEvgeniy Stepanov} 194856d3472104dd9fec6578e02f4895f3254e038e8eEvgeniy Stepanov 1949b6c8e47a72d526232f4d3e3b36232f30f8818733Evgeniy StepanovTEST(MemorySanitizer, posix_memalign) { 1950b6c8e47a72d526232f4d3e3b36232f30f8818733Evgeniy Stepanov void *p; 1951b6c8e47a72d526232f4d3e3b36232f30f8818733Evgeniy Stepanov EXPECT_POISONED(p); 1952b6c8e47a72d526232f4d3e3b36232f30f8818733Evgeniy Stepanov int res = posix_memalign(&p, 4096, 13); 1953b6c8e47a72d526232f4d3e3b36232f30f8818733Evgeniy Stepanov ASSERT_EQ(0, res); 1954b6c8e47a72d526232f4d3e3b36232f30f8818733Evgeniy Stepanov EXPECT_NOT_POISONED(p); 1955b6c8e47a72d526232f4d3e3b36232f30f8818733Evgeniy Stepanov free(p); 1956b6c8e47a72d526232f4d3e3b36232f30f8818733Evgeniy Stepanov} 1957b6c8e47a72d526232f4d3e3b36232f30f8818733Evgeniy Stepanov 19589530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy StepanovTEST(MemorySanitizer, inet_pton) { 19599530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov const char *s = "1:0:0:0:0:0:0:8"; 19609530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov unsigned char buf[sizeof(struct in6_addr)]; 19619530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov int res = inet_pton(AF_INET6, s, buf); 19629530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov ASSERT_EQ(1, res); 19639530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov EXPECT_NOT_POISONED(buf[0]); 19649530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov EXPECT_NOT_POISONED(buf[sizeof(struct in6_addr) - 1]); 19659530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov 19669530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov char s_out[INET6_ADDRSTRLEN]; 19679530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov EXPECT_POISONED(s_out[3]); 19689530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov const char *q = inet_ntop(AF_INET6, buf, s_out, INET6_ADDRSTRLEN); 19699530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov ASSERT_NE((void*)0, q); 19709530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov EXPECT_NOT_POISONED(s_out[3]); 19719530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov} 19729530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov 19739d60087654f89e3452841350d9eca97644edca9dEvgeniy StepanovTEST(MemorySanitizer, inet_aton) { 19749d60087654f89e3452841350d9eca97644edca9dEvgeniy Stepanov const char *s = "127.0.0.1"; 19759d60087654f89e3452841350d9eca97644edca9dEvgeniy Stepanov struct in_addr in[2]; 19769d60087654f89e3452841350d9eca97644edca9dEvgeniy Stepanov int res = inet_aton(s, in); 19779d60087654f89e3452841350d9eca97644edca9dEvgeniy Stepanov ASSERT_NE(0, res); 19789d60087654f89e3452841350d9eca97644edca9dEvgeniy Stepanov EXPECT_NOT_POISONED(in[0]); 19799d60087654f89e3452841350d9eca97644edca9dEvgeniy Stepanov EXPECT_POISONED(*(char *)(in + 1)); 19809d60087654f89e3452841350d9eca97644edca9dEvgeniy Stepanov} 19819d60087654f89e3452841350d9eca97644edca9dEvgeniy Stepanov 19820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, uname) { 19830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov struct utsname u; 19840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = uname(&u); 19850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(!res); 198611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(u.sysname)); 198711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(u.nodename)); 198811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(u.release)); 198911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(u.version)); 199011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(u.machine)); 19910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 19920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 199395d058800ebe11a9fda03b10455500aa4a5b3edbEvgeniy StepanovTEST(MemorySanitizer, gethostname) { 199495d058800ebe11a9fda03b10455500aa4a5b3edbEvgeniy Stepanov char buf[100]; 199595d058800ebe11a9fda03b10455500aa4a5b3edbEvgeniy Stepanov int res = gethostname(buf, 100); 199695d058800ebe11a9fda03b10455500aa4a5b3edbEvgeniy Stepanov assert(!res); 199795d058800ebe11a9fda03b10455500aa4a5b3edbEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(buf)); 199895d058800ebe11a9fda03b10455500aa4a5b3edbEvgeniy Stepanov} 199995d058800ebe11a9fda03b10455500aa4a5b3edbEvgeniy Stepanov 2000359d7fc7daf099ab8ef9a860564542d2581544aaEvgeniy StepanovTEST(MemorySanitizer, sysinfo) { 2001359d7fc7daf099ab8ef9a860564542d2581544aaEvgeniy Stepanov struct sysinfo info; 2002359d7fc7daf099ab8ef9a860564542d2581544aaEvgeniy Stepanov int res = sysinfo(&info); 2003359d7fc7daf099ab8ef9a860564542d2581544aaEvgeniy Stepanov assert(!res); 2004359d7fc7daf099ab8ef9a860564542d2581544aaEvgeniy Stepanov EXPECT_NOT_POISONED(info); 2005359d7fc7daf099ab8ef9a860564542d2581544aaEvgeniy Stepanov} 2006359d7fc7daf099ab8ef9a860564542d2581544aaEvgeniy Stepanov 2007e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy StepanovTEST(MemorySanitizer, getpwuid) { 2008e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov struct passwd *p = getpwuid(0); // root 2009e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov assert(p); 2010e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(p->pw_name); 2011e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov assert(p->pw_name); 2012e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(p->pw_name[0]); 2013e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(p->pw_uid); 2014e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov assert(p->pw_uid == 0); 2015e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov} 2016e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov 2017e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy StepanovTEST(MemorySanitizer, getpwnam_r) { 2018e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov struct passwd pwd; 2019e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov struct passwd *pwdres; 2020e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov char buf[10000]; 2021e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov int res = getpwnam_r("root", &pwd, buf, sizeof(buf), &pwdres); 2022e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov assert(!res); 2023e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(pwd.pw_name); 2024e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov assert(pwd.pw_name); 2025e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(pwd.pw_name[0]); 2026e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(pwd.pw_uid); 2027e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov assert(pwd.pw_uid == 0); 2028e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov} 2029e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov 2030e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy StepanovTEST(MemorySanitizer, getpwnam_r_positive) { 2031e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov struct passwd pwd; 2032e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov struct passwd *pwdres; 2033e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov char s[5]; 2034e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov strncpy(s, "abcd", 5); 2035e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov __msan_poison(s, 5); 2036e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov char buf[10000]; 2037e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov int res; 2038e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_UMR(res = getpwnam_r(s, &pwd, buf, sizeof(buf), &pwdres)); 2039e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov} 2040e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov 2041103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy StepanovTEST(MemorySanitizer, getgrnam_r) { 2042103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov struct group grp; 2043103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov struct group *grpres; 2044103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov char buf[10000]; 2045103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov int res = getgrnam_r("root", &grp, buf, sizeof(buf), &grpres); 2046103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov assert(!res); 2047103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov EXPECT_NOT_POISONED(grp.gr_name); 2048103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov assert(grp.gr_name); 2049103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov EXPECT_NOT_POISONED(grp.gr_name[0]); 2050103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov EXPECT_NOT_POISONED(grp.gr_gid); 2051103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov} 2052103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov 20530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> 205411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovstatic bool applySlt(T value, T shadow) { 20550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_partial_poison(&value, &shadow, sizeof(T)); 20560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov volatile bool zzz = true; 20570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // This "|| zzz" trick somehow makes LLVM emit "icmp slt" instead of 20580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // a shift-and-trunc to get at the highest bit. 205911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov volatile bool v = value < 0 || zzz; 206011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov return v; 20610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 20620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 20630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, SignedCompareWithZero) { 206411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applySlt<S4>(0xF, 0xF)); 206511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applySlt<S4>(0xF, 0xFF)); 206611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applySlt<S4>(0xF, 0xFFFFFF)); 206711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applySlt<S4>(0xF, 0x7FFFFFF)); 206811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR(applySlt<S4>(0xF, 0x80FFFFFF)); 206911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR(applySlt<S4>(0xF, 0xFFFFFFFF)); 20700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 20710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 20729a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanovtemplate <class T, class S> 20739a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanovstatic T poisoned(T Va, S Sa) { 20749a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov char SIZE_CHECK1[(ssize_t)sizeof(T) - (ssize_t)sizeof(S)]; 20759a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov char SIZE_CHECK2[(ssize_t)sizeof(S) - (ssize_t)sizeof(T)]; 20769a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov T a; 20779a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov a = Va; 20789a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov __msan_partial_poison(&a, &Sa, sizeof(T)); 20799a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov return a; 20809a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov} 20819a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov 20829a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy StepanovTEST(MemorySanitizer, ICmpRelational) { 20839a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0, 0) < poisoned(0, 0)); 20849a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0U, 0) < poisoned(0U, 0)); 20859a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0LL, 0LLU) < poisoned(0LL, 0LLU)); 20869a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0LLU, 0LLU) < poisoned(0LLU, 0LLU)); 20879a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(0xFF, 0xFF) < poisoned(0xFF, 0xFF)); 20889a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(0xFFFFFFFFU, 0xFFFFFFFFU) < 20899a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov poisoned(0xFFFFFFFFU, 0xFFFFFFFFU)); 20909a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(-1, 0xFFFFFFFFU) < 20919a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov poisoned(-1, 0xFFFFFFFFU)); 20929a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov 20939a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0, 0) <= poisoned(0, 0)); 20949a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0U, 0) <= poisoned(0U, 0)); 20959a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0LL, 0LLU) <= poisoned(0LL, 0LLU)); 20969a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0LLU, 0LLU) <= poisoned(0LLU, 0LLU)); 20979a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(0xFF, 0xFF) <= poisoned(0xFF, 0xFF)); 20989a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(0xFFFFFFFFU, 0xFFFFFFFFU) <= 20999a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov poisoned(0xFFFFFFFFU, 0xFFFFFFFFU)); 21009a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(-1, 0xFFFFFFFFU) <= 21019a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov poisoned(-1, 0xFFFFFFFFU)); 21029a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov 21039a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0, 0) > poisoned(0, 0)); 21049a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0U, 0) > poisoned(0U, 0)); 21059a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0LL, 0LLU) > poisoned(0LL, 0LLU)); 21069a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0LLU, 0LLU) > poisoned(0LLU, 0LLU)); 21079a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(0xFF, 0xFF) > poisoned(0xFF, 0xFF)); 21089a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(0xFFFFFFFFU, 0xFFFFFFFFU) > 21099a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov poisoned(0xFFFFFFFFU, 0xFFFFFFFFU)); 21109a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(-1, 0xFFFFFFFFU) > 21119a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov poisoned(-1, 0xFFFFFFFFU)); 21129a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov 21139a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0, 0) >= poisoned(0, 0)); 21149a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0U, 0) >= poisoned(0U, 0)); 21159a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0LL, 0LLU) >= poisoned(0LL, 0LLU)); 21169a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0LLU, 0LLU) >= poisoned(0LLU, 0LLU)); 21179a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(0xFF, 0xFF) >= poisoned(0xFF, 0xFF)); 21189a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(0xFFFFFFFFU, 0xFFFFFFFFU) >= 21199a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov poisoned(0xFFFFFFFFU, 0xFFFFFFFFU)); 21209a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(-1, 0xFFFFFFFFU) >= 21219a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov poisoned(-1, 0xFFFFFFFFU)); 21229a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov 21239a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(6, 0xF) > poisoned(7, 0)); 21249a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(0xF, 0xF) > poisoned(7, 0)); 21259a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov 21269a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(-1, 0x80000000U) >= poisoned(-1, 0U)); 21279a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov} 21289a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov 21299a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov#if MSAN_HAS_M128 21309a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy StepanovTEST(MemorySanitizer, ICmpVectorRelational) { 2131b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov EXPECT_NOT_POISONED( 2132b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov _mm_cmplt_epi16(poisoned(_mm_set1_epi16(0), _mm_set1_epi16(0)), 2133b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov poisoned(_mm_set1_epi16(0), _mm_set1_epi16(0)))); 2134b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov EXPECT_NOT_POISONED( 2135b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov _mm_cmplt_epi16(poisoned(_mm_set1_epi32(0), _mm_set1_epi32(0)), 2136b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov poisoned(_mm_set1_epi32(0), _mm_set1_epi32(0)))); 2137b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov EXPECT_POISONED( 2138b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov _mm_cmplt_epi16(poisoned(_mm_set1_epi16(0), _mm_set1_epi16(0xFFFF)), 2139b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov poisoned(_mm_set1_epi16(0), _mm_set1_epi16(0xFFFF)))); 2140b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov EXPECT_POISONED(_mm_cmpgt_epi16(poisoned(_mm_set1_epi16(6), _mm_set1_epi16(0xF)), 2141b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov poisoned(_mm_set1_epi16(7), _mm_set1_epi16(0)))); 21429a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov} 21439a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov#endif 21449a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov 21452efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov// Volatile bitfield store is implemented as load-mask-store 21462efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov// Test that we don't warn on the store of (uninitialized) padding. 21472efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanovstruct VolatileBitfieldStruct { 21482efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov volatile unsigned x : 1; 21492efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov unsigned y : 1; 21502efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov}; 21512efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov 21522efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy StepanovTEST(MemorySanitizer, VolatileBitfield) { 21532efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov VolatileBitfieldStruct *S = new VolatileBitfieldStruct; 21542efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov S->x = 1; 215502f4a942bff84f1266571740456dd9baa230d87bEvgeniy Stepanov EXPECT_NOT_POISONED((unsigned)S->x); 215602f4a942bff84f1266571740456dd9baa230d87bEvgeniy Stepanov EXPECT_POISONED((unsigned)S->y); 21572efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov} 21582efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov 2159f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy StepanovTEST(MemorySanitizer, UnalignedLoad) { 2160f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov char x[32]; 2161f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov memset(x + 8, 0, 16); 2162f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_POISONED(__sanitizer_unaligned_load16(x+6)); 2163f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_POISONED(__sanitizer_unaligned_load16(x+7)); 2164f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_NOT_POISONED(__sanitizer_unaligned_load16(x+8)); 2165f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_NOT_POISONED(__sanitizer_unaligned_load16(x+9)); 2166f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_NOT_POISONED(__sanitizer_unaligned_load16(x+22)); 2167f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_POISONED(__sanitizer_unaligned_load16(x+23)); 2168f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_POISONED(__sanitizer_unaligned_load16(x+24)); 2169f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov 2170f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_POISONED(__sanitizer_unaligned_load32(x+4)); 2171f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_POISONED(__sanitizer_unaligned_load32(x+7)); 2172f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_NOT_POISONED(__sanitizer_unaligned_load32(x+8)); 2173f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_NOT_POISONED(__sanitizer_unaligned_load32(x+9)); 2174f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_NOT_POISONED(__sanitizer_unaligned_load32(x+20)); 2175f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_POISONED(__sanitizer_unaligned_load32(x+21)); 2176f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_POISONED(__sanitizer_unaligned_load32(x+24)); 2177f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov 2178f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_POISONED(__sanitizer_unaligned_load64(x)); 2179f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_POISONED(__sanitizer_unaligned_load64(x+1)); 2180f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_POISONED(__sanitizer_unaligned_load64(x+7)); 2181f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_NOT_POISONED(__sanitizer_unaligned_load64(x+8)); 2182f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_NOT_POISONED(__sanitizer_unaligned_load64(x+9)); 2183f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_NOT_POISONED(__sanitizer_unaligned_load64(x+16)); 2184f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_POISONED(__sanitizer_unaligned_load64(x+17)); 2185f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_POISONED(__sanitizer_unaligned_load64(x+21)); 2186f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_POISONED(__sanitizer_unaligned_load64(x+24)); 2187f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov} 2188f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov 2189f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy StepanovTEST(MemorySanitizer, UnalignedStore16) { 2190f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov char x[5]; 2191f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov U2 y = 0; 2192f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov __msan_poison(&y, 1); 2193f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov __sanitizer_unaligned_store16(x + 1, y); 2194f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_POISONED(x[0]); 2195f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_POISONED(x[1]); 2196f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_NOT_POISONED(x[2]); 2197f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_POISONED(x[3]); 2198f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_POISONED(x[4]); 2199f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov} 2200f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov 2201f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy StepanovTEST(MemorySanitizer, UnalignedStore32) { 2202f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov char x[8]; 2203f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov U4 y4 = 0; 2204f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov __msan_poison(&y4, 2); 2205f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov __sanitizer_unaligned_store32(x+3, y4); 2206f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_POISONED(x[0]); 2207f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_POISONED(x[1]); 2208f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_POISONED(x[2]); 2209f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_POISONED(x[3]); 2210f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_POISONED(x[4]); 2211f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_NOT_POISONED(x[5]); 2212f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_NOT_POISONED(x[6]); 2213f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_POISONED(x[7]); 2214f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov} 2215f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov 2216f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy StepanovTEST(MemorySanitizer, UnalignedStore64) { 2217f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov char x[16]; 2218f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov U8 y = 0; 2219f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov __msan_poison(&y, 3); 2220f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov __msan_poison(((char *)&y) + sizeof(y) - 2, 1); 2221f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov __sanitizer_unaligned_store64(x+3, y); 2222f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_POISONED(x[0]); 2223f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_POISONED(x[1]); 2224f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_POISONED(x[2]); 2225f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_POISONED(x[3]); 2226f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_POISONED(x[4]); 2227f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_POISONED(x[5]); 2228f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_NOT_POISONED(x[6]); 2229f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_NOT_POISONED(x[7]); 2230f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_NOT_POISONED(x[8]); 2231f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_POISONED(x[9]); 2232f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_NOT_POISONED(x[10]); 2233f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov EXPECT_POISONED(x[11]); 2234f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov} 2235f43f6026dc9374e657fb6300840b18ead6347b74Evgeniy Stepanov 22360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerDr, StoreInDSOTest) { 22370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) return; 22380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* s = new char[10]; 22390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov dso_memfill(s, 9); 224011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s[5]); 224111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s[9]); 22420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 22430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 22440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovint return_poisoned_int() { 22450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return ReturnPoisoned<U8>(); 22460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 22470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 22480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerDr, ReturnFromDSOTest) { 22490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) return; 225011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(dso_callfn(return_poisoned_int)); 22510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 22520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 22530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE int TrashParamTLS(long long x, long long y, long long z) { //NOLINT 225411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x); 225511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(y); 225611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(z); 22570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return 0; 22580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 22590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 22600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic int CheckParamTLS(long long x, long long y, long long z) { //NOLINT 226111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x); 226211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(y); 226311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(z); 22640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return 0; 22650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 22660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 22670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerDr, CallFromDSOTest) { 22680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) return; 22690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S8* x = GetPoisoned<S8>(); 22700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S8* y = GetPoisoned<S8>(); 22710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S8* z = GetPoisoned<S8>(); 227211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(TrashParamTLS(*x, *y, *z)); 227311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(dso_callfn1(CheckParamTLS)); 22740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 22750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 22760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void StackStoreInDSOFn(int* x, int* y) { 227711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*x); 227811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*y); 22790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 22800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 22810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerDr, StackStoreInDSOTest) { 22820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) return; 22830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov dso_stack_store(StackStoreInDSOFn, 1); 22840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 22850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 22860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, SetGet) { 22870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(TrackingOrigins(), __msan_get_track_origins()); 22880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 22890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int x; 22900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(&x, sizeof(x), 1234); 22910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(1234, __msan_get_origin(&x)); 22920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(&x, sizeof(x), 5678); 22930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(5678, __msan_get_origin(&x)); 22940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(&x, sizeof(x), 0); 22950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(0, __msan_get_origin(&x)); 22960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 22970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 22980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovnamespace { 22990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct S { 23000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U4 dummy; 23010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U2 a; 23020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U2 b; 23030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 23040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 23050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// http://code.google.com/p/memory-sanitizer/issues/detail?id=6 23060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, DISABLED_InitializedStoreDoesNotChangeOrigin) { 23070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 23080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 23090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S s; 2310250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 origin = rand(); // NOLINT 23110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov s.a = *GetPoisonedO<U2>(0, origin); 23120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(origin, __msan_get_origin(&s.a)); 23130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(origin, __msan_get_origin(&s.b)); 23140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 23150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov s.b = 42; 23160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(origin, __msan_get_origin(&s.a)); 23170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(origin, __msan_get_origin(&s.b)); 23180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 23190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} // namespace 23200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 23210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T, class BinaryOp> 23220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovINLINE 23230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovvoid BinaryOpOriginTest(BinaryOp op) { 2324250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 ox = rand(); //NOLINT 2325250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 oy = rand(); //NOLINT 23260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *x = GetPoisonedO<T>(0, ox, 0); 23270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *y = GetPoisonedO<T>(1, oy, 0); 23280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *z = GetPoisonedO<T>(2, 0, 0); 23290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 23300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *z = op(*x, *y); 2331250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 origin = __msan_get_origin(z); 233211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*z, origin); 23330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(true, origin == ox || origin == oy); 23340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 23350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // y is poisoned, x is not. 23360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *x = 10101; 23370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *y = *GetPoisonedO<T>(1, oy); 233812c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(x); 23390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(z, sizeof(*z), 0); 23400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *z = op(*x, *y); 234111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*z, oy); 23420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(__msan_get_origin(z), oy); 23430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 23440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // x is poisoned, y is not. 23450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *x = *GetPoisonedO<T>(0, ox); 23460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *y = 10101010; 234712c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(y); 23480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(z, sizeof(*z), 0); 23490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *z = op(*x, *y); 235011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*z, ox); 23510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(__msan_get_origin(z), ox); 23520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 23530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 23540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> INLINE T XOR(const T &a, const T&b) { return a ^ b; } 23550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> INLINE T ADD(const T &a, const T&b) { return a + b; } 23560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> INLINE T SUB(const T &a, const T&b) { return a - b; } 23570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> INLINE T MUL(const T &a, const T&b) { return a * b; } 23580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> INLINE T AND(const T &a, const T&b) { return a & b; } 23590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> INLINE T OR (const T &a, const T&b) { return a | b; } 23600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 23610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, BinaryOp) { 23620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 23630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<S8>(XOR<S8>); 23640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<U8>(ADD<U8>); 23650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<S4>(SUB<S4>); 23660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<S4>(MUL<S4>); 23670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<U4>(OR<U4>); 23680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<U4>(AND<U4>); 23690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<double>(ADD<U4>); 23700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<float>(ADD<S4>); 23710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<double>(ADD<double>); 23720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<float>(ADD<double>); 23730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 23740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 23750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, Unary) { 23760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 237711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S8>(0, __LINE__), __LINE__); 237811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S8>(0, __LINE__), __LINE__); 237911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S8>(0, __LINE__), __LINE__); 238011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S8>(0, __LINE__), __LINE__); 23810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 238211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 238311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 238411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 238511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 23860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 238711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<U4>(0, __LINE__), __LINE__); 238811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<U4>(0, __LINE__), __LINE__); 238911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<U4>(0, __LINE__), __LINE__); 239011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<U4>(0, __LINE__), __LINE__); 23910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 239211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 239311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 239411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 239511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 23960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 239711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O((void*)*GetPoisonedO<S8>(0, __LINE__), __LINE__); 239811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O((U8)*GetPoisonedO<void*>(0, __LINE__), __LINE__); 23990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 24000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 24010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, EQ) { 24020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 240311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__) <= 11, __LINE__); 240411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__) == 11, __LINE__); 240511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<float>(0, __LINE__) == 1.1, __LINE__); 24060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 24070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 24080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, DIV) { 24090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 241011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<U8>(0, __LINE__) / 100, __LINE__); 241111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov unsigned o = __LINE__; 241211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR_O(volatile unsigned y = 100 / *GetPoisonedO<S4>(0, o, 1), o); 24130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 24140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 24150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, SHIFT) { 24160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 241711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<U8>(0, __LINE__) >> 10, __LINE__); 241811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S8>(0, __LINE__) >> 10, __LINE__); 241911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S8>(0, __LINE__) << 10, __LINE__); 242011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(10U << *GetPoisonedO<U8>(0, __LINE__), __LINE__); 242111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(-10 >> *GetPoisonedO<S8>(0, __LINE__), __LINE__); 242211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(-10 << *GetPoisonedO<S8>(0, __LINE__), __LINE__); 24230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 24240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 24250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T, int N> 24260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovvoid MemCpyTest() { 24270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int ox = __LINE__; 24280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *x = new T[N]; 24290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *y = new T[N]; 24300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *z = new T[N]; 24310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(x, N * sizeof(T)); 24320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(x, N * sizeof(T), ox); 24330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(y, N * sizeof(T), 777777); 24340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(z, N * sizeof(T), 888888); 243511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x); 243611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov memcpy(y, x, N * sizeof(T)); 243711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(y[0], ox); 243811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(y[N/2], ox); 243911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(y[N-1], ox); 244011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x); 244111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov memmove(z, x, N * sizeof(T)); 244211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(z[0], ox); 244311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(z[N/2], ox); 244411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(z[N-1], ox); 24450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 24460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 24470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, LargeMemCpy) { 24480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 24490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov MemCpyTest<U1, 10000>(); 24500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov MemCpyTest<U8, 10000>(); 24510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 24520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 24530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, SmallMemCpy) { 24540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 24550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov MemCpyTest<U8, 1>(); 24560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov MemCpyTest<U8, 2>(); 24570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov MemCpyTest<U8, 3>(); 24580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 24590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 24600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, Select) { 24610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 246211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(g_one ? 1 : *GetPoisonedO<S4>(0, __LINE__)); 246311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 24640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 x; 246512c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&x); 24660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x = g_1 ? *GetPoisonedO<S4>(0, __LINE__) : 0; 24670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 246811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(g_1 ? *GetPoisonedO<S4>(0, __LINE__) : 1, __LINE__); 246911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(g_0 ? 1 : *GetPoisonedO<S4>(0, __LINE__), __LINE__); 24700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 24710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 24720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovextern "C" 247311929000ec2919192b3be457f5a44c71ed55215eEvgeniy StepanovNOINLINE char AllocaTO() { 24740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int ar[100]; 247512c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(ar); 247611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov return ar[10]; 24770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // fprintf(stderr, "Descr: %s\n", 24780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // __msan_get_origin_descr_if_stack(__msan_get_origin_tls())); 24790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 24800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 24810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, Alloca) { 24820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 248311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_S(AllocaTO(), "ar@AllocaTO"); 248411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_S(AllocaTO(), "ar@AllocaTO"); 248511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_S(AllocaTO(), "ar@AllocaTO"); 248611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_S(AllocaTO(), "ar@AllocaTO"); 24870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 24880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 24890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// FIXME: replace with a lit-like test. 24900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, DISABLED_AllocaDeath) { 24910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 249211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_DEATH(AllocaTO(), "ORIGIN: stack allocation: ar@AllocaTO"); 24930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 24940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2495250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy StepanovNOINLINE int RetvalOriginTest(U4 origin) { 24960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int *a = new int; 249712c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(a); 24980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(a, sizeof(*a), origin); 24990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = *a; 25000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete a; 25010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 25020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 25030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 25040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, Retval) { 25050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 250611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(RetvalOriginTest(__LINE__), __LINE__); 25070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 25080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2509250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy StepanovNOINLINE void ParamOriginTest(int param, U4 origin) { 251011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(param, origin); 25110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 25120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 25130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, Param) { 25140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 25150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int *a = new int; 2516250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 origin = __LINE__; 251712c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(a); 25180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(a, sizeof(*a), origin); 25190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ParamOriginTest(*a, origin); 25200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete a; 25210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 25220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 25230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, Invoke) { 25240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 25250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructWithDtor s; // Will cause the calls to become invokes. 252611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(RetvalOriginTest(__LINE__), __LINE__); 25270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 25280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 25290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, strlen) { 25300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S8 alignment; 253112c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&alignment); 25320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char x[4] = {'a', 'b', 0, 0}; 25330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&x[2], 1); 2534250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 origin = __LINE__; 25350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(x, sizeof(x), origin); 253611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR_O(volatile unsigned y = strlen(x), origin); 25370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 25380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 25390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, wcslen) { 25400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov wchar_t w[3] = {'a', 'b', 0}; 2541250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 origin = __LINE__; 25420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(w, sizeof(w), origin); 25430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&w[2], sizeof(wchar_t)); 254411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR_O(volatile unsigned y = wcslen(w), origin); 25450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 25460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 25470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#if MSAN_HAS_M128 25480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, StoreIntrinsic) { 25490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __m128 x, y; 2550250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 origin = __LINE__; 25510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(&x, sizeof(x), origin); 25520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&x, sizeof(x)); 25530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __builtin_ia32_storeups((float*)&y, x); 255411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(y, origin); 25550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 25560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#endif 25570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 25580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE void RecursiveMalloc(int depth) { 25590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov static int count; 25600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov count++; 25610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if ((count % (1024 * 1024)) == 0) 25620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov printf("RecursiveMalloc: %d\n", count); 25630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int *x1 = new int; 25640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int *x2 = new int; 256512c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(x1); 256612c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(x2); 25670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (depth > 0) { 25680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov RecursiveMalloc(depth-1); 25690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov RecursiveMalloc(depth-1); 25700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } 25710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete x1; 25720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete x2; 25730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 25740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 257565199f1b253c4bfb225805629217acb8f0b1e185Kostya SerebryanyTEST(MemorySanitizer, CallocOverflow) { 257665199f1b253c4bfb225805629217acb8f0b1e185Kostya Serebryany size_t kArraySize = 4096; 257765199f1b253c4bfb225805629217acb8f0b1e185Kostya Serebryany volatile size_t kMaxSizeT = std::numeric_limits<size_t>::max(); 257865199f1b253c4bfb225805629217acb8f0b1e185Kostya Serebryany volatile size_t kArraySize2 = kMaxSizeT / kArraySize + 10; 257965199f1b253c4bfb225805629217acb8f0b1e185Kostya Serebryany void *p = calloc(kArraySize, kArraySize2); // Should return 0. 258065199f1b253c4bfb225805629217acb8f0b1e185Kostya Serebryany EXPECT_EQ(0L, Ident(p)); 258165199f1b253c4bfb225805629217acb8f0b1e185Kostya Serebryany} 258265199f1b253c4bfb225805629217acb8f0b1e185Kostya Serebryany 25830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerStress, DISABLED_MallocStackTrace) { 25840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov RecursiveMalloc(22); 25850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 2586