ntverrsrc.c File Reference

#include "precomp.h"
#include <ndk/ldrtypes.h>
#include <ndk/ldrfuncs.h>
#include "ntverrsrc.h"
#include <debug.h>

Include dependency graph for ntverrsrc.c:

Go to the source code of this file.

Classes
struct	_VS_VERSION_INFO_STRUCT32

Macros
#define	NDEBUG
#define	VS_VERSION_INFO   1
#define	VS_FILE_INFO   RT_VERSION
#define	DWORD_ALIGN(base, ptr)   ( (ULONG_PTR)(base) + ((((ULONG_PTR)(ptr) - (ULONG_PTR)(base)) + 3) & ~3) )
#define	VersionInfo32_Value(ver)   DWORD_ALIGN( (ver), (ver)->szKey + wcslen((ver)->szKey) + 1 )
#define	VersionInfo32_Children(ver)
#define	VersionInfo32_Next(ver)   (PVS_VERSION_INFO_STRUCT32)( (ULONG_PTR)ver + (((ver)->wLength + 3) & ~3) )

Typedefs
typedef struct _VS_VERSION_INFO_STRUCT32	VS_VERSION_INFO_STRUCT32
typedef struct _VS_VERSION_INFO_STRUCT32 *	PVS_VERSION_INFO_STRUCT32
typedef const VS_VERSION_INFO_STRUCT32 *	PCVS_VERSION_INFO_STRUCT32

Functions
NTSTATUS	NtGetVersionResource (IN PVOID BaseAddress, OUT PVOID *Resource, OUT PULONG ResourceSize OPTIONAL)
static PCVS_VERSION_INFO_STRUCT32	VersionInfo32_FindChild (IN PCVS_VERSION_INFO_STRUCT32 info, IN PCWSTR szKey, IN UINT cchKey)
static NTSTATUS	VersionInfo32_QueryValue (IN PCVS_VERSION_INFO_STRUCT32 info, IN PCWSTR lpSubBlock, OUT PVOID *lplpBuffer, OUT PUINT puLen OPTIONAL, OUT BOOL *pbText OPTIONAL)
NTSTATUS	NtVerQueryValue (IN const VOID *pBlock, IN PCWSTR lpSubBlock, OUT PVOID *lplpBuffer, OUT PUINT puLen)

Macro Definition Documentation

DWORD_ALIGN

#define DWORD_ALIGN	(		base,
			ptr
	)		( (ULONG_PTR)(base) + ((((ULONG_PTR)(ptr) - (ULONG_PTR)(base)) + 3) & ~3) )

Definition at line 89 of file ntverrsrc.c.

NDEBUG

#define NDEBUG

Definition at line 19 of file ntverrsrc.c.

VersionInfo32_Children

#define VersionInfo32_Children

(

ver

)

Value:

(PCVS_VERSION_INFO_STRUCT32)( VersionInfo32_Value( ver ) + \
                           ( ( (ver)->wValueLength * \
                               ((ver)->wType? 2 : 1) + 3 ) & ~3 ) )

Definition at line 95 of file ntverrsrc.c.

VersionInfo32_Next

#define VersionInfo32_Next

(

ver

)

(PVS_VERSION_INFO_STRUCT32)( (ULONG_PTR)ver + (((ver)->wLength + 3) & ~3) )

Definition at line 100 of file ntverrsrc.c.

VersionInfo32_Value

#define VersionInfo32_Value

(

ver

)

DWORD_ALIGN( (ver), (ver)->szKey + wcslen((ver)->szKey) + 1 )

Definition at line 92 of file ntverrsrc.c.

VS_FILE_INFO

#define VS_FILE_INFO   RT_VERSION

VS_VERSION_INFO

#define VS_VERSION_INFO   1

Typedef Documentation

PCVS_VERSION_INFO_STRUCT32

typedef const VS_VERSION_INFO_STRUCT32* PCVS_VERSION_INFO_STRUCT32

Definition at line 87 of file ntverrsrc.c.

PVS_VERSION_INFO_STRUCT32

typedef struct _VS_VERSION_INFO_STRUCT32 * PVS_VERSION_INFO_STRUCT32

VS_VERSION_INFO_STRUCT32

typedef struct _VS_VERSION_INFO_STRUCT32 VS_VERSION_INFO_STRUCT32

Function Documentation

NtGetVersionResource()

NTSTATUS NtGetVersionResource	(	IN PVOID	BaseAddress,
		OUT PVOID *	Resource,
		OUT PULONG ResourceSize	OPTIONAL
	)

Definition at line 26 of file ntverrsrc.c.

{
// #define RT_VERSION MAKEINTRESOURCE(16)  // See winuser.h
#define VS_VERSION_INFO         1       // See psdk/verrsrc.h
#define VS_FILE_INFO            RT_VERSION

    NTSTATUS Status;
    LDR_RESOURCE_INFO ResourceInfo;
    PIMAGE_RESOURCE_DATA_ENTRY ResourceDataEntry;
    PVOID Data = NULL;
    ULONG Size = 0;

    /* Try to find the resource */
    ResourceInfo.Type = 16; // RT_VERSION;
    ResourceInfo.Name = VS_VERSION_INFO; // MAKEINTRESOURCEW(VS_VERSION_INFO);
    ResourceInfo.Language = 0; // Don't care about the language

    Status = LdrFindResource_U(BaseAddress,
                               &ResourceInfo,
                               RESOURCE_DATA_LEVEL,
                               &ResourceDataEntry);
    if (!NT_SUCCESS(Status))
    {
        DPRINT1("NtGetVersionResource: Version resource not found, Status 0x%08lx\n", Status);
        return Status;
    }

    /* Access the resource */
    Status = LdrAccessResource(BaseAddress,
                               ResourceDataEntry,
                               &Data,
                               &Size);
    if (!NT_SUCCESS(Status))
    {
        DPRINT1("NtGetVersionResource: Cannot access Version resource, Status 0x%08lx\n", Status);
        return Status;
    }

    *Resource = Data;
    if (ResourceSize) *ResourceSize = Size;

    return STATUS_SUCCESS;
}

Referenced by CheckForValidPEAndVendor().

NtVerQueryValue()

NTSTATUS NtVerQueryValue	(	IN const VOID *	pBlock,
		IN PCWSTR	lpSubBlock,
		OUT PVOID *	lplpBuffer,
		OUT PUINT	puLen
	)

Definition at line 174 of file ntverrsrc.c.

{
    PCVS_VERSION_INFO_STRUCT32 info = pBlock;

    DPRINT("%s (%p, %S, %p, %p)\n", __FUNCTION__, pBlock, lpSubBlock, lplpBuffer, puLen);

    if (!pBlock)
        return FALSE;

    if (!lpSubBlock || !*lpSubBlock)
        lpSubBlock = L"\\";

    return VersionInfo32_QueryValue(info, lpSubBlock, lplpBuffer, puLen, NULL);
}

Referenced by CheckForValidPEAndVendor().

VersionInfo32_FindChild()

static PCVS_VERSION_INFO_STRUCT32 VersionInfo32_FindChild ( IN PCVS_VERSION_INFO_STRUCT32  info, IN PCWSTR  szKey, IN UINT  cchKey  )

static

Definition at line 104 of file ntverrsrc.c.

{
    PCVS_VERSION_INFO_STRUCT32 child = VersionInfo32_Children(info);

    while ((ULONG_PTR)child < (ULONG_PTR)info + info->wLength)
    {
        if (!_wcsnicmp(child->szKey, szKey, cchKey) && !child->szKey[cchKey])
            return child;

        if (child->wLength == 0) return NULL;
        child = VersionInfo32_Next(child);
    }

    return NULL;
}

Referenced by VersionInfo32_QueryValue().

VersionInfo32_QueryValue()

static NTSTATUS VersionInfo32_QueryValue ( IN PCVS_VERSION_INFO_STRUCT32  info, IN PCWSTR  lpSubBlock, OUT PVOID *  lplpBuffer, OUT PUINT puLen  OPTIONAL, OUT BOOL *pbText  OPTIONAL  )

static

Definition at line 124 of file ntverrsrc.c.

{
    PCWSTR lpNextSlash;

    DPRINT("lpSubBlock : (%S)\n", lpSubBlock);

    while (*lpSubBlock)
    {
        /* Find next path component */
        for (lpNextSlash = lpSubBlock; *lpNextSlash; lpNextSlash++)
        {
            if (*lpNextSlash == '\\')
                break;
        }

        /* Skip empty components */
        if (lpNextSlash == lpSubBlock)
        {
            lpSubBlock++;
            continue;
        }

        /* We have a non-empty component: search info for key */
        info = VersionInfo32_FindChild(info, lpSubBlock, lpNextSlash - lpSubBlock);
        if (!info)
        {
            if (puLen) *puLen = 0;
            return STATUS_RESOURCE_TYPE_NOT_FOUND;
        }

        /* Skip path component */
        lpSubBlock = lpNextSlash;
    }

    /* Return value */
    *lplpBuffer = (PVOID)VersionInfo32_Value(info);
    if (puLen)
        *puLen = info->wValueLength;
    if (pbText)
        *pbText = info->wType;

    return STATUS_SUCCESS;
}

Referenced by NtVerQueryValue().