1 /* GLib testing framework examples and tests
3 * Copyright (C) 2008-2011 Red Hat, Inc.
5 * This library is free software; you can redistribute it and/or
6 * modify it under the terms of the GNU Lesser General Public
7 * License as published by the Free Software Foundation; either
8 * version 2 of the License, or (at your option) any later version.
10 * This library is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * Lesser General Public License for more details.
15 * You should have received a copy of the GNU Lesser General
16 * Public License along with this library; if not, write to the
17 * Free Software Foundation, Inc., 59 Temple Place, Suite 330,
18 * Boston, MA 02111-1307, USA.
25 #include <sys/types.h>
26 #include <sys/socket.h>
30 #include <gio/gunixconnection.h>
33 #include "gnetworkingprivate.h"
44 echo_server_thread (gpointer user_data)
46 IPTestData *data = user_data;
52 sock = g_socket_accept (data->server, NULL, &error);
53 g_assert_no_error (error);
57 nread = g_socket_receive (sock, buf, sizeof (buf), NULL, &error);
58 g_assert_no_error (error);
59 g_assert_cmpint (nread, >=, 0);
64 nwrote = g_socket_send (sock, buf, nread, NULL, &error);
65 g_assert_no_error (error);
66 g_assert_cmpint (nwrote, ==, nread);
69 g_socket_close (sock, &error);
70 g_assert_no_error (error);
71 g_object_unref (sock);
76 create_server (GSocketFamily family,
77 GThreadFunc server_thread,
86 data = g_slice_new (IPTestData);
87 data->family = family;
89 data->server = server = g_socket_new (family,
91 G_SOCKET_PROTOCOL_DEFAULT,
93 g_assert_no_error (error);
95 g_assert_cmpint (g_socket_get_family (server), ==, family);
96 g_assert_cmpint (g_socket_get_socket_type (server), ==, G_SOCKET_TYPE_STREAM);
97 g_assert_cmpint (g_socket_get_protocol (server), ==, G_SOCKET_PROTOCOL_DEFAULT);
99 g_socket_set_blocking (server, TRUE);
101 #if defined (IPPROTO_IPV6) && defined (IPV6_V6ONLY)
106 fd = g_socket_get_fd (server);
108 setsockopt (fd, IPPROTO_IPV6, IPV6_V6ONLY, &v6_only, sizeof (v6_only));
113 iaddr = g_inet_address_new_any (family);
115 iaddr = g_inet_address_new_loopback (family);
116 addr = g_inet_socket_address_new (iaddr, 0);
117 g_object_unref (iaddr);
119 g_assert_cmpint (g_inet_socket_address_get_port (G_INET_SOCKET_ADDRESS (addr)), ==, 0);
120 g_socket_bind (server, addr, TRUE, &error);
121 g_assert_no_error (error);
122 g_object_unref (addr);
124 addr = g_socket_get_local_address (server, &error);
125 g_assert_no_error (error);
126 g_assert_cmpint (g_inet_socket_address_get_port (G_INET_SOCKET_ADDRESS (addr)), !=, 0);
127 g_object_unref (addr);
129 g_socket_listen (server, &error);
130 g_assert_no_error (error);
132 data->thread = g_thread_new ("server", server_thread, data);
137 static const gchar *testbuf = "0123456789abcdef";
140 test_ip_async_read_ready (GSocket *client,
144 IPTestData *data = user_data;
145 GError *error = NULL;
149 g_assert_cmpint (cond, ==, G_IO_IN);
151 len = g_socket_receive (client, buf, sizeof (buf), NULL, &error);
152 g_assert_no_error (error);
153 g_assert_cmpint (len, ==, strlen (testbuf) + 1);
155 g_assert_cmpstr (testbuf, ==, buf);
157 g_main_loop_quit (data->loop);
163 test_ip_async_write_ready (GSocket *client,
167 IPTestData *data = user_data;
168 GError *error = NULL;
172 g_assert_cmpint (cond, ==, G_IO_OUT);
174 len = g_socket_send (client, testbuf, strlen (testbuf) + 1, NULL, &error);
175 g_assert_no_error (error);
176 g_assert_cmpint (len, ==, strlen (testbuf) + 1);
178 source = g_socket_create_source (client, G_IO_IN, NULL);
179 g_source_set_callback (source, (GSourceFunc)test_ip_async_read_ready,
181 g_source_attach (source, NULL);
182 g_source_unref (source);
188 test_ip_async_timed_out (GSocket *client,
192 IPTestData *data = user_data;
193 GError *error = NULL;
198 if (data->family == G_SOCKET_FAMILY_IPV4)
200 g_assert_cmpint (cond, ==, G_IO_IN);
201 len = g_socket_receive (client, buf, sizeof (buf), NULL, &error);
202 g_assert_cmpint (len, ==, -1);
203 g_assert_error (error, G_IO_ERROR, G_IO_ERROR_TIMED_OUT);
204 g_clear_error (&error);
207 source = g_socket_create_source (client, G_IO_OUT, NULL);
208 g_source_set_callback (source, (GSourceFunc)test_ip_async_write_ready,
210 g_source_attach (source, NULL);
211 g_source_unref (source);
217 test_ip_async_connected (GSocket *client,
221 IPTestData *data = user_data;
222 GError *error = NULL;
227 g_socket_check_connect_result (client, &error);
228 g_assert_no_error (error);
229 /* We do this after the check_connect_result, since that will give a
230 * more useful assertion in case of error.
232 g_assert_cmpint (cond, ==, G_IO_OUT);
234 g_assert (g_socket_is_connected (client));
236 /* This adds 1 second to "make check", so let's just only do it once. */
237 if (data->family == G_SOCKET_FAMILY_IPV4)
239 len = g_socket_receive (client, buf, sizeof (buf), NULL, &error);
240 g_assert_cmpint (len, ==, -1);
241 g_assert_error (error, G_IO_ERROR, G_IO_ERROR_WOULD_BLOCK);
242 g_clear_error (&error);
244 source = g_socket_create_source (client, G_IO_IN, NULL);
245 g_source_set_callback (source, (GSourceFunc)test_ip_async_timed_out,
247 g_source_attach (source, NULL);
248 g_source_unref (source);
251 test_ip_async_timed_out (client, 0, data);
257 idle_test_ip_async_connected (gpointer user_data)
259 IPTestData *data = user_data;
261 return test_ip_async_connected (data->client, G_IO_OUT, data);
265 test_ip_async (GSocketFamily family)
268 GError *error = NULL;
270 GSocketAddress *addr;
275 data = create_server (family, echo_server_thread, FALSE);
276 addr = g_socket_get_local_address (data->server, &error);
278 client = g_socket_new (family,
279 G_SOCKET_TYPE_STREAM,
280 G_SOCKET_PROTOCOL_DEFAULT,
282 g_assert_no_error (error);
283 data->client = client;
285 g_assert_cmpint (g_socket_get_family (client), ==, family);
286 g_assert_cmpint (g_socket_get_socket_type (client), ==, G_SOCKET_TYPE_STREAM);
287 g_assert_cmpint (g_socket_get_protocol (client), ==, G_SOCKET_PROTOCOL_DEFAULT);
289 g_socket_set_blocking (client, FALSE);
290 g_socket_set_timeout (client, 1);
292 if (g_socket_connect (client, addr, NULL, &error))
294 g_assert_no_error (error);
295 g_idle_add (idle_test_ip_async_connected, data);
299 g_assert_error (error, G_IO_ERROR, G_IO_ERROR_PENDING);
300 g_clear_error (&error);
301 source = g_socket_create_source (client, G_IO_OUT, NULL);
302 g_source_set_callback (source, (GSourceFunc)test_ip_async_connected,
304 g_source_attach (source, NULL);
305 g_source_unref (source);
307 g_object_unref (addr);
309 data->loop = g_main_loop_new (NULL, TRUE);
310 g_main_loop_run (data->loop);
311 g_main_loop_unref (data->loop);
313 g_socket_shutdown (client, FALSE, TRUE, &error);
314 g_assert_no_error (error);
316 g_thread_join (data->thread);
318 len = g_socket_receive (client, buf, sizeof (buf), NULL, &error);
319 g_assert_no_error (error);
320 g_assert_cmpint (len, ==, 0);
322 g_socket_close (client, &error);
323 g_assert_no_error (error);
324 g_socket_close (data->server, &error);
325 g_assert_no_error (error);
327 g_object_unref (data->server);
328 g_object_unref (client);
330 g_slice_free (IPTestData, data);
334 test_ipv4_async (void)
336 test_ip_async (G_SOCKET_FAMILY_IPV4);
340 test_ipv6_async (void)
342 test_ip_async (G_SOCKET_FAMILY_IPV6);
346 test_ip_sync (GSocketFamily family)
349 GError *error = NULL;
351 GSocketAddress *addr;
355 data = create_server (family, echo_server_thread, FALSE);
356 addr = g_socket_get_local_address (data->server, &error);
358 client = g_socket_new (family,
359 G_SOCKET_TYPE_STREAM,
360 G_SOCKET_PROTOCOL_DEFAULT,
362 g_assert_no_error (error);
364 g_assert_cmpint (g_socket_get_family (client), ==, family);
365 g_assert_cmpint (g_socket_get_socket_type (client), ==, G_SOCKET_TYPE_STREAM);
366 g_assert_cmpint (g_socket_get_protocol (client), ==, G_SOCKET_PROTOCOL_DEFAULT);
368 g_socket_set_blocking (client, TRUE);
369 g_socket_set_timeout (client, 1);
371 g_socket_connect (client, addr, NULL, &error);
372 g_assert_no_error (error);
373 g_assert (g_socket_is_connected (client));
374 g_object_unref (addr);
376 /* This adds 1 second to "make check", so let's just only do it once. */
377 if (family == G_SOCKET_FAMILY_IPV4)
379 len = g_socket_receive (client, buf, sizeof (buf), NULL, &error);
380 g_assert_cmpint (len, ==, -1);
381 g_assert_error (error, G_IO_ERROR, G_IO_ERROR_TIMED_OUT);
382 g_clear_error (&error);
385 len = g_socket_send (client, testbuf, strlen (testbuf) + 1, NULL, &error);
386 g_assert_no_error (error);
387 g_assert_cmpint (len, ==, strlen (testbuf) + 1);
389 len = g_socket_receive (client, buf, sizeof (buf), NULL, &error);
390 g_assert_no_error (error);
391 g_assert_cmpint (len, ==, strlen (testbuf) + 1);
393 g_assert_cmpstr (testbuf, ==, buf);
395 g_socket_shutdown (client, FALSE, TRUE, &error);
396 g_assert_no_error (error);
398 g_thread_join (data->thread);
400 len = g_socket_receive (client, buf, sizeof (buf), NULL, &error);
401 g_assert_no_error (error);
402 g_assert_cmpint (len, ==, 0);
404 g_socket_close (client, &error);
405 g_assert_no_error (error);
406 g_socket_close (data->server, &error);
407 g_assert_no_error (error);
409 g_object_unref (data->server);
410 g_object_unref (client);
412 g_slice_free (IPTestData, data);
416 test_ipv4_sync (void)
418 test_ip_sync (G_SOCKET_FAMILY_IPV4);
422 test_ipv6_sync (void)
424 test_ip_sync (G_SOCKET_FAMILY_IPV6);
428 graceful_server_thread (gpointer user_data)
430 IPTestData *data = user_data;
432 GError *error = NULL;
435 sock = g_socket_accept (data->server, NULL, &error);
436 g_assert_no_error (error);
438 len = g_socket_send (sock, testbuf, strlen (testbuf) + 1, NULL, &error);
439 g_assert_no_error (error);
440 g_assert_cmpint (len, ==, strlen (testbuf) + 1);
446 test_close_graceful (void)
448 GSocketFamily family = G_SOCKET_FAMILY_IPV4;
450 GError *error = NULL;
451 GSocket *client, *server;
452 GSocketAddress *addr;
456 data = create_server (family, graceful_server_thread, FALSE);
457 addr = g_socket_get_local_address (data->server, &error);
459 client = g_socket_new (family,
460 G_SOCKET_TYPE_STREAM,
461 G_SOCKET_PROTOCOL_DEFAULT,
463 g_assert_no_error (error);
465 g_assert_cmpint (g_socket_get_family (client), ==, family);
466 g_assert_cmpint (g_socket_get_socket_type (client), ==, G_SOCKET_TYPE_STREAM);
467 g_assert_cmpint (g_socket_get_protocol (client), ==, G_SOCKET_PROTOCOL_DEFAULT);
469 g_socket_set_blocking (client, TRUE);
470 g_socket_set_timeout (client, 1);
472 g_socket_connect (client, addr, NULL, &error);
473 g_assert_no_error (error);
474 g_assert (g_socket_is_connected (client));
475 g_object_unref (addr);
477 server = g_thread_join (data->thread);
479 /* similar to g_tcp_connection_set_graceful_disconnect(), but explicit */
480 g_socket_shutdown (server, FALSE, TRUE, &error);
481 g_assert_no_error (error);
483 /* we must timeout */
484 g_socket_condition_wait (client, G_IO_HUP, NULL, &error);
485 g_assert_error (error, G_IO_ERROR, G_IO_ERROR_TIMED_OUT);
486 g_clear_error (&error);
488 /* check that the remaining data is received */
489 len = g_socket_receive (client, buf, strlen (testbuf) + 1, NULL, &error);
490 g_assert_no_error (error);
491 g_assert_cmpint (len, ==, strlen (testbuf) + 1);
493 /* and only then the connection is closed */
494 len = g_socket_receive (client, buf, sizeof (buf), NULL, &error);
495 g_assert_no_error (error);
496 g_assert_cmpint (len, ==, 0);
498 g_socket_close (server, &error);
499 g_assert_no_error (error);
501 g_socket_close (client, &error);
502 g_assert_no_error (error);
504 g_object_unref (server);
505 g_object_unref (data->server);
506 g_object_unref (client);
508 g_slice_free (IPTestData, data);
511 #if defined (IPPROTO_IPV6) && defined (IPV6_V6ONLY)
513 v4mapped_server_thread (gpointer user_data)
515 IPTestData *data = user_data;
517 GError *error = NULL;
518 GSocketAddress *addr;
520 sock = g_socket_accept (data->server, NULL, &error);
521 g_assert_no_error (error);
523 g_assert_cmpint (g_socket_get_family (sock), ==, G_SOCKET_FAMILY_IPV6);
525 addr = g_socket_get_local_address (sock, &error);
526 g_assert_no_error (error);
527 g_assert_cmpint (g_socket_address_get_family (addr), ==, G_SOCKET_FAMILY_IPV4);
528 g_object_unref (addr);
530 addr = g_socket_get_remote_address (sock, &error);
531 g_assert_no_error (error);
532 g_assert_cmpint (g_socket_address_get_family (addr), ==, G_SOCKET_FAMILY_IPV4);
533 g_object_unref (addr);
535 g_socket_close (sock, &error);
536 g_assert_no_error (error);
537 g_object_unref (sock);
542 test_ipv6_v4mapped (void)
545 GError *error = NULL;
547 GSocketAddress *addr, *v4addr;
550 data = create_server (G_SOCKET_FAMILY_IPV6, v4mapped_server_thread, TRUE);
552 client = g_socket_new (G_SOCKET_FAMILY_IPV4,
553 G_SOCKET_TYPE_STREAM,
554 G_SOCKET_PROTOCOL_DEFAULT,
556 g_assert_no_error (error);
558 g_socket_set_blocking (client, TRUE);
559 g_socket_set_timeout (client, 1);
561 addr = g_socket_get_local_address (data->server, &error);
562 iaddr = g_inet_address_new_loopback (G_SOCKET_FAMILY_IPV4);
563 v4addr = g_inet_socket_address_new (iaddr, g_inet_socket_address_get_port (G_INET_SOCKET_ADDRESS (addr)));
564 g_object_unref (iaddr);
565 g_object_unref (addr);
567 g_socket_connect (client, v4addr, NULL, &error);
568 g_assert_no_error (error);
569 g_assert (g_socket_is_connected (client));
571 g_thread_join (data->thread);
573 g_socket_close (client, &error);
574 g_assert_no_error (error);
575 g_socket_close (data->server, &error);
576 g_assert_no_error (error);
578 g_object_unref (data->server);
579 g_object_unref (client);
580 g_object_unref (v4addr);
582 g_slice_free (IPTestData, data);
587 test_timed_wait (void)
590 GError *error = NULL;
592 GSocketAddress *addr;
596 data = create_server (G_SOCKET_FAMILY_IPV4, echo_server_thread, FALSE);
597 addr = g_socket_get_local_address (data->server, &error);
599 client = g_socket_new (G_SOCKET_FAMILY_IPV4,
600 G_SOCKET_TYPE_STREAM,
601 G_SOCKET_PROTOCOL_DEFAULT,
603 g_assert_no_error (error);
605 g_socket_set_blocking (client, TRUE);
606 g_socket_set_timeout (client, 1);
608 g_socket_connect (client, addr, NULL, &error);
609 g_assert_no_error (error);
610 g_object_unref (addr);
612 start_time = g_get_monotonic_time ();
613 g_socket_condition_timed_wait (client, G_IO_IN, 100000 /* 100 ms */,
615 g_assert_error (error, G_IO_ERROR, G_IO_ERROR_TIMED_OUT);
616 g_clear_error (&error);
617 poll_duration = g_get_monotonic_time () - start_time;
619 g_assert_cmpint (poll_duration, >=, 98000);
620 g_assert_cmpint (poll_duration, <, 112000);
622 g_socket_close (client, &error);
623 g_assert_no_error (error);
625 g_thread_join (data->thread);
627 g_socket_close (data->server, &error);
628 g_assert_no_error (error);
630 g_object_unref (data->server);
631 g_object_unref (client);
633 g_slice_free (IPTestData, data);
639 struct sockaddr_in6 sin6, gsin6;
640 GSocketAddress *saddr;
641 GInetSocketAddress *isaddr;
643 GError *error = NULL;
645 memset (&sin6, 0, sizeof (sin6));
646 sin6.sin6_family = AF_INET6;
647 sin6.sin6_addr = in6addr_loopback;
648 sin6.sin6_port = g_htons (42);
649 sin6.sin6_scope_id = g_htonl (17);
650 sin6.sin6_flowinfo = g_htonl (1729);
652 saddr = g_socket_address_new_from_native (&sin6, sizeof (sin6));
653 g_assert (G_IS_INET_SOCKET_ADDRESS (saddr));
655 isaddr = G_INET_SOCKET_ADDRESS (saddr);
656 iaddr = g_inet_socket_address_get_address (isaddr);
657 g_assert_cmpint (g_inet_address_get_family (iaddr), ==, G_SOCKET_FAMILY_IPV6);
658 g_assert (g_inet_address_get_is_loopback (iaddr));
660 g_assert_cmpint (g_inet_socket_address_get_port (isaddr), ==, 42);
661 g_assert_cmpint (g_inet_socket_address_get_scope_id (isaddr), ==, 17);
662 g_assert_cmpint (g_inet_socket_address_get_flowinfo (isaddr), ==, 1729);
664 g_socket_address_to_native (saddr, &gsin6, sizeof (gsin6), &error);
665 g_assert_no_error (error);
667 g_assert (memcmp (&sin6.sin6_addr, &gsin6.sin6_addr, sizeof (struct in6_addr)) == 0);
668 g_assert_cmpint (sin6.sin6_port, ==, gsin6.sin6_port);
669 g_assert_cmpint (sin6.sin6_scope_id, ==, gsin6.sin6_scope_id);
670 g_assert_cmpint (sin6.sin6_flowinfo, ==, gsin6.sin6_flowinfo);
672 g_object_unref (saddr);
677 test_unix_from_fd (void)
683 fd = socket (AF_UNIX, SOCK_STREAM, 0);
684 g_assert_cmpint (fd, !=, -1);
687 s = g_socket_new_from_fd (fd, &error);
688 g_assert_no_error (error);
689 g_assert_cmpint (g_socket_get_family (s), ==, G_SOCKET_FAMILY_UNIX);
690 g_assert_cmpint (g_socket_get_socket_type (s), ==, G_SOCKET_TYPE_STREAM);
691 g_assert_cmpint (g_socket_get_protocol (s), ==, G_SOCKET_PROTOCOL_DEFAULT);
696 test_unix_connection (void)
701 GSocketConnection *c;
703 fd = socket (AF_UNIX, SOCK_STREAM, 0);
704 g_assert_cmpint (fd, !=, -1);
707 s = g_socket_new_from_fd (fd, &error);
708 g_assert_no_error (error);
709 c = g_socket_connection_factory_create_connection (s);
710 g_assert (G_IS_UNIX_CONNECTION (c));
715 static GSocketConnection *
716 create_connection_for_fd (int fd)
720 GSocketConnection *connection;
722 socket = g_socket_new_from_fd (fd, &err);
723 g_assert_no_error (err);
724 g_assert (G_IS_SOCKET (socket));
725 connection = g_socket_connection_factory_create_connection (socket);
726 g_assert (G_IS_UNIX_CONNECTION (connection));
727 g_object_unref (socket);
731 #define TEST_DATA "failure to say failure to say 'i love gnome-panel!'."
734 test_unix_connection_ancillary_data (void)
738 gint status, fd, len;
743 g_assert_cmpint (status, ==, 0);
745 status = socketpair (PF_UNIX, SOCK_STREAM, 0, sv);
746 g_assert_cmpint (status, ==, 0);
749 g_assert_cmpint (pid, >=, 0);
751 /* Child: close its copy of the write end of the pipe, receive it
752 * again from the parent over the socket, and write some text to it.
754 * Parent: send the write end of the pipe (still open for the
755 * parent) over the socket, close it, and read some text from the
756 * read end of the pipe.
760 GSocketConnection *connection;
763 connection = create_connection_for_fd (sv[0]);
765 status = close (pv[1]);
766 g_assert_cmpint (status, ==, 0);
769 fd = g_unix_connection_receive_fd (G_UNIX_CONNECTION (connection), NULL,
771 g_assert_no_error (err);
772 g_assert_cmpint (fd, >, -1);
773 g_object_unref (connection);
776 len = write (fd, TEST_DATA, sizeof (TEST_DATA));
777 while (len == -1 && errno == EINTR);
778 g_assert_cmpint (len, ==, sizeof (TEST_DATA));
783 GSocketConnection *connection;
786 connection = create_connection_for_fd (sv[1]);
789 g_unix_connection_send_fd (G_UNIX_CONNECTION (connection), pv[1], NULL,
791 g_assert_no_error (err);
792 g_object_unref (connection);
794 status = close (pv[1]);
795 g_assert_cmpint (status, ==, 0);
797 memset (buffer, 0xff, sizeof buffer);
799 len = read (pv[0], buffer, sizeof buffer);
800 while (len == -1 && errno == EINTR);
802 g_assert_cmpint (len, ==, sizeof (TEST_DATA));
803 g_assert_cmpstr (buffer, ==, TEST_DATA);
805 waitpid (pid, &status, 0);
806 g_assert (WIFEXITED (status));
807 g_assert_cmpint (WEXITSTATUS (status), ==, 0);
810 /* TODO: add test for g_unix_connection_send_credentials() and
811 * g_unix_connection_receive_credentials().
814 #endif /* G_OS_UNIX */
820 g_test_init (&argc, &argv, NULL);
822 g_test_add_func ("/socket/ipv4_sync", test_ipv4_sync);
823 g_test_add_func ("/socket/ipv4_async", test_ipv4_async);
824 g_test_add_func ("/socket/ipv6_sync", test_ipv6_sync);
825 g_test_add_func ("/socket/ipv6_async", test_ipv6_async);
826 #if defined (IPPROTO_IPV6) && defined (IPV6_V6ONLY)
827 g_test_add_func ("/socket/ipv6_v4mapped", test_ipv6_v4mapped);
829 g_test_add_func ("/socket/close_graceful", test_close_graceful);
830 g_test_add_func ("/socket/timed_wait", test_timed_wait);
831 g_test_add_func ("/socket/address", test_sockaddr);
833 g_test_add_func ("/socket/unix-from-fd", test_unix_from_fd);
834 g_test_add_func ("/socket/unix-connection", test_unix_connection);
835 g_test_add_func ("/socket/unix-connection-ancillary-data", test_unix_connection_ancillary_data);