dwm-meillo

annotate main.c @ 616:f617c75a8405

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