Support to handle IPF assembly files

[mirror_edk2.git] / Tools / Java / Source / MigrationTools / org / tianocore / migration / ModuleInfo.java

diff --git a/Tools/Java/Source/MigrationTools/org/tianocore/migration/ModuleInfo.java b/Tools/Java/Source/MigrationTools/org/tianocore/migration/ModuleInfo.java
index 69862f67a129dc8ba1442d7d96a45c29c37eebca..df07f836af94dad71cab7a153e55af317fe0ad68 100644 (file)
--- a/Tools/Java/Source/MigrationTools/org/tianocore/migration/ModuleInfo.java
+++ b/Tools/Java/Source/MigrationTools/org/tianocore/migration/ModuleInfo.java
@@ -108,15 +108,17 @@ public final class ModuleInfo {
     \r
     public final void enroll(String filepath) throws Exception {\r
         String temp = null;\r
-        if (filepath.contains(".c") || filepath.contains(".C") || filepath.contains(".h") || \r
-                filepath.contains(".H") || filepath.contains(".dxs") || filepath.contains(".uni")) {\r
-               addSourceFile(filepath.replace(modulepath + File.separator, ""), null);\r
-        } else if (filepath.contains(".inf") || filepath.contains(".msa")) {\r
+         if (filepath.contains(".inf") || filepath.contains(".msa")) {\r
             temp = filepath.replace(modulepath + File.separator, "");\r
             if (!temp.contains(File.separator)) {                            // .inf in subdirectory is not regarded\r
                 msaorinf.add(temp);\r
             }\r
-        }\r
+        } else if (filepath.contains(".c") || filepath.contains(".C") || filepath.contains(".h") || \r
+                filepath.contains(".H") || filepath.contains(".dxs") || filepath.contains(".uni") ||\r
+                filepath.contains(".s") || filepath.contains(".S") || filepath.contains(".i") ||\r
+                filepath.contains(".asm")) {\r
+               addSourceFile(filepath.replace(modulepath + File.separator, ""), null);\r
+        } \r
     }\r
 \r
     public static final boolean isModule(String path) {\r