155 lines
6.6 KiB
Python
155 lines
6.6 KiB
Python
|
from primaite.simulator.network.container import Network
|
||
|
|
from primaite.simulator.network.hardware.base import Switch, NIC
|
||
|
|
from primaite.simulator.network.hardware.nodes.computer import Computer
|
||
|
|
from primaite.simulator.network.hardware.nodes.router import Router, ACLAction
|
||
|
|
from primaite.simulator.network.transmission.network_layer import IPProtocol
|
||
|
|
from primaite.simulator.network.transmission.transport_layer import Port
|
||
|
|
|
||
|
|
|
||
|
|
def arcd_uc2_network() -> Network:
|
||
|
|
"""
|
||
|
|
Models the ARCD Use Case 2 Network.
|
||
|
|
|
||
|
|
+------------+
|
||
|
|
| domain_ |
|
||
|
|
+------------+ controller |
|
||
|
|
| | |
|
||
|
|
| +------------+
|
||
|
|
|
|
||
|
|
|
|
||
|
|
+------------+ | +------------+
|
||
|
|
| | | | |
|
||
|
|
| client_1 +---------+ | +---------+ web_server |
|
||
|
|
| | | | | | |
|
||
|
|
+------------+ | | | +------------+
|
||
|
|
+--+---------+ +------------+ +------+--+--+
|
||
|
|
| | | | | |
|
||
|
|
| switch_2 +------+ router_1 +------+ switch_1 |
|
||
|
|
| | | | | |
|
||
|
|
+--+------+--+ +------------+ +--+---+--+--+
|
||
|
|
+------------+ | | | | | +------------+
|
||
|
|
| | | | | | | | database |
|
||
|
|
| client_2 +---------+ | | | +---------+ _server |
|
||
|
|
| | | | | | |
|
||
|
|
+------------+ | | | +------------+
|
||
|
|
| +------------+ | |
|
||
|
|
| | security | | |
|
||
|
|
+---------+ _suite +---------+ | +------------+
|
||
|
|
| | | | backup_ |
|
||
|
|
+------------+ +------------+ server |
|
||
|
|
| |
|
||
|
|
+------------+
|
||
|
|
|
||
|
|
Example:
|
||
|
|
>>> network = arcd_uc2_network()
|
||
|
|
>>> network.get_node_by_hostname("client_1").ping("192.168.1.10")
|
||
|
|
|
||
|
|
"""
|
||
|
|
network = Network()
|
||
|
|
|
||
|
|
# Router 1
|
||
|
|
router_1 = Router(hostname="router_1", num_ports=5)
|
||
|
|
router_1.power_on()
|
||
|
|
router_1.configure_port(port=1, ip_address="192.168.1.1", subnet_mask="255.255.255.0")
|
||
|
|
router_1.configure_port(port=2, ip_address="192.168.10.1", subnet_mask="255.255.255.0")
|
||
|
|
|
||
|
|
# Switch 1
|
||
|
|
switch_1 = Switch(hostname="switch_1", num_ports=8)
|
||
|
|
switch_1.power_on()
|
||
|
|
network.connect(endpoint_a=router_1.ethernet_ports[1], endpoint_b=switch_1.switch_ports[8])
|
||
|
|
router_1.enable_port(1)
|
||
|
|
|
||
|
|
# Switch 2
|
||
|
|
switch_2 = Switch(hostname="switch_2", num_ports=8)
|
||
|
|
switch_2.power_on()
|
||
|
|
network.connect(endpoint_a=router_1.ethernet_ports[2], endpoint_b=switch_2.switch_ports[8])
|
||
|
|
router_1.enable_port(2)
|
||
|
|
|
||
|
|
# Client 1
|
||
|
|
client_1 = Computer(
|
||
|
|
hostname="client_1",
|
||
|
|
ip_address="192.168.10.21",
|
||
|
|
subnet_mask="255.255.255.0",
|
||
|
|
default_gateway="192.168.10.1"
|
||
|
|
)
|
||
|
|
client_1.power_on()
|
||
|
|
network.connect(endpoint_a=client_1, endpoint_b=switch_2.switch_ports[1])
|
||
|
|
|
||
|
|
# Client 2
|
||
|
|
client_2 = Computer(
|
||
|
|
hostname="client_2",
|
||
|
|
ip_address="192.168.10.22",
|
||
|
|
subnet_mask="255.255.255.0",
|
||
|
|
default_gateway="192.168.10.1"
|
||
|
|
)
|
||
|
|
client_2.power_on()
|
||
|
|
network.connect(endpoint_a=client_2, endpoint_b=switch_2.switch_ports[2])
|
||
|
|
|
||
|
|
# Domain Controller
|
||
|
|
domain_controller = Computer(
|
||
|
|
hostname="domain_controller",
|
||
|
|
ip_address="192.168.1.10",
|
||
|
|
subnet_mask="255.255.255.0",
|
||
|
|
default_gateway="192.168.1.1"
|
||
|
|
)
|
||
|
|
domain_controller.power_on()
|
||
|
|
network.connect(endpoint_a=domain_controller, endpoint_b=switch_1.switch_ports[1])
|
||
|
|
|
||
|
|
# Web Server
|
||
|
|
web_server = Computer(
|
||
|
|
hostname="web_server",
|
||
|
|
ip_address="192.168.1.12",
|
||
|
|
subnet_mask="255.255.255.0",
|
||
|
|
default_gateway="192.168.1.1"
|
||
|
|
)
|
||
|
|
web_server.power_on()
|
||
|
|
network.connect(endpoint_a=web_server, endpoint_b=switch_1.switch_ports[2])
|
||
|
|
|
||
|
|
# Database Server
|
||
|
|
database_server = Computer(
|
||
|
|
hostname="database_server",
|
||
|
|
ip_address="192.168.1.14",
|
||
|
|
subnet_mask="255.255.255.0",
|
||
|
|
default_gateway="192.168.1.1"
|
||
|
|
)
|
||
|
|
database_server.power_on()
|
||
|
|
network.connect(endpoint_a=database_server, endpoint_b=switch_1.switch_ports[3])
|
||
|
|
|
||
|
|
# Backup Server
|
||
|
|
backup_server = Computer(
|
||
|
|
hostname="backup_server",
|
||
|
|
ip_address="192.168.1.16",
|
||
|
|
subnet_mask="255.255.255.0",
|
||
|
|
default_gateway="192.168.1.1"
|
||
|
|
)
|
||
|
|
backup_server.power_on()
|
||
|
|
network.connect(endpoint_a=backup_server, endpoint_b=switch_1.switch_ports[4])
|
||
|
|
|
||
|
|
# Security Suite
|
||
|
|
security_suite = Computer(
|
||
|
|
hostname="security_suite",
|
||
|
|
ip_address="192.168.1.110",
|
||
|
|
subnet_mask="255.255.255.0",
|
||
|
|
default_gateway="192.168.1.1"
|
||
|
|
)
|
||
|
|
security_suite.power_on()
|
||
|
|
network.connect(endpoint_a=security_suite, endpoint_b=switch_1.switch_ports[7])
|
||
|
|
security_suite_external_nic = NIC(ip_address="192.168.10.110", subnet_mask="255.255.255.0")
|
||
|
|
security_suite.connect_nic(security_suite_external_nic)
|
||
|
|
network.connect(endpoint_a=security_suite_external_nic, endpoint_b=switch_2.switch_ports[7])
|
||
|
|
|
||
|
|
router_1.acl.add_rule(
|
||
|
|
action=ACLAction.PERMIT,
|
||
|
|
src_port=Port.ARP,
|
||
|
|
dst_port=Port.ARP,
|
||
|
|
position=22
|
||
|
|
)
|
||
|
|
|
||
|
|
router_1.acl.add_rule(
|
||
|
|
action=ACLAction.PERMIT,
|
||
|
|
protocol=IPProtocol.ICMP,
|
||
|
|
position=23
|
||
|
|
)
|
||
|
|
|
||
|
|
return network
|