libbb/progress.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Progress bar code.
   4  */
   5 /* Original copyright notice which applies to the CONFIG_FEATURE_WGET_STATUSBAR stuff,
   6  * much of which was blatantly stolen from openssh.
   7  */
   8 /*-
   9  * Copyright (c) 1992, 1993
  10  * The Regents of the University of California.  All rights reserved.
  11  *
  12  * Redistribution and use in source and binary forms, with or without
  13  * modification, are permitted provided that the following conditions
  14  * are met:
  15  * 1. Redistributions of source code must retain the above copyright
  16  *    notice, this list of conditions and the following disclaimer.
  17  * 2. Redistributions in binary form must reproduce the above copyright
  18  *    notice, this list of conditions and the following disclaimer in the
  19  *    documentation and/or other materials provided with the distribution.
  20  *
  21  * 3. BSD Advertising Clause omitted per the July 22, 1999 licensing change
  22  *    ftp://ftp.cs.berkeley.edu/pub/4bsd/README.Impt.License.Change
  23  *
  24  * 4. Neither the name of the University nor the names of its contributors
  25  *    may be used to endorse or promote products derived from this software
  26  *    without specific prior written permission.
  27  *
  28  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  29  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  30  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  31  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  32  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  33  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  34  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  35  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  36  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  37  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  38  * SUCH DAMAGE.
  39  */
  40 #include "libbb.h"
  41 #include "unicode.h"
  42
  43 enum {
  44         /* Seconds when xfer considered "stalled" */
  45         STALLTIME = 5
  46 };
  47
  48 static unsigned int get_tty2_width(void)
  49 {
  50         unsigned width;
  51         get_terminal_width_height(2, &width, NULL);
  52         return width;
  53 }
  54
  55 void FAST_FUNC bb_progress_init(bb_progress_t *p)
  56 {
  57         p->start_sec = monotonic_sec();
  58         p->lastupdate_sec = p->start_sec;
  59         p->lastsize = 0;
  60         p->inited = 1;
  61 }
  62
  63 void FAST_FUNC bb_progress_update(bb_progress_t *p,
  64                 const char *curfile,
  65                 off_t beg_range,
  66                 off_t transferred,
  67                 off_t totalsize)
  68 {
  69         uoff_t beg_and_transferred;
  70         unsigned since_last_update, elapsed;
  71         unsigned ratio;
  72         int barlength, i;
  73
  74         /* totalsize == 0 if it is unknown */
  75
  76         elapsed = monotonic_sec();
  77         since_last_update = elapsed - p->lastupdate_sec;
  78         /* Do not update on every call
  79          * (we can be called on every network read!) */
  80         if (since_last_update == 0 && !totalsize)
  81                 return;
  82
  83         beg_and_transferred = beg_range + transferred;
  84         ratio = 100;
  85         if (beg_and_transferred < totalsize) {
  86                 /* Do not update on every call
  87                  * (we can be called on every network read!) */
  88                 if (since_last_update == 0)
  89                         return;
  90                 /* long long helps to have it working even if !LFS */
  91                 ratio = 100ULL * beg_and_transferred / (uoff_t)totalsize;
  92         }
  93
  94 #if ENABLE_UNICODE_SUPPORT
  95         init_unicode();
  96         {
  97                 char *buf = unicode_conv_to_printable_fixedwidth(/*NULL,*/ curfile, 20);
  98                 fprintf(stderr, "\r%s%4u%% ", buf, ratio);
  99                 free(buf);
 100         }
 101 #else
 102         fprintf(stderr, "\r%-20.20s%4u%% ", curfile, ratio);
 103 #endif
 104
 105         barlength = get_tty2_width() - 49;
 106         if (barlength > 0) {
 107                 /* god bless gcc for variable arrays :) */
 108                 char buf[barlength + 1];
 109                 unsigned stars = (unsigned)barlength * ratio / (unsigned)100;
 110                 memset(buf, ' ', barlength);
 111                 buf[barlength] = '\0';
 112                 memset(buf, '*', stars);
 113                 fprintf(stderr, "|%s|", buf);
 114         }
 115
 116         i = 0;
 117         while (beg_and_transferred >= 100000) {
 118                 i++;
 119                 beg_and_transferred >>= 10;
 120         }
 121         /* see http://en.wikipedia.org/wiki/Tera */
 122         fprintf(stderr, "%6u%c ", (unsigned)beg_and_transferred, " kMGTPEZY"[i]);
 123 #define beg_and_transferred dont_use_beg_and_transferred_below
 124
 125         if (transferred > p->lastsize) {
 126                 p->lastupdate_sec = elapsed;
 127                 p->lastsize = transferred;
 128                 if (since_last_update >= STALLTIME) {
 129                         /* We "cut off" these seconds from elapsed time
 130                          * by adjusting start time */
 131                         p->start_sec += since_last_update;
 132                 }
 133                 since_last_update = 0; /* we are un-stalled now */
 134         }
 135         elapsed -= p->start_sec; /* now it's "elapsed since start" */
 136
 137         if (since_last_update >= STALLTIME) {
 138                 fprintf(stderr, " - stalled -");
 139         } else {
 140                 off_t to_download = totalsize - beg_range;
 141                 if (!totalsize || transferred <= 0 || (int)elapsed <= 0 || transferred > to_download) {
 142                         fprintf(stderr, "--:--:-- ETA");
 143                 } else {
 144                         /* to_download / (transferred/elapsed) - elapsed: */
 145                         /* (long long helps to have working ETA even if !LFS) */
 146                         unsigned eta = (unsigned long long)to_download*elapsed/(uoff_t)transferred - elapsed;
 147                         unsigned secs = eta % 3600;
 148                         unsigned hours = eta / 3600;
 149                         fprintf(stderr, "%02u:%02u:%02u ETA", hours, secs / 60, secs % 60);
 150                 }
 151         }
 152 }