Packets/Packet.test.cc

   1 // $Id$
   2 //
   3 // Copyright (C) 2007
   4 // Fraunhofer Institute for Open Communication Systems (FOKUS)
   5 // Competence Center NETwork research (NET), St. Augustin, GERMANY
   6 //     Stefan Bund <g0dil@berlios.de>
   7 //
   8 // This program is free software; you can redistribute it and/or modify
   9 // it under the terms of the GNU General Public License as published by
  10 // the Free Software Foundation; either version 2 of the License, or
  11 // (at your option) any later version.
  12 //
  13 // This program is distributed in the hope that it will be useful,
  14 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  15 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16 // GNU General Public License for more details.
  17 //
  18 // You should have received a copy of the GNU General Public License
  19 // along with this program; if not, write to the
  20 // Free Software Foundation, Inc.,
  21 // 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  22
  23 /** \file
  24     \brief Packet.test unit tests */
  25
  26 //#include "Packet.test.hh"
  27 //#include "Packet.test.ih"
  28
  29 // Custom includes
  30 #include <sstream>
  31 #include "Packets.hh"
  32
  33 #include "../Utils/auto_unit_test.hh"
  34 #include <boost/test/test_tools.hpp>
  35
  36 #define prefix_
  37 ///////////////////////////////cc.p////////////////////////////////////////
  38
  39 namespace {
  40
  41     struct RegTag {
  42         typedef unsigned key_t;
  43     };
  44
  45     struct FooPacketType
  46         : public senf::PacketTypeBase,
  47           public senf::PacketTypeMixin<FooPacketType>
  48     {
  49         using senf::PacketTypeMixin<FooPacketType>::nextPacketRange;
  50         using senf::PacketTypeMixin<FooPacketType>::initSize;
  51         using senf::PacketTypeMixin<FooPacketType>::init;
  52         static size_type initSize()
  53             { return 4u; }
  54
  55         // We need to implement initHeadSize() to force the mixin to switch into 'fixed-size'
  56         // mode. Otherwise, mixin::nextPacketRange() would query the parser for it's size to find
  57         // the header size. Since the parser is VoidPacketParser, the header size would therefore be
  58         // 0
  59         static size_type initHeadSize()
  60             { return initSize(); }
  61     };
  62     typedef senf::ConcretePacket<FooPacketType> FooPacket;
  63
  64     struct BarPacketParser : public senf::PacketParserBase
  65     {
  66 #       include SENF_FIXED_PARSER()
  67
  68         SENF_PARSER_FIELD( type,     senf::UInt16Parser );
  69         SENF_PARSER_FIELD( length,   senf::Int32Parser  );
  70         SENF_PARSER_FIELD( reserved, senf::UInt16Parser );
  71
  72         SENF_PARSER_INIT() {
  73             reserved() << 0xA0A0u;
  74         }
  75
  76         SENF_PARSER_FINALIZE(BarPacketParser);
  77     };
  78
  79     struct BarPacketType
  80         : public senf::PacketTypeBase,
  81           public senf::PacketTypeMixin<BarPacketType,RegTag>
  82     {
  83         typedef senf::PacketTypeMixin<BarPacketType,RegTag> mixin;
  84         typedef senf::ConcretePacket<BarPacketType> packet;
  85         typedef BarPacketParser parser;
  86         using mixin::nextPacketRange;
  87         using mixin::nextPacketType;
  88         using mixin::initSize;
  89         using mixin::init;
  90         static void dump(packet p, std::ostream & os) {
  91             os << "BarPacket:\n"
  92                << "type: " << p->type() << "\n"
  93                << "length: " << p->length() << "\n";
  94         }
  95         static void finalize(packet p) {
  96             if (p.next(senf::nothrow))
  97                 p->type() = senf::PacketRegistry<RegTag>::key(p.next());
  98             else
  99                 p->type() = -1;
 100         }
 101         static key_t nextPacketKey(packet p) {
 102             return p->type();
 103         }
 104     };
 105     typedef BarPacketType::packet BarPacket;
 106
 107     namespace reg {
 108         senf::PacketRegistry<RegTag>::RegistrationProxy<FooPacket> registerFoo(1u);
 109         senf::PacketRegistry<RegTag>::RegistrationProxy<BarPacket> registerBar(2u);
 110     }
 111
 112     struct IntAnnotation {
 113         int value;
 114     };
 115
 116     struct ComplexAnnotation {
 117         std::string s;
 118         int i;
 119     };
 120
 121 }
 122
 123 BOOST_AUTO_UNIT_TEST(packet)
 124 {
 125     senf::Packet packet (FooPacket::create());
 126     BarPacket::createAfter(packet);
 127
 128     SENF_CHECK_NO_THROW( packet.annotation<IntAnnotation>().value = 0xDEADBEEF );
 129     ComplexAnnotation & ca (packet.annotation<ComplexAnnotation>());
 130     ca.s = "dead beef";
 131     ca.i = 0x12345678;
 132     BOOST_REQUIRE( packet );
 133     BOOST_CHECK( packet.next() );
 134     BOOST_CHECK( ! packet.next().next(senf::nothrow) );
 135     BOOST_CHECK( ! packet.prev(senf::nothrow) );
 136     BOOST_CHECK( packet.next().prev() == packet );
 137     BOOST_CHECK( packet.next() != packet );
 138     BOOST_CHECK_EQUAL( std::distance(packet.data().begin(), packet.next().data().begin()), 4 );
 139     BOOST_CHECK_EQUAL( std::distance(packet.data().begin(), packet.data().end()), 12 );
 140     BOOST_CHECK_EQUAL( std::distance(packet.next().data().begin(), packet.next().data().end()), 8 );
 141     BOOST_CHECK( packet.data().end() == packet.next().data().end() );
 142     BOOST_CHECK_EQUAL( packet.size(), 12u );
 143     BOOST_CHECK_EQUAL( packet.next().size(), 8u );
 144     BOOST_CHECK( packet.is<FooPacket>() );
 145     BOOST_CHECK( packet.next().is<BarPacket>() );
 146     BOOST_CHECK( packet.first() == packet );
 147     BOOST_CHECK( packet.last() == packet.next() );
 148
 149     senf::Packet p2 (packet.next());
 150     BOOST_CHECK( p2 );
 151     packet.parseNextAs<FooPacket>();
 152     BOOST_CHECK_EQUAL( packet.size(), 12u );
 153     BOOST_CHECK_EQUAL( packet.next().size(), 8u );
 154     BOOST_CHECK( packet.next().is<FooPacket>() );
 155     BOOST_CHECK( ! p2 );
 156     BOOST_CHECK( packet.next().as<FooPacket>() );
 157
 158     p2 = packet.next().clone();
 159     BOOST_REQUIRE( p2 );
 160     packet.next().append( p2 );
 161     BOOST_REQUIRE( packet.next().next() );
 162     BOOST_CHECK( packet.next().next().next() );
 163     BOOST_CHECK( packet.next().next().next().is<senf::DataPacket>() );
 164     BOOST_CHECK_EQUAL( packet.size(), 16u );
 165
 166     // This calls and checks typeId()
 167     BOOST_CHECK_EQUAL( senf::PacketRegistry<RegTag>::key(packet), 1u );
 168     packet.next().parseNextAs( senf::PacketRegistry<RegTag>::lookup(2u).factory() );
 169     BOOST_CHECK( packet.next().next().is<BarPacket>() );
 170
 171     std::stringstream s;
 172     packet.dump(s);
 173     BOOST_CHECK_EQUAL( s.str(), "BarPacket:\ntype: 0\nlength: 0\n" );
 174
 175     packet.finalizeAll();
 176     BOOST_CHECK_EQUAL( packet.last().as<BarPacket>()->type(),
 177                        BarPacket::Parser::type_t::value_type(-1) );
 178     packet.last().append(FooPacket::create());
 179     packet.finalizeThis();
 180     packet.finalizeTo<BarPacket>();
 181     packet.finalizeTo(packet.find<BarPacket>());
 182     packet.finalizeAll();
 183     BOOST_CHECK_EQUAL( packet.find<BarPacket>()->type(), 1u );
 184
 185     BOOST_CHECK( packet.factory() == FooPacket::factory() );
 186
 187     senf::PacketData::byte data[] = { 0x00, 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
 188                                       0x81, 0x82, 0x83 };
 189
 190     BarPacket::createAfter(packet,data);
 191     BOOST_REQUIRE( packet.next() );
 192     BOOST_REQUIRE( packet.next().is<BarPacket>() );
 193     BOOST_CHECK( packet.last().is<FooPacket>() );
 194     BOOST_CHECK_EQUAL( packet.last().rfind<BarPacket>()->type(), 1u );
 195     BOOST_CHECK_EQUAL( packet.next().size(), 11u );
 196     BOOST_REQUIRE( packet.next().next() );
 197     BOOST_CHECK( packet.next().next().is<FooPacket>() );
 198     BOOST_CHECK( ! packet.next().next().next(senf::nothrow) );
 199     BOOST_CHECK_EQUAL( packet.next().next().data()[0], 0x81u );
 200
 201     BOOST_CHECK( packet.first().find<FooPacket>() == packet );
 202     BOOST_CHECK( packet.last().rfind<BarPacket>() == packet.last().prev() );
 203     BOOST_CHECK( packet.find<FooPacket>() == packet );
 204     BOOST_CHECK( packet.last().rfind<FooPacket>() == packet.last() );
 205     BOOST_CHECK( packet.next<BarPacket>() == packet.next() );
 206     BOOST_CHECK( packet.last().prev().prev<FooPacket>() == packet );
 207 }
 208
 209 BOOST_AUTO_UNIT_TEST(concretePacket)
 210 {
 211     senf::PacketData::byte data[] = { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06 };
 212
 213     BOOST_CHECK_EQUAL( FooPacket::create().size(), 4u );
 214     BOOST_CHECK_EQUAL( FooPacket::create(senf::noinit).size(), 0u );
 215     BOOST_CHECK_THROW( FooPacket::create(2u), senf::TruncatedPacketException );
 216     // No 'u' suffix here to check, that the disable_if works ...
 217     BOOST_CHECK_EQUAL( FooPacket::create(10).size(), 10u );
 218     BOOST_CHECK_EQUAL( FooPacket::create(2u,senf::noinit).size(), 2u );
 219     BOOST_CHECK_EQUAL( FooPacket::create(data).size(), 6u );
 220
 221     senf::Packet packet (FooPacket::create());
 222
 223     BOOST_CHECK_EQUAL( FooPacket::createAfter(packet).size(), 4u );
 224     BOOST_CHECK_EQUAL( packet.size(), 8u );
 225
 226     BOOST_CHECK_EQUAL( FooPacket::createAfter(packet,senf::noinit).size(), 0u );
 227     BOOST_CHECK_EQUAL( packet.size(), 4u );
 228
 229     BOOST_CHECK_THROW( FooPacket::createAfter(packet,2u), senf::TruncatedPacketException );
 230     // No 'u' suffix here to check, that the disable_if works ...
 231     BOOST_CHECK_EQUAL( FooPacket::createAfter(packet,10).size(), 10u );
 232     BOOST_CHECK_EQUAL( packet.size(), 14u );
 233
 234     BOOST_CHECK_EQUAL( FooPacket::createAfter(packet,2u,senf::noinit).size(), 2u );
 235     BOOST_CHECK_EQUAL( packet.size(), 6u );
 236
 237     BOOST_CHECK_EQUAL( FooPacket::createAfter(packet,data).size(), 6u );
 238     BOOST_CHECK_EQUAL( packet.size(), 10u );
 239
 240     BOOST_CHECK_EQUAL( FooPacket::createBefore(packet).size(), 14u );
 241     BOOST_CHECK_EQUAL( packet.size(), 10u );
 242
 243     BOOST_CHECK_EQUAL( FooPacket::createBefore(packet,senf::noinit).size(), 10u );
 244     BOOST_CHECK_EQUAL( packet.size(), 10u );
 245
 246     BOOST_CHECK( packet.clone() != packet );
 247     BOOST_CHECK_EQUAL( BarPacket::create()->reserved(), 0xA0A0u );
 248 }
 249
 250 ///////////////////////////////cc.e////////////////////////////////////////
 251 #undef prefix_
 252
 253 \f
 254 // Local Variables:
 255 // mode: c++
 256 // fill-column: 100
 257 // c-file-style: "senf"
 258 // indent-tabs-mode: nil
 259 // ispell-local-dictionary: "american"
 260 // compile-command: "scons -u test"
 261 // comment-column: 40
 262 // End: