1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
|
// Copyright 2024 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package rtmr
import (
"bytes"
"strings"
"testing"
"github.com/google/go-configfs-tsm/configfs/fakertmr"
)
func TestExtendDigestErr(t *testing.T) {
var sha384Hash [48]byte
tcsErr := []struct {
rtmr int
digest []byte
wantErr string
}{
{rtmr: 1, digest: sha384Hash[:], wantErr: "could not write digest to rmtr1"},
{rtmr: 3, digest: []byte("aaaaaaaa"), wantErr: "the length of the digest must be 48 bytes"},
{rtmr: -1, digest: sha384Hash[:], wantErr: "invalid rtmr index -1. Index can only be a non-negative number"},
}
client := fakertmr.CreateRtmrSubsystem(t.TempDir())
for _, tc := range tcsErr {
err := ExtendDigest(client, tc.rtmr, tc.digest)
if err == nil || !strings.Contains(err.Error(), tc.wantErr) {
t.Fatalf("ExtendtoRtmrClient(%d, %q) failed: %v, want %q", tc.rtmr, tc.digest, err, tc.wantErr)
}
}
}
func TestExtendDigestRtmrOk(t *testing.T) {
var sha384Hash [48]byte
tcsOk := []struct {
rtmr int
digest []byte
}{
{rtmr: 2, digest: sha384Hash[:]},
{rtmr: 3, digest: sha384Hash[:]},
// Test the same rtmr index with an existing entry.
{rtmr: 3, digest: sha384Hash[:]},
}
client := fakertmr.CreateRtmrSubsystem(t.TempDir())
for _, tc := range tcsOk {
err := ExtendDigest(client, tc.rtmr, tc.digest)
if err != nil {
t.Fatalf("ExtendtoRtmrClient (%d, %q) failed: %v", tc.rtmr, tc.digest, err)
}
}
}
func TestGetDigestErr(t *testing.T) {
tcsErr := []struct {
rtmr int
wantErr string
}{
{rtmr: -1, wantErr: "invalid rtmr index -1. Index can only be a non-negative number"},
}
client := fakertmr.CreateRtmrSubsystem(t.TempDir())
for _, tc := range tcsErr {
_, err := GetDigest(client, tc.rtmr)
if err == nil || !strings.Contains(err.Error(), tc.wantErr) {
t.Fatalf("GetDigestRtmr(%d) failed: %v, want %q", tc.rtmr, err, tc.wantErr)
}
}
}
func TestGetDigestOk(t *testing.T) {
var sha384Hash [48]byte
tcsOk := []struct {
rtmr int
digest []byte
tcgMap []byte
}{
{rtmr: 0, digest: sha384Hash[:], tcgMap: []byte("1,7\n")},
{rtmr: 1, digest: sha384Hash[:], tcgMap: []byte("2-6\n")},
{rtmr: 2, digest: sha384Hash[:], tcgMap: []byte("8-15\n")},
{rtmr: 3, digest: sha384Hash[:], tcgMap: []byte("\n")},
// Test the same rtmr index with an existing entry.
{rtmr: 2, digest: sha384Hash[:], tcgMap: []byte("8-15\n")},
}
client := fakertmr.CreateRtmrSubsystem(t.TempDir())
for _, tc := range tcsOk {
r, err := GetDigest(client, tc.rtmr)
if err != nil {
t.Fatalf("GetDigestRtmr(%d) failed: %v", tc.rtmr, err)
}
if r.RtmrIndex != tc.rtmr {
t.Fatalf("GetDigestRtmr(%d) failed: got %d, want %d", tc.rtmr, r.RtmrIndex, tc.rtmr)
}
if !bytes.Equal(r.tcgMap, tc.tcgMap) {
t.Fatalf("GetDigestRtmr(%d) failed: got %q, want %q", tc.rtmr, r.tcgMap, tc.tcgMap)
}
}
}
func TestGetRtmrDigestAndExtendDigest(t *testing.T) {
var sha384Hash [48]byte
sha384Hash[0] = 0x01
client := fakertmr.CreateRtmrSubsystem(t.TempDir())
rtmrIndex := 3
// GetDigest
digest1, err := GetDigest(client, rtmrIndex)
if err != nil {
t.Fatalf("GetDigest(%d) failed: %v", rtmrIndex, err)
}
// ExtendDigest
err = ExtendDigest(client, rtmrIndex, sha384Hash[:])
if err != nil {
t.Fatalf("ExtendDigest(%d) failed: %v", rtmrIndex, err)
}
// GetDigest
digest2, err := GetDigest(client, rtmrIndex)
if err != nil {
t.Fatalf("GetDigest(%d) failed: %v", rtmrIndex, err)
}
if bytes.Equal(digest1.digest, digest2.digest) {
t.Fatalf("rtmr%q does not change after an extend %q", rtmrIndex, digest2.digest)
}
}
|
