xref: /openbmc/qemu/tests/qtest/netdev-socket.c (revision 5f88dd43) 
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,\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,\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,\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,\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,\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 #ifndef _WIN32
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 
418 static void test_dgram_unix(void)
419 {
420     QTestState *qts0, *qts1;
421     char *expect;
422     gchar *path0, *path1;
423 
424     path0 = g_strconcat(tmpdir, "/dgram_unix0", NULL);
425     path1 = g_strconcat(tmpdir, "/dgram_unix1", NULL);
426 
427     qts0 = qtest_initf("-nodefaults -M none "
428                        "-netdev dgram,id=st0,local.type=unix,local.path=%s,"
429                        "remote.type=unix,remote.path=%s",
430                        path0, path1);
431 
432     expect = g_strdup_printf("st0: index=0,type=dgram,udp=%s:%s\r\n",
433                              path0, path1);
434     EXPECT_STATE(qts0, expect, 0);
435     g_free(expect);
436 
437     qts1 = qtest_initf("-nodefaults -M none "
438                        "-netdev dgram,id=st0,local.type=unix,local.path=%s,"
439                        "remote.type=unix,remote.path=%s",
440                        path1, path0);
441 
442 
443     expect = g_strdup_printf("st0: index=0,type=dgram,udp=%s:%s\r\n",
444                              path1, path0);
445     EXPECT_STATE(qts1, expect, 0);
446     g_free(expect);
447 
448     unlink(path0);
449     g_free(path0);
450     unlink(path1);
451     g_free(path1);
452 
453     qtest_quit(qts1);
454     qtest_quit(qts0);
455 }
456 
457 static void test_dgram_fd(void)
458 {
459     QTestState *qts0, *qts1;
460     char *expect;
461     int ret;
462     int sv[2];
463 
464     ret = socketpair(PF_UNIX, SOCK_DGRAM, 0, sv);
465     g_assert_cmpint(ret, !=, -1);
466 
467     qts0 = qtest_initf("-nodefaults -M none "
468                        "-netdev dgram,id=st0,local.type=fd,local.str=%d",
469                        sv[0]);
470 
471     expect = g_strdup_printf("st0: index=0,type=dgram,fd=%d unix\r\n", sv[0]);
472     EXPECT_STATE(qts0, expect, 0);
473     g_free(expect);
474 
475     qts1 = qtest_initf("-nodefaults -M none "
476                        "-netdev dgram,id=st0,local.type=fd,local.str=%d",
477                        sv[1]);
478 
479 
480     expect = g_strdup_printf("st0: index=0,type=dgram,fd=%d unix\r\n", sv[1]);
481     EXPECT_STATE(qts1, expect, 0);
482     g_free(expect);
483 
484     qtest_quit(qts1);
485     qtest_quit(qts0);
486 
487     close(sv[0]);
488     close(sv[1]);
489 }
490 #endif
491 
492 int main(int argc, char **argv)
493 {
494     int ret;
495     bool has_ipv4, has_ipv6, has_afunix;
496     g_autoptr(GError) err = NULL;
497 
498     socket_init();
499     g_test_init(&argc, &argv, NULL);
500 
501     if (socket_check_protocol_support(&has_ipv4, &has_ipv6) < 0) {
502         g_error("socket_check_protocol_support() failed\n");
503     }
504 
505     tmpdir = g_dir_make_tmp("netdev-socket.XXXXXX", &err);
506     if (tmpdir == NULL) {
507         g_error("Can't create temporary directory in %s: %s",
508                 g_get_tmp_dir(), err->message);
509     }
510 
511     if (has_ipv4) {
512         qtest_add_func("/netdev/stream/inet/ipv4", test_stream_inet_ipv4);
513         qtest_add_func("/netdev/dgram/inet", test_dgram_inet);
514 #ifndef _WIN32
515         qtest_add_func("/netdev/dgram/mcast", test_dgram_mcast);
516 #endif
517     }
518     if (has_ipv6) {
519         qtest_add_func("/netdev/stream/inet/ipv6", test_stream_inet_ipv6);
520     }
521 
522     socket_check_afunix_support(&has_afunix);
523     if (has_afunix) {
524 #ifndef _WIN32
525         qtest_add_func("/netdev/dgram/unix", test_dgram_unix);
526 #endif
527         qtest_add_func("/netdev/stream/unix", test_stream_unix);
528         qtest_add_func("/netdev/stream/unix/reconnect",
529                        test_stream_unix_reconnect);
530 #ifdef CONFIG_LINUX
531         qtest_add_func("/netdev/stream/unix/abstract",
532                        test_stream_unix_abstract);
533 #endif
534 #ifndef _WIN32
535         qtest_add_func("/netdev/stream/fd", test_stream_fd);
536         qtest_add_func("/netdev/dgram/fd", test_dgram_fd);
537 #endif
538     }
539 
540     ret = g_test_run();
541 
542     g_rmdir(tmpdir);
543     g_free(tmpdir);
544 
545     return ret;
546 }
547