feat: UDM鉴权用户ki和opc掩码

features/udm_user/api_udm_user.go

@@ -13,6 +13,7 @@ import (
 	"ems.agt/lib/core/file"
 	mmlclient "ems.agt/lib/core/mml_client"
 	"ems.agt/lib/core/utils/ctx"
+	"ems.agt/lib/core/utils/parse"
 	"ems.agt/lib/core/vo/result"
 	"ems.agt/lib/dborm"
 	"ems.agt/lib/log"
@@ -209,6 +210,15 @@ func (s *UdmUserApi) UdmAuthUserList(w http.ResponseWriter, r *http.Request) {
 	querys := ctx.QueryMap(r)
 	querys["neId"] = ""
 	data := s.authUser.Page(querys)
+	// 遍历安全掩码
+	rows := data["rows"].([]model.UdmAuthUser)
+	maskRows := []model.UdmAuthUser{}
+	for _, v := range rows {
+		v.Ki = parse.SafeContent(v.Ki)
+		v.Opc = parse.SafeContent(v.Opc)
+		maskRows = append(maskRows, v)
+	}
+	data["rows"] = maskRows
 	ctx.JSON(w, 200, result.Ok(data))
 }
 
@@ -268,6 +278,8 @@ func (s *UdmUserApi) UdmAuthUserInfo(w http.ResponseWriter, r *http.Request) {
 
 		s.authUser.Insert(neId, userInfo)
 	}
+	userInfo.Ki = parse.SafeContent(userInfo.Ki)
+	userInfo.Opc = parse.SafeContent(userInfo.Opc)
 	ctx.JSON(w, 200, result.OkData(userInfo))
 
 }
@@ -505,7 +517,9 @@ func (s *UdmUserApi) UdmAuthUserExport(w http.ResponseWriter, r *http.Request) {
 		data := [][]string{}
 		data = append(data, []string{"imsi", "ki", "amf", "algo", "opc"})
 		for _, v := range list {
-			data = append(data, []string{v.Imsi, v.Ki, v.Amf, v.AlgoIndex, v.Opc})
+			maskKi := parse.SafeContent(v.Ki)
+			maskOpc := parse.SafeContent(v.Opc)
+			data = append(data, []string{v.Imsi, maskKi, v.Amf, v.AlgoIndex, maskOpc})
 		}
 		// 输出到文件
 		err := file.WriterCSVFile(data, filePath)
@@ -519,7 +533,9 @@ func (s *UdmUserApi) UdmAuthUserExport(w http.ResponseWriter, r *http.Request) {
 		// 转换数据
 		data := [][]string{}
 		for _, v := range list {
-			data = append(data, []string{v.Imsi, v.Ki, v.Amf, v.AlgoIndex, v.Opc})
+			maskKi := parse.SafeContent(v.Ki)
+			maskOpc := parse.SafeContent(v.Opc)
+			data = append(data, []string{v.Imsi, maskKi, v.Amf, v.AlgoIndex, maskOpc})
 		}
 		// 输出到文件
 		err = file.WriterTxtFile(data, filePath)