开发|自动化渗透报告生成系统核心代码

<!-- Spring Expression Language (SpEL) --><dependency><groupId>org.springframework</groupId><artifactId>spring-expression</artifactId></dependency>

<dependency><groupId>e-iceblue</groupId><artifactId>spire.doc</artifactId><version>13.1.3</version></dependency>

<!-- poi-tl (基于 Apache POI 的模板引擎) --><dependency><groupId>com.deepoove</groupId><artifactId>poi-tl</artifactId><version>1.12.1</version><!-- Updated to the latest version --><exclusions><!-- Exclude conflicting dependencies --><exclusion><groupId>org.apache.poi</groupId><artifactId>poi-ooxml</artifactId></exclusion><exclusion><groupId>org.apache.poi</groupId><artifactId>poi</artifactId></exclusion><exclusion><groupId>org.apache.xmlbeans</groupId><artifactId>xmlbeans</artifactId></exclusion><exclusion><groupId>org.apache.xmlgraphics</groupId><artifactId>batik-transcoder</artifactId></exclusion></exclusions></dependency><!-- Apache POI (核心库) --><dependency><groupId>org.apache.poi</groupId><artifactId>poi</artifactId><version>5.2.5</version></dependency><!-- Apache POI (处理 .docx 文件的库) --><dependency><groupId>org.apache.poi</groupId><artifactId>poi-ooxml</artifactId><version>5.2.5</version></dependency><!-- Lombok (简化代码) --><dependency><groupId>org.projectlombok</groupId><artifactId>lombok</artifactId><version>1.18.30</version><scope>provided</scope></dependency><!--这里不进行依赖去除会有冲突执行会报错-》bug1-->

TableRenderData riskTable = new TableRenderData();TableStyle riskTableStyle = new TableStyle();riskTableStyle.setWidth("100%");riskTable.setTableStyle(riskTableStyle);//创建表头RowRenderData row0 = Rows.of("序号", "应用名称","风险说明", "漏洞等级", "修复情况").textColor("FFFFFF")                .bgColor("31322C").center().create();riskTable.addRow(row0);Integer highLevelNum = 0;Integer mediumLevelNum = 0;Integer lowLevelNum = 0;if (vulnerabilityEntityList.size() > 0){for (int i = 0; i < vulnerabilityEntityList.size(); i++) {       String levelName = "";if (vulnerabilityEntityList.get(i).getLevel() == 1){          levelName = "低危";          lowLevelNum += 1;       } elseif (vulnerabilityEntityList.get(i).getLevel() == 2) {          levelName = "中危";          mediumLevelNum += 1;       } elseif (vulnerabilityEntityList.get(i).getLevel() == 3) {          levelName = "高危";          highLevelNum += 1;       }       //生成每行       RowRenderData row = Rows.create(String.valueOf(i+1), wordInfo.getSystemName(), vulnerabilityEntityList.get(i).getVulnTitle(), levelName, "/");       riskTable.addRow(row);    }}wordInfo.setHighNum(highLevelNum);wordInfo.setMediumNum(mediumLevelNum);wordInfo.setLowNum(lowLevelNum);rawData.put("vulnTable", riskTable);

List<Map<String, Object>> dataList = new ArrayList<>();if (vulnerabilityEntityList.size() > 0){for (int i = 0; i < vulnerabilityEntityList.size(); i++) {       Map<String, Object> data = new HashMap<>();       data.put("title", vulnerabilityEntityList.get(i).getVulnTitle());       data.put("url", vulnerabilityEntityList.get(i).getVulnUrl());       data.put("description", vulnerabilityEntityList.get(i).getDescribe());       String levelName = "";       //这里可以换成枚举类，给自己写的小玩意我就没正常处理，弱智一点if (vulnerabilityEntityList.get(i).getLevel() == 1){          levelName = "低危";       } elseif (vulnerabilityEntityList.get(i).getLevel() == 2) {          levelName = "中危";       } elseif (vulnerabilityEntityList.get(i).getLevel() == 3) {           levelName = "高危";       }        data.put("risk", levelName);        data.put("content", wordTemplateService.DealVulnsContent(vulnerabilityEntityList.get(i).getVulnContent()));        data.put("recommendation", vulnerabilityEntityList.get(i).getFixAdvice());        dataList.add(data);    }}rawData.put("dataList", dataList);

publicclassWordUtil {publicstaticvoidupdateTOC(String outputPath) {Document doc = newDocument();        doc.loadFromFile(outputPath);// 更新文档中的所有字段（包括目录）        doc.updateTableOfContents();        // 保存更新后的文档        doc.saveToFile(outputPath, FileFormat.Docx);        restWord(outputPath);    }    public static void restWord(String docFilePath) {        try (FileInputStream in = new FileInputStream(docFilePath)) {            XWPFDocument doc = new XWPFDocument(OPCPackage.open(in));            List<XWPFParagraph> paragraphs = doc.getParagraphs();            if (paragraphs.size() < 1) {                return;            }            XWPFParagraph firstParagraph = paragraphs.get(0);            if (firstParagraph.getText().contains("Spire.Doc")) {                doc.removeBodyElement(doc.getPosOfParagraph(firstParagraph));            }            OutputStream out = new FileOutputStream(docFilePath);            doc.write(out);            out.close();        } catch (Exception e) {            e.printStackTrace();        }    }}