- Don't name-decorate the entrypoint. This lets msvc link the module properly and...
[reactos.git] / reactos / tools / rbuild / module.cpp
index 8bf1465..ab0863b 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;
 }
@@ -102,16 +123,16 @@ GetSubPath (
        const string& att_value )
 {
        if ( !att_value.size() )
-               throw InvalidBuildFileException (
+               throw XMLInvalidBuildFileException (
                        location,
                        "<directory> tag has empty 'name' attribute" );
        if ( strpbrk ( att_value.c_str (), "/\\?*:<>|" ) )
-               throw InvalidBuildFileException (
+               throw XMLInvalidBuildFileException (
                        location,
                        "<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,45 @@ 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;
+}
+
+void IfableData::ExtractModules( std::vector<Module*> &modules )
+{
+       size_t i;
+       for ( i = 0; i < this->modules.size (); i++ )
+               modules.push_back(this->modules[i]);
+}
+
 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 < modules.size(); i++ )
+               delete modules[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 +238,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 +249,7 @@ Module::Module ( const Project& project,
          node (moduleNode),
          importLibrary (NULL),
          bootstrap (NULL),
+         autoRegister(NULL),
          linkerScript (NULL),
          pch (NULL),
          cplusplus (false),
@@ -358,6 +397,12 @@ Module::Module ( const Project& project,
                aliasedModuleName = att->value;
        else
                aliasedModuleName = "";
+
+       if ( type == BootProgram )
+       {
+               att = moduleNode.GetAttribute ( "payload", true );
+               payload = att->value;
+       }
 }
 
 Module::~Module ()
@@ -385,22 +430,29 @@ Module::ProcessXML()
        if ( type == Alias )
        {
                if ( aliasedModuleName == name )
-                       throw InvalidBuildFileException (
+               {
+                       throw XMLInvalidBuildFileException (
                                node.location,
                                "module '%s' cannot link against itself",
                                name.c_str() );
-               const Module* m = project.LocateModule ( aliasedModuleName );
+               }
+               const Module* m = project.LocateModule ( aliasedModuleName );
                if ( !m )
-                       throw InvalidBuildFileException (
+               {
+                       throw XMLInvalidBuildFileException (
                                node.location,
                                "module '%s' trying to alias non-existant module '%s'",
                                name.c_str(),
                                aliasedModuleName.c_str() );
+               }
        }
 
        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++ )
@@ -416,13 +468,17 @@ Module::ProcessXML()
                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 )
@@ -434,9 +490,11 @@ Module::ProcessXMLSubElement ( const XMLElement& e,
                        if ( !stricmp ( att->value.c_str(), "true" ) )
                                first = true;
                        else if ( stricmp ( att->value.c_str(), "false" ) )
-                               throw InvalidBuildFileException (
+                       {
+                               throw XMLInvalidBuildFileException (
                                        e.location,
                                        "attribute 'first' of <file> element can only be 'true' or 'false'" );
+                       }
                }
                string switches = "";
                att = e.GetAttribute ( "switches", false );
@@ -453,12 +511,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;
@@ -466,8 +534,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;
@@ -481,8 +549,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;
@@ -490,68 +558,74 @@ 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 )
-                       throw InvalidBuildFileException (
+               if ( parseContext.ifData )
+               {
+                       throw XMLInvalidBuildFileException (
                                e.location,
                                "<invoke> is not a valid sub-element of <if>" );
+               }
                invocations.push_back ( new Invoke ( e, *this ) );
                subs_invalid = false;
        }
        else if ( e.name == "dependency" )
        {
-               if ( pIf )
-                       throw InvalidBuildFileException (
+               if ( parseContext.ifData )
+               {
+                       throw XMLInvalidBuildFileException (
                                e.location,
                                "<dependency> is not a valid sub-element of <if>" );
+               }
                dependencies.push_back ( new Dependency ( e, *this ) );
                subs_invalid = true;
        }
        else if ( e.name == "importlibrary" )
        {
-               if ( pIf )
-                       throw InvalidBuildFileException (
+               if ( parseContext.ifData )
+               {
+                       throw XMLInvalidBuildFileException (
                                e.location,
                                "<importlibrary> is not a valid sub-element of <if>" );
+               }
                if ( importLibrary )
-                       throw InvalidBuildFileException (
+               {
+                       throw XMLInvalidBuildFileException (
                                e.location,
                                "Only one <importlibrary> is valid per module" );
+               }
                importLibrary = new ImportLibrary ( e, *this );
                subs_invalid = true;
        }
        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;
@@ -564,9 +638,11 @@ Module::ProcessXMLSubElement ( const XMLElement& e,
        else if ( e.name == "linkerscript" )
        {
                if ( linkerScript )
-                       throw InvalidBuildFileException (
+               {
+                       throw XMLInvalidBuildFileException (
                                e.location,
                                "Only one <linkerscript> is valid per module" );
+               }
                linkerScript = new LinkerScript ( project, this, e );
                subs_invalid = true;
        }
@@ -577,7 +653,7 @@ Module::ProcessXMLSubElement ( const XMLElement& e,
        }
        else if ( e.name == "property" )
        {
-               throw InvalidBuildFileException (
+               throw XMLInvalidBuildFileException (
                        e.location,
                        "<property> is not a valid sub-element of <module>" );
        }
@@ -588,25 +664,58 @@ Module::ProcessXMLSubElement ( const XMLElement& e,
        }
        else if ( e.name == "pch" )
        {
-               if ( pIf )
-                       throw InvalidBuildFileException (
+               if ( parseContext.ifData )
+               {
+                       throw XMLInvalidBuildFileException (
                                e.location,
                                "<pch> is not a valid sub-element of <if>" );
+               }
                if ( pch )
-                       throw InvalidBuildFileException (
+               {
+                       throw XMLInvalidBuildFileException (
                                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 XMLInvalidBuildFileException (
+                               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 )
-               throw InvalidBuildFileException (
+       {
+               throw XMLInvalidBuildFileException (
                        e.location,
                        "<%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
@@ -638,6 +747,8 @@ Module::GetModuleType ( const string& location, const XMLAttribute& attribute )
                return BootLoader;
        if ( attribute.value == "bootsector" )
                return BootSector;
+       if ( attribute.value == "bootprogram" )
+               return BootProgram;
        if ( attribute.value == "iso" )
                return Iso;
        if ( attribute.value == "liveiso" )
@@ -661,7 +772,7 @@ Module::GetDefaultModuleExtension () const
        switch (type)
        {
                case BuildTool:
-                       return EXEPOSTFIX;
+                       return ExePostfix;
                case StaticLibrary:
                        return ".a";
                case ObjectLibrary:
@@ -691,6 +802,8 @@ Module::GetDefaultModuleExtension () const
                        return ".o";
                case Alias:
                        return "";
+               case BootProgram:
+                       return "";
        }
        throw InvalidOperationException ( __FILE__,
                                          __LINE__ );
@@ -704,7 +817,7 @@ Module::GetDefaultModuleEntrypoint () const
                case Kernel:
                        return "_NtProcessStartup";
                case KernelModeDLL:
-                       return "_DriverEntry@8";
+                       return "DriverEntry";
                case NativeDLL:
                        return "_DllMainCRTStartup@12";
                case NativeCUI:
@@ -723,7 +836,7 @@ Module::GetDefaultModuleEntrypoint () const
                        else
                                return "_WinMainCRTStartup";
                case KernelModeDriver:
-                       return "_DriverEntry@8";
+                       return "DriverEntry";
                case BuildTool:
                case StaticLibrary:
                case ObjectLibrary:
@@ -734,6 +847,7 @@ Module::GetDefaultModuleEntrypoint () const
                case RpcServer:
                case RpcClient:
                case Alias:
+               case BootProgram:
                        return "";
        }
        throw InvalidOperationException ( __FILE__,
@@ -769,6 +883,7 @@ Module::GetDefaultModuleBaseaddress () const
                case RpcServer:
                case RpcClient:
                case Alias:
+               case BootProgram:
                        return "";
        }
        throw InvalidOperationException ( __FILE__,
@@ -801,6 +916,7 @@ Module::IsDLL () const
                case ObjectLibrary:
                case BootLoader:
                case BootSector:
+               case BootProgram:
                case Iso:
                case LiveIso:
                case RpcServer:
@@ -829,6 +945,7 @@ Module::GenerateInOutputTree () const
                case BuildTool:
                case BootLoader:
                case BootSector:
+               case BootProgram:
                case Iso:
                case LiveIso:
                        return true;
@@ -868,7 +985,7 @@ string
 Module::GetPath () const
 {
        if ( path.length() > 0 )
-               return path + CSEP + GetTargetName ();
+               return path + cSep + GetTargetName ();
        else
                return GetTargetName ();
 }
@@ -876,7 +993,20 @@ Module::GetPath () const
 string
 Module::GetPathWithPrefix ( const string& prefix ) const
 {
-       return path + CSEP + prefix + GetTargetName ();
+       return path + cSep + prefix + GetTargetName ();
+}
+
+string
+Module::GetPathToBaseDir () const
+{
+       string temp_path = path;
+       string result = "..\\";
+       while(temp_path.find ('\\') != string::npos)
+       {
+               temp_path.erase (0, temp_path.find('\\')+1);
+               result += "..\\";
+       }
+       return result;
 }
 
 string
@@ -893,11 +1023,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++ )
@@ -914,7 +1043,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 )
@@ -939,13 +1068,6 @@ File::ProcessXML()
 {
 }
 
-bool
-File::IsGeneratedFile () const
-{
-       string extension = GetExtension ( name );
-       return ( extension == ".spec" || extension == ".SPEC" );
-}
-
 
 Library::Library ( const XMLElement& _node,
                    const Module& _module,
@@ -956,27 +1078,33 @@ Library::Library ( const XMLElement& _node,
          importedModule(_module.project.LocateModule(_name))
 {
        if ( module.name == name )
-               throw InvalidBuildFileException (
+       {
+               throw XMLInvalidBuildFileException (
                        node.location,
                        "module '%s' cannot link against itself",
                        name.c_str() );
+       }
        if ( !importedModule )
-               throw InvalidBuildFileException (
+       {
+               throw XMLInvalidBuildFileException (
                        node.location,
                        "module '%s' trying to import non-existant module '%s'",
                        module.name.c_str(),
                        name.c_str() );
+       }
 }
 
 void
 Library::ProcessXML()
 {
        if ( !module.project.LocateModule ( name ) )
-               throw InvalidBuildFileException (
+       {
+               throw XMLInvalidBuildFileException (
                        node.location,
                        "module '%s' is trying to link against non-existant module '%s'",
                        module.name.c_str(),
                        name.c_str() );
+       }
 }
 
 
@@ -997,11 +1125,13 @@ Invoke::ProcessXML()
        {
                invokeModule = module.project.LocateModule ( att->value );
                if ( invokeModule == NULL )
-                       throw InvalidBuildFileException (
+               {
+                       throw XMLInvalidBuildFileException (
                                node.location,
                                "module '%s' is trying to invoke non-existant module '%s'",
                                module.name.c_str(),
                                att->value.c_str() );
+               }
        }
 
        for ( size_t i = 0; i < node.subElements.size (); i++ )
@@ -1023,9 +1153,12 @@ Invoke::ProcessXMLSubElement ( const XMLElement& e )
                        ProcessXMLSubElementOutput ( *e.subElements[i] );
        }
        if ( subs_invalid && e.subElements.size() > 0 )
-               throw InvalidBuildFileException ( e.location,
-                                                 "<%s> cannot have sub-elements",
-                                                 e.name.c_str() );
+       {
+               throw XMLInvalidBuildFileException (
+                       e.location,
+                       "<%s> cannot have sub-elements",
+                       e.name.c_str() );
+       }
 }
 
 void
@@ -1034,13 +1167,17 @@ 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 )
-               throw InvalidBuildFileException ( e.location,
-                                                 "<%s> cannot have sub-elements",
-                                                 e.name.c_str() );
+       {
+               throw XMLInvalidBuildFileException (
+                       e.location,
+                       "<%s> cannot have sub-elements",
+                       e.name.c_str() );
+       }
 }
 
 void
@@ -1049,14 +1186,17 @@ 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 )
-               throw InvalidBuildFileException (
+       {
+               throw XMLInvalidBuildFileException (
                        e.location,
                        "<%s> cannot have sub-elements",
                        e.name.c_str() );
+       }
 }
 
 void
@@ -1134,10 +1274,13 @@ Dependency::ProcessXML()
 {
        dependencyModule = module.project.LocateModule ( node.value );
        if ( dependencyModule == NULL )
-               throw InvalidBuildFileException ( node.location,
-                                                 "module '%s' depend on non-existant module '%s'",
-                                                 module.name.c_str(),
-                                                 node.value.c_str() );
+       {
+               throw XMLInvalidBuildFileException (
+                       node.location,
+                       "module '%s' depend on non-existant module '%s'",
+                       module.name.c_str(),
+                       node.value.c_str() );
+       }
 }
 
 
@@ -1182,6 +1325,7 @@ If::~If ()
 void
 If::ProcessXML()
 {
+       
 }
 
 
@@ -1219,3 +1363,86 @@ 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 BootProgram:
+               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 XMLInvalidBuildFileException (
+               node.location,
+               "<autoregister> type attribute must be DllRegisterServer, DllInstall or Both." );
+}
+
+void
+AutoRegister::Initialize ()
+{
+       if ( !IsSupportedModuleType ( module->type ) )
+       {
+               throw XMLInvalidBuildFileException (
+                       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()
+{
+}