| #!/usr/bin/python |
| # |
| # Copyright 2016 The Android Open Source Project |
| # |
| # Licensed under the Apache License, Version 2.0 (the "License"); |
| # you may not use this file except in compliance with the License. |
| # You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| # See the License for the specific language governing permissions and |
| # limitations under the License. |
| |
| import unittest |
| |
| import cstruct |
| |
| |
| # These aren't constants, they're classes. So, pylint: disable=invalid-name |
| TestStructA = cstruct.Struct("TestStructA", "=BI", "byte1 int2") |
| TestStructB = cstruct.Struct("TestStructB", "=BI", "byte1 int2") |
| |
| |
| class CstructTest(unittest.TestCase): |
| |
| def CheckEquals(self, a, b): |
| self.assertEquals(a, b) |
| self.assertEquals(b, a) |
| assert a == b |
| assert b == a |
| assert not (a != b) # pylint: disable=g-comparison-negation,superfluous-parens |
| assert not (b != a) # pylint: disable=g-comparison-negation,superfluous-parens |
| |
| def CheckNotEquals(self, a, b): |
| self.assertNotEquals(a, b) |
| self.assertNotEquals(b, a) |
| assert a != b |
| assert b != a |
| assert not (a == b) # pylint: disable=g-comparison-negation,superfluous-parens |
| assert not (b == a) # pylint: disable=g-comparison-negation,superfluous-parens |
| |
| def testEqAndNe(self): |
| a1 = TestStructA((1, 2)) |
| a2 = TestStructA((2, 3)) |
| a3 = TestStructA((1, 2)) |
| b = TestStructB((1, 2)) |
| self.CheckNotEquals(a1, b) |
| self.CheckNotEquals(a2, b) |
| self.CheckNotEquals(a1, a2) |
| self.CheckNotEquals(a2, a3) |
| for i in [a1, a2, a3, b]: |
| self.CheckEquals(i, i) |
| self.CheckEquals(a1, a3) |
| |
| def testNestedStructs(self): |
| Nested = cstruct.Struct("Nested", "!HSSi", |
| "word1 nest2 nest3 int4", |
| [TestStructA, TestStructB]) |
| DoubleNested = cstruct.Struct("DoubleNested", "SSB", |
| "nest1 nest2 byte3", |
| [TestStructA, Nested]) |
| d = DoubleNested((TestStructA((1, 2)), |
| Nested((5, TestStructA((3, 4)), TestStructB((7, 8)), 9)), |
| 6)) |
| |
| expectedlen = (len(TestStructA) + |
| 2 + len(TestStructA) + len(TestStructB) + 4 + |
| 1) |
| self.assertEquals(expectedlen, len(DoubleNested)) |
| |
| self.assertEquals(7, d.nest2.nest3.byte1) |
| |
| d.byte3 = 252 |
| d.nest2.word1 = 33214 |
| n = d.nest2 |
| n.int4 = -55 |
| t = n.nest3 |
| t.int2 = 33627591 |
| |
| self.assertEquals(33627591, d.nest2.nest3.int2) |
| |
| expected = ( |
| "DoubleNested(nest1=TestStructA(byte1=1, int2=2)," |
| " nest2=Nested(word1=33214, nest2=TestStructA(byte1=3, int2=4)," |
| " nest3=TestStructB(byte1=7, int2=33627591), int4=-55), byte3=252)") |
| self.assertEquals(expected, str(d)) |
| expected = ("01" "02000000" |
| "81be" "03" "04000000" |
| "07" "c71d0102" "ffffffc9" "fc").decode("hex") |
| self.assertEquals(expected, d.Pack()) |
| unpacked = DoubleNested(expected) |
| self.CheckEquals(unpacked, d) |
| |
| def testNullTerminatedStrings(self): |
| TestStruct = cstruct.Struct("TestStruct", "B16si16AH", |
| "byte1 string2 int3 ascii4 word5") |
| nullstr = "hello" + (16 - len("hello")) * "\x00" |
| |
| t = TestStruct((2, nullstr, 12345, nullstr, 33210)) |
| expected = ("TestStruct(byte1=2, string2=68656c6c6f0000000000000000000000," |
| " int3=12345, ascii4=hello, word5=33210)") |
| self.assertEquals(expected, str(t)) |
| |
| embeddednull = "hello\x00visible123" |
| t = TestStruct((2, embeddednull, 12345, embeddednull, 33210)) |
| expected = ("TestStruct(byte1=2, string2=68656c6c6f0076697369626c65313233," |
| " int3=12345, ascii4=hello\x00visible123, word5=33210)") |
| self.assertEquals(expected, str(t)) |
| |
| def testZeroInitialization(self): |
| TestStruct = cstruct.Struct("TestStruct", "B16si16AH", |
| "byte1 string2 int3 ascii4 word5") |
| t = TestStruct() |
| self.assertEquals(0, t.byte1) |
| self.assertEquals("\x00" * 16, t.string2) |
| self.assertEquals(0, t.int3) |
| self.assertEquals("\x00" * 16, t.ascii4) |
| self.assertEquals(0, t.word5) |
| self.assertEquals("\x00" * len(TestStruct), t.Pack()) |
| |
| def testKeywordInitialization(self): |
| TestStruct = cstruct.Struct("TestStruct", "=B16sIH", |
| "byte1 string2 int3 word4") |
| text = "hello world! ^_^" |
| text_bytes = text.encode("hex") |
| |
| # Populate all fields |
| t1 = TestStruct(byte1=1, string2=text, int3=0xFEDCBA98, word4=0x1234) |
| expected = ("01" + text_bytes + "98BADCFE" "3412").decode("hex") |
| self.assertEquals(expected, t1.Pack()) |
| |
| # Partially populated |
| t1 = TestStruct(string2=text, word4=0x1234) |
| expected = ("00" + text_bytes + "00000000" "3412").decode("hex") |
| self.assertEquals(expected, t1.Pack()) |
| |
| def testCstructOffset(self): |
| TestStruct = cstruct.Struct("TestStruct", "B16si16AH", |
| "byte1 string2 int3 ascii4 word5") |
| nullstr = "hello" + (16 - len("hello")) * "\x00" |
| t = TestStruct((2, nullstr, 12345, nullstr, 33210)) |
| self.assertEquals(0, t.offset("byte1")) |
| self.assertEquals(1, t.offset("string2")) # sizeof(byte) |
| self.assertEquals(17, t.offset("int3")) # sizeof(byte) + 16*sizeof(char) |
| # The integer is automatically padded by the struct module |
| # to match native alignment. |
| # offset = sizeof(byte) + 16*sizeof(char) + padding + sizeof(int) |
| self.assertEquals(24, t.offset("ascii4")) |
| self.assertEquals(40, t.offset("word5")) |
| self.assertRaises(KeyError, t.offset, "random") |
| |
| # TODO: Add support for nested struct offset |
| Nested = cstruct.Struct("Nested", "!HSSi", "word1 nest2 nest3 int4", |
| [TestStructA, TestStructB]) |
| DoubleNested = cstruct.Struct("DoubleNested", "SSB", "nest1 nest2 byte3", |
| [TestStructA, Nested]) |
| d = DoubleNested((TestStructA((1, 2)), Nested((5, TestStructA((3, 4)), |
| TestStructB((7, 8)), 9)), 6)) |
| self.assertEqual(0, d.offset("nest1")) |
| self.assertEqual(len(TestStructA), d.offset("nest2")) |
| self.assertEqual(len(TestStructA) + len(Nested), d.offset("byte3")) |
| self.assertRaises(KeyError, t.offset, "word1") |
| |
| def testDefinitionFieldMismatch(self): |
| cstruct.Struct("TestA", "=BI", "byte1 int2") |
| cstruct.Struct("TestA", "=BxxxxxIx", "byte1 int2") |
| with self.assertRaises(ValueError): |
| cstruct.Struct("TestA", "=B", "byte1 int2") |
| with self.assertRaises(ValueError): |
| cstruct.Struct("TestA", "=BI", "byte1") |
| |
| Nested = cstruct.Struct("Nested", "!II", "int1 int2") |
| cstruct.Struct("TestB", "=BSI", "byte1 nest2 int3", [Nested]) |
| cstruct.Struct("TestB", "=BxSxIx", "byte1 nest2 int3", [Nested]) |
| with self.assertRaises(ValueError): |
| cstruct.Struct("TestB", "=BSI", "byte1 int3", [Nested]) |
| with self.assertRaises(ValueError): |
| cstruct.Struct("TestB", "=BSI", "byte1 nest2", [Nested]) |
| |
| cstruct.Struct("TestC", "=BSSI", "byte1 nest2 nest3 int4", [Nested, Nested]) |
| with self.assertRaises(ValueError): |
| cstruct.Struct("TestC", "=BSSI", "byte1 nest2 int4", [Nested, Nested]) |
| |
| |
| if __name__ == "__main__": |
| unittest.main() |