mirror of
https://github.com/reactos/reactos.git
synced 2025-07-23 05:43:39 +00:00
398 lines
12 KiB
C++
398 lines
12 KiB
C++
/*
|
|
* Copyright (C) 2005 Casper S. Hornstrup
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
|
|
*/
|
|
#include "pch.h"
|
|
#include <assert.h>
|
|
|
|
#include "rbuild.h"
|
|
|
|
using std::string;
|
|
using std::vector;
|
|
|
|
TestSupportCode::TestSupportCode ( const Project& project )
|
|
: project ( project )
|
|
{
|
|
}
|
|
|
|
TestSupportCode::~TestSupportCode ()
|
|
{
|
|
}
|
|
|
|
bool
|
|
TestSupportCode::IsTestModule ( const Module& module )
|
|
{
|
|
return module.type == Test;
|
|
}
|
|
|
|
void
|
|
TestSupportCode::GenerateTestSupportCode ( bool verbose )
|
|
{
|
|
for ( size_t i = 0; i < project.modules.size (); i++ )
|
|
{
|
|
if ( IsTestModule ( *project.modules[i] ) )
|
|
{
|
|
GenerateTestSupportCodeForModule ( *project.modules[i],
|
|
verbose );
|
|
}
|
|
}
|
|
}
|
|
|
|
void
|
|
TestSupportCode::GenerateTestSupportCodeForModule ( Module& module,
|
|
bool verbose )
|
|
{
|
|
if ( verbose )
|
|
{
|
|
printf ( "\nGenerating test support code for %s",
|
|
module.name.c_str () );
|
|
}
|
|
|
|
WriteHooksFile ( module );
|
|
WriteStubsFile ( module );
|
|
WriteStartupFile ( module );
|
|
}
|
|
|
|
string
|
|
TestSupportCode::GetHooksFilename ( Module& module )
|
|
{
|
|
return NormalizeFilename ( Environment::GetIntermediatePath () + SSEP + module.GetBasePath () + SSEP + "_hooks.c" );
|
|
}
|
|
|
|
char*
|
|
TestSupportCode::WriteStubbedSymbolToHooksFile ( char* buffer,
|
|
const StubbedComponent& component,
|
|
const StubbedSymbol& symbol )
|
|
{
|
|
buffer = buffer + sprintf ( buffer,
|
|
" {\"%s\", \"%s\", NULL, NULL, NULL},\n",
|
|
component.name.c_str (),
|
|
symbol.newname.c_str () );
|
|
return buffer;
|
|
}
|
|
|
|
char*
|
|
TestSupportCode::WriteStubbedComponentToHooksFile ( char* buffer,
|
|
const StubbedComponent& component )
|
|
{
|
|
for ( size_t i = 0; i < component.symbols.size () ; i++ )
|
|
buffer = WriteStubbedSymbolToHooksFile ( buffer,
|
|
component,
|
|
*component.symbols[i] );
|
|
return buffer;
|
|
}
|
|
|
|
void
|
|
TestSupportCode::WriteHooksFile ( Module& module )
|
|
{
|
|
char* buf;
|
|
char* s;
|
|
|
|
buf = (char*) malloc ( 50*1024 );
|
|
if ( buf == NULL )
|
|
throw OutOfMemoryException ();
|
|
|
|
s = buf;
|
|
s = s + sprintf ( s, "/* This file is automatically generated. */\n" );
|
|
s = s + sprintf ( s, "#include <windows.h>\n" );
|
|
s = s + sprintf ( s, "#include \"regtests.h\"\n" );
|
|
s = s + sprintf ( s, "\n" );
|
|
s = s + sprintf ( s, "_API_DESCRIPTION ExternalDependencies[] =\n" );
|
|
s = s + sprintf ( s, "{\n" );
|
|
|
|
int symbolCount = 0;
|
|
for ( size_t i = 0; i < module.stubbedComponents.size () ; i++ )
|
|
{
|
|
s = WriteStubbedComponentToHooksFile ( s,
|
|
*module.stubbedComponents[i] );
|
|
symbolCount += module.stubbedComponents[i]->symbols.size ();
|
|
}
|
|
|
|
s = s + sprintf ( s, "};\n" );
|
|
s = s + sprintf ( s, "\n" );
|
|
s = s + sprintf ( s, "#define ExternalDependencyCount %d\n", symbolCount );
|
|
s = s + sprintf ( s, "ULONG MaxExternalDependency = ExternalDependencyCount - 1;\n" );
|
|
s = s + sprintf ( s, "\n" );
|
|
|
|
FileSupportCode::WriteIfChanged ( buf, GetHooksFilename ( module ) );
|
|
|
|
free ( buf );
|
|
}
|
|
|
|
string
|
|
TestSupportCode::GetStubsFilename ( Module& module )
|
|
{
|
|
return NormalizeFilename ( Environment::GetIntermediatePath () + SSEP + module.GetBasePath () + SSEP + "_stubs.S" );
|
|
}
|
|
|
|
string
|
|
GetImportSymbol ( const StubbedSymbol& symbol )
|
|
{
|
|
if (symbol.symbol[0] == '@')
|
|
return "__imp_" + symbol.symbol;
|
|
else
|
|
return "__imp__" + symbol.symbol;
|
|
}
|
|
|
|
char*
|
|
TestSupportCode::WriteStubbedSymbolToStubsFile ( char* buffer,
|
|
const StubbedComponent& component,
|
|
const StubbedSymbol& symbol,
|
|
int stubIndex )
|
|
{
|
|
string importSymbol = GetImportSymbol( symbol );
|
|
buffer = buffer + sprintf ( buffer,
|
|
".globl _%s\n",
|
|
symbol.symbol.c_str () );
|
|
buffer = buffer + sprintf ( buffer,
|
|
".globl %s\n",
|
|
importSymbol.c_str () );
|
|
buffer = buffer + sprintf ( buffer,
|
|
"_%s:\n",
|
|
symbol.symbol.c_str () );
|
|
buffer = buffer + sprintf ( buffer,
|
|
"%s:\n",
|
|
importSymbol.c_str () );
|
|
buffer = buffer + sprintf ( buffer,
|
|
" pushl $%d\n",
|
|
stubIndex );
|
|
buffer = buffer + sprintf ( buffer,
|
|
" jmp passthrough\n" );
|
|
buffer = buffer + sprintf ( buffer, "\n" );
|
|
return buffer;
|
|
}
|
|
|
|
char*
|
|
TestSupportCode::WriteStubbedComponentToStubsFile ( char* buffer,
|
|
const StubbedComponent& component,
|
|
int* stubIndex )
|
|
{
|
|
for ( size_t i = 0; i < component.symbols.size () ; i++ )
|
|
buffer = WriteStubbedSymbolToStubsFile ( buffer,
|
|
component,
|
|
*component.symbols[i],
|
|
(*stubIndex)++ );
|
|
return buffer;
|
|
}
|
|
|
|
void
|
|
TestSupportCode::WriteStubsFile ( Module& module )
|
|
{
|
|
char* buf;
|
|
char* s;
|
|
|
|
buf = (char*) malloc ( 50*1024 );
|
|
if ( buf == NULL )
|
|
throw OutOfMemoryException ();
|
|
|
|
s = buf;
|
|
s = s + sprintf ( s, "/* This file is automatically generated. */\n" );
|
|
s = s + sprintf ( s, "passthrough:\n" );
|
|
s = s + sprintf ( s, " call _FrameworkGetHook@4\n" );
|
|
s = s + sprintf ( s, " test %%eax, %%eax\n" );
|
|
s = s + sprintf ( s, " je .return\n" );
|
|
s = s + sprintf ( s, " jmp *%%eax\n" );
|
|
s = s + sprintf ( s, ".return:\n" );
|
|
s = s + sprintf ( s, " /* This will most likely corrupt the stack */\n" );
|
|
s = s + sprintf ( s, " ret\n" );
|
|
s = s + sprintf ( s, "\n" );
|
|
|
|
int stubIndex = 0;
|
|
for ( size_t i = 0; i < module.stubbedComponents.size () ; i++ )
|
|
{
|
|
s = WriteStubbedComponentToStubsFile ( s,
|
|
*module.stubbedComponents[i],
|
|
&stubIndex );
|
|
}
|
|
|
|
FileSupportCode::WriteIfChanged ( buf, GetStubsFilename ( module ) );
|
|
|
|
free ( buf );
|
|
}
|
|
|
|
string
|
|
TestSupportCode::GetStartupFilename ( Module& module )
|
|
{
|
|
return NormalizeFilename ( Environment::GetIntermediatePath () + SSEP + module.GetBasePath () + SSEP + "_startup.c" );
|
|
}
|
|
|
|
bool
|
|
TestSupportCode::IsUnknownCharacter ( char ch )
|
|
{
|
|
if ( ch >= 'a' && ch <= 'z' )
|
|
return false;
|
|
if ( ch >= 'A' && ch <= 'Z' )
|
|
return false;
|
|
if ( ch >= '0' && ch <= '9' )
|
|
return false;
|
|
return true;
|
|
}
|
|
|
|
string
|
|
TestSupportCode::GetTestDispatcherName ( string filename )
|
|
{
|
|
string filenamePart = ReplaceExtension ( GetFilename ( filename ), "" );
|
|
if ( filenamePart.length () > 0 )
|
|
filenamePart[0] = toupper ( filenamePart[0] );
|
|
for ( size_t i = 1; i < filenamePart.length (); i++ )
|
|
{
|
|
if ( IsUnknownCharacter ( filenamePart[i] ) )
|
|
filenamePart[i] = '_';
|
|
else
|
|
filenamePart[i] = tolower ( filenamePart[i] );
|
|
}
|
|
return filenamePart + "Test";
|
|
}
|
|
|
|
bool
|
|
TestSupportCode::IsTestFile ( string& filename ) const
|
|
{
|
|
if ( stricmp ( GetFilename ( filename ).c_str (), "setup.c" ) == 0 )
|
|
return false;
|
|
return true;
|
|
}
|
|
|
|
void
|
|
TestSupportCode::GetSourceFilenames ( string_list& list,
|
|
Module& module ) const
|
|
{
|
|
size_t i;
|
|
|
|
const vector<File*>& files = module.non_if_data.files;
|
|
for ( i = 0; i < files.size (); i++ )
|
|
{
|
|
if ( !files[i]->IsGeneratedFile () && IsTestFile ( files[i]->name ) )
|
|
list.push_back ( files[i]->name );
|
|
}
|
|
// intentionally make a copy so that we can append more work in
|
|
// the middle of processing without having to go recursive
|
|
vector<If*> v = module.non_if_data.ifs;
|
|
for ( i = 0; i < v.size (); i++ )
|
|
{
|
|
size_t j;
|
|
If& rIf = *v[i];
|
|
// check for sub-ifs to add to list
|
|
const vector<If*>& ifs = rIf.data.ifs;
|
|
for ( j = 0; j < ifs.size (); j++ )
|
|
v.push_back ( ifs[j] );
|
|
const vector<File*>& files = rIf.data.files;
|
|
for ( j = 0; j < files.size (); j++ )
|
|
{
|
|
File& file = *files[j];
|
|
if ( !file.IsGeneratedFile () && IsTestFile ( file.name ) )
|
|
{
|
|
list.push_back ( file.name );
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
char*
|
|
TestSupportCode::WriteTestDispatcherPrototypesToStartupFile ( char* buffer,
|
|
Module& module )
|
|
{
|
|
string_list files;
|
|
GetSourceFilenames ( files,
|
|
module );
|
|
for ( size_t i = 0; i < files.size (); i++ )
|
|
{
|
|
buffer = buffer + sprintf ( buffer,
|
|
"extern void %s(int Command, char *Buffer);\n",
|
|
GetTestDispatcherName ( files[i] ).c_str () );
|
|
}
|
|
buffer = buffer + sprintf ( buffer, "\n" );
|
|
return buffer;
|
|
}
|
|
|
|
char*
|
|
TestSupportCode::WriteRegisterTestsFunctionToStartupFile ( char* buffer,
|
|
Module& module )
|
|
{
|
|
buffer = buffer + sprintf ( buffer,
|
|
"extern void AddTest(TestRoutine Routine);\n" );
|
|
buffer = buffer + sprintf ( buffer,
|
|
"\n" );
|
|
|
|
buffer = buffer + sprintf ( buffer,
|
|
"void\n" );
|
|
buffer = buffer + sprintf ( buffer,
|
|
"RegisterTests()\n" );
|
|
buffer = buffer + sprintf ( buffer,
|
|
"{\n" );
|
|
|
|
string_list files;
|
|
GetSourceFilenames ( files,
|
|
module );
|
|
for ( size_t i = 0; i < files.size (); i++ )
|
|
{
|
|
buffer = buffer + sprintf ( buffer,
|
|
"AddTest((TestRoutine)%s);\n",
|
|
GetTestDispatcherName ( files[i]).c_str () );
|
|
}
|
|
buffer = buffer + sprintf ( buffer,
|
|
"}\n" );
|
|
buffer = buffer + sprintf ( buffer, "\n" );
|
|
return buffer;
|
|
}
|
|
|
|
void
|
|
TestSupportCode::WriteStartupFile ( Module& module )
|
|
{
|
|
char* buf;
|
|
char* s;
|
|
|
|
buf = (char*) malloc ( 50*1024 );
|
|
if ( buf == NULL )
|
|
throw OutOfMemoryException ();
|
|
|
|
s = buf;
|
|
s = s + sprintf ( s, "/* This file is automatically generated. */\n" );
|
|
s = s + sprintf ( s, "\n" );
|
|
s = s + sprintf ( s, "#include <windows.h>\n" );
|
|
s = s + sprintf ( s, "#include \"regtests.h\"\n" );
|
|
s = s + sprintf ( s, "\n" );
|
|
s = WriteTestDispatcherPrototypesToStartupFile ( s,
|
|
module );
|
|
s = WriteRegisterTestsFunctionToStartupFile ( s,
|
|
module );
|
|
s = s + sprintf ( s, "\n" );
|
|
s = s + sprintf ( s, "void\n" );
|
|
s = s + sprintf ( s, "ConsoleWrite(char *Buffer)\n" );
|
|
s = s + sprintf ( s, "{\n" );
|
|
s = s + sprintf ( s, " printf(Buffer);\n" );
|
|
s = s + sprintf ( s, "}\n" );
|
|
s = s + sprintf ( s, "\n" );
|
|
s = s + sprintf ( s, "int\n" );
|
|
s = s + sprintf ( s, "mainCRTStartup(HANDLE hInstance,\n" );
|
|
s = s + sprintf ( s, " HANDLE hPrevInstance,\n" );
|
|
s = s + sprintf ( s, " LPSTR lpszCmdParam,\n" );
|
|
s = s + sprintf ( s, " int nCmdShow)\n" );
|
|
s = s + sprintf ( s, "{\n" );
|
|
s = s + sprintf ( s, " _SetPriorityClass(_GetCurrentProcess(), HIGH_PRIORITY_CLASS);\n" );
|
|
s = s + sprintf ( s, " _SetThreadPriority(_GetCurrentThread(), THREAD_PRIORITY_TIME_CRITICAL);\n" );
|
|
s = s + sprintf ( s, " InitializeTests();\n" );
|
|
s = s + sprintf ( s, " RegisterTests();\n" );
|
|
s = s + sprintf ( s, " SetupOnce();\n" );
|
|
s = s + sprintf ( s, " PerformTests(ConsoleWrite, NULL);\n" );
|
|
s = s + sprintf ( s, " _ExitProcess(0);\n" );
|
|
s = s + sprintf ( s, " return 0;\n" );
|
|
s = s + sprintf ( s, "}\n" );
|
|
s = s + sprintf ( s, "\n" );
|
|
|
|
FileSupportCode::WriteIfChanged ( buf, GetStartupFilename ( module ) );
|
|
|
|
free ( buf );
|
|
}
|