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 | }
|
---|