Skip to content

Commit

Permalink
testutil: use OpenTPMDevice
Browse files Browse the repository at this point in the history
  • Loading branch information
chrisccoulson committed Jan 12, 2024
1 parent a4e2a29 commit 8f702d5
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions testutil/tpm.go
Original file line number Diff line number Diff line change
Expand Up @@ -257,18 +257,17 @@ func (c *tpmSimulatorLaunchContext) stopAndTerminate() (err error) {
c.captureErr("terminate", c.terminateFn(stopOk))
}()

transport, err := mssim.OpenConnection("", c.port)
device := mssim.NewLocalDevice(c.port)
tpm, err := tpm2.OpenTPMDevice(device)
if err != nil {
return fmt.Errorf("cannot open simulator connection for stop: %w", err)
}

tpm := tpm2.NewTPMContext(transport)
tpm.SetCommandTimeout(5 * time.Second)

c.captureErr("shutdown", func() error {
return tpm.Shutdown(tpm2.StartupClear)
})
if err := transport.Stop(); err != nil {
if err := tpm.Transport().(*mssim.Transport).Stop(); err != nil {
return fmt.Errorf("cannot stop simulator: %w", err)
}
if err := tpm.Close(); err != nil {
Expand Down

0 comments on commit 8f702d5

Please sign in to comment.