diff --git a/features/udm_user/api_udm_user.go b/features/udm_user/api_udm_user.go index b820cb1b..e034acb4 100644 --- a/features/udm_user/api_udm_user.go +++ b/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)