Initialize new RWMutex when creating a SmContext

omec-project · Jan 3, 2024 · 082e74e · 082e74e
1 parent 2d503fa
commit 082e74e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/context/amf_ue.go b/context/amf_ue.go
@@ -905,6 +905,7 @@ func (ue *AmfUe) CopyDataFromUeContextModel(ueContext models.UeContext) {
 	if len(ueContext.SessionContextList) > 0 {
 		for _, pduSessionContext := range ueContext.SessionContextList {
 			smContext := SmContext{
+				Mu:              new(sync.RWMutex),
 				PduSessionIDVal: pduSessionContext.PduSessionId,
 				SmContextRefVal: pduSessionContext.SmContextRef,
 				SnssaiVal:       *pduSessionContext.SNssai,

diff --git a/context/sm_context.go b/context/sm_context.go
@@ -44,7 +44,10 @@ type SmContext struct {
 }
 
 func NewSmContext(pduSessionID int32) *SmContext {
-	c := &SmContext{PduSessionIDVal: pduSessionID}
+	c := &SmContext{
+		PduSessionIDVal: pduSessionID,
+		Mu:              new(sync.RWMutex),
+	}
 	return c
 }