建档

source/Tool/Blob.hpp

@@ -0,0 +1,137 @@
+#pragma once
+
+#include <vector>
+#include <string>
+#include <cstdint>
+#include <stdexcept>
+#include <algorithm>
+
+class Blob
+{
+    using BYTE = unsigned char;
+
+private:
+    const std::vector<BYTE> m_blob;
+    size_t m_offset;
+
+    // 检查是否有足够的剩余字节
+    void checkSize(size_t required) const
+    {
+        if (m_offset + required > m_blob.size())
+        {
+            throw std::out_of_range("Insufficient data in blob");
+        }
+    }
+
+public:
+    // 构造函数，接受一个字节向量的引用
+    Blob(const std::vector<BYTE> blob) : m_blob(blob), m_offset(0) {}
+
+    // 重置解析偏移量
+    void reset() { m_offset = 0; }
+
+    // 获取当前偏移量
+    size_t getOffset() const { return m_offset; }
+
+    // 设置偏移量
+    void setOffset(size_t offset)
+    {
+        if (offset > m_blob.size())
+        {
+            throw std::out_of_range("Offset out of range");
+        }
+        m_offset = offset;
+    }
+
+    // 获取剩余字节数
+    size_t remaining() const { return m_blob.size() - m_offset; }
+
+    // 读取一个T类型的数据（适用于基本类型）
+    template <typename T>
+    T get()
+    {
+        checkSize(sizeof(T));
+
+        T value;
+        std::copy(m_blob.begin() + m_offset,
+                  m_blob.begin() + m_offset + sizeof(T),
+                  reinterpret_cast<BYTE *>(&value));
+        m_offset += sizeof(T);
+
+        return value;
+    }
+
+    // 读取int类型
+    int32_t getInt() { return get<int32_t>(); }
+
+    // 读取无符号int类型
+    uint32_t getUInt() { return get<uint32_t>(); }
+
+    // 读取short类型
+    int16_t getShort() { return get<int16_t>(); }
+
+    // 读取无符号short类型
+    uint16_t getUShort() { return get<uint16_t>(); }
+
+    // 读取float类型
+    float getFloat() { return get<float>(); }
+
+    // 读取double类型
+    double getDouble() { return get<double>(); }
+
+    BYTE getByte() { return get<BYTE>(); }
+
+    float get256()
+    {
+        BYTE buf = get<BYTE>();
+        return static_cast<float>(buf);
+    }
+
+    // 读取指定长度的字符串
+    std::string getString(size_t length)
+    {
+        checkSize(length);
+
+        std::string str(reinterpret_cast<const char *>(m_blob.data() + m_offset), length);
+        m_offset += length;
+
+        return str;
+    }
+
+    // 读取以null结尾的字符串
+    std::string getNullTerminatedString()
+    {
+        size_t length = 0;
+        size_t pos = m_offset;
+
+        // 查找null终止符
+        while (pos < m_blob.size() && m_blob[pos] != '\0')
+        {
+            pos++;
+            length++;
+        }
+
+        // 确保找到了终止符
+        if (pos >= m_blob.size())
+        {
+            throw std::runtime_error("Null-terminated string not found");
+        }
+
+        // 读取字符串（不包含终止符）
+        std::string str(reinterpret_cast<const char *>(m_blob.data() + m_offset), length);
+        m_offset = pos + 1; // 跳过终止符
+
+        return str;
+    }
+
+    // 读取指定数量的字节
+    std::vector<BYTE> getBytes(size_t count)
+    {
+        checkSize(count);
+
+        std::vector<BYTE> bytes(m_blob.begin() + m_offset, m_blob.begin() + m_offset + count);
+        m_offset += count;
+
+        return bytes;
+    }
+};