summaryrefslogtreecommitdiffstats
path: root/gnome-session/gsm-process-helper.c
blob: 01d581d4cc45b0e605c3db75c88574bf00052b1d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
/* -*- Mode: C; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 8 -*-
 *
 * Copyright (C) 2010 Novell, Inc.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License as
 * published by the Free Software Foundation; either version 2 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, see <http://www.gnu.org/licenses/>.
 */

#include <config.h>

#include <gio/gio.h>
#include <glib/gi18n.h>

#include "gsm-process-helper.h"

typedef struct {
        gboolean        done;
        GSubprocess    *process;
        gboolean        caught_error;
        GError        **error;
        GMainContext   *maincontext;
        GSource        *timeout_source;
} GsmProcessHelper;

static void
on_child_exited (GObject        *source,
                 GAsyncResult   *result,
                 gpointer        data)
{
        GsmProcessHelper *helper = data;

        helper->done = TRUE;

        if (!g_subprocess_wait_check_finish ((GSubprocess*)source, result,
                                             helper->caught_error ? NULL : helper->error))
                helper->caught_error = TRUE;

        g_clear_pointer (&helper->timeout_source, g_source_destroy);

        g_main_context_wakeup (helper->maincontext);
}

static gboolean
on_child_timeout (gpointer data)
{
        GsmProcessHelper *helper = data;
        
        g_assert (!helper->done);
        
        g_subprocess_force_exit (helper->process);
        
        g_set_error_literal (helper->error,
                             G_IO_CHANNEL_ERROR,
                             G_IO_CHANNEL_ERROR_FAILED,
                             "Timed out");

        helper->timeout_source = NULL;
        
        return FALSE;
}

gboolean
gsm_process_helper (const char   *command_line,
                    unsigned int  timeout,
                    GError      **error)
{
        gboolean ret = FALSE;
        GsmProcessHelper helper = { 0, };
        gchar **argv = NULL;
        GMainContext *subcontext = NULL;

        if (!g_shell_parse_argv (command_line, NULL, &argv, error))
                goto out;

        helper.error = error;

        subcontext = g_main_context_new ();
        g_main_context_push_thread_default (subcontext);

        helper.process = g_subprocess_newv ((const char*const*)argv, 0, error);
        if (!helper.process)
                goto out;

        g_subprocess_wait_async (helper.process, NULL, on_child_exited, &helper);

        helper.timeout_source = g_timeout_source_new (timeout);

        g_source_set_callback (helper.timeout_source, on_child_timeout, &helper, NULL);
        g_source_attach (helper.timeout_source, subcontext);

        while (!helper.done)
                g_main_context_iteration (subcontext, TRUE);

        ret = helper.caught_error;
 out:
        g_strfreev (argv);
        if (subcontext) {
                g_main_context_pop_thread_default (subcontext);
                g_main_context_unref (subcontext);
        }
        g_clear_object (&helper.process);
        return ret;
}