atexit cleanup

This commit is contained in:
Connor Lane Smith 2010-07-02 03:44:01 +01:00
parent 6ba5103df4
commit d9c24564d2
2 changed files with 36 additions and 46 deletions

View File

@ -26,7 +26,7 @@ static void drawinput(void);
static Bool grabkeyboard(void); static Bool grabkeyboard(void);
static void kpress(XKeyEvent *e); static void kpress(XKeyEvent *e);
static void run(void); static void run(void);
static void setup(Bool topbar); static void setup(void);
#include "config.h" #include "config.h"
@ -34,14 +34,13 @@ static void setup(Bool topbar);
static char *prompt = NULL; static char *prompt = NULL;
static char text[4096]; static char text[4096];
static int promptw = 0; static int promptw = 0;
static int ret = 0;
static int screen; static int screen;
static unsigned int cursor = 0; static unsigned int cursor = 0;
static unsigned int numlockmask = 0; static unsigned int numlockmask = 0;
static unsigned int mw, mh; static unsigned int mw, mh;
static unsigned long normcol[ColLast]; static unsigned long normcol[ColLast];
static unsigned long selcol[ColLast]; static unsigned long selcol[ColLast];
static Bool running = True; static Bool topbar = True;
static DC dc; static DC dc;
static Display *dpy; static Display *dpy;
static Window win, root; static Window win, root;
@ -51,6 +50,7 @@ cleanup(void) {
cleanupdraw(&dc); cleanupdraw(&dc);
XDestroyWindow(dpy, win); XDestroyWindow(dpy, win);
XUngrabKeyboard(dpy, CurrentTime); XUngrabKeyboard(dpy, CurrentTime);
XCloseDisplay(dpy);
} }
void void
@ -81,7 +81,6 @@ drawinput(void)
drawtext(&dc, *text ? text : NULL, normcol, False); drawtext(&dc, *text ? text : NULL, normcol, False);
drawcursor(); drawcursor();
XCopyArea(dpy, dc.drawable, win, dc.gc, 0, 0, mw, mh, 0, 0); XCopyArea(dpy, dc.drawable, win, dc.gc, 0, 0, mw, mh, 0, 0);
XFlush(dpy);
} }
Bool Bool
@ -200,9 +199,7 @@ kpress(XKeyEvent *e) {
cursor = len; cursor = len;
break; break;
case XK_Escape: case XK_Escape:
ret = 1; exit(EXIT_FAILURE);
running = False;
return;
case XK_Home: case XK_Home:
cursor = 0; cursor = 0;
break; break;
@ -214,8 +211,7 @@ kpress(XKeyEvent *e) {
case XK_Return: case XK_Return:
fprintf(stdout, "%s", text); fprintf(stdout, "%s", text);
fflush(stdout); fflush(stdout);
running = False; exit(EXIT_SUCCESS);
return;
case XK_Right: case XK_Right:
if(cursor == len) if(cursor == len)
return; return;
@ -230,7 +226,8 @@ run(void) {
XEvent ev; XEvent ev;
/* main event loop */ /* main event loop */
while(running && !XNextEvent(dpy, &ev)) XSync(dpy, False);
while(!XNextEvent(dpy, &ev))
switch(ev.type) { switch(ev.type) {
case KeyPress: case KeyPress:
kpress(&ev.xkey); kpress(&ev.xkey);
@ -240,14 +237,15 @@ run(void) {
drawinput(); drawinput();
break; break;
case VisibilityNotify: case VisibilityNotify:
if (ev.xvisibility.state != VisibilityUnobscured) if(ev.xvisibility.state != VisibilityUnobscured)
XRaiseWindow(dpy, win); XRaiseWindow(dpy, win);
break; break;
} }
exit(EXIT_FAILURE);
} }
void void
setup(Bool topbar) { setup(void) {
int i, j, x, y; int i, j, x, y;
#if XINERAMA #if XINERAMA
int n; int n;
@ -320,7 +318,6 @@ setup(Bool topbar) {
int int
main(int argc, char *argv[]) { main(int argc, char *argv[]) {
unsigned int i; unsigned int i;
Bool topbar = True;
/* command line args */ /* command line args */
progname = "dinput"; progname = "dinput";
@ -364,15 +361,13 @@ main(int argc, char *argv[]) {
fprintf(stderr, "dinput: warning: no locale support\n"); fprintf(stderr, "dinput: warning: no locale support\n");
if(!(dpy = XOpenDisplay(NULL))) if(!(dpy = XOpenDisplay(NULL)))
eprint("cannot open display\n"); eprint("cannot open display\n");
if(atexit(&cleanup) != 0)
eprint("cannot register cleanup\n");
screen = DefaultScreen(dpy); screen = DefaultScreen(dpy);
root = RootWindow(dpy, screen); root = RootWindow(dpy, screen);
running = grabkeyboard(); grabkeyboard();
setup(topbar); setup();
drawinput();
XSync(dpy, False);
run(); run();
cleanup(); return 0;
XCloseDisplay(dpy);
return ret;
} }

47
dmenu.c
View File

@ -36,7 +36,7 @@ static void dinput(void);
static void drawmenu(void); static void drawmenu(void);
static void drawmenuh(void); static void drawmenuh(void);
static void drawmenuv(void); static void drawmenuv(void);
static Bool grabkeyboard(void); static void grabkeyboard(void);
static void kpress(XKeyEvent *e); static void kpress(XKeyEvent *e);
static void match(char *pattern); static void match(char *pattern);
static void readstdin(void); static void readstdin(void);
@ -52,14 +52,12 @@ static char *prompt = NULL;
static char text[4096]; static char text[4096];
static int cmdw = 0; static int cmdw = 0;
static int promptw = 0; static int promptw = 0;
static int ret = 0;
static int screen; static int screen;
static unsigned int lines = 0; static unsigned int lines = 0;
static unsigned int numlockmask = 0; static unsigned int numlockmask = 0;
static unsigned int mw, mh; static unsigned int mw, mh;
static unsigned long normcol[ColLast]; static unsigned long normcol[ColLast];
static unsigned long selcol[ColLast]; static unsigned long selcol[ColLast];
static Bool running = True;
static Bool topbar = True; static Bool topbar = True;
static DC dc; static DC dc;
static Display *dpy; static Display *dpy;
@ -87,15 +85,15 @@ appenditem(Item *i, Item **list, Item **last) {
void void
calcoffsetsh(void) { calcoffsetsh(void) {
unsigned int w; unsigned int x;
w = promptw + cmdw + (2 * spaceitem); x = promptw + cmdw + (2 * spaceitem);
for(next = curr; next; next = next->right) for(next = curr; next; next = next->right)
if((w += MIN(textw(&dc, next->text), mw / 3)) > mw) if((x += MIN(textw(&dc, next->text), mw / 3)) > mw)
break; break;
w = promptw + cmdw + (2 * spaceitem); x = promptw + cmdw + (2 * spaceitem);
for(prev = curr; prev && prev->left; prev = prev->left) for(prev = curr; prev && prev->left; prev = prev->left)
if((w += MIN(textw(&dc, prev->left->text), mw / 3)) > mw) if((x += MIN(textw(&dc, prev->left->text), mw / 3)) > mw)
break; break;
} }
@ -146,6 +144,7 @@ cleanup(void) {
cleanupdraw(&dc); cleanupdraw(&dc);
XDestroyWindow(dpy, win); XDestroyWindow(dpy, win);
XUngrabKeyboard(dpy, CurrentTime); XUngrabKeyboard(dpy, CurrentTime);
XCloseDisplay(dpy);
} }
void void
@ -182,7 +181,6 @@ drawmenu(void) {
else if(curr) else if(curr)
drawmenuh(); drawmenuh();
XCopyArea(dpy, dc.drawable, win, dc.gc, 0, 0, mw, mh, 0, 0); XCopyArea(dpy, dc.drawable, win, dc.gc, 0, 0, mw, mh, 0, 0);
XFlush(dpy);
} }
void void
@ -219,7 +217,7 @@ drawmenuv(void) {
XMoveResizeWindow(dpy, win, wa.x, wa.y + (topbar ? 0 : wa.height - mh), mw, mh); XMoveResizeWindow(dpy, win, wa.x, wa.y + (topbar ? 0 : wa.height - mh), mw, mh);
} }
Bool void
grabkeyboard(void) { grabkeyboard(void) {
unsigned int len; unsigned int len;
@ -229,7 +227,8 @@ grabkeyboard(void) {
break; break;
usleep(1000); usleep(1000);
} }
return len > 0; if(!len)
exit(EXIT_FAILURE);
} }
void void
@ -326,9 +325,7 @@ kpress(XKeyEvent *e) {
sel = sel->right; sel = sel->right;
break; break;
case XK_Escape: case XK_Escape:
ret = 1; exit(EXIT_FAILURE);
running = False;
return;
case XK_Home: case XK_Home:
sel = curr = item; sel = curr = item;
calcoffsets(); calcoffsets();
@ -360,8 +357,7 @@ kpress(XKeyEvent *e) {
dinput(); dinput();
fprintf(stdout, "%s", sel ? sel->text : text); fprintf(stdout, "%s", sel ? sel->text : text);
fflush(stdout); fflush(stdout);
running = False; exit(EXIT_SUCCESS);
return;
case XK_Right: case XK_Right:
case XK_Down: case XK_Down:
if(!sel || !sel->right) if(!sel || !sel->right)
@ -454,7 +450,8 @@ run(void) {
XEvent ev; XEvent ev;
/* main event loop */ /* main event loop */
while(running && !XNextEvent(dpy, &ev)) XSync(dpy, False);
while(!XNextEvent(dpy, &ev))
switch(ev.type) { switch(ev.type) {
case KeyPress: case KeyPress:
kpress(&ev.xkey); kpress(&ev.xkey);
@ -464,10 +461,11 @@ run(void) {
drawmenu(); drawmenu();
break; break;
case VisibilityNotify: case VisibilityNotify:
if (ev.xvisibility.state != VisibilityUnobscured) if(ev.xvisibility.state != VisibilityUnobscured)
XRaiseWindow(dpy, win); XRaiseWindow(dpy, win);
break; break;
} }
exit(EXIT_FAILURE);
} }
void void
@ -586,13 +584,15 @@ main(int argc, char *argv[]) {
} }
else { else {
fputs("usage: dmenu [-i] [-b] [-l <lines>] [-fn <font>] [-nb <color>]\n" fputs("usage: dmenu [-i] [-b] [-l <lines>] [-fn <font>] [-nb <color>]\n"
" [-nf <color>] [-p <prompt>] [-sb <color>] [-sf <color>] [-v]\n", stderr); " [-nf <color>] [-p <prompt>] [-sb <color>] [-sf <color>] [-v]\n", stderr);
exit(EXIT_FAILURE); exit(EXIT_FAILURE);
} }
if(!setlocale(LC_CTYPE, "") || !XSupportsLocale()) if(!setlocale(LC_CTYPE, "") || !XSupportsLocale())
fprintf(stderr, "dmenu: warning: no locale support\n"); fprintf(stderr, "dmenu: warning: no locale support\n");
if(!(dpy = XOpenDisplay(NULL))) if(!(dpy = XOpenDisplay(NULL)))
eprint("cannot open display\n"); eprint("cannot open display\n");
if(atexit(&cleanup) != 0)
eprint("cannot register cleanup\n");
screen = DefaultScreen(dpy); screen = DefaultScreen(dpy);
root = RootWindow(dpy, screen); root = RootWindow(dpy, screen);
if(!(argp = malloc(sizeof *argp * (argc+2)))) if(!(argp = malloc(sizeof *argp * (argc+2))))
@ -600,13 +600,8 @@ main(int argc, char *argv[]) {
memcpy(argp + 2, argv + 1, sizeof *argp * argc); memcpy(argp + 2, argv + 1, sizeof *argp * argc);
readstdin(); readstdin();
running = grabkeyboard(); grabkeyboard();
setup(); setup();
drawmenu();
XSync(dpy, False);
run(); run();
cleanup(); return 0;
XCloseDisplay(dpy);
return ret;
} }