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 Michael Stapelberg and contributors (see also: LICENSE)
   6  *
   7  */
   8 #include "libi3.h"
   9
  10 #include <stdio.h>
  11 #include <string.h>
  12 #include <stdbool.h>
  13 #include <limits.h>
  14 #include <stdlib.h>
  15 #include <math.h>
  16
  17 #include <xcb/xcb.h>
  18 #include <xcb/xcb_aux.h>
  19
  20 /*
  21  * Try to get the contents of the given atom (for example I3_SOCKET_PATH) from
  22  * the X11 root window and return NULL if it doesn’t work.
  23  *
  24  * If the provided XCB connection is NULL, a new connection will be
  25  * established.
  26  *
  27  * The memory for the contents is dynamically allocated and has to be
  28  * free()d by the caller.
  29  *
  30  */
  31 char *root_atom_contents(const char *atomname, xcb_connection_t *provided_conn, int screen) {
  32     xcb_intern_atom_cookie_t atom_cookie;
  33     xcb_intern_atom_reply_t *atom_reply;
  34     char *content = NULL;
  35     size_t content_max_words = 256;
  36     xcb_connection_t *conn = provided_conn;
  37
  38     if (provided_conn == NULL &&
  39         ((conn = xcb_connect(NULL, &screen)) == NULL ||
  40          xcb_connection_has_error(conn))) {
  41         return NULL;
  42     }
  43
  44     atom_cookie = xcb_intern_atom(conn, 0, strlen(atomname), atomname);
  45
  46     xcb_screen_t *root_screen = xcb_aux_get_screen(conn, screen);
  47     xcb_window_t root = root_screen->root;
  48
  49     atom_reply = xcb_intern_atom_reply(conn, atom_cookie, NULL);
  50     if (atom_reply == NULL) {
  51         goto out_conn;
  52     }
  53
  54     xcb_get_property_cookie_t prop_cookie;
  55     xcb_get_property_reply_t *prop_reply;
  56     prop_cookie = xcb_get_property_unchecked(conn, false, root, atom_reply->atom,
  57                                              XCB_GET_PROPERTY_TYPE_ANY, 0, content_max_words);
  58     prop_reply = xcb_get_property_reply(conn, prop_cookie, NULL);
  59     if (prop_reply == NULL) {
  60         goto out_atom;
  61     }
  62     if (xcb_get_property_value_length(prop_reply) > 0 && prop_reply->bytes_after > 0) {
  63         /* We received an incomplete value. Ask again but with a properly
  64          * adjusted size. */
  65         content_max_words += ceil(prop_reply->bytes_after / 4.0);
  66         /* Repeat the request, with adjusted size */
  67         free(prop_reply);
  68         prop_cookie = xcb_get_property_unchecked(conn, false, root, atom_reply->atom,
  69                                                  XCB_GET_PROPERTY_TYPE_ANY, 0, content_max_words);
  70         prop_reply = xcb_get_property_reply(conn, prop_cookie, NULL);
  71         if (prop_reply == NULL) {
  72             goto out_atom;
  73         }
  74     }
  75     if (xcb_get_property_value_length(prop_reply) == 0) {
  76         goto out;
  77     }
  78     if (prop_reply->type == XCB_ATOM_CARDINAL) {
  79         /* We treat a CARDINAL as a >= 32-bit unsigned int. The only CARDINAL
  80          * we query is I3_PID, which is 32-bit. */
  81         sasprintf(&content, "%u", *((unsigned int *)xcb_get_property_value(prop_reply)));
  82     } else {
  83         sasprintf(&content, "%.*s", xcb_get_property_value_length(prop_reply),
  84                   (char *)xcb_get_property_value(prop_reply));
  85     }
  86
  87 out:
  88     free(prop_reply);
  89 out_atom:
  90     free(atom_reply);
  91 out_conn:
  92     if (provided_conn == NULL)
  93         xcb_disconnect(conn);
  94     return content;
  95 }