RtlGetNtProductType.c File Reference

#include "precomp.h"
#include <winbase.h>
#include <winreg.h>

Include dependency graph for RtlGetNtProductType.c:

Go to the source code of this file.

Functions
static BOOLEAN	ReturnNtProduct (PDWORD ProductNtType)
static BOOLEAN	ChangeNtProductType (DWORD NtProductType)
	START_TEST (RtlGetNtProductType)

Function Documentation

ChangeNtProductType()

static BOOLEAN ChangeNtProductType ( DWORD  NtProductType )

static

Definition at line 73 of file RtlGetNtProductType.c.

{
    PWSTR ProductTypeString;
    LONG Result;
    HKEY Key;
    DWORD dwSize;
 
    if (NtProductType == NtProductWinNt)
    {
        ProductTypeString = L"WinNT";
    }
    else if (NtProductType == NtProductLanManNt)
    {
        ProductTypeString = L"LanmanNT";
    }
    else if (NtProductType == NtProductServer)
    {
        ProductTypeString = L"ServerNT";
    }
    else
    {
        ok(FALSE, "Passed invalid product type to ChangeNtProduct: %lu", NtProductType);
    }
 
    Result = RegOpenKeyExW(HKEY_LOCAL_MACHINE,
                           L"SYSTEM\\CurrentControlSet\\Control\\ProductOptions",
                           0,
                           KEY_ALL_ACCESS,
                           &Key);
    if (Result != ERROR_SUCCESS)
    {
        ok(FALSE, "RegOpenKeyExW failed with 0x%lx\n", Result);
        return FALSE;
    }
 
    dwSize = ((DWORD)wcslen(ProductTypeString) + 1) * sizeof(WCHAR);
    Result = RegSetValueExW(Key,
                            L"ProductType",
                            0,
                            REG_SZ,
                            (PBYTE)ProductTypeString,
                            dwSize);
    if (Result != ERROR_SUCCESS)
    {
        ok(FALSE, "RegSetValueExW failed with 0x%lx\n", Result);
        RegCloseKey(Key);
        return FALSE;
    }
 
    RegCloseKey(Key);
    return TRUE;
}

Referenced by START_TEST().

ReturnNtProduct()

static BOOLEAN ReturnNtProduct ( PDWORD  ProductNtType )

static

Definition at line 14 of file RtlGetNtProductType.c.

{
    LONG Result;
    HKEY Key;
    WCHAR Data[20];
    DWORD Size = sizeof(Data);
 
    Result = RegOpenKeyExW(HKEY_LOCAL_MACHINE,
                           L"SYSTEM\\CurrentControlSet\\Control\\ProductOptions",
                           0,
                           KEY_QUERY_VALUE,
                           &Key);
    if (Result != ERROR_SUCCESS)
    {
        *ProductNtType = 0;
        return FALSE;
    }
 
    Result = RegQueryValueExW(Key,
                              L"ProductType",
                              NULL,
                              NULL,
                              (PBYTE)&Data,
                              &Size);
    if (Result != ERROR_SUCCESS)
    {
        RegCloseKey(Key);
        *ProductNtType = 0;
        return FALSE;
    }
 
    if (wcscmp(Data, L"WinNT") == 0)
    {
        *ProductNtType = NtProductWinNt;
        RegCloseKey(Key);
        return TRUE;
    }
 
    if (wcscmp(Data, L"LanmanNT") == 0)
    {
        *ProductNtType = NtProductLanManNt;
        RegCloseKey(Key);
        return TRUE;
    }
 
    if (wcscmp(Data, L"ServerNT") == 0)
    {
        *ProductNtType = NtProductServer;
        RegCloseKey(Key);
        return TRUE;
    }
 
    *ProductNtType = 0;
    RegCloseKey(Key);
    return FALSE;
}

Referenced by START_TEST().

START_TEST()

START_TEST

(

RtlGetNtProductType

)

Definition at line 126 of file RtlGetNtProductType.c.

{
    BOOLEAN Ret;
    DWORD ProductNtType;
    NT_PRODUCT_TYPE ProductType = NtProductWinNt, ProductType2;
 
    /* Remove ReportAsWorkstation override during tests */
    DWORD ReportAsWorkstation = 0xbaadf00d;
    HKEY hKey;
    if (RegOpenKeyExW(HKEY_LOCAL_MACHINE, L"SYSTEM\\CurrentControlSet\\Control\\ReactOS\\Settings\\Version",
                      0, KEY_READ | KEY_WRITE, &hKey) == ERROR_SUCCESS)
    {
        DWORD cb = sizeof(DWORD);
        if (RegQueryValueExW(hKey, L"ReportAsWorkstation", NULL, NULL, (PBYTE)&ReportAsWorkstation, &cb))
            ReportAsWorkstation = 0xbaadf00d;
        RegDeleteValueW(hKey, L"ReportAsWorkstation");
        RegCloseKey(hKey);
    }
 
    /*
     * Wrap the call in SEH. This ensures the testcase won't crash but also
     * it proves to us that RtlGetNtProductType() throws an exception if a NULL
     * argument is being passed to caller as output.
     */
    StartSeh()
        RtlGetNtProductType(NULL);
    EndSeh(STATUS_ACCESS_VIOLATION);
 
    /* Query the product type normally from the Registry */
    Ret = ReturnNtProduct(&ProductNtType);
    if (!Ret)
    {
        ok(Ret, "Failed to query the product type value!\n");
    }
 
    /* Now, get the product type from the NTDLL system call */
    Ret = RtlGetNtProductType(&ProductType);
    ok(Ret == TRUE, "Expected a valid product type value (and TRUE as returned success code) but got %u as status.\n", Ret);
    ok(ProductNtType == ProductType, "Expected the product type value to be the same but got %lu (original value pointed by RtlGetNtProductType() is %d).\n", ProductNtType, ProductType);
 
    /* Change product type in the registry */
    ok_char(ChangeNtProductType(NtProductWinNt), TRUE);
    ok_char(RtlGetNtProductType(&ProductType2), TRUE);
    ok_long(ProductType2, ProductType);
 
    ok_char(ChangeNtProductType(NtProductLanManNt), TRUE);
    ok_char(RtlGetNtProductType(&ProductType2), TRUE);
    ok_long(ProductType2, ProductType);
 
    ok_char(ChangeNtProductType(NtProductServer), TRUE);
    ok_char(RtlGetNtProductType(&ProductType2), TRUE);
    ok_long(ProductType2, ProductType);
 
    ok_char(ChangeNtProductType(ProductType), TRUE);
 
 
    /* Restore ReportAsWorkstation */
    if (RegOpenKeyExW(HKEY_LOCAL_MACHINE, L"SYSTEM\\CurrentControlSet\\Control\\ReactOS\\Settings\\Version",
                      0, KEY_WRITE, &hKey) == ERROR_SUCCESS)
    {
        if (ReportAsWorkstation != 0xbaadf00d)
            RegSetValueExW(hKey, L"ReportAsWorkstation", 0, REG_DWORD, (PBYTE)&ReportAsWorkstation, sizeof(DWORD));
        RegCloseKey(hKey);
    }
}