ostinato/common/ip6over6.h

92 lines
3.0 KiB
C++

/*
Copyright (C) 2010 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 _IP_6_OVER_6_H
#define _IP_6_OVER_6_H
#include "ip6over6.pb.h"
#include "comboprotocol.h"
#include "ip6.h"
typedef ComboProtocol<OstProto::Protocol::kIp6over6FieldNumber,
Ip6Protocol, Ip6Protocol> Ip6over6Combo;
class Ip6over6Protocol : public Ip6over6Combo
{
public:
Ip6over6Protocol(StreamBase *stream, AbstractProtocol *parent = 0)
: Ip6over6Combo(stream, parent)
{
}
static Ip6over6Protocol* createInstance(StreamBase *stream,
AbstractProtocol *parent)
{
return new Ip6over6Protocol(stream, parent);
}
virtual void protoDataCopyInto(OstProto::Protocol &protocol) const
{
OstProto::Protocol tempProto;
protoA->protoDataCopyInto(tempProto);
protocol.MutableExtension(OstProto::ip6over6)
->MutableExtension(OstProto::ip6_outer)
->CopyFrom(tempProto.GetExtension(OstProto::ip6));
tempProto.Clear();
protoB->protoDataCopyInto(tempProto);
protocol.MutableExtension(OstProto::ip6over6)
->MutableExtension(OstProto::ip6_inner)
->CopyFrom(tempProto.GetExtension(OstProto::ip6));
protocol.mutable_protocol_id()->set_id(protocolNumber());
}
virtual void protoDataCopyFrom(const OstProto::Protocol &protocol)
{
if (protocol.protocol_id().id() == protocolNumber()
&& protocol.HasExtension(OstProto::ip6over6))
{
OstProto::Protocol tempProto;
// NOTE: To use protoX->protoDataCopyFrom() we need to arrange
// so that it sees its own protocolNumber() and its own extension
// in 'protocol'
tempProto.mutable_protocol_id()->set_id(protoA->protocolNumber());
tempProto.MutableExtension(OstProto::ip6)->CopyFrom(
protocol.GetExtension(OstProto::ip6over6).GetExtension(
OstProto::ip6_outer));
protoA->protoDataCopyFrom(tempProto);
tempProto.Clear();
tempProto.mutable_protocol_id()->set_id(protoB->protocolNumber());
tempProto.MutableExtension(OstProto::ip6)->CopyFrom(
protocol.GetExtension(OstProto::ip6over6).GetExtension(
OstProto::ip6_inner));
protoB->protoDataCopyFrom(tempProto);
}
}
};
#endif