1 /***************************************************************************
2 * _ _ ____ _
3 * Project ___| | | | _ \| |
4 * / __| | | | |_) | |
5 * | (__| |_| | _ <| |___
6 * \___|\___/|_| \_\_____|
7 *
8 * Copyright (C) 1998 - 2019, 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 #ifdef HAVE_SYS_IOCTL_H
25 #include <sys/ioctl.h>
26 #endif
27
28 #define ENABLE_CURLX_PRINTF
29 /* use our own printf() functions */
30 #include "curlx.h"
31
32 #include "tool_cfgable.h"
33 #include "tool_cb_prg.h"
34 #include "tool_util.h"
35
36 #include "memdebug.h" /* keep this as LAST include */
37
38 #ifdef HAVE_TERMIOS_H
39 # include <termios.h>
40 #elif defined(HAVE_TERMIO_H)
41 # include <termio.h>
42 #endif
43
44 /* 200 values generated by this perl code:
45
46 my $pi = 3.1415;
47 foreach my $i (1 .. 200) {
48 printf "%d, ", sin($i/200 * 2 * $pi) * 5000 + 5000;
49 }
50 */
51 static const unsigned int sinus[] = {
52 5157, 5313, 5470, 5626, 5782, 5936, 6090, 6243, 6394, 6545, 6693, 6840, 6985,
53 7128, 7269, 7408, 7545, 7679, 7810, 7938, 8064, 8187, 8306, 8422, 8535, 8644,
54 8750, 8852, 8950, 9045, 9135, 9221, 9303, 9381, 9454, 9524, 9588, 9648, 9704,
55 9755, 9801, 9842, 9879, 9911, 9938, 9960, 9977, 9990, 9997, 9999, 9997, 9990,
56 9977, 9960, 9938, 9911, 9879, 9842, 9801, 9755, 9704, 9648, 9588, 9524, 9455,
57 9381, 9303, 9221, 9135, 9045, 8950, 8852, 8750, 8645, 8535, 8422, 8306, 8187,
58 8064, 7939, 7810, 7679, 7545, 7409, 7270, 7129, 6986, 6841, 6694, 6545, 6395,
59 6243, 6091, 5937, 5782, 5627, 5470, 5314, 5157, 5000, 4843, 4686, 4529, 4373,
60 4218, 4063, 3909, 3757, 3605, 3455, 3306, 3159, 3014, 2871, 2730, 2591, 2455,
61 2321, 2190, 2061, 1935, 1813, 1693, 1577, 1464, 1355, 1249, 1147, 1049, 955,
62 864, 778, 696, 618, 545, 476, 411, 351, 295, 244, 198, 157, 120, 88, 61, 39,
63 22, 9, 2, 0, 2, 9, 22, 39, 61, 88, 120, 156, 198, 244, 295, 350, 410, 475,
64 544, 618, 695, 777, 864, 954, 1048, 1146, 1248, 1354, 1463, 1576, 1692, 1812,
65 1934, 2060, 2188, 2320, 2454, 2590, 2729, 2870, 3013, 3158, 3305, 3454, 3604,
66 3755, 3908, 4062, 4216, 4372, 4528, 4685, 4842, 4999
67 };
68
fly(struct ProgressData * bar,bool moved)69 static void fly(struct ProgressData *bar, bool moved)
70 {
71 char buf[256];
72 int pos;
73 int check = bar->width - 2;
74
75 msnprintf(buf, sizeof(buf), "%*s\r", bar->width-1, " ");
76 memcpy(&buf[bar->bar], "-=O=-", 5);
77
78 pos = sinus[bar->tick%200] / (10000 / check);
79 buf[pos] = '#';
80 pos = sinus[(bar->tick + 5)%200] / (10000 / check);
81 buf[pos] = '#';
82 pos = sinus[(bar->tick + 10)%200] / (10000 / check);
83 buf[pos] = '#';
84 pos = sinus[(bar->tick + 15)%200] / (10000 / check);
85 buf[pos] = '#';
86
87 fputs(buf, bar->out);
88 bar->tick += 2;
89 if(bar->tick >= 200)
90 bar->tick -= 200;
91
92 bar->bar += (moved?bar->barmove:0);
93 if(bar->bar >= (bar->width - 6)) {
94 bar->barmove = -1;
95 bar->bar = bar->width - 6;
96 }
97 else if(bar->bar < 0) {
98 bar->barmove = 1;
99 bar->bar = 0;
100 }
101 }
102
103 /*
104 ** callback for CURLOPT_XFERINFOFUNCTION
105 */
106
107 #define MAX_BARLENGTH 256
108
109 #if (SIZEOF_CURL_OFF_T == 4)
110 # define CURL_OFF_T_MAX CURL_OFF_T_C(0x7FFFFFFF)
111 #else
112 /* assume CURL_SIZEOF_CURL_OFF_T == 8 */
113 # define CURL_OFF_T_MAX CURL_OFF_T_C(0x7FFFFFFFFFFFFFFF)
114 #endif
115
tool_progress_cb(void * clientp,curl_off_t dltotal,curl_off_t dlnow,curl_off_t ultotal,curl_off_t ulnow)116 int tool_progress_cb(void *clientp,
117 curl_off_t dltotal, curl_off_t dlnow,
118 curl_off_t ultotal, curl_off_t ulnow)
119 {
120 /* The original progress-bar source code was written for curl by Lars Aas,
121 and this new edition inherits some of his concepts. */
122
123 struct timeval now = tvnow();
124 struct ProgressData *bar = (struct ProgressData *)clientp;
125 curl_off_t total;
126 curl_off_t point;
127
128 /* expected transfer size */
129 if((CURL_OFF_T_MAX - bar->initial_size) < (dltotal + ultotal))
130 total = CURL_OFF_T_MAX;
131 else
132 total = dltotal + ultotal + bar->initial_size;
133
134 /* we've come this far */
135 if((CURL_OFF_T_MAX - bar->initial_size) < (dlnow + ulnow))
136 point = CURL_OFF_T_MAX;
137 else
138 point = dlnow + ulnow + bar->initial_size;
139
140 if(bar->calls) {
141 /* after first call... */
142 if(total) {
143 /* we know the total data to get... */
144 if(bar->prev == point)
145 /* progress didn't change since last invoke */
146 return 0;
147 else if((tvdiff(now, bar->prevtime) < 100L) && point < total)
148 /* limit progress-bar updating to 10 Hz except when we're at 100% */
149 return 0;
150 }
151 else {
152 /* total is unknown */
153 if(tvdiff(now, bar->prevtime) < 100L)
154 /* limit progress-bar updating to 10 Hz */
155 return 0;
156 fly(bar, point != bar->prev);
157 }
158 }
159
160 /* simply count invokes */
161 bar->calls++;
162
163 if((total > 0) && (point != bar->prev)) {
164 char line[MAX_BARLENGTH + 1];
165 char format[40];
166 double frac;
167 double percent;
168 int barwidth;
169 int num;
170 if(point > total)
171 /* we have got more than the expected total! */
172 total = point;
173
174 frac = (double)point / (double)total;
175 percent = frac * 100.0;
176 barwidth = bar->width - 7;
177 num = (int) (((double)barwidth) * frac);
178 if(num > MAX_BARLENGTH)
179 num = MAX_BARLENGTH;
180 memset(line, '#', num);
181 line[num] = '\0';
182 msnprintf(format, sizeof(format), "\r%%-%ds %%5.1f%%%%", barwidth);
183 fprintf(bar->out, format, line, percent);
184 }
185 fflush(bar->out);
186 bar->prev = point;
187 bar->prevtime = now;
188
189 return 0;
190 }
191
progressbarinit(struct ProgressData * bar,struct OperationConfig * config)192 void progressbarinit(struct ProgressData *bar,
193 struct OperationConfig *config)
194 {
195 char *colp;
196 memset(bar, 0, sizeof(struct ProgressData));
197
198 /* pass this through to progress function so
199 * it can display progress towards total file
200 * not just the part that's left. (21-may-03, dbyron) */
201 if(config->use_resume)
202 bar->initial_size = config->resume_from;
203
204 colp = curlx_getenv("COLUMNS");
205 if(colp) {
206 char *endptr;
207 long num = strtol(colp, &endptr, 10);
208 if((endptr != colp) && (endptr == colp + strlen(colp)) && (num > 20))
209 bar->width = (int)num;
210 curl_free(colp);
211 }
212
213 if(!bar->width) {
214 int cols = 0;
215
216 #ifdef TIOCGSIZE
217 struct ttysize ts;
218 if(!ioctl(STDIN_FILENO, TIOCGSIZE, &ts))
219 cols = ts.ts_cols;
220 #elif defined(TIOCGWINSZ)
221 struct winsize ts;
222 if(!ioctl(STDIN_FILENO, TIOCGWINSZ, &ts))
223 cols = ts.ws_col;
224 #elif defined(_WIN32)
225 {
226 HANDLE stderr_hnd = GetStdHandle(STD_ERROR_HANDLE);
227 CONSOLE_SCREEN_BUFFER_INFO console_info;
228
229 if((stderr_hnd != INVALID_HANDLE_VALUE) &&
230 GetConsoleScreenBufferInfo(stderr_hnd, &console_info)) {
231 /*
232 * Do not use +1 to get the true screen-width since writing a
233 * character at the right edge will cause a line wrap.
234 */
235 cols = (int)
236 (console_info.srWindow.Right - console_info.srWindow.Left);
237 }
238 }
239 #endif /* TIOCGSIZE */
240 bar->width = cols;
241 }
242
243 if(!bar->width)
244 bar->width = 79;
245 else if(bar->width > MAX_BARLENGTH)
246 bar->width = MAX_BARLENGTH;
247
248 bar->out = config->global->errors;
249 bar->tick = 150;
250 bar->barmove = 1;
251 }
252