From 5a6ab019b80a652522ac8e321550c064d28a2889 Mon Sep 17 00:00:00 2001
From: yangwei999 <348134071@qq.com>
Date: Tue, 15 Apr 2025 15:11:24 +0800
Subject: [PATCH] add cve source

---
 cve-vulner-manager/common/common.go      | 1 +
 cve-vulner-manager/controllers/upload.go | 2 ++
 cve-vulner-manager/models/modeldb.go     | 2 ++
 3 files changed, 5 insertions(+)

diff --git a/cve-vulner-manager/common/common.go b/cve-vulner-manager/common/common.go
index 27b74f3..a9ac833 100644
--- a/cve-vulner-manager/common/common.go
+++ b/cve-vulner-manager/common/common.go
@@ -346,6 +346,7 @@ type CveFixSuggest struct {
 type CveOriginData struct {
 	Ids            string               `json:"ids"`
 	CveNum         string               `json:"cveNum"`
+	Source         string               `json:"source"`
 	UpdateType     string               `json:"updateType"`
 	CvePackName    []string             `json:"cvePackName"`
 	PackName       []string             `json:"packName"`
diff --git a/cve-vulner-manager/controllers/upload.go b/cve-vulner-manager/controllers/upload.go
index 43965a2..9399b9b 100644
--- a/cve-vulner-manager/controllers/upload.go
+++ b/cve-vulner-manager/controllers/upload.go
@@ -299,6 +299,7 @@ func AddOrgUpstream(source int, CveDataDict common.CveOriginData) (ResDataList [
 		cveNum = common.DeletePreAndSufSpace(cveNum)
 	}
 	orCve.Source = source
+	orCve.CveSource = CveDataDict.Source
 	orCve.CveNum = cveNum
 	orCve.Version = version
 	orCve.UpdateType = updateType
@@ -534,6 +535,7 @@ func AddOrgUpstreamRecord(source int, CveDataDict common.CveOriginData) {
 		orCve.AffectUpdate = 1
 	}
 
+	orCve.CveSource = CveDataDict.Source
 	err := models.InsertOriginCveRecord(&orCve)
 	if err != nil {
 		logs.Error("InsertOriginCveRecord, err: ", err)
diff --git a/cve-vulner-manager/models/modeldb.go b/cve-vulner-manager/models/modeldb.go
index b92d869..689b840 100644
--- a/cve-vulner-manager/models/modeldb.go
+++ b/cve-vulner-manager/models/modeldb.go
@@ -336,6 +336,7 @@ type OriginUpstream struct {
 	CveId         int64  `orm:"pk;auto;column(cve_id)"`
 	Ids           string `orm:"size(256);column(cve_un_ids);unique" description:" 唯一编号,根据此字段去重数据, 唯一识别码，可以填cve编号"`
 	CveNum        string `orm:"size(256);column(cve_num);index" description:"cve编号"`
+	CveSource     string `orm:"size(256);column(cve_source)" description:"cve来源"`
 	UpdateType    string `orm:"size(32);column(update_type);" description:"数据上传类型：insert, update, delete"`
 	CvePackName   string `orm:"size(512);column(cve_packname);null" description:"Cve在上游对应的包名"`
 	PackName      string `orm:"size(512);column(git_packname);index;null" description:"Cve对应的openEuler包名称(或者影响的包名)"`
@@ -996,6 +997,7 @@ type OriginUpstreamRecord struct {
 	CveId         int64  `orm:"pk;auto;column(cve_record_id)"`
 	Ids           string `orm:"size(256);column(cve_un_ids)" description:" 唯一编号,根据此字段去重数据, 唯一识别码，可以填cve编号"`
 	CveNum        string `orm:"size(256);column(cve_num);index" description:"cve编号"`
+	CveSource     string `orm:"size(256);column(cve_source)" description:"cve来源"`
 	UpdateType    string `orm:"size(32);column(update_type);" description:"数据上传类型：insert, update, delete"`
 	CvePackName   string `orm:"size(512);column(cve_packname);null" description:"Cve在上游对应的包名"`
 	PackName      string `orm:"size(512);column(git_packname);index;null" description:"Cve对应的openEuler包名称(或者影响的包名)"`
-- 
Gitee