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
|
#include <errno.h>
#include <netinet/in.h>
#include <wasi/descriptor_table.h>
#include <wasi/sockets_utils.h>
static int tcp_socket(network_ip_address_family_t family, bool blocking)
{
tcp_create_socket_error_code_t error;
tcp_own_tcp_socket_t socket;
if (!tcp_create_socket_create_tcp_socket(family, &socket, &error)) {
errno = __wasi_sockets_utils__map_error(error);
return -1;
}
tcp_borrow_tcp_socket_t socket_borrow = tcp_borrow_tcp_socket(socket);
poll_own_pollable_t socket_pollable =
tcp_method_tcp_socket_subscribe(socket_borrow);
descriptor_table_entry_t
entry = { .tag = DESCRIPTOR_TABLE_ENTRY_TCP_SOCKET,
.tcp_socket = {
.socket = socket,
.socket_pollable = socket_pollable,
.blocking = blocking,
.fake_nodelay = false,
.family = family,
.state = { .tag = TCP_SOCKET_STATE_UNBOUND,
.unbound = {
/* No additional state. */ } },
} };
int fd;
if (!descriptor_table_insert(entry, &fd)) {
errno = EMFILE;
return -1;
}
return fd;
}
static int udp_socket(network_ip_address_family_t family, bool blocking)
{
udp_create_socket_error_code_t error;
udp_own_udp_socket_t socket;
if (!udp_create_socket_create_udp_socket(family, &socket, &error)) {
errno = __wasi_sockets_utils__map_error(error);
return -1;
}
udp_borrow_udp_socket_t socket_borrow = udp_borrow_udp_socket(socket);
poll_own_pollable_t socket_pollable =
udp_method_udp_socket_subscribe(socket_borrow);
descriptor_table_entry_t
entry = { .tag = DESCRIPTOR_TABLE_ENTRY_UDP_SOCKET,
.udp_socket = {
.socket = socket,
.socket_pollable = socket_pollable,
.blocking = blocking,
.family = family,
.state = { .tag = UDP_SOCKET_STATE_UNBOUND,
.unbound = {
/* No additional state. */ } },
} };
int fd;
if (!descriptor_table_insert(entry, &fd)) {
errno = EMFILE;
return -1;
}
return fd;
}
int socket(int domain, int type, int protocol)
{
network_ip_address_family_t family;
switch (domain) {
case PF_INET:
family = NETWORK_IP_ADDRESS_FAMILY_IPV4;
break;
case PF_INET6:
family = NETWORK_IP_ADDRESS_FAMILY_IPV6;
break;
default:
errno = EAFNOSUPPORT;
return -1;
}
int real_type = type & ~(SOCK_NONBLOCK | SOCK_CLOEXEC);
bool blocking = (type & SOCK_NONBLOCK) == 0;
// Ignore SOCK_CLOEXEC flag. That concept does not exist in WASI.
if (real_type == SOCK_STREAM &&
(protocol == 0 || protocol == IPPROTO_TCP)) {
return tcp_socket(family, blocking);
} else if (real_type == SOCK_DGRAM &&
(protocol == 0 || protocol == IPPROTO_UDP)) {
return udp_socket(family, blocking);
} else {
errno = EPROTONOSUPPORT;
return -1;
}
}
|
