添加了爬取cve编号的支持

Author: Kaze-2715

src/main/java/xin/ctkqiang/controller/DatabaseController.java

@@ -142,6 +142,7 @@ public void CreateTableIfNotExists() {
                 + "type VARCHAR(50),"
                 + "platform VARCHAR(50),"
                 + "created_at DATETIME DEFAULT CURRENT_TIMESTAMP,"
+                + "cve VARCHAR(20),"
                 + "PRIMARY KEY (id)"
                 + ");";
 
@@ -211,8 +212,8 @@ public int AddExploit(List<Exploit> exploit_list) {
         long SuccessCount = 0;
 
         // 准备参数化SQL语句
-        stringBuilder.append("INSERT INTO records (id, description, date, author, type, platform) ");
-        stringBuilder.append("VALUES (?, ?, ?, ?, ?, ?)");
+        stringBuilder.append("INSERT INTO records (id, description, date, author, type, platform, cve) ");
+        stringBuilder.append("VALUES (?, ?, ?, ?, ?, ?, ?)");
 
         // 确保输入列表不为空
         assert exploit_list != null;
@@ -235,6 +236,7 @@ public int AddExploit(List<Exploit> exploit_list) {
                     PSTMT.setString(4, exploit.getAuthor());
                     PSTMT.setString(5, exploit.getType());
                     PSTMT.setString(6, exploit.getPlatform());
+                    PSTMT.setString(7, exploit.getCve());
 
                     PSTMT.addBatch();
                 }
@@ -289,6 +291,7 @@ public List<Exploit> GetAllExploits() {
                     exploit.setAuthor(rs.getString("author"));
                     exploit.setType(rs.getString("type"));
                     exploit.setPlatform(rs.getString("platform"));
+                    exploit.setCve(rs.getString("cve"));
 
                     // 添加到结果列表
                     exploits.add(exploit);
@@ -403,13 +406,14 @@ public int ExportToSQL(List<Exploit> exploits) {
             // 为每条记录生成INSERT语句
             for (Exploit e : exploits) {
                 String sql = String.format(
-                        "INSERT INTO exploits (id, description, date, author, type, platform) VALUES (%d, '%s', '%s', '%s', '%s', '%s');\n",
+                        "INSERT INTO exploits (id, description, date, author, type, platform, cve) VALUES (%d, '%s', '%s', '%s', '%s', '%s', '%s');\n",
                         e.getId(),
                         EscapeSql(e.getDescription()),
                         e.getDate(),
                         EscapeSql(e.getAuthor()),
                         EscapeSql(e.getType()),
-                        EscapeSql(e.getPlatform()));
+                        EscapeSql(e.getPlatform()),
+                        EscapeSql(e.getCve()));
                 writer.write(sql);
             }
 
@@ -452,15 +456,16 @@ public int ExportToCSV(List<Exploit> exploits) {
 
             try (FileWriter Writer = new FileWriter(FilePath)) {
                 // 写入CSV表头
-                Writer.write("描述,日期,作者,类型,平台\n");
+                Writer.write("描述,日期,作者,类型,平台,CVE编号\n");
 
                 // 写入每条记录
                 for (Exploit e : exploits) {
                     Writer.write(CsvSafe(e.getDescription()) + "," +
                             CsvSafe(e.getDate()) + "," +
                             CsvSafe(e.getAuthor()) + "," +
                             CsvSafe(e.getType()) + "," +
-                            CsvSafe(e.getPlatform()) + "\n");
+                            CsvSafe(e.getPlatform()) + "," +
+                            CsvSafe(e.getCve()) + "\n");
                 }
 
                 System.out.println("✅ CSV 导出完成，保存路径：" + FilePath);

src/main/java/xin/ctkqiang/controller/ExploitDbController.java

@@ -144,6 +144,8 @@ public void Crawl(int pageSize, boolean isExport, String extension) {
                     JsonNode authorNode = Node.get("author");
                     JsonNode typeNode = Node.get("type");
                     JsonNode platformNode = Node.get("platform");
+                        // 添加获取CVE编号的节点
+                        JsonNode codeNode = Node.get("code");
 
                     // 解析描述信息（数组格式）
                     if (descriptionNode != null && descriptionNode.isArray() && descriptionNode.size() > 1) {
@@ -180,6 +182,23 @@ public void Crawl(int pageSize, boolean isExport, String extension) {
                         exploit.setPlatform("");
                     }
 
+                        // 解析CVE编号
+                        if (codeNode != null && codeNode.isArray()) {
+                            // 遍历code数组，查找code_type为"cve"的项
+                            for (JsonNode codeItem : codeNode) {
+                                JsonNode codeTypeNode = codeItem.get("code_type");
+                                if (codeTypeNode != null && "cve".equals(codeTypeNode.asText())) {
+                                    JsonNode cveCodeNode = codeItem.get("code");
+                                    if (cveCodeNode != null) {
+                                        exploit.setCve("CVE-" + cveCodeNode.asText());
+                                    } else {
+                                        exploit.setCve("N/A");
+                                    }
+                                    break;
+                                }
+                            }
+                        }
+
                     // 添加到结果列表
                     exploits.add(exploit);
                 }
@@ -281,14 +300,14 @@ public int ExportToSQL(List<Exploit> exploits) {
      */
     public static void PrintExploitTable(List<Exploit> exploits) {
         // 定义表格格式
-        String format = "| %-4s | %-40s | %-10s | %-15s | %-10s | %-10s |%n";
+        String format = "| %-2s | %-40s | %-10s | %-15s | %-10s | %-10s | %-20s |%n";
 
         // 打印表头
         System.out.println(
-                "+------+------------------------------------------+------------+-----------------+------------+------------+");
-        System.out.printf(format, "编号", "描述", "日期", "作者", "类型", "平台");
+                "+------+------------------------------------------+------------+-----------------+------------+------------+----------------------+");
+        System.out.printf(format, "编号", "描述", "日期", "作者", "类型", "平台", "CVE编号");
         System.out.println(
-                "+------+------------------------------------------+------------+-----------------+------------+------------+");
+                "+------+------------------------------------------+------------+-----------------+------------+------------+----------------------+");
 
         // 打印表格内容
         int index = 1;
@@ -299,14 +318,15 @@ public static void PrintExploitTable(List<Exploit> exploits) {
             String author = truncateAndPad(e.getAuthor().trim(), 15);
             String type = padRight(e.getType().trim(), 10);
             String platform = padRight(e.getPlatform().trim(), 10);
+            String cve = padRight(e.getCve().trim(), 20);
 
             // 打印一行数据
-            System.out.printf(format, index++, desc, date, author, type, platform);
+            System.out.printf(format, index++, desc, date, author, type, platform, cve);
         }
 
         // 打印表格底部
         System.out.println(
-                "+------+------------------------------------------+------------+-----------------+------------+------------+");
+                "+------+------------------------------------------+------------+-----------------+------------+------------+----------------------+");
     }
 
     /**

src/main/java/xin/ctkqiang/dto/Exploit.java

@@ -7,6 +7,7 @@ public class Exploit {
     private String author;
     private String type;
     private String platform;
+    private String cve;
 
     public String getId() {
         return id;
@@ -56,4 +57,12 @@ public void setPlatform(String platform) {
         this.platform = platform;
     }
 
+    public String getCve() {
+        return cve == null ? "N/A" : cve;
+    }
+
+    public void setCve(String cve) {
+        this.cve = cve;
+    }
+
 }