Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit f1a0ba6c authored by Tadeusz Struk's avatar Tadeusz Struk Committed by James Morris
Browse files

selftests/tpm2: Extend tests to cover partial reads 



Three new tests added:
1. Send get random cmd, read header in 1st read, read the rest in second
   read - expect success
2. Send get random cmd, read only part of the response, send another
   get random command, read the response - expect success
3. Send get random cmd followed by another get random cmd, without
   reading the first response - expect the second cmd to fail with -EBUSY

Signed-off-by: default avatarTadeusz Struk <tadeusz.struk@intel.com>
Reviewed-by: default avatarJarkko Sakkinen <jarkko.sakkinen@linux.intel.com>
Tested-by: default avatarJarkko Sakkinen <jarkko.sakkinen@linux.intel.com>
Signed-off-by: default avatarJarkko Sakkinen <jarkko.sakkinen@linux.intel.com>
Signed-off-by: default avatarJames Morris <james.morris@microsoft.com>
parent be24b37e

tools/testing/selftests/tpm2/tpm2.py

+1 −0
Original line number Diff line number Diff line
@@ -22,6 +22,7 @@ TPM2_CC_UNSEAL = 0x015E 
TPM2_CC_FLUSH_CONTEXT = 0x0165

TPM2_CC_START_AUTH_SESSION = 0x0176

TPM2_CC_GET_CAPABILITY	= 0x017A

TPM2_CC_GET_RANDOM = 0x017B

TPM2_CC_PCR_READ = 0x017E

TPM2_CC_POLICY_PCR = 0x017F

TPM2_CC_PCR_EXTEND = 0x0182

tools/testing/selftests/tpm2/tpm2_tests.py

+63 −0
Original line number Diff line number Diff line
@@ -158,6 +158,69 @@ class SmokeTest(unittest.TestCase): 
            pass

        self.assertEqual(rejected, True)



    def test_read_partial_resp(self):

        try:

            fmt = '>HIIH'

            cmd = struct.pack(fmt,

                              tpm2.TPM2_ST_NO_SESSIONS,

                              struct.calcsize(fmt),

                              tpm2.TPM2_CC_GET_RANDOM,

                              0x20)

            self.client.tpm.write(cmd)

            hdr = self.client.tpm.read(10)

            sz = struct.unpack('>I', hdr[2:6])[0]

            rsp = self.client.tpm.read()

        except:

            pass

        self.assertEqual(sz, 10 + 2 + 32)

        self.assertEqual(len(rsp), 2 + 32)



    def test_read_partial_overwrite(self):

        try:

            fmt = '>HIIH'

            cmd = struct.pack(fmt,

                              tpm2.TPM2_ST_NO_SESSIONS,

                              struct.calcsize(fmt),

                              tpm2.TPM2_CC_GET_RANDOM,

                              0x20)

            self.client.tpm.write(cmd)

            # Read part of the respone

            rsp1 = self.client.tpm.read(15)



            # Send a new cmd

            self.client.tpm.write(cmd)



            # Read the whole respone

            rsp2 = self.client.tpm.read()

        except:

            pass

        self.assertEqual(len(rsp1), 15)

        self.assertEqual(len(rsp2), 10 + 2 + 32)



    def test_send_two_cmds(self):

        rejected = False

        try:

            fmt = '>HIIH'

            cmd = struct.pack(fmt,

                              tpm2.TPM2_ST_NO_SESSIONS,

                              struct.calcsize(fmt),

                              tpm2.TPM2_CC_GET_RANDOM,

                              0x20)

            self.client.tpm.write(cmd)



            # expect the second one to raise -EBUSY error

            self.client.tpm.write(cmd)

            rsp = self.client.tpm.read()



        except IOError, e:

            # read the response

            rsp = self.client.tpm.read()

            rejected = True

            pass

        except:

            pass

        self.assertEqual(rejected, True)



class SpaceTest(unittest.TestCase):

    def setUp(self):

        logging.basicConfig(filename='SpaceTest.log', level=logging.DEBUG)