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 "dwm.h"
   7 #include <errno.h>
   8 #include <stdio.h>
   9 #include <stdlib.h>
  10 #include <string.h>
  11 #include <unistd.h>
  12 #include <sys/select.h>
  13 #include <X11/cursorfont.h>
  14 #include <X11/keysym.h>
  15 #include <X11/Xatom.h>
  16 #include <X11/Xproto.h>
  17
  18 /* extern */
  19
  20 char stext[1024];
  21 Bool *seltag;
  22 int bx, by, bw, bh, bmw, mw, screen, sx, sy, sw, sh;
  23 unsigned int ntags, numlockmask;
  24 Atom wmatom[WMLast], netatom[NetLast];
  25 Bool running = True;
  26 Bool issel = True;
  27 Bool maximized = False;
  28 Client *clients = NULL;
  29 Client *sel = NULL;
  30 Cursor cursor[CurLast];
  31 Display *dpy;
  32 DC dc = {0};
  33 Window root, barwin;
  34
  35 /* static */
  36
  37 static int (*xerrorxlib)(Display *, XErrorEvent *);
  38 static Bool otherwm, readin;
  39
  40 static void
  41 cleanup()
  42 {
  43         close(STDIN_FILENO);
  44         while(sel) {
  45                 resize(sel, True, TopLeft);
  46                 unmanage(sel);
  47         }
  48         if(dc.font.set)
  49                 XFreeFontSet(dpy, dc.font.set);
  50         else
  51                 XFreeFont(dpy, dc.font.xfont);
  52         XUngrabKey(dpy, AnyKey, AnyModifier, root);
  53         XFreePixmap(dpy, dc.drawable);
  54         XFreeGC(dpy, dc.gc);
  55         XDestroyWindow(dpy, barwin);
  56         XSetInputFocus(dpy, PointerRoot, RevertToPointerRoot, CurrentTime);
  57         XSync(dpy, False);
  58 }
  59
  60 static void
  61 scan()
  62 {
  63         unsigned int i, num;
  64         Window *wins, d1, d2;
  65         XWindowAttributes wa;
  66
  67         wins = NULL;
  68         if(XQueryTree(dpy, root, &d1, &d2, &wins, &num)) {
  69                 for(i = 0; i < num; i++) {
  70                         if(!XGetWindowAttributes(dpy, wins[i], &wa))
  71                                 continue;
  72                         if(wa.override_redirect || XGetTransientForHint(dpy, wins[i], &d1))
  73                                 continue;
  74                         if(wa.map_state == IsViewable)
  75                                 manage(wins[i], &wa);
  76                 }
  77         }
  78         if(wins)
  79                 XFree(wins);
  80 }
  81
  82 static void
  83 setup()
  84 {
  85         int i, j;
  86         unsigned int mask;
  87         Window w;
  88         XModifierKeymap *modmap;
  89         XSetWindowAttributes wa;
  90
  91         /* init atoms */
  92         wmatom[WMProtocols] = XInternAtom(dpy, "WM_PROTOCOLS", False);
  93         wmatom[WMDelete] = XInternAtom(dpy, "WM_DELETE_WINDOW", False);
  94         netatom[NetSupported] = XInternAtom(dpy, "_NET_SUPPORTED", False);
  95         netatom[NetWMName] = XInternAtom(dpy, "_NET_WM_NAME", False);
  96         XChangeProperty(dpy, root, netatom[NetSupported], XA_ATOM, 32,
  97                         PropModeReplace, (unsigned char *) netatom, NetLast);
  98
  99         /* init cursors */
 100         cursor[CurNormal] = XCreateFontCursor(dpy, XC_left_ptr);
 101         cursor[CurResize] = XCreateFontCursor(dpy, XC_sizing);
 102         cursor[CurMove] = XCreateFontCursor(dpy, XC_fleur);
 103
 104         modmap = XGetModifierMapping(dpy);
 105         for (i = 0; i < 8; i++) {
 106                 for (j = 0; j < modmap->max_keypermod; j++) {
 107                         if(modmap->modifiermap[i * modmap->max_keypermod + j] == XKeysymToKeycode(dpy, XK_Num_Lock))
 108                                 numlockmask = (1 << i);
 109                 }
 110         }
 111         XFree(modmap);
 112
 113         wa.event_mask = SubstructureRedirectMask | SubstructureNotifyMask | EnterWindowMask | LeaveWindowMask;
 114         wa.cursor = cursor[CurNormal];
 115         XChangeWindowAttributes(dpy, root, CWEventMask | CWCursor, &wa);
 116
 117         grabkeys();
 118         initrregs();
 119
 120         for(ntags = 0; tags[ntags]; ntags++);
 121         seltag = emallocz(sizeof(Bool) * ntags);
 122         seltag[0] = True;
 123
 124         /* style */
 125         dc.norm[ColBG] = getcolor(NORMBGCOLOR);
 126         dc.norm[ColFG] = getcolor(NORMFGCOLOR);
 127         dc.sel[ColBG] = getcolor(SELBGCOLOR);
 128         dc.sel[ColFG] = getcolor(SELFGCOLOR);
 129         dc.status[ColBG] = getcolor(STATUSBGCOLOR);
 130         dc.status[ColFG] = getcolor(STATUSFGCOLOR);
 131         setfont(FONT);
 132
 133         bmw = textw(FLOATSYMBOL) > textw(TILESYMBOL) ? textw(FLOATSYMBOL) : textw(TILESYMBOL);
 134         sx = sy = 0;
 135         sw = DisplayWidth(dpy, screen);
 136         sh = DisplayHeight(dpy, screen);
 137         mw = (sw * MASTERW) / 100;
 138
 139         bx = by = 0;
 140         bw = sw;
 141         dc.h = bh = dc.font.height + 2;
 142         wa.override_redirect = 1;
 143         wa.background_pixmap = ParentRelative;
 144         wa.event_mask = ButtonPressMask | ExposureMask;
 145         barwin = XCreateWindow(dpy, root, bx, by, bw, bh, 0, DefaultDepth(dpy, screen),
 146                         CopyFromParent, DefaultVisual(dpy, screen),
 147                         CWOverrideRedirect | CWBackPixmap | CWEventMask, &wa);
 148         XDefineCursor(dpy, barwin, cursor[CurNormal]);
 149         XMapRaised(dpy, barwin);
 150
 151         dc.drawable = XCreatePixmap(dpy, root, sw, bh, DefaultDepth(dpy, screen));
 152         dc.gc = XCreateGC(dpy, root, 0, 0);
 153         XSetLineAttributes(dpy, dc.gc, 1, LineSolid, CapButt, JoinMiter);
 154
 155         issel = XQueryPointer(dpy, root, &w, &w, &i, &i, &i, &i, &mask);
 156         strcpy(stext, "dwm-"VERSION);
 157 }
 158
 159 /*
 160  * Startup Error handler to check if another window manager
 161  * is already running.
 162  */
 163 static int
 164 xerrorstart(Display *dsply, XErrorEvent *ee)
 165 {
 166         otherwm = True;
 167         return -1;
 168 }
 169
 170 /* extern */
 171
 172 int
 173 getproto(Window w)
 174 {
 175         int i, format, protos, status;
 176         unsigned long extra, res;
 177         Atom *protocols, real;
 178
 179         protos = 0;
 180         status = XGetWindowProperty(dpy, w, wmatom[WMProtocols], 0L, 20L, False,
 181                         XA_ATOM, &real, &format, &res, &extra, (unsigned char **)&protocols);
 182         if(status != Success || protocols == 0)
 183                 return protos;
 184         for(i = 0; i < res; i++)
 185                 if(protocols[i] == wmatom[WMDelete])
 186                         protos |= PROTODELWIN;
 187         free(protocols);
 188         return protos;
 189 }
 190
 191 void
 192 sendevent(Window w, Atom a, long value)
 193 {
 194         XEvent e;
 195
 196         e.type = ClientMessage;
 197         e.xclient.window = w;
 198         e.xclient.message_type = a;
 199         e.xclient.format = 32;
 200         e.xclient.data.l[0] = value;
 201         e.xclient.data.l[1] = CurrentTime;
 202         XSendEvent(dpy, w, False, NoEventMask, &e);
 203         XSync(dpy, False);
 204 }
 205
 206 void
 207 quit(Arg *arg)
 208 {
 209         readin = running = False;
 210 }
 211
 212 /*
 213  * There's no way to check accesses to destroyed windows, thus those cases are
 214  * ignored (especially on UnmapNotify's).  Other types of errors call Xlibs
 215  * default error handler, which calls exit().
 216  */
 217 int
 218 xerror(Display *dpy, XErrorEvent *ee)
 219 {
 220         if(ee->error_code == BadWindow
 221         || (ee->request_code == X_SetInputFocus && ee->error_code == BadMatch)
 222         || (ee->request_code == X_PolyText8 && ee->error_code == BadDrawable)
 223         || (ee->request_code == X_PolyFillRectangle && ee->error_code == BadDrawable)
 224         || (ee->request_code == X_PolySegment && ee->error_code == BadDrawable)
 225         || (ee->request_code == X_ConfigureWindow && ee->error_code == BadMatch)
 226         || (ee->request_code == X_GrabKey && ee->error_code == BadAccess))
 227                 return 0;
 228         fprintf(stderr, "dwm: fatal error: request code=%d, error code=%d\n",
 229                 ee->request_code, ee->error_code);
 230         return xerrorxlib(dpy, ee); /* may call exit() */
 231 }
 232
 233 int
 234 main(int argc, char *argv[])
 235 {
 236         int r, xfd;
 237         fd_set rd;
 238
 239         if(argc == 2 && !strncmp("-v", argv[1], 3)) {
 240                 fputs("dwm-"VERSION", (C)opyright MMVI Anselm R. Garbe\n", stdout);
 241                 exit(EXIT_SUCCESS);
 242         }
 243         else if(argc != 1)
 244                 eprint("usage: dwm [-v]\n");
 245
 246         dpy = XOpenDisplay(0);
 247         if(!dpy)
 248                 eprint("dwm: cannot open display\n");
 249
 250         xfd = ConnectionNumber(dpy);
 251         screen = DefaultScreen(dpy);
 252         root = RootWindow(dpy, screen);
 253
 254         otherwm = False;
 255         XSetErrorHandler(xerrorstart);
 256         /* this causes an error if some other window manager is running */
 257         XSelectInput(dpy, root, SubstructureRedirectMask);
 258         XSync(dpy, False);
 259
 260         if(otherwm)
 261                 eprint("dwm: another window manager is already running\n");
 262
 263         XSync(dpy, False);
 264         XSetErrorHandler(NULL);
 265         xerrorxlib = XSetErrorHandler(xerror);
 266         XSync(dpy, False);
 267
 268         setup();
 269         drawstatus();
 270         scan();
 271
 272         /* main event loop, also reads status text from stdin */
 273         XSync(dpy, False);
 274         procevent();
 275         readin = True;
 276         while(running) {
 277                 FD_ZERO(&rd);
 278                 if(readin)
 279                         FD_SET(STDIN_FILENO, &rd);
 280                 FD_SET(xfd, &rd);
 281                 r = select(xfd + 1, &rd, NULL, NULL, NULL);
 282                 if((r == -1) && (errno == EINTR))
 283                         continue;
 284                 if(r > 0) {
 285                         if(readin && FD_ISSET(STDIN_FILENO, &rd)) {
 286                                 readin = NULL != fgets(stext, sizeof(stext), stdin);
 287                                 if(readin)
 288                                         stext[strlen(stext) - 1] = 0;
 289                                 else
 290                                         strcpy(stext, "broken pipe");
 291                                 drawstatus();
 292                         }
 293                 }
 294                 else if(r < 0)
 295                         eprint("select failed\n");
 296                 procevent();
 297         }
 298         cleanup();
 299         XCloseDisplay(dpy);
 300
 301         return 0;
 302 }