add springboot support

Author: waderwu

src/main/java/org/clyze/doop/common/BasicJavaSupport.java

@@ -72,6 +72,7 @@ private void preprocessInput(Database db, Set<String> tmpDirs,
         boolean isZip = filenameL.endsWith(".zip");
         boolean isClass = filenameL.endsWith(".class");
         boolean isApk = filenameL.endsWith(".apk");
+        boolean isSpringBoot = parameters.isSpringBootJar(filenameL);
 
         ArtifactScanner.EntryProcessor gProc = (jarFile, entry, entryName) -> {
             if (entryName.endsWith(".properties"))
@@ -90,6 +91,12 @@ else if ((isJar || isAar || isZip || isWar) && entryName.endsWith(".xml")) {
             // Process WAR inputs.
             parameters.processFatArchives(tmpDirs);
         }
+
+        if (isSpringBoot) {
+            System.out.println("Processing springBoot: " + filename);
+            parameters.processSpringBootArchives(tmpDirs, filename);
+        }
+
         if (isJar || isApk || isZip || isWar)
             artScanner.processArchive(filename, classSet::add, gProc);
         else if (isClass) {

src/main/java/org/clyze/doop/common/Parameters.java

@@ -7,11 +7,17 @@
 import java.util.concurrent.ConcurrentHashMap;
 import java.util.stream.Collectors;
 import java.util.stream.Stream;
+import java.util.zip.ZipEntry;
+import java.util.zip.ZipInputStream;
+
 import org.apache.log4j.Logger;
 import org.clyze.doop.util.filter.ClassFilter;
 import org.clyze.doop.util.filter.GlobClassFilter;
 import org.clyze.utils.ContainerUtils;
 import org.clyze.utils.JHelper;
+import org.zeroturnaround.zip.ZipUtil;
+
+import static org.clyze.utils.ContainerUtils.createTmpDir;
 
 /**
  * This class handles common parameters for Doop Java front-ends.
@@ -314,4 +320,67 @@ public void processFatArchives(Set<String> tmpDirs) {
         System.out.println("inputs = " + getInputs());
         System.out.println("libraries = " + getDependencies());
     }
+
+    public boolean isSpringBootJar(String jarFile){
+        try{
+            ZipInputStream in = new ZipInputStream(new FileInputStream(jarFile));
+
+            ZipEntry entry = null;
+
+            while((entry = in.getNextEntry()) != null){
+                if (entry.isDirectory() && entry.getName().equals("org/springframework/boot/")){
+                    return true;
+                }
+            }
+        }catch (Exception e){
+            e.printStackTrace();
+        }
+        return false;
+    }
+
+    public void processSpringBootArchives(Set<String> tmpDirs, String springBootJar){
+        try{
+            String tmpDirPath = createTmpDir(tmpDirs);
+            Set<String> jarLibs = ConcurrentHashMap.newKeySet();
+
+            if (tmpDirPath == null) {
+                System.out.println("ERROR: null temporary directory path");
+                return;
+            }
+            File tmpDir = new File(tmpDirPath);
+            File springBootJarFile = new File(springBootJar);
+            ZipUtil.unpack(springBootJarFile, tmpDir);
+            // zip BOOT-INF/classes
+            String jar = tmpDir + "/" + springBootJarFile.getName();
+            File webInfClasses = new File(tmpDir,"BOOT-INF/classes/");
+            if (webInfClasses.exists()) {
+                File zipTarget = new File(jar);
+                ZipUtil.pack(webInfClasses, zipTarget);
+                jarLibs.add(zipTarget.getCanonicalPath());
+            }
+            // collect libs in BOOT-INF/lib
+            File bootInfLibs = new File(tmpDir, "BOOT-INF/lib");
+            if (bootInfLibs.exists()) {
+                File[] files = bootInfLibs.listFiles();
+                if (files != null) {
+                    for (File file : files) {
+                        if (file.getName().endsWith(".jar")) {
+                            jarLibs.add(file.getCanonicalPath());
+                        }
+                    }
+                }
+            }
+
+            jarLibs.addAll(getInputs());
+            jarLibs.remove(springBootJar);
+            setInputs(new ArrayList<String>(jarLibs));
+//            jarLibs.addAll(getDependencies());
+//            setDependencies(new ArrayList<>(jarLibs));
+            System.out.println("inputs = " + getInputs());
+            System.out.println("libraries = " + getDependencies());
+        }catch (Exception e){
+            e.printStackTrace();
+        }
+
+    }
 }