Home | History | Annotate | Download | only in posix 
      1 /* wc.c - Word count
      2  *
      3  * Copyright 2011 Rob Landley <rob (at) landley.net>
      4  *
      5  * See http://opengroup.org/onlinepubs/9699919799/utilities/wc.html
      6 
      7 USE_WC(NEWTOY(wc, USE_TOYBOX_I18N("m")"cwl[!cm]", TOYFLAG_USR|TOYFLAG_BIN|TOYFLAG_LOCALE))
      8 
      9 config WC
     10   bool "wc"
     11   default y
     12   help
     13     usage: wc -lwcm [FILE...]
     14 
     15     Count lines, words, and characters in input.
     16 
     17     -l	show lines
     18     -w	show words
     19     -c	show bytes
     20     -m	show characters
     21 
     22     By default outputs lines, words, bytes, and filename for each
     23     argument (or from stdin if none). Displays only either bytes
     24     or characters.
     25 */
     26 
     27 #define FOR_wc
     28 #include "toys.h"
     29 
     30 GLOBALS(
     31   unsigned long totals[3];
     32 )
     33 
     34 static void show_lengths(unsigned long *lengths, char *name)
     35 {
     36   int i, nospace = 1;
     37   for (i=0; i<3; i++) {
     38     if (!toys.optflags || (toys.optflags&(1<<i))) {
     39       xprintf(" %ld"+nospace, lengths[i]);
     40       nospace = 0;
     41     }
     42     TT.totals[i] += lengths[i];
     43   }
     44   if (*toys.optargs) xprintf(" %s", name);
     45   xputc('\n');
     46 }
     47 
     48 static void do_wc(int fd, char *name)
     49 {
     50   int i, len, clen=1, space;
     51   unsigned long word=0, lengths[]={0,0,0};
     52 
     53   if (toys.optflags == FLAG_c) {
     54     struct stat st;
     55 
     56     // On Linux, files in /proc often report their size as 0.
     57     if (!fstat(fd, &st) && S_ISREG(st.st_mode) && st.st_size > 0) {
     58       lengths[2] = st.st_size;
     59       goto show;
     60     }
     61   }
     62 
     63   for (;;) {
     64     len = read(fd, toybuf, sizeof(toybuf));
     65     if (len<0) perror_msg_raw(name);
     66     if (len<1) break;
     67     if (toys.optflags == FLAG_c) {
     68       lengths[2] += len;
     69       continue;
     70     }
     71     for (i=0; i<len; i+=clen) {
     72       wchar_t wchar;
     73 
     74       if (CFG_TOYBOX_I18N && (toys.optflags&FLAG_m)) {
     75         clen = mbrtowc(&wchar, toybuf+i, len-i, 0);
     76         if (clen == -1) {
     77           clen = 1;
     78           continue;
     79         }
     80         if (clen == -2) break;
     81         if (clen == 0) clen=1;
     82         space = iswspace(wchar);
     83       } else space = isspace(toybuf[i]);
     84 
     85       if (toybuf[i]==10) lengths[0]++;
     86       if (space) word=0;
     87       else {
     88         if (!word) lengths[1]++;
     89         word=1;
     90       }
     91       lengths[2]++;
     92     }
     93   }
     94 
     95 show:
     96   show_lengths(lengths, name);
     97 }
     98 
     99 void wc_main(void)
    100 {
    101   toys.optflags |= (toys.optflags&8)>>1;
    102   loopfiles(toys.optargs, do_wc);
    103   if (toys.optc>1) show_lengths(TT.totals, "total");
    104 }
    105