def __init__(self, ip=None, mac=None):
super(Host, self).__init__(n_tips=1)
self.tip = self.tips[0]
self.nic = NIC(self.tip)
self.libsocket = LibSocket(self)
if mac is None:
mac = self.index
if ip is None:
ip = '1.0.0.{}'.format(self.index)
self.ip = ip_from_user(ip)
self.mac = mac_from_user(mac)
self.ip2mac = CacheTable(entry_duration=1000)
self.arp_table = self.ip2mac
self.avail_ports = defaultdict(lambda: 49152)
self.pending_packets = defaultdict(deque)
self.pending_frames = deque()
self.sockets = {}
mint.worker(self.run)
def __init__(self, n_tips=3):
super(Switch, self).__init__(n_tips)
self.nics = [NIC(tip) for tip in self.tips]
mac2port = CacheTable(entry_duration=1000)
self.ports = [
Port(
host=self,
nic=nic,
mac2port=mac2port,
index=i,
) for i, nic in enumerate(self.nics)
]
each(self.ports).calc_other_ports(self.ports)
mint.worker(self.run, priority=simulation.SWITCH_PRIORITY)
def __init__(self, ip=None, mac=None):
super(Host, self).__init__(n_tips=1)
self.tip = self.tips[0]
self.nic = NIC(self.tip)
self.libsocket = LibSocket(self)
if mac is None:
mac = self.index
if ip is None:
ip = '1.0.0.{}'.format(self.index)
self.ip = ip_from_user(ip)
self.mac = mac_from_user(mac)
self.ip2mac = CacheTable(entry_duration=1000)
self.arp_table = self.ip2mac
self.avail_ports = defaultdict(lambda: 49152)
self.pending_packets = defaultdict(deque)
self.pending_frames = deque()
self.sockets = {}
mint.worker(self.run)
class Host(Entity):
def __init__(self, ip=None, mac=None):
super(Host, self).__init__(n_tips=1)
self.tip = self.tips[0]
self.nic = NIC(self.tip)
self.libsocket = LibSocket(self)
if mac is None:
mac = self.index
if ip is None:
ip = '1.0.0.{}'.format(self.index)
self.ip = ip_from_user(ip)
self.mac = mac_from_user(mac)
self.ip2mac = CacheTable(entry_duration=1000)
self.arp_table = self.ip2mac
self.avail_ports = defaultdict(lambda: 49152)
self.pending_packets = defaultdict(deque)
self.pending_frames = deque()
self.sockets = {}
mint.worker(self.run)
def bind(self, sock, ip, port=None):
# TODO: more realistic implementation
if port is None:
port = port_from_user(self.avail_ports[sock.type])
self.avail_ports[type] += 1
self.sockets[(sock.type, self.ip, port)] = sock
return self.ip, port
def run(self):
while True:
self.send()
self.recv()
mint.elapse(1)
def send(self):
try:
frame = self.pending_frames[0]
self.nic.send(frame.raw, block=False)
self.pending_frames.popleft()
except IndexError:
pass
except Full:
pass
def recv(self):
try:
raw = self.nic.recv(block=False)
except Empty:
pass
else:
self.on_frame(raw)
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)
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)
def on_ipv4(self, frame):
packet = Packet(frame.payload)
src_ip = packet.src_ip
src_mac = frame.src_mac
self.ip2mac[src_ip] = src_mac
#mint.report('!!!')
#print repr(packet.protocol), Packet.Protocol[packet.protocol]
#print self.ip2mac
if packet.protocol == Packet.Protocol.UDP:
datagram = Datagram(packet.payload)
try:
sock_id = (self.libsocket.SOCK_DGRAM, packet.dst_ip,
datagram.dst_port)
sock = self.sockets[sock_id]
remote_addr = (packet.src_ip, datagram.src_port)
sock.feed_datagram(datagram.payload, remote_addr)
except KeyError:
pass
else:
mint.report('{}: unsupport protocol {}', self,
Packet.Protocol[packet.protocol])
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)
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)
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)
def sending(self, at):
return self.nic.sending
def sent(self, at):
return self.nic.sent
@property
def status(self):
nic = OrderedDict([
('ip', str(self.ip)),
('mac', str(self.mac)),
('nic', self.nic),
('tip', self.tip),
])
return [
('nic', nic),
('opdu', Discomposer(lambda: self.nic.odata)),
('ipdu', Discomposer(lambda: self.nic.idata)),
]
def __init__(self, n_interfaces):
super(EntityWithNIC, self).__init__(n_interfaces)
self.nics = [NIC(tip) for tip in self.tips]
self.nic = self.nics[0]
class Host(Entity):
def __init__(self, ip=None, mac=None):
super(Host, self).__init__(n_tips=1)
self.tip = self.tips[0]
self.nic = NIC(self.tip)
self.libsocket = LibSocket(self)
if mac is None:
mac = self.index
if ip is None:
ip = '1.0.0.{}'.format(self.index)
self.ip = ip_from_user(ip)
self.mac = mac_from_user(mac)
self.ip2mac = CacheTable(entry_duration=1000)
self.arp_table = self.ip2mac
self.avail_ports = defaultdict(lambda: 49152)
self.pending_packets = defaultdict(deque)
self.pending_frames = deque()
self.sockets = {}
mint.worker(self.run)
def bind(self, sock, ip, port=None):
# TODO: more realistic implementation
if port is None:
port = port_from_user(self.avail_ports[sock.type])
self.avail_ports[type] += 1
self.sockets[(sock.type, self.ip, port)] = sock
return self.ip, port
def run(self):
while True:
self.send()
self.recv()
mint.elapse(1)
def send(self):
try:
frame = self.pending_frames[0]
self.nic.send(frame.raw, block=False)
self.pending_frames.popleft()
except IndexError:
pass
except Full:
pass
def recv(self):
try:
raw = self.nic.recv(block=False)
except Empty:
pass
else:
self.on_frame(raw)
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)
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)
def on_ipv4(self, frame):
packet = Packet(frame.payload)
src_ip = packet.src_ip
src_mac = frame.src_mac
self.ip2mac[src_ip] = src_mac
#mint.report('!!!')
#print repr(packet.protocol), Packet.Protocol[packet.protocol]
#print self.ip2mac
if packet.protocol == Packet.Protocol.UDP:
datagram = Datagram(packet.payload)
try:
sock_id = (self.libsocket.SOCK_DGRAM,
packet.dst_ip, datagram.dst_port)
sock = self.sockets[sock_id]
remote_addr = (packet.src_ip, datagram.src_port)
sock.feed_datagram(datagram.payload, remote_addr)
except KeyError:
pass
else:
mint.report('{}: unsupport protocol {}',
self,
Packet.Protocol[packet.protocol])
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)
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)
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)
def sending(self, at):
return self.nic.sending
def sent(self, at):
return self.nic.sent
@property
def status(self):
nic = OrderedDict([
('ip', str(self.ip)),
('mac', str(self.mac)),
('nic', self.nic),
('tip', self.tip),
])
return [('nic', nic),
('opdu', Discomposer(lambda: self.nic.odata)),
('ipdu', Discomposer(lambda: self.nic.idata)),
]
