mirror of
https://github.com/edgelesssys/constellation.git
synced 2024-10-01 01:36:09 -04:00
454 lines
13 KiB
Go
454 lines
13 KiB
Go
/*
|
|
Copyright (c) Edgeless Systems GmbH
|
|
|
|
SPDX-License-Identifier: AGPL-3.0-only
|
|
*/
|
|
|
|
package cloudcmd
|
|
|
|
import (
|
|
"crypto/sha256"
|
|
"encoding/base64"
|
|
"encoding/hex"
|
|
"testing"
|
|
|
|
"github.com/spf13/cobra"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/edgelesssys/constellation/v2/internal/atls"
|
|
"github.com/edgelesssys/constellation/v2/internal/attestation/azure/snp"
|
|
"github.com/edgelesssys/constellation/v2/internal/attestation/azure/trustedlaunch"
|
|
"github.com/edgelesssys/constellation/v2/internal/attestation/gcp"
|
|
"github.com/edgelesssys/constellation/v2/internal/attestation/idkeydigest"
|
|
"github.com/edgelesssys/constellation/v2/internal/attestation/measurements"
|
|
"github.com/edgelesssys/constellation/v2/internal/attestation/qemu"
|
|
"github.com/edgelesssys/constellation/v2/internal/config"
|
|
"github.com/edgelesssys/constellation/v2/internal/logger"
|
|
"github.com/edgelesssys/constellation/v2/internal/variant"
|
|
)
|
|
|
|
func TestNewValidator(t *testing.T) {
|
|
testPCRs := measurements.M{
|
|
0: measurements.WithAllBytes(0x00, measurements.Enforce),
|
|
1: measurements.WithAllBytes(0xFF, measurements.Enforce),
|
|
2: measurements.WithAllBytes(0x00, measurements.Enforce),
|
|
3: measurements.WithAllBytes(0xFF, measurements.Enforce),
|
|
4: measurements.WithAllBytes(0x00, measurements.Enforce),
|
|
5: measurements.WithAllBytes(0x00, measurements.Enforce),
|
|
}
|
|
|
|
testCases := map[string]struct {
|
|
config *config.Config
|
|
wantErr bool
|
|
}{
|
|
"gcp": {
|
|
config: &config.Config{
|
|
AttestationVariant: variant.GCPSEVES{}.String(),
|
|
Provider: config.ProviderConfig{
|
|
GCP: &config.GCPConfig{
|
|
Measurements: testPCRs,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
"azure cvm": {
|
|
config: &config.Config{
|
|
AttestationVariant: variant.AzureSEVSNP{}.String(),
|
|
Provider: config.ProviderConfig{
|
|
Azure: &config.AzureConfig{
|
|
Measurements: testPCRs,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
"azure trusted launch": {
|
|
config: &config.Config{
|
|
AttestationVariant: variant.AzureTrustedLaunch{}.String(),
|
|
Provider: config.ProviderConfig{
|
|
Azure: &config.AzureConfig{
|
|
Measurements: testPCRs,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
"openstack": {
|
|
config: &config.Config{
|
|
AttestationVariant: variant.QEMUVTPM{}.String(),
|
|
Provider: config.ProviderConfig{
|
|
OpenStack: &config.OpenStackConfig{
|
|
Measurements: testPCRs,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
"qemu": {
|
|
config: &config.Config{
|
|
AttestationVariant: variant.QEMUVTPM{}.String(),
|
|
Provider: config.ProviderConfig{
|
|
QEMU: &config.QEMUConfig{
|
|
Measurements: testPCRs,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
"no pcrs provided": {
|
|
config: &config.Config{
|
|
AttestationVariant: variant.AzureSEVSNP{}.String(),
|
|
Provider: config.ProviderConfig{
|
|
Azure: &config.AzureConfig{
|
|
Measurements: measurements.M{},
|
|
},
|
|
},
|
|
},
|
|
wantErr: true,
|
|
},
|
|
"unknown variant": {
|
|
config: &config.Config{
|
|
AttestationVariant: "unknown",
|
|
Provider: config.ProviderConfig{
|
|
QEMU: &config.QEMUConfig{
|
|
Measurements: testPCRs,
|
|
},
|
|
},
|
|
},
|
|
wantErr: true,
|
|
},
|
|
"set idkeydigest": {
|
|
config: &config.Config{
|
|
AttestationVariant: variant.AzureSEVSNP{}.String(),
|
|
Provider: config.ProviderConfig{
|
|
Azure: &config.AzureConfig{
|
|
Measurements: testPCRs,
|
|
IDKeyDigest: idkeydigest.List{[]byte("414141414141414141414141414141414141414141414141414141414141414141414141414141414141414141414141")},
|
|
EnforceIDKeyDigest: idkeydigest.Equal,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
for name, tc := range testCases {
|
|
t.Run(name, func(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
validators, err := NewValidator(tc.config, "https://192.0.2.1:8080/maa", logger.NewTest(t))
|
|
|
|
if tc.wantErr {
|
|
assert.Error(err)
|
|
} else {
|
|
assert.NoError(err)
|
|
assert.Equal(tc.config.GetMeasurements(), validators.pcrs)
|
|
variant, err := variant.FromString(tc.config.AttestationVariant)
|
|
require.NoError(t, err)
|
|
assert.Equal(variant, validators.attestationVariant)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestValidatorV(t *testing.T) {
|
|
newTestPCRs := func() measurements.M {
|
|
return measurements.M{
|
|
0: measurements.WithAllBytes(0x00, measurements.WarnOnly),
|
|
1: measurements.WithAllBytes(0x00, measurements.WarnOnly),
|
|
2: measurements.WithAllBytes(0x00, measurements.WarnOnly),
|
|
3: measurements.WithAllBytes(0x00, measurements.WarnOnly),
|
|
4: measurements.WithAllBytes(0x00, measurements.WarnOnly),
|
|
5: measurements.WithAllBytes(0x00, measurements.WarnOnly),
|
|
6: measurements.WithAllBytes(0x00, measurements.WarnOnly),
|
|
7: measurements.WithAllBytes(0x00, measurements.WarnOnly),
|
|
8: measurements.WithAllBytes(0x00, measurements.WarnOnly),
|
|
9: measurements.WithAllBytes(0x00, measurements.WarnOnly),
|
|
10: measurements.WithAllBytes(0x00, measurements.WarnOnly),
|
|
11: measurements.WithAllBytes(0x00, measurements.WarnOnly),
|
|
12: measurements.WithAllBytes(0x00, measurements.WarnOnly),
|
|
}
|
|
}
|
|
|
|
testCases := map[string]struct {
|
|
variant variant.Variant
|
|
pcrs measurements.M
|
|
wantVs atls.Validator
|
|
}{
|
|
"gcp": {
|
|
variant: variant.GCPSEVES{},
|
|
pcrs: newTestPCRs(),
|
|
wantVs: gcp.NewValidator(config.GCPSEVES{Measurements: newTestPCRs()}, nil),
|
|
},
|
|
"azure cvm": {
|
|
variant: variant.AzureSEVSNP{},
|
|
pcrs: newTestPCRs(),
|
|
wantVs: snp.NewValidator(
|
|
config.AzureSEVSNP{
|
|
Measurements: newTestPCRs(),
|
|
FirmwareSignerConfig: config.SNPFirmwareSignerConfig{
|
|
AcceptedKeyDigests: idkeydigest.List{},
|
|
EnforcementPolicy: idkeydigest.WarnOnly,
|
|
},
|
|
},
|
|
nil,
|
|
),
|
|
},
|
|
"azure trusted launch": {
|
|
variant: variant.AzureTrustedLaunch{},
|
|
pcrs: newTestPCRs(),
|
|
wantVs: trustedlaunch.NewValidator(config.AzureTrustedLaunch{Measurements: newTestPCRs()}, nil),
|
|
},
|
|
"qemu": {
|
|
variant: variant.QEMUVTPM{},
|
|
pcrs: newTestPCRs(),
|
|
wantVs: qemu.NewValidator(config.QEMUVTPM{Measurements: newTestPCRs()}, nil),
|
|
},
|
|
}
|
|
|
|
for name, tc := range testCases {
|
|
t.Run(name, func(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
validators := &Validator{attestationVariant: tc.variant, pcrs: tc.pcrs}
|
|
|
|
resultValidator := validators.V(&cobra.Command{})
|
|
|
|
assert.Equal(tc.wantVs.OID(), resultValidator.OID())
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestValidatorUpdateInitPCRs(t *testing.T) {
|
|
zero := measurements.WithAllBytes(0x00, measurements.WarnOnly)
|
|
one := measurements.WithAllBytes(0x11, measurements.WarnOnly)
|
|
one64 := base64.StdEncoding.EncodeToString(one.Expected[:])
|
|
oneHash := sha256.Sum256(one.Expected[:])
|
|
pcrZeroUpdatedOne := sha256.Sum256(append(zero.Expected[:], oneHash[:]...))
|
|
newTestPCRs := func() measurements.M {
|
|
return measurements.M{
|
|
0: measurements.WithAllBytes(0x00, measurements.WarnOnly),
|
|
1: measurements.WithAllBytes(0x00, measurements.WarnOnly),
|
|
2: measurements.WithAllBytes(0x00, measurements.WarnOnly),
|
|
3: measurements.WithAllBytes(0x00, measurements.WarnOnly),
|
|
4: measurements.WithAllBytes(0x00, measurements.WarnOnly),
|
|
5: measurements.WithAllBytes(0x00, measurements.WarnOnly),
|
|
6: measurements.WithAllBytes(0x00, measurements.WarnOnly),
|
|
7: measurements.WithAllBytes(0x00, measurements.WarnOnly),
|
|
8: measurements.WithAllBytes(0x00, measurements.WarnOnly),
|
|
9: measurements.WithAllBytes(0x00, measurements.WarnOnly),
|
|
10: measurements.WithAllBytes(0x00, measurements.WarnOnly),
|
|
11: measurements.WithAllBytes(0x00, measurements.WarnOnly),
|
|
12: measurements.WithAllBytes(0x00, measurements.WarnOnly),
|
|
13: measurements.WithAllBytes(0x00, measurements.WarnOnly),
|
|
14: measurements.WithAllBytes(0x00, measurements.WarnOnly),
|
|
15: measurements.WithAllBytes(0x00, measurements.WarnOnly),
|
|
16: measurements.WithAllBytes(0x00, measurements.WarnOnly),
|
|
17: measurements.WithAllBytes(0x11, measurements.WarnOnly),
|
|
18: measurements.WithAllBytes(0x11, measurements.WarnOnly),
|
|
19: measurements.WithAllBytes(0x11, measurements.WarnOnly),
|
|
20: measurements.WithAllBytes(0x11, measurements.WarnOnly),
|
|
21: measurements.WithAllBytes(0x11, measurements.WarnOnly),
|
|
22: measurements.WithAllBytes(0x11, measurements.WarnOnly),
|
|
23: measurements.WithAllBytes(0x00, measurements.WarnOnly),
|
|
}
|
|
}
|
|
|
|
testCases := map[string]struct {
|
|
variant variant.Variant
|
|
pcrs measurements.M
|
|
ownerID string
|
|
clusterID string
|
|
wantErr bool
|
|
}{
|
|
"gcp update owner ID": {
|
|
variant: variant.GCPSEVES{},
|
|
pcrs: newTestPCRs(),
|
|
ownerID: one64,
|
|
},
|
|
"gcp update cluster ID": {
|
|
variant: variant.GCPSEVES{},
|
|
pcrs: newTestPCRs(),
|
|
clusterID: one64,
|
|
},
|
|
"gcp update both": {
|
|
variant: variant.GCPSEVES{},
|
|
pcrs: newTestPCRs(),
|
|
ownerID: one64,
|
|
clusterID: one64,
|
|
},
|
|
"azure update owner ID": {
|
|
variant: variant.AzureSEVSNP{},
|
|
pcrs: newTestPCRs(),
|
|
ownerID: one64,
|
|
},
|
|
"azure update cluster ID": {
|
|
variant: variant.AzureSEVSNP{},
|
|
pcrs: newTestPCRs(),
|
|
clusterID: one64,
|
|
},
|
|
"azure update both": {
|
|
variant: variant.AzureSEVSNP{},
|
|
pcrs: newTestPCRs(),
|
|
ownerID: one64,
|
|
clusterID: one64,
|
|
},
|
|
"owner ID and cluster ID empty": {
|
|
variant: variant.GCPSEVES{},
|
|
pcrs: newTestPCRs(),
|
|
},
|
|
"invalid encoding": {
|
|
variant: variant.GCPSEVES{},
|
|
pcrs: newTestPCRs(),
|
|
ownerID: "invalid",
|
|
wantErr: true,
|
|
},
|
|
}
|
|
|
|
for name, tc := range testCases {
|
|
t.Run(name, func(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
validators := &Validator{attestationVariant: tc.variant, pcrs: tc.pcrs}
|
|
|
|
err := validators.UpdateInitPCRs(tc.ownerID, tc.clusterID)
|
|
|
|
if tc.wantErr {
|
|
assert.Error(err)
|
|
return
|
|
}
|
|
assert.NoError(err)
|
|
for i := 0; i < len(tc.pcrs); i++ {
|
|
switch {
|
|
case i == int(measurements.PCRIndexClusterID) && tc.clusterID == "":
|
|
// should be deleted
|
|
_, ok := validators.pcrs[uint32(i)]
|
|
assert.False(ok)
|
|
|
|
case i == int(measurements.PCRIndexClusterID):
|
|
pcr, ok := validators.pcrs[uint32(i)]
|
|
assert.True(ok)
|
|
assert.Equal(pcrZeroUpdatedOne, pcr.Expected)
|
|
|
|
case i == int(measurements.PCRIndexOwnerID) && tc.ownerID == "":
|
|
// should be deleted
|
|
_, ok := validators.pcrs[uint32(i)]
|
|
assert.False(ok)
|
|
|
|
case i == int(measurements.PCRIndexOwnerID):
|
|
pcr, ok := validators.pcrs[uint32(i)]
|
|
assert.True(ok)
|
|
assert.Equal(pcrZeroUpdatedOne, pcr.Expected)
|
|
|
|
default:
|
|
if i >= 17 && i <= 22 {
|
|
assert.Equal(one, validators.pcrs[uint32(i)])
|
|
} else {
|
|
assert.Equal(zero, validators.pcrs[uint32(i)])
|
|
}
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestUpdatePCR(t *testing.T) {
|
|
emptyMap := measurements.M{}
|
|
defaultMap := measurements.M{
|
|
0: measurements.WithAllBytes(0xAA, measurements.Enforce),
|
|
1: measurements.WithAllBytes(0xBB, measurements.Enforce),
|
|
}
|
|
|
|
testCases := map[string]struct {
|
|
pcrMap measurements.M
|
|
pcrIndex uint32
|
|
encoded string
|
|
wantEntries int
|
|
wantErr bool
|
|
}{
|
|
"empty input, empty map": {
|
|
pcrMap: emptyMap,
|
|
pcrIndex: 10,
|
|
encoded: "",
|
|
wantEntries: 0,
|
|
wantErr: false,
|
|
},
|
|
"empty input, default map": {
|
|
pcrMap: defaultMap,
|
|
pcrIndex: 10,
|
|
encoded: "",
|
|
wantEntries: len(defaultMap),
|
|
wantErr: false,
|
|
},
|
|
"correct input, empty map": {
|
|
pcrMap: emptyMap,
|
|
pcrIndex: 10,
|
|
encoded: base64.StdEncoding.EncodeToString([]byte("Constellation")),
|
|
wantEntries: 1,
|
|
wantErr: false,
|
|
},
|
|
"correct input, default map": {
|
|
pcrMap: defaultMap,
|
|
pcrIndex: 10,
|
|
encoded: base64.StdEncoding.EncodeToString([]byte("Constellation")),
|
|
wantEntries: len(defaultMap) + 1,
|
|
wantErr: false,
|
|
},
|
|
"hex input, empty map": {
|
|
pcrMap: emptyMap,
|
|
pcrIndex: 10,
|
|
encoded: hex.EncodeToString([]byte("Constellation")),
|
|
wantEntries: 1,
|
|
wantErr: false,
|
|
},
|
|
"hex input, default map": {
|
|
pcrMap: defaultMap,
|
|
pcrIndex: 10,
|
|
encoded: hex.EncodeToString([]byte("Constellation")),
|
|
wantEntries: len(defaultMap) + 1,
|
|
wantErr: false,
|
|
},
|
|
"unencoded input, empty map": {
|
|
pcrMap: emptyMap,
|
|
pcrIndex: 10,
|
|
encoded: "Constellation",
|
|
wantEntries: 0,
|
|
wantErr: true,
|
|
},
|
|
"unencoded input, default map": {
|
|
pcrMap: defaultMap,
|
|
pcrIndex: 10,
|
|
encoded: "Constellation",
|
|
wantEntries: len(defaultMap),
|
|
wantErr: true,
|
|
},
|
|
"empty input at occupied index": {
|
|
pcrMap: defaultMap,
|
|
pcrIndex: 0,
|
|
encoded: "",
|
|
wantEntries: len(defaultMap) - 1,
|
|
wantErr: false,
|
|
},
|
|
}
|
|
|
|
for name, tc := range testCases {
|
|
t.Run(name, func(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
pcrs := make(measurements.M)
|
|
for k, v := range tc.pcrMap {
|
|
pcrs[k] = v
|
|
}
|
|
|
|
validators := &Validator{
|
|
attestationVariant: variant.GCPSEVES{},
|
|
pcrs: pcrs,
|
|
}
|
|
err := validators.updatePCR(tc.pcrIndex, tc.encoded)
|
|
|
|
if tc.wantErr {
|
|
assert.Error(err)
|
|
} else {
|
|
assert.NoError(err)
|
|
}
|
|
assert.Len(pcrs, tc.wantEntries)
|
|
})
|
|
}
|
|
}
|