mirror of
https://github.com/edgelesssys/constellation.git
synced 2024-12-30 09:56:21 -05:00
103 lines
2.7 KiB
Go
103 lines
2.7 KiB
Go
/*
|
|
Copyright (c) Edgeless Systems GmbH
|
|
|
|
SPDX-License-Identifier: AGPL-3.0-only
|
|
*/
|
|
|
|
package initialize
|
|
|
|
import (
|
|
"errors"
|
|
"io"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/edgelesssys/constellation/v2/internal/attestation/measurements"
|
|
"github.com/edgelesssys/constellation/v2/internal/attestation/simulator"
|
|
"github.com/google/go-tpm-tools/client"
|
|
"github.com/google/go-tpm/legacy/tpm2"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// simTPMNOPCloser is a wrapper for the generic TPM simulator with a NOP Close() method.
|
|
type simTPMNOPCloser struct {
|
|
io.ReadWriteCloser
|
|
}
|
|
|
|
func (s simTPMNOPCloser) Close() error {
|
|
return nil
|
|
}
|
|
|
|
func TestMarkNodeAsBootstrapped(t *testing.T) {
|
|
cgo := os.Getenv("CGO_ENABLED")
|
|
if cgo == "0" {
|
|
t.Skip("skipping test because CGO is disabled and tpm simulator requires it")
|
|
}
|
|
assert := assert.New(t)
|
|
require := require.New(t)
|
|
|
|
tpm, err := simulator.OpenSimulatedTPM()
|
|
require.NoError(err)
|
|
defer tpm.Close()
|
|
pcrs, err := client.ReadAllPCRs(tpm)
|
|
require.NoError(err)
|
|
|
|
assert.NoError(MarkNodeAsBootstrapped(func() (io.ReadWriteCloser, error) {
|
|
return &simTPMNOPCloser{tpm}, nil
|
|
}, []byte{0x0, 0x1, 0x2, 0x3}))
|
|
|
|
pcrsInitialized, err := client.ReadAllPCRs(tpm)
|
|
require.NoError(err)
|
|
|
|
for i := range pcrs {
|
|
assert.NotEqual(pcrs[i].Pcrs[uint32(measurements.PCRIndexClusterID)], pcrsInitialized[i].Pcrs[uint32(measurements.PCRIndexClusterID)])
|
|
}
|
|
}
|
|
|
|
func TestFailOpener(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
assert.Error(MarkNodeAsBootstrapped(func() (io.ReadWriteCloser, error) { return nil, errors.New("failed") }, []byte{0x0, 0x1, 0x2, 0x3}))
|
|
}
|
|
|
|
func TestIsNodeInitialized(t *testing.T) {
|
|
cgo := os.Getenv("CGO_ENABLED")
|
|
if cgo == "0" {
|
|
t.Skip("skipping test because CGO is disabled and tpm simulator requires it")
|
|
}
|
|
testCases := map[string]struct {
|
|
pcrValueClusterID []byte
|
|
wantInitialized bool
|
|
wantErr bool
|
|
}{
|
|
"uninitialized PCRs results in uninitialized node": {},
|
|
"initializing PCRs result in initialized node": {
|
|
pcrValueClusterID: []byte{0x4, 0x5, 0x6, 0x7},
|
|
wantInitialized: true,
|
|
},
|
|
}
|
|
|
|
for name, tc := range testCases {
|
|
t.Run(name, func(t *testing.T) {
|
|
assert := require.New(t)
|
|
require := require.New(t)
|
|
tpm, err := simulator.OpenSimulatedTPM()
|
|
require.NoError(err)
|
|
defer tpm.Close()
|
|
if tc.pcrValueClusterID != nil {
|
|
require.NoError(tpm2.PCREvent(tpm, measurements.PCRIndexClusterID, tc.pcrValueClusterID))
|
|
}
|
|
initialized, err := IsNodeBootstrapped(func() (io.ReadWriteCloser, error) {
|
|
return &simTPMNOPCloser{tpm}, nil
|
|
})
|
|
if tc.wantErr {
|
|
assert.Error(err)
|
|
return
|
|
}
|
|
require.NoError(err)
|
|
require.Equal(tc.wantInitialized, initialized)
|
|
})
|
|
}
|
|
}
|