Desktop
/* $Id: progress.c,v 1.42 2010/12/04 17:24:55 khorben Exp $ */
/* Copyright (c) 2010 Pierre Pronchery <khorben@defora.org> */
/* This file is part of DeforaOS Desktop Accessories */
/* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, version 3 of the License.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>. */
/* TODO:
* - fix when EOF occurs on the output side */
#include <sys/stat.h>
#include <sys/time.h>
#include <sys/types.h>
#include <fcntl.h>
#include <unistd.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <errno.h>
#include <gtk/gtk.h>
/* types */
typedef struct _Prefs
{
int flags;
ssize_t bufsiz;
char * filename;
char * title;
size_t length;
} Prefs;
#define PREFS_z 0x1
/* progress */
/* types */
typedef struct _Progress
{
Prefs * prefs; /* preferences */
struct timeval tv; /* start time */
int fd; /* read descriptor */
int eof; /* end of file */
int fds[2]; /* for the pipe */
pid_t pid; /* child's pid */
size_t cnt; /* bytes written */
char * buf;
size_t bufsiz;
size_t buf_cnt;
GIOChannel * in_channel;
guint in_id;
GIOChannel * out_channel;
guint out_id;
/* widgets */
GtkWidget * window;
GtkWidget * done;
GtkWidget * speed;
GtkWidget * progress;
int pulse; /* tells when to pulse */
} Progress;
/* functions */
static int _progress_error(Progress * progress, char const * message, int ret);
static int _progress_gerror(Progress * progress, char const * message,
GError * error, int ret);
static int _progress_exec(Progress * progress, char * argv[]);
/* callbacks */
static gboolean _progress_closex(gpointer data);
static void _progress_cancel(void);
static gboolean _progress_channel(GIOChannel * source, GIOCondition condition,
gpointer data);
static gboolean _progress_idle_in(gpointer data);
static gboolean _progress_idle_out(gpointer data);
static gboolean _progress_timeout(gpointer data);
static int _progress(Prefs * prefs, char * argv[])
{
Progress p;
struct stat st;
GtkWidget * vbox;
GtkWidget * hbox;
GtkSizeGroup * left;
GtkSizeGroup * right;
GtkWidget * widget;
PangoFontDescription * bold;
char * q;
memset(&p, 0, sizeof(p));
p.prefs = prefs;
if((p.buf = malloc(prefs->bufsiz)) == NULL)
return _progress_error(&p, "malloc", 1);
p.bufsiz = prefs->bufsiz;
if(pipe(p.fds) != 0)
return _progress_error(&p, "pipe", 1);
if((p.pid = fork()) == -1)
{
close(p.fds[0]);
close(p.fds[1]);
return _progress_error(&p, "fork", 1);
}
if(p.pid != 0)
return _progress_exec(&p, argv);
close(p.fds[0]);
if(gettimeofday(&p.tv, NULL) != 0)
return _progress_error(&p, "gettimeofday", 1);
if(prefs->filename == NULL)
prefs->filename = "Standard input";
else if((p.fd = open(prefs->filename, O_RDONLY)) < 0)
return _progress_error(&p, prefs->filename, 1);
else if(fstat(p.fd, &st) == 0)
prefs->length = st.st_size;
p.in_channel = g_io_channel_unix_new(p.fd);
g_io_channel_set_encoding(p.in_channel, NULL, NULL);
p.in_id = 0;
g_idle_add(_progress_idle_in, &p);
p.out_channel = g_io_channel_unix_new(p.fds[1]);
g_io_channel_set_encoding(p.out_channel, NULL, NULL);
p.out_id = 0;
/* graphical interface */
p.window = gtk_window_new(GTK_WINDOW_TOPLEVEL);
gtk_window_set_default_size(GTK_WINDOW(p.window), 300, 100);
gtk_window_set_title(GTK_WINDOW(p.window), prefs->title != NULL
? prefs->title : "Progress");
g_signal_connect_swapped(G_OBJECT(p.window), "delete-event", G_CALLBACK(
_progress_closex), p.window);
vbox = gtk_vbox_new(FALSE, 0);
hbox = gtk_hbox_new(FALSE, 0);
left = gtk_size_group_new(GTK_SIZE_GROUP_HORIZONTAL);
right = gtk_size_group_new(GTK_SIZE_GROUP_HORIZONTAL);
/* file */
widget = gtk_label_new("File: ");
bold = pango_font_description_new();
pango_font_description_set_weight(bold, PANGO_WEIGHT_BOLD);
gtk_widget_modify_font(widget, bold);
gtk_misc_set_alignment(GTK_MISC(widget), 0.0, 0.0);
gtk_size_group_add_widget(left, widget);
gtk_box_pack_start(GTK_BOX(hbox), widget, FALSE, TRUE, 0);
if((q = g_filename_to_utf8(prefs->filename, -1, NULL, NULL, NULL))
== NULL)
q = prefs->filename;
widget = gtk_label_new(q);
gtk_label_set_ellipsize(GTK_LABEL(widget), PANGO_ELLIPSIZE_MIDDLE);
gtk_misc_set_alignment(GTK_MISC(widget), 0.0, 0.0);
gtk_size_group_add_widget(right, widget);
gtk_box_pack_start(GTK_BOX(hbox), widget, TRUE, TRUE, 0);
gtk_box_pack_start(GTK_BOX(vbox), hbox, TRUE, TRUE, 0);
/* done */
hbox = gtk_hbox_new(FALSE, 0);
widget = gtk_label_new("Done: ");
gtk_widget_modify_font(widget, bold);
gtk_misc_set_alignment(GTK_MISC(widget), 0.0, 0.0);
gtk_size_group_add_widget(left, widget);
gtk_box_pack_start(GTK_BOX(hbox), widget, FALSE, TRUE, 0);
p.done = gtk_label_new("0.0 kB");
gtk_misc_set_alignment(GTK_MISC(p.done), 0.0, 0.0);
gtk_size_group_add_widget(right, p.done);
gtk_box_pack_start(GTK_BOX(hbox), p.done, TRUE, TRUE, 0);
gtk_box_pack_start(GTK_BOX(vbox), hbox, TRUE, TRUE, 4);
/* speed */
hbox = gtk_hbox_new(FALSE, 0);
widget = gtk_label_new("Speed: ");
gtk_widget_modify_font(widget, bold);
gtk_misc_set_alignment(GTK_MISC(widget), 0.0, 0.0);
gtk_size_group_add_widget(left, widget);
gtk_box_pack_start(GTK_BOX(hbox), widget, FALSE, TRUE, 0);
p.speed = gtk_label_new("0.0 kB/s");
g_timeout_add(250, _progress_timeout, &p);
gtk_misc_set_alignment(GTK_MISC(p.speed), 0.0, 0.0);
gtk_size_group_add_widget(right, p.speed);
gtk_box_pack_start(GTK_BOX(hbox), p.speed, TRUE, TRUE, 0);
gtk_box_pack_start(GTK_BOX(vbox), hbox, TRUE, TRUE, 4);
/* progress */
p.progress = gtk_progress_bar_new();
p.pulse = 0;
gtk_box_pack_start(GTK_BOX(vbox), p.progress, TRUE, TRUE, 4);
hbox = gtk_hbox_new(FALSE, 0);
widget = gtk_button_new_from_stock(GTK_STOCK_CANCEL);
g_signal_connect(G_OBJECT(widget), "clicked", G_CALLBACK(
_progress_cancel), NULL);
gtk_box_pack_end(GTK_BOX(hbox), widget, FALSE, TRUE, 0);
gtk_box_pack_start(GTK_BOX(vbox), hbox, TRUE, TRUE, 0);
gtk_container_add(GTK_CONTAINER(p.window), vbox);
gtk_container_set_border_width(GTK_CONTAINER(p.window), 4);
gtk_widget_show_all(p.window);
gtk_main();
close(p.fd);
close(p.fds[1]);
return 0;
}
static int _error_do(Progress * progress, char const * message,
char const * error, int ret);
static int _progress_error(Progress * progress, char const * message,
int ret)
{
return _error_do(progress, message, strerror(errno), ret);
}
static int _progress_gerror(Progress * progress, char const * message,
GError * error, int ret)
{
_error_do(progress, message, error->message, ret);
g_error_free(error);
return ret;
}
static int _error_do(Progress * progress, char const * message,
char const * error, int ret)
{
GtkWidget * dialog;
if(ret < 0)
{
fputs("progress: ", stderr);
perror(message);
return -ret;
}
dialog = gtk_message_dialog_new((progress != NULL)
? GTK_WINDOW(progress->window) : NULL,
GTK_DIALOG_DESTROY_WITH_PARENT,
GTK_MESSAGE_ERROR, GTK_BUTTONS_CLOSE, "%s: %s", message,
error);
gtk_window_set_title(GTK_WINDOW(dialog), "Error");
gtk_dialog_run(GTK_DIALOG(dialog));
gtk_widget_destroy(dialog);
return ret;
}
static gboolean _progress_idle_in(gpointer data)
{
Progress * p = data;
#ifdef DEBUG
fprintf(stderr, "DEBUG: %s() in_id=%u\n", __func__, p->in_id);
#endif
if(p->in_id == 0)
p->in_id = g_io_add_watch(p->in_channel, G_IO_IN,
_progress_channel, p);
return FALSE;
}
static gboolean _progress_idle_out(gpointer data)
{
Progress * p = data;
#ifdef DEBUG
fprintf(stderr, "DEBUG: %s() out_id=%u\n", __func__, p->out_id);
#endif
if(p->out_id == 0)
p->out_id = g_io_add_watch(p->out_channel, G_IO_OUT,
_progress_channel, p);
return FALSE;
}
/* progress_exec */
static int _exec_gunzip(Progress * progress, char * argv[]);
static int _progress_exec(Progress * progress, char * argv[])
{
close(progress->fds[1]);
if(dup2(progress->fds[0], 0) == -1)
{
perror("dup2");
exit(1); /* FIXME warn user */
}
if(progress->prefs->flags & PREFS_z)
_exec_gunzip(progress, argv);
else
execvp(argv[0], argv);
exit(_progress_error(NULL, argv[0], -1));
return 1;
}
static int _exec_gunzip(Progress * progress, char * argv[])
{
static Progress tmp;
tmp.prefs = progress->prefs;
tmp.prefs->flags -= PREFS_z;
if(pipe(tmp.fds) != 0)
exit(_progress_error(NULL, "pipe", -1));
if((tmp.pid = fork()) == -1)
exit(_progress_error(NULL, "fork", -1));
if(tmp.pid == 0)
return _progress_exec(&tmp, argv);
close(tmp.fds[0]);
if(dup2(tmp.fds[1], 1) == -1)
exit(_progress_error(NULL, "dup2", -1));
execlp("gunzip", "gunzip", NULL);
execlp("gzip", "gzip", "-d", NULL);
exit(_progress_error(NULL, "gunzip", -1));
return 1;
}
/* callbacks */
static gboolean _progress_closex(gpointer data)
{
GtkWidget * widget = data;
gtk_widget_hide(widget);
gtk_main_quit();
return FALSE;
}
static void _progress_cancel(void)
{
gtk_main_quit();
}
/* progress_out */
static gboolean _channel_in(Progress * p, GIOChannel * source);
static gboolean _channel_out(Progress * p, GIOChannel * source);
static void _out_rate(Progress * p);
static gboolean _progress_channel(GIOChannel * source, GIOCondition condition,
gpointer data)
{
Progress * p = data;
if(condition == G_IO_IN)
return _channel_in(p, source);
if(condition == G_IO_OUT)
return _channel_out(p, source);
_progress_error(p, p->prefs->filename, 0);
gtk_main_quit();
return FALSE;
}
static gboolean _channel_in(Progress * p, GIOChannel * source)
{
GIOStatus status;
gsize read;
GError * error = NULL;
#ifdef DEBUG
fprintf(stderr, "DEBUG: %s()\n", __func__);
#endif
p->in_id = 0;
status = g_io_channel_read_chars(source, &p->buf[p->buf_cnt],
p->bufsiz - p->buf_cnt, &read, &error);
if(status == G_IO_STATUS_ERROR)
{
_progress_error(p, p->prefs->filename, 0);
g_io_channel_close(source);
gtk_main_quit();
return FALSE;
}
else if(status == G_IO_STATUS_EOF)
{
p->eof = 1; /* reached end of input file */
g_io_channel_close(source);
}
else if(p->buf_cnt + read != p->bufsiz)
g_idle_add(_progress_idle_in, p); /* continue to read */
if(p->buf_cnt == 0)
g_idle_add(_progress_idle_out, p); /* begin to write */
p->buf_cnt += read;
return FALSE;
}
static gboolean _channel_out(Progress * p, GIOChannel * source)
{
GIOStatus status;
gsize written;
GError * error = NULL;
#ifdef DEBUG
fprintf(stderr, "DEBUG: %s()\n", __func__);
#endif
p->out_id = 0;
/* write data */
status = g_io_channel_write_chars(source, p->buf, p->buf_cnt, &written,
&error);
if(status == G_IO_STATUS_ERROR)
{
_progress_gerror(p, p->prefs->filename, error, 0);
gtk_main_quit();
return FALSE;
}
else if(status == G_IO_STATUS_EOF)
{
p->eof = 1; /* reached end of output file */
g_io_channel_close(source);
}
else if(p->buf_cnt == p->bufsiz)
g_idle_add(_progress_idle_in, p); /* read again */
p->buf_cnt -= written;
memmove(p->buf, &p->buf[written], p->buf_cnt);
p->cnt += written;
_out_rate(p);
if(p->buf_cnt > 0)
g_idle_add(_progress_idle_out, p); /* continue to write */
else if(p->eof == 1) /* reached end of output */
{
g_io_channel_close(p->out_channel);
gtk_main_quit();
}
return FALSE;
}
static void _out_rate(Progress * p)
{
gdouble fraction;
GtkProgressBar * bar = GTK_PROGRESS_BAR(p->progress);
char buf[16];
if(p->prefs->length == 0 || p->cnt == 0)
{
p->pulse = 1;
return;
}
fraction = p->cnt;
fraction /= p->prefs->length;
if(gtk_progress_bar_get_fraction(bar) == fraction)
return;
gtk_progress_bar_set_fraction(bar, fraction);
snprintf(buf, sizeof(buf), "%.1f%%", fraction * 100);
gtk_progress_bar_set_text(bar, buf);
}
/* progress_timeout */
static void _timeout_done(Progress * progress);
static void _timeout_speed(Progress * progress);
static void _timeout_progress(Progress * progress);
static gboolean _progress_timeout(gpointer data)
{
Progress * progress = data;
_timeout_done(progress);
_timeout_speed(progress);
_timeout_progress(progress);
return TRUE;
}
static void _timeout_done(Progress * progress)
{
double cnt = progress->cnt / 1024;
double total = progress->prefs->length / 1024;
char buf[32];
char const * unit = "kB";
if(progress->prefs->length > 1048576 || progress->cnt > 1048576)
{
cnt /= 1024;
total /= 1024;
unit = "MB";
}
if(progress->prefs->length == 0)
snprintf(buf, sizeof(buf), "%.1f %s", cnt, unit);
else
snprintf(buf, sizeof(buf), "%.1f of %.1f %s", cnt, total, unit);
gtk_label_set_text(GTK_LABEL(progress->done), buf);
}
static void _timeout_speed(Progress * progress)
{
struct timeval tv;
double rate;
char buf[16];
char const * unit = "kB";
if(gettimeofday(&tv, NULL) != 0)
{
_progress_error(progress, "gettimeofday", FALSE);
return;
}
if((tv.tv_sec = tv.tv_sec - progress->tv.tv_sec) < 0)
tv.tv_sec = 0;
if((tv.tv_usec = tv.tv_usec - progress->tv.tv_usec) < 0)
{
tv.tv_sec--;
tv.tv_usec += 1000000;
}
rate = (tv.tv_sec * 1000) + (tv.tv_usec / 1000);
unit = "kB";
if((rate = progress->cnt / rate) > 1024)
{
rate /= 1024;
unit = "MB";
}
snprintf(buf, sizeof(buf), "%.1f %s/s", rate, unit);
gtk_label_set_text(GTK_LABEL(progress->speed), buf);
}
static void _timeout_progress(Progress * progress)
{
if(progress->pulse != 1)
return; /* setting the fraction is done somewhere else */
gtk_progress_bar_pulse(GTK_PROGRESS_BAR(progress->progress));
progress->pulse = 0;
}
/* usage */
static int _usage(void)
{
fputs("Usage: progress [-z][-b buffer size][-f file][-l length]"
"[-p prefix]\n"
" [-t title] cmd [args...]\n", stderr);
return 1;
}
/* main */
int main(int argc, char * argv[])
{
Prefs prefs;
int o;
char * p;
memset(&prefs, 0, sizeof(prefs));
prefs.bufsiz = 65536;
gtk_init(&argc, &argv);
while((o = getopt(argc, argv, "b:f:l:t:z")) != -1)
switch(o)
{
case 'b':
prefs.bufsiz = strtol(optarg, &p, 0);
if(optarg[0] == '\0' || *p != '\0'
|| prefs.bufsiz <= 0)
return _usage();
break;
case 'f':
prefs.filename = optarg;
break;
case 'l':
prefs.length = strtol(optarg, &p, 0);
if(optarg[0] == '\0' || *p != '\0')
return _usage();
break;
case 't':
prefs.title = optarg;
break;
case 'z':
prefs.flags |= PREFS_z;
break;
default:
return _usage();
}
if(argc - optind < 1)
return _usage();
return _progress(&prefs, &argv[optind]) == 0 ? 0 : 2;
}