xref: /netbsd-src/usr.bin/head/head.c (revision ce63d6c20fc4ec8ddc95c84bb229e3c4ecf82b69)
1 /*
2  * Copyright (c) 1980, 1987 Regents of the University of California.
3  * 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 char copyright[] =
36 "@(#) Copyright (c) 1980, 1987 Regents of the University of California.\n\
37  All rights reserved.\n";
38 #endif /* not lint */
39 
40 #ifndef lint
41 static char sccsid[] = "@(#)head.c	5.5 (Berkeley) 6/1/90";
42 #endif /* not lint */
43 
44 #include <stdio.h>
45 #include <ctype.h>
46 /*
47  * head - give the first few lines of a stream or of each of a set of files
48  *
49  * Bill Joy UCB August 24, 1977
50  */
51 
52 main(argc, argv)
53 	int	argc;
54 	char	**argv;
55 {
56 	register int	ch, cnt;
57 	int	firsttime, linecnt = 10;
58 
59 	if (argc > 1 && argv[1][0] == '-') {
60 		if (!isdigit(argv[1][1])) {
61 			fprintf(stderr, "head: illegal option -- %c\n", argv[1][1]);
62 			goto usage;
63 		}
64 		if ((linecnt = atoi(argv[1] + 1)) < 0) {
65 usage:			fputs("usage: head [-line_count] [file ...]\n", stderr);
66 			exit(1);
67 		}
68 		--argc; ++argv;
69 	}
70 	/* setlinebuf(stdout); */
71 	for (firsttime = 1, --argc, ++argv;; firsttime = 0) {
72 		if (!*argv) {
73 			if (!firsttime)
74 				exit(0);
75 		}
76 		else {
77 			if (!freopen(*argv, "r", stdin)) {
78 				fprintf(stderr, "head: can't read %s.\n", *argv);
79 				exit(1);
80 			}
81 			if (argc > 1) {
82 				if (!firsttime)
83 					putchar('\n');
84 				printf("==> %s <==\n", *argv);
85 			}
86 			++argv;
87 		}
88 		for (cnt = linecnt; cnt; --cnt)
89 			while ((ch = getchar()) != EOF)
90 				if (putchar(ch) == '\n')
91 					break;
92 	}
93 	/*NOTREACHED*/
94 }
95