gflags.c File Reference

#include "gflags.h"

Include dependency graph for gflags.c:

Go to the source code of this file.

Functions
BOOL	OpenImageFileExecOptions (IN REGSAM SamDesired, IN OPTIONAL PCWSTR ImageName, OUT HKEY *Key)
DWORD	ReadSZFlagsFromRegistry (HKEY SubKey, PWSTR Value)
static BOOL	ParseCmdline (int argc, LPWSTR argv[])
int	wmain (int argc, LPWSTR argv[])

Variables
static BOOL	UsePageHeap = FALSE
static BOOL	UseImageFile = FALSE
const WCHAR	ImageExecOptionsString [] = L"Software\\Microsoft\\Windows NT\\CurrentVersion\\Image File Execution Options"

Function Documentation

OpenImageFileExecOptions()

BOOL OpenImageFileExecOptions	(	IN REGSAM	SamDesired,
		IN OPTIONAL PCWSTR	ImageName,
		OUT HKEY *	Key
	)

Definition at line 16 of file gflags.c.

{
    LONG Ret;
    HKEY HandleKey, HandleSubKey;
 
    Ret = RegOpenKeyExW(HKEY_LOCAL_MACHINE, ImageExecOptionsString, 0, SamDesired, &HandleKey);
    if (Ret != ERROR_SUCCESS)
    {
        wprintf(L"OpenIFEO: RegOpenKeyEx failed (%d)\n", Ret);
        return FALSE;
    }
 
    if (ImageName == NULL)
    {
        *Key = HandleKey;
        return TRUE;
    }
 
    Ret = RegCreateKeyExW(HandleKey, ImageName, 0, NULL, REG_OPTION_NON_VOLATILE, SamDesired, NULL, &HandleSubKey, NULL);
    CloseHandle(HandleKey);
 
    if (Ret != ERROR_SUCCESS)
    {
        wprintf(L"OpenIFEO: RegCreateKeyEx failed (%d)\n", Ret);
        return FALSE;
    }
    *Key = HandleSubKey;
    return TRUE;
}

Referenced by DisplayStatus(), and ModifyStatus().

ParseCmdline()

static BOOL ParseCmdline ( int  argc, LPWSTR  argv[]  )

static

Definition at line 62 of file gflags.c.

{
    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);
            }
            if (argv[i][1] == L'i' && argv[i][2] == UNICODE_NULL)
            {
                UseImageFile = TRUE;
                return ImageFile_ParseCmdline(i + 1, argc, argv);
            }
        }
        else
        {
            wprintf(L"Invalid option: %s\n", argv[i]);
            return FALSE;
        }
    }
 
    if (!UsePageHeap && !UseImageFile)
    {
        wprintf(L"Only page heap / image file flags are supported\n");
        return FALSE;
    }
 
    return TRUE;
}

Referenced by wmain().

ReadSZFlagsFromRegistry()

DWORD ReadSZFlagsFromRegistry	(	HKEY	SubKey,
		PWSTR	Value
	)

Definition at line 47 of file gflags.c.

{
    WCHAR Buffer[20] = { 0 };
    DWORD Len, Flags, Type;
 
    Len = sizeof(Buffer) - sizeof(WCHAR);
    Flags = 0;
    if (RegQueryValueExW(SubKey, Value, NULL, &Type, (BYTE*)Buffer, &Len) == ERROR_SUCCESS && Type == REG_SZ)
    {
        Flags = wcstoul(Buffer, NULL, 16);
    }
 
    return Flags;
}

Referenced by DisplayImageInfo(), DisplayStatus(), and ModifyStatus().

wmain()

int wmain	(	int	argc,
		LPWSTR	argv[]
	)

Definition at line 104 of file gflags.c.

{
    if (!ParseCmdline(argc, argv))
    {
        wprintf(L"Usage: gflags [/p [image.exe] [/enable|/disable [/full]]]\n"
                L"              [/i <image.exe> [<Flags>]]\n"
                L"    image.exe:  Image you want to deal with\n"
                L"    /enable:    enable page heap for the image\n"
                L"    /disable:   disable page heap for the image\n"
                L"    /full:      activate full debug page heap\n"
                L"    <Flags>:    A 32 bit hex number (0x00000001) that specifies\n"
                L"                one or more global flags to set.\n"
                L"                Without any flags, the current settings are shown.\n"
                L"                Specify FFFFFFFF to delete the GlobalFlags entry.\n"
                L"                Additionally, instead of a single hex number,\n"
                L"                specify a list of abbreviations prefixed with\n"
                L"                a '+' to add, and '-' to remove a bit.\n"
                L"                Valid abbreviations:\n");
        PrintFlags(~0, DEST_IMAGE);
        return 1;
    }
 
    if (UsePageHeap)
    {
        return PageHeap_Execute();
    }
    else if (UseImageFile)
    {
        return ImageFile_Execute();
    }
    return 2;
}

Variable Documentation

ImageExecOptionsString

const WCHAR ImageExecOptionsString[] = L"Software\\Microsoft\\Windows NT\\CurrentVersion\\Image File Execution Options"

Definition at line 14 of file gflags.c.

Referenced by OpenImageFileExecOptions().

UseImageFile

BOOL UseImageFile = FALSE

static

Definition at line 12 of file gflags.c.

Referenced by ParseCmdline(), and wmain().

UsePageHeap

BOOL UsePageHeap = FALSE

static

Definition at line 11 of file gflags.c.

Referenced by ParseCmdline(), and wmain().