git.sur5r.net Git - i3/i3/blob - src/startup.c

   1 /*
   2  * vim:ts=4:sw=4:expandtab
   3  *
   4  * i3 - an improved dynamic tiling window manager
   5  * © 2009-2011 Michael Stapelberg and contributors (see also: LICENSE)
   6  *
   7  * startup.c: Startup notification code. Ensures a startup notification context
   8  *            is setup when launching applications. We store the current
   9  *            workspace to open windows in that startup notification context on
  10  *            the appropriate workspace.
  11  *
  12  */
  13 #include "all.h"
  14
  15 #include <sys/types.h>
  16 #include <sys/wait.h>
  17
  18 #define SN_API_NOT_YET_FROZEN 1
  19 #include <libsn/sn-launcher.h>
  20
  21 static TAILQ_HEAD(startup_sequence_head, Startup_Sequence) startup_sequences =
  22     TAILQ_HEAD_INITIALIZER(startup_sequences);
  23
  24 /*
  25  * After 60 seconds, a timeout will be triggered for each startup sequence.
  26  *
  27  * The timeout will just trigger completion of the sequence, so the normal
  28  * completion process takes place (startup_monitor_event will free it).
  29  *
  30  */
  31 static void startup_timeout(EV_P_ ev_timer *w, int revents) {
  32     const char *id = sn_launcher_context_get_startup_id(w->data);
  33     DLOG("Timeout for startup sequence %s\n", id);
  34
  35     struct Startup_Sequence *current, *sequence = NULL;
  36     TAILQ_FOREACH(current, &startup_sequences, sequences) {
  37         if (strcmp(current->id, id) != 0)
  38             continue;
  39
  40         sequence = current;
  41         break;
  42     }
  43
  44     /* Unref the context (for the timeout itself, see start_application) */
  45     sn_launcher_context_unref(w->data);
  46
  47     if (!sequence) {
  48         DLOG("Sequence already deleted, nevermind.\n");
  49         return;
  50     }
  51
  52     /* Complete the startup sequence, will trigger its deletion. */
  53     sn_launcher_context_complete(w->data);
  54     free(w);
  55 }
  56
  57 /*
  58  * Starts the given application by passing it through a shell. We use double fork
  59  * to avoid zombie processes. As the started application’s parent exits (immediately),
  60  * the application is reparented to init (process-id 1), which correctly handles
  61  * childs, so we don’t have to do it :-).
  62  *
  63  * The shell is determined by looking for the SHELL environment variable. If it
  64  * does not exist, /bin/sh is used.
  65  *
  66  */
  67 void start_application(const char *command) {
  68     /* Create a startup notification context to monitor the progress of this
  69      * startup. */
  70     SnLauncherContext *context;
  71     context = sn_launcher_context_new(sndisplay, conn_screen);
  72     sn_launcher_context_set_name(context, "i3");
  73     sn_launcher_context_set_description(context, "exec command in i3");
  74     /* Chop off everything starting from the first space (if there are any
  75      * spaces in the command), since we don’t want the parameters. */
  76     char *first_word = sstrdup(command);
  77     char *space = strchr(first_word, ' ');
  78     if (space)
  79         *space = '\0';
  80     sn_launcher_context_initiate(context, "i3", first_word, last_timestamp);
  81     free(first_word);
  82
  83     /* Trigger a timeout after 60 seconds */
  84     struct ev_timer *timeout = scalloc(sizeof(struct ev_timer));
  85     ev_timer_init(timeout, startup_timeout, 60.0, 0.);
  86     timeout->data = context;
  87     ev_timer_start(main_loop, timeout);
  88
  89     LOG("startup id = %s\n", sn_launcher_context_get_startup_id(context));
  90
  91     /* Save the ID and current workspace in our internal list of startup
  92      * sequences */
  93     Con *ws = con_get_workspace(focused);
  94     struct Startup_Sequence *sequence = scalloc(sizeof(struct Startup_Sequence));
  95     sequence->id = sstrdup(sn_launcher_context_get_startup_id(context));
  96     sequence->workspace = sstrdup(ws->name);
  97     sequence->context = context;
  98     TAILQ_INSERT_TAIL(&startup_sequences, sequence, sequences);
  99
 100     /* Increase the refcount once (it starts with 1, so it will be 2 now) for
 101      * the timeout. Even if the sequence gets completed, the timeout still
 102      * needs the context (but will unref it then) */
 103     sn_launcher_context_ref(context);
 104
 105     LOG("executing: %s\n", command);
 106     if (fork() == 0) {
 107         /* Child process */
 108         setsid();
 109         if (fork() == 0) {
 110             /* Setup the environment variable(s) */
 111             sn_launcher_context_setup_child_process(context);
 112
 113             /* Stores the path of the shell */
 114             static const char *shell = NULL;
 115
 116             if (shell == NULL)
 117                 if ((shell = getenv("SHELL")) == NULL)
 118                     shell = "/bin/sh";
 119
 120             /* This is the child */
 121             execl(shell, shell, "-c", command, (void*)NULL);
 122             /* not reached */
 123         }
 124         _exit(0);
 125     }
 126     wait(0);
 127
 128     /* Change the pointer of the root window to indicate progress */
 129     if (xcursor_supported)
 130         xcursor_set_root_cursor(XCURSOR_CURSOR_WATCH);
 131     else xcb_set_root_cursor(XCURSOR_CURSOR_WATCH);
 132 }
 133
 134 /*
 135  * Called by libstartup-notification when something happens
 136  *
 137  */
 138 void startup_monitor_event(SnMonitorEvent *event, void *userdata) {
 139     SnStartupSequence *snsequence;
 140
 141     snsequence = sn_monitor_event_get_startup_sequence(event);
 142
 143     /* Get the corresponding internal startup sequence */
 144     const char *id = sn_startup_sequence_get_id(snsequence);
 145     struct Startup_Sequence *current, *sequence = NULL;
 146     TAILQ_FOREACH(current, &startup_sequences, sequences) {
 147         if (strcmp(current->id, id) != 0)
 148             continue;
 149
 150         sequence = current;
 151         break;
 152     }
 153
 154     if (!sequence) {
 155         DLOG("Got event for startup sequence that we did not initiate (ID = %s). Ignoring.\n", id);
 156         return;
 157     }
 158
 159     switch (sn_monitor_event_get_type(event)) {
 160         case SN_MONITOR_EVENT_COMPLETED:
 161             DLOG("startup sequence %s completed\n", sn_startup_sequence_get_id(snsequence));
 162
 163             /* Unref the context, will be free()d */
 164             sn_launcher_context_unref(sequence->context);
 165
 166             /* Delete our internal sequence */
 167             TAILQ_REMOVE(&startup_sequences, sequence, sequences);
 168
 169             if (TAILQ_EMPTY(&startup_sequences)) {
 170                 DLOG("No more startup sequences running, changing root window cursor to default pointer.\n");
 171                 /* Change the pointer of the root window to indicate progress */
 172                 if (xcursor_supported)
 173                     xcursor_set_root_cursor(XCURSOR_CURSOR_POINTER);
 174                 else xcb_set_root_cursor(XCURSOR_CURSOR_POINTER);
 175             }
 176             break;
 177         default:
 178             /* ignore */
 179             break;
 180     }
 181 }
 182
 183 /*
 184  * Checks if the given window belongs to a startup notification by checking if
 185  * the _NET_STARTUP_ID property is set on the window (or on its leader, if it’s
 186  * unset).
 187  *
 188  * If so, returns the workspace on which the startup was initiated.
 189  * Returns NULL otherwise.
 190  *
 191  */
 192 char *startup_workspace_for_window(i3Window *cwindow, xcb_get_property_reply_t *startup_id_reply) {
 193     /* The _NET_STARTUP_ID is only needed during this function, so we get it
 194      * here and don’t save it in the 'cwindow'. */
 195     if (startup_id_reply == NULL || xcb_get_property_value_length(startup_id_reply) == 0) {
 196         FREE(startup_id_reply);
 197         DLOG("No _NET_STARTUP_ID set on this window\n");
 198         if (cwindow->leader == XCB_NONE)
 199             return NULL;
 200
 201         xcb_get_property_cookie_t cookie;
 202         cookie = xcb_get_property(conn, false, cwindow->leader, A__NET_STARTUP_ID, XCB_GET_PROPERTY_TYPE_ANY, 0, 512);
 203         DLOG("Checking leader window 0x%08x\n", cwindow->leader);
 204         startup_id_reply = xcb_get_property_reply(conn, cookie, NULL);
 205
 206         if (startup_id_reply == NULL || xcb_get_property_value_length(startup_id_reply) == 0) {
 207             DLOG("No _NET_STARTUP_ID set on the leader either\n");
 208             FREE(startup_id_reply);
 209             return NULL;
 210         }
 211     }
 212
 213     char *startup_id;
 214     if (asprintf(&startup_id, "%.*s", xcb_get_property_value_length(startup_id_reply),
 215                  (char*)xcb_get_property_value(startup_id_reply)) == -1) {
 216         perror("asprintf()");
 217         DLOG("Could not get _NET_STARTUP_ID\n");
 218         free(startup_id_reply);
 219         return NULL;
 220     }
 221
 222     struct Startup_Sequence *current, *sequence = NULL;
 223     TAILQ_FOREACH(current, &startup_sequences, sequences) {
 224         if (strcmp(current->id, startup_id) != 0)
 225             continue;
 226
 227         sequence = current;
 228         break;
 229     }
 230
 231     free(startup_id);
 232     free(startup_id_reply);
 233
 234     if (!sequence) {
 235         DLOG("WARNING: This sequence (ID %s) was not found\n", startup_id);
 236         return NULL;
 237     }
 238
 239     return sequence->workspace;
 240 }