Spawn and read from info script without blocking
This commit is contained in:
		
							
								
								
									
										142
									
								
								main.c
									
									
									
									
									
								
							
							
						
						
									
										142
									
								
								main.c
									
									
									
									
									
								
							@@ -22,10 +22,14 @@
 | 
			
		||||
#include <stdlib.h>
 | 
			
		||||
#include <stdio.h>
 | 
			
		||||
#include <string.h>
 | 
			
		||||
#include <fcntl.h>
 | 
			
		||||
#include <unistd.h>
 | 
			
		||||
#include <errno.h>
 | 
			
		||||
#include <signal.h>
 | 
			
		||||
#include <sys/select.h>
 | 
			
		||||
#include <sys/stat.h>
 | 
			
		||||
#include <sys/time.h>
 | 
			
		||||
#include <sys/wait.h>
 | 
			
		||||
#include <X11/keysym.h>
 | 
			
		||||
 | 
			
		||||
#include "types.h"
 | 
			
		||||
@@ -38,8 +42,6 @@
 | 
			
		||||
#include "config.h"
 | 
			
		||||
 | 
			
		||||
enum {
 | 
			
		||||
	BAR_L_LEN    = 512,
 | 
			
		||||
	BAR_R_LEN    = 64,
 | 
			
		||||
	FILENAME_CNT = 1024,
 | 
			
		||||
	TITLE_LEN    = 256
 | 
			
		||||
};
 | 
			
		||||
@@ -70,12 +72,11 @@ int prefix;
 | 
			
		||||
bool resized = false;
 | 
			
		||||
 | 
			
		||||
const char * const INFO_SCRIPT = ".sxiv/exec/image-info";
 | 
			
		||||
char *info_script;
 | 
			
		||||
 | 
			
		||||
struct {
 | 
			
		||||
	char l[BAR_L_LEN];
 | 
			
		||||
	char r[BAR_R_LEN];
 | 
			
		||||
} bar;
 | 
			
		||||
  char *script;
 | 
			
		||||
  int fd;
 | 
			
		||||
  unsigned int i, lastsep;
 | 
			
		||||
} info;
 | 
			
		||||
 | 
			
		||||
timeout_t timeouts[] = {
 | 
			
		||||
	{ { 0, 0 }, false, redraw },
 | 
			
		||||
@@ -212,35 +213,66 @@ bool check_timeouts(struct timeval *t)
 | 
			
		||||
	return tmin > 0;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void open_info(void)
 | 
			
		||||
{
 | 
			
		||||
	static pid_t pid;
 | 
			
		||||
	int pfd[2];
 | 
			
		||||
 | 
			
		||||
	win.bar.l[0] = '\0';
 | 
			
		||||
 | 
			
		||||
	if (info.fd != -1) {
 | 
			
		||||
		close(info.fd);
 | 
			
		||||
		kill(pid, SIGTERM);
 | 
			
		||||
		while (waitpid(-1, NULL, WNOHANG) > 0);
 | 
			
		||||
		info.fd = -1;
 | 
			
		||||
	}
 | 
			
		||||
	if (info.script == NULL || pipe(pfd) < 0)
 | 
			
		||||
		return;
 | 
			
		||||
 | 
			
		||||
	pid = fork();
 | 
			
		||||
	if (pid > 0) {
 | 
			
		||||
		close(pfd[1]);
 | 
			
		||||
		fcntl(pfd[0], F_SETFL, O_NONBLOCK);
 | 
			
		||||
		info.fd = pfd[0];
 | 
			
		||||
		info.i = info.lastsep = 0;
 | 
			
		||||
	} else if (pid == 0) {
 | 
			
		||||
		close(pfd[0]);
 | 
			
		||||
		dup2(pfd[1], 1);
 | 
			
		||||
		execl(info.script, info.script, files[fileidx].name, NULL);
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void read_info(void)
 | 
			
		||||
{
 | 
			
		||||
	char cmd[4096];
 | 
			
		||||
	FILE *outp;
 | 
			
		||||
	int c, i = 0, n = sizeof(bar.l) - 1;
 | 
			
		||||
	bool lastsep = false;
 | 
			
		||||
	ssize_t i, n;
 | 
			
		||||
	char buf[BAR_L_LEN];
 | 
			
		||||
 | 
			
		||||
	if (info_script != NULL) {
 | 
			
		||||
		snprintf(cmd, sizeof(cmd), "%s \"%s\"", info_script, files[fileidx].name);
 | 
			
		||||
		outp = popen(cmd, "r");
 | 
			
		||||
		if (outp == NULL)
 | 
			
		||||
	while (true) {
 | 
			
		||||
		n = read(info.fd, buf, sizeof(buf));
 | 
			
		||||
		if (n < 0 && errno == EAGAIN)
 | 
			
		||||
			return;
 | 
			
		||||
		else if (n == 0)
 | 
			
		||||
			goto end;
 | 
			
		||||
		while (i < n && (c = fgetc(outp)) != EOF) {
 | 
			
		||||
			if (c == '\n') {
 | 
			
		||||
				if (!lastsep) {
 | 
			
		||||
					bar.l[i++] = ' ';
 | 
			
		||||
					lastsep = true;
 | 
			
		||||
		for (i = 0; i < n; i++) {
 | 
			
		||||
			if (buf[i] == '\n') {
 | 
			
		||||
				if (info.lastsep == 0) {
 | 
			
		||||
					win.bar.l[info.i++] = ' ';
 | 
			
		||||
					info.lastsep = 1;
 | 
			
		||||
				}
 | 
			
		||||
			} else {
 | 
			
		||||
				bar.l[i++] = c;
 | 
			
		||||
				lastsep = false;
 | 
			
		||||
				win.bar.l[info.i++] = buf[i];
 | 
			
		||||
				info.lastsep = 0;
 | 
			
		||||
			}
 | 
			
		||||
			if (info.i + 1 == sizeof(win.bar.l))
 | 
			
		||||
				goto end;
 | 
			
		||||
		}
 | 
			
		||||
		pclose(outp);
 | 
			
		||||
	}
 | 
			
		||||
end:
 | 
			
		||||
	if (lastsep)
 | 
			
		||||
		i--;
 | 
			
		||||
	bar.l[i] = '\0';
 | 
			
		||||
	info.i -= info.lastsep;
 | 
			
		||||
	win.bar.l[info.i] = '\0';
 | 
			
		||||
	win_update_bar(&win);
 | 
			
		||||
	info.fd = -1;
 | 
			
		||||
	while (waitpid(-1, NULL, WNOHANG) > 0);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void load_image(int new)
 | 
			
		||||
@@ -261,7 +293,7 @@ void load_image(int new)
 | 
			
		||||
	alternate = fileidx;
 | 
			
		||||
	fileidx = new;
 | 
			
		||||
 | 
			
		||||
	read_info();
 | 
			
		||||
	open_info();
 | 
			
		||||
 | 
			
		||||
	if (img.multi.cnt > 0 && img.multi.animate)
 | 
			
		||||
		set_timeout(animate, img.multi.frames[img.multi.sel].delay, true);
 | 
			
		||||
@@ -271,9 +303,10 @@ void load_image(int new)
 | 
			
		||||
 | 
			
		||||
void update_info(void)
 | 
			
		||||
{
 | 
			
		||||
	unsigned int i, fn, fw, n, len = sizeof(bar.r);
 | 
			
		||||
	int sel;
 | 
			
		||||
	char *t = bar.r, title[TITLE_LEN];
 | 
			
		||||
	unsigned int i, fn, fw, n;
 | 
			
		||||
	unsigned int llen = sizeof(win.bar.l), rlen = sizeof(win.bar.r);
 | 
			
		||||
	char *lt = win.bar.l, *rt = win.bar.r, title[TITLE_LEN];
 | 
			
		||||
	bool ow_info;
 | 
			
		||||
 | 
			
		||||
	for (fw = 0, i = filecnt; i > 0; fw++, i /= 10);
 | 
			
		||||
@@ -283,39 +316,37 @@ void update_info(void)
 | 
			
		||||
		win_set_title(&win, "sxiv");
 | 
			
		||||
 | 
			
		||||
		if (tns.cnt == filecnt) {
 | 
			
		||||
			n = snprintf(t, len, "%0*d/%d", fw, sel + 1, filecnt);
 | 
			
		||||
			n = snprintf(rt, rlen, "%0*d/%d", fw, sel + 1, filecnt);
 | 
			
		||||
			ow_info = true;
 | 
			
		||||
		} else {
 | 
			
		||||
			snprintf(bar.l, sizeof(bar.l), "Loading... %0*d/%d",
 | 
			
		||||
			         fw, tns.cnt, filecnt);
 | 
			
		||||
			bar.r[0] = '\0';
 | 
			
		||||
			snprintf(lt, llen, "Loading... %0*d/%d", fw, tns.cnt, filecnt);
 | 
			
		||||
			rt[0] = '\0';
 | 
			
		||||
			ow_info = false;
 | 
			
		||||
		}
 | 
			
		||||
	} else {
 | 
			
		||||
		snprintf(title, sizeof(title), "sxiv - %s", files[sel].name);
 | 
			
		||||
		win_set_title(&win, title);
 | 
			
		||||
 | 
			
		||||
		n = snprintf(t, len, "%3d%% ", (int) (img.zoom * 100.0));
 | 
			
		||||
		n = snprintf(rt, rlen, "%3d%% ", (int) (img.zoom * 100.0));
 | 
			
		||||
		if (img.multi.cnt > 0) {
 | 
			
		||||
			for (fn = 0, i = img.multi.cnt; i > 0; fn++, i /= 10);
 | 
			
		||||
			n += snprintf(t + n, len - n, "(%0*d/%d) ",
 | 
			
		||||
			n += snprintf(rt + n, rlen - n, "(%0*d/%d) ",
 | 
			
		||||
			              fn, img.multi.sel + 1, img.multi.cnt);
 | 
			
		||||
		}
 | 
			
		||||
		n += snprintf(t + n, len - n, "%0*d/%d", fw, sel + 1, filecnt);
 | 
			
		||||
		ow_info = bar.l[0] == '\0';
 | 
			
		||||
		n += snprintf(rt + n, rlen - n, "%0*d/%d", fw, sel + 1, filecnt);
 | 
			
		||||
		ow_info = info.script == NULL;
 | 
			
		||||
	}
 | 
			
		||||
	if (ow_info) {
 | 
			
		||||
		fn = strlen(files[sel].name);
 | 
			
		||||
		if (fn < sizeof(bar.l) &&
 | 
			
		||||
		if (fn < llen &&
 | 
			
		||||
		    win_textwidth(files[sel].name, fn, true) +
 | 
			
		||||
		    win_textwidth(bar.r, n, true) < win.w)
 | 
			
		||||
		    win_textwidth(rt, n, true) < win.w)
 | 
			
		||||
		{
 | 
			
		||||
			strncpy(bar.l, files[sel].name, sizeof(bar.l));
 | 
			
		||||
			strncpy(lt, files[sel].name, llen);
 | 
			
		||||
		} else {
 | 
			
		||||
			strncpy(bar.l, files[sel].base, sizeof(bar.l));
 | 
			
		||||
			strncpy(lt, files[sel].base, llen);
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	win_set_bar_info(&win, bar.l, bar.r);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void redraw(void)
 | 
			
		||||
@@ -458,8 +489,8 @@ void run(void)
 | 
			
		||||
	int xfd;
 | 
			
		||||
	fd_set fds;
 | 
			
		||||
	struct timeval timeout;
 | 
			
		||||
	bool discard, to_set;
 | 
			
		||||
	XEvent ev, nextev;
 | 
			
		||||
	bool discard;
 | 
			
		||||
 | 
			
		||||
	redraw();
 | 
			
		||||
 | 
			
		||||
@@ -482,12 +513,20 @@ void run(void)
 | 
			
		||||
				check_timeouts(NULL);
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		while (XPending(win.env.dpy) == 0 && check_timeouts(&timeout)) {
 | 
			
		||||
			/* wait for timeouts */
 | 
			
		||||
		while (XPending(win.env.dpy) == 0
 | 
			
		||||
		       && ((to_set = check_timeouts(&timeout)) || info.fd != -1))
 | 
			
		||||
		{
 | 
			
		||||
			/* check for timeouts & input */
 | 
			
		||||
			xfd = ConnectionNumber(win.env.dpy);
 | 
			
		||||
			FD_ZERO(&fds);
 | 
			
		||||
			FD_SET(xfd, &fds);
 | 
			
		||||
			select(xfd + 1, &fds, 0, 0, &timeout);
 | 
			
		||||
			if (info.fd != -1) {
 | 
			
		||||
				FD_SET(info.fd, &fds);
 | 
			
		||||
				xfd = MAX(xfd, info.fd);
 | 
			
		||||
			}
 | 
			
		||||
			select(xfd + 1, &fds, 0, 0, to_set ? &timeout : NULL);
 | 
			
		||||
			if (FD_ISSET(info.fd, &fds))
 | 
			
		||||
				read_info();
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		do {
 | 
			
		||||
@@ -641,13 +680,14 @@ int main(int argc, char **argv)
 | 
			
		||||
		warn("could not locate home directory");
 | 
			
		||||
	} else {
 | 
			
		||||
		len = strlen(homedir) + strlen(INFO_SCRIPT) + 2;
 | 
			
		||||
		info_script = (char*) s_malloc(len);
 | 
			
		||||
		snprintf(info_script, len, "%s/%s", homedir, INFO_SCRIPT);
 | 
			
		||||
		if (access(info_script, X_OK) != 0) {
 | 
			
		||||
			free(info_script);
 | 
			
		||||
			info_script = NULL;
 | 
			
		||||
		info.script = (char*) s_malloc(len);
 | 
			
		||||
		snprintf(info.script, len, "%s/%s", homedir, INFO_SCRIPT);
 | 
			
		||||
		if (access(info.script, X_OK) != 0) {
 | 
			
		||||
			free(info.script);
 | 
			
		||||
			info.script = NULL;
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	info.fd = -1;
 | 
			
		||||
 | 
			
		||||
	if (options->thumb_mode) {
 | 
			
		||||
		mode = MODE_THUMB;
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										29
									
								
								window.c
									
									
									
									
									
								
							
							
						
						
									
										29
									
								
								window.c
									
									
									
									
									
								
							@@ -414,9 +414,7 @@ void win_draw_bar(win_t *win)
 | 
			
		||||
	XSetForeground(e->dpy, gc, win->bar.fgcol);
 | 
			
		||||
	XSetBackground(e->dpy, gc, win->bar.bgcol);
 | 
			
		||||
 | 
			
		||||
	if (win->bar.r != NULL) {
 | 
			
		||||
		len = strlen(win->bar.r);
 | 
			
		||||
		if (len > 0) {
 | 
			
		||||
	if ((len = strlen(win->bar.r)) > 0) {
 | 
			
		||||
		if ((tw = win_textwidth(win->bar.r, len, true)) > w)
 | 
			
		||||
			return;
 | 
			
		||||
		x = win->w - tw + H_TEXT_PAD;
 | 
			
		||||
@@ -426,9 +424,8 @@ void win_draw_bar(win_t *win)
 | 
			
		||||
		else
 | 
			
		||||
			XDrawString(e->dpy, win->pm, gc, x, y, win->bar.r, len);
 | 
			
		||||
	}
 | 
			
		||||
	}
 | 
			
		||||
	if (win->bar.l != NULL) {
 | 
			
		||||
		olen = len = strlen(win->bar.l);
 | 
			
		||||
	if ((len = strlen(win->bar.l)) > 0) {
 | 
			
		||||
		olen = len;
 | 
			
		||||
		while (len > 0 && (tw = win_textwidth(win->bar.l, len, true)) > w)
 | 
			
		||||
			len--;
 | 
			
		||||
		if (len > 0) {
 | 
			
		||||
@@ -480,6 +477,18 @@ void win_draw_rect(win_t *win, Pixmap pm, int x, int y, int w, int h,
 | 
			
		||||
		XDrawRectangle(win->env.dpy, pm, gc, x, y, w, h);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void win_update_bar(win_t *win)
 | 
			
		||||
{
 | 
			
		||||
	if (win == NULL || win->xwin == None || win->pm == None)
 | 
			
		||||
		return;
 | 
			
		||||
 | 
			
		||||
	if (win->bar.h > 0) {
 | 
			
		||||
		win_draw_bar(win);
 | 
			
		||||
		XCopyArea(win->env.dpy, win->pm, win->xwin, gc,
 | 
			
		||||
		          0, win->h, win->w, win->bar.h, 0, win->h);
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
int win_textwidth(const char *text, unsigned int len, bool with_padding)
 | 
			
		||||
{
 | 
			
		||||
	XRectangle r;
 | 
			
		||||
@@ -514,14 +523,6 @@ void win_set_title(win_t *win, const char *title)
 | 
			
		||||
	                PropModeReplace, (unsigned char *) title, strlen(title));
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void win_set_bar_info(win_t *win, char *linfo, char *rinfo)
 | 
			
		||||
{
 | 
			
		||||
	if (win != NULL) {
 | 
			
		||||
		win->bar.l = linfo;
 | 
			
		||||
		win->bar.r = rinfo;
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void win_set_cursor(win_t *win, cursor_t cursor)
 | 
			
		||||
{
 | 
			
		||||
	if (win == NULL || win->xwin == None)
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										12
									
								
								window.h
									
									
									
									
									
								
							
							
						
						
									
										12
									
								
								window.h
									
									
									
									
									
								
							@@ -24,6 +24,11 @@
 | 
			
		||||
 | 
			
		||||
#include "types.h"
 | 
			
		||||
 | 
			
		||||
enum {
 | 
			
		||||
	BAR_L_LEN = 512,
 | 
			
		||||
	BAR_R_LEN = 64
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
typedef struct {
 | 
			
		||||
	Display *dpy;
 | 
			
		||||
	int scr;
 | 
			
		||||
@@ -55,8 +60,8 @@ typedef struct {
 | 
			
		||||
 | 
			
		||||
	struct {
 | 
			
		||||
		unsigned int h;
 | 
			
		||||
		char *l;
 | 
			
		||||
		char *r;
 | 
			
		||||
		char l[BAR_L_LEN];
 | 
			
		||||
		char r[BAR_R_LEN];
 | 
			
		||||
		unsigned long bgcol;
 | 
			
		||||
		unsigned long fgcol;
 | 
			
		||||
	} bar;
 | 
			
		||||
@@ -80,10 +85,11 @@ void win_draw(win_t*);
 | 
			
		||||
void win_draw_rect(win_t*, Pixmap, int, int, int, int, bool, int,
 | 
			
		||||
                   unsigned long);
 | 
			
		||||
 | 
			
		||||
void win_update_bar(win_t*);
 | 
			
		||||
 | 
			
		||||
int win_textwidth(const char*, unsigned int, bool);
 | 
			
		||||
 | 
			
		||||
void win_set_title(win_t*, const char*);
 | 
			
		||||
void win_set_bar_info(win_t*, char*, char*);
 | 
			
		||||
void win_set_cursor(win_t*, cursor_t);
 | 
			
		||||
 | 
			
		||||
#endif /* WINDOW_H */
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user