Check for valid IP address and prevent to run on non-Windows platforms.

gns3server/compute/__init__.py

@@ -24,8 +24,9 @@ from .virtualbox import VirtualBox
 from .dynamips import Dynamips
 from .qemu import Qemu
 from .vmware import VMware
+from .traceng import TraceNG
 
-MODULES = [Builtin, VPCS, VirtualBox, Dynamips, Qemu, VMware]
+MODULES = [Builtin, VPCS, VirtualBox, Dynamips, Qemu, VMware, TraceNG]
 
 if sys.platform.startswith("linux") or hasattr(sys, "_called_from_test") or os.environ.get("PYTEST_BUILD_DOCUMENTATION") == "1":
     # IOU & Docker only runs on Linux but test suite works on UNIX platform
@@ -34,8 +35,3 @@ if sys.platform.startswith("linux") or hasattr(sys, "_called_from_test") or os.e
         MODULES.append(Docker)
         from .iou import IOU
         MODULES.append(IOU)
-
-#if sys.platform.startswith("win") or hasattr(sys, "_called_from_test") or os.environ.get("PYTEST_BUILD_DOCUMENTATION") == "1":
-# FIXME: TraceNG only runs on Windows but test suite works on UNIX platform
-from .traceng import TraceNG
-MODULES.append(TraceNG)

gns3server/compute/traceng/traceng_vm.py

@@ -19,11 +19,13 @@
 TraceNG VM management  in order to run a TraceNG VM.
 """
 
+import sys
 import os
 import socket
 import subprocess
 import asyncio
 import shutil
+import ipaddress
 
 from gns3server.utils.asyncio import wait_for_process_termination
 from gns3server.utils.asyncio import monitor_process
@@ -152,11 +154,17 @@ class TraceNGVM(BaseNode):
         :param ip_address: IP address
         """
 
+        try:
+            if ip_address:
+                ipaddress.IPv4Address(ip_address)
+        except ipaddress.AddressValueError:
+            raise TraceNGError("Invalid IP address: {}\n".format(ip_address))
+
+        self._ip_address = ip_address
         log.info("{module}: {name} [{id}] set IP address to {ip_address}".format(module=self.manager.module_name,
                                                                                 name=self.name,
                                                                                 id=self.id,
                                                                                 ip_address=ip_address))
-        self._ip_address = ip_address
 
     @asyncio.coroutine
     def start(self, destination=None):
@@ -164,6 +172,8 @@ class TraceNGVM(BaseNode):
         Starts the TraceNG process.
         """
 
+        if not sys.platform.startswith("win"):
+            raise TraceNGError("Sorry, TraceNG can only run on Windows")
         yield from self._check_requirements()
         if not self.is_running():
             nio = self._ethernet_adapter.get_nio(0)
@@ -382,11 +392,10 @@ class TraceNGVM(BaseNode):
         (to be passed to subprocess.Popen())
         """
 
-        # TODO: better validation
         if not destination:
-            raise TraceNGError("Please provide a destination to trace")
+            raise TraceNGError("Please provide a host or IP address to trace")
         if not self._ip_address:
-            raise TraceNGError("Please provide an IP address for this TraceNG node")
+            raise TraceNGError("Please configure an IP address for this TraceNG node")
 
         command = [self._traceng_path()]
         # use the local UDP tunnel to uBridge instead