summaryrefslogtreecommitdiff
path: root/test/lsan/TestCases/Linux/libdl_deadlock.cpp
blob: 7c1f82f10beef8b232d3cf0dac92b3f063a67b6d (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
// Regression test for a deadlock in leak detection,
// where lsan would call dl_iterate_phdr while holding the allocator lock.
// RUN: %clangxx_lsan %s -o %t && %run %t

#include <link.h>
#include <mutex>
#include <stdlib.h>
#include <thread>
#include <unistd.h>

std::mutex in, out;

int Callback(struct dl_phdr_info *info, size_t size, void *data) {
  for (int step = 0; step < 50; ++step) {
    void *p[1000];
    for (int i = 0; i < 1000; ++i)
      p[i] = malloc(10 * i);

    if (step == 0)
      in.unlock();

    for (int i = 0; i < 1000; ++i)
      free(p[i]);
  }
  out.unlock();
  return 1; // just once
}

void Watchdog() {
  // This is just a fail-safe to turn a deadlock (in case the bug reappears)
  // into a (slow) test failure.
  usleep(10000000);
  if (!out.try_lock()) {
    write(2, "DEADLOCK\n", 9);
    exit(1);
  }
}

int main() {
  in.lock();
  out.lock();

  std::thread t([] { dl_iterate_phdr(Callback, nullptr); });
  t.detach();

  std::thread w(Watchdog);
  w.detach();

  // Wait for the malloc thread to preheat, then start leak detection (on exit)
  in.lock();
  return 0;
}