424 lines
15 KiB
Go
424 lines
15 KiB
Go
package nessus
|
||
|
||
import (
|
||
"dt_automate/tool"
|
||
"encoding/json"
|
||
"fmt"
|
||
"log"
|
||
"os"
|
||
"time"
|
||
|
||
"github.com/Esword618/unioffice/color"
|
||
"github.com/Esword618/unioffice/document"
|
||
"github.com/Esword618/unioffice/measurement"
|
||
"github.com/Esword618/unioffice/schema/soo/wml"
|
||
"github.com/gocarina/gocsv"
|
||
|
||
"github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/common"
|
||
"github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/common/errors"
|
||
"github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/common/profile"
|
||
tmt "github.com/tencentcloud/tencentcloud-sdk-go/tencentcloud/tmt/v20180321"
|
||
)
|
||
|
||
// A1,数据
|
||
func CSV_damo() {
|
||
//读取docx模板
|
||
doc, err := document.Open("temp/lousao_temp.docx")
|
||
if err != nil {
|
||
log.Panic("打开文件失败", err)
|
||
}
|
||
// 读取CSV文件示例
|
||
file, err := os.OpenFile("temp/zww_to5b2r.csv", os.O_RDWR|os.O_CREATE, os.ModePerm)
|
||
if err != nil {
|
||
fmt.Println("打开文件失败:", err)
|
||
return
|
||
}
|
||
log.Println("打开数据模板")
|
||
defer file.Close()
|
||
|
||
// var people []interface{}
|
||
var people []*Person
|
||
if err := gocsv.UnmarshalFile(file, &people); err != nil {
|
||
fmt.Println("读取文件失败:", err)
|
||
return
|
||
}
|
||
log.Println("开始将数据填入报告模板")
|
||
|
||
// table := doc.Tables()[3]
|
||
// var host string
|
||
// var con int
|
||
// var tables []document.Table
|
||
// for _, csvstr := range people {
|
||
// if csvstr.Host == host {
|
||
// row := table.AddRow()
|
||
// row.Properties().SetHeight(22, wml.ST_HeightRuleExact)
|
||
// // text_s(row.AddCell(), csvstr.Host) //主机IP
|
||
// text_s(row.AddCell(), Translate(csvstr.Name)) //风险项
|
||
// if csvstr.Risk == "None" {
|
||
// text_s(row.AddCell(), "无") //危险程度
|
||
// } else {
|
||
// text_s(row.AddCell(), Translate(csvstr.Risk)) //危险程度
|
||
// }
|
||
|
||
// if csvstr.Risk_Factor == "None" {
|
||
// text_s(row.AddCell(), "无") //危险因素
|
||
// } else {
|
||
// text_s(row.AddCell(), Translate(csvstr.Risk_Factor)) //危险因素
|
||
// }
|
||
// text_s(row.AddCell(), Translate(csvstr.Synopsisk)) //摘要
|
||
// text_s(row.AddCell(), Translate(csvstr.Description)) //描述
|
||
// if csvstr.Solution == "n/a" {
|
||
// text_s(row.AddCell(), "无") //解决方案
|
||
// } else {
|
||
// text_s(row.AddCell(), Translate(csvstr.Solution)) //解决方案
|
||
// }
|
||
|
||
// } else {
|
||
// con++
|
||
// doc.Tables()[con] = doc.Tables()[3]
|
||
// table = doc.Tables()[con]
|
||
// row := table.AddRow()
|
||
// row.Properties().SetHeight(22, wml.ST_HeightRuleExact)
|
||
// // text_s(row.AddCell(), csvstr.Host) //主机IP
|
||
// text_s(row.AddCell(), Translate(csvstr.Name)) //风险项
|
||
// if csvstr.Risk == "None" {
|
||
// text_s(row.AddCell(), "无") //危险程度
|
||
// } else {
|
||
// text_s(row.AddCell(), Translate(csvstr.Risk)) //危险程度
|
||
// }
|
||
|
||
// if csvstr.Risk_Factor == "None" {
|
||
// text_s(row.AddCell(), "无") //危险因素
|
||
// } else {
|
||
// text_s(row.AddCell(), Translate(csvstr.Risk_Factor)) //危险因素
|
||
// }
|
||
// text_s(row.AddCell(), Translate(csvstr.Synopsisk)) //摘要
|
||
// text_s(row.AddCell(), Translate(csvstr.Description)) //描述
|
||
// if csvstr.Solution == "n/a" {
|
||
// text_s(row.AddCell(), "无") //解决方案
|
||
// } else {
|
||
// text_s(row.AddCell(), Translate(csvstr.Solution)) //解决方案
|
||
// }
|
||
// }
|
||
// host = csvstr.Host
|
||
// }
|
||
//根据ip创建表格
|
||
// 定义一个 map 用于记录已经出现过的 IP 地址
|
||
var ip []string
|
||
ipMap := make(map[string]bool)
|
||
for _, csvstr := range people {
|
||
// 检查当前 IP 是否已经在 map 中
|
||
if _, exists := ipMap[csvstr.Host]; !exists {
|
||
// 如果不存在，则将其添加到数组和 map 中
|
||
ip = append(ip, csvstr.Host)
|
||
ipMap[csvstr.Host] = true
|
||
}
|
||
}
|
||
log.Println(ip)
|
||
for _, ip := range ip {
|
||
//创建一个段落添加IP：
|
||
// 创建一个段落
|
||
paragraph := doc.AddParagraph()
|
||
// 设置段落样式
|
||
paragraph.Properties().SetStyle("Normal")
|
||
// 添加文本
|
||
run := paragraph.AddRun()
|
||
run.AddText("IP：" + ip)
|
||
// 创建表格
|
||
table := doc.AddTable()
|
||
// 设置表格宽度
|
||
table.Properties().SetWidthPercent(100)
|
||
// 设置表格边框
|
||
borderProps := table.Properties().Borders()
|
||
borderProps.SetTop(wml.ST_BorderSingle, color.Black, measurement.Point)
|
||
borderProps.SetBottom(wml.ST_BorderSingle, color.Black, measurement.Point)
|
||
borderProps.SetLeft(wml.ST_BorderSingle, color.Black, measurement.Point)
|
||
borderProps.SetRight(wml.ST_BorderSingle, color.Black, measurement.Point)
|
||
|
||
// 添加表头
|
||
headerRow := table.AddRow()
|
||
headerRow.Properties().SetHeight(22, wml.ST_HeightRuleExact)
|
||
|
||
text_s(headerRow.AddCell(), "风险项")
|
||
text_s(headerRow.AddCell(), "危险程度")
|
||
text_s(headerRow.AddCell(), "危险因素")
|
||
text_s(headerRow.AddCell(), "摘要")
|
||
text_s(headerRow.AddCell(), "描述")
|
||
text_s(headerRow.AddCell(), "解决方案")
|
||
// 设置表头单元格的边框（包含内边框）
|
||
for _, cell := range headerRow.Cells() {
|
||
borderProps := cell.Properties().Borders()
|
||
borderProps.SetTop(wml.ST_BorderSingle, color.Black, measurement.Point)
|
||
borderProps.SetBottom(wml.ST_BorderSingle, color.Black, measurement.Point)
|
||
borderProps.SetLeft(wml.ST_BorderSingle, color.Black, measurement.Point)
|
||
borderProps.SetRight(wml.ST_BorderSingle, color.Black, measurement.Point)
|
||
}
|
||
// 添加数据行
|
||
for _, csvstr := range people {
|
||
if csvstr.Host == ip {
|
||
row := table.AddRow()
|
||
row.Properties().SetHeight(22, wml.ST_HeightRuleExact)
|
||
// text_s(row.AddCell(), csvstr.Host) //主机IP
|
||
text_s(row.AddCell(), csvstr.Name) //风险项
|
||
if csvstr.Risk == "None" {
|
||
text_s(row.AddCell(), "无") //危险程度
|
||
} else {
|
||
text_s(row.AddCell(), csvstr.Risk) //危险程度
|
||
}
|
||
if csvstr.Risk_Factor == "None" {
|
||
text_s(row.AddCell(), "无") //危险因素
|
||
} else {
|
||
text_s(row.AddCell(), csvstr.Risk_Factor) //危险因素
|
||
}
|
||
text_s(row.AddCell(), csvstr.Synopsisk) //摘要
|
||
text_s(row.AddCell(), csvstr.Description) //描述
|
||
if csvstr.Solution == "n/a" {
|
||
text_s(row.AddCell(), "无") //解决方案
|
||
} else {
|
||
text_s(row.AddCell(), csvstr.Solution) //解决方案
|
||
}
|
||
// 设置单元格的边框（包含内边框）
|
||
for _, cell := range row.Cells() {
|
||
borderProps := cell.Properties().Borders()
|
||
borderProps.SetTop(wml.ST_BorderSingle, color.Black, measurement.Point)
|
||
borderProps.SetBottom(wml.ST_BorderSingle, color.Black, measurement.Point)
|
||
borderProps.SetLeft(wml.ST_BorderSingle, color.Black, measurement.Point)
|
||
borderProps.SetRight(wml.ST_BorderSingle, color.Black, measurement.Point)
|
||
}
|
||
}
|
||
}
|
||
|
||
//添加一个空行
|
||
paragraph = doc.AddParagraph()
|
||
paragraph.Properties().SetStyle("Normal")
|
||
run = paragraph.AddRun()
|
||
run.AddBreak()
|
||
}
|
||
// for _, csvstr := range people {
|
||
|
||
// if csvstr.Host == host {
|
||
// row := table.AddRow()
|
||
// row.Properties().SetHeight(22, wml.ST_HeightRuleExact)
|
||
// // text_s(row.AddCell(), csvstr.Host) //主机IP
|
||
// text_s(row.AddCell(), csvstr.Name) //风险项
|
||
// if csvstr.Risk == "None" {
|
||
// text_s(row.AddCell(), "无") //危险程度
|
||
// } else {
|
||
// text_s(row.AddCell(), csvstr.Risk) //危险程度
|
||
// }
|
||
|
||
// if csvstr.Risk_Factor == "None" {
|
||
// text_s(row.AddCell(), "无") //危险因素
|
||
// } else {
|
||
// text_s(row.AddCell(), csvstr.Risk_Factor) //危险因素
|
||
// }
|
||
// text_s(row.AddCell(), csvstr.Synopsisk) //摘要
|
||
// text_s(row.AddCell(), csvstr.Description) //描述
|
||
// if csvstr.Solution == "n/a" {
|
||
// text_s(row.AddCell(), "无") //解决方案
|
||
// } else {
|
||
// text_s(row.AddCell(), csvstr.Solution) //解决方案
|
||
// }
|
||
|
||
// } else {
|
||
// con++
|
||
// tables[con] = doc.Tables()[3]
|
||
// table = tables[con]
|
||
// row := table.AddRow()
|
||
// row.Properties().SetHeight(22, wml.ST_HeightRuleExact)
|
||
// // text_s(row.AddCell(), csvstr.Host) //主机IP
|
||
// text_s(row.AddCell(), csvstr.Name) //风险项
|
||
// if csvstr.Risk == "None" {
|
||
// text_s(row.AddCell(), "无") //危险程度
|
||
// } else {
|
||
// text_s(row.AddCell(), csvstr.Risk) //危险程度
|
||
// }
|
||
|
||
// if csvstr.Risk_Factor == "None" {
|
||
// text_s(row.AddCell(), "无") //危险因素
|
||
// } else {
|
||
// text_s(row.AddCell(), csvstr.Risk_Factor) //危险因素
|
||
// }
|
||
// text_s(row.AddCell(), csvstr.Synopsisk) //摘要
|
||
// text_s(row.AddCell(), csvstr.Description) //描述
|
||
// if csvstr.Solution == "n/a" {
|
||
// text_s(row.AddCell(), "无") //解决方案
|
||
// } else {
|
||
// text_s(row.AddCell(), csvstr.Solution) //解决方案
|
||
// }
|
||
// }
|
||
// host = csvstr.Host
|
||
// }
|
||
// cvs_word()
|
||
//写入CSV
|
||
// file, err := os.OpenFile("output.csv", os.O_RDWR|os.O_CREATE, os.ModePerm)
|
||
// if err != nil {
|
||
// fmt.Println("创建文件失败:", err)
|
||
// return
|
||
// }
|
||
// defer file.Close()
|
||
|
||
// people := []*Person{
|
||
// {Name: "Alice", Age: 21, Gender: "F"},
|
||
// {Name: "Bob", Age: 25, Gender: "M"},
|
||
// }
|
||
// if err := gocsv.MarshalFile(&people, file); err != nil {
|
||
// fmt.Println("写入文件失败:", err)
|
||
// return
|
||
// }
|
||
doc.SaveToFile("大同信创云平台漏扫报告" + time.Unix(tool.Timestamp("second"), 0).Format("20060102") + ".docx")
|
||
}
|
||
|
||
// csv表结构体对应字段
|
||
type Person struct {
|
||
Plugin_ID string `csv:"Plugin ID"`
|
||
CVE string `csv:"CVE"`
|
||
CVSS string `csv:"CVSS"`
|
||
Risk string `csv:"Risk"`
|
||
Host string `csv:"Host"`
|
||
Protocol string `csv:"Protocol"`
|
||
Port string `csv:"Port"`
|
||
Name string `csv:"Name"`
|
||
Synopsisk string `csv:"Synopsis"`
|
||
Description string `csv:"Description"`
|
||
Solution string `csv:"Solution"`
|
||
See_Also string `csv:"See Also"`
|
||
Plugin_Output string `csv:"Plugin Output"`
|
||
STIG_Severity string `csv:"STIG Severity"`
|
||
CVSS_Base_Score string `csv:"CVSS v3.0 Base Score"`
|
||
CVSS_Temporal_Score string `csv:"CVSS v3.0 Temporal Score"`
|
||
Risk_Factor string `csv:"Risk Factor"`
|
||
BID string `csv:"BID"`
|
||
XREF string `csv:"XREF"`
|
||
MSKB string `csv:"MSKB"`
|
||
Plugin_Publication_Date CustomDate `csv:"Plugin Publication Date,layout=2006/01/02"`
|
||
Plugin_Modification_Date CustomDate `csv:"Plugin Modification Date,layout=2006/01/02"`
|
||
Metasploit string `csv:"Metasploit"`
|
||
Core_Impact string `csv:"Core Impact"`
|
||
CANVAS string `csv:"CANVAS"`
|
||
}
|
||
|
||
// 定义一个自定义日期类型
|
||
type CustomDate struct {
|
||
time.Time
|
||
}
|
||
|
||
// 实现TypeUnmarshaller接口，用于将CSV字符串转换为time.Time
|
||
func (cd *CustomDate) UnmarshalCSV(csv string) error {
|
||
date, err := time.Parse("2006/01/02", csv)
|
||
if err != nil {
|
||
return fmt.Errorf("invalid date format: %s", csv)
|
||
}
|
||
cd.Time = date
|
||
return nil
|
||
}
|
||
|
||
// 批量填入表格内容方法
|
||
func text_s(cell document.Cell, ss string) {
|
||
// 设置单元格垂直居中对齐
|
||
// cell.Properties().SetVerticalAlignment(wml.ST_VerticalJcCenter)
|
||
//在单元格中添加段落
|
||
p := cell.AddParagraph().AddRun()
|
||
p.Properties().SetFontFamily("微软雅黑")
|
||
p.Properties().SetSize(12)
|
||
// p.Properties().SetBold(true)
|
||
p.AddText(ss)
|
||
}
|
||
|
||
func Translate(text string) string {
|
||
// 实例化一个认证对象，入参需要传入腾讯云账户 SecretId 和 SecretKey，此处还需注意密钥对的保密
|
||
// 代码泄露可能会导致 SecretId 和 SecretKey 泄露，并威胁账号下所有资源的安全性。以下代码示例仅供参考，建议采用更安全的方式来使用密钥，请参见：https://cloud.tencent.com/document/product/1278/85305
|
||
// 密钥可前往官网控制台 https://console.cloud.tencent.com/cam/capi 进行获取
|
||
credential := common.NewCredential(
|
||
"AKIDgN8k9LNCpGUVJtFNI0UkmbDblhQvCLL8",
|
||
"d2GI7lFXcWGoNjvtd9SXP4XmnaEgz5ye",
|
||
)
|
||
// 实例化一个client选项，可选的，没有特殊需求可以跳过
|
||
cpf := profile.NewClientProfile()
|
||
cpf.HttpProfile.Endpoint = "tmt.tencentcloudapi.com"
|
||
// 实例化要请求产品的client对象,clientProfile是可选的
|
||
client, _ := tmt.NewClient(credential, "ap-beijing", cpf)
|
||
|
||
// 实例化一个请求对象,每个接口都会对应一个request对象
|
||
request := tmt.NewTextTranslateRequest()
|
||
|
||
request.SourceText = common.StringPtr(text)
|
||
request.Source = common.StringPtr("en")
|
||
request.Target = common.StringPtr("zh")
|
||
request.ProjectId = common.Int64Ptr(0)
|
||
|
||
// 返回的resp是一个TextTranslateResponse的实例，与请求对象对应
|
||
response, err := client.TextTranslate(request)
|
||
if _, ok := err.(*errors.TencentCloudSDKError); ok {
|
||
fmt.Printf("An API error has returned: %s", err)
|
||
}
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
// 解析 OutputJSON
|
||
var outputJSON map[string]interface{}
|
||
err = json.Unmarshal([]byte(response.ToJsonString()), &outputJSON)
|
||
if err != nil {
|
||
log.Fatalf("解析 OutputJSON 失败: %v", err)
|
||
}
|
||
return outputJSON["Response"].(map[string]interface{})["TargetText"].(string)
|
||
}
|
||
func cvs_word() {
|
||
doc, err := document.Open("temp/lousao_temp.docx")
|
||
if err != nil {
|
||
log.Panic("打开文件失败", err)
|
||
}
|
||
// // defer doc.Close()
|
||
// // 填充模板中的变量
|
||
// paragraphs := []document.Paragraph{}
|
||
// for _, p := range doc.Paragraphs() {
|
||
// paragraphs = append(paragraphs, p)
|
||
// }
|
||
// //此示例文档使用了不常见的结构化文档标记
|
||
// //文档模板中的除外。通常情况下，您只需迭代
|
||
// //文档的段落。
|
||
// for _, sdt := range doc.StructuredDocumentTags() {
|
||
// for _, p := range sdt.Paragraphs() {
|
||
// paragraphs = append(paragraphs, p)
|
||
// }
|
||
// }
|
||
log.Println("开始生成漏扫报告模板")
|
||
// for _, p := range paragraphs {
|
||
// for _, r := range p.Runs() {
|
||
// if strings.Contains(r.Text(), "{{DATE_A}}") {
|
||
// Replacetext(r, "{{DATE_A}}", time.Unix(tool.Timestamp("second"), 0).Format("2006年01月02日"))
|
||
// } else if strings.Contains(r.Text(), "{{HLW_FW_1}}") { //插入互联网防火墙图片
|
||
// Replaceimg(r, doc, "img/hw_fw1.png")
|
||
// } else if strings.Contains(r.Text(), "{{HLW_AFC2000_1}}") { //插入互联网流量清洗图片
|
||
// Replaceimg(r, doc, "img/afc2000_1.png")
|
||
// } else if strings.Contains(r.Text(), "{{HLW_AFC2000_2}}") { //插入互联网流量清洗图片
|
||
// Replaceimg(r, doc, "img/afc2000_2.png")
|
||
// } else if strings.Contains(r.Text(), "{{HLW_TSGZ}}") { //插入互联网态势感知图片
|
||
// Replaceimg(r, doc, "img/hw_tsgz.png")
|
||
// } else if strings.Contains(r.Text(), "{{HLW_BLJ}}") { //插入互联网堡垒机图片
|
||
// Replaceimg(r, doc, "img/hw_baolj.png")
|
||
// } else if strings.Contains(r.Text(), "{{ZWW_FW_1}}") { //插入政务网防火墙图片
|
||
// Replaceimg(r, doc, "img/zww_fw1.png")
|
||
// } else if strings.Contains(r.Text(), "{{ZWW_TSGZ}}") { //插入政务网态势感知图片
|
||
// Replaceimg(r, doc, "img/zww_tsgz.png")
|
||
// } else if strings.Contains(r.Text(), "{{ZWW_BLJ}}") { //插入政务网堡垒机图片
|
||
// Replaceimg(r, doc, "img/zww_baolj.png")
|
||
// } else {
|
||
// // log.Println("")
|
||
// }
|
||
// }
|
||
// }
|
||
// 遍历文档中的表格，替换{{bianliang}} 中的内容
|
||
table := doc.Tables()[3]
|
||
row := table.AddRow()
|
||
row.Properties().SetHeight(22, wml.ST_HeightRuleExact)
|
||
text_s(row.AddCell(), "192.168.1.1")
|
||
text_s(row.AddCell(), "高危")
|
||
|
||
row = table.AddRow()
|
||
row.Properties().SetHeight(22, wml.ST_HeightRuleExact)
|
||
text_s(row.AddCell(), "192.168.1.2")
|
||
text_s(row.AddCell(), "中危")
|
||
doc.SaveToFile("大同信创云平台漏扫报告" + time.Unix(tool.Timestamp("second"), 0).Format("20060102") + ".docx")
|
||
}
|