def test_tcp_filter_tcp_packet_not_in_src_range_or_dst_range(self):
rule = port_filter.PortRangeRule(protocol='TCP',
src_lo=0,
src_hi=1,
dst_lo=0,
dst_hi=1)
packet = FakePacket(protocol=socket.IPPROTO_TCP,
src_port=5,
dst_port=5)
self.assertFalse(rule.filter_condition(packet))
def test_udp_filter_udp_packet_in_src_range_and_dst_range(self):
rule = port_filter.PortRangeRule(protocol='UDP',
src_lo=0,
src_hi=1,
dst_lo=0,
dst_hi=1)
packet = FakePacket(protocol=socket.IPPROTO_UDP,
src_port=0,
dst_port=0)
self.assertTrue(rule.filter_condition(packet))
def test_tcp_filter_tcp_packet_in_dst_range(self):
rule = port_filter.PortRangeRule(protocol='TCP', dst_lo=0, dst_hi=1)
packet = FakePacket(protocol=socket.IPPROTO_TCP, dst_port=0)
self.assertTrue(rule.filter_condition(packet))
def test_tcp_filter_udp_packet(self):
rule = port_filter.PortRangeRule(protocol='TCP', src_lo=0, src_hi=1)
packet = FakePacket(protocol=socket.IPPROTO_UDP)
self.assertFalse(rule.filter_condition(packet))
def test_udp_filter_udp_packet_out_dst_range(self):
rule = port_filter.PortRangeRule(protocol='UDP', dst_lo=0, dst_hi=1)
packet = FakePacket(protocol=socket.IPPROTO_UDP, dst_port=5)
self.assertFalse(rule.filter_condition(packet))