128 lines
3.1 KiB
C
128 lines
3.1 KiB
C
/*
|
|
* Process Hacker -
|
|
* PE viewer
|
|
*
|
|
* Copyright (C) 2010 wj32
|
|
*
|
|
* This file is part of Process Hacker.
|
|
*
|
|
* Process Hacker is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* Process Hacker is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with Process Hacker. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include <peview.h>
|
|
#include <objbase.h>
|
|
|
|
PPH_STRING PvFileName = NULL;
|
|
|
|
static BOOLEAN NTAPI PvCommandLineCallback(
|
|
_In_opt_ PPH_COMMAND_LINE_OPTION Option,
|
|
_In_opt_ PPH_STRING Value,
|
|
_In_opt_ PVOID Context
|
|
)
|
|
{
|
|
if (!Option)
|
|
PhSwapReference(&PvFileName, Value);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
static VOID PvpInitializeDpi(
|
|
VOID
|
|
)
|
|
{
|
|
HDC hdc;
|
|
|
|
if (hdc = GetDC(NULL))
|
|
{
|
|
PhGlobalDpi = GetDeviceCaps(hdc, LOGPIXELSY);
|
|
ReleaseDC(NULL, hdc);
|
|
}
|
|
}
|
|
|
|
INT WINAPI wWinMain(
|
|
_In_ HINSTANCE hInstance,
|
|
_In_opt_ HINSTANCE hPrevInstance,
|
|
_In_ PWSTR lpCmdLine,
|
|
_In_ INT nCmdShow
|
|
)
|
|
{
|
|
static PH_COMMAND_LINE_OPTION options[] =
|
|
{
|
|
{ 0, L"h", NoArgumentType }
|
|
};
|
|
PH_STRINGREF commandLine;
|
|
|
|
if (!NT_SUCCESS(PhInitializePhLibEx(0, 0, 0)))
|
|
return 1;
|
|
|
|
PhGuiSupportInitialization();
|
|
PvpInitializeDpi();
|
|
|
|
PhApplicationName = L"PE Viewer";
|
|
|
|
PhUnicodeStringToStringRef(&NtCurrentPeb()->ProcessParameters->CommandLine, &commandLine);
|
|
|
|
PhParseCommandLine(
|
|
&commandLine,
|
|
options,
|
|
sizeof(options) / sizeof(PH_COMMAND_LINE_OPTION),
|
|
PH_COMMAND_LINE_IGNORE_FIRST_PART,
|
|
PvCommandLineCallback,
|
|
NULL
|
|
);
|
|
|
|
if (!PvFileName)
|
|
{
|
|
static PH_FILETYPE_FILTER filters[] =
|
|
{
|
|
{ L"Supported files (*.exe;*.dll;*.ocx;*.sys;*.scr;*.cpl;*.ax;*.acm;*.lib;*.winmd;*.efi)", L"*.exe;*.dll;*.ocx;*.sys;*.scr;*.cpl;*.ax;*.acm;*.lib;*.winmd;*.efi" },
|
|
{ L"All files (*.*)", L"*.*" }
|
|
};
|
|
PVOID fileDialog;
|
|
|
|
CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
|
|
|
|
fileDialog = PhCreateOpenFileDialog();
|
|
PhSetFileDialogFilter(fileDialog, filters, sizeof(filters) / sizeof(PH_FILETYPE_FILTER));
|
|
|
|
if (PhShowFileDialog(NULL, fileDialog))
|
|
{
|
|
PvFileName = PhGetFileDialogFileName(fileDialog);
|
|
}
|
|
|
|
PhFreeFileDialog(fileDialog);
|
|
}
|
|
|
|
if (!PvFileName)
|
|
return 1;
|
|
|
|
if (PhEndsWithString2(PvFileName, L".lnk", TRUE))
|
|
{
|
|
PPH_STRING targetFileName;
|
|
|
|
CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
|
|
targetFileName = PvResolveShortcutTarget(PvFileName);
|
|
|
|
if (targetFileName)
|
|
PhMoveReference(&PvFileName, targetFileName);
|
|
}
|
|
|
|
if (!PhEndsWithString2(PvFileName, L".lib", TRUE))
|
|
PvPeProperties();
|
|
else
|
|
PvLibProperties();
|
|
|
|
return 0;
|
|
}
|