Esempio n. 1
0
 def on_frame(self, raw):
     frame = Frame(raw)
     self.mac2port[frame.src_mac] = self
     if frame.dst_mac in (frame.src_mac, MAC_Loopback):
         return
     try:
         port = self.mac2port[frame.dst_mac]
     except KeyError:
         self.flood(frame)
     else:
         port.send(frame)
Esempio n. 2
0
 def on_frame(self, raw):
     frame = Frame(raw)
     if frame.ethertype == Frame.EtherType.ARP:
         self.arp.process(frame.payload)
     elif frame.ethertype == Frame.EtherType.IPv4:
         self.on_ipv4(frame, interface=self)
     elif frame.ethertype in (Frame.EtherType.Raw, ):
         self.report(
             'unsupported frame type "{}", ignored',
             Frame.EtherType[frame.ethertype],
         )
     else:
         self.report('unknown frame type, ignored')
Esempio n. 3
0
 def send_packet(self, data, dst_ip, ethertype):
     try:
         dst_mac = self.ip2mac[dst_ip]
     except KeyError:
         self.pending_packets[dst_ip].append((data, ethertype))
         self.arp.query(dst_ip)
     else:
         frame = Frame(
             dst_mac=dst_mac,
             src_mac=self.mac,
             ethertype=ethertype,
             payload=data,
         )
         self.send_frame(frame)
Esempio n. 4
0
 def _send_pending_packets(self, ip, mac):
     ps = self.pending_packets[ip]
     if not ps:
         return
     self.report('send pending {} packet(s) for {}', len(ps), ip)
     while ps:
         p, ethertype = ps.popleft()
         f = Frame(
             ethertype=ethertype,
             dst_mac=mac,
             src_mac=self.mac,
             payload=p,
         )
         self.send_frame(f)
Esempio n. 5
0
 def on_frame(self, raw):
     frame = Frame(raw)
     self.mac2port[frame.src_mac] = self
     self.host.report('{} <= {}', self, frame.src_mac)
     if frame.dst_mac in (frame.src_mac, MAC_Loopback):
         self.host.report('loopback frame, ignored')
         return
     try:
         port = self.mac2port[frame.dst_mac]
     except KeyError:
         self.flood(frame)
     else:
         self.host.report('{} -> {}', self.index, port.index)
         port.send(frame)
Esempio n. 6
0
 def arp_respond(self, request):
     packet = ARP(
         oper=ARP.Oper.IAm,
         src_ip=self.ip,
         src_mac=self.mac,
         dst_ip=request.src_ip,
         dst_mac=request.src_mac,
     )
     frame = Frame(
         dst_mac=request.src_mac,
         src_mac=self.mac,
         ethertype=Frame.EtherType.ARP,
         payload=packet.raw,
     )
     self.pending_frames.append(frame)
Esempio n. 7
0
 def arp_query(self, dst_ip):
     packet = ARP(
         oper=ARP.Oper.WhoIs,
         src_ip=self.ip,
         src_mac=self.mac,
         dst_ip=dst_ip,
         dst_mac=MAC_Invalid,
     )
     frame = Frame(
         dst_mac=MAC_Broadcast,
         src_mac=self.mac,
         ethertype=Frame.EtherType.ARP,
         payload=packet.raw,
     )
     self.pending_frames.append(frame)
Esempio n. 8
0
 def send_packet(self, packet, ethertype=Frame.EtherType.IPv4):
     try:
         dst_ip = packet.dst_ip
         dst_mac = self.ip2mac[dst_ip]
     except KeyError:
         self.pending_packets[dst_ip].append((packet, ethertype))
         self.arp_query(dst_ip)
     else:
         frame = Frame(
             dst_mac=dst_mac,
             src_mac=self.mac,
             ethertype=ethertype,
             payload=packet.raw,
         )
         self.pending_frames.append(frame)
Esempio n. 9
0
 def query(self, dst_ip):
     pdu = PDU(
         oper=PDU.Oper.WhoIs,
         src_ip=self.host.ip,
         src_mac=self.host.mac,
         dst_ip=dst_ip,
         dst_mac=MAC_Invalid,
     )
     frame = Frame(
         dst_mac=MAC_Broadcast,
         src_mac=self.host.mac,
         ethertype=Frame.EtherType.ARP,
         payload=pdu.raw,
     )
     self.send_frame(frame)
     self.report('Who has {}? Tell {}', pdu.dst_ip, pdu.src_ip)
Esempio n. 10
0
 def run(self):
     while True:
         self.routes.remove_expired_entries()
         for iport in self.ports:
             try:
                 raw = iport.recv(block=False)
             except Empty:
                 continue
             frame = Frame(raw)
             self.routes.register(frame.src, iport)
             oports = self.routes.get(frame.dst, exclude=iport)
             for oport in oports:
                 try:
                     oport.send(raw, block=False)
                 except Full:
                     mint.report('frame dropped {}', repr(raw))
         mint.elapse(1)
Esempio n. 11
0
 def send_packet(self, data, dst_ip, ethertype):
     if not dst_ip:
         self.report('invalid dst_ip {}, ignored', dst_ip)
         return
     try:
         dst_mac = self.ip2mac[dst_ip]
     except KeyError:
         self.pending_packets[dst_ip].append((data, ethertype))
         self.arp.query(dst_ip)
     else:
         frame = Frame(
             dst_mac=dst_mac,
             src_mac=self.mac,
             ethertype=ethertype,
             payload=data,
         )
         self.send_frame(frame)
Esempio n. 12
0
 def on_arp(self, raw):
     arp = ARP(raw)
     self.ip2mac[arp.src_ip] = arp.src_mac
     if arp.oper == ARP.Oper.IAm:
         packets = self.pending_packets[arp.src_ip]
         while packets:
             packet, ethertype = packets.popleft()
             frame = Frame(
                 ethertype=ethertype,
                 dst_mac=arp.src_mac,
                 src_mac=self.mac,
                 payload=packet.raw,
             )
             self.pending_frames.append(frame)
         return
     if arp.dst_ip != self.ip:
         return
     if arp.oper == ARP.Oper.WhoIs:
         self.arp_respond(arp)
Esempio n. 13
0
 def respond(self, request):
     pdu = PDU(
         oper=PDU.Oper.IAm,
         src_ip=self.host.ip,
         src_mac=self.host.mac,
         dst_ip=request.src_ip,
         dst_mac=request.src_mac,
     )
     frame = Frame(
         dst_mac=request.src_mac,
         src_mac=self.host.mac,
         ethertype=Frame.EtherType.ARP,
         payload=pdu.raw,
     )
     self.send_frame(frame)
     self.report(
         'tell {} that {} is {}',
         pdu.dst_ip,
         pdu.src_mac,
         pdu.src_ip,
     )
Esempio n. 14
0
 def on_frame(self, raw):
     frame = Frame(raw)
     if frame.ethertype == Frame.EtherType.ARP:
         self.on_arp(frame.payload)
     elif frame.ethertype == Frame.EtherType.IPv4:
         self.on_ipv4(frame)