Skip to content

Commit a7971c5

Browse files
committed
Adjusts overriding logic for multivalidator
1 parent 708d3d4 commit a7971c5

File tree

2 files changed

+8
-0
lines changed

2 files changed

+8
-0
lines changed

azure-tcbinfo-override/override.go

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@ package azure_tcbinfo_override
33
import (
44
"github.com/flashbots/cvm-reverse-proxy/internal/atls"
55
azure_tdx "github.com/flashbots/cvm-reverse-proxy/internal/attestation/azure/tdx"
6+
"github.com/flashbots/cvm-reverse-proxy/proxy"
67

78
"github.com/google/go-tdx-guest/pcs"
89
)
@@ -23,6 +24,9 @@ func OverrideV6InstanceOutdatedSEAMLoader(tcbInfo pcs.TcbInfo) pcs.TcbInfo {
2324

2425
func OverrideAzureValidatorsForV6SEAMLoader(validators []atls.Validator) {
2526
for _, validator := range validators {
27+
if multiValidator, ok := validator.(*proxy.MultiValidator); ok {
28+
OverrideAzureValidatorsForV6SEAMLoader(multiValidator.Validators())
29+
}
2630
if azureTdxValidator, ok := validator.(*azure_tdx.Validator); ok {
2731
azureTdxValidator.SetTcbOverride(OverrideV6InstanceOutdatedSEAMLoader)
2832
}

proxy/mutli_validator.go

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,10 @@ func NewMultiValidator(validators []atls.Validator) *MultiValidator {
2727
}
2828
}
2929

30+
func (v *MultiValidator) Validators() []atls.Validator {
31+
return v.validators
32+
}
33+
3034
func (v *MultiValidator) OID() asn1.ObjectIdentifier {
3135
return v.oid
3236
}

0 commit comments

Comments
 (0)