1 /* 2 * Copyright (c) 1980, 1987, 1991, 1993 3 * The Regents of the University of California. All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 1. Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 3. All advertising materials mentioning features or use of this software 14 * must display the following acknowledgement: 15 * This product includes software developed by the University of 16 * California, Berkeley and its contributors. 17 * 4. Neither the name of the University nor the names of its contributors 18 * may be used to endorse or promote products derived from this software 19 * without specific prior written permission. 20 * 21 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 22 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 24 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 25 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 31 * SUCH DAMAGE. 32 */ 33 34 #ifndef lint 35 static char copyright[] = 36 "@(#) Copyright (c) 1980, 1987, 1991, 1993\n\ 37 The Regents of the University of California. All rights reserved.\n"; 38 #endif /* not lint */ 39 40 #ifndef lint 41 static char sccsid[] = "@(#)wc.c 8.1 (Berkeley) 6/6/93"; 42 #endif /* not lint */ 43 44 #include <sys/param.h> 45 #include <sys/stat.h> 46 #include <fcntl.h> 47 #include <unistd.h> 48 #include <errno.h> 49 #include <stdio.h> 50 #include <stdlib.h> 51 #include <string.h> 52 #include <ctype.h> 53 54 u_long tlinect, twordct, tcharct; 55 int doline, doword, dochar; 56 57 void cnt __P((char *)); 58 void err __P((const char *, ...)); 59 void usage __P((void)); 60 61 int 62 main(argc, argv) 63 int argc; 64 char *argv[]; 65 { 66 register int ch; 67 int total; 68 69 while ((ch = getopt(argc, argv, "lwc")) != EOF) 70 switch((char)ch) { 71 case 'l': 72 doline = 1; 73 break; 74 case 'w': 75 doword = 1; 76 break; 77 case 'c': 78 dochar = 1; 79 break; 80 case '?': 81 default: 82 usage(); 83 } 84 argv += optind; 85 argc -= optind; 86 87 /* Wc's flags are on by default. */ 88 if (doline + doword + dochar == 0) 89 doline = doword = dochar = 1; 90 91 total = 0; 92 if (!*argv) { 93 cnt(NULL); 94 (void)printf("\n"); 95 } 96 else do { 97 cnt(*argv); 98 (void)printf(" %s\n", *argv); 99 ++total; 100 } while(*++argv); 101 102 if (total > 1) { 103 if (doline) 104 (void)printf(" %7ld", tlinect); 105 if (doword) 106 (void)printf(" %7ld", twordct); 107 if (dochar) 108 (void)printf(" %7ld", tcharct); 109 (void)printf(" total\n"); 110 } 111 exit(0); 112 } 113 114 void 115 cnt(file) 116 char *file; 117 { 118 register u_char *p; 119 register short gotsp; 120 register int ch, len; 121 register u_long linect, wordct, charct; 122 struct stat sb; 123 int fd; 124 u_char buf[MAXBSIZE]; 125 126 fd = STDIN_FILENO; 127 linect = wordct = charct = 0; 128 if (file) { 129 if ((fd = open(file, O_RDONLY, 0)) < 0) 130 err("%s: %s", file, strerror(errno)); 131 if (doword) 132 goto word; 133 /* 134 * Line counting is split out because it's a lot faster to get 135 * lines than to get words, since the word count requires some 136 * logic. 137 */ 138 if (doline) { 139 while (len = read(fd, buf, MAXBSIZE)) { 140 if (len == -1) 141 err("%s: %s", file, strerror(errno)); 142 charct += len; 143 for (p = buf; len--; ++p) 144 if (*p == '\n') 145 ++linect; 146 } 147 tlinect += linect; 148 (void)printf(" %7lu", linect); 149 if (dochar) { 150 tcharct += charct; 151 (void)printf(" %7lu", charct); 152 } 153 (void)close(fd); 154 return; 155 } 156 /* 157 * If all we need is the number of characters and it's a 158 * regular or linked file, just stat the puppy. 159 */ 160 if (dochar) { 161 if (fstat(fd, &sb)) 162 err("%s: %s", file, strerror(errno)); 163 if (S_ISREG(sb.st_mode) || S_ISLNK(sb.st_mode)) { 164 (void)printf(" %7qu", sb.st_size); 165 tcharct += sb.st_size; 166 (void)close(fd); 167 return; 168 } 169 } 170 } 171 172 /* Do it the hard way... */ 173 word: for (gotsp = 1; len = read(fd, buf, MAXBSIZE);) { 174 if (len == -1) 175 err("%s: %s", file, strerror(errno)); 176 /* 177 * This loses in the presence of multi-byte characters. 178 * To do it right would require a function to return a 179 * character while knowing how many bytes it consumed. 180 */ 181 charct += len; 182 for (p = buf; len--;) { 183 ch = *p++; 184 if (ch == '\n') 185 ++linect; 186 if (isspace(ch)) 187 gotsp = 1; 188 else if (gotsp) { 189 gotsp = 0; 190 ++wordct; 191 } 192 } 193 } 194 if (doline) { 195 tlinect += linect; 196 (void)printf(" %7lu", linect); 197 } 198 if (doword) { 199 twordct += wordct; 200 (void)printf(" %7lu", wordct); 201 } 202 if (dochar) { 203 tcharct += charct; 204 (void)printf(" %7lu", charct); 205 } 206 (void)close(fd); 207 } 208 209 void 210 usage() 211 { 212 (void)fprintf(stderr, "usage: wc [-clw] [files]\n"); 213 exit(1); 214 } 215 216 #if __STDC__ 217 #include <stdarg.h> 218 #else 219 #include <varargs.h> 220 #endif 221 222 void 223 #if __STDC__ 224 err(const char *fmt, ...) 225 #else 226 err(fmt, va_alist) 227 char *fmt; 228 va_dcl 229 #endif 230 { 231 va_list ap; 232 #if __STDC__ 233 va_start(ap, fmt); 234 #else 235 va_start(ap); 236 #endif 237 (void)fprintf(stderr, "wc: "); 238 (void)vfprintf(stderr, fmt, ap); 239 va_end(ap); 240 (void)fprintf(stderr, "\n"); 241 exit(1); 242 /* NOTREACHED */ 243 } 244