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
 20static int (*xerrorxlib)(Display *, XErrorEvent *);
 21static Bool otherwm;
 22
 23static void
 24cleanup()
 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
 42static void
 43scan()
 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
 64static int
 65win_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 */
 87static int
 88xerrorstart(Display *dsply, XErrorEvent *ee)
 89{
 90	otherwm = True;
 91	return -1;
 92}
 93
 94/* extern */
 95
 96char stext[1024];
 97Bool *seltag;
 98int screen, sx, sy, sw, sh, bx, by, bw, bh, mw;
 99unsigned int ntags, numlockmask;
100Atom wmatom[WMLast], netatom[NetLast];
101Bool running = True;
102Bool issel = True;
103Client *clients = NULL;
104Client *sel = NULL;
105Cursor cursor[CurLast];
106Display *dpy;
107DC dc = {0};
108Window root, barwin;
109
110int
111getproto(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
131void
132sendevent(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
146void
147quit(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 */
157int
158xerror(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
173int
174main(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}