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
|
/*
* Copyright (c) 2017-2018 Cisco and/or its affiliates.
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at:
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifndef __sock_test_h__
#define __sock_test_h__
#include <netdb.h>
#include <errno.h>
#include <stdlib.h>
#include <string.h>
#include <vcl/vcl_test.h>
#define SOCK_TEST_AF_UNIX_FILENAME "/tmp/ldp_server_af_unix_socket"
#define SOCK_TEST_MIXED_EPOLL_DATA "Hello, world! (over an AF_UNIX socket)"
#define SOCK_TEST_AF_UNIX_ACCEPT_DATA 0xaf0000af
#define SOCK_TEST_AF_UNIX_FD_MASK 0x00af0000
#define SOCK_TEST_BANNER_STRING \
"============================================\n"
static inline int
sock_test_read (int fd, uint8_t *buf, uint32_t nbytes,
vcl_test_stats_t *stats)
{
int rx_bytes, errno_val;
do
{
if (stats)
stats->rx_xacts++;
rx_bytes = read (fd, buf, nbytes);
if (stats)
{
if ((rx_bytes == 0) ||
((rx_bytes < 0) && ((errno == EAGAIN) || (errno == EWOULDBLOCK))))
stats->rx_eagain++;
else if (rx_bytes < nbytes)
stats->rx_incomp++;
}
}
while ((rx_bytes == 0) ||
((rx_bytes < 0) && ((errno == EAGAIN) || (errno == EWOULDBLOCK))));
if (rx_bytes < 0)
{
errno_val = errno;
perror ("ERROR in sock_test_read()");
fprintf (stderr, "SOCK_TEST: ERROR: socket read "
"failed (errno = %d)!\n", errno_val);
errno = errno_val;
}
else if (stats)
stats->rx_bytes += rx_bytes;
return (rx_bytes);
}
static inline int
sock_test_write (int fd, uint8_t *buf, uint32_t nbytes,
vcl_test_stats_t *stats, uint32_t verbose)
{
int tx_bytes = 0;
int nbytes_left = nbytes;
int rv, errno_val;
do
{
if (stats)
stats->tx_xacts++;
rv = write (fd, buf, nbytes_left);
if (rv < 0)
{
if ((errno == EAGAIN) || (errno == EWOULDBLOCK))
{
if (stats)
stats->tx_eagain++;
continue;
}
else
break;
}
tx_bytes += rv;
if (tx_bytes != nbytes)
{
nbytes_left = nbytes_left - rv;
if (stats)
stats->tx_incomp++;
if (verbose)
{
printf ("SOCK_TEST: WARNING: bytes written (%d) "
"!= bytes to write (%d)!\n", tx_bytes, nbytes);
}
}
} while (tx_bytes != nbytes);
if (tx_bytes < 0)
{
errno_val = errno;
perror ("ERROR in sock_test_write()");
fprintf (stderr, "SOCK_TEST: ERROR: socket write failed "
"(errno = %d)!\n", errno_val);
}
else if (stats)
stats->tx_bytes += tx_bytes;
return (tx_bytes);
}
#endif /* __sock_test_h__ */
|
