Merge branch 'develop' into dependabot/pip/daemon/pyyaml-5.4
This commit is contained in:
commit
750489c2e1
145 changed files with 4867 additions and 9313 deletions
101
Dockerfile
Normal file
101
Dockerfile
Normal file
|
@ -0,0 +1,101 @@
|
||||||
|
# syntax=docker/dockerfile:1
|
||||||
|
FROM ubuntu:20.04
|
||||||
|
LABEL Description="CORE Docker Image"
|
||||||
|
|
||||||
|
# define variables
|
||||||
|
ARG DEBIAN_FRONTEND=noninteractive
|
||||||
|
ARG PREFIX=/usr/local
|
||||||
|
ARG BRANCH=develop
|
||||||
|
ARG CORE_TARBALL=core.tar.gz
|
||||||
|
ARG OSPF_TARBALL=ospf.tar.gz
|
||||||
|
|
||||||
|
# install system dependencies
|
||||||
|
RUN apt-get update && \
|
||||||
|
apt-get install -y --no-install-recommends \
|
||||||
|
automake \
|
||||||
|
bash \
|
||||||
|
ca-certificates \
|
||||||
|
ethtool \
|
||||||
|
gawk \
|
||||||
|
gcc \
|
||||||
|
g++ \
|
||||||
|
iproute2 \
|
||||||
|
iputils-ping \
|
||||||
|
libc-dev \
|
||||||
|
libev-dev \
|
||||||
|
libreadline-dev \
|
||||||
|
libtool \
|
||||||
|
libtk-img \
|
||||||
|
make \
|
||||||
|
nftables \
|
||||||
|
python3 \
|
||||||
|
python3-pip \
|
||||||
|
python3-tk \
|
||||||
|
pkg-config \
|
||||||
|
systemctl \
|
||||||
|
tk \
|
||||||
|
wget \
|
||||||
|
xauth \
|
||||||
|
xterm \
|
||||||
|
&& apt-get clean
|
||||||
|
# install python dependencies
|
||||||
|
RUN python3 -m pip install \
|
||||||
|
grpcio==1.27.2 \
|
||||||
|
grpcio-tools==1.27.2 \
|
||||||
|
poetry==1.1.7
|
||||||
|
# retrieve, build, and install core
|
||||||
|
RUN wget -q -O ${CORE_TARBALL} https://api.github.com/repos/coreemu/core/tarball/${BRANCH} && \
|
||||||
|
tar xf ${CORE_TARBALL} && \
|
||||||
|
cd coreemu-core* && \
|
||||||
|
./bootstrap.sh && \
|
||||||
|
./configure && \
|
||||||
|
make -j $(nproc) && \
|
||||||
|
make install && \
|
||||||
|
cd daemon && \
|
||||||
|
python3 -m poetry build -f wheel && \
|
||||||
|
python3 -m pip install dist/* && \
|
||||||
|
cp scripts/* ${PREFIX}/bin && \
|
||||||
|
mkdir /etc/core && \
|
||||||
|
cp -n data/core.conf /etc/core && \
|
||||||
|
cp -n data/logging.conf /etc/core && \
|
||||||
|
mkdir -p ${PREFIX}/share/core && \
|
||||||
|
cp -r examples ${PREFIX}/share/core && \
|
||||||
|
echo '\
|
||||||
|
[Unit]\n\
|
||||||
|
Description=Common Open Research Emulator Service\n\
|
||||||
|
After=network.target\n\
|
||||||
|
\n\
|
||||||
|
[Service]\n\
|
||||||
|
Type=simple\n\
|
||||||
|
ExecStart=/usr/local/bin/core-daemon\n\
|
||||||
|
TasksMax=infinity\n\
|
||||||
|
\n\
|
||||||
|
[Install]\n\
|
||||||
|
WantedBy=multi-user.target\
|
||||||
|
' > /lib/systemd/system/core-daemon.service && \
|
||||||
|
cd ../.. && \
|
||||||
|
rm ${CORE_TARBALL} && \
|
||||||
|
rm -rf coreemu-core*
|
||||||
|
# retrieve, build, and install ospf mdr
|
||||||
|
RUN wget -q -O ${OSPF_TARBALL} https://github.com/USNavalResearchLaboratory/ospf-mdr/tarball/master && \
|
||||||
|
tar xf ${OSPF_TARBALL} && \
|
||||||
|
cd USNavalResearchLaboratory-ospf-mdr* && \
|
||||||
|
./bootstrap.sh && \
|
||||||
|
./configure --disable-doc --enable-user=root --enable-group=root \
|
||||||
|
--with-cflags=-ggdb --sysconfdir=/usr/local/etc/quagga --enable-vtysh \
|
||||||
|
--localstatedir=/var/run/quagga && \
|
||||||
|
make -j $(nproc) && \
|
||||||
|
make install && \
|
||||||
|
cd .. && \
|
||||||
|
rm ${OSPF_TARBALL} && \
|
||||||
|
rm -rf USNavalResearchLaboratory-ospf-mdr*
|
||||||
|
# retrieve and install emane packages
|
||||||
|
RUN wget -q https://adjacentlink.com/downloads/emane/emane-1.2.7-release-1.ubuntu-20_04.amd64.tar.gz && \
|
||||||
|
tar xf emane*.tar.gz && \
|
||||||
|
cd emane-1.2.7-release-1/debs/ubuntu-20_04/amd64 && \
|
||||||
|
apt-get install -y ./emane*.deb ./python3-emane_*.deb && \
|
||||||
|
cd ../../../.. && \
|
||||||
|
rm emane-1.2.7-release-1.ubuntu-20_04.amd64.tar.gz && \
|
||||||
|
rm -rf emane-1.2.7-release-1
|
||||||
|
CMD ["systemctl", "start", "core-daemon"]
|
||||||
|
# sudo docker run -itd --name core -e DISPLAY -v /tmp/.X11-unix:/tmp/.X11-unix:rw --privileged core
|
|
@ -57,7 +57,7 @@ fpm -s dir -t deb -n core-distributed \
|
||||||
-d "procps" \
|
-d "procps" \
|
||||||
-d "libc6 >= 2.14" \
|
-d "libc6 >= 2.14" \
|
||||||
-d "bash >= 3.0" \
|
-d "bash >= 3.0" \
|
||||||
-d "ebtables" \
|
-d "nftables" \
|
||||||
-d "iproute2" \
|
-d "iproute2" \
|
||||||
-d "libev4" \
|
-d "libev4" \
|
||||||
-d "openssh-server" \
|
-d "openssh-server" \
|
||||||
|
@ -77,7 +77,7 @@ fpm -s dir -t rpm -n core-distributed \
|
||||||
-d "ethtool" \
|
-d "ethtool" \
|
||||||
-d "procps-ng" \
|
-d "procps-ng" \
|
||||||
-d "bash >= 3.0" \
|
-d "bash >= 3.0" \
|
||||||
-d "ebtables" \
|
-d "nftables" \
|
||||||
-d "iproute" \
|
-d "iproute" \
|
||||||
-d "libev" \
|
-d "libev" \
|
||||||
-d "net-tools" \
|
-d "net-tools" \
|
||||||
|
@ -123,7 +123,7 @@ all: change-files
|
||||||
|
|
||||||
.PHONY: change-files
|
.PHONY: change-files
|
||||||
change-files:
|
change-files:
|
||||||
$(call change-files,gui/core-gui)
|
$(call change-files,gui/core-gui-legacy)
|
||||||
$(call change-files,daemon/core/constants.py)
|
$(call change-files,daemon/core/constants.py)
|
||||||
$(call change-files,netns/setup.py)
|
$(call change-files,netns/setup.py)
|
||||||
|
|
||||||
|
|
|
@ -2,7 +2,7 @@
|
||||||
# Process this file with autoconf to produce a configure script.
|
# Process this file with autoconf to produce a configure script.
|
||||||
|
|
||||||
# this defines the CORE version number, must be static for AC_INIT
|
# this defines the CORE version number, must be static for AC_INIT
|
||||||
AC_INIT(core, 7.5.2)
|
AC_INIT(core, 8.0.0)
|
||||||
|
|
||||||
# autoconf and automake initialization
|
# autoconf and automake initialization
|
||||||
AC_CONFIG_SRCDIR([netns/version.h.in])
|
AC_CONFIG_SRCDIR([netns/version.h.in])
|
||||||
|
@ -123,9 +123,9 @@ if test "x$enable_daemon" = "xyes"; then
|
||||||
AC_MSG_ERROR([Could not locate sysctl (from procps package).])
|
AC_MSG_ERROR([Could not locate sysctl (from procps package).])
|
||||||
fi
|
fi
|
||||||
|
|
||||||
AC_CHECK_PROG(ebtables_path, ebtables, $as_dir, no, $SEARCHPATH)
|
AC_CHECK_PROG(nftables_path, nft, $as_dir, no, $SEARCHPATH)
|
||||||
if test "x$ebtables_path" = "xno" ; then
|
if test "x$nftables_path" = "xno" ; then
|
||||||
AC_MSG_ERROR([Could not locate ebtables (from ebtables package).])
|
AC_MSG_ERROR([Could not locate nftables (from nftables package).])
|
||||||
fi
|
fi
|
||||||
|
|
||||||
AC_CHECK_PROG(ip_path, ip, $as_dir, no, $SEARCHPATH)
|
AC_CHECK_PROG(ip_path, ip, $as_dir, no, $SEARCHPATH)
|
||||||
|
|
|
@ -2,6 +2,3 @@ import logging.config
|
||||||
|
|
||||||
# setup default null handler
|
# setup default null handler
|
||||||
logging.getLogger(__name__).addHandler(logging.NullHandler())
|
logging.getLogger(__name__).addHandler(logging.NullHandler())
|
||||||
|
|
||||||
# disable paramiko logging
|
|
||||||
logging.getLogger("paramiko").setLevel(logging.WARNING)
|
|
||||||
|
|
File diff suppressed because it is too large
Load diff
File diff suppressed because it is too large
Load diff
|
@ -14,6 +14,8 @@ from core.emulator.data import (
|
||||||
)
|
)
|
||||||
from core.emulator.session import Session
|
from core.emulator.session import Session
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
def handle_node_event(node_data: NodeData) -> core_pb2.Event:
|
def handle_node_event(node_data: NodeData) -> core_pb2.Event:
|
||||||
"""
|
"""
|
||||||
|
@ -199,7 +201,7 @@ class EventStreamer:
|
||||||
elif isinstance(data, FileData):
|
elif isinstance(data, FileData):
|
||||||
event = handle_file_event(data)
|
event = handle_file_event(data)
|
||||||
else:
|
else:
|
||||||
logging.error("unknown event: %s", data)
|
logger.error("unknown event: %s", data)
|
||||||
except Empty:
|
except Empty:
|
||||||
pass
|
pass
|
||||||
if event:
|
if event:
|
||||||
|
|
|
@ -7,10 +7,9 @@ import grpc
|
||||||
from grpc import ServicerContext
|
from grpc import ServicerContext
|
||||||
|
|
||||||
from core import utils
|
from core import utils
|
||||||
from core.api.grpc import common_pb2, core_pb2
|
from core.api.grpc import common_pb2, core_pb2, wrappers
|
||||||
from core.api.grpc.common_pb2 import MappedConfig
|
|
||||||
from core.api.grpc.configservices_pb2 import ConfigServiceConfig
|
from core.api.grpc.configservices_pb2 import ConfigServiceConfig
|
||||||
from core.api.grpc.emane_pb2 import GetEmaneModelConfig
|
from core.api.grpc.emane_pb2 import NodeEmaneConfig
|
||||||
from core.api.grpc.services_pb2 import (
|
from core.api.grpc.services_pb2 import (
|
||||||
NodeServiceConfig,
|
NodeServiceConfig,
|
||||||
NodeServiceData,
|
NodeServiceData,
|
||||||
|
@ -28,9 +27,10 @@ from core.nodes.base import CoreNode, CoreNodeBase, NodeBase
|
||||||
from core.nodes.docker import DockerNode
|
from core.nodes.docker import DockerNode
|
||||||
from core.nodes.interface import CoreInterface
|
from core.nodes.interface import CoreInterface
|
||||||
from core.nodes.lxd import LxcNode
|
from core.nodes.lxd import LxcNode
|
||||||
from core.nodes.network import WlanNode
|
from core.nodes.network import CtrlNet, PtpNet, WlanNode
|
||||||
from core.services.coreservices import CoreService
|
from core.services.coreservices import CoreService
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
WORKERS = 10
|
WORKERS = 10
|
||||||
|
|
||||||
|
|
||||||
|
@ -156,7 +156,7 @@ def create_nodes(
|
||||||
start = time.monotonic()
|
start = time.monotonic()
|
||||||
results, exceptions = utils.threadpool(funcs)
|
results, exceptions = utils.threadpool(funcs)
|
||||||
total = time.monotonic() - start
|
total = time.monotonic() - start
|
||||||
logging.debug("grpc created nodes time: %s", total)
|
logger.debug("grpc created nodes time: %s", total)
|
||||||
return results, exceptions
|
return results, exceptions
|
||||||
|
|
||||||
|
|
||||||
|
@ -180,7 +180,7 @@ def create_links(
|
||||||
start = time.monotonic()
|
start = time.monotonic()
|
||||||
results, exceptions = utils.threadpool(funcs)
|
results, exceptions = utils.threadpool(funcs)
|
||||||
total = time.monotonic() - start
|
total = time.monotonic() - start
|
||||||
logging.debug("grpc created links time: %s", total)
|
logger.debug("grpc created links time: %s", total)
|
||||||
return results, exceptions
|
return results, exceptions
|
||||||
|
|
||||||
|
|
||||||
|
@ -204,7 +204,7 @@ def edit_links(
|
||||||
start = time.monotonic()
|
start = time.monotonic()
|
||||||
results, exceptions = utils.threadpool(funcs)
|
results, exceptions = utils.threadpool(funcs)
|
||||||
total = time.monotonic() - start
|
total = time.monotonic() - start
|
||||||
logging.debug("grpc edit links time: %s", total)
|
logger.debug("grpc edit links time: %s", total)
|
||||||
return results, exceptions
|
return results, exceptions
|
||||||
|
|
||||||
|
|
||||||
|
@ -251,12 +251,15 @@ def get_config_options(
|
||||||
return results
|
return results
|
||||||
|
|
||||||
|
|
||||||
def get_node_proto(session: Session, node: NodeBase) -> core_pb2.Node:
|
def get_node_proto(
|
||||||
|
session: Session, node: NodeBase, emane_configs: List[NodeEmaneConfig]
|
||||||
|
) -> core_pb2.Node:
|
||||||
"""
|
"""
|
||||||
Convert CORE node to protobuf representation.
|
Convert CORE node to protobuf representation.
|
||||||
|
|
||||||
:param session: session containing node
|
:param session: session containing node
|
||||||
:param node: node to convert
|
:param node: node to convert
|
||||||
|
:param emane_configs: emane configs related to node
|
||||||
:return: node proto
|
:return: node proto
|
||||||
"""
|
"""
|
||||||
node_type = session.get_node_type(node.__class__)
|
node_type = session.get_node_type(node.__class__)
|
||||||
|
@ -271,17 +274,53 @@ def get_node_proto(session: Session, node: NodeBase) -> core_pb2.Node:
|
||||||
node_dir = None
|
node_dir = None
|
||||||
config_services = []
|
config_services = []
|
||||||
if isinstance(node, CoreNodeBase):
|
if isinstance(node, CoreNodeBase):
|
||||||
node_dir = node.nodedir
|
node_dir = str(node.directory)
|
||||||
config_services = [x for x in node.config_services]
|
config_services = [x for x in node.config_services]
|
||||||
channel = None
|
channel = None
|
||||||
if isinstance(node, CoreNode):
|
if isinstance(node, CoreNode):
|
||||||
channel = node.ctrlchnlname
|
channel = str(node.ctrlchnlname)
|
||||||
emane_model = None
|
emane_model = None
|
||||||
if isinstance(node, EmaneNet):
|
if isinstance(node, EmaneNet):
|
||||||
emane_model = node.model.name
|
emane_model = node.model.name
|
||||||
image = None
|
image = None
|
||||||
if isinstance(node, (DockerNode, LxcNode)):
|
if isinstance(node, (DockerNode, LxcNode)):
|
||||||
image = node.image
|
image = node.image
|
||||||
|
# check for wlan config
|
||||||
|
wlan_config = session.mobility.get_configs(
|
||||||
|
node.id, config_type=BasicRangeModel.name
|
||||||
|
)
|
||||||
|
if wlan_config:
|
||||||
|
wlan_config = get_config_options(wlan_config, BasicRangeModel)
|
||||||
|
# check for mobility config
|
||||||
|
mobility_config = session.mobility.get_configs(
|
||||||
|
node.id, config_type=Ns2ScriptedMobility.name
|
||||||
|
)
|
||||||
|
if mobility_config:
|
||||||
|
mobility_config = get_config_options(mobility_config, Ns2ScriptedMobility)
|
||||||
|
# check for service configs
|
||||||
|
custom_services = session.services.custom_services.get(node.id)
|
||||||
|
service_configs = {}
|
||||||
|
if custom_services:
|
||||||
|
for service in custom_services.values():
|
||||||
|
service_proto = get_service_configuration(service)
|
||||||
|
service_configs[service.name] = NodeServiceConfig(
|
||||||
|
node_id=node.id,
|
||||||
|
service=service.name,
|
||||||
|
data=service_proto,
|
||||||
|
files=service.config_data,
|
||||||
|
)
|
||||||
|
# check for config service configs
|
||||||
|
config_service_configs = {}
|
||||||
|
if isinstance(node, CoreNode):
|
||||||
|
for service in node.config_services.values():
|
||||||
|
if not service.custom_templates and not service.custom_config:
|
||||||
|
continue
|
||||||
|
config_service_configs[service.name] = ConfigServiceConfig(
|
||||||
|
node_id=node.id,
|
||||||
|
name=service.name,
|
||||||
|
templates=service.custom_templates,
|
||||||
|
config=service.custom_config,
|
||||||
|
)
|
||||||
return core_pb2.Node(
|
return core_pb2.Node(
|
||||||
id=node.id,
|
id=node.id,
|
||||||
name=node.name,
|
name=node.name,
|
||||||
|
@ -297,6 +336,11 @@ def get_node_proto(session: Session, node: NodeBase) -> core_pb2.Node:
|
||||||
dir=node_dir,
|
dir=node_dir,
|
||||||
channel=channel,
|
channel=channel,
|
||||||
canvas=node.canvas,
|
canvas=node.canvas,
|
||||||
|
wlan_config=wlan_config,
|
||||||
|
mobility_config=mobility_config,
|
||||||
|
service_configs=service_configs,
|
||||||
|
config_service_configs=config_service_configs,
|
||||||
|
emane_configs=emane_configs,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@ -529,54 +573,20 @@ def get_nem_id(
|
||||||
return nem_id
|
return nem_id
|
||||||
|
|
||||||
|
|
||||||
def get_emane_model_configs(session: Session) -> List[GetEmaneModelConfig]:
|
def get_emane_model_configs_dict(session: Session) -> Dict[int, List[NodeEmaneConfig]]:
|
||||||
configs = []
|
configs = {}
|
||||||
for _id in session.emane.node_configurations:
|
for _id, model_configs in session.emane.node_configs.items():
|
||||||
if _id == -1:
|
|
||||||
continue
|
|
||||||
model_configs = session.emane.node_configurations[_id]
|
|
||||||
for model_name in model_configs:
|
for model_name in model_configs:
|
||||||
model = session.emane.models[model_name]
|
model_class = session.emane.get_model(model_name)
|
||||||
current_config = session.emane.get_model_config(_id, model_name)
|
current_config = session.emane.get_config(_id, model_name)
|
||||||
config = get_config_options(current_config, model)
|
config = get_config_options(current_config, model_class)
|
||||||
node_id, iface_id = utils.parse_iface_config_id(_id)
|
node_id, iface_id = utils.parse_iface_config_id(_id)
|
||||||
iface_id = iface_id if iface_id is not None else -1
|
iface_id = iface_id if iface_id is not None else -1
|
||||||
model_config = GetEmaneModelConfig(
|
node_config = NodeEmaneConfig(
|
||||||
node_id=node_id, model=model_name, iface_id=iface_id, config=config
|
model=model_name, iface_id=iface_id, config=config
|
||||||
)
|
)
|
||||||
configs.append(model_config)
|
node_configs = configs.setdefault(node_id, [])
|
||||||
return configs
|
node_configs.append(node_config)
|
||||||
|
|
||||||
|
|
||||||
def get_wlan_configs(session: Session) -> Dict[int, MappedConfig]:
|
|
||||||
configs = {}
|
|
||||||
for node_id in session.mobility.node_configurations:
|
|
||||||
model_config = session.mobility.node_configurations[node_id]
|
|
||||||
if node_id == -1:
|
|
||||||
continue
|
|
||||||
for model_name in model_config:
|
|
||||||
if model_name != BasicRangeModel.name:
|
|
||||||
continue
|
|
||||||
current_config = session.mobility.get_model_config(node_id, model_name)
|
|
||||||
config = get_config_options(current_config, BasicRangeModel)
|
|
||||||
mapped_config = MappedConfig(config=config)
|
|
||||||
configs[node_id] = mapped_config
|
|
||||||
return configs
|
|
||||||
|
|
||||||
|
|
||||||
def get_mobility_configs(session: Session) -> Dict[int, MappedConfig]:
|
|
||||||
configs = {}
|
|
||||||
for node_id in session.mobility.node_configurations:
|
|
||||||
model_config = session.mobility.node_configurations[node_id]
|
|
||||||
if node_id == -1:
|
|
||||||
continue
|
|
||||||
for model_name in model_config:
|
|
||||||
if model_name != Ns2ScriptedMobility.name:
|
|
||||||
continue
|
|
||||||
current_config = session.mobility.get_model_config(node_id, model_name)
|
|
||||||
config = get_config_options(current_config, Ns2ScriptedMobility)
|
|
||||||
mapped_config = MappedConfig(config=config)
|
|
||||||
configs[node_id] = mapped_config
|
|
||||||
return configs
|
return configs
|
||||||
|
|
||||||
|
|
||||||
|
@ -590,15 +600,6 @@ def get_hooks(session: Session) -> List[core_pb2.Hook]:
|
||||||
return hooks
|
return hooks
|
||||||
|
|
||||||
|
|
||||||
def get_emane_models(session: Session) -> List[str]:
|
|
||||||
emane_models = []
|
|
||||||
for model in session.emane.models.keys():
|
|
||||||
if len(model.split("_")) != 2:
|
|
||||||
continue
|
|
||||||
emane_models.append(model)
|
|
||||||
return emane_models
|
|
||||||
|
|
||||||
|
|
||||||
def get_default_services(session: Session) -> List[ServiceDefaults]:
|
def get_default_services(session: Session) -> List[ServiceDefaults]:
|
||||||
default_services = []
|
default_services = []
|
||||||
for name, services in session.services.default_services.items():
|
for name, services in session.services.default_services.items():
|
||||||
|
@ -607,45 +608,6 @@ def get_default_services(session: Session) -> List[ServiceDefaults]:
|
||||||
return default_services
|
return default_services
|
||||||
|
|
||||||
|
|
||||||
def get_node_service_configs(session: Session) -> List[NodeServiceConfig]:
|
|
||||||
configs = []
|
|
||||||
for node_id, service_configs in session.services.custom_services.items():
|
|
||||||
for name in service_configs:
|
|
||||||
service = session.services.get_service(node_id, name)
|
|
||||||
service_proto = get_service_configuration(service)
|
|
||||||
config = NodeServiceConfig(
|
|
||||||
node_id=node_id,
|
|
||||||
service=name,
|
|
||||||
data=service_proto,
|
|
||||||
files=service.config_data,
|
|
||||||
)
|
|
||||||
configs.append(config)
|
|
||||||
return configs
|
|
||||||
|
|
||||||
|
|
||||||
def get_node_config_service_configs(session: Session) -> List[ConfigServiceConfig]:
|
|
||||||
configs = []
|
|
||||||
for node in session.nodes.values():
|
|
||||||
if not isinstance(node, CoreNodeBase):
|
|
||||||
continue
|
|
||||||
for name, service in node.config_services.items():
|
|
||||||
if not service.custom_templates and not service.custom_config:
|
|
||||||
continue
|
|
||||||
config_proto = ConfigServiceConfig(
|
|
||||||
node_id=node.id,
|
|
||||||
name=name,
|
|
||||||
templates=service.custom_templates,
|
|
||||||
config=service.custom_config,
|
|
||||||
)
|
|
||||||
configs.append(config_proto)
|
|
||||||
return configs
|
|
||||||
|
|
||||||
|
|
||||||
def get_emane_config(session: Session) -> Dict[str, common_pb2.ConfigOption]:
|
|
||||||
current_config = session.emane.get_configs()
|
|
||||||
return get_config_options(current_config, session.emane.emane_config)
|
|
||||||
|
|
||||||
|
|
||||||
def get_mobility_node(
|
def get_mobility_node(
|
||||||
session: Session, node_id: int, context: ServicerContext
|
session: Session, node_id: int, context: ServicerContext
|
||||||
) -> Union[WlanNode, EmaneNet]:
|
) -> Union[WlanNode, EmaneNet]:
|
||||||
|
@ -656,3 +618,88 @@ def get_mobility_node(
|
||||||
return session.get_node(node_id, EmaneNet)
|
return session.get_node(node_id, EmaneNet)
|
||||||
except CoreError:
|
except CoreError:
|
||||||
context.abort(grpc.StatusCode.NOT_FOUND, "node id is not for wlan or emane")
|
context.abort(grpc.StatusCode.NOT_FOUND, "node id is not for wlan or emane")
|
||||||
|
|
||||||
|
|
||||||
|
def convert_session(session: Session) -> wrappers.Session:
|
||||||
|
links = []
|
||||||
|
nodes = []
|
||||||
|
emane_configs = get_emane_model_configs_dict(session)
|
||||||
|
for _id in session.nodes:
|
||||||
|
node = session.nodes[_id]
|
||||||
|
if not isinstance(node, (PtpNet, CtrlNet)):
|
||||||
|
node_emane_configs = emane_configs.get(node.id, [])
|
||||||
|
node_proto = get_node_proto(session, node, node_emane_configs)
|
||||||
|
nodes.append(node_proto)
|
||||||
|
node_links = get_links(node)
|
||||||
|
links.extend(node_links)
|
||||||
|
default_services = get_default_services(session)
|
||||||
|
x, y, z = session.location.refxyz
|
||||||
|
lat, lon, alt = session.location.refgeo
|
||||||
|
location = core_pb2.SessionLocation(
|
||||||
|
x=x, y=y, z=z, lat=lat, lon=lon, alt=alt, scale=session.location.refscale
|
||||||
|
)
|
||||||
|
hooks = get_hooks(session)
|
||||||
|
session_file = str(session.file_path) if session.file_path else None
|
||||||
|
options = get_config_options(session.options.get_configs(), session.options)
|
||||||
|
servers = [
|
||||||
|
core_pb2.Server(name=x.name, host=x.host)
|
||||||
|
for x in session.distributed.servers.values()
|
||||||
|
]
|
||||||
|
return core_pb2.Session(
|
||||||
|
id=session.id,
|
||||||
|
state=session.state.value,
|
||||||
|
nodes=nodes,
|
||||||
|
links=links,
|
||||||
|
dir=str(session.directory),
|
||||||
|
user=session.user,
|
||||||
|
default_services=default_services,
|
||||||
|
location=location,
|
||||||
|
hooks=hooks,
|
||||||
|
metadata=session.metadata,
|
||||||
|
file=session_file,
|
||||||
|
options=options,
|
||||||
|
servers=servers,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def configure_node(
|
||||||
|
session: Session, node: core_pb2.Node, core_node: NodeBase, context: ServicerContext
|
||||||
|
) -> None:
|
||||||
|
for emane_config in node.emane_configs:
|
||||||
|
_id = utils.iface_config_id(node.id, emane_config.iface_id)
|
||||||
|
config = {k: v.value for k, v in emane_config.config.items()}
|
||||||
|
session.emane.set_config(_id, emane_config.model, config)
|
||||||
|
if node.wlan_config:
|
||||||
|
config = {k: v.value for k, v in node.wlan_config.items()}
|
||||||
|
session.mobility.set_model_config(node.id, BasicRangeModel.name, config)
|
||||||
|
if node.mobility_config:
|
||||||
|
config = {k: v.value for k, v in node.mobility_config.items()}
|
||||||
|
session.mobility.set_model_config(node.id, Ns2ScriptedMobility.name, config)
|
||||||
|
for service_name, service_config in node.service_configs.items():
|
||||||
|
data = service_config.data
|
||||||
|
config = ServiceConfig(
|
||||||
|
node_id=node.id,
|
||||||
|
service=service_name,
|
||||||
|
startup=data.startup,
|
||||||
|
validate=data.validate,
|
||||||
|
shutdown=data.shutdown,
|
||||||
|
files=data.configs,
|
||||||
|
directories=data.dirs,
|
||||||
|
)
|
||||||
|
service_configuration(session, config)
|
||||||
|
for file_name, file_data in service_config.files.items():
|
||||||
|
session.services.set_service_file(
|
||||||
|
node.id, service_name, file_name, file_data
|
||||||
|
)
|
||||||
|
if node.config_service_configs:
|
||||||
|
if not isinstance(core_node, CoreNode):
|
||||||
|
context.abort(
|
||||||
|
grpc.StatusCode.INVALID_ARGUMENT,
|
||||||
|
"invalid node type with config service configs",
|
||||||
|
)
|
||||||
|
for service_name, service_config in node.config_service_configs.items():
|
||||||
|
service = core_node.config_services[service_name]
|
||||||
|
if service_config.config:
|
||||||
|
service.set_config(service_config.config)
|
||||||
|
for name, template in service_config.templates.items():
|
||||||
|
service.set_template(name, template)
|
||||||
|
|
File diff suppressed because it is too large
Load diff
|
@ -171,18 +171,32 @@ class ConfigServiceData:
|
||||||
class ConfigServiceDefaults:
|
class ConfigServiceDefaults:
|
||||||
templates: Dict[str, str]
|
templates: Dict[str, str]
|
||||||
config: Dict[str, "ConfigOption"]
|
config: Dict[str, "ConfigOption"]
|
||||||
modes: List[str]
|
modes: Dict[str, Dict[str, str]]
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def from_proto(
|
def from_proto(
|
||||||
cls, proto: configservices_pb2.GetConfigServicesResponse
|
cls, proto: configservices_pb2.GetConfigServiceDefaultsResponse
|
||||||
) -> "ConfigServiceDefaults":
|
) -> "ConfigServiceDefaults":
|
||||||
config = ConfigOption.from_dict(proto.config)
|
config = ConfigOption.from_dict(proto.config)
|
||||||
|
modes = {x.name: dict(x.config) for x in proto.modes}
|
||||||
return ConfigServiceDefaults(
|
return ConfigServiceDefaults(
|
||||||
templates=dict(proto.templates), config=config, modes=list(proto.modes)
|
templates=dict(proto.templates), config=config, modes=modes
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass
|
||||||
|
class Server:
|
||||||
|
name: str
|
||||||
|
host: str
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def from_proto(cls, proto: core_pb2.Server) -> "Server":
|
||||||
|
return Server(name=proto.name, host=proto.host)
|
||||||
|
|
||||||
|
def to_proto(self) -> core_pb2.Server:
|
||||||
|
return core_pb2.Server(name=self.name, host=self.host)
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
@dataclass
|
||||||
class Service:
|
class Service:
|
||||||
group: str
|
group: str
|
||||||
|
@ -205,16 +219,16 @@ class ServiceDefault:
|
||||||
|
|
||||||
@dataclass
|
@dataclass
|
||||||
class NodeServiceData:
|
class NodeServiceData:
|
||||||
executables: List[str]
|
executables: List[str] = field(default_factory=list)
|
||||||
dependencies: List[str]
|
dependencies: List[str] = field(default_factory=list)
|
||||||
dirs: List[str]
|
dirs: List[str] = field(default_factory=list)
|
||||||
configs: List[str]
|
configs: List[str] = field(default_factory=list)
|
||||||
startup: List[str]
|
startup: List[str] = field(default_factory=list)
|
||||||
validate: List[str]
|
validate: List[str] = field(default_factory=list)
|
||||||
validation_mode: ServiceValidationMode
|
validation_mode: ServiceValidationMode = ServiceValidationMode.NON_BLOCKING
|
||||||
validation_timer: int
|
validation_timer: int = 5
|
||||||
shutdown: List[str]
|
shutdown: List[str] = field(default_factory=list)
|
||||||
meta: str
|
meta: str = None
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def from_proto(cls, proto: services_pb2.NodeServiceData) -> "NodeServiceData":
|
def from_proto(cls, proto: services_pb2.NodeServiceData) -> "NodeServiceData":
|
||||||
|
@ -225,12 +239,43 @@ class NodeServiceData:
|
||||||
configs=proto.configs,
|
configs=proto.configs,
|
||||||
startup=proto.startup,
|
startup=proto.startup,
|
||||||
validate=proto.validate,
|
validate=proto.validate,
|
||||||
validation_mode=proto.validation_mode,
|
validation_mode=ServiceValidationMode(proto.validation_mode),
|
||||||
validation_timer=proto.validation_timer,
|
validation_timer=proto.validation_timer,
|
||||||
shutdown=proto.shutdown,
|
shutdown=proto.shutdown,
|
||||||
meta=proto.meta,
|
meta=proto.meta,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
def to_proto(self) -> services_pb2.NodeServiceData:
|
||||||
|
return services_pb2.NodeServiceData(
|
||||||
|
executables=self.executables,
|
||||||
|
dependencies=self.dependencies,
|
||||||
|
dirs=self.dirs,
|
||||||
|
configs=self.configs,
|
||||||
|
startup=self.startup,
|
||||||
|
validate=self.validate,
|
||||||
|
validation_mode=self.validation_mode.value,
|
||||||
|
validation_timer=self.validation_timer,
|
||||||
|
shutdown=self.shutdown,
|
||||||
|
meta=self.meta,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass
|
||||||
|
class NodeServiceConfig:
|
||||||
|
node_id: int
|
||||||
|
service: str
|
||||||
|
data: NodeServiceData
|
||||||
|
files: Dict[str, str] = field(default_factory=dict)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def from_proto(cls, proto: services_pb2.NodeServiceConfig) -> "NodeServiceConfig":
|
||||||
|
return NodeServiceConfig(
|
||||||
|
node_id=proto.node_id,
|
||||||
|
service=proto.service,
|
||||||
|
data=NodeServiceData.from_proto(proto.data),
|
||||||
|
files=dict(proto.files),
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
@dataclass
|
||||||
class ServiceConfig:
|
class ServiceConfig:
|
||||||
|
@ -254,6 +299,19 @@ class ServiceConfig:
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass
|
||||||
|
class ServiceFileConfig:
|
||||||
|
node_id: int
|
||||||
|
service: str
|
||||||
|
file: str
|
||||||
|
data: str = field(repr=False)
|
||||||
|
|
||||||
|
def to_proto(self) -> services_pb2.ServiceFileConfig:
|
||||||
|
return services_pb2.ServiceFileConfig(
|
||||||
|
node_id=self.node_id, service=self.service, file=self.file, data=self.data
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
@dataclass
|
||||||
class BridgeThroughput:
|
class BridgeThroughput:
|
||||||
node_id: int
|
node_id: int
|
||||||
|
@ -364,11 +422,11 @@ class ExceptionEvent:
|
||||||
|
|
||||||
@dataclass
|
@dataclass
|
||||||
class ConfigOption:
|
class ConfigOption:
|
||||||
label: str
|
|
||||||
name: str
|
name: str
|
||||||
value: str
|
value: str
|
||||||
type: ConfigOptionType
|
label: str = None
|
||||||
group: str
|
type: ConfigOptionType = None
|
||||||
|
group: str = None
|
||||||
select: List[str] = None
|
select: List[str] = None
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
|
@ -386,15 +444,27 @@ class ConfigOption:
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def from_proto(cls, proto: common_pb2.ConfigOption) -> "ConfigOption":
|
def from_proto(cls, proto: common_pb2.ConfigOption) -> "ConfigOption":
|
||||||
|
config_type = ConfigOptionType(proto.type) if proto.type is not None else None
|
||||||
return ConfigOption(
|
return ConfigOption(
|
||||||
label=proto.label,
|
label=proto.label,
|
||||||
name=proto.name,
|
name=proto.name,
|
||||||
value=proto.value,
|
value=proto.value,
|
||||||
type=ConfigOptionType(proto.type),
|
type=config_type,
|
||||||
group=proto.group,
|
group=proto.group,
|
||||||
select=proto.select,
|
select=proto.select,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
def to_proto(self) -> common_pb2.ConfigOption:
|
||||||
|
config_type = self.type.value if self.type is not None else None
|
||||||
|
return common_pb2.ConfigOption(
|
||||||
|
label=self.label,
|
||||||
|
name=self.name,
|
||||||
|
value=self.value,
|
||||||
|
type=config_type,
|
||||||
|
select=self.select,
|
||||||
|
group=self.group,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
@dataclass
|
||||||
class Interface:
|
class Interface:
|
||||||
|
@ -598,11 +668,12 @@ class EmaneModelConfig:
|
||||||
)
|
)
|
||||||
|
|
||||||
def to_proto(self) -> emane_pb2.EmaneModelConfig:
|
def to_proto(self) -> emane_pb2.EmaneModelConfig:
|
||||||
|
config = ConfigOption.to_dict(self.config)
|
||||||
return emane_pb2.EmaneModelConfig(
|
return emane_pb2.EmaneModelConfig(
|
||||||
node_id=self.node_id,
|
node_id=self.node_id,
|
||||||
model=self.model,
|
model=self.model,
|
||||||
iface_id=self.iface_id,
|
iface_id=self.iface_id,
|
||||||
config=self.config,
|
config=config,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@ -635,11 +706,11 @@ class Geo:
|
||||||
|
|
||||||
@dataclass
|
@dataclass
|
||||||
class Node:
|
class Node:
|
||||||
id: int
|
id: int = None
|
||||||
name: str
|
name: str = None
|
||||||
type: NodeType
|
type: NodeType = NodeType.DEFAULT
|
||||||
model: str = None
|
model: str = None
|
||||||
position: Position = None
|
position: Position = Position(x=0, y=0)
|
||||||
services: Set[str] = field(default_factory=set)
|
services: Set[str] = field(default_factory=set)
|
||||||
config_services: Set[str] = field(default_factory=set)
|
config_services: Set[str] = field(default_factory=set)
|
||||||
emane: str = None
|
emane: str = None
|
||||||
|
@ -669,6 +740,23 @@ class Node:
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def from_proto(cls, proto: core_pb2.Node) -> "Node":
|
def from_proto(cls, proto: core_pb2.Node) -> "Node":
|
||||||
|
service_configs = {}
|
||||||
|
service_file_configs = {}
|
||||||
|
for service, node_config in proto.service_configs.items():
|
||||||
|
service_configs[service] = NodeServiceData.from_proto(node_config.data)
|
||||||
|
service_file_configs[service] = dict(node_config.files)
|
||||||
|
emane_configs = {}
|
||||||
|
for emane_config in proto.emane_configs:
|
||||||
|
iface_id = None if emane_config.iface_id == -1 else emane_config.iface_id
|
||||||
|
model = emane_config.model
|
||||||
|
key = (model, iface_id)
|
||||||
|
emane_configs[key] = ConfigOption.from_dict(emane_config.config)
|
||||||
|
config_service_configs = {}
|
||||||
|
for service, service_config in proto.config_service_configs.items():
|
||||||
|
config_service_configs[service] = ConfigServiceData(
|
||||||
|
templates=dict(service_config.templates),
|
||||||
|
config=dict(service_config.config),
|
||||||
|
)
|
||||||
return Node(
|
return Node(
|
||||||
id=proto.id,
|
id=proto.id,
|
||||||
name=proto.name,
|
name=proto.name,
|
||||||
|
@ -685,9 +773,43 @@ class Node:
|
||||||
dir=proto.dir,
|
dir=proto.dir,
|
||||||
channel=proto.channel,
|
channel=proto.channel,
|
||||||
canvas=proto.canvas,
|
canvas=proto.canvas,
|
||||||
|
wlan_config=ConfigOption.from_dict(proto.wlan_config),
|
||||||
|
mobility_config=ConfigOption.from_dict(proto.mobility_config),
|
||||||
|
service_configs=service_configs,
|
||||||
|
service_file_configs=service_file_configs,
|
||||||
|
config_service_configs=config_service_configs,
|
||||||
|
emane_model_configs=emane_configs,
|
||||||
)
|
)
|
||||||
|
|
||||||
def to_proto(self) -> core_pb2.Node:
|
def to_proto(self) -> core_pb2.Node:
|
||||||
|
emane_configs = []
|
||||||
|
for key, config in self.emane_model_configs.items():
|
||||||
|
model, iface_id = key
|
||||||
|
if iface_id is None:
|
||||||
|
iface_id = -1
|
||||||
|
config = {k: v.to_proto() for k, v in config.items()}
|
||||||
|
emane_config = emane_pb2.NodeEmaneConfig(
|
||||||
|
iface_id=iface_id, model=model, config=config
|
||||||
|
)
|
||||||
|
emane_configs.append(emane_config)
|
||||||
|
service_configs = {}
|
||||||
|
for service, service_data in self.service_configs.items():
|
||||||
|
service_configs[service] = services_pb2.NodeServiceConfig(
|
||||||
|
service=service, data=service_data.to_proto()
|
||||||
|
)
|
||||||
|
for service, file_configs in self.service_file_configs.items():
|
||||||
|
service_config = service_configs.get(service)
|
||||||
|
if service_config:
|
||||||
|
service_config.files.update(file_configs)
|
||||||
|
else:
|
||||||
|
service_configs[service] = services_pb2.NodeServiceConfig(
|
||||||
|
service=service, files=file_configs
|
||||||
|
)
|
||||||
|
config_service_configs = {}
|
||||||
|
for service, service_config in self.config_service_configs.items():
|
||||||
|
config_service_configs[service] = configservices_pb2.ConfigServiceConfig(
|
||||||
|
templates=service_config.templates, config=service_config.config
|
||||||
|
)
|
||||||
return core_pb2.Node(
|
return core_pb2.Node(
|
||||||
id=self.id,
|
id=self.id,
|
||||||
name=self.name,
|
name=self.name,
|
||||||
|
@ -703,24 +825,50 @@ class Node:
|
||||||
dir=self.dir,
|
dir=self.dir,
|
||||||
channel=self.channel,
|
channel=self.channel,
|
||||||
canvas=self.canvas,
|
canvas=self.canvas,
|
||||||
|
wlan_config={k: v.to_proto() for k, v in self.wlan_config.items()},
|
||||||
|
mobility_config={k: v.to_proto() for k, v in self.mobility_config.items()},
|
||||||
|
service_configs=service_configs,
|
||||||
|
config_service_configs=config_service_configs,
|
||||||
|
emane_configs=emane_configs,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
def set_wlan(self, config: Dict[str, str]) -> None:
|
||||||
|
for key, value in config.items():
|
||||||
|
option = ConfigOption(name=key, value=value)
|
||||||
|
self.wlan_config[key] = option
|
||||||
|
|
||||||
|
def set_mobility(self, config: Dict[str, str]) -> None:
|
||||||
|
for key, value in config.items():
|
||||||
|
option = ConfigOption(name=key, value=value)
|
||||||
|
self.mobility_config[key] = option
|
||||||
|
|
||||||
|
def set_emane_model(
|
||||||
|
self, model: str, config: Dict[str, str], iface_id: int = None
|
||||||
|
) -> None:
|
||||||
|
key = (model, iface_id)
|
||||||
|
config_options = self.emane_model_configs.setdefault(key, {})
|
||||||
|
for key, value in config.items():
|
||||||
|
option = ConfigOption(name=key, value=value)
|
||||||
|
config_options[key] = option
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
@dataclass
|
||||||
class Session:
|
class Session:
|
||||||
id: int
|
id: int = None
|
||||||
state: SessionState
|
state: SessionState = SessionState.DEFINITION
|
||||||
nodes: Dict[int, Node]
|
nodes: Dict[int, Node] = field(default_factory=dict)
|
||||||
links: List[Link]
|
links: List[Link] = field(default_factory=list)
|
||||||
dir: str
|
dir: str = None
|
||||||
user: str
|
user: str = None
|
||||||
default_services: Dict[str, Set[str]]
|
default_services: Dict[str, Set[str]] = field(default_factory=dict)
|
||||||
location: SessionLocation
|
location: SessionLocation = SessionLocation(
|
||||||
hooks: Dict[str, Hook]
|
x=0.0, y=0.0, z=0.0, lat=47.57917, lon=-122.13232, alt=2.0, scale=150.0
|
||||||
emane_models: List[str]
|
)
|
||||||
emane_config: Dict[str, ConfigOption]
|
hooks: Dict[str, Hook] = field(default_factory=dict)
|
||||||
metadata: Dict[str, str]
|
metadata: Dict[str, str] = field(default_factory=dict)
|
||||||
file: Path
|
file: Path = None
|
||||||
|
options: Dict[str, ConfigOption] = field(default_factory=dict)
|
||||||
|
servers: List[Server] = field(default_factory=list)
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def from_proto(cls, proto: core_pb2.Session) -> "Session":
|
def from_proto(cls, proto: core_pb2.Session) -> "Session":
|
||||||
|
@ -730,33 +878,9 @@ class Session:
|
||||||
x.node_type: set(x.services) for x in proto.default_services
|
x.node_type: set(x.services) for x in proto.default_services
|
||||||
}
|
}
|
||||||
hooks = {x.file: Hook.from_proto(x) for x in proto.hooks}
|
hooks = {x.file: Hook.from_proto(x) for x in proto.hooks}
|
||||||
# update nodes with their current configurations
|
|
||||||
for model in proto.emane_model_configs:
|
|
||||||
iface_id = None
|
|
||||||
if model.iface_id != -1:
|
|
||||||
iface_id = model.iface_id
|
|
||||||
node = nodes[model.node_id]
|
|
||||||
key = (model.model, iface_id)
|
|
||||||
node.emane_model_configs[key] = ConfigOption.from_dict(model.config)
|
|
||||||
for node_id, mapped_config in proto.wlan_configs.items():
|
|
||||||
node = nodes[node_id]
|
|
||||||
node.wlan_config = ConfigOption.from_dict(mapped_config.config)
|
|
||||||
for config in proto.service_configs:
|
|
||||||
service = config.service
|
|
||||||
node = nodes[config.node_id]
|
|
||||||
node.service_configs[service] = NodeServiceData.from_proto(config.data)
|
|
||||||
for file, data in config.files.items():
|
|
||||||
files = node.service_file_configs.setdefault(service, {})
|
|
||||||
files[file] = data
|
|
||||||
for config in proto.config_service_configs:
|
|
||||||
node = nodes[config.node_id]
|
|
||||||
node.config_service_configs[config.name] = ConfigServiceData(
|
|
||||||
templates=dict(config.templates), config=dict(config.config)
|
|
||||||
)
|
|
||||||
for node_id, mapped_config in proto.mobility_configs.items():
|
|
||||||
node = nodes[node_id]
|
|
||||||
node.mobility_config = ConfigOption.from_dict(mapped_config.config)
|
|
||||||
file_path = Path(proto.file) if proto.file else None
|
file_path = Path(proto.file) if proto.file else None
|
||||||
|
options = ConfigOption.from_dict(proto.options)
|
||||||
|
servers = [Server.from_proto(x) for x in proto.servers]
|
||||||
return Session(
|
return Session(
|
||||||
id=proto.id,
|
id=proto.id,
|
||||||
state=SessionState(proto.state),
|
state=SessionState(proto.state),
|
||||||
|
@ -767,10 +891,107 @@ class Session:
|
||||||
default_services=default_services,
|
default_services=default_services,
|
||||||
location=SessionLocation.from_proto(proto.location),
|
location=SessionLocation.from_proto(proto.location),
|
||||||
hooks=hooks,
|
hooks=hooks,
|
||||||
emane_models=list(proto.emane_models),
|
|
||||||
emane_config=ConfigOption.from_dict(proto.emane_config),
|
|
||||||
metadata=dict(proto.metadata),
|
metadata=dict(proto.metadata),
|
||||||
file=file_path,
|
file=file_path,
|
||||||
|
options=options,
|
||||||
|
servers=servers,
|
||||||
|
)
|
||||||
|
|
||||||
|
def to_proto(self) -> core_pb2.Session:
|
||||||
|
nodes = [x.to_proto() for x in self.nodes.values()]
|
||||||
|
links = [x.to_proto() for x in self.links]
|
||||||
|
hooks = [x.to_proto() for x in self.hooks.values()]
|
||||||
|
options = {k: v.to_proto() for k, v in self.options.items()}
|
||||||
|
servers = [x.to_proto() for x in self.servers]
|
||||||
|
default_services = []
|
||||||
|
for node_type, services in self.default_services.items():
|
||||||
|
default_service = services_pb2.ServiceDefaults(
|
||||||
|
node_type=node_type, services=services
|
||||||
|
)
|
||||||
|
default_services.append(default_service)
|
||||||
|
file = str(self.file) if self.file else None
|
||||||
|
return core_pb2.Session(
|
||||||
|
id=self.id,
|
||||||
|
state=self.state.value,
|
||||||
|
nodes=nodes,
|
||||||
|
links=links,
|
||||||
|
dir=self.dir,
|
||||||
|
user=self.user,
|
||||||
|
default_services=default_services,
|
||||||
|
location=self.location.to_proto(),
|
||||||
|
hooks=hooks,
|
||||||
|
metadata=self.metadata,
|
||||||
|
file=file,
|
||||||
|
options=options,
|
||||||
|
servers=servers,
|
||||||
|
)
|
||||||
|
|
||||||
|
def add_node(
|
||||||
|
self,
|
||||||
|
_id: int,
|
||||||
|
*,
|
||||||
|
name: str = None,
|
||||||
|
_type: NodeType = NodeType.DEFAULT,
|
||||||
|
model: str = "PC",
|
||||||
|
position: Position = None,
|
||||||
|
geo: Geo = None,
|
||||||
|
emane: str = None,
|
||||||
|
image: str = None,
|
||||||
|
server: str = None,
|
||||||
|
) -> Node:
|
||||||
|
node = Node(
|
||||||
|
id=_id,
|
||||||
|
name=name,
|
||||||
|
type=_type,
|
||||||
|
model=model,
|
||||||
|
position=position,
|
||||||
|
geo=geo,
|
||||||
|
emane=emane,
|
||||||
|
image=image,
|
||||||
|
server=server,
|
||||||
|
)
|
||||||
|
self.nodes[node.id] = node
|
||||||
|
return node
|
||||||
|
|
||||||
|
def add_link(
|
||||||
|
self,
|
||||||
|
*,
|
||||||
|
node1: Node,
|
||||||
|
node2: Node,
|
||||||
|
iface1: Interface = None,
|
||||||
|
iface2: Interface = None,
|
||||||
|
options: LinkOptions = None,
|
||||||
|
) -> Link:
|
||||||
|
link = Link(
|
||||||
|
node1_id=node1.id,
|
||||||
|
node2_id=node2.id,
|
||||||
|
iface1=iface1,
|
||||||
|
iface2=iface2,
|
||||||
|
options=options,
|
||||||
|
)
|
||||||
|
self.links.append(link)
|
||||||
|
return link
|
||||||
|
|
||||||
|
def set_options(self, config: Dict[str, str]) -> None:
|
||||||
|
for key, value in config.items():
|
||||||
|
option = ConfigOption(name=key, value=value)
|
||||||
|
self.options[key] = option
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass
|
||||||
|
class CoreConfig:
|
||||||
|
services: List[Service] = field(default_factory=list)
|
||||||
|
config_services: List[ConfigService] = field(default_factory=list)
|
||||||
|
emane_models: List[str] = field(default_factory=list)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def from_proto(cls, proto: core_pb2.GetConfigResponse) -> "CoreConfig":
|
||||||
|
services = [Service.from_proto(x) for x in proto.services]
|
||||||
|
config_services = [ConfigService.from_proto(x) for x in proto.config_services]
|
||||||
|
return CoreConfig(
|
||||||
|
services=services,
|
||||||
|
config_services=config_services,
|
||||||
|
emane_models=list(proto.emane_models),
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
|
File diff suppressed because it is too large
Load diff
|
@ -10,6 +10,8 @@ from core.api.tlv.enumerations import ConfigTlvs, NodeTlvs
|
||||||
from core.config import ConfigGroup, ConfigurableOptions
|
from core.config import ConfigGroup, ConfigurableOptions
|
||||||
from core.emulator.data import ConfigData, NodeData
|
from core.emulator.data import ConfigData, NodeData
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
def convert_node(node_data: NodeData):
|
def convert_node(node_data: NodeData):
|
||||||
"""
|
"""
|
||||||
|
@ -139,9 +141,9 @@ class ConfigShim:
|
||||||
captions = None
|
captions = None
|
||||||
data_types = []
|
data_types = []
|
||||||
possible_values = []
|
possible_values = []
|
||||||
logging.debug("configurable: %s", configurable_options)
|
logger.debug("configurable: %s", configurable_options)
|
||||||
logging.debug("configuration options: %s", configurable_options.configurations)
|
logger.debug("configuration options: %s", configurable_options.configurations)
|
||||||
logging.debug("configuration data: %s", config)
|
logger.debug("configuration data: %s", config)
|
||||||
for configuration in configurable_options.configurations():
|
for configuration in configurable_options.configurations():
|
||||||
if not captions:
|
if not captions:
|
||||||
captions = configuration.label
|
captions = configuration.label
|
||||||
|
|
|
@ -4,6 +4,8 @@ Utilities for working with python struct data.
|
||||||
|
|
||||||
import logging
|
import logging
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
def pack_values(clazz, packers):
|
def pack_values(clazz, packers):
|
||||||
"""
|
"""
|
||||||
|
@ -15,7 +17,7 @@ def pack_values(clazz, packers):
|
||||||
"""
|
"""
|
||||||
|
|
||||||
# iterate through tuples of values to pack
|
# iterate through tuples of values to pack
|
||||||
logging.debug("packing: %s", packers)
|
logger.debug("packing: %s", packers)
|
||||||
data = b""
|
data = b""
|
||||||
for packer in packers:
|
for packer in packers:
|
||||||
# check if a transformer was provided for valid values
|
# check if a transformer was provided for valid values
|
||||||
|
@ -37,7 +39,7 @@ def pack_values(clazz, packers):
|
||||||
value = transformer(value)
|
value = transformer(value)
|
||||||
|
|
||||||
# pack and add to existing data
|
# pack and add to existing data
|
||||||
logging.debug("packing: %s - %s type(%s)", tlv_type, value, type(value))
|
logger.debug("packing: %s - %s type(%s)", tlv_type, value, type(value))
|
||||||
data += clazz.pack(tlv_type.value, value)
|
data += clazz.pack(tlv_type.value, value)
|
||||||
|
|
||||||
return data
|
return data
|
||||||
|
|
|
@ -4,73 +4,71 @@ Common support for configurable CORE objects.
|
||||||
|
|
||||||
import logging
|
import logging
|
||||||
from collections import OrderedDict
|
from collections import OrderedDict
|
||||||
from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Type, Union
|
from dataclasses import dataclass, field
|
||||||
|
from typing import TYPE_CHECKING, Any, Dict, List, Optional, Set, Tuple, Type, Union
|
||||||
|
|
||||||
from core.emane.nodes import EmaneNet
|
from core.emane.nodes import EmaneNet
|
||||||
from core.emulator.enumerations import ConfigDataTypes
|
from core.emulator.enumerations import ConfigDataTypes
|
||||||
|
from core.errors import CoreConfigError
|
||||||
from core.nodes.network import WlanNode
|
from core.nodes.network import WlanNode
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.location.mobility import WirelessModel
|
from core.location.mobility import WirelessModel
|
||||||
|
|
||||||
WirelessModelType = Type[WirelessModel]
|
WirelessModelType = Type[WirelessModel]
|
||||||
|
|
||||||
|
_BOOL_OPTIONS: Set[str] = {"0", "1"}
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass
|
||||||
class ConfigGroup:
|
class ConfigGroup:
|
||||||
"""
|
"""
|
||||||
Defines configuration group tabs used for display by ConfigurationOptions.
|
Defines configuration group tabs used for display by ConfigurationOptions.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
def __init__(self, name: str, start: int, stop: int) -> None:
|
name: str
|
||||||
"""
|
start: int
|
||||||
Creates a ConfigGroup object.
|
stop: int
|
||||||
|
|
||||||
:param name: configuration group display name
|
|
||||||
:param start: configurations start index for this group
|
|
||||||
:param stop: configurations stop index for this group
|
|
||||||
"""
|
|
||||||
self.name: str = name
|
|
||||||
self.start: int = start
|
|
||||||
self.stop: int = stop
|
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass
|
||||||
class Configuration:
|
class Configuration:
|
||||||
"""
|
"""
|
||||||
Represents a configuration options.
|
Represents a configuration options.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
def __init__(
|
id: str
|
||||||
self,
|
type: ConfigDataTypes
|
||||||
_id: str,
|
label: str = None
|
||||||
_type: ConfigDataTypes,
|
default: str = ""
|
||||||
label: str = None,
|
options: List[str] = field(default_factory=list)
|
||||||
default: str = "",
|
|
||||||
options: List[str] = None,
|
|
||||||
) -> None:
|
|
||||||
"""
|
|
||||||
Creates a Configuration object.
|
|
||||||
|
|
||||||
:param _id: unique name for configuration
|
def __post_init__(self) -> None:
|
||||||
:param _type: configuration data type
|
self.label = self.label if self.label else self.id
|
||||||
:param label: configuration label for display
|
if self.type == ConfigDataTypes.BOOL:
|
||||||
:param default: default value for configuration
|
if self.default and self.default not in _BOOL_OPTIONS:
|
||||||
:param options: list options if this is a configuration with a combobox
|
raise CoreConfigError(
|
||||||
"""
|
f"{self.id} bool value must be one of: {_BOOL_OPTIONS}: "
|
||||||
self.id: str = _id
|
f"{self.default}"
|
||||||
self.type: ConfigDataTypes = _type
|
)
|
||||||
self.default: str = default
|
elif self.type == ConfigDataTypes.FLOAT:
|
||||||
if not options:
|
if self.default:
|
||||||
options = []
|
try:
|
||||||
self.options: List[str] = options
|
float(self.default)
|
||||||
if not label:
|
except ValueError:
|
||||||
label = _id
|
raise CoreConfigError(
|
||||||
self.label: str = label
|
f"{self.id} is not a valid float: {self.default}"
|
||||||
|
)
|
||||||
def __str__(self):
|
elif self.type != ConfigDataTypes.STRING:
|
||||||
return (
|
if self.default:
|
||||||
f"{self.__class__.__name__}(id={self.id}, type={self.type}, "
|
try:
|
||||||
f"default={self.default}, options={self.options})"
|
int(self.default)
|
||||||
)
|
except ValueError:
|
||||||
|
raise CoreConfigError(
|
||||||
|
f"{self.id} is not a valid int: {self.default}"
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
class ConfigurableOptions:
|
class ConfigurableOptions:
|
||||||
|
@ -182,7 +180,7 @@ class ConfigurableManager:
|
||||||
:param config_type: configuration type to store configuration for
|
:param config_type: configuration type to store configuration for
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
logging.debug(
|
logger.debug(
|
||||||
"setting config for node(%s) type(%s): %s", node_id, config_type, config
|
"setting config for node(%s) type(%s): %s", node_id, config_type, config
|
||||||
)
|
)
|
||||||
node_configs = self.node_configurations.setdefault(node_id, OrderedDict())
|
node_configs = self.node_configurations.setdefault(node_id, OrderedDict())
|
||||||
|
@ -314,7 +312,7 @@ class ModelManager(ConfigurableManager):
|
||||||
:param config: model configuration, None for default configuration
|
:param config: model configuration, None for default configuration
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
logging.debug(
|
logger.debug(
|
||||||
"setting model(%s) for node(%s): %s", model_class.name, node.id, config
|
"setting model(%s) for node(%s): %s", model_class.name, node.id, config
|
||||||
)
|
)
|
||||||
self.set_model_config(node.id, model_class.name, config)
|
self.set_model_config(node.id, model_class.name, config)
|
||||||
|
@ -343,5 +341,5 @@ class ModelManager(ConfigurableManager):
|
||||||
model_class = self.models[model_name]
|
model_class = self.models[model_name]
|
||||||
models.append((model_class, config))
|
models.append((model_class, config))
|
||||||
|
|
||||||
logging.debug("models for node(%s): %s", node.id, models)
|
logger.debug("models for node(%s): %s", node.id, models)
|
||||||
return models
|
return models
|
||||||
|
|
|
@ -2,9 +2,10 @@ import abc
|
||||||
import enum
|
import enum
|
||||||
import inspect
|
import inspect
|
||||||
import logging
|
import logging
|
||||||
import pathlib
|
|
||||||
import time
|
import time
|
||||||
from typing import Any, Dict, List
|
from dataclasses import dataclass
|
||||||
|
from pathlib import Path
|
||||||
|
from typing import Any, Dict, List, Optional
|
||||||
|
|
||||||
from mako import exceptions
|
from mako import exceptions
|
||||||
from mako.lookup import TemplateLookup
|
from mako.lookup import TemplateLookup
|
||||||
|
@ -14,6 +15,7 @@ from core.config import Configuration
|
||||||
from core.errors import CoreCommandError, CoreError
|
from core.errors import CoreCommandError, CoreError
|
||||||
from core.nodes.base import CoreNode
|
from core.nodes.base import CoreNode
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
TEMPLATES_DIR: str = "templates"
|
TEMPLATES_DIR: str = "templates"
|
||||||
|
|
||||||
|
|
||||||
|
@ -27,6 +29,14 @@ class ConfigServiceBootError(Exception):
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass
|
||||||
|
class ShadowDir:
|
||||||
|
path: str
|
||||||
|
src: Optional[str] = None
|
||||||
|
templates: bool = False
|
||||||
|
has_node_paths: bool = False
|
||||||
|
|
||||||
|
|
||||||
class ConfigService(abc.ABC):
|
class ConfigService(abc.ABC):
|
||||||
"""
|
"""
|
||||||
Base class for creating configurable services.
|
Base class for creating configurable services.
|
||||||
|
@ -38,6 +48,9 @@ class ConfigService(abc.ABC):
|
||||||
# time to wait in seconds for determining if service started successfully
|
# time to wait in seconds for determining if service started successfully
|
||||||
validation_timer: int = 5
|
validation_timer: int = 5
|
||||||
|
|
||||||
|
# directories to shadow and copy files from
|
||||||
|
shadow_directories: List[ShadowDir] = []
|
||||||
|
|
||||||
def __init__(self, node: CoreNode) -> None:
|
def __init__(self, node: CoreNode) -> None:
|
||||||
"""
|
"""
|
||||||
Create ConfigService instance.
|
Create ConfigService instance.
|
||||||
|
@ -46,7 +59,7 @@ class ConfigService(abc.ABC):
|
||||||
"""
|
"""
|
||||||
self.node: CoreNode = node
|
self.node: CoreNode = node
|
||||||
class_file = inspect.getfile(self.__class__)
|
class_file = inspect.getfile(self.__class__)
|
||||||
templates_path = pathlib.Path(class_file).parent.joinpath(TEMPLATES_DIR)
|
templates_path = Path(class_file).parent.joinpath(TEMPLATES_DIR)
|
||||||
self.templates: TemplateLookup = TemplateLookup(directories=templates_path)
|
self.templates: TemplateLookup = TemplateLookup(directories=templates_path)
|
||||||
self.config: Dict[str, Configuration] = {}
|
self.config: Dict[str, Configuration] = {}
|
||||||
self.custom_templates: Dict[str, str] = {}
|
self.custom_templates: Dict[str, str] = {}
|
||||||
|
@ -133,7 +146,8 @@ class ConfigService(abc.ABC):
|
||||||
:return: nothing
|
:return: nothing
|
||||||
:raises ConfigServiceBootError: when there is an error starting service
|
:raises ConfigServiceBootError: when there is an error starting service
|
||||||
"""
|
"""
|
||||||
logging.info("node(%s) service(%s) starting...", self.node.name, self.name)
|
logger.info("node(%s) service(%s) starting...", self.node.name, self.name)
|
||||||
|
self.create_shadow_dirs()
|
||||||
self.create_dirs()
|
self.create_dirs()
|
||||||
self.create_files()
|
self.create_files()
|
||||||
wait = self.validation_mode == ConfigServiceMode.BLOCKING
|
wait = self.validation_mode == ConfigServiceMode.BLOCKING
|
||||||
|
@ -154,7 +168,7 @@ class ConfigService(abc.ABC):
|
||||||
try:
|
try:
|
||||||
self.node.cmd(cmd)
|
self.node.cmd(cmd)
|
||||||
except CoreCommandError:
|
except CoreCommandError:
|
||||||
logging.exception(
|
logger.exception(
|
||||||
f"node({self.node.name}) service({self.name}) "
|
f"node({self.node.name}) service({self.name}) "
|
||||||
f"failed shutdown: {cmd}"
|
f"failed shutdown: {cmd}"
|
||||||
)
|
)
|
||||||
|
@ -168,6 +182,64 @@ class ConfigService(abc.ABC):
|
||||||
self.stop()
|
self.stop()
|
||||||
self.start()
|
self.start()
|
||||||
|
|
||||||
|
def create_shadow_dirs(self) -> None:
|
||||||
|
"""
|
||||||
|
Creates a shadow of a host system directory recursively
|
||||||
|
to be mapped and live within a node.
|
||||||
|
|
||||||
|
:return: nothing
|
||||||
|
:raises CoreError: when there is a failure creating a directory or file
|
||||||
|
"""
|
||||||
|
for shadow_dir in self.shadow_directories:
|
||||||
|
# setup shadow and src paths, using node unique paths when configured
|
||||||
|
shadow_path = Path(shadow_dir.path)
|
||||||
|
if shadow_dir.src is None:
|
||||||
|
src_path = shadow_path
|
||||||
|
else:
|
||||||
|
src_path = Path(shadow_dir.src)
|
||||||
|
if shadow_dir.has_node_paths:
|
||||||
|
src_path = src_path / self.node.name
|
||||||
|
# validate shadow and src paths
|
||||||
|
if not shadow_path.is_absolute():
|
||||||
|
raise CoreError(f"shadow dir({shadow_path}) is not absolute")
|
||||||
|
if not src_path.is_absolute():
|
||||||
|
raise CoreError(f"shadow source dir({src_path}) is not absolute")
|
||||||
|
if not src_path.is_dir():
|
||||||
|
raise CoreError(f"shadow source dir({src_path}) does not exist")
|
||||||
|
# create root of the shadow path within node
|
||||||
|
logger.info(
|
||||||
|
"node(%s) creating shadow directory(%s) src(%s) node paths(%s) "
|
||||||
|
"templates(%s)",
|
||||||
|
self.node.name,
|
||||||
|
shadow_path,
|
||||||
|
src_path,
|
||||||
|
shadow_dir.has_node_paths,
|
||||||
|
shadow_dir.templates,
|
||||||
|
)
|
||||||
|
self.node.create_dir(shadow_path)
|
||||||
|
# find all directories and files to create
|
||||||
|
dir_paths = []
|
||||||
|
file_paths = []
|
||||||
|
for path in src_path.rglob("*"):
|
||||||
|
shadow_src_path = shadow_path / path.relative_to(src_path)
|
||||||
|
if path.is_dir():
|
||||||
|
dir_paths.append(shadow_src_path)
|
||||||
|
else:
|
||||||
|
file_paths.append((path, shadow_src_path))
|
||||||
|
# create all directories within node
|
||||||
|
for path in dir_paths:
|
||||||
|
self.node.create_dir(path)
|
||||||
|
# create all files within node, from templates when configured
|
||||||
|
data = self.data()
|
||||||
|
templates = TemplateLookup(directories=src_path)
|
||||||
|
for path, dst_path in file_paths:
|
||||||
|
if shadow_dir.templates:
|
||||||
|
template = templates.get_template(path.name)
|
||||||
|
rendered = self._render(template, data)
|
||||||
|
self.node.create_file(dst_path, rendered)
|
||||||
|
else:
|
||||||
|
self.node.copy_file(path, dst_path)
|
||||||
|
|
||||||
def create_dirs(self) -> None:
|
def create_dirs(self) -> None:
|
||||||
"""
|
"""
|
||||||
Creates directories for service.
|
Creates directories for service.
|
||||||
|
@ -175,10 +247,12 @@ class ConfigService(abc.ABC):
|
||||||
:return: nothing
|
:return: nothing
|
||||||
:raises CoreError: when there is a failure creating a directory
|
:raises CoreError: when there is a failure creating a directory
|
||||||
"""
|
"""
|
||||||
for directory in self.directories:
|
logger.debug("creating config service directories")
|
||||||
|
for directory in sorted(self.directories):
|
||||||
|
dir_path = Path(directory)
|
||||||
try:
|
try:
|
||||||
self.node.privatedir(directory)
|
self.node.create_dir(dir_path)
|
||||||
except (CoreCommandError, ValueError):
|
except (CoreCommandError, CoreError):
|
||||||
raise CoreError(
|
raise CoreError(
|
||||||
f"node({self.node.name}) service({self.name}) "
|
f"node({self.node.name}) service({self.name}) "
|
||||||
f"failure to create service directory: {directory}"
|
f"failure to create service directory: {directory}"
|
||||||
|
@ -219,17 +293,21 @@ class ConfigService(abc.ABC):
|
||||||
:return: mapping of files to templates
|
:return: mapping of files to templates
|
||||||
"""
|
"""
|
||||||
templates = {}
|
templates = {}
|
||||||
for name in self.files:
|
for file in self.files:
|
||||||
basename = pathlib.Path(name).name
|
file_path = Path(file)
|
||||||
if name in self.custom_templates:
|
if file_path.is_absolute():
|
||||||
template = self.custom_templates[name]
|
template_path = str(file_path.relative_to("/"))
|
||||||
template = self.clean_text(template)
|
|
||||||
elif self.templates.has_template(basename):
|
|
||||||
template = self.templates.get_template(basename).source
|
|
||||||
else:
|
else:
|
||||||
template = self.get_text_template(name)
|
template_path = str(file_path)
|
||||||
|
if file in self.custom_templates:
|
||||||
|
template = self.custom_templates[file]
|
||||||
template = self.clean_text(template)
|
template = self.clean_text(template)
|
||||||
templates[name] = template
|
elif self.templates.has_template(template_path):
|
||||||
|
template = self.templates.get_template(template_path).source
|
||||||
|
else:
|
||||||
|
template = self.get_text_template(file)
|
||||||
|
template = self.clean_text(template)
|
||||||
|
templates[file] = template
|
||||||
return templates
|
return templates
|
||||||
|
|
||||||
def create_files(self) -> None:
|
def create_files(self) -> None:
|
||||||
|
@ -239,24 +317,20 @@ class ConfigService(abc.ABC):
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
data = self.data()
|
data = self.data()
|
||||||
for name in self.files:
|
for file in sorted(self.files):
|
||||||
basename = pathlib.Path(name).name
|
logger.debug(
|
||||||
if name in self.custom_templates:
|
"node(%s) service(%s) template(%s)", self.node.name, self.name, file
|
||||||
text = self.custom_templates[name]
|
|
||||||
rendered = self.render_text(text, data)
|
|
||||||
elif self.templates.has_template(basename):
|
|
||||||
rendered = self.render_template(basename, data)
|
|
||||||
else:
|
|
||||||
text = self.get_text_template(name)
|
|
||||||
rendered = self.render_text(text, data)
|
|
||||||
logging.debug(
|
|
||||||
"node(%s) service(%s) template(%s): \n%s",
|
|
||||||
self.node.name,
|
|
||||||
self.name,
|
|
||||||
name,
|
|
||||||
rendered,
|
|
||||||
)
|
)
|
||||||
self.node.nodefile(name, rendered)
|
file_path = Path(file)
|
||||||
|
if file in self.custom_templates:
|
||||||
|
text = self.custom_templates[file]
|
||||||
|
rendered = self.render_text(text, data)
|
||||||
|
elif self.templates.has_template(file_path.name):
|
||||||
|
rendered = self.render_template(file_path.name, data)
|
||||||
|
else:
|
||||||
|
text = self.get_text_template(file)
|
||||||
|
rendered = self.render_text(text, data)
|
||||||
|
self.node.create_file(file_path, rendered)
|
||||||
|
|
||||||
def run_startup(self, wait: bool) -> None:
|
def run_startup(self, wait: bool) -> None:
|
||||||
"""
|
"""
|
||||||
|
@ -300,7 +374,7 @@ class ConfigService(abc.ABC):
|
||||||
del cmds[index]
|
del cmds[index]
|
||||||
index += 1
|
index += 1
|
||||||
except CoreCommandError:
|
except CoreCommandError:
|
||||||
logging.debug(
|
logger.debug(
|
||||||
f"node({self.node.name}) service({self.name}) "
|
f"node({self.node.name}) service({self.name}) "
|
||||||
f"validate command failed: {cmd}"
|
f"validate command failed: {cmd}"
|
||||||
)
|
)
|
||||||
|
|
|
@ -1,6 +1,8 @@
|
||||||
import logging
|
import logging
|
||||||
from typing import TYPE_CHECKING, Dict, List, Set
|
from typing import TYPE_CHECKING, Dict, List, Set
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.configservice.base import ConfigService
|
from core.configservice.base import ConfigService
|
||||||
|
|
||||||
|
@ -41,7 +43,7 @@ class ConfigServiceDependencies:
|
||||||
for name in self.node_services:
|
for name in self.node_services:
|
||||||
service = self.node_services[name]
|
service = self.node_services[name]
|
||||||
if service.name in self.started:
|
if service.name in self.started:
|
||||||
logging.debug(
|
logger.debug(
|
||||||
"skipping service that will already be started: %s", service.name
|
"skipping service that will already be started: %s", service.name
|
||||||
)
|
)
|
||||||
continue
|
continue
|
||||||
|
@ -75,7 +77,7 @@ class ConfigServiceDependencies:
|
||||||
:param service: service to check dependencies for
|
:param service: service to check dependencies for
|
||||||
:return: list of config services to start in order
|
:return: list of config services to start in order
|
||||||
"""
|
"""
|
||||||
logging.debug("starting service dependency check: %s", service.name)
|
logger.debug("starting service dependency check: %s", service.name)
|
||||||
self._reset()
|
self._reset()
|
||||||
return self._visit(service)
|
return self._visit(service)
|
||||||
|
|
||||||
|
@ -86,7 +88,7 @@ class ConfigServiceDependencies:
|
||||||
:param current_service: service being visited
|
:param current_service: service being visited
|
||||||
:return: list of dependent services for a visited service
|
:return: list of dependent services for a visited service
|
||||||
"""
|
"""
|
||||||
logging.debug("visiting service(%s): %s", current_service.name, self.path)
|
logger.debug("visiting service(%s): %s", current_service.name, self.path)
|
||||||
self.visited.add(current_service.name)
|
self.visited.add(current_service.name)
|
||||||
self.visiting.add(current_service.name)
|
self.visiting.add(current_service.name)
|
||||||
|
|
||||||
|
@ -109,7 +111,7 @@ class ConfigServiceDependencies:
|
||||||
self._visit(service)
|
self._visit(service)
|
||||||
|
|
||||||
# add service when bottom is found
|
# add service when bottom is found
|
||||||
logging.debug("adding service to startup path: %s", current_service.name)
|
logger.debug("adding service to startup path: %s", current_service.name)
|
||||||
self.started.add(current_service.name)
|
self.started.add(current_service.name)
|
||||||
self.path.append(current_service)
|
self.path.append(current_service)
|
||||||
self.visiting.remove(current_service.name)
|
self.visiting.remove(current_service.name)
|
||||||
|
|
|
@ -1,11 +1,15 @@
|
||||||
import logging
|
import logging
|
||||||
import pathlib
|
import pathlib
|
||||||
|
import pkgutil
|
||||||
|
from pathlib import Path
|
||||||
from typing import Dict, List, Type
|
from typing import Dict, List, Type
|
||||||
|
|
||||||
from core import utils
|
from core import configservices, utils
|
||||||
from core.configservice.base import ConfigService
|
from core.configservice.base import ConfigService
|
||||||
from core.errors import CoreError
|
from core.errors import CoreError
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
class ConfigServiceManager:
|
class ConfigServiceManager:
|
||||||
"""
|
"""
|
||||||
|
@ -28,7 +32,7 @@ class ConfigServiceManager:
|
||||||
"""
|
"""
|
||||||
service_class = self.services.get(name)
|
service_class = self.services.get(name)
|
||||||
if service_class is None:
|
if service_class is None:
|
||||||
raise CoreError(f"service does not exit {name}")
|
raise CoreError(f"service does not exist {name}")
|
||||||
return service_class
|
return service_class
|
||||||
|
|
||||||
def add(self, service: Type[ConfigService]) -> None:
|
def add(self, service: Type[ConfigService]) -> None:
|
||||||
|
@ -40,7 +44,7 @@ class ConfigServiceManager:
|
||||||
:raises CoreError: when service is a duplicate or has unmet executables
|
:raises CoreError: when service is a duplicate or has unmet executables
|
||||||
"""
|
"""
|
||||||
name = service.name
|
name = service.name
|
||||||
logging.debug(
|
logger.debug(
|
||||||
"loading service: class(%s) name(%s)", service.__class__.__name__, name
|
"loading service: class(%s) name(%s)", service.__class__.__name__, name
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -55,27 +59,46 @@ class ConfigServiceManager:
|
||||||
except CoreError as e:
|
except CoreError as e:
|
||||||
raise CoreError(f"config service({service.name}): {e}")
|
raise CoreError(f"config service({service.name}): {e}")
|
||||||
|
|
||||||
# make service available
|
# make service available
|
||||||
self.services[name] = service
|
self.services[name] = service
|
||||||
|
|
||||||
def load(self, path: str) -> List[str]:
|
def load_locals(self) -> List[str]:
|
||||||
"""
|
"""
|
||||||
Search path provided for configurable services and add them for being managed.
|
Search and add config service from local core module.
|
||||||
|
|
||||||
|
:return: list of errors when loading services
|
||||||
|
"""
|
||||||
|
errors = []
|
||||||
|
for module_info in pkgutil.walk_packages(
|
||||||
|
configservices.__path__, f"{configservices.__name__}."
|
||||||
|
):
|
||||||
|
services = utils.load_module(module_info.name, ConfigService)
|
||||||
|
for service in services:
|
||||||
|
try:
|
||||||
|
self.add(service)
|
||||||
|
except CoreError as e:
|
||||||
|
errors.append(service.name)
|
||||||
|
logger.debug("not loading config service(%s): %s", service.name, e)
|
||||||
|
return errors
|
||||||
|
|
||||||
|
def load(self, path: Path) -> List[str]:
|
||||||
|
"""
|
||||||
|
Search path provided for config services and add them for being managed.
|
||||||
|
|
||||||
:param path: path to search configurable services
|
:param path: path to search configurable services
|
||||||
:return: list errors when loading and adding services
|
:return: list errors when loading services
|
||||||
"""
|
"""
|
||||||
path = pathlib.Path(path)
|
path = pathlib.Path(path)
|
||||||
subdirs = [x for x in path.iterdir() if x.is_dir()]
|
subdirs = [x for x in path.iterdir() if x.is_dir()]
|
||||||
subdirs.append(path)
|
subdirs.append(path)
|
||||||
service_errors = []
|
service_errors = []
|
||||||
for subdir in subdirs:
|
for subdir in subdirs:
|
||||||
logging.debug("loading config services from: %s", subdir)
|
logger.debug("loading config services from: %s", subdir)
|
||||||
services = utils.load_classes(str(subdir), ConfigService)
|
services = utils.load_classes(subdir, ConfigService)
|
||||||
for service in services:
|
for service in services:
|
||||||
try:
|
try:
|
||||||
self.add(service)
|
self.add(service)
|
||||||
except CoreError as e:
|
except CoreError as e:
|
||||||
service_errors.append(service.name)
|
service_errors.append(service.name)
|
||||||
logging.debug("not loading service(%s): %s", service.name, e)
|
logger.debug("not loading service(%s): %s", service.name, e)
|
||||||
return service_errors
|
return service_errors
|
||||||
|
|
|
@ -9,6 +9,7 @@ from core.nodes.base import CoreNodeBase
|
||||||
from core.nodes.interface import DEFAULT_MTU, CoreInterface
|
from core.nodes.interface import DEFAULT_MTU, CoreInterface
|
||||||
from core.nodes.network import WlanNode
|
from core.nodes.network import WlanNode
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
GROUP: str = "Quagga"
|
GROUP: str = "Quagga"
|
||||||
QUAGGA_STATE_DIR: str = "/var/run/quagga"
|
QUAGGA_STATE_DIR: str = "/var/run/quagga"
|
||||||
|
|
||||||
|
@ -101,9 +102,9 @@ class Zebra(ConfigService):
|
||||||
ip4s = []
|
ip4s = []
|
||||||
ip6s = []
|
ip6s = []
|
||||||
for ip4 in iface.ip4s:
|
for ip4 in iface.ip4s:
|
||||||
ip4s.append(str(ip4.ip))
|
ip4s.append(str(ip4))
|
||||||
for ip6 in iface.ip6s:
|
for ip6 in iface.ip6s:
|
||||||
ip6s.append(str(ip6.ip))
|
ip6s.append(str(ip6))
|
||||||
ifaces.append((iface, ip4s, ip6s, iface.control))
|
ifaces.append((iface, ip4s, ip6s, iface.control))
|
||||||
|
|
||||||
return dict(
|
return dict(
|
||||||
|
@ -226,12 +227,6 @@ class Ospfv3mdr(Ospfv3):
|
||||||
|
|
||||||
name: str = "OSPFv3MDR"
|
name: str = "OSPFv3MDR"
|
||||||
|
|
||||||
def data(self) -> Dict[str, Any]:
|
|
||||||
for iface in self.node.get_ifaces():
|
|
||||||
is_wireless = isinstance(iface.net, (WlanNode, EmaneNet))
|
|
||||||
logging.info("MDR wireless: %s", is_wireless)
|
|
||||||
return dict()
|
|
||||||
|
|
||||||
def quagga_iface_config(self, iface: CoreInterface) -> str:
|
def quagga_iface_config(self, iface: CoreInterface) -> str:
|
||||||
config = super().quagga_iface_config(iface)
|
config = super().quagga_iface_config(iface)
|
||||||
if isinstance(iface.net, (WlanNode, EmaneNet)):
|
if isinstance(iface.net, (WlanNode, EmaneNet)):
|
||||||
|
|
|
@ -20,20 +20,20 @@ class VpnClient(ConfigService):
|
||||||
validation_mode: ConfigServiceMode = ConfigServiceMode.BLOCKING
|
validation_mode: ConfigServiceMode = ConfigServiceMode.BLOCKING
|
||||||
default_configs: List[Configuration] = [
|
default_configs: List[Configuration] = [
|
||||||
Configuration(
|
Configuration(
|
||||||
_id="keydir",
|
id="keydir",
|
||||||
_type=ConfigDataTypes.STRING,
|
type=ConfigDataTypes.STRING,
|
||||||
label="Key Dir",
|
label="Key Dir",
|
||||||
default="/etc/core/keys",
|
default="/etc/core/keys",
|
||||||
),
|
),
|
||||||
Configuration(
|
Configuration(
|
||||||
_id="keyname",
|
id="keyname",
|
||||||
_type=ConfigDataTypes.STRING,
|
type=ConfigDataTypes.STRING,
|
||||||
label="Key Name",
|
label="Key Name",
|
||||||
default="client1",
|
default="client1",
|
||||||
),
|
),
|
||||||
Configuration(
|
Configuration(
|
||||||
_id="server",
|
id="server",
|
||||||
_type=ConfigDataTypes.STRING,
|
type=ConfigDataTypes.STRING,
|
||||||
label="Server",
|
label="Server",
|
||||||
default="10.0.2.10",
|
default="10.0.2.10",
|
||||||
),
|
),
|
||||||
|
@ -54,20 +54,20 @@ class VpnServer(ConfigService):
|
||||||
validation_mode: ConfigServiceMode = ConfigServiceMode.BLOCKING
|
validation_mode: ConfigServiceMode = ConfigServiceMode.BLOCKING
|
||||||
default_configs: List[Configuration] = [
|
default_configs: List[Configuration] = [
|
||||||
Configuration(
|
Configuration(
|
||||||
_id="keydir",
|
id="keydir",
|
||||||
_type=ConfigDataTypes.STRING,
|
type=ConfigDataTypes.STRING,
|
||||||
label="Key Dir",
|
label="Key Dir",
|
||||||
default="/etc/core/keys",
|
default="/etc/core/keys",
|
||||||
),
|
),
|
||||||
Configuration(
|
Configuration(
|
||||||
_id="keyname",
|
id="keyname",
|
||||||
_type=ConfigDataTypes.STRING,
|
type=ConfigDataTypes.STRING,
|
||||||
label="Key Name",
|
label="Key Name",
|
||||||
default="server",
|
default="server",
|
||||||
),
|
),
|
||||||
Configuration(
|
Configuration(
|
||||||
_id="subnet",
|
id="subnet",
|
||||||
_type=ConfigDataTypes.STRING,
|
type=ConfigDataTypes.STRING,
|
||||||
label="Subnet",
|
label="Subnet",
|
||||||
default="10.0.200.0",
|
default="10.0.200.0",
|
||||||
),
|
),
|
||||||
|
|
|
@ -1,49 +0,0 @@
|
||||||
from typing import Dict, List
|
|
||||||
|
|
||||||
from core.config import Configuration
|
|
||||||
from core.configservice.base import ConfigService, ConfigServiceMode
|
|
||||||
from core.emulator.enumerations import ConfigDataTypes
|
|
||||||
|
|
||||||
|
|
||||||
class SimpleService(ConfigService):
|
|
||||||
name: str = "Simple"
|
|
||||||
group: str = "SimpleGroup"
|
|
||||||
directories: List[str] = ["/etc/quagga", "/usr/local/lib"]
|
|
||||||
files: List[str] = ["test1.sh", "test2.sh"]
|
|
||||||
executables: List[str] = []
|
|
||||||
dependencies: List[str] = []
|
|
||||||
startup: List[str] = []
|
|
||||||
validate: List[str] = []
|
|
||||||
shutdown: List[str] = []
|
|
||||||
validation_mode: ConfigServiceMode = ConfigServiceMode.BLOCKING
|
|
||||||
default_configs: List[Configuration] = [
|
|
||||||
Configuration(_id="value1", _type=ConfigDataTypes.STRING, label="Text"),
|
|
||||||
Configuration(_id="value2", _type=ConfigDataTypes.BOOL, label="Boolean"),
|
|
||||||
Configuration(
|
|
||||||
_id="value3",
|
|
||||||
_type=ConfigDataTypes.STRING,
|
|
||||||
label="Multiple Choice",
|
|
||||||
options=["value1", "value2", "value3"],
|
|
||||||
),
|
|
||||||
]
|
|
||||||
modes: Dict[str, Dict[str, str]] = {
|
|
||||||
"mode1": {"value1": "value1", "value2": "0", "value3": "value2"},
|
|
||||||
"mode2": {"value1": "value2", "value2": "1", "value3": "value3"},
|
|
||||||
"mode3": {"value1": "value3", "value2": "0", "value3": "value1"},
|
|
||||||
}
|
|
||||||
|
|
||||||
def get_text_template(self, name: str) -> str:
|
|
||||||
if name == "test1.sh":
|
|
||||||
return """
|
|
||||||
# sample script 1
|
|
||||||
# node id(${node.id}) name(${node.name})
|
|
||||||
# config: ${config}
|
|
||||||
echo hello
|
|
||||||
"""
|
|
||||||
elif name == "test2.sh":
|
|
||||||
return """
|
|
||||||
# sample script 2
|
|
||||||
# node id(${node.id}) name(${node.name})
|
|
||||||
# config: ${config}
|
|
||||||
echo hello2
|
|
||||||
"""
|
|
|
@ -149,11 +149,13 @@ class DhcpService(ConfigService):
|
||||||
subnets = []
|
subnets = []
|
||||||
for iface in self.node.get_ifaces(control=False):
|
for iface in self.node.get_ifaces(control=False):
|
||||||
for ip4 in iface.ip4s:
|
for ip4 in iface.ip4s:
|
||||||
|
if ip4.size == 1:
|
||||||
|
continue
|
||||||
# divide the address space in half
|
# divide the address space in half
|
||||||
index = (ip4.size - 2) / 2
|
index = (ip4.size - 2) / 2
|
||||||
rangelow = ip4[index]
|
rangelow = ip4[index]
|
||||||
rangehigh = ip4[-2]
|
rangehigh = ip4[-2]
|
||||||
subnets.append((ip4.ip, ip4.netmask, rangelow, rangehigh, str(ip4.ip)))
|
subnets.append((ip4.cidr.ip, ip4.netmask, rangelow, rangehigh, ip4.ip))
|
||||||
return dict(subnets=subnets)
|
return dict(subnets=subnets)
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -1,3 +1,5 @@
|
||||||
COREDPY_VERSION = "@PACKAGE_VERSION@"
|
from pathlib import Path
|
||||||
CORE_CONF_DIR = "@CORE_CONF_DIR@"
|
|
||||||
CORE_DATA_DIR = "@CORE_DATA_DIR@"
|
COREDPY_VERSION: str = "@PACKAGE_VERSION@"
|
||||||
|
CORE_CONF_DIR: Path = Path("@CORE_CONF_DIR@")
|
||||||
|
CORE_DATA_DIR: Path = Path("@CORE_DATA_DIR@")
|
||||||
|
|
File diff suppressed because it is too large
Load diff
|
@ -1,9 +1,12 @@
|
||||||
import logging
|
import logging
|
||||||
|
from pathlib import Path
|
||||||
from typing import Dict, List
|
from typing import Dict, List
|
||||||
|
|
||||||
from core.config import Configuration
|
from core.config import Configuration
|
||||||
from core.emulator.enumerations import ConfigDataTypes
|
from core.emulator.enumerations import ConfigDataTypes
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
manifest = None
|
manifest = None
|
||||||
try:
|
try:
|
||||||
from emane.shell import manifest
|
from emane.shell import manifest
|
||||||
|
@ -12,7 +15,7 @@ except ImportError:
|
||||||
from emanesh import manifest
|
from emanesh import manifest
|
||||||
except ImportError:
|
except ImportError:
|
||||||
manifest = None
|
manifest = None
|
||||||
logging.debug("compatible emane python bindings not installed")
|
logger.debug("compatible emane python bindings not installed")
|
||||||
|
|
||||||
|
|
||||||
def _type_value(config_type: str) -> ConfigDataTypes:
|
def _type_value(config_type: str) -> ConfigDataTypes:
|
||||||
|
@ -71,9 +74,10 @@ def _get_default(config_type_name: str, config_value: List[str]) -> str:
|
||||||
return config_default
|
return config_default
|
||||||
|
|
||||||
|
|
||||||
def parse(manifest_path: str, defaults: Dict[str, str]) -> List[Configuration]:
|
def parse(manifest_path: Path, defaults: Dict[str, str]) -> List[Configuration]:
|
||||||
"""
|
"""
|
||||||
Parses a valid emane manifest file and converts the provided configuration values into ones used by core.
|
Parses a valid emane manifest file and converts the provided configuration values
|
||||||
|
into ones used by core.
|
||||||
|
|
||||||
:param manifest_path: absolute manifest file path
|
:param manifest_path: absolute manifest file path
|
||||||
:param defaults: used to override default values for configurations
|
:param defaults: used to override default values for configurations
|
||||||
|
@ -85,7 +89,7 @@ def parse(manifest_path: str, defaults: Dict[str, str]) -> List[Configuration]:
|
||||||
return []
|
return []
|
||||||
|
|
||||||
# load configuration file
|
# load configuration file
|
||||||
manifest_file = manifest.Manifest(manifest_path)
|
manifest_file = manifest.Manifest(str(manifest_path))
|
||||||
manifest_configurations = manifest_file.getAllConfiguration()
|
manifest_configurations = manifest_file.getAllConfiguration()
|
||||||
|
|
||||||
configurations = []
|
configurations = []
|
||||||
|
@ -116,8 +120,8 @@ def parse(manifest_path: str, defaults: Dict[str, str]) -> List[Configuration]:
|
||||||
config_descriptions = f"{config_descriptions} file"
|
config_descriptions = f"{config_descriptions} file"
|
||||||
|
|
||||||
configuration = Configuration(
|
configuration = Configuration(
|
||||||
_id=config_name,
|
id=config_name,
|
||||||
_type=config_type_value,
|
type=config_type_value,
|
||||||
default=config_default,
|
default=config_default,
|
||||||
options=possible,
|
options=possible,
|
||||||
label=config_descriptions,
|
label=config_descriptions,
|
||||||
|
|
|
@ -2,12 +2,11 @@
|
||||||
Defines Emane Models used within CORE.
|
Defines Emane Models used within CORE.
|
||||||
"""
|
"""
|
||||||
import logging
|
import logging
|
||||||
import os
|
from pathlib import Path
|
||||||
from typing import Dict, List, Optional, Set
|
from typing import Dict, List, Optional, Set
|
||||||
|
|
||||||
from core.config import ConfigGroup, Configuration
|
from core.config import ConfigGroup, Configuration
|
||||||
from core.emane import emanemanifest
|
from core.emane import emanemanifest
|
||||||
from core.emane.nodes import EmaneNet
|
|
||||||
from core.emulator.data import LinkOptions
|
from core.emulator.data import LinkOptions
|
||||||
from core.emulator.enumerations import ConfigDataTypes
|
from core.emulator.enumerations import ConfigDataTypes
|
||||||
from core.errors import CoreError
|
from core.errors import CoreError
|
||||||
|
@ -15,6 +14,10 @@ from core.location.mobility import WirelessModel
|
||||||
from core.nodes.interface import CoreInterface
|
from core.nodes.interface import CoreInterface
|
||||||
from core.xml import emanexml
|
from core.xml import emanexml
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
DEFAULT_DEV: str = "ctrl0"
|
||||||
|
MANIFEST_PATH: str = "share/emane/manifest"
|
||||||
|
|
||||||
|
|
||||||
class EmaneModel(WirelessModel):
|
class EmaneModel(WirelessModel):
|
||||||
"""
|
"""
|
||||||
|
@ -23,6 +26,17 @@ class EmaneModel(WirelessModel):
|
||||||
configurable parameters. Helper functions also live here.
|
configurable parameters. Helper functions also live here.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
|
# default platform configuration settings
|
||||||
|
platform_controlport: str = "controlportendpoint"
|
||||||
|
platform_xml: str = "nemmanager.xml"
|
||||||
|
platform_defaults: Dict[str, str] = {
|
||||||
|
"eventservicedevice": DEFAULT_DEV,
|
||||||
|
"eventservicegroup": "224.1.2.8:45703",
|
||||||
|
"otamanagerdevice": DEFAULT_DEV,
|
||||||
|
"otamanagergroup": "224.1.2.8:45702",
|
||||||
|
}
|
||||||
|
platform_config: List[Configuration] = []
|
||||||
|
|
||||||
# default mac configuration settings
|
# default mac configuration settings
|
||||||
mac_library: Optional[str] = None
|
mac_library: Optional[str] = None
|
||||||
mac_xml: Optional[str] = None
|
mac_xml: Optional[str] = None
|
||||||
|
@ -53,23 +67,37 @@ class EmaneModel(WirelessModel):
|
||||||
config_ignore: Set[str] = set()
|
config_ignore: Set[str] = set()
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def load(cls, emane_prefix: str) -> None:
|
def load(cls, emane_prefix: Path) -> None:
|
||||||
"""
|
"""
|
||||||
Called after being loaded within the EmaneManager. Provides configured emane_prefix for
|
Called after being loaded within the EmaneManager. Provides configured
|
||||||
parsing xml files.
|
emane_prefix for parsing xml files.
|
||||||
|
|
||||||
:param emane_prefix: configured emane prefix path
|
:param emane_prefix: configured emane prefix path
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
manifest_path = "share/emane/manifest"
|
cls._load_platform_config(emane_prefix)
|
||||||
# load mac configuration
|
# load mac configuration
|
||||||
mac_xml_path = os.path.join(emane_prefix, manifest_path, cls.mac_xml)
|
mac_xml_path = emane_prefix / MANIFEST_PATH / cls.mac_xml
|
||||||
cls.mac_config = emanemanifest.parse(mac_xml_path, cls.mac_defaults)
|
cls.mac_config = emanemanifest.parse(mac_xml_path, cls.mac_defaults)
|
||||||
|
|
||||||
# load phy configuration
|
# load phy configuration
|
||||||
phy_xml_path = os.path.join(emane_prefix, manifest_path, cls.phy_xml)
|
phy_xml_path = emane_prefix / MANIFEST_PATH / cls.phy_xml
|
||||||
cls.phy_config = emanemanifest.parse(phy_xml_path, cls.phy_defaults)
|
cls.phy_config = emanemanifest.parse(phy_xml_path, cls.phy_defaults)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def _load_platform_config(cls, emane_prefix: Path) -> None:
|
||||||
|
platform_xml_path = emane_prefix / MANIFEST_PATH / cls.platform_xml
|
||||||
|
cls.platform_config = emanemanifest.parse(
|
||||||
|
platform_xml_path, cls.platform_defaults
|
||||||
|
)
|
||||||
|
# remove controlport configuration, since core will set this directly
|
||||||
|
controlport_index = None
|
||||||
|
for index, configuration in enumerate(cls.platform_config):
|
||||||
|
if configuration.id == cls.platform_controlport:
|
||||||
|
controlport_index = index
|
||||||
|
break
|
||||||
|
if controlport_index is not None:
|
||||||
|
cls.platform_config.pop(controlport_index)
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def configurations(cls) -> List[Configuration]:
|
def configurations(cls) -> List[Configuration]:
|
||||||
"""
|
"""
|
||||||
|
@ -77,7 +105,9 @@ class EmaneModel(WirelessModel):
|
||||||
|
|
||||||
:return: all configurations
|
:return: all configurations
|
||||||
"""
|
"""
|
||||||
return cls.mac_config + cls.phy_config + cls.external_config
|
return (
|
||||||
|
cls.platform_config + cls.mac_config + cls.phy_config + cls.external_config
|
||||||
|
)
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def config_groups(cls) -> List[ConfigGroup]:
|
def config_groups(cls) -> List[ConfigGroup]:
|
||||||
|
@ -86,11 +116,13 @@ class EmaneModel(WirelessModel):
|
||||||
|
|
||||||
:return: list of configuration groups.
|
:return: list of configuration groups.
|
||||||
"""
|
"""
|
||||||
mac_len = len(cls.mac_config)
|
platform_len = len(cls.platform_config)
|
||||||
|
mac_len = len(cls.mac_config) + platform_len
|
||||||
phy_len = len(cls.phy_config) + mac_len
|
phy_len = len(cls.phy_config) + mac_len
|
||||||
config_len = len(cls.configurations())
|
config_len = len(cls.configurations())
|
||||||
return [
|
return [
|
||||||
ConfigGroup("MAC Parameters", 1, mac_len),
|
ConfigGroup("Platform Parameters", 1, platform_len),
|
||||||
|
ConfigGroup("MAC Parameters", platform_len + 1, mac_len),
|
||||||
ConfigGroup("PHY Parameters", mac_len + 1, phy_len),
|
ConfigGroup("PHY Parameters", mac_len + 1, phy_len),
|
||||||
ConfigGroup("External Parameters", phy_len + 1, config_len),
|
ConfigGroup("External Parameters", phy_len + 1, config_len),
|
||||||
]
|
]
|
||||||
|
@ -110,13 +142,14 @@ class EmaneModel(WirelessModel):
|
||||||
emanexml.create_phy_xml(self, iface, config)
|
emanexml.create_phy_xml(self, iface, config)
|
||||||
emanexml.create_transport_xml(iface, config)
|
emanexml.create_transport_xml(iface, config)
|
||||||
|
|
||||||
def post_startup(self) -> None:
|
def post_startup(self, iface: CoreInterface) -> None:
|
||||||
"""
|
"""
|
||||||
Logic to execute after the emane manager is finished with startup.
|
Logic to execute after the emane manager is finished with startup.
|
||||||
|
|
||||||
|
:param iface: interface for post startup
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
logging.debug("emane model(%s) has no post setup tasks", self.name)
|
logger.debug("emane model(%s) has no post setup tasks", self.name)
|
||||||
|
|
||||||
def update(self, moved_ifaces: List[CoreInterface]) -> None:
|
def update(self, moved_ifaces: List[CoreInterface]) -> None:
|
||||||
"""
|
"""
|
||||||
|
@ -128,10 +161,9 @@ class EmaneModel(WirelessModel):
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
try:
|
try:
|
||||||
emane_net = self.session.get_node(self.id, EmaneNet)
|
self.session.emane.set_nem_positions(moved_ifaces)
|
||||||
emane_net.setnempositions(moved_ifaces)
|
|
||||||
except CoreError:
|
except CoreError:
|
||||||
logging.exception("error during update")
|
logger.exception("error during update")
|
||||||
|
|
||||||
def linkconfig(
|
def linkconfig(
|
||||||
self, iface: CoreInterface, options: LinkOptions, iface2: CoreInterface = None
|
self, iface: CoreInterface, options: LinkOptions, iface2: CoreInterface = None
|
||||||
|
@ -144,4 +176,4 @@ class EmaneModel(WirelessModel):
|
||||||
:param iface2: interface two
|
:param iface2: interface two
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
logging.warning("emane model(%s) does not support link config", self.name)
|
logger.warning("emane model(%s) does not support link config", self.name)
|
||||||
|
|
|
@ -6,10 +6,13 @@ from typing import TYPE_CHECKING, Dict, List, Optional, Set, Tuple
|
||||||
|
|
||||||
from lxml import etree
|
from lxml import etree
|
||||||
|
|
||||||
|
from core.emane.nodes import EmaneNet
|
||||||
from core.emulator.data import LinkData
|
from core.emulator.data import LinkData
|
||||||
from core.emulator.enumerations import LinkTypes, MessageFlags
|
from core.emulator.enumerations import LinkTypes, MessageFlags
|
||||||
from core.nodes.network import CtrlNet
|
from core.nodes.network import CtrlNet
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
from emane import shell
|
from emane import shell
|
||||||
except ImportError:
|
except ImportError:
|
||||||
|
@ -17,12 +20,11 @@ except ImportError:
|
||||||
from emanesh import shell
|
from emanesh import shell
|
||||||
except ImportError:
|
except ImportError:
|
||||||
shell = None
|
shell = None
|
||||||
logging.debug("compatible emane python bindings not installed")
|
logger.debug("compatible emane python bindings not installed")
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.emane.emanemanager import EmaneManager
|
from core.emane.emanemanager import EmaneManager
|
||||||
|
|
||||||
DEFAULT_PORT: int = 47_000
|
|
||||||
MAC_COMPONENT_INDEX: int = 1
|
MAC_COMPONENT_INDEX: int = 1
|
||||||
EMANE_RFPIPE: str = "rfpipemaclayer"
|
EMANE_RFPIPE: str = "rfpipemaclayer"
|
||||||
EMANE_80211: str = "ieee80211abgmaclayer"
|
EMANE_80211: str = "ieee80211abgmaclayer"
|
||||||
|
@ -77,10 +79,10 @@ class EmaneLink:
|
||||||
|
|
||||||
|
|
||||||
class EmaneClient:
|
class EmaneClient:
|
||||||
def __init__(self, address: str) -> None:
|
def __init__(self, address: str, port: int) -> None:
|
||||||
self.address: str = address
|
self.address: str = address
|
||||||
self.client: shell.ControlPortClient = shell.ControlPortClient(
|
self.client: shell.ControlPortClient = shell.ControlPortClient(
|
||||||
self.address, DEFAULT_PORT
|
self.address, port
|
||||||
)
|
)
|
||||||
self.nems: Dict[int, LossTable] = {}
|
self.nems: Dict[int, LossTable] = {}
|
||||||
self.setup()
|
self.setup()
|
||||||
|
@ -91,7 +93,7 @@ class EmaneClient:
|
||||||
# get mac config
|
# get mac config
|
||||||
mac_id, _, emane_model = components[MAC_COMPONENT_INDEX]
|
mac_id, _, emane_model = components[MAC_COMPONENT_INDEX]
|
||||||
mac_config = self.client.getConfiguration(mac_id)
|
mac_config = self.client.getConfiguration(mac_id)
|
||||||
logging.debug(
|
logger.debug(
|
||||||
"address(%s) nem(%s) emane(%s)", self.address, nem_id, emane_model
|
"address(%s) nem(%s) emane(%s)", self.address, nem_id, emane_model
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -101,9 +103,9 @@ class EmaneClient:
|
||||||
elif emane_model == EMANE_RFPIPE:
|
elif emane_model == EMANE_RFPIPE:
|
||||||
loss_table = self.handle_rfpipe(mac_config)
|
loss_table = self.handle_rfpipe(mac_config)
|
||||||
else:
|
else:
|
||||||
logging.warning("unknown emane link model: %s", emane_model)
|
logger.warning("unknown emane link model: %s", emane_model)
|
||||||
continue
|
continue
|
||||||
logging.info("monitoring links nem(%s) model(%s)", nem_id, emane_model)
|
logger.info("monitoring links nem(%s) model(%s)", nem_id, emane_model)
|
||||||
loss_table.mac_id = mac_id
|
loss_table.mac_id = mac_id
|
||||||
self.nems[nem_id] = loss_table
|
self.nems[nem_id] = loss_table
|
||||||
|
|
||||||
|
@ -138,12 +140,12 @@ class EmaneClient:
|
||||||
|
|
||||||
def handle_tdma(self, config: Dict[str, Tuple]):
|
def handle_tdma(self, config: Dict[str, Tuple]):
|
||||||
pcr = config["pcrcurveuri"][0][0]
|
pcr = config["pcrcurveuri"][0][0]
|
||||||
logging.debug("tdma pcr: %s", pcr)
|
logger.debug("tdma pcr: %s", pcr)
|
||||||
|
|
||||||
def handle_80211(self, config: Dict[str, Tuple]) -> LossTable:
|
def handle_80211(self, config: Dict[str, Tuple]) -> LossTable:
|
||||||
unicastrate = config["unicastrate"][0][0]
|
unicastrate = config["unicastrate"][0][0]
|
||||||
pcr = config["pcrcurveuri"][0][0]
|
pcr = config["pcrcurveuri"][0][0]
|
||||||
logging.debug("80211 pcr: %s", pcr)
|
logger.debug("80211 pcr: %s", pcr)
|
||||||
tree = etree.parse(pcr)
|
tree = etree.parse(pcr)
|
||||||
root = tree.getroot()
|
root = tree.getroot()
|
||||||
table = root.find("table")
|
table = root.find("table")
|
||||||
|
@ -159,7 +161,7 @@ class EmaneClient:
|
||||||
|
|
||||||
def handle_rfpipe(self, config: Dict[str, Tuple]) -> LossTable:
|
def handle_rfpipe(self, config: Dict[str, Tuple]) -> LossTable:
|
||||||
pcr = config["pcrcurveuri"][0][0]
|
pcr = config["pcrcurveuri"][0][0]
|
||||||
logging.debug("rfpipe pcr: %s", pcr)
|
logger.debug("rfpipe pcr: %s", pcr)
|
||||||
tree = etree.parse(pcr)
|
tree = etree.parse(pcr)
|
||||||
root = tree.getroot()
|
root = tree.getroot()
|
||||||
table = root.find("table")
|
table = root.find("table")
|
||||||
|
@ -187,12 +189,13 @@ class EmaneLinkMonitor:
|
||||||
self.running: bool = False
|
self.running: bool = False
|
||||||
|
|
||||||
def start(self) -> None:
|
def start(self) -> None:
|
||||||
self.loss_threshold = int(self.emane_manager.get_config("loss_threshold"))
|
options = self.emane_manager.session.options
|
||||||
self.link_interval = int(self.emane_manager.get_config("link_interval"))
|
self.loss_threshold = options.get_config_int("loss_threshold")
|
||||||
self.link_timeout = int(self.emane_manager.get_config("link_timeout"))
|
self.link_interval = options.get_config_int("link_interval")
|
||||||
|
self.link_timeout = options.get_config_int("link_timeout")
|
||||||
self.initialize()
|
self.initialize()
|
||||||
if not self.clients:
|
if not self.clients:
|
||||||
logging.info("no valid emane models to monitor links")
|
logger.info("no valid emane models to monitor links")
|
||||||
return
|
return
|
||||||
self.scheduler = sched.scheduler()
|
self.scheduler = sched.scheduler()
|
||||||
self.scheduler.enter(0, 0, self.check_links)
|
self.scheduler.enter(0, 0, self.check_links)
|
||||||
|
@ -202,22 +205,28 @@ class EmaneLinkMonitor:
|
||||||
|
|
||||||
def initialize(self) -> None:
|
def initialize(self) -> None:
|
||||||
addresses = self.get_addresses()
|
addresses = self.get_addresses()
|
||||||
for address in addresses:
|
for address, port in addresses:
|
||||||
client = EmaneClient(address)
|
client = EmaneClient(address, port)
|
||||||
if client.nems:
|
if client.nems:
|
||||||
self.clients.append(client)
|
self.clients.append(client)
|
||||||
|
|
||||||
def get_addresses(self) -> List[str]:
|
def get_addresses(self) -> List[Tuple[str, int]]:
|
||||||
addresses = []
|
addresses = []
|
||||||
nodes = self.emane_manager.getnodes()
|
nodes = self.emane_manager.getnodes()
|
||||||
for node in nodes:
|
for node in nodes:
|
||||||
|
control = None
|
||||||
|
ports = []
|
||||||
for iface in node.get_ifaces():
|
for iface in node.get_ifaces():
|
||||||
if isinstance(iface.net, CtrlNet):
|
if isinstance(iface.net, CtrlNet):
|
||||||
ip4 = iface.get_ip4()
|
ip4 = iface.get_ip4()
|
||||||
if ip4:
|
if ip4:
|
||||||
address = str(ip4.ip)
|
control = str(ip4.ip)
|
||||||
addresses.append(address)
|
if isinstance(iface.net, EmaneNet):
|
||||||
break
|
port = self.emane_manager.get_nem_port(iface)
|
||||||
|
ports.append(port)
|
||||||
|
if control:
|
||||||
|
for port in ports:
|
||||||
|
addresses.append((control, port))
|
||||||
return addresses
|
return addresses
|
||||||
|
|
||||||
def check_links(self) -> None:
|
def check_links(self) -> None:
|
||||||
|
@ -228,7 +237,7 @@ class EmaneLinkMonitor:
|
||||||
client.check_links(self.links, self.loss_threshold)
|
client.check_links(self.links, self.loss_threshold)
|
||||||
except shell.ControlPortException:
|
except shell.ControlPortException:
|
||||||
if self.running:
|
if self.running:
|
||||||
logging.exception("link monitor error")
|
logger.exception("link monitor error")
|
||||||
|
|
||||||
# find new links
|
# find new links
|
||||||
current_links = set(self.links.keys())
|
current_links = set(self.links.keys())
|
||||||
|
|
70
daemon/core/emane/modelmanager.py
Normal file
70
daemon/core/emane/modelmanager.py
Normal file
|
@ -0,0 +1,70 @@
|
||||||
|
import logging
|
||||||
|
import pkgutil
|
||||||
|
from pathlib import Path
|
||||||
|
from typing import Dict, List, Type
|
||||||
|
|
||||||
|
from core import utils
|
||||||
|
from core.emane import models as emane_models
|
||||||
|
from core.emane.emanemodel import EmaneModel
|
||||||
|
from core.errors import CoreError
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
|
class EmaneModelManager:
|
||||||
|
models: Dict[str, Type[EmaneModel]] = {}
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def load_locals(cls, emane_prefix: Path) -> List[str]:
|
||||||
|
"""
|
||||||
|
Load local core emane models and make them available.
|
||||||
|
|
||||||
|
:param emane_prefix: installed emane prefix
|
||||||
|
:return: list of errors encountered loading emane models
|
||||||
|
"""
|
||||||
|
errors = []
|
||||||
|
for module_info in pkgutil.walk_packages(
|
||||||
|
emane_models.__path__, f"{emane_models.__name__}."
|
||||||
|
):
|
||||||
|
models = utils.load_module(module_info.name, EmaneModel)
|
||||||
|
for model in models:
|
||||||
|
logger.debug("loading emane model: %s", model.name)
|
||||||
|
try:
|
||||||
|
model.load(emane_prefix)
|
||||||
|
cls.models[model.name] = model
|
||||||
|
except CoreError as e:
|
||||||
|
errors.append(model.name)
|
||||||
|
logger.debug("not loading emane model(%s): %s", model.name, e)
|
||||||
|
return errors
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def load(cls, path: Path, emane_prefix: Path) -> List[str]:
|
||||||
|
"""
|
||||||
|
Search and load custom emane models and make them available.
|
||||||
|
|
||||||
|
:param path: path to search for custom emane models
|
||||||
|
:param emane_prefix: installed emane prefix
|
||||||
|
:return: list of errors encountered loading emane models
|
||||||
|
"""
|
||||||
|
subdirs = [x for x in path.iterdir() if x.is_dir()]
|
||||||
|
subdirs.append(path)
|
||||||
|
errors = []
|
||||||
|
for subdir in subdirs:
|
||||||
|
logger.debug("loading emane models from: %s", subdir)
|
||||||
|
models = utils.load_classes(subdir, EmaneModel)
|
||||||
|
for model in models:
|
||||||
|
logger.debug("loading emane model: %s", model.name)
|
||||||
|
try:
|
||||||
|
model.load(emane_prefix)
|
||||||
|
cls.models[model.name] = model
|
||||||
|
except CoreError as e:
|
||||||
|
errors.append(model.name)
|
||||||
|
logger.debug("not loading emane model(%s): %s", model.name, e)
|
||||||
|
return errors
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def get(cls, name: str) -> Type[EmaneModel]:
|
||||||
|
model = cls.models.get(name)
|
||||||
|
if model is None:
|
||||||
|
raise CoreError(f"emame model does not exist {name}")
|
||||||
|
return model
|
0
daemon/core/emane/models/__init__.py
Normal file
0
daemon/core/emane/models/__init__.py
Normal file
|
@ -1,6 +1,7 @@
|
||||||
"""
|
"""
|
||||||
EMANE Bypass model for CORE
|
EMANE Bypass model for CORE
|
||||||
"""
|
"""
|
||||||
|
from pathlib import Path
|
||||||
from typing import List, Set
|
from typing import List, Set
|
||||||
|
|
||||||
from core.config import Configuration
|
from core.config import Configuration
|
||||||
|
@ -18,8 +19,8 @@ class EmaneBypassModel(emanemodel.EmaneModel):
|
||||||
mac_library: str = "bypassmaclayer"
|
mac_library: str = "bypassmaclayer"
|
||||||
mac_config: List[Configuration] = [
|
mac_config: List[Configuration] = [
|
||||||
Configuration(
|
Configuration(
|
||||||
_id="none",
|
id="none",
|
||||||
_type=ConfigDataTypes.BOOL,
|
type=ConfigDataTypes.BOOL,
|
||||||
default="0",
|
default="0",
|
||||||
label="There are no parameters for the bypass model.",
|
label="There are no parameters for the bypass model.",
|
||||||
)
|
)
|
||||||
|
@ -30,6 +31,5 @@ class EmaneBypassModel(emanemodel.EmaneModel):
|
||||||
phy_config: List[Configuration] = []
|
phy_config: List[Configuration] = []
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def load(cls, emane_prefix: str) -> None:
|
def load(cls, emane_prefix: Path) -> None:
|
||||||
# ignore default logic
|
cls._load_platform_config(emane_prefix)
|
||||||
pass
|
|
|
@ -3,7 +3,7 @@ commeffect.py: EMANE CommEffect model for CORE
|
||||||
"""
|
"""
|
||||||
|
|
||||||
import logging
|
import logging
|
||||||
import os
|
from pathlib import Path
|
||||||
from typing import Dict, List
|
from typing import Dict, List
|
||||||
|
|
||||||
from lxml import etree
|
from lxml import etree
|
||||||
|
@ -14,6 +14,8 @@ from core.emulator.data import LinkOptions
|
||||||
from core.nodes.interface import CoreInterface
|
from core.nodes.interface import CoreInterface
|
||||||
from core.xml import emanexml
|
from core.xml import emanexml
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
from emane.events.commeffectevent import CommEffectEvent
|
from emane.events.commeffectevent import CommEffectEvent
|
||||||
except ImportError:
|
except ImportError:
|
||||||
|
@ -21,7 +23,7 @@ except ImportError:
|
||||||
from emanesh.events.commeffectevent import CommEffectEvent
|
from emanesh.events.commeffectevent import CommEffectEvent
|
||||||
except ImportError:
|
except ImportError:
|
||||||
CommEffectEvent = None
|
CommEffectEvent = None
|
||||||
logging.debug("compatible emane python bindings not installed")
|
logger.debug("compatible emane python bindings not installed")
|
||||||
|
|
||||||
|
|
||||||
def convert_none(x: float) -> int:
|
def convert_none(x: float) -> int:
|
||||||
|
@ -48,17 +50,26 @@ class EmaneCommEffectModel(emanemodel.EmaneModel):
|
||||||
external_config: List[Configuration] = []
|
external_config: List[Configuration] = []
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def load(cls, emane_prefix: str) -> None:
|
def load(cls, emane_prefix: Path) -> None:
|
||||||
shim_xml_path = os.path.join(emane_prefix, "share/emane/manifest", cls.shim_xml)
|
cls._load_platform_config(emane_prefix)
|
||||||
|
shim_xml_path = emane_prefix / "share/emane/manifest" / cls.shim_xml
|
||||||
cls.config_shim = emanemanifest.parse(shim_xml_path, cls.shim_defaults)
|
cls.config_shim = emanemanifest.parse(shim_xml_path, cls.shim_defaults)
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def configurations(cls) -> List[Configuration]:
|
def configurations(cls) -> List[Configuration]:
|
||||||
return cls.config_shim
|
return cls.platform_config + cls.config_shim
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def config_groups(cls) -> List[ConfigGroup]:
|
def config_groups(cls) -> List[ConfigGroup]:
|
||||||
return [ConfigGroup("CommEffect SHIM Parameters", 1, len(cls.configurations()))]
|
platform_len = len(cls.platform_config)
|
||||||
|
return [
|
||||||
|
ConfigGroup("Platform Parameters", 1, platform_len),
|
||||||
|
ConfigGroup(
|
||||||
|
"CommEffect SHIM Parameters",
|
||||||
|
platform_len + 1,
|
||||||
|
len(cls.configurations()),
|
||||||
|
),
|
||||||
|
]
|
||||||
|
|
||||||
def build_xml_files(self, config: Dict[str, str], iface: CoreInterface) -> None:
|
def build_xml_files(self, config: Dict[str, str], iface: CoreInterface) -> None:
|
||||||
"""
|
"""
|
||||||
|
@ -111,21 +122,15 @@ class EmaneCommEffectModel(emanemodel.EmaneModel):
|
||||||
Generate CommEffect events when a Link Message is received having
|
Generate CommEffect events when a Link Message is received having
|
||||||
link parameters.
|
link parameters.
|
||||||
"""
|
"""
|
||||||
service = self.session.emane.service
|
|
||||||
if service is None:
|
|
||||||
logging.warning("%s: EMANE event service unavailable", self.name)
|
|
||||||
return
|
|
||||||
|
|
||||||
if iface is None or iface2 is None:
|
if iface is None or iface2 is None:
|
||||||
logging.warning("%s: missing NEM information", self.name)
|
logger.warning("%s: missing NEM information", self.name)
|
||||||
return
|
return
|
||||||
|
|
||||||
# TODO: batch these into multiple events per transmission
|
# TODO: batch these into multiple events per transmission
|
||||||
# TODO: may want to split out seconds portion of delay and jitter
|
# TODO: may want to split out seconds portion of delay and jitter
|
||||||
event = CommEffectEvent()
|
event = CommEffectEvent()
|
||||||
nem1 = self.session.emane.get_nem_id(iface)
|
nem1 = self.session.emane.get_nem_id(iface)
|
||||||
nem2 = self.session.emane.get_nem_id(iface2)
|
nem2 = self.session.emane.get_nem_id(iface2)
|
||||||
logging.info("sending comm effect event")
|
logger.info("sending comm effect event")
|
||||||
event.append(
|
event.append(
|
||||||
nem1,
|
nem1,
|
||||||
latency=convert_none(options.delay),
|
latency=convert_none(options.delay),
|
||||||
|
@ -135,4 +140,4 @@ class EmaneCommEffectModel(emanemodel.EmaneModel):
|
||||||
unicast=int(convert_none(options.bandwidth)),
|
unicast=int(convert_none(options.bandwidth)),
|
||||||
broadcast=int(convert_none(options.bandwidth)),
|
broadcast=int(convert_none(options.bandwidth)),
|
||||||
)
|
)
|
||||||
service.publish(nem2, event)
|
self.session.emane.publish_event(nem2, event)
|
|
@ -1,7 +1,7 @@
|
||||||
"""
|
"""
|
||||||
ieee80211abg.py: EMANE IEEE 802.11abg model for CORE
|
ieee80211abg.py: EMANE IEEE 802.11abg model for CORE
|
||||||
"""
|
"""
|
||||||
import os
|
from pathlib import Path
|
||||||
|
|
||||||
from core.emane import emanemodel
|
from core.emane import emanemodel
|
||||||
|
|
||||||
|
@ -15,8 +15,8 @@ class EmaneIeee80211abgModel(emanemodel.EmaneModel):
|
||||||
mac_xml: str = "ieee80211abgmaclayer.xml"
|
mac_xml: str = "ieee80211abgmaclayer.xml"
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def load(cls, emane_prefix: str) -> None:
|
def load(cls, emane_prefix: Path) -> None:
|
||||||
cls.mac_defaults["pcrcurveuri"] = os.path.join(
|
cls.mac_defaults["pcrcurveuri"] = str(
|
||||||
emane_prefix, "share/emane/xml/models/mac/ieee80211abg/ieee80211pcr.xml"
|
emane_prefix / "share/emane/xml/models/mac/ieee80211abg/ieee80211pcr.xml"
|
||||||
)
|
)
|
||||||
super().load(emane_prefix)
|
super().load(emane_prefix)
|
|
@ -1,7 +1,7 @@
|
||||||
"""
|
"""
|
||||||
rfpipe.py: EMANE RF-PIPE model for CORE
|
rfpipe.py: EMANE RF-PIPE model for CORE
|
||||||
"""
|
"""
|
||||||
import os
|
from pathlib import Path
|
||||||
|
|
||||||
from core.emane import emanemodel
|
from core.emane import emanemodel
|
||||||
|
|
||||||
|
@ -15,8 +15,8 @@ class EmaneRfPipeModel(emanemodel.EmaneModel):
|
||||||
mac_xml: str = "rfpipemaclayer.xml"
|
mac_xml: str = "rfpipemaclayer.xml"
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def load(cls, emane_prefix: str) -> None:
|
def load(cls, emane_prefix: Path) -> None:
|
||||||
cls.mac_defaults["pcrcurveuri"] = os.path.join(
|
cls.mac_defaults["pcrcurveuri"] = str(
|
||||||
emane_prefix, "share/emane/xml/models/mac/rfpipe/rfpipepcr.xml"
|
emane_prefix / "share/emane/xml/models/mac/rfpipe/rfpipepcr.xml"
|
||||||
)
|
)
|
||||||
super().load(emane_prefix)
|
super().load(emane_prefix)
|
68
daemon/core/emane/models/tdma.py
Normal file
68
daemon/core/emane/models/tdma.py
Normal file
|
@ -0,0 +1,68 @@
|
||||||
|
"""
|
||||||
|
tdma.py: EMANE TDMA model bindings for CORE
|
||||||
|
"""
|
||||||
|
|
||||||
|
import logging
|
||||||
|
from pathlib import Path
|
||||||
|
from typing import Set
|
||||||
|
|
||||||
|
from core import constants, utils
|
||||||
|
from core.config import Configuration
|
||||||
|
from core.emane import emanemodel
|
||||||
|
from core.emane.nodes import EmaneNet
|
||||||
|
from core.emulator.enumerations import ConfigDataTypes
|
||||||
|
from core.nodes.interface import CoreInterface
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
|
class EmaneTdmaModel(emanemodel.EmaneModel):
|
||||||
|
# model name
|
||||||
|
name: str = "emane_tdma"
|
||||||
|
|
||||||
|
# mac configuration
|
||||||
|
mac_library: str = "tdmaeventschedulerradiomodel"
|
||||||
|
mac_xml: str = "tdmaeventschedulerradiomodel.xml"
|
||||||
|
|
||||||
|
# add custom schedule options and ignore it when writing emane xml
|
||||||
|
schedule_name: str = "schedule"
|
||||||
|
default_schedule: Path = (
|
||||||
|
constants.CORE_DATA_DIR / "examples" / "tdma" / "schedule.xml"
|
||||||
|
)
|
||||||
|
config_ignore: Set[str] = {schedule_name}
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def load(cls, emane_prefix: Path) -> None:
|
||||||
|
cls.mac_defaults["pcrcurveuri"] = str(
|
||||||
|
emane_prefix
|
||||||
|
/ "share/emane/xml/models/mac/tdmaeventscheduler/tdmabasemodelpcr.xml"
|
||||||
|
)
|
||||||
|
super().load(emane_prefix)
|
||||||
|
config_item = Configuration(
|
||||||
|
id=cls.schedule_name,
|
||||||
|
type=ConfigDataTypes.STRING,
|
||||||
|
default=str(cls.default_schedule),
|
||||||
|
label="TDMA schedule file (core)",
|
||||||
|
)
|
||||||
|
cls.mac_config.insert(0, config_item)
|
||||||
|
|
||||||
|
def post_startup(self, iface: CoreInterface) -> None:
|
||||||
|
# get configured schedule
|
||||||
|
emane_net = self.session.get_node(self.id, EmaneNet)
|
||||||
|
config = self.session.emane.get_iface_config(emane_net, iface)
|
||||||
|
schedule = Path(config[self.schedule_name])
|
||||||
|
if not schedule.is_file():
|
||||||
|
logger.error("ignoring invalid tdma schedule: %s", schedule)
|
||||||
|
return
|
||||||
|
# initiate tdma schedule
|
||||||
|
nem_id = self.session.emane.get_nem_id(iface)
|
||||||
|
if not nem_id:
|
||||||
|
logger.error("could not find nem for interface")
|
||||||
|
return
|
||||||
|
service = self.session.emane.nem_service.get(nem_id)
|
||||||
|
if service:
|
||||||
|
device = service.device
|
||||||
|
logger.info(
|
||||||
|
"setting up tdma schedule: schedule(%s) device(%s)", schedule, device
|
||||||
|
)
|
||||||
|
utils.cmd(f"emaneevent-tdmaschedule -i {device} {schedule}")
|
|
@ -4,7 +4,7 @@ share the same MAC+PHY model.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
import logging
|
import logging
|
||||||
from typing import TYPE_CHECKING, Dict, List, Optional, Tuple, Type
|
from typing import TYPE_CHECKING, Dict, List, Optional, Type
|
||||||
|
|
||||||
from core.emulator.data import InterfaceData, LinkData, LinkOptions
|
from core.emulator.data import InterfaceData, LinkData, LinkOptions
|
||||||
from core.emulator.distributed import DistributedServer
|
from core.emulator.distributed import DistributedServer
|
||||||
|
@ -19,6 +19,8 @@ from core.errors import CoreError
|
||||||
from core.nodes.base import CoreNetworkBase, CoreNode
|
from core.nodes.base import CoreNetworkBase, CoreNode
|
||||||
from core.nodes.interface import CoreInterface
|
from core.nodes.interface import CoreInterface
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.emane.emanemodel import EmaneModel
|
from core.emane.emanemodel import EmaneModel
|
||||||
from core.emulator.session import Session
|
from core.emulator.session import Session
|
||||||
|
@ -34,7 +36,7 @@ except ImportError:
|
||||||
from emanesh.events import LocationEvent
|
from emanesh.events import LocationEvent
|
||||||
except ImportError:
|
except ImportError:
|
||||||
LocationEvent = None
|
LocationEvent = None
|
||||||
logging.debug("compatible emane python bindings not installed")
|
logger.debug("compatible emane python bindings not installed")
|
||||||
|
|
||||||
|
|
||||||
class EmaneNet(CoreNetworkBase):
|
class EmaneNet(CoreNetworkBase):
|
||||||
|
@ -92,9 +94,7 @@ class EmaneNet(CoreNetworkBase):
|
||||||
def updatemodel(self, config: Dict[str, str]) -> None:
|
def updatemodel(self, config: Dict[str, str]) -> None:
|
||||||
if not self.model:
|
if not self.model:
|
||||||
raise CoreError(f"no model set to update for node({self.name})")
|
raise CoreError(f"no model set to update for node({self.name})")
|
||||||
logging.info(
|
logger.info("node(%s) updating model(%s): %s", self.id, self.model.name, config)
|
||||||
"node(%s) updating model(%s): %s", self.id, self.model.name, config
|
|
||||||
)
|
|
||||||
self.model.update_config(config)
|
self.model.update_config(config)
|
||||||
|
|
||||||
def setmodel(self, model: "WirelessModelType", config: Dict[str, str]) -> None:
|
def setmodel(self, model: "WirelessModelType", config: Dict[str, str]) -> None:
|
||||||
|
@ -110,67 +110,6 @@ class EmaneNet(CoreNetworkBase):
|
||||||
self.mobility = model(session=self.session, _id=self.id)
|
self.mobility = model(session=self.session, _id=self.id)
|
||||||
self.mobility.update_config(config)
|
self.mobility.update_config(config)
|
||||||
|
|
||||||
def _nem_position(
|
|
||||||
self, iface: CoreInterface
|
|
||||||
) -> Optional[Tuple[int, float, float, float]]:
|
|
||||||
"""
|
|
||||||
Creates nem position for emane event for a given interface.
|
|
||||||
|
|
||||||
:param iface: interface to get nem emane position for
|
|
||||||
:return: nem position tuple, None otherwise
|
|
||||||
"""
|
|
||||||
nem_id = self.session.emane.get_nem_id(iface)
|
|
||||||
ifname = iface.localname
|
|
||||||
if nem_id is None:
|
|
||||||
logging.info("nemid for %s is unknown", ifname)
|
|
||||||
return
|
|
||||||
node = iface.node
|
|
||||||
x, y, z = node.getposition()
|
|
||||||
lat, lon, alt = self.session.location.getgeo(x, y, z)
|
|
||||||
if node.position.alt is not None:
|
|
||||||
alt = node.position.alt
|
|
||||||
node.position.set_geo(lon, lat, alt)
|
|
||||||
# altitude must be an integer or warning is printed
|
|
||||||
alt = int(round(alt))
|
|
||||||
return nem_id, lon, lat, alt
|
|
||||||
|
|
||||||
def setnemposition(self, iface: CoreInterface) -> None:
|
|
||||||
"""
|
|
||||||
Publish a NEM location change event using the EMANE event service.
|
|
||||||
|
|
||||||
:param iface: interface to set nem position for
|
|
||||||
"""
|
|
||||||
if self.session.emane.service is None:
|
|
||||||
logging.info("position service not available")
|
|
||||||
return
|
|
||||||
position = self._nem_position(iface)
|
|
||||||
if position:
|
|
||||||
nemid, lon, lat, alt = position
|
|
||||||
event = LocationEvent()
|
|
||||||
event.append(nemid, latitude=lat, longitude=lon, altitude=alt)
|
|
||||||
self.session.emane.service.publish(0, event)
|
|
||||||
|
|
||||||
def setnempositions(self, moved_ifaces: List[CoreInterface]) -> None:
|
|
||||||
"""
|
|
||||||
Several NEMs have moved, from e.g. a WaypointMobilityModel
|
|
||||||
calculation. Generate an EMANE Location Event having several
|
|
||||||
entries for each interface that has moved.
|
|
||||||
"""
|
|
||||||
if len(moved_ifaces) == 0:
|
|
||||||
return
|
|
||||||
|
|
||||||
if self.session.emane.service is None:
|
|
||||||
logging.info("position service not available")
|
|
||||||
return
|
|
||||||
|
|
||||||
event = LocationEvent()
|
|
||||||
for iface in moved_ifaces:
|
|
||||||
position = self._nem_position(iface)
|
|
||||||
if position:
|
|
||||||
nemid, lon, lat, alt = position
|
|
||||||
event.append(nemid, latitude=lat, longitude=lon, altitude=alt)
|
|
||||||
self.session.emane.service.publish(0, event)
|
|
||||||
|
|
||||||
def links(self, flags: MessageFlags = MessageFlags.NONE) -> List[LinkData]:
|
def links(self, flags: MessageFlags = MessageFlags.NONE) -> List[LinkData]:
|
||||||
links = super().links(flags)
|
links = super().links(flags)
|
||||||
emane_manager = self.session.emane
|
emane_manager = self.session.emane
|
||||||
|
|
|
@ -1,67 +0,0 @@
|
||||||
"""
|
|
||||||
tdma.py: EMANE TDMA model bindings for CORE
|
|
||||||
"""
|
|
||||||
|
|
||||||
import logging
|
|
||||||
import os
|
|
||||||
from typing import Set
|
|
||||||
|
|
||||||
from core import constants, utils
|
|
||||||
from core.config import Configuration
|
|
||||||
from core.emane import emanemodel
|
|
||||||
from core.emulator.enumerations import ConfigDataTypes
|
|
||||||
|
|
||||||
|
|
||||||
class EmaneTdmaModel(emanemodel.EmaneModel):
|
|
||||||
# model name
|
|
||||||
name: str = "emane_tdma"
|
|
||||||
|
|
||||||
# mac configuration
|
|
||||||
mac_library: str = "tdmaeventschedulerradiomodel"
|
|
||||||
mac_xml: str = "tdmaeventschedulerradiomodel.xml"
|
|
||||||
|
|
||||||
# add custom schedule options and ignore it when writing emane xml
|
|
||||||
schedule_name: str = "schedule"
|
|
||||||
default_schedule: str = os.path.join(
|
|
||||||
constants.CORE_DATA_DIR, "examples", "tdma", "schedule.xml"
|
|
||||||
)
|
|
||||||
config_ignore: Set[str] = {schedule_name}
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def load(cls, emane_prefix: str) -> None:
|
|
||||||
cls.mac_defaults["pcrcurveuri"] = os.path.join(
|
|
||||||
emane_prefix,
|
|
||||||
"share/emane/xml/models/mac/tdmaeventscheduler/tdmabasemodelpcr.xml",
|
|
||||||
)
|
|
||||||
super().load(emane_prefix)
|
|
||||||
cls.mac_config.insert(
|
|
||||||
0,
|
|
||||||
Configuration(
|
|
||||||
_id=cls.schedule_name,
|
|
||||||
_type=ConfigDataTypes.STRING,
|
|
||||||
default=cls.default_schedule,
|
|
||||||
label="TDMA schedule file (core)",
|
|
||||||
),
|
|
||||||
)
|
|
||||||
|
|
||||||
def post_startup(self) -> None:
|
|
||||||
"""
|
|
||||||
Logic to execute after the emane manager is finished with startup.
|
|
||||||
|
|
||||||
:return: nothing
|
|
||||||
"""
|
|
||||||
# get configured schedule
|
|
||||||
config = self.session.emane.get_configs(node_id=self.id, config_type=self.name)
|
|
||||||
if not config:
|
|
||||||
return
|
|
||||||
schedule = config[self.schedule_name]
|
|
||||||
|
|
||||||
# get the set event device
|
|
||||||
event_device = self.session.emane.event_device
|
|
||||||
|
|
||||||
# initiate tdma schedule
|
|
||||||
logging.info(
|
|
||||||
"setting up tdma schedule: schedule(%s) device(%s)", schedule, event_device
|
|
||||||
)
|
|
||||||
args = f"emaneevent-tdmaschedule -i {event_device} {schedule}"
|
|
||||||
utils.cmd(args)
|
|
|
@ -3,15 +3,21 @@ import logging
|
||||||
import os
|
import os
|
||||||
import signal
|
import signal
|
||||||
import sys
|
import sys
|
||||||
|
from pathlib import Path
|
||||||
from typing import Dict, List, Type
|
from typing import Dict, List, Type
|
||||||
|
|
||||||
import core.services
|
import core.services
|
||||||
from core import configservices, utils
|
from core import utils
|
||||||
from core.configservice.manager import ConfigServiceManager
|
from core.configservice.manager import ConfigServiceManager
|
||||||
|
from core.emane.modelmanager import EmaneModelManager
|
||||||
from core.emulator.session import Session
|
from core.emulator.session import Session
|
||||||
from core.executables import get_requirements
|
from core.executables import get_requirements
|
||||||
from core.services.coreservices import ServiceManager
|
from core.services.coreservices import ServiceManager
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
DEFAULT_EMANE_PREFIX: str = "/usr"
|
||||||
|
|
||||||
|
|
||||||
def signal_handler(signal_number: int, _) -> None:
|
def signal_handler(signal_number: int, _) -> None:
|
||||||
"""
|
"""
|
||||||
|
@ -21,7 +27,7 @@ def signal_handler(signal_number: int, _) -> None:
|
||||||
:param _: ignored
|
:param _: ignored
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
logging.info("caught signal: %s", signal_number)
|
logger.info("caught signal: %s", signal_number)
|
||||||
sys.exit(signal_number)
|
sys.exit(signal_number)
|
||||||
|
|
||||||
|
|
||||||
|
@ -47,8 +53,7 @@ class CoreEmu:
|
||||||
os.umask(0)
|
os.umask(0)
|
||||||
|
|
||||||
# configuration
|
# configuration
|
||||||
if config is None:
|
config = config if config else {}
|
||||||
config = {}
|
|
||||||
self.config: Dict[str, str] = config
|
self.config: Dict[str, str] = config
|
||||||
|
|
||||||
# session management
|
# session management
|
||||||
|
@ -56,15 +61,12 @@ class CoreEmu:
|
||||||
|
|
||||||
# load services
|
# load services
|
||||||
self.service_errors: List[str] = []
|
self.service_errors: List[str] = []
|
||||||
self.load_services()
|
|
||||||
|
|
||||||
# config services
|
|
||||||
self.service_manager: ConfigServiceManager = ConfigServiceManager()
|
self.service_manager: ConfigServiceManager = ConfigServiceManager()
|
||||||
config_services_path = os.path.abspath(os.path.dirname(configservices.__file__))
|
self._load_services()
|
||||||
self.service_manager.load(config_services_path)
|
|
||||||
custom_dir = self.config.get("custom_config_services_dir")
|
# check and load emane
|
||||||
if custom_dir:
|
self.has_emane: bool = False
|
||||||
self.service_manager.load(custom_dir)
|
self._load_emane()
|
||||||
|
|
||||||
# check executables exist on path
|
# check executables exist on path
|
||||||
self._validate_env()
|
self._validate_env()
|
||||||
|
@ -83,7 +85,7 @@ class CoreEmu:
|
||||||
for requirement in get_requirements(use_ovs):
|
for requirement in get_requirements(use_ovs):
|
||||||
utils.which(requirement, required=True)
|
utils.which(requirement, required=True)
|
||||||
|
|
||||||
def load_services(self) -> None:
|
def _load_services(self) -> None:
|
||||||
"""
|
"""
|
||||||
Loads default and custom services for use within CORE.
|
Loads default and custom services for use within CORE.
|
||||||
|
|
||||||
|
@ -91,15 +93,46 @@ class CoreEmu:
|
||||||
"""
|
"""
|
||||||
# load default services
|
# load default services
|
||||||
self.service_errors = core.services.load()
|
self.service_errors = core.services.load()
|
||||||
|
|
||||||
# load custom services
|
# load custom services
|
||||||
service_paths = self.config.get("custom_services_dir")
|
service_paths = self.config.get("custom_services_dir")
|
||||||
logging.debug("custom service paths: %s", service_paths)
|
logger.debug("custom service paths: %s", service_paths)
|
||||||
if service_paths:
|
if service_paths is not None:
|
||||||
for service_path in service_paths.split(","):
|
for service_path in service_paths.split(","):
|
||||||
service_path = service_path.strip()
|
service_path = Path(service_path.strip())
|
||||||
custom_service_errors = ServiceManager.add_services(service_path)
|
custom_service_errors = ServiceManager.add_services(service_path)
|
||||||
self.service_errors.extend(custom_service_errors)
|
self.service_errors.extend(custom_service_errors)
|
||||||
|
# load default config services
|
||||||
|
self.service_manager.load_locals()
|
||||||
|
# load custom config services
|
||||||
|
custom_dir = self.config.get("custom_config_services_dir")
|
||||||
|
if custom_dir is not None:
|
||||||
|
custom_dir = Path(custom_dir)
|
||||||
|
self.service_manager.load(custom_dir)
|
||||||
|
|
||||||
|
def _load_emane(self) -> None:
|
||||||
|
"""
|
||||||
|
Check if emane is installed and load models.
|
||||||
|
|
||||||
|
:return: nothing
|
||||||
|
"""
|
||||||
|
# check for emane
|
||||||
|
path = utils.which("emane", required=False)
|
||||||
|
self.has_emane = path is not None
|
||||||
|
if not self.has_emane:
|
||||||
|
logger.info("emane is not installed, emane functionality disabled")
|
||||||
|
return
|
||||||
|
# get version
|
||||||
|
emane_version = utils.cmd("emane --version")
|
||||||
|
logger.info("using emane: %s", emane_version)
|
||||||
|
emane_prefix = self.config.get("emane_prefix", DEFAULT_EMANE_PREFIX)
|
||||||
|
emane_prefix = Path(emane_prefix)
|
||||||
|
EmaneModelManager.load_locals(emane_prefix)
|
||||||
|
# load custom models
|
||||||
|
custom_path = self.config.get("emane_models_dir")
|
||||||
|
if custom_path is not None:
|
||||||
|
logger.info("loading custom emane models: %s", custom_path)
|
||||||
|
custom_path = Path(custom_path)
|
||||||
|
EmaneModelManager.load(custom_path, emane_prefix)
|
||||||
|
|
||||||
def shutdown(self) -> None:
|
def shutdown(self) -> None:
|
||||||
"""
|
"""
|
||||||
|
@ -107,7 +140,7 @@ class CoreEmu:
|
||||||
|
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
logging.info("shutting down all sessions")
|
logger.info("shutting down all sessions")
|
||||||
sessions = self.sessions.copy()
|
sessions = self.sessions.copy()
|
||||||
self.sessions.clear()
|
self.sessions.clear()
|
||||||
for _id in sessions:
|
for _id in sessions:
|
||||||
|
@ -128,7 +161,7 @@ class CoreEmu:
|
||||||
_id += 1
|
_id += 1
|
||||||
session = _cls(_id, config=self.config)
|
session = _cls(_id, config=self.config)
|
||||||
session.service_manager = self.service_manager
|
session.service_manager = self.service_manager
|
||||||
logging.info("created session: %s", _id)
|
logger.info("created session: %s", _id)
|
||||||
self.sessions[_id] = session
|
self.sessions[_id] = session
|
||||||
return session
|
return session
|
||||||
|
|
||||||
|
@ -139,14 +172,14 @@ class CoreEmu:
|
||||||
:param _id: session id to delete
|
:param _id: session id to delete
|
||||||
:return: True if deleted, False otherwise
|
:return: True if deleted, False otherwise
|
||||||
"""
|
"""
|
||||||
logging.info("deleting session: %s", _id)
|
logger.info("deleting session: %s", _id)
|
||||||
session = self.sessions.pop(_id, None)
|
session = self.sessions.pop(_id, None)
|
||||||
result = False
|
result = False
|
||||||
if session:
|
if session:
|
||||||
logging.info("shutting session down: %s", _id)
|
logger.info("shutting session down: %s", _id)
|
||||||
session.data_collect()
|
session.data_collect()
|
||||||
session.shutdown()
|
session.shutdown()
|
||||||
result = True
|
result = True
|
||||||
else:
|
else:
|
||||||
logging.error("session to delete did not exist: %s", _id)
|
logger.error("session to delete did not exist: %s", _id)
|
||||||
return result
|
return result
|
||||||
|
|
|
@ -91,6 +91,7 @@ class NodeOptions:
|
||||||
server: str = None
|
server: str = None
|
||||||
image: str = None
|
image: str = None
|
||||||
emane: str = None
|
emane: str = None
|
||||||
|
legacy: bool = False
|
||||||
|
|
||||||
def set_position(self, x: float, y: float) -> None:
|
def set_position(self, x: float, y: float) -> None:
|
||||||
"""
|
"""
|
||||||
|
|
|
@ -6,6 +6,7 @@ import logging
|
||||||
import os
|
import os
|
||||||
import threading
|
import threading
|
||||||
from collections import OrderedDict
|
from collections import OrderedDict
|
||||||
|
from pathlib import Path
|
||||||
from tempfile import NamedTemporaryFile
|
from tempfile import NamedTemporaryFile
|
||||||
from typing import TYPE_CHECKING, Callable, Dict, Tuple
|
from typing import TYPE_CHECKING, Callable, Dict, Tuple
|
||||||
|
|
||||||
|
@ -19,6 +20,8 @@ from core.executables import get_requirements
|
||||||
from core.nodes.interface import GreTap
|
from core.nodes.interface import GreTap
|
||||||
from core.nodes.network import CoreNetwork, CtrlNet
|
from core.nodes.network import CoreNetwork, CtrlNet
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.emulator.session import Session
|
from core.emulator.session import Session
|
||||||
|
|
||||||
|
@ -61,7 +64,7 @@ class DistributedServer:
|
||||||
replace_env = env is not None
|
replace_env = env is not None
|
||||||
if not wait:
|
if not wait:
|
||||||
cmd += " &"
|
cmd += " &"
|
||||||
logging.debug(
|
logger.debug(
|
||||||
"remote cmd server(%s) cwd(%s) wait(%s): %s", self.host, cwd, wait, cmd
|
"remote cmd server(%s) cwd(%s) wait(%s): %s", self.host, cwd, wait, cmd
|
||||||
)
|
)
|
||||||
try:
|
try:
|
||||||
|
@ -79,23 +82,23 @@ class DistributedServer:
|
||||||
stdout, stderr = e.streams_for_display()
|
stdout, stderr = e.streams_for_display()
|
||||||
raise CoreCommandError(e.result.exited, cmd, stdout, stderr)
|
raise CoreCommandError(e.result.exited, cmd, stdout, stderr)
|
||||||
|
|
||||||
def remote_put(self, source: str, destination: str) -> None:
|
def remote_put(self, src_path: Path, dst_path: Path) -> None:
|
||||||
"""
|
"""
|
||||||
Push file to remote server.
|
Push file to remote server.
|
||||||
|
|
||||||
:param source: source file to push
|
:param src_path: source file to push
|
||||||
:param destination: destination file location
|
:param dst_path: destination file location
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
with self.lock:
|
with self.lock:
|
||||||
self.conn.put(source, destination)
|
self.conn.put(str(src_path), str(dst_path))
|
||||||
|
|
||||||
def remote_put_temp(self, destination: str, data: str) -> None:
|
def remote_put_temp(self, dst_path: Path, data: str) -> None:
|
||||||
"""
|
"""
|
||||||
Remote push file contents to a remote server, using a temp file as an
|
Remote push file contents to a remote server, using a temp file as an
|
||||||
intermediate step.
|
intermediate step.
|
||||||
|
|
||||||
:param destination: file destination for data
|
:param dst_path: file destination for data
|
||||||
:param data: data to store in remote file
|
:param data: data to store in remote file
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
|
@ -103,7 +106,7 @@ class DistributedServer:
|
||||||
temp = NamedTemporaryFile(delete=False)
|
temp = NamedTemporaryFile(delete=False)
|
||||||
temp.write(data.encode("utf-8"))
|
temp.write(data.encode("utf-8"))
|
||||||
temp.close()
|
temp.close()
|
||||||
self.conn.put(temp.name, destination)
|
self.conn.put(temp.name, str(dst_path))
|
||||||
os.unlink(temp.name)
|
os.unlink(temp.name)
|
||||||
|
|
||||||
|
|
||||||
|
@ -144,7 +147,7 @@ class DistributedController:
|
||||||
f"command({requirement})"
|
f"command({requirement})"
|
||||||
)
|
)
|
||||||
self.servers[name] = server
|
self.servers[name] = server
|
||||||
cmd = f"mkdir -p {self.session.session_dir}"
|
cmd = f"mkdir -p {self.session.directory}"
|
||||||
server.remote_cmd(cmd)
|
server.remote_cmd(cmd)
|
||||||
|
|
||||||
def execute(self, func: Callable[[DistributedServer], None]) -> None:
|
def execute(self, func: Callable[[DistributedServer], None]) -> None:
|
||||||
|
@ -170,13 +173,11 @@ class DistributedController:
|
||||||
tunnels = self.tunnels[key]
|
tunnels = self.tunnels[key]
|
||||||
for tunnel in tunnels:
|
for tunnel in tunnels:
|
||||||
tunnel.shutdown()
|
tunnel.shutdown()
|
||||||
|
|
||||||
# remove all remote session directories
|
# remove all remote session directories
|
||||||
for name in self.servers:
|
for name in self.servers:
|
||||||
server = self.servers[name]
|
server = self.servers[name]
|
||||||
cmd = f"rm -rf {self.session.session_dir}"
|
cmd = f"rm -rf {self.session.directory}"
|
||||||
server.remote_cmd(cmd)
|
server.remote_cmd(cmd)
|
||||||
|
|
||||||
# clear tunnels
|
# clear tunnels
|
||||||
self.tunnels.clear()
|
self.tunnels.clear()
|
||||||
|
|
||||||
|
@ -212,23 +213,18 @@ class DistributedController:
|
||||||
tunnel = self.tunnels.get(key)
|
tunnel = self.tunnels.get(key)
|
||||||
if tunnel is not None:
|
if tunnel is not None:
|
||||||
return tunnel
|
return tunnel
|
||||||
|
|
||||||
# local to server
|
# local to server
|
||||||
logging.info(
|
logger.info("local tunnel node(%s) to remote(%s) key(%s)", node.name, host, key)
|
||||||
"local tunnel node(%s) to remote(%s) key(%s)", node.name, host, key
|
|
||||||
)
|
|
||||||
local_tap = GreTap(session=self.session, remoteip=host, key=key)
|
local_tap = GreTap(session=self.session, remoteip=host, key=key)
|
||||||
local_tap.net_client.set_iface_master(node.brname, local_tap.localname)
|
local_tap.net_client.set_iface_master(node.brname, local_tap.localname)
|
||||||
|
|
||||||
# server to local
|
# server to local
|
||||||
logging.info(
|
logger.info(
|
||||||
"remote tunnel node(%s) to local(%s) key(%s)", node.name, self.address, key
|
"remote tunnel node(%s) to local(%s) key(%s)", node.name, self.address, key
|
||||||
)
|
)
|
||||||
remote_tap = GreTap(
|
remote_tap = GreTap(
|
||||||
session=self.session, remoteip=self.address, key=key, server=server
|
session=self.session, remoteip=self.address, key=key, server=server
|
||||||
)
|
)
|
||||||
remote_tap.net_client.set_iface_master(node.brname, remote_tap.localname)
|
remote_tap.net_client.set_iface_master(node.brname, remote_tap.localname)
|
||||||
|
|
||||||
# save tunnels for shutdown
|
# save tunnels for shutdown
|
||||||
tunnel = (local_tap, remote_tap)
|
tunnel = (local_tap, remote_tap)
|
||||||
self.tunnels[key] = tunnel
|
self.tunnels[key] = tunnel
|
||||||
|
@ -244,7 +240,7 @@ class DistributedController:
|
||||||
:param node2_id: node two id
|
:param node2_id: node two id
|
||||||
:return: tunnel key for the node pair
|
:return: tunnel key for the node pair
|
||||||
"""
|
"""
|
||||||
logging.debug("creating tunnel key for: %s, %s", node1_id, node2_id)
|
logger.debug("creating tunnel key for: %s, %s", node1_id, node2_id)
|
||||||
key = (
|
key = (
|
||||||
(self.session.id << 16)
|
(self.session.id << 16)
|
||||||
^ utils.hashkey(node1_id)
|
^ utils.hashkey(node1_id)
|
||||||
|
|
|
@ -4,6 +4,7 @@ that manages a CORE session.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
import logging
|
import logging
|
||||||
|
import math
|
||||||
import os
|
import os
|
||||||
import pwd
|
import pwd
|
||||||
import shutil
|
import shutil
|
||||||
|
@ -62,6 +63,8 @@ from core.services.coreservices import CoreServices
|
||||||
from core.xml import corexml, corexmldeployment
|
from core.xml import corexml, corexmldeployment
|
||||||
from core.xml.corexml import CoreXmlReader, CoreXmlWriter
|
from core.xml.corexml import CoreXmlReader, CoreXmlWriter
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
# maps for converting from API call node type values to classes and vice versa
|
# maps for converting from API call node type values to classes and vice versa
|
||||||
NODES: Dict[NodeTypes, Type[NodeBase]] = {
|
NODES: Dict[NodeTypes, Type[NodeBase]] = {
|
||||||
NodeTypes.DEFAULT: CoreNode,
|
NodeTypes.DEFAULT: CoreNode,
|
||||||
|
@ -103,13 +106,13 @@ class Session:
|
||||||
self.id: int = _id
|
self.id: int = _id
|
||||||
|
|
||||||
# define and create session directory when desired
|
# define and create session directory when desired
|
||||||
self.session_dir: str = os.path.join(tempfile.gettempdir(), f"pycore.{self.id}")
|
self.directory: Path = Path(tempfile.gettempdir()) / f"pycore.{self.id}"
|
||||||
if mkdir:
|
if mkdir:
|
||||||
os.mkdir(self.session_dir)
|
self.directory.mkdir()
|
||||||
|
|
||||||
self.name: Optional[str] = None
|
self.name: Optional[str] = None
|
||||||
self.file_name: Optional[str] = None
|
self.file_path: Optional[Path] = None
|
||||||
self.thumbnail: Optional[str] = None
|
self.thumbnail: Optional[Path] = None
|
||||||
self.user: Optional[str] = None
|
self.user: Optional[str] = None
|
||||||
self.event_loop: EventLoop = EventLoop()
|
self.event_loop: EventLoop = EventLoop()
|
||||||
self.link_colors: Dict[int, str] = {}
|
self.link_colors: Dict[int, str] = {}
|
||||||
|
@ -197,7 +200,7 @@ class Session:
|
||||||
:raises core.CoreError: when objects to link is less than 2, or no common
|
:raises core.CoreError: when objects to link is less than 2, or no common
|
||||||
networks are found
|
networks are found
|
||||||
"""
|
"""
|
||||||
logging.info(
|
logger.info(
|
||||||
"handling wireless linking node1(%s) node2(%s): %s",
|
"handling wireless linking node1(%s) node2(%s): %s",
|
||||||
node1.name,
|
node1.name,
|
||||||
node2.name,
|
node2.name,
|
||||||
|
@ -208,7 +211,7 @@ class Session:
|
||||||
raise CoreError("no common network found for wireless link/unlink")
|
raise CoreError("no common network found for wireless link/unlink")
|
||||||
for common_network, iface1, iface2 in common_networks:
|
for common_network, iface1, iface2 in common_networks:
|
||||||
if not isinstance(common_network, (WlanNode, EmaneNet)):
|
if not isinstance(common_network, (WlanNode, EmaneNet)):
|
||||||
logging.info(
|
logger.info(
|
||||||
"skipping common network that is not wireless/emane: %s",
|
"skipping common network that is not wireless/emane: %s",
|
||||||
common_network,
|
common_network,
|
||||||
)
|
)
|
||||||
|
@ -263,7 +266,7 @@ class Session:
|
||||||
else:
|
else:
|
||||||
# peer to peer link
|
# peer to peer link
|
||||||
if isinstance(node1, CoreNodeBase) and isinstance(node2, CoreNodeBase):
|
if isinstance(node1, CoreNodeBase) and isinstance(node2, CoreNodeBase):
|
||||||
logging.info("linking ptp: %s - %s", node1.name, node2.name)
|
logger.info("linking ptp: %s - %s", node1.name, node2.name)
|
||||||
start = self.state.should_start()
|
start = self.state.should_start()
|
||||||
ptp = self.create_node(PtpNet, start)
|
ptp = self.create_node(PtpNet, start)
|
||||||
iface1 = node1.new_iface(ptp, iface1_data)
|
iface1 = node1.new_iface(ptp, iface1_data)
|
||||||
|
@ -286,7 +289,7 @@ class Session:
|
||||||
elif isinstance(node1, CoreNetworkBase) and isinstance(
|
elif isinstance(node1, CoreNetworkBase) and isinstance(
|
||||||
node2, CoreNetworkBase
|
node2, CoreNetworkBase
|
||||||
):
|
):
|
||||||
logging.info(
|
logger.info(
|
||||||
"linking network to network: %s - %s", node1.name, node2.name
|
"linking network to network: %s - %s", node1.name, node2.name
|
||||||
)
|
)
|
||||||
iface1 = node1.linknet(node2)
|
iface1 = node1.linknet(node2)
|
||||||
|
@ -303,10 +306,10 @@ class Session:
|
||||||
# configure tunnel nodes
|
# configure tunnel nodes
|
||||||
key = options.key
|
key = options.key
|
||||||
if isinstance(node1, TunnelNode):
|
if isinstance(node1, TunnelNode):
|
||||||
logging.info("setting tunnel key for: %s", node1.name)
|
logger.info("setting tunnel key for: %s", node1.name)
|
||||||
node1.setkey(key, iface1_data)
|
node1.setkey(key, iface1_data)
|
||||||
if isinstance(node2, TunnelNode):
|
if isinstance(node2, TunnelNode):
|
||||||
logging.info("setting tunnel key for: %s", node2.name)
|
logger.info("setting tunnel key for: %s", node2.name)
|
||||||
node2.setkey(key, iface2_data)
|
node2.setkey(key, iface2_data)
|
||||||
self.sdt.add_link(node1_id, node2_id)
|
self.sdt.add_link(node1_id, node2_id)
|
||||||
return iface1, iface2
|
return iface1, iface2
|
||||||
|
@ -332,7 +335,7 @@ class Session:
|
||||||
"""
|
"""
|
||||||
node1 = self.get_node(node1_id, NodeBase)
|
node1 = self.get_node(node1_id, NodeBase)
|
||||||
node2 = self.get_node(node2_id, NodeBase)
|
node2 = self.get_node(node2_id, NodeBase)
|
||||||
logging.info(
|
logger.info(
|
||||||
"deleting link(%s) node(%s):interface(%s) node(%s):interface(%s)",
|
"deleting link(%s) node(%s):interface(%s) node(%s):interface(%s)",
|
||||||
link_type.name,
|
link_type.name,
|
||||||
node1.name,
|
node1.name,
|
||||||
|
@ -409,7 +412,7 @@ class Session:
|
||||||
options = LinkOptions()
|
options = LinkOptions()
|
||||||
node1 = self.get_node(node1_id, NodeBase)
|
node1 = self.get_node(node1_id, NodeBase)
|
||||||
node2 = self.get_node(node2_id, NodeBase)
|
node2 = self.get_node(node2_id, NodeBase)
|
||||||
logging.info(
|
logger.info(
|
||||||
"update link(%s) node(%s):interface(%s) node(%s):interface(%s)",
|
"update link(%s) node(%s):interface(%s) node(%s):interface(%s)",
|
||||||
link_type.name,
|
link_type.name,
|
||||||
node1.name,
|
node1.name,
|
||||||
|
@ -525,7 +528,7 @@ class Session:
|
||||||
raise CoreError(f"invalid distributed server: {options.server}")
|
raise CoreError(f"invalid distributed server: {options.server}")
|
||||||
|
|
||||||
# create node
|
# create node
|
||||||
logging.info(
|
logger.info(
|
||||||
"creating node(%s) id(%s) name(%s) start(%s)",
|
"creating node(%s) id(%s) name(%s) start(%s)",
|
||||||
_class.__name__,
|
_class.__name__,
|
||||||
_id,
|
_id,
|
||||||
|
@ -542,28 +545,32 @@ class Session:
|
||||||
node.canvas = options.canvas
|
node.canvas = options.canvas
|
||||||
|
|
||||||
# set node position and broadcast it
|
# set node position and broadcast it
|
||||||
self.set_node_position(node, options)
|
has_geo = all(i is not None for i in [options.lon, options.lat, options.alt])
|
||||||
|
if has_geo:
|
||||||
|
self.set_node_geo(node, options.lon, options.lat, options.alt)
|
||||||
|
else:
|
||||||
|
self.set_node_pos(node, options.x, options.y)
|
||||||
|
|
||||||
# add services to needed nodes
|
# add services to needed nodes
|
||||||
if isinstance(node, (CoreNode, PhysicalNode)):
|
if isinstance(node, (CoreNode, PhysicalNode)):
|
||||||
node.type = options.model
|
node.type = options.model
|
||||||
logging.debug("set node type: %s", node.type)
|
if options.legacy or options.services:
|
||||||
self.services.add_services(node, node.type, options.services)
|
logger.debug("set node type: %s", node.type)
|
||||||
|
self.services.add_services(node, node.type, options.services)
|
||||||
|
|
||||||
# add config services
|
# add config services
|
||||||
logging.info("setting node config services: %s", options.config_services)
|
config_services = options.config_services
|
||||||
for name in options.config_services:
|
if not options.legacy and not config_services and not node.services:
|
||||||
|
config_services = self.services.default_services.get(node.type, [])
|
||||||
|
logger.info("setting node config services: %s", config_services)
|
||||||
|
for name in config_services:
|
||||||
service_class = self.service_manager.get_service(name)
|
service_class = self.service_manager.get_service(name)
|
||||||
node.add_config_service(service_class)
|
node.add_config_service(service_class)
|
||||||
|
|
||||||
# ensure default emane configuration
|
# ensure default emane configuration
|
||||||
if isinstance(node, EmaneNet) and options.emane:
|
if isinstance(node, EmaneNet) and options.emane:
|
||||||
model = self.emane.models.get(options.emane)
|
model_class = self.emane.get_model(options.emane)
|
||||||
if not model:
|
node.model = model_class(self, node.id)
|
||||||
raise CoreError(
|
|
||||||
f"node({node.name}) emane model({options.emane}) does not exist"
|
|
||||||
)
|
|
||||||
node.model = model(self, node.id)
|
|
||||||
if self.state == EventTypes.RUNTIME_STATE:
|
if self.state == EventTypes.RUNTIME_STATE:
|
||||||
self.emane.add_node(node)
|
self.emane.add_node(node)
|
||||||
# set default wlan config if needed
|
# set default wlan config if needed
|
||||||
|
@ -575,51 +582,26 @@ class Session:
|
||||||
if self.state == EventTypes.RUNTIME_STATE and is_boot_node:
|
if self.state == EventTypes.RUNTIME_STATE and is_boot_node:
|
||||||
self.write_nodes()
|
self.write_nodes()
|
||||||
self.add_remove_control_iface(node, remove=False)
|
self.add_remove_control_iface(node, remove=False)
|
||||||
self.services.boot_services(node)
|
self.boot_node(node)
|
||||||
|
|
||||||
self.sdt.add_node(node)
|
self.sdt.add_node(node)
|
||||||
return node
|
return node
|
||||||
|
|
||||||
def edit_node(self, node_id: int, options: NodeOptions) -> None:
|
def set_node_pos(self, node: NodeBase, x: float, y: float) -> None:
|
||||||
"""
|
node.setposition(x, y, None)
|
||||||
Edit node information.
|
self.sdt.edit_node(
|
||||||
|
node, node.position.lon, node.position.lat, node.position.alt
|
||||||
|
)
|
||||||
|
|
||||||
:param node_id: id of node to update
|
def set_node_geo(self, node: NodeBase, lon: float, lat: float, alt: float) -> None:
|
||||||
:param options: data to update node with
|
x, y, _ = self.location.getxyz(lat, lon, alt)
|
||||||
:return: nothing
|
if math.isinf(x) or math.isinf(y):
|
||||||
:raises core.CoreError: when node to update does not exist
|
raise CoreError(
|
||||||
"""
|
f"invalid geo for current reference/scale: {lon},{lat},{alt}"
|
||||||
node = self.get_node(node_id, NodeBase)
|
)
|
||||||
node.icon = options.icon
|
node.setposition(x, y, None)
|
||||||
self.set_node_position(node, options)
|
node.position.set_geo(lon, lat, alt)
|
||||||
self.sdt.edit_node(node, options.lon, options.lat, options.alt)
|
self.sdt.edit_node(node, lon, lat, alt)
|
||||||
|
|
||||||
def set_node_position(self, node: NodeBase, options: NodeOptions) -> None:
|
|
||||||
"""
|
|
||||||
Set position for a node, use lat/lon/alt if needed.
|
|
||||||
|
|
||||||
:param node: node to set position for
|
|
||||||
:param options: data for node
|
|
||||||
:return: nothing
|
|
||||||
"""
|
|
||||||
# extract location values
|
|
||||||
x = options.x
|
|
||||||
y = options.y
|
|
||||||
lat = options.lat
|
|
||||||
lon = options.lon
|
|
||||||
alt = options.alt
|
|
||||||
|
|
||||||
# check if we need to generate position from lat/lon/alt
|
|
||||||
has_empty_position = all(i is None for i in [x, y])
|
|
||||||
has_lat_lon_alt = all(i is not None for i in [lat, lon, alt])
|
|
||||||
using_lat_lon_alt = has_empty_position and has_lat_lon_alt
|
|
||||||
if using_lat_lon_alt:
|
|
||||||
x, y, _ = self.location.getxyz(lat, lon, alt)
|
|
||||||
node.setposition(x, y, None)
|
|
||||||
node.position.set_geo(lon, lat, alt)
|
|
||||||
self.broadcast_node(node)
|
|
||||||
elif not has_empty_position:
|
|
||||||
node.setposition(x, y, None)
|
|
||||||
|
|
||||||
def start_mobility(self, node_ids: List[int] = None) -> None:
|
def start_mobility(self, node_ids: List[int] = None) -> None:
|
||||||
"""
|
"""
|
||||||
|
@ -638,45 +620,42 @@ class Session:
|
||||||
:return: True if active, False otherwise
|
:return: True if active, False otherwise
|
||||||
"""
|
"""
|
||||||
result = self.state in {EventTypes.RUNTIME_STATE, EventTypes.DATACOLLECT_STATE}
|
result = self.state in {EventTypes.RUNTIME_STATE, EventTypes.DATACOLLECT_STATE}
|
||||||
logging.info("session(%s) checking if active: %s", self.id, result)
|
logger.info("session(%s) checking if active: %s", self.id, result)
|
||||||
return result
|
return result
|
||||||
|
|
||||||
def open_xml(self, file_name: str, start: bool = False) -> None:
|
def open_xml(self, file_path: Path, start: bool = False) -> None:
|
||||||
"""
|
"""
|
||||||
Import a session from the EmulationScript XML format.
|
Import a session from the EmulationScript XML format.
|
||||||
|
|
||||||
:param file_name: xml file to load session from
|
:param file_path: xml file to load session from
|
||||||
:param start: instantiate session if true, false otherwise
|
:param start: instantiate session if true, false otherwise
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
logging.info("opening xml: %s", file_name)
|
logger.info("opening xml: %s", file_path)
|
||||||
|
|
||||||
# clear out existing session
|
# clear out existing session
|
||||||
self.clear()
|
self.clear()
|
||||||
|
|
||||||
# set state and read xml
|
# set state and read xml
|
||||||
state = EventTypes.CONFIGURATION_STATE if start else EventTypes.DEFINITION_STATE
|
state = EventTypes.CONFIGURATION_STATE if start else EventTypes.DEFINITION_STATE
|
||||||
self.set_state(state)
|
self.set_state(state)
|
||||||
self.name = os.path.basename(file_name)
|
self.name = file_path.name
|
||||||
self.file_name = file_name
|
self.file_path = file_path
|
||||||
CoreXmlReader(self).read(file_name)
|
CoreXmlReader(self).read(file_path)
|
||||||
|
|
||||||
# start session if needed
|
# start session if needed
|
||||||
if start:
|
if start:
|
||||||
self.set_state(EventTypes.INSTANTIATION_STATE)
|
self.set_state(EventTypes.INSTANTIATION_STATE)
|
||||||
self.instantiate()
|
self.instantiate()
|
||||||
|
|
||||||
def save_xml(self, file_name: str) -> None:
|
def save_xml(self, file_path: Path) -> None:
|
||||||
"""
|
"""
|
||||||
Export a session to the EmulationScript XML format.
|
Export a session to the EmulationScript XML format.
|
||||||
|
|
||||||
:param file_name: file name to write session xml to
|
:param file_path: file name to write session xml to
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
CoreXmlWriter(self).write(file_name)
|
CoreXmlWriter(self).write(file_path)
|
||||||
|
|
||||||
def add_hook(
|
def add_hook(
|
||||||
self, state: EventTypes, file_name: str, data: str, source_name: str = None
|
self, state: EventTypes, file_name: str, data: str, src_name: str = None
|
||||||
) -> None:
|
) -> None:
|
||||||
"""
|
"""
|
||||||
Store a hook from a received file message.
|
Store a hook from a received file message.
|
||||||
|
@ -684,11 +663,11 @@ class Session:
|
||||||
:param state: when to run hook
|
:param state: when to run hook
|
||||||
:param file_name: file name for hook
|
:param file_name: file name for hook
|
||||||
:param data: hook data
|
:param data: hook data
|
||||||
:param source_name: source name
|
:param src_name: source name
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
logging.info(
|
logger.info(
|
||||||
"setting state hook: %s - %s source(%s)", state, file_name, source_name
|
"setting state hook: %s - %s source(%s)", state, file_name, src_name
|
||||||
)
|
)
|
||||||
hook = file_name, data
|
hook = file_name, data
|
||||||
state_hooks = self.hooks.setdefault(state, [])
|
state_hooks = self.hooks.setdefault(state, [])
|
||||||
|
@ -696,26 +675,26 @@ class Session:
|
||||||
|
|
||||||
# immediately run a hook if it is in the current state
|
# immediately run a hook if it is in the current state
|
||||||
if self.state == state:
|
if self.state == state:
|
||||||
logging.info("immediately running new state hook")
|
logger.info("immediately running new state hook")
|
||||||
self.run_hook(hook)
|
self.run_hook(hook)
|
||||||
|
|
||||||
def add_node_file(
|
def add_node_file(
|
||||||
self, node_id: int, source_name: str, file_name: str, data: str
|
self, node_id: int, src_path: Path, file_path: Path, data: str
|
||||||
) -> None:
|
) -> None:
|
||||||
"""
|
"""
|
||||||
Add a file to a node.
|
Add a file to a node.
|
||||||
|
|
||||||
:param node_id: node to add file to
|
:param node_id: node to add file to
|
||||||
:param source_name: source file name
|
:param src_path: source file path
|
||||||
:param file_name: file name to add
|
:param file_path: file path to add
|
||||||
:param data: file data
|
:param data: file data
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
node = self.get_node(node_id, CoreNodeBase)
|
node = self.get_node(node_id, CoreNode)
|
||||||
if source_name is not None:
|
if src_path is not None:
|
||||||
node.addfile(source_name, file_name)
|
node.addfile(src_path, file_path)
|
||||||
elif data is not None:
|
elif data is not None:
|
||||||
node.nodefile(file_name, data)
|
node.create_file(file_path, data)
|
||||||
|
|
||||||
def clear(self) -> None:
|
def clear(self) -> None:
|
||||||
"""
|
"""
|
||||||
|
@ -769,9 +748,9 @@ class Session:
|
||||||
Shutdown all session nodes and remove the session directory.
|
Shutdown all session nodes and remove the session directory.
|
||||||
"""
|
"""
|
||||||
if self.state == EventTypes.SHUTDOWN_STATE:
|
if self.state == EventTypes.SHUTDOWN_STATE:
|
||||||
logging.info("session(%s) state(%s) already shutdown", self.id, self.state)
|
logger.info("session(%s) state(%s) already shutdown", self.id, self.state)
|
||||||
else:
|
else:
|
||||||
logging.info("session(%s) state(%s) shutting down", self.id, self.state)
|
logger.info("session(%s) state(%s) shutting down", self.id, self.state)
|
||||||
self.set_state(EventTypes.SHUTDOWN_STATE, send_event=True)
|
self.set_state(EventTypes.SHUTDOWN_STATE, send_event=True)
|
||||||
# clear out current core session
|
# clear out current core session
|
||||||
self.clear()
|
self.clear()
|
||||||
|
@ -780,7 +759,7 @@ class Session:
|
||||||
# remove this sessions working directory
|
# remove this sessions working directory
|
||||||
preserve = self.options.get_config("preservedir") == "1"
|
preserve = self.options.get_config("preservedir") == "1"
|
||||||
if not preserve:
|
if not preserve:
|
||||||
shutil.rmtree(self.session_dir, ignore_errors=True)
|
shutil.rmtree(self.directory, ignore_errors=True)
|
||||||
|
|
||||||
def broadcast_event(self, event_data: EventData) -> None:
|
def broadcast_event(self, event_data: EventData) -> None:
|
||||||
"""
|
"""
|
||||||
|
@ -864,7 +843,7 @@ class Session:
|
||||||
return
|
return
|
||||||
self.state = state
|
self.state = state
|
||||||
self.state_time = time.monotonic()
|
self.state_time = time.monotonic()
|
||||||
logging.info("changing session(%s) to state %s", self.id, state.name)
|
logger.info("changing session(%s) to state %s", self.id, state.name)
|
||||||
self.write_state(state)
|
self.write_state(state)
|
||||||
self.run_hooks(state)
|
self.run_hooks(state)
|
||||||
self.run_state_hooks(state)
|
self.run_state_hooks(state)
|
||||||
|
@ -879,12 +858,12 @@ class Session:
|
||||||
:param state: state to write to file
|
:param state: state to write to file
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
state_file = os.path.join(self.session_dir, "state")
|
state_file = self.directory / "state"
|
||||||
try:
|
try:
|
||||||
with open(state_file, "w") as f:
|
with state_file.open("w") as f:
|
||||||
f.write(f"{state.value} {state.name}\n")
|
f.write(f"{state.value} {state.name}\n")
|
||||||
except IOError:
|
except IOError:
|
||||||
logging.exception("error writing state file: %s", state.name)
|
logger.exception("error writing state file: %s", state.name)
|
||||||
|
|
||||||
def run_hooks(self, state: EventTypes) -> None:
|
def run_hooks(self, state: EventTypes) -> None:
|
||||||
"""
|
"""
|
||||||
|
@ -906,24 +885,24 @@ class Session:
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
file_name, data = hook
|
file_name, data = hook
|
||||||
logging.info("running hook %s", file_name)
|
logger.info("running hook %s", file_name)
|
||||||
file_path = os.path.join(self.session_dir, file_name)
|
file_path = self.directory / file_name
|
||||||
log_path = os.path.join(self.session_dir, f"{file_name}.log")
|
log_path = self.directory / f"{file_name}.log"
|
||||||
try:
|
try:
|
||||||
with open(file_path, "w") as f:
|
with file_path.open("w") as f:
|
||||||
f.write(data)
|
f.write(data)
|
||||||
with open(log_path, "w") as f:
|
with log_path.open("w") as f:
|
||||||
args = ["/bin/sh", file_name]
|
args = ["/bin/sh", file_name]
|
||||||
subprocess.check_call(
|
subprocess.check_call(
|
||||||
args,
|
args,
|
||||||
stdout=f,
|
stdout=f,
|
||||||
stderr=subprocess.STDOUT,
|
stderr=subprocess.STDOUT,
|
||||||
close_fds=True,
|
close_fds=True,
|
||||||
cwd=self.session_dir,
|
cwd=self.directory,
|
||||||
env=self.get_environment(),
|
env=self.get_environment(),
|
||||||
)
|
)
|
||||||
except (IOError, subprocess.CalledProcessError):
|
except (IOError, subprocess.CalledProcessError):
|
||||||
logging.exception("error running hook: %s", file_path)
|
logger.exception("error running hook: %s", file_path)
|
||||||
|
|
||||||
def run_state_hooks(self, state: EventTypes) -> None:
|
def run_state_hooks(self, state: EventTypes) -> None:
|
||||||
"""
|
"""
|
||||||
|
@ -940,7 +919,7 @@ class Session:
|
||||||
hook(state)
|
hook(state)
|
||||||
except Exception:
|
except Exception:
|
||||||
message = f"exception occurred when running {state.name} state hook: {hook}"
|
message = f"exception occurred when running {state.name} state hook: {hook}"
|
||||||
logging.exception(message)
|
logger.exception(message)
|
||||||
self.exception(ExceptionLevels.ERROR, "Session.run_state_hooks", message)
|
self.exception(ExceptionLevels.ERROR, "Session.run_state_hooks", message)
|
||||||
|
|
||||||
def add_state_hook(
|
def add_state_hook(
|
||||||
|
@ -983,10 +962,10 @@ class Session:
|
||||||
"""
|
"""
|
||||||
self.emane.poststartup()
|
self.emane.poststartup()
|
||||||
# create session deployed xml
|
# create session deployed xml
|
||||||
xml_file_name = os.path.join(self.session_dir, "session-deployed.xml")
|
|
||||||
xml_writer = corexml.CoreXmlWriter(self)
|
xml_writer = corexml.CoreXmlWriter(self)
|
||||||
corexmldeployment.CoreXmlDeployment(self, xml_writer.scenario)
|
corexmldeployment.CoreXmlDeployment(self, xml_writer.scenario)
|
||||||
xml_writer.write(xml_file_name)
|
xml_file_path = self.directory / "session-deployed.xml"
|
||||||
|
xml_writer.write(xml_file_path)
|
||||||
|
|
||||||
def get_environment(self, state: bool = True) -> Dict[str, str]:
|
def get_environment(self, state: bool = True) -> Dict[str, str]:
|
||||||
"""
|
"""
|
||||||
|
@ -1001,9 +980,9 @@ class Session:
|
||||||
env["CORE_PYTHON"] = sys.executable
|
env["CORE_PYTHON"] = sys.executable
|
||||||
env["SESSION"] = str(self.id)
|
env["SESSION"] = str(self.id)
|
||||||
env["SESSION_SHORT"] = self.short_session_id()
|
env["SESSION_SHORT"] = self.short_session_id()
|
||||||
env["SESSION_DIR"] = self.session_dir
|
env["SESSION_DIR"] = str(self.directory)
|
||||||
env["SESSION_NAME"] = str(self.name)
|
env["SESSION_NAME"] = str(self.name)
|
||||||
env["SESSION_FILENAME"] = str(self.file_name)
|
env["SESSION_FILENAME"] = str(self.file_path)
|
||||||
env["SESSION_USER"] = str(self.user)
|
env["SESSION_USER"] = str(self.user)
|
||||||
if state:
|
if state:
|
||||||
env["SESSION_STATE"] = str(self.state)
|
env["SESSION_STATE"] = str(self.state)
|
||||||
|
@ -1011,8 +990,8 @@ class Session:
|
||||||
# /etc/core/environment
|
# /etc/core/environment
|
||||||
# /home/user/.core/environment
|
# /home/user/.core/environment
|
||||||
# /tmp/pycore.<session id>/environment
|
# /tmp/pycore.<session id>/environment
|
||||||
core_env_path = Path(constants.CORE_CONF_DIR) / "environment"
|
core_env_path = constants.CORE_CONF_DIR / "environment"
|
||||||
session_env_path = Path(self.session_dir) / "environment"
|
session_env_path = self.directory / "environment"
|
||||||
if self.user:
|
if self.user:
|
||||||
user_home_path = Path(f"~{self.user}").expanduser()
|
user_home_path = Path(f"~{self.user}").expanduser()
|
||||||
user_env1 = user_home_path / ".core" / "environment"
|
user_env1 = user_home_path / ".core" / "environment"
|
||||||
|
@ -1025,23 +1004,23 @@ class Session:
|
||||||
try:
|
try:
|
||||||
utils.load_config(path, env)
|
utils.load_config(path, env)
|
||||||
except IOError:
|
except IOError:
|
||||||
logging.exception("error reading environment file: %s", path)
|
logger.exception("error reading environment file: %s", path)
|
||||||
return env
|
return env
|
||||||
|
|
||||||
def set_thumbnail(self, thumb_file: str) -> None:
|
def set_thumbnail(self, thumb_file: Path) -> None:
|
||||||
"""
|
"""
|
||||||
Set the thumbnail filename. Move files from /tmp to session dir.
|
Set the thumbnail filename. Move files from /tmp to session dir.
|
||||||
|
|
||||||
:param thumb_file: tumbnail file to set for session
|
:param thumb_file: tumbnail file to set for session
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
if not os.path.exists(thumb_file):
|
if not thumb_file.is_file():
|
||||||
logging.error("thumbnail file to set does not exist: %s", thumb_file)
|
logger.error("thumbnail file to set does not exist: %s", thumb_file)
|
||||||
self.thumbnail = None
|
self.thumbnail = None
|
||||||
return
|
return
|
||||||
destination_file = os.path.join(self.session_dir, os.path.basename(thumb_file))
|
dst_path = self.directory / thumb_file.name
|
||||||
shutil.copy(thumb_file, destination_file)
|
shutil.copy(thumb_file, dst_path)
|
||||||
self.thumbnail = destination_file
|
self.thumbnail = dst_path
|
||||||
|
|
||||||
def set_user(self, user: str) -> None:
|
def set_user(self, user: str) -> None:
|
||||||
"""
|
"""
|
||||||
|
@ -1054,10 +1033,10 @@ class Session:
|
||||||
if user:
|
if user:
|
||||||
try:
|
try:
|
||||||
uid = pwd.getpwnam(user).pw_uid
|
uid = pwd.getpwnam(user).pw_uid
|
||||||
gid = os.stat(self.session_dir).st_gid
|
gid = self.directory.stat().st_gid
|
||||||
os.chown(self.session_dir, uid, gid)
|
os.chown(self.directory, uid, gid)
|
||||||
except IOError:
|
except IOError:
|
||||||
logging.exception("failed to set permission on %s", self.session_dir)
|
logger.exception("failed to set permission on %s", self.directory)
|
||||||
self.user = user
|
self.user = user
|
||||||
|
|
||||||
def create_node(
|
def create_node(
|
||||||
|
@ -1114,7 +1093,7 @@ class Session:
|
||||||
with self.nodes_lock:
|
with self.nodes_lock:
|
||||||
if _id in self.nodes:
|
if _id in self.nodes:
|
||||||
node = self.nodes.pop(_id)
|
node = self.nodes.pop(_id)
|
||||||
logging.info("deleted node(%s)", node.name)
|
logger.info("deleted node(%s)", node.name)
|
||||||
if node:
|
if node:
|
||||||
node.shutdown()
|
node.shutdown()
|
||||||
self.sdt.delete_node(_id)
|
self.sdt.delete_node(_id)
|
||||||
|
@ -1140,14 +1119,14 @@ class Session:
|
||||||
Write nodes to a 'nodes' file in the session dir.
|
Write nodes to a 'nodes' file in the session dir.
|
||||||
The 'nodes' file lists: number, name, api-type, class-type
|
The 'nodes' file lists: number, name, api-type, class-type
|
||||||
"""
|
"""
|
||||||
file_path = os.path.join(self.session_dir, "nodes")
|
file_path = self.directory / "nodes"
|
||||||
try:
|
try:
|
||||||
with self.nodes_lock:
|
with self.nodes_lock:
|
||||||
with open(file_path, "w") as f:
|
with file_path.open("w") as f:
|
||||||
for _id, node in self.nodes.items():
|
for _id, node in self.nodes.items():
|
||||||
f.write(f"{_id} {node.name} {node.apitype} {type(node)}\n")
|
f.write(f"{_id} {node.name} {node.apitype} {type(node)}\n")
|
||||||
except IOError:
|
except IOError:
|
||||||
logging.exception("error writing nodes file")
|
logger.exception("error writing nodes file")
|
||||||
|
|
||||||
def exception(
|
def exception(
|
||||||
self, level: ExceptionLevels, source: str, text: str, node_id: int = None
|
self, level: ExceptionLevels, source: str, text: str, node_id: int = None
|
||||||
|
@ -1238,13 +1217,13 @@ class Session:
|
||||||
"""
|
"""
|
||||||
# this is called from instantiate() after receiving an event message
|
# this is called from instantiate() after receiving an event message
|
||||||
# for the instantiation state
|
# for the instantiation state
|
||||||
logging.debug(
|
logger.debug(
|
||||||
"session(%s) checking if not in runtime state, current state: %s",
|
"session(%s) checking if not in runtime state, current state: %s",
|
||||||
self.id,
|
self.id,
|
||||||
self.state.name,
|
self.state.name,
|
||||||
)
|
)
|
||||||
if self.state == EventTypes.RUNTIME_STATE:
|
if self.state == EventTypes.RUNTIME_STATE:
|
||||||
logging.info("valid runtime state found, returning")
|
logger.info("valid runtime state found, returning")
|
||||||
return
|
return
|
||||||
# start event loop and set to runtime
|
# start event loop and set to runtime
|
||||||
self.event_loop.run()
|
self.event_loop.run()
|
||||||
|
@ -1258,25 +1237,23 @@ class Session:
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
if self.state.already_collected():
|
if self.state.already_collected():
|
||||||
logging.info(
|
logger.info(
|
||||||
"session(%s) state(%s) already data collected", self.id, self.state
|
"session(%s) state(%s) already data collected", self.id, self.state
|
||||||
)
|
)
|
||||||
return
|
return
|
||||||
logging.info("session(%s) state(%s) data collection", self.id, self.state)
|
logger.info("session(%s) state(%s) data collection", self.id, self.state)
|
||||||
self.set_state(EventTypes.DATACOLLECT_STATE, send_event=True)
|
self.set_state(EventTypes.DATACOLLECT_STATE, send_event=True)
|
||||||
|
|
||||||
# stop event loop
|
# stop event loop
|
||||||
self.event_loop.stop()
|
self.event_loop.stop()
|
||||||
|
|
||||||
# stop node services
|
# stop mobility and node services
|
||||||
with self.nodes_lock:
|
with self.nodes_lock:
|
||||||
funcs = []
|
funcs = []
|
||||||
for node_id in self.nodes:
|
for node in self.nodes.values():
|
||||||
node = self.nodes[node_id]
|
if isinstance(node, CoreNodeBase) and node.up:
|
||||||
if not isinstance(node, CoreNodeBase) or not node.up:
|
args = (node,)
|
||||||
continue
|
funcs.append((self.services.stop_services, args, {}))
|
||||||
args = (node,)
|
|
||||||
funcs.append((self.services.stop_services, args, {}))
|
|
||||||
utils.threadpool(funcs)
|
utils.threadpool(funcs)
|
||||||
|
|
||||||
# shutdown emane
|
# shutdown emane
|
||||||
|
@ -1307,7 +1284,7 @@ class Session:
|
||||||
:param node: node to boot
|
:param node: node to boot
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
logging.info("booting node(%s): %s", node.name, [x.name for x in node.services])
|
logger.info("booting node(%s): %s", node.name, [x.name for x in node.services])
|
||||||
self.services.boot_services(node)
|
self.services.boot_services(node)
|
||||||
node.start_config_services()
|
node.start_config_services()
|
||||||
|
|
||||||
|
@ -1328,7 +1305,7 @@ class Session:
|
||||||
funcs.append((self.boot_node, (node,), {}))
|
funcs.append((self.boot_node, (node,), {}))
|
||||||
results, exceptions = utils.threadpool(funcs)
|
results, exceptions = utils.threadpool(funcs)
|
||||||
total = time.monotonic() - start
|
total = time.monotonic() - start
|
||||||
logging.debug("boot run time: %s", total)
|
logger.debug("boot run time: %s", total)
|
||||||
if not exceptions:
|
if not exceptions:
|
||||||
self.update_control_iface_hosts()
|
self.update_control_iface_hosts()
|
||||||
return exceptions
|
return exceptions
|
||||||
|
@ -1356,7 +1333,7 @@ class Session:
|
||||||
"""
|
"""
|
||||||
d0 = self.options.get_config("controlnetif0")
|
d0 = self.options.get_config("controlnetif0")
|
||||||
if d0:
|
if d0:
|
||||||
logging.error("controlnet0 cannot be assigned with a host interface")
|
logger.error("controlnet0 cannot be assigned with a host interface")
|
||||||
d1 = self.options.get_config("controlnetif1")
|
d1 = self.options.get_config("controlnetif1")
|
||||||
d2 = self.options.get_config("controlnetif2")
|
d2 = self.options.get_config("controlnetif2")
|
||||||
d3 = self.options.get_config("controlnetif3")
|
d3 = self.options.get_config("controlnetif3")
|
||||||
|
@ -1401,7 +1378,7 @@ class Session:
|
||||||
:param conf_required: flag to check if conf is required
|
:param conf_required: flag to check if conf is required
|
||||||
:return: control net node
|
:return: control net node
|
||||||
"""
|
"""
|
||||||
logging.debug(
|
logger.debug(
|
||||||
"add/remove control net: index(%s) remove(%s) conf_required(%s)",
|
"add/remove control net: index(%s) remove(%s) conf_required(%s)",
|
||||||
net_index,
|
net_index,
|
||||||
remove,
|
remove,
|
||||||
|
@ -1415,7 +1392,7 @@ class Session:
|
||||||
return None
|
return None
|
||||||
else:
|
else:
|
||||||
prefix_spec = CtrlNet.DEFAULT_PREFIX_LIST[net_index]
|
prefix_spec = CtrlNet.DEFAULT_PREFIX_LIST[net_index]
|
||||||
logging.debug("prefix spec: %s", prefix_spec)
|
logger.debug("prefix spec: %s", prefix_spec)
|
||||||
server_iface = self.get_control_net_server_ifaces()[net_index]
|
server_iface = self.get_control_net_server_ifaces()[net_index]
|
||||||
|
|
||||||
# return any existing controlnet bridge
|
# return any existing controlnet bridge
|
||||||
|
@ -1438,7 +1415,7 @@ class Session:
|
||||||
if net_index == 0:
|
if net_index == 0:
|
||||||
updown_script = self.options.get_config("controlnet_updown_script")
|
updown_script = self.options.get_config("controlnet_updown_script")
|
||||||
if not updown_script:
|
if not updown_script:
|
||||||
logging.debug("controlnet updown script not configured")
|
logger.debug("controlnet updown script not configured")
|
||||||
|
|
||||||
prefixes = prefix_spec.split()
|
prefixes = prefix_spec.split()
|
||||||
if len(prefixes) > 1:
|
if len(prefixes) > 1:
|
||||||
|
@ -1452,7 +1429,7 @@ class Session:
|
||||||
else:
|
else:
|
||||||
prefix = prefixes[0]
|
prefix = prefixes[0]
|
||||||
|
|
||||||
logging.info(
|
logger.info(
|
||||||
"controlnet(%s) prefix(%s) updown(%s) serverintf(%s)",
|
"controlnet(%s) prefix(%s) updown(%s) serverintf(%s)",
|
||||||
_id,
|
_id,
|
||||||
prefix,
|
prefix,
|
||||||
|
@ -1515,7 +1492,7 @@ class Session:
|
||||||
msg = f"Control interface not added to node {node.id}. "
|
msg = f"Control interface not added to node {node.id}. "
|
||||||
msg += f"Invalid control network prefix ({control_net.prefix}). "
|
msg += f"Invalid control network prefix ({control_net.prefix}). "
|
||||||
msg += "A longer prefix length may be required for this many nodes."
|
msg += "A longer prefix length may be required for this many nodes."
|
||||||
logging.exception(msg)
|
logger.exception(msg)
|
||||||
|
|
||||||
def update_control_iface_hosts(
|
def update_control_iface_hosts(
|
||||||
self, net_index: int = 0, remove: bool = False
|
self, net_index: int = 0, remove: bool = False
|
||||||
|
@ -1533,12 +1510,12 @@ class Session:
|
||||||
try:
|
try:
|
||||||
control_net = self.get_control_net(net_index)
|
control_net = self.get_control_net(net_index)
|
||||||
except CoreError:
|
except CoreError:
|
||||||
logging.exception("error retrieving control net node")
|
logger.exception("error retrieving control net node")
|
||||||
return
|
return
|
||||||
|
|
||||||
header = f"CORE session {self.id} host entries"
|
header = f"CORE session {self.id} host entries"
|
||||||
if remove:
|
if remove:
|
||||||
logging.info("Removing /etc/hosts file entries.")
|
logger.info("Removing /etc/hosts file entries.")
|
||||||
utils.file_demunge("/etc/hosts", header)
|
utils.file_demunge("/etc/hosts", header)
|
||||||
return
|
return
|
||||||
|
|
||||||
|
@ -1548,7 +1525,7 @@ class Session:
|
||||||
for ip in iface.ips():
|
for ip in iface.ips():
|
||||||
entries.append(f"{ip.ip} {name}")
|
entries.append(f"{ip.ip} {name}")
|
||||||
|
|
||||||
logging.info("Adding %d /etc/hosts file entries.", len(entries))
|
logger.info("Adding %d /etc/hosts file entries.", len(entries))
|
||||||
utils.file_munge("/etc/hosts", header, "\n".join(entries) + "\n")
|
utils.file_munge("/etc/hosts", header, "\n".join(entries) + "\n")
|
||||||
|
|
||||||
def runtime(self) -> float:
|
def runtime(self) -> float:
|
||||||
|
@ -1577,7 +1554,7 @@ class Session:
|
||||||
current_time = self.runtime()
|
current_time = self.runtime()
|
||||||
if current_time > 0:
|
if current_time > 0:
|
||||||
if event_time <= current_time:
|
if event_time <= current_time:
|
||||||
logging.warning(
|
logger.warning(
|
||||||
"could not schedule past event for time %s (run time is now %s)",
|
"could not schedule past event for time %s (run time is now %s)",
|
||||||
event_time,
|
event_time,
|
||||||
current_time,
|
current_time,
|
||||||
|
@ -1589,7 +1566,7 @@ class Session:
|
||||||
)
|
)
|
||||||
if not name:
|
if not name:
|
||||||
name = ""
|
name = ""
|
||||||
logging.info(
|
logger.info(
|
||||||
"scheduled event %s at time %s data=%s",
|
"scheduled event %s at time %s data=%s",
|
||||||
name,
|
name,
|
||||||
event_time + current_time,
|
event_time + current_time,
|
||||||
|
@ -1608,12 +1585,12 @@ class Session:
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
if data is None:
|
if data is None:
|
||||||
logging.warning("no data for event node(%s) name(%s)", node_id, name)
|
logger.warning("no data for event node(%s) name(%s)", node_id, name)
|
||||||
return
|
return
|
||||||
now = self.runtime()
|
now = self.runtime()
|
||||||
if not name:
|
if not name:
|
||||||
name = ""
|
name = ""
|
||||||
logging.info("running event %s at time %s cmd=%s", name, now, data)
|
logger.info("running event %s at time %s cmd=%s", name, now, data)
|
||||||
if not node_id:
|
if not node_id:
|
||||||
utils.mute_detach(data)
|
utils.mute_detach(data)
|
||||||
else:
|
else:
|
||||||
|
|
|
@ -13,51 +13,87 @@ class SessionConfig(ConfigurableManager, ConfigurableOptions):
|
||||||
name: str = "session"
|
name: str = "session"
|
||||||
options: List[Configuration] = [
|
options: List[Configuration] = [
|
||||||
Configuration(
|
Configuration(
|
||||||
_id="controlnet", _type=ConfigDataTypes.STRING, label="Control Network"
|
id="controlnet", type=ConfigDataTypes.STRING, label="Control Network"
|
||||||
),
|
),
|
||||||
Configuration(
|
Configuration(
|
||||||
_id="controlnet0", _type=ConfigDataTypes.STRING, label="Control Network 0"
|
id="controlnet0", type=ConfigDataTypes.STRING, label="Control Network 0"
|
||||||
),
|
),
|
||||||
Configuration(
|
Configuration(
|
||||||
_id="controlnet1", _type=ConfigDataTypes.STRING, label="Control Network 1"
|
id="controlnet1", type=ConfigDataTypes.STRING, label="Control Network 1"
|
||||||
),
|
),
|
||||||
Configuration(
|
Configuration(
|
||||||
_id="controlnet2", _type=ConfigDataTypes.STRING, label="Control Network 2"
|
id="controlnet2", type=ConfigDataTypes.STRING, label="Control Network 2"
|
||||||
),
|
),
|
||||||
Configuration(
|
Configuration(
|
||||||
_id="controlnet3", _type=ConfigDataTypes.STRING, label="Control Network 3"
|
id="controlnet3", type=ConfigDataTypes.STRING, label="Control Network 3"
|
||||||
),
|
),
|
||||||
Configuration(
|
Configuration(
|
||||||
_id="controlnet_updown_script",
|
id="controlnet_updown_script",
|
||||||
_type=ConfigDataTypes.STRING,
|
type=ConfigDataTypes.STRING,
|
||||||
label="Control Network Script",
|
label="Control Network Script",
|
||||||
),
|
),
|
||||||
Configuration(
|
Configuration(
|
||||||
_id="enablerj45",
|
id="enablerj45",
|
||||||
_type=ConfigDataTypes.BOOL,
|
type=ConfigDataTypes.BOOL,
|
||||||
default="1",
|
default="1",
|
||||||
label="Enable RJ45s",
|
label="Enable RJ45s",
|
||||||
),
|
),
|
||||||
Configuration(
|
Configuration(
|
||||||
_id="preservedir",
|
id="preservedir",
|
||||||
_type=ConfigDataTypes.BOOL,
|
type=ConfigDataTypes.BOOL,
|
||||||
default="0",
|
default="0",
|
||||||
label="Preserve session dir",
|
label="Preserve session dir",
|
||||||
),
|
),
|
||||||
Configuration(
|
Configuration(
|
||||||
_id="enablesdt",
|
id="enablesdt",
|
||||||
_type=ConfigDataTypes.BOOL,
|
type=ConfigDataTypes.BOOL,
|
||||||
default="0",
|
default="0",
|
||||||
label="Enable SDT3D output",
|
label="Enable SDT3D output",
|
||||||
),
|
),
|
||||||
Configuration(
|
Configuration(
|
||||||
_id="sdturl",
|
id="sdturl",
|
||||||
_type=ConfigDataTypes.STRING,
|
type=ConfigDataTypes.STRING,
|
||||||
default=Sdt.DEFAULT_SDT_URL,
|
default=Sdt.DEFAULT_SDT_URL,
|
||||||
label="SDT3D URL",
|
label="SDT3D URL",
|
||||||
),
|
),
|
||||||
Configuration(
|
Configuration(
|
||||||
_id="ovs", _type=ConfigDataTypes.BOOL, default="0", label="Enable OVS"
|
id="ovs", type=ConfigDataTypes.BOOL, default="0", label="Enable OVS"
|
||||||
|
),
|
||||||
|
Configuration(
|
||||||
|
id="platform_id_start",
|
||||||
|
type=ConfigDataTypes.INT32,
|
||||||
|
default="1",
|
||||||
|
label="EMANE Platform ID Start",
|
||||||
|
),
|
||||||
|
Configuration(
|
||||||
|
id="nem_id_start",
|
||||||
|
type=ConfigDataTypes.INT32,
|
||||||
|
default="1",
|
||||||
|
label="EMANE NEM ID Start",
|
||||||
|
),
|
||||||
|
Configuration(
|
||||||
|
id="link_enabled",
|
||||||
|
type=ConfigDataTypes.BOOL,
|
||||||
|
default="1",
|
||||||
|
label="EMANE Links?",
|
||||||
|
),
|
||||||
|
Configuration(
|
||||||
|
id="loss_threshold",
|
||||||
|
type=ConfigDataTypes.INT32,
|
||||||
|
default="30",
|
||||||
|
label="EMANE Link Loss Threshold (%)",
|
||||||
|
),
|
||||||
|
Configuration(
|
||||||
|
id="link_interval",
|
||||||
|
type=ConfigDataTypes.INT32,
|
||||||
|
default="1",
|
||||||
|
label="EMANE Link Check Interval (sec)",
|
||||||
|
),
|
||||||
|
Configuration(
|
||||||
|
id="link_timeout",
|
||||||
|
type=ConfigDataTypes.INT32,
|
||||||
|
default="4",
|
||||||
|
label="EMANE Link Timeout (sec)",
|
||||||
),
|
),
|
||||||
]
|
]
|
||||||
config_type: RegisterTlvs = RegisterTlvs.UTILITY
|
config_type: RegisterTlvs = RegisterTlvs.UTILITY
|
||||||
|
@ -112,3 +148,13 @@ class SessionConfig(ConfigurableManager, ConfigurableOptions):
|
||||||
if value is not None:
|
if value is not None:
|
||||||
value = int(value)
|
value = int(value)
|
||||||
return value
|
return value
|
||||||
|
|
||||||
|
def config_reset(self, node_id: int = None) -> None:
|
||||||
|
"""
|
||||||
|
Clear prior configuration files and reset to default values.
|
||||||
|
|
||||||
|
:param node_id: node id to store configuration for
|
||||||
|
:return: nothing
|
||||||
|
"""
|
||||||
|
super().config_reset(node_id)
|
||||||
|
self.set_configs(self.default_values())
|
||||||
|
|
|
@ -46,3 +46,11 @@ class CoreServiceBootError(Exception):
|
||||||
"""
|
"""
|
||||||
|
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
class CoreConfigError(Exception):
|
||||||
|
"""
|
||||||
|
Used when there is an error defining a configurable option.
|
||||||
|
"""
|
||||||
|
|
||||||
|
pass
|
||||||
|
|
|
@ -7,15 +7,15 @@ SYSCTL: str = "sysctl"
|
||||||
IP: str = "ip"
|
IP: str = "ip"
|
||||||
ETHTOOL: str = "ethtool"
|
ETHTOOL: str = "ethtool"
|
||||||
TC: str = "tc"
|
TC: str = "tc"
|
||||||
EBTABLES: str = "ebtables"
|
|
||||||
MOUNT: str = "mount"
|
MOUNT: str = "mount"
|
||||||
UMOUNT: str = "umount"
|
UMOUNT: str = "umount"
|
||||||
OVS_VSCTL: str = "ovs-vsctl"
|
OVS_VSCTL: str = "ovs-vsctl"
|
||||||
TEST: str = "test"
|
TEST: str = "test"
|
||||||
|
NFTABLES: str = "nft"
|
||||||
|
|
||||||
COMMON_REQUIREMENTS: List[str] = [
|
COMMON_REQUIREMENTS: List[str] = [
|
||||||
BASH,
|
BASH,
|
||||||
EBTABLES,
|
NFTABLES,
|
||||||
ETHTOOL,
|
ETHTOOL,
|
||||||
IP,
|
IP,
|
||||||
MOUNT,
|
MOUNT,
|
||||||
|
|
|
@ -22,6 +22,7 @@ from core.gui.statusbar import StatusBar
|
||||||
from core.gui.themes import PADY
|
from core.gui.themes import PADY
|
||||||
from core.gui.toolbar import Toolbar
|
from core.gui.toolbar import Toolbar
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
WIDTH: int = 1000
|
WIDTH: int = 1000
|
||||||
HEIGHT: int = 800
|
HEIGHT: int = 800
|
||||||
|
|
||||||
|
@ -171,7 +172,7 @@ class Application(ttk.Frame):
|
||||||
def show_grpc_exception(
|
def show_grpc_exception(
|
||||||
self, message: str, e: grpc.RpcError, blocking: bool = False
|
self, message: str, e: grpc.RpcError, blocking: bool = False
|
||||||
) -> None:
|
) -> None:
|
||||||
logging.exception("app grpc exception", exc_info=e)
|
logger.exception("app grpc exception", exc_info=e)
|
||||||
dialog = ErrorDialog(self, "GRPC Exception", message, e.details())
|
dialog = ErrorDialog(self, "GRPC Exception", message, e.details())
|
||||||
if blocking:
|
if blocking:
|
||||||
dialog.show()
|
dialog.show()
|
||||||
|
@ -179,7 +180,7 @@ class Application(ttk.Frame):
|
||||||
self.after(0, lambda: dialog.show())
|
self.after(0, lambda: dialog.show())
|
||||||
|
|
||||||
def show_exception(self, message: str, e: Exception) -> None:
|
def show_exception(self, message: str, e: Exception) -> None:
|
||||||
logging.exception("app exception", exc_info=e)
|
logger.exception("app exception", exc_info=e)
|
||||||
self.after(
|
self.after(
|
||||||
0, lambda: ErrorDialog(self, "App Exception", message, str(e)).show()
|
0, lambda: ErrorDialog(self, "App Exception", message, str(e)).show()
|
||||||
)
|
)
|
||||||
|
|
|
@ -120,25 +120,18 @@ class IpConfigs(yaml.YAMLObject):
|
||||||
yaml_tag: str = "!IpConfigs"
|
yaml_tag: str = "!IpConfigs"
|
||||||
yaml_loader: Type[yaml.SafeLoader] = yaml.SafeLoader
|
yaml_loader: Type[yaml.SafeLoader] = yaml.SafeLoader
|
||||||
|
|
||||||
def __init__(
|
def __init__(self, **kwargs) -> None:
|
||||||
self,
|
self.__setstate__(kwargs)
|
||||||
ip4: str = None,
|
|
||||||
ip6: str = None,
|
def __setstate__(self, kwargs):
|
||||||
ip4s: List[str] = None,
|
self.ip4s: List[str] = kwargs.get(
|
||||||
ip6s: List[str] = None,
|
"ip4s", ["10.0.0.0", "192.168.0.0", "172.16.0.0"]
|
||||||
) -> None:
|
)
|
||||||
if ip4s is None:
|
self.ip4: str = kwargs.get("ip4", self.ip4s[0])
|
||||||
ip4s = ["10.0.0.0", "192.168.0.0", "172.16.0.0"]
|
self.ip6s: List[str] = kwargs.get("ip6s", ["2001::", "2002::", "a::"])
|
||||||
self.ip4s: List[str] = ip4s
|
self.ip6: str = kwargs.get("ip6", self.ip6s[0])
|
||||||
if ip6s is None:
|
self.enable_ip4: bool = kwargs.get("enable_ip4", True)
|
||||||
ip6s = ["2001::", "2002::", "a::"]
|
self.enable_ip6: bool = kwargs.get("enable_ip6", True)
|
||||||
self.ip6s: List[str] = ip6s
|
|
||||||
if ip4 is None:
|
|
||||||
ip4 = self.ip4s[0]
|
|
||||||
self.ip4: str = ip4
|
|
||||||
if ip6 is None:
|
|
||||||
ip6 = self.ip6s[0]
|
|
||||||
self.ip6: str = ip6
|
|
||||||
|
|
||||||
|
|
||||||
class GuiConfig(yaml.YAMLObject):
|
class GuiConfig(yaml.YAMLObject):
|
||||||
|
@ -223,7 +216,7 @@ def check_directory() -> None:
|
||||||
|
|
||||||
def read() -> GuiConfig:
|
def read() -> GuiConfig:
|
||||||
with CONFIG_PATH.open("r") as f:
|
with CONFIG_PATH.open("r") as f:
|
||||||
return yaml.load(f, Loader=yaml.SafeLoader)
|
return yaml.safe_load(f)
|
||||||
|
|
||||||
|
|
||||||
def save(config: GuiConfig) -> None:
|
def save(config: GuiConfig) -> None:
|
||||||
|
|
|
@ -6,23 +6,18 @@ import json
|
||||||
import logging
|
import logging
|
||||||
import os
|
import os
|
||||||
import tkinter as tk
|
import tkinter as tk
|
||||||
|
from pathlib import Path
|
||||||
from tkinter import messagebox
|
from tkinter import messagebox
|
||||||
from typing import TYPE_CHECKING, Dict, Iterable, List, Optional, Set, Tuple
|
from typing import TYPE_CHECKING, Dict, Iterable, List, Optional, Set, Tuple
|
||||||
|
|
||||||
import grpc
|
import grpc
|
||||||
|
|
||||||
from core.api.grpc import (
|
from core.api.grpc import client, configservices_pb2, core_pb2
|
||||||
client,
|
|
||||||
configservices_pb2,
|
|
||||||
core_pb2,
|
|
||||||
emane_pb2,
|
|
||||||
mobility_pb2,
|
|
||||||
services_pb2,
|
|
||||||
wlan_pb2,
|
|
||||||
)
|
|
||||||
from core.api.grpc.wrappers import (
|
from core.api.grpc.wrappers import (
|
||||||
ConfigOption,
|
ConfigOption,
|
||||||
ConfigService,
|
ConfigService,
|
||||||
|
EmaneModelConfig,
|
||||||
|
Event,
|
||||||
ExceptionEvent,
|
ExceptionEvent,
|
||||||
Link,
|
Link,
|
||||||
LinkEvent,
|
LinkEvent,
|
||||||
|
@ -33,6 +28,9 @@ from core.api.grpc.wrappers import (
|
||||||
NodeServiceData,
|
NodeServiceData,
|
||||||
NodeType,
|
NodeType,
|
||||||
Position,
|
Position,
|
||||||
|
Server,
|
||||||
|
ServiceConfig,
|
||||||
|
ServiceFileConfig,
|
||||||
Session,
|
Session,
|
||||||
SessionLocation,
|
SessionLocation,
|
||||||
SessionState,
|
SessionState,
|
||||||
|
@ -45,10 +43,11 @@ from core.gui.dialogs.mobilityplayer import MobilityPlayer
|
||||||
from core.gui.dialogs.sessions import SessionsDialog
|
from core.gui.dialogs.sessions import SessionsDialog
|
||||||
from core.gui.graph.edges import CanvasEdge
|
from core.gui.graph.edges import CanvasEdge
|
||||||
from core.gui.graph.node import CanvasNode
|
from core.gui.graph.node import CanvasNode
|
||||||
from core.gui.graph.shape import Shape
|
|
||||||
from core.gui.interface import InterfaceManager
|
from core.gui.interface import InterfaceManager
|
||||||
from core.gui.nodeutils import NodeDraw
|
from core.gui.nodeutils import NodeDraw
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.gui.app import Application
|
from core.gui.app import Application
|
||||||
|
|
||||||
|
@ -77,6 +76,7 @@ class CoreClient:
|
||||||
self.config_services: Dict[str, ConfigService] = {}
|
self.config_services: Dict[str, ConfigService] = {}
|
||||||
|
|
||||||
# loaded configuration data
|
# loaded configuration data
|
||||||
|
self.emane_models: List[str] = []
|
||||||
self.servers: Dict[str, CoreServer] = {}
|
self.servers: Dict[str, CoreServer] = {}
|
||||||
self.custom_nodes: Dict[str, NodeDraw] = {}
|
self.custom_nodes: Dict[str, NodeDraw] = {}
|
||||||
self.custom_observers: Dict[str, Observer] = {}
|
self.custom_observers: Dict[str, Observer] = {}
|
||||||
|
@ -98,8 +98,7 @@ class CoreClient:
|
||||||
@property
|
@property
|
||||||
def client(self) -> client.CoreGrpcClient:
|
def client(self) -> client.CoreGrpcClient:
|
||||||
if self.session:
|
if self.session:
|
||||||
response = self._client.check_session(self.session.id)
|
if not self._client.check_session(self.session.id):
|
||||||
if not response.result:
|
|
||||||
throughputs_enabled = self.handling_throughputs is not None
|
throughputs_enabled = self.handling_throughputs is not None
|
||||||
self.cancel_throughputs()
|
self.cancel_throughputs()
|
||||||
self.cancel_events()
|
self.cancel_events()
|
||||||
|
@ -150,22 +149,20 @@ class CoreClient:
|
||||||
for observer in self.app.guiconfig.observers:
|
for observer in self.app.guiconfig.observers:
|
||||||
self.custom_observers[observer.name] = observer
|
self.custom_observers[observer.name] = observer
|
||||||
|
|
||||||
def handle_events(self, event: core_pb2.Event) -> None:
|
def handle_events(self, event: Event) -> None:
|
||||||
if not self.session or event.source == GUI_SOURCE:
|
if not self.session or event.source == GUI_SOURCE:
|
||||||
return
|
return
|
||||||
if event.session_id != self.session.id:
|
if event.session_id != self.session.id:
|
||||||
logging.warning(
|
logger.warning(
|
||||||
"ignoring event session(%s) current(%s)",
|
"ignoring event session(%s) current(%s)",
|
||||||
event.session_id,
|
event.session_id,
|
||||||
self.session.id,
|
self.session.id,
|
||||||
)
|
)
|
||||||
return
|
return
|
||||||
|
if event.link_event:
|
||||||
if event.HasField("link_event"):
|
self.app.after(0, self.handle_link_event, event.link_event)
|
||||||
link_event = LinkEvent.from_proto(event.link_event)
|
elif event.session_event:
|
||||||
self.app.after(0, self.handle_link_event, link_event)
|
logger.info("session event: %s", event)
|
||||||
elif event.HasField("session_event"):
|
|
||||||
logging.info("session event: %s", event)
|
|
||||||
session_event = event.session_event
|
session_event = event.session_event
|
||||||
if session_event.event <= SessionState.SHUTDOWN.value:
|
if session_event.event <= SessionState.SHUTDOWN.value:
|
||||||
self.session.state = SessionState(session_event.event)
|
self.session.state = SessionState(session_event.event)
|
||||||
|
@ -180,24 +177,22 @@ class CoreClient:
|
||||||
else:
|
else:
|
||||||
dialog.set_pause()
|
dialog.set_pause()
|
||||||
else:
|
else:
|
||||||
logging.warning("unknown session event: %s", session_event)
|
logger.warning("unknown session event: %s", session_event)
|
||||||
elif event.HasField("node_event"):
|
elif event.node_event:
|
||||||
node_event = NodeEvent.from_proto(event.node_event)
|
self.app.after(0, self.handle_node_event, event.node_event)
|
||||||
self.app.after(0, self.handle_node_event, node_event)
|
elif event.config_event:
|
||||||
elif event.HasField("config_event"):
|
logger.info("config event: %s", event)
|
||||||
logging.info("config event: %s", event)
|
elif event.exception_event:
|
||||||
elif event.HasField("exception_event"):
|
self.handle_exception_event(event.exception_event)
|
||||||
event = ExceptionEvent.from_proto(event.session_id, event.exception_event)
|
|
||||||
self.handle_exception_event(event)
|
|
||||||
else:
|
else:
|
||||||
logging.info("unhandled event: %s", event)
|
logger.info("unhandled event: %s", event)
|
||||||
|
|
||||||
def handle_link_event(self, event: LinkEvent) -> None:
|
def handle_link_event(self, event: LinkEvent) -> None:
|
||||||
logging.debug("Link event: %s", event)
|
logger.debug("Link event: %s", event)
|
||||||
node1_id = event.link.node1_id
|
node1_id = event.link.node1_id
|
||||||
node2_id = event.link.node2_id
|
node2_id = event.link.node2_id
|
||||||
if node1_id == node2_id:
|
if node1_id == node2_id:
|
||||||
logging.warning("ignoring links with loops: %s", event)
|
logger.warning("ignoring links with loops: %s", event)
|
||||||
return
|
return
|
||||||
canvas_node1 = self.canvas_nodes[node1_id]
|
canvas_node1 = self.canvas_nodes[node1_id]
|
||||||
canvas_node2 = self.canvas_nodes[node2_id]
|
canvas_node2 = self.canvas_nodes[node2_id]
|
||||||
|
@ -215,7 +210,7 @@ class CoreClient:
|
||||||
canvas_node1, canvas_node2, event.link
|
canvas_node1, canvas_node2, event.link
|
||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
logging.warning("unknown link event: %s", event)
|
logger.warning("unknown link event: %s", event)
|
||||||
else:
|
else:
|
||||||
if event.message_type == MessageType.ADD:
|
if event.message_type == MessageType.ADD:
|
||||||
self.app.manager.add_wired_edge(canvas_node1, canvas_node2, event.link)
|
self.app.manager.add_wired_edge(canvas_node1, canvas_node2, event.link)
|
||||||
|
@ -224,10 +219,10 @@ class CoreClient:
|
||||||
elif event.message_type == MessageType.NONE:
|
elif event.message_type == MessageType.NONE:
|
||||||
self.app.manager.update_wired_edge(event.link)
|
self.app.manager.update_wired_edge(event.link)
|
||||||
else:
|
else:
|
||||||
logging.warning("unknown link event: %s", event)
|
logger.warning("unknown link event: %s", event)
|
||||||
|
|
||||||
def handle_node_event(self, event: NodeEvent) -> None:
|
def handle_node_event(self, event: NodeEvent) -> None:
|
||||||
logging.debug("node event: %s", event)
|
logger.debug("node event: %s", event)
|
||||||
node = event.node
|
node = event.node
|
||||||
if event.message_type == MessageType.NONE:
|
if event.message_type == MessageType.NONE:
|
||||||
canvas_node = self.canvas_nodes[node.id]
|
canvas_node = self.canvas_nodes[node.id]
|
||||||
|
@ -238,15 +233,13 @@ class CoreClient:
|
||||||
canvas_node.update_icon(node.icon)
|
canvas_node.update_icon(node.icon)
|
||||||
elif event.message_type == MessageType.DELETE:
|
elif event.message_type == MessageType.DELETE:
|
||||||
canvas_node = self.canvas_nodes[node.id]
|
canvas_node = self.canvas_nodes[node.id]
|
||||||
canvas_node.canvas.clear_selection()
|
canvas_node.canvas_delete()
|
||||||
canvas_node.canvas.select_object(canvas_node.id)
|
|
||||||
canvas_node.canvas.delete_selected_objects()
|
|
||||||
elif event.message_type == MessageType.ADD:
|
elif event.message_type == MessageType.ADD:
|
||||||
if node.id in self.session.nodes:
|
if node.id in self.session.nodes:
|
||||||
logging.error("core node already exists: %s", node)
|
logger.error("core node already exists: %s", node)
|
||||||
self.app.manager.add_core_node(node)
|
self.app.manager.add_core_node(node)
|
||||||
else:
|
else:
|
||||||
logging.warning("unknown node event: %s", event)
|
logger.warning("unknown node event: %s", event)
|
||||||
|
|
||||||
def enable_throughputs(self) -> None:
|
def enable_throughputs(self) -> None:
|
||||||
self.handling_throughputs = self.client.throughputs(
|
self.handling_throughputs = self.client.throughputs(
|
||||||
|
@ -278,34 +271,35 @@ class CoreClient:
|
||||||
CPU_USAGE_DELAY, self.handle_cpu_event
|
CPU_USAGE_DELAY, self.handle_cpu_event
|
||||||
)
|
)
|
||||||
|
|
||||||
def handle_throughputs(self, event: core_pb2.ThroughputsEvent) -> None:
|
def handle_throughputs(self, event: ThroughputsEvent) -> None:
|
||||||
event = ThroughputsEvent.from_proto(event)
|
|
||||||
if event.session_id != self.session.id:
|
if event.session_id != self.session.id:
|
||||||
logging.warning(
|
logger.warning(
|
||||||
"ignoring throughput event session(%s) current(%s)",
|
"ignoring throughput event session(%s) current(%s)",
|
||||||
event.session_id,
|
event.session_id,
|
||||||
self.session.id,
|
self.session.id,
|
||||||
)
|
)
|
||||||
return
|
return
|
||||||
logging.debug("handling throughputs event: %s", event)
|
logger.debug("handling throughputs event: %s", event)
|
||||||
self.app.after(0, self.app.manager.set_throughputs, event)
|
self.app.after(0, self.app.manager.set_throughputs, event)
|
||||||
|
|
||||||
def handle_cpu_event(self, event: core_pb2.CpuUsageEvent) -> None:
|
def handle_cpu_event(self, event: core_pb2.CpuUsageEvent) -> None:
|
||||||
self.app.after(0, self.app.statusbar.set_cpu, event.usage)
|
self.app.after(0, self.app.statusbar.set_cpu, event.usage)
|
||||||
|
|
||||||
def handle_exception_event(self, event: ExceptionEvent) -> None:
|
def handle_exception_event(self, event: ExceptionEvent) -> None:
|
||||||
logging.info("exception event: %s", event)
|
logger.info("exception event: %s", event)
|
||||||
self.app.statusbar.add_alert(event)
|
self.app.statusbar.add_alert(event)
|
||||||
|
|
||||||
|
def update_session_title(self) -> None:
|
||||||
|
title_file = self.session.file.name if self.session.file else ""
|
||||||
|
self.master.title(f"CORE Session({self.session.id}) {title_file}")
|
||||||
|
|
||||||
def join_session(self, session_id: int) -> None:
|
def join_session(self, session_id: int) -> None:
|
||||||
logging.info("joining session(%s)", session_id)
|
logger.info("joining session(%s)", session_id)
|
||||||
self.reset()
|
self.reset()
|
||||||
try:
|
try:
|
||||||
response = self.client.get_session(session_id)
|
self.session = self.client.get_session(session_id)
|
||||||
self.session = Session.from_proto(response.session)
|
self.session.user = self.user
|
||||||
self.client.set_session_user(self.session.id, self.user)
|
self.update_session_title()
|
||||||
title_file = self.session.file.name if self.session.file else ""
|
|
||||||
self.master.title(f"CORE Session({self.session.id}) {title_file}")
|
|
||||||
self.handling_events = self.client.events(
|
self.handling_events = self.client.events(
|
||||||
self.session.id, self.handle_events
|
self.session.id, self.handle_events
|
||||||
)
|
)
|
||||||
|
@ -320,53 +314,14 @@ class CoreClient:
|
||||||
def is_runtime(self) -> bool:
|
def is_runtime(self) -> bool:
|
||||||
return self.session and self.session.state == SessionState.RUNTIME
|
return self.session and self.session.state == SessionState.RUNTIME
|
||||||
|
|
||||||
def parse_metadata(self) -> None:
|
|
||||||
# canvas setting
|
|
||||||
config = self.session.metadata
|
|
||||||
canvas_config = config.get("canvas")
|
|
||||||
logging.debug("canvas metadata: %s", canvas_config)
|
|
||||||
if canvas_config:
|
|
||||||
canvas_config = json.loads(canvas_config)
|
|
||||||
self.app.manager.parse_metadata(canvas_config)
|
|
||||||
|
|
||||||
# load saved shapes
|
|
||||||
shapes_config = config.get("shapes")
|
|
||||||
if shapes_config:
|
|
||||||
shapes_config = json.loads(shapes_config)
|
|
||||||
for shape_config in shapes_config:
|
|
||||||
logging.debug("loading shape: %s", shape_config)
|
|
||||||
Shape.from_metadata(self.app, shape_config)
|
|
||||||
|
|
||||||
# load edges config
|
|
||||||
edges_config = config.get("edges")
|
|
||||||
if edges_config:
|
|
||||||
edges_config = json.loads(edges_config)
|
|
||||||
logging.info("edges config: %s", edges_config)
|
|
||||||
for edge_config in edges_config:
|
|
||||||
edge = self.links[edge_config["token"]]
|
|
||||||
edge.width = edge_config["width"]
|
|
||||||
edge.color = edge_config["color"]
|
|
||||||
edge.redraw()
|
|
||||||
|
|
||||||
# read hidden nodes
|
|
||||||
hidden = config.get("hidden")
|
|
||||||
if hidden:
|
|
||||||
hidden = json.loads(hidden)
|
|
||||||
for _id in hidden:
|
|
||||||
canvas_node = self.canvas_nodes.get(_id)
|
|
||||||
if canvas_node:
|
|
||||||
canvas_node.hide()
|
|
||||||
else:
|
|
||||||
logging.warning("invalid node to hide: %s", _id)
|
|
||||||
|
|
||||||
def create_new_session(self) -> None:
|
def create_new_session(self) -> None:
|
||||||
"""
|
"""
|
||||||
Create a new session
|
Create a new session
|
||||||
"""
|
"""
|
||||||
try:
|
try:
|
||||||
response = self.client.create_session()
|
session = self.client.create_session()
|
||||||
logging.info("created session: %s", response)
|
logger.info("created session: %s", session.id)
|
||||||
self.join_session(response.session_id)
|
self.join_session(session.id)
|
||||||
location_config = self.app.guiconfig.location
|
location_config = self.app.guiconfig.location
|
||||||
self.session.location = SessionLocation(
|
self.session.location = SessionLocation(
|
||||||
x=location_config.x,
|
x=location_config.x,
|
||||||
|
@ -387,7 +342,7 @@ class CoreClient:
|
||||||
session_id = self.session.id
|
session_id = self.session.id
|
||||||
try:
|
try:
|
||||||
response = self.client.delete_session(session_id)
|
response = self.client.delete_session(session_id)
|
||||||
logging.info("deleted session(%s), Result: %s", session_id, response)
|
logger.info("deleted session(%s), Result: %s", session_id, response)
|
||||||
except grpc.RpcError as e:
|
except grpc.RpcError as e:
|
||||||
self.app.show_grpc_exception("Delete Session Error", e)
|
self.app.show_grpc_exception("Delete Session Error", e)
|
||||||
|
|
||||||
|
@ -397,23 +352,21 @@ class CoreClient:
|
||||||
"""
|
"""
|
||||||
try:
|
try:
|
||||||
self.client.connect()
|
self.client.connect()
|
||||||
# get all available services
|
# get current core configurations services/config services
|
||||||
response = self.client.get_services()
|
core_config = self.client.get_config()
|
||||||
for service in response.services:
|
self.emane_models = sorted(core_config.emane_models)
|
||||||
|
for service in core_config.services:
|
||||||
group_services = self.services.setdefault(service.group, set())
|
group_services = self.services.setdefault(service.group, set())
|
||||||
group_services.add(service.name)
|
group_services.add(service.name)
|
||||||
# get config service informations
|
for service in core_config.config_services:
|
||||||
response = self.client.get_config_services()
|
self.config_services[service.name] = service
|
||||||
for service in response.services:
|
|
||||||
self.config_services[service.name] = ConfigService.from_proto(service)
|
|
||||||
group_services = self.config_services_groups.setdefault(
|
group_services = self.config_services_groups.setdefault(
|
||||||
service.group, set()
|
service.group, set()
|
||||||
)
|
)
|
||||||
group_services.add(service.name)
|
group_services.add(service.name)
|
||||||
# join provided session, create new session, or show dialog to select an
|
# join provided session, create new session, or show dialog to select an
|
||||||
# existing session
|
# existing session
|
||||||
response = self.client.get_sessions()
|
sessions = self.client.get_sessions()
|
||||||
sessions = response.sessions
|
|
||||||
if session_id:
|
if session_id:
|
||||||
session_ids = set(x.id for x in sessions)
|
session_ids = set(x.id for x in sessions)
|
||||||
if session_id not in session_ids:
|
if session_id not in session_ids:
|
||||||
|
@ -432,71 +385,50 @@ class CoreClient:
|
||||||
dialog = SessionsDialog(self.app, True)
|
dialog = SessionsDialog(self.app, True)
|
||||||
dialog.show()
|
dialog.show()
|
||||||
except grpc.RpcError as e:
|
except grpc.RpcError as e:
|
||||||
logging.exception("core setup error")
|
logger.exception("core setup error")
|
||||||
self.app.show_grpc_exception("Setup Error", e, blocking=True)
|
self.app.show_grpc_exception("Setup Error", e, blocking=True)
|
||||||
self.app.close()
|
self.app.close()
|
||||||
|
|
||||||
def edit_node(self, core_node: Node) -> None:
|
def edit_node(self, core_node: Node) -> None:
|
||||||
try:
|
try:
|
||||||
position = core_node.position.to_proto()
|
self.client.move_node(
|
||||||
self.client.edit_node(
|
self.session.id, core_node.id, core_node.position, source=GUI_SOURCE
|
||||||
self.session.id, core_node.id, position, source=GUI_SOURCE
|
|
||||||
)
|
)
|
||||||
except grpc.RpcError as e:
|
except grpc.RpcError as e:
|
||||||
self.app.show_grpc_exception("Edit Node Error", e)
|
self.app.show_grpc_exception("Edit Node Error", e)
|
||||||
|
|
||||||
def send_servers(self) -> None:
|
def get_links(self, definition: bool = False) -> List[Link]:
|
||||||
for server in self.servers.values():
|
if not definition:
|
||||||
self.client.add_session_server(self.session.id, server.name, server.address)
|
self.ifaces_manager.set_macs([x.link for x in self.links.values()])
|
||||||
|
|
||||||
def start_session(self) -> Tuple[bool, List[str]]:
|
|
||||||
self.ifaces_manager.set_macs([x.link for x in self.links.values()])
|
|
||||||
nodes = [x.to_proto() for x in self.session.nodes.values()]
|
|
||||||
links = []
|
links = []
|
||||||
asymmetric_links = []
|
|
||||||
for edge in self.links.values():
|
for edge in self.links.values():
|
||||||
link = edge.link
|
link = edge.link
|
||||||
if link.iface1 and not link.iface1.mac:
|
if not definition:
|
||||||
link.iface1.mac = self.ifaces_manager.next_mac()
|
if link.iface1 and not link.iface1.mac:
|
||||||
if link.iface2 and not link.iface2.mac:
|
link.iface1.mac = self.ifaces_manager.next_mac()
|
||||||
link.iface2.mac = self.ifaces_manager.next_mac()
|
if link.iface2 and not link.iface2.mac:
|
||||||
links.append(link.to_proto())
|
link.iface2.mac = self.ifaces_manager.next_mac()
|
||||||
|
links.append(link)
|
||||||
if edge.asymmetric_link:
|
if edge.asymmetric_link:
|
||||||
asymmetric_links.append(edge.asymmetric_link.to_proto())
|
links.append(edge.asymmetric_link)
|
||||||
wlan_configs = self.get_wlan_configs_proto()
|
return links
|
||||||
mobility_configs = self.get_mobility_configs_proto()
|
|
||||||
emane_model_configs = self.get_emane_model_configs_proto()
|
def start_session(self, definition: bool = False) -> Tuple[bool, List[str]]:
|
||||||
hooks = [x.to_proto() for x in self.session.hooks.values()]
|
self.session.links = self.get_links(definition)
|
||||||
service_configs = self.get_service_configs_proto()
|
self.session.metadata = self.get_metadata()
|
||||||
file_configs = self.get_service_file_configs_proto()
|
self.session.servers.clear()
|
||||||
config_service_configs = self.get_config_service_configs_proto()
|
for server in self.servers.values():
|
||||||
emane_config = to_dict(self.session.emane_config)
|
self.session.servers.append(Server(name=server.name, host=server.address))
|
||||||
result = False
|
result = False
|
||||||
exceptions = []
|
exceptions = []
|
||||||
try:
|
try:
|
||||||
self.send_servers()
|
result, exceptions = self.client.start_session(self.session, definition)
|
||||||
response = self.client.start_session(
|
logger.info(
|
||||||
|
"start session(%s) definition(%s), result: %s",
|
||||||
self.session.id,
|
self.session.id,
|
||||||
nodes,
|
definition,
|
||||||
links,
|
result,
|
||||||
self.session.location.to_proto(),
|
|
||||||
hooks,
|
|
||||||
emane_config,
|
|
||||||
emane_model_configs,
|
|
||||||
wlan_configs,
|
|
||||||
mobility_configs,
|
|
||||||
service_configs,
|
|
||||||
file_configs,
|
|
||||||
asymmetric_links,
|
|
||||||
config_service_configs,
|
|
||||||
)
|
)
|
||||||
logging.info(
|
|
||||||
"start session(%s), result: %s", self.session.id, response.result
|
|
||||||
)
|
|
||||||
if response.result:
|
|
||||||
self.set_metadata()
|
|
||||||
result = response.result
|
|
||||||
exceptions = response.exceptions
|
|
||||||
except grpc.RpcError as e:
|
except grpc.RpcError as e:
|
||||||
self.app.show_grpc_exception("Start Session Error", e)
|
self.app.show_grpc_exception("Start Session Error", e)
|
||||||
return result, exceptions
|
return result, exceptions
|
||||||
|
@ -506,9 +438,8 @@ class CoreClient:
|
||||||
session_id = self.session.id
|
session_id = self.session.id
|
||||||
result = False
|
result = False
|
||||||
try:
|
try:
|
||||||
response = self.client.stop_session(session_id)
|
result = self.client.stop_session(session_id)
|
||||||
logging.info("stopped session(%s), result: %s", session_id, response)
|
logger.info("stopped session(%s), result: %s", session_id, result)
|
||||||
result = response.result
|
|
||||||
except grpc.RpcError as e:
|
except grpc.RpcError as e:
|
||||||
self.app.show_grpc_exception("Stop Session Error", e)
|
self.app.show_grpc_exception("Stop Session Error", e)
|
||||||
return result
|
return result
|
||||||
|
@ -522,7 +453,7 @@ class CoreClient:
|
||||||
self.mobility_players[node.id] = mobility_player
|
self.mobility_players[node.id] = mobility_player
|
||||||
mobility_player.show()
|
mobility_player.show()
|
||||||
|
|
||||||
def set_metadata(self) -> None:
|
def get_metadata(self) -> Dict[str, str]:
|
||||||
# create canvas data
|
# create canvas data
|
||||||
canvas_config = self.app.manager.get_metadata()
|
canvas_config = self.app.manager.get_metadata()
|
||||||
canvas_config = json.dumps(canvas_config)
|
canvas_config = json.dumps(canvas_config)
|
||||||
|
@ -548,11 +479,9 @@ class CoreClient:
|
||||||
hidden = json.dumps(hidden)
|
hidden = json.dumps(hidden)
|
||||||
|
|
||||||
# save metadata
|
# save metadata
|
||||||
metadata = dict(
|
return dict(
|
||||||
canvas=canvas_config, shapes=shapes, edges=edges_config, hidden=hidden
|
canvas=canvas_config, shapes=shapes, edges=edges_config, hidden=hidden
|
||||||
)
|
)
|
||||||
response = self.client.set_session_metadata(self.session.id, metadata)
|
|
||||||
logging.debug("set session metadata %s, result: %s", metadata, response)
|
|
||||||
|
|
||||||
def launch_terminal(self, node_id: int) -> None:
|
def launch_terminal(self, node_id: int) -> None:
|
||||||
try:
|
try:
|
||||||
|
@ -564,9 +493,9 @@ class CoreClient:
|
||||||
parent=self.app,
|
parent=self.app,
|
||||||
)
|
)
|
||||||
return
|
return
|
||||||
response = self.client.get_node_terminal(self.session.id, node_id)
|
node_term = self.client.get_node_terminal(self.session.id, node_id)
|
||||||
cmd = f"{terminal} {response.terminal} &"
|
cmd = f"{terminal} {node_term} &"
|
||||||
logging.info("launching terminal %s", cmd)
|
logger.info("launching terminal %s", cmd)
|
||||||
os.system(cmd)
|
os.system(cmd)
|
||||||
except grpc.RpcError as e:
|
except grpc.RpcError as e:
|
||||||
self.app.show_grpc_exception("Node Terminal Error", e)
|
self.app.show_grpc_exception("Node Terminal Error", e)
|
||||||
|
@ -574,189 +503,82 @@ class CoreClient:
|
||||||
def get_xml_dir(self) -> str:
|
def get_xml_dir(self) -> str:
|
||||||
return str(self.session.file.parent) if self.session.file else str(XMLS_PATH)
|
return str(self.session.file.parent) if self.session.file else str(XMLS_PATH)
|
||||||
|
|
||||||
def save_xml(self, file_path: str = None) -> None:
|
def save_xml(self, file_path: Path = None) -> bool:
|
||||||
"""
|
"""
|
||||||
Save core session as to an xml file
|
Save core session as to an xml file
|
||||||
"""
|
"""
|
||||||
if not file_path and not self.session.file:
|
if not file_path and not self.session.file:
|
||||||
logging.error("trying to save xml for session with no file")
|
logger.error("trying to save xml for session with no file")
|
||||||
return
|
return False
|
||||||
if not file_path:
|
if not file_path:
|
||||||
file_path = str(self.session.file)
|
file_path = self.session.file
|
||||||
|
result = False
|
||||||
try:
|
try:
|
||||||
if not self.is_runtime():
|
if not self.is_runtime():
|
||||||
logging.debug("Send session data to the daemon")
|
logger.debug("sending session data to the daemon")
|
||||||
self.send_data()
|
result, exceptions = self.start_session(definition=True)
|
||||||
response = self.client.save_xml(self.session.id, file_path)
|
if not result:
|
||||||
logging.info("saved xml file %s, result: %s", file_path, response)
|
message = "\n".join(exceptions)
|
||||||
|
self.app.show_exception_data(
|
||||||
|
"Session Definition Exception",
|
||||||
|
"Failed to define session",
|
||||||
|
message,
|
||||||
|
)
|
||||||
|
self.client.save_xml(self.session.id, str(file_path))
|
||||||
|
if self.session.file != file_path:
|
||||||
|
self.session.file = file_path
|
||||||
|
self.update_session_title()
|
||||||
|
logger.info("saved xml file %s", file_path)
|
||||||
|
result = True
|
||||||
except grpc.RpcError as e:
|
except grpc.RpcError as e:
|
||||||
self.app.show_grpc_exception("Save XML Error", e)
|
self.app.show_grpc_exception("Save XML Error", e)
|
||||||
|
return result
|
||||||
|
|
||||||
def open_xml(self, file_path: str) -> None:
|
def open_xml(self, file_path: Path) -> None:
|
||||||
"""
|
"""
|
||||||
Open core xml
|
Open core xml
|
||||||
"""
|
"""
|
||||||
try:
|
try:
|
||||||
response = self._client.open_xml(file_path)
|
result, session_id = self._client.open_xml(file_path)
|
||||||
logging.info("open xml file %s, response: %s", file_path, response)
|
logger.info(
|
||||||
self.join_session(response.session_id)
|
"open xml file %s, result(%s) session(%s)",
|
||||||
|
file_path,
|
||||||
|
result,
|
||||||
|
session_id,
|
||||||
|
)
|
||||||
|
self.join_session(session_id)
|
||||||
except grpc.RpcError as e:
|
except grpc.RpcError as e:
|
||||||
self.app.show_grpc_exception("Open XML Error", e)
|
self.app.show_grpc_exception("Open XML Error", e)
|
||||||
|
|
||||||
def get_node_service(self, node_id: int, service_name: str) -> NodeServiceData:
|
def get_node_service(self, node_id: int, service_name: str) -> NodeServiceData:
|
||||||
response = self.client.get_node_service(self.session.id, node_id, service_name)
|
node_service = self.client.get_node_service(
|
||||||
logging.debug(
|
self.session.id, node_id, service_name
|
||||||
"get node(%s) %s service, response: %s", node_id, service_name, response
|
|
||||||
)
|
)
|
||||||
return NodeServiceData.from_proto(response.service)
|
logger.debug(
|
||||||
|
"get node(%s) service(%s): %s", node_id, service_name, node_service
|
||||||
def set_node_service(
|
|
||||||
self,
|
|
||||||
node_id: int,
|
|
||||||
service_name: str,
|
|
||||||
dirs: List[str],
|
|
||||||
files: List[str],
|
|
||||||
startups: List[str],
|
|
||||||
validations: List[str],
|
|
||||||
shutdowns: List[str],
|
|
||||||
) -> NodeServiceData:
|
|
||||||
response = self.client.set_node_service(
|
|
||||||
self.session.id,
|
|
||||||
node_id,
|
|
||||||
service_name,
|
|
||||||
directories=dirs,
|
|
||||||
files=files,
|
|
||||||
startup=startups,
|
|
||||||
validate=validations,
|
|
||||||
shutdown=shutdowns,
|
|
||||||
)
|
)
|
||||||
logging.info(
|
return node_service
|
||||||
"Set %s service for node(%s), files: %s, Startup: %s, "
|
|
||||||
"Validation: %s, Shutdown: %s, Result: %s",
|
|
||||||
service_name,
|
|
||||||
node_id,
|
|
||||||
files,
|
|
||||||
startups,
|
|
||||||
validations,
|
|
||||||
shutdowns,
|
|
||||||
response,
|
|
||||||
)
|
|
||||||
response = self.client.get_node_service(self.session.id, node_id, service_name)
|
|
||||||
return NodeServiceData.from_proto(response.service)
|
|
||||||
|
|
||||||
def get_node_service_file(
|
def get_node_service_file(
|
||||||
self, node_id: int, service_name: str, file_name: str
|
self, node_id: int, service_name: str, file_name: str
|
||||||
) -> str:
|
) -> str:
|
||||||
response = self.client.get_node_service_file(
|
data = self.client.get_node_service_file(
|
||||||
self.session.id, node_id, service_name, file_name
|
self.session.id, node_id, service_name, file_name
|
||||||
)
|
)
|
||||||
logging.debug(
|
logger.debug(
|
||||||
"get service file for node(%s), service: %s, file: %s, result: %s",
|
"get service file for node(%s), service: %s, file: %s, data: %s",
|
||||||
node_id,
|
|
||||||
service_name,
|
|
||||||
file_name,
|
|
||||||
response,
|
|
||||||
)
|
|
||||||
return response.data
|
|
||||||
|
|
||||||
def set_node_service_file(
|
|
||||||
self, node_id: int, service_name: str, file_name: str, data: str
|
|
||||||
) -> None:
|
|
||||||
response = self.client.set_node_service_file(
|
|
||||||
self.session.id, node_id, service_name, file_name, data
|
|
||||||
)
|
|
||||||
logging.info(
|
|
||||||
"set node(%s) service file, service: %s, file: %s, data: %s, result: %s",
|
|
||||||
node_id,
|
node_id,
|
||||||
service_name,
|
service_name,
|
||||||
file_name,
|
file_name,
|
||||||
data,
|
data,
|
||||||
response,
|
|
||||||
)
|
)
|
||||||
|
return data
|
||||||
def create_nodes_and_links(self) -> None:
|
|
||||||
"""
|
|
||||||
create nodes and links that have not been created yet
|
|
||||||
"""
|
|
||||||
self.client.set_session_state(self.session.id, SessionState.DEFINITION.value)
|
|
||||||
for node in self.session.nodes.values():
|
|
||||||
response = self.client.add_node(
|
|
||||||
self.session.id, node.to_proto(), source=GUI_SOURCE
|
|
||||||
)
|
|
||||||
logging.debug("created node: %s", response)
|
|
||||||
asymmetric_links = []
|
|
||||||
for edge in self.links.values():
|
|
||||||
self.add_link(edge.link)
|
|
||||||
if edge.asymmetric_link:
|
|
||||||
asymmetric_links.append(edge.asymmetric_link)
|
|
||||||
for link in asymmetric_links:
|
|
||||||
self.add_link(link)
|
|
||||||
|
|
||||||
def send_data(self) -> None:
|
|
||||||
"""
|
|
||||||
Send to daemon all session info, but don't start the session
|
|
||||||
"""
|
|
||||||
self.send_servers()
|
|
||||||
self.create_nodes_and_links()
|
|
||||||
for config_proto in self.get_wlan_configs_proto():
|
|
||||||
self.client.set_wlan_config(
|
|
||||||
self.session.id, config_proto.node_id, config_proto.config
|
|
||||||
)
|
|
||||||
for config_proto in self.get_mobility_configs_proto():
|
|
||||||
self.client.set_mobility_config(
|
|
||||||
self.session.id, config_proto.node_id, config_proto.config
|
|
||||||
)
|
|
||||||
for config_proto in self.get_service_configs_proto():
|
|
||||||
self.client.set_node_service(
|
|
||||||
self.session.id,
|
|
||||||
config_proto.node_id,
|
|
||||||
config_proto.service,
|
|
||||||
config_proto.files,
|
|
||||||
config_proto.directories,
|
|
||||||
config_proto.startup,
|
|
||||||
config_proto.validate,
|
|
||||||
config_proto.shutdown,
|
|
||||||
)
|
|
||||||
for config_proto in self.get_service_file_configs_proto():
|
|
||||||
self.client.set_node_service_file(
|
|
||||||
self.session.id,
|
|
||||||
config_proto.node_id,
|
|
||||||
config_proto.service,
|
|
||||||
config_proto.file,
|
|
||||||
config_proto.data,
|
|
||||||
)
|
|
||||||
for hook in self.session.hooks.values():
|
|
||||||
self.client.add_hook(
|
|
||||||
self.session.id, hook.state.value, hook.file, hook.data
|
|
||||||
)
|
|
||||||
for config_proto in self.get_emane_model_configs_proto():
|
|
||||||
self.client.set_emane_model_config(
|
|
||||||
self.session.id,
|
|
||||||
config_proto.node_id,
|
|
||||||
config_proto.model,
|
|
||||||
config_proto.config,
|
|
||||||
config_proto.iface_id,
|
|
||||||
)
|
|
||||||
config = to_dict(self.session.emane_config)
|
|
||||||
self.client.set_emane_config(self.session.id, config)
|
|
||||||
location = self.session.location
|
|
||||||
self.client.set_session_location(
|
|
||||||
self.session.id,
|
|
||||||
location.x,
|
|
||||||
location.y,
|
|
||||||
location.z,
|
|
||||||
location.lat,
|
|
||||||
location.lon,
|
|
||||||
location.alt,
|
|
||||||
location.scale,
|
|
||||||
)
|
|
||||||
self.set_metadata()
|
|
||||||
|
|
||||||
def close(self) -> None:
|
def close(self) -> None:
|
||||||
"""
|
"""
|
||||||
Clean ups when done using grpc
|
Clean ups when done using grpc
|
||||||
"""
|
"""
|
||||||
logging.debug("close grpc")
|
logger.debug("close grpc")
|
||||||
self.client.close()
|
self.client.close()
|
||||||
|
|
||||||
def next_node_id(self) -> int:
|
def next_node_id(self) -> int:
|
||||||
|
@ -783,11 +605,11 @@ class CoreClient:
|
||||||
image = "ubuntu:latest"
|
image = "ubuntu:latest"
|
||||||
emane = None
|
emane = None
|
||||||
if node_type == NodeType.EMANE:
|
if node_type == NodeType.EMANE:
|
||||||
if not self.session.emane_models:
|
if not self.emane_models:
|
||||||
dialog = EmaneInstallDialog(self.app)
|
dialog = EmaneInstallDialog(self.app)
|
||||||
dialog.show()
|
dialog.show()
|
||||||
return
|
return
|
||||||
emane = self.session.emane_models[0]
|
emane = self.emane_models[0]
|
||||||
name = f"emane{node_id}"
|
name = f"emane{node_id}"
|
||||||
elif node_type == NodeType.WIRELESS_LAN:
|
elif node_type == NodeType.WIRELESS_LAN:
|
||||||
name = f"wlan{node_id}"
|
name = f"wlan{node_id}"
|
||||||
|
@ -806,13 +628,13 @@ class CoreClient:
|
||||||
)
|
)
|
||||||
if nutils.is_custom(node):
|
if nutils.is_custom(node):
|
||||||
services = nutils.get_custom_services(self.app.guiconfig, model)
|
services = nutils.get_custom_services(self.app.guiconfig, model)
|
||||||
node.services = set(services)
|
node.config_services = set(services)
|
||||||
# assign default services to CORE node
|
# assign default services to CORE node
|
||||||
else:
|
else:
|
||||||
services = self.session.default_services.get(model)
|
services = self.session.default_services.get(model)
|
||||||
if services:
|
if services:
|
||||||
node.services = services.copy()
|
node.config_services = set(services)
|
||||||
logging.info(
|
logger.info(
|
||||||
"add node(%s) to session(%s), coordinates(%s, %s)",
|
"add node(%s) to session(%s), coordinates(%s, %s)",
|
||||||
node.name,
|
node.name,
|
||||||
self.session.id,
|
self.session.id,
|
||||||
|
@ -850,7 +672,7 @@ class CoreClient:
|
||||||
dst_iface_id = edge.link.iface2.id
|
dst_iface_id = edge.link.iface2.id
|
||||||
self.iface_to_edge[(dst_node.id, dst_iface_id)] = edge
|
self.iface_to_edge[(dst_node.id, dst_iface_id)] = edge
|
||||||
|
|
||||||
def get_wlan_configs_proto(self) -> List[wlan_pb2.WlanConfig]:
|
def get_wlan_configs(self) -> List[Tuple[int, Dict[str, str]]]:
|
||||||
configs = []
|
configs = []
|
||||||
for node in self.session.nodes.values():
|
for node in self.session.nodes.values():
|
||||||
if node.type != NodeType.WIRELESS_LAN:
|
if node.type != NodeType.WIRELESS_LAN:
|
||||||
|
@ -858,11 +680,10 @@ class CoreClient:
|
||||||
if not node.wlan_config:
|
if not node.wlan_config:
|
||||||
continue
|
continue
|
||||||
config = ConfigOption.to_dict(node.wlan_config)
|
config = ConfigOption.to_dict(node.wlan_config)
|
||||||
wlan_config = wlan_pb2.WlanConfig(node_id=node.id, config=config)
|
configs.append((node.id, config))
|
||||||
configs.append(wlan_config)
|
|
||||||
return configs
|
return configs
|
||||||
|
|
||||||
def get_mobility_configs_proto(self) -> List[mobility_pb2.MobilityConfig]:
|
def get_mobility_configs(self) -> List[Tuple[int, Dict[str, str]]]:
|
||||||
configs = []
|
configs = []
|
||||||
for node in self.session.nodes.values():
|
for node in self.session.nodes.values():
|
||||||
if not nutils.is_mobility(node):
|
if not nutils.is_mobility(node):
|
||||||
|
@ -870,27 +691,24 @@ class CoreClient:
|
||||||
if not node.mobility_config:
|
if not node.mobility_config:
|
||||||
continue
|
continue
|
||||||
config = ConfigOption.to_dict(node.mobility_config)
|
config = ConfigOption.to_dict(node.mobility_config)
|
||||||
mobility_config = mobility_pb2.MobilityConfig(
|
configs.append((node.id, config))
|
||||||
node_id=node.id, config=config
|
|
||||||
)
|
|
||||||
configs.append(mobility_config)
|
|
||||||
return configs
|
return configs
|
||||||
|
|
||||||
def get_emane_model_configs_proto(self) -> List[emane_pb2.EmaneModelConfig]:
|
def get_emane_model_configs(self) -> List[EmaneModelConfig]:
|
||||||
configs = []
|
configs = []
|
||||||
for node in self.session.nodes.values():
|
for node in self.session.nodes.values():
|
||||||
for key, config in node.emane_model_configs.items():
|
for key, config in node.emane_model_configs.items():
|
||||||
model, iface_id = key
|
model, iface_id = key
|
||||||
config = ConfigOption.to_dict(config)
|
# config = ConfigOption.to_dict(config)
|
||||||
if iface_id is None:
|
if iface_id is None:
|
||||||
iface_id = -1
|
iface_id = -1
|
||||||
config_proto = emane_pb2.EmaneModelConfig(
|
config = EmaneModelConfig(
|
||||||
node_id=node.id, iface_id=iface_id, model=model, config=config
|
node_id=node.id, model=model, iface_id=iface_id, config=config
|
||||||
)
|
)
|
||||||
configs.append(config_proto)
|
configs.append(config)
|
||||||
return configs
|
return configs
|
||||||
|
|
||||||
def get_service_configs_proto(self) -> List[services_pb2.ServiceConfig]:
|
def get_service_configs(self) -> List[ServiceConfig]:
|
||||||
configs = []
|
configs = []
|
||||||
for node in self.session.nodes.values():
|
for node in self.session.nodes.values():
|
||||||
if not nutils.is_container(node):
|
if not nutils.is_container(node):
|
||||||
|
@ -898,19 +716,19 @@ class CoreClient:
|
||||||
if not node.service_configs:
|
if not node.service_configs:
|
||||||
continue
|
continue
|
||||||
for name, config in node.service_configs.items():
|
for name, config in node.service_configs.items():
|
||||||
config_proto = services_pb2.ServiceConfig(
|
config = ServiceConfig(
|
||||||
node_id=node.id,
|
node_id=node.id,
|
||||||
service=name,
|
service=name,
|
||||||
directories=config.dirs,
|
|
||||||
files=config.configs,
|
files=config.configs,
|
||||||
|
directories=config.dirs,
|
||||||
startup=config.startup,
|
startup=config.startup,
|
||||||
validate=config.validate,
|
validate=config.validate,
|
||||||
shutdown=config.shutdown,
|
shutdown=config.shutdown,
|
||||||
)
|
)
|
||||||
configs.append(config_proto)
|
configs.append(config)
|
||||||
return configs
|
return configs
|
||||||
|
|
||||||
def get_service_file_configs_proto(self) -> List[services_pb2.ServiceFileConfig]:
|
def get_service_file_configs(self) -> List[ServiceFileConfig]:
|
||||||
configs = []
|
configs = []
|
||||||
for node in self.session.nodes.values():
|
for node in self.session.nodes.values():
|
||||||
if not nutils.is_container(node):
|
if not nutils.is_container(node):
|
||||||
|
@ -919,10 +737,8 @@ class CoreClient:
|
||||||
continue
|
continue
|
||||||
for service, file_configs in node.service_file_configs.items():
|
for service, file_configs in node.service_file_configs.items():
|
||||||
for file, data in file_configs.items():
|
for file, data in file_configs.items():
|
||||||
config_proto = services_pb2.ServiceFileConfig(
|
config = ServiceFileConfig(node.id, service, file, data)
|
||||||
node_id=node.id, service=service, file=file, data=data
|
configs.append(config)
|
||||||
)
|
|
||||||
configs.append(config_proto)
|
|
||||||
return configs
|
return configs
|
||||||
|
|
||||||
def get_config_service_configs_proto(
|
def get_config_service_configs_proto(
|
||||||
|
@ -945,39 +761,37 @@ class CoreClient:
|
||||||
return config_service_protos
|
return config_service_protos
|
||||||
|
|
||||||
def run(self, node_id: int) -> str:
|
def run(self, node_id: int) -> str:
|
||||||
logging.info("running node(%s) cmd: %s", node_id, self.observer)
|
logger.info("running node(%s) cmd: %s", node_id, self.observer)
|
||||||
return self.client.node_command(self.session.id, node_id, self.observer).output
|
_, output = self.client.node_command(self.session.id, node_id, self.observer)
|
||||||
|
return output
|
||||||
|
|
||||||
def get_wlan_config(self, node_id: int) -> Dict[str, ConfigOption]:
|
def get_wlan_config(self, node_id: int) -> Dict[str, ConfigOption]:
|
||||||
response = self.client.get_wlan_config(self.session.id, node_id)
|
config = self.client.get_wlan_config(self.session.id, node_id)
|
||||||
config = response.config
|
logger.debug(
|
||||||
logging.debug(
|
|
||||||
"get wlan configuration from node %s, result configuration: %s",
|
"get wlan configuration from node %s, result configuration: %s",
|
||||||
node_id,
|
node_id,
|
||||||
config,
|
config,
|
||||||
)
|
)
|
||||||
return ConfigOption.from_dict(config)
|
return config
|
||||||
|
|
||||||
def get_mobility_config(self, node_id: int) -> Dict[str, ConfigOption]:
|
def get_mobility_config(self, node_id: int) -> Dict[str, ConfigOption]:
|
||||||
response = self.client.get_mobility_config(self.session.id, node_id)
|
config = self.client.get_mobility_config(self.session.id, node_id)
|
||||||
config = response.config
|
logger.debug(
|
||||||
logging.debug(
|
|
||||||
"get mobility config from node %s, result configuration: %s",
|
"get mobility config from node %s, result configuration: %s",
|
||||||
node_id,
|
node_id,
|
||||||
config,
|
config,
|
||||||
)
|
)
|
||||||
return ConfigOption.from_dict(config)
|
return config
|
||||||
|
|
||||||
def get_emane_model_config(
|
def get_emane_model_config(
|
||||||
self, node_id: int, model: str, iface_id: int = None
|
self, node_id: int, model: str, iface_id: int = None
|
||||||
) -> Dict[str, ConfigOption]:
|
) -> Dict[str, ConfigOption]:
|
||||||
if iface_id is None:
|
if iface_id is None:
|
||||||
iface_id = -1
|
iface_id = -1
|
||||||
response = self.client.get_emane_model_config(
|
config = self.client.get_emane_model_config(
|
||||||
self.session.id, node_id, model, iface_id
|
self.session.id, node_id, model, iface_id
|
||||||
)
|
)
|
||||||
config = response.config
|
logger.debug(
|
||||||
logging.debug(
|
|
||||||
"get emane model config: node id: %s, EMANE model: %s, "
|
"get emane model config: node id: %s, EMANE model: %s, "
|
||||||
"interface: %s, config: %s",
|
"interface: %s, config: %s",
|
||||||
node_id,
|
node_id,
|
||||||
|
@ -985,42 +799,21 @@ class CoreClient:
|
||||||
iface_id,
|
iface_id,
|
||||||
config,
|
config,
|
||||||
)
|
)
|
||||||
return ConfigOption.from_dict(config)
|
return config
|
||||||
|
|
||||||
def execute_script(self, script) -> None:
|
def execute_script(self, script) -> None:
|
||||||
response = self.client.execute_script(script)
|
session_id = self.client.execute_script(script)
|
||||||
logging.info("execute python script %s", response)
|
logger.info("execute python script %s", session_id)
|
||||||
if response.session_id != -1:
|
if session_id != -1:
|
||||||
self.join_session(response.session_id)
|
self.join_session(session_id)
|
||||||
|
|
||||||
def add_link(self, link: Link) -> None:
|
def add_link(self, link: Link) -> None:
|
||||||
iface1 = link.iface1.to_proto() if link.iface1 else None
|
result, _, _ = self.client.add_link(self.session.id, link, source=GUI_SOURCE)
|
||||||
iface2 = link.iface2.to_proto() if link.iface2 else None
|
logger.debug("added link: %s", result)
|
||||||
options = link.options.to_proto() if link.options else None
|
if not result:
|
||||||
response = self.client.add_link(
|
logger.error("error adding link: %s", link)
|
||||||
self.session.id,
|
|
||||||
link.node1_id,
|
|
||||||
link.node2_id,
|
|
||||||
iface1,
|
|
||||||
iface2,
|
|
||||||
options,
|
|
||||||
source=GUI_SOURCE,
|
|
||||||
)
|
|
||||||
logging.debug("added link: %s", response)
|
|
||||||
if not response.result:
|
|
||||||
logging.error("error adding link: %s", link)
|
|
||||||
|
|
||||||
def edit_link(self, link: Link) -> None:
|
def edit_link(self, link: Link) -> None:
|
||||||
iface1_id = link.iface1.id if link.iface1 else None
|
result = self.client.edit_link(self.session.id, link, source=GUI_SOURCE)
|
||||||
iface2_id = link.iface2.id if link.iface2 else None
|
if not result:
|
||||||
response = self.client.edit_link(
|
logger.error("error editing link: %s", link)
|
||||||
self.session.id,
|
|
||||||
link.node1_id,
|
|
||||||
link.node2_id,
|
|
||||||
link.options.to_proto(),
|
|
||||||
iface1_id,
|
|
||||||
iface2_id,
|
|
||||||
source=GUI_SOURCE,
|
|
||||||
)
|
|
||||||
if not response.result:
|
|
||||||
logging.error("error editing link: %s", link)
|
|
||||||
|
|
File diff suppressed because it is too large
Load diff
|
@ -24,7 +24,7 @@ class SizeAndScaleDialog(Dialog):
|
||||||
super().__init__(app, "Canvas Size and Scale")
|
super().__init__(app, "Canvas Size and Scale")
|
||||||
self.manager: CanvasManager = self.app.manager
|
self.manager: CanvasManager = self.app.manager
|
||||||
self.section_font: font.Font = font.Font(weight=font.BOLD)
|
self.section_font: font.Font = font.Font(weight=font.BOLD)
|
||||||
width, height = self.manager.current_dimensions
|
width, height = self.manager.current().current_dimensions
|
||||||
self.pixel_width: tk.IntVar = tk.IntVar(value=width)
|
self.pixel_width: tk.IntVar = tk.IntVar(value=width)
|
||||||
self.pixel_height: tk.IntVar = tk.IntVar(value=height)
|
self.pixel_height: tk.IntVar = tk.IntVar(value=height)
|
||||||
location = self.app.core.session.location
|
location = self.app.core.session.location
|
||||||
|
@ -189,7 +189,7 @@ class SizeAndScaleDialog(Dialog):
|
||||||
|
|
||||||
def click_apply(self) -> None:
|
def click_apply(self) -> None:
|
||||||
width, height = self.pixel_width.get(), self.pixel_height.get()
|
width, height = self.pixel_width.get(), self.pixel_height.get()
|
||||||
self.manager.redraw_canvases((width, height))
|
self.manager.redraw_canvas((width, height))
|
||||||
location = self.app.core.session.location
|
location = self.app.core.session.location
|
||||||
location.x = self.x.get()
|
location.x = self.x.get()
|
||||||
location.y = self.y.get()
|
location.y = self.y.get()
|
||||||
|
|
|
@ -13,6 +13,8 @@ from core.gui.graph.graph import CanvasGraph
|
||||||
from core.gui.themes import PADX, PADY
|
from core.gui.themes import PADX, PADY
|
||||||
from core.gui.widgets import image_chooser
|
from core.gui.widgets import image_chooser
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.gui.app import Application
|
from core.gui.app import Application
|
||||||
|
|
||||||
|
@ -167,5 +169,5 @@ class CanvasWallpaperDialog(Dialog):
|
||||||
try:
|
try:
|
||||||
self.canvas.set_wallpaper(filename)
|
self.canvas.set_wallpaper(filename)
|
||||||
except FileNotFoundError:
|
except FileNotFoundError:
|
||||||
logging.error("invalid background: %s", filename)
|
logger.error("invalid background: %s", filename)
|
||||||
self.destroy()
|
self.destroy()
|
||||||
|
|
|
@ -18,6 +18,8 @@ from core.gui.dialogs.dialog import Dialog
|
||||||
from core.gui.themes import FRAME_PAD, PADX, PADY
|
from core.gui.themes import FRAME_PAD, PADX, PADY
|
||||||
from core.gui.widgets import CodeText, ConfigFrame, ListboxScroll
|
from core.gui.widgets import CodeText, ConfigFrame, ListboxScroll
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.gui.app import Application
|
from core.gui.app import Application
|
||||||
from core.gui.coreclient import CoreClient
|
from core.gui.coreclient import CoreClient
|
||||||
|
@ -73,7 +75,7 @@ class ConfigServiceConfigDialog(Dialog):
|
||||||
|
|
||||||
def load(self) -> None:
|
def load(self) -> None:
|
||||||
try:
|
try:
|
||||||
self.core.create_nodes_and_links()
|
self.core.start_session(definition=True)
|
||||||
service = self.core.config_services[self.service_name]
|
service = self.core.config_services[self.service_name]
|
||||||
self.dependencies = service.dependencies[:]
|
self.dependencies = service.dependencies[:]
|
||||||
self.executables = service.executables[:]
|
self.executables = service.executables[:]
|
||||||
|
@ -86,18 +88,18 @@ class ConfigServiceConfigDialog(Dialog):
|
||||||
self.validation_time = service.validation_timer
|
self.validation_time = service.validation_timer
|
||||||
self.validation_period.set(service.validation_period)
|
self.validation_period.set(service.validation_period)
|
||||||
|
|
||||||
response = self.core.client.get_config_service_defaults(self.service_name)
|
defaults = self.core.client.get_config_service_defaults(self.service_name)
|
||||||
self.original_service_files = response.templates
|
self.original_service_files = defaults.templates
|
||||||
self.temp_service_files = dict(self.original_service_files)
|
self.temp_service_files = dict(self.original_service_files)
|
||||||
self.modes = sorted(x.name for x in response.modes)
|
self.modes = sorted(defaults.modes)
|
||||||
self.mode_configs = {x.name: x.config for x in response.modes}
|
self.mode_configs = defaults.modes
|
||||||
self.config = ConfigOption.from_dict(response.config)
|
self.config = ConfigOption.from_dict(defaults.config)
|
||||||
self.default_config = {x.name: x.value for x in self.config.values()}
|
self.default_config = {x.name: x.value for x in self.config.values()}
|
||||||
service_config = self.node.config_service_configs.get(self.service_name)
|
service_config = self.node.config_service_configs.get(self.service_name)
|
||||||
if service_config:
|
if service_config:
|
||||||
for key, value in service_config.config.items():
|
for key, value in service_config.config.items():
|
||||||
self.config[key].value = value
|
self.config[key].value = value
|
||||||
logging.info("default config: %s", self.default_config)
|
logger.info("default config: %s", self.default_config)
|
||||||
for file, data in service_config.templates.items():
|
for file, data in service_config.templates.items():
|
||||||
self.modified_files.add(file)
|
self.modified_files.add(file)
|
||||||
self.temp_service_files[file] = data
|
self.temp_service_files[file] = data
|
||||||
|
@ -181,7 +183,7 @@ class ConfigServiceConfigDialog(Dialog):
|
||||||
self.modes_combobox.bind("<<ComboboxSelected>>", self.handle_mode_changed)
|
self.modes_combobox.bind("<<ComboboxSelected>>", self.handle_mode_changed)
|
||||||
self.modes_combobox.grid(row=0, column=1, sticky=tk.EW, pady=PADY)
|
self.modes_combobox.grid(row=0, column=1, sticky=tk.EW, pady=PADY)
|
||||||
|
|
||||||
logging.info("config service config: %s", self.config)
|
logger.info("config service config: %s", self.config)
|
||||||
self.config_frame = ConfigFrame(tab, self.app, self.config)
|
self.config_frame = ConfigFrame(tab, self.app, self.config)
|
||||||
self.config_frame.draw_config()
|
self.config_frame.draw_config()
|
||||||
self.config_frame.grid(sticky=tk.NSEW, pady=PADY)
|
self.config_frame.grid(sticky=tk.NSEW, pady=PADY)
|
||||||
|
@ -308,9 +310,9 @@ class ConfigServiceConfigDialog(Dialog):
|
||||||
current_listbox.itemconfig(current_listbox.curselection()[0], bg="")
|
current_listbox.itemconfig(current_listbox.curselection()[0], bg="")
|
||||||
self.destroy()
|
self.destroy()
|
||||||
return
|
return
|
||||||
service_config = self.node.config_service_configs.get(self.service_name)
|
service_config = self.node.config_service_configs.setdefault(
|
||||||
if not service_config:
|
self.service_name, ConfigServiceData()
|
||||||
service_config = ConfigServiceData()
|
)
|
||||||
if self.config_frame:
|
if self.config_frame:
|
||||||
self.config_frame.parse_config()
|
self.config_frame.parse_config()
|
||||||
service_config.config = {x.name: x.value for x in self.config.values()}
|
service_config.config = {x.name: x.value for x in self.config.values()}
|
||||||
|
@ -328,7 +330,7 @@ class ConfigServiceConfigDialog(Dialog):
|
||||||
def handle_mode_changed(self, event: tk.Event) -> None:
|
def handle_mode_changed(self, event: tk.Event) -> None:
|
||||||
mode = self.modes_combobox.get()
|
mode = self.modes_combobox.get()
|
||||||
config = self.mode_configs[mode]
|
config = self.mode_configs[mode]
|
||||||
logging.info("mode config: %s", config)
|
logger.info("mode config: %s", config)
|
||||||
self.config_frame.set_values(config)
|
self.config_frame.set_values(config)
|
||||||
|
|
||||||
def update_template_file_data(self, event: tk.Event) -> None:
|
def update_template_file_data(self, event: tk.Event) -> None:
|
||||||
|
@ -350,7 +352,7 @@ class ConfigServiceConfigDialog(Dialog):
|
||||||
|
|
||||||
def click_defaults(self) -> None:
|
def click_defaults(self) -> None:
|
||||||
self.node.config_service_configs.pop(self.service_name, None)
|
self.node.config_service_configs.pop(self.service_name, None)
|
||||||
logging.info(
|
logger.info(
|
||||||
"cleared config service config: %s", self.node.config_service_configs
|
"cleared config service config: %s", self.node.config_service_configs
|
||||||
)
|
)
|
||||||
self.temp_service_files = dict(self.original_service_files)
|
self.temp_service_files = dict(self.original_service_files)
|
||||||
|
@ -358,7 +360,7 @@ class ConfigServiceConfigDialog(Dialog):
|
||||||
self.template_text.text.delete(1.0, "end")
|
self.template_text.text.delete(1.0, "end")
|
||||||
self.template_text.text.insert("end", self.temp_service_files[filename])
|
self.template_text.text.insert("end", self.temp_service_files[filename])
|
||||||
if self.config_frame:
|
if self.config_frame:
|
||||||
logging.info("resetting defaults: %s", self.default_config)
|
logger.info("resetting defaults: %s", self.default_config)
|
||||||
self.config_frame.set_values(self.default_config)
|
self.config_frame.set_values(self.default_config)
|
||||||
|
|
||||||
def click_copy(self) -> None:
|
def click_copy(self) -> None:
|
||||||
|
|
|
@ -13,6 +13,8 @@ from core.gui.nodeutils import NodeDraw
|
||||||
from core.gui.themes import FRAME_PAD, PADX, PADY
|
from core.gui.themes import FRAME_PAD, PADX, PADY
|
||||||
from core.gui.widgets import CheckboxList, ListboxScroll, image_chooser
|
from core.gui.widgets import CheckboxList, ListboxScroll, image_chooser
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.gui.app import Application
|
from core.gui.app import Application
|
||||||
|
|
||||||
|
@ -209,7 +211,7 @@ class CustomNodesDialog(Dialog):
|
||||||
name, node_draw.image_file, list(node_draw.services)
|
name, node_draw.image_file, list(node_draw.services)
|
||||||
)
|
)
|
||||||
self.app.guiconfig.nodes.append(custom_node)
|
self.app.guiconfig.nodes.append(custom_node)
|
||||||
logging.info("saving custom nodes: %s", self.app.guiconfig.nodes)
|
logger.info("saving custom nodes: %s", self.app.guiconfig.nodes)
|
||||||
self.app.save_config()
|
self.app.save_config()
|
||||||
self.destroy()
|
self.destroy()
|
||||||
|
|
||||||
|
@ -219,7 +221,7 @@ class CustomNodesDialog(Dialog):
|
||||||
image_file = str(Path(self.image_file).absolute())
|
image_file = str(Path(self.image_file).absolute())
|
||||||
custom_node = CustomNode(name, image_file, list(self.services))
|
custom_node = CustomNode(name, image_file, list(self.services))
|
||||||
node_draw = NodeDraw.from_custom(custom_node)
|
node_draw = NodeDraw.from_custom(custom_node)
|
||||||
logging.info(
|
logger.info(
|
||||||
"created new custom node (%s), image file (%s), services: (%s)",
|
"created new custom node (%s), image file (%s), services: (%s)",
|
||||||
name,
|
name,
|
||||||
image_file,
|
image_file,
|
||||||
|
@ -239,7 +241,7 @@ class CustomNodesDialog(Dialog):
|
||||||
node_draw.image_file = str(Path(self.image_file).absolute())
|
node_draw.image_file = str(Path(self.image_file).absolute())
|
||||||
node_draw.image = self.image
|
node_draw.image = self.image
|
||||||
node_draw.services = set(self.services)
|
node_draw.services = set(self.services)
|
||||||
logging.debug(
|
logger.debug(
|
||||||
"edit custom node (%s), image: (%s), services (%s)",
|
"edit custom node (%s), image: (%s), services (%s)",
|
||||||
node_draw.model,
|
node_draw.model,
|
||||||
node_draw.image_file,
|
node_draw.image_file,
|
||||||
|
|
|
@ -19,40 +19,6 @@ if TYPE_CHECKING:
|
||||||
from core.gui.app import Application
|
from core.gui.app import Application
|
||||||
|
|
||||||
|
|
||||||
class GlobalEmaneDialog(Dialog):
|
|
||||||
def __init__(self, master: tk.BaseWidget, app: "Application") -> None:
|
|
||||||
super().__init__(app, "EMANE Configuration", master=master)
|
|
||||||
self.config_frame: Optional[ConfigFrame] = None
|
|
||||||
self.enabled: bool = not self.app.core.is_runtime()
|
|
||||||
self.draw()
|
|
||||||
|
|
||||||
def draw(self) -> None:
|
|
||||||
self.top.columnconfigure(0, weight=1)
|
|
||||||
self.top.rowconfigure(0, weight=1)
|
|
||||||
session = self.app.core.session
|
|
||||||
self.config_frame = ConfigFrame(
|
|
||||||
self.top, self.app, session.emane_config, self.enabled
|
|
||||||
)
|
|
||||||
self.config_frame.draw_config()
|
|
||||||
self.config_frame.grid(sticky=tk.NSEW, pady=PADY)
|
|
||||||
self.draw_buttons()
|
|
||||||
|
|
||||||
def draw_buttons(self) -> None:
|
|
||||||
frame = ttk.Frame(self.top)
|
|
||||||
frame.grid(sticky=tk.EW)
|
|
||||||
for i in range(2):
|
|
||||||
frame.columnconfigure(i, weight=1)
|
|
||||||
state = tk.NORMAL if self.enabled else tk.DISABLED
|
|
||||||
button = ttk.Button(frame, text="Apply", command=self.click_apply, state=state)
|
|
||||||
button.grid(row=0, column=0, sticky=tk.EW, padx=PADX)
|
|
||||||
button = ttk.Button(frame, text="Cancel", command=self.destroy)
|
|
||||||
button.grid(row=0, column=1, sticky=tk.EW)
|
|
||||||
|
|
||||||
def click_apply(self) -> None:
|
|
||||||
self.config_frame.parse_config()
|
|
||||||
self.destroy()
|
|
||||||
|
|
||||||
|
|
||||||
class EmaneModelDialog(Dialog):
|
class EmaneModelDialog(Dialog):
|
||||||
def __init__(
|
def __init__(
|
||||||
self,
|
self,
|
||||||
|
@ -115,7 +81,7 @@ class EmaneConfigDialog(Dialog):
|
||||||
self.radiovar: tk.IntVar = tk.IntVar()
|
self.radiovar: tk.IntVar = tk.IntVar()
|
||||||
self.radiovar.set(1)
|
self.radiovar.set(1)
|
||||||
self.emane_models: List[str] = [
|
self.emane_models: List[str] = [
|
||||||
x.split("_")[1] for x in self.app.core.session.emane_models
|
x.split("_")[1] for x in self.app.core.emane_models
|
||||||
]
|
]
|
||||||
model = self.node.emane.split("_")[1]
|
model = self.node.emane.split("_")[1]
|
||||||
self.emane_model: tk.StringVar = tk.StringVar(value=model)
|
self.emane_model: tk.StringVar = tk.StringVar(value=model)
|
||||||
|
@ -179,9 +145,7 @@ class EmaneConfigDialog(Dialog):
|
||||||
def draw_emane_buttons(self) -> None:
|
def draw_emane_buttons(self) -> None:
|
||||||
frame = ttk.Frame(self.top)
|
frame = ttk.Frame(self.top)
|
||||||
frame.grid(sticky=tk.EW, pady=PADY)
|
frame.grid(sticky=tk.EW, pady=PADY)
|
||||||
for i in range(2):
|
frame.columnconfigure(0, weight=1)
|
||||||
frame.columnconfigure(i, weight=1)
|
|
||||||
|
|
||||||
image = images.from_enum(ImageEnum.EDITNODE, width=images.BUTTON_SIZE)
|
image = images.from_enum(ImageEnum.EDITNODE, width=images.BUTTON_SIZE)
|
||||||
self.emane_model_button = ttk.Button(
|
self.emane_model_button = ttk.Button(
|
||||||
frame,
|
frame,
|
||||||
|
@ -191,18 +155,7 @@ class EmaneConfigDialog(Dialog):
|
||||||
command=self.click_model_config,
|
command=self.click_model_config,
|
||||||
)
|
)
|
||||||
self.emane_model_button.image = image
|
self.emane_model_button.image = image
|
||||||
self.emane_model_button.grid(row=0, column=0, padx=PADX, sticky=tk.EW)
|
self.emane_model_button.grid(padx=PADX, sticky=tk.EW)
|
||||||
|
|
||||||
image = images.from_enum(ImageEnum.EDITNODE, width=images.BUTTON_SIZE)
|
|
||||||
button = ttk.Button(
|
|
||||||
frame,
|
|
||||||
text="EMANE options",
|
|
||||||
image=image,
|
|
||||||
compound=tk.RIGHT,
|
|
||||||
command=self.click_emane_config,
|
|
||||||
)
|
|
||||||
button.image = image
|
|
||||||
button.grid(row=0, column=1, sticky=tk.EW)
|
|
||||||
|
|
||||||
def draw_apply_and_cancel(self) -> None:
|
def draw_apply_and_cancel(self) -> None:
|
||||||
frame = ttk.Frame(self.top)
|
frame = ttk.Frame(self.top)
|
||||||
|
@ -215,10 +168,6 @@ class EmaneConfigDialog(Dialog):
|
||||||
button = ttk.Button(frame, text="Cancel", command=self.destroy)
|
button = ttk.Button(frame, text="Cancel", command=self.destroy)
|
||||||
button.grid(row=0, column=1, sticky=tk.EW)
|
button.grid(row=0, column=1, sticky=tk.EW)
|
||||||
|
|
||||||
def click_emane_config(self) -> None:
|
|
||||||
dialog = GlobalEmaneDialog(self, self.app)
|
|
||||||
dialog.show()
|
|
||||||
|
|
||||||
def click_model_config(self) -> None:
|
def click_model_config(self) -> None:
|
||||||
"""
|
"""
|
||||||
draw emane model configuration
|
draw emane model configuration
|
||||||
|
|
|
@ -7,6 +7,8 @@ from core.gui.appconfig import SCRIPT_PATH
|
||||||
from core.gui.dialogs.dialog import Dialog
|
from core.gui.dialogs.dialog import Dialog
|
||||||
from core.gui.themes import FRAME_PAD, PADX
|
from core.gui.themes import FRAME_PAD, PADX
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.gui.app import Application
|
from core.gui.app import Application
|
||||||
|
|
||||||
|
@ -83,6 +85,6 @@ class ExecutePythonDialog(Dialog):
|
||||||
def script_execute(self) -> None:
|
def script_execute(self) -> None:
|
||||||
file = self.file_entry.get()
|
file = self.file_entry.get()
|
||||||
options = self.option_entry.get()
|
options = self.option_entry.get()
|
||||||
logging.info("Execute %s with options %s", file, options)
|
logger.info("Execute %s with options %s", file, options)
|
||||||
self.app.core.execute_script(file)
|
self.app.core.execute_script(file)
|
||||||
self.destroy()
|
self.destroy()
|
||||||
|
|
|
@ -6,6 +6,8 @@ from typing import TYPE_CHECKING, Optional
|
||||||
from core.gui.dialogs.dialog import Dialog
|
from core.gui.dialogs.dialog import Dialog
|
||||||
from core.gui.themes import FRAME_PAD, PADX, PADY
|
from core.gui.themes import FRAME_PAD, PADX, PADY
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.gui.app import Application
|
from core.gui.app import Application
|
||||||
|
|
||||||
|
@ -139,8 +141,8 @@ class FindDialog(Dialog):
|
||||||
_x, _y, _, _ = canvas_node.canvas.bbox(canvas_node.id)
|
_x, _y, _, _ = canvas_node.canvas.bbox(canvas_node.id)
|
||||||
oid = canvas_node.canvas.find_withtag("rectangle")
|
oid = canvas_node.canvas.find_withtag("rectangle")
|
||||||
x0, y0, x1, y1 = canvas_node.canvas.bbox(oid[0])
|
x0, y0, x1, y1 = canvas_node.canvas.bbox(oid[0])
|
||||||
logging.debug("Dist to most left: %s", abs(x0 - _x))
|
logger.debug("Dist to most left: %s", abs(x0 - _x))
|
||||||
logging.debug("White canvas width: %s", abs(x0 - x1))
|
logger.debug("White canvas width: %s", abs(x0 - x1))
|
||||||
|
|
||||||
# calculate the node's location
|
# calculate the node's location
|
||||||
# (as fractions of white canvas's width and height)
|
# (as fractions of white canvas's width and height)
|
||||||
|
|
|
@ -23,6 +23,8 @@ class IpConfigDialog(Dialog):
|
||||||
self.ip4_listbox: Optional[ListboxScroll] = None
|
self.ip4_listbox: Optional[ListboxScroll] = None
|
||||||
self.ip6_entry: Optional[ttk.Entry] = None
|
self.ip6_entry: Optional[ttk.Entry] = None
|
||||||
self.ip6_listbox: Optional[ListboxScroll] = None
|
self.ip6_listbox: Optional[ListboxScroll] = None
|
||||||
|
self.enable_ip4 = tk.BooleanVar(value=self.app.guiconfig.ips.enable_ip4)
|
||||||
|
self.enable_ip6 = tk.BooleanVar(value=self.app.guiconfig.ips.enable_ip6)
|
||||||
self.draw()
|
self.draw()
|
||||||
|
|
||||||
def draw(self) -> None:
|
def draw(self) -> None:
|
||||||
|
@ -36,10 +38,19 @@ class IpConfigDialog(Dialog):
|
||||||
frame.rowconfigure(0, weight=1)
|
frame.rowconfigure(0, weight=1)
|
||||||
frame.grid(sticky=tk.NSEW, pady=PADY)
|
frame.grid(sticky=tk.NSEW, pady=PADY)
|
||||||
|
|
||||||
|
ip4_checkbox = ttk.Checkbutton(
|
||||||
|
frame, text="Enable IP4?", variable=self.enable_ip4
|
||||||
|
)
|
||||||
|
ip4_checkbox.grid(row=0, column=0, sticky=tk.EW)
|
||||||
|
ip6_checkbox = ttk.Checkbutton(
|
||||||
|
frame, text="Enable IP6?", variable=self.enable_ip6
|
||||||
|
)
|
||||||
|
ip6_checkbox.grid(row=0, column=1, sticky=tk.EW)
|
||||||
|
|
||||||
ip4_frame = ttk.LabelFrame(frame, text="IPv4", padding=FRAME_PAD)
|
ip4_frame = ttk.LabelFrame(frame, text="IPv4", padding=FRAME_PAD)
|
||||||
ip4_frame.columnconfigure(0, weight=1)
|
ip4_frame.columnconfigure(0, weight=1)
|
||||||
ip4_frame.rowconfigure(0, weight=1)
|
ip4_frame.rowconfigure(1, weight=1)
|
||||||
ip4_frame.grid(row=0, column=0, stick="nsew")
|
ip4_frame.grid(row=1, column=0, stick=tk.NSEW)
|
||||||
self.ip4_listbox = ListboxScroll(ip4_frame)
|
self.ip4_listbox = ListboxScroll(ip4_frame)
|
||||||
self.ip4_listbox.listbox.bind("<<ListboxSelect>>", self.select_ip4)
|
self.ip4_listbox.listbox.bind("<<ListboxSelect>>", self.select_ip4)
|
||||||
self.ip4_listbox.grid(sticky=tk.NSEW, pady=PADY)
|
self.ip4_listbox.grid(sticky=tk.NSEW, pady=PADY)
|
||||||
|
@ -63,7 +74,7 @@ class IpConfigDialog(Dialog):
|
||||||
ip6_frame = ttk.LabelFrame(frame, text="IPv6", padding=FRAME_PAD)
|
ip6_frame = ttk.LabelFrame(frame, text="IPv6", padding=FRAME_PAD)
|
||||||
ip6_frame.columnconfigure(0, weight=1)
|
ip6_frame.columnconfigure(0, weight=1)
|
||||||
ip6_frame.rowconfigure(0, weight=1)
|
ip6_frame.rowconfigure(0, weight=1)
|
||||||
ip6_frame.grid(row=0, column=1, stick="nsew")
|
ip6_frame.grid(row=1, column=1, stick=tk.NSEW)
|
||||||
self.ip6_listbox = ListboxScroll(ip6_frame)
|
self.ip6_listbox = ListboxScroll(ip6_frame)
|
||||||
self.ip6_listbox.listbox.bind("<<ListboxSelect>>", self.select_ip6)
|
self.ip6_listbox.listbox.bind("<<ListboxSelect>>", self.select_ip6)
|
||||||
self.ip6_listbox.grid(sticky=tk.NSEW, pady=PADY)
|
self.ip6_listbox.grid(sticky=tk.NSEW, pady=PADY)
|
||||||
|
@ -86,7 +97,7 @@ class IpConfigDialog(Dialog):
|
||||||
|
|
||||||
# draw buttons
|
# draw buttons
|
||||||
frame = ttk.Frame(self.top)
|
frame = ttk.Frame(self.top)
|
||||||
frame.grid(stick="ew")
|
frame.grid(stick=tk.EW)
|
||||||
for i in range(2):
|
for i in range(2):
|
||||||
frame.columnconfigure(i, weight=1)
|
frame.columnconfigure(i, weight=1)
|
||||||
button = ttk.Button(frame, text="Save", command=self.click_save)
|
button = ttk.Button(frame, text="Save", command=self.click_save)
|
||||||
|
@ -142,10 +153,18 @@ class IpConfigDialog(Dialog):
|
||||||
ip6 = self.ip6_listbox.listbox.get(index)
|
ip6 = self.ip6_listbox.listbox.get(index)
|
||||||
ip6s.append(ip6)
|
ip6s.append(ip6)
|
||||||
ip_config = self.app.guiconfig.ips
|
ip_config = self.app.guiconfig.ips
|
||||||
ip_config.ip4 = self.ip4
|
ip_changed = False
|
||||||
ip_config.ip6 = self.ip6
|
if ip_config.ip4 != self.ip4:
|
||||||
|
ip_config.ip4 = self.ip4
|
||||||
|
ip_changed = True
|
||||||
|
if ip_config.ip6 != self.ip6:
|
||||||
|
ip_config.ip6 = self.ip6
|
||||||
|
ip_changed = True
|
||||||
ip_config.ip4s = ip4s
|
ip_config.ip4s = ip4s
|
||||||
ip_config.ip6s = ip6s
|
ip_config.ip6s = ip6s
|
||||||
self.app.core.ifaces_manager.update_ips(self.ip4, self.ip6)
|
ip_config.enable_ip4 = self.enable_ip4.get()
|
||||||
|
ip_config.enable_ip6 = self.enable_ip6.get()
|
||||||
|
if ip_changed:
|
||||||
|
self.app.core.ifaces_manager.update_ips(self.ip4, self.ip6)
|
||||||
self.app.save_config()
|
self.app.save_config()
|
||||||
self.destroy()
|
self.destroy()
|
||||||
|
|
|
@ -134,7 +134,7 @@ class MobilityPlayerDialog(Dialog):
|
||||||
session_id = self.app.core.session.id
|
session_id = self.app.core.session.id
|
||||||
try:
|
try:
|
||||||
self.app.core.client.mobility_action(
|
self.app.core.client.mobility_action(
|
||||||
session_id, self.node.id, MobilityAction.START.value
|
session_id, self.node.id, MobilityAction.START
|
||||||
)
|
)
|
||||||
except grpc.RpcError as e:
|
except grpc.RpcError as e:
|
||||||
self.app.show_grpc_exception("Mobility Error", e)
|
self.app.show_grpc_exception("Mobility Error", e)
|
||||||
|
@ -144,7 +144,7 @@ class MobilityPlayerDialog(Dialog):
|
||||||
session_id = self.app.core.session.id
|
session_id = self.app.core.session.id
|
||||||
try:
|
try:
|
||||||
self.app.core.client.mobility_action(
|
self.app.core.client.mobility_action(
|
||||||
session_id, self.node.id, MobilityAction.PAUSE.value
|
session_id, self.node.id, MobilityAction.PAUSE
|
||||||
)
|
)
|
||||||
except grpc.RpcError as e:
|
except grpc.RpcError as e:
|
||||||
self.app.show_grpc_exception("Mobility Error", e)
|
self.app.show_grpc_exception("Mobility Error", e)
|
||||||
|
@ -154,7 +154,7 @@ class MobilityPlayerDialog(Dialog):
|
||||||
session_id = self.app.core.session.id
|
session_id = self.app.core.session.id
|
||||||
try:
|
try:
|
||||||
self.app.core.client.mobility_action(
|
self.app.core.client.mobility_action(
|
||||||
session_id, self.node.id, MobilityAction.STOP.value
|
session_id, self.node.id, MobilityAction.STOP
|
||||||
)
|
)
|
||||||
except grpc.RpcError as e:
|
except grpc.RpcError as e:
|
||||||
self.app.show_grpc_exception("Mobility Error", e)
|
self.app.show_grpc_exception("Mobility Error", e)
|
||||||
|
|
|
@ -17,6 +17,8 @@ from core.gui.dialogs.emaneconfig import EmaneModelDialog
|
||||||
from core.gui.themes import FRAME_PAD, PADX, PADY
|
from core.gui.themes import FRAME_PAD, PADX, PADY
|
||||||
from core.gui.widgets import ListboxScroll, image_chooser
|
from core.gui.widgets import ListboxScroll, image_chooser
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.gui.app import Application
|
from core.gui.app import Application
|
||||||
from core.gui.graph.node import CanvasNode
|
from core.gui.graph.node import CanvasNode
|
||||||
|
@ -260,17 +262,17 @@ class NodeConfigDialog(Dialog):
|
||||||
row += 1
|
row += 1
|
||||||
|
|
||||||
if nutils.is_rj45(self.node):
|
if nutils.is_rj45(self.node):
|
||||||
response = self.app.core.client.get_ifaces()
|
ifaces = self.app.core.client.get_ifaces()
|
||||||
logging.debug("host machine available interfaces: %s", response)
|
logger.debug("host machine available interfaces: %s", ifaces)
|
||||||
ifaces = ListboxScroll(frame)
|
ifaces_scroll = ListboxScroll(frame)
|
||||||
ifaces.listbox.config(state=state)
|
ifaces_scroll.listbox.config(state=state)
|
||||||
ifaces.grid(
|
ifaces_scroll.grid(
|
||||||
row=row, column=0, columnspan=2, sticky=tk.EW, padx=PADX, pady=PADY
|
row=row, column=0, columnspan=2, sticky=tk.EW, padx=PADX, pady=PADY
|
||||||
)
|
)
|
||||||
for inf in sorted(response.ifaces[:]):
|
for inf in sorted(ifaces):
|
||||||
ifaces.listbox.insert(tk.END, inf)
|
ifaces_scroll.listbox.insert(tk.END, inf)
|
||||||
row += 1
|
row += 1
|
||||||
ifaces.listbox.bind("<<ListboxSelect>>", self.iface_select)
|
ifaces_scroll.listbox.bind("<<ListboxSelect>>", self.iface_select)
|
||||||
|
|
||||||
# interfaces
|
# interfaces
|
||||||
if self.canvas_node.ifaces:
|
if self.canvas_node.ifaces:
|
||||||
|
@ -296,10 +298,9 @@ class NodeConfigDialog(Dialog):
|
||||||
emane_node = self.canvas_node.has_emane_link(iface.id)
|
emane_node = self.canvas_node.has_emane_link(iface.id)
|
||||||
if emane_node:
|
if emane_node:
|
||||||
emane_model = emane_node.emane.split("_")[1]
|
emane_model = emane_node.emane.split("_")[1]
|
||||||
|
command = partial(self.click_emane_config, emane_model, iface.id)
|
||||||
button = ttk.Button(
|
button = ttk.Button(
|
||||||
tab,
|
tab, text=f"Configure EMANE {emane_model}", command=command
|
||||||
text=f"Configure EMANE {emane_model}",
|
|
||||||
command=lambda: self.click_emane_config(emane_model, iface.id),
|
|
||||||
)
|
)
|
||||||
button.grid(row=row, sticky=tk.EW, columnspan=3, pady=PADY)
|
button.grid(row=row, sticky=tk.EW, columnspan=3, pady=PADY)
|
||||||
row += 1
|
row += 1
|
||||||
|
@ -365,6 +366,7 @@ class NodeConfigDialog(Dialog):
|
||||||
button.grid(row=0, column=1, sticky=tk.EW)
|
button.grid(row=0, column=1, sticky=tk.EW)
|
||||||
|
|
||||||
def click_emane_config(self, emane_model: str, iface_id: int) -> None:
|
def click_emane_config(self, emane_model: str, iface_id: int) -> None:
|
||||||
|
logger.info("configuring emane: %s - %s", emane_model, iface_id)
|
||||||
dialog = EmaneModelDialog(self, self.app, self.node, emane_model, iface_id)
|
dialog = EmaneModelDialog(self, self.app, self.node, emane_model, iface_id)
|
||||||
dialog.show()
|
dialog.show()
|
||||||
|
|
||||||
|
|
|
@ -12,6 +12,8 @@ from core.gui.dialogs.dialog import Dialog
|
||||||
from core.gui.themes import FRAME_PAD, PADX, PADY
|
from core.gui.themes import FRAME_PAD, PADX, PADY
|
||||||
from core.gui.widgets import CheckboxList, ListboxScroll
|
from core.gui.widgets import CheckboxList, ListboxScroll
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.gui.app import Application
|
from core.gui.app import Application
|
||||||
|
|
||||||
|
@ -29,6 +31,7 @@ class NodeConfigServiceDialog(Dialog):
|
||||||
if services is None:
|
if services is None:
|
||||||
services = set(node.config_services)
|
services = set(node.config_services)
|
||||||
self.current_services: Set[str] = services
|
self.current_services: Set[str] = services
|
||||||
|
self.protocol("WM_DELETE_WINDOW", self.click_cancel)
|
||||||
self.draw()
|
self.draw()
|
||||||
|
|
||||||
def draw(self) -> None:
|
def draw(self) -> None:
|
||||||
|
@ -100,6 +103,7 @@ class NodeConfigServiceDialog(Dialog):
|
||||||
self.current_services.add(name)
|
self.current_services.add(name)
|
||||||
elif not var.get() and name in self.current_services:
|
elif not var.get() and name in self.current_services:
|
||||||
self.current_services.remove(name)
|
self.current_services.remove(name)
|
||||||
|
self.node.config_service_configs.pop(name, None)
|
||||||
self.draw_current_services()
|
self.draw_current_services()
|
||||||
self.node.config_services = self.current_services.copy()
|
self.node.config_services = self.current_services.copy()
|
||||||
|
|
||||||
|
@ -131,7 +135,7 @@ class NodeConfigServiceDialog(Dialog):
|
||||||
|
|
||||||
def click_save(self) -> None:
|
def click_save(self) -> None:
|
||||||
self.node.config_services = self.current_services.copy()
|
self.node.config_services = self.current_services.copy()
|
||||||
logging.info("saved node config services: %s", self.node.config_services)
|
logger.info("saved node config services: %s", self.node.config_services)
|
||||||
self.destroy()
|
self.destroy()
|
||||||
|
|
||||||
def click_cancel(self) -> None:
|
def click_cancel(self) -> None:
|
||||||
|
@ -144,6 +148,7 @@ class NodeConfigServiceDialog(Dialog):
|
||||||
service = self.current.listbox.get(cur[0])
|
service = self.current.listbox.get(cur[0])
|
||||||
self.current.listbox.delete(cur[0])
|
self.current.listbox.delete(cur[0])
|
||||||
self.current_services.remove(service)
|
self.current_services.remove(service)
|
||||||
|
self.node.config_service_configs.pop(service, None)
|
||||||
for checkbutton in self.services.frame.winfo_children():
|
for checkbutton in self.services.frame.winfo_children():
|
||||||
if checkbutton["text"] == service:
|
if checkbutton["text"] == service:
|
||||||
checkbutton.invoke()
|
checkbutton.invoke()
|
||||||
|
|
|
@ -17,7 +17,7 @@ if TYPE_CHECKING:
|
||||||
|
|
||||||
class NodeServiceDialog(Dialog):
|
class NodeServiceDialog(Dialog):
|
||||||
def __init__(self, app: "Application", node: Node) -> None:
|
def __init__(self, app: "Application", node: Node) -> None:
|
||||||
title = f"{node.name} Services"
|
title = f"{node.name} Services (Deprecated)"
|
||||||
super().__init__(app, title)
|
super().__init__(app, title)
|
||||||
self.node: Node = node
|
self.node: Node = node
|
||||||
self.groups: Optional[ListboxScroll] = None
|
self.groups: Optional[ListboxScroll] = None
|
||||||
|
@ -25,6 +25,7 @@ class NodeServiceDialog(Dialog):
|
||||||
self.current: Optional[ListboxScroll] = None
|
self.current: Optional[ListboxScroll] = None
|
||||||
services = set(node.services)
|
services = set(node.services)
|
||||||
self.current_services: Set[str] = services
|
self.current_services: Set[str] = services
|
||||||
|
self.protocol("WM_DELETE_WINDOW", self.click_cancel)
|
||||||
self.draw()
|
self.draw()
|
||||||
|
|
||||||
def draw(self) -> None:
|
def draw(self) -> None:
|
||||||
|
@ -77,7 +78,7 @@ class NodeServiceDialog(Dialog):
|
||||||
button.grid(row=0, column=1, sticky=tk.EW, padx=PADX)
|
button.grid(row=0, column=1, sticky=tk.EW, padx=PADX)
|
||||||
button = ttk.Button(frame, text="Remove", command=self.click_remove)
|
button = ttk.Button(frame, text="Remove", command=self.click_remove)
|
||||||
button.grid(row=0, column=2, sticky=tk.EW, padx=PADX)
|
button.grid(row=0, column=2, sticky=tk.EW, padx=PADX)
|
||||||
button = ttk.Button(frame, text="Cancel", command=self.destroy)
|
button = ttk.Button(frame, text="Cancel", command=self.click_cancel)
|
||||||
button.grid(row=0, column=3, sticky=tk.EW)
|
button.grid(row=0, column=3, sticky=tk.EW)
|
||||||
|
|
||||||
# trigger group change
|
# trigger group change
|
||||||
|
@ -98,6 +99,8 @@ class NodeServiceDialog(Dialog):
|
||||||
self.current_services.add(name)
|
self.current_services.add(name)
|
||||||
elif not var.get() and name in self.current_services:
|
elif not var.get() and name in self.current_services:
|
||||||
self.current_services.remove(name)
|
self.current_services.remove(name)
|
||||||
|
self.node.service_configs.pop(name, None)
|
||||||
|
self.node.service_file_configs.pop(name, None)
|
||||||
self.current.listbox.delete(0, tk.END)
|
self.current.listbox.delete(0, tk.END)
|
||||||
for name in sorted(self.current_services):
|
for name in sorted(self.current_services):
|
||||||
self.current.listbox.insert(tk.END, name)
|
self.current.listbox.insert(tk.END, name)
|
||||||
|
@ -125,6 +128,9 @@ class NodeServiceDialog(Dialog):
|
||||||
"Service Configuration", "Select a service to configure", parent=self
|
"Service Configuration", "Select a service to configure", parent=self
|
||||||
)
|
)
|
||||||
|
|
||||||
|
def click_cancel(self) -> None:
|
||||||
|
self.destroy()
|
||||||
|
|
||||||
def click_save(self) -> None:
|
def click_save(self) -> None:
|
||||||
self.node.services = self.current_services.copy()
|
self.node.services = self.current_services.copy()
|
||||||
self.destroy()
|
self.destroy()
|
||||||
|
@ -135,6 +141,8 @@ class NodeServiceDialog(Dialog):
|
||||||
service = self.current.listbox.get(cur[0])
|
service = self.current.listbox.get(cur[0])
|
||||||
self.current.listbox.delete(cur[0])
|
self.current.listbox.delete(cur[0])
|
||||||
self.current_services.remove(service)
|
self.current_services.remove(service)
|
||||||
|
self.node.service_configs.pop(service, None)
|
||||||
|
self.node.service_file_configs.pop(service, None)
|
||||||
for checkbutton in self.services.frame.winfo_children():
|
for checkbutton in self.services.frame.winfo_children():
|
||||||
if checkbutton["text"] == service:
|
if checkbutton["text"] == service:
|
||||||
checkbutton.invoke()
|
checkbutton.invoke()
|
||||||
|
|
|
@ -9,6 +9,8 @@ from core.gui.dialogs.dialog import Dialog
|
||||||
from core.gui.themes import FRAME_PAD, PADX, PADY, scale_fonts
|
from core.gui.themes import FRAME_PAD, PADX, PADY, scale_fonts
|
||||||
from core.gui.validation import LARGEST_SCALE, SMALLEST_SCALE
|
from core.gui.validation import LARGEST_SCALE, SMALLEST_SCALE
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.gui.app import Application
|
from core.gui.app import Application
|
||||||
|
|
||||||
|
@ -102,7 +104,7 @@ class PreferencesDialog(Dialog):
|
||||||
|
|
||||||
def theme_change(self, event: tk.Event) -> None:
|
def theme_change(self, event: tk.Event) -> None:
|
||||||
theme = self.theme.get()
|
theme = self.theme.get()
|
||||||
logging.info("changing theme: %s", theme)
|
logger.info("changing theme: %s", theme)
|
||||||
self.app.style.theme_use(theme)
|
self.app.style.theme_use(theme)
|
||||||
|
|
||||||
def click_save(self) -> None:
|
def click_save(self) -> None:
|
||||||
|
|
|
@ -106,10 +106,8 @@ class RunToolDialog(Dialog):
|
||||||
for selection in self.node_list.listbox.curselection():
|
for selection in self.node_list.listbox.curselection():
|
||||||
node_name = self.node_list.listbox.get(selection)
|
node_name = self.node_list.listbox.get(selection)
|
||||||
node_id = self.executable_nodes[node_name]
|
node_id = self.executable_nodes[node_name]
|
||||||
response = self.app.core.client.node_command(
|
_, output = self.app.core.client.node_command(
|
||||||
self.app.core.session.id, node_id, command
|
self.app.core.session.id, node_id, command
|
||||||
)
|
)
|
||||||
self.result.text.insert(
|
self.result.text.insert(tk.END, f"> {node_name} > {command}:\n{output}\n")
|
||||||
tk.END, f"> {node_name} > {command}:\n{response.output}\n"
|
|
||||||
)
|
|
||||||
self.result.text.config(state=tk.DISABLED)
|
self.result.text.config(state=tk.DISABLED)
|
||||||
|
|
|
@ -1,7 +1,7 @@
|
||||||
import logging
|
import logging
|
||||||
import os
|
|
||||||
import tkinter as tk
|
import tkinter as tk
|
||||||
from tkinter import filedialog, ttk
|
from pathlib import Path
|
||||||
|
from tkinter import filedialog, messagebox, ttk
|
||||||
from typing import TYPE_CHECKING, Dict, List, Optional, Set, Tuple
|
from typing import TYPE_CHECKING, Dict, List, Optional, Set, Tuple
|
||||||
|
|
||||||
import grpc
|
import grpc
|
||||||
|
@ -15,6 +15,8 @@ from core.gui.images import ImageEnum
|
||||||
from core.gui.themes import FRAME_PAD, PADX, PADY
|
from core.gui.themes import FRAME_PAD, PADX, PADY
|
||||||
from core.gui.widgets import CodeText, ListboxScroll
|
from core.gui.widgets import CodeText, ListboxScroll
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.gui.app import Application
|
from core.gui.app import Application
|
||||||
from core.gui.coreclient import CoreClient
|
from core.gui.coreclient import CoreClient
|
||||||
|
@ -26,7 +28,7 @@ class ServiceConfigDialog(Dialog):
|
||||||
def __init__(
|
def __init__(
|
||||||
self, master: tk.BaseWidget, app: "Application", service_name: str, node: Node
|
self, master: tk.BaseWidget, app: "Application", service_name: str, node: Node
|
||||||
) -> None:
|
) -> None:
|
||||||
title = f"{service_name} Service"
|
title = f"{service_name} Service (Deprecated)"
|
||||||
super().__init__(app, title, master=master)
|
super().__init__(app, title, master=master)
|
||||||
self.core: "CoreClient" = app.core
|
self.core: "CoreClient" = app.core
|
||||||
self.node: Node = node
|
self.node: Node = node
|
||||||
|
@ -76,7 +78,7 @@ class ServiceConfigDialog(Dialog):
|
||||||
|
|
||||||
def load(self) -> None:
|
def load(self) -> None:
|
||||||
try:
|
try:
|
||||||
self.app.core.create_nodes_and_links()
|
self.core.start_session(definition=True)
|
||||||
default_config = self.app.core.get_node_service(
|
default_config = self.app.core.get_node_service(
|
||||||
self.node.id, self.service_name
|
self.node.id, self.service_name
|
||||||
)
|
)
|
||||||
|
@ -388,7 +390,7 @@ class ServiceConfigDialog(Dialog):
|
||||||
1.0, "end"
|
1.0, "end"
|
||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
logging.debug("file already existed")
|
logger.debug("file already existed")
|
||||||
|
|
||||||
def delete_filename(self) -> None:
|
def delete_filename(self) -> None:
|
||||||
cbb = self.filename_combobox
|
cbb = self.filename_combobox
|
||||||
|
@ -447,36 +449,31 @@ class ServiceConfigDialog(Dialog):
|
||||||
self.current_service_color("")
|
self.current_service_color("")
|
||||||
self.destroy()
|
self.destroy()
|
||||||
return
|
return
|
||||||
|
files = set(self.filenames)
|
||||||
try:
|
if (
|
||||||
if (
|
self.is_custom_command()
|
||||||
self.is_custom_command()
|
or self.has_new_files()
|
||||||
or self.has_new_files()
|
or self.is_custom_directory()
|
||||||
or self.is_custom_directory()
|
):
|
||||||
):
|
startup, validate, shutdown = self.get_commands()
|
||||||
startup, validate, shutdown = self.get_commands()
|
files = set(self.filename_combobox["values"])
|
||||||
config = self.core.set_node_service(
|
service_data = NodeServiceData(
|
||||||
self.node.id,
|
configs=list(files),
|
||||||
self.service_name,
|
dirs=self.temp_directories,
|
||||||
dirs=self.temp_directories,
|
startup=startup,
|
||||||
files=list(self.filename_combobox["values"]),
|
validate=validate,
|
||||||
startups=startup,
|
shutdown=shutdown,
|
||||||
validations=validate,
|
)
|
||||||
shutdowns=shutdown,
|
logger.info("setting service data: %s", service_data)
|
||||||
)
|
self.node.service_configs[self.service_name] = service_data
|
||||||
self.node.service_configs[self.service_name] = config
|
for file in self.modified_files:
|
||||||
for file in self.modified_files:
|
if file not in files:
|
||||||
file_configs = self.node.service_file_configs.setdefault(
|
continue
|
||||||
self.service_name, {}
|
file_configs = self.node.service_file_configs.setdefault(
|
||||||
)
|
self.service_name, {}
|
||||||
file_configs[file] = self.temp_service_files[file]
|
)
|
||||||
# TODO: check if this is really needed
|
file_configs[file] = self.temp_service_files[file]
|
||||||
self.app.core.set_node_service_file(
|
self.current_service_color("green")
|
||||||
self.node.id, self.service_name, file, self.temp_service_files[file]
|
|
||||||
)
|
|
||||||
self.current_service_color("green")
|
|
||||||
except grpc.RpcError as e:
|
|
||||||
self.app.show_grpc_exception("Save Service Config Error", e)
|
|
||||||
self.destroy()
|
self.destroy()
|
||||||
|
|
||||||
def display_service_file_data(self, event: tk.Event) -> None:
|
def display_service_file_data(self, event: tk.Event) -> None:
|
||||||
|
@ -579,11 +576,13 @@ class ServiceConfigDialog(Dialog):
|
||||||
self.directory_entry.insert("end", d)
|
self.directory_entry.insert("end", d)
|
||||||
|
|
||||||
def add_directory(self) -> None:
|
def add_directory(self) -> None:
|
||||||
d = self.directory_entry.get()
|
directory = Path(self.directory_entry.get())
|
||||||
if os.path.isdir(d):
|
if directory.is_absolute():
|
||||||
if d not in self.temp_directories:
|
if str(directory) not in self.temp_directories:
|
||||||
self.dir_list.listbox.insert("end", d)
|
self.dir_list.listbox.insert("end", directory)
|
||||||
self.temp_directories.append(d)
|
self.temp_directories.append(str(directory))
|
||||||
|
else:
|
||||||
|
messagebox.showerror("Add Directory", "Path must be absolute!", parent=self)
|
||||||
|
|
||||||
def remove_directory(self) -> None:
|
def remove_directory(self) -> None:
|
||||||
d = self.directory_entry.get()
|
d = self.directory_entry.get()
|
||||||
|
@ -594,7 +593,7 @@ class ServiceConfigDialog(Dialog):
|
||||||
i = dirs.index(d)
|
i = dirs.index(d)
|
||||||
self.dir_list.listbox.delete(i)
|
self.dir_list.listbox.delete(i)
|
||||||
except ValueError:
|
except ValueError:
|
||||||
logging.debug("directory is not in the list")
|
logger.debug("directory is not in the list")
|
||||||
self.directory_entry.delete(0, "end")
|
self.directory_entry.delete(0, "end")
|
||||||
|
|
||||||
def directory_select(self, event) -> None:
|
def directory_select(self, event) -> None:
|
||||||
|
|
|
@ -1,15 +1,14 @@
|
||||||
import logging
|
import logging
|
||||||
import tkinter as tk
|
import tkinter as tk
|
||||||
from tkinter import ttk
|
from tkinter import ttk
|
||||||
from typing import TYPE_CHECKING, Dict, Optional
|
from typing import TYPE_CHECKING, Optional
|
||||||
|
|
||||||
import grpc
|
|
||||||
|
|
||||||
from core.api.grpc.wrappers import ConfigOption
|
|
||||||
from core.gui.dialogs.dialog import Dialog
|
from core.gui.dialogs.dialog import Dialog
|
||||||
from core.gui.themes import PADX, PADY
|
from core.gui.themes import PADX, PADY
|
||||||
from core.gui.widgets import ConfigFrame
|
from core.gui.widgets import ConfigFrame
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.gui.app import Application
|
from core.gui.app import Application
|
||||||
|
|
||||||
|
@ -19,25 +18,15 @@ class SessionOptionsDialog(Dialog):
|
||||||
super().__init__(app, "Session Options")
|
super().__init__(app, "Session Options")
|
||||||
self.config_frame: Optional[ConfigFrame] = None
|
self.config_frame: Optional[ConfigFrame] = None
|
||||||
self.has_error: bool = False
|
self.has_error: bool = False
|
||||||
self.config: Dict[str, ConfigOption] = self.get_config()
|
|
||||||
self.enabled: bool = not self.app.core.is_runtime()
|
self.enabled: bool = not self.app.core.is_runtime()
|
||||||
if not self.has_error:
|
if not self.has_error:
|
||||||
self.draw()
|
self.draw()
|
||||||
|
|
||||||
def get_config(self) -> Dict[str, ConfigOption]:
|
|
||||||
try:
|
|
||||||
session_id = self.app.core.session.id
|
|
||||||
response = self.app.core.client.get_session_options(session_id)
|
|
||||||
return ConfigOption.from_dict(response.config)
|
|
||||||
except grpc.RpcError as e:
|
|
||||||
self.app.show_grpc_exception("Get Session Options Error", e)
|
|
||||||
self.has_error = True
|
|
||||||
self.destroy()
|
|
||||||
|
|
||||||
def draw(self) -> None:
|
def draw(self) -> None:
|
||||||
self.top.columnconfigure(0, weight=1)
|
self.top.columnconfigure(0, weight=1)
|
||||||
self.top.rowconfigure(0, weight=1)
|
self.top.rowconfigure(0, weight=1)
|
||||||
self.config_frame = ConfigFrame(self.top, self.app, self.config, self.enabled)
|
options = self.app.core.session.options
|
||||||
|
self.config_frame = ConfigFrame(self.top, self.app, options, self.enabled)
|
||||||
self.config_frame.draw_config()
|
self.config_frame.draw_config()
|
||||||
self.config_frame.grid(sticky=tk.NSEW, pady=PADY)
|
self.config_frame.grid(sticky=tk.NSEW, pady=PADY)
|
||||||
|
|
||||||
|
@ -53,10 +42,6 @@ class SessionOptionsDialog(Dialog):
|
||||||
|
|
||||||
def save(self) -> None:
|
def save(self) -> None:
|
||||||
config = self.config_frame.parse_config()
|
config = self.config_frame.parse_config()
|
||||||
try:
|
for key, value in config.items():
|
||||||
session_id = self.app.core.session.id
|
self.app.core.session.options[key].value = value
|
||||||
response = self.app.core.client.set_session_options(session_id, config)
|
|
||||||
logging.info("saved session config: %s", response)
|
|
||||||
except grpc.RpcError as e:
|
|
||||||
self.app.show_grpc_exception("Set Session Options Error", e)
|
|
||||||
self.destroy()
|
self.destroy()
|
||||||
|
|
|
@ -12,6 +12,8 @@ from core.gui.images import ImageEnum
|
||||||
from core.gui.task import ProgressTask
|
from core.gui.task import ProgressTask
|
||||||
from core.gui.themes import PADX, PADY
|
from core.gui.themes import PADX, PADY
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.gui.app import Application
|
from core.gui.app import Application
|
||||||
|
|
||||||
|
@ -30,10 +32,9 @@ class SessionsDialog(Dialog):
|
||||||
|
|
||||||
def get_sessions(self) -> List[SessionSummary]:
|
def get_sessions(self) -> List[SessionSummary]:
|
||||||
try:
|
try:
|
||||||
response = self.app.core.client.get_sessions()
|
sessions = self.app.core.client.get_sessions()
|
||||||
logging.info("sessions: %s", response)
|
logger.info("sessions: %s", sessions)
|
||||||
sessions = sorted(response.sessions, key=lambda x: x.id)
|
return sorted(sessions, key=lambda x: x.id)
|
||||||
return [SessionSummary.from_proto(x) for x in sessions]
|
|
||||||
except grpc.RpcError as e:
|
except grpc.RpcError as e:
|
||||||
self.app.show_grpc_exception("Get Sessions Error", e)
|
self.app.show_grpc_exception("Get Sessions Error", e)
|
||||||
self.destroy()
|
self.destroy()
|
||||||
|
@ -176,7 +177,7 @@ class SessionsDialog(Dialog):
|
||||||
self.selected_id = None
|
self.selected_id = None
|
||||||
self.delete_button.config(state=tk.DISABLED)
|
self.delete_button.config(state=tk.DISABLED)
|
||||||
self.connect_button.config(state=tk.DISABLED)
|
self.connect_button.config(state=tk.DISABLED)
|
||||||
logging.debug("selected session: %s", self.selected_session)
|
logger.debug("selected session: %s", self.selected_session)
|
||||||
|
|
||||||
def click_connect(self) -> None:
|
def click_connect(self) -> None:
|
||||||
if not self.selected_session:
|
if not self.selected_session:
|
||||||
|
@ -200,7 +201,7 @@ class SessionsDialog(Dialog):
|
||||||
def click_delete(self) -> None:
|
def click_delete(self) -> None:
|
||||||
if not self.selected_session:
|
if not self.selected_session:
|
||||||
return
|
return
|
||||||
logging.info("click delete session: %s", self.selected_session)
|
logger.info("click delete session: %s", self.selected_session)
|
||||||
self.tree.delete(self.selected_id)
|
self.tree.delete(self.selected_id)
|
||||||
self.app.core.delete_session(self.selected_session)
|
self.app.core.delete_session(self.selected_session)
|
||||||
session_id = None
|
session_id = None
|
||||||
|
|
|
@ -11,6 +11,8 @@ from core.gui.frames.link import EdgeInfoFrame, WirelessEdgeInfoFrame
|
||||||
from core.gui.graph import tags
|
from core.gui.graph import tags
|
||||||
from core.gui.utils import bandwidth_text, delay_jitter_text
|
from core.gui.utils import bandwidth_text, delay_jitter_text
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.gui.app import Application
|
from core.gui.app import Application
|
||||||
from core.gui.graph.graph import CanvasGraph
|
from core.gui.graph.graph import CanvasGraph
|
||||||
|
@ -393,7 +395,7 @@ class Edge:
|
||||||
self.dst.canvas.coords(self.dst_label2, *dst_pos)
|
self.dst.canvas.coords(self.dst_label2, *dst_pos)
|
||||||
|
|
||||||
def delete(self) -> None:
|
def delete(self) -> None:
|
||||||
logging.debug("deleting canvas edge, id: %s", self.id)
|
logger.debug("deleting canvas edge, id: %s", self.id)
|
||||||
self.src.canvas.delete(self.id)
|
self.src.canvas.delete(self.id)
|
||||||
self.src.canvas.delete(self.src_label)
|
self.src.canvas.delete(self.src_label)
|
||||||
self.src.canvas.delete(self.dst_label)
|
self.src.canvas.delete(self.dst_label)
|
||||||
|
@ -488,7 +490,7 @@ class CanvasWirelessEdge(Edge):
|
||||||
token: str,
|
token: str,
|
||||||
link: Link,
|
link: Link,
|
||||||
) -> None:
|
) -> None:
|
||||||
logging.debug("drawing wireless link from node %s to node %s", src, dst)
|
logger.debug("drawing wireless link from node %s to node %s", src, dst)
|
||||||
super().__init__(app, src, dst)
|
super().__init__(app, src, dst)
|
||||||
self.src.wireless_edges.add(self)
|
self.src.wireless_edges.add(self)
|
||||||
self.dst.wireless_edges.add(self)
|
self.dst.wireless_edges.add(self)
|
||||||
|
@ -622,7 +624,7 @@ class CanvasEdge(Edge):
|
||||||
self.draw_link_options()
|
self.draw_link_options()
|
||||||
|
|
||||||
def complete(self, dst: "CanvasNode", link: Link = None) -> None:
|
def complete(self, dst: "CanvasNode", link: Link = None) -> None:
|
||||||
logging.debug(
|
logger.debug(
|
||||||
"completing wired link from node(%s) to node(%s)",
|
"completing wired link from node(%s) to node(%s)",
|
||||||
self.src.core_node.name,
|
self.src.core_node.name,
|
||||||
dst.core_node.name,
|
dst.core_node.name,
|
||||||
|
|
|
@ -18,6 +18,8 @@ from core.gui.graph.node import CanvasNode, ShadowNode
|
||||||
from core.gui.graph.shape import Shape
|
from core.gui.graph.shape import Shape
|
||||||
from core.gui.graph.shapeutils import ShapeType, is_draw_shape, is_marker
|
from core.gui.graph.shapeutils import ShapeType, is_draw_shape, is_marker
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.gui.app import Application
|
from core.gui.app import Application
|
||||||
from core.gui.graph.manager import CanvasManager
|
from core.gui.graph.manager import CanvasManager
|
||||||
|
@ -101,10 +103,15 @@ class CanvasGraph(tk.Canvas):
|
||||||
"""
|
"""
|
||||||
Bind any mouse events or hot keys to the matching action
|
Bind any mouse events or hot keys to the matching action
|
||||||
"""
|
"""
|
||||||
|
self.bind("<Control-c>", self.copy_selected)
|
||||||
|
self.bind("<Control-v>", self.paste_selected)
|
||||||
|
self.bind("<Control-x>", self.cut_selected)
|
||||||
|
self.bind("<Control-d>", self.delete_selected)
|
||||||
|
self.bind("<Control-h>", self.hide_selected)
|
||||||
self.bind("<ButtonPress-1>", self.click_press)
|
self.bind("<ButtonPress-1>", self.click_press)
|
||||||
self.bind("<ButtonRelease-1>", self.click_release)
|
self.bind("<ButtonRelease-1>", self.click_release)
|
||||||
self.bind("<B1-Motion>", self.click_motion)
|
self.bind("<B1-Motion>", self.click_motion)
|
||||||
self.bind("<Delete>", self.press_delete)
|
self.bind("<Delete>", self.delete_selected)
|
||||||
self.bind("<Control-1>", self.ctrl_click)
|
self.bind("<Control-1>", self.ctrl_click)
|
||||||
self.bind("<Double-Button-1>", self.double_click)
|
self.bind("<Double-Button-1>", self.double_click)
|
||||||
self.bind("<MouseWheel>", self.zoom)
|
self.bind("<MouseWheel>", self.zoom)
|
||||||
|
@ -184,7 +191,7 @@ class CanvasGraph(tk.Canvas):
|
||||||
"""
|
"""
|
||||||
Draw a node or finish drawing an edge according to the current graph mode
|
Draw a node or finish drawing an edge according to the current graph mode
|
||||||
"""
|
"""
|
||||||
logging.debug("click release")
|
logger.debug("click release")
|
||||||
x, y = self.canvas_xy(event)
|
x, y = self.canvas_xy(event)
|
||||||
if not self.inside_canvas(x, y):
|
if not self.inside_canvas(x, y):
|
||||||
return
|
return
|
||||||
|
@ -210,7 +217,7 @@ class CanvasGraph(tk.Canvas):
|
||||||
else:
|
else:
|
||||||
self.focus_set()
|
self.focus_set()
|
||||||
self.selected = self.get_selected(event)
|
self.selected = self.get_selected(event)
|
||||||
logging.debug(
|
logger.debug(
|
||||||
"click release selected(%s) mode(%s)", self.selected, self.manager.mode
|
"click release selected(%s) mode(%s)", self.selected, self.manager.mode
|
||||||
)
|
)
|
||||||
if self.manager.mode == GraphMode.EDGE:
|
if self.manager.mode == GraphMode.EDGE:
|
||||||
|
@ -228,7 +235,7 @@ class CanvasGraph(tk.Canvas):
|
||||||
edge = self.drawing_edge
|
edge = self.drawing_edge
|
||||||
self.drawing_edge = None
|
self.drawing_edge = None
|
||||||
# edge dst must be a node
|
# edge dst must be a node
|
||||||
logging.debug("current selected: %s", self.selected)
|
logger.debug("current selected: %s", self.selected)
|
||||||
dst_node = self.nodes.get(self.selected)
|
dst_node = self.nodes.get(self.selected)
|
||||||
if not dst_node:
|
if not dst_node:
|
||||||
edge.delete()
|
edge.delete()
|
||||||
|
@ -275,7 +282,7 @@ class CanvasGraph(tk.Canvas):
|
||||||
if select_id is not None:
|
if select_id is not None:
|
||||||
self.move(select_id, x_offset, y_offset)
|
self.move(select_id, x_offset, y_offset)
|
||||||
|
|
||||||
def delete_selected_objects(self) -> None:
|
def delete_selected_objects(self, _event: tk.Event = None) -> None:
|
||||||
edges = set()
|
edges = set()
|
||||||
nodes = []
|
nodes = []
|
||||||
for object_id in self.selection:
|
for object_id in self.selection:
|
||||||
|
@ -305,7 +312,7 @@ class CanvasGraph(tk.Canvas):
|
||||||
self.selection.clear()
|
self.selection.clear()
|
||||||
self.core.deleted_canvas_nodes(nodes)
|
self.core.deleted_canvas_nodes(nodes)
|
||||||
|
|
||||||
def hide_selected_objects(self) -> None:
|
def hide_selected(self, _event: tk.Event = None) -> None:
|
||||||
for object_id in self.selection:
|
for object_id in self.selection:
|
||||||
# delete selection box
|
# delete selection box
|
||||||
selection_id = self.selection[object_id]
|
selection_id = self.selection[object_id]
|
||||||
|
@ -331,8 +338,8 @@ class CanvasGraph(tk.Canvas):
|
||||||
self.offset[0] * factor + event.x * (1 - factor),
|
self.offset[0] * factor + event.x * (1 - factor),
|
||||||
self.offset[1] * factor + event.y * (1 - factor),
|
self.offset[1] * factor + event.y * (1 - factor),
|
||||||
)
|
)
|
||||||
logging.debug("ratio: %s", self.ratio)
|
logger.debug("ratio: %s", self.ratio)
|
||||||
logging.debug("offset: %s", self.offset)
|
logger.debug("offset: %s", self.offset)
|
||||||
self.app.statusbar.set_zoom(self.ratio)
|
self.app.statusbar.set_zoom(self.ratio)
|
||||||
if self.wallpaper:
|
if self.wallpaper:
|
||||||
self.redraw_wallpaper()
|
self.redraw_wallpaper()
|
||||||
|
@ -347,10 +354,10 @@ class CanvasGraph(tk.Canvas):
|
||||||
|
|
||||||
self.cursor = x, y
|
self.cursor = x, y
|
||||||
selected = self.get_selected(event)
|
selected = self.get_selected(event)
|
||||||
logging.debug("click press(%s): %s", self.cursor, selected)
|
logger.debug("click press(%s): %s", self.cursor, selected)
|
||||||
x_check = self.cursor[0] - self.offset[0]
|
x_check = self.cursor[0] - self.offset[0]
|
||||||
y_check = self.cursor[1] - self.offset[1]
|
y_check = self.cursor[1] - self.offset[1]
|
||||||
logging.debug("click press offset(%s, %s)", x_check, y_check)
|
logger.debug("click press offset(%s, %s)", x_check, y_check)
|
||||||
is_node = selected in self.nodes
|
is_node = selected in self.nodes
|
||||||
if self.manager.mode == GraphMode.EDGE and is_node:
|
if self.manager.mode == GraphMode.EDGE and is_node:
|
||||||
node = self.nodes[selected]
|
node = self.nodes[selected]
|
||||||
|
@ -387,7 +394,7 @@ class CanvasGraph(tk.Canvas):
|
||||||
node = self.nodes[selected]
|
node = self.nodes[selected]
|
||||||
self.select_object(node.id)
|
self.select_object(node.id)
|
||||||
self.selected = selected
|
self.selected = selected
|
||||||
logging.debug(
|
logger.debug(
|
||||||
"selected node(%s), coords: (%s, %s)",
|
"selected node(%s), coords: (%s, %s)",
|
||||||
node.core_node.name,
|
node.core_node.name,
|
||||||
node.core_node.position.x,
|
node.core_node.position.x,
|
||||||
|
@ -397,7 +404,7 @@ class CanvasGraph(tk.Canvas):
|
||||||
shadow_node = self.shadow_nodes[selected]
|
shadow_node = self.shadow_nodes[selected]
|
||||||
self.select_object(shadow_node.id)
|
self.select_object(shadow_node.id)
|
||||||
self.selected = selected
|
self.selected = selected
|
||||||
logging.debug(
|
logger.debug(
|
||||||
"selected shadow node(%s), coords: (%s, %s)",
|
"selected shadow node(%s), coords: (%s, %s)",
|
||||||
shadow_node.node.core_node.name,
|
shadow_node.node.core_node.name,
|
||||||
shadow_node.node.core_node.position.x,
|
shadow_node.node.core_node.position.x,
|
||||||
|
@ -418,7 +425,7 @@ class CanvasGraph(tk.Canvas):
|
||||||
self.cursor = x, y
|
self.cursor = x, y
|
||||||
|
|
||||||
# handle multiple selections
|
# handle multiple selections
|
||||||
logging.debug("control left click: %s", event)
|
logger.debug("control left click: %s", event)
|
||||||
selected = self.get_selected(event)
|
selected = self.get_selected(event)
|
||||||
if (
|
if (
|
||||||
selected not in self.selection
|
selected not in self.selection
|
||||||
|
@ -485,17 +492,6 @@ class CanvasGraph(tk.Canvas):
|
||||||
if self.select_box and self.manager.mode == GraphMode.SELECT:
|
if self.select_box and self.manager.mode == GraphMode.SELECT:
|
||||||
self.select_box.shape_motion(x, y)
|
self.select_box.shape_motion(x, y)
|
||||||
|
|
||||||
def press_delete(self, _event: tk.Event) -> None:
|
|
||||||
"""
|
|
||||||
delete selected nodes and any data that relates to it
|
|
||||||
"""
|
|
||||||
logging.debug("press delete key")
|
|
||||||
if not self.app.core.is_runtime():
|
|
||||||
self.delete_selected_objects()
|
|
||||||
self.app.default_info()
|
|
||||||
else:
|
|
||||||
logging.debug("node deletion is disabled during runtime state")
|
|
||||||
|
|
||||||
def double_click(self, event: tk.Event) -> None:
|
def double_click(self, event: tk.Event) -> None:
|
||||||
selected = self.get_selected(event)
|
selected = self.get_selected(event)
|
||||||
if selected is not None and selected in self.shapes:
|
if selected is not None and selected in self.shapes:
|
||||||
|
@ -606,10 +602,10 @@ class CanvasGraph(tk.Canvas):
|
||||||
self.draw_wallpaper(image)
|
self.draw_wallpaper(image)
|
||||||
|
|
||||||
def redraw_canvas(self, dimensions: Tuple[int, int] = None) -> None:
|
def redraw_canvas(self, dimensions: Tuple[int, int] = None) -> None:
|
||||||
logging.debug("redrawing canvas to dimensions: %s", dimensions)
|
logger.debug("redrawing canvas to dimensions: %s", dimensions)
|
||||||
|
|
||||||
# reset scale and move back to original position
|
# reset scale and move back to original position
|
||||||
logging.debug("resetting scaling: %s %s", self.ratio, self.offset)
|
logger.debug("resetting scaling: %s %s", self.ratio, self.offset)
|
||||||
factor = 1 / self.ratio
|
factor = 1 / self.ratio
|
||||||
self.scale(tk.ALL, self.offset[0], self.offset[1], factor, factor)
|
self.scale(tk.ALL, self.offset[0], self.offset[1], factor, factor)
|
||||||
self.move(tk.ALL, -self.offset[0], -self.offset[1])
|
self.move(tk.ALL, -self.offset[0], -self.offset[1])
|
||||||
|
@ -628,11 +624,11 @@ class CanvasGraph(tk.Canvas):
|
||||||
|
|
||||||
def redraw_wallpaper(self) -> None:
|
def redraw_wallpaper(self) -> None:
|
||||||
if self.adjust_to_dim.get():
|
if self.adjust_to_dim.get():
|
||||||
logging.debug("drawing wallpaper to canvas dimensions")
|
logger.debug("drawing wallpaper to canvas dimensions")
|
||||||
self.resize_to_wallpaper()
|
self.resize_to_wallpaper()
|
||||||
else:
|
else:
|
||||||
option = ScaleOption(self.scale_option.get())
|
option = ScaleOption(self.scale_option.get())
|
||||||
logging.debug("drawing canvas using scaling option: %s", option)
|
logger.debug("drawing canvas using scaling option: %s", option)
|
||||||
if option == ScaleOption.UPPER_LEFT:
|
if option == ScaleOption.UPPER_LEFT:
|
||||||
self.wallpaper_upper_left()
|
self.wallpaper_upper_left()
|
||||||
elif option == ScaleOption.CENTERED:
|
elif option == ScaleOption.CENTERED:
|
||||||
|
@ -640,7 +636,7 @@ class CanvasGraph(tk.Canvas):
|
||||||
elif option == ScaleOption.SCALED:
|
elif option == ScaleOption.SCALED:
|
||||||
self.wallpaper_scaled()
|
self.wallpaper_scaled()
|
||||||
elif option == ScaleOption.TILED:
|
elif option == ScaleOption.TILED:
|
||||||
logging.warning("tiled background not implemented yet")
|
logger.warning("tiled background not implemented yet")
|
||||||
self.organize()
|
self.organize()
|
||||||
|
|
||||||
def organize(self) -> None:
|
def organize(self) -> None:
|
||||||
|
@ -648,7 +644,7 @@ class CanvasGraph(tk.Canvas):
|
||||||
self.tag_raise(tag)
|
self.tag_raise(tag)
|
||||||
|
|
||||||
def set_wallpaper(self, filename: Optional[str]) -> None:
|
def set_wallpaper(self, filename: Optional[str]) -> None:
|
||||||
logging.info("setting canvas(%s) background: %s", self.id, filename)
|
logger.info("setting canvas(%s) background: %s", self.id, filename)
|
||||||
if filename:
|
if filename:
|
||||||
img = Image.open(filename)
|
img = Image.open(filename)
|
||||||
self.wallpaper = img
|
self.wallpaper = img
|
||||||
|
@ -671,20 +667,38 @@ class CanvasGraph(tk.Canvas):
|
||||||
edge.complete(dst)
|
edge.complete(dst)
|
||||||
return edge
|
return edge
|
||||||
|
|
||||||
def copy(self) -> None:
|
def copy_selected(self, _event: tk.Event = None) -> None:
|
||||||
if self.core.is_runtime():
|
if self.core.is_runtime():
|
||||||
logging.debug("copy is disabled during runtime state")
|
logger.debug("copy is disabled during runtime state")
|
||||||
return
|
return
|
||||||
if self.selection:
|
if self.selection:
|
||||||
logging.debug("to copy nodes: %s", self.selection)
|
logger.debug("to copy nodes: %s", self.selection)
|
||||||
self.to_copy.clear()
|
self.to_copy.clear()
|
||||||
for node_id in self.selection.keys():
|
for node_id in self.selection.keys():
|
||||||
canvas_node = self.nodes[node_id]
|
canvas_node = self.nodes[node_id]
|
||||||
self.to_copy.append(canvas_node)
|
self.to_copy.append(canvas_node)
|
||||||
|
|
||||||
def paste(self) -> None:
|
def cut_selected(self, _event: tk.Event = None) -> None:
|
||||||
if self.core.is_runtime():
|
if self.core.is_runtime():
|
||||||
logging.debug("paste is disabled during runtime state")
|
logger.debug("cut is disabled during runtime state")
|
||||||
|
return
|
||||||
|
self.copy_selected()
|
||||||
|
self.delete_selected()
|
||||||
|
|
||||||
|
def delete_selected(self, _event: tk.Event = None) -> None:
|
||||||
|
"""
|
||||||
|
delete selected nodes and any data that relates to it
|
||||||
|
"""
|
||||||
|
logger.debug("press delete key")
|
||||||
|
if self.core.is_runtime():
|
||||||
|
logger.debug("node deletion is disabled during runtime state")
|
||||||
|
return
|
||||||
|
self.delete_selected_objects()
|
||||||
|
self.app.default_info()
|
||||||
|
|
||||||
|
def paste_selected(self, _event: tk.Event = None) -> None:
|
||||||
|
if self.core.is_runtime():
|
||||||
|
logger.debug("paste is disabled during runtime state")
|
||||||
return
|
return
|
||||||
# maps original node canvas id to copy node canvas id
|
# maps original node canvas id to copy node canvas id
|
||||||
copy_map = {}
|
copy_map = {}
|
||||||
|
@ -813,6 +827,7 @@ class CanvasGraph(tk.Canvas):
|
||||||
wallpaper=wallpaper_path,
|
wallpaper=wallpaper_path,
|
||||||
wallpaper_style=self.scale_option.get(),
|
wallpaper_style=self.scale_option.get(),
|
||||||
fit_image=self.adjust_to_dim.get(),
|
fit_image=self.adjust_to_dim.get(),
|
||||||
|
dimensions=self.current_dimensions,
|
||||||
)
|
)
|
||||||
|
|
||||||
def parse_metadata(self, config: Dict[str, Any]) -> None:
|
def parse_metadata(self, config: Dict[str, Any]) -> None:
|
||||||
|
@ -820,12 +835,15 @@ class CanvasGraph(tk.Canvas):
|
||||||
self.adjust_to_dim.set(fit_image)
|
self.adjust_to_dim.set(fit_image)
|
||||||
wallpaper_style = config.get("wallpaper_style", 1)
|
wallpaper_style = config.get("wallpaper_style", 1)
|
||||||
self.scale_option.set(wallpaper_style)
|
self.scale_option.set(wallpaper_style)
|
||||||
|
dimensions = config.get("dimensions")
|
||||||
|
if dimensions:
|
||||||
|
self.redraw_canvas(dimensions)
|
||||||
wallpaper = config.get("wallpaper")
|
wallpaper = config.get("wallpaper")
|
||||||
if wallpaper:
|
if wallpaper:
|
||||||
wallpaper = Path(wallpaper)
|
wallpaper = Path(wallpaper)
|
||||||
if not wallpaper.is_file():
|
if not wallpaper.is_file():
|
||||||
wallpaper = appconfig.BACKGROUNDS_PATH.joinpath(wallpaper)
|
wallpaper = appconfig.BACKGROUNDS_PATH.joinpath(wallpaper)
|
||||||
logging.info("canvas(%s), wallpaper: %s", self.id, wallpaper)
|
logger.info("canvas(%s), wallpaper: %s", self.id, wallpaper)
|
||||||
if wallpaper.is_file():
|
if wallpaper.is_file():
|
||||||
self.set_wallpaper(str(wallpaper))
|
self.set_wallpaper(str(wallpaper))
|
||||||
else:
|
else:
|
||||||
|
|
|
@ -1,3 +1,4 @@
|
||||||
|
import json
|
||||||
import logging
|
import logging
|
||||||
import tkinter as tk
|
import tkinter as tk
|
||||||
from copy import deepcopy
|
from copy import deepcopy
|
||||||
|
@ -16,9 +17,12 @@ from core.gui.graph.edges import (
|
||||||
from core.gui.graph.enums import GraphMode
|
from core.gui.graph.enums import GraphMode
|
||||||
from core.gui.graph.graph import CanvasGraph
|
from core.gui.graph.graph import CanvasGraph
|
||||||
from core.gui.graph.node import CanvasNode
|
from core.gui.graph.node import CanvasNode
|
||||||
|
from core.gui.graph.shape import Shape
|
||||||
from core.gui.graph.shapeutils import ShapeType
|
from core.gui.graph.shapeutils import ShapeType
|
||||||
from core.gui.nodeutils import NodeDraw
|
from core.gui.nodeutils import NodeDraw
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.gui.app import Application
|
from core.gui.app import Application
|
||||||
from core.gui.coreclient import CoreClient
|
from core.gui.coreclient import CoreClient
|
||||||
|
@ -85,7 +89,6 @@ class CanvasManager:
|
||||||
self.app.guiconfig.preferences.width,
|
self.app.guiconfig.preferences.width,
|
||||||
self.app.guiconfig.preferences.height,
|
self.app.guiconfig.preferences.height,
|
||||||
)
|
)
|
||||||
self.current_dimensions: Tuple[int, int] = self.default_dimensions
|
|
||||||
self.show_node_labels: ShowVar = ShowNodeLabels(
|
self.show_node_labels: ShowVar = ShowNodeLabels(
|
||||||
self, tags.NODE_LABEL, value=True
|
self, tags.NODE_LABEL, value=True
|
||||||
)
|
)
|
||||||
|
@ -166,7 +169,7 @@ class CanvasManager:
|
||||||
canvas_id = self._next_id()
|
canvas_id = self._next_id()
|
||||||
self.notebook.add(tab, text=f"Canvas {canvas_id}")
|
self.notebook.add(tab, text=f"Canvas {canvas_id}")
|
||||||
unique_id = self.notebook.tabs()[-1]
|
unique_id = self.notebook.tabs()[-1]
|
||||||
logging.info("creating canvas(%s)", canvas_id)
|
logger.info("creating canvas(%s)", canvas_id)
|
||||||
self.canvas_ids[unique_id] = canvas_id
|
self.canvas_ids[unique_id] = canvas_id
|
||||||
self.unique_ids[canvas_id] = unique_id
|
self.unique_ids[canvas_id] = unique_id
|
||||||
|
|
||||||
|
@ -205,7 +208,7 @@ class CanvasManager:
|
||||||
edge.delete()
|
edge.delete()
|
||||||
|
|
||||||
def join(self, session: Session) -> None:
|
def join(self, session: Session) -> None:
|
||||||
# clear out all canvas
|
# clear out all canvases
|
||||||
for canvas_id in self.notebook.tabs():
|
for canvas_id in self.notebook.tabs():
|
||||||
self.notebook.forget(canvas_id)
|
self.notebook.forget(canvas_id)
|
||||||
self.canvases.clear()
|
self.canvases.clear()
|
||||||
|
@ -213,7 +216,7 @@ class CanvasManager:
|
||||||
self.unique_ids.clear()
|
self.unique_ids.clear()
|
||||||
self.edges.clear()
|
self.edges.clear()
|
||||||
self.wireless_edges.clear()
|
self.wireless_edges.clear()
|
||||||
logging.info("cleared canvases")
|
logger.info("cleared canvases")
|
||||||
|
|
||||||
# reset settings
|
# reset settings
|
||||||
self.show_node_labels.set(True)
|
self.show_node_labels.set(True)
|
||||||
|
@ -232,6 +235,10 @@ class CanvasManager:
|
||||||
self.draw_session(session)
|
self.draw_session(session)
|
||||||
|
|
||||||
def draw_session(self, session: Session) -> None:
|
def draw_session(self, session: Session) -> None:
|
||||||
|
# draw canvas configurations and shapes
|
||||||
|
self.parse_metadata_canvas(session.metadata)
|
||||||
|
self.parse_metadata_shapes(session.metadata)
|
||||||
|
|
||||||
# create session nodes
|
# create session nodes
|
||||||
for core_node in session.nodes.values():
|
for core_node in session.nodes.values():
|
||||||
# add node, avoiding ignored nodes
|
# add node, avoiding ignored nodes
|
||||||
|
@ -254,50 +261,92 @@ class CanvasManager:
|
||||||
else:
|
else:
|
||||||
self.add_wired_edge(node1, node2, link)
|
self.add_wired_edge(node1, node2, link)
|
||||||
|
|
||||||
# parse metadata and organize canvases
|
# organize canvas order
|
||||||
self.core.parse_metadata()
|
|
||||||
for canvas in self.canvases.values():
|
for canvas in self.canvases.values():
|
||||||
canvas.organize()
|
canvas.organize()
|
||||||
|
|
||||||
|
# parse metada for edge configs and hidden nodes
|
||||||
|
self.parse_metadata_edges(session.metadata)
|
||||||
|
self.parse_metadata_hidden(session.metadata)
|
||||||
|
|
||||||
# create a default canvas if none were created prior
|
# create a default canvas if none were created prior
|
||||||
if not self.canvases:
|
if not self.canvases:
|
||||||
self.add_canvas()
|
self.add_canvas()
|
||||||
|
|
||||||
def redraw_canvases(self, dimensions: Tuple[int, int]) -> None:
|
def redraw_canvas(self, dimensions: Tuple[int, int]) -> None:
|
||||||
for canvas in self.canvases.values():
|
canvas = self.current()
|
||||||
canvas.redraw_canvas(dimensions)
|
canvas.redraw_canvas(dimensions)
|
||||||
if canvas.wallpaper:
|
if canvas.wallpaper:
|
||||||
canvas.redraw_wallpaper()
|
canvas.redraw_wallpaper()
|
||||||
|
|
||||||
def get_metadata(self) -> Dict[str, Any]:
|
def get_metadata(self) -> Dict[str, Any]:
|
||||||
canvases = [x.get_metadata() for x in self.all()]
|
canvases = [x.get_metadata() for x in self.all()]
|
||||||
return dict(
|
return dict(gridlines=self.show_grid.get(), canvases=canvases)
|
||||||
gridlines=self.app.manager.show_grid.get(),
|
|
||||||
dimensions=self.app.manager.current_dimensions,
|
|
||||||
canvases=canvases,
|
|
||||||
)
|
|
||||||
|
|
||||||
def parse_metadata(self, config: Dict[str, Any]) -> None:
|
def parse_metadata_canvas(self, metadata: Dict[str, Any]) -> None:
|
||||||
|
# canvas setting
|
||||||
|
canvas_config = metadata.get("canvas")
|
||||||
|
logger.debug("canvas metadata: %s", canvas_config)
|
||||||
|
if not canvas_config:
|
||||||
|
return
|
||||||
|
canvas_config = json.loads(canvas_config)
|
||||||
# get configured dimensions and gridlines option
|
# get configured dimensions and gridlines option
|
||||||
dimensions = self.default_dimensions
|
gridlines = canvas_config.get("gridlines", True)
|
||||||
dimensions = config.get("dimensions", dimensions)
|
|
||||||
gridlines = config.get("gridlines", True)
|
|
||||||
self.show_grid.set(gridlines)
|
self.show_grid.set(gridlines)
|
||||||
self.redraw_canvases(dimensions)
|
|
||||||
|
|
||||||
# get background configurations
|
# get background configurations
|
||||||
for canvas_config in config.get("canvases", []):
|
for canvas_config in canvas_config.get("canvases", []):
|
||||||
canvas_id = canvas_config.get("id")
|
canvas_id = canvas_config.get("id")
|
||||||
if canvas_id is None:
|
if canvas_id is None:
|
||||||
logging.error("canvas config id not provided")
|
logger.error("canvas config id not provided")
|
||||||
continue
|
continue
|
||||||
canvas = self.get(canvas_id)
|
canvas = self.get(canvas_id)
|
||||||
canvas.parse_metadata(canvas_config)
|
canvas.parse_metadata(canvas_config)
|
||||||
|
|
||||||
|
def parse_metadata_shapes(self, metadata: Dict[str, Any]) -> None:
|
||||||
|
# load saved shapes
|
||||||
|
shapes_config = metadata.get("shapes")
|
||||||
|
if not shapes_config:
|
||||||
|
return
|
||||||
|
shapes_config = json.loads(shapes_config)
|
||||||
|
for shape_config in shapes_config:
|
||||||
|
logger.debug("loading shape: %s", shape_config)
|
||||||
|
Shape.from_metadata(self.app, shape_config)
|
||||||
|
|
||||||
|
def parse_metadata_edges(self, metadata: Dict[str, Any]) -> None:
|
||||||
|
# load edges config
|
||||||
|
edges_config = metadata.get("edges")
|
||||||
|
if not edges_config:
|
||||||
|
return
|
||||||
|
edges_config = json.loads(edges_config)
|
||||||
|
logger.info("edges config: %s", edges_config)
|
||||||
|
for edge_config in edges_config:
|
||||||
|
edge_token = edge_config["token"]
|
||||||
|
edge = self.core.links.get(edge_token)
|
||||||
|
if edge:
|
||||||
|
edge.width = edge_config["width"]
|
||||||
|
edge.color = edge_config["color"]
|
||||||
|
edge.redraw()
|
||||||
|
else:
|
||||||
|
logger.warning("invalid edge token to configure: %s", edge_token)
|
||||||
|
|
||||||
|
def parse_metadata_hidden(self, metadata: Dict[str, Any]) -> None:
|
||||||
|
# read hidden nodes
|
||||||
|
hidden_config = metadata.get("hidden")
|
||||||
|
if not hidden_config:
|
||||||
|
return
|
||||||
|
hidden_config = json.loads(hidden_config)
|
||||||
|
for node_id in hidden_config:
|
||||||
|
canvas_node = self.core.canvas_nodes.get(node_id)
|
||||||
|
if canvas_node:
|
||||||
|
canvas_node.hide()
|
||||||
|
else:
|
||||||
|
logger.warning("invalid node to hide: %s", node_id)
|
||||||
|
|
||||||
def add_core_node(self, core_node: Node) -> None:
|
def add_core_node(self, core_node: Node) -> None:
|
||||||
# get canvas tab for node
|
# get canvas tab for node
|
||||||
canvas_id = core_node.canvas if core_node.canvas > 0 else 1
|
canvas_id = core_node.canvas if core_node.canvas > 0 else 1
|
||||||
logging.info("adding core node canvas(%s): %s", core_node.name, canvas_id)
|
logger.info("adding core node canvas(%s): %s", core_node.name, canvas_id)
|
||||||
canvas = self.get(canvas_id)
|
canvas = self.get(canvas_id)
|
||||||
image = nutils.get_icon(core_node, self.app)
|
image = nutils.get_icon(core_node, self.app)
|
||||||
x = core_node.position.x
|
x = core_node.position.x
|
||||||
|
@ -354,7 +403,7 @@ class CanvasManager:
|
||||||
network_id = link.network_id if link.network_id else None
|
network_id = link.network_id if link.network_id else None
|
||||||
token = create_wireless_token(src.id, dst.id, network_id)
|
token = create_wireless_token(src.id, dst.id, network_id)
|
||||||
if token in self.wireless_edges:
|
if token in self.wireless_edges:
|
||||||
logging.warning("ignoring link that already exists: %s", link)
|
logger.warning("ignoring link that already exists: %s", link)
|
||||||
return
|
return
|
||||||
edge = CanvasWirelessEdge(self.app, src, dst, network_id, token, link)
|
edge = CanvasWirelessEdge(self.app, src, dst, network_id, token, link)
|
||||||
self.wireless_edges[token] = edge
|
self.wireless_edges[token] = edge
|
||||||
|
|
|
@ -24,6 +24,8 @@ from core.gui.graph.edges import CanvasEdge, CanvasWirelessEdge
|
||||||
from core.gui.graph.tooltip import CanvasTooltip
|
from core.gui.graph.tooltip import CanvasTooltip
|
||||||
from core.gui.images import ImageEnum
|
from core.gui.images import ImageEnum
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.gui.app import Application
|
from core.gui.app import Application
|
||||||
from core.gui.graph.graph import CanvasGraph
|
from core.gui.graph.graph import CanvasGraph
|
||||||
|
@ -87,7 +89,7 @@ class CanvasNode:
|
||||||
self.canvas.tag_bind(self.id, "<Button-1>", self.show_info)
|
self.canvas.tag_bind(self.id, "<Button-1>", self.show_info)
|
||||||
|
|
||||||
def delete(self) -> None:
|
def delete(self) -> None:
|
||||||
logging.debug("Delete canvas node for %s", self.core_node)
|
logger.debug("Delete canvas node for %s", self.core_node)
|
||||||
self.canvas.delete(self.id)
|
self.canvas.delete(self.id)
|
||||||
self.canvas.delete(self.text_id)
|
self.canvas.delete(self.text_id)
|
||||||
self.delete_antennas()
|
self.delete_antennas()
|
||||||
|
@ -110,7 +112,7 @@ class CanvasNode:
|
||||||
"""
|
"""
|
||||||
delete one antenna
|
delete one antenna
|
||||||
"""
|
"""
|
||||||
logging.debug("Delete an antenna on %s", self.core_node.name)
|
logger.debug("Delete an antenna on %s", self.core_node.name)
|
||||||
if self.antennas:
|
if self.antennas:
|
||||||
antenna_id = self.antennas.pop()
|
antenna_id = self.antennas.pop()
|
||||||
self.canvas.delete(antenna_id)
|
self.canvas.delete(antenna_id)
|
||||||
|
@ -120,7 +122,7 @@ class CanvasNode:
|
||||||
"""
|
"""
|
||||||
delete all antennas
|
delete all antennas
|
||||||
"""
|
"""
|
||||||
logging.debug("Remove all antennas for %s", self.core_node.name)
|
logger.debug("Remove all antennas for %s", self.core_node.name)
|
||||||
for antenna_id in self.antennas:
|
for antenna_id in self.antennas:
|
||||||
self.canvas.delete(antenna_id)
|
self.canvas.delete(antenna_id)
|
||||||
self.antennas.clear()
|
self.antennas.clear()
|
||||||
|
@ -253,10 +255,12 @@ class CanvasNode:
|
||||||
else:
|
else:
|
||||||
self.context.add_command(label="Configure", command=self.show_config)
|
self.context.add_command(label="Configure", command=self.show_config)
|
||||||
if nutils.is_container(self.core_node):
|
if nutils.is_container(self.core_node):
|
||||||
self.context.add_command(label="Services", command=self.show_services)
|
|
||||||
self.context.add_command(
|
self.context.add_command(
|
||||||
label="Config Services", command=self.show_config_services
|
label="Config Services", command=self.show_config_services
|
||||||
)
|
)
|
||||||
|
self.context.add_command(
|
||||||
|
label="Services (Deprecated)", command=self.show_services
|
||||||
|
)
|
||||||
if is_emane:
|
if is_emane:
|
||||||
self.context.add_command(
|
self.context.add_command(
|
||||||
label="EMANE Config", command=self.show_emane_config
|
label="EMANE Config", command=self.show_emane_config
|
||||||
|
@ -334,7 +338,7 @@ class CanvasNode:
|
||||||
def canvas_copy(self) -> None:
|
def canvas_copy(self) -> None:
|
||||||
self.canvas.clear_selection()
|
self.canvas.clear_selection()
|
||||||
self.canvas.select_object(self.id)
|
self.canvas.select_object(self.id)
|
||||||
self.canvas.copy()
|
self.canvas.copy_selected()
|
||||||
|
|
||||||
def show_config(self) -> None:
|
def show_config(self) -> None:
|
||||||
dialog = NodeConfigDialog(self.app, self)
|
dialog = NodeConfigDialog(self.app, self)
|
||||||
|
@ -400,7 +404,7 @@ class CanvasNode:
|
||||||
|
|
||||||
def update_icon(self, icon_path: str) -> None:
|
def update_icon(self, icon_path: str) -> None:
|
||||||
if not Path(icon_path).exists():
|
if not Path(icon_path).exists():
|
||||||
logging.error(f"node icon does not exist: {icon_path}")
|
logger.error(f"node icon does not exist: {icon_path}")
|
||||||
return
|
return
|
||||||
self.core_node.icon = icon_path
|
self.core_node.icon = icon_path
|
||||||
self.image = images.from_file(icon_path, width=images.NODE_SIZE)
|
self.image = images.from_file(icon_path, width=images.NODE_SIZE)
|
||||||
|
@ -459,10 +463,10 @@ class CanvasNode:
|
||||||
def _service_action(self, service: str, action: ServiceAction) -> None:
|
def _service_action(self, service: str, action: ServiceAction) -> None:
|
||||||
session_id = self.app.core.session.id
|
session_id = self.app.core.session.id
|
||||||
try:
|
try:
|
||||||
response = self.app.core.client.service_action(
|
result = self.app.core.client.service_action(
|
||||||
session_id, self.core_node.id, service, action
|
session_id, self.core_node.id, service, action
|
||||||
)
|
)
|
||||||
if not response.result:
|
if not result:
|
||||||
self.app.show_error("Service Action Error", "Action Failed!")
|
self.app.show_error("Service Action Error", "Action Failed!")
|
||||||
except grpc.RpcError as e:
|
except grpc.RpcError as e:
|
||||||
self.app.show_grpc_exception("Service Error", e)
|
self.app.show_grpc_exception("Service Error", e)
|
||||||
|
|
|
@ -5,6 +5,8 @@ from core.gui.dialogs.shapemod import ShapeDialog
|
||||||
from core.gui.graph import tags
|
from core.gui.graph import tags
|
||||||
from core.gui.graph.shapeutils import ShapeType
|
from core.gui.graph.shapeutils import ShapeType
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.gui.app import Application
|
from core.gui.app import Application
|
||||||
from core.gui.graph.graph import CanvasGraph
|
from core.gui.graph.graph import CanvasGraph
|
||||||
|
@ -92,7 +94,7 @@ class Shape:
|
||||||
shape = Shape(app, canvas, shape_type, *coords, data=data)
|
shape = Shape(app, canvas, shape_type, *coords, data=data)
|
||||||
canvas.shapes[shape.id] = shape
|
canvas.shapes[shape.id] = shape
|
||||||
except ValueError:
|
except ValueError:
|
||||||
logging.exception("unknown shape: %s", shape_type)
|
logger.exception("unknown shape: %s", shape_type)
|
||||||
|
|
||||||
def draw(self) -> None:
|
def draw(self) -> None:
|
||||||
if self.created:
|
if self.created:
|
||||||
|
@ -139,7 +141,7 @@ class Shape:
|
||||||
state=self.app.manager.show_annotations.state(),
|
state=self.app.manager.show_annotations.state(),
|
||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
logging.error("unknown shape type: %s", self.shape_type)
|
logger.error("unknown shape type: %s", self.shape_type)
|
||||||
self.created = True
|
self.created = True
|
||||||
|
|
||||||
def get_font(self) -> List[Union[int, str]]:
|
def get_font(self) -> List[Union[int, str]]:
|
||||||
|
@ -192,7 +194,7 @@ class Shape:
|
||||||
self.canvas.move(self.text_id, x_offset, y_offset)
|
self.canvas.move(self.text_id, x_offset, y_offset)
|
||||||
|
|
||||||
def delete(self) -> None:
|
def delete(self) -> None:
|
||||||
logging.debug("Delete shape, id(%s)", self.id)
|
logger.debug("Delete shape, id(%s)", self.id)
|
||||||
self.canvas.delete(self.id)
|
self.canvas.delete(self.id)
|
||||||
self.canvas.delete(self.text_id)
|
self.canvas.delete(self.text_id)
|
||||||
|
|
||||||
|
|
|
@ -9,6 +9,8 @@ from core.gui import nodeutils as nutils
|
||||||
from core.gui.graph.edges import CanvasEdge
|
from core.gui.graph.edges import CanvasEdge
|
||||||
from core.gui.graph.node import CanvasNode
|
from core.gui.graph.node import CanvasNode
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.gui.app import Application
|
from core.gui.app import Application
|
||||||
|
|
||||||
|
@ -158,11 +160,18 @@ class InterfaceManager:
|
||||||
index += 1
|
index += 1
|
||||||
return index
|
return index
|
||||||
|
|
||||||
def get_ips(self, node: Node) -> [str, str]:
|
def get_ips(self, node: Node) -> [Optional[str], Optional[str]]:
|
||||||
|
enable_ip4 = self.app.guiconfig.ips.enable_ip4
|
||||||
|
enable_ip6 = self.app.guiconfig.ips.enable_ip6
|
||||||
|
ip4, ip6 = None, None
|
||||||
|
if not enable_ip4 and not enable_ip6:
|
||||||
|
return ip4, ip6
|
||||||
index = self.next_index(node)
|
index = self.next_index(node)
|
||||||
ip4 = self.current_subnets.ip4[index]
|
if enable_ip4:
|
||||||
ip6 = self.current_subnets.ip6[index]
|
ip4 = str(self.current_subnets.ip4[index])
|
||||||
return str(ip4), str(ip6)
|
if enable_ip6:
|
||||||
|
ip6 = str(self.current_subnets.ip6[index])
|
||||||
|
return ip4, ip6
|
||||||
|
|
||||||
def get_subnets(self, iface: Interface) -> Subnets:
|
def get_subnets(self, iface: Interface) -> Subnets:
|
||||||
ip4_subnet = self.ip4_subnets
|
ip4_subnet = self.ip4_subnets
|
||||||
|
@ -196,12 +205,12 @@ class InterfaceManager:
|
||||||
else:
|
else:
|
||||||
self.current_subnets = self.next_subnets()
|
self.current_subnets = self.next_subnets()
|
||||||
else:
|
else:
|
||||||
logging.info("ignoring subnet change for link between network nodes")
|
logger.info("ignoring subnet change for link between network nodes")
|
||||||
|
|
||||||
def find_subnets(
|
def find_subnets(
|
||||||
self, canvas_node: CanvasNode, visited: Set[int] = None
|
self, canvas_node: CanvasNode, visited: Set[int] = None
|
||||||
) -> Optional[IPNetwork]:
|
) -> Optional[IPNetwork]:
|
||||||
logging.info("finding subnet for node: %s", canvas_node.core_node.name)
|
logger.info("finding subnet for node: %s", canvas_node.core_node.name)
|
||||||
subnets = None
|
subnets = None
|
||||||
if not visited:
|
if not visited:
|
||||||
visited = set()
|
visited = set()
|
||||||
|
@ -220,7 +229,7 @@ class InterfaceManager:
|
||||||
else:
|
else:
|
||||||
subnets = self.find_subnets(check_node, visited)
|
subnets = self.find_subnets(check_node, visited)
|
||||||
if subnets:
|
if subnets:
|
||||||
logging.info("found subnets: %s", subnets)
|
logger.info("found subnets: %s", subnets)
|
||||||
break
|
break
|
||||||
return subnets
|
return subnets
|
||||||
|
|
||||||
|
@ -244,7 +253,7 @@ class InterfaceManager:
|
||||||
iface1=src_iface,
|
iface1=src_iface,
|
||||||
iface2=dst_iface,
|
iface2=dst_iface,
|
||||||
)
|
)
|
||||||
logging.info("added link between %s and %s", src_node.name, dst_node.name)
|
logger.info("added link between %s and %s", src_node.name, dst_node.name)
|
||||||
return link
|
return link
|
||||||
|
|
||||||
def create_iface(self, canvas_node: CanvasNode, wireless_link: bool) -> Interface:
|
def create_iface(self, canvas_node: CanvasNode, wireless_link: bool) -> Interface:
|
||||||
|
@ -266,5 +275,5 @@ class InterfaceManager:
|
||||||
ip6=ip6,
|
ip6=ip6,
|
||||||
ip6_mask=ip6_mask,
|
ip6_mask=ip6_mask,
|
||||||
)
|
)
|
||||||
logging.info("create node(%s) interface(%s)", node.name, iface)
|
logger.info("create node(%s) interface(%s)", node.name, iface)
|
||||||
return iface
|
return iface
|
||||||
|
|
|
@ -1,8 +1,8 @@
|
||||||
import logging
|
import logging
|
||||||
import os
|
|
||||||
import tkinter as tk
|
import tkinter as tk
|
||||||
import webbrowser
|
import webbrowser
|
||||||
from functools import partial
|
from functools import partial
|
||||||
|
from pathlib import Path
|
||||||
from tkinter import filedialog, messagebox
|
from tkinter import filedialog, messagebox
|
||||||
from typing import TYPE_CHECKING, Optional
|
from typing import TYPE_CHECKING, Optional
|
||||||
|
|
||||||
|
@ -27,6 +27,8 @@ from core.gui.graph.manager import CanvasManager
|
||||||
from core.gui.observers import ObserversMenu
|
from core.gui.observers import ObserversMenu
|
||||||
from core.gui.task import ProgressTask
|
from core.gui.task import ProgressTask
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.gui.app import Application
|
from core.gui.app import Application
|
||||||
|
|
||||||
|
@ -76,7 +78,7 @@ class Menubar(tk.Menu):
|
||||||
self.app.bind_all("<Control-n>", lambda e: self.click_new())
|
self.app.bind_all("<Control-n>", lambda e: self.click_new())
|
||||||
menu.add_command(label="Save", accelerator="Ctrl+S", command=self.click_save)
|
menu.add_command(label="Save", accelerator="Ctrl+S", command=self.click_save)
|
||||||
self.app.bind_all("<Control-s>", self.click_save)
|
self.app.bind_all("<Control-s>", self.click_save)
|
||||||
menu.add_command(label="Save As...", command=self.click_save_xml)
|
menu.add_command(label="Save As...", command=self.click_save_as)
|
||||||
menu.add_command(
|
menu.add_command(
|
||||||
label="Open...", command=self.click_open_xml, accelerator="Ctrl+O"
|
label="Open...", command=self.click_open_xml, accelerator="Ctrl+O"
|
||||||
)
|
)
|
||||||
|
@ -84,7 +86,7 @@ class Menubar(tk.Menu):
|
||||||
self.recent_menu = tk.Menu(menu)
|
self.recent_menu = tk.Menu(menu)
|
||||||
for i in self.app.guiconfig.recentfiles:
|
for i in self.app.guiconfig.recentfiles:
|
||||||
self.recent_menu.add_command(
|
self.recent_menu.add_command(
|
||||||
label=i, command=partial(self.open_recent_files, i)
|
label=i, command=partial(self.open_recent_files, Path(i))
|
||||||
)
|
)
|
||||||
menu.add_cascade(label="Recent Files", menu=self.recent_menu)
|
menu.add_cascade(label="Recent Files", menu=self.recent_menu)
|
||||||
menu.add_separator()
|
menu.add_separator()
|
||||||
|
@ -120,11 +122,6 @@ class Menubar(tk.Menu):
|
||||||
)
|
)
|
||||||
menu.add_command(label="Hide", accelerator="Ctrl+H", command=self.click_hide)
|
menu.add_command(label="Hide", accelerator="Ctrl+H", command=self.click_hide)
|
||||||
self.add_cascade(label="Edit", menu=menu)
|
self.add_cascade(label="Edit", menu=menu)
|
||||||
self.app.master.bind_all("<Control-x>", self.click_cut)
|
|
||||||
self.app.master.bind_all("<Control-c>", self.click_copy)
|
|
||||||
self.app.master.bind_all("<Control-v>", self.click_paste)
|
|
||||||
self.app.master.bind_all("<Control-d>", self.click_delete)
|
|
||||||
self.app.master.bind_all("<Control-h>", self.click_hide)
|
|
||||||
self.edit_menu = menu
|
self.edit_menu = menu
|
||||||
|
|
||||||
def draw_canvas_menu(self) -> None:
|
def draw_canvas_menu(self) -> None:
|
||||||
|
@ -272,27 +269,28 @@ class Menubar(tk.Menu):
|
||||||
menu.add_command(label="About", command=self.click_about)
|
menu.add_command(label="About", command=self.click_about)
|
||||||
self.add_cascade(label="Help", menu=menu)
|
self.add_cascade(label="Help", menu=menu)
|
||||||
|
|
||||||
def open_recent_files(self, filename: str) -> None:
|
def open_recent_files(self, file_path: Path) -> None:
|
||||||
if os.path.isfile(filename):
|
if file_path.is_file():
|
||||||
logging.debug("Open recent file %s", filename)
|
logger.debug("Open recent file %s", file_path)
|
||||||
self.open_xml_task(filename)
|
self.open_xml_task(file_path)
|
||||||
else:
|
else:
|
||||||
logging.warning("File does not exist %s", filename)
|
logger.warning("File does not exist %s", file_path)
|
||||||
|
|
||||||
def update_recent_files(self) -> None:
|
def update_recent_files(self) -> None:
|
||||||
self.recent_menu.delete(0, tk.END)
|
self.recent_menu.delete(0, tk.END)
|
||||||
for i in self.app.guiconfig.recentfiles:
|
for i in self.app.guiconfig.recentfiles:
|
||||||
self.recent_menu.add_command(
|
self.recent_menu.add_command(
|
||||||
label=i, command=partial(self.open_recent_files, i)
|
label=i, command=partial(self.open_recent_files, Path(i))
|
||||||
)
|
)
|
||||||
|
|
||||||
def click_save(self, _event=None) -> None:
|
def click_save(self, _event: tk.Event = None) -> None:
|
||||||
if self.core.session.file:
|
if self.core.session.file:
|
||||||
self.core.save_xml()
|
if self.core.save_xml():
|
||||||
|
self.add_recent_file_to_gui_config(self.core.session.file)
|
||||||
else:
|
else:
|
||||||
self.click_save_xml()
|
self.click_save_as()
|
||||||
|
|
||||||
def click_save_xml(self, _event: tk.Event = None) -> None:
|
def click_save_as(self, _event: tk.Event = None) -> None:
|
||||||
init_dir = self.core.get_xml_dir()
|
init_dir = self.core.get_xml_dir()
|
||||||
file_path = filedialog.asksaveasfilename(
|
file_path = filedialog.asksaveasfilename(
|
||||||
initialdir=init_dir,
|
initialdir=init_dir,
|
||||||
|
@ -301,8 +299,9 @@ class Menubar(tk.Menu):
|
||||||
defaultextension=".xml",
|
defaultextension=".xml",
|
||||||
)
|
)
|
||||||
if file_path:
|
if file_path:
|
||||||
self.add_recent_file_to_gui_config(file_path)
|
file_path = Path(file_path)
|
||||||
self.core.save_xml(file_path)
|
if self.core.save_xml(file_path):
|
||||||
|
self.add_recent_file_to_gui_config(file_path)
|
||||||
|
|
||||||
def click_open_xml(self, _event: tk.Event = None) -> None:
|
def click_open_xml(self, _event: tk.Event = None) -> None:
|
||||||
init_dir = self.core.get_xml_dir()
|
init_dir = self.core.get_xml_dir()
|
||||||
|
@ -312,9 +311,10 @@ class Menubar(tk.Menu):
|
||||||
filetypes=(("XML Files", "*.xml"), ("All Files", "*")),
|
filetypes=(("XML Files", "*.xml"), ("All Files", "*")),
|
||||||
)
|
)
|
||||||
if file_path:
|
if file_path:
|
||||||
|
file_path = Path(file_path)
|
||||||
self.open_xml_task(file_path)
|
self.open_xml_task(file_path)
|
||||||
|
|
||||||
def open_xml_task(self, file_path: str) -> None:
|
def open_xml_task(self, file_path: Path) -> None:
|
||||||
self.add_recent_file_to_gui_config(file_path)
|
self.add_recent_file_to_gui_config(file_path)
|
||||||
self.prompt_save_running_session()
|
self.prompt_save_running_session()
|
||||||
task = ProgressTask(self.app, "Open XML", self.core.open_xml, args=(file_path,))
|
task = ProgressTask(self.app, "Open XML", self.core.open_xml, args=(file_path,))
|
||||||
|
@ -324,21 +324,14 @@ class Menubar(tk.Menu):
|
||||||
dialog = ExecutePythonDialog(self.app)
|
dialog = ExecutePythonDialog(self.app)
|
||||||
dialog.show()
|
dialog.show()
|
||||||
|
|
||||||
def add_recent_file_to_gui_config(self, file_path) -> None:
|
def add_recent_file_to_gui_config(self, file_path: Path) -> None:
|
||||||
recent_files = self.app.guiconfig.recentfiles
|
recent_files = self.app.guiconfig.recentfiles
|
||||||
num_files = len(recent_files)
|
file_path = str(file_path)
|
||||||
if num_files == 0:
|
if file_path in recent_files:
|
||||||
recent_files.insert(0, file_path)
|
recent_files.remove(file_path)
|
||||||
elif 0 < num_files <= MAX_FILES:
|
recent_files.insert(0, file_path)
|
||||||
if file_path in recent_files:
|
if len(recent_files) > MAX_FILES:
|
||||||
recent_files.remove(file_path)
|
recent_files.pop()
|
||||||
recent_files.insert(0, file_path)
|
|
||||||
else:
|
|
||||||
if num_files == MAX_FILES:
|
|
||||||
recent_files.pop()
|
|
||||||
recent_files.insert(0, file_path)
|
|
||||||
else:
|
|
||||||
logging.error("unexpected number of recent files")
|
|
||||||
self.app.save_config()
|
self.app.save_config()
|
||||||
self.app.menubar.update_recent_files()
|
self.app.menubar.update_recent_files()
|
||||||
|
|
||||||
|
@ -411,47 +404,47 @@ class Menubar(tk.Menu):
|
||||||
|
|
||||||
def click_copy(self, _event: tk.Event = None) -> None:
|
def click_copy(self, _event: tk.Event = None) -> None:
|
||||||
canvas = self.manager.current()
|
canvas = self.manager.current()
|
||||||
canvas.copy()
|
canvas.copy_selected()
|
||||||
|
|
||||||
def click_paste(self, _event: tk.Event = None) -> None:
|
def click_paste(self, event: tk.Event = None) -> None:
|
||||||
canvas = self.manager.current()
|
canvas = self.manager.current()
|
||||||
canvas.paste()
|
canvas.paste_selected(event)
|
||||||
|
|
||||||
def click_delete(self, _event: tk.Event = None) -> None:
|
def click_delete(self, event: tk.Event = None) -> None:
|
||||||
canvas = self.manager.current()
|
canvas = self.manager.current()
|
||||||
canvas.delete_selected_objects()
|
canvas.delete_selected(event)
|
||||||
|
|
||||||
def click_hide(self, _event: tk.Event = None) -> None:
|
def click_hide(self, event: tk.Event = None) -> None:
|
||||||
canvas = self.manager.current()
|
canvas = self.manager.current()
|
||||||
canvas.hide_selected_objects()
|
canvas.hide_selected(event)
|
||||||
|
|
||||||
def click_cut(self, _event: tk.Event = None) -> None:
|
def click_cut(self, event: tk.Event = None) -> None:
|
||||||
canvas = self.manager.current()
|
canvas = self.manager.current()
|
||||||
canvas.copy()
|
canvas.copy_selected(event)
|
||||||
canvas.delete_selected_objects()
|
canvas.delete_selected(event)
|
||||||
|
|
||||||
def click_show_hidden(self, _event: tk.Event = None) -> None:
|
def click_show_hidden(self, _event: tk.Event = None) -> None:
|
||||||
for canvas in self.manager.all():
|
for canvas in self.manager.all():
|
||||||
canvas.show_hidden()
|
canvas.show_hidden()
|
||||||
|
|
||||||
def click_session_options(self) -> None:
|
def click_session_options(self) -> None:
|
||||||
logging.debug("Click options")
|
logger.debug("Click options")
|
||||||
dialog = SessionOptionsDialog(self.app)
|
dialog = SessionOptionsDialog(self.app)
|
||||||
if not dialog.has_error:
|
if not dialog.has_error:
|
||||||
dialog.show()
|
dialog.show()
|
||||||
|
|
||||||
def click_sessions(self) -> None:
|
def click_sessions(self) -> None:
|
||||||
logging.debug("Click change sessions")
|
logger.debug("Click change sessions")
|
||||||
dialog = SessionsDialog(self.app)
|
dialog = SessionsDialog(self.app)
|
||||||
dialog.show()
|
dialog.show()
|
||||||
|
|
||||||
def click_hooks(self) -> None:
|
def click_hooks(self) -> None:
|
||||||
logging.debug("Click hooks")
|
logger.debug("Click hooks")
|
||||||
dialog = HooksDialog(self.app)
|
dialog = HooksDialog(self.app)
|
||||||
dialog.show()
|
dialog.show()
|
||||||
|
|
||||||
def click_servers(self) -> None:
|
def click_servers(self) -> None:
|
||||||
logging.debug("Click emulation servers")
|
logger.debug("Click emulation servers")
|
||||||
dialog = ServersDialog(self.app)
|
dialog = ServersDialog(self.app)
|
||||||
dialog.show()
|
dialog.show()
|
||||||
|
|
||||||
|
@ -460,11 +453,11 @@ class Menubar(tk.Menu):
|
||||||
dialog.show()
|
dialog.show()
|
||||||
|
|
||||||
def click_autogrid(self) -> None:
|
def click_autogrid(self) -> None:
|
||||||
width, height = self.manager.current_dimensions
|
width, height = self.manager.current().current_dimensions
|
||||||
padding = (images.NODE_SIZE / 2) + 10
|
padding = (images.NODE_SIZE / 2) + 10
|
||||||
layout_size = padding + images.NODE_SIZE
|
layout_size = padding + images.NODE_SIZE
|
||||||
col_count = width // layout_size
|
col_count = width // layout_size
|
||||||
logging.info(
|
logger.info(
|
||||||
"auto grid layout: dimension(%s, %s) col(%s)", width, height, col_count
|
"auto grid layout: dimension(%s, %s) col(%s)", width, height, col_count
|
||||||
)
|
)
|
||||||
canvas = self.manager.current()
|
canvas = self.manager.current()
|
||||||
|
|
|
@ -8,6 +8,8 @@ from core.gui import images
|
||||||
from core.gui.appconfig import CustomNode, GuiConfig
|
from core.gui.appconfig import CustomNode, GuiConfig
|
||||||
from core.gui.images import ImageEnum
|
from core.gui.images import ImageEnum
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.gui.app import Application
|
from core.gui.app import Application
|
||||||
|
|
||||||
|
@ -29,10 +31,9 @@ ANTENNA_ICON: Optional[PhotoImage] = None
|
||||||
def setup() -> None:
|
def setup() -> None:
|
||||||
global ANTENNA_ICON
|
global ANTENNA_ICON
|
||||||
nodes = [
|
nodes = [
|
||||||
(ImageEnum.ROUTER, NodeType.DEFAULT, "Router", "router"),
|
|
||||||
(ImageEnum.HOST, NodeType.DEFAULT, "Host", "host"),
|
|
||||||
(ImageEnum.PC, NodeType.DEFAULT, "PC", "PC"),
|
(ImageEnum.PC, NodeType.DEFAULT, "PC", "PC"),
|
||||||
(ImageEnum.MDR, NodeType.DEFAULT, "MDR", "mdr"),
|
(ImageEnum.MDR, NodeType.DEFAULT, "MDR", "mdr"),
|
||||||
|
(ImageEnum.ROUTER, NodeType.DEFAULT, "Router", "router"),
|
||||||
(ImageEnum.PROUTER, NodeType.DEFAULT, "PRouter", "prouter"),
|
(ImageEnum.PROUTER, NodeType.DEFAULT, "PRouter", "prouter"),
|
||||||
(ImageEnum.DOCKER, NodeType.DOCKER, "Docker", None),
|
(ImageEnum.DOCKER, NodeType.DOCKER, "Docker", None),
|
||||||
(ImageEnum.LXC, NodeType.LXC, "LXC", None),
|
(ImageEnum.LXC, NodeType.LXC, "LXC", None),
|
||||||
|
@ -118,11 +119,11 @@ def get_icon(node: Node, app: "Application") -> PhotoImage:
|
||||||
try:
|
try:
|
||||||
image = images.from_file(node.icon, width=images.NODE_SIZE, scale=scale)
|
image = images.from_file(node.icon, width=images.NODE_SIZE, scale=scale)
|
||||||
except OSError:
|
except OSError:
|
||||||
logging.error("invalid icon: %s", node.icon)
|
logger.error("invalid icon: %s", node.icon)
|
||||||
# custom node
|
# custom node
|
||||||
elif is_custom(node):
|
elif is_custom(node):
|
||||||
image_file = _get_custom_file(app.guiconfig, node.model)
|
image_file = _get_custom_file(app.guiconfig, node.model)
|
||||||
logging.info("custom node file: %s", image_file)
|
logger.info("custom node file: %s", image_file)
|
||||||
if image_file:
|
if image_file:
|
||||||
image = images.from_file(image_file, width=images.NODE_SIZE, scale=scale)
|
image = images.from_file(image_file, width=images.NODE_SIZE, scale=scale)
|
||||||
# built in node
|
# built in node
|
||||||
|
|
|
@ -4,6 +4,8 @@ import time
|
||||||
import tkinter as tk
|
import tkinter as tk
|
||||||
from typing import TYPE_CHECKING, Any, Callable, Optional, Tuple
|
from typing import TYPE_CHECKING, Any, Callable, Optional, Tuple
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.gui.app import Application
|
from core.gui.app import Application
|
||||||
|
|
||||||
|
@ -43,7 +45,7 @@ class ProgressTask:
|
||||||
if self.callback:
|
if self.callback:
|
||||||
self.app.after(0, self.callback, *values)
|
self.app.after(0, self.callback, *values)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logging.exception("progress task exception")
|
logger.exception("progress task exception")
|
||||||
self.app.show_exception("Task Error", e)
|
self.app.show_exception("Task Error", e)
|
||||||
finally:
|
finally:
|
||||||
self.app.after(0, self.complete)
|
self.app.after(0, self.complete)
|
||||||
|
|
|
@ -20,6 +20,8 @@ from core.gui.task import ProgressTask
|
||||||
from core.gui.themes import Styles
|
from core.gui.themes import Styles
|
||||||
from core.gui.tooltip import Tooltip
|
from core.gui.tooltip import Tooltip
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.gui.app import Application
|
from core.gui.app import Application
|
||||||
|
|
||||||
|
@ -304,7 +306,6 @@ class Toolbar(ttk.Frame):
|
||||||
def start_callback(self, result: bool, exceptions: List[str]) -> None:
|
def start_callback(self, result: bool, exceptions: List[str]) -> None:
|
||||||
if result:
|
if result:
|
||||||
self.set_runtime()
|
self.set_runtime()
|
||||||
self.app.core.set_metadata()
|
|
||||||
self.app.core.show_mobility_players()
|
self.app.core.show_mobility_players()
|
||||||
else:
|
else:
|
||||||
enable_buttons(self.design_frame, enabled=True)
|
enable_buttons(self.design_frame, enabled=True)
|
||||||
|
@ -338,7 +339,7 @@ class Toolbar(ttk.Frame):
|
||||||
type_enum: NodeTypeEnum,
|
type_enum: NodeTypeEnum,
|
||||||
image: PhotoImage,
|
image: PhotoImage,
|
||||||
) -> None:
|
) -> None:
|
||||||
logging.debug("update button(%s): %s", button, node_draw)
|
logger.debug("update button(%s): %s", button, node_draw)
|
||||||
button.configure(image=image)
|
button.configure(image=image)
|
||||||
button.image = image
|
button.image = image
|
||||||
self.app.manager.node_draw = node_draw
|
self.app.manager.node_draw = node_draw
|
||||||
|
@ -399,7 +400,7 @@ class Toolbar(ttk.Frame):
|
||||||
"""
|
"""
|
||||||
redraw buttons on the toolbar, send node and link messages to grpc server
|
redraw buttons on the toolbar, send node and link messages to grpc server
|
||||||
"""
|
"""
|
||||||
logging.info("clicked stop button")
|
logger.info("clicked stop button")
|
||||||
self.app.menubar.set_state(is_runtime=False)
|
self.app.menubar.set_state(is_runtime=False)
|
||||||
self.app.core.close_mobility_players()
|
self.app.core.close_mobility_players()
|
||||||
enable_buttons(self.runtime_frame, enabled=False)
|
enable_buttons(self.runtime_frame, enabled=False)
|
||||||
|
@ -415,7 +416,7 @@ class Toolbar(ttk.Frame):
|
||||||
def update_annotation(
|
def update_annotation(
|
||||||
self, shape_type: ShapeType, image_enum: ImageEnum, image: PhotoImage
|
self, shape_type: ShapeType, image_enum: ImageEnum, image: PhotoImage
|
||||||
) -> None:
|
) -> None:
|
||||||
logging.debug("clicked annotation")
|
logger.debug("clicked annotation")
|
||||||
self.annotation_button.configure(image=image)
|
self.annotation_button.configure(image=image)
|
||||||
self.annotation_button.image = image
|
self.annotation_button.image = image
|
||||||
self.app.manager.annotation_type = shape_type
|
self.app.manager.annotation_type = shape_type
|
||||||
|
@ -433,7 +434,7 @@ class Toolbar(ttk.Frame):
|
||||||
self.marker_frame.grid()
|
self.marker_frame.grid()
|
||||||
|
|
||||||
def click_run_button(self) -> None:
|
def click_run_button(self) -> None:
|
||||||
logging.debug("Click on RUN button")
|
logger.debug("Click on RUN button")
|
||||||
dialog = RunToolDialog(self.app)
|
dialog = RunToolDialog(self.app)
|
||||||
dialog.show()
|
dialog.show()
|
||||||
|
|
||||||
|
|
|
@ -10,6 +10,8 @@ from core.gui import appconfig, themes, validation
|
||||||
from core.gui.dialogs.dialog import Dialog
|
from core.gui.dialogs.dialog import Dialog
|
||||||
from core.gui.themes import FRAME_PAD, PADX, PADY
|
from core.gui.themes import FRAME_PAD, PADX, PADY
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.gui.app import Application
|
from core.gui.app import Application
|
||||||
|
|
||||||
|
@ -161,7 +163,7 @@ class ConfigFrame(ttk.Notebook):
|
||||||
)
|
)
|
||||||
entry.grid(row=index, column=1, sticky=tk.EW)
|
entry.grid(row=index, column=1, sticky=tk.EW)
|
||||||
else:
|
else:
|
||||||
logging.error("unhandled config option type: %s", option.type)
|
logger.error("unhandled config option type: %s", option.type)
|
||||||
self.values[option.name] = value
|
self.values[option.name] = value
|
||||||
|
|
||||||
def parse_config(self) -> Dict[str, str]:
|
def parse_config(self) -> Dict[str, str]:
|
||||||
|
|
|
@ -10,6 +10,7 @@ from pyproj import Transformer
|
||||||
|
|
||||||
from core.emulator.enumerations import RegisterTlvs
|
from core.emulator.enumerations import RegisterTlvs
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
SCALE_FACTOR: float = 100.0
|
SCALE_FACTOR: float = 100.0
|
||||||
CRS_WGS84: int = 4326
|
CRS_WGS84: int = 4326
|
||||||
CRS_PROJ: int = 3857
|
CRS_PROJ: int = 3857
|
||||||
|
@ -92,7 +93,7 @@ class GeoLocation:
|
||||||
:param alt: altitude value
|
:param alt: altitude value
|
||||||
:return: x,y,z representation of provided values
|
:return: x,y,z representation of provided values
|
||||||
"""
|
"""
|
||||||
logging.debug("input lon,lat,alt(%s, %s, %s)", lon, lat, alt)
|
logger.debug("input lon,lat,alt(%s, %s, %s)", lon, lat, alt)
|
||||||
px, py = self.to_pixels.transform(lon, lat)
|
px, py = self.to_pixels.transform(lon, lat)
|
||||||
px -= self.refproj[0]
|
px -= self.refproj[0]
|
||||||
py -= self.refproj[1]
|
py -= self.refproj[1]
|
||||||
|
@ -100,7 +101,7 @@ class GeoLocation:
|
||||||
x = self.meters2pixels(px) + self.refxyz[0]
|
x = self.meters2pixels(px) + self.refxyz[0]
|
||||||
y = -(self.meters2pixels(py) + self.refxyz[1])
|
y = -(self.meters2pixels(py) + self.refxyz[1])
|
||||||
z = self.meters2pixels(pz) + self.refxyz[2]
|
z = self.meters2pixels(pz) + self.refxyz[2]
|
||||||
logging.debug("result x,y,z(%s, %s, %s)", x, y, z)
|
logger.debug("result x,y,z(%s, %s, %s)", x, y, z)
|
||||||
return x, y, z
|
return x, y, z
|
||||||
|
|
||||||
def getgeo(self, x: float, y: float, z: float) -> Tuple[float, float, float]:
|
def getgeo(self, x: float, y: float, z: float) -> Tuple[float, float, float]:
|
||||||
|
@ -112,7 +113,7 @@ class GeoLocation:
|
||||||
:param z: z value
|
:param z: z value
|
||||||
:return: lat,lon,alt representation of provided values
|
:return: lat,lon,alt representation of provided values
|
||||||
"""
|
"""
|
||||||
logging.debug("input x,y(%s, %s)", x, y)
|
logger.debug("input x,y(%s, %s)", x, y)
|
||||||
x -= self.refxyz[0]
|
x -= self.refxyz[0]
|
||||||
y = -(y - self.refxyz[1])
|
y = -(y - self.refxyz[1])
|
||||||
if z is None:
|
if z is None:
|
||||||
|
@ -123,5 +124,5 @@ class GeoLocation:
|
||||||
py = self.refproj[1] + self.pixels2meters(y)
|
py = self.refproj[1] + self.pixels2meters(y)
|
||||||
lon, lat = self.to_geo.transform(px, py)
|
lon, lat = self.to_geo.transform(px, py)
|
||||||
alt = self.refgeo[2] + self.pixels2meters(z)
|
alt = self.refgeo[2] + self.pixels2meters(z)
|
||||||
logging.debug("result lon,lat,alt(%s, %s, %s)", lon, lat, alt)
|
logger.debug("result lon,lat,alt(%s, %s, %s)", lon, lat, alt)
|
||||||
return lat, lon, alt
|
return lat, lon, alt
|
||||||
|
|
|
@ -28,6 +28,8 @@ from core.nodes.base import CoreNode
|
||||||
from core.nodes.interface import CoreInterface
|
from core.nodes.interface import CoreInterface
|
||||||
from core.nodes.network import WlanNode
|
from core.nodes.network import WlanNode
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.emulator.session import Session
|
from core.emulator.session import Session
|
||||||
|
|
||||||
|
@ -42,6 +44,43 @@ def get_mobility_node(session: "Session", node_id: int) -> Union[WlanNode, Emane
|
||||||
return session.get_node(node_id, EmaneNet)
|
return session.get_node(node_id, EmaneNet)
|
||||||
|
|
||||||
|
|
||||||
|
def get_config_int(current: int, config: Dict[str, str], name: str) -> Optional[int]:
|
||||||
|
"""
|
||||||
|
Convenience function to get config values as int.
|
||||||
|
|
||||||
|
:param current: current config value to use when one is not provided
|
||||||
|
:param config: config to get values from
|
||||||
|
:param name: name of config value to get
|
||||||
|
:return: current config value when not provided, new value otherwise
|
||||||
|
"""
|
||||||
|
value = get_config_float(current, config, name)
|
||||||
|
if value is not None:
|
||||||
|
value = int(value)
|
||||||
|
return value
|
||||||
|
|
||||||
|
|
||||||
|
def get_config_float(
|
||||||
|
current: Union[int, float], config: Dict[str, str], name: str
|
||||||
|
) -> Optional[float]:
|
||||||
|
"""
|
||||||
|
Convenience function to get config values as float.
|
||||||
|
|
||||||
|
:param current: current config value to use when one is not provided
|
||||||
|
:param config: config to get values from
|
||||||
|
:param name: name of config value to get
|
||||||
|
:return: current config value when not provided, new value otherwise
|
||||||
|
"""
|
||||||
|
value = config.get(name)
|
||||||
|
if value is not None:
|
||||||
|
if value == "":
|
||||||
|
value = None
|
||||||
|
else:
|
||||||
|
value = float(value)
|
||||||
|
else:
|
||||||
|
value = current
|
||||||
|
return value
|
||||||
|
|
||||||
|
|
||||||
class MobilityManager(ModelManager):
|
class MobilityManager(ModelManager):
|
||||||
"""
|
"""
|
||||||
Member of session class for handling configuration data for mobility and
|
Member of session class for handling configuration data for mobility and
|
||||||
|
@ -81,7 +120,7 @@ class MobilityManager(ModelManager):
|
||||||
if node_ids is None:
|
if node_ids is None:
|
||||||
node_ids = self.nodes()
|
node_ids = self.nodes()
|
||||||
for node_id in node_ids:
|
for node_id in node_ids:
|
||||||
logging.debug(
|
logger.debug(
|
||||||
"node(%s) mobility startup: %s", node_id, self.get_all_configs(node_id)
|
"node(%s) mobility startup: %s", node_id, self.get_all_configs(node_id)
|
||||||
)
|
)
|
||||||
try:
|
try:
|
||||||
|
@ -95,8 +134,8 @@ class MobilityManager(ModelManager):
|
||||||
if node.mobility:
|
if node.mobility:
|
||||||
self.session.event_loop.add_event(0.0, node.mobility.startup)
|
self.session.event_loop.add_event(0.0, node.mobility.startup)
|
||||||
except CoreError:
|
except CoreError:
|
||||||
logging.exception("mobility startup error")
|
logger.exception("mobility startup error")
|
||||||
logging.warning(
|
logger.warning(
|
||||||
"skipping mobility configuration for unknown node: %s", node_id
|
"skipping mobility configuration for unknown node: %s", node_id
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -114,7 +153,7 @@ class MobilityManager(ModelManager):
|
||||||
try:
|
try:
|
||||||
node = get_mobility_node(self.session, node_id)
|
node = get_mobility_node(self.session, node_id)
|
||||||
except CoreError:
|
except CoreError:
|
||||||
logging.exception(
|
logger.exception(
|
||||||
"ignoring event for model(%s), unknown node(%s)", name, node_id
|
"ignoring event for model(%s), unknown node(%s)", name, node_id
|
||||||
)
|
)
|
||||||
return
|
return
|
||||||
|
@ -124,17 +163,17 @@ class MobilityManager(ModelManager):
|
||||||
for model in models:
|
for model in models:
|
||||||
cls = self.models.get(model)
|
cls = self.models.get(model)
|
||||||
if not cls:
|
if not cls:
|
||||||
logging.warning("ignoring event for unknown model '%s'", model)
|
logger.warning("ignoring event for unknown model '%s'", model)
|
||||||
continue
|
continue
|
||||||
if cls.config_type in [RegisterTlvs.WIRELESS, RegisterTlvs.MOBILITY]:
|
if cls.config_type in [RegisterTlvs.WIRELESS, RegisterTlvs.MOBILITY]:
|
||||||
model = node.mobility
|
model = node.mobility
|
||||||
else:
|
else:
|
||||||
continue
|
continue
|
||||||
if model is None:
|
if model is None:
|
||||||
logging.warning("ignoring event, %s has no model", node.name)
|
logger.warning("ignoring event, %s has no model", node.name)
|
||||||
continue
|
continue
|
||||||
if cls.name != model.name:
|
if cls.name != model.name:
|
||||||
logging.warning(
|
logger.warning(
|
||||||
"ignoring event for %s wrong model %s,%s",
|
"ignoring event for %s wrong model %s,%s",
|
||||||
node.name,
|
node.name,
|
||||||
cls.name,
|
cls.name,
|
||||||
|
@ -236,35 +275,35 @@ class BasicRangeModel(WirelessModel):
|
||||||
name: str = "basic_range"
|
name: str = "basic_range"
|
||||||
options: List[Configuration] = [
|
options: List[Configuration] = [
|
||||||
Configuration(
|
Configuration(
|
||||||
_id="range",
|
id="range",
|
||||||
_type=ConfigDataTypes.UINT32,
|
type=ConfigDataTypes.UINT32,
|
||||||
default="275",
|
default="275",
|
||||||
label="wireless range (pixels)",
|
label="wireless range (pixels)",
|
||||||
),
|
),
|
||||||
Configuration(
|
Configuration(
|
||||||
_id="bandwidth",
|
id="bandwidth",
|
||||||
_type=ConfigDataTypes.UINT64,
|
type=ConfigDataTypes.UINT64,
|
||||||
default="54000000",
|
default="54000000",
|
||||||
label="bandwidth (bps)",
|
label="bandwidth (bps)",
|
||||||
),
|
),
|
||||||
Configuration(
|
Configuration(
|
||||||
_id="jitter",
|
id="jitter",
|
||||||
_type=ConfigDataTypes.UINT64,
|
type=ConfigDataTypes.UINT64,
|
||||||
default="0",
|
default="0",
|
||||||
label="transmission jitter (usec)",
|
label="transmission jitter (usec)",
|
||||||
),
|
),
|
||||||
Configuration(
|
Configuration(
|
||||||
_id="delay",
|
id="delay",
|
||||||
_type=ConfigDataTypes.UINT64,
|
type=ConfigDataTypes.UINT64,
|
||||||
default="5000",
|
default="5000",
|
||||||
label="transmission delay (usec)",
|
label="transmission delay (usec)",
|
||||||
),
|
),
|
||||||
Configuration(
|
Configuration(
|
||||||
_id="error", _type=ConfigDataTypes.STRING, default="0", label="loss (%)"
|
id="error", type=ConfigDataTypes.STRING, default="0", label="loss (%)"
|
||||||
),
|
),
|
||||||
Configuration(
|
Configuration(
|
||||||
_id="promiscuous",
|
id="promiscuous",
|
||||||
_type=ConfigDataTypes.BOOL,
|
type=ConfigDataTypes.BOOL,
|
||||||
default="0",
|
default="0",
|
||||||
label="promiscuous mode",
|
label="promiscuous mode",
|
||||||
),
|
),
|
||||||
|
@ -293,25 +332,6 @@ class BasicRangeModel(WirelessModel):
|
||||||
self.jitter: Optional[int] = None
|
self.jitter: Optional[int] = None
|
||||||
self.promiscuous: bool = False
|
self.promiscuous: bool = False
|
||||||
|
|
||||||
def _get_config(self, current_value: int, config: Dict[str, str], name: str) -> int:
|
|
||||||
"""
|
|
||||||
Convenience for updating value to use from a provided configuration.
|
|
||||||
|
|
||||||
:param current_value: current config value to use when one is not provided
|
|
||||||
:param config: config to get values from
|
|
||||||
:param name: name of config value to get
|
|
||||||
:return: current config value when not provided, new value otherwise
|
|
||||||
"""
|
|
||||||
value = config.get(name)
|
|
||||||
if value is not None:
|
|
||||||
if value == "":
|
|
||||||
value = None
|
|
||||||
else:
|
|
||||||
value = int(float(value))
|
|
||||||
else:
|
|
||||||
value = current_value
|
|
||||||
return value
|
|
||||||
|
|
||||||
def setlinkparams(self) -> None:
|
def setlinkparams(self) -> None:
|
||||||
"""
|
"""
|
||||||
Apply link parameters to all interfaces. This is invoked from
|
Apply link parameters to all interfaces. This is invoked from
|
||||||
|
@ -406,20 +426,20 @@ class BasicRangeModel(WirelessModel):
|
||||||
a = min(iface, iface2)
|
a = min(iface, iface2)
|
||||||
b = max(iface, iface2)
|
b = max(iface, iface2)
|
||||||
|
|
||||||
with self.wlan._linked_lock:
|
with self.wlan.linked_lock:
|
||||||
linked = self.wlan.linked(a, b)
|
linked = self.wlan.is_linked(a, b)
|
||||||
if d > self.range:
|
if d > self.range:
|
||||||
if linked:
|
if linked:
|
||||||
logging.debug("was linked, unlinking")
|
logger.debug("was linked, unlinking")
|
||||||
self.wlan.unlink(a, b)
|
self.wlan.unlink(a, b)
|
||||||
self.sendlinkmsg(a, b, unlink=True)
|
self.sendlinkmsg(a, b, unlink=True)
|
||||||
else:
|
else:
|
||||||
if not linked:
|
if not linked:
|
||||||
logging.debug("was not linked, linking")
|
logger.debug("was not linked, linking")
|
||||||
self.wlan.link(a, b)
|
self.wlan.link(a, b)
|
||||||
self.sendlinkmsg(a, b)
|
self.sendlinkmsg(a, b)
|
||||||
except KeyError:
|
except KeyError:
|
||||||
logging.exception("error getting interfaces during calclinkS")
|
logger.exception("error getting interfaces during calclink")
|
||||||
|
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def calcdistance(
|
def calcdistance(
|
||||||
|
@ -446,15 +466,15 @@ class BasicRangeModel(WirelessModel):
|
||||||
:param config: values to update configuration
|
:param config: values to update configuration
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
self.range = self._get_config(self.range, config, "range")
|
self.range = get_config_int(self.range, config, "range")
|
||||||
if self.range is None:
|
if self.range is None:
|
||||||
self.range = 0
|
self.range = 0
|
||||||
logging.debug("wlan %s set range to %s", self.wlan.name, self.range)
|
logger.debug("wlan %s set range to %s", self.wlan.name, self.range)
|
||||||
self.bw = self._get_config(self.bw, config, "bandwidth")
|
self.bw = get_config_int(self.bw, config, "bandwidth")
|
||||||
self.delay = self._get_config(self.delay, config, "delay")
|
self.delay = get_config_int(self.delay, config, "delay")
|
||||||
self.loss = self._get_config(self.loss, config, "error")
|
self.loss = get_config_float(self.loss, config, "error")
|
||||||
self.jitter = self._get_config(self.jitter, config, "jitter")
|
self.jitter = get_config_int(self.jitter, config, "jitter")
|
||||||
promiscuous = config["promiscuous"] == "1"
|
promiscuous = config.get("promiscuous", "0") == "1"
|
||||||
if self.promiscuous and not promiscuous:
|
if self.promiscuous and not promiscuous:
|
||||||
self.wlan.net_client.set_mac_learning(self.wlan.brname, LEARNING_ENABLED)
|
self.wlan.net_client.set_mac_learning(self.wlan.brname, LEARNING_ENABLED)
|
||||||
elif not self.promiscuous and promiscuous:
|
elif not self.promiscuous and promiscuous:
|
||||||
|
@ -506,10 +526,10 @@ class BasicRangeModel(WirelessModel):
|
||||||
:return: all link data
|
:return: all link data
|
||||||
"""
|
"""
|
||||||
all_links = []
|
all_links = []
|
||||||
with self.wlan._linked_lock:
|
with self.wlan.linked_lock:
|
||||||
for a in self.wlan._linked:
|
for a in self.wlan.linked:
|
||||||
for b in self.wlan._linked[a]:
|
for b in self.wlan.linked[a]:
|
||||||
if self.wlan._linked[a][b]:
|
if self.wlan.linked[a][b]:
|
||||||
all_links.append(self.create_link_data(a, b, flags))
|
all_links.append(self.create_link_data(a, b, flags))
|
||||||
return all_links
|
return all_links
|
||||||
|
|
||||||
|
@ -868,40 +888,38 @@ class Ns2ScriptedMobility(WayPointMobility):
|
||||||
name: str = "ns2script"
|
name: str = "ns2script"
|
||||||
options: List[Configuration] = [
|
options: List[Configuration] = [
|
||||||
Configuration(
|
Configuration(
|
||||||
_id="file", _type=ConfigDataTypes.STRING, label="mobility script file"
|
id="file", type=ConfigDataTypes.STRING, label="mobility script file"
|
||||||
),
|
),
|
||||||
Configuration(
|
Configuration(
|
||||||
_id="refresh_ms",
|
id="refresh_ms",
|
||||||
_type=ConfigDataTypes.UINT32,
|
type=ConfigDataTypes.UINT32,
|
||||||
default="50",
|
default="50",
|
||||||
label="refresh time (ms)",
|
label="refresh time (ms)",
|
||||||
),
|
),
|
||||||
|
Configuration(id="loop", type=ConfigDataTypes.BOOL, default="1", label="loop"),
|
||||||
Configuration(
|
Configuration(
|
||||||
_id="loop", _type=ConfigDataTypes.BOOL, default="1", label="loop"
|
id="autostart",
|
||||||
),
|
type=ConfigDataTypes.STRING,
|
||||||
Configuration(
|
|
||||||
_id="autostart",
|
|
||||||
_type=ConfigDataTypes.STRING,
|
|
||||||
label="auto-start seconds (0.0 for runtime)",
|
label="auto-start seconds (0.0 for runtime)",
|
||||||
),
|
),
|
||||||
Configuration(
|
Configuration(
|
||||||
_id="map",
|
id="map",
|
||||||
_type=ConfigDataTypes.STRING,
|
type=ConfigDataTypes.STRING,
|
||||||
label="node mapping (optional, e.g. 0:1,1:2,2:3)",
|
label="node mapping (optional, e.g. 0:1,1:2,2:3)",
|
||||||
),
|
),
|
||||||
Configuration(
|
Configuration(
|
||||||
_id="script_start",
|
id="script_start",
|
||||||
_type=ConfigDataTypes.STRING,
|
type=ConfigDataTypes.STRING,
|
||||||
label="script file to run upon start",
|
label="script file to run upon start",
|
||||||
),
|
),
|
||||||
Configuration(
|
Configuration(
|
||||||
_id="script_pause",
|
id="script_pause",
|
||||||
_type=ConfigDataTypes.STRING,
|
type=ConfigDataTypes.STRING,
|
||||||
label="script file to run upon pause",
|
label="script file to run upon pause",
|
||||||
),
|
),
|
||||||
Configuration(
|
Configuration(
|
||||||
_id="script_stop",
|
id="script_stop",
|
||||||
_type=ConfigDataTypes.STRING,
|
type=ConfigDataTypes.STRING,
|
||||||
label="script file to run upon stop",
|
label="script file to run upon stop",
|
||||||
),
|
),
|
||||||
]
|
]
|
||||||
|
@ -920,7 +938,7 @@ class Ns2ScriptedMobility(WayPointMobility):
|
||||||
:param _id: object id
|
:param _id: object id
|
||||||
"""
|
"""
|
||||||
super().__init__(session, _id)
|
super().__init__(session, _id)
|
||||||
self.file: Optional[str] = None
|
self.file: Optional[Path] = None
|
||||||
self.autostart: Optional[str] = None
|
self.autostart: Optional[str] = None
|
||||||
self.nodemap: Dict[int, int] = {}
|
self.nodemap: Dict[int, int] = {}
|
||||||
self.script_start: Optional[str] = None
|
self.script_start: Optional[str] = None
|
||||||
|
@ -928,8 +946,8 @@ class Ns2ScriptedMobility(WayPointMobility):
|
||||||
self.script_stop: Optional[str] = None
|
self.script_stop: Optional[str] = None
|
||||||
|
|
||||||
def update_config(self, config: Dict[str, str]) -> None:
|
def update_config(self, config: Dict[str, str]) -> None:
|
||||||
self.file = config["file"]
|
self.file = Path(config["file"])
|
||||||
logging.info(
|
logger.info(
|
||||||
"ns-2 scripted mobility configured for WLAN %d using file: %s",
|
"ns-2 scripted mobility configured for WLAN %d using file: %s",
|
||||||
self.id,
|
self.id,
|
||||||
self.file,
|
self.file,
|
||||||
|
@ -953,15 +971,15 @@ class Ns2ScriptedMobility(WayPointMobility):
|
||||||
|
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
filename = self.findfile(self.file)
|
file_path = self.findfile(self.file)
|
||||||
try:
|
try:
|
||||||
f = open(filename, "r")
|
f = file_path.open("r")
|
||||||
except IOError:
|
except IOError:
|
||||||
logging.exception(
|
logger.exception(
|
||||||
"ns-2 scripted mobility failed to load file: %s", self.file
|
"ns-2 scripted mobility failed to load file: %s", self.file
|
||||||
)
|
)
|
||||||
return
|
return
|
||||||
logging.info("reading ns-2 script file: %s", filename)
|
logger.info("reading ns-2 script file: %s", file_path)
|
||||||
ln = 0
|
ln = 0
|
||||||
ix = iy = iz = None
|
ix = iy = iz = None
|
||||||
inodenum = None
|
inodenum = None
|
||||||
|
@ -977,13 +995,13 @@ class Ns2ScriptedMobility(WayPointMobility):
|
||||||
# waypoints:
|
# waypoints:
|
||||||
# $ns_ at 1.00 "$node_(6) setdest 500.0 178.0 25.0"
|
# $ns_ at 1.00 "$node_(6) setdest 500.0 178.0 25.0"
|
||||||
parts = line.split()
|
parts = line.split()
|
||||||
time = float(parts[2])
|
line_time = float(parts[2])
|
||||||
nodenum = parts[3][1 + parts[3].index("(") : parts[3].index(")")]
|
nodenum = parts[3][1 + parts[3].index("(") : parts[3].index(")")]
|
||||||
x = float(parts[5])
|
x = float(parts[5])
|
||||||
y = float(parts[6])
|
y = float(parts[6])
|
||||||
z = None
|
z = None
|
||||||
speed = float(parts[7].strip('"'))
|
speed = float(parts[7].strip('"'))
|
||||||
self.addwaypoint(time, self.map(nodenum), x, y, z, speed)
|
self.addwaypoint(line_time, self.map(nodenum), x, y, z, speed)
|
||||||
elif line[:7] == "$node_(":
|
elif line[:7] == "$node_(":
|
||||||
# initial position (time=0, speed=0):
|
# initial position (time=0, speed=0):
|
||||||
# $node_(6) set X_ 780.0
|
# $node_(6) set X_ 780.0
|
||||||
|
@ -1004,38 +1022,38 @@ class Ns2ScriptedMobility(WayPointMobility):
|
||||||
else:
|
else:
|
||||||
raise ValueError
|
raise ValueError
|
||||||
except ValueError:
|
except ValueError:
|
||||||
logging.exception(
|
logger.exception(
|
||||||
"skipping line %d of file %s '%s'", ln, self.file, line
|
"skipping line %d of file %s '%s'", ln, self.file, line
|
||||||
)
|
)
|
||||||
continue
|
continue
|
||||||
if ix is not None and iy is not None:
|
if ix is not None and iy is not None:
|
||||||
self.addinitial(self.map(inodenum), ix, iy, iz)
|
self.addinitial(self.map(inodenum), ix, iy, iz)
|
||||||
|
|
||||||
def findfile(self, file_name: str) -> str:
|
def findfile(self, file_path: Path) -> Path:
|
||||||
"""
|
"""
|
||||||
Locate a script file. If the specified file doesn't exist, look in the
|
Locate a script file. If the specified file doesn't exist, look in the
|
||||||
same directory as the scenario file, or in gui directories.
|
same directory as the scenario file, or in gui directories.
|
||||||
|
|
||||||
:param file_name: file name to find
|
:param file_path: file name to find
|
||||||
:return: absolute path to the file
|
:return: absolute path to the file
|
||||||
:raises CoreError: when file is not found
|
:raises CoreError: when file is not found
|
||||||
"""
|
"""
|
||||||
file_path = Path(file_name).expanduser()
|
file_path = file_path.expanduser()
|
||||||
if file_path.exists():
|
if file_path.exists():
|
||||||
return str(file_path)
|
return file_path
|
||||||
if self.session.file_name:
|
if self.session.file_path:
|
||||||
file_path = Path(self.session.file_name).parent / file_name
|
session_file_path = self.session.file_path.parent / file_path
|
||||||
if file_path.exists():
|
if session_file_path.exists():
|
||||||
return str(file_path)
|
return session_file_path
|
||||||
if self.session.user:
|
if self.session.user:
|
||||||
user_path = Path(f"~{self.session.user}").expanduser()
|
user_path = Path(f"~{self.session.user}").expanduser()
|
||||||
file_path = user_path / ".core" / "configs" / file_name
|
configs_path = user_path / ".core" / "configs" / file_path
|
||||||
if file_path.exists():
|
if configs_path.exists():
|
||||||
return str(file_path)
|
return configs_path
|
||||||
file_path = user_path / ".coregui" / "mobility" / file_name
|
mobility_path = user_path / ".coregui" / "mobility" / file_path
|
||||||
if file_path.exists():
|
if mobility_path.exists():
|
||||||
return str(file_path)
|
return mobility_path
|
||||||
raise CoreError(f"invalid file: {file_name}")
|
raise CoreError(f"invalid file: {file_path}")
|
||||||
|
|
||||||
def parsemap(self, mapstr: str) -> None:
|
def parsemap(self, mapstr: str) -> None:
|
||||||
"""
|
"""
|
||||||
|
@ -1047,7 +1065,6 @@ class Ns2ScriptedMobility(WayPointMobility):
|
||||||
self.nodemap = {}
|
self.nodemap = {}
|
||||||
if mapstr.strip() == "":
|
if mapstr.strip() == "":
|
||||||
return
|
return
|
||||||
|
|
||||||
for pair in mapstr.split(","):
|
for pair in mapstr.split(","):
|
||||||
parts = pair.split(":")
|
parts = pair.split(":")
|
||||||
try:
|
try:
|
||||||
|
@ -1055,7 +1072,7 @@ class Ns2ScriptedMobility(WayPointMobility):
|
||||||
raise ValueError
|
raise ValueError
|
||||||
self.nodemap[int(parts[0])] = int(parts[1])
|
self.nodemap[int(parts[0])] = int(parts[1])
|
||||||
except ValueError:
|
except ValueError:
|
||||||
logging.exception("ns-2 mobility node map error")
|
logger.exception("ns-2 mobility node map error")
|
||||||
|
|
||||||
def map(self, nodenum: str) -> int:
|
def map(self, nodenum: str) -> int:
|
||||||
"""
|
"""
|
||||||
|
@ -1077,19 +1094,19 @@ class Ns2ScriptedMobility(WayPointMobility):
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
if self.autostart == "":
|
if self.autostart == "":
|
||||||
logging.info("not auto-starting ns-2 script for %s", self.net.name)
|
logger.info("not auto-starting ns-2 script for %s", self.net.name)
|
||||||
return
|
return
|
||||||
try:
|
try:
|
||||||
t = float(self.autostart)
|
t = float(self.autostart)
|
||||||
except ValueError:
|
except ValueError:
|
||||||
logging.exception(
|
logger.exception(
|
||||||
"Invalid auto-start seconds specified '%s' for %s",
|
"Invalid auto-start seconds specified '%s' for %s",
|
||||||
self.autostart,
|
self.autostart,
|
||||||
self.net.name,
|
self.net.name,
|
||||||
)
|
)
|
||||||
return
|
return
|
||||||
self.movenodesinitial()
|
self.movenodesinitial()
|
||||||
logging.info("scheduling ns-2 script for %s autostart at %s", self.net.name, t)
|
logger.info("scheduling ns-2 script for %s autostart at %s", self.net.name, t)
|
||||||
self.state = self.STATE_RUNNING
|
self.state = self.STATE_RUNNING
|
||||||
self.session.event_loop.add_event(t, self.run)
|
self.session.event_loop.add_event(t, self.run)
|
||||||
|
|
||||||
|
@ -1099,7 +1116,7 @@ class Ns2ScriptedMobility(WayPointMobility):
|
||||||
|
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
logging.info("starting script: %s", self.file)
|
logger.info("starting script: %s", self.file)
|
||||||
laststate = self.state
|
laststate = self.state
|
||||||
super().start()
|
super().start()
|
||||||
if laststate == self.STATE_PAUSED:
|
if laststate == self.STATE_PAUSED:
|
||||||
|
@ -1120,7 +1137,7 @@ class Ns2ScriptedMobility(WayPointMobility):
|
||||||
|
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
logging.info("pausing script: %s", self.file)
|
logger.info("pausing script: %s", self.file)
|
||||||
super().pause()
|
super().pause()
|
||||||
self.statescript("pause")
|
self.statescript("pause")
|
||||||
|
|
||||||
|
@ -1132,7 +1149,7 @@ class Ns2ScriptedMobility(WayPointMobility):
|
||||||
position
|
position
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
logging.info("stopping script: %s", self.file)
|
logger.info("stopping script: %s", self.file)
|
||||||
super().stop(move_initial=move_initial)
|
super().stop(move_initial=move_initial)
|
||||||
self.statescript("stop")
|
self.statescript("stop")
|
||||||
|
|
||||||
|
@ -1152,8 +1169,7 @@ class Ns2ScriptedMobility(WayPointMobility):
|
||||||
filename = self.script_stop
|
filename = self.script_stop
|
||||||
if filename is None or filename == "":
|
if filename is None or filename == "":
|
||||||
return
|
return
|
||||||
|
filename = Path(filename)
|
||||||
filename = self.findfile(filename)
|
filename = self.findfile(filename)
|
||||||
args = f"{BASH} {filename} {typestr}"
|
args = f"{BASH} {filename} {typestr}"
|
||||||
utils.cmd(
|
utils.cmd(args, cwd=self.session.directory, env=self.session.get_environment())
|
||||||
args, cwd=self.session.session_dir, env=self.session.get_environment()
|
|
||||||
)
|
|
||||||
|
|
|
@ -3,9 +3,9 @@ Defines the base logic for nodes used within core.
|
||||||
"""
|
"""
|
||||||
import abc
|
import abc
|
||||||
import logging
|
import logging
|
||||||
import os
|
|
||||||
import shutil
|
import shutil
|
||||||
import threading
|
import threading
|
||||||
|
from pathlib import Path
|
||||||
from threading import RLock
|
from threading import RLock
|
||||||
from typing import TYPE_CHECKING, Dict, List, Optional, Tuple, Type, Union
|
from typing import TYPE_CHECKING, Dict, List, Optional, Tuple, Type, Union
|
||||||
|
|
||||||
|
@ -21,6 +21,8 @@ from core.nodes.client import VnodeClient
|
||||||
from core.nodes.interface import CoreInterface, TunTap, Veth
|
from core.nodes.interface import CoreInterface, TunTap, Veth
|
||||||
from core.nodes.netclient import LinuxNetClient, get_net_client
|
from core.nodes.netclient import LinuxNetClient, get_net_client
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.emulator.distributed import DistributedServer
|
from core.emulator.distributed import DistributedServer
|
||||||
from core.emulator.session import Session
|
from core.emulator.session import Session
|
||||||
|
@ -30,6 +32,8 @@ if TYPE_CHECKING:
|
||||||
CoreServices = List[Union[CoreService, Type[CoreService]]]
|
CoreServices = List[Union[CoreService, Type[CoreService]]]
|
||||||
ConfigServiceType = Type[ConfigService]
|
ConfigServiceType = Type[ConfigService]
|
||||||
|
|
||||||
|
PRIVATE_DIRS: List[Path] = [Path("/var/run"), Path("/var/log")]
|
||||||
|
|
||||||
|
|
||||||
class NodeBase(abc.ABC):
|
class NodeBase(abc.ABC):
|
||||||
"""
|
"""
|
||||||
|
@ -97,7 +101,7 @@ class NodeBase(abc.ABC):
|
||||||
self,
|
self,
|
||||||
args: str,
|
args: str,
|
||||||
env: Dict[str, str] = None,
|
env: Dict[str, str] = None,
|
||||||
cwd: str = None,
|
cwd: Path = None,
|
||||||
wait: bool = True,
|
wait: bool = True,
|
||||||
shell: bool = False,
|
shell: bool = False,
|
||||||
) -> str:
|
) -> str:
|
||||||
|
@ -221,7 +225,7 @@ class CoreNodeBase(NodeBase):
|
||||||
"""
|
"""
|
||||||
super().__init__(session, _id, name, server)
|
super().__init__(session, _id, name, server)
|
||||||
self.config_services: Dict[str, "ConfigService"] = {}
|
self.config_services: Dict[str, "ConfigService"] = {}
|
||||||
self.nodedir: Optional[str] = None
|
self.directory: Optional[Path] = None
|
||||||
self.tmpnodedir: bool = False
|
self.tmpnodedir: bool = False
|
||||||
|
|
||||||
@abc.abstractmethod
|
@abc.abstractmethod
|
||||||
|
@ -233,11 +237,21 @@ class CoreNodeBase(NodeBase):
|
||||||
raise NotImplementedError
|
raise NotImplementedError
|
||||||
|
|
||||||
@abc.abstractmethod
|
@abc.abstractmethod
|
||||||
def nodefile(self, filename: str, contents: str, mode: int = 0o644) -> None:
|
def create_dir(self, dir_path: Path) -> None:
|
||||||
|
"""
|
||||||
|
Create a node private directory.
|
||||||
|
|
||||||
|
:param dir_path: path to create
|
||||||
|
:return: nothing
|
||||||
|
"""
|
||||||
|
raise NotImplementedError
|
||||||
|
|
||||||
|
@abc.abstractmethod
|
||||||
|
def create_file(self, file_path: Path, contents: str, mode: int = 0o644) -> None:
|
||||||
"""
|
"""
|
||||||
Create a node file with a given mode.
|
Create a node file with a given mode.
|
||||||
|
|
||||||
:param filename: name of file to create
|
:param file_path: name of file to create
|
||||||
:param contents: contents of file
|
:param contents: contents of file
|
||||||
:param mode: mode for file
|
:param mode: mode for file
|
||||||
:return: nothing
|
:return: nothing
|
||||||
|
@ -245,12 +259,25 @@ class CoreNodeBase(NodeBase):
|
||||||
raise NotImplementedError
|
raise NotImplementedError
|
||||||
|
|
||||||
@abc.abstractmethod
|
@abc.abstractmethod
|
||||||
def addfile(self, srcname: str, filename: str) -> None:
|
def copy_file(self, src_path: Path, dst_path: Path, mode: int = None) -> None:
|
||||||
|
"""
|
||||||
|
Copy source file to node host destination, updating the file mode when
|
||||||
|
provided.
|
||||||
|
|
||||||
|
:param src_path: source file to copy
|
||||||
|
:param dst_path: node host destination
|
||||||
|
:param mode: file mode
|
||||||
|
:return: nothing
|
||||||
|
"""
|
||||||
|
raise NotImplementedError
|
||||||
|
|
||||||
|
@abc.abstractmethod
|
||||||
|
def addfile(self, src_path: Path, file_path: Path) -> None:
|
||||||
"""
|
"""
|
||||||
Add a file.
|
Add a file.
|
||||||
|
|
||||||
:param srcname: source file name
|
:param src_path: source file path
|
||||||
:param filename: file name to add
|
:param file_path: file name to add
|
||||||
:return: nothing
|
:return: nothing
|
||||||
:raises CoreCommandError: when a non-zero exit status occurs
|
:raises CoreCommandError: when a non-zero exit status occurs
|
||||||
"""
|
"""
|
||||||
|
@ -302,6 +329,21 @@ class CoreNodeBase(NodeBase):
|
||||||
"""
|
"""
|
||||||
raise NotImplementedError
|
raise NotImplementedError
|
||||||
|
|
||||||
|
def host_path(self, path: Path, is_dir: bool = False) -> Path:
|
||||||
|
"""
|
||||||
|
Return the name of a node"s file on the host filesystem.
|
||||||
|
|
||||||
|
:param path: path to translate to host path
|
||||||
|
:param is_dir: True if path is a directory path, False otherwise
|
||||||
|
:return: path to file
|
||||||
|
"""
|
||||||
|
if is_dir:
|
||||||
|
directory = str(path).strip("/").replace("/", ".")
|
||||||
|
return self.directory / directory
|
||||||
|
else:
|
||||||
|
directory = str(path.parent).strip("/").replace("/", ".")
|
||||||
|
return self.directory / directory / path.name
|
||||||
|
|
||||||
def add_config_service(self, service_class: "ConfigServiceType") -> None:
|
def add_config_service(self, service_class: "ConfigServiceType") -> None:
|
||||||
"""
|
"""
|
||||||
Adds a configuration service to the node.
|
Adds a configuration service to the node.
|
||||||
|
@ -345,9 +387,9 @@ class CoreNodeBase(NodeBase):
|
||||||
|
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
if self.nodedir is None:
|
if self.directory is None:
|
||||||
self.nodedir = os.path.join(self.session.session_dir, self.name + ".conf")
|
self.directory = self.session.directory / f"{self.name}.conf"
|
||||||
self.host_cmd(f"mkdir -p {self.nodedir}")
|
self.host_cmd(f"mkdir -p {self.directory}")
|
||||||
self.tmpnodedir = True
|
self.tmpnodedir = True
|
||||||
else:
|
else:
|
||||||
self.tmpnodedir = False
|
self.tmpnodedir = False
|
||||||
|
@ -362,7 +404,7 @@ class CoreNodeBase(NodeBase):
|
||||||
if preserve:
|
if preserve:
|
||||||
return
|
return
|
||||||
if self.tmpnodedir:
|
if self.tmpnodedir:
|
||||||
self.host_cmd(f"rm -rf {self.nodedir}")
|
self.host_cmd(f"rm -rf {self.directory}")
|
||||||
|
|
||||||
def add_iface(self, iface: CoreInterface, iface_id: int) -> None:
|
def add_iface(self, iface: CoreInterface, iface_id: int) -> None:
|
||||||
"""
|
"""
|
||||||
|
@ -387,7 +429,7 @@ class CoreNodeBase(NodeBase):
|
||||||
if iface_id not in self.ifaces:
|
if iface_id not in self.ifaces:
|
||||||
raise CoreError(f"node({self.name}) interface({iface_id}) does not exist")
|
raise CoreError(f"node({self.name}) interface({iface_id}) does not exist")
|
||||||
iface = self.ifaces.pop(iface_id)
|
iface = self.ifaces.pop(iface_id)
|
||||||
logging.info("node(%s) removing interface(%s)", self.name, iface.name)
|
logger.info("node(%s) removing interface(%s)", self.name, iface.name)
|
||||||
iface.detachnet()
|
iface.detachnet()
|
||||||
iface.shutdown()
|
iface.shutdown()
|
||||||
|
|
||||||
|
@ -458,7 +500,7 @@ class CoreNode(CoreNodeBase):
|
||||||
session: "Session",
|
session: "Session",
|
||||||
_id: int = None,
|
_id: int = None,
|
||||||
name: str = None,
|
name: str = None,
|
||||||
nodedir: str = None,
|
directory: Path = None,
|
||||||
server: "DistributedServer" = None,
|
server: "DistributedServer" = None,
|
||||||
) -> None:
|
) -> None:
|
||||||
"""
|
"""
|
||||||
|
@ -467,19 +509,17 @@ class CoreNode(CoreNodeBase):
|
||||||
:param session: core session instance
|
:param session: core session instance
|
||||||
:param _id: object id
|
:param _id: object id
|
||||||
:param name: object name
|
:param name: object name
|
||||||
:param nodedir: node directory
|
:param directory: node directory
|
||||||
:param server: remote server node
|
:param server: remote server node
|
||||||
will run on, default is None for localhost
|
will run on, default is None for localhost
|
||||||
"""
|
"""
|
||||||
super().__init__(session, _id, name, server)
|
super().__init__(session, _id, name, server)
|
||||||
self.nodedir: Optional[str] = nodedir
|
self.directory: Optional[Path] = directory
|
||||||
self.ctrlchnlname: str = os.path.abspath(
|
self.ctrlchnlname: Path = self.session.directory / self.name
|
||||||
os.path.join(self.session.session_dir, self.name)
|
|
||||||
)
|
|
||||||
self.client: Optional[VnodeClient] = None
|
self.client: Optional[VnodeClient] = None
|
||||||
self.pid: Optional[int] = None
|
self.pid: Optional[int] = None
|
||||||
self.lock: RLock = RLock()
|
self.lock: RLock = RLock()
|
||||||
self._mounts: List[Tuple[str, str]] = []
|
self._mounts: List[Tuple[Path, Path]] = []
|
||||||
self.node_net_client: LinuxNetClient = self.create_node_net_client(
|
self.node_net_client: LinuxNetClient = self.create_node_net_client(
|
||||||
self.session.use_ovs()
|
self.session.use_ovs()
|
||||||
)
|
)
|
||||||
|
@ -524,33 +564,33 @@ class CoreNode(CoreNodeBase):
|
||||||
f"{VNODED} -v -c {self.ctrlchnlname} -l {self.ctrlchnlname}.log "
|
f"{VNODED} -v -c {self.ctrlchnlname} -l {self.ctrlchnlname}.log "
|
||||||
f"-p {self.ctrlchnlname}.pid"
|
f"-p {self.ctrlchnlname}.pid"
|
||||||
)
|
)
|
||||||
if self.nodedir:
|
if self.directory:
|
||||||
vnoded += f" -C {self.nodedir}"
|
vnoded += f" -C {self.directory}"
|
||||||
env = self.session.get_environment(state=False)
|
env = self.session.get_environment(state=False)
|
||||||
env["NODE_NUMBER"] = str(self.id)
|
env["NODE_NUMBER"] = str(self.id)
|
||||||
env["NODE_NAME"] = str(self.name)
|
env["NODE_NAME"] = str(self.name)
|
||||||
|
|
||||||
output = self.host_cmd(vnoded, env=env)
|
output = self.host_cmd(vnoded, env=env)
|
||||||
self.pid = int(output)
|
self.pid = int(output)
|
||||||
logging.debug("node(%s) pid: %s", self.name, self.pid)
|
logger.debug("node(%s) pid: %s", self.name, self.pid)
|
||||||
|
|
||||||
# create vnode client
|
# create vnode client
|
||||||
self.client = VnodeClient(self.name, self.ctrlchnlname)
|
self.client = VnodeClient(self.name, self.ctrlchnlname)
|
||||||
|
|
||||||
# bring up the loopback interface
|
# bring up the loopback interface
|
||||||
logging.debug("bringing up loopback interface")
|
logger.debug("bringing up loopback interface")
|
||||||
self.node_net_client.device_up("lo")
|
self.node_net_client.device_up("lo")
|
||||||
|
|
||||||
# set hostname for node
|
# set hostname for node
|
||||||
logging.debug("setting hostname: %s", self.name)
|
logger.debug("setting hostname: %s", self.name)
|
||||||
self.node_net_client.set_hostname(self.name)
|
self.node_net_client.set_hostname(self.name)
|
||||||
|
|
||||||
# mark node as up
|
# mark node as up
|
||||||
self.up = True
|
self.up = True
|
||||||
|
|
||||||
# create private directories
|
# create private directories
|
||||||
self.privatedir("/var/run")
|
for dir_path in PRIVATE_DIRS:
|
||||||
self.privatedir("/var/log")
|
self.create_dir(dir_path)
|
||||||
|
|
||||||
def shutdown(self) -> None:
|
def shutdown(self) -> None:
|
||||||
"""
|
"""
|
||||||
|
@ -561,35 +601,30 @@ class CoreNode(CoreNodeBase):
|
||||||
# nothing to do if node is not up
|
# nothing to do if node is not up
|
||||||
if not self.up:
|
if not self.up:
|
||||||
return
|
return
|
||||||
|
|
||||||
with self.lock:
|
with self.lock:
|
||||||
try:
|
try:
|
||||||
# unmount all targets (NOTE: non-persistent mount namespaces are
|
# unmount all targets (NOTE: non-persistent mount namespaces are
|
||||||
# removed by the kernel when last referencing process is killed)
|
# removed by the kernel when last referencing process is killed)
|
||||||
self._mounts = []
|
self._mounts = []
|
||||||
|
|
||||||
# shutdown all interfaces
|
# shutdown all interfaces
|
||||||
for iface in self.get_ifaces():
|
for iface in self.get_ifaces():
|
||||||
iface.shutdown()
|
iface.shutdown()
|
||||||
|
|
||||||
# kill node process if present
|
# kill node process if present
|
||||||
try:
|
try:
|
||||||
self.host_cmd(f"kill -9 {self.pid}")
|
self.host_cmd(f"kill -9 {self.pid}")
|
||||||
except CoreCommandError:
|
except CoreCommandError:
|
||||||
logging.exception("error killing process")
|
logger.exception("error killing process")
|
||||||
|
|
||||||
# remove node directory if present
|
# remove node directory if present
|
||||||
try:
|
try:
|
||||||
self.host_cmd(f"rm -rf {self.ctrlchnlname}")
|
self.host_cmd(f"rm -rf {self.ctrlchnlname}")
|
||||||
except CoreCommandError:
|
except CoreCommandError:
|
||||||
logging.exception("error removing node directory")
|
logger.exception("error removing node directory")
|
||||||
|
|
||||||
# clear interface data, close client, and mark self and not up
|
# clear interface data, close client, and mark self and not up
|
||||||
self.ifaces.clear()
|
self.ifaces.clear()
|
||||||
self.client.close()
|
self.client.close()
|
||||||
self.up = False
|
self.up = False
|
||||||
except OSError:
|
except OSError:
|
||||||
logging.exception("error during shutdown")
|
logger.exception("error during shutdown")
|
||||||
finally:
|
finally:
|
||||||
self.rmnodedir()
|
self.rmnodedir()
|
||||||
|
|
||||||
|
@ -636,35 +671,37 @@ class CoreNode(CoreNodeBase):
|
||||||
else:
|
else:
|
||||||
return f"ssh -X -f {self.server.host} xterm -e {terminal}"
|
return f"ssh -X -f {self.server.host} xterm -e {terminal}"
|
||||||
|
|
||||||
def privatedir(self, path: str) -> None:
|
def create_dir(self, dir_path: Path) -> None:
|
||||||
"""
|
"""
|
||||||
Create a private directory.
|
Create a node private directory.
|
||||||
|
|
||||||
:param path: path to create
|
:param dir_path: path to create
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
if path[0] != "/":
|
if not dir_path.is_absolute():
|
||||||
raise ValueError(f"path not fully qualified: {path}")
|
raise CoreError(f"private directory path not fully qualified: {dir_path}")
|
||||||
hostpath = os.path.join(
|
logger.debug("node(%s) creating private directory: %s", self.name, dir_path)
|
||||||
self.nodedir, os.path.normpath(path).strip("/").replace("/", ".")
|
parent_path = self._find_parent_path(dir_path)
|
||||||
)
|
if parent_path:
|
||||||
self.host_cmd(f"mkdir -p {hostpath}")
|
self.host_cmd(f"mkdir -p {parent_path}")
|
||||||
self.mount(hostpath, path)
|
else:
|
||||||
|
host_path = self.host_path(dir_path, is_dir=True)
|
||||||
|
self.host_cmd(f"mkdir -p {host_path}")
|
||||||
|
self.mount(host_path, dir_path)
|
||||||
|
|
||||||
def mount(self, source: str, target: str) -> None:
|
def mount(self, src_path: Path, target_path: Path) -> None:
|
||||||
"""
|
"""
|
||||||
Create and mount a directory.
|
Create and mount a directory.
|
||||||
|
|
||||||
:param source: source directory to mount
|
:param src_path: source directory to mount
|
||||||
:param target: target directory to create
|
:param target_path: target directory to create
|
||||||
:return: nothing
|
:return: nothing
|
||||||
:raises CoreCommandError: when a non-zero exit status occurs
|
:raises CoreCommandError: when a non-zero exit status occurs
|
||||||
"""
|
"""
|
||||||
source = os.path.abspath(source)
|
logger.debug("node(%s) mounting: %s at %s", self.name, src_path, target_path)
|
||||||
logging.debug("node(%s) mounting: %s at %s", self.name, source, target)
|
self.cmd(f"mkdir -p {target_path}")
|
||||||
self.cmd(f"mkdir -p {target}")
|
self.cmd(f"{MOUNT} -n --bind {src_path} {target_path}")
|
||||||
self.cmd(f"{MOUNT} -n --bind {source} {target}")
|
self._mounts.append((src_path, target_path))
|
||||||
self._mounts.append((source, target))
|
|
||||||
|
|
||||||
def next_iface_id(self) -> int:
|
def next_iface_id(self) -> int:
|
||||||
"""
|
"""
|
||||||
|
@ -719,9 +756,9 @@ class CoreNode(CoreNodeBase):
|
||||||
if self.up:
|
if self.up:
|
||||||
flow_id = self.node_net_client.get_ifindex(veth.name)
|
flow_id = self.node_net_client.get_ifindex(veth.name)
|
||||||
veth.flow_id = int(flow_id)
|
veth.flow_id = int(flow_id)
|
||||||
logging.debug("interface flow index: %s - %s", veth.name, veth.flow_id)
|
logger.debug("interface flow index: %s - %s", veth.name, veth.flow_id)
|
||||||
mac = self.node_net_client.get_mac(veth.name)
|
mac = self.node_net_client.get_mac(veth.name)
|
||||||
logging.debug("interface mac: %s - %s", veth.name, mac)
|
logger.debug("interface mac: %s - %s", veth.name, mac)
|
||||||
veth.set_mac(mac)
|
veth.set_mac(mac)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
|
@ -851,86 +888,99 @@ class CoreNode(CoreNodeBase):
|
||||||
self.ifup(iface_id)
|
self.ifup(iface_id)
|
||||||
return self.get_iface(iface_id)
|
return self.get_iface(iface_id)
|
||||||
|
|
||||||
def addfile(self, srcname: str, filename: str) -> None:
|
def addfile(self, src_path: Path, file_path: Path) -> None:
|
||||||
"""
|
"""
|
||||||
Add a file.
|
Add a file.
|
||||||
|
|
||||||
:param srcname: source file name
|
:param src_path: source file path
|
||||||
:param filename: file name to add
|
:param file_path: file name to add
|
||||||
:return: nothing
|
:return: nothing
|
||||||
:raises CoreCommandError: when a non-zero exit status occurs
|
:raises CoreCommandError: when a non-zero exit status occurs
|
||||||
"""
|
"""
|
||||||
logging.info("adding file from %s to %s", srcname, filename)
|
logger.info("adding file from %s to %s", src_path, file_path)
|
||||||
directory = os.path.dirname(filename)
|
directory = file_path.parent
|
||||||
if self.server is None:
|
if self.server is None:
|
||||||
self.client.check_cmd(f"mkdir -p {directory}")
|
self.client.check_cmd(f"mkdir -p {directory}")
|
||||||
self.client.check_cmd(f"mv {srcname} {filename}")
|
self.client.check_cmd(f"mv {src_path} {file_path}")
|
||||||
self.client.check_cmd("sync")
|
self.client.check_cmd("sync")
|
||||||
else:
|
else:
|
||||||
self.host_cmd(f"mkdir -p {directory}")
|
self.host_cmd(f"mkdir -p {directory}")
|
||||||
self.server.remote_put(srcname, filename)
|
self.server.remote_put(src_path, file_path)
|
||||||
|
|
||||||
def hostfilename(self, filename: str) -> str:
|
def _find_parent_path(self, path: Path) -> Optional[Path]:
|
||||||
"""
|
"""
|
||||||
Return the name of a node"s file on the host filesystem.
|
Check if there is an existing mounted parent directory created for this node.
|
||||||
|
|
||||||
:param filename: host file name
|
:param path: existing parent path to use
|
||||||
:return: path to file
|
:return: exist parent path if exists, None otherwise
|
||||||
"""
|
"""
|
||||||
dirname, basename = os.path.split(filename)
|
logger.debug("looking for existing parent: %s", path)
|
||||||
if not basename:
|
existing_path = None
|
||||||
raise ValueError(f"no basename for filename: {filename}")
|
for parent in path.parents:
|
||||||
if dirname and dirname[0] == "/":
|
node_path = self.host_path(parent, is_dir=True)
|
||||||
dirname = dirname[1:]
|
if node_path == self.directory:
|
||||||
dirname = dirname.replace("/", ".")
|
break
|
||||||
dirname = os.path.join(self.nodedir, dirname)
|
if self.path_exists(str(node_path)):
|
||||||
return os.path.join(dirname, basename)
|
relative_path = path.relative_to(parent)
|
||||||
|
existing_path = node_path / relative_path
|
||||||
|
break
|
||||||
|
return existing_path
|
||||||
|
|
||||||
def nodefile(self, filename: str, contents: str, mode: int = 0o644) -> None:
|
def create_file(self, file_path: Path, contents: str, mode: int = 0o644) -> None:
|
||||||
"""
|
"""
|
||||||
Create a node file with a given mode.
|
Create file within a node at the given path, using contents and mode.
|
||||||
|
|
||||||
:param filename: name of file to create
|
:param file_path: desired path for file
|
||||||
:param contents: contents of file
|
:param contents: contents of file
|
||||||
:param mode: mode for file
|
:param mode: mode to create file with
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
hostfilename = self.hostfilename(filename)
|
logger.debug("node(%s) create file(%s) mode(%o)", self.name, file_path, mode)
|
||||||
dirname, _basename = os.path.split(hostfilename)
|
host_path = self._find_parent_path(file_path)
|
||||||
if self.server is None:
|
if host_path:
|
||||||
if not os.path.isdir(dirname):
|
self.host_cmd(f"mkdir -p {host_path.parent}")
|
||||||
os.makedirs(dirname, mode=0o755)
|
|
||||||
with open(hostfilename, "w") as open_file:
|
|
||||||
open_file.write(contents)
|
|
||||||
os.chmod(open_file.name, mode)
|
|
||||||
else:
|
else:
|
||||||
self.host_cmd(f"mkdir -m {0o755:o} -p {dirname}")
|
host_path = self.host_path(file_path)
|
||||||
self.server.remote_put_temp(hostfilename, contents)
|
directory = host_path.parent
|
||||||
self.host_cmd(f"chmod {mode:o} {hostfilename}")
|
if self.server is None:
|
||||||
logging.debug(
|
if not directory.exists():
|
||||||
"node(%s) added file: %s; mode: 0%o", self.name, hostfilename, mode
|
directory.mkdir(parents=True, mode=0o755)
|
||||||
|
with host_path.open("w") as f:
|
||||||
|
f.write(contents)
|
||||||
|
host_path.chmod(mode)
|
||||||
|
else:
|
||||||
|
self.host_cmd(f"mkdir -m {0o755:o} -p {directory}")
|
||||||
|
self.server.remote_put_temp(host_path, contents)
|
||||||
|
self.host_cmd(f"chmod {mode:o} {host_path}")
|
||||||
|
|
||||||
|
def copy_file(self, src_path: Path, dst_path: Path, mode: int = None) -> None:
|
||||||
|
"""
|
||||||
|
Copy source file to node host destination, updating the file mode when
|
||||||
|
provided.
|
||||||
|
|
||||||
|
:param src_path: source file to copy
|
||||||
|
:param dst_path: node host destination
|
||||||
|
:param mode: file mode
|
||||||
|
:return: nothing
|
||||||
|
"""
|
||||||
|
logger.debug(
|
||||||
|
"node(%s) copying file src(%s) to dst(%s) mode(%o)",
|
||||||
|
self.name,
|
||||||
|
src_path,
|
||||||
|
dst_path,
|
||||||
|
mode or 0,
|
||||||
)
|
)
|
||||||
|
host_path = self._find_parent_path(dst_path)
|
||||||
def nodefilecopy(self, filename: str, srcfilename: str, mode: int = None) -> None:
|
if host_path:
|
||||||
"""
|
self.host_cmd(f"mkdir -p {host_path.parent}")
|
||||||
Copy a file to a node, following symlinks and preserving metadata.
|
|
||||||
Change file mode if specified.
|
|
||||||
|
|
||||||
:param filename: file name to copy file to
|
|
||||||
:param srcfilename: file to copy
|
|
||||||
:param mode: mode to copy to
|
|
||||||
:return: nothing
|
|
||||||
"""
|
|
||||||
hostfilename = self.hostfilename(filename)
|
|
||||||
if self.server is None:
|
|
||||||
shutil.copy2(srcfilename, hostfilename)
|
|
||||||
else:
|
else:
|
||||||
self.server.remote_put(srcfilename, hostfilename)
|
host_path = self.host_path(dst_path)
|
||||||
|
if self.server is None:
|
||||||
|
shutil.copy2(src_path, host_path)
|
||||||
|
else:
|
||||||
|
self.server.remote_put(src_path, host_path)
|
||||||
if mode is not None:
|
if mode is not None:
|
||||||
self.host_cmd(f"chmod {mode:o} {hostfilename}")
|
self.host_cmd(f"chmod {mode:o} {host_path}")
|
||||||
logging.info(
|
|
||||||
"node(%s) copied file: %s; mode: %s", self.name, hostfilename, mode
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
class CoreNetworkBase(NodeBase):
|
class CoreNetworkBase(NodeBase):
|
||||||
|
@ -958,9 +1008,9 @@ class CoreNetworkBase(NodeBase):
|
||||||
will run on, default is None for localhost
|
will run on, default is None for localhost
|
||||||
"""
|
"""
|
||||||
super().__init__(session, _id, name, server)
|
super().__init__(session, _id, name, server)
|
||||||
self.brname = None
|
self.brname: Optional[str] = None
|
||||||
self._linked = {}
|
self.linked: Dict[CoreInterface, Dict[CoreInterface, bool]] = {}
|
||||||
self._linked_lock = threading.Lock()
|
self.linked_lock: threading.Lock = threading.Lock()
|
||||||
|
|
||||||
@abc.abstractmethod
|
@abc.abstractmethod
|
||||||
def startup(self) -> None:
|
def startup(self) -> None:
|
||||||
|
@ -1029,8 +1079,8 @@ class CoreNetworkBase(NodeBase):
|
||||||
i = self.next_iface_id()
|
i = self.next_iface_id()
|
||||||
self.ifaces[i] = iface
|
self.ifaces[i] = iface
|
||||||
iface.net_id = i
|
iface.net_id = i
|
||||||
with self._linked_lock:
|
with self.linked_lock:
|
||||||
self._linked[iface] = {}
|
self.linked[iface] = {}
|
||||||
|
|
||||||
def detach(self, iface: CoreInterface) -> None:
|
def detach(self, iface: CoreInterface) -> None:
|
||||||
"""
|
"""
|
||||||
|
@ -1041,8 +1091,8 @@ class CoreNetworkBase(NodeBase):
|
||||||
"""
|
"""
|
||||||
del self.ifaces[iface.net_id]
|
del self.ifaces[iface.net_id]
|
||||||
iface.net_id = None
|
iface.net_id = None
|
||||||
with self._linked_lock:
|
with self.linked_lock:
|
||||||
del self._linked[iface]
|
del self.linked[iface]
|
||||||
|
|
||||||
def links(self, flags: MessageFlags = MessageFlags.NONE) -> List[LinkData]:
|
def links(self, flags: MessageFlags = MessageFlags.NONE) -> List[LinkData]:
|
||||||
"""
|
"""
|
||||||
|
|
|
@ -3,6 +3,7 @@ client.py: implementation of the VnodeClient class for issuing commands
|
||||||
over a control channel to the vnoded process running in a network namespace.
|
over a control channel to the vnoded process running in a network namespace.
|
||||||
The control channel can be accessed via calls using the vcmd shell.
|
The control channel can be accessed via calls using the vcmd shell.
|
||||||
"""
|
"""
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
from core import utils
|
from core import utils
|
||||||
from core.executables import BASH, VCMD
|
from core.executables import BASH, VCMD
|
||||||
|
@ -13,7 +14,7 @@ class VnodeClient:
|
||||||
Provides client functionality for interacting with a virtual node.
|
Provides client functionality for interacting with a virtual node.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
def __init__(self, name: str, ctrlchnlname: str) -> None:
|
def __init__(self, name: str, ctrlchnlname: Path) -> None:
|
||||||
"""
|
"""
|
||||||
Create a VnodeClient instance.
|
Create a VnodeClient instance.
|
||||||
|
|
||||||
|
@ -21,7 +22,7 @@ class VnodeClient:
|
||||||
:param ctrlchnlname: control channel name
|
:param ctrlchnlname: control channel name
|
||||||
"""
|
"""
|
||||||
self.name: str = name
|
self.name: str = name
|
||||||
self.ctrlchnlname: str = ctrlchnlname
|
self.ctrlchnlname: Path = ctrlchnlname
|
||||||
|
|
||||||
def _verify_connection(self) -> None:
|
def _verify_connection(self) -> None:
|
||||||
"""
|
"""
|
||||||
|
|
|
@ -1,6 +1,6 @@
|
||||||
import json
|
import json
|
||||||
import logging
|
import logging
|
||||||
import os
|
from pathlib import Path
|
||||||
from tempfile import NamedTemporaryFile
|
from tempfile import NamedTemporaryFile
|
||||||
from typing import TYPE_CHECKING, Callable, Dict, Optional
|
from typing import TYPE_CHECKING, Callable, Dict, Optional
|
||||||
|
|
||||||
|
@ -11,6 +11,8 @@ from core.errors import CoreCommandError
|
||||||
from core.nodes.base import CoreNode
|
from core.nodes.base import CoreNode
|
||||||
from core.nodes.netclient import LinuxNetClient, get_net_client
|
from core.nodes.netclient import LinuxNetClient, get_net_client
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.emulator.session import Session
|
from core.emulator.session import Session
|
||||||
|
|
||||||
|
@ -50,7 +52,7 @@ class DockerClient:
|
||||||
self.run(f"docker rm -f {self.name}")
|
self.run(f"docker rm -f {self.name}")
|
||||||
|
|
||||||
def check_cmd(self, cmd: str, wait: bool = True, shell: bool = False) -> str:
|
def check_cmd(self, cmd: str, wait: bool = True, shell: bool = False) -> str:
|
||||||
logging.info("docker cmd output: %s", cmd)
|
logger.info("docker cmd output: %s", cmd)
|
||||||
return utils.cmd(f"docker exec {self.name} {cmd}", wait=wait, shell=shell)
|
return utils.cmd(f"docker exec {self.name} {cmd}", wait=wait, shell=shell)
|
||||||
|
|
||||||
def create_ns_cmd(self, cmd: str) -> str:
|
def create_ns_cmd(self, cmd: str) -> str:
|
||||||
|
@ -60,11 +62,11 @@ class DockerClient:
|
||||||
args = f"docker inspect -f '{{{{.State.Pid}}}}' {self.name}"
|
args = f"docker inspect -f '{{{{.State.Pid}}}}' {self.name}"
|
||||||
output = self.run(args)
|
output = self.run(args)
|
||||||
self.pid = output
|
self.pid = output
|
||||||
logging.debug("node(%s) pid: %s", self.name, self.pid)
|
logger.debug("node(%s) pid: %s", self.name, self.pid)
|
||||||
return output
|
return output
|
||||||
|
|
||||||
def copy_file(self, source: str, destination: str) -> str:
|
def copy_file(self, src_path: Path, dst_path: Path) -> str:
|
||||||
args = f"docker cp {source} {self.name}:{destination}"
|
args = f"docker cp {src_path} {self.name}:{dst_path}"
|
||||||
return self.run(args)
|
return self.run(args)
|
||||||
|
|
||||||
|
|
||||||
|
@ -76,7 +78,7 @@ class DockerNode(CoreNode):
|
||||||
session: "Session",
|
session: "Session",
|
||||||
_id: int = None,
|
_id: int = None,
|
||||||
name: str = None,
|
name: str = None,
|
||||||
nodedir: str = None,
|
directory: str = None,
|
||||||
server: DistributedServer = None,
|
server: DistributedServer = None,
|
||||||
image: str = None,
|
image: str = None,
|
||||||
) -> None:
|
) -> None:
|
||||||
|
@ -86,7 +88,7 @@ class DockerNode(CoreNode):
|
||||||
:param session: core session instance
|
:param session: core session instance
|
||||||
:param _id: object id
|
:param _id: object id
|
||||||
:param name: object name
|
:param name: object name
|
||||||
:param nodedir: node directory
|
:param directory: node directory
|
||||||
:param server: remote server node
|
:param server: remote server node
|
||||||
will run on, default is None for localhost
|
will run on, default is None for localhost
|
||||||
:param image: image to start container with
|
:param image: image to start container with
|
||||||
|
@ -94,7 +96,7 @@ class DockerNode(CoreNode):
|
||||||
if image is None:
|
if image is None:
|
||||||
image = "ubuntu"
|
image = "ubuntu"
|
||||||
self.image: str = image
|
self.image: str = image
|
||||||
super().__init__(session, _id, name, nodedir, server)
|
super().__init__(session, _id, name, directory, server)
|
||||||
|
|
||||||
def create_node_net_client(self, use_ovs: bool) -> LinuxNetClient:
|
def create_node_net_client(self, use_ovs: bool) -> LinuxNetClient:
|
||||||
"""
|
"""
|
||||||
|
@ -162,77 +164,73 @@ class DockerNode(CoreNode):
|
||||||
"""
|
"""
|
||||||
return f"docker exec -it {self.name} bash"
|
return f"docker exec -it {self.name} bash"
|
||||||
|
|
||||||
def privatedir(self, path: str) -> None:
|
def create_dir(self, dir_path: Path) -> None:
|
||||||
"""
|
"""
|
||||||
Create a private directory.
|
Create a private directory.
|
||||||
|
|
||||||
:param path: path to create
|
:param dir_path: path to create
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
logging.debug("creating node dir: %s", path)
|
logger.debug("creating node dir: %s", dir_path)
|
||||||
args = f"mkdir -p {path}"
|
args = f"mkdir -p {dir_path}"
|
||||||
self.cmd(args)
|
self.cmd(args)
|
||||||
|
|
||||||
def mount(self, source: str, target: str) -> None:
|
def mount(self, src_path: str, target_path: str) -> None:
|
||||||
"""
|
"""
|
||||||
Create and mount a directory.
|
Create and mount a directory.
|
||||||
|
|
||||||
:param source: source directory to mount
|
:param src_path: source directory to mount
|
||||||
:param target: target directory to create
|
:param target_path: target directory to create
|
||||||
:return: nothing
|
:return: nothing
|
||||||
:raises CoreCommandError: when a non-zero exit status occurs
|
:raises CoreCommandError: when a non-zero exit status occurs
|
||||||
"""
|
"""
|
||||||
logging.debug("mounting source(%s) target(%s)", source, target)
|
logger.debug("mounting source(%s) target(%s)", src_path, target_path)
|
||||||
raise Exception("not supported")
|
raise Exception("not supported")
|
||||||
|
|
||||||
def nodefile(self, filename: str, contents: str, mode: int = 0o644) -> None:
|
def create_file(self, file_path: Path, contents: str, mode: int = 0o644) -> None:
|
||||||
"""
|
"""
|
||||||
Create a node file with a given mode.
|
Create a node file with a given mode.
|
||||||
|
|
||||||
:param filename: name of file to create
|
:param file_path: name of file to create
|
||||||
:param contents: contents of file
|
:param contents: contents of file
|
||||||
:param mode: mode for file
|
:param mode: mode for file
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
logging.debug("nodefile filename(%s) mode(%s)", filename, mode)
|
logger.debug("node(%s) create file(%s) mode(%o)", self.name, file_path, mode)
|
||||||
directory = os.path.dirname(filename)
|
|
||||||
temp = NamedTemporaryFile(delete=False)
|
temp = NamedTemporaryFile(delete=False)
|
||||||
temp.write(contents.encode("utf-8"))
|
temp.write(contents.encode("utf-8"))
|
||||||
temp.close()
|
temp.close()
|
||||||
|
temp_path = Path(temp.name)
|
||||||
if directory:
|
directory = file_path.name
|
||||||
|
if str(directory) != ".":
|
||||||
self.cmd(f"mkdir -m {0o755:o} -p {directory}")
|
self.cmd(f"mkdir -m {0o755:o} -p {directory}")
|
||||||
if self.server is not None:
|
if self.server is not None:
|
||||||
self.server.remote_put(temp.name, temp.name)
|
self.server.remote_put(temp_path, temp_path)
|
||||||
self.client.copy_file(temp.name, filename)
|
self.client.copy_file(temp_path, file_path)
|
||||||
self.cmd(f"chmod {mode:o} {filename}")
|
self.cmd(f"chmod {mode:o} {file_path}")
|
||||||
if self.server is not None:
|
if self.server is not None:
|
||||||
self.host_cmd(f"rm -f {temp.name}")
|
self.host_cmd(f"rm -f {temp_path}")
|
||||||
os.unlink(temp.name)
|
temp_path.unlink()
|
||||||
logging.debug("node(%s) added file: %s; mode: 0%o", self.name, filename, mode)
|
|
||||||
|
|
||||||
def nodefilecopy(self, filename: str, srcfilename: str, mode: int = None) -> None:
|
def copy_file(self, src_path: Path, dst_path: Path, mode: int = None) -> None:
|
||||||
"""
|
"""
|
||||||
Copy a file to a node, following symlinks and preserving metadata.
|
Copy a file to a node, following symlinks and preserving metadata.
|
||||||
Change file mode if specified.
|
Change file mode if specified.
|
||||||
|
|
||||||
:param filename: file name to copy file to
|
:param dst_path: file name to copy file to
|
||||||
:param srcfilename: file to copy
|
:param src_path: file to copy
|
||||||
:param mode: mode to copy to
|
:param mode: mode to copy to
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
logging.info(
|
logger.info(
|
||||||
"node file copy file(%s) source(%s) mode(%s)", filename, srcfilename, mode
|
"node file copy file(%s) source(%s) mode(%o)", dst_path, src_path, mode or 0
|
||||||
)
|
)
|
||||||
directory = os.path.dirname(filename)
|
self.cmd(f"mkdir -p {dst_path.parent}")
|
||||||
self.cmd(f"mkdir -p {directory}")
|
if self.server:
|
||||||
|
|
||||||
if self.server is None:
|
|
||||||
source = srcfilename
|
|
||||||
else:
|
|
||||||
temp = NamedTemporaryFile(delete=False)
|
temp = NamedTemporaryFile(delete=False)
|
||||||
source = temp.name
|
temp_path = Path(temp.name)
|
||||||
self.server.remote_put(source, temp.name)
|
src_path = temp_path
|
||||||
|
self.server.remote_put(src_path, temp_path)
|
||||||
self.client.copy_file(source, filename)
|
self.client.copy_file(src_path, dst_path)
|
||||||
self.cmd(f"chmod {mode:o} {filename}")
|
if mode is not None:
|
||||||
|
self.cmd(f"chmod {mode:o} {dst_path}")
|
||||||
|
|
|
@ -4,6 +4,7 @@ virtual ethernet classes that implement the interfaces available under Linux.
|
||||||
|
|
||||||
import logging
|
import logging
|
||||||
import time
|
import time
|
||||||
|
from pathlib import Path
|
||||||
from typing import TYPE_CHECKING, Callable, Dict, List, Optional, Tuple
|
from typing import TYPE_CHECKING, Callable, Dict, List, Optional, Tuple
|
||||||
|
|
||||||
import netaddr
|
import netaddr
|
||||||
|
@ -14,6 +15,8 @@ from core.emulator.enumerations import TransportType
|
||||||
from core.errors import CoreCommandError, CoreError
|
from core.errors import CoreCommandError, CoreError
|
||||||
from core.nodes.netclient import LinuxNetClient, get_net_client
|
from core.nodes.netclient import LinuxNetClient, get_net_client
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.emulator.distributed import DistributedServer
|
from core.emulator.distributed import DistributedServer
|
||||||
from core.emulator.session import Session
|
from core.emulator.session import Session
|
||||||
|
@ -79,7 +82,7 @@ class CoreInterface:
|
||||||
self,
|
self,
|
||||||
args: str,
|
args: str,
|
||||||
env: Dict[str, str] = None,
|
env: Dict[str, str] = None,
|
||||||
cwd: str = None,
|
cwd: Path = None,
|
||||||
wait: bool = True,
|
wait: bool = True,
|
||||||
shell: bool = False,
|
shell: bool = False,
|
||||||
) -> str:
|
) -> str:
|
||||||
|
@ -273,7 +276,7 @@ class CoreInterface:
|
||||||
:return: True if parameter changed, False otherwise
|
:return: True if parameter changed, False otherwise
|
||||||
"""
|
"""
|
||||||
# treat None and 0 as unchanged values
|
# treat None and 0 as unchanged values
|
||||||
logging.debug("setting param: %s - %s", key, value)
|
logger.debug("setting param: %s - %s", key, value)
|
||||||
if value is None or value < 0:
|
if value is None or value < 0:
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
@ -456,7 +459,7 @@ class TunTap(CoreInterface):
|
||||||
try:
|
try:
|
||||||
self.node.node_net_client.device_flush(self.name)
|
self.node.node_net_client.device_flush(self.name)
|
||||||
except CoreCommandError:
|
except CoreCommandError:
|
||||||
logging.exception("error shutting down tunnel tap")
|
logger.exception("error shutting down tunnel tap")
|
||||||
|
|
||||||
self.up = False
|
self.up = False
|
||||||
|
|
||||||
|
@ -481,14 +484,14 @@ class TunTap(CoreInterface):
|
||||||
msg = f"attempt {i} failed with nonzero exit status {r}"
|
msg = f"attempt {i} failed with nonzero exit status {r}"
|
||||||
if i < attempts + 1:
|
if i < attempts + 1:
|
||||||
msg += ", retrying..."
|
msg += ", retrying..."
|
||||||
logging.info(msg)
|
logger.info(msg)
|
||||||
time.sleep(delay)
|
time.sleep(delay)
|
||||||
delay += delay
|
delay += delay
|
||||||
if delay > maxretrydelay:
|
if delay > maxretrydelay:
|
||||||
delay = maxretrydelay
|
delay = maxretrydelay
|
||||||
else:
|
else:
|
||||||
msg += ", giving up"
|
msg += ", giving up"
|
||||||
logging.info(msg)
|
logger.info(msg)
|
||||||
|
|
||||||
return result
|
return result
|
||||||
|
|
||||||
|
@ -499,7 +502,7 @@ class TunTap(CoreInterface):
|
||||||
|
|
||||||
:return: wait for device local response
|
:return: wait for device local response
|
||||||
"""
|
"""
|
||||||
logging.debug("waiting for device local: %s", self.localname)
|
logger.debug("waiting for device local: %s", self.localname)
|
||||||
|
|
||||||
def localdevexists():
|
def localdevexists():
|
||||||
try:
|
try:
|
||||||
|
@ -516,7 +519,7 @@ class TunTap(CoreInterface):
|
||||||
|
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
logging.debug("waiting for device node: %s", self.name)
|
logger.debug("waiting for device node: %s", self.name)
|
||||||
|
|
||||||
def nodedevexists():
|
def nodedevexists():
|
||||||
try:
|
try:
|
||||||
|
@ -633,5 +636,5 @@ class GreTap(CoreInterface):
|
||||||
self.net_client.device_down(self.localname)
|
self.net_client.device_down(self.localname)
|
||||||
self.net_client.delete_device(self.localname)
|
self.net_client.delete_device(self.localname)
|
||||||
except CoreCommandError:
|
except CoreCommandError:
|
||||||
logging.exception("error during shutdown")
|
logger.exception("error during shutdown")
|
||||||
self.localname = None
|
self.localname = None
|
||||||
|
|
|
@ -1,7 +1,7 @@
|
||||||
import json
|
import json
|
||||||
import logging
|
import logging
|
||||||
import os
|
|
||||||
import time
|
import time
|
||||||
|
from pathlib import Path
|
||||||
from tempfile import NamedTemporaryFile
|
from tempfile import NamedTemporaryFile
|
||||||
from typing import TYPE_CHECKING, Callable, Dict, Optional
|
from typing import TYPE_CHECKING, Callable, Dict, Optional
|
||||||
|
|
||||||
|
@ -12,6 +12,8 @@ from core.errors import CoreCommandError
|
||||||
from core.nodes.base import CoreNode
|
from core.nodes.base import CoreNode
|
||||||
from core.nodes.interface import CoreInterface
|
from core.nodes.interface import CoreInterface
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.emulator.session import Session
|
from core.emulator.session import Session
|
||||||
|
|
||||||
|
@ -57,11 +59,10 @@ class LxdClient:
|
||||||
args = self.create_cmd(cmd)
|
args = self.create_cmd(cmd)
|
||||||
return utils.cmd(args, wait=wait, shell=shell)
|
return utils.cmd(args, wait=wait, shell=shell)
|
||||||
|
|
||||||
def copy_file(self, source: str, destination: str) -> None:
|
def copy_file(self, src_path: Path, dst_path: Path) -> None:
|
||||||
if destination[0] != "/":
|
if not str(dst_path).startswith("/"):
|
||||||
destination = os.path.join("/root/", destination)
|
dst_path = Path("/root/") / dst_path
|
||||||
|
args = f"lxc file push {src_path} {self.name}/{dst_path}"
|
||||||
args = f"lxc file push {source} {self.name}/{destination}"
|
|
||||||
self.run(args)
|
self.run(args)
|
||||||
|
|
||||||
|
|
||||||
|
@ -73,7 +74,7 @@ class LxcNode(CoreNode):
|
||||||
session: "Session",
|
session: "Session",
|
||||||
_id: int = None,
|
_id: int = None,
|
||||||
name: str = None,
|
name: str = None,
|
||||||
nodedir: str = None,
|
directory: str = None,
|
||||||
server: DistributedServer = None,
|
server: DistributedServer = None,
|
||||||
image: str = None,
|
image: str = None,
|
||||||
) -> None:
|
) -> None:
|
||||||
|
@ -83,7 +84,7 @@ class LxcNode(CoreNode):
|
||||||
:param session: core session instance
|
:param session: core session instance
|
||||||
:param _id: object id
|
:param _id: object id
|
||||||
:param name: object name
|
:param name: object name
|
||||||
:param nodedir: node directory
|
:param directory: node directory
|
||||||
:param server: remote server node
|
:param server: remote server node
|
||||||
will run on, default is None for localhost
|
will run on, default is None for localhost
|
||||||
:param image: image to start container with
|
:param image: image to start container with
|
||||||
|
@ -91,7 +92,7 @@ class LxcNode(CoreNode):
|
||||||
if image is None:
|
if image is None:
|
||||||
image = "ubuntu"
|
image = "ubuntu"
|
||||||
self.image: str = image
|
self.image: str = image
|
||||||
super().__init__(session, _id, name, nodedir, server)
|
super().__init__(session, _id, name, directory, server)
|
||||||
|
|
||||||
def alive(self) -> bool:
|
def alive(self) -> bool:
|
||||||
"""
|
"""
|
||||||
|
@ -139,81 +140,77 @@ class LxcNode(CoreNode):
|
||||||
"""
|
"""
|
||||||
return f"lxc exec {self.name} -- {sh}"
|
return f"lxc exec {self.name} -- {sh}"
|
||||||
|
|
||||||
def privatedir(self, path: str) -> None:
|
def create_dir(self, dir_path: Path) -> None:
|
||||||
"""
|
"""
|
||||||
Create a private directory.
|
Create a private directory.
|
||||||
|
|
||||||
:param path: path to create
|
:param dir_path: path to create
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
logging.info("creating node dir: %s", path)
|
logger.info("creating node dir: %s", dir_path)
|
||||||
args = f"mkdir -p {path}"
|
args = f"mkdir -p {dir_path}"
|
||||||
self.cmd(args)
|
self.cmd(args)
|
||||||
|
|
||||||
def mount(self, source: str, target: str) -> None:
|
def mount(self, src_path: Path, target_path: Path) -> None:
|
||||||
"""
|
"""
|
||||||
Create and mount a directory.
|
Create and mount a directory.
|
||||||
|
|
||||||
:param source: source directory to mount
|
:param src_path: source directory to mount
|
||||||
:param target: target directory to create
|
:param target_path: target directory to create
|
||||||
:return: nothing
|
:return: nothing
|
||||||
:raises CoreCommandError: when a non-zero exit status occurs
|
:raises CoreCommandError: when a non-zero exit status occurs
|
||||||
"""
|
"""
|
||||||
logging.debug("mounting source(%s) target(%s)", source, target)
|
logger.debug("mounting source(%s) target(%s)", src_path, target_path)
|
||||||
raise Exception("not supported")
|
raise Exception("not supported")
|
||||||
|
|
||||||
def nodefile(self, filename: str, contents: str, mode: int = 0o644) -> None:
|
def create_file(self, file_path: Path, contents: str, mode: int = 0o644) -> None:
|
||||||
"""
|
"""
|
||||||
Create a node file with a given mode.
|
Create a node file with a given mode.
|
||||||
|
|
||||||
:param filename: name of file to create
|
:param file_path: name of file to create
|
||||||
:param contents: contents of file
|
:param contents: contents of file
|
||||||
:param mode: mode for file
|
:param mode: mode for file
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
logging.debug("nodefile filename(%s) mode(%s)", filename, mode)
|
logger.debug("node(%s) create file(%s) mode(%o)", self.name, file_path, mode)
|
||||||
|
|
||||||
directory = os.path.dirname(filename)
|
|
||||||
temp = NamedTemporaryFile(delete=False)
|
temp = NamedTemporaryFile(delete=False)
|
||||||
temp.write(contents.encode("utf-8"))
|
temp.write(contents.encode("utf-8"))
|
||||||
temp.close()
|
temp.close()
|
||||||
|
temp_path = Path(temp.name)
|
||||||
if directory:
|
directory = file_path.parent
|
||||||
|
if str(directory) != ".":
|
||||||
self.cmd(f"mkdir -m {0o755:o} -p {directory}")
|
self.cmd(f"mkdir -m {0o755:o} -p {directory}")
|
||||||
if self.server is not None:
|
if self.server is not None:
|
||||||
self.server.remote_put(temp.name, temp.name)
|
self.server.remote_put(temp_path, temp_path)
|
||||||
self.client.copy_file(temp.name, filename)
|
self.client.copy_file(temp_path, file_path)
|
||||||
self.cmd(f"chmod {mode:o} {filename}")
|
self.cmd(f"chmod {mode:o} {file_path}")
|
||||||
if self.server is not None:
|
if self.server is not None:
|
||||||
self.host_cmd(f"rm -f {temp.name}")
|
self.host_cmd(f"rm -f {temp_path}")
|
||||||
os.unlink(temp.name)
|
temp_path.unlink()
|
||||||
logging.debug("node(%s) added file: %s; mode: 0%o", self.name, filename, mode)
|
logger.debug("node(%s) added file: %s; mode: 0%o", self.name, file_path, mode)
|
||||||
|
|
||||||
def nodefilecopy(self, filename: str, srcfilename: str, mode: int = None) -> None:
|
def copy_file(self, src_path: Path, dst_path: Path, mode: int = None) -> None:
|
||||||
"""
|
"""
|
||||||
Copy a file to a node, following symlinks and preserving metadata.
|
Copy a file to a node, following symlinks and preserving metadata.
|
||||||
Change file mode if specified.
|
Change file mode if specified.
|
||||||
|
|
||||||
:param filename: file name to copy file to
|
:param dst_path: file name to copy file to
|
||||||
:param srcfilename: file to copy
|
:param src_path: file to copy
|
||||||
:param mode: mode to copy to
|
:param mode: mode to copy to
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
logging.info(
|
logger.info(
|
||||||
"node file copy file(%s) source(%s) mode(%s)", filename, srcfilename, mode
|
"node file copy file(%s) source(%s) mode(%o)", dst_path, src_path, mode or 0
|
||||||
)
|
)
|
||||||
directory = os.path.dirname(filename)
|
self.cmd(f"mkdir -p {dst_path.parent}")
|
||||||
self.cmd(f"mkdir -p {directory}")
|
if self.server:
|
||||||
|
|
||||||
if self.server is None:
|
|
||||||
source = srcfilename
|
|
||||||
else:
|
|
||||||
temp = NamedTemporaryFile(delete=False)
|
temp = NamedTemporaryFile(delete=False)
|
||||||
source = temp.name
|
temp_path = Path(temp.name)
|
||||||
self.server.remote_put(source, temp.name)
|
src_path = temp_path
|
||||||
|
self.server.remote_put(src_path, temp_path)
|
||||||
self.client.copy_file(source, filename)
|
self.client.copy_file(src_path, dst_path)
|
||||||
self.cmd(f"chmod {mode:o} {filename}")
|
if mode is not None:
|
||||||
|
self.cmd(f"chmod {mode:o} {dst_path}")
|
||||||
|
|
||||||
def add_iface(self, iface: CoreInterface, iface_id: int) -> None:
|
def add_iface(self, iface: CoreInterface, iface_id: int) -> None:
|
||||||
super().add_iface(iface, iface_id)
|
super().add_iface(iface, iface_id)
|
||||||
|
|
|
@ -38,7 +38,7 @@ class LinuxNetClient:
|
||||||
:param device: device to add route to
|
:param device: device to add route to
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
self.run(f"{IP} route add {route} dev {device}")
|
self.run(f"{IP} route replace {route} dev {device}")
|
||||||
|
|
||||||
def device_up(self, device: str) -> None:
|
def device_up(self, device: str) -> None:
|
||||||
"""
|
"""
|
||||||
|
@ -361,14 +361,15 @@ class OvsNetClient(LinuxNetClient):
|
||||||
return True
|
return True
|
||||||
return False
|
return False
|
||||||
|
|
||||||
def disable_mac_learning(self, name: str) -> None:
|
def set_mac_learning(self, name: str, value: int) -> None:
|
||||||
"""
|
"""
|
||||||
Disable mac learning for a OVS bridge.
|
Set mac learning for an OVS bridge.
|
||||||
|
|
||||||
:param name: bridge name
|
:param name: bridge name
|
||||||
|
:param value: ageing time value
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
self.run(f"{OVS_VSCTL} set bridge {name} other_config:mac-aging-time=0")
|
self.run(f"{OVS_VSCTL} set bridge {name} other_config:mac-aging-time={value}")
|
||||||
|
|
||||||
|
|
||||||
def get_net_client(use_ovs: bool, run: Callable[..., str]) -> LinuxNetClient:
|
def get_net_client(use_ovs: bool, run: Callable[..., str]) -> LinuxNetClient:
|
||||||
|
|
|
@ -6,7 +6,10 @@ import logging
|
||||||
import math
|
import math
|
||||||
import threading
|
import threading
|
||||||
import time
|
import time
|
||||||
from typing import TYPE_CHECKING, Callable, Dict, List, Optional, Type
|
from collections import OrderedDict
|
||||||
|
from pathlib import Path
|
||||||
|
from queue import Queue
|
||||||
|
from typing import TYPE_CHECKING, Dict, List, Optional, Type
|
||||||
|
|
||||||
import netaddr
|
import netaddr
|
||||||
|
|
||||||
|
@ -20,11 +23,13 @@ from core.emulator.enumerations import (
|
||||||
RegisterTlvs,
|
RegisterTlvs,
|
||||||
)
|
)
|
||||||
from core.errors import CoreCommandError, CoreError
|
from core.errors import CoreCommandError, CoreError
|
||||||
from core.executables import EBTABLES, TC
|
from core.executables import NFTABLES, TC
|
||||||
from core.nodes.base import CoreNetworkBase
|
from core.nodes.base import CoreNetworkBase
|
||||||
from core.nodes.interface import CoreInterface, GreTap, Veth
|
from core.nodes.interface import CoreInterface, GreTap, Veth
|
||||||
from core.nodes.netclient import get_net_client
|
from core.nodes.netclient import get_net_client
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.emulator.distributed import DistributedServer
|
from core.emulator.distributed import DistributedServer
|
||||||
from core.emulator.session import Session
|
from core.emulator.session import Session
|
||||||
|
@ -33,224 +38,194 @@ if TYPE_CHECKING:
|
||||||
WirelessModelType = Type[WirelessModel]
|
WirelessModelType = Type[WirelessModel]
|
||||||
|
|
||||||
LEARNING_DISABLED: int = 0
|
LEARNING_DISABLED: int = 0
|
||||||
ebtables_lock: threading.Lock = threading.Lock()
|
|
||||||
|
|
||||||
|
|
||||||
class EbtablesQueue:
|
class SetQueue(Queue):
|
||||||
"""
|
"""
|
||||||
Helper class for queuing up ebtables commands into rate-limited
|
Set backed queue to avoid duplicate submissions.
|
||||||
|
"""
|
||||||
|
|
||||||
|
def _init(self, maxsize):
|
||||||
|
self.queue: OrderedDict = OrderedDict()
|
||||||
|
|
||||||
|
def _put(self, item):
|
||||||
|
self.queue[item] = None
|
||||||
|
|
||||||
|
def _get(self):
|
||||||
|
key, _ = self.queue.popitem(last=False)
|
||||||
|
return key
|
||||||
|
|
||||||
|
|
||||||
|
class NftablesQueue:
|
||||||
|
"""
|
||||||
|
Helper class for queuing up nftables commands into rate-limited
|
||||||
atomic commits. This improves performance and reliability when there are
|
atomic commits. This improves performance and reliability when there are
|
||||||
many WLAN link updates.
|
many WLAN link updates.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
# update rate is every 300ms
|
# update rate is every 300ms
|
||||||
rate: float = 0.3
|
rate: float = 0.3
|
||||||
# ebtables
|
atomic_file: str = "/tmp/pycore.nftables.atomic"
|
||||||
atomic_file: str = "/tmp/pycore.ebtables.atomic"
|
chain: str = "forward"
|
||||||
|
|
||||||
def __init__(self) -> None:
|
def __init__(self) -> None:
|
||||||
"""
|
"""
|
||||||
Initialize the helper class, but don't start the update thread
|
Initialize the helper class, but don't start the update thread
|
||||||
until a WLAN is instantiated.
|
until a WLAN is instantiated.
|
||||||
"""
|
"""
|
||||||
self.doupdateloop: bool = False
|
self.running: bool = False
|
||||||
self.updatethread: Optional[threading.Thread] = None
|
self.run_thread: Optional[threading.Thread] = None
|
||||||
# this lock protects cmds and updates lists
|
# this lock protects cmds and updates lists
|
||||||
self.updatelock: threading.Lock = threading.Lock()
|
self.lock: threading.Lock = threading.Lock()
|
||||||
# list of pending ebtables commands
|
# list of pending nftables commands
|
||||||
self.cmds: List[str] = []
|
self.cmds: List[str] = []
|
||||||
# list of WLANs requiring update
|
# list of WLANs requiring update
|
||||||
self.updates: List["CoreNetwork"] = []
|
self.updates: SetQueue = SetQueue()
|
||||||
# timestamps of last WLAN update; this keeps track of WLANs that are
|
# timestamps of last WLAN update; this keeps track of WLANs that are
|
||||||
# using this queue
|
# using this queue
|
||||||
self.last_update_time: Dict["CoreNetwork", float] = {}
|
self.last_update_time: Dict["CoreNetwork", float] = {}
|
||||||
|
|
||||||
def startupdateloop(self, wlan: "CoreNetwork") -> None:
|
def start(self, net: "CoreNetwork") -> None:
|
||||||
"""
|
"""
|
||||||
Kick off the update loop; only needs to be invoked once.
|
Start thread to listen for updates for the provided network.
|
||||||
|
:param net: network to start checking updates
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
with self.updatelock:
|
with self.lock:
|
||||||
self.last_update_time[wlan] = time.monotonic()
|
self.last_update_time[net] = time.monotonic()
|
||||||
if self.doupdateloop:
|
if self.running:
|
||||||
return
|
return
|
||||||
self.doupdateloop = True
|
self.running = True
|
||||||
self.updatethread = threading.Thread(target=self.updateloop, daemon=True)
|
self.run_thread = threading.Thread(target=self.run, daemon=True)
|
||||||
self.updatethread.start()
|
self.run_thread.start()
|
||||||
|
|
||||||
def stopupdateloop(self, wlan: "CoreNetwork") -> None:
|
def stop(self, net: "CoreNetwork") -> None:
|
||||||
"""
|
"""
|
||||||
Kill the update loop thread if there are no more WLANs using it.
|
Stop updates for network, when no networks remain, stop update thread.
|
||||||
|
:param net: network to stop watching updates
|
||||||
|
:return: nothing
|
||||||
|
"""
|
||||||
|
with self.lock:
|
||||||
|
self.last_update_time.pop(net, None)
|
||||||
|
if self.last_update_time:
|
||||||
|
return
|
||||||
|
self.running = False
|
||||||
|
if self.run_thread:
|
||||||
|
self.updates.put(None)
|
||||||
|
self.run_thread.join()
|
||||||
|
self.run_thread = None
|
||||||
|
|
||||||
|
def last_update(self, net: "CoreNetwork") -> float:
|
||||||
|
"""
|
||||||
|
Return the time elapsed since this network was last updated.
|
||||||
|
:param net: network node
|
||||||
|
:return: elapsed time
|
||||||
|
"""
|
||||||
|
now = time.monotonic()
|
||||||
|
last_update = self.last_update_time.setdefault(net, now)
|
||||||
|
return now - last_update
|
||||||
|
|
||||||
|
def run(self) -> None:
|
||||||
|
"""
|
||||||
|
Thread target that looks for networks needing update, and
|
||||||
|
rate limits the amount of nftables activity. Only one userspace program
|
||||||
|
should use nftables at any given time, or results can be unpredictable.
|
||||||
|
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
with self.updatelock:
|
while self.running:
|
||||||
try:
|
net = self.updates.get()
|
||||||
del self.last_update_time[wlan]
|
if net is None:
|
||||||
except KeyError:
|
break
|
||||||
logging.exception(
|
if not net.up:
|
||||||
"error deleting last update time for wlan, ignored before: %s", wlan
|
self.last_update_time[net] = time.monotonic()
|
||||||
)
|
elif self.last_update(net) > self.rate:
|
||||||
if len(self.last_update_time) > 0:
|
with self.lock:
|
||||||
|
self.build_cmds(net)
|
||||||
|
self.commit(net)
|
||||||
|
self.last_update_time[net] = time.monotonic()
|
||||||
|
|
||||||
|
def commit(self, net: "CoreNetwork") -> None:
|
||||||
|
"""
|
||||||
|
Commit changes to nftables for the provided network.
|
||||||
|
:param net: network to commit nftables changes
|
||||||
|
:return: nothing
|
||||||
|
"""
|
||||||
|
if not self.cmds:
|
||||||
return
|
return
|
||||||
self.doupdateloop = False
|
# write out nft commands to file
|
||||||
if self.updatethread:
|
for cmd in self.cmds:
|
||||||
self.updatethread.join()
|
net.host_cmd(f"echo {cmd} >> {self.atomic_file}", shell=True)
|
||||||
self.updatethread = None
|
# read file as atomic change
|
||||||
|
net.host_cmd(f"{NFTABLES} -f {self.atomic_file}")
|
||||||
|
# remove file
|
||||||
|
net.host_cmd(f"rm -f {self.atomic_file}")
|
||||||
|
self.cmds.clear()
|
||||||
|
|
||||||
def ebatomiccmd(self, cmd: str) -> str:
|
def update(self, net: "CoreNetwork") -> None:
|
||||||
"""
|
"""
|
||||||
Helper for building ebtables atomic file command list.
|
Flag this network has an update, so the nftables chain will be rebuilt.
|
||||||
|
:param net: wlan network
|
||||||
:param cmd: ebtable command
|
|
||||||
:return: ebtable atomic command
|
|
||||||
"""
|
|
||||||
return f"{EBTABLES} --atomic-file {self.atomic_file} {cmd}"
|
|
||||||
|
|
||||||
def lastupdate(self, wlan: "CoreNetwork") -> float:
|
|
||||||
"""
|
|
||||||
Return the time elapsed since this WLAN was last updated.
|
|
||||||
|
|
||||||
:param wlan: wlan entity
|
|
||||||
:return: elpased time
|
|
||||||
"""
|
|
||||||
try:
|
|
||||||
elapsed = time.monotonic() - self.last_update_time[wlan]
|
|
||||||
except KeyError:
|
|
||||||
self.last_update_time[wlan] = time.monotonic()
|
|
||||||
elapsed = 0.0
|
|
||||||
|
|
||||||
return elapsed
|
|
||||||
|
|
||||||
def updated(self, wlan: "CoreNetwork") -> None:
|
|
||||||
"""
|
|
||||||
Keep track of when this WLAN was last updated.
|
|
||||||
|
|
||||||
:param wlan: wlan entity
|
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
self.last_update_time[wlan] = time.monotonic()
|
self.updates.put(net)
|
||||||
self.updates.remove(wlan)
|
|
||||||
|
|
||||||
def updateloop(self) -> None:
|
def delete_table(self, net: "CoreNetwork") -> None:
|
||||||
"""
|
"""
|
||||||
Thread target that looks for WLANs needing update, and
|
Delete nftable bridge rule table.
|
||||||
rate limits the amount of ebtables activity. Only one userspace program
|
|
||||||
should use ebtables at any given time, or results can be unpredictable.
|
|
||||||
|
|
||||||
|
:param net: network to delete table for
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
while self.doupdateloop:
|
with self.lock:
|
||||||
with self.updatelock:
|
net.host_cmd(f"{NFTABLES} delete table bridge {net.brname}")
|
||||||
for wlan in self.updates:
|
|
||||||
# Check if wlan is from a previously closed session. Because of the
|
|
||||||
# rate limiting scheme employed here, this may happen if a new session
|
|
||||||
# is started soon after closing a previous session.
|
|
||||||
# TODO: if these are WlanNodes, this will never throw an exception
|
|
||||||
try:
|
|
||||||
wlan.session
|
|
||||||
except Exception:
|
|
||||||
# Just mark as updated to remove from self.updates.
|
|
||||||
self.updated(wlan)
|
|
||||||
continue
|
|
||||||
|
|
||||||
if self.lastupdate(wlan) > self.rate:
|
def build_cmds(self, net: "CoreNetwork") -> None:
|
||||||
self.buildcmds(wlan)
|
|
||||||
self.ebcommit(wlan)
|
|
||||||
self.updated(wlan)
|
|
||||||
|
|
||||||
time.sleep(self.rate)
|
|
||||||
|
|
||||||
def ebcommit(self, wlan: "CoreNetwork") -> None:
|
|
||||||
"""
|
"""
|
||||||
Perform ebtables atomic commit using commands built in the self.cmds list.
|
Inspect linked nodes for a network, and rebuild the nftables chain commands.
|
||||||
|
:param net: network to build commands for
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
# save kernel ebtables snapshot to a file
|
with net.linked_lock:
|
||||||
args = self.ebatomiccmd("--atomic-save")
|
if net.has_nftables_chain:
|
||||||
wlan.host_cmd(args)
|
self.cmds.append(f"flush table bridge {net.brname}")
|
||||||
|
|
||||||
# modify the table file using queued ebtables commands
|
|
||||||
for c in self.cmds:
|
|
||||||
args = self.ebatomiccmd(c)
|
|
||||||
wlan.host_cmd(args)
|
|
||||||
self.cmds = []
|
|
||||||
|
|
||||||
# commit the table file to the kernel
|
|
||||||
args = self.ebatomiccmd("--atomic-commit")
|
|
||||||
wlan.host_cmd(args)
|
|
||||||
|
|
||||||
try:
|
|
||||||
wlan.host_cmd(f"rm -f {self.atomic_file}")
|
|
||||||
except CoreCommandError:
|
|
||||||
logging.exception("error removing atomic file: %s", self.atomic_file)
|
|
||||||
|
|
||||||
def ebchange(self, wlan: "CoreNetwork") -> None:
|
|
||||||
"""
|
|
||||||
Flag a change to the given WLAN's _linked dict, so the ebtables
|
|
||||||
chain will be rebuilt at the next interval.
|
|
||||||
|
|
||||||
:return: nothing
|
|
||||||
"""
|
|
||||||
with self.updatelock:
|
|
||||||
if wlan not in self.updates:
|
|
||||||
self.updates.append(wlan)
|
|
||||||
|
|
||||||
def buildcmds(self, wlan: "CoreNetwork") -> None:
|
|
||||||
"""
|
|
||||||
Inspect a _linked dict from a wlan, and rebuild the ebtables chain for that WLAN.
|
|
||||||
|
|
||||||
:return: nothing
|
|
||||||
"""
|
|
||||||
with wlan._linked_lock:
|
|
||||||
if wlan.has_ebtables_chain:
|
|
||||||
# flush the chain
|
|
||||||
self.cmds.append(f"-F {wlan.brname}")
|
|
||||||
else:
|
else:
|
||||||
wlan.has_ebtables_chain = True
|
net.has_nftables_chain = True
|
||||||
self.cmds.extend(
|
policy = net.policy.value.lower()
|
||||||
[
|
self.cmds.append(f"add table bridge {net.brname}")
|
||||||
f"-N {wlan.brname} -P {wlan.policy.value}",
|
self.cmds.append(
|
||||||
f"-A FORWARD --logical-in {wlan.brname} -j {wlan.brname}",
|
f"add chain bridge {net.brname} {self.chain} {{type filter hook "
|
||||||
]
|
f"forward priority 0\\; policy {policy}\\;}}"
|
||||||
)
|
)
|
||||||
|
# add default rule to accept all traffic not for this bridge
|
||||||
|
self.cmds.append(
|
||||||
|
f"add rule bridge {net.brname} {self.chain} "
|
||||||
|
f"ibriport != {net.brname} accept"
|
||||||
|
)
|
||||||
# rebuild the chain
|
# rebuild the chain
|
||||||
for iface1, v in wlan._linked.items():
|
for iface1, v in net.linked.items():
|
||||||
for oface2, linked in v.items():
|
for iface2, linked in v.items():
|
||||||
if wlan.policy == NetworkPolicy.DROP and linked:
|
policy = None
|
||||||
self.cmds.extend(
|
if net.policy == NetworkPolicy.DROP and linked:
|
||||||
[
|
policy = "accept"
|
||||||
f"-A {wlan.brname} -i {iface1.localname} -o {oface2.localname} -j ACCEPT",
|
elif net.policy == NetworkPolicy.ACCEPT and not linked:
|
||||||
f"-A {wlan.brname} -o {iface1.localname} -i {oface2.localname} -j ACCEPT",
|
policy = "drop"
|
||||||
]
|
if policy:
|
||||||
|
self.cmds.append(
|
||||||
|
f"add rule bridge {net.brname} {self.chain} "
|
||||||
|
f"iif {iface1.localname} oif {iface2.localname} "
|
||||||
|
f"{policy}"
|
||||||
)
|
)
|
||||||
elif wlan.policy == NetworkPolicy.ACCEPT and not linked:
|
self.cmds.append(
|
||||||
self.cmds.extend(
|
f"add rule bridge {net.brname} {self.chain} "
|
||||||
[
|
f"oif {iface1.localname} iif {iface2.localname} "
|
||||||
f"-A {wlan.brname} -i {iface1.localname} -o {oface2.localname} -j DROP",
|
f"{policy}"
|
||||||
f"-A {wlan.brname} -o {iface1.localname} -i {oface2.localname} -j DROP",
|
|
||||||
]
|
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
# a global object because all WLANs share the same queue
|
# a global object because all networks share the same queue
|
||||||
# cannot have multiple threads invoking the ebtables commnd
|
# cannot have multiple threads invoking the nftables commnd
|
||||||
ebq: EbtablesQueue = EbtablesQueue()
|
nft_queue: NftablesQueue = NftablesQueue()
|
||||||
|
|
||||||
|
|
||||||
def ebtablescmds(call: Callable[..., str], cmds: List[str]) -> None:
|
|
||||||
"""
|
|
||||||
Run ebtable commands.
|
|
||||||
|
|
||||||
:param call: function to call commands
|
|
||||||
:param cmds: commands to call
|
|
||||||
:return: nothing
|
|
||||||
"""
|
|
||||||
with ebtables_lock:
|
|
||||||
for args in cmds:
|
|
||||||
call(args)
|
|
||||||
|
|
||||||
|
|
||||||
class CoreNetwork(CoreNetworkBase):
|
class CoreNetwork(CoreNetworkBase):
|
||||||
|
@ -282,17 +257,17 @@ class CoreNetwork(CoreNetworkBase):
|
||||||
if name is None:
|
if name is None:
|
||||||
name = str(self.id)
|
name = str(self.id)
|
||||||
if policy is not None:
|
if policy is not None:
|
||||||
self.policy = policy
|
self.policy: NetworkPolicy = policy
|
||||||
self.name: Optional[str] = name
|
self.name: Optional[str] = name
|
||||||
sessionid = self.session.short_session_id()
|
sessionid = self.session.short_session_id()
|
||||||
self.brname: str = f"b.{self.id}.{sessionid}"
|
self.brname: str = f"b.{self.id}.{sessionid}"
|
||||||
self.has_ebtables_chain: bool = False
|
self.has_nftables_chain: bool = False
|
||||||
|
|
||||||
def host_cmd(
|
def host_cmd(
|
||||||
self,
|
self,
|
||||||
args: str,
|
args: str,
|
||||||
env: Dict[str, str] = None,
|
env: Dict[str, str] = None,
|
||||||
cwd: str = None,
|
cwd: Path = None,
|
||||||
wait: bool = True,
|
wait: bool = True,
|
||||||
shell: bool = False,
|
shell: bool = False,
|
||||||
) -> str:
|
) -> str:
|
||||||
|
@ -308,7 +283,7 @@ class CoreNetwork(CoreNetworkBase):
|
||||||
:return: combined stdout and stderr
|
:return: combined stdout and stderr
|
||||||
:raises CoreCommandError: when a non-zero exit status occurs
|
:raises CoreCommandError: when a non-zero exit status occurs
|
||||||
"""
|
"""
|
||||||
logging.debug("network node(%s) cmd", self.name)
|
logger.debug("network node(%s) cmd", self.name)
|
||||||
output = utils.cmd(args, env, cwd, wait, shell)
|
output = utils.cmd(args, env, cwd, wait, shell)
|
||||||
self.session.distributed.execute(lambda x: x.remote_cmd(args, env, cwd, wait))
|
self.session.distributed.execute(lambda x: x.remote_cmd(args, env, cwd, wait))
|
||||||
return output
|
return output
|
||||||
|
@ -321,9 +296,9 @@ class CoreNetwork(CoreNetworkBase):
|
||||||
:raises CoreCommandError: when there is a command exception
|
:raises CoreCommandError: when there is a command exception
|
||||||
"""
|
"""
|
||||||
self.net_client.create_bridge(self.brname)
|
self.net_client.create_bridge(self.brname)
|
||||||
self.has_ebtables_chain = False
|
self.has_nftables_chain = False
|
||||||
self.up = True
|
self.up = True
|
||||||
ebq.startupdateloop(self)
|
nft_queue.start(self)
|
||||||
|
|
||||||
def shutdown(self) -> None:
|
def shutdown(self) -> None:
|
||||||
"""
|
"""
|
||||||
|
@ -333,27 +308,18 @@ class CoreNetwork(CoreNetworkBase):
|
||||||
"""
|
"""
|
||||||
if not self.up:
|
if not self.up:
|
||||||
return
|
return
|
||||||
|
nft_queue.stop(self)
|
||||||
ebq.stopupdateloop(self)
|
|
||||||
|
|
||||||
try:
|
try:
|
||||||
self.net_client.delete_bridge(self.brname)
|
self.net_client.delete_bridge(self.brname)
|
||||||
if self.has_ebtables_chain:
|
if self.has_nftables_chain:
|
||||||
cmds = [
|
nft_queue.delete_table(self)
|
||||||
f"{EBTABLES} -D FORWARD --logical-in {self.brname} -j {self.brname}",
|
|
||||||
f"{EBTABLES} -X {self.brname}",
|
|
||||||
]
|
|
||||||
ebtablescmds(self.host_cmd, cmds)
|
|
||||||
except CoreCommandError:
|
except CoreCommandError:
|
||||||
logging.exception("error during shutdown")
|
logging.exception("error during shutdown")
|
||||||
|
|
||||||
# removes veth pairs used for bridge-to-bridge connections
|
# removes veth pairs used for bridge-to-bridge connections
|
||||||
for iface in self.get_ifaces():
|
for iface in self.get_ifaces():
|
||||||
iface.shutdown()
|
iface.shutdown()
|
||||||
|
|
||||||
self.ifaces.clear()
|
self.ifaces.clear()
|
||||||
self._linked.clear()
|
self.linked.clear()
|
||||||
del self.session
|
|
||||||
self.up = False
|
self.up = False
|
||||||
|
|
||||||
def attach(self, iface: CoreInterface) -> None:
|
def attach(self, iface: CoreInterface) -> None:
|
||||||
|
@ -378,7 +344,7 @@ class CoreNetwork(CoreNetworkBase):
|
||||||
iface.net_client.delete_iface(self.brname, iface.localname)
|
iface.net_client.delete_iface(self.brname, iface.localname)
|
||||||
super().detach(iface)
|
super().detach(iface)
|
||||||
|
|
||||||
def linked(self, iface1: CoreInterface, iface2: CoreInterface) -> bool:
|
def is_linked(self, iface1: CoreInterface, iface2: CoreInterface) -> bool:
|
||||||
"""
|
"""
|
||||||
Determine if the provided network interfaces are linked.
|
Determine if the provided network interfaces are linked.
|
||||||
|
|
||||||
|
@ -389,12 +355,10 @@ class CoreNetwork(CoreNetworkBase):
|
||||||
# check if the network interfaces are attached to this network
|
# check if the network interfaces are attached to this network
|
||||||
if self.ifaces[iface1.net_id] != iface1:
|
if self.ifaces[iface1.net_id] != iface1:
|
||||||
raise ValueError(f"inconsistency for interface {iface1.name}")
|
raise ValueError(f"inconsistency for interface {iface1.name}")
|
||||||
|
|
||||||
if self.ifaces[iface2.net_id] != iface2:
|
if self.ifaces[iface2.net_id] != iface2:
|
||||||
raise ValueError(f"inconsistency for interface {iface2.name}")
|
raise ValueError(f"inconsistency for interface {iface2.name}")
|
||||||
|
|
||||||
try:
|
try:
|
||||||
linked = self._linked[iface1][iface2]
|
linked = self.linked[iface1][iface2]
|
||||||
except KeyError:
|
except KeyError:
|
||||||
if self.policy == NetworkPolicy.ACCEPT:
|
if self.policy == NetworkPolicy.ACCEPT:
|
||||||
linked = True
|
linked = True
|
||||||
|
@ -402,41 +366,37 @@ class CoreNetwork(CoreNetworkBase):
|
||||||
linked = False
|
linked = False
|
||||||
else:
|
else:
|
||||||
raise Exception(f"unknown policy: {self.policy.value}")
|
raise Exception(f"unknown policy: {self.policy.value}")
|
||||||
self._linked[iface1][iface2] = linked
|
self.linked[iface1][iface2] = linked
|
||||||
|
|
||||||
return linked
|
return linked
|
||||||
|
|
||||||
def unlink(self, iface1: CoreInterface, iface2: CoreInterface) -> None:
|
def unlink(self, iface1: CoreInterface, iface2: CoreInterface) -> None:
|
||||||
"""
|
"""
|
||||||
Unlink two interfaces, resulting in adding or removing ebtables
|
Unlink two interfaces, resulting in adding or removing filtering rules.
|
||||||
|
|
||||||
|
:param iface1: interface one
|
||||||
|
:param iface2: interface two
|
||||||
|
:return: nothing
|
||||||
|
"""
|
||||||
|
with self.linked_lock:
|
||||||
|
if not self.is_linked(iface1, iface2):
|
||||||
|
return
|
||||||
|
self.linked[iface1][iface2] = False
|
||||||
|
nft_queue.update(self)
|
||||||
|
|
||||||
|
def link(self, iface1: CoreInterface, iface2: CoreInterface) -> None:
|
||||||
|
"""
|
||||||
|
Link two interfaces together, resulting in adding or removing
|
||||||
filtering rules.
|
filtering rules.
|
||||||
|
|
||||||
:param iface1: interface one
|
:param iface1: interface one
|
||||||
:param iface2: interface two
|
:param iface2: interface two
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
with self._linked_lock:
|
with self.linked_lock:
|
||||||
if not self.linked(iface1, iface2):
|
if self.is_linked(iface1, iface2):
|
||||||
return
|
return
|
||||||
self._linked[iface1][iface2] = False
|
self.linked[iface1][iface2] = True
|
||||||
|
nft_queue.update(self)
|
||||||
ebq.ebchange(self)
|
|
||||||
|
|
||||||
def link(self, iface1: CoreInterface, iface2: CoreInterface) -> None:
|
|
||||||
"""
|
|
||||||
Link two interfaces together, resulting in adding or removing
|
|
||||||
ebtables filtering rules.
|
|
||||||
|
|
||||||
:param iface1: interface one
|
|
||||||
:param iface2: interface two
|
|
||||||
:return: nothing
|
|
||||||
"""
|
|
||||||
with self._linked_lock:
|
|
||||||
if self.linked(iface1, iface2):
|
|
||||||
return
|
|
||||||
self._linked[iface1][iface2] = True
|
|
||||||
|
|
||||||
ebq.ebchange(self)
|
|
||||||
|
|
||||||
def linkconfig(
|
def linkconfig(
|
||||||
self, iface: CoreInterface, options: LinkOptions, iface2: CoreInterface = None
|
self, iface: CoreInterface, options: LinkOptions, iface2: CoreInterface = None
|
||||||
|
@ -542,8 +502,8 @@ class CoreNetwork(CoreNetworkBase):
|
||||||
iface.net_client.set_iface_master(net.brname, iface.name)
|
iface.net_client.set_iface_master(net.brname, iface.name)
|
||||||
i = net.next_iface_id()
|
i = net.next_iface_id()
|
||||||
net.ifaces[i] = iface
|
net.ifaces[i] = iface
|
||||||
with net._linked_lock:
|
with net.linked_lock:
|
||||||
net._linked[iface] = {}
|
net.linked[iface] = {}
|
||||||
iface.net = self
|
iface.net = self
|
||||||
iface.othernet = net
|
iface.othernet = net
|
||||||
return iface
|
return iface
|
||||||
|
@ -769,7 +729,7 @@ class CtrlNet(CoreNetwork):
|
||||||
raise CoreError(f"old bridges exist for node: {self.id}")
|
raise CoreError(f"old bridges exist for node: {self.id}")
|
||||||
|
|
||||||
super().startup()
|
super().startup()
|
||||||
logging.info("added control network bridge: %s %s", self.brname, self.prefix)
|
logger.info("added control network bridge: %s %s", self.brname, self.prefix)
|
||||||
|
|
||||||
if self.hostid and self.assign_address:
|
if self.hostid and self.assign_address:
|
||||||
self.add_addresses(self.hostid)
|
self.add_addresses(self.hostid)
|
||||||
|
@ -777,7 +737,7 @@ class CtrlNet(CoreNetwork):
|
||||||
self.add_addresses(-2)
|
self.add_addresses(-2)
|
||||||
|
|
||||||
if self.updown_script:
|
if self.updown_script:
|
||||||
logging.info(
|
logger.info(
|
||||||
"interface %s updown script (%s startup) called",
|
"interface %s updown script (%s startup) called",
|
||||||
self.brname,
|
self.brname,
|
||||||
self.updown_script,
|
self.updown_script,
|
||||||
|
@ -797,7 +757,7 @@ class CtrlNet(CoreNetwork):
|
||||||
try:
|
try:
|
||||||
self.net_client.delete_iface(self.brname, self.serverintf)
|
self.net_client.delete_iface(self.brname, self.serverintf)
|
||||||
except CoreCommandError:
|
except CoreCommandError:
|
||||||
logging.exception(
|
logger.exception(
|
||||||
"error deleting server interface %s from bridge %s",
|
"error deleting server interface %s from bridge %s",
|
||||||
self.serverintf,
|
self.serverintf,
|
||||||
self.brname,
|
self.brname,
|
||||||
|
@ -805,14 +765,14 @@ class CtrlNet(CoreNetwork):
|
||||||
|
|
||||||
if self.updown_script is not None:
|
if self.updown_script is not None:
|
||||||
try:
|
try:
|
||||||
logging.info(
|
logger.info(
|
||||||
"interface %s updown script (%s shutdown) called",
|
"interface %s updown script (%s shutdown) called",
|
||||||
self.brname,
|
self.brname,
|
||||||
self.updown_script,
|
self.updown_script,
|
||||||
)
|
)
|
||||||
self.host_cmd(f"{self.updown_script} {self.brname} shutdown")
|
self.host_cmd(f"{self.updown_script} {self.brname} shutdown")
|
||||||
except CoreCommandError:
|
except CoreCommandError:
|
||||||
logging.exception("error issuing shutdown script shutdown")
|
logger.exception("error issuing shutdown script shutdown")
|
||||||
|
|
||||||
super().shutdown()
|
super().shutdown()
|
||||||
|
|
||||||
|
@ -990,7 +950,7 @@ class WlanNode(CoreNetwork):
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
super().startup()
|
super().startup()
|
||||||
ebq.ebchange(self)
|
nft_queue.update(self)
|
||||||
|
|
||||||
def attach(self, iface: CoreInterface) -> None:
|
def attach(self, iface: CoreInterface) -> None:
|
||||||
"""
|
"""
|
||||||
|
@ -1012,7 +972,7 @@ class WlanNode(CoreNetwork):
|
||||||
:param config: configuration for model being set
|
:param config: configuration for model being set
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
logging.debug("node(%s) setting model: %s", self.name, model.name)
|
logger.debug("node(%s) setting model: %s", self.name, model.name)
|
||||||
if model.config_type == RegisterTlvs.WIRELESS:
|
if model.config_type == RegisterTlvs.WIRELESS:
|
||||||
self.model = model(session=self.session, _id=self.id)
|
self.model = model(session=self.session, _id=self.id)
|
||||||
for iface in self.get_ifaces():
|
for iface in self.get_ifaces():
|
||||||
|
@ -1031,7 +991,7 @@ class WlanNode(CoreNetwork):
|
||||||
def updatemodel(self, config: Dict[str, str]) -> None:
|
def updatemodel(self, config: Dict[str, str]) -> None:
|
||||||
if not self.model:
|
if not self.model:
|
||||||
raise CoreError(f"no model set to update for node({self.name})")
|
raise CoreError(f"no model set to update for node({self.name})")
|
||||||
logging.debug(
|
logger.debug(
|
||||||
"node(%s) updating model(%s): %s", self.id, self.model.name, config
|
"node(%s) updating model(%s): %s", self.id, self.model.name, config
|
||||||
)
|
)
|
||||||
self.model.update_config(config)
|
self.model.update_config(config)
|
||||||
|
|
|
@ -3,9 +3,9 @@ PhysicalNode class for including real systems in the emulated network.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
import logging
|
import logging
|
||||||
import os
|
|
||||||
import threading
|
import threading
|
||||||
from typing import IO, TYPE_CHECKING, List, Optional, Tuple
|
from pathlib import Path
|
||||||
|
from typing import TYPE_CHECKING, List, Optional, Tuple
|
||||||
|
|
||||||
from core.emulator.data import InterfaceData, LinkOptions
|
from core.emulator.data import InterfaceData, LinkOptions
|
||||||
from core.emulator.distributed import DistributedServer
|
from core.emulator.distributed import DistributedServer
|
||||||
|
@ -16,6 +16,8 @@ from core.nodes.base import CoreNetworkBase, CoreNodeBase
|
||||||
from core.nodes.interface import DEFAULT_MTU, CoreInterface
|
from core.nodes.interface import DEFAULT_MTU, CoreInterface
|
||||||
from core.nodes.network import CoreNetwork, GreTap
|
from core.nodes.network import CoreNetwork, GreTap
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.emulator.session import Session
|
from core.emulator.session import Session
|
||||||
|
|
||||||
|
@ -26,15 +28,15 @@ class PhysicalNode(CoreNodeBase):
|
||||||
session: "Session",
|
session: "Session",
|
||||||
_id: int = None,
|
_id: int = None,
|
||||||
name: str = None,
|
name: str = None,
|
||||||
nodedir: str = None,
|
directory: Path = None,
|
||||||
server: DistributedServer = None,
|
server: DistributedServer = None,
|
||||||
) -> None:
|
) -> None:
|
||||||
super().__init__(session, _id, name, server)
|
super().__init__(session, _id, name, server)
|
||||||
if not self.server:
|
if not self.server:
|
||||||
raise CoreError("physical nodes must be assigned to a remote server")
|
raise CoreError("physical nodes must be assigned to a remote server")
|
||||||
self.nodedir: Optional[str] = nodedir
|
self.directory: Optional[Path] = directory
|
||||||
self.lock: threading.RLock = threading.RLock()
|
self.lock: threading.RLock = threading.RLock()
|
||||||
self._mounts: List[Tuple[str, str]] = []
|
self._mounts: List[Tuple[Path, Path]] = []
|
||||||
|
|
||||||
def startup(self) -> None:
|
def startup(self) -> None:
|
||||||
with self.lock:
|
with self.lock:
|
||||||
|
@ -44,15 +46,12 @@ class PhysicalNode(CoreNodeBase):
|
||||||
def shutdown(self) -> None:
|
def shutdown(self) -> None:
|
||||||
if not self.up:
|
if not self.up:
|
||||||
return
|
return
|
||||||
|
|
||||||
with self.lock:
|
with self.lock:
|
||||||
while self._mounts:
|
while self._mounts:
|
||||||
_source, target = self._mounts.pop(-1)
|
_, target_path = self._mounts.pop(-1)
|
||||||
self.umount(target)
|
self.umount(target_path)
|
||||||
|
|
||||||
for iface in self.get_ifaces():
|
for iface in self.get_ifaces():
|
||||||
iface.shutdown()
|
iface.shutdown()
|
||||||
|
|
||||||
self.rmnodedir()
|
self.rmnodedir()
|
||||||
|
|
||||||
def path_exists(self, path: str) -> bool:
|
def path_exists(self, path: str) -> bool:
|
||||||
|
@ -166,7 +165,7 @@ class PhysicalNode(CoreNodeBase):
|
||||||
def new_iface(
|
def new_iface(
|
||||||
self, net: CoreNetworkBase, iface_data: InterfaceData
|
self, net: CoreNetworkBase, iface_data: InterfaceData
|
||||||
) -> CoreInterface:
|
) -> CoreInterface:
|
||||||
logging.info("creating interface")
|
logger.info("creating interface")
|
||||||
ips = iface_data.get_ips()
|
ips = iface_data.get_ips()
|
||||||
iface_id = iface_data.id
|
iface_id = iface_data.id
|
||||||
if iface_id is None:
|
if iface_id is None:
|
||||||
|
@ -186,55 +185,40 @@ class PhysicalNode(CoreNodeBase):
|
||||||
self.adopt_iface(iface, iface_id, iface_data.mac, ips)
|
self.adopt_iface(iface, iface_id, iface_data.mac, ips)
|
||||||
return iface
|
return iface
|
||||||
|
|
||||||
def privatedir(self, path: str) -> None:
|
def privatedir(self, dir_path: Path) -> None:
|
||||||
if path[0] != "/":
|
if not str(dir_path).startswith("/"):
|
||||||
raise ValueError(f"path not fully qualified: {path}")
|
raise CoreError(f"private directory path not fully qualified: {dir_path}")
|
||||||
hostpath = os.path.join(
|
host_path = self.host_path(dir_path, is_dir=True)
|
||||||
self.nodedir, os.path.normpath(path).strip("/").replace("/", ".")
|
self.host_cmd(f"mkdir -p {host_path}")
|
||||||
)
|
self.mount(host_path, dir_path)
|
||||||
os.mkdir(hostpath)
|
|
||||||
self.mount(hostpath, path)
|
|
||||||
|
|
||||||
def mount(self, source: str, target: str) -> None:
|
def mount(self, src_path: Path, target_path: Path) -> None:
|
||||||
source = os.path.abspath(source)
|
logger.debug("node(%s) mounting: %s at %s", self.name, src_path, target_path)
|
||||||
logging.info("mounting %s at %s", source, target)
|
self.cmd(f"mkdir -p {target_path}")
|
||||||
os.makedirs(target)
|
self.host_cmd(f"{MOUNT} --bind {src_path} {target_path}", cwd=self.directory)
|
||||||
self.host_cmd(f"{MOUNT} --bind {source} {target}", cwd=self.nodedir)
|
self._mounts.append((src_path, target_path))
|
||||||
self._mounts.append((source, target))
|
|
||||||
|
|
||||||
def umount(self, target: str) -> None:
|
def umount(self, target_path: Path) -> None:
|
||||||
logging.info("unmounting '%s'", target)
|
logger.info("unmounting '%s'", target_path)
|
||||||
try:
|
try:
|
||||||
self.host_cmd(f"{UMOUNT} -l {target}", cwd=self.nodedir)
|
self.host_cmd(f"{UMOUNT} -l {target_path}", cwd=self.directory)
|
||||||
except CoreCommandError:
|
except CoreCommandError:
|
||||||
logging.exception("unmounting failed for %s", target)
|
logger.exception("unmounting failed for %s", target_path)
|
||||||
|
|
||||||
def opennodefile(self, filename: str, mode: str = "w") -> IO:
|
def nodefile(self, file_path: Path, contents: str, mode: int = 0o644) -> None:
|
||||||
dirname, basename = os.path.split(filename)
|
host_path = self.host_path(file_path)
|
||||||
if not basename:
|
directory = host_path.parent
|
||||||
raise ValueError("no basename for filename: " + filename)
|
if not directory.is_dir():
|
||||||
|
directory.mkdir(parents=True, mode=0o755)
|
||||||
if dirname and dirname[0] == "/":
|
with host_path.open("w") as f:
|
||||||
dirname = dirname[1:]
|
|
||||||
|
|
||||||
dirname = dirname.replace("/", ".")
|
|
||||||
dirname = os.path.join(self.nodedir, dirname)
|
|
||||||
if not os.path.isdir(dirname):
|
|
||||||
os.makedirs(dirname, mode=0o755)
|
|
||||||
|
|
||||||
hostfilename = os.path.join(dirname, basename)
|
|
||||||
return open(hostfilename, mode)
|
|
||||||
|
|
||||||
def nodefile(self, filename: str, contents: str, mode: int = 0o644) -> None:
|
|
||||||
with self.opennodefile(filename, "w") as f:
|
|
||||||
f.write(contents)
|
f.write(contents)
|
||||||
os.chmod(f.name, mode)
|
host_path.chmod(mode)
|
||||||
logging.info("created nodefile: '%s'; mode: 0%o", f.name, mode)
|
logger.info("created nodefile: '%s'; mode: 0%o", host_path, mode)
|
||||||
|
|
||||||
def cmd(self, args: str, wait: bool = True, shell: bool = False) -> str:
|
def cmd(self, args: str, wait: bool = True, shell: bool = False) -> str:
|
||||||
return self.host_cmd(args, wait=wait)
|
return self.host_cmd(args, wait=wait)
|
||||||
|
|
||||||
def addfile(self, srcname: str, filename: str) -> None:
|
def addfile(self, src_path: str, file_path: str) -> None:
|
||||||
raise CoreError("physical node does not support addfile")
|
raise CoreError("physical node does not support addfile")
|
||||||
|
|
||||||
|
|
||||||
|
@ -433,7 +417,7 @@ class Rj45Node(CoreNodeBase):
|
||||||
if items[1][:4] == "fe80":
|
if items[1][:4] == "fe80":
|
||||||
continue
|
continue
|
||||||
self.old_addrs.append((items[1], None))
|
self.old_addrs.append((items[1], None))
|
||||||
logging.info("saved rj45 state: addrs(%s) up(%s)", self.old_addrs, self.old_up)
|
logger.info("saved rj45 state: addrs(%s) up(%s)", self.old_addrs, self.old_up)
|
||||||
|
|
||||||
def restorestate(self) -> None:
|
def restorestate(self) -> None:
|
||||||
"""
|
"""
|
||||||
|
@ -443,7 +427,7 @@ class Rj45Node(CoreNodeBase):
|
||||||
:raises CoreCommandError: when there is a command exception
|
:raises CoreCommandError: when there is a command exception
|
||||||
"""
|
"""
|
||||||
localname = self.iface.localname
|
localname = self.iface.localname
|
||||||
logging.info("restoring rj45 state: %s", localname)
|
logger.info("restoring rj45 state: %s", localname)
|
||||||
for addr in self.old_addrs:
|
for addr in self.old_addrs:
|
||||||
self.net_client.create_address(localname, addr[0], addr[1])
|
self.net_client.create_address(localname, addr[0], addr[1])
|
||||||
if self.old_up:
|
if self.old_up:
|
||||||
|
@ -464,10 +448,10 @@ class Rj45Node(CoreNodeBase):
|
||||||
def termcmdstring(self, sh: str) -> str:
|
def termcmdstring(self, sh: str) -> str:
|
||||||
raise CoreError("rj45 does not support terminal commands")
|
raise CoreError("rj45 does not support terminal commands")
|
||||||
|
|
||||||
def addfile(self, srcname: str, filename: str) -> None:
|
def addfile(self, src_path: str, file_path: str) -> None:
|
||||||
raise CoreError("rj45 does not support addfile")
|
raise CoreError("rj45 does not support addfile")
|
||||||
|
|
||||||
def nodefile(self, filename: str, contents: str, mode: int = 0o644) -> None:
|
def nodefile(self, file_path: str, contents: str, mode: int = 0o644) -> None:
|
||||||
raise CoreError("rj45 does not support nodefile")
|
raise CoreError("rj45 does not support nodefile")
|
||||||
|
|
||||||
def cmd(self, args: str, wait: bool = True, shell: bool = False) -> str:
|
def cmd(self, args: str, wait: bool = True, shell: bool = False) -> str:
|
||||||
|
|
|
@ -15,6 +15,8 @@ from core.errors import CoreError
|
||||||
from core.nodes.base import CoreNetworkBase, NodeBase
|
from core.nodes.base import CoreNetworkBase, NodeBase
|
||||||
from core.nodes.network import WlanNode
|
from core.nodes.network import WlanNode
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.emulator.session import Session
|
from core.emulator.session import Session
|
||||||
|
|
||||||
|
@ -109,7 +111,7 @@ class Sdt:
|
||||||
return False
|
return False
|
||||||
|
|
||||||
self.seturl()
|
self.seturl()
|
||||||
logging.info("connecting to SDT at %s://%s", self.protocol, self.address)
|
logger.info("connecting to SDT at %s://%s", self.protocol, self.address)
|
||||||
if self.sock is None:
|
if self.sock is None:
|
||||||
try:
|
try:
|
||||||
if self.protocol.lower() == "udp":
|
if self.protocol.lower() == "udp":
|
||||||
|
@ -119,7 +121,7 @@ class Sdt:
|
||||||
# Default to tcp
|
# Default to tcp
|
||||||
self.sock = socket.create_connection(self.address, 5)
|
self.sock = socket.create_connection(self.address, 5)
|
||||||
except IOError:
|
except IOError:
|
||||||
logging.exception("SDT socket connect error")
|
logger.exception("SDT socket connect error")
|
||||||
return False
|
return False
|
||||||
|
|
||||||
if not self.initialize():
|
if not self.initialize():
|
||||||
|
@ -157,7 +159,7 @@ class Sdt:
|
||||||
try:
|
try:
|
||||||
self.sock.close()
|
self.sock.close()
|
||||||
except IOError:
|
except IOError:
|
||||||
logging.error("error closing socket")
|
logger.error("error closing socket")
|
||||||
finally:
|
finally:
|
||||||
self.sock = None
|
self.sock = None
|
||||||
|
|
||||||
|
@ -191,11 +193,11 @@ class Sdt:
|
||||||
|
|
||||||
try:
|
try:
|
||||||
cmd = f"{cmdstr}\n".encode()
|
cmd = f"{cmdstr}\n".encode()
|
||||||
logging.debug("sdt cmd: %s", cmd)
|
logger.debug("sdt cmd: %s", cmd)
|
||||||
self.sock.sendall(cmd)
|
self.sock.sendall(cmd)
|
||||||
return True
|
return True
|
||||||
except IOError:
|
except IOError:
|
||||||
logging.exception("SDT connection error")
|
logger.exception("SDT connection error")
|
||||||
self.sock = None
|
self.sock = None
|
||||||
self.connected = False
|
self.connected = False
|
||||||
return False
|
return False
|
||||||
|
@ -250,7 +252,7 @@ class Sdt:
|
||||||
:param node: node to add
|
:param node: node to add
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
logging.debug("sdt add node: %s - %s", node.id, node.name)
|
logger.debug("sdt add node: %s - %s", node.id, node.name)
|
||||||
if not self.connect():
|
if not self.connect():
|
||||||
return
|
return
|
||||||
pos = self.get_node_position(node)
|
pos = self.get_node_position(node)
|
||||||
|
@ -262,8 +264,8 @@ class Sdt:
|
||||||
icon = node.icon
|
icon = node.icon
|
||||||
if icon:
|
if icon:
|
||||||
node_type = node.name
|
node_type = node.name
|
||||||
icon = icon.replace("$CORE_DATA_DIR", CORE_DATA_DIR)
|
icon = icon.replace("$CORE_DATA_DIR", str(CORE_DATA_DIR))
|
||||||
icon = icon.replace("$CORE_CONF_DIR", CORE_CONF_DIR)
|
icon = icon.replace("$CORE_CONF_DIR", str(CORE_CONF_DIR))
|
||||||
self.cmd(f"sprite {node_type} image {icon}")
|
self.cmd(f"sprite {node_type} image {icon}")
|
||||||
self.cmd(
|
self.cmd(
|
||||||
f'node {node.id} nodeLayer "{NODE_LAYER}" '
|
f'node {node.id} nodeLayer "{NODE_LAYER}" '
|
||||||
|
@ -280,7 +282,7 @@ class Sdt:
|
||||||
:param alt: node altitude
|
:param alt: node altitude
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
logging.debug("sdt update node: %s - %s", node.id, node.name)
|
logger.debug("sdt update node: %s - %s", node.id, node.name)
|
||||||
if not self.connect():
|
if not self.connect():
|
||||||
return
|
return
|
||||||
|
|
||||||
|
@ -300,7 +302,7 @@ class Sdt:
|
||||||
:param node_id: node id to delete
|
:param node_id: node id to delete
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
logging.debug("sdt delete node: %s", node_id)
|
logger.debug("sdt delete node: %s", node_id)
|
||||||
if not self.connect():
|
if not self.connect():
|
||||||
return
|
return
|
||||||
self.cmd(f"delete node,{node_id}")
|
self.cmd(f"delete node,{node_id}")
|
||||||
|
@ -315,7 +317,7 @@ class Sdt:
|
||||||
if not self.connect():
|
if not self.connect():
|
||||||
return
|
return
|
||||||
node = node_data.node
|
node = node_data.node
|
||||||
logging.debug("sdt handle node update: %s - %s", node.id, node.name)
|
logger.debug("sdt handle node update: %s - %s", node.id, node.name)
|
||||||
if node_data.message_type == MessageFlags.DELETE:
|
if node_data.message_type == MessageFlags.DELETE:
|
||||||
self.cmd(f"delete node,{node.id}")
|
self.cmd(f"delete node,{node.id}")
|
||||||
else:
|
else:
|
||||||
|
@ -356,7 +358,7 @@ class Sdt:
|
||||||
:param label: label for link
|
:param label: label for link
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
logging.debug("sdt add link: %s, %s, %s", node1_id, node2_id, network_id)
|
logger.debug("sdt add link: %s, %s, %s", node1_id, node2_id, network_id)
|
||||||
if not self.connect():
|
if not self.connect():
|
||||||
return
|
return
|
||||||
if self.wireless_net_check(node1_id) or self.wireless_net_check(node2_id):
|
if self.wireless_net_check(node1_id) or self.wireless_net_check(node2_id):
|
||||||
|
@ -396,7 +398,7 @@ class Sdt:
|
||||||
:param network_id: network link is associated with, None otherwise
|
:param network_id: network link is associated with, None otherwise
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
logging.debug("sdt delete link: %s, %s, %s", node1_id, node2_id, network_id)
|
logger.debug("sdt delete link: %s, %s, %s", node1_id, node2_id, network_id)
|
||||||
if not self.connect():
|
if not self.connect():
|
||||||
return
|
return
|
||||||
if self.wireless_net_check(node1_id) or self.wireless_net_check(node2_id):
|
if self.wireless_net_check(node1_id) or self.wireless_net_check(node2_id):
|
||||||
|
@ -416,7 +418,7 @@ class Sdt:
|
||||||
:param label: label to update
|
:param label: label to update
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
logging.debug("sdt edit link: %s, %s, %s", node1_id, node2_id, network_id)
|
logger.debug("sdt edit link: %s, %s, %s", node1_id, node2_id, network_id)
|
||||||
if not self.connect():
|
if not self.connect():
|
||||||
return
|
return
|
||||||
if self.wireless_net_check(node1_id) or self.wireless_net_check(node2_id):
|
if self.wireless_net_check(node1_id) or self.wireless_net_check(node2_id):
|
||||||
|
|
|
@ -4,11 +4,11 @@ Services
|
||||||
Services available to nodes can be put in this directory. Everything listed in
|
Services available to nodes can be put in this directory. Everything listed in
|
||||||
__all__ is automatically loaded by the main core module.
|
__all__ is automatically loaded by the main core module.
|
||||||
"""
|
"""
|
||||||
import os
|
from pathlib import Path
|
||||||
|
|
||||||
from core.services.coreservices import ServiceManager
|
from core.services.coreservices import ServiceManager
|
||||||
|
|
||||||
_PATH = os.path.abspath(os.path.dirname(__file__))
|
_PATH: Path = Path(__file__).resolve().parent
|
||||||
|
|
||||||
|
|
||||||
def load():
|
def load():
|
||||||
|
|
|
@ -10,6 +10,7 @@ services.
|
||||||
import enum
|
import enum
|
||||||
import logging
|
import logging
|
||||||
import time
|
import time
|
||||||
|
from pathlib import Path
|
||||||
from typing import (
|
from typing import (
|
||||||
TYPE_CHECKING,
|
TYPE_CHECKING,
|
||||||
Dict,
|
Dict,
|
||||||
|
@ -33,6 +34,8 @@ from core.errors import (
|
||||||
)
|
)
|
||||||
from core.nodes.base import CoreNode
|
from core.nodes.base import CoreNode
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.emulator.session import Session
|
from core.emulator.session import Session
|
||||||
|
|
||||||
|
@ -160,7 +163,7 @@ class ServiceShim:
|
||||||
cls.setvalue(service, key, values[cls.keys.index(key)])
|
cls.setvalue(service, key, values[cls.keys.index(key)])
|
||||||
except IndexError:
|
except IndexError:
|
||||||
# old config does not need to have new keys
|
# old config does not need to have new keys
|
||||||
logging.exception("error indexing into key")
|
logger.exception("error indexing into key")
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def setvalue(cls, service: "CoreService", key: str, value: str) -> None:
|
def setvalue(cls, service: "CoreService", key: str, value: str) -> None:
|
||||||
|
@ -227,7 +230,7 @@ class ServiceManager:
|
||||||
:raises ValueError: when service cannot be loaded
|
:raises ValueError: when service cannot be loaded
|
||||||
"""
|
"""
|
||||||
name = service.name
|
name = service.name
|
||||||
logging.debug("loading service: class(%s) name(%s)", service.__name__, name)
|
logger.debug("loading service: class(%s) name(%s)", service.__name__, name)
|
||||||
|
|
||||||
# avoid duplicate services
|
# avoid duplicate services
|
||||||
if name in cls.services:
|
if name in cls.services:
|
||||||
|
@ -244,7 +247,7 @@ class ServiceManager:
|
||||||
try:
|
try:
|
||||||
service.on_load()
|
service.on_load()
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logging.exception("error during service(%s) on load", service.name)
|
logger.exception("error during service(%s) on load", service.name)
|
||||||
raise ValueError(e)
|
raise ValueError(e)
|
||||||
|
|
||||||
# make service available
|
# make service available
|
||||||
|
@ -264,7 +267,7 @@ class ServiceManager:
|
||||||
return service
|
return service
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def add_services(cls, path: str) -> List[str]:
|
def add_services(cls, path: Path) -> List[str]:
|
||||||
"""
|
"""
|
||||||
Method for retrieving all CoreServices from a given path.
|
Method for retrieving all CoreServices from a given path.
|
||||||
|
|
||||||
|
@ -276,12 +279,11 @@ class ServiceManager:
|
||||||
for service in services:
|
for service in services:
|
||||||
if not service.name:
|
if not service.name:
|
||||||
continue
|
continue
|
||||||
|
|
||||||
try:
|
try:
|
||||||
cls.add(service)
|
cls.add(service)
|
||||||
except (CoreError, ValueError) as e:
|
except (CoreError, ValueError) as e:
|
||||||
service_errors.append(service.name)
|
service_errors.append(service.name)
|
||||||
logging.debug("not loading service(%s): %s", service.name, e)
|
logger.debug("not loading service(%s): %s", service.name, e)
|
||||||
return service_errors
|
return service_errors
|
||||||
|
|
||||||
|
|
||||||
|
@ -329,14 +331,14 @@ class CoreServices:
|
||||||
:param node_type: node type to get default services for
|
:param node_type: node type to get default services for
|
||||||
:return: default services
|
:return: default services
|
||||||
"""
|
"""
|
||||||
logging.debug("getting default services for type: %s", node_type)
|
logger.debug("getting default services for type: %s", node_type)
|
||||||
results = []
|
results = []
|
||||||
defaults = self.default_services.get(node_type, [])
|
defaults = self.default_services.get(node_type, [])
|
||||||
for name in defaults:
|
for name in defaults:
|
||||||
logging.debug("checking for service with service manager: %s", name)
|
logger.debug("checking for service with service manager: %s", name)
|
||||||
service = ServiceManager.get(name)
|
service = ServiceManager.get(name)
|
||||||
if not service:
|
if not service:
|
||||||
logging.warning("default service %s is unknown", name)
|
logger.warning("default service %s is unknown", name)
|
||||||
else:
|
else:
|
||||||
results.append(service)
|
results.append(service)
|
||||||
return results
|
return results
|
||||||
|
@ -369,7 +371,7 @@ class CoreServices:
|
||||||
:param service_name: name of service to set
|
:param service_name: name of service to set
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
logging.debug("setting custom service(%s) for node: %s", service_name, node_id)
|
logger.debug("setting custom service(%s) for node: %s", service_name, node_id)
|
||||||
service = self.get_service(node_id, service_name)
|
service = self.get_service(node_id, service_name)
|
||||||
if not service:
|
if not service:
|
||||||
service_class = ServiceManager.get(service_name)
|
service_class = ServiceManager.get(service_name)
|
||||||
|
@ -391,15 +393,15 @@ class CoreServices:
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
if not services:
|
if not services:
|
||||||
logging.info(
|
logger.info(
|
||||||
"using default services for node(%s) type(%s)", node.name, node_type
|
"using default services for node(%s) type(%s)", node.name, node_type
|
||||||
)
|
)
|
||||||
services = self.default_services.get(node_type, [])
|
services = self.default_services.get(node_type, [])
|
||||||
logging.info("setting services for node(%s): %s", node.name, services)
|
logger.info("setting services for node(%s): %s", node.name, services)
|
||||||
for service_name in services:
|
for service_name in services:
|
||||||
service = self.get_service(node.id, service_name, default_service=True)
|
service = self.get_service(node.id, service_name, default_service=True)
|
||||||
if not service:
|
if not service:
|
||||||
logging.warning(
|
logger.warning(
|
||||||
"unknown service(%s) for node(%s)", service_name, node.name
|
"unknown service(%s) for node(%s)", service_name, node.name
|
||||||
)
|
)
|
||||||
continue
|
continue
|
||||||
|
@ -457,7 +459,7 @@ class CoreServices:
|
||||||
raise CoreServiceBootError(*exceptions)
|
raise CoreServiceBootError(*exceptions)
|
||||||
|
|
||||||
def _boot_service_path(self, node: CoreNode, boot_path: List["CoreServiceType"]):
|
def _boot_service_path(self, node: CoreNode, boot_path: List["CoreServiceType"]):
|
||||||
logging.info(
|
logger.info(
|
||||||
"booting node(%s) services: %s",
|
"booting node(%s) services: %s",
|
||||||
node.name,
|
node.name,
|
||||||
" -> ".join([x.name for x in boot_path]),
|
" -> ".join([x.name for x in boot_path]),
|
||||||
|
@ -467,7 +469,7 @@ class CoreServices:
|
||||||
try:
|
try:
|
||||||
self.boot_service(node, service)
|
self.boot_service(node, service)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logging.exception("exception booting service: %s", service.name)
|
logger.exception("exception booting service: %s", service.name)
|
||||||
raise CoreServiceBootError(e)
|
raise CoreServiceBootError(e)
|
||||||
|
|
||||||
def boot_service(self, node: CoreNode, service: "CoreServiceType") -> None:
|
def boot_service(self, node: CoreNode, service: "CoreServiceType") -> None:
|
||||||
|
@ -479,7 +481,7 @@ class CoreServices:
|
||||||
:param service: service to start
|
:param service: service to start
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
logging.info(
|
logger.info(
|
||||||
"starting node(%s) service(%s) validation(%s)",
|
"starting node(%s) service(%s) validation(%s)",
|
||||||
node.name,
|
node.name,
|
||||||
service.name,
|
service.name,
|
||||||
|
@ -488,10 +490,11 @@ class CoreServices:
|
||||||
|
|
||||||
# create service directories
|
# create service directories
|
||||||
for directory in service.dirs:
|
for directory in service.dirs:
|
||||||
|
dir_path = Path(directory)
|
||||||
try:
|
try:
|
||||||
node.privatedir(directory)
|
node.create_dir(dir_path)
|
||||||
except (CoreCommandError, ValueError) as e:
|
except (CoreCommandError, CoreError) as e:
|
||||||
logging.warning(
|
logger.warning(
|
||||||
"error mounting private dir '%s' for service '%s': %s",
|
"error mounting private dir '%s' for service '%s': %s",
|
||||||
directory,
|
directory,
|
||||||
service.name,
|
service.name,
|
||||||
|
@ -534,14 +537,14 @@ class CoreServices:
|
||||||
"node(%s) service(%s) failed validation" % (node.name, service.name)
|
"node(%s) service(%s) failed validation" % (node.name, service.name)
|
||||||
)
|
)
|
||||||
|
|
||||||
def copy_service_file(self, node: CoreNode, filename: str, cfg: str) -> bool:
|
def copy_service_file(self, node: CoreNode, file_path: Path, cfg: str) -> bool:
|
||||||
"""
|
"""
|
||||||
Given a configured service filename and config, determine if the
|
Given a configured service filename and config, determine if the
|
||||||
config references an existing file that should be copied.
|
config references an existing file that should be copied.
|
||||||
Returns True for local files, False for generated.
|
Returns True for local files, False for generated.
|
||||||
|
|
||||||
:param node: node to copy service for
|
:param node: node to copy service for
|
||||||
:param filename: file name for a configured service
|
:param file_path: file name for a configured service
|
||||||
:param cfg: configuration string
|
:param cfg: configuration string
|
||||||
:return: True if successful, False otherwise
|
:return: True if successful, False otherwise
|
||||||
"""
|
"""
|
||||||
|
@ -550,7 +553,7 @@ class CoreServices:
|
||||||
src = src.split("\n")[0]
|
src = src.split("\n")[0]
|
||||||
src = utils.expand_corepath(src, node.session, node)
|
src = utils.expand_corepath(src, node.session, node)
|
||||||
# TODO: glob here
|
# TODO: glob here
|
||||||
node.nodefilecopy(filename, src, mode=0o644)
|
node.copy_file(src, file_path, mode=0o644)
|
||||||
return True
|
return True
|
||||||
return False
|
return False
|
||||||
|
|
||||||
|
@ -562,21 +565,21 @@ class CoreServices:
|
||||||
:param service: service to validate
|
:param service: service to validate
|
||||||
:return: service validation status
|
:return: service validation status
|
||||||
"""
|
"""
|
||||||
logging.debug("validating node(%s) service(%s)", node.name, service.name)
|
logger.debug("validating node(%s) service(%s)", node.name, service.name)
|
||||||
cmds = service.validate
|
cmds = service.validate
|
||||||
if not service.custom:
|
if not service.custom:
|
||||||
cmds = service.get_validate(node)
|
cmds = service.get_validate(node)
|
||||||
|
|
||||||
status = 0
|
status = 0
|
||||||
for cmd in cmds:
|
for cmd in cmds:
|
||||||
logging.debug("validating service(%s) using: %s", service.name, cmd)
|
logger.debug("validating service(%s) using: %s", service.name, cmd)
|
||||||
try:
|
try:
|
||||||
node.cmd(cmd)
|
node.cmd(cmd)
|
||||||
except CoreCommandError as e:
|
except CoreCommandError as e:
|
||||||
logging.debug(
|
logger.debug(
|
||||||
"node(%s) service(%s) validate failed", node.name, service.name
|
"node(%s) service(%s) validate failed", node.name, service.name
|
||||||
)
|
)
|
||||||
logging.debug("cmd(%s): %s", e.cmd, e.output)
|
logger.debug("cmd(%s): %s", e.cmd, e.output)
|
||||||
status = -1
|
status = -1
|
||||||
break
|
break
|
||||||
|
|
||||||
|
@ -611,7 +614,7 @@ class CoreServices:
|
||||||
f"error stopping service {service.name}: {e.stderr}",
|
f"error stopping service {service.name}: {e.stderr}",
|
||||||
node.id,
|
node.id,
|
||||||
)
|
)
|
||||||
logging.exception("error running stop command %s", args)
|
logger.exception("error running stop command %s", args)
|
||||||
status = -1
|
status = -1
|
||||||
return status
|
return status
|
||||||
|
|
||||||
|
@ -679,13 +682,13 @@ class CoreServices:
|
||||||
# retrieve custom service
|
# retrieve custom service
|
||||||
service = self.get_service(node_id, service_name)
|
service = self.get_service(node_id, service_name)
|
||||||
if service is None:
|
if service is None:
|
||||||
logging.warning("received file name for unknown service: %s", service_name)
|
logger.warning("received file name for unknown service: %s", service_name)
|
||||||
return
|
return
|
||||||
|
|
||||||
# validate file being set is valid
|
# validate file being set is valid
|
||||||
config_files = service.configs
|
config_files = service.configs
|
||||||
if file_name not in config_files:
|
if file_name not in config_files:
|
||||||
logging.warning(
|
logger.warning(
|
||||||
"received unknown file(%s) for service(%s)", file_name, service_name
|
"received unknown file(%s) for service(%s)", file_name, service_name
|
||||||
)
|
)
|
||||||
return
|
return
|
||||||
|
@ -713,7 +716,7 @@ class CoreServices:
|
||||||
try:
|
try:
|
||||||
node.cmd(cmd, wait)
|
node.cmd(cmd, wait)
|
||||||
except CoreCommandError:
|
except CoreCommandError:
|
||||||
logging.exception("error starting command")
|
logger.exception("error starting command")
|
||||||
status = -1
|
status = -1
|
||||||
return status
|
return status
|
||||||
|
|
||||||
|
@ -729,27 +732,25 @@ class CoreServices:
|
||||||
config_files = service.configs
|
config_files = service.configs
|
||||||
if not service.custom:
|
if not service.custom:
|
||||||
config_files = service.get_configs(node)
|
config_files = service.get_configs(node)
|
||||||
|
|
||||||
for file_name in config_files:
|
for file_name in config_files:
|
||||||
logging.debug(
|
file_path = Path(file_name)
|
||||||
|
logger.debug(
|
||||||
"generating service config custom(%s): %s", service.custom, file_name
|
"generating service config custom(%s): %s", service.custom, file_name
|
||||||
)
|
)
|
||||||
if service.custom:
|
if service.custom:
|
||||||
cfg = service.config_data.get(file_name)
|
cfg = service.config_data.get(file_name)
|
||||||
if cfg is None:
|
if cfg is None:
|
||||||
cfg = service.generate_config(node, file_name)
|
cfg = service.generate_config(node, file_name)
|
||||||
|
|
||||||
# cfg may have a file:/// url for copying from a file
|
# cfg may have a file:/// url for copying from a file
|
||||||
try:
|
try:
|
||||||
if self.copy_service_file(node, file_name, cfg):
|
if self.copy_service_file(node, file_path, cfg):
|
||||||
continue
|
continue
|
||||||
except IOError:
|
except IOError:
|
||||||
logging.exception("error copying service file: %s", file_name)
|
logger.exception("error copying service file: %s", file_name)
|
||||||
continue
|
continue
|
||||||
else:
|
else:
|
||||||
cfg = service.generate_config(node, file_name)
|
cfg = service.generate_config(node, file_name)
|
||||||
|
node.create_file(file_path, cfg)
|
||||||
node.nodefile(file_name, cfg)
|
|
||||||
|
|
||||||
def service_reconfigure(self, node: CoreNode, service: "CoreService") -> None:
|
def service_reconfigure(self, node: CoreNode, service: "CoreService") -> None:
|
||||||
"""
|
"""
|
||||||
|
@ -762,17 +763,15 @@ class CoreServices:
|
||||||
config_files = service.configs
|
config_files = service.configs
|
||||||
if not service.custom:
|
if not service.custom:
|
||||||
config_files = service.get_configs(node)
|
config_files = service.get_configs(node)
|
||||||
|
|
||||||
for file_name in config_files:
|
for file_name in config_files:
|
||||||
|
file_path = Path(file_name)
|
||||||
if file_name[:7] == "file:///":
|
if file_name[:7] == "file:///":
|
||||||
# TODO: implement this
|
# TODO: implement this
|
||||||
raise NotImplementedError
|
raise NotImplementedError
|
||||||
|
|
||||||
cfg = service.config_data.get(file_name)
|
cfg = service.config_data.get(file_name)
|
||||||
if cfg is None:
|
if cfg is None:
|
||||||
cfg = service.generate_config(node, file_name)
|
cfg = service.generate_config(node, file_name)
|
||||||
|
node.create_file(file_path, cfg)
|
||||||
node.nodefile(file_name, cfg)
|
|
||||||
|
|
||||||
|
|
||||||
class CoreService:
|
class CoreService:
|
||||||
|
|
|
@ -11,6 +11,8 @@ from core.nodes.base import CoreNode
|
||||||
from core.nodes.interface import CoreInterface
|
from core.nodes.interface import CoreInterface
|
||||||
from core.services.coreservices import CoreService
|
from core.services.coreservices import CoreService
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
class VPNClient(CoreService):
|
class VPNClient(CoreService):
|
||||||
name: str = "VPNClient"
|
name: str = "VPNClient"
|
||||||
|
@ -33,7 +35,7 @@ class VPNClient(CoreService):
|
||||||
with open(fname, "r") as f:
|
with open(fname, "r") as f:
|
||||||
cfg += f.read()
|
cfg += f.read()
|
||||||
except IOError:
|
except IOError:
|
||||||
logging.exception(
|
logger.exception(
|
||||||
"error opening VPN client configuration template (%s)", fname
|
"error opening VPN client configuration template (%s)", fname
|
||||||
)
|
)
|
||||||
return cfg
|
return cfg
|
||||||
|
@ -61,7 +63,7 @@ class VPNServer(CoreService):
|
||||||
with open(fname, "r") as f:
|
with open(fname, "r") as f:
|
||||||
cfg += f.read()
|
cfg += f.read()
|
||||||
except IOError:
|
except IOError:
|
||||||
logging.exception(
|
logger.exception(
|
||||||
"Error opening VPN server configuration template (%s)", fname
|
"Error opening VPN server configuration template (%s)", fname
|
||||||
)
|
)
|
||||||
return cfg
|
return cfg
|
||||||
|
@ -89,7 +91,7 @@ class IPsec(CoreService):
|
||||||
with open(fname, "r") as f:
|
with open(fname, "r") as f:
|
||||||
cfg += f.read()
|
cfg += f.read()
|
||||||
except IOError:
|
except IOError:
|
||||||
logging.exception("Error opening IPsec configuration template (%s)", fname)
|
logger.exception("Error opening IPsec configuration template (%s)", fname)
|
||||||
return cfg
|
return cfg
|
||||||
|
|
||||||
|
|
||||||
|
@ -112,7 +114,7 @@ class Firewall(CoreService):
|
||||||
with open(fname, "r") as f:
|
with open(fname, "r") as f:
|
||||||
cfg += f.read()
|
cfg += f.read()
|
||||||
except IOError:
|
except IOError:
|
||||||
logging.exception(
|
logger.exception(
|
||||||
"Error opening Firewall configuration template (%s)", fname
|
"Error opening Firewall configuration template (%s)", fname
|
||||||
)
|
)
|
||||||
return cfg
|
return cfg
|
||||||
|
|
|
@ -236,7 +236,7 @@ max-lease-time 7200;
|
||||||
ddns-update-style none;
|
ddns-update-style none;
|
||||||
"""
|
"""
|
||||||
for iface in node.get_ifaces(control=False):
|
for iface in node.get_ifaces(control=False):
|
||||||
cfg += "\n".join(map(cls.subnetentry, iface.ips()))
|
cfg += "\n".join(map(cls.subnetentry, iface.ip4s))
|
||||||
cfg += "\n"
|
cfg += "\n"
|
||||||
return cfg
|
return cfg
|
||||||
|
|
||||||
|
@ -246,15 +246,13 @@ ddns-update-style none;
|
||||||
Generate a subnet declaration block given an IPv4 prefix string
|
Generate a subnet declaration block given an IPv4 prefix string
|
||||||
for inclusion in the dhcpd3 config file.
|
for inclusion in the dhcpd3 config file.
|
||||||
"""
|
"""
|
||||||
address = str(ip.ip)
|
if ip.size == 1:
|
||||||
if netaddr.valid_ipv6(address):
|
|
||||||
return ""
|
return ""
|
||||||
else:
|
# divide the address space in half
|
||||||
# divide the address space in half
|
index = (ip.size - 2) / 2
|
||||||
index = (ip.size - 2) / 2
|
rangelow = ip[index]
|
||||||
rangelow = ip[index]
|
rangehigh = ip[-2]
|
||||||
rangehigh = ip[-2]
|
return """
|
||||||
return """
|
|
||||||
subnet %s netmask %s {
|
subnet %s netmask %s {
|
||||||
pool {
|
pool {
|
||||||
range %s %s;
|
range %s %s;
|
||||||
|
@ -263,12 +261,12 @@ subnet %s netmask %s {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
""" % (
|
""" % (
|
||||||
ip.ip,
|
ip.cidr.ip,
|
||||||
ip.netmask,
|
ip.netmask,
|
||||||
rangelow,
|
rangelow,
|
||||||
rangehigh,
|
rangehigh,
|
||||||
address,
|
ip.ip,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
class DhcpClientService(UtilService):
|
class DhcpClientService(UtilService):
|
||||||
|
|
|
@ -36,6 +36,8 @@ import netaddr
|
||||||
|
|
||||||
from core.errors import CoreCommandError, CoreError
|
from core.errors import CoreCommandError, CoreError
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.emulator.session import Session
|
from core.emulator.session import Session
|
||||||
from core.nodes.base import CoreNode
|
from core.nodes.base import CoreNode
|
||||||
|
@ -46,11 +48,10 @@ IFACE_CONFIG_FACTOR: int = 1000
|
||||||
|
|
||||||
|
|
||||||
def execute_file(
|
def execute_file(
|
||||||
path: str, exec_globals: Dict[str, str] = None, exec_locals: Dict[str, str] = None
|
path: Path, exec_globals: Dict[str, str] = None, exec_locals: Dict[str, str] = None
|
||||||
) -> None:
|
) -> None:
|
||||||
"""
|
"""
|
||||||
Provides an alternative way to run execfile to be compatible for
|
Provides a way to execute a file.
|
||||||
both python2/3.
|
|
||||||
|
|
||||||
:param path: path of file to execute
|
:param path: path of file to execute
|
||||||
:param exec_globals: globals values to pass to execution
|
:param exec_globals: globals values to pass to execution
|
||||||
|
@ -59,10 +60,10 @@ def execute_file(
|
||||||
"""
|
"""
|
||||||
if exec_globals is None:
|
if exec_globals is None:
|
||||||
exec_globals = {}
|
exec_globals = {}
|
||||||
exec_globals.update({"__file__": path, "__name__": "__main__"})
|
exec_globals.update({"__file__": str(path), "__name__": "__main__"})
|
||||||
with open(path, "rb") as f:
|
with path.open("rb") as f:
|
||||||
data = compile(f.read(), path, "exec")
|
data = compile(f.read(), path, "exec")
|
||||||
exec(data, exec_globals, exec_locals)
|
exec(data, exec_globals, exec_locals)
|
||||||
|
|
||||||
|
|
||||||
def hashkey(value: Union[str, int]) -> int:
|
def hashkey(value: Union[str, int]) -> int:
|
||||||
|
@ -92,24 +93,19 @@ def _detach_init() -> None:
|
||||||
os.setsid()
|
os.setsid()
|
||||||
|
|
||||||
|
|
||||||
def _valid_module(path: str, file_name: str) -> bool:
|
def _valid_module(path: Path) -> bool:
|
||||||
"""
|
"""
|
||||||
Check if file is a valid python module.
|
Check if file is a valid python module.
|
||||||
|
|
||||||
:param path: path to file
|
:param path: path to file
|
||||||
:param file_name: file name to check
|
|
||||||
:return: True if a valid python module file, False otherwise
|
:return: True if a valid python module file, False otherwise
|
||||||
"""
|
"""
|
||||||
file_path = os.path.join(path, file_name)
|
if not path.is_file():
|
||||||
if not os.path.isfile(file_path):
|
|
||||||
return False
|
return False
|
||||||
|
if path.name.startswith("_"):
|
||||||
if file_name.startswith("_"):
|
|
||||||
return False
|
return False
|
||||||
|
if not path.suffix == ".py":
|
||||||
if not file_name.endswith(".py"):
|
|
||||||
return False
|
return False
|
||||||
|
|
||||||
return True
|
return True
|
||||||
|
|
||||||
|
|
||||||
|
@ -124,13 +120,10 @@ def _is_class(module: Any, member: Type, clazz: Type) -> bool:
|
||||||
"""
|
"""
|
||||||
if not inspect.isclass(member):
|
if not inspect.isclass(member):
|
||||||
return False
|
return False
|
||||||
|
|
||||||
if not issubclass(member, clazz):
|
if not issubclass(member, clazz):
|
||||||
return False
|
return False
|
||||||
|
|
||||||
if member.__module__ != module.__name__:
|
if member.__module__ != module.__name__:
|
||||||
return False
|
return False
|
||||||
|
|
||||||
return True
|
return True
|
||||||
|
|
||||||
|
|
||||||
|
@ -196,7 +189,7 @@ def mute_detach(args: str, **kwargs: Dict[str, Any]) -> int:
|
||||||
def cmd(
|
def cmd(
|
||||||
args: str,
|
args: str,
|
||||||
env: Dict[str, str] = None,
|
env: Dict[str, str] = None,
|
||||||
cwd: str = None,
|
cwd: Path = None,
|
||||||
wait: bool = True,
|
wait: bool = True,
|
||||||
shell: bool = False,
|
shell: bool = False,
|
||||||
) -> str:
|
) -> str:
|
||||||
|
@ -213,7 +206,7 @@ def cmd(
|
||||||
:raises CoreCommandError: when there is a non-zero exit status or the file to
|
:raises CoreCommandError: when there is a non-zero exit status or the file to
|
||||||
execute is not found
|
execute is not found
|
||||||
"""
|
"""
|
||||||
logging.debug("command cwd(%s) wait(%s): %s", cwd, wait, args)
|
logger.debug("command cwd(%s) wait(%s): %s", cwd, wait, args)
|
||||||
if shell is False:
|
if shell is False:
|
||||||
args = shlex.split(args)
|
args = shlex.split(args)
|
||||||
try:
|
try:
|
||||||
|
@ -230,7 +223,7 @@ def cmd(
|
||||||
else:
|
else:
|
||||||
return ""
|
return ""
|
||||||
except OSError as e:
|
except OSError as e:
|
||||||
logging.error("cmd error: %s", e.strerror)
|
logger.error("cmd error: %s", e.strerror)
|
||||||
raise CoreCommandError(1, args, "", e.strerror)
|
raise CoreCommandError(1, args, "", e.strerror)
|
||||||
|
|
||||||
|
|
||||||
|
@ -282,7 +275,7 @@ def file_demunge(pathname: str, header: str) -> None:
|
||||||
|
|
||||||
def expand_corepath(
|
def expand_corepath(
|
||||||
pathname: str, session: "Session" = None, node: "CoreNode" = None
|
pathname: str, session: "Session" = None, node: "CoreNode" = None
|
||||||
) -> str:
|
) -> Path:
|
||||||
"""
|
"""
|
||||||
Expand a file path given session information.
|
Expand a file path given session information.
|
||||||
|
|
||||||
|
@ -294,14 +287,12 @@ def expand_corepath(
|
||||||
if session is not None:
|
if session is not None:
|
||||||
pathname = pathname.replace("~", f"/home/{session.user}")
|
pathname = pathname.replace("~", f"/home/{session.user}")
|
||||||
pathname = pathname.replace("%SESSION%", str(session.id))
|
pathname = pathname.replace("%SESSION%", str(session.id))
|
||||||
pathname = pathname.replace("%SESSION_DIR%", session.session_dir)
|
pathname = pathname.replace("%SESSION_DIR%", str(session.directory))
|
||||||
pathname = pathname.replace("%SESSION_USER%", session.user)
|
pathname = pathname.replace("%SESSION_USER%", session.user)
|
||||||
|
|
||||||
if node is not None:
|
if node is not None:
|
||||||
pathname = pathname.replace("%NODE%", str(node.id))
|
pathname = pathname.replace("%NODE%", str(node.id))
|
||||||
pathname = pathname.replace("%NODENAME%", node.name)
|
pathname = pathname.replace("%NODENAME%", node.name)
|
||||||
|
return Path(pathname)
|
||||||
return pathname
|
|
||||||
|
|
||||||
|
|
||||||
def sysctl_devname(devname: str) -> Optional[str]:
|
def sysctl_devname(devname: str) -> Optional[str]:
|
||||||
|
@ -334,10 +325,25 @@ def load_config(file_path: Path, d: Dict[str, str]) -> None:
|
||||||
key, value = line.split("=", 1)
|
key, value = line.split("=", 1)
|
||||||
d[key] = value.strip()
|
d[key] = value.strip()
|
||||||
except ValueError:
|
except ValueError:
|
||||||
logging.exception("error reading file to dict: %s", file_path)
|
logger.exception("error reading file to dict: %s", file_path)
|
||||||
|
|
||||||
|
|
||||||
def load_classes(path: str, clazz: Generic[T]) -> T:
|
def load_module(import_statement: str, clazz: Generic[T]) -> List[T]:
|
||||||
|
classes = []
|
||||||
|
try:
|
||||||
|
module = importlib.import_module(import_statement)
|
||||||
|
members = inspect.getmembers(module, lambda x: _is_class(module, x, clazz))
|
||||||
|
for member in members:
|
||||||
|
valid_class = member[1]
|
||||||
|
classes.append(valid_class)
|
||||||
|
except Exception:
|
||||||
|
logger.exception(
|
||||||
|
"unexpected error during import, skipping: %s", import_statement
|
||||||
|
)
|
||||||
|
return classes
|
||||||
|
|
||||||
|
|
||||||
|
def load_classes(path: Path, clazz: Generic[T]) -> List[T]:
|
||||||
"""
|
"""
|
||||||
Dynamically load classes for use within CORE.
|
Dynamically load classes for use within CORE.
|
||||||
|
|
||||||
|
@ -346,50 +352,36 @@ def load_classes(path: str, clazz: Generic[T]) -> T:
|
||||||
:return: list of classes loaded
|
:return: list of classes loaded
|
||||||
"""
|
"""
|
||||||
# validate path exists
|
# validate path exists
|
||||||
logging.debug("attempting to load modules from path: %s", path)
|
logger.debug("attempting to load modules from path: %s", path)
|
||||||
if not os.path.isdir(path):
|
if not path.is_dir():
|
||||||
logging.warning("invalid custom module directory specified" ": %s", path)
|
logger.warning("invalid custom module directory specified" ": %s", path)
|
||||||
# check if path is in sys.path
|
# check if path is in sys.path
|
||||||
parent_path = os.path.dirname(path)
|
parent = str(path.parent)
|
||||||
if parent_path not in sys.path:
|
if parent not in sys.path:
|
||||||
logging.debug("adding parent path to allow imports: %s", parent_path)
|
logger.debug("adding parent path to allow imports: %s", parent)
|
||||||
sys.path.append(parent_path)
|
sys.path.append(parent)
|
||||||
|
|
||||||
# retrieve potential service modules, and filter out invalid modules
|
|
||||||
base_module = os.path.basename(path)
|
|
||||||
module_names = os.listdir(path)
|
|
||||||
module_names = filter(lambda x: _valid_module(path, x), module_names)
|
|
||||||
module_names = map(lambda x: x[:-3], module_names)
|
|
||||||
|
|
||||||
# import and add all service modules in the path
|
# import and add all service modules in the path
|
||||||
classes = []
|
classes = []
|
||||||
for module_name in module_names:
|
for p in path.iterdir():
|
||||||
import_statement = f"{base_module}.{module_name}"
|
if not _valid_module(p):
|
||||||
logging.debug("importing custom module: %s", import_statement)
|
continue
|
||||||
try:
|
import_statement = f"{path.name}.{p.stem}"
|
||||||
module = importlib.import_module(import_statement)
|
logger.debug("importing custom module: %s", import_statement)
|
||||||
members = inspect.getmembers(module, lambda x: _is_class(module, x, clazz))
|
loaded = load_module(import_statement, clazz)
|
||||||
for member in members:
|
classes.extend(loaded)
|
||||||
valid_class = member[1]
|
|
||||||
classes.append(valid_class)
|
|
||||||
except Exception:
|
|
||||||
logging.exception(
|
|
||||||
"unexpected error during import, skipping: %s", import_statement
|
|
||||||
)
|
|
||||||
|
|
||||||
return classes
|
return classes
|
||||||
|
|
||||||
|
|
||||||
def load_logging_config(config_path: str) -> None:
|
def load_logging_config(config_path: Path) -> None:
|
||||||
"""
|
"""
|
||||||
Load CORE logging configuration file.
|
Load CORE logging configuration file.
|
||||||
|
|
||||||
:param config_path: path to logging config file
|
:param config_path: path to logging config file
|
||||||
:return: nothing
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
with open(config_path, "r") as log_config_file:
|
with config_path.open("r") as f:
|
||||||
log_config = json.load(log_config_file)
|
log_config = json.load(f)
|
||||||
logging.config.dictConfig(log_config)
|
logging.config.dictConfig(log_config)
|
||||||
|
|
||||||
|
|
||||||
def threadpool(
|
def threadpool(
|
||||||
|
@ -415,7 +407,7 @@ def threadpool(
|
||||||
result = future.result()
|
result = future.result()
|
||||||
results.append(result)
|
results.append(result)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logging.exception("thread pool exception")
|
logger.exception("thread pool exception")
|
||||||
exceptions.append(e)
|
exceptions.append(e)
|
||||||
return results, exceptions
|
return results, exceptions
|
||||||
|
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
import logging
|
import logging
|
||||||
|
from pathlib import Path
|
||||||
from typing import TYPE_CHECKING, Any, Dict, Generic, List, Optional, Type, TypeVar
|
from typing import TYPE_CHECKING, Any, Dict, Generic, List, Optional, Type, TypeVar
|
||||||
|
|
||||||
from lxml import etree
|
from lxml import etree
|
||||||
|
@ -16,6 +17,8 @@ from core.nodes.lxd import LxcNode
|
||||||
from core.nodes.network import CtrlNet, GreTapBridge, WlanNode
|
from core.nodes.network import CtrlNet, GreTapBridge, WlanNode
|
||||||
from core.services.coreservices import CoreService
|
from core.services.coreservices import CoreService
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.emane.emanemodel import EmaneModel
|
from core.emane.emanemodel import EmaneModel
|
||||||
from core.emulator.session import Session
|
from core.emulator.session import Session
|
||||||
|
@ -25,7 +28,7 @@ T = TypeVar("T")
|
||||||
|
|
||||||
|
|
||||||
def write_xml_file(
|
def write_xml_file(
|
||||||
xml_element: etree.Element, file_path: str, doctype: str = None
|
xml_element: etree.Element, file_path: Path, doctype: str = None
|
||||||
) -> None:
|
) -> None:
|
||||||
xml_data = etree.tostring(
|
xml_data = etree.tostring(
|
||||||
xml_element,
|
xml_element,
|
||||||
|
@ -34,8 +37,8 @@ def write_xml_file(
|
||||||
encoding="UTF-8",
|
encoding="UTF-8",
|
||||||
doctype=doctype,
|
doctype=doctype,
|
||||||
)
|
)
|
||||||
with open(file_path, "wb") as xml_file:
|
with file_path.open("wb") as f:
|
||||||
xml_file.write(xml_data)
|
f.write(xml_data)
|
||||||
|
|
||||||
|
|
||||||
def get_type(element: etree.Element, name: str, _type: Generic[T]) -> Optional[T]:
|
def get_type(element: etree.Element, name: str, _type: Generic[T]) -> Optional[T]:
|
||||||
|
@ -77,20 +80,6 @@ def create_iface_data(iface_element: etree.Element) -> InterfaceData:
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
def create_emane_config(session: "Session") -> etree.Element:
|
|
||||||
emane_configuration = etree.Element("emane_global_configuration")
|
|
||||||
config = session.emane.get_configs()
|
|
||||||
emulator_element = etree.SubElement(emane_configuration, "emulator")
|
|
||||||
for emulator_config in session.emane.emane_config.emulator_config:
|
|
||||||
value = config[emulator_config.id]
|
|
||||||
add_configuration(emulator_element, emulator_config.id, value)
|
|
||||||
core_element = etree.SubElement(emane_configuration, "core")
|
|
||||||
for core_config in session.emane.emane_config.core_config:
|
|
||||||
value = config[core_config.id]
|
|
||||||
add_configuration(core_element, core_config.id, value)
|
|
||||||
return emane_configuration
|
|
||||||
|
|
||||||
|
|
||||||
def create_emane_model_config(
|
def create_emane_model_config(
|
||||||
node_id: int,
|
node_id: int,
|
||||||
model: "EmaneModelType",
|
model: "EmaneModelType",
|
||||||
|
@ -101,22 +90,22 @@ def create_emane_model_config(
|
||||||
add_attribute(emane_element, "node", node_id)
|
add_attribute(emane_element, "node", node_id)
|
||||||
add_attribute(emane_element, "iface", iface_id)
|
add_attribute(emane_element, "iface", iface_id)
|
||||||
add_attribute(emane_element, "model", model.name)
|
add_attribute(emane_element, "model", model.name)
|
||||||
|
platform_element = etree.SubElement(emane_element, "platform")
|
||||||
|
for platform_config in model.platform_config:
|
||||||
|
value = config[platform_config.id]
|
||||||
|
add_configuration(platform_element, platform_config.id, value)
|
||||||
mac_element = etree.SubElement(emane_element, "mac")
|
mac_element = etree.SubElement(emane_element, "mac")
|
||||||
for mac_config in model.mac_config:
|
for mac_config in model.mac_config:
|
||||||
value = config[mac_config.id]
|
value = config[mac_config.id]
|
||||||
add_configuration(mac_element, mac_config.id, value)
|
add_configuration(mac_element, mac_config.id, value)
|
||||||
|
|
||||||
phy_element = etree.SubElement(emane_element, "phy")
|
phy_element = etree.SubElement(emane_element, "phy")
|
||||||
for phy_config in model.phy_config:
|
for phy_config in model.phy_config:
|
||||||
value = config[phy_config.id]
|
value = config[phy_config.id]
|
||||||
add_configuration(phy_element, phy_config.id, value)
|
add_configuration(phy_element, phy_config.id, value)
|
||||||
|
|
||||||
external_element = etree.SubElement(emane_element, "external")
|
external_element = etree.SubElement(emane_element, "external")
|
||||||
for external_config in model.external_config:
|
for external_config in model.external_config:
|
||||||
value = config[external_config.id]
|
value = config[external_config.id]
|
||||||
add_configuration(external_element, external_config.id, value)
|
add_configuration(external_element, external_config.id, value)
|
||||||
|
|
||||||
return emane_element
|
return emane_element
|
||||||
|
|
||||||
|
|
||||||
|
@ -293,13 +282,12 @@ class CoreXmlWriter:
|
||||||
self.write_session_metadata()
|
self.write_session_metadata()
|
||||||
self.write_default_services()
|
self.write_default_services()
|
||||||
|
|
||||||
def write(self, file_name: str) -> None:
|
def write(self, path: Path) -> None:
|
||||||
self.scenario.set("name", file_name)
|
self.scenario.set("name", str(path))
|
||||||
|
|
||||||
# write out generated xml
|
# write out generated xml
|
||||||
xml_tree = etree.ElementTree(self.scenario)
|
xml_tree = etree.ElementTree(self.scenario)
|
||||||
xml_tree.write(
|
xml_tree.write(
|
||||||
file_name, xml_declaration=True, pretty_print=True, encoding="UTF-8"
|
str(path), xml_declaration=True, pretty_print=True, encoding="UTF-8"
|
||||||
)
|
)
|
||||||
|
|
||||||
def write_session_origin(self) -> None:
|
def write_session_origin(self) -> None:
|
||||||
|
@ -374,22 +362,16 @@ class CoreXmlWriter:
|
||||||
self.scenario.append(metadata_elements)
|
self.scenario.append(metadata_elements)
|
||||||
|
|
||||||
def write_emane_configs(self) -> None:
|
def write_emane_configs(self) -> None:
|
||||||
emane_global_configuration = create_emane_config(self.session)
|
|
||||||
self.scenario.append(emane_global_configuration)
|
|
||||||
emane_configurations = etree.Element("emane_configurations")
|
emane_configurations = etree.Element("emane_configurations")
|
||||||
for node_id in self.session.emane.nodes():
|
for node_id, model_configs in self.session.emane.node_configs.items():
|
||||||
all_configs = self.session.emane.get_all_configs(node_id)
|
|
||||||
if not all_configs:
|
|
||||||
continue
|
|
||||||
node_id, iface_id = utils.parse_iface_config_id(node_id)
|
node_id, iface_id = utils.parse_iface_config_id(node_id)
|
||||||
for model_name in all_configs:
|
for model_name, config in model_configs.items():
|
||||||
config = all_configs[model_name]
|
logger.debug(
|
||||||
logging.debug(
|
|
||||||
"writing emane config node(%s) model(%s)", node_id, model_name
|
"writing emane config node(%s) model(%s)", node_id, model_name
|
||||||
)
|
)
|
||||||
model = self.session.emane.models[model_name]
|
model_class = self.session.emane.get_model(model_name)
|
||||||
emane_configuration = create_emane_model_config(
|
emane_configuration = create_emane_model_config(
|
||||||
node_id, model, config, iface_id
|
node_id, model_class, config, iface_id
|
||||||
)
|
)
|
||||||
emane_configurations.append(emane_configuration)
|
emane_configurations.append(emane_configuration)
|
||||||
if emane_configurations.getchildren():
|
if emane_configurations.getchildren():
|
||||||
|
@ -404,7 +386,7 @@ class CoreXmlWriter:
|
||||||
|
|
||||||
for model_name in all_configs:
|
for model_name in all_configs:
|
||||||
config = all_configs[model_name]
|
config = all_configs[model_name]
|
||||||
logging.debug(
|
logger.debug(
|
||||||
"writing mobility config node(%s) model(%s)", node_id, model_name
|
"writing mobility config node(%s) model(%s)", node_id, model_name
|
||||||
)
|
)
|
||||||
mobility_configuration = etree.SubElement(
|
mobility_configuration = etree.SubElement(
|
||||||
|
@ -580,8 +562,8 @@ class CoreXmlReader:
|
||||||
self.session: "Session" = session
|
self.session: "Session" = session
|
||||||
self.scenario: Optional[etree.ElementTree] = None
|
self.scenario: Optional[etree.ElementTree] = None
|
||||||
|
|
||||||
def read(self, file_name: str) -> None:
|
def read(self, file_path: Path) -> None:
|
||||||
xml_tree = etree.parse(file_name)
|
xml_tree = etree.parse(str(file_path))
|
||||||
self.scenario = xml_tree.getroot()
|
self.scenario = xml_tree.getroot()
|
||||||
|
|
||||||
# read xml session content
|
# read xml session content
|
||||||
|
@ -593,7 +575,6 @@ class CoreXmlReader:
|
||||||
self.read_session_origin()
|
self.read_session_origin()
|
||||||
self.read_service_configs()
|
self.read_service_configs()
|
||||||
self.read_mobility_configs()
|
self.read_mobility_configs()
|
||||||
self.read_emane_global_config()
|
|
||||||
self.read_nodes()
|
self.read_nodes()
|
||||||
self.read_links()
|
self.read_links()
|
||||||
self.read_emane_configs()
|
self.read_emane_configs()
|
||||||
|
@ -609,7 +590,7 @@ class CoreXmlReader:
|
||||||
services = []
|
services = []
|
||||||
for service in node.iterchildren():
|
for service in node.iterchildren():
|
||||||
services.append(service.get("name"))
|
services.append(service.get("name"))
|
||||||
logging.info(
|
logger.info(
|
||||||
"reading default services for nodes(%s): %s", node_type, services
|
"reading default services for nodes(%s): %s", node_type, services
|
||||||
)
|
)
|
||||||
self.session.services.default_services[node_type] = services
|
self.session.services.default_services[node_type] = services
|
||||||
|
@ -624,7 +605,7 @@ class CoreXmlReader:
|
||||||
name = data.get("name")
|
name = data.get("name")
|
||||||
value = data.get("value")
|
value = data.get("value")
|
||||||
configs[name] = value
|
configs[name] = value
|
||||||
logging.info("reading session metadata: %s", configs)
|
logger.info("reading session metadata: %s", configs)
|
||||||
self.session.metadata = configs
|
self.session.metadata = configs
|
||||||
|
|
||||||
def read_session_options(self) -> None:
|
def read_session_options(self) -> None:
|
||||||
|
@ -636,7 +617,7 @@ class CoreXmlReader:
|
||||||
name = configuration.get("name")
|
name = configuration.get("name")
|
||||||
value = configuration.get("value")
|
value = configuration.get("value")
|
||||||
xml_config[name] = value
|
xml_config[name] = value
|
||||||
logging.info("reading session options: %s", xml_config)
|
logger.info("reading session options: %s", xml_config)
|
||||||
config = self.session.options.get_configs()
|
config = self.session.options.get_configs()
|
||||||
config.update(xml_config)
|
config.update(xml_config)
|
||||||
|
|
||||||
|
@ -650,7 +631,7 @@ class CoreXmlReader:
|
||||||
state = get_int(hook, "state")
|
state = get_int(hook, "state")
|
||||||
state = EventTypes(state)
|
state = EventTypes(state)
|
||||||
data = hook.text
|
data = hook.text
|
||||||
logging.info("reading hook: state(%s) name(%s)", state, name)
|
logger.info("reading hook: state(%s) name(%s)", state, name)
|
||||||
self.session.add_hook(state, name, data)
|
self.session.add_hook(state, name, data)
|
||||||
|
|
||||||
def read_servers(self) -> None:
|
def read_servers(self) -> None:
|
||||||
|
@ -660,7 +641,7 @@ class CoreXmlReader:
|
||||||
for server in servers.iterchildren():
|
for server in servers.iterchildren():
|
||||||
name = server.get("name")
|
name = server.get("name")
|
||||||
address = server.get("address")
|
address = server.get("address")
|
||||||
logging.info("reading server: name(%s) address(%s)", name, address)
|
logger.info("reading server: name(%s) address(%s)", name, address)
|
||||||
self.session.distributed.add_server(name, address)
|
self.session.distributed.add_server(name, address)
|
||||||
|
|
||||||
def read_session_origin(self) -> None:
|
def read_session_origin(self) -> None:
|
||||||
|
@ -672,19 +653,19 @@ class CoreXmlReader:
|
||||||
lon = get_float(session_origin, "lon")
|
lon = get_float(session_origin, "lon")
|
||||||
alt = get_float(session_origin, "alt")
|
alt = get_float(session_origin, "alt")
|
||||||
if all([lat, lon, alt]):
|
if all([lat, lon, alt]):
|
||||||
logging.info("reading session reference geo: %s, %s, %s", lat, lon, alt)
|
logger.info("reading session reference geo: %s, %s, %s", lat, lon, alt)
|
||||||
self.session.location.setrefgeo(lat, lon, alt)
|
self.session.location.setrefgeo(lat, lon, alt)
|
||||||
|
|
||||||
scale = get_float(session_origin, "scale")
|
scale = get_float(session_origin, "scale")
|
||||||
if scale:
|
if scale:
|
||||||
logging.info("reading session reference scale: %s", scale)
|
logger.info("reading session reference scale: %s", scale)
|
||||||
self.session.location.refscale = scale
|
self.session.location.refscale = scale
|
||||||
|
|
||||||
x = get_float(session_origin, "x")
|
x = get_float(session_origin, "x")
|
||||||
y = get_float(session_origin, "y")
|
y = get_float(session_origin, "y")
|
||||||
z = get_float(session_origin, "z")
|
z = get_float(session_origin, "z")
|
||||||
if all([x, y]):
|
if all([x, y]):
|
||||||
logging.info("reading session reference xyz: %s, %s, %s", x, y, z)
|
logger.info("reading session reference xyz: %s, %s, %s", x, y, z)
|
||||||
self.session.location.refxyz = (x, y, z)
|
self.session.location.refxyz = (x, y, z)
|
||||||
|
|
||||||
def read_service_configs(self) -> None:
|
def read_service_configs(self) -> None:
|
||||||
|
@ -695,7 +676,7 @@ class CoreXmlReader:
|
||||||
for service_configuration in service_configurations.iterchildren():
|
for service_configuration in service_configurations.iterchildren():
|
||||||
node_id = get_int(service_configuration, "node")
|
node_id = get_int(service_configuration, "node")
|
||||||
service_name = service_configuration.get("name")
|
service_name = service_configuration.get("name")
|
||||||
logging.info(
|
logger.info(
|
||||||
"reading custom service(%s) for node(%s)", service_name, node_id
|
"reading custom service(%s) for node(%s)", service_name, node_id
|
||||||
)
|
)
|
||||||
self.session.services.set_service(node_id, service_name)
|
self.session.services.set_service(node_id, service_name)
|
||||||
|
@ -731,28 +712,10 @@ class CoreXmlReader:
|
||||||
files.add(name)
|
files.add(name)
|
||||||
service.configs = tuple(files)
|
service.configs = tuple(files)
|
||||||
|
|
||||||
def read_emane_global_config(self) -> None:
|
|
||||||
emane_global_configuration = self.scenario.find("emane_global_configuration")
|
|
||||||
if emane_global_configuration is None:
|
|
||||||
return
|
|
||||||
emulator_configuration = emane_global_configuration.find("emulator")
|
|
||||||
configs = {}
|
|
||||||
for config in emulator_configuration.iterchildren():
|
|
||||||
name = config.get("name")
|
|
||||||
value = config.get("value")
|
|
||||||
configs[name] = value
|
|
||||||
core_configuration = emane_global_configuration.find("core")
|
|
||||||
for config in core_configuration.iterchildren():
|
|
||||||
name = config.get("name")
|
|
||||||
value = config.get("value")
|
|
||||||
configs[name] = value
|
|
||||||
self.session.emane.set_configs(config=configs)
|
|
||||||
|
|
||||||
def read_emane_configs(self) -> None:
|
def read_emane_configs(self) -> None:
|
||||||
emane_configurations = self.scenario.find("emane_configurations")
|
emane_configurations = self.scenario.find("emane_configurations")
|
||||||
if emane_configurations is None:
|
if emane_configurations is None:
|
||||||
return
|
return
|
||||||
|
|
||||||
for emane_configuration in emane_configurations.iterchildren():
|
for emane_configuration in emane_configurations.iterchildren():
|
||||||
node_id = get_int(emane_configuration, "node")
|
node_id = get_int(emane_configuration, "node")
|
||||||
iface_id = get_int(emane_configuration, "iface")
|
iface_id = get_int(emane_configuration, "iface")
|
||||||
|
@ -763,38 +726,39 @@ class CoreXmlReader:
|
||||||
node = self.session.nodes.get(node_id)
|
node = self.session.nodes.get(node_id)
|
||||||
if not node:
|
if not node:
|
||||||
raise CoreXmlError(f"node for emane config doesn't exist: {node_id}")
|
raise CoreXmlError(f"node for emane config doesn't exist: {node_id}")
|
||||||
model = self.session.emane.models.get(model_name)
|
self.session.emane.get_model(model_name)
|
||||||
if not model:
|
|
||||||
raise CoreXmlError(f"invalid emane model: {model_name}")
|
|
||||||
if iface_id is not None and iface_id not in node.ifaces:
|
if iface_id is not None and iface_id not in node.ifaces:
|
||||||
raise CoreXmlError(
|
raise CoreXmlError(
|
||||||
f"invalid interface id({iface_id}) for node({node.name})"
|
f"invalid interface id({iface_id}) for node({node.name})"
|
||||||
)
|
)
|
||||||
|
|
||||||
# read and set emane model configuration
|
# read and set emane model configuration
|
||||||
|
platform_configuration = emane_configuration.find("platform")
|
||||||
|
for config in platform_configuration.iterchildren():
|
||||||
|
name = config.get("name")
|
||||||
|
value = config.get("value")
|
||||||
|
configs[name] = value
|
||||||
mac_configuration = emane_configuration.find("mac")
|
mac_configuration = emane_configuration.find("mac")
|
||||||
for config in mac_configuration.iterchildren():
|
for config in mac_configuration.iterchildren():
|
||||||
name = config.get("name")
|
name = config.get("name")
|
||||||
value = config.get("value")
|
value = config.get("value")
|
||||||
configs[name] = value
|
configs[name] = value
|
||||||
|
|
||||||
phy_configuration = emane_configuration.find("phy")
|
phy_configuration = emane_configuration.find("phy")
|
||||||
for config in phy_configuration.iterchildren():
|
for config in phy_configuration.iterchildren():
|
||||||
name = config.get("name")
|
name = config.get("name")
|
||||||
value = config.get("value")
|
value = config.get("value")
|
||||||
configs[name] = value
|
configs[name] = value
|
||||||
|
|
||||||
external_configuration = emane_configuration.find("external")
|
external_configuration = emane_configuration.find("external")
|
||||||
for config in external_configuration.iterchildren():
|
for config in external_configuration.iterchildren():
|
||||||
name = config.get("name")
|
name = config.get("name")
|
||||||
value = config.get("value")
|
value = config.get("value")
|
||||||
configs[name] = value
|
configs[name] = value
|
||||||
|
|
||||||
logging.info(
|
logger.info(
|
||||||
"reading emane configuration node(%s) model(%s)", node_id, model_name
|
"reading emane configuration node(%s) model(%s)", node_id, model_name
|
||||||
)
|
)
|
||||||
node_id = utils.iface_config_id(node_id, iface_id)
|
node_id = utils.iface_config_id(node_id, iface_id)
|
||||||
self.session.emane.set_model_config(node_id, model_name, configs)
|
self.session.emane.set_config(node_id, model_name, configs)
|
||||||
|
|
||||||
def read_mobility_configs(self) -> None:
|
def read_mobility_configs(self) -> None:
|
||||||
mobility_configurations = self.scenario.find("mobility_configurations")
|
mobility_configurations = self.scenario.find("mobility_configurations")
|
||||||
|
@ -811,7 +775,7 @@ class CoreXmlReader:
|
||||||
value = config.get("value")
|
value = config.get("value")
|
||||||
configs[name] = value
|
configs[name] = value
|
||||||
|
|
||||||
logging.info(
|
logger.info(
|
||||||
"reading mobility configuration node(%s) model(%s)", node_id, model_name
|
"reading mobility configuration node(%s) model(%s)", node_id, model_name
|
||||||
)
|
)
|
||||||
self.session.mobility.set_model_config(node_id, model_name, configs)
|
self.session.mobility.set_model_config(node_id, model_name, configs)
|
||||||
|
@ -868,7 +832,7 @@ class CoreXmlReader:
|
||||||
if all([lat, lon, alt]):
|
if all([lat, lon, alt]):
|
||||||
options.set_location(lat, lon, alt)
|
options.set_location(lat, lon, alt)
|
||||||
|
|
||||||
logging.info("reading node id(%s) model(%s) name(%s)", node_id, model, name)
|
logger.info("reading node id(%s) model(%s) name(%s)", node_id, model, name)
|
||||||
self.session.add_node(_class, node_id, options)
|
self.session.add_node(_class, node_id, options)
|
||||||
|
|
||||||
def read_network(self, network_element: etree.Element) -> None:
|
def read_network(self, network_element: etree.Element) -> None:
|
||||||
|
@ -896,7 +860,7 @@ class CoreXmlReader:
|
||||||
if all([lat, lon, alt]):
|
if all([lat, lon, alt]):
|
||||||
options.set_location(lat, lon, alt)
|
options.set_location(lat, lon, alt)
|
||||||
|
|
||||||
logging.info(
|
logger.info(
|
||||||
"reading node id(%s) node_type(%s) name(%s)", node_id, node_type, name
|
"reading node id(%s) node_type(%s) name(%s)", node_id, node_type, name
|
||||||
)
|
)
|
||||||
self.session.add_node(_class, node_id, options)
|
self.session.add_node(_class, node_id, options)
|
||||||
|
@ -926,7 +890,7 @@ class CoreXmlReader:
|
||||||
for template_element in templates_element.iterchildren():
|
for template_element in templates_element.iterchildren():
|
||||||
name = template_element.get("name")
|
name = template_element.get("name")
|
||||||
template = template_element.text
|
template = template_element.text
|
||||||
logging.info(
|
logger.info(
|
||||||
"loading xml template(%s): %s", type(template), template
|
"loading xml template(%s): %s", type(template), template
|
||||||
)
|
)
|
||||||
service.set_template(name, template)
|
service.set_template(name, template)
|
||||||
|
@ -978,12 +942,12 @@ class CoreXmlReader:
|
||||||
options.buffer = get_int(options_element, "buffer")
|
options.buffer = get_int(options_element, "buffer")
|
||||||
|
|
||||||
if options.unidirectional == 1 and node_set in node_sets:
|
if options.unidirectional == 1 and node_set in node_sets:
|
||||||
logging.info("updating link node1(%s) node2(%s)", node1_id, node2_id)
|
logger.info("updating link node1(%s) node2(%s)", node1_id, node2_id)
|
||||||
self.session.update_link(
|
self.session.update_link(
|
||||||
node1_id, node2_id, iface1_data.id, iface2_data.id, options
|
node1_id, node2_id, iface1_data.id, iface2_data.id, options
|
||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
logging.info("adding link node1(%s) node2(%s)", node1_id, node2_id)
|
logger.info("adding link node1(%s) node2(%s)", node1_id, node2_id)
|
||||||
self.session.add_link(
|
self.session.add_link(
|
||||||
node1_id, node2_id, iface1_data, iface2_data, options
|
node1_id, node2_id, iface1_data, iface2_data, options
|
||||||
)
|
)
|
||||||
|
|
|
@ -1,5 +1,5 @@
|
||||||
import logging
|
import logging
|
||||||
import os
|
from pathlib import Path
|
||||||
from tempfile import NamedTemporaryFile
|
from tempfile import NamedTemporaryFile
|
||||||
from typing import TYPE_CHECKING, Dict, List, Optional, Set, Tuple
|
from typing import TYPE_CHECKING, Dict, List, Optional, Set, Tuple
|
||||||
|
|
||||||
|
@ -12,11 +12,11 @@ from core.emulator.distributed import DistributedServer
|
||||||
from core.errors import CoreError
|
from core.errors import CoreError
|
||||||
from core.nodes.base import CoreNode, CoreNodeBase
|
from core.nodes.base import CoreNode, CoreNodeBase
|
||||||
from core.nodes.interface import CoreInterface
|
from core.nodes.interface import CoreInterface
|
||||||
from core.nodes.network import CtrlNet
|
|
||||||
from core.xml import corexml
|
from core.xml import corexml
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
if TYPE_CHECKING:
|
if TYPE_CHECKING:
|
||||||
from core.emane.emanemanager import EmaneManager, StartData
|
|
||||||
from core.emane.emanemodel import EmaneModel
|
from core.emane.emanemodel import EmaneModel
|
||||||
|
|
||||||
_MAC_PREFIX = "02:02"
|
_MAC_PREFIX = "02:02"
|
||||||
|
@ -47,14 +47,14 @@ def _value_to_params(value: str) -> Optional[Tuple[str]]:
|
||||||
return None
|
return None
|
||||||
return values
|
return values
|
||||||
except SyntaxError:
|
except SyntaxError:
|
||||||
logging.exception("error in value string to param list")
|
logger.exception("error in value string to param list")
|
||||||
return None
|
return None
|
||||||
|
|
||||||
|
|
||||||
def create_file(
|
def create_file(
|
||||||
xml_element: etree.Element,
|
xml_element: etree.Element,
|
||||||
doc_name: str,
|
doc_name: str,
|
||||||
file_path: str,
|
file_path: Path,
|
||||||
server: DistributedServer = None,
|
server: DistributedServer = None,
|
||||||
) -> None:
|
) -> None:
|
||||||
"""
|
"""
|
||||||
|
@ -71,10 +71,11 @@ def create_file(
|
||||||
)
|
)
|
||||||
if server:
|
if server:
|
||||||
temp = NamedTemporaryFile(delete=False)
|
temp = NamedTemporaryFile(delete=False)
|
||||||
corexml.write_xml_file(xml_element, temp.name, doctype=doctype)
|
temp_path = Path(temp.name)
|
||||||
|
corexml.write_xml_file(xml_element, temp_path, doctype=doctype)
|
||||||
temp.close()
|
temp.close()
|
||||||
server.remote_put(temp.name, file_path)
|
server.remote_put(temp_path, file_path)
|
||||||
os.unlink(temp.name)
|
temp_path.unlink()
|
||||||
else:
|
else:
|
||||||
corexml.write_xml_file(xml_element, file_path, doctype=doctype)
|
corexml.write_xml_file(xml_element, file_path, doctype=doctype)
|
||||||
|
|
||||||
|
@ -92,9 +93,9 @@ def create_node_file(
|
||||||
:return:
|
:return:
|
||||||
"""
|
"""
|
||||||
if isinstance(node, CoreNode):
|
if isinstance(node, CoreNode):
|
||||||
file_path = os.path.join(node.nodedir, file_name)
|
file_path = node.directory / file_name
|
||||||
else:
|
else:
|
||||||
file_path = os.path.join(node.session.session_dir, file_name)
|
file_path = node.session.directory / file_name
|
||||||
create_file(xml_element, doc_name, file_path, node.server)
|
create_file(xml_element, doc_name, file_path, node.server)
|
||||||
|
|
||||||
|
|
||||||
|
@ -143,74 +144,67 @@ def add_configurations(
|
||||||
|
|
||||||
|
|
||||||
def build_platform_xml(
|
def build_platform_xml(
|
||||||
emane_manager: "EmaneManager", control_net: CtrlNet, data: "StartData"
|
nem_id: int,
|
||||||
|
nem_port: int,
|
||||||
|
emane_net: EmaneNet,
|
||||||
|
iface: CoreInterface,
|
||||||
|
config: Dict[str, str],
|
||||||
) -> None:
|
) -> None:
|
||||||
"""
|
"""
|
||||||
Create platform xml for a specific node.
|
Create platform xml for a nem/interface.
|
||||||
|
|
||||||
:param emane_manager: emane manager with emane
|
:param nem_id: nem id for current node/interface
|
||||||
configurations
|
:param nem_port: control port to configure for emane
|
||||||
:param control_net: control net node for this emane
|
:param emane_net: emane network associate with node and interface
|
||||||
network
|
:param iface: node interface to create platform xml for
|
||||||
:param data: start data for a node connected to emane and associated interfaces
|
:param config: emane configuration for interface
|
||||||
:return: the next nem id that can be used for creating platform xml files
|
:return: nothing
|
||||||
"""
|
"""
|
||||||
# create top level platform element
|
# create top level platform element
|
||||||
transport_configs = {"otamanagerdevice", "eventservicedevice"}
|
|
||||||
platform_element = etree.Element("platform")
|
platform_element = etree.Element("platform")
|
||||||
for configuration in emane_manager.emane_config.emulator_config:
|
for configuration in emane_net.model.platform_config:
|
||||||
name = configuration.id
|
name = configuration.id
|
||||||
if not isinstance(data.node, CoreNode) and name in transport_configs:
|
value = config[configuration.id]
|
||||||
value = control_net.brname
|
|
||||||
else:
|
|
||||||
value = emane_manager.get_config(name)
|
|
||||||
add_param(platform_element, name, value)
|
add_param(platform_element, name, value)
|
||||||
|
add_param(
|
||||||
|
platform_element, emane_net.model.platform_controlport, f"0.0.0.0:{nem_port}"
|
||||||
|
)
|
||||||
|
|
||||||
# create nem xml entries for all interfaces
|
# build nem xml
|
||||||
for iface in data.ifaces:
|
nem_definition = nem_file_name(iface)
|
||||||
emane_net = iface.net
|
nem_element = etree.Element(
|
||||||
if not isinstance(emane_net, EmaneNet):
|
"nem", id=str(nem_id), name=iface.localname, definition=nem_definition
|
||||||
raise CoreError(
|
)
|
||||||
f"emane interface not connected to emane net: {emane_net.name}"
|
|
||||||
)
|
|
||||||
nem_id = emane_manager.next_nem_id()
|
|
||||||
emane_manager.set_nem(nem_id, iface)
|
|
||||||
emane_manager.write_nem(iface, nem_id)
|
|
||||||
config = emane_manager.get_iface_config(emane_net, iface)
|
|
||||||
emane_net.model.build_xml_files(config, iface)
|
|
||||||
|
|
||||||
# build nem xml
|
# create model based xml files
|
||||||
nem_definition = nem_file_name(iface)
|
emane_net.model.build_xml_files(config, iface)
|
||||||
nem_element = etree.Element(
|
|
||||||
"nem", id=str(nem_id), name=iface.localname, definition=nem_definition
|
|
||||||
)
|
|
||||||
|
|
||||||
# check if this is an external transport
|
# check if this is an external transport
|
||||||
if is_external(config):
|
if is_external(config):
|
||||||
nem_element.set("transport", "external")
|
nem_element.set("transport", "external")
|
||||||
platform_endpoint = "platformendpoint"
|
platform_endpoint = "platformendpoint"
|
||||||
add_param(nem_element, platform_endpoint, config[platform_endpoint])
|
add_param(nem_element, platform_endpoint, config[platform_endpoint])
|
||||||
transport_endpoint = "transportendpoint"
|
transport_endpoint = "transportendpoint"
|
||||||
add_param(nem_element, transport_endpoint, config[transport_endpoint])
|
add_param(nem_element, transport_endpoint, config[transport_endpoint])
|
||||||
|
|
||||||
# define transport element
|
# define transport element
|
||||||
transport_name = transport_file_name(iface)
|
transport_name = transport_file_name(iface)
|
||||||
transport_element = etree.SubElement(
|
transport_element = etree.SubElement(
|
||||||
nem_element, "transport", definition=transport_name
|
nem_element, "transport", definition=transport_name
|
||||||
)
|
)
|
||||||
add_param(transport_element, "device", iface.name)
|
add_param(transport_element, "device", iface.name)
|
||||||
|
|
||||||
# add nem element to platform element
|
# add nem element to platform element
|
||||||
platform_element.append(nem_element)
|
platform_element.append(nem_element)
|
||||||
|
|
||||||
# generate and assign interface mac address based on nem id
|
# generate and assign interface mac address based on nem id
|
||||||
mac = _MAC_PREFIX + ":00:00:"
|
mac = _MAC_PREFIX + ":00:00:"
|
||||||
mac += f"{(nem_id >> 8) & 0xFF:02X}:{nem_id & 0xFF:02X}"
|
mac += f"{(nem_id >> 8) & 0xFF:02X}:{nem_id & 0xFF:02X}"
|
||||||
iface.set_mac(mac)
|
iface.set_mac(mac)
|
||||||
|
|
||||||
doc_name = "platform"
|
doc_name = "platform"
|
||||||
file_name = f"{data.node.name}-platform.xml"
|
file_name = platform_file_name(iface)
|
||||||
create_node_file(data.node, platform_element, doc_name, file_name)
|
create_node_file(iface.node, platform_element, doc_name, file_name)
|
||||||
|
|
||||||
|
|
||||||
def create_transport_xml(iface: CoreInterface, config: Dict[str, str]) -> None:
|
def create_transport_xml(iface: CoreInterface, config: Dict[str, str]) -> None:
|
||||||
|
@ -316,7 +310,7 @@ def create_event_service_xml(
|
||||||
group: str,
|
group: str,
|
||||||
port: str,
|
port: str,
|
||||||
device: str,
|
device: str,
|
||||||
file_directory: str,
|
file_directory: Path,
|
||||||
server: DistributedServer = None,
|
server: DistributedServer = None,
|
||||||
) -> None:
|
) -> None:
|
||||||
"""
|
"""
|
||||||
|
@ -340,8 +334,7 @@ def create_event_service_xml(
|
||||||
):
|
):
|
||||||
sub_element = etree.SubElement(event_element, name)
|
sub_element = etree.SubElement(event_element, name)
|
||||||
sub_element.text = value
|
sub_element.text = value
|
||||||
file_name = "libemaneeventservice.xml"
|
file_path = file_directory / "libemaneeventservice.xml"
|
||||||
file_path = os.path.join(file_directory, file_name)
|
|
||||||
create_file(event_element, "emaneeventmsgsvc", file_path, server)
|
create_file(event_element, "emaneeventmsgsvc", file_path, server)
|
||||||
|
|
||||||
|
|
||||||
|
@ -394,3 +387,7 @@ def phy_file_name(iface: CoreInterface) -> str:
|
||||||
:return: phy xml file name
|
:return: phy xml file name
|
||||||
"""
|
"""
|
||||||
return f"{iface.name}-phy.xml"
|
return f"{iface.name}-phy.xml"
|
||||||
|
|
||||||
|
|
||||||
|
def platform_file_name(iface: CoreInterface) -> str:
|
||||||
|
return f"{iface.name}-platform.xml"
|
||||||
|
|
|
@ -13,8 +13,21 @@
|
||||||
"format": "%(asctime)s - %(levelname)s - %(module)s:%(funcName)s - %(message)s"
|
"format": "%(asctime)s - %(levelname)s - %(module)s:%(funcName)s - %(message)s"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
"root": {
|
"loggers": {
|
||||||
"level": "INFO",
|
"": {
|
||||||
"handlers": ["console"]
|
"level": "WARNING",
|
||||||
}
|
"handlers": ["console"],
|
||||||
|
"propagate": false
|
||||||
|
},
|
||||||
|
"core": {
|
||||||
|
"level": "INFO",
|
||||||
|
"handlers": ["console"],
|
||||||
|
"propagate": false
|
||||||
|
},
|
||||||
|
"__main__": {
|
||||||
|
"level": "INFO",
|
||||||
|
"handlers": ["console"],
|
||||||
|
"propagate": false
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,5 +0,0 @@
|
||||||
{
|
|
||||||
"bridge": "none",
|
|
||||||
"iptables": false
|
|
||||||
|
|
||||||
}
|
|
|
@ -2,7 +2,7 @@ import argparse
|
||||||
import logging
|
import logging
|
||||||
|
|
||||||
from core.api.grpc import client
|
from core.api.grpc import client
|
||||||
from core.api.grpc.core_pb2 import Node, NodeType, Position, SessionState
|
from core.api.grpc.wrappers import NodeType, Position, Server
|
||||||
|
|
||||||
|
|
||||||
def log_event(event):
|
def log_event(event):
|
||||||
|
@ -10,62 +10,39 @@ def log_event(event):
|
||||||
|
|
||||||
|
|
||||||
def main(args):
|
def main(args):
|
||||||
|
# helper to create interfaces
|
||||||
|
interface_helper = client.InterfaceHelper(ip4_prefix="10.83.0.0/16")
|
||||||
|
|
||||||
|
# create grpc client and connect
|
||||||
core = client.CoreGrpcClient()
|
core = client.CoreGrpcClient()
|
||||||
|
core.connect()
|
||||||
|
|
||||||
with core.context_connect():
|
# create session
|
||||||
# create session
|
session = core.create_session()
|
||||||
response = core.create_session()
|
|
||||||
session_id = response.session_id
|
|
||||||
logging.info("created session: %s", response)
|
|
||||||
|
|
||||||
# add distributed server
|
# add distributed server
|
||||||
server_name = "core2"
|
server = Server(name="core2", host=args.server)
|
||||||
response = core.add_session_server(session_id, server_name, args.server)
|
session.servers.append(server)
|
||||||
logging.info("added session server: %s", response)
|
|
||||||
|
|
||||||
# handle events session may broadcast
|
# handle events session may broadcast
|
||||||
core.events(session_id, log_event)
|
core.events(session.id, log_event)
|
||||||
|
|
||||||
# change session state
|
# create switch node
|
||||||
response = core.set_session_state(session_id, SessionState.CONFIGURATION)
|
position = Position(x=150, y=100)
|
||||||
logging.info("set session state: %s", response)
|
switch = session.add_node(1, _type=NodeType.SWITCH, position=position)
|
||||||
|
position = Position(x=100, y=50)
|
||||||
|
node1 = session.add_node(2, position=position)
|
||||||
|
position = Position(x=200, y=50)
|
||||||
|
node2 = session.add_node(3, position=position, server=server.name)
|
||||||
|
|
||||||
# create switch node
|
# create links
|
||||||
switch = Node(type=NodeType.SWITCH)
|
iface1 = interface_helper.create_iface(node1.id, 0)
|
||||||
response = core.add_node(session_id, switch)
|
session.add_link(node1=node1, node2=switch, iface1=iface1)
|
||||||
logging.info("created switch: %s", response)
|
iface1 = interface_helper.create_iface(node2.id, 0)
|
||||||
switch_id = response.node_id
|
session.add_link(node1=node2, node2=switch, iface1=iface1)
|
||||||
|
|
||||||
# helper to create interfaces
|
# start session
|
||||||
interface_helper = client.InterfaceHelper(ip4_prefix="10.83.0.0/16")
|
core.start_session(session)
|
||||||
|
|
||||||
# create node one
|
|
||||||
position = Position(x=100, y=50)
|
|
||||||
node = Node(position=position)
|
|
||||||
response = core.add_node(session_id, node)
|
|
||||||
logging.info("created node one: %s", response)
|
|
||||||
node1_id = response.node_id
|
|
||||||
|
|
||||||
# create link
|
|
||||||
interface1 = interface_helper.create_iface(node1_id, 0)
|
|
||||||
response = core.add_link(session_id, node1_id, switch_id, interface1)
|
|
||||||
logging.info("created link from node one to switch: %s", response)
|
|
||||||
|
|
||||||
# create node two
|
|
||||||
position = Position(x=200, y=50)
|
|
||||||
node = Node(position=position, server=server_name)
|
|
||||||
response = core.add_node(session_id, node)
|
|
||||||
logging.info("created node two: %s", response)
|
|
||||||
node2_id = response.node_id
|
|
||||||
|
|
||||||
# create link
|
|
||||||
interface1 = interface_helper.create_iface(node2_id, 0)
|
|
||||||
response = core.add_link(session_id, node2_id, switch_id, interface1)
|
|
||||||
logging.info("created link from node two to switch: %s", response)
|
|
||||||
|
|
||||||
# change session state
|
|
||||||
response = core.set_session_state(session_id, SessionState.INSTANTIATION)
|
|
||||||
logging.info("set session state: %s", response)
|
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
|
|
|
@ -1,7 +1,7 @@
|
||||||
# required imports
|
# required imports
|
||||||
from core.api.grpc import client
|
from core.api.grpc import client
|
||||||
from core.api.grpc.core_pb2 import Node, NodeType, Position, SessionState
|
from core.api.grpc.wrappers import NodeType, Position
|
||||||
from core.emane.ieee80211abg import EmaneIeee80211abgModel
|
from core.emane.models.ieee80211abg import EmaneIeee80211abgModel
|
||||||
|
|
||||||
# interface helper
|
# interface helper
|
||||||
iface_helper = client.InterfaceHelper(ip4_prefix="10.0.0.0/24", ip6_prefix="2001::/64")
|
iface_helper = client.InterfaceHelper(ip4_prefix="10.0.0.0/24", ip6_prefix="2001::/64")
|
||||||
|
@ -10,45 +10,29 @@ iface_helper = client.InterfaceHelper(ip4_prefix="10.0.0.0/24", ip6_prefix="2001
|
||||||
core = client.CoreGrpcClient()
|
core = client.CoreGrpcClient()
|
||||||
core.connect()
|
core.connect()
|
||||||
|
|
||||||
# create session and get id
|
# add session
|
||||||
response = core.create_session()
|
session = core.create_session()
|
||||||
session_id = response.session_id
|
|
||||||
|
|
||||||
# change session state to configuration so that nodes get started when added
|
# create nodes
|
||||||
core.set_session_state(session_id, SessionState.CONFIGURATION)
|
|
||||||
|
|
||||||
# create emane node
|
|
||||||
position = Position(x=200, y=200)
|
position = Position(x=200, y=200)
|
||||||
emane = Node(type=NodeType.EMANE, position=position, emane=EmaneIeee80211abgModel.name)
|
emane = session.add_node(
|
||||||
response = core.add_node(session_id, emane)
|
1, _type=NodeType.EMANE, position=position, emane=EmaneIeee80211abgModel.name
|
||||||
emane_id = response.node_id
|
)
|
||||||
|
|
||||||
# create node one
|
|
||||||
position = Position(x=100, y=100)
|
position = Position(x=100, y=100)
|
||||||
n1 = Node(type=NodeType.DEFAULT, position=position, model="mdr")
|
node1 = session.add_node(2, model="mdr", position=position)
|
||||||
response = core.add_node(session_id, n1)
|
|
||||||
n1_id = response.node_id
|
|
||||||
|
|
||||||
# create node two
|
|
||||||
position = Position(x=300, y=100)
|
position = Position(x=300, y=100)
|
||||||
n2 = Node(type=NodeType.DEFAULT, position=position, model="mdr")
|
node2 = session.add_node(3, model="mdr", position=position)
|
||||||
response = core.add_node(session_id, n2)
|
|
||||||
n2_id = response.node_id
|
|
||||||
|
|
||||||
# configure general emane settings
|
# create links
|
||||||
core.set_emane_config(session_id, {"eventservicettl": "2"})
|
iface1 = iface_helper.create_iface(node1.id, 0)
|
||||||
|
session.add_link(node1=node1, node2=emane, iface1=iface1)
|
||||||
|
iface1 = iface_helper.create_iface(node2.id, 0)
|
||||||
|
session.add_link(node1=node2, node2=emane, iface1=iface1)
|
||||||
|
|
||||||
# configure emane model settings
|
# setup emane configurations using a dict mapping currently support values as strings
|
||||||
# using a dict mapping currently support values as strings
|
emane.set_emane_model(
|
||||||
core.set_emane_model_config(
|
EmaneIeee80211abgModel.name, {"eventservicettl": "2", "unicastrate": "3"}
|
||||||
session_id, emane_id, EmaneIeee80211abgModel.name, {"unicastrate": "3"}
|
|
||||||
)
|
)
|
||||||
|
|
||||||
# links nodes to emane
|
# start session
|
||||||
iface1 = iface_helper.create_iface(n1_id, 0)
|
core.start_session(session)
|
||||||
core.add_link(session_id, n1_id, emane_id, iface1)
|
|
||||||
iface1 = iface_helper.create_iface(n2_id, 0)
|
|
||||||
core.add_link(session_id, n2_id, emane_id, iface1)
|
|
||||||
|
|
||||||
# change session state
|
|
||||||
core.set_session_state(session_id, SessionState.INSTANTIATION)
|
|
||||||
|
|
|
@ -1,5 +1,5 @@
|
||||||
from core.api.grpc import client
|
from core.api.grpc import client
|
||||||
from core.api.grpc.core_pb2 import Node, NodeType, Position, SessionState
|
from core.api.grpc.wrappers import Position
|
||||||
|
|
||||||
# interface helper
|
# interface helper
|
||||||
iface_helper = client.InterfaceHelper(ip4_prefix="10.0.0.0/24", ip6_prefix="2001::/64")
|
iface_helper = client.InterfaceHelper(ip4_prefix="10.0.0.0/24", ip6_prefix="2001::/64")
|
||||||
|
@ -8,29 +8,19 @@ iface_helper = client.InterfaceHelper(ip4_prefix="10.0.0.0/24", ip6_prefix="2001
|
||||||
core = client.CoreGrpcClient()
|
core = client.CoreGrpcClient()
|
||||||
core.connect()
|
core.connect()
|
||||||
|
|
||||||
# create session and get id
|
# add session
|
||||||
response = core.create_session()
|
session = core.create_session()
|
||||||
session_id = response.session_id
|
|
||||||
|
|
||||||
# change session state to configuration so that nodes get started when added
|
# create nodes
|
||||||
core.set_session_state(session_id, SessionState.CONFIGURATION)
|
|
||||||
|
|
||||||
# create node one
|
|
||||||
position = Position(x=100, y=100)
|
position = Position(x=100, y=100)
|
||||||
n1 = Node(type=NodeType.DEFAULT, position=position, model="PC")
|
node1 = session.add_node(1, position=position)
|
||||||
response = core.add_node(session_id, n1)
|
|
||||||
n1_id = response.node_id
|
|
||||||
|
|
||||||
# create node two
|
|
||||||
position = Position(x=300, y=100)
|
position = Position(x=300, y=100)
|
||||||
n2 = Node(type=NodeType.DEFAULT, position=position, model="PC")
|
node2 = session.add_node(2, position=position)
|
||||||
response = core.add_node(session_id, n2)
|
|
||||||
n2_id = response.node_id
|
|
||||||
|
|
||||||
# links nodes together
|
# create link
|
||||||
iface1 = iface_helper.create_iface(n1_id, 0)
|
iface1 = iface_helper.create_iface(node1.id, 0)
|
||||||
iface2 = iface_helper.create_iface(n2_id, 0)
|
iface2 = iface_helper.create_iface(node2.id, 0)
|
||||||
core.add_link(session_id, n1_id, n2_id, iface1, iface2)
|
session.add_link(node1=node1, node2=node2, iface1=iface1, iface2=iface2)
|
||||||
|
|
||||||
# change session state
|
# start session
|
||||||
core.set_session_state(session_id, SessionState.INSTANTIATION)
|
core.start_session(session)
|
||||||
|
|
|
@ -1,6 +1,5 @@
|
||||||
# required imports
|
|
||||||
from core.api.grpc import client
|
from core.api.grpc import client
|
||||||
from core.api.grpc.core_pb2 import Node, NodeType, Position, SessionState
|
from core.api.grpc.wrappers import NodeType, Position
|
||||||
|
|
||||||
# interface helper
|
# interface helper
|
||||||
iface_helper = client.InterfaceHelper(ip4_prefix="10.0.0.0/24", ip6_prefix="2001::/64")
|
iface_helper = client.InterfaceHelper(ip4_prefix="10.0.0.0/24", ip6_prefix="2001::/64")
|
||||||
|
@ -9,36 +8,22 @@ iface_helper = client.InterfaceHelper(ip4_prefix="10.0.0.0/24", ip6_prefix="2001
|
||||||
core = client.CoreGrpcClient()
|
core = client.CoreGrpcClient()
|
||||||
core.connect()
|
core.connect()
|
||||||
|
|
||||||
# create session and get id
|
# add session
|
||||||
response = core.create_session()
|
session = core.create_session()
|
||||||
session_id = response.session_id
|
|
||||||
|
|
||||||
# change session state to configuration so that nodes get started when added
|
# create nodes
|
||||||
core.set_session_state(session_id, SessionState.CONFIGURATION)
|
|
||||||
|
|
||||||
# create switch node
|
|
||||||
position = Position(x=200, y=200)
|
position = Position(x=200, y=200)
|
||||||
switch = Node(type=NodeType.SWITCH, position=position)
|
switch = session.add_node(1, _type=NodeType.SWITCH, position=position)
|
||||||
response = core.add_node(session_id, switch)
|
|
||||||
switch_id = response.node_id
|
|
||||||
|
|
||||||
# create node one
|
|
||||||
position = Position(x=100, y=100)
|
position = Position(x=100, y=100)
|
||||||
n1 = Node(type=NodeType.DEFAULT, position=position, model="PC")
|
node1 = session.add_node(2, position=position)
|
||||||
response = core.add_node(session_id, n1)
|
|
||||||
n1_id = response.node_id
|
|
||||||
|
|
||||||
# create node two
|
|
||||||
position = Position(x=300, y=100)
|
position = Position(x=300, y=100)
|
||||||
n2 = Node(type=NodeType.DEFAULT, position=position, model="PC")
|
node2 = session.add_node(3, position=position)
|
||||||
response = core.add_node(session_id, n2)
|
|
||||||
n2_id = response.node_id
|
|
||||||
|
|
||||||
# links nodes to switch
|
# create links
|
||||||
iface1 = iface_helper.create_iface(n1_id, 0)
|
iface1 = iface_helper.create_iface(node1.id, 0)
|
||||||
core.add_link(session_id, n1_id, switch_id, iface1)
|
session.add_link(node1=node1, node2=switch, iface1=iface1)
|
||||||
iface1 = iface_helper.create_iface(n2_id, 0)
|
iface1 = iface_helper.create_iface(node2.id, 0)
|
||||||
core.add_link(session_id, n2_id, switch_id, iface1)
|
session.add_link(node1=node2, node2=switch, iface1=iface1)
|
||||||
|
|
||||||
# change session state
|
# start session
|
||||||
core.set_session_state(session_id, SessionState.INSTANTIATION)
|
core.start_session(session)
|
||||||
|
|
|
@ -1,6 +1,5 @@
|
||||||
# required imports
|
|
||||||
from core.api.grpc import client
|
from core.api.grpc import client
|
||||||
from core.api.grpc.core_pb2 import Node, NodeType, Position, SessionState
|
from core.api.grpc.wrappers import NodeType, Position
|
||||||
|
|
||||||
# interface helper
|
# interface helper
|
||||||
iface_helper = client.InterfaceHelper(ip4_prefix="10.0.0.0/24", ip6_prefix="2001::/64")
|
iface_helper = client.InterfaceHelper(ip4_prefix="10.0.0.0/24", ip6_prefix="2001::/64")
|
||||||
|
@ -9,50 +8,34 @@ iface_helper = client.InterfaceHelper(ip4_prefix="10.0.0.0/24", ip6_prefix="2001
|
||||||
core = client.CoreGrpcClient()
|
core = client.CoreGrpcClient()
|
||||||
core.connect()
|
core.connect()
|
||||||
|
|
||||||
# create session and get id
|
# add session
|
||||||
response = core.create_session()
|
session = core.create_session()
|
||||||
session_id = response.session_id
|
|
||||||
|
|
||||||
# change session state to configuration so that nodes get started when added
|
# create nodes
|
||||||
core.set_session_state(session_id, SessionState.CONFIGURATION)
|
|
||||||
|
|
||||||
# create wlan node
|
|
||||||
position = Position(x=200, y=200)
|
position = Position(x=200, y=200)
|
||||||
wlan = Node(type=NodeType.WIRELESS_LAN, position=position)
|
wlan = session.add_node(1, _type=NodeType.WIRELESS_LAN, position=position)
|
||||||
response = core.add_node(session_id, wlan)
|
|
||||||
wlan_id = response.node_id
|
|
||||||
|
|
||||||
# create node one
|
|
||||||
position = Position(x=100, y=100)
|
position = Position(x=100, y=100)
|
||||||
n1 = Node(type=NodeType.DEFAULT, position=position, model="mdr")
|
node1 = session.add_node(2, model="mdr", position=position)
|
||||||
response = core.add_node(session_id, n1)
|
|
||||||
n1_id = response.node_id
|
|
||||||
|
|
||||||
# create node two
|
|
||||||
position = Position(x=300, y=100)
|
position = Position(x=300, y=100)
|
||||||
n2 = Node(type=NodeType.DEFAULT, position=position, model="mdr")
|
node2 = session.add_node(3, model="mdr", position=position)
|
||||||
response = core.add_node(session_id, n2)
|
|
||||||
n2_id = response.node_id
|
|
||||||
|
|
||||||
# configure wlan using a dict mapping currently
|
# create links
|
||||||
|
iface1 = iface_helper.create_iface(node1.id, 0)
|
||||||
|
session.add_link(node1=node1, node2=wlan, iface1=iface1)
|
||||||
|
iface1 = iface_helper.create_iface(node2.id, 0)
|
||||||
|
session.add_link(node1=node2, node2=wlan, iface1=iface1)
|
||||||
|
|
||||||
|
# set wlan config using a dict mapping currently
|
||||||
# support values as strings
|
# support values as strings
|
||||||
core.set_wlan_config(
|
wlan.set_wlan(
|
||||||
session_id,
|
|
||||||
wlan_id,
|
|
||||||
{
|
{
|
||||||
"range": "280",
|
"range": "280",
|
||||||
"bandwidth": "55000000",
|
"bandwidth": "55000000",
|
||||||
"delay": "6000",
|
"delay": "6000",
|
||||||
"jitter": "5",
|
"jitter": "5",
|
||||||
"error": "5",
|
"error": "5",
|
||||||
},
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
# links nodes to wlan
|
# start session
|
||||||
iface1 = iface_helper.create_iface(n1_id, 0)
|
core.start_session(session)
|
||||||
core.add_link(session_id, n1_id, wlan_id, iface1)
|
|
||||||
iface1 = iface_helper.create_iface(n2_id, 0)
|
|
||||||
core.add_link(session_id, n2_id, wlan_id, iface1)
|
|
||||||
|
|
||||||
# change session state
|
|
||||||
core.set_session_state(session_id, SessionState.INSTANTIATION)
|
|
||||||
|
|
|
@ -1,6 +1,7 @@
|
||||||
"""
|
"""
|
||||||
Example custom emane model.
|
Example custom emane model.
|
||||||
"""
|
"""
|
||||||
|
from pathlib import Path
|
||||||
from typing import Dict, List, Optional, Set
|
from typing import Dict, List, Optional, Set
|
||||||
|
|
||||||
from core.config import Configuration
|
from core.config import Configuration
|
||||||
|
@ -39,17 +40,34 @@ class ExampleModel(emanemodel.EmaneModel):
|
||||||
|
|
||||||
name: str = "emane_example"
|
name: str = "emane_example"
|
||||||
mac_library: str = "rfpipemaclayer"
|
mac_library: str = "rfpipemaclayer"
|
||||||
mac_xml: str = "/usr/share/emane/manifest/rfpipemaclayer.xml"
|
mac_xml: str = "rfpipemaclayer.xml"
|
||||||
mac_defaults: Dict[str, str] = {
|
mac_defaults: Dict[str, str] = {
|
||||||
"pcrcurveuri": "/usr/share/emane/xml/models/mac/rfpipe/rfpipepcr.xml"
|
"pcrcurveuri": "/usr/share/emane/xml/models/mac/rfpipe/rfpipepcr.xml"
|
||||||
}
|
}
|
||||||
mac_config: List[Configuration] = emanemanifest.parse(mac_xml, mac_defaults)
|
mac_config: List[Configuration] = []
|
||||||
phy_library: Optional[str] = None
|
phy_library: Optional[str] = None
|
||||||
phy_xml: str = "/usr/share/emane/manifest/emanephy.xml"
|
phy_xml: str = "emanephy.xml"
|
||||||
phy_defaults: Dict[str, str] = {
|
phy_defaults: Dict[str, str] = {
|
||||||
"subid": "1",
|
"subid": "1",
|
||||||
"propagationmodel": "2ray",
|
"propagationmodel": "2ray",
|
||||||
"noisemode": "none",
|
"noisemode": "none",
|
||||||
}
|
}
|
||||||
phy_config: List[Configuration] = emanemanifest.parse(phy_xml, phy_defaults)
|
phy_config: List[Configuration] = []
|
||||||
config_ignore: Set[str] = set()
|
config_ignore: Set[str] = set()
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def load(cls, emane_prefix: Path) -> None:
|
||||||
|
"""
|
||||||
|
Called after being loaded within the EmaneManager. Provides configured
|
||||||
|
emane_prefix for parsing xml files.
|
||||||
|
|
||||||
|
:param emane_prefix: configured emane prefix path
|
||||||
|
:return: nothing
|
||||||
|
"""
|
||||||
|
manifest_path = "share/emane/manifest"
|
||||||
|
# load mac configuration
|
||||||
|
mac_xml_path = emane_prefix / manifest_path / cls.mac_xml
|
||||||
|
cls.mac_config = emanemanifest.parse(mac_xml_path, cls.mac_defaults)
|
||||||
|
# load phy configuration
|
||||||
|
phy_xml_path = emane_prefix / manifest_path / cls.phy_xml
|
||||||
|
cls.phy_config = emanemanifest.parse(phy_xml_path, cls.phy_defaults)
|
||||||
|
|
Some files were not shown because too many files have changed in this diff Show more
Loading…
Reference in a new issue