def verify_decode(self, test_vector, test_message):
serialized = "".join(test_vector).decode("hex")
message = test_message_pb2.TestMessage()
tls_message.decode(serialized, message)
self.assertEqual(test_message, message,
msg = "%s vs %s" % (test_message, message))
def test_encode_repeated_fixed_length_too_many_fails(self):
test_message = test_message_pb2.TestMessage()
test_message.CopyFrom(valid_test_message)
# The test message has 2, add 3 more to go over the total
# allowed which is 4.
for i in range(3):
test_message.vector_fixed_bytes.append("\xab\x00\xa0\xa0")
self.verify_encode_fails(test_message)
def test_encode_ignores_bad_select(self):
test_vector = serialized_valid_test_message[:]
test_vector[11] = "0000"
test_vector[12] = ""
test_message = test_message_pb2.TestMessage()
test_message.CopyFrom(valid_test_message)
test_message.test_enum = test_message_pb2.TestMessage.ENUM_0
self.verify_encode(test_message, test_vector)
def test_decode_empty(self):
test_vector = serialized_valid_test_message[:]
# var_bytes has no min length
test_vector[7] = "00"
test_message = test_message_pb2.TestMessage()
test_message.CopyFrom(valid_test_message)
test_message.var_bytes = ""
self.verify_decode(test_vector, test_message)
def test_decode_valid_select(self):
test_vector = serialized_valid_test_message[:]
test_vector[11] = "0000"
test_vector[12] = ""
test_message = test_message_pb2.TestMessage()
test_message.CopyFrom(valid_test_message)
test_message.test_enum = test_message_pb2.TestMessage.ENUM_0
test_message.ClearField("select_uint32")
self.verify_decode(test_vector, test_message)
def test_read_longer_buffer(self):
test_vector = serialized_valid_test_message[:]
test_vector.append("somegarbageintheend".encode("hex"))
serialized = "".join(test_vector).decode("hex")
message = test_message_pb2.TestMessage()
reader = tls_message.TLSReader(serialized)
reader.read(message)
self.assertEqual(valid_test_message, message,
msg = "%s vs %s" % (valid_test_message, message))
self.assertFalse(reader.finished())
def verify_decode_fail(self, test_vector):
serialized = "".join(test_vector).decode("hex")
message = test_message_pb2.TestMessage()
self.assertRaises(tls_message.TLSDecodingError,
tls_message.decode, serialized, message)
def test_encode_repeated_too_short_fails(self):
test_message = test_message_pb2.TestMessage()
test_message.CopyFrom(valid_test_message)
test_message.ClearField("vector_uint32")
self.verify_encode_fails(test_message)
def test_encode_repeated_too_long_fails(self):
test_message = test_message_pb2.TestMessage()
test_message.CopyFrom(valid_test_message)
test_message.vector_uint32.extend([1, 2, 3, 4])
self.verify_encode_fails(test_message)
def test_encode_too_long_fails(self):
test_message = test_message_pb2.TestMessage()
test_message.CopyFrom(valid_test_message)
test_message.var_bytes = "Iamtoolongformyowngood"
self.verify_encode_fails(test_message)
def test_encode_too_short_fails(self):
test_message = test_message_pb2.TestMessage()
test_message.CopyFrom(valid_test_message)
test_message.var_bytes2 = "sho"
self.verify_encode_fails(test_message)
def test_encode_repeated_fixed_length_wrong_length_fails(self):
test_message = test_message_pb2.TestMessage()
test_message.CopyFrom(valid_test_message)
test_message.vector_fixed_bytes.append("\xab\x00")
self.verify_encode_fails(test_message)
def test_encode_too_large_value_fails(self):
test_message = test_message_pb2.TestMessage()
test_message.CopyFrom(valid_test_message)
test_message.uint_8 = 65000
self.verify_encode_fails(test_message)
def test_encode_ignores_skipped_fields(self):
test_message = test_message_pb2.TestMessage()
test_message.CopyFrom(valid_test_message)
test_message.skip_uint32 = 42
self.verify_encode(test_message, serialized_valid_test_message)
#!/usr/bin/env python
import unittest
from ct.client import tls_message
from ct.proto import client_pb2
from ct.proto import test_message_pb2
valid_test_message = test_message_pb2.TestMessage()
valid_test_message.uint_8 = 0
valid_test_message.uint_16 = 258
valid_test_message.uint_24 = 197637
valid_test_message.uint_32 = 101124105
valid_test_message.uint_48 = 11042563100175
valid_test_message.uint_64 = 255
valid_test_message.fixed_bytes = "\xff\x00"
valid_test_message.var_bytes = "hello"
valid_test_message.var_bytes2 = "world"
valid_test_message.vector_bytes.append("hello")
valid_test_message.vector_bytes.append("world")
valid_test_message.vector_uint32.append(1)
valid_test_message.vector_uint32.append(255)
valid_test_message.test_enum = test_message_pb2.TestMessage.ENUM_1
valid_test_message.select_uint32 = 2
valid_test_message.embedded_message.uint_32 = 3
valid_test_message.repeated_message.add().uint_32 = 4
valid_test_message.repeated_message.add().uint_32 = 256
# Test vectors are given as a list of serialized, hex-encoded components.
def test_encode_bad_length_fails(self):
test_message = test_message_pb2.TestMessage()
test_message.CopyFrom(valid_test_message)
test_message.fixed_bytes = "hello"
self.verify_encode_fails(test_message)
def test_encode_repeated_fixed_length_too_few_fails(self):
test_message = test_message_pb2.TestMessage()
test_message.CopyFrom(valid_test_message)
test_message.ClearField("vector_fixed_bytes")
self.verify_encode_fails(test_message)
