go/common: Add compute runtime policy sanity test

Author: martintomazic

go/common/node/sgx_test.go

@@ -4,6 +4,7 @@ import (
 	"encoding/hex"
 	"os"
 	"testing"
+	"time"
 
 	"github.com/stretchr/testify/require"
 
@@ -248,6 +249,55 @@ func TestHashAttestation(t *testing.T) {
 	require.EqualValues("9a288bd33ba7a4c2eefdee68e4c08c1a34c369302ef8176a3bfdb4fedcec333e", hex.EncodeToString(h))
 }
 
+// TestComputePolicySanity checks that attestation verification uses the stricter
+// compute policy when requested and falls back to the default policy otherwise.
+func TestComputePolicySanity(t *testing.T) {
+	require := require.New(t)
+
+	pcs.SetSkipVerify()
+	defer pcs.UnsetSkipVerify()
+
+	// Build a raw SGX report (384 bytes) with a known RAK hash in ReportData.
+	var rak signature.PublicKey
+	rakHash := HashRAK(rak)
+
+	var rawReport [384]byte
+	copy(rawReport[320:], rakHash[:])
+
+	mockQuote, err := pcs.NewMockQuote(rawReport[:])
+	require.NoError(err, "NewMockQuote")
+
+	sa := SGXAttestation{
+		Versioned: cbor.NewVersioned(LatestSGXAttestationVersion),
+		Quote: quote.Quote{
+			PCS: &pcs.QuoteBundle{
+				Quote: mockQuote,
+			},
+		},
+	}
+
+	sc := SGXConstraints{
+		Versioned: cbor.NewVersioned(1),
+		Enclaves:  []sgx.EnclaveIdentity{{}},
+		Policy: &quote.Policy{
+			PCS: &pcs.QuotePolicy{},
+		},
+		ComputePolicy: &quote.Policy{
+			PCS: &pcs.QuotePolicy{Disabled: true},
+		},
+	}
+
+	var nodeID signature.PublicKey
+	cfg := &TEEFeatures{SGX: TEEFeaturesSGX{PCS: true}}
+
+	err = sa.Verify(cfg, time.Now(), 0, &sc, rak, nil, nodeID, true)
+	require.Error(err, "attestation should be rejected when compute policy is used")
+	require.ErrorContains(err, "PCS quotes are disabled by policy")
+
+	err = sa.Verify(cfg, time.Now(), 0, &sc, rak, nil, nodeID, false)
+	require.NoError(err, "attestation should pass when falling back to default policy")
+}
+
 func FuzzSGXConstraints(f *testing.F) {
 	// Add some V0 constraints.
 	raw, err := os.ReadFile("testdata/sgx_constraints_v0.bin")

go/common/sgx/pcs/pcs.go

@@ -55,6 +55,12 @@ func SetSkipVerify() {
 	unsafeSkipVerify = true
 }
 
+// UnsetSkipVerify will enable quote signature verification for the remainder of the process'
+// lifetime.
+func UnsetSkipVerify() {
+	unsafeSkipVerify = false
+}
+
 // SetAllowDebugEnclaves will enable running and communicating with enclaves with debug flag enabled
 // in report body for the remainder of the process' lifetime.
 func SetAllowDebugEnclaves() {