summaryrefslogtreecommitdiff
path: root/test/asan/TestCases/Windows/bind_io_completion_callback.cc
blob: ef7e45867ede0dcc38ac029fe03bfc9cb6a80f09 (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
// Make sure we can throw exceptions from work items executed via
// BindIoCompletionCallback.
//
// RUN: %clangxx_asan %s -o %t.exe
// RUN: %run %t.exe 2>&1 | FileCheck %s

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

void ThrowAndCatch();

__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
  }
}

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, 10 * 1000))
    return 5;
  fprintf(stderr, "Done!\n");
// CHECK: Done!
}