mirror of
https://github.com/reactos/reactos.git
synced 2024-09-27 21:16:34 +00:00
83 lines
1.8 KiB
C
83 lines
1.8 KiB
C
/*
|
|
* PROJECT: Global Flags utility
|
|
* LICENSE: GPL-2.0 (https://spdx.org/licenses/GPL-2.0)
|
|
* PURPOSE: Global Flags utility entrypoint
|
|
* COPYRIGHT: Copyright 2017 Pierre Schweitzer (pierre@reactos.org)
|
|
* Copyright 2018 Mark Jansen (mark.jansen@reactos.org)
|
|
*/
|
|
|
|
#include "gflags.h"
|
|
|
|
static BOOL UsePageHeap = FALSE;
|
|
|
|
|
|
DWORD ReagFlagsFromRegistry(HKEY SubKey, PVOID Buffer, PWSTR Value, DWORD MaxLen)
|
|
{
|
|
DWORD Len, Flags, Type;
|
|
|
|
Len = MaxLen;
|
|
Flags = 0;
|
|
if (RegQueryValueEx(SubKey, Value, NULL, &Type, Buffer, &Len) == ERROR_SUCCESS && Type == REG_SZ)
|
|
{
|
|
Flags = wcstoul(Buffer, NULL, 16);
|
|
}
|
|
|
|
return Flags;
|
|
}
|
|
|
|
static BOOL ParseCmdline(int argc, LPWSTR argv[])
|
|
{
|
|
INT i;
|
|
|
|
if (argc < 2)
|
|
{
|
|
wprintf(L"Not enough args!\n");
|
|
return FALSE;
|
|
}
|
|
|
|
for (i = 1; i < argc; i++)
|
|
{
|
|
if (argv[i][0] == L'/')
|
|
{
|
|
if (argv[i][1] == L'p' && argv[i][2] == UNICODE_NULL)
|
|
{
|
|
UsePageHeap = TRUE;
|
|
return PageHeap_ParseCmdline(i + 1, argc, argv);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
wprintf(L"Invalid option: %s\n", argv[i]);
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
if (!UsePageHeap)
|
|
{
|
|
wprintf(L"Only page heap flags are supported\n");
|
|
return FALSE;
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
|
|
int wmain(int argc, LPWSTR argv[])
|
|
{
|
|
if (!ParseCmdline(argc, argv))
|
|
{
|
|
wprintf(L"Usage: gflags /p [image.exe] [/enable|/disable [/full]]\n"
|
|
L"\timage.exe:\tImage you want to deal with\n"
|
|
L"\t/enable:\tenable page heap for the image\n"
|
|
L"\t/disable:\tdisable page heap for the image\n"
|
|
L"\t/full:\t\tactivate full debug page heap\n");
|
|
return 1;
|
|
}
|
|
|
|
if (UsePageHeap)
|
|
{
|
|
return PageHeap_Execute();
|
|
}
|
|
return 2;
|
|
}
|