git.sur5r.net Git - i3/i3/blob - libi3/root_atom_contents.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  */
   8 #include <stdio.h>
   9 #include <string.h>
  10 #include <stdbool.h>
  11 #include <limits.h>
  12 #include <stdlib.h>
  13
  14 #include <xcb/xcb.h>
  15 #include <xcb/xcb_aux.h>
  16
  17 #include "libi3.h"
  18
  19 /*
  20  * Try to get the contents of the given atom (for example I3_SOCKET_PATH) from
  21  * the X11 root window and return NULL if it doesn’t work.
  22  *
  23  * If the provided XCB connection is NULL, a new connection will be
  24  * established.
  25  *
  26  * The memory for the contents is dynamically allocated and has to be
  27  * free()d by the caller.
  28  *
  29  */
  30 char *root_atom_contents(const char *atomname, xcb_connection_t *provided_conn, int screen) {
  31     xcb_intern_atom_cookie_t atom_cookie;
  32     xcb_intern_atom_reply_t *atom_reply;
  33     char *content;
  34     xcb_connection_t *conn = provided_conn;
  35
  36     if (provided_conn == NULL &&
  37         ((conn = xcb_connect(NULL, &screen)) == NULL ||
  38          xcb_connection_has_error(conn)))
  39         return NULL;
  40
  41     atom_cookie = xcb_intern_atom(conn, 0, strlen(atomname), atomname);
  42
  43     xcb_screen_t *root_screen = xcb_aux_get_screen(conn, screen);
  44     xcb_window_t root = root_screen->root;
  45
  46     atom_reply = xcb_intern_atom_reply(conn, atom_cookie, NULL);
  47     if (atom_reply == NULL)
  48         return NULL;
  49
  50     xcb_get_property_cookie_t prop_cookie;
  51     xcb_get_property_reply_t *prop_reply;
  52     prop_cookie = xcb_get_property_unchecked(conn, false, root, atom_reply->atom,
  53                                              XCB_GET_PROPERTY_TYPE_ANY, 0, PATH_MAX);
  54     prop_reply = xcb_get_property_reply(conn, prop_cookie, NULL);
  55     if (prop_reply == NULL) {
  56         free(atom_reply);
  57         return NULL;
  58     }
  59     if (xcb_get_property_value_length(prop_reply) == 0) {
  60         free(atom_reply);
  61         free(prop_reply);
  62         return NULL;
  63     }
  64     if (prop_reply->type == XCB_ATOM_CARDINAL) {
  65         /* We treat a CARDINAL as a >= 32-bit unsigned int. The only CARDINAL
  66          * we query is I3_PID, which is 32-bit. */
  67         if (asprintf(&content, "%u", *((unsigned int*)xcb_get_property_value(prop_reply))) == -1) {
  68             free(atom_reply);
  69             free(prop_reply);
  70             return NULL;
  71         }
  72     } else {
  73         if (asprintf(&content, "%.*s", xcb_get_property_value_length(prop_reply),
  74                      (char*)xcb_get_property_value(prop_reply)) == -1) {
  75             free(atom_reply);
  76             free(prop_reply);
  77             return NULL;
  78         }
  79     }
  80     if (provided_conn == NULL)
  81         xcb_disconnect(conn);
  82     free(atom_reply);
  83     free(prop_reply);
  84     return content;
  85 }
  86