Verify cause status for association response (#59)

omec-project · Mar 17, 2022 · 6be4ce2 · 6be4ce2
1 parent b303ec1
commit 6be4ce2
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/pkg/pfcpsim/pfcpsim.go b/pkg/pfcpsim/pfcpsim.go
@@ -317,7 +317,17 @@ func (c *PFCPClient) SetupAssociation() error {
 		return err
 	}
 
-	if _, ok := resp.(*message.AssociationSetupResponse); !ok {
+	assocResp, ok := resp.(*message.AssociationSetupResponse)
+	if !ok {
+		return NewInvalidResponseError()
+	}
+
+	cause, err := assocResp.Cause.Cause()
+	if err != nil {
+		return NewInvalidResponseError(err)
+	}
+
+	if cause != ieLib.CauseRequestAccepted {
 		return NewInvalidResponseError()
 	}