Add ModifyInftask in FrameworkTask.

[mirror_edk2.git] / Tools / Source / GenBuild / org / tianocore / build / autogen / AutoGen.java

diff --git a/Tools/Source/GenBuild/org/tianocore/build/autogen/AutoGen.java b/Tools/Source/GenBuild/org/tianocore/build/autogen/AutoGen.java
index 7f9b40e65cd7af1388bf997302e2365e471b542a..3147a245389169fe68761915beac6b1d9a5c3b07 100644 (file)
--- a/Tools/Source/GenBuild/org/tianocore/build/autogen/AutoGen.java
+++ b/Tools/Source/GenBuild/org/tianocore/build/autogen/AutoGen.java
@@ -812,7 +812,7 @@ public class AutoGen {
                        int entryPointCount = 0;\r
                        fileBuffer\r
                                        .append("GLOBAL_REMOVE_IF_UNREFERENCED const UINT32 _gPeimRevision = 0;\r\n");\r
-                       if (entryPointList == null) {\r
+                       if (entryPointList == null || entryPointList.length == 0) {\r
                                fileBuffer.append("EFI_STATUS\r\n");\r
                                fileBuffer.append("EFIAPI\r\n");\r
                                fileBuffer.append("ProcessModuleEntryPointList (\r\n");\r
@@ -899,8 +899,8 @@ public class AutoGen {
                         fileBuffer.append("EFIAPI\r\n");\r
                         fileBuffer.append(entryPointList[i]);\r
                         fileBuffer.append(" (\r\n");\r
-                        fileBuffer.append("  EFI_HANDLE        ImageHandle,\r\n");\r
-                        fileBuffer.append("  EFI_SYSTEM_TABLE  *SystemTable\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
                         entryPointCount++;\r
                     } else {\r
@@ -975,7 +975,7 @@ public class AutoGen {
                                                fileBuffer.append(entryPointList[i]);\r
                                                fileBuffer.append(" (\r\n");\r
                                                fileBuffer\r
-                                                               .append("  EFI_HANDLE        ImageHandle\r\n");\r
+                                                               .append("  IN EFI_HANDLE        ImageHandle\r\n");\r
                                                fileBuffer.append("  );\r\n");\r
                                        } else {\r
                                                break;\r
@@ -1051,8 +1051,8 @@ public class AutoGen {
                     fileBuffer.append("EFIAPI\r\n");\r
                     fileBuffer.append(entryPointList[i]);\r
                     fileBuffer.append(" (\r\n");\r
-                    fileBuffer.append("  EFI_HANDLE        ImageHandle,\r\n");\r
-                    fileBuffer.append("  EFI_SYSTEM_TABLE  *SystemTable\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
                     entryPointCount++;\r
                 }\r
@@ -1141,7 +1141,7 @@ public class AutoGen {
                                                fileBuffer.append(entryPointList[i]);\r
                                                fileBuffer.append(" (\r\n");\r
                                                fileBuffer\r
-                                                               .append("  EFI_HANDLE        ImageHandle\r\n");\r
+                                                               .append("  IN EFI_HANDLE        ImageHandle\r\n");\r
                                                fileBuffer.append("  );\r\n");\r
                                                entryPointCount++;\r
                                        } else {\r
@@ -1163,7 +1163,7 @@ public class AutoGen {
                                                fileBuffer.append(entryPointList[i]);\r
                                                fileBuffer.append(" (\r\n");\r
                                                fileBuffer\r
-                                                               .append("  EFI_HANDLE        ImageHandle\r\n");\r
+                                                               .append("  IN EFI_HANDLE        ImageHandle\r\n");\r
                                                fileBuffer.append("  );\r\n");\r
                                        } else {\r
                                                break;\r