sbase/wc.c

102 lines
1.6 KiB
C
Raw Normal View History

2011-05-23 01:36:34 +00:00
/* See LICENSE file for copyright and license details. */
#include "utf.h"
2011-05-23 01:36:34 +00:00
#include "util.h"
static int lflag = 0;
static int wflag = 0;
static char cmode = 0;
static size_t tc = 0, tl = 0, tw = 0;
2011-05-23 01:36:34 +00:00
2015-03-07 12:33:39 +00:00
static void
output(const char *str, size_t nc, size_t nl, size_t nw)
{
int noflags = !cmode && !lflag && !wflag;
2015-02-06 19:46:37 +00:00
int first = 1;
if (lflag || noflags)
printf("%*.1zu", first ? (first = 0) : 7, nl);
if (wflag || noflags)
printf("%*.1zu", first ? (first = 0) : 7, nw);
if (cmode || noflags)
printf("%*.1zu", first ? (first = 0) : 7, nc);
if (str)
printf(" %s", str);
putchar('\n');
}
2015-03-07 12:33:39 +00:00
static void
wc(FILE *fp, const char *str)
{
2015-02-01 10:22:11 +00:00
int word = 0, rlen;
Rune c;
size_t nc = 0, nl = 0, nw = 0;
while ((rlen = efgetrune(&c, fp, str))) {
2015-02-01 10:22:11 +00:00
nc += (cmode == 'c') ? rlen :
(c != Runeerror) ? 1 : 0;
if (c == '\n')
nl++;
2015-02-11 12:24:59 +00:00
if (!isspacerune(c))
word = 1;
else if (word) {
word = 0;
nw++;
}
}
if (word)
nw++;
tc += nc;
tl += nl;
tw += nw;
output(str, nc, nl, nw);
}
2011-05-23 01:36:34 +00:00
static void
usage(void)
{
eprintf("usage: %s [-c | -m] [-lw] [file ...]\n", argv0);
}
2011-05-23 01:36:34 +00:00
int
main(int argc, char *argv[])
{
FILE *fp;
int many;
2015-02-06 19:11:33 +00:00
int ret = 0;
2013-03-11 00:12:10 +00:00
ARGBEGIN {
case 'c':
cmode = 'c';
break;
case 'm':
cmode = 'm';
break;
case 'l':
lflag = 1;
2013-03-11 00:12:10 +00:00
break;
case 'w':
wflag = 1;
2013-03-11 00:12:10 +00:00
break;
default:
usage();
2013-03-11 00:12:10 +00:00
} ARGEND;
if (!argc) {
2011-05-23 01:36:34 +00:00
wc(stdin, NULL);
2013-03-11 00:12:10 +00:00
} else {
for (many = (argc > 1); *argv; argc--, argv++) {
if (!(fp = fopen(*argv, "r"))) {
weprintf("fopen %s:", *argv);
2015-02-06 19:11:33 +00:00
ret = 1;
continue;
}
wc(fp, *argv);
2013-03-11 00:12:10 +00:00
fclose(fp);
}
if (many)
2013-03-11 00:12:10 +00:00
output("total", tc, tl, tw);
2011-05-23 01:36:34 +00:00
}
2015-02-06 19:11:33 +00:00
return ret;
2011-05-23 01:36:34 +00:00
}