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
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
|
// $Id$
#include "RMCast_Reassembly.h"
#include "RMCast_Partial_Message.h"
#include "ace/Message_Block.h"
#if !defined (ACE_LACKS_PRAGMA_ONCE)
# pragma once
#endif /* ACE_LACKS_PRAGMA_ONCE */
#if !defined (__ACE_INLINE__)
#include "RMCast_Reassembly.i"
#endif /* __ACE_INLINE__ */
ACE_RCSID(ace, RMCast_Reassembly, "$Id$")
ACE_RMCast_Reassembly::
ACE_RMCast_Reassembly (void)
: ACE_RMCast_Module ()
{
}
ACE_RMCast_Reassembly::~ACE_RMCast_Reassembly (void)
{
(void) this->close_i ();
}
/**
*
* We cleanup the resources in the destructor
*/
int
ACE_RMCast_Reassembly::close ()
{
this->close_i ();
return this->ACE_RMCast_Module::close ();
}
int
ACE_RMCast_Reassembly::data (ACE_RMCast::Data &data)
{
if (this->next () == 0)
return 0;
// ACE_DEBUG ((LM_DEBUG,
// "Reassembly::data - %d,%d,%d\n",
// data.sequence_number,
// data.total_size,
// data.fragment_offset));
if (data.payload->length () + data.fragment_offset > data.total_size)
{
ACE_ERROR ((LM_ERROR,
"RMCast_Reassembly::data - invalid size\n"));
return -1; // Corrupt message?
}
ACE_RMCast_Partial_Message *message;
{
ACE_GUARD_RETURN (ACE_SYNCH_MUTEX, ace_mon, this->mutex_, -1);
if (this->messages_.find (data.sequence_number, message) == -1)
{
ACE_NEW_RETURN (message,
ACE_RMCast_Partial_Message (data.total_size),
-1);
// ACE_DEBUG ((LM_DEBUG,
// "Reassembly::data - new message\n"));
if (this->messages_.bind (data.sequence_number,
message) == -1)
return -1; // Internal error?
}
if (message->fragment_received (data.total_size,
data.fragment_offset,
data.payload) == -1)
{
// ACE_DEBUG ((LM_DEBUG,
// "Error in fragment_received\n"));
return -1;
}
if (!message->is_complete ())
{
// ACE_DEBUG ((LM_DEBUG,
// "Reassembly::data - message still incomplete\n"));
return 0;
}
if (this->messages_.unbind (data.sequence_number) == -1)
{
// ACE_DEBUG ((LM_DEBUG,
// "Reassembly::data - message now complete\n"));
return -1;
}
}
// Push the message...
ACE_RMCast::Data downstream_data;
downstream_data.source = data.source;
downstream_data.sequence_number = data.sequence_number;
downstream_data.total_size = message->message_body ()->length ();
downstream_data.fragment_offset = 0;
downstream_data.payload = message->message_body ();
int r = this->next ()->data (downstream_data);
delete message;
return r;
}
void
ACE_RMCast_Reassembly::close_i (void)
{
for (Message_Map_Iterator i = this->messages_.begin ();
i != this->messages_.end ();
++i)
{
ACE_RMCast_Partial_Message *message = (*i).int_id_;
if (message != 0)
delete message;
}
this->messages_.unbind_all ();
}
#if defined (ACE_HAS_EXPLICIT_TEMPLATE_INSTANTIATION)
template class ACE_Hash_Map_Manager<ACE_UINT32,ACE_RMCast_Partial_Message*,ACE_Null_Mutex>;
template class ACE_Hash_Map_Manager_Ex<ACE_UINT32,ACE_RMCast_Partial_Message*,ACE_Hash<ACE_UINT32>,ACE_Equal_To<ACE_UINT32>,ACE_Null_Mutex>;
template class ACE_Hash_Map_Iterator<ACE_UINT32,ACE_RMCast_Partial_Message*,ACE_Null_Mutex>;
template class ACE_Hash_Map_Iterator_Ex<ACE_UINT32,ACE_RMCast_Partial_Message*,ACE_Hash<ACE_UINT32>,ACE_Equal_To<ACE_UINT32>,ACE_Null_Mutex >;
template class ACE_Hash_Map_Reverse_Iterator_Ex<ACE_UINT32,ACE_RMCast_Partial_Message*,ACE_Hash<ACE_UINT32>,ACE_Equal_To<ACE_UINT32>,ACE_Null_Mutex >;
template class ACE_Hash_Map_Iterator_Base_Ex<ACE_UINT32,ACE_RMCast_Partial_Message*,ACE_Hash<ACE_UINT32>,ACE_Equal_To<ACE_UINT32>,ACE_Null_Mutex >;
template class ACE_Hash_Map_Entry<ACE_UINT32,ACE_RMCast_Partial_Message*>;
template class ACE_Less_Than<ACE_UINT32>;
#elif defined (ACE_HAS_TEMPLATE_INSTANTIATION_PRAGMA)
#pragma instantiate ACE_Hash_Map_Manager<ACE_UINT32,ACE_RMCast_Partial_Message*,ACE_Null_Mutex>
#pragma instantiate ACE_Hash_Map_Manager_Ex<ACE_UINT32,ACE_RMCast_Partial_Message*,ACE_Hash<ACE_UINT32>,ACE_Equal_To<ACE_UINT32>,ACE_Null_Mutex>
#pragma instantiate ACE_Hash_Map_Iterator<ACE_UINT32,ACE_RMCast_Partial_Message*,ACE_Null_Mutex>
#pragma instantiate ACE_Hash_Map_Iterator_Ex<ACE_UINT32,ACE_RMCast_Partial_Message*,ACE_Hash<ACE_UINT32>,ACE_Equal_To<ACE_UINT32>,ACE_Null_Mutex >
#pragma instantiate ACE_Hash_Map_Reverse_Iterator_Ex<ACE_UINT32,ACE_RMCast_Partial_Message*,ACE_Hash<ACE_UINT32>,ACE_Equal_To<ACE_UINT32>,ACE_Null_Mutex >
#pragma instantiate ACE_Hash_Map_Iterator_Base_Ex<ACE_UINT32,ACE_RMCast_Partial_Message*,ACE_Hash<ACE_UINT32>,ACE_Equal_To<ACE_UINT32>,ACE_Null_Mutex >
#pragma instantiate ACE_Hash_Map_Entry<ACE_UINT32,ACE_RMCast_Partial_Message*>
#pragma instantiate ACE_Less_Than<ACE_UINT32>
#endif /* ACE_HAS_EXPLICIT_TEMPLATE_INSTANTIATION */
|
