tpm: selftest: add test covering async mode (8f84bddc) · Commits · 戴 / test

tools/testing/selftests/tpm2/test_smoke.sh

+1 −0

Original line number	Diff line number	Diff line
		@@ -2,3 +2,4 @@
		# SPDX-License-Identifier: (GPL-2.0 OR BSD-3-Clause)

		python -m unittest -v tpm2_tests.SmokeTest
		python -m unittest -v tpm2_tests.AsyncTest

+17 −2

Original line number	Diff line number	Diff line
		@@ -6,8 +6,8 @@ import socket
		import struct
		import sys
		import unittest
		from fcntl import ioctl

		import fcntl
		import select

		TPM2_ST_NO_SESSIONS = 0x8001
		TPM2_ST_SESSIONS = 0x8002
		@@ -352,6 +352,7 @@ def hex_dump(d):
		class Client:
		FLAG_DEBUG = 0x01
		FLAG_SPACE = 0x02
		FLAG_NONBLOCK = 0x04
		TPM_IOC_NEW_SPACE = 0xa200

		def __init__(self, flags = 0):
		@@ -362,13 +363,27 @@ class Client:
		else:
		self.tpm = open('/dev/tpmrm0', 'r+b', buffering=0)

		if (self.flags & Client.FLAG_NONBLOCK):
		flags = fcntl.fcntl(self.tpm, fcntl.F_GETFL)
		flags \|= os.O_NONBLOCK
		fcntl.fcntl(self.tpm, fcntl.F_SETFL, flags)
		self.tpm_poll = select.poll()

		def close(self):
		self.tpm.close()

		def send_cmd(self, cmd):
		self.tpm.write(cmd)

		if (self.flags & Client.FLAG_NONBLOCK):
		self.tpm_poll.register(self.tpm, select.POLLIN)
		self.tpm_poll.poll(10000)

		rsp = self.tpm.read()

		if (self.flags & Client.FLAG_NONBLOCK):
		self.tpm_poll.unregister(self.tpm)

		if (self.flags & Client.FLAG_DEBUG) != 0:
		sys.stderr.write('cmd' + os.linesep)
		sys.stderr.write(hex_dump(cmd) + os.linesep)

+13 −0

Original line number	Diff line number	Diff line
		@@ -288,3 +288,16 @@ class SpaceTest(unittest.TestCase):

		self.assertEqual(rc, tpm2.TPM2_RC_COMMAND_CODE \|
		tpm2.TSS2_RESMGR_TPM_RC_LAYER)

		class AsyncTest(unittest.TestCase):
		def setUp(self):
		logging.basicConfig(filename='AsyncTest.log', level=logging.DEBUG)

		def test_async(self):
		log = logging.getLogger(__name__)
		log.debug(sys._getframe().f_code.co_name)

		async_client = tpm2.Client(tpm2.Client.FLAG_NONBLOCK)
		log.debug("Calling get_cap in a NON_BLOCKING mode")
		async_client.get_cap(tpm2.TPM2_CAP_HANDLES, tpm2.HR_LOADED_SESSION)
		async_client.close()