update to 1.1.1

Acmesec · Oct 9, 2022 · e5daaed · e5daaed
1 parent 860e302
commit e5daaed
Show file tree

Hide file tree

Showing 6 changed files with 181 additions and 137 deletions.
diff --git a/build.gradle b/build.gradle
@@ -1,4 +1,4 @@
-version '1.1.0'
+version '1.1.1'
 
 buildscript {
     ext.kotlin_version = '1.4.10'
@@ -58,4 +58,7 @@ compileKotlin {
 }
 compileTestKotlin {
     kotlinOptions.jvmTarget = "1.8"
+}
+tasks.withType(JavaCompile) {
+    options.encoding = "UTF-8"
 }
diff --git a/src/burp/BurpExtender.java b/src/burp/BurpExtender.java
@@ -46,7 +46,7 @@ public class BurpExtender implements IBurpExtender, ITab, IHttpListener{
     public static Map<String,String> config = Config.getInitDatabaseSetting();
     public static DomainConsumer domainConsumer = new DomainConsumer();
     public static WebConsumer webConsumer = new WebConsumer();
-    public static final String VERSION = "1.1.0";
+    public static final String VERSION = "1.1.1";
     public static final String EXTENSION_NAME = "Sylas";
 
     @Override

diff --git a/src/domain/DomainConsumer.java b/src/domain/DomainConsumer.java
@@ -1,6 +1,10 @@
 package domain;
 
 import burp.BurpExtender;
+import ui.Sylas;
+import utils.DbUtil;
+
+import java.util.HashMap;
 
 //负责将收集的域名信息写入数据库
 public class DomainConsumer extends Thread {
@@ -12,8 +16,19 @@ public void run() {
             try {
                 QueueToResult();
                 sleep(5000);
+                if(BurpExtender.db.mode == DbUtil.MYSQL_DB && BurpExtender.db.isConnect){
+                    HashMap<String, HashMap<String, String>> map = BurpExtender.db.getSubDomainMap(BurpExtender.config.get("currentProject"));
+                    for (String key : map.keySet()) {
+                        if (!BurpExtender.subDomainMap.containsKey(key)) {
+                            HashMap<String, String> domain = map.get(key);
+                            BurpExtender.subDomainMap.put(key, domain);
+                            BurpExtender.subDomainCount += 1;
+                            Sylas.addSubDomainToUI(key, domain.get("ipAddress"), domain.get("createTime"));
+                        }
+                    }
+                }
             } catch (Exception error) {
-                error.printStackTrace(BurpExtender.getStderr());
+                BurpExtender.getStderr().println(error);
             }
         }
     }

diff --git a/src/ui/NewProject.java b/src/ui/NewProject.java
@@ -52,6 +52,17 @@ private void addActionPerformed(ActionEvent e) {
                 }
                 break;
             case DOMAIN_MODE:
+                if(item.contains(",")){
+                    String[] arr = item.split(",");
+                    for (int i=0;i<arr.length;i++){
+                        if(!"".equals(arr[i]) && !BurpExtender.currentRootDomainSet.contains(arr[i]) && !list.contains(arr[i])){
+                            list.addElement(arr[i]);
+                            BurpExtender.currentRootDomainSet.add(arr[i]);
+                            BurpExtender.db.addRootDomain(BurpExtender.config.get("currentProject"), arr[i]);
+                        }
+                    }
+                    break;
+                }
                 if(!"".equals(item) && !BurpExtender.currentRootDomainSet.contains(item) && !list.contains(item)){
                     list.addElement(item);
                     BurpExtender.currentRootDomainSet.add(item);