main.c

   1 /*
   2  * (C)opyright MMVI Anselm R. Garbe <garbeam at gmail dot com>
   3  * See LICENSE file for license details.
   4  */
   5
   6 #include <errno.h>
   7 #include <stdarg.h>
   8 #include <stdio.h>
   9 #include <stdlib.h>
  10 #include <string.h>
  11 #include <unistd.h>
  12
  13 #include <X11/cursorfont.h>
  14 #include <X11/Xatom.h>
  15 #include <X11/Xproto.h>
  16
  17 #include "dwm.h"
  18
  19 /********** CUSTOMIZE **********/
  20
  21 char *tags[TLast] = {
  22         [Tscratch] = "scratch",
  23         [Tdev] = "dev",
  24         [Twww] = "www",
  25         [Twork] = "work",
  26 };
  27
  28 /********** CUSTOMIZE **********/
  29
  30 /* X structs */
  31 Display *dpy;
  32 Window root, barwin;
  33 Atom wm_atom[WMLast], net_atom[NetLast];
  34 Cursor cursor[CurLast];
  35 Bool running = True;
  36 Bool issel;
  37
  38 int tsel = Tdev; /* default tag */
  39 int screen, sx, sy, sw, sh, bx, by, bw, bh, mw;
  40 char stext[1024];
  41
  42 DC dc = {0};
  43 Client *clients = NULL;
  44 Client *sel = NULL;
  45
  46 static Bool other_wm_running;
  47 static const char version[] =
  48         "dwm-" VERSION ", (C)opyright MMVI Anselm R. Garbe\n";
  49 static int (*x_xerror) (Display *, XErrorEvent *);
  50
  51 static void
  52 usage() {       error("usage: dwm [-v]\n"); }
  53
  54 static void
  55 scan_wins()
  56 {
  57         unsigned int i, num;
  58         Window *wins;
  59         XWindowAttributes wa;
  60         Window d1, d2;
  61
  62         if(XQueryTree(dpy, root, &d1, &d2, &wins, &num)) {
  63                 for(i = 0; i < num; i++) {
  64                         if(!XGetWindowAttributes(dpy, wins[i], &wa))
  65                                 continue;
  66                         if(wa.override_redirect || XGetTransientForHint(dpy, wins[i], &d1))
  67                                 continue;
  68                         if(wa.map_state == IsViewable)
  69                                 manage(wins[i], &wa);
  70                 }
  71         }
  72         if(wins)
  73                 XFree(wins);
  74 }
  75
  76 static int
  77 win_property(Window w, Atom a, Atom t, long l, unsigned char **prop)
  78 {
  79         Atom real;
  80         int format;
  81         unsigned long res, extra;
  82         int status;
  83
  84         status = XGetWindowProperty(dpy, w, a, 0L, l, False, t, &real, &format,
  85                         &res, &extra, prop);
  86
  87         if(status != Success || *prop == 0) {
  88                 return 0;
  89         }
  90         if(res == 0) {
  91                 free((void *) *prop);
  92         }
  93         return res;
  94 }
  95
  96 int
  97 proto(Window w)
  98 {
  99         unsigned char *protocols;
 100         long res;
 101         int protos = 0;
 102         int i;
 103
 104         res = win_property(w, wm_atom[WMProtocols], XA_ATOM, 20L, &protocols);
 105         if(res <= 0) {
 106                 return protos;
 107         }
 108         for(i = 0; i < res; i++) {
 109                 if(protocols[i] == wm_atom[WMDelete])
 110                         protos |= WM_PROTOCOL_DELWIN;
 111         }
 112         free((char *) protocols);
 113         return protos;
 114 }
 115
 116 void
 117 sendevent(Window w, Atom a, long value)
 118 {
 119         XEvent e;
 120
 121         e.type = ClientMessage;
 122         e.xclient.window = w;
 123         e.xclient.message_type = a;
 124         e.xclient.format = 32;
 125         e.xclient.data.l[0] = value;
 126         e.xclient.data.l[1] = CurrentTime;
 127         XSendEvent(dpy, w, False, NoEventMask, &e);
 128         XFlush(dpy);
 129 }
 130
 131 /*
 132  * There's no way to check accesses to destroyed windows, thus
 133  * those cases are ignored (especially on UnmapNotify's).
 134  * Other types of errors call Xlib's default error handler, which
 135  * calls exit().
 136  */
 137 int
 138 xerror(Display *dpy, XErrorEvent *error)
 139 {
 140         if(error->error_code == BadWindow
 141                         || (error->request_code == X_SetInputFocus
 142                                 && error->error_code == BadMatch)
 143                         || (error->request_code == X_PolyText8
 144                                 && error->error_code == BadDrawable)
 145                         || (error->request_code == X_PolyFillRectangle
 146                                 && error->error_code == BadDrawable)
 147                         || (error->request_code == X_PolySegment
 148                                 && error->error_code == BadDrawable)
 149                         || (error->request_code == X_ConfigureWindow
 150                                 && error->error_code == BadMatch)
 151                         || (error->request_code == X_GrabKey
 152                                 && error->error_code == BadAccess))
 153                 return 0;
 154         fprintf(stderr, "dwm: fatal error: request code=%d, error code=%d\n",
 155                         error->request_code, error->error_code);
 156         return x_xerror(dpy, error); /* may call exit() */
 157 }
 158
 159 /*
 160  * Startup Error handler to check if another window manager
 161  * is already running.
 162  */
 163 static int
 164 startup_xerror(Display *dpy, XErrorEvent *error)
 165 {
 166         other_wm_running = True;
 167         return -1;
 168 }
 169
 170 static void
 171 cleanup()
 172 {
 173         while(sel) {
 174                 resize(sel, True);
 175                 unmanage(sel);
 176         }
 177         XSetInputFocus(dpy, PointerRoot, RevertToPointerRoot, CurrentTime);
 178 }
 179
 180 void
 181 quit(Arg *arg)
 182 {
 183         running = False;
 184 }
 185
 186 int
 187 main(int argc, char *argv[])
 188 {
 189         int i, n;
 190         fd_set rd;
 191         XSetWindowAttributes wa;
 192         unsigned int mask;
 193         Bool readstdin = True;
 194         Window w;
 195         XEvent ev;
 196
 197         for(i = 1; (i < argc) && (argv[i][0] == '-'); i++) {
 198                 switch (argv[i][1]) {
 199                 case 'v':
 200                         fprintf(stdout, "%s", version);
 201                         exit(0);
 202                         break;
 203                 default:
 204                         usage();
 205                         break;
 206                 }
 207         }
 208
 209         dpy = XOpenDisplay(0);
 210         if(!dpy)
 211                 error("dwm: cannot connect X server\n");
 212
 213         screen = DefaultScreen(dpy);
 214         root = RootWindow(dpy, screen);
 215
 216         /* check if another WM is already running */
 217         other_wm_running = False;
 218         XSetErrorHandler(startup_xerror);
 219         /* this causes an error if some other WM is running */
 220         XSelectInput(dpy, root, SubstructureRedirectMask);
 221         XFlush(dpy);
 222
 223         if(other_wm_running)
 224                 error("dwm: another window manager is already running\n");
 225
 226         XSetErrorHandler(0);
 227         x_xerror = XSetErrorHandler(xerror);
 228
 229         /* init atoms */
 230         wm_atom[WMProtocols] = XInternAtom(dpy, "WM_PROTOCOLS", False);
 231         wm_atom[WMDelete] = XInternAtom(dpy, "WM_DELETE_WINDOW", False);
 232         net_atom[NetSupported] = XInternAtom(dpy, "_NET_SUPPORTED", False);
 233         net_atom[NetWMName] = XInternAtom(dpy, "_NET_WM_NAME", False);
 234         XChangeProperty(dpy, root, net_atom[NetSupported], XA_ATOM, 32,
 235                         PropModeReplace, (unsigned char *) net_atom, NetLast);
 236
 237         /* init cursors */
 238         cursor[CurNormal] = XCreateFontCursor(dpy, XC_left_ptr);
 239         cursor[CurResize] = XCreateFontCursor(dpy, XC_sizing);
 240         cursor[CurMove] = XCreateFontCursor(dpy, XC_fleur);
 241
 242         grabkeys();
 243
 244         /* style */
 245         dc.bg = getcolor(BGCOLOR);
 246         dc.fg = getcolor(FGCOLOR);
 247         dc.border = getcolor(BORDERCOLOR);
 248         setfont(FONT);
 249
 250         sx = sy = 0;
 251         sw = DisplayWidth(dpy, screen);
 252         sh = DisplayHeight(dpy, screen);
 253         mw = (sw * MASTERW) / 100;
 254
 255         wa.override_redirect = 1;
 256         wa.background_pixmap = ParentRelative;
 257         wa.event_mask = ButtonPressMask | ExposureMask;
 258
 259         bx = by = 0;
 260         bw = sw;
 261         dc.h = bh = dc.font.height + 4;
 262         barwin = XCreateWindow(dpy, root, bx, by, bw, bh, 0, DefaultDepth(dpy, screen),
 263                         CopyFromParent, DefaultVisual(dpy, screen),
 264                         CWOverrideRedirect | CWBackPixmap | CWEventMask, &wa);
 265         XDefineCursor(dpy, barwin, cursor[CurNormal]);
 266         XMapRaised(dpy, barwin);
 267
 268         dc.drawable = XCreatePixmap(dpy, root, sw, bh, DefaultDepth(dpy, screen));
 269         dc.gc = XCreateGC(dpy, root, 0, 0);
 270         drawstatus();
 271
 272         issel = XQueryPointer(dpy, root, &w, &w, &i, &i, &i, &i, &mask);
 273
 274         wa.event_mask = SubstructureRedirectMask | EnterWindowMask \
 275                                         | LeaveWindowMask;
 276         wa.cursor = cursor[CurNormal];
 277
 278         XChangeWindowAttributes(dpy, root, CWEventMask | CWCursor, &wa);
 279
 280         strcpy(stext, "dwm-"VERSION);
 281         scan_wins();
 282
 283         /* main event loop, reads status text from stdin as well */
 284 Mainloop:
 285         while(running) {
 286                 FD_ZERO(&rd);
 287                 if(readstdin)
 288                         FD_SET(STDIN_FILENO, &rd);
 289                 FD_SET(ConnectionNumber(dpy), &rd);
 290
 291                 i = select(ConnectionNumber(dpy) + 1, &rd, 0, 0, 0);
 292                 if(i == -1 && errno == EINTR)
 293                         continue;
 294                 if(i < 0)
 295                         error("select failed\n");
 296                 else if(i > 0) {
 297                         if(FD_ISSET(ConnectionNumber(dpy), &rd)) {
 298                                 while(XPending(dpy)) {
 299                                         XNextEvent(dpy, &ev);
 300                                         if(handler[ev.type])
 301                                                 (handler[ev.type])(&ev); /* call handler */
 302                                 }
 303                         }
 304                         if(readstdin && FD_ISSET(STDIN_FILENO, &rd)) {
 305                                 i = n = 0;
 306                                 for(;;) {
 307                                         if((i = getchar()) == EOF) {
 308                                                 /* broken pipe/end of producer */
 309                                                 readstdin = False;
 310                                                 strcpy(stext, "broken pipe");
 311                                                 goto Mainloop;
 312                                         }
 313                                         if(i == '\n' || n >= sizeof(stext) - 1)
 314                                                 break;
 315                                         stext[n++] = i;
 316                                 }
 317                                 stext[n] = 0;
 318                                 drawstatus();
 319                         }
 320                 }
 321         }
 322
 323         cleanup();
 324         XCloseDisplay(dpy);
 325
 326         return 0;
 327 }