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
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
|
/*
* Copyright (c) 2008-2018 Cisco Systems, Inc. All rights reserved
* Copyright (c) 2009 Sandia National Laboratories. All rights reserved.
* Copyright (c) 2017 Mellanox Technologies. All rights reserved.
*
* $COPYRIGHT$
*
* Additional copyrights may follow
*
* $HEADER$
*/
#include "opal_config.h"
#ifdef HAVE_SYS_TYPES_H
# include <sys/types.h>
#endif
#ifdef HAVE_SYS_STAT_H
# include <sys/stat.h>
#endif
#ifdef HAVE_SYS_SOCKET_H
# include <sys/socket.h>
#endif
#ifdef HAVE_ARPA_INET_H
# include <arpa/inet.h>
#endif
#ifdef HAVE_NETINET_IN_H
# include <netinet/in.h>
#endif
#ifdef HAVE_UNISTD_H
# include <unistd.h>
#endif
#include <errno.h>
#include <fcntl.h>
#include <stdlib.h>
#include <string.h>
#include "opal/constants.h"
#include "opal/util/fd.h"
#include "opal/util/string_copy.h"
/*
* Simple loop over reading from a fd
*/
int opal_fd_read(int fd, int len, void *buffer)
{
int rc;
char *b = buffer;
while (len > 0) {
rc = read(fd, b, len);
if (rc < 0 && (EAGAIN == errno || EINTR == errno)) {
continue;
} else if (rc > 0) {
len -= rc;
b += rc;
} else if (0 == rc) {
return OPAL_ERR_TIMEOUT;
} else {
return OPAL_ERR_IN_ERRNO;
}
}
return OPAL_SUCCESS;
}
/*
* Simple loop over writing to an fd
*/
int opal_fd_write(int fd, int len, const void *buffer)
{
int rc;
const char *b = buffer;
while (len > 0) {
rc = write(fd, b, len);
if (rc < 0 && (EAGAIN == errno || EINTR == errno)) {
continue;
} else if (rc > 0) {
len -= rc;
b += rc;
} else {
return OPAL_ERR_IN_ERRNO;
}
}
return OPAL_SUCCESS;
}
int opal_fd_set_cloexec(int fd)
{
#ifdef FD_CLOEXEC
int flags;
/* Stevens says that we should get the fd's flags before we set
them. So say we all. */
flags = fcntl(fd, F_GETFD, 0);
if (-1 == flags) {
return OPAL_ERR_IN_ERRNO;
}
if (fcntl(fd, F_SETFD, FD_CLOEXEC | flags) == -1) {
return OPAL_ERR_IN_ERRNO;
}
#endif
return OPAL_SUCCESS;
}
bool opal_fd_is_regular(int fd)
{
struct stat buf;
if (fstat(fd, &buf)) {
return false;
}
return S_ISREG(buf.st_mode);
}
bool opal_fd_is_chardev(int fd)
{
struct stat buf;
if (fstat(fd, &buf)) {
return false;
}
return S_ISCHR(buf.st_mode);
}
bool opal_fd_is_blkdev(int fd)
{
struct stat buf;
if (fstat(fd, &buf)) {
return false;
}
return S_ISBLK(buf.st_mode);
}
const char *opal_fd_get_peer_name(int fd)
{
char *str;
const char *ret = NULL;
struct sockaddr sa;
socklen_t slt = (socklen_t) sizeof(sa);
int rc = getpeername(fd, &sa, &slt);
if (0 != rc) {
ret = strdup("Unknown");
return ret;
}
size_t len = INET_ADDRSTRLEN;
#if OPAL_ENABLE_IPV6
len = INET6_ADDRSTRLEN;
#endif
str = calloc(1, len);
if (NULL == str) {
return NULL;
}
if (sa.sa_family == AF_INET) {
struct sockaddr_in *si;
si = (struct sockaddr_in *) &sa;
ret = inet_ntop(AF_INET, &(si->sin_addr), str, INET_ADDRSTRLEN);
if (NULL == ret) {
free(str);
}
}
#if OPAL_ENABLE_IPV6
else if (sa.sa_family == AF_INET6) {
struct sockaddr_in6 *si6;
si6 = (struct sockaddr_in6 *) &sa;
ret = inet_ntop(AF_INET6, &(si6->sin6_addr), str, INET6_ADDRSTRLEN);
if (NULL == ret) {
free(str);
}
}
#endif
else {
// This string is guaranteed to be <= INET_ADDRSTRLEN
opal_string_copy(str, "Unknown", len);
ret = str;
}
return ret;
}
|
