glib/gio/tests/unix-fd.c
Ryan Lortie 3d7edc137e Bug 601637 - add GUnixFDList
change GUnixFDMessage to contain a GUnixFDList.

add test case for GUnixFDMessage and GUnixFDList.

update docs.
2009-11-24 17:55:57 -05:00

198 lines
4.8 KiB
C

#include <gio/gio.h>
#include <gio/gunixfdmessage.h>
#include <sys/socket.h>
#include <string.h>
#include <unistd.h>
/* ensures that no FDs are left open at the end */
static void
check_fd_list (const gint *fd_list)
{
gint i;
for (i = 0; i < 40; i++)
{
int my_fd;
my_fd = dup (0);
g_assert_cmpint (fd_list[i], ==, my_fd);
}
for (i = 0; i < 40; i++)
close (fd_list[i]);
}
static void
create_fd_list (gint *fd_list)
{
gint i;
for (i = 0; i < 40; i++)
{
fd_list[i] = dup (0);
g_assert_cmpint (fd_list[i], >, 0);
}
for (i = 0; i < 40; i++)
close (fd_list[i]);
}
static void
test_fds (void)
{
GUnixFDMessage *message;
GUnixFDMessage **mv;
GUnixFDList *list, *l2;
GSocket *sockets[2];
const gint *peek;
char buffer[1024];
gint fd_list[40];
GOutputVector ov;
GInputVector iv;
gint *stolen;
gint sv[3];
gint flags;
gint nm;
gint s;
create_fd_list (fd_list);
s = socketpair (PF_UNIX, SOCK_STREAM, 0, sv);
g_assert_cmpint (s, ==, 0);
sv[2] = -1;
list = g_unix_fd_list_new_from_array (sv, -1);
message = G_UNIX_FD_MESSAGE (g_unix_fd_message_new_with_fd_list (list));
g_object_unref (list);
g_assert (g_unix_fd_message_get_fd_list (message) == list);
g_object_get (message, "fd-list", &l2, NULL);
g_assert (l2 == list);
g_assert_cmpint (g_unix_fd_list_get_length (list), ==, 2);
peek = g_unix_fd_list_peek_fds (list, &s);
g_assert_cmpint (s, ==, 2);
stolen = g_unix_fd_message_steal_fds (message, &s);
g_assert_cmpint (s, ==, 2);
g_assert (stolen == peek);
g_assert_cmpint (stolen[0], ==, sv[0]);
g_assert_cmpint (stolen[1], ==, sv[1]);
g_assert_cmpint (stolen[2], ==, sv[2]);
g_free (stolen);
g_unix_fd_message_append_fd (message, sv[0], NULL);
s = close (sv[0]);
g_assert_cmpint (s, ==, 0);
g_unix_fd_message_append_fd (message, sv[1], NULL);
s = close (sv[1]);
g_assert_cmpint (s, ==, 0);
s = close (g_unix_fd_list_get (list, 0, NULL));
g_assert_cmpint (s, ==, 0);
s = close (g_unix_fd_list_get (list, 1, NULL));
g_assert_cmpint (s, ==, 0);
s = close (g_unix_fd_list_get (list, 0, NULL));
g_assert_cmpint (s, ==, 0);
s = close (g_unix_fd_list_get (list, 1, NULL));
g_assert_cmpint (s, ==, 0);
s = close (g_unix_fd_list_get (list, 0, NULL));
g_assert_cmpint (s, ==, 0);
s = close (g_unix_fd_list_get (list, 1, NULL));
g_assert_cmpint (s, ==, 0);
g_object_unref (message);
g_object_unref (list);
message = G_UNIX_FD_MESSAGE (g_unix_fd_message_new ());
list = g_unix_fd_message_get_fd_list (message);
s = pipe (sv);
g_assert_cmpint (s, ==, 0);
s = g_unix_fd_list_append (list, sv[0], NULL);
g_assert_cmpint (s, >=, 0);
s = g_unix_fd_list_append (list, sv[1], NULL);
g_assert_cmpint (s, >=, 0);
s = close (sv[0]);
g_assert_cmpint (s, ==, 0);
s = close (sv[1]);
g_assert_cmpint (s, ==, 0);
s = close (g_unix_fd_list_get (list, 0, NULL));
g_assert_cmpint (s, ==, 0);
s = close (g_unix_fd_list_get (list, 1, NULL));
g_assert_cmpint (s, ==, 0);
s = socketpair (PF_UNIX, SOCK_STREAM, 0, sv);
g_assert_cmpint (s, ==, 0);
sockets[0] = g_socket_new_from_fd (sv[0], NULL);
g_assert (G_IS_SOCKET (sockets[0]));
sockets[1] = g_socket_new_from_fd (sv[1], NULL);
g_assert (G_IS_SOCKET (sockets[1]));
buffer[0] = 0xff;
ov.buffer = buffer;
ov.size = 1;
s = g_socket_send_message (sockets[0], NULL, &ov, 1,
(GSocketControlMessage **) &message,
1, 0, NULL, NULL);
g_assert_cmpint (s, ==, 1);
g_object_unref (message);
message = NULL;
flags = 0;
iv.buffer = buffer;
iv.size = 1;
s = g_socket_receive_message (sockets[1], NULL, &iv, 1,
(GSocketControlMessage ***) &mv,
&nm, &flags, NULL, NULL);
g_assert_cmpint (s, ==, 1);
g_object_unref (sockets[0]);
g_object_unref (sockets[1]);
g_assert_cmpint (nm, ==, 1);
message = mv[0];
g_free (mv);
g_assert (G_IS_UNIX_FD_MESSAGE (message));
list = g_object_ref (g_unix_fd_message_get_fd_list (message));
g_object_unref (message);
peek = g_unix_fd_list_peek_fds (list, &s);
g_assert_cmpint (s, ==, 2);
sv[0] = g_unix_fd_list_get (list, 1, NULL);
strcpy (buffer, "failure to say failure to say 'i love gnome-panel!'.");
s = write (sv[0], buffer, strlen (buffer) + 1);
g_assert_cmpint (s, ==, strlen (buffer) + 1);
close (sv[0]);
memset (buffer, 0xff, sizeof buffer);
s = read (peek[0], buffer, sizeof buffer);
g_assert_cmpint (s, ==, 53);
g_assert_cmpstr (buffer, ==,
"failure to say failure to say 'i love gnome-panel!'.");
g_object_unref (list);
check_fd_list (fd_list);
}
int
main (int argc, char **argv)
{
g_test_init (&argc, &argv, NULL);
g_type_init ();
g_test_add_func ("/unix-streams/file-descriptors", test_fds);
return g_test_run();
}