reactos/rostests/rosautotest/tools.cpp

248 lines
5.8 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: Various helper functions
* COPYRIGHT: Copyright 2008-2015 Colin Finck <colin@reactos.org>
*/
#include "precomp.h"
static const char HexCharacters[] = "0123456789ABCDEF";
/**
* Escapes a string according to RFC 1738.
* Required for passing parameters to the web service.
*
* @param Input
* Constant pointer to a char array, which contains the input buffer to escape.
*
* @return
* The escaped string as std::string.
*/
string
EscapeString(const char* Input)
{
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;
}
/**
* Escapes a string according to RFC 1738.
* Required for passing parameters to the web service.
*
* @param Input
* Pointer to a std::string, which contains the input buffer to escape.
*
* @return
* The escaped string as std::string.
*/
string
EscapeString(const string& Input)
{
return EscapeString(Input.c_str());
}
/**
* Determines whether a string contains entirely numeric values.
*
* @param Input
* Constant pointer to a char array containing the input to check.
*
* @return
* true if the string is entirely numeric, false otherwise.
*/
bool
IsNumber(const char* Input)
{
do
{
if(!isdigit(*Input))
return false;
++Input;
}
while(*Input);
return true;
}
/**
* Outputs a string through the standard output and the debug output.
* The input string may have LF or CRLF line endings.
*
* @param InputString
* The std::string to output
*/
void
StringOut(const string& InputString)
{
const char* pInput = InputString.c_str();
char* OutputString = new char[InputString.size() + 1];
char* pOutput = OutputString;
/* Unify the line endings (the piped output of the tests may use CRLF) */
while (*pInput)
{
/* If this is a CRLF line-ending, only copy a \n to the new string and skip the next character */
if (*pInput == '\r' && *(pInput + 1) == '\n')
{
*pOutput = '\n';
++pInput;
}
else
{
*pOutput = *pInput;
}
++pInput;
++pOutput;
}
*pOutput = 0;
OutputDebugStringA(OutputString);
if (Configuration.DoPrint())
cout << OutputString << flush;
delete[] OutputString;
}
/**
* Gets a value from a specified INI file and returns it converted to ASCII.
*
* @param AppName
* Constant pointer to a WCHAR array with the INI section to look in (lpAppName parameter passed to GetPrivateProfileStringW)
*
* @param KeyName
* Constant pointer to a WCHAR array containing the key to look for in the specified section (lpKeyName parameter passed to GetPrivateProfileStringW)
*
* @param FileName
* Constant pointer to a WCHAR array containing the path to the INI file
*
* @return
* Returns the data of the value as std::string or an empty string if no data could be retrieved.
*/
string
GetINIValue(PCWCH AppName, PCWCH KeyName, PCWCH FileName)
{
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;
}
/**
* Converts an ASCII string to a Unicode one.
*
* @param AsciiString
* Constant pointer to a char array containing the ASCII string
*
* @return
* The Unicode string as std::wstring
*/
wstring
AsciiToUnicode(const char* AsciiString)
{
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;
}
/**
* Converts an ASCII string to a Unicode one.
*
* @param AsciiString
* Pointer to a std::string containing the ASCII string
*
* @return
* The Unicode string as std::wstring
*/
wstring
AsciiToUnicode(const string& AsciiString)
{
return AsciiToUnicode(AsciiString.c_str());
}
/**
* Converts a Unicode string to an ASCII one.
*
* @param UnicodeString
* Constant pointer to a WCHAR array containing the Unicode string
*
* @return
* The ASCII string as std::string
*/
string
UnicodeToAscii(PCWSTR UnicodeString)
{
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;
}
/**
* Converts a Unicode string to an ASCII one.
*
* @param UnicodeString
* Pointer to a std::wstring containing the Unicode string
*
* @return
* The ASCII string as std::string
*/
string
UnicodeToAscii(const wstring& UnicodeString)
{
return UnicodeToAscii(UnicodeString.c_str());
}