aboutsummaryrefslogtreecommitdiff
path: root/test/asan/TestCases/Windows/bind_io_completion_callback.cc
blob: c062a799fdccd317d7d263a6a056ea91903863dc (plain) (blame)
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
// Make sure we can throw exceptions from work items executed via
// BindIoCompletionCallback.
//
// Clang doesn't support exceptions on Windows yet, so for the time being we
// build this program in two parts: the code with exceptions is built with CL,
// the rest is built with Clang.  This represents the typical scenario when we
// build a large project using "clang-cl -fallback -fsanitize=address".
//
// RUN: cl -c %s -Fo%t.obj
// RUN: %clangxx_asan -o %t.exe %s %t.obj
// RUN: %run %t.exe 2>&1 | FileCheck %s

#include <windows.h>
#include <stdio.h>

void ThrowAndCatch();

#if !defined(__clang__)
__declspec(noinline)
void Throw() {
  fprintf(stderr, "Throw\n");
// CHECK: Throw
  throw 1;
}

void ThrowAndCatch() {
  int local;
  try {
    Throw();
  } catch(...) {
    fprintf(stderr, "Catch\n");
// CHECK: Catch
  }
}
#else

char buffer[65536];
HANDLE done;
OVERLAPPED ov;

void CALLBACK completion_callback(DWORD error, DWORD bytesRead,
                                  LPOVERLAPPED pov) {
  ThrowAndCatch();
  SetEvent(done);
}

int main(int argc, char **argv) {
  done = CreateEvent(0, false, false, "job is done");
  if (!done)
    return 1;
  HANDLE file = CreateFile(
      argv[0], GENERIC_READ, FILE_SHARE_READ | FILE_SHARE_WRITE, NULL,
      OPEN_EXISTING,
      FILE_ATTRIBUTE_NORMAL | FILE_FLAG_NO_BUFFERING | FILE_FLAG_OVERLAPPED,
      NULL);
  if (!file)
    return 2;
  if (!BindIoCompletionCallback(file, completion_callback, 0))
    return 3;

  if (!ReadFile(file, buffer, sizeof(buffer), NULL, &ov) &&
      GetLastError() != ERROR_IO_PENDING)
    return 4;

  if (WAIT_OBJECT_0 != WaitForSingleObject(done, INFINITE))
    return 5;
  fprintf(stderr, "Done!\n");
// CHECK: Done!
}
#endif