fdsdump: storageRecord: define component

Author: Lukas Hutak

src/tools/fdsdump/src/storageRecord.cpp

@@ -0,0 +1,28 @@
+
+#include <cstring>
+#include <cstdint>
+#include <stdexcept>
+
+#include "storageRecord.hpp"
+
+StorageRecord::StorageRecord(
+        const struct fds_drec &rec,
+        enum Direction dir,
+        const shared_tsnapshot &snapshot)
+    : m_data{new uint8_t[rec.size]}, m_snapshot{snapshot}
+{
+    const uint16_t tmplt_id = rec.tmplt->id;
+    const fds_template *tmplt = fds_tsnapshot_template_get(m_snapshot.get(), tmplt_id);
+
+    if (!tmplt) {
+        throw std::runtime_error("Snapshot doesn't contain required template");;
+    }
+
+    std::memcpy(m_data.get(), rec.data, rec.size);
+
+    m_flow.dir = dir;
+    m_flow.rec.data = m_data.get();
+    m_flow.rec.size = rec.size;
+    m_flow.rec.tmplt = tmplt;
+    m_flow.rec.snap = m_snapshot.get();
+}

src/tools/fdsdump/src/storageRecord.hpp

@@ -0,0 +1,42 @@
+
+#pragma once
+
+#include <libfds.h>
+#include <memory>
+
+#include "common.hpp"
+#include "flow.hpp"
+
+/**
+ * @brief Flow storage record.
+ *
+ * The flow record contains the whole IPFIX Data Record and a reference to
+ * a template snapshot necessary for its interpretation.
+ */
+class StorageRecord {
+public:
+    /**
+     * @brief Create a storage record by creating a copy of IPFIX Data Record
+     * extracted, for example, from FDS File.
+     *
+     * @param rec      Flow data record to be stored
+     * @param dir      Direction of the record to be considered.
+     * @param snapshot Template snapshot that should be used
+     */
+    StorageRecord(
+        const struct fds_drec &rec,
+        enum Direction dir,
+        const shared_tsnapshot &snapshot);
+    ~StorageRecord() = default;
+
+    Flow &
+    get_flow() { return m_flow; };
+
+    const Flow &
+    get_flow_const() const { return m_flow; };
+
+private:
+    std::unique_ptr<uint8_t[]> m_data;
+    shared_tsnapshot m_snapshot;
+    Flow m_flow;
+};