Merge pull request #1129 from GNS3/filters_virtualbox

Packet filters support for VirtualBox

gns3server/compute/virtualbox/virtualbox_vm.py

@@ -94,6 +94,10 @@ class VirtualBoxVM(BaseNode):
             json["node_directory"] = None
         return json
 
+    @property
+    def ethernet_adapters(self):
+        return self._ethernet_adapters
+
     @asyncio.coroutine
     def _get_system_properties(self):
 
@@ -973,16 +977,39 @@ class VirtualBoxVM(BaseNode):
                                                            self._local_udp_tunnels[adapter_number][1],
                                                            nio)
             except KeyError:
-                raise VirtualBoxError("Adapter {adapter_number} doesn't exist on VirtualBox VM '{name}'".format(name=self.name,
+                raise VirtualBoxError("Adapter {adapter_number} doesn't exist on VirtualBox VM '{name}'".format(
+                    name=self.name,
                     adapter_number=adapter_number))
             yield from self._control_vm("setlinkstate{} on".format(adapter_number + 1))
 
         adapter.add_nio(0, nio)
-        log.info("VirtualBox VM '{name}' [{id}]: {nio} added to adapter {adapter_number}".format(name=self.name,
+        log.info("VirtualBox VM '{name}' [{id}]: {nio} added to adapter {adapter_number}".format(
+            name=self.name,
             id=self.id,
             nio=nio,
             adapter_number=adapter_number))
 
+    @asyncio.coroutine
+    def adapter_update_nio_binding(self, adapter_number, nio):
+        """
+        Update a port NIO binding.
+
+        :param adapter_number: adapter number
+        :param nio: NIO instance to add to the adapter
+        """
+
+        if self.is_running():
+            try:
+                yield from self.update_ubridge_udp_connection(
+                    "VBOX-{}-{}".format(self._id, adapter_number),
+                    self._local_udp_tunnels[adapter_number][1],
+                    nio)
+            except IndexError:
+                raise VirtualBoxError('Adapter {adapter_number} does not exist on VirtualBox VM "{name}"'.format(
+                    name=self._name,
+                    adapter_number=adapter_number
+                ))
+
     @asyncio.coroutine
     def adapter_remove_nio_binding(self, adapter_number):
         """

gns3server/controller/link.py

@@ -371,6 +371,7 @@ class Link:
                                           'iou',
                                           'cloud',
                                           'nat',
+                                          'virtualbox',
                                           'docker'):
                 return node["node"]
         return None

gns3server/controller/project.py

@@ -746,11 +746,11 @@ class Project:
                 yield from self.add_node(compute, name, node_id, dump=False, **node)
             for link_data in topology.get("links", []):
                 link = yield from self.add_link(link_id=link_data["link_id"])
+                if "filters" in link_data:
+                    yield from link.update_filters(link_data["filters"])
                 for node_link in link_data["nodes"]:
                     node = self.get_node(node_link["node_id"])
                     yield from link.add_node(node, node_link["adapter_number"], node_link["port_number"], label=node_link.get("label"), dump=False)
-                if "filters" in link_data:
-                    yield from link.update_filters(link_data["filters"])
             for drawing_data in topology.get("drawings", []):
                 yield from self.add_drawing(dump=False, **drawing_data)
 

gns3server/handlers/api/compute/qemu_handler.py

@@ -294,7 +294,7 @@ class QEMUHandler:
         nio = vm.ethernet_adapters[int(request.match_info["adapter_number"])]
         if "filters" in request.json and nio:
             nio.filters = request.json["filters"]
-        yield from vm.adapter_update_nio_binding(int(request.match_info["port_number"]), nio)
+        yield from vm.adapter_update_nio_binding(int(request.match_info["adapter_number"]), nio)
         response.set_status(201)
         response.json(request.json)
 

gns3server/handlers/api/compute/virtualbox_handler.py

@@ -290,6 +290,33 @@ class VirtualBoxHandler:
         response.set_status(201)
         response.json(nio)
 
+    @Route.put(
+        r"/projects/{project_id}/virtualbox/nodes/{node_id}/adapters/{adapter_number:\d+}/ports/{port_number:\d+}/nio",
+        parameters={
+            "project_id": "Project UUID",
+            "node_id": "Node UUID",
+            "adapter_number": "Network adapter where the nio is located",
+            "port_number": "Port from where the nio should be updated"
+        },
+        status_codes={
+            201: "NIO updated",
+            400: "Invalid request",
+            404: "Instance doesn't exist"
+        },
+        input=NIO_SCHEMA,
+        output=NIO_SCHEMA,
+        description="Update a NIO from a Virtualbox instance")
+    def update_nio(request, response):
+
+        virtualbox_manager = VirtualBox.instance()
+        vm = virtualbox_manager.get_node(request.match_info["node_id"], project_id=request.match_info["project_id"])
+        nio = vm.ethernet_adapters[int(request.match_info["adapter_number"])]
+        if "filters" in request.json and nio:
+            nio.filters = request.json["filters"]
+        yield from vm.adapter_update_nio_binding(int(request.match_info["adapter_number"]), nio)
+        response.set_status(201)
+        response.json(request.json)
+
     @Route.delete(
         r"/projects/{project_id}/virtualbox/nodes/{node_id}/adapters/{adapter_number:\d+}/ports/{port_number:\d+}/nio",
         parameters={

tests/handlers/api/compute/test_virtualbox.py

@@ -26,7 +26,8 @@ def vm(http_compute, project, monkeypatch):
     vboxmanage_path = "/fake/VboxManage"
 
     with asyncio_patch("gns3server.compute.virtualbox.virtualbox_vm.VirtualBoxVM.create", return_value=True) as mock:
-        response = http_compute.post("/projects/{project_id}/virtualbox/nodes".format(project_id=project.id), {"name": "VMTEST",
+        response = http_compute.post("/projects/{project_id}/virtualbox/nodes".format(
+            project_id=project.id), {"name": "VMTEST",
                                      "vmname": "VMTEST",
                                      "linked_clone": False})
     assert mock.called
@@ -111,6 +112,24 @@ def test_vbox_nio_create_udp(http_compute, vm):
     assert response.json["type"] == "nio_udp"
 
 
+def test_virtualbox_nio_update_udp(http_compute, vm):
+    with asyncio_patch('gns3server.compute.virtualbox.virtualbox_vm.VirtualBoxVM.ethernet_adapters'):
+        with asyncio_patch('gns3server.compute.virtualbox.virtualbox_vm.VirtualBoxVM.adapter_remove_nio_binding') as mock:
+            response = http_compute.put("/projects/{project_id}/virtualbox/nodes/{node_id}/adapters/0/ports/0/nio".format(
+                project_id=vm["project_id"],
+                node_id=vm["node_id"]),
+                {
+                    "type": "nio_udp",
+                    "lport": 4242,
+                    "rport": 4343,
+                    "rhost": "127.0.0.1",
+                    "filters": {}},
+                example=True)
+    assert response.status == 201, response.body.decode()
+    assert response.route == "/projects/{project_id}/virtualbox/nodes/{node_id}/adapters/{adapter_number:\d+}/ports/{port_number:\d+}/nio"
+    assert response.json["type"] == "nio_udp"
+
+
 def test_vbox_delete_nio(http_compute, vm):
 
     with asyncio_patch('gns3server.compute.virtualbox.virtualbox_vm.VirtualBoxVM.adapter_remove_nio_binding') as mock: