xref: /openbmc/qemu/tests/qtest/netdev-socket.c (revision 2df1eb27) 
1 /*
2  * QTest testcase for netdev stream and dgram
3  *
4  * Copyright (c) 2022 Red Hat, Inc.
5  *
6  * SPDX-License-Identifier: GPL-2.0-or-later
7  */
8 
9 #include "qemu/osdep.h"
10 #include "qemu/sockets.h"
11 #include <glib/gstdio.h>
12 #include "../unit/socket-helpers.h"
13 #include "libqtest.h"
14 #include "qapi/qmp/qstring.h"
15 #include "qemu/sockets.h"
16 #include "qapi/qobject-input-visitor.h"
17 #include "qapi/qapi-visit-sockets.h"
18 
19 #define CONNECTION_TIMEOUT    60
20 
21 #define EXPECT_STATE(q, e, t)                             \
22 do {                                                      \
23     char *resp = NULL;                                    \
24     g_test_timer_start();                                 \
25     do {                                                  \
26         g_free(resp);                                     \
27         resp = qtest_hmp(q, "info network");              \
28         if (t) {                                          \
29             strrchr(resp, t)[0] = 0;                      \
30         }                                                 \
31         if (g_str_equal(resp, e)) {                       \
32             break;                                        \
33         }                                                 \
34     } while (g_test_timer_elapsed() < CONNECTION_TIMEOUT); \
35     g_assert_cmpstr(resp, ==, e);                         \
36     g_free(resp);                                         \
37 } while (0)
38 
39 static gchar *tmpdir;
40 
41 static int inet_get_free_port_socket_ipv4(int sock)
42 {
43     struct sockaddr_in addr;
44     socklen_t len;
45 
46     memset(&addr, 0, sizeof(addr));
47     addr.sin_family = AF_INET;
48     addr.sin_addr.s_addr = INADDR_ANY;
49     addr.sin_port = 0;
50     if (bind(sock, (struct sockaddr *)&addr, sizeof(addr)) < 0) {
51         return -1;
52     }
53 
54     len = sizeof(addr);
55     if (getsockname(sock,  (struct sockaddr *)&addr, &len) < 0) {
56         return -1;
57     }
58 
59     return ntohs(addr.sin_port);
60 }
61 
62 static int inet_get_free_port_socket_ipv6(int sock)
63 {
64     struct sockaddr_in6 addr;
65     socklen_t len;
66 
67     memset(&addr, 0, sizeof(addr));
68     addr.sin6_family = AF_INET6;
69     addr.sin6_addr = in6addr_any;
70     addr.sin6_port = 0;
71     if (bind(sock, (struct sockaddr *)&addr, sizeof(addr)) < 0) {
72         return -1;
73     }
74 
75     len = sizeof(addr);
76     if (getsockname(sock,  (struct sockaddr *)&addr, &len) < 0) {
77         return -1;
78     }
79 
80     return ntohs(addr.sin6_port);
81 }
82 
83 static int inet_get_free_port_multiple(int nb, int *port, bool ipv6)
84 {
85     g_autofree int *sock = g_new(int, nb);
86     int i;
87 
88     for (i = 0; i < nb; i++) {
89         sock[i] = socket(ipv6 ? AF_INET6 : AF_INET, SOCK_STREAM, 0);
90         if (sock[i] < 0) {
91             break;
92         }
93         port[i] = ipv6 ? inet_get_free_port_socket_ipv6(sock[i]) :
94                          inet_get_free_port_socket_ipv4(sock[i]);
95         if (port[i] == -1) {
96             break;
97         }
98     }
99 
100     nb = i;
101     for (i = 0; i < nb; i++) {
102         close(sock[i]);
103     }
104 
105     return nb;
106 }
107 
108 static int inet_get_free_port(bool ipv6)
109 {
110     int nb, port;
111 
112     nb = inet_get_free_port_multiple(1, &port, ipv6);
113     g_assert_cmpint(nb, ==, 1);
114 
115     return port;
116 }
117 
118 static void test_stream_inet_ipv4(void)
119 {
120     QTestState *qts0, *qts1;
121     char *expect;
122     int port;
123 
124     port = inet_get_free_port(false);
125     qts0 = qtest_initf("-nodefaults -M none "
126                        "-netdev stream,id=st0,server=true,addr.type=inet,"
127                        "addr.ipv4=on,addr.ipv6=off,"
128                        "addr.host=127.0.0.1,addr.port=%d", port);
129 
130     EXPECT_STATE(qts0, "st0: index=0,type=stream,listening\r\n", 0);
131 
132     qts1 = qtest_initf("-nodefaults -M none "
133                        "-netdev stream,server=false,id=st0,addr.type=inet,"
134                        "addr.ipv4=on,addr.ipv6=off,"
135                        "addr.host=127.0.0.1,addr.port=%d", port);
136 
137     expect = g_strdup_printf("st0: index=0,type=stream,tcp:127.0.0.1:%d\r\n",
138                              port);
139     EXPECT_STATE(qts1, expect, 0);
140     g_free(expect);
141 
142     /* the port is unknown, check only the address */
143     EXPECT_STATE(qts0, "st0: index=0,type=stream,tcp:127.0.0.1", ':');
144 
145     qtest_quit(qts1);
146     qtest_quit(qts0);
147 }
148 
149 static void wait_stream_connected(QTestState *qts, const char *id,
150                                   SocketAddress **addr)
151 {
152     QDict *resp, *data;
153     QString *qstr;
154     QObject *obj;
155     Visitor *v = NULL;
156 
157     resp = qtest_qmp_eventwait_ref(qts, "NETDEV_STREAM_CONNECTED");
158     g_assert_nonnull(resp);
159     data = qdict_get_qdict(resp, "data");
160     g_assert_nonnull(data);
161 
162     qstr = qobject_to(QString, qdict_get(data, "netdev-id"));
163     g_assert_nonnull(data);
164 
165     g_assert(!strcmp(qstring_get_str(qstr), id));
166 
167     obj = qdict_get(data, "addr");
168 
169     v = qobject_input_visitor_new(obj);
170     visit_type_SocketAddress(v, NULL, addr, NULL);
171     visit_free(v);
172     qobject_unref(resp);
173 }
174 
175 static void wait_stream_disconnected(QTestState *qts, const char *id)
176 {
177     QDict *resp, *data;
178     QString *qstr;
179 
180     resp = qtest_qmp_eventwait_ref(qts, "NETDEV_STREAM_DISCONNECTED");
181     g_assert_nonnull(resp);
182     data = qdict_get_qdict(resp, "data");
183     g_assert_nonnull(data);
184 
185     qstr = qobject_to(QString, qdict_get(data, "netdev-id"));
186     g_assert_nonnull(data);
187 
188     g_assert(!strcmp(qstring_get_str(qstr), id));
189     qobject_unref(resp);
190 }
191 
192 static void test_stream_unix_reconnect(void)
193 {
194     QTestState *qts0, *qts1;
195     SocketAddress *addr;
196     gchar *path;
197 
198     path = g_strconcat(tmpdir, "/stream_unix_reconnect", NULL);
199     qts0 = qtest_initf("-nodefaults -M none "
200                        "-netdev stream,id=st0,server=true,addr.type=unix,"
201                        "addr.path=%s", path);
202 
203     EXPECT_STATE(qts0, "st0: index=0,type=stream,listening\r\n", 0);
204 
205     qts1 = qtest_initf("-nodefaults -M none "
206                        "-netdev stream,server=false,id=st0,addr.type=unix,"
207                        "addr.path=%s,reconnect=1", path);
208 
209     wait_stream_connected(qts0, "st0", &addr);
210     g_assert_cmpint(addr->type, ==, SOCKET_ADDRESS_TYPE_UNIX);
211     g_assert_cmpstr(addr->u.q_unix.path, ==, path);
212     qapi_free_SocketAddress(addr);
213 
214     /* kill server */
215     qtest_quit(qts0);
216 
217     /* check client has been disconnected */
218     wait_stream_disconnected(qts1, "st0");
219 
220     /* restart server */
221     qts0 = qtest_initf("-nodefaults -M none "
222                        "-netdev stream,id=st0,server=true,addr.type=unix,"
223                        "addr.path=%s", path);
224 
225     /* wait connection events*/
226     wait_stream_connected(qts0, "st0", &addr);
227     g_assert_cmpint(addr->type, ==, SOCKET_ADDRESS_TYPE_UNIX);
228     g_assert_cmpstr(addr->u.q_unix.path, ==, path);
229     qapi_free_SocketAddress(addr);
230 
231     wait_stream_connected(qts1, "st0", &addr);
232     g_assert_cmpint(addr->type, ==, SOCKET_ADDRESS_TYPE_UNIX);
233     g_assert_cmpstr(addr->u.q_unix.path, ==, path);
234     qapi_free_SocketAddress(addr);
235 
236     qtest_quit(qts1);
237     qtest_quit(qts0);
238     g_free(path);
239 }
240 
241 static void test_stream_inet_ipv6(void)
242 {
243     QTestState *qts0, *qts1;
244     char *expect;
245     int port;
246 
247     port = inet_get_free_port(true);
248     qts0 = qtest_initf("-nodefaults -M none "
249                        "-netdev stream,id=st0,server=true,addr.type=inet,"
250                        "addr.ipv4=off,addr.ipv6=on,"
251                        "addr.host=::1,addr.port=%d", port);
252 
253     EXPECT_STATE(qts0, "st0: index=0,type=stream,listening\r\n", 0);
254 
255     qts1 = qtest_initf("-nodefaults -M none "
256                        "-netdev stream,server=false,id=st0,addr.type=inet,"
257                        "addr.ipv4=off,addr.ipv6=on,"
258                        "addr.host=::1,addr.port=%d", port);
259 
260     expect = g_strdup_printf("st0: index=0,type=stream,tcp:::1:%d\r\n",
261                              port);
262     EXPECT_STATE(qts1, expect, 0);
263     g_free(expect);
264 
265     /* the port is unknown, check only the address */
266     EXPECT_STATE(qts0, "st0: index=0,type=stream,tcp:::1", ':');
267 
268     qtest_quit(qts1);
269     qtest_quit(qts0);
270 }
271 
272 static void test_stream_unix(void)
273 {
274     QTestState *qts0, *qts1;
275     char *expect;
276     gchar *path;
277 
278     path = g_strconcat(tmpdir, "/stream_unix", NULL);
279 
280     qts0 = qtest_initf("-nodefaults -M none "
281                        "-netdev stream,id=st0,server=true,"
282                        "addr.type=unix,addr.path=%s,",
283                        path);
284 
285     EXPECT_STATE(qts0, "st0: index=0,type=stream,listening\r\n", 0);
286 
287     qts1 = qtest_initf("-nodefaults -M none "
288                        "-netdev stream,id=st0,server=false,"
289                        "addr.type=unix,addr.path=%s",
290                        path);
291 
292     expect = g_strdup_printf("st0: index=0,type=stream,unix:%s\r\n", path);
293     EXPECT_STATE(qts1, expect, 0);
294     EXPECT_STATE(qts0, expect, 0);
295     g_free(expect);
296     g_free(path);
297 
298     qtest_quit(qts1);
299     qtest_quit(qts0);
300 }
301 
302 #ifdef CONFIG_LINUX
303 static void test_stream_unix_abstract(void)
304 {
305     QTestState *qts0, *qts1;
306     char *expect;
307     gchar *path;
308 
309     path = g_strconcat(tmpdir, "/stream_unix_abstract", NULL);
310 
311     qts0 = qtest_initf("-nodefaults -M none "
312                        "-netdev stream,id=st0,server=true,"
313                        "addr.type=unix,addr.path=%s,"
314                        "addr.abstract=on",
315                        path);
316 
317     EXPECT_STATE(qts0, "st0: index=0,type=stream,listening\r\n", 0);
318 
319     qts1 = qtest_initf("-nodefaults -M none "
320                        "-netdev stream,id=st0,server=false,"
321                        "addr.type=unix,addr.path=%s,addr.abstract=on",
322                        path);
323 
324     expect = g_strdup_printf("st0: index=0,type=stream,unix:%s\r\n", path);
325     EXPECT_STATE(qts1, expect, 0);
326     EXPECT_STATE(qts0, expect, 0);
327     g_free(expect);
328     g_free(path);
329 
330     qtest_quit(qts1);
331     qtest_quit(qts0);
332 }
333 #endif
334 
335 #ifndef _WIN32
336 static void test_stream_fd(void)
337 {
338     QTestState *qts0, *qts1;
339     int sock[2];
340     int ret;
341 
342     ret = socketpair(AF_LOCAL, SOCK_STREAM, 0, sock);
343     g_assert_true(ret == 0);
344 
345     qts0 = qtest_initf("-nodefaults -M none "
346                        "-netdev stream,id=st0,addr.type=fd,addr.str=%d",
347                        sock[0]);
348 
349     EXPECT_STATE(qts0, "st0: index=0,type=stream,unix:\r\n", 0);
350 
351     qts1 = qtest_initf("-nodefaults -M none "
352                        "-netdev stream,id=st0,addr.type=fd,addr.str=%d",
353                        sock[1]);
354 
355     EXPECT_STATE(qts1, "st0: index=0,type=stream,unix:\r\n", 0);
356     EXPECT_STATE(qts0, "st0: index=0,type=stream,unix:\r\n", 0);
357 
358     qtest_quit(qts1);
359     qtest_quit(qts0);
360 
361     close(sock[0]);
362     close(sock[1]);
363 }
364 #endif
365 
366 static void test_dgram_inet(void)
367 {
368     QTestState *qts0, *qts1;
369     char *expect;
370     int port[2];
371     int nb;
372 
373     nb = inet_get_free_port_multiple(2, port, false);
374     g_assert_cmpint(nb, ==, 2);
375 
376     qts0 = qtest_initf("-nodefaults -M none "
377                        "-netdev dgram,id=st0,"
378                        "local.type=inet,local.host=127.0.0.1,local.port=%d,"
379                        "remote.type=inet,remote.host=127.0.0.1,remote.port=%d",
380                         port[0], port[1]);
381 
382     expect = g_strdup_printf("st0: index=0,type=dgram,"
383                              "udp=127.0.0.1:%d/127.0.0.1:%d\r\n",
384                              port[0], port[1]);
385     EXPECT_STATE(qts0, expect, 0);
386     g_free(expect);
387 
388     qts1 = qtest_initf("-nodefaults -M none "
389                        "-netdev dgram,id=st0,"
390                        "local.type=inet,local.host=127.0.0.1,local.port=%d,"
391                        "remote.type=inet,remote.host=127.0.0.1,remote.port=%d",
392                         port[1], port[0]);
393 
394     expect = g_strdup_printf("st0: index=0,type=dgram,"
395                              "udp=127.0.0.1:%d/127.0.0.1:%d\r\n",
396                              port[1], port[0]);
397     EXPECT_STATE(qts1, expect, 0);
398     g_free(expect);
399 
400     qtest_quit(qts1);
401     qtest_quit(qts0);
402 }
403 
404 #if !defined(_WIN32) && !defined(CONFIG_DARWIN)
405 static void test_dgram_mcast(void)
406 {
407     QTestState *qts;
408 
409     qts = qtest_initf("-nodefaults -M none "
410                       "-netdev dgram,id=st0,"
411                       "remote.type=inet,remote.host=230.0.0.1,remote.port=1234");
412 
413     EXPECT_STATE(qts, "st0: index=0,type=dgram,mcast=230.0.0.1:1234\r\n", 0);
414 
415     qtest_quit(qts);
416 }
417 #endif
418 
419 #ifndef _WIN32
420 static void test_dgram_unix(void)
421 {
422     QTestState *qts0, *qts1;
423     char *expect;
424     gchar *path0, *path1;
425 
426     path0 = g_strconcat(tmpdir, "/dgram_unix0", NULL);
427     path1 = g_strconcat(tmpdir, "/dgram_unix1", NULL);
428 
429     qts0 = qtest_initf("-nodefaults -M none "
430                        "-netdev dgram,id=st0,local.type=unix,local.path=%s,"
431                        "remote.type=unix,remote.path=%s",
432                        path0, path1);
433 
434     expect = g_strdup_printf("st0: index=0,type=dgram,udp=%s:%s\r\n",
435                              path0, path1);
436     EXPECT_STATE(qts0, expect, 0);
437     g_free(expect);
438 
439     qts1 = qtest_initf("-nodefaults -M none "
440                        "-netdev dgram,id=st0,local.type=unix,local.path=%s,"
441                        "remote.type=unix,remote.path=%s",
442                        path1, path0);
443 
444 
445     expect = g_strdup_printf("st0: index=0,type=dgram,udp=%s:%s\r\n",
446                              path1, path0);
447     EXPECT_STATE(qts1, expect, 0);
448     g_free(expect);
449 
450     unlink(path0);
451     g_free(path0);
452     unlink(path1);
453     g_free(path1);
454 
455     qtest_quit(qts1);
456     qtest_quit(qts0);
457 }
458 
459 static void test_dgram_fd(void)
460 {
461     QTestState *qts0, *qts1;
462     char *expect;
463     int ret;
464     int sv[2];
465 
466     ret = socketpair(PF_UNIX, SOCK_DGRAM, 0, sv);
467     g_assert_cmpint(ret, !=, -1);
468 
469     qts0 = qtest_initf("-nodefaults -M none "
470                        "-netdev dgram,id=st0,local.type=fd,local.str=%d",
471                        sv[0]);
472 
473     expect = g_strdup_printf("st0: index=0,type=dgram,fd=%d unix\r\n", sv[0]);
474     EXPECT_STATE(qts0, expect, 0);
475     g_free(expect);
476 
477     qts1 = qtest_initf("-nodefaults -M none "
478                        "-netdev dgram,id=st0,local.type=fd,local.str=%d",
479                        sv[1]);
480 
481 
482     expect = g_strdup_printf("st0: index=0,type=dgram,fd=%d unix\r\n", sv[1]);
483     EXPECT_STATE(qts1, expect, 0);
484     g_free(expect);
485 
486     qtest_quit(qts1);
487     qtest_quit(qts0);
488 
489     close(sv[0]);
490     close(sv[1]);
491 }
492 #endif
493 
494 int main(int argc, char **argv)
495 {
496     int ret;
497     bool has_ipv4, has_ipv6, has_afunix;
498     g_autoptr(GError) err = NULL;
499 
500     socket_init();
501     g_test_init(&argc, &argv, NULL);
502 
503     if (socket_check_protocol_support(&has_ipv4, &has_ipv6) < 0) {
504         g_error("socket_check_protocol_support() failed\n");
505     }
506 
507     tmpdir = g_dir_make_tmp("netdev-socket.XXXXXX", &err);
508     if (tmpdir == NULL) {
509         g_error("Can't create temporary directory in %s: %s",
510                 g_get_tmp_dir(), err->message);
511     }
512 
513     if (has_ipv4) {
514         qtest_add_func("/netdev/stream/inet/ipv4", test_stream_inet_ipv4);
515         qtest_add_func("/netdev/dgram/inet", test_dgram_inet);
516 #if !defined(_WIN32) && !defined(CONFIG_DARWIN)
517         qtest_add_func("/netdev/dgram/mcast", test_dgram_mcast);
518 #endif
519     }
520     if (has_ipv6) {
521         qtest_add_func("/netdev/stream/inet/ipv6", test_stream_inet_ipv6);
522     }
523 
524     socket_check_afunix_support(&has_afunix);
525     if (has_afunix) {
526 #ifndef _WIN32
527         qtest_add_func("/netdev/dgram/unix", test_dgram_unix);
528 #endif
529         qtest_add_func("/netdev/stream/unix/oneshot", test_stream_unix);
530         qtest_add_func("/netdev/stream/unix/reconnect",
531                        test_stream_unix_reconnect);
532 #ifdef CONFIG_LINUX
533         qtest_add_func("/netdev/stream/unix/abstract",
534                        test_stream_unix_abstract);
535 #endif
536 #ifndef _WIN32
537         qtest_add_func("/netdev/stream/fd", test_stream_fd);
538         qtest_add_func("/netdev/dgram/fd", test_dgram_fd);
539 #endif
540     }
541 
542     ret = g_test_run();
543 
544     g_rmdir(tmpdir);
545     g_free(tmpdir);
546 
547     return ret;
548 }
549