mirror of
git://git.suckless.org/sbase
synced 2024-12-26 08:52:36 +00:00
b8edf3b4ee
There is still some programs left to be updated for this. Many of these programs would stop on the first file that they could not open.
103 lines
1.6 KiB
C
103 lines
1.6 KiB
C
/* See LICENSE file for copyright and license details. */
|
|
#include <ctype.h>
|
|
#include <stdbool.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
#include "util.h"
|
|
|
|
static void output(const char *, long, long, long);
|
|
static void wc(FILE *, const char *);
|
|
|
|
static bool lflag = false;
|
|
static bool wflag = false;
|
|
static char cmode = 0;
|
|
static long tc = 0, tl = 0, tw = 0;
|
|
|
|
static void
|
|
usage(void)
|
|
{
|
|
eprintf("usage: %s [-clmw] [files...]\n", argv0);
|
|
}
|
|
|
|
int
|
|
main(int argc, char *argv[])
|
|
{
|
|
FILE *fp;
|
|
int i;
|
|
|
|
ARGBEGIN {
|
|
case 'c':
|
|
cmode = 'c';
|
|
break;
|
|
case 'm':
|
|
cmode = 'm';
|
|
break;
|
|
case 'l':
|
|
lflag = true;
|
|
break;
|
|
case 'w':
|
|
wflag = true;
|
|
break;
|
|
default:
|
|
usage();
|
|
} ARGEND;
|
|
|
|
if (argc == 0) {
|
|
wc(stdin, NULL);
|
|
} else {
|
|
for (i = 0; i < argc; i++) {
|
|
if(!(fp = fopen(argv[i], "r"))) {
|
|
weprintf("fopen %s:", argv[i]);
|
|
continue;
|
|
}
|
|
wc(fp, argv[i]);
|
|
fclose(fp);
|
|
}
|
|
if (argc > 1)
|
|
output("total", tc, tl, tw);
|
|
}
|
|
return EXIT_SUCCESS;
|
|
}
|
|
|
|
void
|
|
output(const char *str, long nc, long nl, long nw)
|
|
{
|
|
bool noflags = !cmode && !lflag && !wflag;
|
|
|
|
if(lflag || noflags)
|
|
printf(" %5ld", nl);
|
|
if(wflag || noflags)
|
|
printf(" %5ld", nw);
|
|
if(cmode || noflags)
|
|
printf(" %5ld", nc);
|
|
if(str)
|
|
printf(" %s", str);
|
|
putchar('\n');
|
|
}
|
|
|
|
void
|
|
wc(FILE *fp, const char *str)
|
|
{
|
|
bool word = false;
|
|
int c;
|
|
long nc = 0, nl = 0, nw = 0;
|
|
|
|
while((c = getc(fp)) != EOF) {
|
|
if(cmode != 'm' || UTF8_POINT(c))
|
|
nc++;
|
|
if(c == '\n')
|
|
nl++;
|
|
if(!isspace(c))
|
|
word = true;
|
|
else if(word) {
|
|
word = false;
|
|
nw++;
|
|
}
|
|
}
|
|
tc += nc;
|
|
tl += nl;
|
|
tw += nw;
|
|
output(str, nc, nl, nw);
|
|
}
|