1211

test/Detours/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