src/tool_cb_prg.c

   1 /***************************************************************************
   2  *                                  _   _ ____  _
   3  *  Project                     ___| | | |  _ \| |
   4  *                             / __| | | | |_) | |
   5  *                            | (__| |_| |  _ <| |___
   6  *                             \___|\___/|_| \_\_____|
   7  *
   8  * Copyright (C) 1998 - 2014, Daniel Stenberg, <daniel@haxx.se>, et al.
   9  *
  10  * This software is licensed as described in the file COPYING, which
  11  * you should have received as part of this distribution. The terms
  12  * are also available at http://curl.haxx.se/docs/copyright.html.
  13  *
  14  * You may opt to use, copy, modify, merge, publish, distribute and/or sell
  15  * copies of the Software, and permit persons to whom the Software is
  16  * furnished to do so, under the terms of the COPYING file.
  17  *
  18  * This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
  19  * KIND, either express or implied.
  20  *
  21  ***************************************************************************/
  22 #include "tool_setup.h"
  23
  24 #define ENABLE_CURLX_PRINTF
  25 /* use our own printf() functions */
  26 #include "curlx.h"
  27
  28 #include "tool_cfgable.h"
  29 #include "tool_cb_prg.h"
  30 #include "tool_util.h"
  31
  32 #include "memdebug.h" /* keep this as LAST include */
  33
  34 /*
  35 ** callback for CURLOPT_XFERINFOFUNCTION
  36 */
  37
  38 #define MAX_BARLENGTH 256
  39
  40 int tool_progress_cb(void *clientp,
  41                      curl_off_t dltotal, curl_off_t dlnow,
  42                      curl_off_t ultotal, curl_off_t ulnow)
  43 {
  44   /* The original progress-bar source code was written for curl by Lars Aas,
  45      and this new edition inherits some of his concepts. */
  46
  47   char line[MAX_BARLENGTH+1];
  48   char format[40];
  49   double frac;
  50   double percent;
  51   int barwidth;
  52   int num;
  53   struct timeval now = tvnow();
  54   struct ProgressData *bar = (struct ProgressData *)clientp;
  55   curl_off_t total;
  56   curl_off_t point;
  57
  58   /* expected transfer size */
  59   total = dltotal + ultotal + bar->initial_size;
  60
  61   /* we've come this far */
  62   point = dlnow + ulnow + bar->initial_size;
  63
  64   if(bar->calls && (tvdiff(now, bar->prevtime) < 100L) && point < total)
  65     /* after first call, limit progress-bar updating to 10 Hz */
  66     /* update when we're at 100% even if last update is less than 200ms ago */
  67     return 0;
  68
  69   if(point > total)
  70     /* we have got more than the expected total! */
  71     total = point;
  72
  73   /* simply count invokes */
  74   bar->calls++;
  75
  76   if(total < 1) {
  77     curl_off_t prevblock = bar->prev / 1024;
  78     curl_off_t thisblock = point / 1024;
  79     while(thisblock > prevblock) {
  80       fprintf(bar->out, "#");
  81       prevblock++;
  82     }
  83   }
  84   else if(point != bar->prev) {
  85     frac = (double)point / (double)total;
  86     percent = frac * 100.0f;
  87     barwidth = bar->width - 7;
  88     num = (int) (((double)barwidth) * frac);
  89     if(num > MAX_BARLENGTH)
  90       num = MAX_BARLENGTH;
  91     memset(line, '#', num);
  92     line[num] = '\0';
  93     snprintf(format, sizeof(format), "\r%%-%ds %%5.1f%%%%", barwidth);
  94     fprintf(bar->out, format, line, percent);
  95   }
  96   fflush(bar->out);
  97   bar->prev = point;
  98   bar->prevtime = now;
  99
 100   return 0;
 101 }
 102
 103 void progressbarinit(struct ProgressData *bar,
 104                      struct OperationConfig *config)
 105 {
 106 #ifdef __EMX__
 107   /* 20000318 mgs */
 108   int scr_size[2];
 109 #endif
 110   char *colp;
 111
 112   memset(bar, 0, sizeof(struct ProgressData));
 113
 114   /* pass this through to progress function so
 115    * it can display progress towards total file
 116    * not just the part that's left. (21-may-03, dbyron) */
 117   if(config->use_resume)
 118     bar->initial_size = config->resume_from;
 119
 120 /* TODO: get terminal width through ansi escapes or something similar.
 121    try to update width when xterm is resized... - 19990617 larsa */
 122 #ifndef __EMX__
 123   /* 20000318 mgs
 124    * OS/2 users most likely won't have this env var set, and besides that
 125    * we're using our own way to determine screen width */
 126   colp = curlx_getenv("COLUMNS");
 127   if(colp) {
 128     char *endptr;
 129     long num = strtol(colp, &endptr, 10);
 130     if((endptr != colp) && (endptr == colp + strlen(colp)) && (num > 0))
 131       bar->width = (int)num;
 132     else
 133       bar->width = 79;
 134     curl_free(colp);
 135   }
 136   else
 137     bar->width = 79;
 138 #else
 139   /* 20000318 mgs
 140    * We use this emx library call to get the screen width, and subtract
 141    * one from what we got in order to avoid a problem with the cursor
 142    * advancing to the next line if we print a string that is as long as
 143    * the screen is wide. */
 144
 145   _scrsize(scr_size);
 146   bar->width = scr_size[0] - 1;
 147 #endif
 148
 149   bar->out = config->global->errors;
 150 }