Files
clang-p2996/compiler-rt/test/asan/TestCases/Windows/dll_host.cpp
Charlie Barto 0360f3218a [asan][windows] Make tests more flexable (#85274)
Contains test changes from
https://github.com/llvm/llvm-project/pull/81677 that are simple test
changes. For the most part just makes the tests allow more flexibility
in the callstacks produced by asan.

Note: this PR has the exact changes from
https://github.com/llvm/llvm-project/pull/81677 in addition to a revert
commit to remove the ones that require other things from that PR. This
will be squashed, ofc. I just left things unsquashed for reviewing
pleasure :).

This is a non-functional-change
2024-03-15 10:42:21 -07:00

52 lines
1.2 KiB
C++

// UNSUPPORTED: target={{.*-windows-gnu}}
// 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 -Od %s -Fe%t
#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) {
DWORD err = GetLastError();
printf("Could not load DLL: %s (code: %lu)!\n", dll_name, err);
LPSTR buf;
FormatMessageA(FORMAT_MESSAGE_ALLOCATE_BUFFER | FORMAT_MESSAGE_FROM_SYSTEM |
FORMAT_MESSAGE_IGNORE_INSERTS,
NULL, err, MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), buf, 0,
NULL);
printf("Error: %s\n", buf);
LocalFree(buf);
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;
}