2015-01-14 01:26:32 +00:00
# -*- coding: utf-8 -*-
#
# Copyright (C) 2015 GNS3 Technologies Inc.
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
2016-11-13 09:28:14 +00:00
import sys
2015-02-16 05:13:24 +00:00
import os
2016-05-31 19:08:41 +00:00
import stat
2015-01-14 17:52:02 +00:00
import logging
2015-02-16 05:13:24 +00:00
import aiohttp
import shutil
import asyncio
2015-03-17 15:31:45 +00:00
import tempfile
2015-10-12 21:57:37 +00:00
import psutil
import platform
2018-03-12 06:38:50 +00:00
import re
2015-02-16 05:13:24 +00:00
2020-01-31 09:31:27 +00:00
from aiohttp . web import WebSocketResponse
2016-11-13 09:28:14 +00:00
from gns3server . utils . interfaces import interfaces
2016-06-23 22:56:06 +00:00
from . . compute . port_manager import PortManager
2018-08-25 07:10:47 +00:00
from . . utils . asyncio import wait_run_in_executor , locking
2016-11-08 18:44:12 +00:00
from . . utils . asyncio . telnet_server import AsyncioTelnetServer
2015-09-13 20:52:25 +00:00
from . . ubridge . hypervisor import Hypervisor
2016-05-31 04:07:37 +00:00
from . . ubridge . ubridge_error import UbridgeError
2016-06-23 22:56:06 +00:00
from . nios . nio_udp import NIOUDP
2016-06-07 13:34:04 +00:00
from . error import NodeError
2015-02-16 05:13:24 +00:00
2015-09-13 20:52:25 +00:00
2015-01-14 17:52:02 +00:00
log = logging . getLogger ( __name__ )
2015-01-14 01:26:32 +00:00
2015-01-15 23:50:36 +00:00
2016-05-11 17:35:36 +00:00
class BaseNode :
2015-01-18 22:41:53 +00:00
2015-02-19 10:33:25 +00:00
"""
2016-05-11 17:35:36 +00:00
Base node implementation .
2015-02-19 10:33:25 +00:00
2016-05-11 17:35:36 +00:00
: param name : name of this node
: param node_id : Node instance identifier
2015-02-19 10:33:25 +00:00
: param project : Project instance
2016-05-11 17:35:36 +00:00
: param manager : parent node manager
2015-02-19 10:33:25 +00:00
: param console : TCP console port
2016-02-29 09:38:30 +00:00
: param aux : TCP aux console port
: param allocate_aux : Boolean if true will allocate an aux console port
2016-10-24 19:39:35 +00:00
: param linked_clone : The node base image is duplicate / overlay ( Each node data are independent )
2016-11-08 18:44:12 +00:00
: param wrap_console : The console is wrapped using AsyncioTelnetServer
2015-02-19 10:33:25 +00:00
"""
2016-11-08 18:44:12 +00:00
def __init__ ( self , name , node_id , project , manager , console = None , console_type = " telnet " , aux = None , allocate_aux = False , linked_clone = True , wrap_console = False ) :
2015-01-16 00:43:06 +00:00
2015-01-14 01:26:32 +00:00
self . _name = name
2015-12-01 09:54:51 +00:00
self . _usage = " "
2016-05-11 17:35:36 +00:00
self . _id = node_id
2016-10-24 19:39:35 +00:00
self . _linked_clone = linked_clone
2015-01-20 11:46:15 +00:00
self . _project = project
2015-01-19 10:22:24 +00:00
self . _manager = manager
2015-02-19 10:33:25 +00:00
self . _console = console
2016-02-29 09:38:30 +00:00
self . _aux = aux
2015-07-03 22:06:25 +00:00
self . _console_type = console_type
2015-03-17 15:31:45 +00:00
self . _temporary_directory = None
2015-07-22 04:58:28 +00:00
self . _hw_virtualization = False
2015-09-13 20:52:25 +00:00
self . _ubridge_hypervisor = None
2016-02-29 09:38:30 +00:00
self . _closed = False
2016-05-11 17:35:36 +00:00
self . _node_status = " stopped "
2016-02-02 17:25:17 +00:00
self . _command_line = " "
2016-03-01 14:01:45 +00:00
self . _allocate_aux = allocate_aux
2016-11-08 18:44:12 +00:00
self . _wrap_console = wrap_console
self . _wrapper_telnet_server = None
2022-12-31 01:43:17 +00:00
self . _wrap_console_reader = None
self . _wrap_console_writer = None
2018-03-23 08:44:16 +00:00
self . _internal_console_port = None
2018-04-02 15:27:12 +00:00
self . _custom_adapters = [ ]
2018-11-19 17:22:16 +00:00
self . _ubridge_require_privileged_access = False
2015-02-19 10:33:25 +00:00
if self . _console is not None :
2015-12-07 11:26:46 +00:00
if console_type == " vnc " :
2020-12-03 07:34:42 +00:00
vnc_console_start_port_range , vnc_console_end_port_range = self . _get_vnc_console_port_range ( )
self . _console = self . _manager . port_manager . reserve_tcp_port (
self . _console ,
self . _project ,
port_range_start = vnc_console_start_port_range ,
port_range_end = vnc_console_end_port_range ,
)
2018-03-24 11:11:21 +00:00
elif console_type == " none " :
self . _console = None
2015-12-07 11:26:46 +00:00
else :
self . _console = self . _manager . port_manager . reserve_tcp_port ( self . _console , self . _project )
2016-02-29 09:38:30 +00:00
2016-05-18 09:23:45 +00:00
# We need to allocate aux before giving a random console port
2016-02-29 09:38:30 +00:00
if self . _aux is not None :
self . _aux = self . _manager . port_manager . reserve_tcp_port ( self . _aux , self . _project )
2018-03-23 14:29:39 +00:00
if self . _console is None :
if console_type == " vnc " :
2020-12-03 07:34:42 +00:00
vnc_console_start_port_range , vnc_console_end_port_range = self . _get_vnc_console_port_range ( )
self . _console = self . _manager . port_manager . get_free_tcp_port (
self . _project ,
port_range_start = vnc_console_start_port_range ,
port_range_end = vnc_console_end_port_range )
2018-03-23 14:29:39 +00:00
elif console_type != " none " :
self . _console = self . _manager . port_manager . get_free_tcp_port ( self . _project )
2016-11-08 18:44:12 +00:00
if self . _wrap_console :
2016-12-01 10:47:05 +00:00
self . _internal_console_port = self . _manager . port_manager . get_free_tcp_port ( self . _project )
2016-11-08 18:44:12 +00:00
2016-02-29 09:38:30 +00:00
if self . _aux is None and allocate_aux :
self . _aux = self . _manager . port_manager . get_free_tcp_port ( self . _project )
2015-04-08 17:17:34 +00:00
log . debug ( " {module} : {name} [ {id} ] initialized. Console port {console} " . format ( module = self . manager . module_name ,
name = self . name ,
id = self . id ,
console = self . _console ) )
2015-01-21 22:21:15 +00:00
2015-01-22 10:49:22 +00:00
def __del__ ( self ) :
2015-01-23 02:06:17 +00:00
2016-11-03 13:21:28 +00:00
if hasattr ( self , " _temporary_directory " ) and self . _temporary_directory is not None :
2015-03-17 15:31:45 +00:00
if os . path . exists ( self . _temporary_directory ) :
2015-04-08 17:17:34 +00:00
shutil . rmtree ( self . _temporary_directory , ignore_errors = True )
2015-01-19 10:22:24 +00:00
2016-10-24 19:39:35 +00:00
@property
def linked_clone ( self ) :
return self . _linked_clone
@linked_clone.setter
def linked_clone ( self , val ) :
self . _linked_clone = val
2018-04-02 15:27:12 +00:00
@property
def custom_adapters ( self ) :
return self . _custom_adapters
@custom_adapters.setter
def custom_adapters ( self , val ) :
self . _custom_adapters = val
2015-03-04 15:01:56 +00:00
@property
def status ( self ) :
2016-06-24 21:04:58 +00:00
"""
Returns current node status
"""
2015-03-04 15:01:56 +00:00
2016-05-11 17:35:36 +00:00
return self . _node_status
2015-03-04 15:01:56 +00:00
@status.setter
def status ( self , status ) :
2016-05-11 17:35:36 +00:00
self . _node_status = status
2016-05-18 09:00:35 +00:00
self . updated ( )
def updated ( self ) :
"""
2016-06-24 21:04:58 +00:00
Sends an updated event
2016-05-18 09:00:35 +00:00
"""
2016-05-16 19:12:32 +00:00
self . project . emit ( " node.updated " , self )
2015-03-04 15:01:56 +00:00
2016-02-02 17:25:17 +00:00
@property
def command_line ( self ) :
2016-06-24 21:04:58 +00:00
"""
Returns command used to start the node
"""
2016-02-02 17:25:17 +00:00
return self . _command_line
@command_line.setter
def command_line ( self , command_line ) :
self . _command_line = command_line
2015-01-20 11:46:15 +00:00
@property
def project ( self ) :
2015-01-21 02:02:22 +00:00
"""
2016-05-11 17:35:36 +00:00
Returns the node current project .
2015-01-21 02:02:22 +00:00
: returns : Project instance .
"""
2015-01-20 11:46:15 +00:00
return self . _project
2015-01-14 01:26:32 +00:00
@property
2015-01-20 01:30:57 +00:00
def name ( self ) :
2015-01-14 01:26:32 +00:00
"""
2016-05-11 17:35:36 +00:00
Returns the name for this node .
2015-01-14 01:26:32 +00:00
2015-01-20 01:30:57 +00:00
: returns : name
2015-01-14 01:26:32 +00:00
"""
2015-01-20 01:30:57 +00:00
return self . _name
@name.setter
def name ( self , new_name ) :
"""
2016-05-11 17:35:36 +00:00
Sets the name of this node .
2015-01-20 01:30:57 +00:00
: param new_name : name
"""
2015-02-04 20:48:29 +00:00
log . info ( " {module} : {name} [ {id} ] renamed to {new_name} " . format ( module = self . manager . module_name ,
name = self . name ,
id = self . id ,
new_name = new_name ) )
2015-01-20 01:30:57 +00:00
self . _name = new_name
2015-01-14 01:26:32 +00:00
2015-12-01 09:54:51 +00:00
@property
def usage ( self ) :
"""
2016-05-11 17:35:36 +00:00
Returns the usage for this node .
2015-12-01 09:54:51 +00:00
: returns : usage
"""
return self . _usage
@usage.setter
def usage ( self , new_usage ) :
"""
2016-05-11 17:35:36 +00:00
Sets the usage of this node .
2015-12-01 09:54:51 +00:00
: param new_usage : usage
"""
self . _usage = new_usage
2015-01-14 01:26:32 +00:00
@property
2015-02-04 20:48:29 +00:00
def id ( self ) :
2015-01-14 01:26:32 +00:00
"""
2016-05-11 17:35:36 +00:00
Returns the ID for this node .
2015-01-14 01:26:32 +00:00
2016-05-11 17:35:36 +00:00
: returns : Node identifier ( string )
2015-01-14 01:26:32 +00:00
"""
2015-02-04 20:48:29 +00:00
return self . _id
2015-01-14 01:26:32 +00:00
2015-01-20 01:30:57 +00:00
@property
def manager ( self ) :
2015-01-18 22:41:53 +00:00
"""
2016-05-11 17:35:36 +00:00
Returns the manager for this node .
2015-01-20 01:30:57 +00:00
: returns : instance of manager
2015-01-18 22:41:53 +00:00
"""
2015-01-20 01:30:57 +00:00
return self . _manager
2015-01-14 01:26:32 +00:00
2015-01-20 13:31:47 +00:00
@property
def working_dir ( self ) :
"""
2016-05-11 17:35:36 +00:00
Return the node working directory
2015-01-20 13:31:47 +00:00
"""
2016-05-11 17:35:36 +00:00
return self . _project . node_working_directory ( self )
2015-01-20 13:31:47 +00:00
2017-10-02 08:41:57 +00:00
@property
def working_path ( self ) :
"""
Return the node working path . Doesn ' t create structure of directories when not present.
"""
return self . _project . node_working_path ( self )
2015-03-17 15:31:45 +00:00
@property
def temporary_directory ( self ) :
if self . _temporary_directory is None :
2015-03-17 21:18:55 +00:00
try :
self . _temporary_directory = tempfile . mkdtemp ( )
except OSError as e :
2016-05-11 17:35:36 +00:00
raise NodeError ( " Can ' t create temporary directory: {} " . format ( e ) )
2015-03-17 15:31:45 +00:00
return self . _temporary_directory
2015-01-20 01:30:57 +00:00
def create ( self ) :
2015-01-18 22:41:53 +00:00
"""
2016-05-11 17:35:36 +00:00
Creates the node .
2015-01-18 22:41:53 +00:00
"""
2015-01-14 01:26:32 +00:00
2015-02-04 20:48:29 +00:00
log . info ( " {module} : {name} [ {id} ] created " . format ( module = self . manager . module_name ,
name = self . name ,
id = self . id ) )
2015-01-14 01:26:32 +00:00
2018-10-15 10:05:49 +00:00
async def delete ( self ) :
2015-02-16 05:13:24 +00:00
"""
2016-05-11 17:35:36 +00:00
Delete the node ( including all its files ) .
2015-02-16 05:13:24 +00:00
"""
2018-03-15 07:17:39 +00:00
2016-05-31 19:08:41 +00:00
def set_rw ( operation , name , exc ) :
os . chmod ( name , stat . S_IWRITE )
2015-02-16 05:13:24 +00:00
2016-05-11 17:35:36 +00:00
directory = self . project . node_working_directory ( self )
2015-02-16 05:13:24 +00:00
if os . path . exists ( directory ) :
try :
2018-10-15 10:05:49 +00:00
await wait_run_in_executor ( shutil . rmtree , directory , onerror = set_rw )
2015-02-16 05:13:24 +00:00
except OSError as e :
2016-05-11 17:35:36 +00:00
raise aiohttp . web . HTTPInternalServerError ( text = " Could not delete the node working directory: {} " . format ( e ) )
2015-02-16 05:13:24 +00:00
2015-01-15 23:50:36 +00:00
def start ( self ) :
2015-01-14 17:52:02 +00:00
"""
2016-05-11 17:35:36 +00:00
Starts the node process .
2015-01-14 17:52:02 +00:00
"""
2015-01-18 22:41:53 +00:00
2015-01-14 17:52:02 +00:00
raise NotImplementedError
2018-10-15 10:05:49 +00:00
async def stop ( self ) :
2015-01-19 12:47:20 +00:00
"""
2016-11-08 18:44:12 +00:00
Stop the node process .
2015-01-14 01:26:32 +00:00
"""
2018-03-15 07:17:39 +00:00
2018-10-15 10:05:49 +00:00
await self . stop_wrap_console ( )
2016-11-08 18:44:12 +00:00
self . status = " stopped "
2015-01-22 10:49:22 +00:00
2016-05-14 01:26:50 +00:00
def suspend ( self ) :
"""
Suspends the node process .
"""
raise NotImplementedError
2018-10-15 10:05:49 +00:00
async def close ( self ) :
2015-01-22 10:49:22 +00:00
"""
2016-05-11 17:35:36 +00:00
Close the node process .
2015-01-22 10:49:22 +00:00
"""
2016-02-29 09:38:30 +00:00
if self . _closed :
return False
2016-06-23 01:40:46 +00:00
log . info ( " {module} : ' {name} ' [ {id} ]: is closing " . format ( module = self . manager . module_name ,
name = self . name ,
id = self . id ) )
2016-02-29 09:38:30 +00:00
if self . _console :
self . _manager . port_manager . release_tcp_port ( self . _console , self . _project )
self . _console = None
2016-11-08 18:44:12 +00:00
if self . _wrap_console :
self . _manager . port_manager . release_tcp_port ( self . _internal_console_port , self . _project )
self . _internal_console_port = None
2016-02-29 09:38:30 +00:00
if self . _aux :
self . _manager . port_manager . release_tcp_port ( self . _aux , self . _project )
self . _aux = None
self . _closed = True
return True
2020-12-03 07:34:42 +00:00
def _get_vnc_console_port_range ( self ) :
"""
Returns the VNC console port range .
"""
server_config = self . _manager . config . get_section_config ( " Server " )
vnc_console_start_port_range = server_config . getint ( " vnc_console_start_port_range " , 5900 )
vnc_console_end_port_range = server_config . getint ( " vnc_console_end_port_range " , 10000 )
if not 5900 < = vnc_console_start_port_range < = 65535 :
raise NodeError ( " The VNC console start port range must be between 5900 and 65535 " )
if not 5900 < = vnc_console_end_port_range < = 65535 :
raise NodeError ( " The VNC console start port range must be between 5900 and 65535 " )
if vnc_console_start_port_range > = vnc_console_end_port_range :
raise NodeError ( f " The VNC console start port range value ( { vnc_console_start_port_range } ) "
f " cannot be above or equal to the end value ( { vnc_console_end_port_range } ) " )
return vnc_console_start_port_range , vnc_console_end_port_range
2018-10-15 10:05:49 +00:00
async def start_wrap_console ( self ) :
2016-11-08 18:44:12 +00:00
"""
2020-01-31 09:31:27 +00:00
Start a telnet proxy for the console allowing multiple telnet clients
to be connected at the same time
2016-11-08 18:44:12 +00:00
"""
2018-03-15 07:17:39 +00:00
2016-12-01 10:47:05 +00:00
if not self . _wrap_console or self . _console_type != " telnet " :
2016-11-08 18:44:12 +00:00
return
2016-11-16 12:06:43 +00:00
remaining_trial = 60
while True :
try :
2022-12-31 01:43:17 +00:00
( self . _wrap_console_reader , self . _wrap_console_writer ) = await asyncio . open_connection (
host = " 127.0.0.1 " ,
port = self . _internal_console_port
)
2016-11-16 12:06:43 +00:00
break
except ( OSError , ConnectionRefusedError ) as e :
if remaining_trial < = 0 :
raise e
2018-10-15 10:05:49 +00:00
await asyncio . sleep ( 0.1 )
2016-11-16 12:06:43 +00:00
remaining_trial - = 1
2022-12-31 01:43:17 +00:00
await AsyncioTelnetServer . write_client_intro ( self . _wrap_console_writer , echo = True )
server = AsyncioTelnetServer (
reader = self . _wrap_console_reader ,
writer = self . _wrap_console_writer ,
binary = True ,
echo = True
)
2018-04-16 07:30:06 +00:00
# warning: this will raise OSError exception if there is a problem...
2022-12-30 12:54:37 +00:00
self . _wrapper_telnet_server = await asyncio . start_server (
server . run ,
self . _manager . port_manager . console_host ,
self . console
)
2016-11-08 18:44:12 +00:00
2018-10-15 10:05:49 +00:00
async def stop_wrap_console ( self ) :
2018-03-24 11:11:21 +00:00
"""
Stops the telnet proxy .
"""
if self . _wrapper_telnet_server :
2022-12-31 01:43:17 +00:00
self . _wrap_console_writer . close ( )
2023-01-20 05:14:57 +00:00
if sys . version_info > = ( 3 , 7 , 0 ) :
2023-05-25 10:29:19 +00:00
try :
await self . _wrap_console_writer . wait_closed ( )
except ConnectionResetError :
pass
2018-03-24 11:11:21 +00:00
self . _wrapper_telnet_server . close ( )
2018-10-15 10:05:49 +00:00
await self . _wrapper_telnet_server . wait_closed ( )
2022-12-31 01:43:17 +00:00
self . _wrapper_telnet_server = None
async def reset_wrap_console ( self ) :
"""
Reset the wrap console ( restarts the Telnet proxy )
"""
await self . stop_wrap_console ( )
await self . start_wrap_console ( )
2018-03-24 11:11:21 +00:00
2020-01-31 09:31:27 +00:00
async def start_websocket_console ( self , request ) :
"""
Connect to console using Websocket .
: param ws : Websocket object
"""
if self . status != " started " :
raise NodeError ( " Node {} is not started " . format ( self . name ) )
if self . _console_type != " telnet " :
raise NodeError ( " Node {} console type is not telnet " . format ( self . name ) )
try :
( telnet_reader , telnet_writer ) = await asyncio . open_connection ( self . _manager . port_manager . console_host , self . console )
except ConnectionError as e :
raise NodeError ( " Cannot connect to node {} telnet server: {} " . format ( self . name , e ) )
log . info ( " Connected to Telnet server " )
ws = WebSocketResponse ( )
await ws . prepare ( request )
request . app [ ' websockets ' ] . add ( ws )
log . info ( " New client has connected to console WebSocket " )
async def ws_forward ( telnet_writer ) :
async for msg in ws :
if msg . type == aiohttp . WSMsgType . TEXT :
telnet_writer . write ( msg . data . encode ( ) )
await telnet_writer . drain ( )
elif msg . type == aiohttp . WSMsgType . BINARY :
2021-07-26 10:58:35 +00:00
telnet_writer . write ( msg . data )
2020-01-31 09:31:27 +00:00
await telnet_writer . drain ( )
elif msg . type == aiohttp . WSMsgType . ERROR :
log . debug ( " Websocket connection closed with exception {} " . format ( ws . exception ( ) ) )
async def telnet_forward ( telnet_reader ) :
while not ws . closed and not telnet_reader . at_eof ( ) :
data = await telnet_reader . read ( 1024 )
if data :
await ws . send_bytes ( data )
try :
# keep forwarding websocket data in both direction
2023-02-18 07:32:57 +00:00
if sys . version_info > = ( 3 , 11 , 0 ) :
# Starting with Python 3.11, passing coroutine objects to wait() directly is forbidden.
aws = [ asyncio . create_task ( ws_forward ( telnet_writer ) ) , asyncio . create_task ( telnet_forward ( telnet_reader ) ) ]
else :
aws = [ ws_forward ( telnet_writer ) , telnet_forward ( telnet_reader ) ]
done , pending = await asyncio . wait ( aws , return_when = asyncio . FIRST_COMPLETED )
for task in done :
if task . exception ( ) :
log . warning ( f " Exception while forwarding WebSocket data to Telnet server { task . exception ( ) } " )
for task in pending :
task . cancel ( )
2020-01-31 09:31:27 +00:00
finally :
log . info ( " Client has disconnected from console WebSocket " )
if not ws . closed :
await ws . close ( )
request . app [ ' websockets ' ] . discard ( ws )
return ws
2016-03-01 14:01:45 +00:00
@property
def allocate_aux ( self ) :
"""
: returns : Boolean allocate or not an aux console
"""
2018-03-15 07:17:39 +00:00
2016-03-01 14:01:45 +00:00
return self . _allocate_aux
2016-03-01 14:33:07 +00:00
@allocate_aux.setter
def allocate_aux ( self , allocate_aux ) :
"""
: returns : Boolean allocate or not an aux console
"""
2018-03-15 07:17:39 +00:00
2016-03-01 14:33:07 +00:00
self . _allocate_aux = allocate_aux
2016-02-29 09:38:30 +00:00
@property
def aux ( self ) :
"""
2016-05-11 17:35:36 +00:00
Returns the aux console port of this node .
2016-02-29 09:38:30 +00:00
: returns : aux console port
"""
return self . _aux
@aux.setter
def aux ( self , aux ) :
"""
Changes the aux port
: params aux : Console port ( integer ) or None to free the port
"""
if aux == self . _aux :
return
if self . _aux :
self . _manager . port_manager . release_tcp_port ( self . _aux , self . _project )
self . _aux = None
if aux is not None :
self . _aux = self . _manager . port_manager . reserve_tcp_port ( aux , self . _project )
log . info ( " {module} : ' {name} ' [ {id} ]: aux port set to {port} " . format ( module = self . manager . module_name ,
name = self . name ,
id = self . id ,
port = aux ) )
2015-02-19 10:33:25 +00:00
@property
def console ( self ) :
"""
2016-05-11 17:35:36 +00:00
Returns the console port of this node .
2015-02-19 10:33:25 +00:00
: returns : console port
"""
return self . _console
@console.setter
def console ( self , console ) :
"""
2015-04-08 17:17:34 +00:00
Changes the console port
2015-02-19 10:33:25 +00:00
2016-02-29 09:38:30 +00:00
: params console : Console port ( integer ) or None to free the port
2015-02-19 10:33:25 +00:00
"""
2018-03-24 11:11:21 +00:00
if console == self . _console or self . _console_type == " none " :
2015-02-19 10:33:25 +00:00
return
2015-10-16 16:15:27 +00:00
2016-02-29 09:38:30 +00:00
if self . _console_type == " vnc " and console is not None and console < 5900 :
2018-06-09 14:47:36 +00:00
raise NodeError ( " VNC console require a port superior or equal to 5900, current port is {} " . format ( console ) )
2015-10-16 16:15:27 +00:00
2015-02-19 10:33:25 +00:00
if self . _console :
2015-03-21 23:19:12 +00:00
self . _manager . port_manager . release_tcp_port ( self . _console , self . _project )
2016-02-29 09:38:30 +00:00
self . _console = None
if console is not None :
2016-04-06 12:57:52 +00:00
if self . console_type == " vnc " :
2020-12-03 07:34:42 +00:00
vnc_console_start_port_range , vnc_console_end_port_range = self . _get_vnc_console_port_range ( )
self . _console = self . _manager . port_manager . reserve_tcp_port (
console ,
self . _project ,
port_range_start = vnc_console_start_port_range ,
port_range_end = vnc_console_end_port_range
)
2016-04-06 12:57:52 +00:00
else :
self . _console = self . _manager . port_manager . reserve_tcp_port ( console , self . _project )
2016-02-29 09:38:30 +00:00
log . info ( " {module} : ' {name} ' [ {id} ]: console port set to {port} " . format ( module = self . manager . module_name ,
name = self . name ,
id = self . id ,
port = console ) )
2015-07-03 22:06:25 +00:00
@property
def console_type ( self ) :
"""
2016-05-11 17:35:36 +00:00
Returns the console type for this node .
2015-07-03 22:06:25 +00:00
: returns : console type ( string )
"""
return self . _console_type
@console_type.setter
def console_type ( self , console_type ) :
"""
2016-05-11 17:35:36 +00:00
Sets the console type for this node .
2015-07-03 22:06:25 +00:00
: param console_type : console type ( string )
"""
if console_type != self . _console_type :
# get a new port if the console type change
2018-03-23 14:29:39 +00:00
if self . _console :
self . _manager . port_manager . release_tcp_port ( self . _console , self . _project )
2018-03-23 08:44:16 +00:00
if console_type == " none " :
# no need to allocate a port when the console type is none
self . _console = None
elif console_type == " vnc " :
2022-11-06 11:36:31 +00:00
vnc_console_start_port_range , vnc_console_end_port_range = self . _get_vnc_console_port_range ( )
self . _console = self . _manager . port_manager . get_free_tcp_port (
self . _project ,
vnc_console_start_port_range ,
vnc_console_end_port_range
)
2015-07-03 22:06:25 +00:00
else :
self . _console = self . _manager . port_manager . get_free_tcp_port ( self . _project )
self . _console_type = console_type
2018-03-24 11:11:21 +00:00
log . info ( " {module} : ' {name} ' [ {id} ]: console type set to {console_type} (console port is {console} ) " . format ( module = self . manager . module_name ,
name = self . name ,
id = self . id ,
console_type = console_type ,
console = self . console ) )
2015-07-22 04:58:28 +00:00
2016-06-23 01:40:46 +00:00
@property
def ubridge ( self ) :
"""
Returns the uBridge hypervisor .
2017-07-10 18:38:28 +00:00
: returns : instance of uBridge
2016-06-23 01:40:46 +00:00
"""
2016-12-14 11:01:34 +00:00
if self . _ubridge_hypervisor and not self . _ubridge_hypervisor . is_running ( ) :
self . _ubridge_hypervisor = None
2016-06-23 01:40:46 +00:00
return self . _ubridge_hypervisor
@ubridge.setter
def ubridge ( self , ubride_hypervisor ) :
"""
Set an uBridge hypervisor .
: param ubride_hypervisor : uBridge hypervisor
"""
self . _ubridge_hypervisor = ubride_hypervisor
2015-09-13 20:52:25 +00:00
@property
def ubridge_path ( self ) :
"""
Returns the uBridge executable path .
: returns : path to uBridge
"""
path = self . _manager . config . get_section_config ( " Server " ) . get ( " ubridge_path " , " ubridge " )
2023-12-05 11:24:40 +00:00
if sys . platform . startswith ( " win " ) and hasattr ( sys , " frozen " ) :
ubridge_dir = os . path . normpath ( os . path . join ( os . path . dirname ( os . path . abspath ( sys . executable ) ) , " ubridge " ) )
os . environ [ " PATH " ] = os . pathsep . join ( ubridge_dir ) + os . pathsep + os . environ . get ( " PATH " , " " )
2016-08-29 09:27:35 +00:00
path = shutil . which ( path )
2015-09-13 20:52:25 +00:00
return path
2018-10-15 10:05:49 +00:00
async def _ubridge_send ( self , command ) :
2016-05-29 22:35:07 +00:00
"""
Sends a command to uBridge hypervisor .
: param command : command to send
"""
2017-07-10 18:38:28 +00:00
if not self . _ubridge_hypervisor or not self . _ubridge_hypervisor . is_running ( ) :
2018-11-19 17:22:16 +00:00
await self . _start_ubridge ( self . _ubridge_require_privileged_access )
2016-05-29 22:35:07 +00:00
if not self . _ubridge_hypervisor or not self . _ubridge_hypervisor . is_running ( ) :
raise NodeError ( " Cannot send command ' {} ' : uBridge is not running " . format ( command ) )
2016-05-31 04:07:37 +00:00
try :
2018-10-15 10:05:49 +00:00
await self . _ubridge_hypervisor . send ( command )
2016-05-31 04:07:37 +00:00
except UbridgeError as e :
2018-04-16 08:36:36 +00:00
raise UbridgeError ( " Error while sending command ' {} ' : {} : {} " . format ( command , e , self . _ubridge_hypervisor . read_stdout ( ) ) )
2016-05-29 22:35:07 +00:00
2018-08-25 07:10:47 +00:00
@locking
2018-11-19 17:22:16 +00:00
async def _start_ubridge ( self , require_privileged_access = False ) :
2015-09-13 20:52:25 +00:00
"""
2016-05-11 17:35:36 +00:00
Starts uBridge ( handles connections to and from this node ) .
2015-09-13 20:52:25 +00:00
"""
2017-03-28 14:27:09 +00:00
# Prevent us to start multiple ubridge
if self . _ubridge_hypervisor and self . _ubridge_hypervisor . is_running ( ) :
return
2016-07-12 15:38:13 +00:00
if self . ubridge_path is None :
2016-12-05 16:30:09 +00:00
raise NodeError ( " uBridge is not available, path doesn ' t exist, or you just installed GNS3 and need to restart your user session to refresh user permissions. " )
2016-07-12 15:38:13 +00:00
2018-11-19 17:22:16 +00:00
if require_privileged_access and not self . _manager . has_privileged_access ( self . ubridge_path ) :
raise NodeError ( " uBridge requires root access or the capability to interact with network adapters " )
2015-09-14 21:05:25 +00:00
2015-09-13 20:52:25 +00:00
server_config = self . _manager . config . get_section_config ( " Server " )
server_host = server_config . get ( " host " )
2017-03-03 17:40:26 +00:00
if not self . ubridge :
2016-06-23 01:40:46 +00:00
self . _ubridge_hypervisor = Hypervisor ( self . _project , self . ubridge_path , self . working_dir , server_host )
2015-09-13 20:52:25 +00:00
log . info ( " Starting new uBridge hypervisor {} : {} " . format ( self . _ubridge_hypervisor . host , self . _ubridge_hypervisor . port ) )
2018-10-15 10:05:49 +00:00
await self . _ubridge_hypervisor . start ( )
2017-05-11 15:26:18 +00:00
if self . _ubridge_hypervisor :
log . info ( " Hypervisor {} : {} has successfully started " . format ( self . _ubridge_hypervisor . host , self . _ubridge_hypervisor . port ) )
2018-10-15 10:05:49 +00:00
await self . _ubridge_hypervisor . connect ( )
2018-11-19 17:22:16 +00:00
# save if privileged are required in case uBridge needs to be restarted in self._ubridge_send()
self . _ubridge_require_privileged_access = require_privileged_access
2015-09-13 20:52:25 +00:00
2018-10-15 10:05:49 +00:00
async def _stop_ubridge ( self ) :
2016-05-29 22:35:07 +00:00
"""
Stops uBridge .
"""
if self . _ubridge_hypervisor and self . _ubridge_hypervisor . is_running ( ) :
2016-10-03 10:31:01 +00:00
log . info ( " Stopping uBridge hypervisor {} : {} " . format ( self . _ubridge_hypervisor . host , self . _ubridge_hypervisor . port ) )
2018-10-15 10:05:49 +00:00
await self . _ubridge_hypervisor . stop ( )
2016-12-14 11:01:34 +00:00
self . _ubridge_hypervisor = None
2016-05-29 22:35:07 +00:00
2018-10-15 10:05:49 +00:00
async def add_ubridge_udp_connection ( self , bridge_name , source_nio , destination_nio ) :
2016-06-23 22:56:06 +00:00
"""
2016-11-13 09:28:14 +00:00
Creates an UDP connection in uBridge .
2016-06-23 22:56:06 +00:00
: param bridge_name : bridge name in uBridge
: param source_nio : source NIO instance
: param destination_nio : destination NIO instance
"""
2018-10-15 10:05:49 +00:00
await self . _ubridge_send ( " bridge create {name} " . format ( name = bridge_name ) )
2016-06-23 22:56:06 +00:00
if not isinstance ( destination_nio , NIOUDP ) :
raise NodeError ( " Destination NIO is not UDP " )
2018-10-15 10:05:49 +00:00
await self . _ubridge_send ( ' bridge add_nio_udp {name} {lport} {rhost} {rport} ' . format ( name = bridge_name ,
2016-06-23 22:56:06 +00:00
lport = source_nio . lport ,
rhost = source_nio . rhost ,
rport = source_nio . rport ) )
2018-10-15 10:05:49 +00:00
await self . _ubridge_send ( ' bridge add_nio_udp {name} {lport} {rhost} {rport} ' . format ( name = bridge_name ,
2016-06-23 22:56:06 +00:00
lport = destination_nio . lport ,
rhost = destination_nio . rhost ,
rport = destination_nio . rport ) )
if destination_nio . capturing :
2018-10-15 10:05:49 +00:00
await self . _ubridge_send ( ' bridge start_capture {name} " {pcap_file} " ' . format ( name = bridge_name ,
2016-06-23 22:56:06 +00:00
pcap_file = destination_nio . pcap_output_file ) )
2018-10-15 10:05:49 +00:00
await self . _ubridge_send ( ' bridge start {name} ' . format ( name = bridge_name ) )
await self . _ubridge_apply_filters ( bridge_name , destination_nio . filters )
2017-06-30 08:22:30 +00:00
2018-10-15 10:05:49 +00:00
async def update_ubridge_udp_connection ( self , bridge_name , source_nio , destination_nio ) :
2017-07-11 11:42:47 +00:00
if destination_nio :
2018-10-15 10:05:49 +00:00
await self . _ubridge_apply_filters ( bridge_name , destination_nio . filters )
2017-06-30 08:22:30 +00:00
2018-10-15 10:05:49 +00:00
async def ubridge_delete_bridge ( self , name ) :
2017-07-12 09:42:37 +00:00
"""
: params name : Delete the bridge with this name
"""
2018-03-15 07:17:39 +00:00
2017-07-12 09:42:37 +00:00
if self . ubridge :
2018-10-15 10:05:49 +00:00
await self . _ubridge_send ( " bridge delete {name} " . format ( name = name ) )
2017-07-12 09:42:37 +00:00
2018-10-15 10:05:49 +00:00
async def _ubridge_apply_filters ( self , bridge_name , filters ) :
2017-06-30 08:22:30 +00:00
"""
2018-03-12 06:38:50 +00:00
Apply packet filters
2017-06-30 08:22:30 +00:00
: param bridge_name : bridge name in uBridge
2018-03-12 06:38:50 +00:00
: param filters : Array of filter dictionary
2017-06-30 08:22:30 +00:00
"""
2018-03-15 07:17:39 +00:00
2018-10-15 10:05:49 +00:00
await self . _ubridge_send ( ' bridge reset_packet_filters ' + bridge_name )
2018-03-12 06:38:50 +00:00
for packet_filter in self . _build_filter_list ( filters ) :
cmd = ' bridge add_packet_filter {} {} ' . format ( bridge_name , packet_filter )
try :
2018-10-15 10:05:49 +00:00
await self . _ubridge_send ( cmd )
2018-03-12 06:38:50 +00:00
except UbridgeError as e :
2019-01-17 11:01:58 +00:00
match = re . search ( r " Cannot compile filter ' (.*) ' : syntax error " , str ( e ) )
2018-03-12 06:38:50 +00:00
if match :
message = " Warning: ignoring BPF packet filter ' {} ' due to syntax error " . format ( self . name , match . group ( 1 ) )
log . warning ( message )
self . project . emit ( " log.warning " , { " message " : message } )
else :
raise
2017-07-17 09:21:54 +00:00
def _build_filter_list ( self , filters ) :
"""
: returns : Iterator building a list of filter
"""
2018-03-15 07:17:39 +00:00
2017-06-30 08:22:30 +00:00
i = 0
2017-07-05 14:36:39 +00:00
for ( filter_type , values ) in filters . items ( ) :
2017-07-11 15:30:29 +00:00
if isinstance ( values [ 0 ] , str ) :
for line in values [ 0 ] . split ( ' \n ' ) :
line = line . strip ( )
2017-07-17 09:21:54 +00:00
yield " {filter_name} {filter_type} {filter_value} " . format (
2017-07-11 15:30:29 +00:00
filter_name = " filter " + str ( i ) ,
filter_type = filter_type ,
filter_value = ' " {} " {} ' . format ( line , " " . join ( [ str ( v ) for v in values [ 1 : ] ] ) ) ) . strip ( )
i + = 1
else :
2017-07-17 09:21:54 +00:00
yield " {filter_name} {filter_type} {filter_value} " . format (
2017-07-11 15:30:29 +00:00
filter_name = " filter " + str ( i ) ,
filter_type = filter_type ,
filter_value = " " . join ( [ str ( v ) for v in values ] ) )
i + = 1
2016-06-23 22:56:06 +00:00
2018-10-15 10:05:49 +00:00
async def _add_ubridge_ethernet_connection ( self , bridge_name , ethernet_interface , block_host_traffic = False ) :
2016-11-13 09:28:14 +00:00
"""
Creates a connection with an Ethernet interface in uBridge .
: param bridge_name : bridge name in uBridge
: param ethernet_interface : Ethernet interface name
: param block_host_traffic : block network traffic originating from the host OS ( Windows only )
"""
2017-07-19 09:56:24 +00:00
if sys . platform . startswith ( " linux " ) and block_host_traffic is False :
# on Linux we use RAW sockets by default excepting if host traffic must be blocked
2018-10-15 10:05:49 +00:00
await self . _ubridge_send ( ' bridge add_nio_linux_raw {name} " {interface} " ' . format ( name = bridge_name , interface = ethernet_interface ) )
2016-11-13 09:28:14 +00:00
elif sys . platform . startswith ( " win " ) :
2022-07-06 06:22:11 +00:00
npf_id , source_mac = self . _find_windows_interface ( ethernet_interface )
2016-11-13 09:28:14 +00:00
if npf_id :
2018-10-15 10:05:49 +00:00
await self . _ubridge_send ( ' bridge add_nio_ethernet {name} " {interface} " ' . format ( name = bridge_name ,
2016-11-13 09:28:14 +00:00
interface = npf_id ) )
else :
2016-11-15 06:36:51 +00:00
raise NodeError ( " Could not find NPF id for interface {} " . format ( ethernet_interface ) )
2016-11-13 09:28:14 +00:00
if block_host_traffic :
if source_mac :
2018-10-15 10:05:49 +00:00
await self . _ubridge_send ( ' bridge set_pcap_filter {name} " not ether src {mac} " ' . format ( name = bridge_name , mac = source_mac ) )
2017-07-19 09:56:24 +00:00
log . info ( ' PCAP filter applied on " {interface} " for source MAC {mac} ' . format ( interface = ethernet_interface , mac = source_mac ) )
2016-11-13 09:28:14 +00:00
else :
2017-07-19 09:56:24 +00:00
log . warning ( " Could not block host network traffic on {} (no MAC address found) " . format ( ethernet_interface ) )
2016-11-13 09:28:14 +00:00
else :
# on other platforms we just rely on the pcap library
2018-10-15 10:05:49 +00:00
await self . _ubridge_send ( ' bridge add_nio_ethernet {name} " {interface} " ' . format ( name = bridge_name , interface = ethernet_interface ) )
2017-07-19 09:56:24 +00:00
source_mac = None
for interface in interfaces ( ) :
if interface [ " name " ] == ethernet_interface :
source_mac = interface [ " mac_address " ]
if source_mac :
2018-10-15 10:05:49 +00:00
await self . _ubridge_send ( ' bridge set_pcap_filter {name} " not ether src {mac} " ' . format ( name = bridge_name , mac = source_mac ) )
2017-07-19 09:56:24 +00:00
log . info ( ' PCAP filter applied on " {interface} " for source MAC {mac} ' . format ( interface = ethernet_interface , mac = source_mac ) )
2016-11-13 09:28:14 +00:00
2022-07-06 06:22:11 +00:00
@staticmethod
def _find_windows_interface ( ethernet_interface ) :
"""
Get NPF ID and MAC address by input ethernet interface name .
Return None , None when not match any interface
: returns : NPF ID and MAC address
"""
# on Windows we use Winpcap/Npcap
windows_interfaces = interfaces ( )
for interface in windows_interfaces :
if str . strip ( ethernet_interface ) == str . strip ( interface [ " name " ] ) :
return interface [ " id " ] , interface [ " mac_address " ]
for interface in windows_interfaces :
if " netcard " in interface and ethernet_interface in interface [ " netcard " ] :
return interface [ " id " ] , interface [ " mac_address " ]
return None , None
2016-06-23 22:56:06 +00:00
def _create_local_udp_tunnel ( self ) :
"""
Creates a local UDP tunnel ( pair of 2 NIOs , one for each direction )
: returns : source NIO and destination NIO .
"""
m = PortManager . instance ( )
lport = m . get_free_udp_port ( self . project )
rport = m . get_free_udp_port ( self . project )
source_nio_settings = { ' lport ' : lport , ' rhost ' : ' 127.0.0.1 ' , ' rport ' : rport , ' type ' : ' nio_udp ' }
destination_nio_settings = { ' lport ' : rport , ' rhost ' : ' 127.0.0.1 ' , ' rport ' : lport , ' type ' : ' nio_udp ' }
2016-06-25 00:35:39 +00:00
source_nio = self . manager . create_nio ( source_nio_settings )
destination_nio = self . manager . create_nio ( destination_nio_settings )
2016-06-23 22:56:06 +00:00
log . info ( " {module} : ' {name} ' [ {id} ]:local UDP tunnel created between port {port1} and {port2} " . format ( module = self . manager . module_name ,
name = self . name ,
id = self . id ,
port1 = lport ,
port2 = rport ) )
return source_nio , destination_nio
2015-07-22 04:58:28 +00:00
@property
def hw_virtualization ( self ) :
"""
2016-05-11 17:35:36 +00:00
Returns either the node is using hardware virtualization or not .
2015-07-22 04:58:28 +00:00
: return : boolean
"""
return self . _hw_virtualization
2015-10-12 21:57:37 +00:00
def check_available_ram ( self , requested_ram ) :
"""
Sends a warning notification if there is not enough RAM on the system to allocate requested RAM .
: param requested_ram : requested amount of RAM in MB
"""
available_ram = int ( psutil . virtual_memory ( ) . available / ( 1024 * 1024 ) )
2021-09-20 08:38:02 +00:00
percentage_left = 100 - psutil . virtual_memory ( ) . percent
2015-10-12 21:57:37 +00:00
if requested_ram > available_ram :
message = ' " {} " requires {} MB of RAM to run but there is only {} MB - {} % o f RAM left on " {} " ' . format ( self . name ,
requested_ram ,
available_ram ,
percentage_left ,
platform . node ( ) )
self . project . emit ( " log.warning " , { " message " : message } )
2018-04-02 15:27:12 +00:00
def _get_custom_adapter_settings ( self , adapter_number ) :
for custom_adapter in self . custom_adapters :
if custom_adapter [ " adapter_number " ] == adapter_number :
return custom_adapter
return { }