]> git.proxmox.com Git - mirror_edk2.git/blobdiff - Tools/Source/GenBuild/org/tianocore/build/autogen/AutoGen.java
Fixed EDKT318
[mirror_edk2.git] / Tools / Source / GenBuild / org / tianocore / build / autogen / AutoGen.java
index ad4a5c74829a1abf1f37e32c21c73d6198a402d6..5b6620b50dfc8d5f3672bcb012fe825a4fd482ed 100644 (file)
@@ -22,6 +22,7 @@ import java.io.FileInputStream;
 import java.io.FileOutputStream;\r
 import java.io.FileReader;\r
 import java.io.FileWriter;\r
+import java.io.IOException;\r
 import java.util.ArrayList;\r
 import java.util.HashSet;\r
 import java.util.Iterator;\r
@@ -33,24 +34,21 @@ import java.util.Set;
 \r
 import org.apache.tools.ant.BuildException;\r
 import org.apache.xmlbeans.XmlObject;\r
-import org.tianocore.GuidsDocument;\r
-import org.tianocore.LibraryClassDocument.LibraryClass;\r
-import org.tianocore.PPIsDocument;\r
-import org.tianocore.ProtocolsDocument;\r
-import org.tianocore.build.exception.*;\r
+import org.tianocore.build.exception.AutoGenException;\r
 import org.tianocore.build.global.GlobalData;\r
-import org.tianocore.build.global.Spd;\r
 import org.tianocore.build.global.SurfaceAreaQuery;\r
 import org.tianocore.build.id.ModuleIdentification;\r
 import org.tianocore.build.id.PackageIdentification;\r
 import org.tianocore.build.pcd.action.PCDAutoGenAction;\r
-\r
+import org.tianocore.common.definitions.ToolDefinitions;\r
+import org.tianocore.common.definitions.EdkDefinitions;\r
+import org.tianocore.common.exception.EdkException;\r
 import org.tianocore.common.logger.EdkLog;\r
 \r
 /**\r
* This class is to generate Autogen.h and Autogen.c according to module surface\r
* area or library surface area.\r
- */\r
+  This class is to generate Autogen.h and Autogen.c according to module surface\r
+  area or library surface area.\r
+**/\r
 public class AutoGen {\r
     ///\r
     /// The output path of Autogen.h and Autogen.c\r
@@ -101,170 +99,180 @@ public class AutoGen {
     ///\r
     private Set<String> mGuidList = new HashSet<String>();\r
 \r
-    //\r
-    // The dependence package list which recoreded in module or library surface\r
-    // area and it's dependence on library instance surface are.\r
-    //\r
+    ///\r
+    /// The dependence package list which recoreded in module or library surface\r
+    /// area and it's dependence on library instance surface area.\r
+    ///\r
     private List<PackageIdentification> mDepPkgList = new LinkedList<PackageIdentification>();\r
 \r
-    //\r
-    //  For non library module, add its library instance's construct and destructor to\r
-    //  list.\r
-    //\r
-    private List<String> libConstructList = new ArrayList<String>();\r
-    private List<String> libDestructList = new ArrayList<String>();\r
+    ///\r
+    ///  For non library module, add its library instance's construct and destructor to\r
+    ///  list. String[0] recode LibConstructor name, String[1] recode Lib instance \r
+       ///  module type.\r
+    ///\r
+    private List<String[]> libConstructList = new ArrayList<String[]>();\r
+    private List<String[]> libDestructList = new ArrayList<String[]>();\r
 \r
-    //\r
-    // List to store SetVirtalAddressMapCallBack, ExitBootServiceCallBack\r
-    //\r
+    ///\r
+    /// List to store SetVirtalAddressMapCallBack, ExitBootServiceCallBack\r
+    ///\r
     private List<String> setVirtalAddList = new ArrayList<String>();\r
     private List<String> exitBootServiceList = new ArrayList<String>();\r
 \r
+    private SurfaceAreaQuery saq = null;\r
+    \r
+    private ModuleIdentification parentId = null;\r
 \r
     /**\r
-     * Construct function\r
-     *\r
-     * This function mainly initialize some member variable.\r
-     *\r
-     * @param outputPath\r
-     *            Output path of AutoGen file.\r
-     * @param baseName\r
-     *            Module base name.\r
-     * @param arch\r
-     *            Target architecture.\r
-     */\r
-    public AutoGen(String fvDir, String outputPath, ModuleIdentification moduleId, String arch) {\r
+      Construct function\r
+     \r
+      This function mainly initialize some member variable.\r
+      \r
+      @param fvDir\r
+                 Absolute path of FV directory.\r
+      @param outputPath\r
+                 Output path of AutoGen file.\r
+      @param moduleId\r
+                 Module identification.\r
+      @param arch\r
+                 Target architecture.\r
+    **/\r
+    public AutoGen(String fvDir, String outputPath, ModuleIdentification moduleId, String arch, SurfaceAreaQuery saq, ModuleIdentification parentId) {\r
         this.outputPath = outputPath;\r
         this.moduleId = moduleId;\r
         this.arch = arch;\r
         this.fvDir = fvDir;\r
-\r
+        this.saq = saq;\r
+        this.parentId = parentId;\r
     }\r
 \r
     /**\r
-     * saveFile function\r
-     *\r
-     * This function save the content in stringBuffer to file.\r
-     *\r
-     * @param fileName\r
-     *            The name of file.\r
-     * @param fileBuffer\r
-     *            The content of AutoGen file in buffer.\r
-     * @return "true" successful, "false" failed.\r
-     */\r
+      saveFile function\r
+     \r
+      This function save the content in stringBuffer to file.\r
+     \r
+      @param fileName\r
+                 The name of file.\r
+      @param fileBuffer\r
+                 The content of AutoGen file in buffer.\r
+      @return boolean           \r
+                 "true"  successful\r
+                 "false" failed\r
+    **/\r
     private boolean saveFile(String fileName, StringBuffer fileBuffer) {\r
-        try {\r
-            File autoGenH = new File(fileName);\r
+        \r
+        File autoGenH = new File(fileName);\r
 \r
             //\r
             // if the file exists, compare their content\r
             //\r
-            if (autoGenH.exists()) {\r
-                FileReader fIn = new FileReader(autoGenH);\r
-                char[] oldFileBuffer = new char[(int) autoGenH.length()];\r
-                fIn.read(oldFileBuffer, 0, (int) autoGenH.length());\r
+        if (autoGenH.exists()) {\r
+            char[] oldFileBuffer = new char[(int) autoGenH.length()];\r
+                       try {\r
+                               FileReader fIn = new FileReader(autoGenH);\r
+                               fIn.read(oldFileBuffer, 0, (int) autoGenH.length());\r
                 fIn.close();\r
+                       } catch (IOException e) {\r
+                               EdkLog.log(EdkLog.EDK_INFO, this.moduleId.getName() \r
+                                                               + "'s " \r
+                                                               + fileName \r
+                                                               + " is exist, but can't be open!!");\r
+                               return false;\r
+                       }\r
 \r
-                //\r
-                // if we got the same file, don't re-generate it to prevent\r
-                // sources depending on it from re-building\r
-                //\r
-                if (fileBuffer.toString().compareTo(new String(oldFileBuffer)) == 0) {\r
-                    return true;\r
-                }\r
+            //\r
+            // if we got the same file, don't re-generate it to prevent\r
+            // sources depending on it from re-building\r
+            //\r
+            if (fileBuffer.toString().compareTo(new String(oldFileBuffer)) == 0) {\r
+                return true;\r
             }\r
-            FileWriter fOut = new FileWriter(autoGenH);\r
+        }\r
+        \r
+               try {\r
+                       FileWriter fOut = new FileWriter(autoGenH);\r
             fOut.write(fileBuffer.toString());\r
+                       fOut.flush();\r
             fOut.close();\r
-        } catch (Exception e) {\r
-            return false;\r
-        }\r
+               } catch (IOException e) {\r
+                       EdkLog.log(EdkLog.EDK_INFO, this.moduleId.getName() \r
+                                                               + "'s " \r
+                                                               + fileName \r
+                                                               + " can't be create!!");\r
+                       return false;\r
+               } \r
         return true;\r
     }\r
 \r
     /**\r
-     * genAutogen function\r
-     *\r
-     * This function call libGenAutoGen or moduleGenAutogen function, which\r
-     * dependence on generate library autogen or module autogen.\r
-     *\r
-     * @throws BuildException\r
-     *             Failed to creat AutoGen.c & AutoGen.h.\r
-     */\r
-    public void genAutogen() throws BuildException {\r
+      genAutogen function\r
+     \r
+      This function call libGenAutoGen or moduleGenAutogen function, which\r
+      dependence on generate library autogen or module autogen.\r
+     \r
+      @throws BuildException\r
+                  Failed to creat AutoGen.c & AutoGen.h.\r
+    **/\r
+    public void genAutogen() throws EdkException {\r
         try {\r
             //\r
             // If outputPath do not exist, create it.\r
             //\r
             File path = new File(outputPath);\r
             path.mkdirs();\r
-\r
-            //\r
-            // Check current is library or not, then call the corresponding\r
-            // function.\r
-            //\r
-            if (this.moduleId.isLibrary()) {\r
-                libGenAutogen();\r
-            } else {\r
-                moduleGenAutogen();\r
-            }\r
-\r
-        } catch (Exception e) {\r
-            throw new BuildException(\r
-                                    "Failed to create AutoGen.c & AutoGen.h!\n"\r
-                                    + e.getMessage());\r
+               } catch (Exception e) {\r
+            throw new AutoGenException(\r
+                                    "Failed to create "\r
+                                    + outputPath + " directory");\r
         }\r
+            \r
+               //\r
+               // Check current is library or not, then call the corresponding\r
+               // function.\r
+               //\r
+               if (this.moduleId.isLibrary()) {\r
+                       libGenAutogen();\r
+               } else {\r
+                       moduleGenAutogen();\r
+               }\r
     }\r
 \r
     /**\r
-     * moduleGenAutogen function\r
-     *\r
-     * This function generates AutoGen.c & AutoGen.h for module.\r
-     *\r
-     * @throws BuildException\r
-     *             Faile to create module AutoGen.c & AutoGen.h.\r
-     */\r
-    void moduleGenAutogen() throws BuildException {\r
-\r
-        try {\r
-            collectLibInstanceInfo();\r
-            moduleGenAutogenC();\r
-            moduleGenAutogenH();\r
-        } catch (Exception e) {\r
-            throw new BuildException(\r
-                                    "Faile to create module AutoGen.c & AutoGen.h!\n"\r
-                                    + e.getMessage());\r
-        }\r
+      moduleGenAutogen function\r
+     \r
+      This function generates AutoGen.c & AutoGen.h for module.\r
+     \r
+      @throws BuildException\r
+                  Faile to create module AutoGen.c & AutoGen.h.\r
+    **/\r
+    void moduleGenAutogen() throws EdkException {\r
+               collectLibInstanceInfo();\r
+               moduleGenAutogenC();\r
+               moduleGenAutogenH();\r
     }\r
 \r
     /**\r
-     * libGenAutogen function\r
-     *\r
-     * This function generates AutoGen.c & AutoGen.h for library.\r
-     *\r
-     * @throws BuildException\r
-     *             Faile to create library AutoGen.c & AutoGen.h\r
-     */\r
-    void libGenAutogen() throws BuildException {\r
-        try {\r
-            libGenAutogenC();\r
-            libGenAutogenH();\r
-        } catch (Exception e) {\r
-            throw new BuildException(\r
-                                    "Failed to create library AutoGen.c & AutoGen.h!\n"\r
-                                    + e.getMessage());\r
-        }\r
+      libGenAutogen function\r
+     \r
+      This function generates AutoGen.c & AutoGen.h for library.\r
+     \r
+      @throws BuildException\r
+                  Faile to create library AutoGen.c & AutoGen.h\r
+    **/\r
+    void libGenAutogen() throws EdkException {\r
+               libGenAutogenC();\r
+               libGenAutogenH();\r
     }\r
 \r
     /**\r
-     * moduleGenAutogenH\r
-     *\r
-     * This function generates AutoGen.h for module.\r
-     *\r
-     * @throws BuildException\r
-     *             Failed to generate AutoGen.h.\r
-     */\r
-    void moduleGenAutogenH() throws AutoGenException {\r
+      moduleGenAutogenH\r
+     \r
+      This function generates AutoGen.h for module.\r
+     \r
+      @throws BuildException\r
+                  Failed to generate AutoGen.h.\r
+    **/\r
+    void moduleGenAutogenH() throws EdkException {\r
 \r
         Set<String> libClassIncludeH;\r
         String moduleType;\r
@@ -276,14 +284,21 @@ public class AutoGen {
         //\r
         // Write Autogen.h header notation\r
         //\r
-        fileBuffer.append(CommonDefinition.autogenHNotation);\r
+        fileBuffer.append(CommonDefinition.AUTOGENHNOTATION);\r
 \r
         //\r
         // Add #ifndef ${BaseName}_AUTOGENH\r
         // #def ${BseeName}_AUTOGENH\r
         //\r
-        fileBuffer.append("#ifndef    " + "_AUTOGENH_" + this.moduleId.getGuid().replaceAll("-", "_") +"\r\n");\r
-        fileBuffer.append("#define    " + "_AUTOGENH_" + this.moduleId.getGuid().replaceAll("-", "_") +"\r\n\r\n");\r
+        fileBuffer.append(CommonDefinition.IFNDEF \r
+                                     + CommonDefinition.AUTOGENH\r
+                                     + this.moduleId.getGuid().replaceAll("-", "_") \r
+                                     + ToolDefinitions.LINE_SEPARATOR);\r
+        fileBuffer.append(CommonDefinition.DEFINE \r
+                                     + CommonDefinition.AUTOGENH\r
+                                     + this.moduleId.getGuid().replaceAll("-", "_") \r
+                                     + ToolDefinitions.LINE_SEPARATOR \r
+                                     + ToolDefinitions.LINE_SEPARATOR);\r
 \r
         //\r
         // Write the specification version and release version at the begine\r
@@ -292,20 +307,20 @@ public class AutoGen {
         // be got from module surface area instead of hard code by it's\r
         // moduleType.\r
         //\r
-        moduleType = SurfaceAreaQuery.getModuleType();\r
+        moduleType = saq.getModuleType();\r
 \r
         //\r
         // Add "extern int __make_me_compile_correctly;" at begin of\r
         // AutoGen.h.\r
         //\r
-        fileBuffer.append(CommonDefinition.autoGenHbegin);\r
+        fileBuffer.append(CommonDefinition.AUTOGENHBEGIN);\r
 \r
         //\r
         // Put EFI_SPECIFICATION_VERSION, and EDK_RELEASE_VERSION.\r
         //\r
-        String[] specList = SurfaceAreaQuery.getExternSpecificaiton();\r
+        String[] specList = saq.getExternSpecificaiton();\r
         for (int i = 0; i < specList.length; i++) {\r
-            fileBuffer.append(CommonDefinition.marcDefineStr + specList[i]\r
+            fileBuffer.append(CommonDefinition.DEFINE + specList[i]\r
                               + "\r\n");\r
         }\r
         //\r
@@ -313,8 +328,7 @@ public class AutoGen {
         //\r
         // PackageIdentification[] consumedPkgIdList = SurfaceAreaQuery\r
         // .getDependencePkg(this.arch);\r
-        PackageIdentification[] consumedPkgIdList = SurfaceAreaQuery\r
-                                                    .getDependencePkg(this.arch);\r
+        PackageIdentification[] consumedPkgIdList = saq.getDependencePkg(this.arch);\r
         if (consumedPkgIdList != null) {\r
             headerFileList = depPkgToAutogenH(consumedPkgIdList, moduleType);\r
             item = headerFileList.iterator();\r
@@ -326,8 +340,7 @@ public class AutoGen {
         //\r
         // Write library class's related *.h file to autogen.h.\r
         //\r
-        String[] libClassList = SurfaceAreaQuery\r
-                                .getLibraryClasses(CommonDefinition.AlwaysConsumed,this.arch);\r
+        String[] libClassList = saq.getLibraryClasses(CommonDefinition.ALWAYSCONSUMED,this.arch);\r
         if (libClassList != null) {\r
             libClassIncludeH = LibraryClassToAutogenH(libClassList);\r
             item = libClassIncludeH.iterator();\r
@@ -336,8 +349,7 @@ public class AutoGen {
             }\r
         }\r
 \r
-        libClassList = SurfaceAreaQuery\r
-                       .getLibraryClasses(CommonDefinition.AlwaysProduced, this.arch);\r
+        libClassList = saq.getLibraryClasses(CommonDefinition.ALWAYSPRODUCED, this.arch);\r
         if (libClassList != null) {\r
             libClassIncludeH = LibraryClassToAutogenH(libClassList);\r
             item = libClassIncludeH.iterator();\r
@@ -351,10 +363,10 @@ public class AutoGen {
         //  If is TianoR8FlashMap, copy {Fv_DIR}/FlashMap.h to\r
         // {DEST_DIR_DRBUG}/FlashMap.h\r
         //\r
-        if (SurfaceAreaQuery.isHaveTianoR8FlashMap()) {\r
-            fileBuffer.append(CommonDefinition.include);\r
+        if (saq.isHaveTianoR8FlashMap()) {\r
+            fileBuffer.append(CommonDefinition.INCLUDE);\r
             fileBuffer.append("  <");\r
-            fileBuffer.append(CommonDefinition.tianoR8FlashMapH + ">\r\n");\r
+            fileBuffer.append(CommonDefinition.TIANOR8PLASHMAPH + ">\r\n");\r
             copyFlashMapHToDebugDir();\r
         }\r
 \r
@@ -362,7 +374,7 @@ public class AutoGen {
         //\r
         if (this.myPcdAutogen != null) {\r
             fileBuffer.append("\r\n");\r
-            fileBuffer.append(this.myPcdAutogen.OutputH());\r
+            fileBuffer.append(this.myPcdAutogen.getHAutoGenString());\r
         }\r
 \r
         //\r
@@ -374,30 +386,30 @@ public class AutoGen {
         // Save string buffer content in AutoGen.h.\r
         //\r
         if (!saveFile(outputPath + File.separatorChar + "AutoGen.h", fileBuffer)) {\r
-            throw new BuildException("Failed to generate AutoGen.h !!!");\r
+            throw new AutoGenException("Failed to generate AutoGen.h !!!");\r
         }\r
     }\r
 \r
     /**\r
-     * moduleGenAutogenC\r
-     *\r
-     * This function generates AutoGen.c for module.\r
-     *\r
-     * @throws BuildException\r
-     *             Failed to generate AutoGen.c.\r
-     */\r
-    void moduleGenAutogenC() throws AutoGenException {\r
+      moduleGenAutogenC\r
+     \r
+      This function generates AutoGen.c for module.\r
+     \r
+      @throws BuildException\r
+                  Failed to generate AutoGen.c.\r
+    **/\r
+    void moduleGenAutogenC() throws EdkException {\r
 \r
         StringBuffer fileBuffer = new StringBuffer(8192);\r
         //\r
         // Write Autogen.c header notation\r
         //\r
-        fileBuffer.append(CommonDefinition.autogenCNotation);\r
+        fileBuffer.append(CommonDefinition.AUTOGENCNOTATION);\r
 \r
         //\r
         // Write #include <AutoGen.h> at beginning of AutoGen.c\r
         //\r
-        fileBuffer.append(CommonDefinition.includeAutogenH);\r
+        fileBuffer.append(CommonDefinition.INCLUDEAUTOGENH);\r
 \r
         //\r
         // Get the native MSA file infomation. Since before call autogen,\r
@@ -405,7 +417,7 @@ public class AutoGen {
         // process <Externs> it should be set the DOC as the Native MSA info.\r
         //\r
         Map<String, XmlObject> doc = GlobalData.getNativeMsa(this.moduleId);\r
-        SurfaceAreaQuery.push(doc);\r
+        saq.push(doc);\r
         //\r
         // Write <Extern>\r
         // DriverBinding/ComponentName/DriverConfiguration/DriverDialog\r
@@ -423,21 +435,23 @@ public class AutoGen {
         //\r
         // Write EntryPoint to autgoGen.c\r
         //\r
-        String[] entryPointList = SurfaceAreaQuery.getModuleEntryPointArray();\r
-        EntryPointToAutoGen(CommonDefinition.remDupString(entryPointList), fileBuffer);\r
+        String[] entryPointList = saq.getModuleEntryPointArray();\r
+               String[] unloadImageList = saq.getModuleUnloadImageArray();\r
+        EntryPointToAutoGen(CommonDefinition.remDupString(entryPointList), \r
+                                       CommonDefinition.remDupString(unloadImageList),\r
+                                       fileBuffer);\r
 \r
-        pcdDriverType = SurfaceAreaQuery.getPcdDriverType();\r
+        pcdDriverType = saq.getPcdDriverType();\r
 \r
         //\r
         // Restore the DOC which include the FPD module info.\r
         //\r
-        SurfaceAreaQuery.pop();\r
+        saq.pop();\r
 \r
         //\r
         // Write Guid to autogen.c\r
         //\r
-        String guid = CommonDefinition.formatGuidName(SurfaceAreaQuery\r
-                                                      .getModuleGuid());\r
+        String guid = CommonDefinition.formatGuidName(saq.getModuleGuid());\r
 \r
         fileBuffer\r
         .append("GLOBAL_REMOVE_IF_UNREFERENCED EFI_GUID gEfiCallerIdGuid = {");\r
@@ -465,7 +479,7 @@ public class AutoGen {
         //\r
         // Get module dependent Package identification.\r
         //\r
-        PackageIdentification[] packages = SurfaceAreaQuery.getDependencePkg(this.arch);\r
+        PackageIdentification[] packages = saq.getDependencePkg(this.arch);\r
         for (int i = 0; i < packages.length; i++) {\r
             if (!this.mDepPkgList.contains(packages[i])) {\r
                 this.mDepPkgList.add(packages[i]);\r
@@ -487,33 +501,30 @@ public class AutoGen {
                                                  arch, \r
                                                  false, \r
                                                  null,\r
-                                                 pcdDriverType);\r
-        try {\r
-            this.myPcdAutogen.execute();\r
-        } catch (Exception exp) {\r
-            throw new PcdAutogenException (exp.getMessage());\r
-        }\r
-\r
+                                                 pcdDriverType, \r
+                                                 parentId);\r
+        \r
+        this.myPcdAutogen.execute();\r
         if (this.myPcdAutogen != null) {\r
             fileBuffer.append("\r\n");\r
-            fileBuffer.append(this.myPcdAutogen.OutputC());\r
+            fileBuffer.append(this.myPcdAutogen.getCAutoGenString());\r
         }\r
 \r
         if (!saveFile(outputPath + File.separatorChar + "AutoGen.c", fileBuffer)) {\r
-            throw new BuildException("Failed to generate AutoGen.c !!!");\r
+            throw new AutoGenException("Failed to generate AutoGen.c !!!");\r
         }\r
 \r
     }\r
 \r
     /**\r
-     * libGenAutogenH\r
-     *\r
-     * This function generates AutoGen.h for library.\r
-     *\r
-     * @throws BuildException\r
-     *             Failed to generate AutoGen.c.\r
-     */\r
-    void libGenAutogenH() throws AutoGenException {\r
+      libGenAutogenH\r
+     \r
+      This function generates AutoGen.h for library.\r
+     \r
+      @throws BuildException\r
+                  Failed to generate AutoGen.c.\r
+    **/\r
+    void libGenAutogenH() throws EdkException {\r
 \r
         Set<String> libClassIncludeH;\r
         String moduleType;\r
@@ -524,14 +535,21 @@ public class AutoGen {
         //\r
         // Write Autogen.h header notation\r
         //\r
-        fileBuffer.append(CommonDefinition.autogenHNotation);\r
+        fileBuffer.append(CommonDefinition.AUTOGENHNOTATION);\r
 \r
         //\r
         // Add #ifndef ${BaseName}_AUTOGENH\r
         // #def ${BseeName}_AUTOGENH\r
         //\r
-        fileBuffer.append("#ifndef    " + "_AUTOGENH_" + this.moduleId.getGuid().replaceAll("-", "_") + "\r\n");\r
-        fileBuffer.append("#define    " + "_AUTOGENH_" + this.moduleId.getGuid().replaceAll("-", "_") + "\r\n\r\n");\r
+        fileBuffer.append(CommonDefinition.IFNDEF \r
+                                     + CommonDefinition.AUTOGENH\r
+                                     + this.moduleId.getGuid().replaceAll("-", "_") \r
+                                     + ToolDefinitions.LINE_SEPARATOR);\r
+        fileBuffer.append(CommonDefinition.DEFINE \r
+                                     + CommonDefinition.AUTOGENH\r
+                                     + this.moduleId.getGuid().replaceAll("-", "_") \r
+                                     + ToolDefinitions.LINE_SEPARATOR \r
+                                     + ToolDefinitions.LINE_SEPARATOR);\r
 \r
         //\r
         // Write EFI_SPECIFICATION_VERSION and EDK_RELEASE_VERSION\r
@@ -539,10 +557,10 @@ public class AutoGen {
         // Note: the specification version and release version should\r
         // be get from module surface area instead of hard code.\r
         //\r
-        fileBuffer.append(CommonDefinition.autoGenHbegin);\r
-        String[] specList = SurfaceAreaQuery.getExternSpecificaiton();\r
+        fileBuffer.append(CommonDefinition.AUTOGENHBEGIN);\r
+        String[] specList = saq.getExternSpecificaiton();\r
         for (int i = 0; i < specList.length; i++) {\r
-            fileBuffer.append(CommonDefinition.marcDefineStr + specList[i]\r
+            fileBuffer.append(CommonDefinition.DEFINE + specList[i]\r
                               + "\r\n");\r
         }\r
         // fileBuffer.append(CommonDefinition.autoGenHLine1);\r
@@ -551,8 +569,8 @@ public class AutoGen {
         //\r
         // Write consumed package's mdouleInfo related *.h file to autogen.h.\r
         //\r
-        moduleType = SurfaceAreaQuery.getModuleType();\r
-        PackageIdentification[] cosumedPkglist = SurfaceAreaQuery\r
+        moduleType = saq.getModuleType();\r
+        PackageIdentification[] cosumedPkglist = saq\r
                                                  .getDependencePkg(this.arch);\r
         headerFileList = depPkgToAutogenH(cosumedPkglist, moduleType);\r
         item = headerFileList.iterator();\r
@@ -562,8 +580,8 @@ public class AutoGen {
         //\r
         // Write library class's related *.h file to autogen.h\r
         //\r
-        String[] libClassList = SurfaceAreaQuery\r
-                                .getLibraryClasses(CommonDefinition.AlwaysConsumed, this.arch);\r
+        String[] libClassList = saq\r
+                                .getLibraryClasses(CommonDefinition.ALWAYSCONSUMED, this.arch);\r
         if (libClassList != null) {\r
             libClassIncludeH = LibraryClassToAutogenH(libClassList);\r
             item = libClassIncludeH.iterator();\r
@@ -572,8 +590,8 @@ public class AutoGen {
             }\r
         }\r
 \r
-        libClassList = SurfaceAreaQuery\r
-                       .getLibraryClasses(CommonDefinition.AlwaysProduced, this.arch);\r
+        libClassList = saq\r
+                       .getLibraryClasses(CommonDefinition.ALWAYSPRODUCED, this.arch);\r
         if (libClassList != null) {\r
             libClassIncludeH = LibraryClassToAutogenH(libClassList);\r
             item = libClassIncludeH.iterator();\r
@@ -581,16 +599,16 @@ public class AutoGen {
                 fileBuffer.append(item.next().toString());\r
             }\r
         }\r
-        fileBuffer.append("\r\n");\r
+        fileBuffer.append(ToolDefinitions.LINE_SEPARATOR);\r
 \r
         //\r
         //  If is TianoR8FlashMap, copy {Fv_DIR}/FlashMap.h to\r
         // {DEST_DIR_DRBUG}/FlashMap.h\r
         //\r
-        if (SurfaceAreaQuery.isHaveTianoR8FlashMap()) {\r
-            fileBuffer.append(CommonDefinition.include);\r
+        if (saq.isHaveTianoR8FlashMap()) {\r
+            fileBuffer.append(CommonDefinition.INCLUDE);\r
             fileBuffer.append("  <");\r
-            fileBuffer.append(CommonDefinition.tianoR8FlashMapH + ">\r\n");\r
+            fileBuffer.append(CommonDefinition.TIANOR8PLASHMAPH + ">\r\n");\r
             copyFlashMapHToDebugDir();\r
         }\r
 \r
@@ -599,7 +617,7 @@ public class AutoGen {
         //\r
         if (this.myPcdAutogen != null) {\r
             fileBuffer.append("\r\n");\r
-            fileBuffer.append(this.myPcdAutogen.OutputH());\r
+            fileBuffer.append(this.myPcdAutogen.getHAutoGenString());\r
         }\r
 \r
         //\r
@@ -611,28 +629,28 @@ public class AutoGen {
         // Save content of string buffer to AutoGen.h file.\r
         //\r
         if (!saveFile(outputPath + File.separatorChar + "AutoGen.h", fileBuffer)) {\r
-            throw new BuildException("Failed to generate AutoGen.h !!!");\r
+            throw new AutoGenException("Failed to generate AutoGen.h !!!");\r
         }\r
     }\r
 \r
     /**\r
-     * libGenAutogenC\r
-     *\r
-     * This function generates AutoGen.h for library.\r
-     *\r
-     * @throws BuildException\r
-     *             Failed to generate AutoGen.c.\r
-     */\r
-    void libGenAutogenC() throws BuildException, PcdAutogenException {\r
+      libGenAutogenC\r
+     \r
+      This function generates AutoGen.h for library.\r
+     \r
+      @throws BuildException\r
+                  Failed to generate AutoGen.c.\r
+    **/\r
+    void libGenAutogenC() throws EdkException {\r
         StringBuffer fileBuffer = new StringBuffer(10240);\r
 \r
         //\r
         // Write Autogen.c header notation\r
         //\r
-        fileBuffer.append(CommonDefinition.autogenCNotation);\r
+        fileBuffer.append(CommonDefinition.AUTOGENCNOTATION);\r
 \r
-        fileBuffer.append(CommonDefinition.autoGenCLine1);\r
-        fileBuffer.append("\r\n");\r
+        fileBuffer.append(ToolDefinitions.LINE_SEPARATOR);\r
+        fileBuffer.append(ToolDefinitions.LINE_SEPARATOR);\r
 \r
         //\r
         // Call pcd autogen.\r
@@ -640,78 +658,73 @@ public class AutoGen {
         this.myPcdAutogen = new PCDAutoGenAction(moduleId,\r
                                                  arch,\r
                                                  true,\r
-                                                 SurfaceAreaQuery.getModulePcdEntryNameArray(),\r
-                                                 pcdDriverType);\r
-        try {\r
-            this.myPcdAutogen.execute();\r
-        } catch (Exception e) {\r
-            throw new PcdAutogenException(e.getMessage());\r
-        }\r
-\r
+                                                 saq.getModulePcdEntryNameArray(),\r
+                                                 pcdDriverType, \r
+                                                 parentId);\r
+        this.myPcdAutogen.execute();\r
         if (this.myPcdAutogen != null) {\r
-            fileBuffer.append("\r\n");\r
-            fileBuffer.append(this.myPcdAutogen.OutputC());\r
+            fileBuffer.append(ToolDefinitions.LINE_SEPARATOR);\r
+            fileBuffer.append(this.myPcdAutogen.getCAutoGenString());\r
         }\r
 \r
         if (!saveFile(outputPath + File.separatorChar + "AutoGen.c", fileBuffer)) {\r
-            throw new BuildException("Failed to generate AutoGen.c !!!");\r
+            throw new AutoGenException("Failed to generate AutoGen.c !!!");\r
         }\r
     }\r
 \r
     /**\r
-     * LibraryClassToAutogenH\r
-     *\r
-     * This function returns *.h files declared by library classes which are\r
-     * consumed or produced by current build module or library.\r
-     *\r
-     * @param libClassList\r
-     *            List of library class which consumed or produce by current\r
-     *            build module or library.\r
-     * @return includeStrList List of *.h file.\r
-     */\r
+      LibraryClassToAutogenH\r
+     \r
+      This function returns *.h files declared by library classes which are\r
+      consumed or produced by current build module or library.\r
+     \r
+      @param libClassList\r
+                 List of library class which consumed or produce by current\r
+                 build module or library.\r
+      @return includeStrList List of *.h file.\r
+    **/\r
     Set<String> LibraryClassToAutogenH(String[] libClassList)\r
-    throws AutoGenException {\r
-        Set<String> includStrList = new LinkedHashSet<String>();\r
-        String includerName[];\r
+    throws EdkException {\r
+        Set<String> includeStrList = new LinkedHashSet<String>();\r
+        String includeName[];\r
         String str = "";\r
 \r
         //\r
         // Get include file from GlobalData's SPDTable according to\r
         // library class name.\r
         //\r
-\r
         for (int i = 0; i < libClassList.length; i++) {\r
-            includerName = GlobalData.getLibraryClassHeaderFiles(\r
-                                                                SurfaceAreaQuery.getDependencePkg(this.arch),\r
+            includeName = GlobalData.getLibraryClassHeaderFiles(\r
+                            saq.getDependencePkg(this.arch),\r
                                                                 libClassList[i]);\r
-            if (includerName == null) {\r
+            if (includeName == null) {\r
                 throw new AutoGenException("Can not find library class ["\r
                                            + libClassList[i] + "] declaration in any SPD package. ");\r
             }\r
-            for (int j = 0; j < includerName.length; j++) {\r
-                String includeNameStr = includerName[j];\r
+            for (int j = 0; j < includeName.length; j++) {\r
+                String includeNameStr = includeName[j];\r
                 if (includeNameStr != null) {\r
-                    str = CommonDefinition.include + " " + "<";\r
+                    str = CommonDefinition.INCLUDE + " " + "<";\r
                     str = str + includeNameStr + ">\r\n";\r
-                    includStrList.add(str);\r
+                    includeStrList.add(str);\r
                     includeNameStr = null;\r
                 }\r
             }\r
         }\r
-        return includStrList;\r
+        return includeStrList;\r
     }\r
 \r
     /**\r
-     * IncludesToAutogenH\r
-     *\r
-     * This function add include file in AutoGen.h file.\r
-     *\r
-     * @param packageNameList\r
-     *            List of module depended package.\r
-     * @param moduleType\r
-     *            Module type.\r
-     * @return\r
-     */\r
+      IncludesToAutogenH\r
+     \r
+      This function add include file in AutoGen.h file.\r
+     \r
+      @param packageNameList\r
+                 List of module depended package.\r
+      @param moduleType\r
+                 Module type.\r
+      @return\r
+    **/\r
     List<String> depPkgToAutogenH(PackageIdentification[] packageNameList,\r
                                   String moduleType) throws AutoGenException {\r
 \r
@@ -730,7 +743,7 @@ public class AutoGen {
                                            + packageNameList[i]\r
                                            + "] declaration in any SPD package. ");\r
             } else if (!pkgHeader.equalsIgnoreCase("")) {\r
-                includeStr = CommonDefinition.include + " <" + pkgHeader\r
+                includeStr = CommonDefinition.INCLUDE + " <" + pkgHeader\r
                              + ">\r\n";\r
                 includeStrList.add(includeStr);\r
             }\r
@@ -740,21 +753,23 @@ public class AutoGen {
     }\r
 \r
     /**\r
-     * EntryPointToAutoGen\r
-     *\r
-     * This function convert <ModuleEntryPoint> & <ModuleUnloadImage>\r
-     * information in mas to AutoGen.c\r
-     *\r
-     * @param entryPointList\r
-     *            List of entry point.\r
-     * @param fileBuffer\r
-     *            String buffer fo AutoGen.c.\r
-     * @throws Exception\r
-     */\r
-    void EntryPointToAutoGen(String[] entryPointList, StringBuffer fileBuffer)\r
-    throws BuildException {\r
-\r
-        String typeStr = SurfaceAreaQuery.getModuleType();\r
+      EntryPointToAutoGen\r
+     \r
+      This function convert <ModuleEntryPoint> & <ModuleUnloadImage>\r
+      information in mas to AutoGen.c\r
+     \r
+      @param entryPointList\r
+                 List of entry point.\r
+      @param fileBuffer\r
+                 String buffer fo AutoGen.c.\r
+      @throws Exception\r
+    **/\r
+    void EntryPointToAutoGen(String[] entryPointList, String[] unloadImageList, StringBuffer fileBuffer)\r
+    throws EdkException {\r
+\r
+        String typeStr = saq.getModuleType();\r
+               int unloadImageCount = 0;\r
+        int entryPointCount  = 0;\r
 \r
         //\r
         // The parameters and return value of entryPoint is difference\r
@@ -764,11 +779,10 @@ public class AutoGen {
         \r
         case CommonDefinition.ModuleTypePeiCore:\r
             if (entryPointList == null ||entryPointList.length != 1 ) {\r
-                throw new BuildException(\r
+                throw new AutoGenException(\r
                                         "Module type = 'PEI_CORE', can have only one module entry point!");\r
             } else {\r
                 fileBuffer.append("EFI_STATUS\r\n");\r
-                fileBuffer.append("EFIAPI\r\n");\r
                 fileBuffer.append(entryPointList[0]);\r
                 fileBuffer.append(" (\r\n");\r
                 fileBuffer\r
@@ -796,12 +810,11 @@ public class AutoGen {
         case CommonDefinition.ModuleTypeDxeCore:\r
             fileBuffer.append("const UINT32 _gUefiDriverRevision = 0;\r\n");\r
             if (entryPointList == null || entryPointList.length != 1) {\r
-                throw new BuildException(\r
+                throw new AutoGenException(\r
                                         "Module type = 'DXE_CORE', can have only one module entry point!");\r
             } else {\r
 \r
                 fileBuffer.append("VOID\r\n");\r
-                fileBuffer.append("EFIAPI\r\n");\r
                 fileBuffer.append(entryPointList[0]);\r
                 fileBuffer.append(" (\n");\r
                 fileBuffer.append("  IN VOID  *HobStart\r\n");\r
@@ -821,7 +834,7 @@ public class AutoGen {
             break;\r
 \r
         case CommonDefinition.ModuleTypePeim:\r
-            int entryPointCount = 0;\r
+            entryPointCount = 0;\r
             fileBuffer\r
             .append("GLOBAL_REMOVE_IF_UNREFERENCED const UINT32 _gPeimRevision = 0;\r\n");\r
             if (entryPointList == null || entryPointList.length == 0) {\r
@@ -838,7 +851,6 @@ public class AutoGen {
             }\r
             for (int i = 0; i < entryPointList.length; i++) {\r
                 fileBuffer.append("EFI_STATUS\r\n");\r
-                fileBuffer.append("EFIAPI\r\n");\r
                 fileBuffer.append(entryPointList[i]);\r
                 fileBuffer.append(" (\r\n");\r
                 fileBuffer\r
@@ -907,7 +919,6 @@ public class AutoGen {
             } else {\r
                 for (int i = 0; i < entryPointList.length; i++) {\r
                     fileBuffer.append("EFI_STATUS\r\n");\r
-                    fileBuffer.append("EFIAPI\r\n");\r
                     fileBuffer.append(entryPointList[i]);\r
                     fileBuffer.append(" (\r\n");\r
                     fileBuffer.append("  IN EFI_HANDLE        ImageHandle,\r\n");\r
@@ -966,26 +977,26 @@ public class AutoGen {
             //\r
             // Add "ModuleUnloadImage" for DxeSmmDriver module type;\r
             //\r
-            entryPointList = SurfaceAreaQuery.getModuleUnloadImageArray();\r
-            entryPointList = CommonDefinition.remDupString(entryPointList);\r
-            entryPointCount = 0;\r
+            //entryPointList = SurfaceAreaQuery.getModuleUnloadImageArray();\r
+            //entryPointList = CommonDefinition.remDupString(entryPointList);\r
+            //entryPointCount = 0;\r
 \r
-            if (entryPointList != null) {\r
-                for (int i = 0; i < entryPointList.length; i++) {\r
+                       unloadImageCount = 0;\r
+            if (unloadImageList != null) {\r
+                for (int i = 0; i < unloadImageList.length; i++) {\r
                     fileBuffer.append("EFI_STATUS\r\n");\r
-                    fileBuffer.append("EFIAPI\r\n");\r
-                    fileBuffer.append(entryPointList[i]);\r
+                    fileBuffer.append(unloadImageList[i]);\r
                     fileBuffer.append(" (\r\n");\r
                     fileBuffer\r
                     .append("  IN EFI_HANDLE        ImageHandle\r\n");\r
                     fileBuffer.append("  );\r\n");\r
-                    entryPointCount++;\r
+                    unloadImageCount++;\r
                 }\r
             }\r
 \r
             fileBuffer\r
             .append("GLOBAL_REMOVE_IF_UNREFERENCED const UINT8  _gDriverUnloadImageCount = ");\r
-            fileBuffer.append(Integer.toString(entryPointCount));\r
+            fileBuffer.append(Integer.toString(unloadImageCount));\r
             fileBuffer.append(";\r\n\r\n");\r
 \r
             fileBuffer.append("EFI_STATUS\r\n");\r
@@ -995,28 +1006,28 @@ public class AutoGen {
             fileBuffer.append("  )\r\n");\r
             fileBuffer.append("{\r\n");\r
 \r
-            if (entryPointCount == 0) {\r
+            if (unloadImageCount == 0) {\r
                 fileBuffer.append("  return EFI_SUCCESS;\r\n");\r
-            } else if (entryPointCount == 1) {\r
+            } else if (unloadImageCount == 1) {\r
                 fileBuffer.append("  return ");\r
-                fileBuffer.append(entryPointList[0]);\r
+                fileBuffer.append(unloadImageList[0]);\r
                 fileBuffer.append("(ImageHandle);\r\n");\r
             } else {\r
                 fileBuffer.append("  EFI_STATUS  Status;\r\n\r\n");\r
                 fileBuffer.append("  Status = EFI_SUCCESS;\r\n\r\n");\r
-                for (int i = 0; i < entryPointList.length; i++) {\r
+                for (int i = 0; i < unloadImageList.length; i++) {\r
                     if (i == 0) {\r
                         fileBuffer.append("     Status = ");\r
-                        fileBuffer.append(entryPointList[i]);\r
+                        fileBuffer.append(unloadImageList[i]);\r
                         fileBuffer.append("(ImageHandle);\r\n");\r
                     } else {\r
                         fileBuffer.append("  if (EFI_ERROR (Status)) {\r\n");\r
                         fileBuffer.append("    ");\r
-                        fileBuffer.append(entryPointList[i]);\r
+                        fileBuffer.append(unloadImageList[i]);\r
                         fileBuffer.append("(ImageHandle);\r\n");\r
                         fileBuffer.append("  } else {\r\n");\r
                         fileBuffer.append("    Status = ");\r
-                        fileBuffer.append(entryPointList[i]);\r
+                        fileBuffer.append(unloadImageList[i]);\r
                         fileBuffer.append("(ImageHandle);\r\n");\r
                         fileBuffer.append("  }\r\n");\r
                     }\r
@@ -1053,7 +1064,6 @@ public class AutoGen {
                 for (int i = 0; i < entryPointList.length; i++) {\r
 \r
                     fileBuffer.append("EFI_STATUS\r\n");\r
-                    fileBuffer.append("EFIAPI\r\n");\r
                     fileBuffer.append(entryPointList[i]);\r
                     fileBuffer.append(" (\r\n");\r
                     fileBuffer.append("  IN EFI_HANDLE        ImageHandle,\r\n");\r
@@ -1132,28 +1142,29 @@ public class AutoGen {
             //\r
             // Add ModuleUnloadImage for DxeDriver and UefiDriver module type.\r
             //\r
-            entryPointList = SurfaceAreaQuery.getModuleUnloadImageArray();\r
+            //entryPointList = SurfaceAreaQuery.getModuleUnloadImageArray();\r
             //\r
             // Remover duplicate unload entry point.\r
             //\r
-            entryPointList = CommonDefinition.remDupString(entryPointList);\r
-            entryPointCount = 0;\r
-            if (entryPointList != null) {\r
-                for (int i = 0; i < entryPointList.length; i++) {\r
+            //entryPointList = CommonDefinition.remDupString(entryPointList);\r
+            //entryPointCount = 0;\r
+                       unloadImageCount = 0;\r
+            if (unloadImageList != null) {\r
+                for (int i = 0; i < unloadImageList.length; i++) {\r
                     fileBuffer.append("EFI_STATUS\r\n");\r
                     fileBuffer.append("EFIAPI\r\n");\r
-                    fileBuffer.append(entryPointList[i]);\r
+                    fileBuffer.append(unloadImageList[i]);\r
                     fileBuffer.append(" (\r\n");\r
                     fileBuffer\r
                     .append("  IN EFI_HANDLE        ImageHandle\r\n");\r
                     fileBuffer.append("  );\r\n");\r
-                    entryPointCount++;\r
+                    unloadImageCount++;\r
                 }\r
             }\r
 \r
             fileBuffer\r
             .append("GLOBAL_REMOVE_IF_UNREFERENCED const UINT8  _gDriverUnloadImageCount = ");\r
-            fileBuffer.append(Integer.toString(entryPointCount));\r
+            fileBuffer.append(Integer.toString(unloadImageCount));\r
             fileBuffer.append(";\r\n\r\n");\r
 \r
             fileBuffer.append("EFI_STATUS\n");\r
@@ -1163,28 +1174,28 @@ public class AutoGen {
             fileBuffer.append("  )\r\n");\r
             fileBuffer.append("{\r\n");\r
 \r
-            if (entryPointCount == 0) {\r
+            if (unloadImageCount == 0) {\r
                 fileBuffer.append("  return EFI_SUCCESS;\r\n");\r
-            } else if (entryPointCount == 1) {\r
+            } else if (unloadImageCount == 1) {\r
                 fileBuffer.append("  return ");\r
-                fileBuffer.append(entryPointList[0]);\r
+                fileBuffer.append(unloadImageList[0]);\r
                 fileBuffer.append("(ImageHandle);\r\n");\r
             } else {\r
                 fileBuffer.append("  EFI_STATUS  Status;\r\n\r\n");\r
                 fileBuffer.append("  Status = EFI_SUCCESS;\r\n\r\n");\r
-                for (int i = 0; i < entryPointList.length; i++) {\r
+                for (int i = 0; i < unloadImageList.length; i++) {\r
                     if (i == 0) {\r
                         fileBuffer.append("  Status = ");\r
-                        fileBuffer.append(entryPointList[i]);\r
+                        fileBuffer.append(unloadImageList[i]);\r
                         fileBuffer.append("(ImageHandle);\r\n");\r
                     } else {\r
                         fileBuffer.append("  if (EFI_ERROR (Status)) {\r\n");\r
                         fileBuffer.append("    ");\r
-                        fileBuffer.append(entryPointList[i]);\r
+                        fileBuffer.append(unloadImageList[i]);\r
                         fileBuffer.append("(ImageHandle);\r\n");\r
                         fileBuffer.append("  } else {\r\n");\r
                         fileBuffer.append("    Status = ");\r
-                        fileBuffer.append(entryPointList[i]);\r
+                        fileBuffer.append(unloadImageList[i]);\r
                         fileBuffer.append("(ImageHandle);\r\n");\r
                         fileBuffer.append("  }\r\n");\r
                     }\r
@@ -1197,16 +1208,16 @@ public class AutoGen {
     }\r
 \r
     /**\r
-     * PpiGuidToAutogenc\r
-     *\r
-     * This function gets GUIDs from SPD file accrodeing to <PPIs> information\r
-     * and write those GUIDs to AutoGen.c.\r
-     *\r
-     * @param fileBuffer\r
-     *            String Buffer for Autogen.c file.\r
-     * @throws BuildException\r
-     *             Guid must set value!\r
-     */\r
+      PpiGuidToAutogenc\r
+     \r
+      This function gets GUIDs from SPD file accrodeing to <PPIs> information\r
+      and write those GUIDs to AutoGen.c.\r
+     \r
+      @param fileBuffer\r
+                 String Buffer for Autogen.c file.\r
+      @throws BuildException\r
+                  Guid must set value!\r
+    **/\r
     void PpiGuidToAutogenC(StringBuffer fileBuffer) throws AutoGenException {\r
         String[] cNameGuid = null;\r
 \r
@@ -1215,12 +1226,12 @@ public class AutoGen {
         // then add those PPI ,and PPI Notify name to list.\r
         //\r
 \r
-        String[] ppiList = SurfaceAreaQuery.getPpiArray(this.arch);\r
+        String[] ppiList = saq.getPpiArray(this.arch);\r
         for (int i = 0; i < ppiList.length; i++) {\r
             this.mPpiList.add(ppiList[i]);\r
         }\r
 \r
-        String[] ppiNotifyList = SurfaceAreaQuery.getPpiNotifyArray(this.arch);\r
+        String[] ppiNotifyList = saq.getPpiNotifyArray(this.arch);\r
         for (int i = 0; i < ppiNotifyList.length; i++) {\r
             this.mPpiList.add(ppiNotifyList[i]);\r
         }\r
@@ -1251,20 +1262,20 @@ public class AutoGen {
     }\r
 \r
     /**\r
-     * ProtocolGuidToAutogenc\r
-     *\r
-     * This function gets GUIDs from SPD file accrodeing to <Protocol>\r
-     * information and write those GUIDs to AutoGen.c.\r
-     *\r
-     * @param fileBuffer\r
-     *            String Buffer for Autogen.c file.\r
-     * @throws BuildException\r
-     *             Protocol name must set.\r
-     */\r
-    void ProtocolGuidToAutogenC(StringBuffer fileBuffer) throws BuildException {\r
+      ProtocolGuidToAutogenc\r
+     \r
+      This function gets GUIDs from SPD file accrodeing to <Protocol>\r
+      information and write those GUIDs to AutoGen.c.\r
+     \r
+      @param fileBuffer\r
+                 String Buffer for Autogen.c file.\r
+      @throws BuildException\r
+                  Protocol name must set.\r
+    **/\r
+    void ProtocolGuidToAutogenC(StringBuffer fileBuffer) throws EdkException {\r
         String[] cNameGuid = null;\r
 \r
-        String[] protocolList = SurfaceAreaQuery.getProtocolArray(this.arch);\r
+        String[] protocolList = saq.getProtocolArray(this.arch);\r
 \r
         //\r
         // Add result to Autogen global list.\r
@@ -1273,7 +1284,7 @@ public class AutoGen {
             this.mProtocolList.add(protocolList[i]);\r
         }\r
 \r
-        String[] protocolNotifyList = SurfaceAreaQuery\r
+        String[] protocolNotifyList = saq\r
                                       .getProtocolNotifyArray(this.arch);\r
 \r
         for (int i = 0; i < protocolNotifyList.length; i++) {\r
@@ -1301,27 +1312,27 @@ public class AutoGen {
                 //\r
                 // If can't find protocol GUID declaration in every package\r
                 //\r
-                throw new BuildException("Can not find protocol Guid ["\r
+                throw new AutoGenException("Can not find protocol Guid ["\r
                                          + protocolKeyWord + "] declaration in any SPD package!");\r
             }\r
         }\r
     }\r
 \r
     /**\r
-     * GuidGuidToAutogenc\r
-     *\r
-     * This function gets GUIDs from SPD file accrodeing to <Guids> information\r
-     * and write those GUIDs to AutoGen.c.\r
-     *\r
-     * @param fileBuffer\r
-     *            String Buffer for Autogen.c file.\r
-     *\r
-     */\r
+      GuidGuidToAutogenc\r
+     \r
+      This function gets GUIDs from SPD file accrodeing to <Guids> information\r
+      and write those GUIDs to AutoGen.c.\r
+     \r
+      @param fileBuffer\r
+                 String Buffer for Autogen.c file.\r
+     \r
+    **/\r
     void GuidGuidToAutogenC(StringBuffer fileBuffer) throws AutoGenException {\r
         String[] cNameGuid = null;\r
         String guidKeyWord = null;\r
 \r
-        String[] guidList = SurfaceAreaQuery.getGuidEntryArray(this.arch);\r
+        String[] guidList = saq.getGuidEntryArray(this.arch);\r
 \r
         for (int i = 0; i < guidList.length; i++) {\r
             this.mGuidList.add(guidList[i]);\r
@@ -1352,49 +1363,45 @@ public class AutoGen {
     }\r
 \r
     /**\r
-     * LibInstanceToAutogenC\r
-     *\r
-     * This function adds dependent library instance to autogen.c,which\r
-     * includeing library's constructor, destructor, and library dependent ppi,\r
-     * protocol, guid, pcd information.\r
-     *\r
-     * @param fileBuffer\r
-     *            String buffer for AutoGen.c\r
-     * @throws BuildException\r
-     */\r
-    void LibInstanceToAutogenC(StringBuffer fileBuffer) throws BuildException {\r
-        try {\r
-            String moduleType = this.moduleId.getModuleType();\r
-            //\r
-            // Add library constructor to AutoGen.c\r
-            //\r
-            LibConstructorToAutogenC(libConstructList, moduleType,\r
-                                     fileBuffer/* autogenC */);\r
-            //\r
-            // Add library destructor to AutoGen.c\r
-            //\r
-            LibDestructorToAutogenC(libDestructList, moduleType, fileBuffer/* autogenC */);\r
-        } catch (Exception e) {\r
-            throw new BuildException(e.getMessage());\r
-        }\r
+      LibInstanceToAutogenC\r
+     \r
+      This function adds dependent library instance to autogen.c,which\r
+      includeing library's constructor, destructor, and library dependent ppi,\r
+      protocol, guid, pcd information.\r
+     \r
+      @param fileBuffer\r
+                 String buffer for AutoGen.c\r
+      @throws BuildException\r
+    **/\r
+    void LibInstanceToAutogenC(StringBuffer fileBuffer) throws EdkException {\r
+               String moduleType = this.moduleId.getModuleType();\r
+               //\r
+               // Add library constructor to AutoGen.c\r
+               //\r
+               LibConstructorToAutogenC(libConstructList, moduleType,\r
+                                                                fileBuffer/* autogenC */);\r
+               //\r
+               // Add library destructor to AutoGen.c\r
+               //\r
+               LibDestructorToAutogenC(libDestructList, moduleType, fileBuffer/* autogenC */);\r
     }\r
 \r
     /**\r
-     * LibConstructorToAutogenc\r
-     *\r
-     * This function writes library constructor list to AutoGen.c. The library\r
-     * constructor's parameter and return value depend on module type.\r
-     *\r
-     * @param libInstanceList\r
-     *            List of library construct name.\r
-     * @param moduleType\r
-     *            Module type.\r
-     * @param fileBuffer\r
-     *            String buffer for AutoGen.c\r
-     * @throws Exception\r
-     */\r
-    void LibConstructorToAutogenC(List<String> libInstanceList,\r
-                                  String moduleType, StringBuffer fileBuffer) throws Exception {\r
+      LibConstructorToAutogenc\r
+     \r
+      This function writes library constructor list to AutoGen.c. The library\r
+      constructor's parameter and return value depend on module type.\r
+     \r
+      @param libInstanceList\r
+                 List of library construct name.\r
+      @param moduleType\r
+                 Module type.\r
+      @param fileBuffer\r
+                 String buffer for AutoGen.c\r
+      @throws Exception\r
+    **/\r
+    void LibConstructorToAutogenC(List<String[]> libInstanceList,\r
+                                  String moduleType, StringBuffer fileBuffer) throws EdkException {\r
         boolean isFirst = true;\r
 \r
         //\r
@@ -1402,44 +1409,54 @@ public class AutoGen {
         // module type.\r
         //\r
         for (int i = 0; i < libInstanceList.size(); i++) {\r
-            switch (CommonDefinition.getModuleType(moduleType)) {\r
-            case CommonDefinition.ModuleTypeBase:\r
-                fileBuffer.append("RETURN_STATUS\r\n");\r
+                       if (libInstanceList.get(i)[1].equalsIgnoreCase(EdkDefinitions.MODULE_TYPE_BASE)) {\r
+                               fileBuffer.append("RETURN_STATUS\r\n");\r
                 fileBuffer.append("EFIAPI\r\n");\r
-                fileBuffer.append(libInstanceList.get(i));\r
+                fileBuffer.append(libInstanceList.get(i)[0]);\r
                 fileBuffer.append(" (\r\n");\r
                 fileBuffer.append("  VOID\r\n");\r
                 fileBuffer.append("  );\r\n");\r
-                break;\r
-\r
-            case CommonDefinition.ModuleTypePeiCore:\r
-            case CommonDefinition.ModuleTypePeim:\r
-                fileBuffer.append("EFI_STATUS\r\n");\r
-                fileBuffer.append("EFIAPI\r\n");\r
-                fileBuffer.append(libInstanceList.get(i));\r
-                fileBuffer.append(" (\r\n");\r
-                fileBuffer\r
-                .append("  IN EFI_FFS_FILE_HEADER       *FfsHeader,\r\n");\r
-                fileBuffer\r
-                .append("  IN EFI_PEI_SERVICES          **PeiServices\r\n");\r
-                fileBuffer.append("  );\r\n");\r
-                break;\r
+                       } else {\r
+                               switch (CommonDefinition.getModuleType(moduleType)) {\r
+                case CommonDefinition.ModuleTypeBase:\r
+                    fileBuffer.append("RETURN_STATUS\r\n");\r
+                    fileBuffer.append("EFIAPI\r\n");\r
+                    fileBuffer.append(libInstanceList.get(i)[0]);\r
+                    fileBuffer.append(" (\r\n");\r
+                    fileBuffer.append("  VOID\r\n");\r
+                    fileBuffer.append("  );\r\n");\r
+                    break;\r
 \r
-            case CommonDefinition.ModuleTypeDxeCore:\r
-            case CommonDefinition.ModuleTypeDxeDriver:\r
-            case CommonDefinition.ModuleTypeDxeRuntimeDriver:\r
-            case CommonDefinition.ModuleTypeDxeSmmDriver:\r
-            case CommonDefinition.ModuleTypeDxeSalDriver:\r
-            case CommonDefinition.ModuleTypeUefiDriver:\r
-            case CommonDefinition.ModuleTypeUefiApplication:\r
-                fileBuffer.append("EFI_STATUS\r\n");\r
-                fileBuffer.append("EFIAPI\r\n");\r
-                fileBuffer.append(libInstanceList.get(i));\r
-                fileBuffer.append(" (\r\n");\r
-                fileBuffer.append("  IN EFI_HANDLE        ImageHandle,\r\n");\r
-                fileBuffer.append("  IN EFI_SYSTEM_TABLE  *SystemTable\r\n");\r
-                fileBuffer.append("  );\r\n");\r
-                break;\r
+                case CommonDefinition.ModuleTypePeiCore:\r
+                case CommonDefinition.ModuleTypePeim:\r
+                                       fileBuffer.append("EFI_STATUS\r\n");\r
+                                       fileBuffer.append("EFIAPI\r\n");\r
+                                       fileBuffer.append(libInstanceList.get(i)[0]);\r
+                                       fileBuffer.append(" (\r\n");\r
+                                       fileBuffer\r
+                                       .append("  IN EFI_FFS_FILE_HEADER       *FfsHeader,\r\n");\r
+                                       fileBuffer\r
+                                       .append("  IN EFI_PEI_SERVICES          **PeiServices\r\n");\r
+                                       fileBuffer.append("  );\r\n");\r
+                                       break;\r
+       \r
+                               case CommonDefinition.ModuleTypeDxeCore:\r
+                               case CommonDefinition.ModuleTypeDxeDriver:\r
+                               case CommonDefinition.ModuleTypeDxeRuntimeDriver:\r
+                               case CommonDefinition.ModuleTypeDxeSmmDriver:\r
+                               case CommonDefinition.ModuleTypeDxeSalDriver:\r
+                               case CommonDefinition.ModuleTypeUefiDriver:\r
+                               case CommonDefinition.ModuleTypeUefiApplication:\r
+                                       fileBuffer.append("EFI_STATUS\r\n");\r
+                                       fileBuffer.append("EFIAPI\r\n");\r
+                                       fileBuffer.append(libInstanceList.get(i)[0]);\r
+                                       fileBuffer.append(" (\r\n");\r
+                                       fileBuffer.append("  IN EFI_HANDLE        ImageHandle,\r\n");\r
+                                       fileBuffer.append("  IN EFI_SYSTEM_TABLE  *SystemTable\r\n");\r
+                                       fileBuffer.append("  );\r\n");\r
+                                       break;\r
+\r
+                           }\r
             }\r
         }\r
 \r
@@ -1488,32 +1505,40 @@ public class AutoGen {
                 fileBuffer.append("\r\n");\r
                 isFirst = false;\r
             }\r
-            switch (CommonDefinition.getModuleType(moduleType)) {\r
-            case CommonDefinition.ModuleTypeBase:\r
+                       if (libInstanceList.get(i)[1].equalsIgnoreCase(EdkDefinitions.MODULE_TYPE_BASE)) {\r
                 fileBuffer.append("  Status = ");\r
-                fileBuffer.append(libInstanceList.get(i));\r
+                fileBuffer.append(libInstanceList.get(i)[0]);\r
                 fileBuffer.append("();\r\n");\r
                 fileBuffer.append("  VOID\r\n");\r
-                break;\r
-            case CommonDefinition.ModuleTypePeiCore:\r
-            case CommonDefinition.ModuleTypePeim:\r
-                fileBuffer.append("  Status = ");\r
-                fileBuffer.append(libInstanceList.get(i));\r
-                fileBuffer.append(" (FfsHeader, PeiServices);\r\n");\r
-                break;\r
-            case CommonDefinition.ModuleTypeDxeCore:\r
-            case CommonDefinition.ModuleTypeDxeDriver:\r
-            case CommonDefinition.ModuleTypeDxeRuntimeDriver:\r
-            case CommonDefinition.ModuleTypeDxeSmmDriver:\r
-            case CommonDefinition.ModuleTypeDxeSalDriver:\r
-            case CommonDefinition.ModuleTypeUefiDriver:\r
-            case CommonDefinition.ModuleTypeUefiApplication:\r
-                fileBuffer.append("  Status = ");\r
-                fileBuffer.append(libInstanceList.get(i));\r
-                fileBuffer.append(" (ImageHandle, SystemTable);\r\n");\r
-                break;\r
-            default:\r
-                EdkLog.log(EdkLog.EDK_INFO,"Autogen doesn't know how to deal with module type - " + moduleType + "!");\r
+                       } else {\r
+                               switch (CommonDefinition.getModuleType(moduleType)) {\r
+                                       case CommonDefinition.ModuleTypeBase:\r
+                                               fileBuffer.append("  Status = ");\r
+                                               fileBuffer.append(libInstanceList.get(i)[0]);\r
+                                               fileBuffer.append("();\r\n");\r
+                                               fileBuffer.append("  VOID\r\n");\r
+                                               break;\r
+                                       case CommonDefinition.ModuleTypePeiCore:\r
+                                       case CommonDefinition.ModuleTypePeim:\r
+                                               fileBuffer.append("  Status = ");\r
+                                               fileBuffer.append(libInstanceList.get(i)[0]);\r
+                                               fileBuffer.append(" (FfsHeader, PeiServices);\r\n");\r
+                                               break;\r
+                                       case CommonDefinition.ModuleTypeDxeCore:\r
+                                       case CommonDefinition.ModuleTypeDxeDriver:\r
+                                       case CommonDefinition.ModuleTypeDxeRuntimeDriver:\r
+                                       case CommonDefinition.ModuleTypeDxeSmmDriver:\r
+                                       case CommonDefinition.ModuleTypeDxeSalDriver:\r
+                                       case CommonDefinition.ModuleTypeUefiDriver:\r
+                                       case CommonDefinition.ModuleTypeUefiApplication:\r
+                                               fileBuffer.append("  Status = ");\r
+                                               fileBuffer.append(libInstanceList.get(i)[0]);\r
+                                               fileBuffer.append(" (ImageHandle, SystemTable);\r\n");\r
+                                               break;\r
+                                       default:\r
+                                               EdkLog.log(EdkLog.EDK_INFO,"Autogen doesn't know how to deal with module type - " + moduleType + "!");\r
+                          }\r
+            \r
             }\r
             fileBuffer.append("  ASSERT_EFI_ERROR (Status);\r\n");\r
         }\r
@@ -1521,59 +1546,68 @@ public class AutoGen {
     }\r
 \r
     /**\r
-     * LibDestructorToAutogenc\r
-     *\r
-     * This function writes library destructor list to AutoGen.c. The library\r
-     * destructor's parameter and return value depend on module type.\r
-     *\r
-     * @param libInstanceList\r
-     *            List of library destructor name.\r
-     * @param moduleType\r
-     *            Module type.\r
-     * @param fileBuffer\r
-     *            String buffer for AutoGen.c\r
-     * @throws Exception\r
-     */\r
-    void LibDestructorToAutogenC(List<String> libInstanceList,\r
-                                 String moduleType, StringBuffer fileBuffer) throws Exception {\r
+      LibDestructorToAutogenc\r
+     \r
+      This function writes library destructor list to AutoGen.c. The library\r
+      destructor's parameter and return value depend on module type.\r
+     \r
+      @param libInstanceList\r
+                 List of library destructor name.\r
+      @param moduleType\r
+                 Module type.\r
+      @param fileBuffer\r
+                 String buffer for AutoGen.c\r
+      @throws Exception\r
+    **/\r
+    void LibDestructorToAutogenC(List<String[]> libInstanceList,\r
+                                 String moduleType, StringBuffer fileBuffer) throws EdkException {\r
         boolean isFirst = true;\r
         for (int i = 0; i < libInstanceList.size(); i++) {\r
-            switch (CommonDefinition.getModuleType(moduleType)) {\r
-            case CommonDefinition.ModuleTypeBase:\r
+                       if (libInstanceList.get(i)[1].equalsIgnoreCase(EdkDefinitions.MODULE_TYPE_BASE)) {\r
                 fileBuffer.append("RETURN_STATUS\r\n");\r
                 fileBuffer.append("EFIAPI\r\n");\r
-                fileBuffer.append(libInstanceList.get(i));\r
+                fileBuffer.append(libInstanceList.get(i)[0]);\r
                 fileBuffer.append(" (\r\n");\r
                 fileBuffer.append("  VOID\r\n");\r
                 fileBuffer.append("  );\r\n");\r
-                break;\r
-            case CommonDefinition.ModuleTypePeiCore:\r
-            case CommonDefinition.ModuleTypePeim:\r
-                fileBuffer.append("EFI_STATUS\r\n");\r
-                fileBuffer.append("EFIAPI\r\n");\r
-                fileBuffer.append(libInstanceList.get(i));\r
-                fileBuffer.append(" (\r\n");\r
-                fileBuffer\r
-                .append("  IN EFI_FFS_FILE_HEADER       *FfsHeader,\r\n");\r
-                fileBuffer\r
-                .append("  IN EFI_PEI_SERVICES          **PeiServices\r\n");\r
-                fileBuffer.append("  );\r\n");\r
-                break;\r
-            case CommonDefinition.ModuleTypeDxeCore:\r
-            case CommonDefinition.ModuleTypeDxeDriver:\r
-            case CommonDefinition.ModuleTypeDxeRuntimeDriver:\r
-            case CommonDefinition.ModuleTypeDxeSmmDriver:\r
-            case CommonDefinition.ModuleTypeDxeSalDriver:\r
-            case CommonDefinition.ModuleTypeUefiDriver:\r
-            case CommonDefinition.ModuleTypeUefiApplication:\r
-                fileBuffer.append("EFI_STATUS\r\n");\r
-                fileBuffer.append("EFIAPI\r\n");\r
-                fileBuffer.append(libInstanceList.get(i));\r
-                fileBuffer.append(" (\r\n");\r
-                fileBuffer.append("  IN EFI_HANDLE        ImageHandle,\r\n");\r
-                fileBuffer.append("  IN EFI_SYSTEM_TABLE  *SystemTable\r\n");\r
-                fileBuffer.append("  );\r\n");\r
-                break;\r
+                       } else {\r
+                               switch (CommonDefinition.getModuleType(moduleType)) {\r
+                                       case CommonDefinition.ModuleTypeBase:\r
+                                               fileBuffer.append("RETURN_STATUS\r\n");\r
+                                               fileBuffer.append("EFIAPI\r\n");\r
+                                               fileBuffer.append(libInstanceList.get(i)[0]);\r
+                                               fileBuffer.append(" (\r\n");\r
+                                               fileBuffer.append("  VOID\r\n");\r
+                                               fileBuffer.append("  );\r\n");\r
+                                               break;\r
+                                       case CommonDefinition.ModuleTypePeiCore:\r
+                                       case CommonDefinition.ModuleTypePeim:\r
+                                               fileBuffer.append("EFI_STATUS\r\n");\r
+                                               fileBuffer.append("EFIAPI\r\n");\r
+                                               fileBuffer.append(libInstanceList.get(i)[0]);\r
+                                               fileBuffer.append(" (\r\n");\r
+                                               fileBuffer\r
+                                               .append("  IN EFI_FFS_FILE_HEADER       *FfsHeader,\r\n");\r
+                                               fileBuffer\r
+                                               .append("  IN EFI_PEI_SERVICES          **PeiServices\r\n");\r
+                                               fileBuffer.append("  );\r\n");\r
+                                               break;\r
+                                       case CommonDefinition.ModuleTypeDxeCore:\r
+                                       case CommonDefinition.ModuleTypeDxeDriver:\r
+                                       case CommonDefinition.ModuleTypeDxeRuntimeDriver:\r
+                                       case CommonDefinition.ModuleTypeDxeSmmDriver:\r
+                                       case CommonDefinition.ModuleTypeDxeSalDriver:\r
+                                       case CommonDefinition.ModuleTypeUefiDriver:\r
+                                       case CommonDefinition.ModuleTypeUefiApplication:\r
+                                               fileBuffer.append("EFI_STATUS\r\n");\r
+                                               fileBuffer.append("EFIAPI\r\n");\r
+                                               fileBuffer.append(libInstanceList.get(i)[0]);\r
+                                               fileBuffer.append(" (\r\n");\r
+                                               fileBuffer.append("  IN EFI_HANDLE        ImageHandle,\r\n");\r
+                                               fileBuffer.append("  IN EFI_SYSTEM_TABLE  *SystemTable\r\n");\r
+                                               fileBuffer.append("  );\r\n");\r
+                                               break;\r
+                           }\r
             }\r
         }\r
 \r
@@ -1610,10 +1644,17 @@ public class AutoGen {
                     fileBuffer.append("\r\n");\r
                     isFirst = false;\r
                 }\r
-                fileBuffer.append("  Status = ");\r
-                fileBuffer.append(libInstanceList.get(i));\r
-                fileBuffer.append("(ImageHandle, SystemTable);\r\n");\r
-                fileBuffer.append("  ASSERT_EFI_ERROR (Status);\r\n");\r
+                               if (libInstanceList.get(i)[1].equalsIgnoreCase(EdkDefinitions.MODULE_TYPE_BASE)) {\r
+                    fileBuffer.append("  Status = ");\r
+                    fileBuffer.append(libInstanceList.get(i)[0]);\r
+                    fileBuffer.append("();\r\n");\r
+                    fileBuffer.append("  VOID\r\n");\r
+                               } else {\r
+                                       fileBuffer.append("  Status = ");\r
+                                       fileBuffer.append(libInstanceList.get(i)[0]);\r
+                                       fileBuffer.append("(ImageHandle, SystemTable);\r\n");\r
+                                       fileBuffer.append("  ASSERT_EFI_ERROR (Status);\r\n");\r
+                               }\r
             }\r
             fileBuffer.append("}\r\n");\r
             break;\r
@@ -1621,100 +1662,88 @@ public class AutoGen {
     }\r
 \r
     /**\r
-     * ExternsDriverBindingToAutoGenC\r
-     *\r
-     * This function is to write DRIVER_BINDING, COMPONENT_NAME,\r
-     * DRIVER_CONFIGURATION, DRIVER_DIAGNOSTIC in AutoGen.c.\r
-     *\r
-     * @param fileBuffer\r
-     *            String buffer for AutoGen.c\r
-     */\r
+      ExternsDriverBindingToAutoGenC\r
+     \r
+      This function is to write DRIVER_BINDING, COMPONENT_NAME,\r
+      DRIVER_CONFIGURATION, DRIVER_DIAGNOSTIC in AutoGen.c.\r
+     \r
+      @param fileBuffer\r
+                 String buffer for AutoGen.c\r
+    **/\r
     void ExternsDriverBindingToAutoGenC(StringBuffer fileBuffer)\r
-    throws BuildException {\r
+    throws EdkException {\r
 \r
         //\r
-        // Check what <extern> contains. And the number of following elements\r
-        // under <extern> should be same. 1. DRIVER_BINDING 2. COMPONENT_NAME\r
-        // 3.DRIVER_CONFIGURATION 4. DRIVER_DIAGNOSTIC\r
-        //\r
-\r
-        String[] drvBindList = SurfaceAreaQuery.getDriverBindingArray();\r
+        // Get the arry of extern. The driverBindingGroup is a 2 dimension array.\r
+               // The second dimension is include following element: DriverBinding, \r
+               // ComponentName, DriverConfiguration, DriverDiag;\r
+               // \r
+        String[][] driverBindingGroup = this.saq.getExternProtocolGroup();\r
 \r
         //\r
-        // If component name protocol,component configuration protocol,\r
-        // component diagnostic protocol is not null or empty, check\r
-        // if every one have the same number of the driver binding protocol.\r
-        //\r
-        if (drvBindList == null || drvBindList.length == 0) {\r
-            return;\r
-        }\r
-\r
-        String[] compNamList = SurfaceAreaQuery.getComponentNameArray();\r
-        String[] compConfList = SurfaceAreaQuery.getDriverConfigArray();\r
-        String[] compDiagList = SurfaceAreaQuery.getDriverDiagArray();\r
-\r
+        // inital BitMask;\r
+               // \r
         int BitMask = 0;\r
 \r
         //\r
         // Write driver binding protocol extern to autogen.c\r
         //\r
-        for (int i = 0; i < drvBindList.length; i++) {\r
-            fileBuffer.append("extern EFI_DRIVER_BINDING_PROTOCOL ");\r
-            fileBuffer.append(drvBindList[i]);\r
-            fileBuffer.append(";\r\n");\r
+        for (int i = 0; i < driverBindingGroup.length; i++) {\r
+                       if (driverBindingGroup[i][0] != null) {\r
+                               fileBuffer.append("extern EFI_DRIVER_BINDING_PROTOCOL ");\r
+                fileBuffer.append(driverBindingGroup[i][0]);\r
+                fileBuffer.append(";\r\n");\r
+                       } \r
         }\r
 \r
         //\r
         // Write component name protocol extern to autogen.c\r
         //\r
-        if (compNamList != null && compNamList.length != 0) {\r
-            if (drvBindList.length != compNamList.length) {\r
-                throw new BuildException(\r
-                                        "Different number of Driver Binding and Component Name protocols!");\r
-            }\r
-\r
-            BitMask |= 0x01;\r
-            for (int i = 0; i < compNamList.length; i++) {\r
-                fileBuffer.append("extern EFI_COMPONENT_NAME_PROTOCOL ");\r
-                fileBuffer.append(compNamList[i]);\r
-                fileBuffer.append(";\r\n");\r
-            }\r
-        }\r
+               for (int i = 0; i < driverBindingGroup.length; i++) {\r
+                       if (driverBindingGroup[i][1]!= null) {\r
+                               if (driverBindingGroup[i][0] != null) {\r
+                                       BitMask |= 0x01;\r
+                                   fileBuffer.append("extern EFI_COMPONENT_NAME_PROTOCOL ");\r
+                    fileBuffer.append(driverBindingGroup[i][1]);\r
+                    fileBuffer.append(";\r\n");\r
+                               } else {\r
+                    throw new AutoGenException("DriverBinding can't be empty!!");\r
+                               }\r
+                       }\r
+               }\r
 \r
         //\r
         // Write driver configration protocol extern to autogen.c\r
         //\r
-        if (compConfList != null && compConfList.length != 0) {\r
-            if (drvBindList.length != compConfList.length) {\r
-                throw new BuildException(\r
-                                        "Different number of Driver Binding and Driver Configuration protocols!");\r
-            }\r
-\r
-            BitMask |= 0x02;\r
-            for (int i = 0; i < compConfList.length; i++) {\r
-                fileBuffer.append("extern EFI_DRIVER_CONFIGURATION_PROTOCOL ");\r
-                fileBuffer.append(compConfList[i]);\r
-                fileBuffer.append(";\r\n");\r
-            }\r
-        }\r
-\r
+               for (int i = 0; i < driverBindingGroup.length; i++) {\r
+                       if (driverBindingGroup[i][2] != null) {\r
+                               if (driverBindingGroup[i][0] != null) {\r
+                    BitMask |= 0x02;\r
+                                   fileBuffer.append("extern EFI_DRIVER_CONFIGURATION_PROTOCOL ");\r
+                    fileBuffer.append(driverBindingGroup[i][2]);\r
+                    fileBuffer.append(";\r\n");\r
+                               } else {\r
+                    throw new AutoGenException("DriverBinding can't be empty!!");\r
+                               }\r
+                       }\r
+               }\r
+        \r
         //\r
         // Write driver dignastic protocol extern to autogen.c\r
         //\r
-        if (compDiagList != null && compDiagList.length != 0) {\r
-            if (drvBindList.length != compDiagList.length) {\r
-                throw new BuildException(\r
-                                        "Different number of Driver Binding and Driver Diagnosis protocols!");\r
-            }\r
-\r
-            BitMask |= 0x04;\r
-            for (int i = 0; i < compDiagList.length; i++) {\r
-                fileBuffer.append("extern EFI_DRIVER_DIAGNOSTICS_PROTOCOL ");\r
-                fileBuffer.append(compDiagList[i]);\r
-                fileBuffer.append(";\r\n");\r
-            }\r
-        }\r
-\r
+               for (int i = 0; i < driverBindingGroup.length; i++) {\r
+                       if (driverBindingGroup[i][3] != null) {\r
+                               if (driverBindingGroup[i][0] != null) {\r
+                    BitMask |= 0x04;\r
+                    fileBuffer.append("extern EFI_DRIVER_DIAGNOSTICS_PROTOCOL ");\r
+                    fileBuffer.append(driverBindingGroup[i][3]);\r
+                    fileBuffer.append(";\r\n");\r
+                               } else {\r
+                    throw new AutoGenException("DriverBinding can't be empty!!");\r
+                               }\r
+                       }\r
+               }\r
+      \r
         //\r
         // Write driver module protocol bitmask.\r
         //\r
@@ -1729,70 +1758,79 @@ public class AutoGen {
         fileBuffer\r
         .append("GLOBAL_REMOVE_IF_UNREFERENCED const UINTN  _gDriverModelProtocolListEntries = ");\r
 \r
-        fileBuffer.append(Integer.toString(drvBindList.length));\r
+        fileBuffer.append(Integer.toString(driverBindingGroup.length));\r
         fileBuffer.append(";\r\n");\r
 \r
         //\r
         // Write drive module protocol list to autogen.c\r
         //\r
-        fileBuffer\r
+               if (driverBindingGroup.length > 0) {\r
+                       fileBuffer\r
         .append("GLOBAL_REMOVE_IF_UNREFERENCED const EFI_DRIVER_MODEL_PROTOCOL_LIST  _gDriverModelProtocolList[] = {");\r
-        for (int i = 0; i < drvBindList.length; i++) {\r
+               }\r
+        \r
+               \r
+        for (int i = 0; i < driverBindingGroup.length; i++) {\r
             if (i != 0) {\r
                 fileBuffer.append(",");\r
             }\r
-            fileBuffer.append("\r\n {\r\n");\r
+                       \r
+                       fileBuffer.append("\r\n {\r\n");\r
             fileBuffer.append("  &");\r
-            fileBuffer.append(drvBindList[i]);\r
+            fileBuffer.append(driverBindingGroup[i][0]);\r
             fileBuffer.append(", \r\n");\r
+                       \r
 \r
-            if (compNamList != null) {\r
+            if (driverBindingGroup[i][1] != null) {\r
                 fileBuffer.append("  &");\r
-                fileBuffer.append(compNamList[i]);\r
+                fileBuffer.append(driverBindingGroup[i][1]);\r
                 fileBuffer.append(", \r\n");\r
             } else {\r
                 fileBuffer.append("  NULL, \r\n");\r
             }\r
 \r
-            if (compConfList != null) {\r
+            if (driverBindingGroup[i][2] != null) {\r
                 fileBuffer.append("  &");\r
-                fileBuffer.append(compConfList[i]);\r
+                fileBuffer.append(driverBindingGroup[i][2]);\r
                 fileBuffer.append(", \r\n");\r
             } else {\r
                 fileBuffer.append("  NULL, \r\n");\r
             }\r
 \r
-            if (compDiagList != null) {\r
+            if (driverBindingGroup[i][3] != null) {\r
                 fileBuffer.append("  &");\r
-                fileBuffer.append(compDiagList[i]);\r
+                fileBuffer.append(driverBindingGroup[i][3]);\r
                 fileBuffer.append(", \r\n");\r
             } else {\r
                 fileBuffer.append("  NULL, \r\n");\r
             }\r
             fileBuffer.append("  }");\r
         }\r
-        fileBuffer.append("\r\n};\r\n");\r
+\r
+               if (driverBindingGroup.length > 0) {\r
+                       fileBuffer.append("\r\n};\r\n");\r
+               }\r
     }\r
 \r
     /**\r
-     * ExternCallBackToAutoGenC\r
-     *\r
-     * This function adds <SetVirtualAddressMapCallBack> and\r
-     * <ExitBootServicesCallBack> infomation to AutoGen.c\r
-     *\r
-     * @param fileBuffer\r
-     *            String buffer for AutoGen.c\r
-     * @throws BuildException\r
-     */\r
+      ExternCallBackToAutoGenC\r
+     \r
+      This function adds <SetVirtualAddressMapCallBack> and\r
+      <ExitBootServicesCallBack> infomation to AutoGen.c\r
+     \r
+      @param fileBuffer\r
+                 String buffer for AutoGen.c\r
+      @throws BuildException\r
+    **/\r
     void ExternCallBackToAutoGenC(StringBuffer fileBuffer)\r
-    throws BuildException {\r
+    throws EdkException {\r
         //\r
         // Collect module's <SetVirtualAddressMapCallBack> and\r
         // <ExitBootServiceCallBack> and add to setVirtualAddList\r
         //  exitBootServiceList.\r
         //\r
-        String[] setVirtuals = SurfaceAreaQuery.getSetVirtualAddressMapCallBackArray();\r
-        String[] exitBoots = SurfaceAreaQuery.getExitBootServicesCallBackArray();\r
+        String[] setVirtuals = saq.getSetVirtualAddressMapCallBackArray();\r
+        String[] exitBoots = saq.getExitBootServicesCallBackArray();\r
         if (setVirtuals != null) {\r
             for (int j = 0; j < setVirtuals.length; j++) {\r
                 this.setVirtalAddList.add(setVirtuals[j]);\r
@@ -1808,28 +1846,205 @@ public class AutoGen {
         //  and <ExitBootServicesCallBack>\r
         //\r
         String moduleType = this.moduleId.getModuleType();\r
-        boolean UefiOrDxeModule = false;\r
-        int Count = 0;\r
-        int i;\r
         switch (CommonDefinition.getModuleType(moduleType)) {\r
         case CommonDefinition.ModuleTypeDxeDriver:\r
         case CommonDefinition.ModuleTypeDxeRuntimeDriver:\r
         case CommonDefinition.ModuleTypeDxeSalDriver:\r
         case CommonDefinition.ModuleTypeUefiDriver:\r
-        case CommonDefinition.ModuleTypeUefiApplication:\r
-            //\r
-            // Entry point lib for these module types needs to know the count\r
-            // of entryPoint.\r
-            //\r
-            UefiOrDxeModule = true;\r
-            fileBuffer\r
+               case CommonDefinition.ModuleTypeUefiApplication:\r
+                       //\r
+                       //  If moduleType is one of above, call setVirtualAddressToAutogenC,\r
+                       //  and setExitBootServiceToAutogenC.\r
+                       // \r
+            setVirtualAddressToAutogenC(fileBuffer);\r
+                       setExitBootServiceToAutogenC(fileBuffer);\r
+            break;\r
+        default:\r
+            break;\r
+        }\r
+    }\r
+\r
+    /**\r
+      copyFlashMapHToDebugDir\r
+      \r
+      This function is to copy the falshmap.h to debug directory and change \r
+      its name to TianoR8FlashMap.h\r
+      \r
+      @param \r
+      @return\r
+    **/\r
+    private void copyFlashMapHToDebugDir() throws  AutoGenException{\r
+\r
+        File inFile = new File(fvDir + File.separatorChar + CommonDefinition.FLASHMAPH);\r
+        int size = (int)inFile.length();\r
+        byte[] buffer = new byte[size];\r
+        File outFile = new File (this.outputPath + File.separatorChar + CommonDefinition.TIANOR8PLASHMAPH);\r
+        //\r
+        //  If TianoR8FlashMap.h existed and the flashMap.h don't change,\r
+        //  do nothing.\r
+        //\r
+        if ((!outFile.exists()) ||(inFile.lastModified() - outFile.lastModified()) >= 0) {\r
+                       if (inFile.exists()) {\r
+                               try{\r
+                                       FileInputStream fis = new FileInputStream (inFile);\r
+                                   fis.read(buffer);\r
+                                   FileOutputStream fos = new FileOutputStream(outFile);\r
+                                   fos.write(buffer);\r
+                                   fis.close();\r
+                                   fos.close();\r
+                               } catch (IOException e){\r
+                    throw new AutoGenException("The file, flashMap.h can't be open!");\r
+                               }\r
+                               \r
+                       } else {\r
+                               throw new AutoGenException("The file, flashMap.h doesn't exist!");\r
+                       }\r
+        }\r
+    }\r
+\r
+    /**\r
+      This function first order the library instances, then collect\r
+      library instance 's PPI, Protocol, GUID,\r
+      SetVirtalAddressMapCallBack, ExitBootServiceCallBack, and\r
+      Destructor, Constructor.\r
+      \r
+      @param\r
+      @return    \r
+    **/\r
+    private void collectLibInstanceInfo() throws EdkException{\r
+        int index;\r
+\r
+        String libConstructName = null;\r
+        String libDestructName = null;\r
+               String libModuleType   = null;\r
+        String[] setVirtuals = null;\r
+        String[] exitBoots = null;\r
+\r
+        ModuleIdentification[] libraryIdList = saq.getLibraryInstance(this.arch);\r
+\r
+               if (libraryIdList != null) {\r
+                       //\r
+                       // Reorder library instance sequence.\r
+                       //\r
+                       AutogenLibOrder libOrder = new AutogenLibOrder(libraryIdList,\r
+                                                                                                                  this.arch);\r
+                       List<ModuleIdentification> orderList = libOrder\r
+                                                                                                  .orderLibInstance();\r
+\r
+                       if (orderList != null) {\r
+                               //\r
+                               // Process library instance one by one.\r
+                               //\r
+                               for (int i = 0; i < orderList.size(); i++) {\r
+\r
+                                       //\r
+                                       // Get library instance basename.\r
+                                       //\r
+                                       ModuleIdentification libInstanceId = orderList.get(i);\r
+\r
+                                       //\r
+                                       // Get override map\r
+                                       //\r
+\r
+                                       Map<String, XmlObject> libDoc = GlobalData.getDoc(libInstanceId, this.arch);\r
+                                       saq.push(libDoc);\r
+                                       //\r
+                                       // Get <PPis>, <Protocols>, <Guids> list of this library\r
+                                       // instance.\r
+                                       //\r
+                                       String[] ppiList = saq.getPpiArray(this.arch);\r
+                                       String[] ppiNotifyList = saq.getPpiNotifyArray(this.arch);\r
+                                       String[] protocolList = saq.getProtocolArray(this.arch);\r
+                                       String[] protocolNotifyList = saq.getProtocolNotifyArray(this.arch);\r
+                                       String[] guidList = saq.getGuidEntryArray(this.arch);\r
+                                       PackageIdentification[] pkgList = saq.getDependencePkg(this.arch);\r
+\r
+                                       //\r
+                                       // Add those ppi, protocol, guid in global ppi,\r
+                                       // protocol, guid\r
+                                       // list.\r
+                                       //\r
+                                       for (index = 0; index < ppiList.length; index++) {\r
+                                               this.mPpiList.add(ppiList[index]);\r
+                                       }\r
+\r
+                                       for (index = 0; index < ppiNotifyList.length; index++) {\r
+                                               this.mPpiList.add(ppiNotifyList[index]);\r
+                                       }\r
+\r
+                                       for (index = 0; index < protocolList.length; index++) {\r
+                                               this.mProtocolList.add(protocolList[index]);\r
+                                       }\r
+\r
+                                       for (index = 0; index < protocolNotifyList.length; index++) {\r
+                                               this.mProtocolList.add(protocolNotifyList[index]);\r
+                                       }\r
+\r
+                                       for (index = 0; index < guidList.length; index++) {\r
+                                               this.mGuidList.add(guidList[index]);\r
+                                       }\r
+                                       for (index = 0; index < pkgList.length; index++) {\r
+                                               if (!this.mDepPkgList.contains(pkgList[index])) {\r
+                                                       this.mDepPkgList.add(pkgList[index]);\r
+                                               }\r
+                                       }\r
+\r
+                                       //\r
+                                       // If not yet parse this library instance's constructor\r
+                                       // element,parse it.\r
+                                       //\r
+                                       libConstructName = saq.getLibConstructorName();\r
+                                       libDestructName = saq.getLibDestructorName();\r
+                                       libModuleType = saq.getModuleType();\r
+\r
+                                       //\r
+                                       // Collect SetVirtualAddressMapCallBack and\r
+                                       // ExitBootServiceCallBack.\r
+                                       //\r
+                                       setVirtuals = saq.getSetVirtualAddressMapCallBackArray();\r
+                                       exitBoots = saq.getExitBootServicesCallBackArray();\r
+                                       if (setVirtuals != null) {\r
+                                               for (int j = 0; j < setVirtuals.length; j++) {\r
+                                                       this.setVirtalAddList.add(setVirtuals[j]);\r
+                                               }\r
+                                       }\r
+                                       if (exitBoots != null) {\r
+                                               for (int k = 0; k < exitBoots.length; k++) {\r
+                                                       this.exitBootServiceList.add(exitBoots[k]);\r
+                                               }\r
+                                       }\r
+                                       saq.pop();\r
+                                       //\r
+                                       // Add dependent library instance constructor function.\r
+                                       //\r
+                                       if (libConstructName != null) {\r
+                                               this.libConstructList.add(new String[] {libConstructName, libModuleType});\r
+                                       }\r
+                                       //\r
+                                       // Add dependent library instance destructor fuction.\r
+                                       //\r
+                                       if (libDestructName != null) {\r
+                                               this.libDestructList.add(new String[] {libDestructName, libModuleType});\r
+                                       }\r
+                               }\r
+                       }\r
+\r
+               }\r
+    }\r
+       private void setVirtualAddressToAutogenC(StringBuffer fileBuffer){\r
+        //\r
+        // Entry point lib for these module types needs to know the count\r
+        // of entryPoint.\r
+        //\r
+               fileBuffer\r
             .append("\r\nGLOBAL_REMOVE_IF_UNREFERENCED  const UINTN _gDriverSetVirtualAddressMapEventCount = ");\r
 \r
             //\r
             // If the list is not valid or has no entries set count to zero else\r
             // set count to the number of valid entries\r
             //\r
-            Count = 0;\r
+            int Count = 0;\r
+                       int i = 0;\r
             if (this.setVirtalAddList != null) {\r
                 for (i = 0; i < this.setVirtalAddList.size(); i++) {\r
                     if (this.setVirtalAddList.get(i).equalsIgnoreCase("")) {\r
@@ -1841,13 +2056,7 @@ public class AutoGen {
 \r
             fileBuffer.append(Integer.toString(Count));\r
             fileBuffer.append(";\r\n\r\n");\r
-            break;\r
-        default:\r
-            break;\r
-        }\r
-\r
-        if (this.setVirtalAddList == null || this.setVirtalAddList.size() == 0) {\r
-            if (UefiOrDxeModule) {\r
+            if (this.setVirtalAddList == null || this.setVirtalAddList.size() == 0) {\r
                 //\r
                 // No data so make a NULL list\r
                 //\r
@@ -1855,87 +2064,82 @@ public class AutoGen {
                 .append("\r\nGLOBAL_REMOVE_IF_UNREFERENCED const EFI_EVENT_NOTIFY _gDriverSetVirtualAddressMapEvent[] = {\r\n");\r
                 fileBuffer.append("  NULL\r\n");\r
                 fileBuffer.append("};\r\n\r\n");\r
-            }\r
-        } else {\r
-            //\r
-            // Write SetVirtualAddressMap function definition.\r
-            //\r
-            for (i = 0; i < this.setVirtalAddList.size(); i++) {\r
-                if (this.setVirtalAddList.get(i).equalsIgnoreCase("")) {\r
-                    break;\r
-                }\r
-                fileBuffer.append("VOID\r\n");\r
-                fileBuffer.append("EFIAPI\r\n");\r
-                fileBuffer.append(this.setVirtalAddList.get(i));\r
-                fileBuffer.append(" (\r\n");\r
-                fileBuffer.append("  IN EFI_EVENT  Event,\r\n");\r
-                fileBuffer.append("  IN VOID       *Context\r\n");\r
-                fileBuffer.append("  );\r\n\r\n");\r
-            }\r
+            } else {\r
+                //\r
+                // Write SetVirtualAddressMap function definition.\r
+                //\r
+                for (i = 0; i < this.setVirtalAddList.size(); i++) {\r
+                    if (this.setVirtalAddList.get(i).equalsIgnoreCase("")) {\r
+                        break;\r
+                    }\r
+                    fileBuffer.append("VOID\r\n");\r
+                    fileBuffer.append("EFIAPI\r\n");\r
+                    fileBuffer.append(this.setVirtalAddList.get(i));\r
+                    fileBuffer.append(" (\r\n");\r
+                    fileBuffer.append("  IN EFI_EVENT  Event,\r\n");\r
+                    fileBuffer.append("  IN VOID       *Context\r\n");\r
+                    fileBuffer.append("  );\r\n\r\n");\r
+                 }\r
 \r
-            //\r
-            // Write SetVirtualAddressMap entry point array.\r
-            //\r
-            fileBuffer\r
-            .append("\r\nGLOBAL_REMOVE_IF_UNREFERENCED const EFI_EVENT_NOTIFY _gDriverSetVirtualAddressMapEvent[] = {");\r
-            for (i = 0; i < this.setVirtalAddList.size(); i++) {\r
-                if (this.setVirtalAddList.get(i).equalsIgnoreCase("")) {\r
-                    break;\r
-                }\r
+                //\r
+                // Write SetVirtualAddressMap entry point array.\r
+                //\r
+                fileBuffer\r
+                .append("\r\nGLOBAL_REMOVE_IF_UNREFERENCED const EFI_EVENT_NOTIFY _gDriverSetVirtualAddressMapEvent[] = {");\r
+                for (i = 0; i < this.setVirtalAddList.size(); i++) {\r
+                    if (this.setVirtalAddList.get(i).equalsIgnoreCase("")) {\r
+                        break;\r
+                    }\r
 \r
-                if (i == 0) {\r
-                    fileBuffer.append("\r\n  ");\r
-                } else {\r
-                    fileBuffer.append(",\r\n  ");\r
-                }\r
+                    if (i == 0) {\r
+                        fileBuffer.append("\r\n  ");\r
+                    } else {\r
+                        fileBuffer.append(",\r\n  ");\r
+                    }\r
 \r
-                fileBuffer.append(this.setVirtalAddList.get(i));\r
-            }\r
-            //\r
-            //  If module is not DXE_DRIVER, DXE_RUNTIME_DIRVER, UEFI_DRIVER\r
-            //  UEFI_APPLICATION and DXE_SAL_DRIVER add the NULL at the end of\r
-            //  _gDriverSetVirtualAddressMapEvent list.\r
-            //\r
-            if (!UefiOrDxeModule) {\r
+                    fileBuffer.append(this.setVirtalAddList.get(i));\r
+                }\r
+                //\r
+                // add the NULL at the end of _gDriverSetVirtualAddressMapEvent list.\r
+                //\r
                 fileBuffer.append(",\r\n  NULL");\r
-            }\r
-            fileBuffer.append("\r\n};\r\n\r\n");\r
-        }\r
+                fileBuffer.append("\r\n};\r\n\r\n");\r
+                       }\r
+       }\r
 \r
-        if (UefiOrDxeModule) {\r
-            //\r
-            // Entry point lib for these module types needs to know the count.\r
-            //\r
-            fileBuffer\r
-            .append("\r\nGLOBAL_REMOVE_IF_UNREFERENCED  const UINTN _gDriverExitBootServicesEventCount = ");\r
 \r
-            //\r
-            // If the list is not valid or has no entries set count to zero else\r
-            // set count to the number of valid entries.\r
-            //\r
-            Count = 0;\r
-            if (this.exitBootServiceList != null) {\r
-                for (i = 0; i < this.exitBootServiceList.size(); i++) {\r
-                    if (this.exitBootServiceList.get(i).equalsIgnoreCase("")) {\r
-                        break;\r
-                    }\r
+       private void setExitBootServiceToAutogenC(StringBuffer fileBuffer){\r
+               //\r
+        // Entry point lib for these module types needs to know the count.\r
+        //\r
+        fileBuffer\r
+        .append("\r\nGLOBAL_REMOVE_IF_UNREFERENCED  const UINTN _gDriverExitBootServicesEventCount = ");\r
+\r
+        //\r
+        // If the list is not valid or has no entries set count to zero else\r
+        // set count to the number of valid entries.\r
+        //\r
+        int Count = 0;\r
+               int i = 0; \r
+        if (this.exitBootServiceList != null) {\r
+            for (i = 0; i < this.exitBootServiceList.size(); i++) {\r
+                if (this.exitBootServiceList.get(i).equalsIgnoreCase("")) {\r
+                    break;\r
                 }\r
-                Count = i;\r
             }\r
-            fileBuffer.append(Integer.toString(Count));\r
-            fileBuffer.append(";\r\n\r\n");\r
+            Count = i;\r
         }\r
+        fileBuffer.append(Integer.toString(Count));\r
+        fileBuffer.append(";\r\n\r\n");\r
 \r
-        if (this.exitBootServiceList == null || this.exitBootServiceList.size() == 0) {\r
-            if (UefiOrDxeModule) {\r
-                //\r
-                // No data so make a NULL list.\r
-                //\r
-                fileBuffer\r
-                .append("\r\nGLOBAL_REMOVE_IF_UNREFERENCED const EFI_EVENT_NOTIFY _gDriverExitBootServicesEvent[] = {\r\n");\r
-                fileBuffer.append("  NULL\r\n");\r
-                fileBuffer.append("};\r\n\r\n");\r
-            }\r
+               if (this.exitBootServiceList == null || this.exitBootServiceList.size() == 0) {\r
+            //      \r
+            // No data so make a NULL list.\r
+            //\r
+            fileBuffer\r
+            .append("\r\nGLOBAL_REMOVE_IF_UNREFERENCED const EFI_EVENT_NOTIFY _gDriverExitBootServicesEvent[] = {\r\n");\r
+            fileBuffer.append("  NULL\r\n");\r
+            fileBuffer.append("};\r\n\r\n");\r
         } else {\r
             //\r
             // Write DriverExitBootServices function definition.\r
@@ -1971,179 +2175,11 @@ public class AutoGen {
                 }\r
                 fileBuffer.append(this.exitBootServiceList.get(i));\r
             }\r
-            if (!UefiOrDxeModule) {\r
-                fileBuffer.append(",\r\n  NULL");\r
-            }\r
+            \r
+            fileBuffer.append(",\r\n  NULL");\r
             fileBuffer.append("\r\n};\r\n\r\n");\r
         }\r
 \r
-    }\r
-\r
-    private void copyFlashMapHToDebugDir() throws  AutoGenException{\r
-\r
-        File inFile = new File(fvDir + File.separatorChar + CommonDefinition.flashMapH);\r
-        int size = (int)inFile.length();\r
-        byte[] buffer = new byte[size];\r
-        File outFile = new File (this.outputPath + File.separatorChar + CommonDefinition.tianoR8FlashMapH);\r
-        //\r
-        //  If TianoR8FlashMap.h existed and the flashMap.h don't change,\r
-        //  do nothing.\r
-        //\r
-        if ((!outFile.exists()) ||(inFile.lastModified() - outFile.lastModified()) >= 0) {\r
-            try {\r
-                if (inFile.exists()) {\r
-                    FileInputStream fis = new FileInputStream (inFile);\r
-                    fis.read(buffer);\r
-                    FileOutputStream fos = new FileOutputStream(outFile);\r
-                    fos.write(buffer);\r
-                    fis.close();\r
-                    fos.close();\r
-                } else {\r
-                    throw new AutoGenException("The file, flashMap.h doesn't exist!");\r
-                }\r
-            } catch (Exception e) {\r
-                throw new AutoGenException(e.getMessage());\r
-            }\r
-        }\r
-    }\r
-\r
-    /**\r
-    *This function first order the library instances, then collect\r
-    *library instance 's PPI, Protocol, GUID,\r
-    *SetVirtalAddressMapCallBack, ExitBootServiceCallBack, and\r
-    *Destructor, Constructor.\r
-    *\r
-    **/\r
-    private void collectLibInstanceInfo(){\r
-        int index;\r
-\r
-        String moduleType = SurfaceAreaQuery.getModuleType();\r
-        String libConstructName = null;\r
-        String libDestructName = null;\r
-        String[] setVirtuals = null;\r
-        String[] exitBoots = null;\r
-\r
-        ModuleIdentification[] libraryIdList = SurfaceAreaQuery\r
-                                               .getLibraryInstance(this.arch);\r
-        try {\r
-            if (libraryIdList != null) {\r
-                //\r
-                // Reorder library instance sequence.\r
-                //\r
-                AutogenLibOrder libOrder = new AutogenLibOrder(libraryIdList,\r
-                                                               this.arch);\r
-                List<ModuleIdentification> orderList = libOrder\r
-                                                       .orderLibInstance();\r
-\r
-                if (orderList != null) {\r
-                    //\r
-                    // Process library instance one by one.\r
-                    //\r
-                    for (int i = 0; i < orderList.size(); i++) {\r
-\r
-                        //\r
-                        // Get library instance basename.\r
-                        //\r
-                        ModuleIdentification libInstanceId = orderList.get(i);\r
-\r
-                        //\r
-                        // Get override map\r
-                        //\r
-\r
-                        Map<String, XmlObject> libDoc = GlobalData.getDoc(\r
-                                                                         libInstanceId, this.arch);\r
-                        SurfaceAreaQuery.push(libDoc);\r
-                        //\r
-                        // Get <PPis>, <Protocols>, <Guids> list of this library\r
-                        // instance.\r
-                        //\r
-                        String[] ppiList = SurfaceAreaQuery.getPpiArray(this.arch);\r
-                        String[] ppiNotifyList = SurfaceAreaQuery\r
-                                                 .getPpiNotifyArray(this.arch);\r
-                        String[] protocolList = SurfaceAreaQuery\r
-                                                .getProtocolArray(this.arch);\r
-                        String[] protocolNotifyList = SurfaceAreaQuery\r
-                                                      .getProtocolNotifyArray(this.arch);\r
-                        String[] guidList = SurfaceAreaQuery\r
-                                            .getGuidEntryArray(this.arch);\r
-                        PackageIdentification[] pkgList = SurfaceAreaQuery.getDependencePkg(this.arch);\r
-\r
-                        //\r
-                        // Add those ppi, protocol, guid in global ppi,\r
-                        // protocol, guid\r
-                        // list.\r
-                        //\r
-                        for (index = 0; index < ppiList.length; index++) {\r
-                            this.mPpiList.add(ppiList[index]);\r
-                        }\r
-\r
-                        for (index = 0; index < ppiNotifyList.length; index++) {\r
-                            this.mPpiList.add(ppiNotifyList[index]);\r
-                        }\r
-\r
-                        for (index = 0; index < protocolList.length; index++) {\r
-                            this.mProtocolList.add(protocolList[index]);\r
-                        }\r
+       }\r
 \r
-                        for (index = 0; index < protocolNotifyList.length; index++) {\r
-                            this.mProtocolList.add(protocolNotifyList[index]);\r
-                        }\r
-\r
-                        for (index = 0; index < guidList.length; index++) {\r
-                            this.mGuidList.add(guidList[index]);\r
-                        }\r
-                        for (index = 0; index < pkgList.length; index++) {\r
-                            if (!this.mDepPkgList.contains(pkgList[index])) {\r
-                                this.mDepPkgList.add(pkgList[index]);\r
-                            }\r
-                        }\r
-\r
-                        //\r
-                        // If not yet parse this library instance's constructor\r
-                        // element,parse it.\r
-                        //\r
-                        libConstructName = SurfaceAreaQuery\r
-                                           .getLibConstructorName();\r
-                        libDestructName = SurfaceAreaQuery\r
-                                          .getLibDestructorName();\r
-\r
-                        //\r
-                        // Collect SetVirtualAddressMapCallBack and\r
-                        // ExitBootServiceCallBack.\r
-                        //\r
-                        setVirtuals = SurfaceAreaQuery.getSetVirtualAddressMapCallBackArray();\r
-                        exitBoots = SurfaceAreaQuery.getExitBootServicesCallBackArray();\r
-                        if (setVirtuals != null) {\r
-                            for (int j = 0; j < setVirtuals.length; j++) {\r
-                                this.setVirtalAddList.add(setVirtuals[j]);\r
-                            }\r
-                        }\r
-                        if (exitBoots != null) {\r
-                            for (int k = 0; k < exitBoots.length; k++) {\r
-                                this.exitBootServiceList.add(exitBoots[k]);\r
-                            }\r
-                        }\r
-                        SurfaceAreaQuery.pop();\r
-                        //\r
-                        // Add dependent library instance constructor function.\r
-                        //\r
-                        if (libConstructName != null) {\r
-                            this.libConstructList.add(libConstructName);\r
-                        }\r
-                        //\r
-                        // Add dependent library instance destructor fuction.\r
-                        //\r
-                        if (libDestructName != null) {\r
-                            this.libDestructList.add(libDestructName);\r
-                        }\r
-                    }\r
-                }\r
-\r
-            }\r
-\r
-        } catch (Exception e) {\r
-            System.out.println(e.getMessage());\r
-            System.out.println("Collect library instance failed!");\r
-        }\r
-    }\r
 }\r