177 lines
3.8 KiB
C++
177 lines
3.8 KiB
C++
#include <qendian.h>
|
|
#include <QHostAddress>
|
|
|
|
#include "../client/stream.h"
|
|
#include "payload.h"
|
|
|
|
PayloadConfigForm *PayloadProtocol::configForm = NULL;
|
|
|
|
PayloadConfigForm::PayloadConfigForm(QWidget *parent)
|
|
: QWidget(parent)
|
|
{
|
|
setupUi(this);
|
|
}
|
|
|
|
void PayloadConfigForm::on_cmbPatternMode_currentIndexChanged(int index)
|
|
{
|
|
switch(index)
|
|
{
|
|
case OstProto::Payload::e_dp_fixed_word:
|
|
lePattern->setEnabled(true);
|
|
break;
|
|
case OstProto::Payload::e_dp_inc_byte:
|
|
case OstProto::Payload::e_dp_dec_byte:
|
|
case OstProto::Payload::e_dp_random:
|
|
lePattern->setDisabled(true);
|
|
break;
|
|
default:
|
|
qWarning("Unhandled/Unknown PatternMode = %d",index);
|
|
}
|
|
}
|
|
|
|
PayloadProtocol::PayloadProtocol(Stream *parent)
|
|
: AbstractProtocol(parent)
|
|
{
|
|
if (configForm == NULL)
|
|
configForm = new PayloadConfigForm;
|
|
}
|
|
|
|
PayloadProtocol::~PayloadProtocol()
|
|
{
|
|
}
|
|
|
|
void PayloadProtocol::protoDataCopyInto(OstProto::Stream &stream)
|
|
{
|
|
// FIXME: multiple headers
|
|
stream.MutableExtension(OstProto::payload)->CopyFrom(data);
|
|
}
|
|
|
|
void PayloadProtocol::protoDataCopyFrom(const OstProto::Stream &stream)
|
|
{
|
|
// FIXME: multiple headers
|
|
if (stream.HasExtension(OstProto::payload))
|
|
data.MergeFrom(stream.GetExtension(OstProto::payload));
|
|
}
|
|
|
|
QString PayloadProtocol::name() const
|
|
{
|
|
return QString("Payload Data");
|
|
}
|
|
|
|
QString PayloadProtocol::shortName() const
|
|
{
|
|
return QString("DATA");
|
|
}
|
|
|
|
int PayloadProtocol::fieldCount() const
|
|
{
|
|
return payload_fieldCount;
|
|
}
|
|
|
|
QVariant PayloadProtocol::fieldData(int index, FieldAttrib attrib,
|
|
int streamIndex) const
|
|
{
|
|
switch (index)
|
|
{
|
|
case payload_dataPattern:
|
|
switch(attrib)
|
|
{
|
|
case FieldName:
|
|
return QString("Data");
|
|
case FieldValue:
|
|
return data.pattern();
|
|
case FieldTextValue:
|
|
return QString(fieldData(index, FieldFrameValue,
|
|
streamIndex).toByteArray().toHex());
|
|
case FieldFrameValue:
|
|
{
|
|
QByteArray fv;
|
|
int dataLen;
|
|
|
|
// FIXME: cannot use stream since it is only on client not
|
|
// on server
|
|
//dataLen = stream->frameLen() - stream->protocolHeaderSize();
|
|
dataLen = 64;
|
|
fv.resize(dataLen+4);
|
|
switch(data.pattern_mode())
|
|
{
|
|
case OstProto::Payload::e_dp_fixed_word:
|
|
for (int i = 0; i < (dataLen/4)+1; i++)
|
|
qToBigEndian((quint32) data.pattern(),
|
|
(uchar*)(fv.data()+(i*4)) );
|
|
break;
|
|
case OstProto::Payload::e_dp_inc_byte:
|
|
for (int i = 0; i < dataLen; i++)
|
|
fv[i] = i % (0xFF + 1);
|
|
break;
|
|
case OstProto::Payload::e_dp_dec_byte:
|
|
for (int i = 0; i < dataLen; i++)
|
|
fv[i] = 0xFF - (i % (0xFF + 1));
|
|
break;
|
|
case OstProto::Payload::e_dp_random:
|
|
for (int i = 0; i < dataLen; i++)
|
|
fv[i] = qrand() % (0xFF + 1);
|
|
break;
|
|
default:
|
|
qWarning("Unhandled data pattern %d",
|
|
data.pattern_mode());
|
|
}
|
|
fv.resize(dataLen);
|
|
return fv;
|
|
}
|
|
default:
|
|
break;
|
|
}
|
|
break;
|
|
|
|
// Meta fields
|
|
|
|
case payload_dataPatternMode:
|
|
switch(attrib)
|
|
{
|
|
case FieldIsMeta:
|
|
return true;
|
|
default:
|
|
break;
|
|
}
|
|
break;
|
|
|
|
default:
|
|
break;
|
|
}
|
|
|
|
return AbstractProtocol::fieldData(index, attrib, streamIndex);
|
|
}
|
|
|
|
bool PayloadProtocol::setFieldData(int index, const QVariant &value,
|
|
FieldAttrib attrib)
|
|
{
|
|
// FIXME
|
|
return false;
|
|
}
|
|
|
|
|
|
QWidget* PayloadProtocol::configWidget()
|
|
{
|
|
return configForm;
|
|
//return new PayloadConfigForm;
|
|
}
|
|
|
|
void PayloadProtocol::loadConfigWidget()
|
|
{
|
|
#define uintToHexStr(num, str, size) QString().setNum(num, 16)
|
|
|
|
configForm->cmbPatternMode->setCurrentIndex(data.pattern_mode());
|
|
configForm->lePattern->setText(uintToHexStr(data.pattern(), QString(), 4));
|
|
}
|
|
|
|
void PayloadProtocol::storeConfigWidget()
|
|
{
|
|
bool isOk;
|
|
|
|
data.set_pattern_mode((OstProto::Payload::DataPatternMode)
|
|
configForm->cmbPatternMode->currentIndex());
|
|
data.set_pattern(configForm->lePattern->text().remove(QChar(' ')).toULong(&isOk, 16));
|
|
}
|
|
|