From 1ed15ac50b96e7e3638bb16f1e9175e8ba1b1fcb Mon Sep 17 00:00:00 2001
From: TsMask <340112800@qq.com>
Date: Tue, 28 Nov 2023 11:41:02 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20UDM=E7=94=A8=E6=88=B7=E6=95=B0=E6=8D=AE?=
 =?UTF-8?q?=E5=AF=BC=E5=85=A5imsi=E7=A9=BA=E5=80=BC=E8=B7=B3=E8=BF=87?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/modules/network_element/service/udm_auth.impl.go | 6 ++++++
 src/modules/network_element/service/udm_sub.impl.go  | 6 ++++++
 2 files changed, 12 insertions(+)

diff --git a/src/modules/network_element/service/udm_auth.impl.go b/src/modules/network_element/service/udm_auth.impl.go
index ed7cc4ab..75a051a1 100644
--- a/src/modules/network_element/service/udm_auth.impl.go
+++ b/src/modules/network_element/service/udm_auth.impl.go
@@ -113,6 +113,9 @@ func (r *UDMAuthImpl) InsertCSV(neID string, data []map[string]string) int64 {
 	prefixes := make(map[string]struct{})
 	for _, v := range data {
 		imsi := v["imsi"]
+		if len(imsi) < 5 {
+			continue
+		}
 		prefix := imsi[:len(imsi)-4]
 		prefixes[prefix] = struct{}{}
 	}
@@ -133,6 +136,9 @@ func (r *UDMAuthImpl) InsertTxt(neID string, data [][]string) int64 {
 	prefixes := make(map[string]struct{})
 	for _, v := range data {
 		imsi := v[0]
+		if len(imsi) < 5 {
+			continue
+		}
 		prefix := imsi[:len(imsi)-4]
 		prefixes[prefix] = struct{}{}
 	}
diff --git a/src/modules/network_element/service/udm_sub.impl.go b/src/modules/network_element/service/udm_sub.impl.go
index 617a0a39..45fc5cd4 100644
--- a/src/modules/network_element/service/udm_sub.impl.go
+++ b/src/modules/network_element/service/udm_sub.impl.go
@@ -129,6 +129,9 @@ func (r *UDMSubImpl) InsertCSV(neID string, data []map[string]string) int64 {
 	prefixes := make(map[string]struct{})
 	for _, v := range data {
 		imsi := v["imsi"]
+		if len(imsi) < 5 {
+			continue
+		}
 		prefix := imsi[:len(imsi)-4]
 		prefixes[prefix] = struct{}{}
 	}
@@ -149,6 +152,9 @@ func (r *UDMSubImpl) InsertTxt(neID string, data [][]string) int64 {
 	prefixes := make(map[string]struct{})
 	for _, v := range data {
 		imsi := v[0]
+		if len(imsi) < 5 {
+			continue
+		}
 		prefix := imsi[:len(imsi)-4]
 		prefixes[prefix] = struct{}{}
 	}