Fix crash by invalid NSSAI (#15)

free5gc · Mar 20, 2022 · 18b1635 · 18b1635
1 parent 6700f6e
commit 18b1635
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/nasConvert/Nssai.go b/nasConvert/Nssai.go
@@ -34,6 +34,10 @@ func RequestedNssaiToModels(nasNssai *nasType.RequestedNSSAI) ([]models.MappingO
 func snssaiToModels(lengthOfSnssaiContents uint8, buf []byte) (models.MappingOfSnssai, error) {
 	snssai := models.MappingOfSnssai{}
 
+	if alen, elen := len(buf), int(lengthOfSnssaiContents)+1; alen < elen {
+		return snssai, fmt.Errorf("S-NSSAI contents is too short: expected = %d, actual = %d", elen, alen)
+	}
+
 	switch lengthOfSnssaiContents {
 	case 0x01: // SST
 		snssai.ServingSnssai = &models.Snssai{