72e05696beb14cb70471e967d41ecdf2ccce921b
[senf.git] / senf / 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 unit tests */
25
26 //#include "Packet.test.hh"
27 //#include "Packet.test.ih"
28
29 // Custom includes
30 #include <sstream>
31 #include <boost/static_assert.hpp>
32 #include <boost/cstdint.hpp>
33 #include "Packets.hh"
34
35 #include <senf/Utils/auto_unit_test.hh>
36 #include <boost/test/test_tools.hpp>
37
38 #define prefix_
39 ///////////////////////////////cc.p////////////////////////////////////////
40
41 namespace {
42
43     struct RegTag {
44         typedef unsigned key_t;
45     };
46
47     struct FooPacketType
48         : public senf::PacketTypeBase,
49           public senf::PacketTypeMixin<FooPacketType>
50     {
51         using senf::PacketTypeMixin<FooPacketType>::nextPacketRange;
52         using senf::PacketTypeMixin<FooPacketType>::initSize;
53         using senf::PacketTypeMixin<FooPacketType>::init;
54         static size_type initSize()
55             { return 4u; }
56
57         // We need to implement initHeadSize() to force the mixin to switch into 'fixed-size'
58         // mode. Otherwise, mixin::nextPacketRange() would query the parser for it's size to find
59         // the header size. Since the parser is VoidPacketParser, the header size would therefore be
60         // 0
61         static size_type initHeadSize()
62             { return initSize(); }
63     };
64     typedef senf::ConcretePacket<FooPacketType> FooPacket;
65
66     struct BarPacketParser : public senf::PacketParserBase
67     {
68 #       include SENF_FIXED_PARSER()
69
70         SENF_PARSER_FIELD( type,     senf::UInt16Parser );
71         SENF_PARSER_FIELD( length,   senf::Int32Parser  );
72         SENF_PARSER_FIELD( reserved, senf::UInt16Parser );
73
74         SENF_PARSER_INIT() {
75             reserved() << 0xA0A0u;
76         }
77
78         SENF_PARSER_FINALIZE(BarPacketParser);
79     };
80
81     struct BarPacketType
82         : public senf::PacketTypeBase,
83           public senf::PacketTypeMixin<BarPacketType,RegTag>
84     {
85         typedef senf::PacketTypeMixin<BarPacketType,RegTag> mixin;
86         typedef senf::ConcretePacket<BarPacketType> packet;
87         typedef BarPacketParser parser;
88         using mixin::nextPacketRange;
89         using mixin::nextPacketType;
90         using mixin::initSize;
91         using mixin::init;
92         static void dump(packet p, std::ostream & os) {
93             os << "BarPacket:\n"
94                << senf::fieldName("type") << p->type() << "\n"
95                << senf::fieldName("length") << p->length() << "\n";
96         }
97         static void finalize(packet p) {
98             if (p.next(senf::nothrow))
99                 p->type() = senf::PacketRegistry<RegTag>::key(p.next());
100             else
101                 p->type() = -1;
102         }
103         static key_t nextPacketKey(packet p) {
104             return p->type();
105         }
106     };
107     typedef BarPacketType::packet BarPacket;
108
109     namespace reg {
110         senf::PacketRegistry<RegTag>::ScopedRegistrationProxy<FooPacket> registerFoo(1u);
111         senf::PacketRegistry<RegTag>::ScopedRegistrationProxy<BarPacket> registerBar(2u);
112     }
113
114     struct IntAnnotation {
115         boost::uint32_t value;
116     };
117
118     std::ostream & operator<<(std::ostream & os, IntAnnotation const & v)
119     { os << v.value; return os; }
120
121     struct LargeAnnotation {
122         char value[32];
123     };
124
125     std::ostream & operator<<(std::ostream & os, LargeAnnotation const & v)
126     { os << v.value; return os; }
127
128     struct ComplexAnnotation : senf::ComplexAnnotation
129     {
130         ComplexAnnotation() : s("empty"), i(-1) {}
131         std::string s;
132         boost::int32_t i;
133         // padding so the size does not depend on the platform ...
134         struct _ {std::string s;boost::int32_t i;};
135         char __ [32-sizeof(_)];
136     };
137
138     std::ostream & operator<<(std::ostream & os, ComplexAnnotation const & v)
139     { os << "('" << v.s << "' " << v.i << ')'; return os; }
140
141     struct ComplexEmptyAnnotation : senf::ComplexAnnotation
142     {};
143
144     std::ostream & operator<<(std::ostream & os, ComplexEmptyAnnotation const & v)
145     { os << "(empty)"; return os; }
146
147     struct InvalidAnnotation
148     {
149         std::string value;
150     };
151
152     std::ostream & operator<<(std::ostream & os, InvalidAnnotation const & v)
153     { os << v.value; return os; }
154
155 }
156
157 SENF_AUTO_UNIT_TEST(packet)
158 {
159     BOOST_CHECK(! senf::Packet().is<BarPacket>() );
160     senf::Packet packet (FooPacket::create());
161     BarPacket::createAfter(packet);
162
163     BOOST_REQUIRE( packet );
164     BOOST_CHECK( packet.next() );
165     BOOST_CHECK( ! packet.next().next(senf::nothrow) );
166     BOOST_CHECK( ! packet.next().next(senf::nothrow).is<BarPacket>() );
167     BOOST_CHECK( ! packet.prev(senf::nothrow) );
168     BOOST_CHECK( packet.next().prev() == packet );
169     SENF_CHECK_NOT_EQUAL( packet.next(), packet );
170     BOOST_CHECK_EQUAL( std::distance(packet.data().begin(), packet.next().data().begin()), 4 );
171     BOOST_CHECK_EQUAL( std::distance(packet.data().begin(), packet.data().end()), 12 );
172     BOOST_CHECK_EQUAL( std::distance(packet.next().data().begin(), packet.next().data().end()), 8 );
173     BOOST_CHECK( packet.data().end() == packet.next().data().end() );
174     BOOST_CHECK_EQUAL( packet.size(), 12u );
175     BOOST_CHECK_EQUAL( packet.next().size(), 8u );
176     BOOST_CHECK( packet.is<FooPacket>() );
177     BOOST_CHECK( packet.next().is<BarPacket>() );
178     BOOST_CHECK( packet.first() == packet );
179     BOOST_CHECK( packet.last() == packet.next() );
180
181     senf::Packet p2 (packet.next());
182     BOOST_CHECK( p2 );
183     packet.parseNextAs<FooPacket>();
184     BOOST_CHECK_EQUAL( packet.size(), 12u );
185     BOOST_CHECK_EQUAL( packet.next().size(), 8u );
186     BOOST_CHECK( packet.next().is<FooPacket>() );
187     BOOST_CHECK( ! p2 );
188     BOOST_CHECK( packet.next().as<FooPacket>() );
189
190     p2 = packet.next().clone();
191     BOOST_REQUIRE( p2 );
192     packet.next().append( p2 );
193     BOOST_REQUIRE( packet.next().next() );
194     BOOST_CHECK( packet.next().next().next() );
195     BOOST_CHECK( packet.next().next().next().is<senf::DataPacket>() );
196     BOOST_CHECK_EQUAL( packet.size(), 16u );
197
198     // This calls and checks typeId()
199     BOOST_CHECK_EQUAL( senf::PacketRegistry<RegTag>::key(packet), 1u );
200     packet.next().parseNextAs( senf::PacketRegistry<RegTag>::lookup(2u).factory() );
201     BOOST_CHECK( packet.next().next().is<BarPacket>() );
202
203     std::stringstream s;
204     packet.dump(s);
205     BOOST_CHECK_EQUAL( s.str(),
206                        "Annotations:\n"
207                        "  (anonymous namespace)::IntAnnotation : 0\n"
208                        "BarPacket:\n"
209                        "  type                    : 0\n"
210                        "  length                  : 0\n" );
211
212     packet.finalizeAll();
213     BOOST_CHECK_EQUAL( packet.last().as<BarPacket>()->type(),
214                        BarPacket::Parser::type_t::value_type(-1) );
215     packet.last().append(FooPacket::create());
216     packet.finalizeThis();
217     packet.finalizeTo<BarPacket>();
218     packet.finalizeTo(packet.find<BarPacket>());
219     packet.finalizeAll();
220     BOOST_CHECK_EQUAL( packet.find<BarPacket>()->type(), 1u );
221
222     BOOST_CHECK( packet.factory() == FooPacket::factory() );
223
224     senf::PacketData::byte data[] = { 0x00, 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
225                                       0x81, 0x82, 0x83 };
226
227     BarPacket::createAfter(packet, data);
228     BOOST_REQUIRE( packet.next() );
229     BOOST_REQUIRE( packet.next().is<BarPacket>() );
230     BOOST_CHECK( packet.last().is<FooPacket>() );
231     BOOST_CHECK_EQUAL( packet.last().rfind<BarPacket>()->type(), 1u );
232     BOOST_CHECK_EQUAL( packet.next().size(), 11u );
233     BOOST_REQUIRE( packet.next().next() );
234     BOOST_CHECK( packet.next().next().is<FooPacket>() );
235     BOOST_CHECK( ! packet.next().next().next(senf::nothrow) );
236     BOOST_CHECK_EQUAL( packet.next().next().data()[0], 0x81u );
237
238     BOOST_CHECK( packet.first().find<FooPacket>() == packet );
239     BOOST_CHECK( packet.last().rfind<BarPacket>() == packet.last().prev() );
240     BOOST_CHECK( packet.find<FooPacket>() == packet );
241     BOOST_CHECK( packet.last().rfind<FooPacket>() == packet.last() );
242     BOOST_CHECK( packet.next<BarPacket>() == packet.next() );
243     BOOST_CHECK( packet.last().prev().prev<FooPacket>() == packet );
244
245     senf::DataPacket::createAfter(packet);
246     BOOST_CHECK_THROW( packet.next().next().next().parseNextAs<BarPacket>(),
247             senf::InvalidPacketChainException );
248
249     SENF_CHECK_NO_THROW( BarPacket::create(senf::noinit).dump(s));
250 }
251
252 SENF_AUTO_UNIT_TEST(concretePacket)
253 {
254     senf::PacketData::byte data[] = { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06 };
255
256     BOOST_CHECK_EQUAL( FooPacket::create().size(), 4u );
257     BOOST_CHECK_EQUAL( FooPacket::create(senf::noinit).size(), 0u );
258     BOOST_CHECK_THROW( FooPacket::create(2u), senf::TruncatedPacketException );
259     // No 'u' suffix here to check, that the disable_if works ...
260     BOOST_CHECK_EQUAL( FooPacket::create(10).size(), 10u );
261     BOOST_CHECK_EQUAL( FooPacket::create(2u,senf::noinit).size(), 2u );
262     BOOST_CHECK_EQUAL( FooPacket::create(data).size(), 6u );
263
264     senf::Packet packet (FooPacket::create());
265
266     BOOST_CHECK_EQUAL( FooPacket::createAfter(packet).size(), 4u );
267     BOOST_CHECK_EQUAL( packet.size(), 8u );
268
269     BOOST_CHECK_EQUAL( FooPacket::createAfter(packet,senf::noinit).size(), 0u );
270     BOOST_CHECK_EQUAL( packet.size(), 4u );
271
272     BOOST_CHECK_THROW( FooPacket::createAfter(packet,2u), senf::TruncatedPacketException );
273     // No 'u' suffix here to check, that the disable_if works ...
274     BOOST_CHECK_EQUAL( FooPacket::createAfter(packet,10).size(), 10u );
275     BOOST_CHECK_EQUAL( packet.size(), 14u );
276
277     BOOST_CHECK_EQUAL( FooPacket::createAfter(packet,2u,senf::noinit).size(), 2u );
278     BOOST_CHECK_EQUAL( packet.size(), 6u );
279
280     BOOST_CHECK_EQUAL( FooPacket::createAfter(packet,data).size(), 6u );
281     BOOST_CHECK_EQUAL( packet.size(), 10u );
282
283     BOOST_CHECK_EQUAL( FooPacket::createBefore(packet).size(), 14u );
284     BOOST_CHECK_EQUAL( packet.size(), 10u );
285
286     BOOST_CHECK_EQUAL( FooPacket::createBefore(packet,senf::noinit).size(), 10u );
287     BOOST_CHECK_EQUAL( packet.size(), 10u );
288
289     BOOST_CHECK_EQUAL( FooPacket::createInsertBefore(packet).size(), 14u );
290     BOOST_CHECK_EQUAL( packet.size(), 10u );
291     BOOST_REQUIRE( packet.prev() );
292     BOOST_CHECK_EQUAL( packet.prev().size(), 14u );
293     BOOST_REQUIRE( packet.prev().prev() );
294     BOOST_CHECK_EQUAL( packet.prev().prev().size(), 14u );
295
296     BOOST_CHECK_EQUAL( FooPacket::createInsertBefore(packet,senf::noinit).size(), 10u );
297     BOOST_CHECK_EQUAL( packet.size(), 10u );
298     BOOST_REQUIRE_NO_THROW( packet.prev().prev().prev() );
299     BOOST_CHECK_THROW( packet.prev().prev().prev().prev(), senf::InvalidPacketChainException );
300     BOOST_CHECK_EQUAL( packet.prev().size(), 10u );
301     BOOST_CHECK_EQUAL( packet.prev().prev().size(), 14u );
302     BOOST_CHECK_EQUAL( packet.prev().prev().prev().size(), 14u );
303
304     SENF_CHECK_NOT_EQUAL( packet.clone(), packet );
305     BOOST_CHECK_EQUAL( BarPacket::create()->reserved(), 0xA0A0u );
306 }
307
308 SENF_AUTO_UNIT_TEST(packetAssign)
309 {
310     BarPacket bar1 (BarPacket::create());
311     BarPacket bar2 (BarPacket::create());
312
313     bar2->type() << 0x2A2Bu;
314     bar1.parser() << bar2;
315
316     BOOST_CHECK_EQUAL( bar1->type(), 0x2A2Bu );
317 }
318
319 SENF_AUTO_UNIT_TEST(packetAnnotation)
320 {
321     typedef senf::detail::AnnotationRegistry Reg;
322
323     senf::Packet packet (FooPacket::create());
324     BarPacket::createAfter(packet);
325
326     ComplexAnnotation & ca (packet.annotation<ComplexAnnotation>());
327
328     BOOST_CHECK_EQUAL( ca.s, "empty" );
329     BOOST_CHECK_EQUAL( ca.i, -1 );
330
331     ca.s = "dead beef";
332     ca.i = 0x12345678;
333     SENF_CHECK_NO_THROW( packet.annotation<IntAnnotation>().value = 0xDEADBEEF );
334
335     senf::Packet p2 (packet.next());
336
337     BOOST_CHECK_EQUAL( p2.annotation<IntAnnotation>().value, 0xDEADBEEFu );
338     BOOST_CHECK_EQUAL( p2.annotation<ComplexAnnotation>().s, "dead beef" );
339     BOOST_CHECK_EQUAL( p2.annotation<ComplexAnnotation>().i, 0x12345678 );
340
341     BOOST_CHECK( Reg::lookup<IntAnnotation>() >= 0 );
342     BOOST_CHECK( Reg::lookup<LargeAnnotation>() < 0 );
343     BOOST_CHECK( Reg::lookup<ComplexAnnotation>() < 0 );
344     BOOST_CHECK( Reg::lookup<ComplexEmptyAnnotation>() < 0 );
345
346 #ifdef SENF_DEBUG
347     std::stringstream ss;
348
349     senf::dumpPacketAnnotationRegistry(ss);
350     BOOST_CHECK_EQUAL(
351         ss.str(),
352         "SENF_PACKET_ANNOTATION_SLOTS = 8\n"
353         "SENF_PACKET_ANNOTATION_SLOTSIZE = 16\n"
354         "TYPE                                                      FAST  COMPLEX   SIZE\n"
355         "(anonymous namespace)::ComplexAnnotation                  no    yes         32\n"
356         "(anonymous namespace)::ComplexEmptyAnnotation             no    yes          1\n"
357         "(anonymous namespace)::IntAnnotation                      yes   no           4\n"
358         "(anonymous namespace)::LargeAnnotation                    no    no          32\n" );
359 #endif
360 }
361
362 #ifdef COMPILE_CHECK
363
364 COMPILE_FAIL(invalidAnnotation)
365 {
366 #if 0 // The traits check fails for user defined but trivial constructors so ...
367 #   ifdef BOOST_HAS_TYPE_TRAITS_INTRINSICS
368
369     senf::Packet packet (FooPacket::create());
370     senf::IGNORE( packet.annotation<InvalidAnnotation>() );
371
372 #   else
373 #   endif
374 #endif
375
376     invalid_annotation_check_disabled();
377
378 }
379
380 #endif
381
382 ///////////////////////////////cc.e////////////////////////////////////////
383 #undef prefix_
384
385 \f
386 // Local Variables:
387 // mode: c++
388 // fill-column: 100
389 // c-file-style: "senf"
390 // indent-tabs-mode: nil
391 // ispell-local-dictionary: "american"
392 // compile-command: "scons -u test"
393 // comment-column: 40
394 // End: