1+ /*
2+ * Copyright (C) 2023 Amazon.com Inc. or its affiliates. All rights reserved.
3+ * SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
4+ */
5+ #include <unistd.h>
6+ #include <string.h>
7+ #include <assert.h>
8+ #ifdef __wasi__
9+ #include <wasi/api.h>
10+ #include <sys/socket.h>
11+ #include <wasi_socket_ext.h>
12+ #endif
13+ #include <arpa/inet.h>
14+ #include <pthread.h>
15+ #define SERVER_MSG "Message from server."
16+ #define PORT 8989
17+ pthread_mutex_t mut ;
18+ pthread_cond_t cond ;
19+ int server_init_complete = 0 ;
20+ char buffer [sizeof (SERVER_MSG ) + 1 ];
21+
22+ struct socket_info {
23+ union {
24+ struct sockaddr_in addr_ipv4 ;
25+ struct sockaddr_in6 addr_ipv6 ;
26+ } addr ;
27+ int sock ;
28+ };
29+
30+ struct thread_args {
31+ int family ;
32+ int protocol ;
33+ };
34+
35+ struct socket_info
36+ init_socket_addr (int family , int protocol )
37+ {
38+ int sock = socket (family , protocol , 0 );
39+ assert (sock != -1 );
40+
41+ struct socket_info info ;
42+ if (family == AF_INET ) {
43+ struct sockaddr_in addr ;
44+ memset (& addr , 0 , sizeof (addr ));
45+ addr .sin_family = AF_INET ;
46+ addr .sin_port = htons (PORT );
47+ addr .sin_addr .s_addr = htonl (INADDR_LOOPBACK );
48+ info .addr .addr_ipv4 = addr ;
49+ }
50+ else if (family == AF_INET6 ) {
51+ struct sockaddr_in6 addr ;
52+ memset (& addr , 0 , sizeof (addr ));
53+ addr .sin6_family = AF_INET6 ;
54+ addr .sin6_port = htons (PORT );
55+ addr .sin6_addr = in6addr_loopback ;
56+ info .addr .addr_ipv6 = addr ;
57+ }
58+ info .sock = sock ;
59+ return info ;
60+ }
61+
62+ void
63+ assert_thread_args (struct thread_args * args )
64+ {
65+ assert (args -> family == AF_INET || args -> family == AF_INET6 );
66+ assert (args -> protocol == SOCK_STREAM || args -> protocol == SOCK_DGRAM );
67+ }
68+
69+ void *
70+ server (void * arg )
71+ {
72+ server_init_complete = 0 ;
73+ struct thread_args * args = (struct thread_args * )arg ;
74+ assert_thread_args (args );
75+
76+ struct socket_info init_server_sock =
77+ init_socket_addr (args -> family , args -> protocol );
78+
79+ int server_sock = init_server_sock .sock ;
80+ socklen_t addr_size ;
81+ struct sockaddr_storage client_addr ;
82+ strcpy (buffer , SERVER_MSG );
83+
84+ struct sockaddr * server_addr = (struct sockaddr * )& init_server_sock .addr ;
85+ int ret = bind (server_sock , server_addr ,
86+ args -> family == AF_INET ? sizeof (struct sockaddr_in )
87+ : sizeof (struct sockaddr_in6 ));
88+ assert (ret == 0 );
89+
90+ (args -> protocol == SOCK_STREAM ) && listen (server_sock , 1 );
91+ pthread_mutex_lock (& mut );
92+ server_init_complete = 1 ;
93+ pthread_mutex_unlock (& mut );
94+ pthread_cond_signal (& cond );
95+
96+ addr_size = sizeof (client_addr );
97+ if (args -> protocol == SOCK_STREAM ) {
98+ int client_sock =
99+ accept (server_sock , (struct sockaddr * )& client_addr , & addr_size );
100+ assert (client_sock >= 0 );
101+ sendto (client_sock , buffer , strlen (buffer ), 0 ,
102+ (struct sockaddr * )& client_addr , addr_size );
103+
104+ assert (close (client_sock ) == 0 );
105+ }
106+ else {
107+ recvfrom (server_sock , buffer , sizeof (buffer ), 0 ,
108+ (struct sockaddr * )& client_addr , & addr_size );
109+ sendto (server_sock , buffer , strlen (buffer ), 0 ,
110+ (struct sockaddr * )& client_addr , addr_size );
111+
112+ assert (close (server_sock ) == 0 );
113+ }
114+
115+ return NULL ;
116+ }
117+
118+ void *
119+ client (void * arg )
120+ {
121+ struct thread_args * args = (struct thread_args * )arg ;
122+ assert_thread_args (args );
123+
124+ pthread_mutex_lock (& mut );
125+
126+ while (server_init_complete == 0 ) {
127+ pthread_cond_wait (& cond , & mut );
128+ }
129+
130+ struct socket_info init_client_sock =
131+ init_socket_addr (args -> family , args -> protocol );
132+ int sock = init_client_sock .sock ;
133+ pthread_mutex_unlock (& mut );
134+
135+ if (args -> family == AF_INET ) {
136+ struct sockaddr_in addr = init_client_sock .addr .addr_ipv4 ;
137+ if (args -> protocol == SOCK_STREAM ) {
138+ assert (connect (sock , (struct sockaddr * )& addr , sizeof (addr )) != -1 );
139+ }
140+ else {
141+ assert (sendto (sock , buffer , strlen (buffer ), 0 ,
142+ (struct sockaddr * )& addr , sizeof (addr ))
143+ != -1 );
144+ }
145+ }
146+ else {
147+ struct sockaddr_in6 addr = init_client_sock .addr .addr_ipv6 ;
148+ if (args -> protocol == SOCK_STREAM ) {
149+ assert (connect (sock , (struct sockaddr * )& addr , sizeof (addr )) != -1 );
150+ }
151+ else {
152+ assert (sendto (sock , buffer , strlen (buffer ), 0 ,
153+ (struct sockaddr * )& addr , sizeof (addr ))
154+ != -1 );
155+ }
156+ }
157+
158+ recv (sock , buffer , sizeof (buffer ), 0 );
159+ assert (strcmp (buffer , SERVER_MSG ) == 0 );
160+ assert (close (sock ) == 0 );
161+ return NULL ;
162+ }
163+
164+ void
165+ test_protocol (int family , int protocol )
166+ {
167+ pthread_t server_thread , client_thread ;
168+ assert (pthread_cond_init (& cond , NULL ) == 0 );
169+ assert (pthread_mutex_init (& mut , NULL ) == 0 );
170+
171+ struct thread_args args = { family , protocol };
172+ assert (pthread_create (& server_thread , NULL , server , (void * )& args ) == 0 );
173+ assert (pthread_create (& client_thread , NULL , client , (void * )& args ) == 0 );
174+ assert (pthread_join (server_thread , NULL ) == 0 );
175+ assert (pthread_join (client_thread , NULL ) == 0 );
176+
177+ assert (pthread_mutex_destroy (& mut ) == 0 );
178+ assert (pthread_cond_destroy (& cond ) == 0 );
179+ }
180+
181+ int
182+ main (int argc , char * * argv )
183+ {
184+ /* test tcp with ipv4 and ipv6 */
185+ test_protocol (AF_INET , SOCK_STREAM );
186+ test_protocol (AF_INET6 , SOCK_STREAM );
187+
188+ /* test udp with ipv4 and ipv6 */
189+ test_protocol (AF_INET , SOCK_DGRAM );
190+ test_protocol (AF_INET6 , SOCK_DGRAM );
191+
192+ return 0 ;
193+ }
0 commit comments