tests: Add a systray program.

CMakeLists.txt

@@ -402,6 +402,11 @@ add_custom_target(check DEPENDS check-integration)
 add_executable(test-gravity tests/test-gravity.c)
 target_link_libraries(test-gravity
     ${AWESOME_COMMON_REQUIRED_LDFLAGS} ${AWESOME_REQUIRED_LDFLAGS})
+
+add_executable(test-systray tests/test-systray.c)
+target_link_libraries(test-systray
+    ${AWESOME_COMMON_REQUIRED_LDFLAGS} ${AWESOME_REQUIRED_LDFLAGS})
+
 if(DO_COVERAGE)
     set(TESTS_RUN_ENV DO_COVERAGE=1)
 endif()

tests/test-systray.c

@@ -0,0 +1,118 @@
+/*
+ * A simple client icon that "does nothing".
+ *
+ * Copyright © 2021 Uli Schlachter <psychon@znc.in>
+ *
+ * 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 General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+
+#include <stdarg.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <xcb/xcb.h>
+#include <xcb/xcb_atom.h>
+#include <xcb/xcb_aux.h>
+
+#include "common/atoms.h"
+#include "common/util.h"
+
+// Include all the C code we might need directly into this file - I am lazy
+#include "common/atoms.c"
+#include "common/util.c"
+
+#define SYSTEM_TRAY_REQUEST_DOCK 0
+
+static xcb_atom_t systray_atom(xcb_connection_t *conn, int screen) {
+    char *atom_name;
+    xcb_intern_atom_reply_t *reply;
+    xcb_atom_t atom;
+
+    atom_name = xcb_atom_name_by_screen("_NET_SYSTEM_TRAY", screen);
+    if(!atom_name)
+        fatal("Error getting systray atom name\n");
+
+    reply = xcb_intern_atom_reply(conn, xcb_intern_atom(conn, 0, strlen(atom_name), atom_name), NULL);
+    free(atom_name);
+    if (!reply)
+        fatal("Error interning systray icon\n");
+
+    atom = reply->atom;
+    free(reply);
+    return atom;
+}
+
+static xcb_window_t find_systray(xcb_connection_t *conn, xcb_atom_t net_system_tray) {
+    xcb_get_selection_owner_reply_t* reply = xcb_get_selection_owner_reply(conn, xcb_get_selection_owner(conn, net_system_tray), NULL);
+    if (!reply)
+        fatal("Failed to request selection owner\n");
+    xcb_window_t owner = reply->owner;
+    if (owner == XCB_NONE)
+        fatal("No systray running\n");
+    free(reply);
+    return owner;
+}
+
+static uint32_t get_color(xcb_connection_t *conn, xcb_screen_t *screen, uint16_t red, uint16_t green, uint16_t blue) {
+    xcb_alloc_color_reply_t *reply = xcb_alloc_color_reply(conn, xcb_alloc_color(conn, screen->default_colormap, red, green, blue), NULL);
+    if (!reply)
+        fatal("Error allocating color");
+    uint32_t pixel = reply->pixel;
+    free(reply);
+    return pixel;
+}
+
+int main() {
+    int default_screen;
+    xcb_connection_t* conn = xcb_connect(NULL, &default_screen);
+    if (xcb_connection_has_error(conn)) {
+        fatal("Could not connect to X11 server: %d\n", xcb_connection_has_error(conn));
+    }
+    atoms_init(conn);
+    xcb_screen_t* screen = xcb_aux_get_screen(conn, default_screen);
+
+    // Create a window for the systray icon
+    xcb_window_t window = xcb_generate_id(conn);
+    xcb_create_window(conn, screen->root_depth, window, screen->root, 0, 0, 10, 10, 0, XCB_WINDOW_CLASS_INPUT_OUTPUT,
+            screen->root_visual, XCB_CW_BACK_PIXEL, (uint32_t[]) { get_color(conn, screen, 0xffff, 0x9999, 0x0000) });
+    xcb_change_property(conn, XCB_PROP_MODE_REPLACE, window, _XEMBED_INFO, _XEMBED_INFO, 32, 2, (uint32_t[]) { 0, 1 });
+
+    // Make our window a systray icon
+    xcb_window_t systray_owner = find_systray(conn, systray_atom(conn, default_screen));
+    xcb_client_message_event_t ev;
+
+    p_clear(&ev, 1);
+    ev.response_type = XCB_CLIENT_MESSAGE;
+    ev.window = systray_owner;
+    ev.format = 32;
+    ev.data.data32[0] = XCB_CURRENT_TIME;
+    ev.data.data32[1] = SYSTEM_TRAY_REQUEST_DOCK;
+    ev.data.data32[2] = window;
+    ev.data.data32[3] = 0;
+    ev.data.data32[4] = 0;
+    ev.type = _NET_SYSTEM_TRAY_OPCODE;
+    xcb_send_event(conn, false, systray_owner, XCB_EVENT_MASK_NO_EVENT, (char *) &ev);
+
+    xcb_flush(conn);
+    xcb_generic_event_t *event;
+    while ((event = xcb_wait_for_event(conn)) != NULL) {
+        free(event);
+    }
+
+    xcb_disconnect(conn);
+    return 0;
+}
+
+// vim: filetype=c:expandtab:shiftwidth=4:tabstop=8:softtabstop=4:textwidth=80