[352] | 1 | /***************************************************************************
|
---|
| 2 | * _ _ ____ _
|
---|
| 3 | * Project ___| | | | _ \| |
|
---|
| 4 | * / __| | | | |_) | |
|
---|
| 5 | * | (__| |_| | _ <| |___
|
---|
| 6 | * \___|\___/|_| \_\_____|
|
---|
| 7 | *
|
---|
| 8 | * Copyright (C) 1998 - 2014, 2017, 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 https://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.0;
|
---|
| 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 | }
|
---|