Пример #1
0
    def test_it_should_not_return_traffic_to_the_previous_hop(self):
        l = Link('A')
        i = Interface(link=l)

        l.accept("This is a packet.", previous_hop=i)

        self.assertEqual(0, len(i.received()))
Пример #2
0
    def test_it_should_send_packets_to_a_link(self):
        l = Link('A')
        i = Interface(link=l)

        i.send("This is a packet.")

        self.assertEqual(1, len(l.forwarded()))
        self.assertEqual("This is a packet.", l.forwarded()[0])
Пример #3
0
    def test_it_should_get_received_packets_filtered_by_lambda(self):
        l = Link('A')
        i = Interface(link=l)

        l.accept(IPv6(src="2001:db8::2", dst="2001:db8::1")/ICMPv6EchoRequest())
        l.accept(IP(src="192.168.0.1", dst="192.168.0.2")/ICMP())
        
        self.assertEqual(1, len(i.received(lbda=lambda p: p.haslayer(IPv6))))
Пример #4
0
    def test_it_should_forward_traffic_to_an_interface(self):
        l = Link('A')
        i = Interface(link=l)
        j = Interface(link=l)

        l.accept("This is a packet.", previous_hop=i)
        
        self.assertEqual(1, len(j.received()))
        self.assertEqual("This is a packet.", j.received()[0])
Пример #5
0
    def test_it_should_get_received_packets_filtered_by_sequence(self):
        l = Link('A')
        i = Interface(link=l)

        l.accept(IPv6(src="2001:db8::2", dst="2001:db8::1")/ICMPv6EchoRequest(seq=2))
        l.accept(IPv6(src="2001:db8::1", dst="2001:db8::2")/ICMPv6EchoReply(seq=2))
        
        self.assertEqual(2, len(i.received(seq=2)))
        self.assertEqual(0, len(i.received(seq=1)))
Пример #6
0
    def test_it_should_get_received_packets_filtered_by_type(self):
        l = Link('A')
        i = Interface(link=l)

        l.accept(IPv6(src="2001:db8::2", dst="2001:db8::1")/ICMPv6EchoRequest())
        l.accept(IPv6(src="2001:db8::1", dst="2001:db8::2")/ICMPv6EchoReply())

        self.assertEqual(1, len(i.received(type=ICMPv6EchoRequest)))
        self.assertTrue(i.received(type=ICMPv6EchoRequest)[0].haslayer(ICMPv6EchoRequest))
Пример #7
0
    def test_it_should_strip_layer_2_frames_from_packets_on_receive(self):
        l = Link('A')
        i = Interface(link=l)

        l.accept(Ether()/IPv6())
        
        self.assertEqual(1, len(i.received()))
        self.assertFalse(i.received()[0].haslayer('Ether'))
        self.assertTrue(i.received()[0].haslayer('IPv6'))
Пример #8
0
    def test_it_should_encapsulate_packets_with_layer_2_frames_on_send(self):
        l = Link('A')
        i = Interface(link=l)

        i.send(IPv6())
        
        self.assertEqual(1, len(l.forwarded()))
        self.assertTrue(l.forwarded()[0].haslayer(Ether))
        self.assertTrue(l.forwarded()[0].haslayer(IPv6))
Пример #9
0
    def test_it_should_set_the_source_and_destination_mac_addresses_to_defaults_for_ethernet_link_layer(self):
        l = Link('A')
        i = Interface(link=l)

        i.send(IPv6())

        self.assertEqual(1, len(l.forwarded()))
        self.assertTrue(l.forwarded()[0].haslayer(Ether))
        self.assertEqual("00:00:00:00:00:00", l.forwarded()[0].getlayer(Ether).src)
        self.assertEqual("ff:ff:ff:ff:ff:ff", l.forwarded()[0].getlayer(Ether).dst)
Пример #10
0
    def test_it_should_overwrite_the_destination_mac_of_an_ethernet_frame(self):
        l = Link('A')
        i = Interface(link=l)
        j = MockInterface('if0', '00:b0:d0:86:bb:f7')
        k = TargetInterface(link_addr="00:b0:d0:bb:cc:ff")
        t = Tap(l, j, k)

        i.send(IPv6())

        self.assertEqual(1, len(j.sent))

        self.assertTrue(j.sent[0].haslayer(Ether))
        self.assertTrue(j.sent[0].haslayer(IPv6))
        self.assertEqual("00:b0:d0:bb:cc:ff", j.sent[0].getlayer(Ether).dst)
Пример #11
0
    def test_it_should_invoke_on_send_callbacks(self):
        l = Link('A')
        i = Interface(link=l)

        self.called = False
        self.packet = None
        def c(p, iface):
            self.called = True
            self.packet = p

        i.on_send(c)
        i.send("This is a packet.")

        self.assertTrue(self.called)
        self.assertEqual("This is a packet.", self.packet)
Пример #12
0
    def test_it_should_accept_packets_from_a_link(self):
        l = Link('A')
        i = Interface(link=l)
        j = Interface(link=l)

        i.send("This is a packet.")

        self.assertEqual(1, len(j.received()))
        self.assertEqual("This is a packet.", j.received()[0])
Пример #13
0
    def test_it_should_get_the_second_v6_global_ip(self):
        l = Link('A')
        i = Interface(link=l, ips=[IPAddress.identify("2001:500:88:200::10"), IPAddress.identify("2001:500:88:200::11"), IPAddress.identify("fe80:500:88:200::10"), IPAddress.identify("192.0.43.10")])

        self.assertEqual("2001:500:88:200::11", i.global_ip(offset=1).short_form())
Пример #14
0
    def test_it_should_get_the_default_v6_link_local_ip(self):
        l = Link('A')
        i = Interface(link=l, ips=[IPAddress.identify("2001:500:88:200::10"), IPAddress.identify("2001:500:88:200::11"), IPAddress.identify("fe80:500:88:200::10"), IPAddress.identify("192.0.43.10")])

        self.assertEqual("fe80:500:88:200::10", i.link_local_ip().short_form())