Added API/CLIs to immediately disable/enable IPFIX plugin for a client - both its generators and exporter.

Signed-off-by: obarash <obarash@gmail.com>

Author: odedba

scripts/automation/trex_control_plane/interactive/trex/emu/emu_plugins/emu_plugin_ipfix.py

@@ -225,6 +225,28 @@ def enable_generator(self, c_key, gen_name, enable):
         EMUValidator.verify(ver_args)
         return self.emu_c._send_plugin_cmd_to_client('ipfix_c_set_gen_state', c_key=c_key, gen_name=gen_name, enable=enable)
 
+    @client_api('command', True)
+    def enable_ipfix(self, c_key, enable):
+        """
+            Enable/disable IPFix plugin for a client.
+            It enable/disable all generators and the exporter. 
+
+            :parameters:
+
+                c_key: :class:`trex.emu.trex_emu_profile.EMUClientKey`
+                    EMUClientKey
+
+                enable: bool
+                    True if we wish to enable ipfix for the client, False if we wish to disable it.
+
+            :returns:
+               bool : Flag indicating the result of the operation.
+        """
+        ver_args = [{'name': 'c_key', 'arg': c_key, 't': EMUClientKey},
+                    {'name': 'enable', 'arg': enable, 't': bool}]
+        EMUValidator.verify(ver_args)
+        return self.emu_c._send_plugin_cmd_to_client('ipfix_c_set_state', c_key=c_key, enable=enable)
+
     # Plugins methods
     @plugin_api('ipfix_show_counters', 'emu')
     def ipfix_show_counters_line(self, line):
@@ -307,6 +329,32 @@ def _enable_disable_gen_line(self, line, caller_func, enable_disable):
         c_key = EMUClientKey(ns_key, opts.mac)
         return self.enable_generator(c_key, opts.gen_name, enable_disable == 'enable')
 
+    @plugin_api('ipfix_enable', 'emu')
+    def ipfix_enable_line(self, line):
+        """Enable IPFix plugin for a client.\n"""
+        res = self._enable_disable_line(line, self.ipfix_enable_line, "enable")
+        self.logger.post_cmd(res)
+
+    @plugin_api('ipfix_disable', 'emu')
+    def ipfix_disable_line(self, line):
+        """Disable IPFix plugin for a client.\n"""
+        res = self._enable_disable_line(line, self.ipfix_enable_line, "disable")
+        self.logger.post_cmd(res)
+
+    def _enable_disable_line(self, line, caller_func, enable_disable):
+        parser = parsing_opts.gen_parser(self,
+                                        "ipfix_enable",
+                                        caller_func.__doc__,
+                                        parsing_opts.EMU_NS_GROUP_NOT_REQ,
+                                        parsing_opts.MAC_ADDRESS,
+                                        )
+
+        opts = parser.parse_args(line.split())
+        self._validate_port(opts)
+        ns_key = EMUNamespaceKey(opts.port, opts.vlan, opts.tpid)
+        c_key = EMUClientKey(ns_key, opts.mac)
+        return self.enable_ipfix(c_key, enable_disable == 'enable')
+
     @plugin_api('ipfix_set_data_rate', 'emu')
     def ipfix_set_data_rate_line(self, line):
         """Set IPFix generator data rate.\n"""