* Implement <autoregister>
[reactos.git] / reactos / tools / rbuild / module.cpp
index 30ec36e..3bd1a5e 100644 (file)
@@ -51,15 +51,36 @@ Replace ( const string& s, const string& find, const string& with )
        return ret;
 }
 
+string
+ChangeSeparator ( const string& s,
+                  const char fromSeparator,
+                  const char toSeparator )
+{
+       string s2(s);
+       char* p = strchr ( &s2[0], fromSeparator );
+       while ( p )
+       {
+               *p++ = toSeparator;
+               p = strchr ( p, fromSeparator );
+       }
+       return s2;
+}
+
 string
 FixSeparator ( const string& s )
+{
+       return ChangeSeparator ( s, cBadSep, cSep );
+}
+
+string
+FixSeparatorForSystemCommand ( const string& s )
 {
        string s2(s);
-       char* p = strchr ( &s2[0], CBAD_SEP );
+       char* p = strchr ( &s2[0], DEF_CBAD_SEP );
        while ( p )
        {
-               *p++ = CSEP;
-               p = strchr ( p, CBAD_SEP );
+               *p++ = DEF_CSEP;
+               p = strchr ( p, DEF_CBAD_SEP );
        }
        return s2;
 }
@@ -111,7 +132,7 @@ GetSubPath (
                        "<directory> tag has invalid characters in 'name' attribute" );
        if ( !path.size() )
                return att_value;
-       return FixSeparator(path + CSEP + att_value);
+       return FixSeparator(path + cSep + att_value);
 }
 
 string
@@ -129,7 +150,7 @@ GetExtension ( const string& filename )
 string
 GetDirectory ( const string& filename )
 {
-       size_t index = filename.find_last_of ( CSEP );
+       size_t index = filename.find_last_of ( cSep );
        if ( index == string::npos )
                return "";
        else
@@ -139,7 +160,7 @@ GetDirectory ( const string& filename )
 string
 GetFilename ( const string& filename )
 {
-       size_t index = filename.find_last_of ( CSEP );
+       size_t index = filename.find_last_of ( cSep );
        if ( index == string::npos )
                return filename;
        else
@@ -166,30 +187,36 @@ GetBooleanValue ( const string& value )
                return false;
 }
 
+string
+ToLower ( string filename )
+{
+       for ( size_t i = 1; i < filename.length (); i++ )
+               filename[i] = tolower ( filename[i] );
+       return filename;
+}
+
 IfableData::~IfableData()
 {
        size_t i;
-       for ( i = 0; i < files.size(); i++ )
-               delete files[i];
-       for ( i = 0; i < includes.size(); i++ )
+       for ( i = 0; i < includes.size (); i++ )
                delete includes[i];
-       for ( i = 0; i < defines.size(); i++ )
+       for ( i = 0; i < defines.size (); i++ )
                delete defines[i];
-       for ( i = 0; i < libraries.size(); i++ )
+       for ( i = 0; i < libraries.size (); i++ )
                delete libraries[i];
-       for ( i = 0; i < properties.size(); i++ )
+       for ( i = 0; i < properties.size (); i++ )
                delete properties[i];
-       for ( i = 0; i < compilerFlags.size(); i++ )
+       for ( i = 0; i < compilerFlags.size (); i++ )
                delete compilerFlags[i];
-       for ( i = 0; i < ifs.size(); i++ )
+       for ( i = 0; i < ifs.size (); i++ )
                delete ifs[i];
+       for ( i = 0; i < compilationUnits.size (); i++ )
+               delete compilationUnits[i];
 }
 
 void IfableData::ProcessXML ()
 {
        size_t i;
-       for ( i = 0; i < files.size (); i++ )
-               files[i]->ProcessXML ();
        for ( i = 0; i < includes.size (); i++ )
                includes[i]->ProcessXML ();
        for ( i = 0; i < defines.size (); i++ )
@@ -202,6 +229,8 @@ void IfableData::ProcessXML ()
                compilerFlags[i]->ProcessXML ();
        for ( i = 0; i < ifs.size (); i++ )
                ifs[i]->ProcessXML ();
+       for ( i = 0; i < compilationUnits.size (); i++ )
+               compilationUnits[i]->ProcessXML ();
 }
 
 Module::Module ( const Project& project,
@@ -211,6 +240,8 @@ Module::Module ( const Project& project,
          node (moduleNode),
          importLibrary (NULL),
          bootstrap (NULL),
+         autoRegister(NULL),
+         linkerScript (NULL),
          pch (NULL),
          cplusplus (false),
          host (HostDefault)
@@ -248,6 +279,25 @@ Module::Module ( const Project& project,
        else
                extension = GetDefaultModuleExtension ();
 
+       att = moduleNode.GetAttribute ( "unicode", false );
+       if ( att != NULL )
+       {
+               const char* p = att->value.c_str();
+               if ( !stricmp ( p, "true" ) || !stricmp ( p, "yes" ) )
+                       isUnicode = true;
+               else if ( !stricmp ( p, "false" ) || !stricmp ( p, "no" ) )
+                       isUnicode = false;
+               else
+               {
+                       throw InvalidAttributeValueException (
+                               moduleNode.location,
+                               "unicode",
+                               att->value );
+               }
+       }
+       else
+               isUnicode = false;
+
        att = moduleNode.GetAttribute ( "entrypoint", false );
        if ( att != NULL )
                entrypoint = att->value;
@@ -353,6 +403,8 @@ Module::~Module ()
                delete linkerFlags[i];
        for ( i = 0; i < stubbedComponents.size(); i++ )
                delete stubbedComponents[i];
+       if ( linkerScript )
+               delete linkerScript;
        if ( pch )
                delete pch;
 }
@@ -378,7 +430,10 @@ Module::ProcessXML()
 
        size_t i;
        for ( i = 0; i < node.subElements.size(); i++ )
-               ProcessXMLSubElement ( *node.subElements[i], path );
+       {
+               ParseContext parseContext;
+               ProcessXMLSubElement ( *node.subElements[i], path, parseContext );
+       }
        for ( i = 0; i < invocations.size(); i++ )
                invocations[i]->ProcessXML ();
        for ( i = 0; i < dependencies.size(); i++ )
@@ -390,15 +445,21 @@ Module::ProcessXML()
        for ( i = 0; i < stubbedComponents.size(); i++ )
                stubbedComponents[i]->ProcessXML();
        non_if_data.ProcessXML();
+       if ( linkerScript )
+               linkerScript->ProcessXML();
        if ( pch )
                pch->ProcessXML();
+       if ( autoRegister )
+               autoRegister->ProcessXML();
 }
 
 void
 Module::ProcessXMLSubElement ( const XMLElement& e,
                                const string& path,
-                               If* pIf /*= NULL*/ )
+                               ParseContext& parseContext )
 {
+       If* pOldIf = parseContext.ifData;
+       CompilationUnit* pOldCompilationUnit = parseContext.compilationUnit;
        bool subs_invalid = false;
        string subpath ( path );
        if ( e.name == "file" && e.value.size () > 0 )
@@ -429,12 +490,22 @@ Module::ProcessXMLSubElement ( const XMLElement& e,
                        else if ( !stricmp ( ext.c_str(), ".cxx" ) )
                                cplusplus = true;
                }
-               File* pFile = new File ( FixSeparator ( path + CSEP + e.value ),
+               File* pFile = new File ( FixSeparator ( path + cSep + e.value ),
                                         first,
                                         switches,
                                         false );
-               if ( pIf )
-                       pIf->data.files.push_back ( pFile );
+               if ( parseContext.compilationUnit )
+                       parseContext.compilationUnit->files.push_back ( pFile );
+               else
+               {
+                       CompilationUnit* pCompilationUnit = new CompilationUnit ( pFile );
+                       if ( parseContext.ifData )
+                               parseContext.ifData->data.compilationUnits.push_back ( pCompilationUnit );
+                       else
+                               non_if_data.compilationUnits.push_back ( pCompilationUnit );
+               }
+               if ( parseContext.ifData )
+                       parseContext.ifData->data.files.push_back ( pFile );
                else
                        non_if_data.files.push_back ( pFile );
                subs_invalid = true;
@@ -442,8 +513,8 @@ Module::ProcessXMLSubElement ( const XMLElement& e,
        else if ( e.name == "library" && e.value.size () )
        {
                Library* pLibrary = new Library ( e, *this, e.value );
-               if ( pIf )
-                       pIf->data.libraries.push_back ( pLibrary );
+               if ( parseContext.ifData )
+                       parseContext.ifData->data.libraries.push_back ( pLibrary );
                else
                        non_if_data.libraries.push_back ( pLibrary );
                subs_invalid = true;
@@ -457,8 +528,8 @@ Module::ProcessXMLSubElement ( const XMLElement& e,
        else if ( e.name == "include" )
        {
                Include* include = new Include ( project, this, &e );
-               if ( pIf )
-                       pIf->data.includes.push_back ( include );
+               if ( parseContext.ifData )
+                       parseContext.ifData->data.includes.push_back ( include );
                else
                        non_if_data.includes.push_back ( include );
                subs_invalid = true;
@@ -466,15 +537,15 @@ Module::ProcessXMLSubElement ( const XMLElement& e,
        else if ( e.name == "define" )
        {
                Define* pDefine = new Define ( project, this, e );
-               if ( pIf )
-                       pIf->data.defines.push_back ( pDefine );
+               if ( parseContext.ifData )
+                       parseContext.ifData->data.defines.push_back ( pDefine );
                else
                        non_if_data.defines.push_back ( pDefine );
                subs_invalid = true;
        }
        else if ( e.name == "invoke" )
        {
-               if ( pIf )
+               if ( parseContext.ifData )
                        throw InvalidBuildFileException (
                                e.location,
                                "<invoke> is not a valid sub-element of <if>" );
@@ -483,7 +554,7 @@ Module::ProcessXMLSubElement ( const XMLElement& e,
        }
        else if ( e.name == "dependency" )
        {
-               if ( pIf )
+               if ( parseContext.ifData )
                        throw InvalidBuildFileException (
                                e.location,
                                "<dependency> is not a valid sub-element of <if>" );
@@ -492,7 +563,7 @@ Module::ProcessXMLSubElement ( const XMLElement& e,
        }
        else if ( e.name == "importlibrary" )
        {
-               if ( pIf )
+               if ( parseContext.ifData )
                        throw InvalidBuildFileException (
                                e.location,
                                "<importlibrary> is not a valid sub-element of <if>" );
@@ -505,29 +576,27 @@ Module::ProcessXMLSubElement ( const XMLElement& e,
        }
        else if ( e.name == "if" )
        {
-               If* pOldIf = pIf;
-               pIf = new If ( e, project, this );
+               parseContext.ifData = new If ( e, project, this );
                if ( pOldIf )
-                       pOldIf->data.ifs.push_back ( pIf );
+                       pOldIf->data.ifs.push_back ( parseContext.ifData );
                else
-                       non_if_data.ifs.push_back ( pIf );
+                       non_if_data.ifs.push_back ( parseContext.ifData );
                subs_invalid = false;
        }
        else if ( e.name == "ifnot" )
        {
-               If* pOldIf = pIf;
-               pIf = new If ( e, project, this, true );
+               parseContext.ifData = new If ( e, project, this, true );
                if ( pOldIf )
-                       pOldIf->data.ifs.push_back ( pIf );
+                       pOldIf->data.ifs.push_back ( parseContext.ifData );
                else
-                       non_if_data.ifs.push_back ( pIf );
+                       non_if_data.ifs.push_back ( parseContext.ifData );
                subs_invalid = false;
        }
        else if ( e.name == "compilerflag" )
        {
                CompilerFlag* pCompilerFlag = new CompilerFlag ( project, this, e );
-               if ( pIf )
-                       pIf->data.compilerFlags.push_back ( pCompilerFlag );
+               if ( parseContext.ifData )
+                       parseContext.ifData->data.compilerFlags.push_back ( pCompilerFlag );
                else
                        non_if_data.compilerFlags.push_back ( pCompilerFlag );
                subs_invalid = true;
@@ -537,6 +606,15 @@ Module::ProcessXMLSubElement ( const XMLElement& e,
                linkerFlags.push_back ( new LinkerFlag ( project, this, e ) );
                subs_invalid = true;
        }
+       else if ( e.name == "linkerscript" )
+       {
+               if ( linkerScript )
+                       throw InvalidBuildFileException (
+                               e.location,
+                               "Only one <linkerscript> is valid per module" );
+               linkerScript = new LinkerScript ( project, this, e );
+               subs_invalid = true;
+       }
        else if ( e.name == "component" )
        {
                stubbedComponents.push_back ( new StubbedComponent ( this, e ) );
@@ -555,7 +633,7 @@ Module::ProcessXMLSubElement ( const XMLElement& e,
        }
        else if ( e.name == "pch" )
        {
-               if ( pIf )
+               if ( parseContext.ifData )
                        throw InvalidBuildFileException (
                                e.location,
                                "<pch> is not a valid sub-element of <if>" );
@@ -564,7 +642,31 @@ Module::ProcessXMLSubElement ( const XMLElement& e,
                                e.location,
                                "Only one <pch> is valid per module" );
                pch = new PchFile (
-                       e, *this, File ( FixSeparator ( path + CSEP + e.value ), false, "", true ) );
+                       e, *this, File ( FixSeparator ( path + cSep + e.value ), false, "", true ) );
+               subs_invalid = true;
+       }
+       else if ( e.name == "compilationunit" )
+       {
+               if ( project.configuration.CompilationUnitsEnabled )
+               {
+                       CompilationUnit* pCompilationUnit = new CompilationUnit ( &project, this, &e );
+                       if ( parseContext.ifData )
+                               parseContext.ifData->data.compilationUnits.push_back ( pCompilationUnit );
+                       else
+                               non_if_data.compilationUnits.push_back ( pCompilationUnit );
+                       parseContext.compilationUnit = pCompilationUnit;
+               }
+               subs_invalid = false;
+       }
+       else if ( e.name == "autoregister" )
+       {
+               if ( autoRegister != NULL)
+               {
+                       throw InvalidBuildFileException ( e.location,
+                                                         "there can be only one <%s> element for a module",
+                                                         e.name.c_str() );
+               }
+               autoRegister = new AutoRegister ( project, this, e );
                subs_invalid = true;
        }
        if ( subs_invalid && e.subElements.size() > 0 )
@@ -573,7 +675,9 @@ Module::ProcessXMLSubElement ( const XMLElement& e,
                        "<%s> cannot have sub-elements",
                        e.name.c_str() );
        for ( size_t i = 0; i < e.subElements.size (); i++ )
-               ProcessXMLSubElement ( *e.subElements[i], subpath, pIf );
+               ProcessXMLSubElement ( *e.subElements[i], subpath, parseContext );
+       parseContext.ifData = pOldIf;
+       parseContext.compilationUnit = pOldCompilationUnit;
 }
 
 ModuleType
@@ -628,7 +732,7 @@ Module::GetDefaultModuleExtension () const
        switch (type)
        {
                case BuildTool:
-                       return EXEPOSTFIX;
+                       return ExePostfix;
                case StaticLibrary:
                        return ".a";
                case ObjectLibrary:
@@ -680,9 +784,15 @@ Module::GetDefaultModuleEntrypoint () const
                        return "_DllMain@12";
                case Win32CUI:
                case Test:
-                       return "_mainCRTStartup";
+                       if ( isUnicode )
+                               return "_wmainCRTStartup";
+                       else
+                               return "_mainCRTStartup";
                case Win32GUI:
-                       return "_WinMainCRTStartup";
+                       if ( isUnicode )
+                               return "_wWinMainCRTStartup";
+                       else
+                               return "_WinMainCRTStartup";
                case KernelModeDriver:
                        return "_DriverEntry@8";
                case BuildTool:
@@ -829,7 +939,7 @@ string
 Module::GetPath () const
 {
        if ( path.length() > 0 )
-               return path + CSEP + GetTargetName ();
+               return path + cSep + GetTargetName ();
        else
                return GetTargetName ();
 }
@@ -837,7 +947,7 @@ Module::GetPath () const
 string
 Module::GetPathWithPrefix ( const string& prefix ) const
 {
-       return path + CSEP + prefix + GetTargetName ();
+       return path + cSep + prefix + GetTargetName ();
 }
 
 string
@@ -854,11 +964,10 @@ Module::HasFileWithExtension (
        const std::string& extension ) const
 {
        size_t i;
-       for ( i = 0; i < data.files.size (); i++ )
+       for ( i = 0; i < data.compilationUnits.size (); i++ )
        {
-               File& file = *data.files[i];
-               string file_ext = GetExtension ( file.name );
-               if ( !stricmp ( file_ext.c_str (), extension.c_str () ) )
+               CompilationUnit* compilationUnit = data.compilationUnits[i];
+               if ( compilationUnit->HasFileWithExtension ( extension ) )
                        return true;
        }
        for ( i = 0; i < data.ifs.size (); i++ )
@@ -875,7 +984,7 @@ Module::InvokeModule () const
        for ( size_t i = 0; i < invocations.size (); i++ )
        {
                Invoke& invoke = *invocations[i];
-               string command = invoke.invokeModule->GetPath () + " " + invoke.GetParameters ();
+               string command = FixSeparatorForSystemCommand(invoke.invokeModule->GetPath ()) + " " + invoke.GetParameters ();
                printf ( "Executing '%s'\n\n", command.c_str () );
                int exitcode = system ( command.c_str () );
                if ( exitcode != 0 )
@@ -900,13 +1009,6 @@ File::ProcessXML()
 {
 }
 
-bool
-File::IsGeneratedFile () const
-{
-       string extension = GetExtension ( name );
-       return ( extension == ".spec" || extension == ".SPEC" );
-}
-
 
 Library::Library ( const XMLElement& _node,
                    const Module& _module,
@@ -995,7 +1097,7 @@ Invoke::ProcessXMLSubElementInput ( const XMLElement& e )
        bool subs_invalid = false;
        if ( e.name == "inputfile" && e.value.size () > 0 )
        {
-               input.push_back ( new InvokeFile ( e, FixSeparator ( module.path + CSEP + e.value ) ) );
+               input.push_back ( new InvokeFile ( e, FixSeparator ( module.path + cSep + e.value ) ) );
                subs_invalid = true;
        }
        if ( subs_invalid && e.subElements.size() > 0 )
@@ -1010,7 +1112,7 @@ Invoke::ProcessXMLSubElementOutput ( const XMLElement& e )
        bool subs_invalid = false;
        if ( e.name == "outputfile" && e.value.size () > 0 )
        {
-               output.push_back ( new InvokeFile ( e, FixSeparator ( module.path + CSEP + e.value ) ) );
+               output.push_back ( new InvokeFile ( e, FixSeparator ( module.path + cSep + e.value ) ) );
                subs_invalid = true;
        }
        if ( subs_invalid && e.subElements.size() > 0 )
@@ -1180,3 +1282,85 @@ void
 PchFile::ProcessXML()
 {
 }
+
+
+AutoRegister::AutoRegister ( const Project& project_,
+                             const Module* module_,
+                             const XMLElement& node_ )
+       : project(project_),
+         module(module_),
+         node(node_)
+{
+       Initialize();
+}
+
+AutoRegister::~AutoRegister ()
+{
+}
+
+bool
+AutoRegister::IsSupportedModuleType ( ModuleType type )
+{
+       switch ( type )
+       {
+               case Win32DLL:
+                       return true;
+               case Kernel:
+               case KernelModeDLL:
+               case NativeDLL:
+               case NativeCUI:
+               case Win32CUI:
+               case Win32GUI:
+               case KernelModeDriver:
+               case BootSector:
+               case BootLoader:
+               case BuildTool:
+               case StaticLibrary:
+               case ObjectLibrary:
+               case Iso:
+               case LiveIso:
+               case Test:
+               case RpcServer:
+               case RpcClient:
+               case Alias:
+                       return false;
+       }
+       throw InvalidOperationException ( __FILE__,
+                                         __LINE__ );
+}
+
+AutoRegisterType
+AutoRegister::GetAutoRegisterType( string type )
+{
+       if ( type == "DllRegisterServer" )
+               return DllRegisterServer;
+       if ( type == "DllInstall" )
+               return DllInstall;
+       if ( type == "Both" )
+               return Both;
+       throw InvalidBuildFileException (
+               node.location,
+               "<autoregister> type attribute must be DllRegisterServer, DllInstall or Both." );
+}
+
+void
+AutoRegister::Initialize ()
+{
+       if ( !IsSupportedModuleType ( module->type ) )
+       {
+               throw InvalidBuildFileException (
+                       node.location,
+                       "<autoregister> is not applicable for this module type." );
+       }
+
+       const XMLAttribute* att = node.GetAttribute ( "infsection", true );
+       infSection = att->value;
+
+       att = node.GetAttribute ( "type", true );
+       type = GetAutoRegisterType ( att->value );
+}
+
+void
+AutoRegister::ProcessXML()
+{
+}