From a2d0b2cd66639e983b5a4fb191d8afe5cac77855 Mon Sep 17 00:00:00 2001 From: Andrew Lamoureux Date: Wed, 20 Mar 2019 15:35:00 -0400 Subject: kaitai: more format compatibility, testing --- python/examples/kaitai/ipv6_packet.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'python/examples/kaitai/ipv6_packet.py') diff --git a/python/examples/kaitai/ipv6_packet.py b/python/examples/kaitai/ipv6_packet.py index 3d828e64..70bac74f 100644 --- a/python/examples/kaitai/ipv6_packet.py +++ b/python/examples/kaitai/ipv6_packet.py @@ -8,7 +8,7 @@ import collections if parse_version(ks_version) < parse_version('0.7'): raise Exception("Incompatible Kaitai Struct Python API: 0.7 or later is required, but you have %s" % (ks_version)) -from protocol_body import ProtocolBody +from . import protocol_body class Ipv6Packet(KaitaiStruct): SEQ_FIELDS = ["version", "traffic_class", "flow_label", "payload_length", "next_header_type", "hop_limit", "src_ipv6_addr", "dst_ipv6_addr", "next_header", "rest"] def __init__(self, _io, _parent=None, _root=None): @@ -44,7 +44,7 @@ class Ipv6Packet(KaitaiStruct): self.dst_ipv6_addr = self._io.read_bytes(16) self._debug['dst_ipv6_addr']['end'] = self._io.pos() self._debug['next_header']['start'] = self._io.pos() - self.next_header = ProtocolBody(self.next_header_type, self._io) + self.next_header = protocol_body.ProtocolBody(self.next_header_type, self._io) self.next_header._read() self._debug['next_header']['end'] = self._io.pos() self._debug['rest']['start'] = self._io.pos() -- cgit v1.3.1