Python Host.receive Examples

Example #1

File: host_arp_reply_test.py Project: jefflai2/sts

 def test_arp_reply(self):
   '''Receive a valid ARP packet and ensure the correct reply'''
   interfaces = [HostInterface(EthAddr(H2_I1_ETH),[IPAddr(H2_I1_IP1),IPAddr(H2_I1_IP2)]),HostInterface(EthAddr(H2_I2_ETH),[IPAddr(H2_I2_IP)])]
   h = Host(interfaces)
   arp_req = arp()
   arp_req.hwsrc = EthAddr(H1_I1_ETH)
   arp_req.hwdst = EthAddr(b"\xff\xff\xff\xff\xff\xff")
   arp_req.opcode = arp.REQUEST
   arp_req.protosrc = IPAddr(H1_I1_IP)
   arp_req.protodst = IPAddr(H2_I1_IP1)
   ether = ethernet()
   ether.type = ethernet.ARP_TYPE
   ether.dst = EthAddr(b"\xff\xff\xff\xff\xff\xff")
   ether.src = EthAddr(H1_I1_ETH)
   ether.payload = arp_req
   # Get the action and arp reply packet
   arp_reply = h.receive(interfaces[0],ether)
   self.assertTrue(arp_reply is not None)
   self.assertEqual(arp_reply.src,EthAddr(H2_I1_ETH))
   self.assertEqual(arp_reply.dst,EthAddr(H1_I1_ETH))
   self.assertEqual(arp_reply.type,ethernet.ARP_TYPE)
   reply_payload = arp_reply.payload
   self.assertEqual(reply_payload.opcode,arp.REPLY)
   self.assertEqual(reply_payload.hwsrc,EthAddr(H2_I1_ETH))
   self.assertEqual(reply_payload.hwdst,EthAddr(H1_I1_ETH))
   self.assertTrue(reply_payload.protosrc == H2_I1_IP1)
   self.assertTrue(reply_payload.protodst == H1_I1_IP)

Example #2

File: host_arp_reply_test.py Project: jefflai2/sts

 def test_none_arp(self):
   '''Receive a non-ARP packet and ensure there is no reply'''
   interfaces = [HostInterface(EthAddr(H2_I1_ETH),[IPAddr(H2_I1_IP1),IPAddr(H2_I1_IP2)]),HostInterface(EthAddr(H2_I2_ETH),[IPAddr(H2_I2_IP)])]
   h = Host(interfaces)
   ether = ethernet()
   ether.type = ethernet.IP_TYPE
   ether.dst = EthAddr(H2_I1_ETH)
   ether.src = EthAddr(H1_I1_ETH)
   # Get the action and reply packet
   reply_packet = h.receive(interfaces[0],ether)
   self.assertTrue(reply_packet is None)

Example #3

File: host_arp_reply_test.py Project: jefflai2/sts

 def test_invalid_arp(self):
   '''Receive a ARP packet that isn't desinated to it and ensure there is no reply'''
   interfaces = [HostInterface(EthAddr(H2_I1_ETH),[IPAddr(H2_I1_IP1),IPAddr(H2_I1_IP2)]),HostInterface(EthAddr(H2_I2_ETH),[IPAddr(H2_I2_IP)])]
   h = Host(interfaces)
   arp_req = arp()
   arp_req.hwsrc = EthAddr(H1_I1_ETH)
   arp_req.hwdst = EthAddr(b"\xff\xff\xff\xff\xff\xff")
   arp_req.opcode = arp.REQUEST
   arp_req.protosrc = IPAddr(H1_I1_IP)
   arp_req.protodst = IPAddr(H3_I3_IP)
   ether = ethernet()
   ether.type = ethernet.ARP_TYPE
   ether.dst = EthAddr(b"\xff\xff\xff\xff\xff\xff")
   ether.src = EthAddr(H1_I1_ETH)
   ether.payload = arp_req
   # Get the action and reply packet
   reply_packet = h.receive(interfaces[0],ether)
   self.assertTrue(reply_packet is None)