]> git.sur5r.net Git - i3/i3/blob - libi3/root_atom_contents.c
Merge pull request #3452 from orestisf1993/title_align
[i3/i3] / 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 }