core-extra/docs/python.md

5.1 KiB

Python API

  • Table of Contents {:toc}

Overview

Writing your own Python scripts offers a rich programming environment with complete control over all aspects of the emulation. This chapter provides a brief introduction to scripting. Most of the documentation is available from sample scripts, or online via interactive Python.

The best starting point is the sample scripts that are included with CORE. If you have a CORE source tree, the example script files can be found under core/daemon/examples/python/. When CORE is installed from packages, the example script files will be in /usr/share/core/examples/python/ (or /usr/local/ prefix when installed from source.) For the most part, the example scripts are self-documenting; see the comments contained within the Python code.

The scripts should be run with root privileges because they create new network namespaces. In general, a CORE Python script does not connect to the CORE daemon, in fact the core-daemon is just another Python script that uses the CORE Python modules and exchanges messages with the GUI. To connect the GUI to your scripts, see the included sample scripts that allow for GUI connections.

Here are the basic elements of a CORE Python script:

"""
This is a standalone script to run a small switch based scenario and will not
interact with the GUI.
"""

import logging

from core.emulator.coreemu import CoreEmu
from core.emulator.data import IpPrefixes
from core.emulator.enumerations import EventTypes
from core.nodes.base import CoreNode
from core.nodes.network import SwitchNode

NODES = 2


def main():
    # ip generator for example
    prefixes = IpPrefixes(ip4_prefix="10.83.0.0/16")

    # create emulator instance for creating sessions and utility methods
    coreemu = CoreEmu()
    session = coreemu.create_session()

    # must be in configuration state for nodes to start, when using "node_add" below
    session.set_state(EventTypes.CONFIGURATION_STATE)

    # create switch network node
    switch = session.add_node(SwitchNode, _id=100)

    # create nodes
    for _ in range(NODES):
        node = session.add_node(CoreNode)
        iface_data = prefixes.create_iface(node)
        session.add_link(node.id, switch.id, iface1_data=iface_data)

    # instantiate session
    session.instantiate()

    # run any desired logic here

    # shutdown session
    coreemu.shutdown()


if __name__ == "__main__":
    logging.basicConfig(level=logging.INFO)
    main()

The above script creates a CORE session having two nodes connected with a switch, Then immediately shutsdown.

The CORE Python modules are documented with comments in the code. From an interactive Python shell, you can retrieve online help about the various classes and methods; for example help(CoreNode) or help(Session).

NOTE: The CORE daemon core-daemon manages a list of sessions and allows the GUI to connect and control sessions. Your Python script uses the same CORE modules but runs independently of the daemon. The daemon does not need to be running for your script to work.

The session created by a Python script may be viewed in the GUI if certain steps are followed. The GUI has a File Menu, Execute Python script... option for running a script and automatically connecting to it. Once connected, normal GUI interaction is possible, such as moving and double-clicking nodes, activating Widgets, etc.

The script should have a line such as the following for running it from the GUI.

if __name__ in ["__main__", "__builtin__"]:
    main()

A script can add sessions to the core-daemon. A global coreemu variable is exposed to the script pointing to the CoreEmu object.

The example below has a fallback to a new CoreEmu object, in the case you would like to run the script standalone, outside of the core-daemon.

coreemu = globals().get("coreemu", CoreEmu())
session = coreemu.create_session()

Finally, nodes and networks need to have their coordinates set to something, otherwise they will be grouped at the coordinates <0, 0>. First sketching the topology in the GUI and then using the Export Python script option may help here.

switch.setposition(x=80,y=50)

A fully-worked example script that you can launch from the GUI is available in the examples directory.

Configuring Services

Examples setting or configuring custom services for a node.

# create session and node
coreemu = CoreEmu()
session = coreemu.create_session()

# create node with custom services
options = NodeOptions(services=["ServiceName"])
node = session.add_node(CoreNode, options=options)

# set custom file data
session.services.set_service_file(node.id, "ServiceName", "FileName", "custom file data")

Configuring EMANE Models

Examples for configuring custom emane model settings.

# create session and emane network
coreemu = CoreEmu()
session = coreemu.create_session()
session.set_location(47.57917, -122.13232, 2.00000, 1.0)
options = NodeOptions()
options.set_position(80, 50)
emane_network = session.add_node(EmaneNet, options=options)

# set custom emane model config defaults
session.emane.set_model(emane_network, EmaneIeee80211abgModel)