• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*===- count.c - The 'count' testing tool ---------------------------------===*\
2  *
3  *                     The LLVM Compiler Infrastructure
4  *
5  * This file is distributed under the University of Illinois Open Source
6  * License. See LICENSE.TXT for details.
7  *
8 \*===----------------------------------------------------------------------===*/
9 
10 #include <stdlib.h>
11 #include <stdio.h>
12 
main(int argc,char ** argv)13 int main(int argc, char **argv) {
14   unsigned Count, NumLines, NumRead;
15   char Buffer[4096], *End;
16 
17   if (argc != 2) {
18     fprintf(stderr, "usage: %s <expected line count>\n", argv[0]);
19     return 2;
20   }
21 
22   Count = strtol(argv[1], &End, 10);
23   if (*End != '\0' && End != argv[1]) {
24     fprintf(stderr, "%s: invalid count argument '%s'\n", argv[0], argv[1]);
25     return 2;
26   }
27 
28   NumLines = 0;
29   do {
30     unsigned i;
31 
32     NumRead = fread(Buffer, 1, sizeof(Buffer), stdin);
33 
34     for (i = 0; i != NumRead; ++i)
35       if (Buffer[i] == '\n')
36         ++NumLines;
37   } while (NumRead == sizeof(Buffer));
38 
39   if (!feof(stdin)) {
40     fprintf(stderr, "%s: error reading stdin\n", argv[0]);
41     return 3;
42   }
43 
44   if (Count != NumLines) {
45     fprintf(stderr, "Expected %d lines, got %d.\n", Count, NumLines);
46     return 1;
47   }
48 
49   return 0;
50 }
51