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