precomp.h File Reference

#include <iostream>
#include <memory>
#include <sstream>
#include <string>
#include <vector>
#include <iomanip>
#include <windef.h>
#include <winbase.h>
#include <winnls.h>
#include <reason.h>
#include <shlobj.h>
#include <wininet.h>
#include <winreg.h>
#include <ndk/rtlfuncs.h>
#include <reactos/buildno.h>
#include "auto_array_ptr.h"
#include "CConfiguration.h"
#include "CFatalException.h"
#include "CInvalidParameterException.h"
#include "CPipe.h"
#include "CProcess.h"
#include "CPipedProcess.h"
#include "CSimpleException.h"
#include "CTestException.h"
#include "CTestInfo.h"
#include "CTest.h"
#include "CTestList.h"
#include "CJournaledTestList.h"
#include "CVirtualTestList.h"
#include "CWebService.h"
#include "CWineTest.h"
#include <rosautotestmsg.h>

Include dependency graph for precomp.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	_INC_WINDOWS
#define	COM_NO_WINDOWS_H
#define	EXCEPTION(Message)   throw CSimpleException(Message)
#define	FATAL(Message)   throw CFatalException(__FILE__, __LINE__, Message)
#define	SSEXCEPTION   throw CSimpleException(ss.str())
#define	TESTEXCEPTION(Message)   throw CTestException(Message)

Functions
VOID	FreeLogs (VOID)
VOID	InitLogs (VOID)
bool	ShutdownSystem ()
wstring	AsciiToUnicode (const char *AsciiString)
wstring	AsciiToUnicode (const string &AsciiString)
string	EscapeString (const char *Input)
string	EscapeString (const string &Input)
string	GetINIValue (PCWCH AppName, PCWCH KeyName, PCWCH FileName)
bool	IsNumber (const char *Input)
string	StringOut (const string &String, bool forcePrint=true)
string	UnicodeToAscii (PCWSTR UnicodeString)
string	UnicodeToAscii (const wstring &UnicodeString)

Variables
CConfiguration	Configuration
HANDLE	hLog

Macro Definition Documentation

_INC_WINDOWS

#define _INC_WINDOWS

Definition at line 22 of file precomp.h.

COM_NO_WINDOWS_H

#define COM_NO_WINDOWS_H

Definition at line 23 of file precomp.h.

EXCEPTION

#define EXCEPTION

(

Message

)

throw CSimpleException(Message)

Definition at line 56 of file precomp.h.

FATAL

#define FATAL

(

Message

)

throw CFatalException(__FILE__, __LINE__, Message)

Definition at line 57 of file precomp.h.

SSEXCEPTION

#define SSEXCEPTION   throw CSimpleException(ss.str())

Definition at line 58 of file precomp.h.

TESTEXCEPTION

#define TESTEXCEPTION

(

Message

)

throw CTestException(Message)

Definition at line 59 of file precomp.h.

WIN32_NO_STATUS

#define WIN32_NO_STATUS

Definition at line 21 of file precomp.h.

Function Documentation

AsciiToUnicode() [1/2]

wstring AsciiToUnicode

(

const char *

AsciiString

)

Converts an ASCII string to a Unicode one.

Parameters

AsciiString

Constant pointer to a char array containing the ASCII string

Returns

The Unicode string as std::wstring

Definition at line 220 of file tools.cpp.

{
    DWORD Length;
    PWSTR UnicodeString;
    wstring ReturnString;
 
    Length = MultiByteToWideChar(CP_ACP, 0, AsciiString, -1, NULL, 0);
 
    UnicodeString = new WCHAR[Length];
    MultiByteToWideChar(CP_ACP, 0, AsciiString, -1, UnicodeString, Length);
    ReturnString = UnicodeString;
    delete UnicodeString;
 
    return ReturnString;
}

Referenced by AddPrinterA(), AddPrintProcessorA(), AsciiToUnicode(), DeleteFormA(), DeletePrinterDriverExA(), DeletePrintProcessorA(), DeletePrintProvidorA(), EnumPrinterDriversA(), CWineTest::GetNextTestInfo(), ResetPrinterA(), SetJobA(), SetPrinterA(), and StartDocDlgA().

AsciiToUnicode() [2/2]

wstring AsciiToUnicode

(

const string &

AsciiString

)

Converts an ASCII string to a Unicode one.

Parameters

AsciiString

Pointer to a std::string containing the ASCII string

Returns

The Unicode string as std::wstring

Definition at line 246 of file tools.cpp.

{
    return AsciiToUnicode(AsciiString.c_str());
}

EscapeString() [1/2]

string EscapeString

(

const char *

Input

)

Escapes a string according to RFC 1738. Required for passing parameters to the web service.

Parameters

Input

Constant pointer to a char array, which contains the input buffer to escape.

Returns

The escaped string as std::string.

Definition at line 24 of file tools.cpp.

{
    string ReturnedString;
 
    do
    {
        if(strchr("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789-_.~", *Input))
        {
            /* It's a character we don't need to escape, just add it to the output string */
            ReturnedString += *Input;
        }
        else
        {
            /* We need to escape this character */
            ReturnedString += '%';
            ReturnedString += HexCharacters[((UCHAR)*Input >> 4) % 16];
            ReturnedString += HexCharacters[(UCHAR)*Input % 16];
        }
    }
    while(*++Input);
 
    return ReturnedString;
}

Referenced by EscapeString(), CConfiguration::GetConfigurationFromFile(), and CWebService::Submit().

EscapeString() [2/2]

string EscapeString

(

const string &

Input

)

Escapes a string according to RFC 1738. Required for passing parameters to the web service.

Parameters

Input

Pointer to a std::string, which contains the input buffer to escape.

Returns

The escaped string as std::string.

Definition at line 59 of file tools.cpp.

{
    return EscapeString(Input.c_str());
}

FreeLogs()

VOID FreeLogs

(

VOID

)

Definition at line 85 of file misc.cpp.

{
    if (hLog) DeregisterEventSource(hLog);
}

GetINIValue()

string GetINIValue	(	PCWCH	AppName,
		PCWCH	KeyName,
		PCWCH	FileName
	)

Gets a value from a specified INI file and returns it converted to ASCII.

Parameters

AppName	Constant pointer to a WCHAR array with the INI section to look in (lpAppName parameter passed to GetPrivateProfileStringW)
KeyName	Constant pointer to a WCHAR array containing the key to look for in the specified section (lpKeyName parameter passed to GetPrivateProfileStringW)
FileName	Constant pointer to a WCHAR array containing the path to the INI file

Returns

Returns the data of the value as std::string or an empty string if no data could be retrieved.

Definition at line 187 of file tools.cpp.

{
    DWORD Length;
    PCHAR AsciiBuffer;
    string ReturnedString;
    WCHAR Buffer[2048];
 
    /* Load the value into a temporary Unicode buffer */
    Length = GetPrivateProfileStringW(AppName, KeyName, NULL, Buffer, sizeof(Buffer) / sizeof(WCHAR), FileName);
 
    if(Length)
    {
        /* Convert the string to ASCII charset */
        AsciiBuffer = new char[Length + 1];
        WideCharToMultiByte(CP_ACP, 0, Buffer, Length + 1, AsciiBuffer, Length + 1, NULL, NULL);
 
        ReturnedString = AsciiBuffer;
        delete[] AsciiBuffer;
    }
 
    return ReturnedString;
}

Referenced by CConfiguration::GetConfigurationFromFile().

InitLogs()

VOID InitLogs

(

VOID

)

Definition at line 12 of file misc.cpp.

{
    WCHAR szBuf[MAX_PATH] = L"SYSTEM\\CurrentControlSet\\Services\\EventLog\\Application\\RosAutotest";
    WCHAR szPath[MAX_PATH];
    DWORD dwCategoryNum = 1;
    DWORD dwDisp, dwData;
    HKEY hKey;
 
    if (RegCreateKeyExW(HKEY_LOCAL_MACHINE,
                        szBuf, 0, NULL,
                        REG_OPTION_NON_VOLATILE,
                        KEY_WRITE, NULL, &hKey, &dwDisp) != ERROR_SUCCESS)
    {
        return;
    }
 
    if (!GetModuleFileName(NULL, szPath, sizeof(szPath) / sizeof(szPath[0])))
        return;
 
    if (RegSetValueExW(hKey,
                       L"EventMessageFile",
                       0,
                       REG_EXPAND_SZ,
                       (LPBYTE)szPath,
                       (DWORD)(wcslen(szPath) + 1) * sizeof(WCHAR)) != ERROR_SUCCESS)
    {
        RegCloseKey(hKey);
        return;
    }
 
    dwData = EVENTLOG_ERROR_TYPE | EVENTLOG_WARNING_TYPE |
             EVENTLOG_INFORMATION_TYPE;
 
    if (RegSetValueExW(hKey,
                       L"TypesSupported",
                       0,
                       REG_DWORD,
                       (LPBYTE)&dwData,
                       sizeof(DWORD)) != ERROR_SUCCESS)
    {
        RegCloseKey(hKey);
        return;
    }
 
    if (RegSetValueExW(hKey,
                       L"CategoryMessageFile",
                       0,
                       REG_EXPAND_SZ,
                       (LPBYTE)szPath,
                       (DWORD)(wcslen(szPath) + 1) * sizeof(WCHAR)) != ERROR_SUCCESS)
    {
        RegCloseKey(hKey);
        return;
    }
 
    if (RegSetValueExW(hKey,
                       L"CategoryCount",
                       0,
                       REG_DWORD,
                       (LPBYTE)&dwCategoryNum,
                       sizeof(DWORD)) != ERROR_SUCCESS)
    {
        RegCloseKey(hKey);
        return;
    }
 
    RegCloseKey(hKey);
 
    hLog = RegisterEventSourceW(NULL, L"RosAutotest");
}

IsNumber()

bool IsNumber

(

const char *

Input

)

Determines whether a string contains entirely numeric values.

Parameters

Input

Constant pointer to a char array containing the input to check.

Returns

true if the string is entirely numeric, false otherwise.

Definition at line 74 of file tools.cpp.

{
    do
    {
        if(!isdigit(*Input))
            return false;
 
        ++Input;
    }
    while(*Input);
 
    return true;
}

Referenced by CWebService::GetSuiteID(), and CWebService::GetTestID().

ShutdownSystem()

bool ShutdownSystem

(

)

Shuts down the system.

Returns

true if everything went well, false if there was a problem while trying to shut down the system.

Definition at line 16 of file shutdown.cpp.

{
    HANDLE hToken;
    TOKEN_PRIVILEGES Privileges;
 
    if (!OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES, &hToken))
    {
        StringOut("OpenProcessToken failed\n");
        return false;
    }
 
    /* Get the LUID for the Shutdown privilege */
    if (!LookupPrivilegeValueW(NULL, SE_SHUTDOWN_NAME, &Privileges.Privileges[0].Luid))
    {
        StringOut("LookupPrivilegeValue failed\n");
        return false;
    }
 
    /* Assign the Shutdown privilege to our process */
    Privileges.PrivilegeCount = 1;
    Privileges.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
 
    if (!AdjustTokenPrivileges(hToken, FALSE, &Privileges, 0, NULL, NULL))
    {
        StringOut("AdjustTokenPrivileges failed\n");
        return false;
    }
 
    /* Finally shut down the system */
    if(!ExitWindowsEx(EWX_POWEROFF, SHTDN_REASON_MAJOR_OTHER | SHTDN_REASON_MINOR_OTHER | SHTDN_REASON_FLAG_PLANNED))
    {
        StringOut("ExitWindowsEx failed\n");
        return false;
    }
 
    return true;
}

Referenced by wmain().

StringOut()

string StringOut	(	const string &	String,
		bool	forcePrint
	)

Outputs a string through the standard output and the debug output. The string may have LF or CRLF line endings.

Parameters

String

The std::string to output

Definition at line 96 of file tools.cpp.

{
    char DbgString[DBGPRINT_BUFSIZE + 1];
    size_t i, start = 0, last_newline = 0, size = 0, curr_pos = 0;
    string NewString;
 
    /* Unify the line endings (the piped output of the tests may use CRLF) */
    for(i = 0; i < String.size(); i++)
    {
        /* If this is a CRLF line-ending, only copy a \n to the new string and skip the next character */
        if(String[i] == '\r' && String[i + 1] == '\n')
        {
            NewString += '\n';
            ++i;
        }
        else
        {
            /* Otherwise copy the string */
            NewString += String[i];
        }
 
        curr_pos = NewString.size();
 
        /* Try to print whole lines but obey the 512 bytes chunk size limit*/
        if(NewString[curr_pos - 1] == '\n' || (curr_pos - start) == DBGPRINT_BUFSIZE)
        {
            if((curr_pos - start) >= DBGPRINT_BUFSIZE)
            {
                /* No newlines so far, or the string just fits */
                if(last_newline <= start || ((curr_pos - start == DBGPRINT_BUFSIZE) && NewString[curr_pos - 1] == '\n'))
                {
                    size = curr_pos - start;
                    memcpy(DbgString, NewString.c_str() + start, size);
                    start = curr_pos;
                }
                else
                {
                    size = last_newline - start;
                    memcpy(DbgString, NewString.c_str() + start, size);
                    start = last_newline;
                }
 
                DbgString[size] = 0;
                DbgPrint("%s", DbgString);
            }
 
            last_newline = curr_pos;
        }
    }
 
    size = curr_pos - start;
 
    /* Only print if forced to or if the rest is a whole line */
    if(forcePrint == true || NewString[curr_pos - 1] == '\n')
    {
        /* Output the whole string */
        if(Configuration.DoPrint())
            cout << NewString << flush;
 
        memcpy(DbgString, NewString.c_str() + start, size);
        DbgString[size] = 0;
        DbgPrint("%s", DbgString);
 
        NewString.clear();
        return NewString;
    }
 
    /* Output full lines only */
    if(Configuration.DoPrint())
        cout << NewString.substr(0, start) << flush;
 
    /* Return the remaining chunk */
    return NewString.substr(start, size);
}

Referenced by CWineTest::GetNextTestInfo(), LdrpAllocateUnicodeString(), CJournaledTestList::LoadJournalFile(), CWineTest::Run(), CWineTest::RunTest(), ShutdownSystem(), wmain(), and CJournaledTestList::WriteInitialJournalFile().

UnicodeToAscii() [1/2]

string UnicodeToAscii

(

const wstring &

UnicodeString

)

Converts a Unicode string to an ASCII one.

Parameters

UnicodeString

Pointer to a std::wstring containing the Unicode string

Returns

The ASCII string as std::string

Definition at line 287 of file tools.cpp.

{
    return UnicodeToAscii(UnicodeString.c_str());
}

UnicodeToAscii() [2/2]

string UnicodeToAscii

(

PCWSTR

UnicodeString

)

Converts a Unicode string to an ASCII one.

Parameters

UnicodeString

Constant pointer to a WCHAR array containing the Unicode string

Returns

The ASCII string as std::string

Definition at line 261 of file tools.cpp.

{
    DWORD Length;
    PCHAR AsciiString;
    string ReturnString;
 
    Length = WideCharToMultiByte(CP_ACP, 0, UnicodeString, -1, NULL, 0, NULL, NULL);
 
    AsciiString = new char[Length];
    WideCharToMultiByte(CP_ACP, 0, UnicodeString, -1, AsciiString, Length, NULL, NULL);
    ReturnString = AsciiString;
    delete AsciiString;
 
    return ReturnString;
}

Referenced by CWineTest::DoListCommand(), CWineTest::GetNextTestInfo(), CConfiguration::ParseParameters(), and UnicodeToAscii().

Variable Documentation

Configuration

CConfiguration Configuration

extern

Definition at line 11 of file main.cpp.

hLog

HANDLE hLog

extern

Definition at line 9 of file misc.cpp.

Referenced by FreeLogs(), InitLogs(), and WriteLogMessage().