reactos/rostests/rosautotest/CConfiguration.cpp

208 lines
6 KiB
C++
Raw Normal View History

/*
* PROJECT: ReactOS Automatic Testing Utility
* LICENSE: GNU GPLv2 or any later version as published by the Free Software Foundation
* PURPOSE: Class for managing all the configuration parameters
* COPYRIGHT: Copyright 2009-2011 Colin Finck <colin@reactos.org>
*/
#include "precomp.h"
#define CONFIGURATION_FILENAMEA "rosautotest.ini"
#define CONFIGURATION_FILENAMEW L"rosautotest.ini"
typedef void (WINAPI *GETSYSINFO)(LPSYSTEM_INFO);
/**
* Constructs an empty CConfiguration object
*/
CConfiguration::CConfiguration()
{
WCHAR WindowsDirectory[MAX_PATH];
/* Zero-initialize variables */
m_CrashRecovery = false;
m_Shutdown = false;
m_Submit = false;
/* Check if we are running under ReactOS from the SystemRoot directory */
if(!GetWindowsDirectoryW(WindowsDirectory, MAX_PATH))
FATAL("GetWindowsDirectoryW failed");
m_IsReactOS = !_wcsnicmp(&WindowsDirectory[3], L"reactos", 7);
}
/**
* Parses the passed parameters and sets the appropriate configuration settings.
*
* @param argc
* The number of parameters (argc parameter of the wmain function)
*
* @param argv
* Pointer to a wchar_t array containing the parameters (argv parameter of the wmain function)
*/
void
CConfiguration::ParseParameters(int argc, wchar_t* argv[])
{
/* Parse the command line arguments */
for(int i = 1; i < argc; i++)
{
if(argv[i][0] == '-' || argv[i][0] == '/')
{
switch(argv[i][1])
{
case 'c':
++i;
m_Comment = UnicodeToAscii(argv[i]);
break;
case 'r':
m_CrashRecovery = true;
break;
case 's':
m_Shutdown = true;
break;
case 'w':
m_Submit = true;
break;
default:
throw CInvalidParameterException();
}
}
else
{
/* Which parameter is this? */
if(m_Module.empty())
{
/* Copy the parameter */
m_Module = argv[i];
}
else if(m_Test.empty())
{
/* Copy the parameter converted to ASCII */
m_Test = UnicodeToAscii(argv[i]);
}
else
{
throw CInvalidParameterException();
}
}
}
/* The /r and /w options shouldn't be used in conjunction */
if(m_CrashRecovery && m_Submit)
throw CInvalidParameterException();
}
/**
* Gets information about the running system and sets the appropriate configuration settings.
*/
void
CConfiguration::GetSystemInformation()
{
char ProductType;
GETSYSINFO GetSysInfo;
HMODULE hKernel32;
OSVERSIONINFOEXW os;
stringstream ss;
SYSTEM_INFO si;
/* Get the build from the define */
ss << "&revision=";
ss << KERNEL_VERSION_BUILD_HEX;
ss << "&platform=";
if(m_IsReactOS)
{
ss << "reactos";
}
else
{
/* No, then use the info from GetVersionExW */
os.dwOSVersionInfoSize = sizeof(os);
if(!GetVersionExW((LPOSVERSIONINFOW)&os))
FATAL("GetVersionExW failed\n");
if(os.dwMajorVersion < 5)
EXCEPTION("Application requires at least Windows 2000!\n");
if(os.wProductType == VER_NT_WORKSTATION)
ProductType = 'w';
else
ProductType = 's';
/* Print all necessary identification information into the Platform string */
ss << os.dwMajorVersion << '.'
<< os.dwMinorVersion << '.'
<< os.dwBuildNumber << '.'
<< os.wServicePackMajor << '.'
<< os.wServicePackMinor << '.'
<< ProductType << '.';
}
/* We also need to know about the processor architecture.
To retrieve this information accurately, check whether "GetNativeSystemInfo" is exported and use it then, otherwise fall back to "GetSystemInfo". */
hKernel32 = GetModuleHandleW(L"KERNEL32.DLL");
GetSysInfo = (GETSYSINFO)GetProcAddress(hKernel32, "GetNativeSystemInfo");
if(!GetSysInfo)
GetSysInfo = (GETSYSINFO)GetProcAddress(hKernel32, "GetSystemInfo");
GetSysInfo(&si);
ss << si.wProcessorArchitecture;
m_SystemInfoRequestString = ss.str();
}
/**
* Reads additional configuration options from the INI file.
*
* ParseParameters should be called before this function to get the desired result.
*/
void
CConfiguration::GetConfigurationFromFile()
{
DWORD Length;
string Value;
WCHAR ConfigFile[MAX_PATH];
/* Most values are only needed if we're going to submit anything */
if(m_Submit)
{
/* Build the path to the configuration file from the application's path */
GetModuleFileNameW(NULL, ConfigFile, MAX_PATH);
Length = wcsrchr(ConfigFile, '\\') - ConfigFile + 1;
wcscpy(&ConfigFile[Length], CONFIGURATION_FILENAMEW);
/* Check if it exists */
if(GetFileAttributesW(ConfigFile) == INVALID_FILE_ATTRIBUTES)
EXCEPTION("Missing \"" CONFIGURATION_FILENAMEA "\" configuration file!\n");
/* Get the user name */
m_AuthenticationRequestString = "&sourceid=";
Value = GetINIValue(L"Login", L"SourceID", ConfigFile);
if(Value.empty())
EXCEPTION("SourceID is missing in the configuration file\n");
m_AuthenticationRequestString += EscapeString(Value);
/* Get the password */
m_AuthenticationRequestString += "&password=";
Value = GetINIValue(L"Login", L"Password", ConfigFile);
if(Value.empty())
EXCEPTION("Password is missing in the configuration file\n");
m_AuthenticationRequestString += EscapeString(Value);
/* If we don't have any Comment string yet, try to find one in the INI file */
if(m_Comment.empty())
m_Comment = GetINIValue(L"Submission", L"Comment", ConfigFile);
}
}