tests/closure: Fix invalid unref
[platform/upstream/glib.git] / gobject / tests / closure.c
1 #include <glib-object.h>
2
3 #ifdef G_OS_UNIX
4 #include <glib-unix.h>
5
6 #include <fcntl.h>
7 #include <signal.h>
8 #include <unistd.h>
9 #endif
10
11 static void
12 test_source (GSource *one, GCallback quit_callback)
13 {
14   GClosure *closure;
15   GMainLoop *loop;
16
17   /* Callback with GMainLoop user_data */
18   loop = g_main_loop_new (NULL, FALSE);
19
20   closure = g_cclosure_new (quit_callback, loop, NULL);
21   g_source_set_closure (one, closure);
22
23   g_source_attach (one, NULL);
24   g_main_loop_run (loop);
25
26   g_source_destroy (one);
27   g_main_loop_unref (loop);
28 }
29
30 static gboolean
31 simple_quit_callback (gpointer user_data)
32 {
33   GMainLoop *loop = user_data;
34
35   g_main_loop_quit (loop);
36
37   return TRUE;
38 }
39
40 static void
41 test_closure_idle (void)
42 {
43   GSource *source;
44
45   source = g_idle_source_new ();
46   test_source (source, G_CALLBACK (simple_quit_callback));
47   g_source_unref (source);
48 }
49
50 static void
51 test_closure_timeout (void)
52 {
53   GSource *source;
54
55   source = g_timeout_source_new (10);
56   test_source (source, G_CALLBACK (simple_quit_callback));
57   g_source_unref (source);
58 }
59
60 static gboolean
61 iochannel_quit_callback (GIOChannel   *channel,
62                          GIOCondition  cond,
63                          gpointer      user_data)
64 {
65   GMainLoop *loop = user_data;
66
67   g_main_loop_quit (loop);
68
69   return TRUE;
70 }
71
72 static void
73 test_closure_iochannel (void)
74 {
75   GIOChannel *chan;
76   GSource *source;
77   char *path;
78   GError *error = NULL;
79
80   if (g_path_is_absolute (g_get_prgname ()))
81     path = g_strdup (g_get_prgname ());
82   else
83     {
84       path = g_test_build_filename (G_TEST_BUILT,
85                                     g_get_prgname (),
86                                     NULL);
87     }
88   chan = g_io_channel_new_file (path, "r", &error);
89   g_assert_no_error (error);
90   g_free (path);
91
92   source = g_io_create_watch (chan, G_IO_IN);
93   test_source (source, G_CALLBACK (iochannel_quit_callback));
94   g_source_unref (source);
95
96   g_io_channel_unref (chan);
97 }
98
99 static void
100 test_closure_child (void)
101 {
102   GSource *source;
103   GPid pid;
104   GError *error = NULL;
105   gchar *argv[3];
106
107   g_assert (getenv ("DO_NOT_ACCIDENTALLY_RECURSE") == NULL);
108   g_setenv ("DO_NOT_ACCIDENTALLY_RECURSE", "1", TRUE);
109
110   if (g_path_is_absolute (g_get_prgname ()))
111     argv[0] = g_strdup (g_get_prgname ());
112   else
113     {
114       argv[0] = g_test_build_filename (G_TEST_BUILT,
115                                        g_get_prgname (),
116                                        NULL);
117     }
118   argv[1] = "-l";
119   argv[2] = NULL;
120
121   g_spawn_async (NULL, argv, NULL,
122                  G_SPAWN_STDOUT_TO_DEV_NULL |
123                  G_SPAWN_STDERR_TO_DEV_NULL |
124                  G_SPAWN_DO_NOT_REAP_CHILD,
125                  NULL, NULL,
126                  &pid, &error);
127   g_assert_no_error (error);
128
129   g_free (argv[0]);
130
131   source = g_child_watch_source_new (pid);
132   test_source (source, G_CALLBACK (iochannel_quit_callback));
133   g_source_unref (source);
134 }
135
136 #ifdef G_OS_UNIX
137 static gboolean
138 fd_quit_callback (gint         fd,
139                   GIOCondition condition,
140                   gpointer     user_data)
141 {
142   GMainLoop *loop = user_data;
143
144   g_main_loop_quit (loop);
145
146   return TRUE;
147 }
148
149 static void
150 test_closure_fd (void)
151 {
152   gint fd;
153   GSource *source;
154
155   fd = open ("/dev/null", O_RDONLY);
156   g_assert (fd != -1);
157
158   source = g_unix_fd_source_new (fd, G_IO_IN);
159   test_source (source, G_CALLBACK (fd_quit_callback));
160   g_source_unref (source);
161
162   close (fd);
163 }
164
165 static gboolean
166 send_usr1 (gpointer user_data)
167 {
168   kill (0, SIGUSR1);
169   return FALSE;
170 }
171
172 static void
173 test_closure_signal (void)
174 {
175   GSource *source;
176
177   g_idle_add_full (G_PRIORITY_LOW, send_usr1, NULL, NULL);
178
179   source = g_unix_signal_source_new (SIGUSR1);
180   test_source (source, G_CALLBACK (fd_quit_callback));
181   g_source_unref (source);
182 }
183 #endif
184
185 int
186 main (int argc,
187       char *argv[])
188 {
189   g_test_init (&argc, &argv, NULL);
190
191   g_test_add_func ("/closure/idle", test_closure_idle);
192   g_test_add_func ("/closure/timeout", test_closure_timeout);
193   g_test_add_func ("/closure/iochannel", test_closure_iochannel);
194   g_test_add_func ("/closure/child", test_closure_child);
195 #ifdef G_OS_UNIX
196   g_test_add_func ("/closure/fd", test_closure_fd);
197   g_test_add_func ("/closure/signal", test_closure_signal);
198 #endif
199
200   return g_test_run ();
201 }