617 lines
22 KiB
Python
617 lines
22 KiB
Python
"""
|
|
Incorporate grpc into python tkinter GUI
|
|
"""
|
|
import logging
|
|
import os
|
|
|
|
from core.api.grpc import client, core_pb2
|
|
from coretk.dialogs.sessions import SessionsDialog
|
|
from coretk.emaneodelnodeconfig import EmaneModelNodeConfig
|
|
from coretk.interface import InterfaceManager
|
|
from coretk.mobilitynodeconfig import MobilityNodeConfig
|
|
from coretk.nodeutils import NodeDraw
|
|
from coretk.servicenodeconfig import ServiceNodeConfig
|
|
from coretk.wlannodeconfig import WlanNodeConfig
|
|
|
|
DEFAULT_NODES = {"router", "host", "PC", "mdr", "prouter"}
|
|
OBSERVERS = {
|
|
"processes": "ps",
|
|
"ifconfig": "ifconfig",
|
|
"IPV4 Routes": "ip -4 ro",
|
|
"IPV6 Routes": "ip -6 ro",
|
|
"Listening sockets": "netstat -tuwnl",
|
|
"IPv4 MFC entries": "ip -4 mroute show",
|
|
"IPv6 MFC entries": "ip -6 mroute show",
|
|
"firewall rules": "iptables -L",
|
|
"IPSec policies": "setkey -DP",
|
|
}
|
|
|
|
|
|
class CoreServer:
|
|
def __init__(self, name, address, port):
|
|
self.name = name
|
|
self.address = address
|
|
self.port = port
|
|
|
|
|
|
class Observer:
|
|
def __init__(self, name, cmd):
|
|
self.name = name
|
|
self.cmd = cmd
|
|
|
|
|
|
class CoreClient:
|
|
def __init__(self, app):
|
|
"""
|
|
Create a CoreGrpc instance
|
|
"""
|
|
self.client = client.CoreGrpcClient()
|
|
self.session_id = None
|
|
self.node_ids = []
|
|
self.app = app
|
|
self.master = app.master
|
|
self.interface_helper = None
|
|
self.services = {}
|
|
self.observer = None
|
|
|
|
# loaded configuration data
|
|
self.servers = {}
|
|
self.custom_nodes = {}
|
|
self.custom_observers = {}
|
|
self.read_config()
|
|
|
|
# data for managing the current session
|
|
self.canvas_nodes = {}
|
|
self.interface_to_edge = {}
|
|
self.state = None
|
|
self.links = {}
|
|
self.hooks = {}
|
|
self.id = 1
|
|
self.reusable = []
|
|
self.preexisting = set()
|
|
self.interfaces_manager = InterfaceManager()
|
|
self.wlanconfig_management = WlanNodeConfig()
|
|
self.mobilityconfig_management = MobilityNodeConfig()
|
|
self.emaneconfig_management = EmaneModelNodeConfig(app)
|
|
self.emane_config = None
|
|
self.serviceconfig_manager = ServiceNodeConfig(app)
|
|
|
|
def set_observer(self, value):
|
|
self.observer = value
|
|
|
|
def read_config(self):
|
|
# read distributed server
|
|
for config in self.app.config.get("servers", []):
|
|
server = CoreServer(config["name"], config["address"], config["port"])
|
|
self.servers[server.name] = server
|
|
|
|
# read custom nodes
|
|
for config in self.app.config.get("nodes", []):
|
|
name = config["name"]
|
|
image_file = config["image"]
|
|
services = set(config["services"])
|
|
node_draw = NodeDraw.from_custom(name, image_file, services)
|
|
self.custom_nodes[name] = node_draw
|
|
|
|
# read observers
|
|
for config in self.app.config.get("observers", []):
|
|
observer = Observer(config["name"], config["cmd"])
|
|
self.custom_observers[observer.name] = observer
|
|
|
|
def handle_events(self, event):
|
|
logging.info("event: %s", event)
|
|
if event.HasField("link_event"):
|
|
self.app.canvas.wireless_draw.hangle_link_event(event.link_event)
|
|
elif event.HasField("session_event"):
|
|
if event.session_event.event <= core_pb2.SessionState.SHUTDOWN:
|
|
self.state = event.session_event.event
|
|
|
|
def handle_throughputs(self, event):
|
|
interface_throughputs = event.interface_throughputs
|
|
for i in interface_throughputs:
|
|
print("")
|
|
return
|
|
throughputs_belong_to_session = []
|
|
for if_tp in interface_throughputs:
|
|
if if_tp.node_id in self.node_ids:
|
|
throughputs_belong_to_session.append(if_tp)
|
|
self.throughput_draw.process_grpc_throughput_event(
|
|
throughputs_belong_to_session
|
|
)
|
|
|
|
def join_session(self, session_id):
|
|
# update session and title
|
|
self.session_id = session_id
|
|
self.master.title(f"CORE Session({self.session_id})")
|
|
|
|
# clear session data
|
|
self.reusable.clear()
|
|
self.preexisting.clear()
|
|
self.canvas_nodes.clear()
|
|
self.links.clear()
|
|
self.hooks.clear()
|
|
self.wlanconfig_management.configurations.clear()
|
|
self.mobilityconfig_management.configurations.clear()
|
|
self.emane_config = None
|
|
|
|
# get session data
|
|
response = self.client.get_session(self.session_id)
|
|
logging.info("joining session(%s): %s", self.session_id, response)
|
|
session = response.session
|
|
self.state = session.state
|
|
self.client.events(self.session_id, self.handle_events)
|
|
|
|
# get hooks
|
|
response = self.client.get_hooks(self.session_id)
|
|
logging.info("joined session hooks: %s", response)
|
|
for hook in response.hooks:
|
|
self.hooks[hook.file] = hook
|
|
|
|
# get wlan configs
|
|
for node in session.nodes:
|
|
if node.type == core_pb2.NodeType.WIRELESS_LAN:
|
|
response = self.client.get_wlan_config(self.session_id, node.id)
|
|
logging.debug("wlan config(%s): %s", node.id, response)
|
|
node_config = response.config
|
|
config = {x: node_config[x].value for x in node_config}
|
|
self.wlanconfig_management.configurations[node.id] = config
|
|
|
|
# get mobility configs
|
|
response = self.client.get_mobility_configs(self.session_id)
|
|
logging.debug("mobility configs: %s", response)
|
|
for node_id in response.configs:
|
|
node_config = response.configs[node_id].config
|
|
config = {x: node_config[x].value for x in node_config}
|
|
self.mobilityconfig_management.configurations[node_id] = config
|
|
|
|
# get emane config
|
|
response = self.client.get_emane_config(self.session_id)
|
|
logging.debug("emane config: %s", response)
|
|
self.emane_config = response.config
|
|
|
|
# get emane model config
|
|
|
|
# determine next node id and reusable nodes
|
|
max_id = 1
|
|
for node in session.nodes:
|
|
if node.id > max_id:
|
|
max_id = node.id
|
|
self.preexisting.add(node.id)
|
|
self.id = max_id
|
|
for i in range(1, self.id):
|
|
if i not in self.preexisting:
|
|
self.reusable.append(i)
|
|
|
|
# draw session
|
|
self.app.canvas.reset_and_redraw(session)
|
|
|
|
# draw tool bar appropritate with session state
|
|
if self.is_runtime():
|
|
self.app.toolbar.runtime_frame.tkraise()
|
|
else:
|
|
self.app.toolbar.design_frame.tkraise()
|
|
|
|
def is_runtime(self):
|
|
return self.state == core_pb2.SessionState.RUNTIME
|
|
|
|
def create_new_session(self):
|
|
"""
|
|
Create a new session
|
|
|
|
:return: nothing
|
|
"""
|
|
response = self.client.create_session()
|
|
logging.info("created session: %s", response)
|
|
self.join_session(response.session_id)
|
|
|
|
def delete_session(self, custom_sid=None):
|
|
if custom_sid is None:
|
|
sid = self.session_id
|
|
else:
|
|
sid = custom_sid
|
|
response = self.client.delete_session(sid)
|
|
logging.info("Deleted session result: %s", response)
|
|
|
|
def shutdown_session(self, custom_sid=None):
|
|
if custom_sid is None:
|
|
sid = self.session_id
|
|
else:
|
|
sid = custom_sid
|
|
s = self.client.get_session(sid).session
|
|
# delete links and nodes from running session
|
|
if s.state == core_pb2.SessionState.RUNTIME:
|
|
self.client.set_session_state(
|
|
self.session_id, core_pb2.SessionState.DATACOLLECT
|
|
)
|
|
self.delete_links(sid)
|
|
self.delete_nodes(sid)
|
|
self.delete_session(sid)
|
|
|
|
def set_up(self):
|
|
"""
|
|
Query sessions, if there exist any, prompt whether to join one
|
|
|
|
:return: existing sessions
|
|
"""
|
|
self.client.connect()
|
|
|
|
# get service information
|
|
response = self.client.get_services()
|
|
for service in response.services:
|
|
group_services = self.services.setdefault(service.group, [])
|
|
group_services.append(service)
|
|
|
|
# if there are no sessions, create a new session, else join a session
|
|
response = self.client.get_sessions()
|
|
logging.info("current sessions: %s", response)
|
|
sessions = response.sessions
|
|
if len(sessions) == 0:
|
|
self.create_new_session()
|
|
else:
|
|
dialog = SessionsDialog(self.app, self.app)
|
|
dialog.show()
|
|
|
|
def get_session_state(self):
|
|
response = self.client.get_session(self.session_id)
|
|
# logging.info("get session: %s", response)
|
|
return response.session.state
|
|
|
|
def edit_node(self, node_id, x, y):
|
|
position = core_pb2.Position(x=x, y=y)
|
|
response = self.client.edit_node(self.session_id, node_id, position)
|
|
logging.info("updated node id %s: %s", node_id, response)
|
|
|
|
def delete_nodes(self, delete_session=None):
|
|
if delete_session is None:
|
|
sid = self.session_id
|
|
else:
|
|
sid = delete_session
|
|
for node in self.client.get_session(sid).session.nodes:
|
|
response = self.client.delete_node(self.session_id, node.id)
|
|
logging.info("delete nodes %s", response)
|
|
|
|
def delete_links(self, delete_session=None):
|
|
if delete_session is None:
|
|
sid = self.session_id
|
|
else:
|
|
sid = delete_session
|
|
|
|
for link in self.client.get_session(sid).session.links:
|
|
response = self.client.delete_link(
|
|
self.session_id,
|
|
link.node_one_id,
|
|
link.node_two_id,
|
|
link.interface_one.id,
|
|
link.interface_two.id,
|
|
)
|
|
logging.info("delete links %s", response)
|
|
|
|
def start_session(self):
|
|
nodes = [x.core_node for x in self.canvas_nodes.values()]
|
|
links = list(self.links.values())
|
|
wlan_configs = self.get_wlan_configs_proto()
|
|
mobility_configs = self.get_mobility_configs_proto()
|
|
emane_model_configs = self.get_emane_model_configs_proto()
|
|
hooks = list(self.hooks.values())
|
|
if self.emane_config:
|
|
emane_config = {x: self.emane_config[x].value for x in self.emane_config}
|
|
else:
|
|
emane_config = None
|
|
response = self.client.start_session(
|
|
self.session_id,
|
|
nodes,
|
|
links,
|
|
hooks=hooks,
|
|
wlan_configs=wlan_configs,
|
|
emane_config=emane_config,
|
|
emane_model_configs=emane_model_configs,
|
|
mobility_configs=mobility_configs,
|
|
)
|
|
logging.debug("Start session %s, result: %s", self.session_id, response.result)
|
|
|
|
response = self.client.get_service_defaults(self.session_id)
|
|
for default in response.defaults:
|
|
print(default.node_type)
|
|
print(default.services)
|
|
response = self.client.get_node_service(self.session_id, 5, "FTP")
|
|
print(response)
|
|
|
|
def stop_session(self):
|
|
response = self.client.stop_session(session_id=self.session_id)
|
|
logging.debug("coregrpc.py Stop session, result: %s", response.result)
|
|
|
|
def launch_terminal(self, node_id):
|
|
response = self.client.get_node_terminal(self.session_id, node_id)
|
|
logging.info("get terminal %s", response.terminal)
|
|
os.system("xterm -e %s &" % response.terminal)
|
|
|
|
def save_xml(self, file_path):
|
|
"""
|
|
Save core session as to an xml file
|
|
|
|
:param str file_path: file path that user pick
|
|
:return: nothing
|
|
"""
|
|
response = self.client.save_xml(self.session_id, file_path)
|
|
logging.info("coregrpc.py save xml %s", response)
|
|
self.client.events(self.session_id, self.handle_events)
|
|
|
|
def open_xml(self, file_path):
|
|
"""
|
|
Open core xml
|
|
|
|
:param str file_path: file to open
|
|
:return: session id
|
|
"""
|
|
response = self.client.open_xml(file_path)
|
|
logging.debug("open xml: %s", response)
|
|
self.join_session(response.session_id)
|
|
|
|
def get_node_service(self, node_id, service_name):
|
|
response = self.client.get_node_service(self.session_id, node_id, service_name)
|
|
logging.debug("get node service %s", response)
|
|
return response.service
|
|
|
|
def set_node_service(self, node_id, service_name, startups, validations, shutdowns):
|
|
response = self.client.set_node_service(
|
|
self.session_id, node_id, service_name, startups, validations, shutdowns
|
|
)
|
|
logging.debug("set node service %s", response)
|
|
|
|
def get_node_service_file(self, node_id, service_name, file_name):
|
|
response = self.client.get_node_service_file(
|
|
self.session_id, node_id, service_name, file_name
|
|
)
|
|
logging.debug("get service file %s", response)
|
|
return response.data
|
|
|
|
def create_nodes_and_links(self):
|
|
node_protos = [x.core_node for x in self.canvas_nodes.values()]
|
|
link_protos = list(self.links.values())
|
|
self.client.set_session_state(self.session_id, core_pb2.SessionState.DEFINITION)
|
|
for node_proto in node_protos:
|
|
response = self.client.add_node(self.session_id, node_proto)
|
|
logging.debug("create node: %s", response)
|
|
for link_proto in link_protos:
|
|
response = self.client.add_link(
|
|
self.session_id,
|
|
link_proto.node_one_id,
|
|
link_proto.node_two_id,
|
|
link_proto.interface_one,
|
|
link_proto.interface_two,
|
|
link_proto.options,
|
|
)
|
|
logging.debug("create link: %s", response)
|
|
|
|
def close(self):
|
|
"""
|
|
Clean ups when done using grpc
|
|
|
|
:return: nothing
|
|
"""
|
|
logging.debug("Close grpc")
|
|
self.client.close()
|
|
|
|
def get_id(self):
|
|
"""
|
|
Get the next node id as well as update id status and reusable ids
|
|
|
|
:rtype: int
|
|
:return: the next id to be used
|
|
"""
|
|
if len(self.reusable) == 0:
|
|
new_id = self.id
|
|
self.id = self.id + 1
|
|
return new_id
|
|
else:
|
|
return self.reusable.pop(0)
|
|
|
|
def is_model_node(self, name):
|
|
return name in DEFAULT_NODES or name in self.custom_nodes
|
|
|
|
def create_node(self, x, y, node_type, model):
|
|
"""
|
|
Add node, with information filled in, to grpc manager
|
|
|
|
:param int x: x coord
|
|
:param int y: y coord
|
|
:param core_pb2.NodeType node_type: node type
|
|
:param str model: node model
|
|
:return: nothing
|
|
"""
|
|
node_id = self.get_id()
|
|
position = core_pb2.Position(x=x, y=y)
|
|
node = core_pb2.Node(
|
|
id=node_id,
|
|
type=node_type,
|
|
name=f"n{node_id}",
|
|
model=model,
|
|
position=position,
|
|
)
|
|
|
|
# set default configuration for wireless node
|
|
self.wlanconfig_management.set_default_config(node_type, node_id)
|
|
self.mobilityconfig_management.set_default_configuration(node_type, node_id)
|
|
|
|
# set default emane configuration for emane node
|
|
if node_type == core_pb2.NodeType.EMANE:
|
|
self.emaneconfig_management.set_default_config(node_id)
|
|
|
|
# set default service configurations
|
|
# TODO: need to deal with this and custom node cases
|
|
if node_type == core_pb2.NodeType.DEFAULT:
|
|
self.serviceconfig_manager.node_default_services_configuration(
|
|
node_id=node_id, node_model=model
|
|
)
|
|
|
|
logging.debug(
|
|
"adding node to core session: %s, coords: (%s, %s), name: %s",
|
|
self.session_id,
|
|
x,
|
|
y,
|
|
node.name,
|
|
)
|
|
return node
|
|
|
|
def delete_wanted_graph_nodes(self, node_ids, edge_tokens):
|
|
"""
|
|
remove the nodes selected by the user and anything related to that node
|
|
such as link, configurations, interfaces
|
|
|
|
:param list[int] node_ids: list of nodes to delete
|
|
:param list edge_tokens: list of edges to delete
|
|
:return: nothing
|
|
"""
|
|
# delete the nodes
|
|
for node_id in node_ids:
|
|
try:
|
|
del self.canvas_nodes[node_id]
|
|
self.reusable.append(node_id)
|
|
except KeyError:
|
|
logging.error("invalid canvas id: %s", node_id)
|
|
self.reusable.sort()
|
|
|
|
# delete the edges and interfaces
|
|
node_interface_pairs = []
|
|
for i in edge_tokens:
|
|
try:
|
|
link = self.links.pop(i)
|
|
if link.interface_one is not None:
|
|
node_interface_pairs.append(
|
|
(link.node_one_id, link.interface_one.id)
|
|
)
|
|
if link.interface_two is not None:
|
|
node_interface_pairs.append(
|
|
(link.node_two_id, link.interface_two.id)
|
|
)
|
|
except KeyError:
|
|
logging.error("coreclient.py invalid edge token ")
|
|
|
|
# delete global emane config if there no longer exist any emane cloud
|
|
# TODO: should not need to worry about this
|
|
node_types = [x.core_node.type for x in self.canvas_nodes.values()]
|
|
if core_pb2.NodeType.EMANE not in node_types:
|
|
self.emane_config = None
|
|
|
|
# delete any mobility configuration, wlan configuration
|
|
for i in node_ids:
|
|
if i in self.mobilityconfig_management.configurations:
|
|
self.mobilityconfig_management.configurations.pop(i)
|
|
if i in self.wlanconfig_management.configurations:
|
|
self.wlanconfig_management.configurations.pop(i)
|
|
|
|
# delete emane configurations
|
|
for i in node_interface_pairs:
|
|
if i in self.emaneconfig_management.configurations:
|
|
self.emaneconfig_management.configurations.pop(i)
|
|
for i in node_ids:
|
|
if tuple([i, None]) in self.emaneconfig_management.configurations:
|
|
self.emaneconfig_management.configurations.pop(tuple([i, None]))
|
|
|
|
def create_interface(self, canvas_node):
|
|
interface = None
|
|
core_node = canvas_node.core_node
|
|
if self.is_model_node(core_node.model):
|
|
ifid = len(canvas_node.interfaces)
|
|
name = f"eth{ifid}"
|
|
interface = core_pb2.Interface(
|
|
id=ifid,
|
|
name=name,
|
|
ip4=str(self.interfaces_manager.get_address()),
|
|
ip4mask=24,
|
|
)
|
|
canvas_node.interfaces.append(interface)
|
|
logging.debug(
|
|
"create node(%s) interface IPv4: %s, name: %s",
|
|
core_node.name,
|
|
interface.ip4,
|
|
interface.name,
|
|
)
|
|
return interface
|
|
|
|
def create_link(self, token, canvas_node_one, canvas_node_two):
|
|
"""
|
|
Create core link for a pair of canvas nodes, with token referencing
|
|
the canvas edge.
|
|
|
|
:param tuple(int, int) token: edge's identification in the canvas
|
|
:param canvas_node_one: canvas node one
|
|
:param canvas_node_two: canvas node two
|
|
|
|
:return: nothing
|
|
"""
|
|
node_one = canvas_node_one.core_node
|
|
node_two = canvas_node_two.core_node
|
|
|
|
# create interfaces
|
|
self.interfaces_manager.new_subnet()
|
|
interface_one = self.create_interface(canvas_node_one)
|
|
if interface_one is not None:
|
|
self.interface_to_edge[(node_one.id, interface_one.id)] = token
|
|
interface_two = self.create_interface(canvas_node_two)
|
|
if interface_two is not None:
|
|
self.interface_to_edge[(node_two.id, interface_two.id)] = token
|
|
|
|
# emane setup
|
|
# TODO: determine if this is needed
|
|
if (
|
|
node_one.type == core_pb2.NodeType.EMANE
|
|
and node_two.type == core_pb2.NodeType.DEFAULT
|
|
):
|
|
if node_two.model == "mdr":
|
|
self.emaneconfig_management.set_default_for_mdr(
|
|
node_one.node_id, node_two.node_id, interface_two.id
|
|
)
|
|
elif (
|
|
node_two.type == core_pb2.NodeType.EMANE
|
|
and node_one.type == core_pb2.NodeType.DEFAULT
|
|
):
|
|
if node_one.model == "mdr":
|
|
self.emaneconfig_management.set_default_for_mdr(
|
|
node_two.node_id, node_one.node_id, interface_one.id
|
|
)
|
|
|
|
link = core_pb2.Link(
|
|
type=core_pb2.LinkType.WIRED,
|
|
node_one_id=node_one.id,
|
|
node_two_id=node_two.id,
|
|
interface_one=interface_one,
|
|
interface_two=interface_two,
|
|
)
|
|
self.links[token] = link
|
|
return link
|
|
|
|
def get_wlan_configs_proto(self):
|
|
configs = []
|
|
wlan_configs = self.wlanconfig_management.configurations
|
|
for node_id in wlan_configs:
|
|
config = wlan_configs[node_id]
|
|
config_proto = core_pb2.WlanConfig(node_id=node_id, config=config)
|
|
configs.append(config_proto)
|
|
return configs
|
|
|
|
def get_mobility_configs_proto(self):
|
|
configs = []
|
|
mobility_configs = self.mobilityconfig_management.configurations
|
|
for node_id in mobility_configs:
|
|
config = mobility_configs[node_id]
|
|
config_proto = core_pb2.MobilityConfig(node_id=node_id, config=config)
|
|
configs.append(config_proto)
|
|
return configs
|
|
|
|
def get_emane_model_configs_proto(self):
|
|
configs = []
|
|
emane_configs = self.emaneconfig_management.configurations
|
|
for key, value in emane_configs.items():
|
|
node_id, interface_id = key
|
|
model, options = value
|
|
config = {x: options[x].value for x in options}
|
|
config_proto = core_pb2.EmaneModelConfig(
|
|
node_id=node_id, interface_id=interface_id, model=model, config=config
|
|
)
|
|
configs.append(config_proto)
|
|
return configs
|
|
|
|
def run(self, node_id):
|
|
logging.info("running node(%s) cmd: %s", node_id, self.observer)
|
|
return self.client.node_command(self.session_id, node_id, self.observer).output
|