2ec7fb30c2
- New Classes: o ProtocolManager - singleton with which all protocols register o ProtocolCollection - Aggregates all registered protocols; exports methods to work on all protocols o StreamBase - aggregates ProtocolCollection with Stream Core and Control; the client/server side stream classes now derive from StreamBase leading to major reduction in their code (more cleanup pending) - AbstractProtocol now supports the additional methods o createInstance() o protocolFrameSize() o protocolFrameOffset(), protocolFramePayloadSize() o protocolId(), payloadProtocolId() o protocolFrameCksum(), protocolFramePayloadCksum() 0 constructor takes an extra param - frameProtoList - Specific protocols - eth2, llc, snap, ip4, udp, tcp now return length, protocol id and cksums correctly (tcp/udp cksum pending) - StreamConfigDialog - protocol controls for length, cksum and protocolid are automatically updated (not fully working yet)
544 lines
13 KiB
C++
544 lines
13 KiB
C++
#include <qendian.h>
|
|
#include <QHostAddress>
|
|
|
|
#include "ip4.h"
|
|
|
|
Ip4ConfigForm *Ip4Protocol::configForm = NULL;
|
|
|
|
Ip4ConfigForm::Ip4ConfigForm(QWidget *parent)
|
|
: QWidget(parent)
|
|
{
|
|
setupUi(this);
|
|
|
|
connect(cmbIpSrcAddrMode, SIGNAL(currentIndexChanged(int)),
|
|
this, SLOT(on_cmbIpSrcAddrMode_currentIndexChanged(int)));
|
|
connect(cmbIpDstAddrMode, SIGNAL(currentIndexChanged(int)),
|
|
this, SLOT(on_cmbIpDstAddrMode_currentIndexChanged(int)));
|
|
}
|
|
|
|
void Ip4ConfigForm::on_cmbIpSrcAddrMode_currentIndexChanged(int index)
|
|
{
|
|
if (index == OstProto::Ip4::e_im_fixed)
|
|
{
|
|
leIpSrcAddrCount->setDisabled(true);
|
|
leIpSrcAddrMask->setDisabled(true);
|
|
}
|
|
else
|
|
{
|
|
leIpSrcAddrCount->setEnabled(true);
|
|
leIpSrcAddrMask->setEnabled(true);
|
|
}
|
|
}
|
|
|
|
void Ip4ConfigForm::on_cmbIpDstAddrMode_currentIndexChanged(int index)
|
|
{
|
|
if (index == OstProto::Ip4::e_im_fixed)
|
|
{
|
|
leIpDstAddrCount->setDisabled(true);
|
|
leIpDstAddrMask->setDisabled(true);
|
|
}
|
|
else
|
|
{
|
|
leIpDstAddrCount->setEnabled(true);
|
|
leIpDstAddrMask->setEnabled(true);
|
|
}
|
|
}
|
|
|
|
Ip4Protocol::Ip4Protocol(
|
|
ProtocolList &frameProtoList,
|
|
OstProto::StreamCore *parent)
|
|
: AbstractProtocol(frameProtoList, parent)
|
|
{
|
|
#if 0
|
|
PbHelper pbh;
|
|
|
|
pbh.ForceSetSingularDefault(&data);
|
|
#endif
|
|
if (configForm == NULL)
|
|
configForm = new Ip4ConfigForm;
|
|
}
|
|
|
|
Ip4Protocol::~Ip4Protocol()
|
|
{
|
|
}
|
|
|
|
AbstractProtocol* Ip4Protocol::createInstance(
|
|
ProtocolList &frameProtoList,
|
|
OstProto::StreamCore *streamCore)
|
|
{
|
|
return new Ip4Protocol(frameProtoList, streamCore);
|
|
}
|
|
|
|
void Ip4Protocol::protoDataCopyInto(OstProto::Stream &stream)
|
|
{
|
|
// FIXME: multiple headers
|
|
stream.MutableExtension(OstProto::ip4)->CopyFrom(data);
|
|
}
|
|
|
|
void Ip4Protocol::protoDataCopyFrom(const OstProto::Stream &stream)
|
|
{
|
|
// FIXME: multiple headers
|
|
if (stream.HasExtension(OstProto::ip4))
|
|
data.MergeFrom(stream.GetExtension(OstProto::ip4));
|
|
}
|
|
|
|
QString Ip4Protocol::name() const
|
|
{
|
|
return QString("Internet Protocol ver 4");
|
|
}
|
|
|
|
QString Ip4Protocol::shortName() const
|
|
{
|
|
return QString("IPv4");
|
|
}
|
|
|
|
quint32 Ip4Protocol::protocolId(ProtocolIdType type) const
|
|
{
|
|
switch(type)
|
|
{
|
|
case ProtocolIdLlc: return 0x060603;
|
|
case ProtocolIdEth: return 0x0800;
|
|
case ProtocolIdIp: return 0x04;
|
|
default:break;
|
|
}
|
|
|
|
return AbstractProtocol::protocolId(type);
|
|
}
|
|
|
|
int Ip4Protocol::fieldCount() const
|
|
{
|
|
return ip4_fieldCount;
|
|
}
|
|
|
|
QVariant Ip4Protocol::fieldData(int index, FieldAttrib attrib,
|
|
int streamIndex) const
|
|
{
|
|
switch (index)
|
|
{
|
|
case ip4_ver:
|
|
switch(attrib)
|
|
{
|
|
case FieldName:
|
|
return QString("Version");
|
|
case FieldValue:
|
|
return (data.ver_hdrlen() >> 4) & 0x0F;
|
|
case FieldTextValue:
|
|
return QString("%1").arg((data.ver_hdrlen() >> 4) & 0x0F);
|
|
case FieldFrameValue:
|
|
return QByteArray(1, (char)(data.ver_hdrlen() & 0xF0));
|
|
case FieldBitSize:
|
|
return 4;
|
|
default:
|
|
break;
|
|
}
|
|
break;
|
|
case ip4_hdrLen:
|
|
switch(attrib)
|
|
{
|
|
case FieldName:
|
|
return QString("Header Length");
|
|
case FieldValue:
|
|
return data.ver_hdrlen() & 0x0F;
|
|
case FieldTextValue:
|
|
return QString("%1").arg(data.ver_hdrlen() & 0x0F);
|
|
case FieldFrameValue:
|
|
return QByteArray(1, (char)(data.ver_hdrlen() << 4));
|
|
case FieldBitSize:
|
|
return 4;
|
|
default:
|
|
break;
|
|
}
|
|
break;
|
|
case ip4_tos:
|
|
switch(attrib)
|
|
{
|
|
case FieldName:
|
|
return QString("TOS/DSCP");
|
|
case FieldValue:
|
|
return data.tos();
|
|
case FieldFrameValue:
|
|
return QByteArray(1, (char) data.tos());
|
|
case FieldTextValue:
|
|
return QString("0x%1").
|
|
arg(data.tos(), 2, BASE_HEX, QChar('0'));;
|
|
default:
|
|
break;
|
|
}
|
|
break;
|
|
case ip4_totLen:
|
|
{
|
|
switch(attrib)
|
|
{
|
|
case FieldName:
|
|
return QString("Total Length");
|
|
case FieldValue:
|
|
{
|
|
int totlen;
|
|
totlen = data.is_override_totlen() ? data.totlen() :
|
|
(protocolFramePayloadSize() + 20);
|
|
return totlen;
|
|
}
|
|
case FieldFrameValue:
|
|
{
|
|
QByteArray fv;
|
|
int totlen;
|
|
totlen = data.is_override_totlen() ? data.totlen() :
|
|
(protocolFramePayloadSize() + 20);
|
|
fv.resize(2);
|
|
qToBigEndian((quint16) totlen, (uchar*) fv.data());
|
|
return fv;
|
|
}
|
|
case FieldTextValue:
|
|
{
|
|
int totlen;
|
|
totlen = data.is_override_totlen() ? data.totlen() :
|
|
(protocolFramePayloadSize() + 20);
|
|
return QString("%1").arg(totlen);
|
|
}
|
|
case FieldBitSize:
|
|
return 16;
|
|
default:
|
|
break;
|
|
}
|
|
break;
|
|
}
|
|
case ip4_id:
|
|
switch(attrib)
|
|
{
|
|
case FieldName:
|
|
return QString("Identification");
|
|
case FieldValue:
|
|
return data.id();
|
|
case FieldFrameValue:
|
|
{
|
|
QByteArray fv;
|
|
fv.resize(2);
|
|
qToBigEndian((quint16) data.id(), (uchar*)fv.data());
|
|
return fv;
|
|
}
|
|
case FieldTextValue:
|
|
return QString("0x%1").
|
|
arg(data.id(), 2, BASE_HEX, QChar('0'));;
|
|
default:
|
|
break;
|
|
}
|
|
break;
|
|
case ip4_flags:
|
|
switch(attrib)
|
|
{
|
|
case FieldName:
|
|
return QString("Flags");
|
|
case FieldValue:
|
|
return data.flags();
|
|
case FieldFrameValue:
|
|
return QByteArray(1, (char) data.flags());
|
|
case FieldTextValue:
|
|
{
|
|
QString s;
|
|
s.append("Unused:");
|
|
s.append(data.flags() & IP_FLAG_UNUSED ? "1" : "0");
|
|
s.append(" Don't Fragment:");
|
|
s.append(data.flags() & IP_FLAG_DF ? "1" : "0");
|
|
s.append(" More Fragments:");
|
|
s.append(data.flags() & IP_FLAG_MF ? "1" : "0");
|
|
return s;
|
|
}
|
|
case FieldBitSize:
|
|
return 3;
|
|
default:
|
|
break;
|
|
}
|
|
break;
|
|
case ip4_fragOfs:
|
|
switch(attrib)
|
|
{
|
|
case FieldName:
|
|
return QString("Fragment Offset");
|
|
case FieldValue:
|
|
return data.frag_ofs();
|
|
case FieldFrameValue:
|
|
{
|
|
QByteArray fv;
|
|
// FIXME need to shift for 13 bits
|
|
fv.resize(2);
|
|
qToBigEndian((quint16) data.frag_ofs(), (uchar*) fv.data());
|
|
return fv;
|
|
}
|
|
case FieldTextValue:
|
|
return QString("%1").arg(data.frag_ofs());
|
|
case FieldBitSize:
|
|
return 13;
|
|
default:
|
|
break;
|
|
}
|
|
break;
|
|
case ip4_ttl:
|
|
switch(attrib)
|
|
{
|
|
case FieldName:
|
|
return QString("Time to Live");
|
|
case FieldValue:
|
|
return data.ttl();
|
|
case FieldFrameValue:
|
|
return QByteArray(1, (char)data.ttl());
|
|
case FieldTextValue:
|
|
return QString("%1").arg(data.ttl());
|
|
default:
|
|
break;
|
|
}
|
|
break;
|
|
case ip4_proto:
|
|
{
|
|
switch(attrib)
|
|
{
|
|
case FieldName:
|
|
return QString("Protocol");
|
|
case FieldValue:
|
|
{
|
|
unsigned char id = payloadProtocolId(ProtocolIdIp);
|
|
return id;
|
|
}
|
|
case FieldFrameValue:
|
|
{
|
|
unsigned char id = payloadProtocolId(ProtocolIdIp);
|
|
return QByteArray(1, (char) id);
|
|
}
|
|
case FieldTextValue:
|
|
{
|
|
unsigned char id = payloadProtocolId(ProtocolIdIp);
|
|
return QString("0x%1").
|
|
arg(id, 2, BASE_HEX, QChar('0'));
|
|
}
|
|
default:
|
|
break;
|
|
}
|
|
break;
|
|
}
|
|
case ip4_cksum:
|
|
{
|
|
|
|
switch(attrib)
|
|
{
|
|
case FieldName:
|
|
return QString("Header Checksum");
|
|
case FieldValue:
|
|
{
|
|
quint16 cksum;
|
|
|
|
if (streamIndex < 0)
|
|
cksum = 0;
|
|
else if (data.is_override_cksum())
|
|
cksum = data.cksum();
|
|
else
|
|
cksum = protocolFrameCksum().toUInt();
|
|
return cksum;
|
|
}
|
|
case FieldFrameValue:
|
|
{
|
|
QByteArray fv;
|
|
quint16 cksum;
|
|
|
|
if (streamIndex < 0)
|
|
cksum = 0;
|
|
else if (data.is_override_cksum())
|
|
cksum = data.cksum();
|
|
else
|
|
cksum = protocolFrameCksum().toUInt();
|
|
fv.resize(2);
|
|
qToBigEndian((quint16) cksum, (uchar*) fv.data());
|
|
return fv;
|
|
}
|
|
case FieldTextValue:
|
|
{
|
|
quint16 cksum;
|
|
|
|
if (streamIndex < 0)
|
|
cksum = 0;
|
|
else if (data.is_override_cksum())
|
|
cksum = data.cksum();
|
|
else
|
|
cksum = protocolFrameCksum().toUInt();
|
|
return QString("0x%1").
|
|
arg(cksum, 4, BASE_HEX, QChar('0'));;
|
|
}
|
|
case FieldBitSize:
|
|
return 16;
|
|
default:
|
|
break;
|
|
}
|
|
break;
|
|
}
|
|
case ip4_srcAddr:
|
|
switch(attrib)
|
|
{
|
|
case FieldName:
|
|
return QString("Source");
|
|
case FieldValue:
|
|
return data.src_ip();
|
|
case FieldFrameValue:
|
|
{
|
|
QByteArray fv;
|
|
fv.resize(4);
|
|
qToBigEndian((quint32) data.src_ip(), (uchar*) fv.data());
|
|
return fv;
|
|
}
|
|
case FieldTextValue:
|
|
return QHostAddress(data.src_ip()).toString();
|
|
default:
|
|
break;
|
|
}
|
|
break;
|
|
case ip4_dstAddr:
|
|
switch(attrib)
|
|
{
|
|
case FieldName:
|
|
return QString("Destination");
|
|
case FieldValue:
|
|
return data.dst_ip();
|
|
case FieldFrameValue:
|
|
{
|
|
QByteArray fv;
|
|
fv.resize(4);
|
|
qToBigEndian((quint32) data.dst_ip(), (uchar*) fv.data());
|
|
return fv;
|
|
}
|
|
case FieldTextValue:
|
|
return QHostAddress(data.dst_ip()).toString();
|
|
default:
|
|
break;
|
|
}
|
|
break;
|
|
|
|
// Meta fields
|
|
|
|
case ip4_isOverrideVer:
|
|
case ip4_isOverrideHdrLen:
|
|
case ip4_isOverrideTotLen:
|
|
case ip4_isOverrideCksum:
|
|
|
|
case ip4_srcAddrMode:
|
|
case ip4_srcAddrCount:
|
|
case ip4_srcAddrMask:
|
|
|
|
case ip4_dstAddrMode:
|
|
case ip4_dstAddrCount:
|
|
case ip4_dstAddrMask:
|
|
switch(attrib)
|
|
{
|
|
case FieldIsMeta:
|
|
return true;
|
|
default:
|
|
break;
|
|
}
|
|
break;
|
|
|
|
default:
|
|
break;
|
|
}
|
|
|
|
return AbstractProtocol::fieldData(index, attrib, streamIndex);
|
|
}
|
|
|
|
bool Ip4Protocol::setFieldData(int index, const QVariant &value,
|
|
FieldAttrib attrib)
|
|
{
|
|
bool isOk = false;
|
|
|
|
if (attrib != FieldValue)
|
|
return false;
|
|
|
|
switch (index)
|
|
{
|
|
case ip4_proto:
|
|
{
|
|
uint proto = value.toUInt(&isOk);
|
|
if (isOk)
|
|
data.set_proto(proto);
|
|
}
|
|
default:
|
|
break;
|
|
}
|
|
return isOk;
|
|
}
|
|
|
|
|
|
QWidget* Ip4Protocol::configWidget()
|
|
{
|
|
return configForm;
|
|
}
|
|
|
|
void Ip4Protocol::loadConfigWidget()
|
|
{
|
|
configForm->leIpVersion->setText(QString().setNum(data.ver_hdrlen() >> 4));
|
|
configForm->cbIpVersionOverride->setChecked(data.is_override_ver());
|
|
|
|
configForm->leIpHdrLen->setText(QString().setNum(data.ver_hdrlen() & 0x0F));
|
|
configForm->cbIpHdrLenOverride->setChecked(data.is_override_hdrlen());
|
|
|
|
configForm->leIpTos->setText(uintToHexStr(data.tos(), 1));
|
|
configForm->leIpLength->setText(fieldData(ip4_totLen, FieldValue).toString());
|
|
configForm->cbIpLengthOverride->setChecked(data.is_override_totlen());
|
|
|
|
configForm->leIpId->setText(uintToHexStr(data.id(), 2));
|
|
configForm->leIpFragOfs->setText(QString().setNum(data.frag_ofs()));
|
|
configForm->cbIpFlagsDf->setChecked((data.flags() & IP_FLAG_DF) > 0);
|
|
configForm->cbIpFlagsMf->setChecked((data.flags() & IP_FLAG_MF) > 0);
|
|
|
|
configForm->leIpTtl->setText(QString().setNum(data.ttl()));
|
|
configForm->leIpProto->setText(uintToHexStr(
|
|
fieldData(ip4_proto, FieldValue).toUInt(), 1));
|
|
|
|
configForm->leIpCksum->setText(uintToHexStr(
|
|
fieldData(ip4_cksum, FieldValue).toUInt(), 2));
|
|
configForm->cbIpCksumOverride->setChecked(data.is_override_cksum());
|
|
|
|
configForm->leIpSrcAddr->setText(QHostAddress(data.src_ip()).toString());
|
|
configForm->cmbIpSrcAddrMode->setCurrentIndex(data.src_ip_mode());
|
|
configForm->leIpSrcAddrCount->setText(QString().setNum(data.src_ip_count()));
|
|
configForm->leIpSrcAddrMask->setText(QHostAddress(data.src_ip_mask()).toString());
|
|
|
|
configForm->leIpDstAddr->setText(QHostAddress(data.dst_ip()).toString());
|
|
configForm->cmbIpDstAddrMode->setCurrentIndex(data.dst_ip_mode());
|
|
configForm->leIpDstAddrCount->setText(QString().setNum(data.dst_ip_count()));
|
|
configForm->leIpDstAddrMask->setText(QHostAddress(data.dst_ip_mask()).toString());
|
|
}
|
|
|
|
void Ip4Protocol::storeConfigWidget()
|
|
{
|
|
uint ff = 0;
|
|
bool isOk;
|
|
|
|
data.set_is_override_ver(configForm->cbIpVersionOverride->isChecked());
|
|
data.set_ver_hdrlen(((configForm->leIpVersion->text().toULong(&isOk) & 0x0F) << 4) |
|
|
(configForm->leIpHdrLen->text().toULong(&isOk) & 0x0F));
|
|
data.set_is_override_hdrlen(configForm->cbIpHdrLenOverride->isChecked());
|
|
|
|
data.set_tos(configForm->leIpTos->text().toULong(&isOk, 16));
|
|
|
|
data.set_totlen(configForm->leIpLength->text().toULong(&isOk));
|
|
data.set_is_override_totlen(configForm->cbIpLengthOverride->isChecked());
|
|
|
|
data.set_id(configForm->leIpId->text().remove(QChar(' ')).toULong(&isOk, 16));
|
|
data.set_frag_ofs(configForm->leIpFragOfs->text().toULong(&isOk));
|
|
|
|
if (configForm->cbIpFlagsDf->isChecked()) ff |= IP_FLAG_DF;
|
|
if (configForm->cbIpFlagsMf->isChecked()) ff |= IP_FLAG_MF;
|
|
data.set_flags(ff);
|
|
|
|
data.set_ttl(configForm->leIpTtl->text().toULong(&isOk));
|
|
data.set_proto(configForm->leIpProto->text().remove(QChar(' ')).toULong(&isOk, 16));
|
|
|
|
data.set_cksum(configForm->leIpCksum->text().remove(QChar(' ')).toULong(&isOk));
|
|
data.set_is_override_cksum(configForm->cbIpCksumOverride->isChecked());
|
|
|
|
data.set_src_ip(QHostAddress(configForm->leIpSrcAddr->text()).toIPv4Address());
|
|
data.set_src_ip_mode((OstProto::Ip4_IpAddrMode)configForm->cmbIpSrcAddrMode->currentIndex());
|
|
data.set_src_ip_count(configForm->leIpSrcAddrCount->text().toULong(&isOk));
|
|
data.set_src_ip_mask(QHostAddress(configForm->leIpSrcAddrMask->text()).toIPv4Address());
|
|
|
|
data.set_dst_ip(QHostAddress(configForm->leIpDstAddr->text()).toIPv4Address());
|
|
data.set_dst_ip_mode((OstProto::Ip4_IpAddrMode)configForm->cmbIpDstAddrMode->currentIndex());
|
|
data.set_dst_ip_count(configForm->leIpDstAddrCount->text().toULong(&isOk));
|
|
}
|
|
|