es traceroute resource script

Author: katzy687

resource-scripts/es_ping/readme.MD

@@ -0,0 +1,2 @@
+from es_traceroute import run_es_traceroute
+run_es_traceroute()

resource-scripts/es_traceroute/__main__.py

@@ -0,0 +1,14 @@
+from cloudshell.helpers.scripts.cloudshell_dev_helpers import attach_to_cloudshell_as
+from es_traceroute import run_es_traceroute
+
+LIVE_SANDBOX_ID = "87df6742-0fd6-43e4-b068-48764b97d7e5"
+TARGET_RESOURCE_NAME = "DUT mock 1"
+
+attach_to_cloudshell_as(user="admin",
+                        password="admin",
+                        domain="Global",
+                        reservation_id=LIVE_SANDBOX_ID,
+                        server_address="localhost",
+                        resource_name=TARGET_RESOURCE_NAME)
+
+run_es_traceroute()

resource-scripts/es_traceroute/debug.py

@@ -0,0 +1,33 @@
+from cloudshell.helpers.scripts.cloudshell_scripts_helpers import get_reservation_context_details, get_api_session, \
+    get_resource_context_details
+from cloudshell.logging.qs_logger import get_qs_logger
+from cloudshell.helpers.sandbox_reporter.reporter import SandboxReporter
+from traceroute_helper import run_traceroute_to_ip, get_local_ip
+
+
+class TraceRouteException(Exception):
+    pass
+
+
+def run_es_traceroute():
+    api = get_api_session()
+    sb_details = get_reservation_context_details()
+    sb_id = sb_details.id
+
+    resource_details = get_resource_context_details()
+    resource_name = resource_details.name
+    logger = get_qs_logger(log_group=sb_id, log_category=resource_details.model, log_file_prefix=resource_name)
+    reporter = SandboxReporter(api, sb_id, logger)
+    ip = resource_details.address
+    es_ip = get_local_ip()
+
+    reporter.info(f"Running traceroute to '{resource_name}' at IP '{ip}' from Execution Server...")
+
+    try:
+        tr_output = run_traceroute_to_ip(ip)
+    except Exception as e:
+        msg = f"Failed ping to '{resource_name}' at IP '{ip}' from ES at IP '{es_ip}'.\n{str(e)}"
+        reporter.error(msg)
+        raise TraceRouteException(msg)
+
+    print(tr_output)

resource-scripts/es_traceroute/es_traceroute.py

@@ -0,0 +1,4 @@
+cloudshell-automation-api
+cloudshell-logging
+cloudshell-sandbox-reporter
+icmplib

resource-scripts/es_traceroute/requirements.txt

@@ -0,0 +1,32 @@
+import socket
+from icmplib import traceroute
+
+
+def get_local_ip():
+    """
+    Use socket to get the local IP, default to loopback 127.0.0.1
+    https://stackoverflow.com/a/28950776
+    :return:
+    """
+    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+    s.settimeout(0)
+    try:
+        # doesn't even have to be reachable
+        s.connect(('10.255.255.255', 1))
+        IP = s.getsockname()[0]
+    except Exception:
+        IP = '127.0.0.1'
+    finally:
+        s.close()
+    return IP
+
+
+def run_traceroute_to_ip(target_ip: str) -> str:
+    tr_hops = traceroute(address=target_ip)
+    full_output = "\n".join([str(hop) for hop in tr_hops])
+    return full_output
+
+
+if __name__ == "__main__":
+    res = run_traceroute_to_ip("google.com")
+    print(res)