1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
/* count.c - Progress indicator from stdin to stdout
*
* Copyright 2002 Rob Landley <rob@landley.net>
USE_COUNT(NEWTOY(count, NULL, TOYFLAG_USR|TOYFLAG_BIN))
config COUNT
bool "count"
default y
help
usage: count
Copy stdin to stdout, displaying simple progress indicator to stderr.
*/
#include "toys.h"
void count_main(void)
{
struct pollfd pfd = {0, POLLIN, 0};
unsigned long long size = 0, last = 0, then = 0, now;
char *buf = xmalloc(65536);
int len;
// poll, print if data not ready, update 4x/second otherwise
for (;;) {
if (!(len = poll(&pfd, 1, (last != size) ? 250 : 0))) continue;
if (len<0 && errno != EINTR && errno != ENOMEM) perror_exit(0);
if ((len = xread(0, buf, 65536))) {
xwrite(1, buf, len);
size += len;
if ((now = millitime())-then<250) continue;
}
dprintf(2, "%llu bytes\r", size);
if (!len) break;
}
dprintf(2, "\n");
}
|
