a498dbf21a
For UDP encaps like VxLAN or Geneve, which can contain IP as payload, the UDP checksum was incorrect because when summing UDP payload (i.e. IP), we skipped the IPv4 checksum field, which should not be skipped in this case. Similar issue, for ICMP with IP as payload, ICMP checksum was incorrect. Essentially, any protocol which checksums over its payload and the payload contains protocols with checksum fields.
163 lines
4.7 KiB
C++
163 lines
4.7 KiB
C++
/*
|
|
Copyright (C) 2010, 2014 Srivats P.
|
|
|
|
This file is part of "Ostinato"
|
|
|
|
This 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/>
|
|
*/
|
|
|
|
#ifndef _USER_SCRIPT_H
|
|
#define _USER_SCRIPT_H
|
|
|
|
#include "abstractprotocol.h"
|
|
#include "userscript.pb.h"
|
|
|
|
#include <QScriptEngine>
|
|
#include <QScriptValue>
|
|
|
|
class UserScriptProtocol;
|
|
|
|
class UserProtocol : public QObject
|
|
{
|
|
Q_OBJECT;
|
|
Q_ENUMS(ProtocolIdType);
|
|
Q_ENUMS(CksumType);
|
|
|
|
Q_PROPERTY(QString name READ name WRITE setName);
|
|
Q_PROPERTY(bool protocolFrameSizeVariable
|
|
READ isProtocolFrameSizeVariable
|
|
WRITE setProtocolFrameSizeVariable);
|
|
Q_PROPERTY(int protocolFrameVariableCount
|
|
READ protocolFrameVariableCount
|
|
WRITE setProtocolFrameVariableCount);
|
|
|
|
public:
|
|
enum ProtocolIdType
|
|
{
|
|
ProtocolIdLlc = AbstractProtocol::ProtocolIdLlc,
|
|
ProtocolIdEth = AbstractProtocol::ProtocolIdEth,
|
|
ProtocolIdIp = AbstractProtocol::ProtocolIdIp,
|
|
ProtocolIdTcpUdp = AbstractProtocol::ProtocolIdTcpUdp
|
|
};
|
|
|
|
enum CksumType
|
|
{
|
|
CksumIp = AbstractProtocol::CksumIp,
|
|
CksumIpPseudo = AbstractProtocol::CksumIpPseudo,
|
|
CksumTcpUdp = AbstractProtocol::CksumTcpUdp
|
|
};
|
|
|
|
enum CksumFlag
|
|
{
|
|
IncludeCksumField = AbstractProtocol::IncludeCksumField
|
|
};
|
|
Q_DECLARE_FLAGS(CksumFlags, CksumFlag);
|
|
Q_FLAG(CksumFlags);
|
|
|
|
UserProtocol(AbstractProtocol *parent);
|
|
|
|
public slots:
|
|
void reset();
|
|
|
|
QString name() const;
|
|
void setName(QString &name);
|
|
|
|
bool isProtocolFrameSizeVariable() const;
|
|
void setProtocolFrameSizeVariable(bool variable);
|
|
int protocolFrameVariableCount() const;
|
|
void setProtocolFrameVariableCount(int count);
|
|
|
|
quint32 payloadProtocolId(UserProtocol::ProtocolIdType type) const;
|
|
int protocolFrameOffset(int streamIndex = 0) const;
|
|
int protocolFramePayloadSize(int streamIndex = 0) const;
|
|
|
|
bool isProtocolFramePayloadValueVariable() const;
|
|
bool isProtocolFramePayloadSizeVariable() const;
|
|
int protocolFramePayloadVariableCount() const;
|
|
|
|
quint32 protocolFrameHeaderCksum(int streamIndex = 0,
|
|
AbstractProtocol::CksumType cksumType = AbstractProtocol::CksumIp) const;
|
|
quint32 protocolFramePayloadCksum(int streamIndex = 0,
|
|
AbstractProtocol::CksumType cksumType = AbstractProtocol::CksumIp) const;
|
|
|
|
private:
|
|
AbstractProtocol *parent_;
|
|
|
|
QString name_;
|
|
bool protocolFrameSizeVariable_;
|
|
int protocolFrameVariableCount_;
|
|
};
|
|
|
|
class UserScriptProtocol : public AbstractProtocol
|
|
{
|
|
public:
|
|
enum userScriptfield
|
|
{
|
|
// Frame Fields
|
|
userScript_program = 0,
|
|
|
|
userScript_fieldCount
|
|
};
|
|
|
|
UserScriptProtocol(StreamBase *stream, AbstractProtocol *parent = 0);
|
|
virtual ~UserScriptProtocol();
|
|
|
|
static AbstractProtocol* createInstance(StreamBase *stream,
|
|
AbstractProtocol *parent = 0);
|
|
virtual quint32 protocolNumber() const;
|
|
|
|
virtual void protoDataCopyInto(OstProto::Protocol &protocol) const;
|
|
virtual void protoDataCopyFrom(const OstProto::Protocol &protocol);
|
|
|
|
virtual quint32 protocolId(ProtocolIdType type) const;
|
|
|
|
virtual QString name() const;
|
|
virtual QString shortName() const;
|
|
|
|
virtual int fieldCount() const;
|
|
|
|
virtual QVariant fieldData(int index, FieldAttrib attrib,
|
|
int streamIndex = 0) const;
|
|
virtual bool setFieldData(int index, const QVariant &value,
|
|
FieldAttrib attrib = FieldValue);
|
|
|
|
virtual int protocolFrameSize(int streamIndex = 0) const;
|
|
|
|
virtual bool isProtocolFrameSizeVariable() const;
|
|
virtual int protocolFrameVariableCount() const;
|
|
|
|
virtual quint32 protocolFrameCksum(int streamIndex = 0,
|
|
CksumType cksumType = CksumIp, CksumFlags cksumFlags = 0) const;
|
|
|
|
void evaluateUserScript() const;
|
|
bool isScriptValid() const;
|
|
int userScriptErrorLineNumber() const;
|
|
QString userScriptErrorText() const;
|
|
|
|
private:
|
|
int userScriptLineCount() const;
|
|
|
|
OstProto::UserScript data;
|
|
|
|
mutable QScriptEngine engine_;
|
|
mutable UserProtocol userProtocol_;
|
|
mutable QScriptValue userProtocolScriptValue_;
|
|
|
|
mutable bool isScriptValid_;
|
|
mutable int errorLineNumber_;
|
|
mutable QString errorText_;
|
|
};
|
|
|
|
#endif
|