imagefile.c File Reference

#include "gflags.h"

Include dependency graph for imagefile.c:

Go to the source code of this file.

Classes
struct	FlagInfo

Macros
#define	FLG_DISABLE_DBGPRINT   0x8000000
#define	FLG_CRITSEC_EVENT_CREATION   0x10000000
#define	FLG_STOP_ON_UNHANDLED_EXCEPTION   0x20000000
#define	FLG_ENABLE_HANDLE_EXCEPTIONS   0x40000000
#define	FLG_DISABLE_PROTDLLS   0x80000000

Typedefs
typedef struct FlagInfo	FlagInfo

Functions
void	PrintFlags (IN DWORD GlobalFlags, IN OPTIONAL WORD Dest)
static void	ShowStatus (DWORD GlobalFlags, DWORD Ignored)
static DWORD	ValidateFlags (DWORD GlobalFlags, WORD Dest)
static DWORD	FindFlag (PCWSTR Name, WORD Dest)
static VOID	ModifyStatus (VOID)
static VOID	DisplayStatus (VOID)
BOOL	ImageFile_ParseCmdline (INT i, int argc, LPWSTR argv[])
INT	ImageFile_Execute ()

Variables
static PWSTR	ImageFile = NULL
static DWORD	OptionsAdd = 0
static DWORD	OptionsRemove = 0
static BOOL	OptionsSet = FALSE
static const FlagInfo	g_Flags []

Macro Definition Documentation

FLG_CRITSEC_EVENT_CREATION

#define FLG_CRITSEC_EVENT_CREATION   0x10000000

Definition at line 24 of file imagefile.c.

FLG_DISABLE_DBGPRINT

#define FLG_DISABLE_DBGPRINT   0x8000000

Definition at line 23 of file imagefile.c.

FLG_DISABLE_PROTDLLS

#define FLG_DISABLE_PROTDLLS   0x80000000

Definition at line 27 of file imagefile.c.

FLG_ENABLE_HANDLE_EXCEPTIONS

#define FLG_ENABLE_HANDLE_EXCEPTIONS   0x40000000

Definition at line 26 of file imagefile.c.

FLG_STOP_ON_UNHANDLED_EXCEPTION

#define FLG_STOP_ON_UNHANDLED_EXCEPTION   0x20000000

Definition at line 25 of file imagefile.c.

Typedef Documentation

FlagInfo

typedef struct FlagInfo FlagInfo

Function Documentation

DisplayStatus()

static VOID DisplayStatus ( VOID  )

static

Definition at line 190 of file imagefile.c.

{
    HKEY IFEOKey;
    DWORD GlobalFlags;
 
    if (!OpenImageFileExecOptions(KEY_READ, ImageFile, &IFEOKey))
    {
        return;
    }
 
    GlobalFlags = ReadSZFlagsFromRegistry(IFEOKey, L"GlobalFlag");
    ShowStatus(GlobalFlags, 0);
 
    CloseHandle(IFEOKey);
}

Referenced by ImageFile_Execute().

FindFlag()

static DWORD FindFlag ( PCWSTR  Name, WORD  Dest  )

static

Definition at line 117 of file imagefile.c.

{
    DWORD n;
 
    for (n = 0; n < ARRAYSIZE(g_Flags); ++n)
    {
        if (g_Flags[n].wDest & Dest)
        {
            if (!wcsicmp(Name, g_Flags[n].szAbbr))
            {
                return g_Flags[n].dwFlag;
            }
        }
    }
 
    return 0;
}

Referenced by ImageFile_ParseCmdline().

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

ModifyStatus()

static VOID ModifyStatus ( VOID  )

static

Definition at line 135 of file imagefile.c.

{
    LONG Ret;
    DWORD GlobalFlags, Requested, Ignored;
    HKEY IFEOKey;
    WCHAR Buffer[11];
 
    if (!OpenImageFileExecOptions(KEY_WRITE | KEY_READ, ImageFile, &IFEOKey))
    {
        return;
    }
 
    if (OptionsSet)
    {
        Requested = OptionsAdd;
    }
    else
    {
        Requested = ReadSZFlagsFromRegistry(IFEOKey, L"GlobalFlag");
        Requested &= ~OptionsRemove;
        Requested |= OptionsAdd;
    }
 
    GlobalFlags = ValidateFlags(Requested, DEST_IMAGE);
    Ignored = GlobalFlags ^ Requested;
 
    if (GlobalFlags)
    {
        wsprintf(Buffer, L"0x%08x", GlobalFlags);
        Ret = RegSetValueExW(IFEOKey, L"GlobalFlag", 0, REG_SZ, (BYTE*)Buffer, (wcslen(Buffer) + 1) * sizeof(WCHAR));
        if (Ret != ERROR_SUCCESS)
        {
            wprintf(L"MS: RegSetValueEx failed (%d)\n", Ret);
        }
        else
        {
            ShowStatus(GlobalFlags, Ignored);
        }
    }
    else
    {
        Ret = RegDeleteValueW(IFEOKey, L"GlobalFlag");
        if (Ret != ERROR_SUCCESS)
        {
            wprintf(L"MS: RegDeleteValue failed (%d)\n", Ret);
        }
        else
        {
            ShowStatus(GlobalFlags, Ignored);
        }
    }
    CloseHandle(IFEOKey);
}

Referenced by ImageFile_Execute().

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

ShowStatus()

static void ShowStatus ( DWORD  GlobalFlags, DWORD  Ignored  )

static

Definition at line 83 of file imagefile.c.

{
    if (GlobalFlags)
    {
        wprintf(L"Current Registry Settings for %s executable are: %08x\n", ImageFile, GlobalFlags);
        PrintFlags(GlobalFlags, 0);
    }
    else
    {
        wprintf(L"No Registry Settings for %s executable\n", ImageFile);
    }
    if (Ignored)
    {
        wprintf(L"The following settings were ignored: %08x\n", Ignored);
        PrintFlags(Ignored, 0);
    }
}

Referenced by DisplayStatus(), and ModifyStatus().

ValidateFlags()

static DWORD ValidateFlags ( DWORD  GlobalFlags, WORD  Dest  )

static

Definition at line 101 of file imagefile.c.

{
    DWORD n;
    DWORD Valid = 0;
 
    for (n = 0; n < ARRAYSIZE(g_Flags); ++n)
    {
        if (g_Flags[n].wDest & Dest)
        {
            Valid |= g_Flags[n].dwFlag;
        }
    }
 
    return GlobalFlags & Valid;
}

Referenced by ModifyStatus().

Variable Documentation

g_Flags

const FlagInfo g_Flags[]

static

Definition at line 30 of file imagefile.c.

Referenced by FindFlag(), PrintFlags(), and ValidateFlags().

ImageFile

PWSTR ImageFile = NULL

static

Definition at line 10 of file imagefile.c.

Referenced by BlImgLoadBootApplication(), BlImgLoadPEImageEx(), DisplayStatus(), ImageFile_ParseCmdline(), ImgpLoadPEImage(), ModifyStatus(), ShowStatus(), Test_RawSize(), and WriteToFlash().

OptionsAdd

DWORD OptionsAdd = 0

static

Definition at line 11 of file imagefile.c.

Referenced by ImageFile_Execute(), ImageFile_ParseCmdline(), and ModifyStatus().

OptionsRemove

DWORD OptionsRemove = 0

static

Definition at line 12 of file imagefile.c.

Referenced by ImageFile_Execute(), and ImageFile_ParseCmdline().

OptionsSet

BOOL OptionsSet = FALSE

static

Definition at line 13 of file imagefile.c.

Referenced by ImageFile_Execute(), ImageFile_ParseCmdline(), and ModifyStatus().