summaryrefslogtreecommitdiffstats
path: root/common/entitybuffer.cpp
blob: 4af84efbe6df3714314669e4cd65ca2ed0f2ef50 (plain)
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
#include "entitybuffer.h"

#include "entity_generated.h"
#include "metadata_generated.h"
#include <QDebug>

using namespace Akonadi2;

EntityBuffer::EntityBuffer(void *dataValue, int dataSize)
    : mEntity(nullptr)
{
    flatbuffers::Verifier verifyer(reinterpret_cast<const uint8_t *>(dataValue), dataSize);
    // Q_ASSERT(Akonadi2::VerifyEntity(verifyer));
    if (!Akonadi2::VerifyEntityBuffer(verifyer)) {
        qWarning() << "invalid buffer";
    } else {
        mEntity = Akonadi2::GetEntity(dataValue);
    }
}

const Akonadi2::Entity &EntityBuffer::entity()
{
    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)
{
    Akonadi2::EntityBuffer buffer(dataValue, dataSize);
    if (auto resourceData = buffer.entity().resource()) {
        handler(resourceData->Data(), resourceData->size());
    }
}

void EntityBuffer::assembleEntityBuffer(flatbuffers::FlatBufferBuilder &fbb, void const *metadataData, size_t metadataSize, void const *resourceData, size_t resourceSize, void const *localData, size_t localSize)
{
    qDebug() << "res size: " << resourceSize;
    auto metadata = fbb.CreateVector<uint8_t>(static_cast<uint8_t const*>(metadataData), metadataSize);
    auto resource = fbb.CreateVector<uint8_t>(static_cast<uint8_t const*>(resourceData), resourceSize);
    auto local = fbb.CreateVector<uint8_t>(static_cast<uint8_t const*>(localData), localSize);
    auto builder = Akonadi2::EntityBuilder(fbb);
    builder.add_metadata(metadata);
    builder.add_resource(resource);
    builder.add_local(local);

    auto buffer = builder.Finish();
    Akonadi2::FinishEntityBuffer(fbb, buffer);
}