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 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
40int 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
103void 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}
151