17c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar// Make sure we can throw exceptions from work items executed via
27c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar// QueueUserWorkItem.
37c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar//
47c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar// Clang doesn't support exceptions on Windows yet, so for the time being we
57c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar// build this program in two parts: the code with exceptions is built with CL,
67c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar// the rest is built with Clang.  This represents the typical scenario when we
77c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar// build a large project using "clang-cl -fallback -fsanitize=address".
87c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar//
97c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar// RUN: cl -c %s -Fo%t.obj
107c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar// RUN: %clangxx_asan -o %t.exe %s %t.obj
117c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar// RUN: %run %t.exe 2>&1 | FileCheck %s
127c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar
137c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar#include <windows.h>
147c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar#include <stdio.h>
157c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar
167c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainarvoid ThrowAndCatch();
177c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar
187c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar#if !defined(__clang__)
197c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar__declspec(noinline)
207c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainarvoid Throw() {
217c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar  fprintf(stderr, "Throw\n");
227c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar// CHECK: Throw
237c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar  throw 1;
247c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar}
257c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar
267c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainarvoid ThrowAndCatch() {
277c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar  int local;
287c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar  try {
297c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar    Throw();
307c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar  } catch(...) {
317c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar    fprintf(stderr, "Catch\n");
327c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar// CHECK: Catch
337c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar  }
347c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar}
357c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar#else
367c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar
377c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga NainarHANDLE done;
387c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar
397c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga NainarDWORD CALLBACK work_item(LPVOID) {
407c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar  ThrowAndCatch();
417c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar  SetEvent(done);
427c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar  return 0;
437c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar}
447c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar
457c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainarint main(int argc, char **argv) {
467c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar  done = CreateEvent(0, false, false, "job is done");
477c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar  if (!done)
487c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar    return 1;
497c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar  QueueUserWorkItem(&work_item, nullptr, 0);
507c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar  if (WAIT_OBJECT_0 != WaitForSingleObject(done, INFINITE))
517c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar    return 2;
527c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar  fprintf(stderr, "Done!\n");
537c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar// CHECK: Done!
547c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar}
557c9150579ed0278492f51cc8434b1d63a44b9bd1Pirama Arumuga Nainar#endif
56