bspwm.c

   1 #include <stdlib.h>
   2 #include <stdio.h>
   3 #include <string.h>
   4 #include <unistd.h>
   5 #include <signal.h>
   6 #include <fcntl.h>
   7 #include <sys/socket.h>
   8 #include <sys/un.h>
   9 #include <sys/select.h>
  10 #include <sys/stat.h>
  11 #include <sys/wait.h>
  12 #include <xcb/xcb.h>
  13 #include <xcb/xcb_event.h>
  14 #include <xcb/xcb_ewmh.h>
  15 #include <xcb/xinerama.h>
  16 #include "types.h"
  17 #include "settings.h"
  18 #include "messages.h"
  19 #include "rules.h"
  20 #include "events.h"
  21 #include "common.h"
  22 #include "helpers.h"
  23 #include "bspwm.h"
  24 #include "tree.h"
  25 #include "ewmh.h"
  26
  27 void quit(void)
  28 {
  29     running = false;
  30 }
  31
  32 void register_events(void)
  33 {
  34     uint32_t values[] = {ROOT_EVENT_MASK};
  35     xcb_generic_error_t *e = xcb_request_check(dpy, xcb_change_window_attributes_checked(dpy, screen->root, XCB_CW_EVENT_MASK, values));
  36     if (e != NULL) {
  37         xcb_disconnect(dpy);
  38         err("another WM is already running\n");
  39     }
  40
  41     xcb_grab_button(dpy, false, screen->root, XCB_EVENT_MASK_BUTTON_PRESS, XCB_GRAB_MODE_ASYNC, XCB_GRAB_MODE_ASYNC, XCB_NONE, XCB_NONE, XCB_BUTTON_INDEX_1, BUTTON_MODIFIER);
  42     xcb_grab_button(dpy, false, screen->root, XCB_EVENT_MASK_BUTTON_PRESS, XCB_GRAB_MODE_ASYNC, XCB_GRAB_MODE_ASYNC, XCB_NONE, XCB_NONE, XCB_BUTTON_INDEX_2, BUTTON_MODIFIER);
  43     xcb_grab_button(dpy, false, screen->root, XCB_EVENT_MASK_BUTTON_PRESS, XCB_GRAB_MODE_ASYNC, XCB_GRAB_MODE_ASYNC, XCB_NONE, XCB_NONE, XCB_BUTTON_INDEX_3, BUTTON_MODIFIER);
  44 }
  45
  46 void setup(void)
  47 {
  48     ewmh_init();
  49     screen = xcb_setup_roots_iterator(xcb_get_setup(dpy)).data;
  50     if (!screen)
  51         err("error: cannot aquire screen\n");
  52
  53     screen_width = screen->width_in_pixels;
  54     screen_height = screen->height_in_pixels;
  55     root_depth = screen->root_depth;
  56
  57     xcb_atom_t net_atoms[] = {ewmh->_NET_SUPPORTED,
  58                               ewmh->_NET_DESKTOP_NAMES,
  59                               ewmh->_NET_NUMBER_OF_DESKTOPS,
  60                               ewmh->_NET_CURRENT_DESKTOP,
  61                               ewmh->_NET_CLIENT_LIST,
  62                               ewmh->_NET_ACTIVE_WINDOW,
  63                               ewmh->_NET_WM_DESKTOP,
  64                               ewmh->_NET_WM_STATE,
  65                               ewmh->_NET_WM_STATE_FULLSCREEN,
  66                               ewmh->_NET_WM_WINDOW_TYPE,
  67                               ewmh->_NET_WM_WINDOW_TYPE_DOCK,
  68                               ewmh->_NET_WM_WINDOW_TYPE_NOTIFICATION,
  69                               ewmh->_NET_WM_WINDOW_TYPE_DIALOG,
  70                               ewmh->_NET_WM_WINDOW_TYPE_UTILITY,
  71                               ewmh->_NET_WM_WINDOW_TYPE_TOOLBAR};
  72
  73     xcb_ewmh_set_supported(ewmh, default_screen, LENGTH(net_atoms), net_atoms);
  74
  75     monitor_uid = desktop_uid = 0;
  76     mon = last_mon = mon_head = mon_tail = NULL;
  77
  78     bool xinerama_is_active = false;
  79
  80     if (xcb_get_extension_data(dpy, &xcb_xinerama_id)->present) {
  81         xcb_xinerama_is_active_reply_t *xia = xcb_xinerama_is_active_reply(dpy, xcb_xinerama_is_active(dpy), NULL);
  82         if (xia != NULL) {
  83             xinerama_is_active = xia->state;
  84             free(xia);
  85         }
  86     }
  87
  88     if (xinerama_is_active) {
  89         xcb_xinerama_query_screens_reply_t *xsq = xcb_xinerama_query_screens_reply(dpy, xcb_xinerama_query_screens(dpy), NULL);
  90         xcb_xinerama_screen_info_t *xsi = xcb_xinerama_query_screens_screen_info(xsq);
  91         int n = xcb_xinerama_query_screens_screen_info_length(xsq);
  92         PRINTF("number of monitors: %d\n", n);
  93         for (int i = 0; i < n; i++) {
  94             xcb_xinerama_screen_info_t info = xsi[i];
  95             xcb_rectangle_t rect = (xcb_rectangle_t) {info.x_org, info.y_org, info.width, info.height};
  96             add_monitor(&rect);
  97         }
  98         free(xsq);
  99     } else {
 100         warn("Xinerama is inactive");
 101         xcb_rectangle_t rect = (xcb_rectangle_t) {0, 0, screen_width, screen_height};
 102         add_monitor(&rect);
 103     }
 104
 105     for (monitor_t *m = mon_head; m != NULL; m = m->next)
 106         add_desktop(m, NULL);
 107
 108     ewmh_update_number_of_desktops();
 109     ewmh_update_desktop_names();
 110     rule_head = make_rule();
 111     frozen_pointer = make_pointer_state();
 112     split_mode = MODE_AUTOMATIC;
 113 }
 114
 115 int main(void)
 116 {
 117     fd_set descriptors;
 118     char socket_path[MAXLEN];
 119     int sock_fd, ret_fd, dpy_fd, sel, n;
 120     struct sockaddr_un sock_address;
 121     size_t rsplen = 0;
 122     char msg[BUFSIZ] = {0};
 123     char rsp[BUFSIZ] = {0};
 124
 125     xcb_generic_event_t *event;
 126
 127     running = true;
 128
 129     dpy = xcb_connect(NULL, &default_screen);
 130
 131     if (xcb_connection_has_error(dpy))
 132         err("error: cannot open display\n");
 133
 134     setup();
 135     register_events();
 136
 137     dpy_fd = xcb_get_file_descriptor(dpy);
 138
 139     char *sp = getenv(SOCKET_ENV_VAR);
 140
 141     strncpy(socket_path, (sp == NULL ? DEFAULT_SOCKET_PATH : sp), sizeof(socket_path));
 142
 143     sock_address.sun_family = AF_UNIX;
 144     strncpy(sock_address.sun_path, socket_path, sizeof(sock_address.sun_path));
 145     unlink(socket_path);
 146
 147     sock_fd = socket(AF_UNIX, SOCK_STREAM, 0);
 148
 149     if (sock_fd == -1)
 150         err("error: could not create socket\n");
 151
 152     bind(sock_fd, (struct sockaddr *) &sock_address, sizeof(sock_address));
 153     listen(sock_fd, SOMAXCONN);
 154
 155     sel = MAX(sock_fd, dpy_fd) + 1;
 156
 157     load_settings();
 158     run_autostart();
 159     ewmh_update_wm_name();
 160
 161     while (running) {
 162
 163         xcb_flush(dpy);
 164
 165         FD_ZERO(&descriptors);
 166         FD_SET(sock_fd, &descriptors);
 167         FD_SET(dpy_fd, &descriptors);
 168
 169         if (select(sel, &descriptors, NULL, NULL, NULL)) {
 170
 171             if (FD_ISSET(sock_fd, &descriptors)) {
 172                 ret_fd = accept(sock_fd, NULL, 0);
 173                 if (ret_fd > 0 && (n = recv(ret_fd, msg, sizeof(msg), 0)) > 0) {
 174                     msg[n] = '\0';
 175                     process_message(msg, rsp);
 176                     rsplen = strlen(rsp);
 177                     if (rsp[rsplen - 1] == '\n')
 178                         rsp[--rsplen] = '\0';
 179                     send(ret_fd, rsp, rsplen, 0);
 180                     close(ret_fd);
 181                     rsp[0] = '\0';
 182                 }
 183             }
 184
 185             if (FD_ISSET(dpy_fd, &descriptors)) {
 186                 while ((event = xcb_poll_for_event(dpy)) != NULL) {
 187                     handle_event(event);
 188                     free(event);
 189                 }
 190             }
 191
 192         }
 193
 194         if (xcb_connection_has_error(dpy)) {
 195             err("connection has errors\n");
 196         }
 197     }
 198
 199     close(sock_fd);
 200     xcb_ewmh_connection_wipe(ewmh);
 201     free(ewmh);
 202     xcb_flush(dpy);
 203     xcb_disconnect(dpy);
 204     return 0;
 205 }