2020-04-18 16:11:12 +01:00
|
|
|
"""
|
|
|
|
This is a script to run a small switch based scenario and depends on
|
|
|
|
the user running this script through the "Execute Python Script" option
|
|
|
|
in the GUI. The usage of globals() below allows this script to leverage the
|
|
|
|
same CoreEmu instance the GUI is using.
|
|
|
|
"""
|
|
|
|
|
2019-10-04 20:36:15 +01:00
|
|
|
import logging
|
2019-05-06 00:19:12 +01:00
|
|
|
|
2020-05-21 06:14:03 +01:00
|
|
|
from core.emulator.coreemu import CoreEmu
|
2020-06-16 20:50:24 +01:00
|
|
|
from core.emulator.data import IpPrefixes
|
2020-05-21 06:14:03 +01:00
|
|
|
from core.emulator.enumerations import EventTypes
|
|
|
|
from core.nodes.base import CoreNode
|
|
|
|
from core.nodes.network import SwitchNode
|
2019-02-16 17:50:19 +00:00
|
|
|
|
2020-04-18 16:11:12 +01:00
|
|
|
NODES = 2
|
|
|
|
|
2018-04-21 01:00:47 +01:00
|
|
|
|
2020-04-18 16:11:12 +01:00
|
|
|
def main():
|
2018-04-21 01:00:47 +01:00
|
|
|
# ip generator for example
|
2018-04-25 00:24:54 +01:00
|
|
|
prefixes = IpPrefixes("10.83.0.0/16")
|
2018-04-21 01:00:47 +01:00
|
|
|
|
|
|
|
# create emulator instance for creating sessions and utility methods
|
2020-05-21 06:14:03 +01:00
|
|
|
coreemu: CoreEmu = globals()["coreemu"]
|
2018-04-25 18:55:48 +01:00
|
|
|
session = coreemu.create_session()
|
2018-04-21 01:00:47 +01:00
|
|
|
|
|
|
|
# must be in configuration state for nodes to start, when using "node_add" below
|
2018-04-26 00:33:58 +01:00
|
|
|
session.set_state(EventTypes.CONFIGURATION_STATE)
|
2018-04-21 01:00:47 +01:00
|
|
|
|
|
|
|
# create switch network node
|
2020-05-21 06:14:03 +01:00
|
|
|
switch = session.add_node(SwitchNode)
|
2018-04-21 01:00:47 +01:00
|
|
|
|
|
|
|
# create nodes
|
2020-04-18 16:11:12 +01:00
|
|
|
for _ in range(NODES):
|
2020-05-21 06:14:03 +01:00
|
|
|
node = session.add_node(CoreNode)
|
2020-06-16 17:30:16 +01:00
|
|
|
interface = prefixes.create_iface(node)
|
|
|
|
session.add_link(node.id, switch.id, iface1_data=interface)
|
2018-04-21 01:00:47 +01:00
|
|
|
|
|
|
|
# instantiate session
|
|
|
|
session.instantiate()
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ in {"__main__", "__builtin__"}:
|
2019-10-04 20:36:15 +01:00
|
|
|
logging.basicConfig(level=logging.INFO)
|
2020-04-18 16:11:12 +01:00
|
|
|
main()
|