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
|
// SOCK_IO.cpp
// $Id$
#define ACE_BUILD_DLL
#include "ace/SOCK_IO.h"
ACE_ALLOC_HOOK_DEFINE(ACE_SOCK_IO)
void
ACE_SOCK_IO::dump (void) const
{
ACE_TRACE ("ACE_SOCK_IO::dump");
}
// Allows a client to read from a socket without having to provide
// a buffer to read. This method determines how much data is in the
// socket, allocates a buffer of this size, reads in the data, and
// returns the number of bytes read.
ssize_t
ACE_SOCK_IO::recv (iovec *io_vec)
{
#if defined (FIONREAD)
u_long inlen;
if (ACE_OS::ioctl (this->get_handle (), FIONREAD,
(u_long *) &inlen) == -1)
return -1;
else if (inlen > 0)
{
ACE_NEW_RETURN (io_vec->iov_base, char[inlen], -1);
io_vec->iov_len = this->recv (io_vec->iov_base, inlen);
return io_vec->iov_len;
}
else
return 0;
#else
io_vec = io_vec;
ACE_NOTSUP_RETURN (-1);
#endif /* FIONREAD */
}
// Send N char *ptrs and int lengths. Note that the char *'s precede
// the ints (basically, an varargs version of writev). The count N is
// the *total* number of trailing arguments, *not* a couple of the
// number of tuple pairs!
ssize_t
ACE_SOCK_IO::send (size_t n, ...) const
{
ACE_TRACE ("ACE_SOCK_IO::send");
va_list argp;
size_t total_tuples = n / 2;
ssize_t result;
#if defined (ACE_HAS_ALLOCA)
iovec *iovp = (iovec *) alloca (total_tuples * sizeof (iovec));
#else
iovec *iovp;
ACE_NEW_RETURN (iovp, iovec[total_tuples], -1);
#endif /* !defined (ACE_HAS_ALLOCA) */
va_start (argp, n);
for (size_t i = 0; i < total_tuples; i++)
{
iovp[i].iov_base = va_arg (argp, char *);
iovp[i].iov_len = va_arg (argp, int);
}
result = ACE_OS::writev (this->get_handle (), iovp, total_tuples);
#if !defined (ACE_HAS_ALLOCA)
delete [] iovp;
#endif /* !defined (ACE_HAS_ALLOCA) */
va_end (argp);
return result;
}
// This is basically an interface to ACE_OS::readv, that doesn't use
// the struct iovec explicitly. The ... can be passed as an arbitrary
// number of (char *ptr, int len) tuples. However, the count N is the
// *total* number of trailing arguments, *not* a couple of the number
// of tuple pairs!
ssize_t
ACE_SOCK_IO::recv (size_t n, ...) const
{
ACE_TRACE ("ACE_SOCK_IO::recv");
va_list argp;
size_t total_tuples = n / 2;
ssize_t result;
#if defined (ACE_HAS_ALLOCA)
iovec *iovp = (iovec *) alloca (total_tuples * sizeof (iovec));
#else
iovec *iovp;
ACE_NEW_RETURN (iovp, iovec[total_tuples], -1);
#endif /* !defined (ACE_HAS_ALLOCA) */
va_start (argp, n);
for (size_t i = 0; i < total_tuples; i++)
{
iovp[i].iov_base = va_arg (argp, char *);
iovp[i].iov_len = va_arg (argp, int);
}
result = ACE_OS::readv (this->get_handle (), iovp, total_tuples);
#if !defined (ACE_HAS_ALLOCA)
delete [] iovp;
#endif /* !defined (ACE_HAS_ALLOCA) */
va_end (argp);
return result;
}
|
