mirror of
https://github.com/checktheroads/hyperglass
synced 2024-05-11 05:55:08 +00:00
408 lines
15 KiB
Python
408 lines
15 KiB
Python
"""Execute validated & constructed query on device.
|
|
|
|
Accepts input from front end application, validates the input and
|
|
returns errors if input is invalid. Passes validated parameters to
|
|
construct.py, which is used to build & run the Netmiko connectoins or
|
|
hyperglass-frr API calls, returns the output back to the front end.
|
|
"""
|
|
|
|
# Standard Library Imports
|
|
import re
|
|
import signal
|
|
|
|
# Third Party Imports
|
|
import httpx
|
|
import sshtunnel
|
|
from netmiko import ConnectHandler
|
|
from netmiko import NetMikoAuthenticationException
|
|
from netmiko import NetmikoAuthError
|
|
from netmiko import NetmikoTimeoutError
|
|
from netmiko import NetMikoTimeoutException
|
|
|
|
# Project Imports
|
|
from hyperglass.configuration import devices
|
|
from hyperglass.configuration import params
|
|
from hyperglass.constants import Supported
|
|
from hyperglass.exceptions import AuthError
|
|
from hyperglass.exceptions import DeviceTimeout
|
|
from hyperglass.exceptions import ResponseEmpty
|
|
from hyperglass.exceptions import RestError
|
|
from hyperglass.exceptions import ScrapeError
|
|
from hyperglass.execution.construct import Construct
|
|
from hyperglass.execution.encode import jwt_decode
|
|
from hyperglass.execution.encode import jwt_encode
|
|
from hyperglass.execution.validate import Validate
|
|
from hyperglass.util import log
|
|
|
|
|
|
class Connect:
|
|
"""Connect to target device via specified transport.
|
|
|
|
scrape_direct() directly connects to devices via SSH
|
|
|
|
scrape_proxied() connects to devices via an SSH proxy
|
|
|
|
rest() connects to devices via HTTP for RESTful API communication
|
|
"""
|
|
|
|
def __init__(self, device, query_data, transport):
|
|
"""Initialize connection to device.
|
|
|
|
Arguments:
|
|
device {object} -- Matched device object
|
|
query_data {object} -- Query object
|
|
transport {str} -- 'scrape' or 'rest'
|
|
"""
|
|
self.device = device
|
|
self.query_data = query_data
|
|
self.query_type = self.query_data.query_type
|
|
self.query_target = self.query_data.query_target
|
|
self.transport = transport
|
|
self.query = getattr(
|
|
Construct(
|
|
device=self.device, query_data=self.query_data, transport=self.transport
|
|
),
|
|
self.query_type,
|
|
)()
|
|
|
|
async def scrape_proxied(self):
|
|
"""Connect to a device via an SSH proxy.
|
|
|
|
Connects to the router via Netmiko library via the sshtunnel
|
|
library, returns the command output.
|
|
"""
|
|
log.debug(f"Connecting to {self.device.proxy} via sshtunnel library...")
|
|
try:
|
|
tunnel = sshtunnel.open_tunnel(
|
|
self.device.proxy.address,
|
|
self.device.proxy.port,
|
|
ssh_username=self.device.proxy.credential.username,
|
|
ssh_password=self.device.proxy.credential.password.get_secret_value(),
|
|
remote_bind_address=(self.device.address, self.device.port),
|
|
local_bind_address=("localhost", 0),
|
|
skip_tunnel_checkup=False,
|
|
)
|
|
except sshtunnel.BaseSSHTunnelForwarderError as scrape_proxy_error:
|
|
log.error(
|
|
f"Error connecting to device {self.device.location} via "
|
|
f"proxy {self.device.proxy.name}"
|
|
)
|
|
raise ScrapeError(
|
|
params.messages.connection_error,
|
|
device_name=self.device.display_name,
|
|
proxy=self.device.proxy.name,
|
|
error=str(scrape_proxy_error),
|
|
)
|
|
|
|
def handle_timeout(*args, **kwargs):
|
|
tunnel.close()
|
|
raise DeviceTimeout(
|
|
params.messages.connection_error,
|
|
device_name=self.device.display_name,
|
|
proxy=self.device.proxy.name,
|
|
error=params.messages.request_timeout,
|
|
)
|
|
|
|
signal.signal(signal.SIGALRM, handle_timeout)
|
|
signal.alarm(params.request_timeout - 1)
|
|
|
|
with tunnel:
|
|
log.debug(
|
|
"Established tunnel with {d}. Local bind port: {p}",
|
|
d=self.device.proxy,
|
|
p=tunnel.local_bind_port,
|
|
)
|
|
scrape_host = {
|
|
"host": "localhost",
|
|
"port": tunnel.local_bind_port,
|
|
"device_type": self.device.nos,
|
|
"username": self.device.credential.username,
|
|
"password": self.device.credential.password.get_secret_value(),
|
|
"global_delay_factor": 0.2,
|
|
"timeout": params.request_timeout - 1,
|
|
}
|
|
|
|
try:
|
|
log.debug("Connecting to {loc}...", loc=self.device.location)
|
|
|
|
nm_connect_direct = ConnectHandler(**scrape_host)
|
|
|
|
responses = []
|
|
for query in self.query:
|
|
raw = nm_connect_direct.send_command(query)
|
|
responses.append(raw)
|
|
log.debug(f'Raw response for command "{query}":\n{raw}')
|
|
response = "\n\n".join(responses)
|
|
|
|
nm_connect_direct.disconnect()
|
|
|
|
log.debug(f"Response type:\n{type(response)}")
|
|
|
|
except (NetMikoTimeoutException, NetmikoTimeoutError) as scrape_error:
|
|
log.error(
|
|
"Timeout connecting to device {loc}: {e}",
|
|
loc=self.device.location,
|
|
e=str(scrape_error),
|
|
)
|
|
raise DeviceTimeout(
|
|
params.messages.connection_error,
|
|
device_name=self.device.display_name,
|
|
proxy=self.device.proxy.name,
|
|
error=params.messages.request_timeout,
|
|
)
|
|
except (NetMikoAuthenticationException, NetmikoAuthError) as auth_error:
|
|
log.error(
|
|
"Error authenticating to device {loc}: {e}",
|
|
loc=self.device.location,
|
|
e=str(auth_error),
|
|
)
|
|
raise AuthError(
|
|
params.messages.connection_error,
|
|
device_name=self.device.display_name,
|
|
proxy=self.device.proxy.name,
|
|
error=params.messages.authentication_error,
|
|
) from None
|
|
except sshtunnel.BaseSSHTunnelForwarderError:
|
|
raise ScrapeError(
|
|
params.messages.connection_error,
|
|
device_name=self.device.display_name,
|
|
proxy=self.device.proxy.name,
|
|
error=params.messages.general,
|
|
)
|
|
if response is None:
|
|
raise ScrapeError(
|
|
params.messages.connection_error,
|
|
device_name=self.device.display_name,
|
|
proxy=None,
|
|
error=params.messages.noresponse_error,
|
|
)
|
|
signal.alarm(0)
|
|
return response
|
|
|
|
async def scrape_direct(self):
|
|
"""Connect directly to a device.
|
|
|
|
Directly connects to the router via Netmiko library, returns the
|
|
command output.
|
|
"""
|
|
log.debug(f"Connecting directly to {self.device.location}...")
|
|
|
|
scrape_host = {
|
|
"host": self.device.address,
|
|
"port": self.device.port,
|
|
"device_type": self.device.nos,
|
|
"username": self.device.credential.username,
|
|
"password": self.device.credential.password.get_secret_value(),
|
|
"global_delay_factor": 0.2,
|
|
"timeout": params.request_timeout,
|
|
}
|
|
|
|
try:
|
|
nm_connect_direct = ConnectHandler(**scrape_host)
|
|
|
|
def handle_timeout(*args, **kwargs):
|
|
nm_connect_direct.disconnect()
|
|
raise DeviceTimeout(
|
|
params.messages.connection_error,
|
|
device_name=self.device.display_name,
|
|
proxy=self.device.proxy.name,
|
|
error=params.messages.request_timeout,
|
|
)
|
|
|
|
signal.signal(signal.SIGALRM, handle_timeout)
|
|
signal.alarm(params.request_timeout - 1)
|
|
|
|
responses = []
|
|
|
|
for query in self.query:
|
|
raw = nm_connect_direct.send_command(query)
|
|
responses.append(raw)
|
|
log.debug(f'Raw response for command "{query}":\n{raw}')
|
|
response = "\n\n".join(responses)
|
|
|
|
nm_connect_direct.diconnect()
|
|
|
|
except (NetMikoTimeoutException, NetmikoTimeoutError) as scrape_error:
|
|
log.error(str(scrape_error))
|
|
raise DeviceTimeout(
|
|
params.messages.connection_error,
|
|
device_name=self.device.display_name,
|
|
proxy=None,
|
|
error=params.messages.request_timeout,
|
|
)
|
|
except (NetMikoAuthenticationException, NetmikoAuthError) as auth_error:
|
|
log.error(
|
|
"Error authenticating to device {loc}: {e}",
|
|
loc=self.device.location,
|
|
e=str(auth_error),
|
|
)
|
|
|
|
raise AuthError(
|
|
params.messages.connection_error,
|
|
device_name=self.device.display_name,
|
|
proxy=None,
|
|
error=params.messages.authentication_error,
|
|
)
|
|
if response is None:
|
|
raise ScrapeError(
|
|
params.messages.connection_error,
|
|
device_name=self.device.display_name,
|
|
proxy=None,
|
|
error=params.messages.noresponse_error,
|
|
)
|
|
signal.alarm(0)
|
|
return response
|
|
|
|
async def rest(self):
|
|
"""Connect to a device running hyperglass-agent via HTTP."""
|
|
log.debug(f"Query parameters: {self.query}")
|
|
|
|
client_params = {
|
|
"headers": {"Content-Type": "application/json"},
|
|
"timeout": params.request_timeout,
|
|
}
|
|
if self.device.ssl is not None and self.device.ssl.enable:
|
|
http_protocol = "https"
|
|
client_params.update({"verify": str(self.device.ssl.cert)})
|
|
log.debug(
|
|
(
|
|
f"Using {str(self.device.ssl.cert)} to validate connection "
|
|
f"to {self.device.name}"
|
|
)
|
|
)
|
|
else:
|
|
http_protocol = "http"
|
|
endpoint = "{protocol}://{address}:{port}/query/".format(
|
|
protocol=http_protocol, address=self.device.address, port=self.device.port
|
|
)
|
|
|
|
log.debug(f"URL endpoint: {endpoint}")
|
|
|
|
try:
|
|
async with httpx.Client(**client_params) as http_client:
|
|
responses = []
|
|
|
|
for query in self.query:
|
|
encoded_query = await jwt_encode(
|
|
payload=query,
|
|
secret=self.device.credential.password.get_secret_value(),
|
|
duration=params.request_timeout,
|
|
)
|
|
log.debug(f"Encoded JWT: {encoded_query}")
|
|
|
|
raw_response = await http_client.post(
|
|
endpoint, json={"encoded": encoded_query}
|
|
)
|
|
log.debug(f"HTTP status code: {raw_response.status_code}")
|
|
|
|
raw = raw_response.text
|
|
log.debug(f"Raw Response: {raw}")
|
|
|
|
if raw_response.status_code == 200:
|
|
decoded = await jwt_decode(
|
|
payload=raw_response.json()["encoded"],
|
|
secret=self.device.credential.password.get_secret_value(),
|
|
)
|
|
log.debug(f"Decoded Response: {decoded}")
|
|
|
|
responses.append(decoded)
|
|
else:
|
|
log.error(raw_response.text)
|
|
|
|
response = "\n\n".join(responses)
|
|
log.debug(f"Output for query {self.query}:\n{response}")
|
|
except httpx.exceptions.HTTPError as rest_error:
|
|
rest_msg = " ".join(
|
|
re.findall(r"[A-Z][^A-Z]*", rest_error.__class__.__name__)
|
|
)
|
|
log.error(f"Error connecting to device {self.device.location}: {rest_msg}")
|
|
raise RestError(
|
|
params.messages.connection_error,
|
|
device_name=self.device.display_name,
|
|
error=rest_msg,
|
|
)
|
|
except OSError as ose:
|
|
log.critical(str(ose))
|
|
raise RestError(
|
|
params.messages.connection_error,
|
|
device_name=self.device.display_name,
|
|
error="System error",
|
|
)
|
|
|
|
if raw_response.status_code != 200:
|
|
log.error(f"Response code is {raw_response.status_code}")
|
|
raise RestError(
|
|
params.messages.connection_error,
|
|
device_name=self.device.display_name,
|
|
error=params.messages.general,
|
|
)
|
|
|
|
if not response:
|
|
log.error(f"No response from device {self.device.location}")
|
|
raise RestError(
|
|
params.messages.connection_error,
|
|
device_name=self.device.display_name,
|
|
error=params.messages.noresponse_error,
|
|
)
|
|
|
|
return response
|
|
|
|
|
|
class Execute:
|
|
"""Perform query execution on device.
|
|
|
|
Ingests raw user input, performs validation of target input, pulls
|
|
all configuraiton variables for the input router and connects to the
|
|
selected device to execute the query.
|
|
"""
|
|
|
|
def __init__(self, lg_data):
|
|
"""Initialize execution object.
|
|
|
|
Arguments:
|
|
lg_data {object} -- Validated query object
|
|
"""
|
|
self.query_data = lg_data
|
|
self.query_location = self.query_data.query_location
|
|
self.query_type = self.query_data.query_type
|
|
self.query_target = self.query_data.query_target
|
|
|
|
async def response(self):
|
|
"""Initiate query validation and execution."""
|
|
device = getattr(devices, self.query_location)
|
|
|
|
log.debug(f"Received query for {self.query_data}")
|
|
log.debug(f"Matched device config: {device}")
|
|
|
|
# Run query parameters through validity checks
|
|
validation = Validate(device, self.query_data, self.query_target)
|
|
valid_input = validation.validate_query()
|
|
|
|
if valid_input:
|
|
log.debug(f"Validation passed for query: {self.query_data}")
|
|
pass
|
|
|
|
connect = None
|
|
output = params.messages.general
|
|
|
|
transport = Supported.map_transport(device.nos)
|
|
connect = Connect(device, self.query_data, transport)
|
|
|
|
if Supported.is_rest(device.nos):
|
|
output = await connect.rest()
|
|
|
|
elif Supported.is_scrape(device.nos):
|
|
if device.proxy:
|
|
output = await connect.scrape_proxied()
|
|
else:
|
|
output = await connect.scrape_direct()
|
|
|
|
if output == "":
|
|
raise ResponseEmpty(
|
|
params.messages.no_output, device_name=device.display_name
|
|
)
|
|
|
|
log.debug(f"Output for query: {self.query}:\n{output}")
|
|
|
|
return output
|