blob: 0d9051ba70f2e22db5fc0ef1365e762904a79afc (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
|
// Test indirect call wrapping in MemorySanitizer runtime.
// RUN: %clangxx_msan -O0 -g -rdynamic %s -o %t && %t
#include <assert.h>
#include <math.h>
#include <pthread.h>
#include <stdio.h>
#include <stdint.h>
#include <sys/time.h>
extern "C" void __msan_set_indirect_call_wrapper(uintptr_t);
bool pthread_create_done;
void *ThreadFn(void *) {
printf("bad threadfn\n");
return 0;
}
void *ThreadFn2(void *) {
printf("good threadfn\n");
pthread_create_done = true;
return 0;
}
bool in_gettimeofday;
bool in_lgamma;
int my_gettimeofday(struct timeval *p, void *q) {
p->tv_sec = 1;
p->tv_usec = 2;
return 42;
}
double my_lgamma(double x) {
printf("zzz\n");
return x;
}
extern "C" uintptr_t my_wrapper(uintptr_t f) {
if (f == (uintptr_t)ThreadFn)
return (uintptr_t)&ThreadFn2;
if (in_gettimeofday)
return (uintptr_t)my_gettimeofday;
if (in_lgamma)
return (uintptr_t)my_lgamma;
return f;
}
int main(void) {
__msan_set_indirect_call_wrapper((uintptr_t)my_wrapper);
// ThreadFn is called indirectly from a wrapper function in MSan rtl and
// is subject to indirect call wrapping (it could be an native-to-translated
// edge).
pthread_t t;
pthread_create(&t, 0, ThreadFn, 0);
pthread_join(t, 0);
assert(pthread_create_done);
// gettimeofday is intercepted in msan_interceptors.cc and the real one (from
// libc) is called indirectly.
struct timeval tv;
in_gettimeofday = true;
int res = gettimeofday(&tv, NULL);
in_gettimeofday = false;
assert(tv.tv_sec == 1);
assert(tv.tv_usec == 2);
assert(res == 42);
// lgamma is intercepted in sanitizer_common_interceptors.inc and is also
// called indirectly.
in_lgamma = true;
double dres = lgamma(1.1);
in_lgamma = false;
assert(dres == 1.1);
return 0;
}
|
