main.cpp File Reference

#include "xml2sdb.h"
#include "sdbpapi.h"
#include "sdbstringtable.h"
#include <time.h>
#include <stdio.h>
#include <stdarg.h>

Include dependency graph for main.cpp:

Go to the source code of this file.

Macros
#define	TICKSPERSEC   10000000
#define	TICKSTO1970   0x019db1ded53e8000i64

Functions
LPVOID WINAPI	SdbpAlloc (SIZE_T size)
LPVOID WINAPI	SdbpReAlloc (LPVOID mem, SIZE_T size, SIZE_T oldSize)
void WINAPI	SdbpFree (LPVOID mem)
DWORD	SdbpStrlen (PCWSTR string)
DWORD WINAPI	SdbpStrsize (PCWSTR string)
PDB WINAPI	SdbpCreate (LPCWSTR path, PATH_TYPE type, BOOL write)
void WINAPI	SdbpFlush (PDB pdb)
void WINAPI	SdbCloseDatabase (PDB pdb)
BOOL WINAPI	SdbpCheckTagType (TAG tag, WORD type)
BOOL WINAPI	SdbpReadData (PDB pdb, PVOID dest, DWORD offset, DWORD num)
TAG WINAPI	SdbGetTagFromTagID (PDB pdb, TAGID tagid)
BOOL WINAPI	SdbpCheckTagIDType (PDB pdb, TAGID tagid, WORD type)
BOOL WINAPIV	ShimDbgPrint (SHIM_LOG_LEVEL Level, PCSTR FunctionName, PCSTR Format,...)
VOID NTAPI	RtlSecondsSince1970ToTime (IN ULONG SecondsSince1970, OUT PLARGE_INTEGER Time)
static bool	convert (const std::string &input, const std::string &output, PlatformType platform)
static std::string	get_strarg (int argc, char *argv[], int &i)
static void	update_loglevel (int argc, char *argv[], int &i)
static PlatformType	parse_platform (const std::string &input)
int	main (int argc, char *argv[])

Variables
ULONG	g_ShimDebugLevel = SHIM_WARN

Macro Definition Documentation

TICKSPERSEC

#define TICKSPERSEC   10000000

Definition at line 161 of file main.cpp.

TICKSTO1970

#define TICKSTO1970   0x019db1ded53e8000i64

Definition at line 165 of file main.cpp.

Function Documentation

convert()

static bool convert ( const std::string &  input, const std::string &  output, PlatformType  platform  )

static

Definition at line 177 of file main.cpp.

{
    sdbstring outputW(output.begin(), output.end());
    Database db;
    if (!db.fromXml(input.c_str(), platform))
    {
        printf("Failed to read XML file '%s'\n", input.c_str());
        return false;
    }
    if (!db.toSdb(outputW.c_str()))
    {
        printf("Failed to write SDB file '%s'\n", output.c_str());
        return false;
    }
    return true;
}

get_strarg()

static std::string get_strarg ( int  argc, char *  argv[], int &  i  )

static

Definition at line 194 of file main.cpp.

{
    if (argv[i][2] != 0)
        return std::string(argv[i] + 2);
 
    ++i;
    if (i >= argc || !argv[i])
        return std::string();
    return argv[i];
}

Referenced by main(), and update_loglevel().

main()

int main	(	int argc	,
		char *	argv[]
	)

Definition at line 217 of file main.cpp.

{
    std::string input, output;
    PlatformType platform = PLATFORM_ANY;
    srand(time(0));
 
    for (int i = 1; i < argc; ++i)
    {
        if (argv[i][0] != '/' && argv[i][0] != '-')
            continue;
 
        switch(argv[i][1])
        {
        case 'i':
            input = get_strarg(argc, argv, i);
            break;
        case 'o':
            output = get_strarg(argc, argv, i);
            break;
        case 'l':
            update_loglevel(argc, argv, i);
            break;
        case 'p':
            platform = parse_platform(get_strarg(argc, argv, i));
            break;
        }
    }
    if (input.empty() || output.empty())
    {
        printf("Usage: %s -i <input.xml> -o <output.sdb> [-l <loglevel>] [-v <version>]\n", argv[0]);
        printf("  -i <input.xml>   : Input XML file to convert\n");
        printf("  -o <output.sdb>  : Output SDB file to create\n");
        printf("  -l <loglevel>    : Set log level (1=ERR, 2=WARN, 3=INFO)\n");
        printf("  -p <platform>    : Set the runtime platform (X86, AMD64, ANY)\n");
        return 1;
    }
 
    if (!convert(input, output, platform))
    {
        printf("Failed converting '%s' to '%s'\n", input.c_str(), output.c_str());
        return 1;
    }
    return 0;
}

parse_platform()

static PlatformType parse_platform ( const std::string &  input )

static

Definition at line 212 of file main.cpp.

{
    return (PlatformType)str_to_enum(input, platform_to_flag);
}

Referenced by main().

RtlSecondsSince1970ToTime()

VOID NTAPI RtlSecondsSince1970ToTime	(	IN ULONG	SecondsSince1970,
		OUT PLARGE_INTEGER	Time
	)

Definition at line 167 of file main.cpp.

{
    Time->QuadPart = ((LONGLONG)SecondsSince1970 * TICKSPERSEC) + TICKSTO1970;
}

SdbCloseDatabase()

void WINAPI SdbCloseDatabase

(

PDB

pdb

)

Closes specified database and frees its memory.

Parameters

[in]

pdb

Handle to the shim database.

Definition at line 76 of file main.cpp.

{
    if (!pdb)
        return;
 
    if (pdb->file)
        fclose((FILE*)pdb->file);
    if (pdb->string_buffer)
        SdbCloseDatabase(pdb->string_buffer);
    if (pdb->string_lookup)
        SdbpTableDestroy(&pdb->string_lookup);
    SdbFree(pdb->data);
    SdbFree(pdb);
}

Referenced by SdbCloseDatabase(), SdbCloseDatabaseWrite(), SdbInstall(), SdbRegisterDatabaseEx(), SdbReleaseDatabase(), SdbUninstall(), and CLayerStringList::~CLayerStringList().

SdbGetTagFromTagID()

TAG WINAPI SdbGetTagFromTagID	(	PDB	pdb,
		TAGID	tagid
	)

Searches shim database for the tag associated with specified tagid.

Parameters

[in]	pdb	Handle to the shim database.
[in]	tagid	The TAGID of the tag.

Returns

Success: The tag associated with specified tagid, Failure: TAG_NULL.

Definition at line 114 of file main.cpp.

{
    TAG data;
    if (!SdbpReadData(pdb, &data, tagid, sizeof(data)))
        return TAG_NULL;
    return data;
}

Referenced by SdbpCheckTagIDType(), and SdbpMatchFileAttributes().

SdbpAlloc()

LPVOID WINAPI SdbpAlloc

(

SIZE_T

size

)

Definition at line 19 of file main.cpp.

{
    return ::calloc(1, size);
}

SdbpCheckTagIDType()

BOOL WINAPI SdbpCheckTagIDType	(	PDB	pdb,
		TAGID	tagid,
		WORD	type
	)

Definition at line 122 of file main.cpp.

{
    TAG tag = SdbGetTagFromTagID(pdb, tagid);
    if (tag == TAG_NULL)
        return FALSE;
    return SdbpCheckTagType(tag, type);
}

SdbpCheckTagType()

BOOL WINAPI SdbpCheckTagType	(	TAG	tag,
		WORD	type
	)

Definition at line 91 of file main.cpp.

{
    if ((tag & TAG_TYPE_MASK) != type)
        return FALSE;
    return TRUE;
}

Referenced by SdbpCheckTagIDType().

SdbpCreate()

PDB WINAPI SdbpCreate	(	LPCWSTR	path,
		PATH_TYPE	type,
		BOOL	write
	)

Definition at line 52 of file main.cpp.

{
    PDB pdb;
    FILE* f;
    std::string pathA(path, path + SdbpStrlen(path));
 
    f = fopen(pathA.c_str(), write ? "wb" : "rb");
    if (!f)
        return NULL;
 
    pdb = (PDB)SdbAlloc(sizeof(DB));
    pdb->file = f;
    pdb->for_write = write;
 
    return pdb;
}

SdbpFlush()

void WINAPI SdbpFlush

(

PDB

pdb

)

Definition at line 69 of file main.cpp.

{
    ASSERT(pdb->for_write);
 
    fwrite(pdb->data, pdb->write_iter, 1, (FILE*)pdb->file);
}

SdbpFree()

void WINAPI SdbpFree

(

LPVOID

mem

)

Definition at line 34 of file main.cpp.

{
    return ::free(mem);
}

SdbpReadData()

BOOL WINAPI SdbpReadData	(	PDB	pdb,
		PVOID	dest,
		DWORD	offset,
		DWORD	num
	)

Definition at line 98 of file main.cpp.

{
    DWORD size = offset + num;
 
    /* Either overflow or no data to read */
    if (size <= offset)
        return FALSE;
 
    /* Overflow */
    if (pdb->size < size)
        return FALSE;
 
    memcpy(dest, pdb->data + offset, num);
    return TRUE;
}

Referenced by SdbGetTagFromTagID(), SdbpOpenDatabase(), and SdbQueryDataExTagID().

SdbpReAlloc()

LPVOID WINAPI SdbpReAlloc	(	LPVOID	mem,
		SIZE_T	size,
		SIZE_T	oldSize
	)

Definition at line 24 of file main.cpp.

{
    LPVOID newMem = ::realloc(mem, size);
    if (newMem && size > oldSize)
    {
        memset((BYTE*)newMem + oldSize, 0, size - oldSize);
    }
    return newMem;
}

SdbpStrlen()

DWORD SdbpStrlen

(

PCWSTR

string

)

Definition at line 39 of file main.cpp.

{
    size_t len = 0;
    while (string[len])
        len++;
    return len;
}

Referenced by SdbpCreate(), and SdbpStrsize().

SdbpStrsize()

DWORD WINAPI SdbpStrsize

(

PCWSTR

string

)

Definition at line 47 of file main.cpp.

{
    return (SdbpStrlen(string) + 1) * sizeof(WCHAR);
}

Referenced by SdbpOpenKey(), SdbSetPermLayerKeys(), SdbWriteStringTag(), and SeiInitPaths().

ShimDbgPrint()

BOOL WINAPIV ShimDbgPrint	(	SHIM_LOG_LEVEL	Level,
		PCSTR	FunctionName,
		PCSTR	Format,
			...
	)

Outputs diagnostic info.

Parameters

[in]	Level	The level to log this message with, choose any of [SHIM_ERR, SHIM_WARN, SHIM_INFO].
[in]	FunctionName	The function this log should be attributed to.
[in]	Format	The format string.
	...	Variable arguments providing additional information.

Returns

Success: TRUE Failure: FALSE.

Definition at line 130 of file main.cpp.

{
    va_list ArgList;
    const char* LevelStr;
 
    if ((ULONG)Level > g_ShimDebugLevel)
        return FALSE;
 
    switch (Level)
    {
    case SHIM_ERR:
        LevelStr = "Err ";
        break;
    case SHIM_WARN:
        LevelStr = "Warn";
        break;
    case SHIM_INFO:
        LevelStr = "Info";
        break;
    default:
        LevelStr = "User";
        break;
    }
    printf("[%s][%-20s] ", LevelStr, FunctionName);
    va_start(ArgList, Format);
    vprintf(Format, ArgList);
    va_end(ArgList);
    return TRUE;
}

update_loglevel()

static void update_loglevel ( int  argc, char *  argv[], int &  i  )

static

Definition at line 205 of file main.cpp.

{
    std::string value = get_strarg(argc, argv, i);
    g_ShimDebugLevel = strtoul(value.c_str(), NULL, 10);
}

Referenced by main().

Variable Documentation

g_ShimDebugLevel

ULONG g_ShimDebugLevel = SHIM_WARN

Definition at line 17 of file main.cpp.

Referenced by ShimDbgPrint(), and update_loglevel().