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 "helpers.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 "utils.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 int register_events(void)
  33 {
  34     xcb_generic_error_t *err;
  35     uint32_t values[] = {XCB_EVENT_MASK_SUBSTRUCTURE_REDIRECT | XCB_EVENT_MASK_SUBSTRUCTURE_NOTIFY | XCB_EVENT_MASK_PROPERTY_CHANGE | XCB_EVENT_MASK_BUTTON_PRESS};
  36     err = xcb_request_check(dpy, xcb_change_window_attributes_checked(dpy, screen->root, XCB_CW_EVENT_MASK, values));
  37     if (err != NULL)
  38         return 1;
  39     return 0;
  40 }
  41
  42 void handle_zombie(int sig)
  43 {
  44     while (waitpid(-1, NULL, WNOHANG) > 0)
  45         ;
  46     signal(sig, handle_zombie);
  47 }
  48
  49 void setup(int default_screen)
  50 {
  51     signal(SIGCHLD, handle_zombie);
  52     ewmh_init();
  53     screen = screen_of_display(dpy, default_screen);
  54     if (!screen)
  55         die("error: cannot aquire screen\n");
  56
  57     screen_width = screen->width_in_pixels;
  58     screen_height = screen->height_in_pixels;
  59
  60     xcb_atom_t net_atoms[] = {ewmh->_NET_SUPPORTED, ewmh->_NET_WM_STATE_FULLSCREEN, ewmh->_NET_WM_STATE, ewmh->_NET_ACTIVE_WINDOW};
  61
  62     xcb_ewmh_set_supported(ewmh, default_screen, LENGTH(net_atoms), net_atoms);
  63
  64     xcb_intern_atom_reply_t *reply;
  65     reply = xcb_intern_atom_reply(dpy, xcb_intern_atom(dpy, 0, strlen("WM_DELETE_WINDOW"), "WM_DELETE_WINDOW"), NULL);
  66     if (reply) {
  67         WM_DELETE_WINDOW = reply->atom;
  68         free(reply);
  69     }
  70
  71     desk = make_desktop(DESK_NAME);
  72     last_desk = NULL;
  73     desk_head = desk;
  74     desk_tail = desk;
  75     num_desktops++;
  76     /* ewmh_update_number_of_desktops(); */
  77     /* ewmh_update_desktop_names(); */
  78
  79     rule_head = make_rule();
  80
  81     split_mode = MODE_AUTOMATIC;
  82 }
  83
  84 int main(void)
  85 {
  86     fd_set descriptors;
  87     int sock_fd, ret_fd, dpy_fd, sel, nbr;
  88     struct sockaddr_un sock_address;
  89     char *sock_path;
  90     char msg[BUFSIZ], rsp[BUFSIZ];
  91
  92     xcb_generic_event_t *event;
  93
  94     running = true;
  95
  96     dpy = xcb_connect(NULL, &default_screen);
  97
  98     if (xcb_connection_has_error(dpy))
  99         die("error: cannot open display\n");
 100
 101     setup(default_screen);
 102
 103     if (register_events() == 1) {
 104         xcb_disconnect(dpy);
 105         die("another WM is already running\n");
 106     }
 107
 108     dpy_fd = xcb_get_file_descriptor(dpy);
 109
 110     sock_path = getenv(SOCK_PATH);
 111
 112     if (sock_path == NULL)
 113         die("BSPWM_SOCKET environment variable is not set\n");
 114
 115     sock_address.sun_family = AF_UNIX;
 116     strcpy(sock_address.sun_path, sock_path);
 117     unlink(sock_path);
 118
 119     sock_fd = socket(AF_UNIX, SOCK_STREAM, 0);
 120
 121     if (sock_fd == -1)
 122         die("error: could not create socket\n");
 123
 124     bind(sock_fd, (struct sockaddr *) &sock_address, sizeof(sock_address));
 125     listen(sock_fd, SOMAXCONN);
 126
 127     sel = MAX(sock_fd, dpy_fd) + 1;
 128
 129     load_settings();
 130     run_autostart();
 131     ewmh_update_wm_name();
 132     update_root_dimensions();
 133
 134     while (running) {
 135
 136         xcb_flush(dpy);
 137
 138         FD_ZERO(&descriptors);
 139         FD_SET(sock_fd, &descriptors);
 140         FD_SET(dpy_fd, &descriptors);
 141
 142         if (select(sel, &descriptors, NULL, NULL, NULL)) {
 143
 144             if (FD_ISSET(dpy_fd, &descriptors)) {
 145                 while ((event = xcb_poll_for_event(dpy)) != NULL) {
 146                     PUTS("got one X event\n");
 147                     handle_event(event);
 148                     free(event);
 149                 }
 150                 if (xcb_connection_has_error(dpy)) {
 151                     die("connection has errors\n");
 152                 }
 153             }
 154
 155             if (FD_ISSET(sock_fd, &descriptors)) {
 156                 ret_fd = accept(sock_fd, NULL, 0);
 157                 if (ret_fd > 0 && (nbr = recv(ret_fd, msg, sizeof(msg), 0)) > 0) {
 158                     msg[nbr] = '\0';
 159                     strcpy(rsp, EMPTY_RESPONSE);
 160                     process_message(msg, rsp);
 161                     send(ret_fd, rsp, strlen(rsp), 0);
 162                     close(ret_fd);
 163                 }
 164             }
 165         }
 166     }
 167
 168     close(sock_fd);
 169     xcb_ewmh_connection_wipe(ewmh);
 170     free(ewmh);
 171     xcb_disconnect(dpy);
 172     return 0;
 173 }