da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin/***********************************************************************
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* This software is part of the ast package *
3e14f97f673e8a630f076077de35afdd43dc1587Roger A. Faulkner* Copyright (c) 1992-2010 AT&T Intellectual Property *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* and is licensed under the *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* Common Public License, Version 1.0 *
7c2fbfb345896881c631598ee3852ce9ce33fb07April Chin* by AT&T Intellectual Property *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* A copy of the License is available at *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* http://www.opensource.org/licenses/cpl1.0.txt *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* (with md5 checksum 059e8cd6165cb4c31e351f2b69388fd9) *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* Information and Software Systems Research *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* AT&T Research *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* Florham Park NJ *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* Glenn Fowler <gsf@research.att.com> *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* David Korn <dgk@research.att.com> *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin* *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin***********************************************************************/
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin#pragma prototyped
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin/*
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin * David Korn
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin * AT&T Bell Laboratories
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin *
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin * count the number of bytes, words, and lines in a file
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin */
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chinstatic const char usage[] =
3e14f97f673e8a630f076077de35afdd43dc1587Roger A. Faulkner"[-?\n@(#)$Id: wc (AT&T Research) 2009-11-28 $\n]"
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chinUSAGE_LICENSE
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin"[+NAME?wc - print the number of bytes, words, and lines in files]"
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin"[+DESCRIPTION?\bwc\b reads one or more input files and, by default, "
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin "for each file writes a line containing the number of newlines, "
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin "\aword\as, and bytes contained in each file followed by the "
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin "file name to standard output in that order. A \aword\a is "
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin "defined to be a non-zero length string delimited by \bisspace\b(3) "
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin "characters.]"
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin"[+?If more than one file is specified, \bwc\b writes a total count "
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin "for all of the named files with \btotal\b written instead "
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin "of the file name.]"
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin"[+?By default, \bwc\b writes all three counts. Options can specified "
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin "so that only certain counts are written. The options \b-c\b "
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin "and \b-m\b are mutually exclusive.]"
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin"[+?If no \afile\a is given, or if the \afile\a is \b-\b, \bwc\b "
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin "reads from standard input and no filename is written to standard "
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin "output. The start of the file is defined as the current offset.]"
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin"[l:lines?List the line counts.]"
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin"[w:words?List the word counts.]"
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin"[c:bytes|chars:chars?List the byte counts.]"
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin"[m|C:multibyte-chars?List the character counts.]"
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin"[q:quiet?Suppress invalid multibyte character warnings.]"
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz"[L:longest-line|max-line-length?List the longest line length; the newline,"
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz "if any, is not counted in the length.]"
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz"[N!:utf8?For \bUTF-8\b locales \b--noutf8\b disables \bUTF-8\b "
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz "optimzations and relies on the native \bmbtowc\b(3).]"
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin"\n"
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin"\n[file ...]\n"
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin"\n"
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin"[+EXIT STATUS?]{"
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin "[+0?All files processed successfully.]"
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin "[+>0?One or more files failed to open or could not be read.]"
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin"}"
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin"[+SEE ALSO?\bcat\b(1), \bisspace\b(3)]"
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin#include <cmd.h>
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin#include <wc.h>
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin#include <ls.h>
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin#define ERRORMAX 125
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chinstatic void printout(register Wc_t *wp, register char *name,register int mode)
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin{
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz if (mode&WC_LINES)
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin sfprintf(sfstdout," %7I*d",sizeof(wp->lines),wp->lines);
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz if (mode&WC_WORDS)
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin sfprintf(sfstdout," %7I*d",sizeof(wp->words),wp->words);
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz if (mode&WC_CHARS)
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin sfprintf(sfstdout," %7I*d",sizeof(wp->chars),wp->chars);
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz if (mode&WC_LONGEST)
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin sfprintf(sfstdout," %7I*d",sizeof(wp->chars),wp->longest);
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz if (name)
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin sfprintf(sfstdout," %s",name);
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin sfputc(sfstdout,'\n');
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin}
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chinint
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chinb_wc(int argc,register char **argv, void* context)
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin{
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin register char *cp;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin register int mode=0, n;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin register Wc_t *wp;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin Sfio_t *fp;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin Sfoff_t tlines=0, twords=0, tchars=0;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin struct stat statb;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin cmdinit(argc, argv, context, ERROR_CATALOG, 0);
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz for (;;)
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin {
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz switch (optget(argv, usage))
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz {
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz case 'c':
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz mode |= WC_CHARS;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz continue;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz case 'l':
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz mode |= WC_LINES;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz continue;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz case 'L':
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz mode |= WC_LONGEST;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz continue;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz case 'N':
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz if (!opt_info.num)
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz mode |= WC_NOUTF8;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz continue;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz case 'm':
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz case 'C':
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz mode |= WC_MBYTE;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz continue;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz case 'q':
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz mode |= WC_QUIET;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz continue;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz case 'w':
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz mode |= WC_WORDS;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz continue;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz case ':':
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz error(2, "%s", opt_info.arg);
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz break;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz case '?':
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz error(ERROR_usage(2), "%s", opt_info.arg);
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz break;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz }
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin break;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin argv += opt_info.index;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin if (error_info.errors)
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin error(ERROR_usage(2), "%s", optusage(NiL));
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz if (mode&WC_MBYTE)
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin {
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz if (mode&WC_CHARS)
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin error(2, "-c and -C are mutually exclusive");
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz if (!mbwide())
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin mode &= ~WC_MBYTE;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz mode |= WC_CHARS;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz if (!(mode&(WC_WORDS|WC_CHARS|WC_LINES|WC_MBYTE|WC_LONGEST)))
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin mode |= (WC_WORDS|WC_CHARS|WC_LINES);
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz if (!(wp = wc_init(mode)))
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin error(3,"internal error");
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz if (cp = *argv)
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin argv++;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz n = 0;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin do
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin {
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz if (!cp || streq(cp,"-"))
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin fp = sfstdin;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz else if (!(fp = sfopen(NiL,cp,"r")))
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin {
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin error(ERROR_system(0),"%s: cannot open",cp);
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin continue;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz if (cp)
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin n++;
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz if (!(mode&(WC_WORDS|WC_LINES|WC_MBYTE|WC_LONGEST)) && fstat(sffileno(fp),&statb)>=0
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin && S_ISREG(statb.st_mode))
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin {
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin wp->chars = statb.st_size - lseek(sffileno(fp),0L,1);
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin lseek(sffileno(fp),0L,2);
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin else
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin wc_count(wp, fp, cp);
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz if (fp!=sfstdin)
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin sfclose(fp);
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin tchars += wp->chars;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin twords += wp->words;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin tlines += wp->lines;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin printout(wp,cp,mode);
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz } while (cp= *argv++);
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz if (n > 1)
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin {
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin wp->lines = tlines;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin wp->chars = tchars;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin wp->words = twords;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin printout(wp,"total",mode);
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin }
34f9b3eef6fdadbda0a846aa4d68691ac40eace5Roland Mainz return error_info.errors<ERRORMAX?error_info.errors:ERRORMAX;
da2e3ebdc1edfbc5028edf1354e7dd2fa69a7968chin}