feat: implement first phase of TDX Measurement Extensions

comid/digests_test.go

@@ -76,7 +76,6 @@ func TestDigests_AddDigest_OK(t *testing.T) {
 		assert.Nil(t, d.Valid())
 	}
 }
-
 func TestDigests_Valid_empty(t *testing.T) {
 	d := NewDigests()
 	require.NotNil(t, d)
@@ -89,7 +88,6 @@ func TestDigests_Valid_empty(t *testing.T) {
 
 	assert.EqualError(t, d.Valid(), "digest at index 0: unknown hash algorithm 666")
 }
-
 func TestDigests_AddDigest_unknown_algo(t *testing.T) {
 	d := NewDigests()
 	require.NotNil(t, d)

comid/tdx-profile/example_pce_refval_test.go

@@ -0,0 +1,339 @@
+// Copyright 2025 Contributors to the Veraison project.
+// SPDX-License-Identifier: Apache-2.0
+
+package tdx
+
+import (
+	_ "embed"
+	"fmt"
+
+	"github.com/veraison/corim/comid"
+	"github.com/veraison/corim/corim"
+	"github.com/veraison/eat"
+)
+
+// Example_decode_PCE_JSON decodes the TDX Provisioning Certification Enclave Measurement Extensions from the given JSON Template
+func Example_decode_PCE_JSON() {
+	profileID, err := eat.NewProfile("2.16.840.1.113741.1.16.1")
+	if err != nil {
+		panic(err) // will not error, as the hard-coded string above is valid
+	}
+	manifest, found := corim.GetProfileManifest(profileID)
+	if !found {
+		fmt.Printf("CoRIM Profile NOT FOUND")
+		return
+	}
+
+	m := manifest.GetComid()
+	if err := m.FromJSON([]byte(TDXPCERefValTemplate)); err != nil {
+		panic(err)
+	}
+
+	if err := m.Valid(); err != nil {
+		panic(err)
+	}
+
+	if err := extractPCERefVals(m); err != nil {
+		panic(err)
+	}
+
+	// output:
+	// OID: 2.16.840.1.113741.1.2.3.4.6
+	// Vendor: Intel Corporation
+	// Model: 0123456789ABCDEF
+	// InstanceID: 11
+	// pceID: 0000
+	// SVN[0]: 10
+	// SVN[1]: 10
+	// SVN[2]: 2
+	// SVN[3]: 2
+	// SVN[4]: 2
+	// SVN[5]: 1
+	// SVN[6]: 4
+	// SVN[7]: 0
+	// SVN[8]: 0
+	// SVN[9]: 0
+	// SVN[10]: 0
+	// SVN[11]: 0
+	// SVN[12]: 0
+	// SVN[13]: 0
+	// SVN[14]: 0
+	// SVN[15]: 0
+	// CryptoKey Type: pkix-base64-key
+	// CryptoKey Value: -----BEGIN PUBLIC KEY-----
+	// MFkwEwYHKoZIzj0CAQYIKoZIzj0DAQcDQgAEFn0taoAwR3PmrKkYLtAsD9o05KSM6mbgfNCgpuL0g6VpTHkZl73wk5BDxoV7n+Oeee0iIqkW3HMZT3ETiniJdg==
+	// -----END PUBLIC KEY-----
+}
+
+func extractPCERefVals(c *comid.Comid) error {
+	if c.Triples.ReferenceValues == nil {
+		return fmt.Errorf("no reference values triples")
+	}
+
+	for i, rv := range c.Triples.ReferenceValues.Values {
+		if err := extractPCERefVal(rv); err != nil {
+			return fmt.Errorf("bad PSA reference value at index %d: %w", i, err)
+		}
+	}
+
+	return nil
+}
+
+func extractPCERefVal(rv comid.ValueTriple) error {
+	class := rv.Environment.Class
+
+	if err := extractClassElements(class); err != nil {
+		return fmt.Errorf("extracting class: %w", err)
+	}
+
+	measurements := rv.Measurements
+	if err := extractPCEMeasurements(&measurements); err != nil {
+		return fmt.Errorf("extracting measurements: %w", err)
+	}
+
+	return nil
+}
+
+func extractPCEMeasurements(meas *comid.Measurements) error {
+	if len(meas.Values) == 0 {
+		return fmt.Errorf("no measurements")
+	}
+	for i := range meas.Values {
+		m := &meas.Values[0]
+		if err := decodePCEMValExtensions(m); err != nil {
+			return fmt.Errorf("extracting measurement at index %d: %w", i, err)
+		}
+
+		if m.AuthorizedBy != nil {
+			err := decodeAuthorisedBy(m)
+			if err != nil {
+				return fmt.Errorf("extracting measurement at index %d: %w", i, err)
+			}
+		}
+	}
+	return nil
+}
+
+func decodePCEMValExtensions(m *comid.Measurement) error {
+	val, err := m.Val.Extensions.Get("instanceid")
+	if err != nil {
+		return fmt.Errorf("failed to decode instanceid from measurement extensions")
+	}
+	i, ok := val.(*TeeInstanceID)
+	if !ok {
+		fmt.Printf("val was not pointer to teeInstanceID")
+	}
+
+	if i.IsBytes() {
+		val, err = i.GetBytes()
+		if err != nil {
+			return fmt.Errorf("failed to decode teeinstanceid: %w", err)
+		}
+		fmt.Printf("\nInstanceID: %x", val)
+	} else if i.IsUint() {
+		val, err = i.GetUint()
+		if err != nil {
+			return fmt.Errorf("failed to decode teeinstanceid: %w", err)
+		}
+		fmt.Printf("\nInstanceID: %d", val)
+	} else {
+		return fmt.Errorf("teeinstanceid is neither integer or byte string")
+	}
+
+	val, err = m.Val.Extensions.Get("tcbcompsvn")
+	if err != nil {
+		return fmt.Errorf("failed to decode teetcbcompsvn from measurement extensions")
+	}
+
+	tD, ok := val.(*TeeTcbCompSvn)
+	if !ok {
+		fmt.Printf("val was not pointer to teetcbcompsvn")
+	}
+	if err = tD.Valid(); err != nil {
+		return fmt.Errorf("invalid computed SVN: %w", err)
+	}
+	val, err = m.Val.Extensions.Get("pceid")
+	if err != nil {
+		return fmt.Errorf("failed to decode tcbevalnum from measurement extensions")
+	}
+	t, ok := val.(*TeePCEID)
+	if !ok {
+		fmt.Printf("val was not pointer to TeeTcbEvalNum")
+	}
+	if err = t.Valid(); err != nil {
+		return fmt.Errorf("invalid PCEID: %w", err)
+	}
+	pceID := *t
+	fmt.Printf("\npceID: %s", pceID)
+
+	err = extractSVN(tD)
+	if err != nil {
+		return fmt.Errorf("unable to extract TEE Digest: %w", err)
+	}
+	return nil
+}
+
+func extractSVN(s *TeeTcbCompSvn) error {
+	if s == nil {
+		return fmt.Errorf("no TEE TCB Comp SVN")
+	}
+
+	if len(*s) > 16 {
+		return fmt.Errorf("computed SVN cannot be greater than 16")
+	}
+
+	for i, svn := range *s {
+		fmt.Printf("\nSVN[%d]: %d", i, svn)
+	}
+
+	return nil
+}
+
+var (
+	// test cases are based on diag files here:
+	// https://github.com/ietf-rats-wg/draft-ietf-rats-corim/tree/main/cddl/examples
+
+	//go:embed testcases/comid_pce_refval.cbor
+	testComid3 []byte
+)
+
+func Example_decode_PCE_CBOR() {
+	profileID, err := eat.NewProfile("2.16.840.1.113741.1.16.1")
+	if err != nil {
+		panic(err) // will not error, as the hard-coded string above is valid
+	}
+	manifest, found := corim.GetProfileManifest(profileID)
+	if !found {
+		fmt.Printf("CoRIM Profile NOT FOUND")
+		return
+	}
+
+	m := manifest.GetComid()
+
+	if err := m.FromCBOR(testComid3); err != nil {
+		panic(err)
+	}
+	if err := m.Valid(); err != nil {
+		panic(err)
+	}
+
+	if err := extractPCERefVals(m); err != nil {
+		panic(err)
+	}
+
+	// Output:
+	// OID: 2.16.840.1.113741.1.2.3.4.5
+	// Vendor: Intel Corporation
+	// Model: TDX PCE TCB
+	// InstanceID: 00112233445566778899aabbccddeeff
+	// pceID: 0000
+	// SVN[0]: 10
+	// SVN[1]: 10
+	// SVN[2]: 2
+	// SVN[3]: 2
+	// SVN[4]: 2
+	// SVN[5]: 1
+	// SVN[6]: 4
+	// SVN[7]: 0
+	// SVN[8]: 0
+	// SVN[9]: 0
+	// SVN[10]: 0
+	// SVN[11]: 0
+	// SVN[12]: 0
+	// SVN[13]: 0
+	// SVN[14]: 0
+	// SVN[15]: 0
+	// CryptoKey Type: pkix-base64-key
+	// CryptoKey Value: -----BEGIN PUBLIC KEY-----
+	// MFkwEwYHKoZIzj0CAQYIKoZIzj0DAQcDQgAEFn0taoAwR3PmrKkYLtAsD9o05KSM6mbgfNCgpuL0g6VpTHkZl73wk5BDxoV7n+Oeee0iIqkW3HMZT3ETiniJdg==
+	// -----END PUBLIC KEY-----
+}
+
+func Example_encode_tdx_pce_refval_with_profile() {
+	profileID, err := eat.NewProfile("2.16.840.1.113741.1.16.1")
+	if err != nil {
+		panic(err) // will not error, as the hard-coded string above is valid
+	}
+	manifest, found := corim.GetProfileManifest(profileID)
+	if !found {
+		fmt.Printf("CoRIM Profile NOT FOUND")
+		return
+	}
+
+	m := manifest.GetComid()
+	if m == nil {
+		panic(err)
+	}
+	m.SetTagIdentity("43BBE37F-2E61-4B33-AED3-53CFF1428B20", 0).
+		AddEntity("INTEL", &TestRegID, comid.RoleCreator, comid.RoleTagCreator, comid.RoleMaintainer)
+
+	refVal := &comid.ValueTriple{}
+	measurement := &comid.Measurement{}
+	refVal.Environment = comid.Environment{
+		Class: comid.NewClassOID(TestOID).
+			SetVendor("Intel Corporation").
+			SetModel("TDX PCE TCB"),
+	}
+
+	refVal.Measurements.Add(measurement)
+	m.Triples.AddReferenceValue(*refVal)
+
+	err = setTDXPCEMvalExtension(&m.Triples.ReferenceValues.Values[0].Measurements.Values[0].Val)
+	if err != nil {
+		fmt.Printf("unable to set extensions :%s", err.Error())
+	}
+
+	err = m.Valid()
+	if err != nil {
+		fmt.Printf("CoMID is not Valid :%s", err.Error())
+	}
+
+	cbor, err := m.ToCBOR()
+	if err == nil {
+		fmt.Printf("%x\n", cbor)
+	} else {
+		fmt.Printf("\n To CBOR Failed: %s \n", err.Error())
+	}
+
+	json, err := m.ToJSON()
+	if err == nil {
+		fmt.Printf("%s\n", string(json))
+	} else {
+		fmt.Printf("\n To JSON Failed \n")
+	}
+
+	// Output:
+	// a301a1005043bbe37f2e614b33aed353cff1428b200281a30065494e54454c01d8207168747470733a2f2f696e74656c2e636f6d028301000204a1008182a100a300d86f4c6086480186f84d01020304050171496e74656c20436f72706f726174696f6e026b544458205043452054434281a101a3384c182d384f685043454944303031387c900102030405060708090a0b0c0d0e0f10
+	// {"tag-identity":{"id":"43bbe37f-2e61-4b33-aed3-53cff1428b20"},"entities":[{"name":"INTEL","regid":"https://intel.com","roles":["creator","tagCreator","maintainer"]}],"triples":{"reference-values":[{"environment":{"class":{"id":{"type":"oid","value":"2.16.840.1.113741.1.2.3.4.5"},"vendor":"Intel Corporation","model":"TDX PCE TCB"}},"measurements":[{"value":{"instanceid":{"type":"uint","value":45},"pceid":"PCEID001","tcbcompsvn":[1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16]}}]}]}}
+}
+
+func setTDXPCEMvalExtension(val *comid.Mval) error {
+	instanceID, err := NewTeeInstanceID(TestUIntInstance)
+	if err != nil {
+		return fmt.Errorf("unable to get teeinstanceID %w", err)
+	}
+	err = val.Extensions.Set("instanceid", instanceID)
+	if err != nil {
+		return fmt.Errorf("unable to set teeinstanceID %w", err)
+	}
+
+	p, err := NewTeePCEID(TestPCEID)
+	if err != nil {
+		return fmt.Errorf("unable to get NewTeepceID %w", err)
+	}
+	err = val.Extensions.Set("pceid", p)
+	if err != nil {
+		return fmt.Errorf("unable to set teepceID %w", err)
+	}
+
+	c, err := NewTeeTcbCompSVN(TestCompSVN)
+	if err != nil {
+		return fmt.Errorf("failed to get TeeTcbCompSVN %w", err)
+	}
+
+	err = val.Extensions.Set("tcbcompsvn", c)
+	if err != nil {
+		return fmt.Errorf("unable to set teetcbcompsvn: %w", err)
+	}
+	return nil
+}