aboutsummaryrefslogtreecommitdiffhomepage
path: root/samples/dumpe/dumpe.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'samples/dumpe/dumpe.cpp')
-rw-r--r--samples/dumpe/dumpe.cpp129
1 files changed, 129 insertions, 0 deletions
diff --git a/samples/dumpe/dumpe.cpp b/samples/dumpe/dumpe.cpp
new file mode 100644
index 0000000..2eac099
--- /dev/null
+++ b/samples/dumpe/dumpe.cpp
@@ -0,0 +1,129 @@
+//////////////////////////////////////////////////////////////////////////////
+//
+// Detours Test Program (dumpe.cpp of dumpe.exe)
+//
+// Microsoft Research Detours Package
+//
+// Copyright (c) Microsoft Corporation. All rights reserved.
+//
+#include <stdio.h>
+#include <stdlib.h>
+#include <windows.h>
+#include <ole2.h>
+#include <shellapi.h>
+#include "detours.h"
+
+//////////////////////////////////////////////////////////////////////////////
+//
+#ifndef NODEBUG
+#undef ASSERT
+VOID DetourAssertMessage(CONST PCHAR szMsg, CONST PCHAR szFile, DWORD nLine);
+
+#define ASSERT(x) \
+do { if (!(x)) { DetourAssertMessage(#x, __FILE__, __LINE__); DebugBreak(); }} while (0)
+ ;
+#undef ASSERTX
+#define ASSERTX(x) \
+do { if (!(x)) { DetourAssertMessage(#x, __FILE__, __LINE__); PCHAR p=(PCHAR)(x); *p = 1; }} while (0)
+ ;
+#else // NODEBUG
+#undef ASSERT
+#define ASSERT(x)
+#undef ASSERTX
+#define ASSERTX(x)
+#endif // NODEBUG
+//
+//////////////////////////////////////////////////////////////////////////////
+
+////////////////////////////////////////////////////////////// Error Messages.
+//
+VOID DetourAssertMessage(CONST PCHAR szMsg, CONST PCHAR szFile, DWORD nLine)
+{
+ printf("ASSERT(%s) failed in %s, line %ld.", szMsg, szFile, nLine);
+}
+
+
+
+static BOOL CALLBACK ExportCallback(PVOID pContext,
+ ULONG nOrdinal,
+ LPCSTR pszSymbol,
+ PVOID pbTarget)
+{
+ (void)pContext;
+
+ printf(" %7ld %p %-30s\n",
+ (ULONG)nOrdinal,
+ pbTarget,
+ pszSymbol ? pszSymbol : "[NONAME]");
+ return TRUE;
+}
+
+BOOL DumpFile(PCHAR pszPath)
+{
+ HINSTANCE hInst = LoadLibraryA(pszPath);
+ if (hInst == NULL) {
+ printf("Unable to load %s: Error %ld\n", pszPath, GetLastError());
+ return FALSE;
+ }
+
+ printf("%s @ %p\n", pszPath, hInst);
+
+ PVOID pbEntry = DetourGetEntryPoint(hInst);
+ printf(" EntryPoint: %p\n", pbEntry);
+
+ printf(" Ordinal RVA Name\n");
+ DetourEnumerateExports(hInst, NULL, ExportCallback);
+
+ return TRUE;
+}
+
+//////////////////////////////////////////////////////////////////////////////
+//
+void PrintUsage(void)
+{
+ printf("Usage:\n"
+ " dumpe [.dll files]\n"
+ "Misc. Options:\n"
+ " /? : Help screen.\n");
+}
+
+//////////////////////////////////////////////////////////////////////// main.
+//
+int CDECL main(int argc, char **argv)
+{
+ BOOL fNeedHelp = FALSE;
+
+ int arg = 1;
+ for (; arg < argc; arg++) {
+ if (argv[arg][0] == '-' || argv[arg][0] == '/') {
+ CHAR *argn = argv[arg] + 1;
+ CHAR *argp = argn;
+ while (*argp && *argp != ':')
+ argp++;
+ if (*argp == ':')
+ *argp++ = '\0';
+
+ switch (argn[0]) {
+
+ case '?': // Help.
+ fNeedHelp = TRUE;
+ break;
+
+ default:
+ fNeedHelp = TRUE;
+ printf("Bad argument: %s:%s\n", argn, argp);
+ break;
+ }
+ }
+ else {
+ DumpFile(argv[arg]);
+ }
+ }
+ if (fNeedHelp || argc == 1) {
+ PrintUsage();
+ return 1;
+ }
+ return 0;
+}
+
+// End of File