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
|
/* socket.c
* Socket wrappers
*
* Copyright 2019, Gerald Combs
*
* Wireshark - Network traffic analyzer
* By Gerald Combs <gerald@wireshark.org>
* Copyright 1998 Gerald Combs
*
* SPDX-License-Identifier: GPL-2.0-or-later
*/
#include "config.h"
#include "socket.h"
#include <stdlib.h>
#include <errno.h>
#include <wsutil/inet_addr.h>
#ifdef _WIN32
#include <wsutil/win32-utils.h>
#define in_port_t uint16_t
#endif
char *
ws_init_sockets(void)
{
char *errmsg = NULL;
#ifdef _WIN32
int err;
WORD wVersionRequested;
WSADATA wsaData;
wVersionRequested = MAKEWORD(2, 2);
err = WSAStartup(wVersionRequested, &wsaData);
if (err != 0) {
errmsg = ws_strdup_printf("Couldn't initialize Windows Sockets: %s",
win32strerror(err));
}
#endif
return errmsg;
}
void
ws_cleanup_sockets(void)
{
#ifdef _WIN32
/* XXX - any reason to check the error return? */
WSACleanup();
#endif
}
int
ws_socket_ptoa(struct sockaddr_storage *dst, const char *src,
uint16_t def_port)
{
int ret = -1, af = -1;
char *addr_src, *p;
char *addr_str = NULL, *port_str = NULL;
union {
ws_in4_addr ip4;
ws_in6_addr ip6;
} addr;
char *endptr;
long num;
in_port_t port;
addr_src = g_strdup(src);
/* Is it an IPv6/IPv4 literal address enclosed in braces? */
if (*addr_src == '[') {
addr_str = addr_src + 1;
if ((p = strchr(addr_str, ']')) == NULL) {
errno = EINVAL;
goto out;
}
*p++ = '\0';
if (*p == ':') {
port_str = p + 1;
}
else if (*p != '\0') {
errno = EINVAL;
goto out;
}
if (ws_inet_pton6(addr_str, &addr.ip6)) {
af = AF_INET6;
}
else if (ws_inet_pton4(addr_str, &addr.ip4)) {
af = AF_INET;
}
else {
errno = EINVAL;
goto out;
}
}
else {
/* It is an IPv4 dotted decimal. */
addr_str = addr_src;
if ((p = strchr(addr_str, ':')) != NULL) {
*p++ = '\0';
port_str = p;
}
if (ws_inet_pton4(addr_str, &addr.ip4)) {
af = AF_INET;
}
else {
errno = EINVAL;
goto out;
}
}
if (port_str != NULL && *port_str != '\0') {
num = strtol(port_str, &endptr, 10);
/* We want the entire string to be a valid decimal representation. */
if (endptr == port_str || *endptr != '\0' || num < 0 || num > UINT16_MAX) {
errno = EINVAL;
goto out;
}
port = g_htons(num);
}
else {
port = g_htons(def_port);
}
/* sockaddr_storage is guaranteed to fit any sockaddr type. */
if (af == AF_INET6) {
struct sockaddr_in6 *sa = (struct sockaddr_in6 *)dst;
memset(sa, 0, sizeof(struct sockaddr_in6));
sa->sin6_family = AF_INET6;
sa->sin6_port = port;
memcpy(&sa->sin6_addr, &addr.ip6, sizeof(struct in6_addr));
ret = 0;
}
else if (af == AF_INET) {
struct sockaddr_in *sa = (struct sockaddr_in *)dst;
memset(sa, 0, sizeof(struct sockaddr_in));
sa->sin_family = AF_INET;
sa->sin_port = port;
memcpy(&sa->sin_addr, &addr.ip4, sizeof(struct in_addr));
ret = 0;
}
else {
ws_assert_not_reached();
}
out:
g_free(addr_src);
return ret;
}
|
