fix: check capability type

smartcontractkit · Nov 11, 2024 · 6535231 · 6535231
1 parent 422e385
commit 6535231
Show file tree

Hide file tree

Showing 2 changed files with 343 additions and 114 deletions.
diff --git a/core/capabilities/launcher.go b/core/capabilities/launcher.go
@@ -396,9 +396,14 @@ func (w *launcher) exposeCapabilities(ctx context.Context, myPeerID p2ptypes.Pee
 		switch capability.CapabilityType {
 		case capabilities.CapabilityTypeTrigger:
 			newTriggerPublisher := func(cap capabilities.BaseCapability, info capabilities.CapabilityInfo) (remotetypes.ReceiverService, error) {
+				triggerCapability, ok := (cap).(capabilities.TriggerCapability)
+				if !ok {
+					return nil, fmt.Errorf("capability does not implement TriggerCapability")
+				}
+
 				publisher := remote.NewTriggerPublisher(
 					capabilityConfig.RemoteTriggerConfig,
-					cap.(capabilities.TriggerCapability),
+					triggerCapability,
 					info,
 					don.DON,
 					idsToDONs,
@@ -419,10 +424,15 @@ func (w *launcher) exposeCapabilities(ctx context.Context, myPeerID p2ptypes.Pee
 			w.lggr.Warn("no remote client configured for capability type consensus, skipping configuration")
 		case capabilities.CapabilityTypeTarget:
 			newTargetServer := func(cap capabilities.BaseCapability, info capabilities.CapabilityInfo) (remotetypes.ReceiverService, error) {
+				targetCapability, ok := (cap).(capabilities.TargetCapability)
+				if !ok {
+					return nil, fmt.Errorf("capability does not implement TargetCapability")
+				}
+
 				return target.NewServer(
 					capabilityConfig.RemoteTargetConfig,
 					myPeerID,
-					cap.(capabilities.TargetCapability),
+					targetCapability,
 					info,
 					don.DON,
 					idsToDONs,