IPFIX output: Introduce support for periodic message

Author: Adrian Duriska

src/plugins/output/ipfix/src/IPFIXOutput.cpp

@@ -527,6 +527,19 @@ IPFIXOutput::on_session_message(ipx_msg_session *message)
     }
 }
 
+void
+IPFIXOutput::on_periodic_message(ipx_msg_periodic_t *message)
+{
+    if (config->split_on_export_time) {
+        return;
+    }
+    // Start a new file, if needed
+    std::time_t time_now = std::time(NULL);
+    if (should_start_new_file(time_now)) {
+        new_file(time_now); // This will make sure that templates will be written to the file
+    }
+}
+
 IPFIXOutput::IPFIXOutput(const Config *config, const ipx_ctx *ctx) : plugin_context(ctx), config(config)
 {
     buffer.reset(new uint8_t[UINT16_MAX]);

src/plugins/output/ipfix/src/IPFIXOutput.hpp

@@ -120,6 +120,13 @@ class IPFIXOutput {
      */
     void
     on_session_message(ipx_msg_session *message);
+
+    /**
+     * \brief Processes an incoming periodic message from the collector
+     * \param[in] message The periodic message
+     */
+    void
+    on_periodic_message(ipx_msg_periodic_t *message);
 };
 
 

src/plugins/output/ipfix/src/IPFIXOutputPlugin.cpp

@@ -71,7 +71,7 @@ IPX_API struct ipx_plugin_info ipx_plugin_info = {
 int
 ipx_plugin_init(ipx_ctx_t *ctx, const char *params)
 {
-    ipx_msg_mask_t mask = IPX_MSG_IPFIX | IPX_MSG_SESSION;
+    ipx_msg_mask_t mask = IPX_MSG_IPFIX | IPX_MSG_SESSION | IPX_MSG_PERIODIC;
     if (ipx_ctx_subscribe(ctx, &mask, nullptr) != IPX_OK) {
         IPX_CTX_ERROR(ctx, "Error subscribing to messages", 0);
         return IPX_ERR_DENIED;
@@ -124,6 +124,8 @@ ipx_plugin_process(ipx_ctx_t *ctx, void *cfg, ipx_msg_t *msg)
             instance->ipfix_output->on_session_message(ipx_msg_base2session(msg));
         } else if (msg_type == IPX_MSG_IPFIX) {
             instance->ipfix_output->on_ipfix_message(ipx_msg_base2ipfix(msg));
+        } else if (msg_type == IPX_MSG_PERIODIC) {
+            instance->ipfix_output->on_periodic_message(ipx_msg_base2periodic(msg));
         }
 
     } catch (std::exception &ex) {