summaryrefslogtreecommitdiff
path: root/libcult/examples/os/net/ipv4/datagram/server.cxx
blob: e788056412a155374595a2aa64c4907168b897af (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
// file      : examples/os/net/ipv4/datagram/server.cxx
// author    : Boris Kolpackov <boris@kolpackov.net>
// copyright : Copyright (c) 2005-2010 Boris Kolpackov
// license   : GNU GPL v2 + exceptions; see accompanying LICENSE file

#include <cult/types.hxx>
#include <cult/containers/vector.hxx>

#include <cult/os/net/ipv4/address.hxx>
#include <cult/os/net/ipv4/datagram-socket.hxx>

#include <iostream>
#include <cstring> // memcmp

#include "protocol.hxx"

using std::cerr;
using std::endl;

using namespace Cult;
using namespace OS::Net::IPv4;

typedef Containers::Vector<Boolean> StatusList;

Int
main ()
{
  try
  {
    Address addr (INADDR_ANY, 10000);

    DatagramSocket socket (addr);

    Message expected_msg;
    expected_msg.sn = 0;

    for (UnsignedShort i = 0; i < payload_size; ++i)
    {
      expected_msg.payload[i] = i;
    }

    StatusList received (message_count, 0);
    StatusList damaged (message_count, 0);
    StatusList duplicate (message_count, 0);

    Message msg;

    while (true)
    {
      socket.recv (&msg, sizeof (msg));

      if (received[msg.sn])
      {
        duplicate[msg.sn] = true;
      }
      else
      {
        received[msg.sn] = true;

        if (std::memcmp (expected_msg.payload, msg.payload, payload_size) != 0)
        {
          damaged[msg.sn] = true;
        }
      }

      if (msg.sn + 1 == message_count) break;
    }

    UnsignedLong lost_count (0), damaged_count (0), duplicate_count (0);

    for (StatusList::Iterator i (received.begin ()), end (received.end ());
         i != end;
         ++i)
      if (!*i) ++lost_count;

    for (StatusList::Iterator i (damaged.begin ()), end (damaged.end ());
         i != end;
         ++i)
      if (*i) ++damaged_count;

    for (StatusList::Iterator i (duplicate.begin ()), end (duplicate.end ());
         i != end;
         ++i)
      if (*i) ++duplicate_count;

    cerr << "lost      : " << lost_count << endl
         << "damaged   : " << damaged_count << endl
         << "duplicate : " << duplicate_count << endl << endl;

    if (lost_count != 0)
    {
      cerr << "lost message dump:" << endl;

      UnsignedLong total = 0;

      for (StatusList::Iterator
             begin (received.begin ()), i (begin), end (received.end ());
           i != end;)
      {
        if (!*i)
        {
          UnsignedLong count = 1;

          for (StatusList::Iterator j = i + 1; j < end && !*j; j++, count++) ;

          cerr << '\t' << i - begin << " : " << count << endl;

          i += count;
          total += count;
        }
        else
          ++i;
      }

      if (total != lost_count)
        cerr << "trouble" << endl;
    }
  }
  catch (OS::Exception const& e)
  {
    cerr << "errno: " << e.code () << endl;
  }
}