1211

test/Detours/samples/dumpe/Makefile

@@ -0,0 +1,53 @@
+##############################################################################
+##
+##  Makefile for Detours Test Programs.
+##
+##  Microsoft Research Detours Package
+##
+##  Copyright (c) Microsoft Corporation.  All rights reserved.
+##
+
+!include ..\common.mak
+
+LIBS=$(LIBS) kernel32.lib
+
+all: dirs \
+    $(BIND)\dumpe.exe \
+!IF $(DETOURS_SOURCE_BROWSING)==1
+    $(OBJD)\dumpe.bsc
+!ENDIF
+
+clean:
+    -del *~ 2>nul
+    -del $(BIND)\dumpe.* 2>nul
+    -rmdir /q /s $(OBJD) 2>nul
+
+realclean: clean
+    -rmdir /q /s $(OBJDS) 2>nul
+
+dirs:
+    @if not exist $(BIND) mkdir $(BIND) && echo.   Created $(BIND)
+    @if not exist $(OBJD) mkdir $(OBJD) && echo.   Created $(OBJD)
+
+$(OBJD)\dumpe.obj : dumpe.cpp
+
+$(BIND)\dumpe.exe : $(OBJD)\dumpe.obj $(DEPS)
+    cl $(CFLAGS) /Fe$@ /Fd$(@R).pdb $(OBJD)\dumpe.obj \
+        /link $(LINKFLAGS) $(LIBS) \
+        /subsystem:console
+
+$(OBJD)\dumpe.bsc : $(OBJD)\dumpe.obj
+    bscmake /v /n /o $@ $(OBJD)\dumpe.sbr
+
+##############################################################################
+
+test: $(BIND)\dumpe.exe
+    $(BIND)\dumpe.exe $(BIND)\slept.dll
+
+testx: $(BIND)\dumpe.exe
+    cd $(MAKEDIR)\..\..\src
+    nmake
+    cd $(MAKEDIR)
+    if exist $(SYSTEMROOT)\system32\browseui.dll $(BIND)\dumpe.exe browseui.dll
+
+################################################################# End of File.

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