You've already forked cmf-bindings
TomZ
113 lines
3.1 KiB
C++
113 lines
3.1 KiB
C++
/*
|
|
* This file is part of the Bitcoin project
|
|
* Copyright (C) 2014-2016 Tom Zander <tomz@freedommail.ch>
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
#include "MessageBuilder.h"
|
|
#include "CMF.h"
|
|
|
|
#include <QBuffer>
|
|
#include <QDebug>
|
|
|
|
namespace {
|
|
int write(char *data, quint32 tag, CMF::ValueType type) {
|
|
Q_ASSERT(type < 8);
|
|
if (tag >= 31) { // use more than 1 byte
|
|
quint8 byte = type | 0xF8; // set the 'tag' to all 1s
|
|
data[0] = byte;
|
|
return CMF::serialize(data +1, tag) + 1;
|
|
}
|
|
Q_ASSERT(tag < 32);
|
|
quint8 byte = tag;
|
|
byte = byte << 3;
|
|
byte += type;
|
|
data[0] = byte;
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
MessageBuilder::MessageBuilder(QIODevice *device)
|
|
: m_device(device),
|
|
m_ownsBuffer(false)
|
|
{
|
|
Q_ASSERT(device);
|
|
}
|
|
|
|
MessageBuilder::MessageBuilder(QByteArray *byteArray)
|
|
: m_device(new QBuffer(byteArray)),
|
|
m_ownsBuffer(true)
|
|
{
|
|
Q_ASSERT(byteArray);
|
|
m_device->open(QIODevice::WriteOnly);
|
|
}
|
|
|
|
MessageBuilder::~MessageBuilder()
|
|
{
|
|
if (m_ownsBuffer) {
|
|
qobject_cast<QBuffer*>(m_device)->close();
|
|
delete m_device;
|
|
}
|
|
}
|
|
|
|
void MessageBuilder::add(quint32 tag, qint64 value)
|
|
{
|
|
CMF::ValueType vt;
|
|
if (value >= 0) {
|
|
vt = CMF::PositiveNumber;
|
|
} else {
|
|
vt = CMF::NegativeNumber;
|
|
value *= -1;
|
|
}
|
|
int tagSize = write(m_data, tag, vt);
|
|
tagSize += CMF::serialize(m_data + tagSize, value);
|
|
m_device->write(m_data, tagSize);
|
|
}
|
|
|
|
void MessageBuilder::add(quint32 tag, quint64 value)
|
|
{
|
|
int tagSize = write(m_data, tag, CMF::PositiveNumber);
|
|
tagSize += CMF::serialize(m_data + tagSize, value);
|
|
m_device->write(m_data, tagSize);
|
|
}
|
|
|
|
void MessageBuilder::add(quint32 tag, const QString &value)
|
|
{
|
|
int tagSize = write(m_data, tag, CMF::String);
|
|
const QByteArray serializedData = value.toUtf8();
|
|
tagSize += CMF::serialize(m_data + tagSize, serializedData.count());
|
|
m_device->write(m_data, tagSize);
|
|
m_device->write(serializedData);
|
|
}
|
|
|
|
void MessageBuilder::add(quint32 tag, const QByteArray &data)
|
|
{
|
|
int tagSize = write(m_data, tag, CMF::ByteArray);
|
|
tagSize += CMF::serialize(m_data + tagSize, data.count());
|
|
m_device->write(m_data, tagSize);
|
|
m_device->write(data);
|
|
}
|
|
|
|
void MessageBuilder::add(quint32 tag, bool value)
|
|
{
|
|
int tagSize = write(m_data, tag, value ? CMF::BoolTrue : CMF::BoolFalse);
|
|
m_device->write(m_data, tagSize);
|
|
}
|
|
|
|
void MessageBuilder::close()
|
|
{
|
|
if (m_ownsBuffer)
|
|
qobject_cast<QBuffer*>(m_device)->close();
|
|
}
|