Merge pull request #1130 from GNS3/vmware_filters

Filters support for VMware

gns3server/compute/vmware/vmware_vm.py

@@ -68,6 +68,10 @@ class VMwareVM(BaseNode):
         if not os.path.exists(vmx_path):
             raise VMwareError('VMware VM "{name}" [{id}]: could not find VMX file "{vmx_path}"'.format(name=name, id=node_id, vmx_path=vmx_path))
 
+    @property
+    def ethernet_adapters(self):
+        return self._ethernet_adapters
+
     def __json__(self):
 
         json = {"name": self.name,
@@ -290,7 +294,7 @@ class VMwareVM(BaseNode):
             if allocate_vmnet:
                 try:
                     vmnet = self.manager.allocate_vmnet()
-                except:
+                except BaseException:
                     # clear everything up in case of error (e.g. no enough vmnets)
                     self._vmnets.clear()
                     raise
@@ -306,6 +310,15 @@ class VMwareVM(BaseNode):
                 log.debug("disabling remaining adapter {}".format(adapter_number))
                 self._vmx_pairs["ethernet{}.startconnected".format(adapter_number)] = "FALSE"
 
+    def _get_vnet(self, adapter_number):
+        """
+        Return the vnet will use in ubridge
+        """
+        vnet = "ethernet{}.vnet".format(adapter_number)
+        if vnet not in self._vmx_pairs:
+            raise VMwareError("vnet {} not in VMX file".format(vnet))
+        return vnet
+
     @asyncio.coroutine
     def _add_ubridge_connection(self, nio, adapter_number):
         """
@@ -316,9 +329,7 @@ class VMwareVM(BaseNode):
         """
 
         block_host_traffic = self.manager.config.get_section_config("VMware").getboolean("block_host_traffic", False)
-        vnet = "ethernet{}.vnet".format(adapter_number)
-        if vnet not in self._vmx_pairs:
-            raise VMwareError("vnet {} not in VMX file".format(vnet))
+        vnet = self._get_vnet(adapter_number)
         yield from self._ubridge_send("bridge create {name}".format(name=vnet))
         vmnet_interface = os.path.basename(self._vmx_pairs[vnet])
 
@@ -346,6 +357,21 @@ class VMwareVM(BaseNode):
         #         source_mac = interface["mac_address"]
         # if source_mac:
         #     yield from self._ubridge_send('bridge set_pcap_filter {name} "not ether src {mac}"'.format(name=vnet, mac=source_mac))
+        yield from self._ubridge_apply_filters(vnet, nio.filters)
+
+    @asyncio.coroutine
+    def _update_ubridge_connection(self, adapter_number, nio):
+        """
+        Update a connection in uBridge.
+
+        :param nio: NIO instance
+        :param adapter_number: adapter number
+        """
+        try:
+            bridge_name = self._get_vnet(adapter_number)
+        except VMwareError:
+            return  # vnet not yet available
+        yield from self._ubridge_apply_filters(bridge_name, nio.filters)
 
     @asyncio.coroutine
     def _delete_ubridge_connection(self, adapter_number):
@@ -735,6 +761,24 @@ class VMwareVM(BaseNode):
                                                                                              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._ubridge_hypervisor:
+            try:
+                yield from self._update_ubridge_connection(adapter_number, nio)
+            except IndexError:
+                raise VMwareError('Adapter {adapter_number} does not exist on VMware VM "{name}"'.format(
+                    name=self._name,
+                    adapter_number=adapter_number
+                ))
+
     @asyncio.coroutine
     def adapter_remove_nio_binding(self, adapter_number):
         """

gns3server/controller/link.py

@@ -366,6 +366,7 @@ class Link:
         """
         for node in self._nodes:
             if node["node"].node_type in ('vpcs',
+                                          'vmware',
                                           'dynamips',
                                           'qemu',
                                           'iou',

gns3server/controller/project.py

@@ -19,7 +19,6 @@ import re
 import os
 import json
 import uuid
-import copy
 import shutil
 import asyncio
 import aiohttp

gns3server/handlers/api/compute/vmware_handler.py

@@ -259,6 +259,33 @@ class VMwareHandler:
         response.set_status(201)
         response.json(nio)
 
+    @Route.put(
+        r"/projects/{project_id}/vmware/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):
+
+        vmware_manager = VMware.instance()
+        vm = vmware_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}/vmware/nodes/{node_id}/adapters/{adapter_number:\d+}/ports/{port_number:\d+}/nio",
         parameters={

tests/handlers/api/compute/test_vmware.py

@@ -120,6 +120,25 @@ def test_vmware_nio_create_udp(http_compute, vm):
     assert response.json["type"] == "nio_udp"
 
 
+def test_vmware_nio_update_udp(http_compute, vm):
+    with asyncio_patch('gns3server.compute.vmware.vmware_vm.VMwareVM._ubridge_send'):
+        with asyncio_patch('gns3server.compute.vmware.vmware_vm.VMwareVM.ethernet_adapters'):
+            with patch('gns3server.compute.vmware.vmware_vm.VMwareVM._get_vnet') as mock:
+                response = http_compute.put("/projects/{project_id}/vmware/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}/vmware/nodes/{node_id}/adapters/{adapter_number:\d+}/ports/{port_number:\d+}/nio"
+                assert response.json["type"] == "nio_udp"
+
+
 def test_vmware_delete_nio(http_compute, vm):
 
     with asyncio_patch('gns3server.compute.vmware.vmware_vm.VMwareVM.adapter_remove_nio_binding') as mock: