blob: 626d629829002ee6e7e22b7a9b23cf487ef81f72 (
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
|
// $Id$
#include "Recv.h"
#include "ace/SOCK_Stream.h"
/* Construct the object with the peer reference and other appropriate
initializations.
*/
Recv::Recv( ACE_SOCK_Stream & _peer )
: Protocol_Task(0), peer_(_peer), error_(0)
{
// Create the tickler that get() will use to trigger recv()
// through the baseclass. Since we're single-threaded this is
// probably overkill but it makes multi-threading easier if we
// choose to do that.
tickler_ = new ACE_Message_Block(1);
}
/* Be sure we manage the lifetime of the tickler to prevent a memory
leak.
*/
Recv::~Recv(void)
{
tickler_->release();
}
/* By putting the tickler to ourselves we cause things to happen in
the baseclass that will invoke recv(). If we know we're single
threaded we could directly call recv() and be done with it but then
we'd have to do something else if we're multi-threaded. Just let
the baseclass worry about those things!
*/
int Recv::get(void)
{
return this->put( tickler_, 0 );
}
int Recv::recv(ACE_Message_Block * message, ACE_Time_Value *timeout)
{
int rval;
/* Xmit will send us the message length in clear-text. I
assume that will be less than 32-bytes!
*/
char msize[32];
int b = 0;
/* Read from the socket one byte at a time until we see then
end-of-string NULL character. Since the OS layers (at leas
in Unix) will provide some buffering this isn't as bad as
it may seem at first.
*/
do
{
rval = this->peer().recv( &msize[b], 1, timeout );
if( rval == -1 )
{
error_ = 1;
ACE_ERROR_RETURN ((LM_ERROR, "%p\n", "Recv::recv() Failed to get message size."), -1);
}
}
while( msize[b++] != 0 );
int size = ACE_OS::atoi(msize);
// Make a block big enough to contain the data we'll read
message = new ACE_Message_Block( size );
// Read the actual message data into our new message block
rval = this->peer().recv_n( message->wr_ptr(), size, 0, timeout );
// If we got the data correctly then send it on upstream.
if( rval > 0 )
{
message->wr_ptr( rval );
return( this->put_next( message ) );
}
// Something bad happend on the recv_n(). Set an error flag
// and return error.
error_ = 1;
return( -1 );
}
|