1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
#include "entitybuffer.h"
#include "entity_generated.h"
#include "metadata_generated.h"
#include <QDebug>
using namespace Sink;
EntityBuffer::EntityBuffer(const void *dataValue, int dataSize) : mEntity(nullptr)
{
flatbuffers::Verifier verifyer(reinterpret_cast<const uint8_t *>(dataValue), dataSize);
// Q_ASSERT(Sink::VerifyEntity(verifyer));
if (!Sink::VerifyEntityBuffer(verifyer)) {
qWarning() << "invalid buffer";
} else {
mEntity = Sink::GetEntity(dataValue);
}
}
EntityBuffer::EntityBuffer(const QByteArray &data) : EntityBuffer(data.constData(), data.size())
{
}
bool EntityBuffer::isValid() const
{
return mEntity;
}
const Sink::Entity &EntityBuffer::entity() const
{
Q_ASSERT(mEntity);
return *mEntity;
}
const uint8_t *EntityBuffer::resourceBuffer()
{
if (!mEntity) {
qDebug() << "no buffer";
return nullptr;
}
return mEntity->resource()->Data();
}
const uint8_t *EntityBuffer::metadataBuffer()
{
if (!mEntity) {
return nullptr;
}
return mEntity->metadata()->Data();
}
const uint8_t *EntityBuffer::localBuffer()
{
if (!mEntity) {
return nullptr;
}
return mEntity->local()->Data();
}
void EntityBuffer::extractResourceBuffer(void *dataValue, int dataSize, const std::function<void(const uint8_t *, size_t size)> &handler)
{
Sink::EntityBuffer buffer(dataValue, dataSize);
if (auto resourceData = buffer.entity().resource()) {
handler(resourceData->Data(), resourceData->size());
}
}
flatbuffers::Offset<flatbuffers::Vector<uint8_t>> EntityBuffer::appendAsVector(flatbuffers::FlatBufferBuilder &fbb, void const *data, size_t size)
{
// Since we do memcpy trickery, this will only work on little endian
assert(FLATBUFFERS_LITTLEENDIAN);
uint8_t *rawDataPtr = Q_NULLPTR;
const auto pos = fbb.CreateUninitializedVector<uint8_t>(size, &rawDataPtr);
std::memcpy((void *)rawDataPtr, data, size);
return pos;
}
void EntityBuffer::assembleEntityBuffer(
flatbuffers::FlatBufferBuilder &fbb, void const *metadataData, size_t metadataSize, void const *resourceData, size_t resourceSize, void const *localData, size_t localSize)
{
auto metadata = appendAsVector(fbb, metadataData, metadataSize);
auto resource = appendAsVector(fbb, resourceData, resourceSize);
auto local = appendAsVector(fbb, localData, localSize);
auto entity = Sink::CreateEntity(fbb, metadata, resource, local);
Sink::FinishEntityBuffer(fbb, entity);
}
Sink::Operation EntityBuffer::operation() const
{
const auto metadataBuffer = readBuffer<Sink::Metadata>(mEntity->metadata());
return metadataBuffer ? metadataBuffer->operation() : Sink::Operation_Creation;
}
qint64 EntityBuffer::revision() const
{
const auto metadataBuffer = readBuffer<Sink::Metadata>(mEntity->metadata());
return metadataBuffer ? metadataBuffer->revision() : -1;
}
|
