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)
107 lines
2.0 KiB
C++
107 lines
2.0 KiB
C++
#include "protocolcollection.h"
|
|
|
|
extern ProtocolManager OstProtocolManager;
|
|
|
|
ProtocolCollection::ProtocolCollection(ProtocolList &streamProtocols,
|
|
OstProto::StreamCore *streamCore)
|
|
: protoManager(OstProtocolManager)
|
|
{
|
|
// Create an instance of each registered protocol
|
|
|
|
QMapIterator<int, void*> iter(protoManager.factory);
|
|
|
|
while (iter.hasNext())
|
|
{
|
|
AbstractProtocol* (*p)(ProtocolList&, OstProto::StreamCore*);
|
|
AbstractProtocol* q;
|
|
|
|
iter.next();
|
|
p = (AbstractProtocol* (*)(ProtocolList&, OstProto::StreamCore*))
|
|
iter.value();
|
|
q = (*p)(streamProtocols, streamCore);
|
|
|
|
protocols.insert(iter.key(), q);
|
|
}
|
|
}
|
|
|
|
ProtocolCollection::~ProtocolCollection()
|
|
{
|
|
QMutableMapIterator<int,AbstractProtocol*> iter(protocols);
|
|
|
|
while (iter.hasNext())
|
|
{
|
|
iter.next();
|
|
if (iter.value())
|
|
{
|
|
delete iter.value();
|
|
iter.remove();
|
|
}
|
|
}
|
|
}
|
|
|
|
void ProtocolCollection::protoDataCopyFrom(const OstProto::Stream &stream) const
|
|
{
|
|
QMapIterator<int,AbstractProtocol*> iter(protocols);
|
|
|
|
while (iter.hasNext())
|
|
{
|
|
iter.next();
|
|
if (iter.value())
|
|
{
|
|
iter.value()->protoDataCopyFrom(stream);
|
|
}
|
|
}
|
|
}
|
|
|
|
void ProtocolCollection::protoDataCopyInto(OstProto::Stream &stream) const
|
|
{
|
|
QMapIterator<int,AbstractProtocol*> iter(protocols);
|
|
|
|
while (iter.hasNext())
|
|
{
|
|
iter.next();
|
|
if (iter.value())
|
|
{
|
|
iter.value()->protoDataCopyInto(stream);
|
|
}
|
|
}
|
|
}
|
|
|
|
void ProtocolCollection::loadConfigWidgets() const
|
|
{
|
|
QMapIterator<int,AbstractProtocol*> iter(protocols);
|
|
|
|
while (iter.hasNext())
|
|
{
|
|
iter.next();
|
|
if (iter.value())
|
|
{
|
|
iter.value()->loadConfigWidget();
|
|
}
|
|
}
|
|
}
|
|
|
|
void ProtocolCollection::storeConfigWidgets() const
|
|
{
|
|
QMapIterator<int,AbstractProtocol*> iter(protocols);
|
|
|
|
while (iter.hasNext())
|
|
{
|
|
iter.next();
|
|
if (iter.value())
|
|
{
|
|
iter.value()->storeConfigWidget();
|
|
}
|
|
}
|
|
}
|
|
|
|
AbstractProtocol* ProtocolCollection::protocol(int protoNum)
|
|
{
|
|
return protocols.value(protoNum);
|
|
}
|
|
|
|
AbstractProtocol* ProtocolCollection::protocol(QString protoName)
|
|
{
|
|
return protocols.value(protoManager.nameToNumberMap.value(protoName));
|
|
}
|