36b293ab5bb1e3005b9775c840bfbcab0725217b
[gnupg.git] / g10 / progress.c
1 /* progress.c
2  * Copyright (C) 2003 Free Software Foundation, Inc.
3  *
4  * This file is part of GnuPG.
5  *
6  * GnuPG is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 2 of the License, or
9  * (at your option) any later version.
10  *
11  * GnuPG is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software
18  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
19  */
20
21 #include <config.h>
22 #include <stdio.h>
23
24 #include "iobuf.h"
25 #include "filter.h"
26 #include "status.h"
27 #include "util.h"
28 #include "options.h"
29
30 /****************
31  * The filter is used to report progress to the user.
32  */
33 int
34 progress_filter (void *opaque, int control,
35                  IOBUF a, byte *buf, size_t *ret_len)
36 {
37   int rc = 0;
38   progress_filter_context_t *pfx = opaque;
39
40   if (control == IOBUFCTRL_INIT)
41     {
42       char buffer[50];
43
44       pfx->last = 0;
45       pfx->offset = 0;
46       pfx->last_time = make_timestamp ();
47
48       sprintf (buffer, "%.20s ? %lu %lu", pfx->what, pfx->offset,
49                pfx->total);
50       write_status_text (STATUS_PROGRESS, buffer);
51     }
52   else if (control == IOBUFCTRL_UNDERFLOW)
53     {
54       u32 timestamp = make_timestamp ();
55       int len = iobuf_read (a, buf, *ret_len);
56
57       if (len >= 0)
58         {
59           pfx->offset += len;
60           *ret_len = len;
61         }
62       else
63         {
64           *ret_len = 0;
65           rc = -1;
66         }
67       if ((len == -1 && pfx->offset != pfx->last)
68           || timestamp - pfx->last_time > 0)
69         {
70           char buffer[50];
71           
72           sprintf (buffer, "%.20s ? %lu %lu", pfx->what, pfx->offset,
73                    pfx->total);
74           write_status_text (STATUS_PROGRESS, buffer);
75
76           pfx->last = pfx->offset;
77           pfx->last_time = timestamp;
78         }
79     }
80   else if (control == IOBUFCTRL_DESC)
81     *(char**)buf = "progress_filter";
82   return rc;
83 }
84
85 void
86 handle_progress (progress_filter_context_t *pfx, IOBUF inp, const char *name)
87 {
88   off_t filesize = 0;
89
90   if (!opt.enable_progress_filter)
91     return;
92
93   if (!is_status_enabled ())
94     return;
95
96   if (name && *name && !(*name == '-' && !name[1]))
97     filesize = iobuf_get_filelength (inp);
98   else if (opt.set_filesize)
99     filesize = opt.set_filesize;
100
101   /* register the progress filter */
102   pfx->what = name ? name : "stdin";
103   pfx->total = filesize;
104   iobuf_push_filter (inp, progress_filter, pfx);
105 }