fix double free bug, simplify error handling, reduce X11 calls

This commit is contained in:
MvA 2022-09-11 15:04:39 +02:00 committed by Dan Cross
parent 3e764832bc
commit 846f724983

View file

@ -24,7 +24,6 @@ struct Win {
XDisplay *dpy;
XWindow root;
Atom net_active_window;
Reprog *exclude = nil;
Win *win;
@ -38,8 +37,7 @@ Font *font;
Image *lightblue;
enum
{
enum {
PAD = 3,
MARGIN = 5
};
@ -60,7 +58,7 @@ void*
erealloc(void *v, ulong n)
{
v = realloc(v, n);
if(v == nil)
if(v==nil)
sysfatal("out of memory reallocating");
return v;
}
@ -69,9 +67,9 @@ char*
estrdup(char *s)
{
s = strdup(s);
if(s == nil)
if(s==nil)
sysfatal("out of memory allocating");
return s;
return(s);
}
char*
@ -88,12 +86,12 @@ getproperty(XWindow w, Atom a)
s = XGetWindowProperty(dpy, w, a, 0, 100L, 0,
AnyPropertyType, &type, &fmt, &n, &dummy, &p);
if(s != 0){
if(s!=0){
XFree(p);
return nil;
return(nil);
}
return (char*)p;
return((char*)p);
}
XWindow
@ -109,27 +107,27 @@ findname(XWindow w)
p = getproperty(w, XA_WM_NAME);
if(p){
free(p);
return w;
return(w);
}
net_wm_name = XInternAtom(dpy, "_NET_WM_NAME", FALSE);
p = getproperty(w, net_wm_name);
if(p){
free(p);
return w;
return(w);
}
rwin = 0;
s = XQueryTree(dpy, w, &dw1, &dw2, &xwin, &nxwin);
if(s != 0){
if(s!=0){
for (i = 0; i < nxwin; i++){
w = findname(xwin[i]);
if(w != 0){
rwin = w;
break ;
}
w = findname(xwin[i]);
if(w != 0){
rwin = w;
break ;
}
}
XFree(xwin);
}
@ -144,6 +142,7 @@ wcmp(const void *w1, const void *w2)
}
/* unicode-aware case-insensitive strcmp, taken from golangs gc/subr.c */
int
_cistrcmp(char *p, char *q)
{
@ -189,49 +188,48 @@ refreshwin(void)
s = XQueryTree(dpy, root, &dw1, &dw2, &xwin, &nxwin);
if(s == 0){
if(xwin != NULL)
if(s==0){
if(xwin!=NULL)
XFree(xwin);
return;
}
qsort(xwin, nxwin, sizeof(xwin[0]), wcmp);
nw = 0;
for(i=0; i < nxwin; i++){
for(i=0; i<nxwin; i++){
memset(&attr, 0, sizeof attr);
xwin[i] = findname(xwin[i]);
if(xwin[i] == 0)
if(xwin[i]==0)
continue;
s = XGetWindowAttributes(dpy, xwin[i], &attr);
if(s == 0)
if(s==0)
continue;
if(attr.width <= 0 ||
if (attr.width <= 0 ||
attr.override_redirect ||
attr.map_state != IsViewable)
continue;
s = XGetClassHint(dpy, xwin[i], &class);
if(s == 0)
if(s==0)
continue;
if (exclude != nil && regexec(exclude, class.res_name, nil, 0)){
free(class.res_name);
free(class.res_class);
if (exclude!=nil && regexec(exclude, class.res_name, nil, 0)) {
free(class.res_name);
free(class.res_class);
continue;
}
net_wm_name = XInternAtom(dpy, "_NET_WM_NAME", FALSE);
wmname = getproperty(xwin[i], net_wm_name);
if(wmname == nil){
if(wmname==nil){
wmname = getproperty(xwin[i], XA_WM_NAME);
if(wmname == nil){
free(class.res_name);
free(class.res_class);
if(wmname==nil){
free(class.res_name);
free(class.res_class);
continue;
}
}
@ -240,24 +238,23 @@ refreshwin(void)
if(showwmnames==1)
label = wmname;
if(nw < nwin && win[nw].n == xwin[i] && strcmp(win[nw].label, label) == 0){
if(nw<nwin && win[nw].n==xwin[i] && strcmp(win[nw].label, label)==0) {
nw++;
free(wmname);
free(class.res_name);
free(class.res_class);
free(wmname);
free(class.res_name);
free(class.res_class);
continue;
}
if(nw < nwin){
free(win[nw].label);
if(nw<nwin){
free(win[nw].label);
win[nw].label = nil;
}
if(nw >= mwin){
if(nw>=mwin){
mwin += 8;
win = erealloc(win, mwin * sizeof(win[0]));
}
win[nw].n = xwin[i];
win[nw].label = estrdup(label);
win[nw].dirty = 1;
@ -270,11 +267,11 @@ refreshwin(void)
XFree(xwin);
while(nwin > nw)
while(nwin>nw)
free(win[--nwin].label);
nwin = nw;
if(sortlabels == 1)
if(sortlabels==1)
qsort(win, nwin, sizeof(struct Win), winlabelcmp);
}
@ -313,16 +310,16 @@ geometry(void)
rows = (Dy(screen->r) - 2 * MARGIN + PAD) / (font->height + PAD);
if(rows*cols<nwin || rows*cols>=nwin*2){
ncols = 1;
if(nwin > 0)
if(nwin>0)
ncols = (nwin + rows - 1) / rows;
if(ncols != cols){
if(ncols!=cols){
cols = ncols;
z = 1;
}
}
r = Rect(0, 0, (Dx(screen->r) - 2 * MARGIN + PAD) / cols - PAD, font->height);
for(i=0; i <nwin; i++)
for(i=0; i<nwin; i++)
win[i].r =
rectaddpt(
rectaddpt(r,
@ -341,11 +338,11 @@ redraw(Image *screen, int all)
all |= geometry();
if(all)
draw(screen, screen->r, lightblue, nil, ZP);
for(i=0; i < nwin; i++)
for(i=0; i<nwin; i++)
if(all || win[i].dirty)
drawwin(i);
if(!all)
for (; i < onwin; i++)
for (; i<onwin; i++)
drawnowin(i);
onwin = nwin;
}
@ -385,29 +382,29 @@ click(Mouse m)
{
int i, j;
if(m.buttons == 0 || (m.buttons & ~4))
if(m.buttons==0 || (m.buttons&~4))
return;
for(i=0; i < nwin; i++)
for(i=0; i<nwin; i++)
if(ptinrect(m.xy, win[i].r))
break;
if(i == nwin)
if(i==nwin)
return;
do
m = emouse();
while(m.buttons == 4);
while(m.buttons==4);
if(m.buttons != 0){
if(m.buttons!=0){
do
m = emouse();
while(m.buttons);
return;
}
for(j=0; j < nwin; j++)
for(j=0; j<nwin; j++)
if(ptinrect(m.xy, win[j].r))
break;
if(j == i)
if(j==i)
selectwin(win[i].n);
}
@ -439,7 +436,7 @@ main(int argc, char **argv)
break;
case 'e':
exclude = regcomp(EARGF(usage()));
if(exclude == nil)
if(exclude==nil)
sysfatal("Bad regexp");
break;
case 's':
@ -455,9 +452,11 @@ main(int argc, char **argv)
if(argc)
usage();
einit(Emouse | Ekeyboard);
Etimer = etimer(0, 1000);
dpy = XOpenDisplay("");
if(dpy == nil)
if(dpy==nil)
sysfatal("open display: %r");
root = DefaultRootWindow(dpy);
@ -465,24 +464,20 @@ main(int argc, char **argv)
initdraw(0, 0, "winwatch");
lightblue = allocimagemix(display, DPalebluegreen, DWhite);
if(lightblue == nil)
if(lightblue==nil)
sysfatal("allocimagemix: %r");
font = openfont(display, fontname);
if(font == nil)
if(font==nil)
sysfatal("font '%s' not found", fontname);
einit(Emouse | Ekeyboard);
Etimer = etimer(0, 1000);
XSetErrorHandler(winwatchxerrorhandler);
refreshwin();
redraw(screen, 1);
for(;;){
switch(eread(Emouse|Ekeyboard|Etimer, &e)){
case Ekeyboard:
if(e.kbdc==0x7F || e.kbdc == 'q')
if(e.kbdc==0x7F || e.kbdc=='q')
exits(0);
break;
case Emouse: