gflags.h File Reference

#include <stdarg.h>
#include <windef.h>
#include <winbase.h>
#include <winuser.h>
#include <winreg.h>
#include <stdio.h>
#include <stdlib.h>
#include <pstypes.h>

Include dependency graph for gflags.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Macros
#define	WIN32_NO_STATUS
#define	DEST_REGISTRY   1
#define	DEST_KERNEL   2
#define	DEST_IMAGE   4

Functions
BOOL	PageHeap_ParseCmdline (INT i, int argc, LPWSTR argv[])
BOOL	ImageFile_ParseCmdline (INT i, int argc, LPWSTR argv[])
INT	PageHeap_Execute ()
INT	ImageFile_Execute ()
DWORD	ReadSZFlagsFromRegistry (HKEY SubKey, PWSTR Value)
BOOL	OpenImageFileExecOptions (IN REGSAM SamDesired, IN OPTIONAL PCWSTR ImageName, OUT HKEY *Key)
void	PrintFlags (IN DWORD GlobalFlags, IN OPTIONAL WORD Dest)

Variables
const WCHAR	ImageExecOptionsString []

Macro Definition Documentation

DEST_IMAGE

#define DEST_IMAGE   4

Definition at line 35 of file gflags.h.

DEST_KERNEL

#define DEST_KERNEL   2

Definition at line 34 of file gflags.h.

DEST_REGISTRY

#define DEST_REGISTRY   1

Definition at line 33 of file gflags.h.

WIN32_NO_STATUS

#define WIN32_NO_STATUS

Definition at line 8 of file gflags.h.

Function Documentation

ImageFile_Execute()

INT ImageFile_Execute

(

)

Definition at line 251 of file imagefile.c.

{
    if (!OptionsAdd && !OptionsRemove && !OptionsSet)
    {
        DisplayStatus();
    }
    else
    {
        ModifyStatus();
    }

    return 0;
}

Referenced by wmain().

ImageFile_ParseCmdline()

BOOL ImageFile_ParseCmdline	(	INT	i,
		int	argc,
		LPWSTR	argv[]
	)

Definition at line 207 of file imagefile.c.

{
    for (; i < argc; i++)
    {
        if (ImageFile == NULL)
        {
            ImageFile = argv[i];
        }
        else if (argv[i][0] == '+')
        {
            if (OptionsSet)
            {
                wprintf(L"Unexpected argument - '%s'\n", argv[i]);
                return FALSE;
            }
            OptionsAdd |= FindFlag(argv[i] + 1, DEST_IMAGE);
        }
        else if (argv[i][0] == '-')
        {
            if (OptionsSet)
            {
                wprintf(L"Unexpected argument - '%s'\n", argv[i]);
                return FALSE;
            }
            OptionsRemove |= FindFlag(argv[i] + 1, DEST_IMAGE);
        }
        else
        {
            OptionsSet = TRUE;
            OptionsAdd = wcstoul(argv[i], NULL, 16);
            if (OptionsAdd == ~0)
                OptionsAdd = 0;
        }
    }

    if (ImageFile == NULL)
    {
        wprintf(L"No Image specified\n");
        return FALSE;
    }

    return TRUE;
}

Referenced by ParseCmdline().

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().

PageHeap_Execute()

INT PageHeap_Execute

(

)

Definition at line 295 of file pageheap.c.

{
    if (!Set && !Unset)
    {
        DisplayStatus();
    }
    else
    {
        ModifyStatus();
    }

    return 0;
}

Referenced by wmain().

PageHeap_ParseCmdline()

BOOL PageHeap_ParseCmdline	(	INT	i,
		int	argc,
		LPWSTR	argv[]
	)

Definition at line 244 of file pageheap.c.

{
    for (; i < argc; i++)
    {
        if (argv[i][0] == L'/')
        {
            if (wcscmp(argv[i], L"/enable") == 0)
            {
                Set = TRUE;
            }
            else if (wcscmp(argv[i], L"/disable") == 0)
            {
                Unset = TRUE;
            }
            else if (wcscmp(argv[i], L"/full") == 0)
            {
                Full = TRUE;
            }
        }
        else if (Image == NULL)
        {
            Image = argv[i];
        }
        else
        {
            wprintf(L"Invalid option: %s\n", argv[i]);
            return FALSE;
        }
    }

    if (Set && Unset)
    {
        wprintf(L"ENABLE and DISABLED cannot be set together\n");
        return FALSE;
    }

    if (Image == NULL && (Set || Unset || Full))
    {
        wprintf(L"Can't ENABLE or DISABLE with no image\n");
        return FALSE;
    }

    if (!Set && !Unset && Full)
    {
        wprintf(L"Cannot deal with full traces with no other indication\n");
        return FALSE;
    }

    return TRUE;
}

Referenced by ParseCmdline().

PrintFlags()

void PrintFlags	(	IN DWORD	GlobalFlags,
		IN OPTIONAL WORD	Dest
	)

Definition at line 67 of file imagefile.c.

{
    DWORD n;

    for (n = 0; n < ARRAYSIZE(g_Flags); ++n)
    {
        if (!Dest || (g_Flags[n].wDest & Dest))
        {
            if (g_Flags[n].dwFlag & GlobalFlags)
            {
                wprintf(L"    %s - %s\n", g_Flags[n].szAbbr, g_Flags[n].szDesc);
            }
        }
    }
}

Referenced by ShowStatus(), and 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().

Variable Documentation

ImageExecOptionsString

const WCHAR ImageExecOptionsString[]

Definition at line 24 of file ldrinit.c.

Referenced by DisplayImageInfo(), LdrOpenImageFileOptionsKey(), ModifyStatus(), and OpenImageFileExecOptions().