diff options
author | Yan Yan <evitayan@google.com> | 2021-04-02 18:58:06 +0000 |
---|---|---|
committer | Automerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com> | 2021-04-02 18:58:06 +0000 |
commit | 983ae625e727f0ce289f985a1d90f7d2424d03d8 (patch) | |
tree | 7d8eb8f1424f5a2cc90e06779c6d49e21d5cb233 | |
parent | 27308eafe35ffa51d9e9737a1a9e3aa229738dff (diff) | |
parent | c3bc6d71b3525ce39caccf096053e2acb31d5d0c (diff) | |
download | tests-983ae625e727f0ce289f985a1d90f7d2424d03d8.tar.gz |
Ignore ICMPv6 packets in _ExpectEspPacketOn am: c3bc6d71b3
Original change: https://android-review.googlesource.com/c/kernel/tests/+/1662144
Change-Id: I97f2050155136940e8d18ce063a041a515e5c138
-rw-r--r-- | net/test/xfrm_base.py | 13 |
1 files changed, 12 insertions, 1 deletions
diff --git a/net/test/xfrm_base.py b/net/test/xfrm_base.py index 03e15c4..e61322e 100644 --- a/net/test/xfrm_base.py +++ b/net/test/xfrm_base.py @@ -270,6 +270,13 @@ def DecryptPacketWithNull(packet): class XfrmBaseTest(multinetwork_base.MultiNetworkBaseTest): """Base test class for all XFRM-related testing.""" + def _isIcmpv6(self, payload): + if not isinstance(payload, scapy.IPv6): + return False + if payload.nh == IPPROTO_ICMPV6: + return True + return payload.nh == IPPROTO_HOPOPTS and payload.payload.nh == IPPROTO_ICMPV6 + def _ExpectEspPacketOn(self, netid, spi, seq, length, src_addr, dst_addr): """Read a packet from a netid and verify its properties. @@ -284,7 +291,11 @@ class XfrmBaseTest(multinetwork_base.MultiNetworkBaseTest): Returns: scapy.IP/IPv6: the read packet """ - packets = self.ReadAllPacketsOn(netid) + packets = [] + for packet in self.ReadAllPacketsOn(netid): + if not self._isIcmpv6(packet): + packets.append(packet) + self.assertEqual(1, len(packets)) packet = packets[0] if length is not None: |