fix: handle type assertions safely (#268)

Signed-off-by: gatici <[email protected]>
omec-project · Dec 16, 2024 · b48c395 · b48c395
1 parent a79c2c2
commit b48c395
Showing 1 changed file with 14 additions and 4 deletions.
diff --git a/simue/handler.go b/simue/handler.go
@@ -181,10 +181,20 @@ func HandleDeregAcceptEvent(ue *simuectx.SimUe,
 	return nil
 }
 
-func HandlePduSessEstRequestEvent(ue *simuectx.SimUe,
-	intfcMsg common.InterfaceMessage,
-) (err error) {
-	msg := intfcMsg.(*common.UuMessage)
+func HandlePduSessEstRequestEvent(ue *simuectx.SimUe, intfcMsg common.InterfaceMessage) (err error) {
+	// Safe type assertions
+	if intfcMsg == nil {
+		err := fmt.Errorf("HandlePduSessEstRequestEvent: intfcMsg is nil")
+		ue.Log.Errorln(err)
+		return err
+	}
+
+	msg, ok := intfcMsg.(*common.UuMessage)
+	if !ok {
+		err := fmt.Errorf("HandlePduSessEstRequestEvent: expected *common.UuMessage, got %T", intfcMsg)
+		ue.Log.Errorln(err)
+		return err
+	}
 	msg.Event = common.UL_INFO_TRANSFER_EVENT
 	SendToGnbUe(ue, msg)
 	return nil