blob: cbd9a0867c982f241834438e898726331452af84 (
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
|
// Test that dylibs interposing write, and then calling functions intercepted
// by TSan don't deadlock (self-lock)
// RUN: %clang_tsan %s -o %t
// RUN: %clang_tsan %s -o %t.dylib -fno-sanitize=thread -dynamiclib -DSHARED_LIB
// Note that running the below command with out `lock_during_write` should
// deadlock (self-lock)
// RUN: env DYLD_INSERT_LIBRARIES=%t.dylib TSAN_OPTIONS=verbosity=2:lock_during_write=disable_for_current_process %run %t 2>&1 | FileCheck %s
#include <stdio.h>
#if defined(SHARED_LIB)
// dylib implementation - interposes write() calls
# include <os/lock.h>
# include <unistd.h>
struct interpose_substitution {
const void *replacement;
const void *original;
};
# define INTERPOSE(replacement, original) \
__attribute__((used)) static const struct interpose_substitution \
substitution_##original[] \
__attribute__((section("__DATA, __interpose"))) = { \
{(const void *)(replacement), (const void *)(original)}}
static ssize_t my_write(int fd, const void *buf, size_t count) {
struct os_unfair_lock_s lock = OS_UNFAIR_LOCK_INIT;
os_unfair_lock_lock(&lock);
printf("Interposed write called: fd=%d, count=%zu\n", fd, count);
ssize_t res = write(fd, buf, count);
os_unfair_lock_unlock(&lock);
return res;
}
INTERPOSE(my_write, write);
#else // defined(SHARED_LIB)
int main() {
printf("Write test completed\n");
return 0;
}
#endif // defined(SHARED_LIB)
// CHECK: Interposed write called: fd={{[0-9]+}}, count={{[0-9]+}}
// CHECK: Write test completed
|
