def test_get_buffer(self): self.assertEqual( t2pc.PolicyLocality([0, 2, 4]).get_buffer_for_digest(), bytearray.fromhex("0000016F" + "15")) self.assertEqual( t2pc.PolicyLocality([34]).get_buffer_for_digest(), bytearray.fromhex("0000016F" + "22"))
def test_create_with_invalid_localites(self): with self.assertRaises(ValueError): t2pc.PolicyLocality([5]) with self.assertRaises(ValueError): t2pc.PolicyLocality([12]) with self.assertRaises(ValueError): t2pc.PolicyLocality([31]) with self.assertRaises(ValueError): t2pc.PolicyLocality([256])
def test_create_with_base_localities(self): policy = t2pc.PolicyLocality([0, 2, 4]) self.assertEqual(policy.get_bitfield(), 0b00010101) policy2 = t2pc.PolicyLocality([1, 2]) self.assertEqual(policy2.get_bitfield(), 0b00000110) policy3 = t2pc.PolicyLocality([3]) self.assertEqual(policy3.get_bitfield(), 0b00001000) policy4 = t2pc.PolicyLocality([57]) self.assertEqual(policy4.get_bitfield(), 57)
def test_policy_locality(self): expected_result = bytearray.fromhex( "07039B45BAF2CC169B0D84AF7C53FD1622B033DF0A5DCDA66360AA99E54947CD") test = t2pc.PolicyTreeSolo(t2pc.PolicyLocality([3, 4])) phash = t2pc.PolicyHasher('sha256') self.assertEqual(test.get_policy(phash), expected_result)
def test_complex_policy_1(self): expected_result = bytearray.fromhex( "DFFDB6C8EAFCBE691E358882B18703121EAB40DE2386F7A8E7B4A06591E1F0EE") # Computation details: # A = TPM2_PolicyLocality(3 & 4) # B = TPM2_PolicyCommandCode(TPM_CC_NV_UndefineSpaceSpecial) # C = TPM2_PolicyCommandCode(TPM_CC_NV_Write) # policy = {{A} AND {C}} OR {{A} AND {B}} a = t2pc.PolicyLocality([3, 4]) b = t2pc.PolicyCommandCode('TPM_CC_NV_UndefineSpaceSpecial') c = t2pc.PolicyCommandCode('TPM_CC_NV_Write') leg1 = t2pc.PolicyTreeAnd([a, c]) leg2 = t2pc.PolicyTreeAnd([a, b]) final = t2pc.PolicyTreeOr([leg1, leg2]) phash = t2pc.PolicyHasher('sha256') self.assertEqual(final.get_policy(phash), expected_result)
def test_create_with_mixed_lower_and_upper(self): with self.assertRaises(ValueError): t2pc.PolicyLocality([1, 4, 35]) with self.assertRaises(ValueError): t2pc.PolicyLocality([36, 128])
def test_create_with_empty_list(self): policy = t2pc.PolicyLocality(None) self.assertEqual(policy.get_bitfield(), 0) policy2 = t2pc.PolicyLocality(()) self.assertEqual(policy2.get_bitfield(), 0)