matthew/ostinato
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Finished basic PacketModel to feed actual data into PacketView

Browse Source 
TODO: PacketModel to feed data into DumpView
This commit is contained in:
Srivats P. 2008-06-14 06:35:17 +00:00
parent 03ab9c4349
commit a480be5a23
2 changed files with 519 additions and 53 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

537
client/packetmodel.cpp
View File

@ -1,9 +1,11 @@
#include <QHostAddress>
#include "packetmodel.h" #include "packetmodel.h"
PacketModel::PacketModel(Stream *pStream, QObject *parent) PacketModel::PacketModel(Stream *pStream, QObject *parent)
{ {
mpStream = pStream; mpStream = pStream;
populatePacketProtocols(); populatePacketProtocols();
#if 1
registerFrameTypeProto(); registerFrameTypeProto();
registerVlanProto(); registerVlanProto();
registerIpProto(); registerIpProto();
@ -14,6 +16,7 @@ PacketModel::PacketModel(Stream *pStream, QObject *parent)
registerIgmpProto(); registerIgmpProto();
registerData(); registerData();
registerInvalidProto(); registerInvalidProto();
#endif
} }
int PacketModel::rowCount(const QModelIndex &parent) const int PacketModel::rowCount(const QModelIndex &parent) const
@ -23,7 +26,7 @@ int PacketModel::rowCount(const QModelIndex &parent) const
// Parent - Invalid i.e. Invisible Root. // Parent - Invalid i.e. Invisible Root.
// Children - Protocol (Top Level) Items // Children - Protocol (Top Level) Items
if (!parent.isValid()) if (!parent.isValid())
return mPacketProtocols.count(); return protoCount();
// Parent - Valid Item // Parent - Valid Item
parentId.w = parent.internalId(); parentId.w = parent.internalId();
@ -32,13 +35,12 @@ int PacketModel::rowCount(const QModelIndex &parent) const
case ITYP_PROTOCOL: case ITYP_PROTOCOL:
return fieldCount(parentId.ws.protocol); return fieldCount(parentId.ws.protocol);
case ITYP_FIELD: case ITYP_FIELD:
return subfieldCount(parentId.ws.protocol, parentId.ws.field);
case ITYP_SUBFIELD:
return 0; return 0;
default: default:
qWarning("%s: Unhandled ItemType", __FUNCTION__); qWarning("%s: Unhandled ItemType", __FUNCTION__);
} }
Q_ASSERT(1 == 1); // Unreachable code
qWarning("%s: Catch all - need to investigate", __FUNCTION__); qWarning("%s: Catch all - need to investigate", __FUNCTION__);
return 0; // catch all return 0; // catch all
} }
@ -62,7 +64,7 @@ QModelIndex PacketModel::index(int row, int col, const QModelIndex &parent) cons
{ {
id.w = 0; id.w = 0;
id.ws.type = ITYP_PROTOCOL; id.ws.type = ITYP_PROTOCOL;
id.ws.protocol = mPacketProtocols.at(row); id.ws.protocol = row;
index = createIndex(row, col, id.w); index = createIndex(row, col, id.w);
goto _exit; goto _exit;
} }
@ -74,21 +76,18 @@ QModelIndex PacketModel::index(int row, int col, const QModelIndex &parent) cons
{ {
case ITYP_PROTOCOL: case ITYP_PROTOCOL:
id.ws.type = ITYP_FIELD; id.ws.type = ITYP_FIELD;
id.ws.field = row;
index = createIndex(row, col, id.w); index = createIndex(row, col, id.w);
goto _exit; goto _exit;
case ITYP_FIELD: case ITYP_FIELD:
// TODO(MED): Nothing till subfield support is added
goto _exit;
case ITYP_SUBFIELD:
goto _exit; goto _exit;
default: default:
qWarning("%s: Unhandled ItemType", __FUNCTION__); qWarning("%s: Unhandled ItemType", __FUNCTION__);
} }
Q_ASSERT(1 == 1); // Unreachable code
_exit: _exit:
return index; return index;
} }
@ -111,19 +110,15 @@ QModelIndex PacketModel::parent(const QModelIndex &index) const
case ITYP_FIELD: case ITYP_FIELD:
parentId.ws.type = ITYP_PROTOCOL; parentId.ws.type = ITYP_PROTOCOL;
parentId.ws.field = 0; parentIndex = createIndex(id.ws.protocol, 0, parentId.w);
parentIndex = createIndex(mPacketProtocols.indexOf(parentId.ws.protocol), 0,
parentId.w);
goto _exit;
case ITYP_SUBFIELD:
// TODO(MED): invalid index till subfield support is added
goto _exit; goto _exit;
default: default:
qWarning("%s: Unhandled ItemType", __FUNCTION__); qWarning("%s: Unhandled ItemType", __FUNCTION__);
} }
Q_ASSERT(1 == 1); // Unreachable code
_exit: _exit:
return parentIndex; return parentIndex;
} }
@ -140,35 +135,30 @@ QVariant PacketModel::data(const QModelIndex &index, int role) const
return QVariant(); return QVariant();
id.w = index.internalId(); id.w = index.internalId();
foreach(proto, mProtocols)
{
if (proto.handle == id.ws.protocol)
goto _found;
}
return QVariant();
_found:
switch(id.ws.type) switch(id.ws.type)
{ {
case ITYP_PROTOCOL: case ITYP_PROTOCOL:
return proto.name; return protoName(id.ws.protocol);
case ITYP_FIELD: case ITYP_FIELD:
return proto.fieldList.at(id.ws.field).name; return fieldName(id.ws.protocol, index.row()) +
QString(" : ") +
case ITYP_SUBFIELD: fieldTextValue(id.ws.protocol, index.row()).toString();
return QVariant(); // TODO(MED): Till subfield support is added
default: default:
qWarning("%s: Unhandled ItemType", __FUNCTION__); qWarning("%s: Unhandled ItemType", __FUNCTION__);
} }
Q_ASSERT(1 == 1); // Unreachable code
return QVariant(); return QVariant();
} }
/* /*
** --------------- Private Stuff ----------------- ** --------------- Private Stuff -----------------
** FIXME(MED): Move these to the Stream Class
**
*/ */
void PacketModel::populatePacketProtocols() void PacketModel::populatePacketProtocols()
{ {
@ -193,10 +183,13 @@ void PacketModel::populatePacketProtocols()
break; break;
case Stream::e_ft_802_3_llc: case Stream::e_ft_802_3_llc:
mPacketProtocols.append(PTYP_L2_NONE);
proto = PTYP_L2_802_3_LLC; proto = PTYP_L2_802_3_LLC;
break; break;
case Stream::e_ft_snap: case Stream::e_ft_snap:
mPacketProtocols.append(PTYP_L2_NONE);
mPacketProtocols.append(PTYP_L2_802_3_LLC);
proto = PTYP_L2_SNAP; proto = PTYP_L2_SNAP;
break; break;
@ -260,29 +253,494 @@ _data:
mPacketProtocols.append(PTYP_DATA); mPacketProtocols.append(PTYP_DATA);
} }
int PacketModel::fieldCount(uint protocol) const int PacketModel::protoCount() const
{
return mPacketProtocols.count();
}
int PacketModel::fieldCount(int protocol) const
{ {
ProtocolInfo proto; ProtocolInfo proto;
if (protocol >= mPacketProtocols.count())
return 0;
foreach(proto, mProtocols) foreach(proto, mProtocols)
{ {
if (proto.handle == protocol) if (proto.handle == mPacketProtocols.at(protocol))
{ {
qDebug("proto=%d, name=%s",protocol,proto.name.toAscii().data()); qDebug("proto=%d, name=%s",protocol,proto.name.toAscii().data());
qDebug("fieldcount = %d", proto.fieldList.size()); qDebug("fieldcount = %d", proto.fieldList.size());
return proto.fieldList.size(); return proto.fieldList.count();
} }
} }
return 0; return 0;
} }
int PacketModel::subfieldCount(uint protocol, int field) const QString PacketModel::protoName(int protocol) const
{ {
// TODO(MED): Till subfield support is added ProtocolInfo proto;
return 0;
if (protocol >= mPacketProtocols.count())
return 0;
foreach(proto, mProtocols)
{
if (proto.handle == mPacketProtocols.at(protocol))
{
qDebug("proto=%d, name=%s",protocol,proto.name.toAscii().data());
qDebug("fieldcount = %d", proto.fieldList.size());
return proto.name;
}
}
return QString();
} }
QString PacketModel::fieldName(int protocol, int field) const
{
ProtocolInfo proto;
if (protocol >= mPacketProtocols.count())
return 0;
foreach(proto, mProtocols)
{
if (proto.handle == mPacketProtocols.at(protocol))
{
qDebug("proto=%d, name=%s",protocol,proto.name.toAscii().data());
qDebug("fieldcount = %d", proto.fieldList.size());
if (field >= proto.fieldList.count())
return QString();
return proto.fieldList.at(field).name;
}
}
return QString();
}
QVariant PacketModel::fieldTextValue(int protocol, int field) const
{
if (protocol >= mPacketProtocols.count())
return QVariant();
switch(mPacketProtocols.at(protocol))
{
case PTYP_L2_NONE:
case PTYP_L2_ETH_2:
return ethField(field, FROL_TEXT_VALUE);
case PTYP_L2_802_3_RAW:
//return dot3Field(field, FROL_TEXT_VALUE); // FIXME(HIGH)
return ethField(field, FROL_TEXT_VALUE);
case PTYP_L2_802_3_LLC:
return llcField(field, FROL_TEXT_VALUE);
case PTYP_L2_SNAP:
return snapField(field, FROL_TEXT_VALUE);
case PTYP_SVLAN:
return svlanField(field, FROL_TEXT_VALUE);
case PTYP_CVLAN:
// return cvlanField(field, FROL_TEXT_VALUE); // FIXME(HIGH)
return svlanField(field, FROL_TEXT_VALUE);
case PTYP_L3_IP:
return ipField(field, FROL_TEXT_VALUE);
case PTYP_L3_ARP:
return QString(); // FIXME(HIGH)
case PTYP_L4_TCP:
return tcpField(field, FROL_TEXT_VALUE);
case PTYP_L4_UDP:
return udpField(field, FROL_TEXT_VALUE);
case PTYP_L4_ICMP:
return QString(); // FIXME(HIGH)
case PTYP_L4_IGMP:
return QString(); // FIXME(HIGH)
case PTYP_INVALID:
return QString(); // FIXME(HIGH)
case PTYP_DATA:
return QString(); // FIXME(HIGH)
}
return QString();
}
QVariant PacketModel::ethField(int field, int role) const
{
FieldInfo info;
// FIXME(MED): Mac Addr formatting
switch(field)
{
case 0:
info.name = QString("Destination Mac Address");
info.textValue = QString("%1%2").
arg(mpStream->l2.eth.dstMacMshw, 4, BASE_HEX, QChar('0')).
arg(mpStream->l2.eth.dstMacLsw, 8, BASE_HEX, QChar('0'));
break;
case 1:
info.name = QString("Source Mac Address");
info.textValue = QString("%1%2").
arg(mpStream->l2.eth.srcMacMshw, 4, BASE_HEX, QChar('0')).
arg(mpStream->l2.eth.srcMacLsw, 8, BASE_HEX, QChar('0'));
break;
case 2:
info.name = QString("Type");
info.textValue = QString("0x%1").
arg(mpStream->proto.etherType, 4, BASE_HEX, QChar('0'));
break;
default:
info.name = QString();
info.textValue = QString();
}
switch(role)
{
case FROL_NAME:
return info.name;
case FROL_TEXT_VALUE:
return info.textValue;
default:
;
}
Q_ASSERT(1 == 1); // Unreachable code
return QVariant();
}
QVariant PacketModel::llcField(int field, int role) const
{
FieldInfo info;
switch(field)
{
case 0:
info.name = QString("DSAP");
info.textValue = QString("0x%1").
arg(mpStream->proto.dsap, 2, BASE_HEX, QChar('0'));
break;
case 1:
info.name = QString("SSAP");
info.textValue = QString("0x%1").
arg(mpStream->proto.ssap, 2, BASE_HEX, QChar('0'));
break;
case 2:
info.name = QString("Control");
info.textValue = QString("0x%1").
arg(mpStream->proto.ctl, 2, BASE_HEX, QChar('0'));
break;
default:
info.name = QString();
info.textValue = QString();
}
switch(role)
{
case FROL_NAME:
return info.name;
case FROL_TEXT_VALUE:
return info.textValue;
default:
;
}
Q_ASSERT(1 == 1); // Unreachable code
return QVariant();
}
QVariant PacketModel::snapField(int field, int role) const
{
FieldInfo info;
switch(field)
{
case 0:
info.name = QString("OUI");
info.textValue = QString("0x%1%2").
arg(mpStream->proto.ouiMsb, 2, BASE_HEX, QChar('0')).
arg(mpStream->proto.ouiLshw, 4, BASE_HEX, QChar('0'));
break;
case 1:
info.name = QString("Type");
info.textValue = QString("0x%1").
arg(mpStream->proto.etherType, 4, BASE_HEX, QChar('0'));
break;
default:
info.name = QString();
info.textValue = QString();
}
switch(role)
{
case FROL_NAME:
return info.name;
case FROL_TEXT_VALUE:
return info.textValue;
default:
;
}
Q_ASSERT(1 == 1); // Unreachable code
return QVariant();
}
QVariant PacketModel::svlanField(int field, int role) const
{
FieldInfo info;
switch(field)
{
case 0:
info.name = QString("TPID");
info.textValue = QString("0x%1").
arg(mpStream->l2.eth.stpid, 4, BASE_HEX, QChar('0'));
break;
case 1:
info.name = QString("PCP");
info.textValue = QString("%1").
arg(mpStream->l2.eth.svlanPrio);
break;
case 2:
info.name = QString("DE");
info.textValue = QString("%1").
arg(mpStream->l2.eth.svlanCfi);
break;
case 3:
info.name = QString("VlanId");
info.textValue = QString("%1").
arg(mpStream->l2.eth.svlanId);
break;
default:
info.name = QString();
info.textValue = QString();
}
switch(role)
{
case FROL_NAME:
return info.name;
case FROL_TEXT_VALUE:
return info.textValue;
default:
;
}
Q_ASSERT(1 == 1); // Unreachable code
return QVariant();
}
QVariant PacketModel::ipField(int field, int role) const
{
FieldInfo info;
switch(field)
{
case 0:
info.name = QString("Version");
info.textValue = QString("%1").
arg(mpStream->l3.ip.ver);
break;
case 1:
info.name = QString("Header Length");
info.textValue = QString("%1").
arg(mpStream->l3.ip.hdrLen);
break;
case 2:
info.name = QString("TOS/DSCP");
info.textValue = QString("0x%1").
arg(mpStream->l3.ip.tos, 2, BASE_HEX, QChar('0'));
break;
case 3:
info.name = QString("Total Length");
info.textValue = QString("%1").
arg(mpStream->l3.ip.totLen);
break;
case 4:
info.name = QString("ID");
info.textValue = QString("0x%1").
arg(mpStream->l3.ip.id, 2, BASE_HEX, QChar('0'));
break;
case 5:
info.name = QString("Flags");
info.textValue = QString("0x%1").
arg(mpStream->l3.ip.flags, 2, BASE_HEX, QChar('0')); // FIXME(HIGH)
break;
case 6:
info.name = QString("Fragment Offset");
info.textValue = QString("%1").
arg(mpStream->l3.ip.fragOfs);
break;
case 7:
info.name = QString("TTL");
info.textValue = QString("%1").
arg(mpStream->l3.ip.ttl);
break;
case 8:
info.name = QString("Protocol Type");
info.textValue = QString("0x%1").
arg(mpStream->l3.ip.proto, 2, BASE_HEX, QChar('0'));
break;
case 9:
info.name = QString("Checksum");
info.textValue = QString("0x%1").
arg(mpStream->l3.ip.cksum, 4, BASE_HEX, QChar('0'));
break;
case 10:
info.name = QString("Source IP");
info.textValue = QHostAddress(mpStream->l3.ip.srcIp).toString();
break;
case 11:
info.name = QString("Destination IP");
info.textValue = QHostAddress(mpStream->l3.ip.dstIp).toString();
break;
default:
info.name = QString();
info.textValue = QString();
}
switch(role)
{
case FROL_NAME:
return info.name;
case FROL_TEXT_VALUE:
return info.textValue;
default:
;
}
Q_ASSERT(1 == 1); // Unreachable code
return QVariant();
}
QVariant PacketModel::tcpField(int field, int role) const
{
FieldInfo info;
switch(field)
{
case 0:
info.name = QString("Source Port");
info.textValue = QString("%1").
arg(mpStream->l4.tcp.srcPort);
break;
case 1:
info.name = QString("Destination Port");
info.textValue = QString("%1").
arg(mpStream->l4.tcp.dstPort);
break;
case 2:
info.name = QString("Seq Number");
info.textValue = QString("%1").
arg(mpStream->l4.tcp.seqNum);
break;
case 3:
info.name = QString("Ack Number");
info.textValue = QString("%1").
arg(mpStream->l4.tcp.ackNum);
break;
case 4:
info.name = QString("Header Length");
info.textValue = QString("%1").
arg(mpStream->l4.tcp.hdrLen);
break;
case 5:
info.name = QString("Reserved");
info.textValue = QString("%1").
arg(mpStream->l4.tcp.rsvd);
break;
case 6:
info.name = QString("Flags");
info.textValue = QString("0x%1").
arg(mpStream->l4.tcp.flags, 2, BASE_HEX, QChar('0'));
break;
case 7:
info.name = QString("Window");
info.textValue = QString("%1").
arg(mpStream->l4.tcp.flags);
break;
case 8:
info.name = QString("Checksum");
info.textValue = QString("0x%1").
arg(mpStream->l4.tcp.cksum, 4, BASE_HEX, QChar('0'));
break;
case 9:
info.name = QString("Urgent Pointer");
info.textValue = QString("%1").
arg(mpStream->l4.tcp.urgPtr);
break;
default:
info.name = QString();
info.textValue = QString();
}
switch(role)
{
case FROL_NAME:
return info.name;
case FROL_TEXT_VALUE:
return info.textValue;
default:
;
}
Q_ASSERT(1 == 1); // Unreachable code
return QVariant();
}
QVariant PacketModel::udpField(int field, int role) const
{
FieldInfo info;
switch(field)
{
case 0:
info.name = QString("Source Port");
info.textValue = QString("%1").
arg(mpStream->l4.udp.srcPort);
break;
case 1:
info.name = QString("Destination Port");
info.textValue = QString("%1").
arg(mpStream->l4.udp.dstPort);
break;
case 2:
info.name = QString("Total Length");
info.textValue = QString("%1").
arg(mpStream->l4.udp.totLen);
break;
case 3:
info.name = QString("Checksum");
info.textValue = QString("0x%1").
arg(mpStream->l4.udp.cksum, 4, BASE_HEX, QChar('0'));
break;
default:
info.name = QString();
info.textValue = QString();
}
switch(role)
{
case FROL_NAME:
return info.name;
case FROL_TEXT_VALUE:
return info.textValue;
default:
;
}
Q_ASSERT(1 == 1); // Unreachable code
return QVariant();
}
/* /*
** ------------- Registration Functions --------------- ** ------------- Registration Functions ---------------
*/ */
@ -331,18 +789,11 @@ void PacketModel::registerFrameTypeProto()
registerField(PTYP_L2_802_3_RAW, "Length", "len"); registerField(PTYP_L2_802_3_RAW, "Length", "len");
registerProto(PTYP_L2_802_3_LLC, "802.3 LLC", "dot3llc"); registerProto(PTYP_L2_802_3_LLC, "802.3 LLC", "dot3llc");
registerField(PTYP_L2_802_3_LLC, "Destination Mac", "dstMac");
registerField(PTYP_L2_802_3_LLC, "Source Mac", "srcMac");
registerField(PTYP_L2_802_3_LLC, "Destination Service Acces Point", "dsap"); registerField(PTYP_L2_802_3_LLC, "Destination Service Acces Point", "dsap");
registerField(PTYP_L2_802_3_LLC, "Source Service Acces Point", "ssap"); registerField(PTYP_L2_802_3_LLC, "Source Service Acces Point", "ssap");
registerField(PTYP_L2_802_3_LLC, "Control", "ctl"); registerField(PTYP_L2_802_3_LLC, "Control", "ctl");
registerProto(PTYP_L2_SNAP, "802.3 LLC SNAP", "dot3snap"); registerProto(PTYP_L2_SNAP, "SNAP", "dot3snap");
registerField(PTYP_L2_SNAP, "Destination Mac", "dstMac");
registerField(PTYP_L2_SNAP, "Source Mac", "srcMac");
registerField(PTYP_L2_SNAP, "Destination Service Acces Point", "dsap");
registerField(PTYP_L2_SNAP, "Source Service Access Point", "ssap");
registerField(PTYP_L2_SNAP, "Control", "ctl");
registerField(PTYP_L2_SNAP, "Organisationally Unique Identifier", "oui"); registerField(PTYP_L2_SNAP, "Organisationally Unique Identifier", "oui");
registerField(PTYP_L2_SNAP, "Type", "type"); registerField(PTYP_L2_SNAP, "Type", "type");

35
client/packetmodel.h
View File

@ -24,20 +24,21 @@ private:
quint32 w; quint32 w;
struct struct
{ {
quint8 type; quint16 type;
#define ITYP_PROTOCOL 1 #define ITYP_PROTOCOL 1
#define ITYP_FIELD 2 #define ITYP_FIELD 2
#define ITYP_SUBFIELD 3 quint16 protocol;
quint8 protocol;
quint8 field;
quint8 subfield;
} ws; } ws;
} IndexId; } IndexId;
Stream *mpStream;
QList<uint> mPacketProtocols;
typedef struct typedef struct
{ {
QString name; QString name;
QString abbr; QString abbr;
QString textValue;
} FieldInfo; } FieldInfo;
typedef struct typedef struct
@ -48,10 +49,7 @@ private:
QList<FieldInfo> fieldList; QList<FieldInfo> fieldList;
} ProtocolInfo; } ProtocolInfo;
Stream *mpStream;
QList<uint> mPacketProtocols;
QList<ProtocolInfo> mProtocols; QList<ProtocolInfo> mProtocols;
void registerProto(uint handle, char *name, char *abbr); void registerProto(uint handle, char *name, char *abbr);
void registerField(uint protoHandle, char *name, char *abbr); void registerField(uint protoHandle, char *name, char *abbr);
@ -67,13 +65,26 @@ private:
void registerInvalidProto(); void registerInvalidProto();
void populatePacketProtocols(); void populatePacketProtocols();
int fieldCount(uint protocol) const;
int subfieldCount(uint protocol, int field) const; int protoCount() const;
int fieldCount(int protocol) const;
QString protoName(int protocol) const;
QString fieldName(int protocol, int field) const;
QVariant fieldTextValue(int protocol, int field) const;
QVariant ethField(int field, int role) const;
QVariant llcField(int field, int role) const;
QVariant snapField(int field, int role) const;
QVariant svlanField(int field, int role) const;
QVariant ipField(int field, int role) const;
QVariant tcpField(int field, int role) const;
QVariant udpField(int field, int role) const;
// FIXME(HIGH): Is this how I want this? // FIXME(HIGH): Is this how I want this?
#define PTYP_L2_NONE 1 #define PTYP_L2_NONE 1
#define PTYP_L2_ETH_2 2 #define PTYP_L2_ETH_2 2
#define PTYP_L2_802_3_RAW 3 #define PTYP_L2_802_3_RAW 3
#define PTYP_L2_802_3_LLC 4 #define PTYP_L2_802_3_LLC 4
#define PTYP_L2_SNAP 5 #define PTYP_L2_SNAP 5
@ -91,6 +102,10 @@ private:
#define PTYP_INVALID 0 #define PTYP_INVALID 0
#define PTYP_DATA 0xFF #define PTYP_DATA 0xFF
#define FROL_NAME 1
#define FROL_TEXT_VALUE 2
#define BASE_HEX 16
}; };
#endif #endif