msan_test.cc revision 512c616cacf70ca029a2bf719a482b902f3687cd
10231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov//===-- msan_test.cc ------------------------------------------------------===// 20231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// 30231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// The LLVM Compiler Infrastructure 40231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// 50231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// This file is distributed under the University of Illinois Open Source 60231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// License. See LICENSE.TXT for details. 70231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// 80231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov//===----------------------------------------------------------------------===// 90231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// 100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// This file is a part of MemorySanitizer. 110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// 120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// MemorySanitizer unit tests. 130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov//===----------------------------------------------------------------------===// 140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 156e5ff89e8980b7c2ca0a39811433fb7ac2c74372Evgeniy Stepanov#ifndef MSAN_EXTERNAL_TEST_CONFIG 166e5ff89e8980b7c2ca0a39811433fb7ac2c74372Evgeniy Stepanov#include "msan_test_config.h" 176e5ff89e8980b7c2ca0a39811433fb7ac2c74372Evgeniy Stepanov#endif // MSAN_EXTERNAL_TEST_CONFIG 186e5ff89e8980b7c2ca0a39811433fb7ac2c74372Evgeniy Stepanov 190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include "sanitizer/msan_interface.h" 200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include "msandr_test_so.h" 210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <stdlib.h> 230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <stdarg.h> 240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <stdio.h> 250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <assert.h> 260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <wchar.h> 277cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov#include <math.h> 280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 299530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov#include <arpa/inet.h> 30e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov#include <dlfcn.h> 31103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov#include <grp.h> 320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <unistd.h> 332bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov#include <link.h> 340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <limits.h> 350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <sys/time.h> 360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <sys/types.h> 370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <sys/stat.h> 380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <fcntl.h> 390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <sys/resource.h> 400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <sys/ioctl.h> 410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <sys/utsname.h> 420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <sys/mman.h> 430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#include <sys/vfs.h> 44d97a15a931ae6696a1071e1471c9a019e821904bEvgeniy Stepanov#include <dirent.h> 45e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov#include <pwd.h> 46134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov#include <sys/socket.h> 470a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov#include <netdb.h> 480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#if defined(__i386__) || defined(__x86_64__) 500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov# include <emmintrin.h> 510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov# define MSAN_HAS_M128 1 520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#else 530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov# define MSAN_HAS_M128 0 540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#endif 550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtypedef unsigned char U1; 570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtypedef unsigned short U2; // NOLINT 580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtypedef unsigned int U4; 590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtypedef unsigned long long U8; // NOLINT 600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtypedef signed char S1; 610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtypedef signed short S2; // NOLINT 620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtypedef signed int S4; 630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtypedef signed long long S8; // NOLINT 640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#define NOINLINE __attribute__((noinline)) 650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#define INLINE __attribute__((always_inline)) 660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 6711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovstatic bool TrackingOrigins() { 6811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov S8 x; 6911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov __msan_set_origin(&x, sizeof(x), 0x1234); 70250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 origin = __msan_get_origin(&x); 7111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov __msan_set_origin(&x, sizeof(x), 0); 7211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov return origin == 0x1234; 7311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov} 740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 7511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov#define EXPECT_UMR(action) \ 760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov do { \ 770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_expect_umr(1); \ 780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov action; \ 790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_expect_umr(0); \ 800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } while (0) 810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 8211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov#define EXPECT_UMR_O(action, origin) \ 830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov do { \ 840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_expect_umr(1); \ 850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov action; \ 860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_expect_umr(0); \ 870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (TrackingOrigins()) \ 8812c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov EXPECT_EQ(origin, __msan_get_umr_origin()); \ 890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } while (0) 900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov#define EXPECT_UMR_S(action, stack_origin) \ 920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov do { \ 930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_expect_umr(1); \ 940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov action; \ 950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_expect_umr(0); \ 96250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 id = __msan_get_umr_origin(); \ 970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov const char *str = __msan_get_origin_descr_if_stack(id); \ 980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!str || strcmp(str, stack_origin)) { \ 990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov fprintf(stderr, "EXPECT_POISONED_S: id=%u %s, %s", \ 1000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov id, stack_origin, str); \ 1010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(1, 0); \ 1020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } \ 1030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } while (0) 1040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 10511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov#define EXPECT_POISONED(x) ExpectPoisoned(x) 10611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 10711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovtemplate<typename T> 10811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovvoid ExpectPoisoned(const T& t) { 10911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NE(-1, __msan_test_shadow((void*)&t, sizeof(t))); 11011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov} 11111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 11211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov#define EXPECT_POISONED_O(x, origin) \ 11311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov ExpectPoisonedWithOrigin(x, origin) 11411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 11511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovtemplate<typename T> 11611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovvoid ExpectPoisonedWithOrigin(const T& t, unsigned origin) { 11711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NE(-1, __msan_test_shadow((void*)&t, sizeof(t))); 11811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov if (TrackingOrigins()) 11911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_EQ(origin, __msan_get_origin((void*)&t)); 12011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov} 12111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 12211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov#define EXPECT_POISONED_S(x, stack_origin) \ 12311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov ExpectPoisonedWithStackOrigin(x, stack_origin) 12411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 12511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovtemplate<typename T> 12611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovvoid ExpectPoisonedWithStackOrigin(const T& t, const char *stack_origin) { 12711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NE(-1, __msan_test_shadow((void*)&t, sizeof(t))); 128250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 id = __msan_get_origin((void*)&t); 12911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov const char *str = __msan_get_origin_descr_if_stack(id); 13011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov if (!str || strcmp(str, stack_origin)) { 13111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov fprintf(stderr, "EXPECT_POISONED_S: id=%u %s, %s", 13211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov id, stack_origin, str); 13311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_EQ(1, 0); 13411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov } 13511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov} 13611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 13711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov#define EXPECT_NOT_POISONED(x) ExpectNotPoisoned(x) 13811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 13911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovtemplate<typename T> 14011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovvoid ExpectNotPoisoned(const T& t) { 14111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_EQ(-1, __msan_test_shadow((void*)&t, sizeof(t))); 14211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov} 1430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic U8 poisoned_array[100]; 1450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> 1460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovT *GetPoisoned(int i = 0, T val = 0) { 1470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *res = (T*)&poisoned_array[i]; 1480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *res = val; 1490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&poisoned_array[i], sizeof(T)); 1500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 1510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 1520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> 154250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy StepanovT *GetPoisonedO(int i, U4 origin, T val = 0) { 1550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *res = (T*)&poisoned_array[i]; 1560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *res = val; 1570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&poisoned_array[i], sizeof(T)); 1580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(&poisoned_array[i], sizeof(T), origin); 1590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 1600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 1610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// This function returns its parameter but in such a way that compiler 1630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// can not prove it. 1640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> 1650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE 1660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic T Ident(T t) { 1670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov volatile T ret = t; 1680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return ret; 1690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 1700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> NOINLINE T ReturnPoisoned() { return *GetPoisoned<T>(); } 1720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic volatile int g_one = 1; 1740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic volatile int g_zero = 0; 1750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic volatile int g_0 = 0; 1760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic volatile int g_1 = 1; 1770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovS4 a_s4[100]; 1790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovS8 a_s8[100]; 1800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 18112c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov// Check that malloc poisons memory. 18212c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov// A lot of tests below depend on this. 18312c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy StepanovTEST(MemorySanitizerSanity, PoisonInMalloc) { 18412c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov int *x = (int*)malloc(sizeof(int)); 18512c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov EXPECT_POISONED(*x); 18612c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov free(x); 18712c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov} 18812c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov 1890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, NegativeTest1) { 1900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 *x = GetPoisoned<S4>(); 1910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (g_one) 1920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *x = 0; 19311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*x); 1940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 1950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, PositiveTest1) { 1970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // Load to store. 19811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S1>()); 19911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S2>()); 20011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S4>()); 20111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S8>()); 2020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // S->S conversions. 20411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S1>()); 20511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S1>()); 20611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S1>()); 2070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 20811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S2>()); 20911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S2>()); 21011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S2>()); 2110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 21211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S4>()); 21311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S4>()); 21411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S4>()); 2150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 21611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S8>()); 21711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S8>()); 21811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<S8>()); 2190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // ZExt 22111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<U1>()); 22211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<U1>()); 22311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<U1>()); 22411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<U2>()); 22511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<U2>()); 22611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<U4>()); 2270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // Unary ops. 22911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(- *GetPoisoned<S4>()); 2300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 23111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR(a_s4[g_zero] = 100 / *GetPoisoned<S4>(0, 1)); 2320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov a_s4[g_zero] = 1 - *GetPoisoned<S4>(); 2350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov a_s4[g_zero] = 1 + *GetPoisoned<S4>(); 2360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 2370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, Phi1) { 2390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 c; 2400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (g_one) { 2410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov c = *GetPoisoned<S4>(); 2420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } else { 24312c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(0); 2440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov c = 0; 2450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } 24611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(c); 2470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 2480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, Phi2) { 2500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 i = *GetPoisoned<S4>(); 2510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 n = g_one; 25211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR(for (; i < g_one; i++);); 25311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(i); 2540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 2550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 25611929000ec2919192b3be457f5a44c71ed55215eEvgeniy StepanovNOINLINE void Arg1ExpectUMR(S4 a1) { EXPECT_POISONED(a1); } 25711929000ec2919192b3be457f5a44c71ed55215eEvgeniy StepanovNOINLINE void Arg2ExpectUMR(S4 a1, S4 a2) { EXPECT_POISONED(a2); } 25811929000ec2919192b3be457f5a44c71ed55215eEvgeniy StepanovNOINLINE void Arg3ExpectUMR(S1 a1, S4 a2, S8 a3) { EXPECT_POISONED(a3); } 2590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, ArgTest) { 2610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov Arg1ExpectUMR(*GetPoisoned<S4>()); 2620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov Arg2ExpectUMR(0, *GetPoisoned<S4>()); 2630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov Arg3ExpectUMR(0, 1, *GetPoisoned<S8>()); 2640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 2650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, CallAndRet) { 2680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) return; 2690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ReturnPoisoned<S1>(); 2700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ReturnPoisoned<S2>(); 2710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ReturnPoisoned<S4>(); 2720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ReturnPoisoned<S8>(); 2730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 27411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(ReturnPoisoned<S1>()); 27511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(ReturnPoisoned<S2>()); 27611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(ReturnPoisoned<S4>()); 27711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(ReturnPoisoned<S8>()); 2780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 2790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// malloc() in the following test may be optimized to produce a compile-time 2810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// undef value. Check that we trap on the volatile assignment anyway. 2820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, DISABLED_MallocNoIdent) { 2830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 *x = (int*)malloc(sizeof(S4)); 28411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*x); 2850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov free(x); 2860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 2870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, Malloc) { 2890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 *x = (int*)Ident(malloc(sizeof(S4))); 29011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*x); 2910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov free(x); 2920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 2930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, Realloc) { 2950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 *x = (int*)Ident(realloc(0, sizeof(S4))); 29611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[0]); 2970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[0] = 1; 2980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x = (int*)Ident(realloc(x, 2 * sizeof(S4))); 29911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[0]); // Ok, was inited before. 30011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[1]); 3010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x = (int*)Ident(realloc(x, 3 * sizeof(S4))); 30211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[0]); // Ok, was inited before. 30311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[2]); 30411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[1]); 3050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[2] = 1; // Init this here. Check that after realloc it is poisoned again. 3060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x = (int*)Ident(realloc(x, 2 * sizeof(S4))); 30711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[0]); // Ok, was inited before. 30811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[1]); 3090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x = (int*)Ident(realloc(x, 3 * sizeof(S4))); 31011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[1]); 31111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[2]); 3120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov free(x); 3130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 3140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 3150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, Calloc) { 3160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 *x = (int*)Ident(calloc(1, sizeof(S4))); 31711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*x); // Should not be poisoned. 3180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // EXPECT_EQ(0, *x); 3190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov free(x); 3200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 3210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 3220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, AndOr) { 3230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U4 *p = GetPoisoned<U4>(); 3240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // We poison two bytes in the midle of a 4-byte word to make the test 3250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // correct regardless of endianness. 3260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ((U1*)p)[1] = 0; 3270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ((U1*)p)[2] = 0xff; 32811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*p & 0x00ffff00); 32911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*p & 0x00ff0000); 33011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*p & 0x0000ff00); 33111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*p & 0xff000000); 33211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*p & 0x000000ff); 33311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*p & 0x0000ffff); 33411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*p & 0xffff0000); 3350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 33611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*p | 0xff0000ff); 33711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*p | 0xff00ffff); 33811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*p | 0xffff00ff); 33911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*p | 0xff000000); 34011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*p | 0x000000ff); 34111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*p | 0x0000ffff); 34211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*p | 0xffff0000); 3430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 34411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<bool>() & *GetPoisoned<bool>()); 3450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 3460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 3470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> 34811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovstatic bool applyNot(T value, T shadow) { 3490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_partial_poison(&value, &shadow, sizeof(T)); 35011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov return !value; 3510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 3520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 3530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, Not) { 35411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U4>(0x0, 0x0)); 35511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U4>(0xFFFFFFFF, 0x0)); 35611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(applyNot<U4>(0xFFFFFFFF, 0xFFFFFFFF)); 35711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U4>(0xFF000000, 0x0FFFFFFF)); 35811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U4>(0xFF000000, 0x00FFFFFF)); 35911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U4>(0xFF000000, 0x0000FFFF)); 36011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U4>(0xFF000000, 0x00000000)); 36111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(applyNot<U4>(0xFF000000, 0xFF000000)); 36211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U4>(0xFF800000, 0xFF000000)); 36311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(applyNot<U4>(0x00008000, 0x00008000)); 36411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 36511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U1>(0x0, 0x0)); 36611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U1>(0xFF, 0xFE)); 36711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<U1>(0xFF, 0x0)); 36811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(applyNot<U1>(0xFF, 0xFF)); 36911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 37011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(applyNot<void*>((void*)0xFFFFFF, (void*)(-1))); 37111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applyNot<void*>((void*)0xFFFFFF, (void*)(-2))); 3720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 3730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 3740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, Shift) { 3750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U4 *up = GetPoisoned<U4>(); 3760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ((U1*)up)[0] = 0; 3770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ((U1*)up)[3] = 0xff; 37811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*up >> 30); 37911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*up >> 24); 38011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*up >> 23); 38111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*up >> 10); 3820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 38311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*up << 30); 38411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*up << 24); 38511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*up << 23); 38611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*up << 10); 3870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 3880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 *sp = (S4*)up; 38911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*sp >> 30); 39011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*sp >> 24); 39111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*sp >> 23); 39211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*sp >> 10); 3930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 3940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov sp = GetPoisoned<S4>(); 3950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ((S1*)sp)[1] = 0; 3960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ((S1*)sp)[2] = 0; 39711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*sp >> 31); 3980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 39911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(100 >> *GetPoisoned<S4>()); 40011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(100U >> *GetPoisoned<S4>()); 4010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE static int GetPoisonedZero() { 4040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int *zero = new int; 4050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *zero = 0; 4060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(zero, sizeof(*zero)); 4070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = *zero; 4080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete zero; 4090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 4100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, LoadFromDirtyAddress) { 4130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int *a = new int; 4140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *a = 0; 41512c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov EXPECT_UMR(break_optimization((void*)(U8)a[GetPoisonedZero()])); 4160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete a; 4170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, StoreToDirtyAddress) { 4200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int *a = new int; 42111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR(a[GetPoisonedZero()] = 0); 42212c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(a); 4230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete a; 4240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE void StackTestFunc() { 4280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 p4; 4290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 ok4 = 1; 4300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S2 p2; 4310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S2 ok2 = 1; 4320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S1 p1; 4330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S1 ok1 = 1; 43412c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&p4); 43512c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&ok4); 43612c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&p2); 43712c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&ok2); 43812c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&p1); 43912c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&ok1); 4400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 44111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(p4); 44211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(p2); 44311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(p1); 44411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(ok1); 44511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(ok2); 44611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(ok4); 4470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, StackTest) { 4500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StackTestFunc(); 4510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE void StackStressFunc() { 4540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int foo[10000]; 45512c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(foo); 4560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, DISABLED_StackStressTest) { 4590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov for (int i = 0; i < 1000000; i++) 4600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StackStressFunc(); 4610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> 4640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovvoid TestFloatingPoint() { 4650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov static volatile T v; 4660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov static T g[100]; 46712c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&g); 4680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *x = GetPoisoned<T>(); 4690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *y = GetPoisoned<T>(1); 47011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*x); 47111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED((long long)*x); 47211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED((int)*x); 4730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov g[0] = *x; 4740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov g[1] = *x + *y; 4750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov g[2] = *x - *y; 4760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov g[3] = *x * *y; 4770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, FloatingPointTest) { 4800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestFloatingPoint<float>(); 4810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestFloatingPoint<double>(); 4820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, DynMem) { 4850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 x = 0; 4860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 *y = GetPoisoned<S4>(); 4870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov memcpy(y, &x, g_one * sizeof(S4)); 48811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*y); 4890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic char *DynRetTestStr; 4920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, DynRet) { 4940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) return; 4950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ReturnPoisoned<S8>(); 49611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(clearenv()); 4970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 4980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 4990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, DynRet1) { 5010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) return; 5020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ReturnPoisoned<S8>(); 5030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct LargeStruct { 5060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 x[10]; 5070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 5080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE 5100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovLargeStruct LargeRetTest() { 5110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov LargeStruct res; 5120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[0] = *GetPoisoned<S4>(); 5130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[1] = *GetPoisoned<S4>(); 5140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[2] = *GetPoisoned<S4>(); 5150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[3] = *GetPoisoned<S4>(); 5160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[4] = *GetPoisoned<S4>(); 5170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[5] = *GetPoisoned<S4>(); 5180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[6] = *GetPoisoned<S4>(); 5190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[7] = *GetPoisoned<S4>(); 5200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[8] = *GetPoisoned<S4>(); 5210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.x[9] = *GetPoisoned<S4>(); 5220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 5230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, LargeRet) { 5260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov LargeStruct a = LargeRetTest(); 52711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(a.x[0]); 52811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(a.x[9]); 5290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, fread) { 5320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *x = new char[32]; 5330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov FILE *f = fopen("/proc/self/stat", "r"); 5340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(f); 5350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov fread(x, 1, 32, f); 53611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[0]); 53711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[16]); 53811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[31]); 5390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov fclose(f); 5400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete x; 5410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, read) { 5440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *x = new char[32]; 5450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int fd = open("/proc/self/stat", O_RDONLY); 5460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(fd > 0); 5470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int sz = read(fd, x, 32); 5480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(sz == 32); 54911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[0]); 55011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[16]); 55111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[31]); 5520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov close(fd); 5530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete x; 5540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, pread) { 5570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *x = new char[32]; 5580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int fd = open("/proc/self/stat", O_RDONLY); 5590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(fd > 0); 5600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int sz = pread(fd, x, 32, 0); 5610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(sz == 32); 56211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[0]); 56311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[16]); 56411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[31]); 5650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov close(fd); 5660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete x; 5670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// FIXME: fails now. 5700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, DISABLED_ioctl) { 5710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov struct winsize ws; 5720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(ioctl(2, TIOCGWINSZ, &ws), 0); 57311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(ws.ws_col); 5740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, readlink) { 5770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *x = new char[1000]; 5780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov readlink("/proc/self/exe", x, 1000); 57911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[0]); 5800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete [] x; 5810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, stat) { 5850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov struct stat* st = new struct stat; 5860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = stat("/proc/self/stat", st); 5870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(!res); 58811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(st->st_dev); 58911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(st->st_mode); 59011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(st->st_size); 5910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 5920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 5930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, statfs) { 5940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov struct statfs* st = new struct statfs; 5950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = statfs("/", st); 5960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(!res); 59711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(st->f_type); 59811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(st->f_bfree); 59911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(st->f_namelen); 6000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 6010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 6020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, pipe) { 6030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* pipefd = new int[2]; 6040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = pipe(pipefd); 6050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(!res); 60611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(pipefd[0]); 60711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(pipefd[1]); 6080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov close(pipefd[0]); 6090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov close(pipefd[1]); 6100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 6110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 612134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy StepanovTEST(MemorySanitizer, pipe2) { 613134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov int* pipefd = new int[2]; 614134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov int res = pipe2(pipefd, O_NONBLOCK); 615134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov assert(!res); 616134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov EXPECT_NOT_POISONED(pipefd[0]); 617134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov EXPECT_NOT_POISONED(pipefd[1]); 618134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov close(pipefd[0]); 619134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov close(pipefd[1]); 620134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov} 621134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov 622134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy StepanovTEST(MemorySanitizer, socketpair) { 623134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov int sv[2]; 624134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov int res = socketpair(AF_UNIX, SOCK_STREAM, 0, sv); 625134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov assert(!res); 626134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov EXPECT_NOT_POISONED(sv[0]); 627134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov EXPECT_NOT_POISONED(sv[1]); 628134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov close(sv[0]); 629134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov close(sv[1]); 630134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov} 631134fe8ae5090d39aa2d37a5db5298e49467399c8Evgeniy Stepanov 6329f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy StepanovTEST(MemorySanitizer, bind_getsockname) { 6339f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov int sock = socket(AF_UNIX, SOCK_STREAM, 0); 6349f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov 6359f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov struct sockaddr_in sai; 6369f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov memset(&sai, 0, sizeof(sai)); 6379f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov sai.sin_family = AF_UNIX; 6389f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov int res = bind(sock, (struct sockaddr *)&sai, sizeof(sai)); 6399f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov 6409f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov assert(!res); 6419f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov char buf[200]; 6429f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov socklen_t addrlen; 6439f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov EXPECT_UMR(getsockname(sock, (struct sockaddr *)&buf, &addrlen)); 6449f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov 6459f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov addrlen = sizeof(buf); 6469f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov res = getsockname(sock, (struct sockaddr *)&buf, &addrlen); 6479f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov EXPECT_NOT_POISONED(addrlen); 6489f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov EXPECT_NOT_POISONED(buf[0]); 6499f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov EXPECT_NOT_POISONED(buf[addrlen - 1]); 6509f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov EXPECT_POISONED(buf[addrlen]); 6519f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov close(sock); 6529f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov} 6539f58c5c60a56d9c39d36b5313dc87ad4bb713163Evgeniy Stepanov 6549d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy StepanovTEST(MemorySanitizer, accept) { 6559d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov int listen_socket = socket(AF_INET, SOCK_STREAM, 0); 6569d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_LT(0, listen_socket); 6579d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov 6589d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov struct sockaddr_in sai; 6599d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov sai.sin_family = AF_INET; 6609d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov sai.sin_port = 0; 6619d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov sai.sin_addr.s_addr = htonl(INADDR_LOOPBACK); 6629d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov int res = bind(listen_socket, (struct sockaddr *)&sai, sizeof(sai)); 6639d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_EQ(0, res); 6649d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov 6659d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov res = listen(listen_socket, 1); 6669d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_EQ(0, res); 6679d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov 6689d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov socklen_t sz = sizeof(sai); 6699d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov res = getsockname(listen_socket, (struct sockaddr *)&sai, &sz); 6709d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_EQ(0, res); 6719d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_EQ(sizeof(sai), sz); 672bc33e138d82759074f8333239f96506027731e20Evgeniy Stepanov 6739d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov int connect_socket = socket(AF_INET, SOCK_STREAM, 0); 6749d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_LT(0, connect_socket); 6759d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov res = fcntl(connect_socket, F_SETFL, O_NONBLOCK); 6769d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_EQ(0, res); 6779d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov res = connect(connect_socket, (struct sockaddr *)&sai, sizeof(sai)); 6789d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_EQ(-1, res); 6799d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_EQ(EINPROGRESS, errno); 6809d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov 6819d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov __msan_poison(&sai, sizeof(sai)); 682bc33e138d82759074f8333239f96506027731e20Evgeniy Stepanov int new_sock = accept(listen_socket, (struct sockaddr *)&sai, &sz); 6839d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_LT(0, new_sock); 6849d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov ASSERT_EQ(sizeof(sai), sz); 6859d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov EXPECT_NOT_POISONED(sai); 6869d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov 687bc33e138d82759074f8333239f96506027731e20Evgeniy Stepanov __msan_poison(&sai, sizeof(sai)); 688bc33e138d82759074f8333239f96506027731e20Evgeniy Stepanov res = getpeername(new_sock, (struct sockaddr *)&sai, &sz); 689bc33e138d82759074f8333239f96506027731e20Evgeniy Stepanov ASSERT_EQ(0, res); 690bc33e138d82759074f8333239f96506027731e20Evgeniy Stepanov ASSERT_EQ(sizeof(sai), sz); 691bc33e138d82759074f8333239f96506027731e20Evgeniy Stepanov EXPECT_NOT_POISONED(sai); 692bc33e138d82759074f8333239f96506027731e20Evgeniy Stepanov 6939d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov close(new_sock); 6949d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov close(connect_socket); 6959d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov close(listen_socket); 6969d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov} 6979d1525ec52430d0b8ffd6d0893b7f5529105b321Evgeniy Stepanov 698512c616cacf70ca029a2bf719a482b902f3687cdEvgeniy StepanovTEST(MemorySanitizer, getaddrinfo) { 699512c616cacf70ca029a2bf719a482b902f3687cdEvgeniy Stepanov struct addrinfo *ai; 700512c616cacf70ca029a2bf719a482b902f3687cdEvgeniy Stepanov struct addrinfo hints; 701512c616cacf70ca029a2bf719a482b902f3687cdEvgeniy Stepanov memset(&hints, 0, sizeof(hints)); 702512c616cacf70ca029a2bf719a482b902f3687cdEvgeniy Stepanov hints.ai_family = AF_INET; 703512c616cacf70ca029a2bf719a482b902f3687cdEvgeniy Stepanov int res = getaddrinfo("localhost", NULL, &hints, &ai); 704512c616cacf70ca029a2bf719a482b902f3687cdEvgeniy Stepanov ASSERT_EQ(0, res); 705512c616cacf70ca029a2bf719a482b902f3687cdEvgeniy Stepanov EXPECT_NOT_POISONED(*ai); 706512c616cacf70ca029a2bf719a482b902f3687cdEvgeniy Stepanov ASSERT_EQ(sizeof(sockaddr_in), ai->ai_addrlen); 707512c616cacf70ca029a2bf719a482b902f3687cdEvgeniy Stepanov EXPECT_NOT_POISONED(*(sockaddr_in*)ai->ai_addr); 708512c616cacf70ca029a2bf719a482b902f3687cdEvgeniy Stepanov} 709512c616cacf70ca029a2bf719a482b902f3687cdEvgeniy Stepanov 7100a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov#define EXPECT_HOSTENT_NOT_POISONED(he) \ 7110a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov do { \ 7120a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(*(he)); \ 7130a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *) 0, (he)->h_name); \ 7140a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *) 0, (he)->h_aliases); \ 7150a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *) 0, (he)->h_addr_list); \ 7160a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(strlen((he)->h_name)); \ 7170a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov char **p = (he)->h_aliases; \ 7180a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov while (*p) { \ 7190a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(*p)); \ 7200a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ++p; \ 7210a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov } \ 7220a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov char **q = (he)->h_addr_list; \ 7230a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov while (*q) { \ 7240a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(*q[0]); \ 7250a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ++q; \ 7260a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov } \ 7270a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(*q); \ 7280a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov } while (0) 7290a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov 7300a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy StepanovTEST(MemorySanitizer, gethostent) { 7310a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent *he = gethostent(); 7320a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *)NULL, he); 7330a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_HOSTENT_NOT_POISONED(he); 7340a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov} 7350a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov 7360a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy StepanovTEST(MemorySanitizer, gethostbyname) { 7370a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent *he = gethostbyname("localhost"); 7380a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *)NULL, he); 7390a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_HOSTENT_NOT_POISONED(he); 7400a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov} 7410a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov 7420a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy StepanovTEST(MemorySanitizer, gethostbyname2) { 7430a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent *he = gethostbyname2("localhost", AF_INET); 7440a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *)NULL, he); 7450a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_HOSTENT_NOT_POISONED(he); 7460a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov} 7470a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov 7480a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy StepanovTEST(MemorySanitizer, gethostbyaddr) { 7490a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov in_addr_t addr = inet_addr("127.0.0.1"); 7500a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(addr); 7510a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent *he = gethostbyaddr(&addr, sizeof(addr), AF_INET); 7520a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *)NULL, he); 7530a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_HOSTENT_NOT_POISONED(he); 7540a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov} 7550a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov 7560a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy StepanovTEST(MemorySanitizer, gethostent_r) { 7570a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov char buf[2000]; 7580a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent he; 7590a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent *result; 7600a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov int err; 7610a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov int res = gethostent_r(&he, buf, sizeof(buf), &result, &err); 7620a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_EQ(0, res); 7630a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(result); 7640a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *)NULL, result); 7650a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_HOSTENT_NOT_POISONED(result); 7660a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(err); 7670a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov} 7680a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov 7690a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy StepanovTEST(MemorySanitizer, gethostbyname_r) { 7700a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov char buf[2000]; 7710a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent he; 7720a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent *result; 7730a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov int err; 7740a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov int res = gethostbyname_r("localhost", &he, buf, sizeof(buf), &result, &err); 7750a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_EQ(0, res); 7760a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(result); 7770a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *)NULL, result); 7780a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_HOSTENT_NOT_POISONED(result); 7790a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(err); 7800a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov} 7810a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov 7820a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy StepanovTEST(MemorySanitizer, gethostbyname2_r) { 7830a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov char buf[2000]; 7840a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent he; 7850a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent *result; 7860a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov int err; 7870a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov int res = gethostbyname2_r("localhost", AF_INET, &he, buf, sizeof(buf), 7880a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov &result, &err); 7890a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_EQ(0, res); 7900a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(result); 7910a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *)NULL, result); 7920a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_HOSTENT_NOT_POISONED(result); 7930a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(err); 7940a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov} 7950a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov 7960a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy StepanovTEST(MemorySanitizer, gethostbyaddr_r) { 7970a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov char buf[2000]; 7980a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent he; 7990a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov struct hostent *result; 8000a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov int err; 8010a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov in_addr_t addr = inet_addr("127.0.0.1"); 8020a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(addr); 8030a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov int res = gethostbyaddr_r(&addr, sizeof(addr), AF_INET, &he, buf, sizeof(buf), 8040a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov &result, &err); 8050a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_EQ(0, res); 8060a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(result); 8070a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov ASSERT_NE((void *)NULL, result); 8080a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_HOSTENT_NOT_POISONED(result); 8090a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov EXPECT_NOT_POISONED(err); 8100a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov} 8110a2cc37712a452525f9f03b3bf67b1f0a97c8d3aEvgeniy Stepanov 812f32be42523a199674ea665a499db131591e64e08Evgeniy StepanovTEST(MemorySanitizer, getsockopt) { 813f32be42523a199674ea665a499db131591e64e08Evgeniy Stepanov int sock = socket(AF_UNIX, SOCK_STREAM, 0); 814f32be42523a199674ea665a499db131591e64e08Evgeniy Stepanov struct linger l[2]; 815f32be42523a199674ea665a499db131591e64e08Evgeniy Stepanov socklen_t sz = sizeof(l[0]); 816f32be42523a199674ea665a499db131591e64e08Evgeniy Stepanov int res = getsockopt(sock, SOL_SOCKET, SO_LINGER, &l[0], &sz); 817f32be42523a199674ea665a499db131591e64e08Evgeniy Stepanov ASSERT_EQ(0, res); 818f32be42523a199674ea665a499db131591e64e08Evgeniy Stepanov ASSERT_EQ(sizeof(l[0]), sz); 819f32be42523a199674ea665a499db131591e64e08Evgeniy Stepanov EXPECT_NOT_POISONED(l[0]); 820f32be42523a199674ea665a499db131591e64e08Evgeniy Stepanov EXPECT_POISONED(*(char *)(l + 1)); 821f32be42523a199674ea665a499db131591e64e08Evgeniy Stepanov} 822f32be42523a199674ea665a499db131591e64e08Evgeniy Stepanov 8230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, getcwd) { 8240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char path[PATH_MAX + 1]; 8250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* res = getcwd(path, sizeof(path)); 8260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(res); 82711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(path[0]); 8280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 8290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 8307eed04c4dce69ad1e485edbf6dd963e176b52e0dEvgeniy StepanovTEST(MemorySanitizer, getcwd_gnu) { 8317eed04c4dce69ad1e485edbf6dd963e176b52e0dEvgeniy Stepanov char* res = getcwd(NULL, 0); 8327eed04c4dce69ad1e485edbf6dd963e176b52e0dEvgeniy Stepanov assert(res); 8337eed04c4dce69ad1e485edbf6dd963e176b52e0dEvgeniy Stepanov EXPECT_NOT_POISONED(res[0]); 8347eed04c4dce69ad1e485edbf6dd963e176b52e0dEvgeniy Stepanov free(res); 8357eed04c4dce69ad1e485edbf6dd963e176b52e0dEvgeniy Stepanov} 8367eed04c4dce69ad1e485edbf6dd963e176b52e0dEvgeniy Stepanov 837d97a15a931ae6696a1071e1471c9a019e821904bEvgeniy StepanovTEST(MemorySanitizer, readdir) { 838d97a15a931ae6696a1071e1471c9a019e821904bEvgeniy Stepanov DIR *dir = opendir("."); 839d97a15a931ae6696a1071e1471c9a019e821904bEvgeniy Stepanov struct dirent *d = readdir(dir); 840d97a15a931ae6696a1071e1471c9a019e821904bEvgeniy Stepanov assert(d); 841d97a15a931ae6696a1071e1471c9a019e821904bEvgeniy Stepanov EXPECT_NOT_POISONED(d->d_name[0]); 842d97a15a931ae6696a1071e1471c9a019e821904bEvgeniy Stepanov closedir(dir); 843d97a15a931ae6696a1071e1471c9a019e821904bEvgeniy Stepanov} 844d97a15a931ae6696a1071e1471c9a019e821904bEvgeniy Stepanov 8450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, realpath) { 8460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov const char* relpath = "."; 8470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char path[PATH_MAX + 1]; 8480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* res = realpath(relpath, path); 8490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(res); 85011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(path[0]); 8510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 8520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 8530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, memcpy) { 8540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* x = new char[2]; 8550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* y = new char[2]; 8560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[0] = 1; 8570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[1] = *GetPoisoned<char>(); 8580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov memcpy(y, x, 2); 85911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(y[0]); 86011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(y[1]); 8610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 8620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 8630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, memmove) { 8640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* x = new char[2]; 8650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* y = new char[2]; 8660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[0] = 1; 8670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[1] = *GetPoisoned<char>(); 8680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov memmove(y, x, 2); 86911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(y[0]); 87011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(y[1]); 8710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 8720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 8730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, strdup) { 8748aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov char buf[4] = "abc"; 8758aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov __msan_poison(buf + 2, sizeof(*buf)); 8768aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov char *x = strdup(buf); 8778aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov EXPECT_NOT_POISONED(x[0]); 8788aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov EXPECT_NOT_POISONED(x[1]); 8798aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov EXPECT_POISONED(x[2]); 8808aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov EXPECT_NOT_POISONED(x[3]); 8818aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov free(x); 8828aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov} 8838aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov 8848aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy StepanovTEST(MemorySanitizer, strndup) { 8858aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov char buf[4] = "abc"; 8868aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov __msan_poison(buf + 2, sizeof(*buf)); 8878aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov char *x = strndup(buf, 3); 8888aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov EXPECT_NOT_POISONED(x[0]); 8898aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov EXPECT_NOT_POISONED(x[1]); 8908aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov EXPECT_POISONED(x[2]); 8918aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov EXPECT_NOT_POISONED(x[3]); 8920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov free(x); 8930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 8940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 8958aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy StepanovTEST(MemorySanitizer, strndup_short) { 8968aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov char buf[4] = "abc"; 8978aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov __msan_poison(buf + 1, sizeof(*buf)); 8988aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov __msan_poison(buf + 2, sizeof(*buf)); 8998aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov char *x = strndup(buf, 2); 9008aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov EXPECT_NOT_POISONED(x[0]); 9018aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov EXPECT_POISONED(x[1]); 9021d21bd159c3830dd7a6c05854408442d91a336adEvgeniy Stepanov EXPECT_NOT_POISONED(x[2]); 9038aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov free(x); 9048aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov} 9058aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov 9068aa1ae03e339ee9365c05dcf831f59b3a817bd84Evgeniy Stepanov 9070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T, int size> 9080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovvoid TestOverlapMemmove() { 9090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *x = new T[size]; 9100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(size >= 3); 9110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[2] = 0; 9120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov memmove(x, x + 1, (size - 1) * sizeof(T)); 91311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x[1]); 9140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) { 9150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // FIXME: under DR we will lose this information 9160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // because accesses in memmove will unpoisin the shadow. 9170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // We need to use our own memove implementation instead of libc's. 91811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[0]); 91911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x[2]); 9200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } 9210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete [] x; 9220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, overlap_memmove) { 9250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestOverlapMemmove<U1, 10>(); 9260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestOverlapMemmove<U1, 1000>(); 9270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestOverlapMemmove<U8, 4>(); 9280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestOverlapMemmove<U8, 1000>(); 9290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, strcpy) { // NOLINT 9320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* x = new char[3]; 9330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* y = new char[3]; 9340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[0] = 'a'; 9350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[1] = *GetPoisoned<char>(1, 1); 9360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[2] = 0; 9370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov strcpy(y, x); // NOLINT 93811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(y[0]); 93911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(y[1]); 94011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(y[2]); 9410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, strncpy) { // NOLINT 9440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* x = new char[3]; 9450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* y = new char[3]; 9460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[0] = 'a'; 9470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[1] = *GetPoisoned<char>(1, 1); 9480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x[2] = 0; 9490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov strncpy(y, x, 2); // NOLINT 95011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(y[0]); 95111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(y[1]); 95211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(y[2]); 9530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, strtol) { 9560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *e; 9570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(1 == strtol("1", &e, 10)); 95811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((S8) e); 9590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, strtoll) { 9620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *e; 9630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(1 == strtoll("1", &e, 10)); 96411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((S8) e); 9650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, strtoul) { 9680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *e; 9690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(1 == strtoul("1", &e, 10)); 97011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((S8) e); 9710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 9730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, strtoull) { 9740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *e; 9750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(1 == strtoull("1", &e, 10)); 97611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((S8) e); 9770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 9780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 979e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy StepanovTEST(MemorySanitizer, strtod) { 980e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov char *e; 981e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov assert(0 != strtod("1.5", &e)); 98211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((S8) e); 983e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov} 984e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov 985e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy StepanovTEST(MemorySanitizer, strtof) { 986e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov char *e; 987e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov assert(0 != strtof("1.5", &e)); 98811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((S8) e); 989e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov} 990e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov 991e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy StepanovTEST(MemorySanitizer, strtold) { 992e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov char *e; 993e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov assert(0 != strtold("1.5", &e)); 99411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((S8) e); 995e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov} 996e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov 997c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy StepanovTEST(MemorySanitizer, modf) { 998c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy Stepanov double x, y; 999c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy Stepanov x = modf(2.1, &y); 1000c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy Stepanov EXPECT_NOT_POISONED(y); 1001c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy Stepanov} 1002c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy Stepanov 1003c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy StepanovTEST(MemorySanitizer, modff) { 1004c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy Stepanov float x, y; 1005c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy Stepanov x = modff(2.1, &y); 1006c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy Stepanov EXPECT_NOT_POISONED(y); 1007c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy Stepanov} 1008c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy Stepanov 1009c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy StepanovTEST(MemorySanitizer, modfl) { 1010c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy Stepanov long double x, y; 1011c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy Stepanov x = modfl(2.1, &y); 1012c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy Stepanov EXPECT_NOT_POISONED(y); 1013c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy Stepanov} 1014c87088b54f7438d589119d314853a5f3b00d02b5Evgeniy Stepanov 10150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, sprintf) { // NOLINT 10160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char buff[10]; 101712c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(buff); 101811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(buff[0]); 10190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = sprintf(buff, "%d", 1234567); // NOLINT 10200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(res == 7); 10210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[0] == '1'); 10220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[1] == '2'); 10230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[2] == '3'); 10240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[6] == '7'); 10250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[7] == 0); 102611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(buff[8]); 10270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 10280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 10290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, snprintf) { 10300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char buff[10]; 103112c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(buff); 103211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(buff[0]); 10330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = snprintf(buff, sizeof(buff), "%d", 1234567); 10340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(res == 7); 10350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[0] == '1'); 10360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[1] == '2'); 10370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[2] == '3'); 10380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[6] == '7'); 10390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[7] == 0); 104011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(buff[8]); 10410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 10420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 10430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, swprintf) { 10440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov wchar_t buff[10]; 10450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(sizeof(wchar_t) == 4); 104612c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(buff); 104711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(buff[0]); 10480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = swprintf(buff, 9, L"%d", 1234567); 10490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(res == 7); 10500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[0] == '1'); 10510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[1] == '2'); 10520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[2] == '3'); 10530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[6] == '7'); 10540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(buff[7] == 0); 105511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(buff[8]); 10560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 10570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 10580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, wcstombs) { 10590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov const wchar_t *x = L"abc"; 10600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char buff[10]; 10610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = wcstombs(buff, x, 4); 10620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(res, 3); 10630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(buff[0], 'a'); 10640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(buff[1], 'b'); 10650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(buff[2], 'c'); 10660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 10670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 10680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, gettimeofday) { 10690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov struct timeval tv; 10700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov struct timezone tz; 107112c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&tv); 107212c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&tz); 10730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(sizeof(tv) == 16); 10740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(sizeof(tz) == 8); 107511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(tv.tv_sec); 107611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(tv.tv_usec); 107711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(tz.tz_minuteswest); 107811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(tz.tz_dsttime); 10790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(0 == gettimeofday(&tv, &tz)); 108011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(tv.tv_sec); 108111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(tv.tv_usec); 108211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(tz.tz_minuteswest); 108311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(tz.tz_dsttime); 10840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 10850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1086e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy StepanovTEST(MemorySanitizer, clock_gettime) { 1087e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov struct timespec tp; 1088e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_POISONED(tp.tv_sec); 1089e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_POISONED(tp.tv_nsec); 1090e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov assert(0 == clock_gettime(CLOCK_REALTIME, &tp)); 1091e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(tp.tv_sec); 1092e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(tp.tv_nsec); 1093e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov} 1094e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov 10957cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy StepanovTEST(MemorySanitizer, clock_getres) { 10967cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov struct timespec tp; 10977cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov EXPECT_POISONED(tp.tv_sec); 10987cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov EXPECT_POISONED(tp.tv_nsec); 10997cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov assert(0 == clock_getres(CLOCK_REALTIME, 0)); 11007cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov EXPECT_POISONED(tp.tv_sec); 11017cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov EXPECT_POISONED(tp.tv_nsec); 11027cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov assert(0 == clock_getres(CLOCK_REALTIME, &tp)); 11037cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov EXPECT_NOT_POISONED(tp.tv_sec); 11047cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov EXPECT_NOT_POISONED(tp.tv_nsec); 11057cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov} 11067cdae1683c9c2fcd4473a5862c90c64be3a8c5fdEvgeniy Stepanov 1107e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy StepanovTEST(MemorySanitizer, getitimer) { 1108e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov struct itimerval it1, it2; 1109e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov int res; 1110e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_POISONED(it1.it_interval.tv_sec); 1111e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_POISONED(it1.it_interval.tv_usec); 1112e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_POISONED(it1.it_value.tv_sec); 1113e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_POISONED(it1.it_value.tv_usec); 1114e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov res = getitimer(ITIMER_VIRTUAL, &it1); 1115e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov assert(!res); 1116e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(it1.it_interval.tv_sec); 1117e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(it1.it_interval.tv_usec); 1118e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(it1.it_value.tv_sec); 1119e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(it1.it_value.tv_usec); 1120e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov 1121e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov it1.it_interval.tv_sec = it1.it_value.tv_sec = 10000; 1122e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov it1.it_interval.tv_usec = it1.it_value.tv_usec = 0; 1123e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov 1124e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov res = setitimer(ITIMER_VIRTUAL, &it1, &it2); 1125e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov assert(!res); 1126e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(it2.it_interval.tv_sec); 1127e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(it2.it_interval.tv_usec); 1128e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(it2.it_value.tv_sec); 1129e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(it2.it_value.tv_usec); 1130e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov 1131e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov // Check that old_value can be 0, and disable the timer. 1132e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov memset(&it1, 0, sizeof(it1)); 1133e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov res = setitimer(ITIMER_VIRTUAL, &it1, 0); 1134e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov assert(!res); 1135e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov} 1136e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov 1137fef660506e9e5703fedfee01d614abd4b741c738Evgeniy StepanovTEST(MemorySanitizer, time) { 1138fef660506e9e5703fedfee01d614abd4b741c738Evgeniy Stepanov time_t t; 1139fef660506e9e5703fedfee01d614abd4b741c738Evgeniy Stepanov EXPECT_POISONED(t); 1140fef660506e9e5703fedfee01d614abd4b741c738Evgeniy Stepanov time_t t2 = time(&t); 1141fef660506e9e5703fedfee01d614abd4b741c738Evgeniy Stepanov assert(t2 != (time_t)-1); 1142fef660506e9e5703fedfee01d614abd4b741c738Evgeniy Stepanov EXPECT_NOT_POISONED(t); 1143fef660506e9e5703fedfee01d614abd4b741c738Evgeniy Stepanov} 1144fef660506e9e5703fedfee01d614abd4b741c738Evgeniy Stepanov 11459358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy StepanovTEST(MemorySanitizer, localtime) { 11469358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov time_t t = 123; 11479358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov struct tm *time = localtime(&t); 11489358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov assert(time != 0); 11499358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov EXPECT_NOT_POISONED(time->tm_sec); 11509358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov EXPECT_NOT_POISONED(time->tm_hour); 11519358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov EXPECT_NOT_POISONED(time->tm_year); 11529358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov EXPECT_NOT_POISONED(time->tm_isdst); 11539358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov} 11549358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov 11559358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy StepanovTEST(MemorySanitizer, localtime_r) { 11569358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov time_t t = 123; 11579358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov struct tm time; 11589358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov struct tm *res = localtime_r(&t, &time); 11599358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov assert(res != 0); 11609358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov EXPECT_NOT_POISONED(time.tm_sec); 11619358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov EXPECT_NOT_POISONED(time.tm_hour); 11629358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov EXPECT_NOT_POISONED(time.tm_year); 11639358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov EXPECT_NOT_POISONED(time.tm_isdst); 11649358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov} 11659358c58d0aaf1b20f17362af354d4c3c1309276aEvgeniy Stepanov 11660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, mmap) { 11670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov const int size = 4096; 11680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov void *p1, *p2; 11690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov p1 = mmap(0, size, PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANON, -1, 0); 11700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(p1, size); 11710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov munmap(p1, size); 11720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov for (int i = 0; i < 1000; i++) { 11730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov p2 = mmap(0, size, PROT_READ|PROT_WRITE, MAP_PRIVATE|MAP_ANON, -1, 0); 11740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (p2 == p1) 11750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov break; 11760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov else 11770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov munmap(p2, size); 11780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } 11790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (p1 == p2) { 118011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*(char*)p2); 11810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov munmap(p2, size); 11820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } 11830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 11840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 11850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// FIXME: enable and add ecvt. 11860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// FIXME: check why msandr does nt handle fcvt. 11870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, fcvt) { 11880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int a, b; 118912c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&a); 119012c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&b); 119111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(a); 119211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(b); 11930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char *str = fcvt(12345.6789, 10, &a, &b); 119411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(a); 119511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(b); 11960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 11970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 11987cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy StepanovTEST(MemorySanitizer, frexp) { 11997cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov int x; 12007cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov x = *GetPoisoned<int>(); 12017cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov double r = frexp(1.1, &x); 12027cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov EXPECT_NOT_POISONED(r); 12037cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov EXPECT_NOT_POISONED(x); 12047cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov 12057cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov x = *GetPoisoned<int>(); 12067cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov float rf = frexpf(1.1, &x); 12077cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov EXPECT_NOT_POISONED(rf); 12087cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov EXPECT_NOT_POISONED(x); 12097cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov 12107cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov x = *GetPoisoned<int>(); 12117cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov double rl = frexpl(1.1, &x); 12127cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov EXPECT_NOT_POISONED(rl); 12137cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov EXPECT_NOT_POISONED(x); 12147cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov} 12157cbbb2943527ff852bdace822c8592cfc7e450d7Evgeniy Stepanov 121606658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanovnamespace { 121706658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov 121806658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanovstatic int cnt; 121906658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov 122006658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanovvoid SigactionHandler(int signo, siginfo_t* si, void* uc) { 122106658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov assert(signo == SIGPROF); 122206658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov assert(si); 122306658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov EXPECT_NOT_POISONED(si->si_errno); 122406658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov EXPECT_NOT_POISONED(si->si_pid); 122506658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov#if __linux__ 122606658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov# if defined(__x86_64__) 122706658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov EXPECT_NOT_POISONED(((ucontext_t*)uc)->uc_mcontext.gregs[REG_RIP]); 122806658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov# elif defined(__i386__) 122906658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov EXPECT_NOT_POISONED(((ucontext_t*)uc)->uc_mcontext.gregs[REG_EIP]); 123006658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov# endif 123106658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov#endif 123206658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov ++cnt; 123306658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov} 123406658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov 123506658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy StepanovTEST(MemorySanitizer, sigaction) { 123606658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov struct sigaction act = {}; 1237cd3049da150124156502b1a8c05e4c4887786cc5Evgeniy Stepanov struct sigaction oldact = {}; 12386d0b7f6c41bed64419dd74a8daf83de5e9f54de9Evgeniy Stepanov struct sigaction origact = {}; 12396d0b7f6c41bed64419dd74a8daf83de5e9f54de9Evgeniy Stepanov 12406d0b7f6c41bed64419dd74a8daf83de5e9f54de9Evgeniy Stepanov sigaction(SIGPROF, 0, &origact); 12416d0b7f6c41bed64419dd74a8daf83de5e9f54de9Evgeniy Stepanov 124206658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov act.sa_flags |= SA_SIGINFO; 124306658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov act.sa_sigaction = &SigactionHandler; 124406658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov sigaction(SIGPROF, &act, 0); 124506658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov 124606658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov kill(getpid(), SIGPROF); 124706658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov 124806658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov act.sa_flags &= ~SA_SIGINFO; 124906658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov act.sa_handler = SIG_DFL; 125006658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov sigaction(SIGPROF, &act, 0); 125106658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov 125206658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov act.sa_flags &= ~SA_SIGINFO; 125306658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov act.sa_handler = SIG_IGN; 1254cd3049da150124156502b1a8c05e4c4887786cc5Evgeniy Stepanov sigaction(SIGPROF, &act, &oldact); 1255cd3049da150124156502b1a8c05e4c4887786cc5Evgeniy Stepanov EXPECT_FALSE(oldact.sa_flags & SA_SIGINFO); 1256cd3049da150124156502b1a8c05e4c4887786cc5Evgeniy Stepanov EXPECT_EQ(SIG_DFL, oldact.sa_handler); 125706658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov kill(getpid(), SIGPROF); 125806658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov 125906658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov act.sa_flags |= SA_SIGINFO; 126006658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov act.sa_sigaction = &SigactionHandler; 1261cd3049da150124156502b1a8c05e4c4887786cc5Evgeniy Stepanov sigaction(SIGPROF, &act, &oldact); 1262cd3049da150124156502b1a8c05e4c4887786cc5Evgeniy Stepanov EXPECT_FALSE(oldact.sa_flags & SA_SIGINFO); 1263cd3049da150124156502b1a8c05e4c4887786cc5Evgeniy Stepanov EXPECT_EQ(SIG_IGN, oldact.sa_handler); 126406658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov kill(getpid(), SIGPROF); 126506658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov 126606658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov act.sa_flags &= ~SA_SIGINFO; 126706658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov act.sa_handler = SIG_DFL; 1268cd3049da150124156502b1a8c05e4c4887786cc5Evgeniy Stepanov sigaction(SIGPROF, &act, &oldact); 1269cd3049da150124156502b1a8c05e4c4887786cc5Evgeniy Stepanov EXPECT_TRUE(oldact.sa_flags & SA_SIGINFO); 1270cd3049da150124156502b1a8c05e4c4887786cc5Evgeniy Stepanov EXPECT_EQ(&SigactionHandler, oldact.sa_sigaction); 127106658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov EXPECT_EQ(2, cnt); 12726d0b7f6c41bed64419dd74a8daf83de5e9f54de9Evgeniy Stepanov 12736d0b7f6c41bed64419dd74a8daf83de5e9f54de9Evgeniy Stepanov sigaction(SIGPROF, &origact, 0); 127406658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov} 127506658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov 127606658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov} // namespace 127706658ea68c4912fffafef6e65f36dce7faa4a174Evgeniy Stepanov 12780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct StructWithDtor { 12790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ~StructWithDtor(); 12800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 12810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 12820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE StructWithDtor::~StructWithDtor() { 128312c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(0); 12840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 12850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 12860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, Invoke) { 12870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructWithDtor s; // Will cause the calls to become invokes. 128811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(0); 128911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<int>()); 129011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(0); 129111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(*GetPoisoned<int>()); 129211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(ReturnPoisoned<S4>()); 12930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 12940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 12950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, ptrtoint) { 12960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // Test that shadow is propagated through pointer-to-integer conversion. 12970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov void* p = (void*)0xABCD; 12980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(((char*)&p) + 1, sizeof(p)); 1299250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov EXPECT_NOT_POISONED((((uintptr_t)p) & 0xFF) == 0); 13000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov void* q = (void*)0xABCD; 13020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&q, sizeof(q) - 1); 1303250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov EXPECT_POISONED((((uintptr_t)q) & 0xFF) == 0); 13040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn2(int guard, ...) { 13070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 13080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 130911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 131011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 131111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 131211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, double)); 13130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 13140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn(int guard, ...) { 13170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 13180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 131911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 132011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 13210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // The following call will overwrite __msan_param_tls. 13220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // Checks after it test that arg shadow was somehow saved across the call. 13230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn2(1, 2, 3, 4, *GetPoisoned<double>()); 132411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 132511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 13260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 13270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, VAArgTest) { 13300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* x = GetPoisoned<int>(); 13310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* y = GetPoisoned<int>(4); 13320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn(1, 13, *x, 42, *y); 13330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_many(int guard, ...) { 13360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 13370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 133811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 133911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 134011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 134111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 134211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 134311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 134411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 134511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 134611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 134711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 13480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 13490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, VAArgManyTest) { 13520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* x = GetPoisoned<int>(); 13530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* y = GetPoisoned<int>(4); 13540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_many(1, 2, *x, 3, 4, 5, 6, 7, 8, 9, *y); 13550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_pass2(va_list vl) { 135811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 135911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 136011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 13610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_pass(int guard, ...) { 13640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 13650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 136611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 13670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_pass2(vl); 13680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 13690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, VAArgPass) { 13720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* x = GetPoisoned<int>(); 13730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* y = GetPoisoned<int>(4); 13740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_pass(1, *x, 2, 3, *y); 13750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_copy2(va_list vl) { 137811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 137911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 13800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_copy(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_POISONED(va_arg(vl, int)); 13870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl2; 13880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_copy(vl2, vl); 13890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_copy2(vl2); 139011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 139111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 13920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 13930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 13940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 13950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, VAArgCopy) { 13960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* x = GetPoisoned<int>(); 13970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* y = GetPoisoned<int>(4); 13980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_copy(1, 2, *x, 3, *y); 13990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_ptr(int guard, ...) { 14020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 14030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 140411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int*)); 140511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int*)); 140611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int*)); 140711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, double*)); 14080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 14090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, VAArgPtr) { 14120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int** x = GetPoisoned<int*>(); 14130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov double** y = GetPoisoned<double*>(8); 14140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int z; 14150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_ptr(1, &z, *x, &z, *y); 14160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_overflow(int guard, ...) { 14190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 14200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 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)); 142411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 142511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 142611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 142711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 142811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, double)); 142911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, double)); 143011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, double)); 143111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, double)); 143211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, double)); 143311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int*)); 143411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, double)); 143511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, double)); 143611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 143711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 143811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, double)); 143911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int*)); 144011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 144111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 144211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, double)); 144311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int*)); 144411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov 144511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 144611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, double)); 144711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int*)); 14480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 14500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, VAArgOverflow) { 14530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* x = GetPoisoned<int>(); 14540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov double* y = GetPoisoned<double>(8); 14550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int** p = GetPoisoned<int*>(16); 14560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int z; 14570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_overflow(1, 14580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1, 2, *x, 4, 5, 6, 14590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1.1, 2.2, 3.3, *y, 5.5, *p, 7.7, 8.8, 14600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // the following args will overflow for sure 14610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *x, *y, *p, 14620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 7, 9.9, &z, 14630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *x, *y, *p); 14640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_tlsoverwrite2(int guard, ...) { 14670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 14680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 146911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(va_arg(vl, int)); 14700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 14710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void vaargsfn_tlsoverwrite(int guard, ...) { 14740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // This call will overwrite TLS contents unless it's backed up somewhere. 14750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_tlsoverwrite2(2, 42); 14760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_list vl; 14770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_start(vl, guard); 147811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(va_arg(vl, int)); 14790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov va_end(vl); 14800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, VAArgTLSOverwrite) { 14830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int* x = GetPoisoned<int>(); 14840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov vaargsfn_tlsoverwrite(1, *x); 14850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct StructByVal { 14880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int a, b, c, d, e, f; 14890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 14900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 14910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE void StructByValTestFunc(struct StructByVal s) { 149211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s.a); 149311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s.b); 149411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s.c); 149511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s.d); 149611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s.e); 149711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s.f); 14980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 14990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE void StructByValTestFunc1(struct StructByVal s) { 15010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructByValTestFunc(s); 15020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE void StructByValTestFunc2(int z, struct StructByVal s) { 15050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructByValTestFunc(s); 15060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, StructByVal) { 15090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // Large aggregates are passed as "byval" pointer argument in LLVM. 15100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov struct StructByVal s; 15110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov s.a = 1; 15120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov s.b = *GetPoisoned<int>(); 15130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov s.c = 2; 15140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov s.d = *GetPoisoned<int>(); 15150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov s.e = 3; 15160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov s.f = *GetPoisoned<int>(); 15170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructByValTestFunc(s); 15180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructByValTestFunc1(s); 15190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructByValTestFunc2(0, s); 15200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#if MSAN_HAS_M128 1524b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy StepanovNOINLINE __m128i m128Eq(__m128i *a, __m128i *b) { return _mm_cmpeq_epi16(*a, *b); } 1525b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy StepanovNOINLINE __m128i m128Lt(__m128i *a, __m128i *b) { return _mm_cmplt_epi16(*a, *b); } 15260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, m128) { 15270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __m128i a = _mm_set1_epi16(0x1234); 15280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __m128i b = _mm_set1_epi16(0x7890); 152911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(m128Eq(&a, &b)); 153011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(m128Lt(&a, &b)); 15310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// FIXME: add more tests for __m128i. 15330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#endif // MSAN_HAS_M128 15340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// We should not complain when copying this poisoned hole. 15360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct StructWithHole { 15370231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U4 a; 15380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // 4-byte hole. 15390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U8 b; 15400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 15410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE StructWithHole ReturnStructWithHole() { 15430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructWithHole res; 15440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&res, sizeof(res)); 15450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.a = 1; 15460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.b = 2; 15470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 15480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, StructWithHole) { 15510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructWithHole a = ReturnStructWithHole(); 155212c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&a); 15530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate <class T> 15560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE T ReturnStruct() { 15570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T res; 15580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&res, sizeof(res)); 15590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.a = 1; 15600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 15610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate <class T> 15640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE void TestReturnStruct() { 15650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T s1 = ReturnStruct<T>(); 156611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s1.a); 156711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s1.b); 15680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 15690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct SSS1 { 15710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int a, b, c; 15720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 15730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct SSS2 { 15740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int b, a, c; 15750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 15760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct SSS3 { 15770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int b, c, a; 15780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 15790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct SSS4 { 15800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int c, b, a; 15810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 15820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 15830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct SSS5 { 15840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int a; 15850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov float b; 15860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 15870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct SSS6 { 15880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int a; 15890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov double b; 15900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 15910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct SSS7 { 15920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S8 b; 15930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int a; 15940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 15950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct SSS8 { 15960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S2 b; 15970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S8 a; 15980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 15990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 16000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, IntStruct3) { 16010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestReturnStruct<SSS1>(); 16020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestReturnStruct<SSS2>(); 16030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestReturnStruct<SSS3>(); 16040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestReturnStruct<SSS4>(); 16050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestReturnStruct<SSS5>(); 16060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestReturnStruct<SSS6>(); 16070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestReturnStruct<SSS7>(); 16080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov TestReturnStruct<SSS8>(); 16090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 16100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 16110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct LongStruct { 16120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U1 a1, b1; 16130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U2 a2, b2; 16140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U4 a4, b4; 16150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U8 a8, b8; 16160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 16170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 16180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE LongStruct ReturnLongStruct1() { 16190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov LongStruct res; 16200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&res, sizeof(res)); 16210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.a1 = res.a2 = res.a4 = res.a8 = 111; 16220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // leaves b1, .., b8 poisoned. 16230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 16240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 16250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 16260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE LongStruct ReturnLongStruct2() { 16270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov LongStruct res; 16280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&res, sizeof(res)); 16290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res.b1 = res.b2 = res.b4 = res.b8 = 111; 16300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // leaves a1, .., a8 poisoned. 16310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 16320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 16330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 16340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, LongStruct) { 16350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov LongStruct s1 = ReturnLongStruct1(); 16360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_print_shadow(&s1, sizeof(s1)); 163711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s1.a1); 163811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s1.a2); 163911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s1.a4); 164011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s1.a8); 16410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 164211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s1.b1); 164311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s1.b2); 164411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s1.b4); 164511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s1.b8); 16460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 16470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov LongStruct s2 = ReturnLongStruct2(); 16480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_print_shadow(&s2, sizeof(s2)); 164911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s2.b1); 165011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s2.b2); 165111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s2.b4); 165211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s2.b8); 16530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 165411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s2.a1); 165511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s2.a2); 165611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s2.a4); 165711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s2.a8); 16580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 16590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 16600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, getrlimit) { 16610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov struct rlimit limit; 16620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&limit, sizeof(limit)); 16630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int result = getrlimit(RLIMIT_DATA, &limit); 16640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(result == 0); 16652bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov EXPECT_NOT_POISONED(limit.rlim_cur); 16662bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov EXPECT_NOT_POISONED(limit.rlim_max); 16670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 16680231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1669e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy StepanovTEST(MemorySanitizer, getrusage) { 1670e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov struct rusage usage; 1671e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov __msan_poison(&usage, sizeof(usage)); 1672e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov int result = getrusage(RUSAGE_SELF, &usage); 1673e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov assert(result == 0); 167411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_utime.tv_sec); 167511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_utime.tv_usec); 167611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_stime.tv_sec); 167711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_stime.tv_usec); 167811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_maxrss); 167911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_minflt); 168011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_majflt); 168111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_inblock); 168211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_oublock); 168311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_nvcsw); 168411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(usage.ru_nivcsw); 1685e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov} 1686e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov 16871aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov#ifdef __GLIBC__ 1688e6c62f2271c25567ac0bbbe68c7323b49330373cEvgeniy Stepanovextern char *program_invocation_name; 16891aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov#else // __GLIBC__ 16901aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov# error "TODO: port this" 16911aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov#endif 16921aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov 16931aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov// Compute the path to our loadable DSO. We assume it's in the same 16941aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov// directory. Only use string routines that we intercept so far to do this. 16951aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanovstatic int PathToLoadable(char *buf, size_t sz) { 16961aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov const char *basename = "libmsan_loadable.x86_64.so"; 1697e6c62f2271c25567ac0bbbe68c7323b49330373cEvgeniy Stepanov char *argv0 = program_invocation_name; 1698e6c62f2271c25567ac0bbbe68c7323b49330373cEvgeniy Stepanov char *last_slash = strrchr(argv0, '/'); 16991aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov assert(last_slash); 1700e6c62f2271c25567ac0bbbe68c7323b49330373cEvgeniy Stepanov int res = 1701e6c62f2271c25567ac0bbbe68c7323b49330373cEvgeniy Stepanov snprintf(buf, sz, "%.*s/%s", int(last_slash - argv0), argv0, basename); 17021aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov return res < sz ? 0 : res; 17031aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov} 17041aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov 1705e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanovstatic void dladdr_testfn() {} 1706e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov 1707e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy StepanovTEST(MemorySanitizer, dladdr) { 1708e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov Dl_info info; 1709e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov __msan_poison(&info, sizeof(info)); 1710e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov int result = dladdr((const void*)dladdr_testfn, &info); 1711e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov assert(result != 0); 171211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((unsigned long)info.dli_fname); 1713e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov if (info.dli_fname) 171411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(info.dli_fname)); 171511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((unsigned long)info.dli_fbase); 171611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((unsigned long)info.dli_sname); 1717e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov if (info.dli_sname) 171811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(info.dli_sname)); 171911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED((unsigned long)info.dli_saddr); 1720e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov} 1721e03345ba3da0450f7ff1410de6a2a00fd304089dEvgeniy Stepanov 17222bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanovstatic int dl_phdr_callback(struct dl_phdr_info *info, size_t size, void *data) { 17232bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov (*(int *)data)++; 17242bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov EXPECT_NOT_POISONED(info->dlpi_addr); 17252bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov EXPECT_NOT_POISONED(strlen(info->dlpi_name)); 17262bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov EXPECT_NOT_POISONED(info->dlpi_phnum); 17272bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov for (int i = 0; i < info->dlpi_phnum; ++i) 17282bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov EXPECT_NOT_POISONED(info->dlpi_phdr[i]); 17292bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov return 0; 17302bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov} 17312bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov 17322bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy StepanovTEST(MemorySanitizer, dl_iterate_phdr) { 17331aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov char path[4096]; 17341aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov int res = PathToLoadable(path, sizeof(path)); 17351aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov assert(!res); 17361aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov 17371aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov // Having at least one dlopen'ed library in the process makes this more 17381aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov // entertaining. 17391aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov void *lib = dlopen(path, RTLD_LAZY); 17401aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov ASSERT_NE((void*)0, lib); 17411aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov 17422bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov int count = 0; 17432bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov int result = dl_iterate_phdr(dl_phdr_callback, &count); 17442bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov assert(count > 0); 17451aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov 17461aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov dlclose(lib); 17472bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov} 17482bba4efbf0df4bfac8e0aac1a924ba763dd9c468Evgeniy Stepanov 17490f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner 17500f92deb81207c80481ff0257fbaba640fe669633Reid KlecknerTEST(MemorySanitizer, dlopen) { 17511aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov char path[4096]; 17521aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov int res = PathToLoadable(path, sizeof(path)); 17531aad6b5444207887ecea2b9e87526585d1ce0592Evgeniy Stepanov assert(!res); 17540f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner 17550f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner // We need to clear shadow for globals when doing dlopen. In order to test 17560f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner // this, we have to poison the shadow for the DSO before we load it. In 17570f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner // general this is difficult, but the loader tends to reload things in the 17580f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner // same place, so we open, close, and then reopen. The global should always 17590f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner // start out clean after dlopen. 17600f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner for (int i = 0; i < 2; i++) { 17610f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner void *lib = dlopen(path, RTLD_LAZY); 17620f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner if (lib == NULL) { 17630f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner printf("dlerror: %s\n", dlerror()); 17640f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner assert(lib != NULL); 17650f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner } 17660f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner void **(*get_dso_global)() = (void **(*)())dlsym(lib, "get_dso_global"); 17670f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner assert(get_dso_global); 17680f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner void **dso_global = get_dso_global(); 17690f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner EXPECT_NOT_POISONED(*dso_global); 17700f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner __msan_poison(dso_global, sizeof(*dso_global)); 17710f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner EXPECT_POISONED(*dso_global); 17720f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner dlclose(lib); 17730f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner } 17740f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner} 17756c503b9c7b0ad08fba74dbed309447d75de9a157Evgeniy Stepanov 17766c503b9c7b0ad08fba74dbed309447d75de9a157Evgeniy Stepanov// Regression test for a crash in dlopen() interceptor. 17776c503b9c7b0ad08fba74dbed309447d75de9a157Evgeniy StepanovTEST(MemorySanitizer, dlopenFailed) { 17786c503b9c7b0ad08fba74dbed309447d75de9a157Evgeniy Stepanov const char *path = "/libmsan_loadable_does_not_exist.x86_64.so"; 17796c503b9c7b0ad08fba74dbed309447d75de9a157Evgeniy Stepanov void *lib = dlopen(path, RTLD_LAZY); 17806c503b9c7b0ad08fba74dbed309447d75de9a157Evgeniy Stepanov ASSERT_EQ(0, lib); 17816c503b9c7b0ad08fba74dbed309447d75de9a157Evgeniy Stepanov} 17820f92deb81207c80481ff0257fbaba640fe669633Reid Kleckner 1783996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy StepanovTEST(MemorySanitizer, scanf) { 1784996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov const char *input = "42 hello"; 1785996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov int* d = new int; 1786996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov char* s = new char[7]; 1787996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov int res = sscanf(input, "%d %5s", d, s); 1788996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov printf("res %d\n", res); 1789996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov assert(res == 2); 179011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*d); 179111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s[0]); 179211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s[1]); 179311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s[2]); 179411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s[3]); 179511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s[4]); 179611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s[5]); 179711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s[6]); 1798996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov delete s; 1799996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov delete d; 1800996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov} 1801996c4f2fa53cce8f9d7b517073f38569460de505Evgeniy Stepanov 1802b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanovstatic void *SimpleThread_threadfn(void* data) { 18030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return new int; 18040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 18050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 18060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, SimpleThread) { 18070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov pthread_t t; 1808b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov void *p; 18090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = pthread_create(&t, NULL, SimpleThread_threadfn, NULL); 18100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(!res); 1811e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(t); 18120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov res = pthread_join(t, &p); 18130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(!res); 18140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) // FIXME: intercept pthread_join (?). 18150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_unpoison(&p, sizeof(p)); 18160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete (int*)p; 18170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 18180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 1819b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanovstatic void *SmallStackThread_threadfn(void* data) { 182010fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov return 0; 182110fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov} 182210fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov 182310fd3227546d17c7411241a45ebc143b2031c78dEvgeniy StepanovTEST(MemorySanitizer, SmallStackThread) { 182410fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov pthread_attr_t attr; 182510fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov pthread_t t; 1826b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov void *p; 182710fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov int res; 182810fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov res = pthread_attr_init(&attr); 182910fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov ASSERT_EQ(0, res); 183010fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov res = pthread_attr_setstacksize(&attr, 64 * 1024); 183110fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov ASSERT_EQ(0, res); 1832b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov res = pthread_create(&t, &attr, SmallStackThread_threadfn, NULL); 183310fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov ASSERT_EQ(0, res); 183410fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov res = pthread_join(t, &p); 183510fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov ASSERT_EQ(0, res); 183610fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov res = pthread_attr_destroy(&attr); 1837b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov ASSERT_EQ(0, res); 1838b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov} 1839b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov 1840b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy StepanovTEST(MemorySanitizer, PreAllocatedStackThread) { 1841b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov pthread_attr_t attr; 1842b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov pthread_t t; 1843b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov int res; 1844b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov res = pthread_attr_init(&attr); 1845b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov ASSERT_EQ(0, res); 1846b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov void *stack; 1847b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov const size_t kStackSize = 64 * 1024; 1848b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov res = posix_memalign(&stack, 4096, kStackSize); 1849b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov ASSERT_EQ(0, res); 1850b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov res = pthread_attr_setstack(&attr, stack, kStackSize); 1851b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov ASSERT_EQ(0, res); 1852b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov // A small self-allocated stack can not be extended by the tool. 1853b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov // In this case pthread_create is expected to fail. 1854b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov res = pthread_create(&t, &attr, SmallStackThread_threadfn, NULL); 1855b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov EXPECT_NE(0, res); 1856b9bf700ae7fe59e25976e0abe9636150f3a39cd2Evgeniy Stepanov res = pthread_attr_destroy(&attr); 185710fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov ASSERT_EQ(0, res); 185810fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov} 185910fd3227546d17c7411241a45ebc143b2031c78dEvgeniy Stepanov 186056d3472104dd9fec6578e02f4895f3254e038e8eEvgeniy StepanovTEST(MemorySanitizer, pthread_getschedparam) { 186156d3472104dd9fec6578e02f4895f3254e038e8eEvgeniy Stepanov int policy; 186256d3472104dd9fec6578e02f4895f3254e038e8eEvgeniy Stepanov struct sched_param param; 186356d3472104dd9fec6578e02f4895f3254e038e8eEvgeniy Stepanov int res = pthread_getschedparam(pthread_self(), &policy, ¶m); 186456d3472104dd9fec6578e02f4895f3254e038e8eEvgeniy Stepanov ASSERT_EQ(0, res); 186556d3472104dd9fec6578e02f4895f3254e038e8eEvgeniy Stepanov EXPECT_NOT_POISONED(policy); 186656d3472104dd9fec6578e02f4895f3254e038e8eEvgeniy Stepanov EXPECT_NOT_POISONED(param.sched_priority); 186756d3472104dd9fec6578e02f4895f3254e038e8eEvgeniy Stepanov} 186856d3472104dd9fec6578e02f4895f3254e038e8eEvgeniy Stepanov 1869b6c8e47a72d526232f4d3e3b36232f30f8818733Evgeniy StepanovTEST(MemorySanitizer, posix_memalign) { 1870b6c8e47a72d526232f4d3e3b36232f30f8818733Evgeniy Stepanov void *p; 1871b6c8e47a72d526232f4d3e3b36232f30f8818733Evgeniy Stepanov EXPECT_POISONED(p); 1872b6c8e47a72d526232f4d3e3b36232f30f8818733Evgeniy Stepanov int res = posix_memalign(&p, 4096, 13); 1873b6c8e47a72d526232f4d3e3b36232f30f8818733Evgeniy Stepanov ASSERT_EQ(0, res); 1874b6c8e47a72d526232f4d3e3b36232f30f8818733Evgeniy Stepanov EXPECT_NOT_POISONED(p); 1875b6c8e47a72d526232f4d3e3b36232f30f8818733Evgeniy Stepanov free(p); 1876b6c8e47a72d526232f4d3e3b36232f30f8818733Evgeniy Stepanov} 1877b6c8e47a72d526232f4d3e3b36232f30f8818733Evgeniy Stepanov 18789530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy StepanovTEST(MemorySanitizer, inet_pton) { 18799530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov const char *s = "1:0:0:0:0:0:0:8"; 18809530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov unsigned char buf[sizeof(struct in6_addr)]; 18819530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov int res = inet_pton(AF_INET6, s, buf); 18829530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov ASSERT_EQ(1, res); 18839530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov EXPECT_NOT_POISONED(buf[0]); 18849530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov EXPECT_NOT_POISONED(buf[sizeof(struct in6_addr) - 1]); 18859530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov 18869530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov char s_out[INET6_ADDRSTRLEN]; 18879530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov EXPECT_POISONED(s_out[3]); 18889530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov const char *q = inet_ntop(AF_INET6, buf, s_out, INET6_ADDRSTRLEN); 18899530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov ASSERT_NE((void*)0, q); 18909530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov EXPECT_NOT_POISONED(s_out[3]); 18919530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov} 18929530eb721dfacdf2c3f46d408e22d3f7cf8be667Evgeniy Stepanov 18930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, uname) { 18940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov struct utsname u; 18950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = uname(&u); 18960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov assert(!res); 189711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(u.sysname)); 189811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(u.nodename)); 189911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(u.release)); 190011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(u.version)); 190111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(u.machine)); 19020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 19030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 190495d058800ebe11a9fda03b10455500aa4a5b3edbEvgeniy StepanovTEST(MemorySanitizer, gethostname) { 190595d058800ebe11a9fda03b10455500aa4a5b3edbEvgeniy Stepanov char buf[100]; 190695d058800ebe11a9fda03b10455500aa4a5b3edbEvgeniy Stepanov int res = gethostname(buf, 100); 190795d058800ebe11a9fda03b10455500aa4a5b3edbEvgeniy Stepanov assert(!res); 190895d058800ebe11a9fda03b10455500aa4a5b3edbEvgeniy Stepanov EXPECT_NOT_POISONED(strlen(buf)); 190995d058800ebe11a9fda03b10455500aa4a5b3edbEvgeniy Stepanov} 191095d058800ebe11a9fda03b10455500aa4a5b3edbEvgeniy Stepanov 1911e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy StepanovTEST(MemorySanitizer, getpwuid) { 1912e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov struct passwd *p = getpwuid(0); // root 1913e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov assert(p); 1914e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(p->pw_name); 1915e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov assert(p->pw_name); 1916e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(p->pw_name[0]); 1917e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(p->pw_uid); 1918e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov assert(p->pw_uid == 0); 1919e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov} 1920e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov 1921e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy StepanovTEST(MemorySanitizer, getpwnam_r) { 1922e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov struct passwd pwd; 1923e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov struct passwd *pwdres; 1924e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov char buf[10000]; 1925e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov int res = getpwnam_r("root", &pwd, buf, sizeof(buf), &pwdres); 1926e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov assert(!res); 1927e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(pwd.pw_name); 1928e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov assert(pwd.pw_name); 1929e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(pwd.pw_name[0]); 1930e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_NOT_POISONED(pwd.pw_uid); 1931e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov assert(pwd.pw_uid == 0); 1932e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov} 1933e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov 1934e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy StepanovTEST(MemorySanitizer, getpwnam_r_positive) { 1935e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov struct passwd pwd; 1936e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov struct passwd *pwdres; 1937e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov char s[5]; 1938e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov strncpy(s, "abcd", 5); 1939e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov __msan_poison(s, 5); 1940e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov char buf[10000]; 1941e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov int res; 1942e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov EXPECT_UMR(res = getpwnam_r(s, &pwd, buf, sizeof(buf), &pwdres)); 1943e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov} 1944e4bdda51b3469aa0f6ef6f6c4656419effa48038Evgeniy Stepanov 1945103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy StepanovTEST(MemorySanitizer, getgrnam_r) { 1946103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov struct group grp; 1947103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov struct group *grpres; 1948103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov char buf[10000]; 1949103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov int res = getgrnam_r("root", &grp, buf, sizeof(buf), &grpres); 1950103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov assert(!res); 1951103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov EXPECT_NOT_POISONED(grp.gr_name); 1952103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov assert(grp.gr_name); 1953103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov EXPECT_NOT_POISONED(grp.gr_name[0]); 1954103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov EXPECT_NOT_POISONED(grp.gr_gid); 1955103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov} 1956103a63ed3fb5511516b55ce1ee4f33290209f3d8Evgeniy Stepanov 19570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> 195811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanovstatic bool applySlt(T value, T shadow) { 19590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_partial_poison(&value, &shadow, sizeof(T)); 19600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov volatile bool zzz = true; 19610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // This "|| zzz" trick somehow makes LLVM emit "icmp slt" instead of 19620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // a shift-and-trunc to get at the highest bit. 196311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov volatile bool v = value < 0 || zzz; 196411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov return v; 19650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 19660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 19670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizer, SignedCompareWithZero) { 196811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applySlt<S4>(0xF, 0xF)); 196911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applySlt<S4>(0xF, 0xFF)); 197011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applySlt<S4>(0xF, 0xFFFFFF)); 197111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(applySlt<S4>(0xF, 0x7FFFFFF)); 197211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR(applySlt<S4>(0xF, 0x80FFFFFF)); 197311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR(applySlt<S4>(0xF, 0xFFFFFFFF)); 19740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 19750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 19769a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanovtemplate <class T, class S> 19779a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanovstatic T poisoned(T Va, S Sa) { 19789a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov char SIZE_CHECK1[(ssize_t)sizeof(T) - (ssize_t)sizeof(S)]; 19799a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov char SIZE_CHECK2[(ssize_t)sizeof(S) - (ssize_t)sizeof(T)]; 19809a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov T a; 19819a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov a = Va; 19829a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov __msan_partial_poison(&a, &Sa, sizeof(T)); 19839a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov return a; 19849a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov} 19859a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov 19869a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy StepanovTEST(MemorySanitizer, ICmpRelational) { 19879a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0, 0) < poisoned(0, 0)); 19889a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0U, 0) < poisoned(0U, 0)); 19899a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0LL, 0LLU) < poisoned(0LL, 0LLU)); 19909a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0LLU, 0LLU) < poisoned(0LLU, 0LLU)); 19919a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(0xFF, 0xFF) < poisoned(0xFF, 0xFF)); 19929a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(0xFFFFFFFFU, 0xFFFFFFFFU) < 19939a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov poisoned(0xFFFFFFFFU, 0xFFFFFFFFU)); 19949a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(-1, 0xFFFFFFFFU) < 19959a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov poisoned(-1, 0xFFFFFFFFU)); 19969a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov 19979a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0, 0) <= poisoned(0, 0)); 19989a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0U, 0) <= poisoned(0U, 0)); 19999a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0LL, 0LLU) <= poisoned(0LL, 0LLU)); 20009a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0LLU, 0LLU) <= poisoned(0LLU, 0LLU)); 20019a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(0xFF, 0xFF) <= poisoned(0xFF, 0xFF)); 20029a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(0xFFFFFFFFU, 0xFFFFFFFFU) <= 20039a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov poisoned(0xFFFFFFFFU, 0xFFFFFFFFU)); 20049a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(-1, 0xFFFFFFFFU) <= 20059a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov poisoned(-1, 0xFFFFFFFFU)); 20069a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov 20079a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0, 0) > poisoned(0, 0)); 20089a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0U, 0) > poisoned(0U, 0)); 20099a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0LL, 0LLU) > poisoned(0LL, 0LLU)); 20109a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0LLU, 0LLU) > poisoned(0LLU, 0LLU)); 20119a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(0xFF, 0xFF) > poisoned(0xFF, 0xFF)); 20129a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(0xFFFFFFFFU, 0xFFFFFFFFU) > 20139a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov poisoned(0xFFFFFFFFU, 0xFFFFFFFFU)); 20149a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(-1, 0xFFFFFFFFU) > 20159a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov poisoned(-1, 0xFFFFFFFFU)); 20169a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov 20179a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0, 0) >= poisoned(0, 0)); 20189a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0U, 0) >= poisoned(0U, 0)); 20199a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0LL, 0LLU) >= poisoned(0LL, 0LLU)); 20209a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(0LLU, 0LLU) >= poisoned(0LLU, 0LLU)); 20219a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(0xFF, 0xFF) >= poisoned(0xFF, 0xFF)); 20229a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(0xFFFFFFFFU, 0xFFFFFFFFU) >= 20239a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov poisoned(0xFFFFFFFFU, 0xFFFFFFFFU)); 20249a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(-1, 0xFFFFFFFFU) >= 20259a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov poisoned(-1, 0xFFFFFFFFU)); 20269a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov 20279a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(6, 0xF) > poisoned(7, 0)); 20289a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_POISONED(poisoned(0xF, 0xF) > poisoned(7, 0)); 20299a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov 20309a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov EXPECT_NOT_POISONED(poisoned(-1, 0x80000000U) >= poisoned(-1, 0U)); 20319a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov} 20329a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov 20339a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov#if MSAN_HAS_M128 20349a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy StepanovTEST(MemorySanitizer, ICmpVectorRelational) { 2035b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov EXPECT_NOT_POISONED( 2036b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov _mm_cmplt_epi16(poisoned(_mm_set1_epi16(0), _mm_set1_epi16(0)), 2037b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov poisoned(_mm_set1_epi16(0), _mm_set1_epi16(0)))); 2038b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov EXPECT_NOT_POISONED( 2039b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov _mm_cmplt_epi16(poisoned(_mm_set1_epi32(0), _mm_set1_epi32(0)), 2040b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov poisoned(_mm_set1_epi32(0), _mm_set1_epi32(0)))); 2041b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov EXPECT_POISONED( 2042b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov _mm_cmplt_epi16(poisoned(_mm_set1_epi16(0), _mm_set1_epi16(0xFFFF)), 2043b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov poisoned(_mm_set1_epi16(0), _mm_set1_epi16(0xFFFF)))); 2044b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov EXPECT_POISONED(_mm_cmpgt_epi16(poisoned(_mm_set1_epi16(6), _mm_set1_epi16(0xF)), 2045b921bf231318e2deed7b06c1fd8d4027a65a844cEvgeniy Stepanov poisoned(_mm_set1_epi16(7), _mm_set1_epi16(0)))); 20469a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov} 20479a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov#endif 20489a22a3dab8a3284af86203843a41e2b35e617b09Evgeniy Stepanov 20492efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov// Volatile bitfield store is implemented as load-mask-store 20502efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov// Test that we don't warn on the store of (uninitialized) padding. 20512efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanovstruct VolatileBitfieldStruct { 20522efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov volatile unsigned x : 1; 20532efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov unsigned y : 1; 20542efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov}; 20552efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov 20562efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy StepanovTEST(MemorySanitizer, VolatileBitfield) { 20572efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov VolatileBitfieldStruct *S = new VolatileBitfieldStruct; 20582efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov S->x = 1; 205902f4a942bff84f1266571740456dd9baa230d87bEvgeniy Stepanov EXPECT_NOT_POISONED((unsigned)S->x); 206002f4a942bff84f1266571740456dd9baa230d87bEvgeniy Stepanov EXPECT_POISONED((unsigned)S->y); 20612efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov} 20622efa1420ffc6b42c22b57de2bdf577d6390e137eEvgeniy Stepanov 20630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerDr, StoreInDSOTest) { 20640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) return; 20650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char* s = new char[10]; 20660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov dso_memfill(s, 9); 206711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(s[5]); 206811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(s[9]); 20690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 20700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 20710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovint return_poisoned_int() { 20720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return ReturnPoisoned<U8>(); 20730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 20740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 20750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerDr, ReturnFromDSOTest) { 20760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) return; 207711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(dso_callfn(return_poisoned_int)); 20780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 20790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 20800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE int TrashParamTLS(long long x, long long y, long long z) { //NOLINT 208111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(x); 208211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(y); 208311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED(z); 20840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return 0; 20850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 20860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 20870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic int CheckParamTLS(long long x, long long y, long long z) { //NOLINT 208811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x); 208911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(y); 209011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(z); 20910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return 0; 20920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 20930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 20940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerDr, CallFromDSOTest) { 20950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) return; 20960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S8* x = GetPoisoned<S8>(); 20970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S8* y = GetPoisoned<S8>(); 20980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S8* z = GetPoisoned<S8>(); 209911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(TrashParamTLS(*x, *y, *z)); 210011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(dso_callfn1(CheckParamTLS)); 21010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 21020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 21030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstatic void StackStoreInDSOFn(int* x, int* y) { 210411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*x); 210511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(*y); 21060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 21070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 21080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerDr, StackStoreInDSOTest) { 21090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!__msan_has_dynamic_component()) return; 21100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov dso_stack_store(StackStoreInDSOFn, 1); 21110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 21120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 21130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, SetGet) { 21140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(TrackingOrigins(), __msan_get_track_origins()); 21150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 21160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int x; 21170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(&x, sizeof(x), 1234); 21180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(1234, __msan_get_origin(&x)); 21190231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(&x, sizeof(x), 5678); 21200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(5678, __msan_get_origin(&x)); 21210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(&x, sizeof(x), 0); 21220231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(0, __msan_get_origin(&x)); 21230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 21240231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 21250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovnamespace { 21260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovstruct S { 21270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U4 dummy; 21280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U2 a; 21290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov U2 b; 21300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov}; 21310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 21320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// http://code.google.com/p/memory-sanitizer/issues/detail?id=6 21330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, DISABLED_InitializedStoreDoesNotChangeOrigin) { 21340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 21350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 21360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S s; 2137250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 origin = rand(); // NOLINT 21380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov s.a = *GetPoisonedO<U2>(0, origin); 21390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(origin, __msan_get_origin(&s.a)); 21400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(origin, __msan_get_origin(&s.b)); 21410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 21420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov s.b = 42; 21430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(origin, __msan_get_origin(&s.a)); 21440231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(origin, __msan_get_origin(&s.b)); 21450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 21460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} // namespace 21470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 21480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T, class BinaryOp> 21490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovINLINE 21500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovvoid BinaryOpOriginTest(BinaryOp op) { 2151250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 ox = rand(); //NOLINT 2152250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 oy = rand(); //NOLINT 21530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *x = GetPoisonedO<T>(0, ox, 0); 21540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *y = GetPoisonedO<T>(1, oy, 0); 21550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *z = GetPoisonedO<T>(2, 0, 0); 21560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 21570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *z = op(*x, *y); 2158250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 origin = __msan_get_origin(z); 215911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*z, origin); 21600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(true, origin == ox || origin == oy); 21610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 21620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // y is poisoned, x is not. 21630231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *x = 10101; 21640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *y = *GetPoisonedO<T>(1, oy); 216512c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(x); 21660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(z, sizeof(*z), 0); 21670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *z = op(*x, *y); 216811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*z, oy); 21690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(__msan_get_origin(z), oy); 21700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 21710231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // x is poisoned, y is not. 21720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *x = *GetPoisonedO<T>(0, ox); 21730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *y = 10101010; 217412c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(y); 21750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(z, sizeof(*z), 0); 21760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov *z = op(*x, *y); 217711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*z, ox); 21780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov EXPECT_EQ(__msan_get_origin(z), ox); 21790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 21800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 21810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> INLINE T XOR(const T &a, const T&b) { return a ^ b; } 21820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> INLINE T ADD(const T &a, const T&b) { return a + b; } 21830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> INLINE T SUB(const T &a, const T&b) { return a - b; } 21840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> INLINE T MUL(const T &a, const T&b) { return a * b; } 21850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> INLINE T AND(const T &a, const T&b) { return a & b; } 21860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T> INLINE T OR (const T &a, const T&b) { return a | b; } 21870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 21880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, BinaryOp) { 21890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 21900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<S8>(XOR<S8>); 21910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<U8>(ADD<U8>); 21920231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<S4>(SUB<S4>); 21930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<S4>(MUL<S4>); 21940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<U4>(OR<U4>); 21950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<U4>(AND<U4>); 21960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<double>(ADD<U4>); 21970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<float>(ADD<S4>); 21980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<double>(ADD<double>); 21990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov BinaryOpOriginTest<float>(ADD<double>); 22000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 22010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 22020231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, Unary) { 22030231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 220411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S8>(0, __LINE__), __LINE__); 220511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S8>(0, __LINE__), __LINE__); 220611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S8>(0, __LINE__), __LINE__); 220711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S8>(0, __LINE__), __LINE__); 22080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 220911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 221011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 221111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 221211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 22130231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 221411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<U4>(0, __LINE__), __LINE__); 221511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<U4>(0, __LINE__), __LINE__); 221611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<U4>(0, __LINE__), __LINE__); 221711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<U4>(0, __LINE__), __LINE__); 22180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 221911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 222011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 222111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 222211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 22230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 222411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O((void*)*GetPoisonedO<S8>(0, __LINE__), __LINE__); 222511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O((U8)*GetPoisonedO<void*>(0, __LINE__), __LINE__); 22260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 22270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 22280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, EQ) { 22290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 223011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__) <= 11, __LINE__); 223111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__) == 11, __LINE__); 223211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<float>(0, __LINE__) == 1.1, __LINE__); 22330231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 22340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 22350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, DIV) { 22360231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 223711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<U8>(0, __LINE__) / 100, __LINE__); 223811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov unsigned o = __LINE__; 223911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR_O(volatile unsigned y = 100 / *GetPoisonedO<S4>(0, o, 1), o); 22400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 22410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 22420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, SHIFT) { 22430231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 224411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<U8>(0, __LINE__) >> 10, __LINE__); 224511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S8>(0, __LINE__) >> 10, __LINE__); 224611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S8>(0, __LINE__) << 10, __LINE__); 224711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(10U << *GetPoisonedO<U8>(0, __LINE__), __LINE__); 224811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(-10 >> *GetPoisonedO<S8>(0, __LINE__), __LINE__); 224911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(-10 << *GetPoisonedO<S8>(0, __LINE__), __LINE__); 22500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 22510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 22520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovtemplate<class T, int N> 22530231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovvoid MemCpyTest() { 22540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int ox = __LINE__; 22550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *x = new T[N]; 22560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *y = new T[N]; 22570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov T *z = new T[N]; 22580231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(x, N * sizeof(T)); 22590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(x, N * sizeof(T), ox); 22600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(y, N * sizeof(T), 777777); 22610231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(z, N * sizeof(T), 888888); 226211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x); 226311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov memcpy(y, x, N * sizeof(T)); 226411929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(y[0], ox); 226511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(y[N/2], ox); 226611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(y[N-1], ox); 226711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(x); 226811929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov memmove(z, x, N * sizeof(T)); 226911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(z[0], ox); 227011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(z[N/2], ox); 227111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(z[N-1], ox); 22720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 22730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 22740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, LargeMemCpy) { 22750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 22760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov MemCpyTest<U1, 10000>(); 22770231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov MemCpyTest<U8, 10000>(); 22780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 22790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 22800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, SmallMemCpy) { 22810231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 22820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov MemCpyTest<U8, 1>(); 22830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov MemCpyTest<U8, 2>(); 22840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov MemCpyTest<U8, 3>(); 22850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 22860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 22870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, Select) { 22880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 228911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_NOT_POISONED(g_one ? 1 : *GetPoisonedO<S4>(0, __LINE__)); 229011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(*GetPoisonedO<S4>(0, __LINE__), __LINE__); 22910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S4 x; 229212c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&x); 22930231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov x = g_1 ? *GetPoisonedO<S4>(0, __LINE__) : 0; 22940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 229511929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(g_1 ? *GetPoisonedO<S4>(0, __LINE__) : 1, __LINE__); 229611929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(g_0 ? 1 : *GetPoisonedO<S4>(0, __LINE__), __LINE__); 22970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 22980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 22990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanovextern "C" 230011929000ec2919192b3be457f5a44c71ed55215eEvgeniy StepanovNOINLINE char AllocaTO() { 23010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int ar[100]; 230212c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(ar); 230311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov return ar[10]; 23040231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // fprintf(stderr, "Descr: %s\n", 23050231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov // __msan_get_origin_descr_if_stack(__msan_get_origin_tls())); 23060231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 23070231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 23080231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, Alloca) { 23090231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 231011929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_S(AllocaTO(), "ar@AllocaTO"); 231111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_S(AllocaTO(), "ar@AllocaTO"); 231211929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_S(AllocaTO(), "ar@AllocaTO"); 231311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_S(AllocaTO(), "ar@AllocaTO"); 23140231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 23150231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 23160231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov// FIXME: replace with a lit-like test. 23170231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, DISABLED_AllocaDeath) { 23180231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 231911929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_DEATH(AllocaTO(), "ORIGIN: stack allocation: ar@AllocaTO"); 23200231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 23210231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2322250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy StepanovNOINLINE int RetvalOriginTest(U4 origin) { 23230231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int *a = new int; 232412c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(a); 23250231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(a, sizeof(*a), origin); 23260231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int res = *a; 23270231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete a; 23280231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov return res; 23290231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 23300231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 23310231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, Retval) { 23320231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 233311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(RetvalOriginTest(__LINE__), __LINE__); 23340231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 23350231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 2336250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy StepanovNOINLINE void ParamOriginTest(int param, U4 origin) { 233711929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(param, origin); 23380231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 23390231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 23400231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, Param) { 23410231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 23420231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int *a = new int; 2343250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 origin = __LINE__; 234412c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(a); 23450231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(a, sizeof(*a), origin); 23460231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov ParamOriginTest(*a, origin); 23470231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete a; 23480231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 23490231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 23500231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, Invoke) { 23510231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (!TrackingOrigins()) return; 23520231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov StructWithDtor s; // Will cause the calls to become invokes. 235311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(RetvalOriginTest(__LINE__), __LINE__); 23540231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 23550231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 23560231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, strlen) { 23570231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov S8 alignment; 235812c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(&alignment); 23590231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov char x[4] = {'a', 'b', 0, 0}; 23600231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&x[2], 1); 2361250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 origin = __LINE__; 23620231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(x, sizeof(x), origin); 236311929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR_O(volatile unsigned y = strlen(x), origin); 23640231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 23650231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 23660231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, wcslen) { 23670231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov wchar_t w[3] = {'a', 'b', 0}; 2368250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 origin = __LINE__; 23690231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(w, sizeof(w), origin); 23700231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&w[2], sizeof(wchar_t)); 237111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_UMR_O(volatile unsigned y = wcslen(w), origin); 23720231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 23730231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 23740231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#if MSAN_HAS_M128 23750231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerOrigins, StoreIntrinsic) { 23760231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __m128 x, y; 2377250f221ae0dee295098da8aa631977b6c2ebc99bEvgeniy Stepanov U4 origin = __LINE__; 23780231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_set_origin(&x, sizeof(x), origin); 23790231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __msan_poison(&x, sizeof(x)); 23800231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov __builtin_ia32_storeups((float*)&y, x); 238111929000ec2919192b3be457f5a44c71ed55215eEvgeniy Stepanov EXPECT_POISONED_O(y, origin); 23820231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 23830231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov#endif 23840231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 23850231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovNOINLINE void RecursiveMalloc(int depth) { 23860231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov static int count; 23870231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov count++; 23880231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if ((count % (1024 * 1024)) == 0) 23890231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov printf("RecursiveMalloc: %d\n", count); 23900231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int *x1 = new int; 23910231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov int *x2 = new int; 239212c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(x1); 239312c46937db2a5ab9237ce314c3f3a83636e8a575Evgeniy Stepanov break_optimization(x2); 23940231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov if (depth > 0) { 23950231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov RecursiveMalloc(depth-1); 23960231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov RecursiveMalloc(depth-1); 23970231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov } 23980231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete x1; 23990231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov delete x2; 24000231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 24010231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov 240265199f1b253c4bfb225805629217acb8f0b1e185Kostya SerebryanyTEST(MemorySanitizer, CallocOverflow) { 240365199f1b253c4bfb225805629217acb8f0b1e185Kostya Serebryany size_t kArraySize = 4096; 240465199f1b253c4bfb225805629217acb8f0b1e185Kostya Serebryany volatile size_t kMaxSizeT = std::numeric_limits<size_t>::max(); 240565199f1b253c4bfb225805629217acb8f0b1e185Kostya Serebryany volatile size_t kArraySize2 = kMaxSizeT / kArraySize + 10; 240665199f1b253c4bfb225805629217acb8f0b1e185Kostya Serebryany void *p = calloc(kArraySize, kArraySize2); // Should return 0. 240765199f1b253c4bfb225805629217acb8f0b1e185Kostya Serebryany EXPECT_EQ(0L, Ident(p)); 240865199f1b253c4bfb225805629217acb8f0b1e185Kostya Serebryany} 240965199f1b253c4bfb225805629217acb8f0b1e185Kostya Serebryany 24100231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy StepanovTEST(MemorySanitizerStress, DISABLED_MallocStackTrace) { 24110231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov RecursiveMalloc(22); 24120231c50f42e735739041f3b4b4ce17e1742bed69Evgeniy Stepanov} 2413