UniRec output: Enable ODID field in translator

Author: TheSableCZ

extra_plugins/output/unirec/src/translator.c

@@ -1483,6 +1483,34 @@ translator_table_fill_internal(translator_t *trans, const struct map_rec *map_re
         return IPX_OK;
     }
 
+    /* Internal "ODID field" function
+     * Implemented as a special converter that must be enabled in the translator because
+     * ODID is not defined in the IPFIX record but in the IPFIX Message header
+     */
+    if (src == MAP_SRC_INTERNAL_ODID) {
+        if (trans->extra_conv.odid.en) {
+            // The function is already enabled!
+            IPX_CTX_ERROR(trans->ctx, "Internal 'odid field' function can be mapped only "
+                "to one UniRec field!", '\0');
+            return IPX_ERR_DENIED;
+        }
+
+        if (ur_type != UR_TYPE_UINT32) {
+            IPX_CTX_ERROR(trans->ctx, "Internal 'odid field' function supports only UniRec "
+                "uint32 type but UniRec field '%s' is '%s'!", map_rec->unirec.name,
+                map_rec->unirec.type_str);
+            return IPX_ERR_DENIED;
+        }
+
+        // Enable the internal converter
+        trans->extra_conv.odid.en = true;
+        trans->extra_conv.odid.req_idx = field_idx;
+        trans->extra_conv.odid.field_id = ur_id;
+        IPX_CTX_DEBUG(trans->ctx, "Added conversion from internal 'odid field' to UniRec '%s'",
+            map_rec->unirec.name);
+        return IPX_ERR_NOTFOUND; // Do NOT add the record!
+    }
+
     IPX_CTX_ERROR(trans->ctx, "Unimplemented internal mapping function!", '\0');
     return IPX_ERR_DENIED;
 }
@@ -1630,6 +1658,19 @@ translator_call_internals(translator_t *trans)
         }
     }
 
+    if (trans->extra_conv.odid.en) {
+        // Call internal 'odid field' converter
+        int field_idx = trans->extra_conv.odid.req_idx;
+        if (translate_internal_odid(trans) == 0) {
+            // Success
+            trans->progress.req_fields[field_idx] = 0; // Filled!
+            converted_fields++;
+        } else {
+            IPX_CTX_WARNING(trans->ctx, "Internal function 'odid field' failed to fill "
+                "UniRec field '%s'", trans->progress.req_names[field_idx]);
+        }
+    }
+
     return converted_fields;
 }