| // This is a host program for DLL tests. |
| // |
| // Just make sure we can compile this. |
| // The actual compile&run sequence is to be done by the DLL tests. |
| // RUN: %clang_cl_asan -O0 %s -Fe%t |
| // |
| // Get the list of ASan wrappers exported by the main module RTL: |
| // note: The mangling decoration (i.e. @4 )is removed because calling convention |
| // differ from 32-bit and 64-bit. |
| // RUN: dumpbin /EXPORTS %t | grep -o "__asan_wrap[^ ]*" | sed -e s/@.*// > %t.exports1 |
| // |
| // The exception handlers differ in 32-bit and 64-bit, so we ignore them: |
| // RUN: grep '[E]XPORT:' %s | sed -e 's/.*[E]XPORT: //' > %t.exports2 |
| // EXPORT: __asan_wrap__except_handler3 |
| // EXPORT: __asan_wrap__except_handler4 |
| // EXPORT: __asan_wrap___C_specific_handler |
| // |
| // Get the list of ASan wrappers imported by the DLL RTL: |
| // [BEWARE: be really careful with the sed commands, as this test can be run |
| // from different environemnts with different shells and seds] |
| // RUN: grep INTERCEPT_LIBRARY_FUNCTION %p/../../../../lib/asan/asan_win_dll_thunk.cc \ |
| // RUN: | grep -v define | sed -e s/.*(/__asan_wrap_/ -e s/).*// \ |
| // RUN: > %t.imports1 |
| // |
| // Add functions interecepted in asan_malloc.win.cc and asan_win.cc. |
| // RUN: grep '[I]MPORT:' %s | sed -e 's/.*[I]MPORT: //' > %t.imports2 |
| // IMPORT: __asan_wrap_HeapAlloc |
| // IMPORT: __asan_wrap_HeapFree |
| // IMPORT: __asan_wrap_HeapReAlloc |
| // IMPORT: __asan_wrap_HeapSize |
| // IMPORT: __asan_wrap_CreateThread |
| // IMPORT: __asan_wrap_RaiseException |
| // IMPORT: __asan_wrap_RtlRaiseException |
| // IMPORT: __asan_wrap_SetUnhandledExceptionFilter |
| // |
| // RUN: cat %t.imports1 %t.imports2 | sort | uniq > %t.imports-sorted |
| // RUN: cat %t.exports1 %t.exports2 | sort | uniq > %t.exports-sorted |
| // |
| // Now make sure the DLL thunk imports everything: |
| // RUN: echo |
| // RUN: echo "=== NOTE === If you see a mismatch below, please update asan_win_dll_thunk.cc" |
| // RUN: diff %t.imports-sorted %t.exports-sorted |
| // REQUIRES: asan-static-runtime |
| |
| #include <stdio.h> |
| #include <windows.h> |
| |
| int main(int argc, char **argv) { |
| if (argc != 2) { |
| printf("Usage: %s [client].dll\n", argv[0]); |
| return 101; |
| } |
| |
| const char *dll_name = argv[1]; |
| |
| HMODULE h = LoadLibrary(dll_name); |
| if (!h) { |
| printf("Could not load DLL: %s (code: %lu)!\n", |
| dll_name, GetLastError()); |
| return 102; |
| } |
| |
| typedef int (*test_function)(); |
| test_function gf = (test_function)GetProcAddress(h, "test_function"); |
| if (!gf) { |
| printf("Could not locate test_function in the DLL!\n"); |
| FreeLibrary(h); |
| return 103; |
| } |
| |
| int ret = gf(); |
| |
| FreeLibrary(h); |
| return ret; |
| } |