1 |
|
|
/* $OpenBSD: head.c,v 1.21 2016/03/20 17:14:51 tb Exp $ */ |
2 |
|
|
|
3 |
|
|
/* |
4 |
|
|
* Copyright (c) 1980, 1987 Regents of the University of California. |
5 |
|
|
* All rights reserved. |
6 |
|
|
* |
7 |
|
|
* Redistribution and use in source and binary forms, with or without |
8 |
|
|
* modification, are permitted provided that the following conditions |
9 |
|
|
* are met: |
10 |
|
|
* 1. Redistributions of source code must retain the above copyright |
11 |
|
|
* notice, this list of conditions and the following disclaimer. |
12 |
|
|
* 2. Redistributions in binary form must reproduce the above copyright |
13 |
|
|
* notice, this list of conditions and the following disclaimer in the |
14 |
|
|
* documentation and/or other materials provided with the distribution. |
15 |
|
|
* 3. Neither the name of the University nor the names of its contributors |
16 |
|
|
* may be used to endorse or promote products derived from this software |
17 |
|
|
* without specific prior written permission. |
18 |
|
|
* |
19 |
|
|
* THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND |
20 |
|
|
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
21 |
|
|
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
22 |
|
|
* ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE |
23 |
|
|
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL |
24 |
|
|
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS |
25 |
|
|
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) |
26 |
|
|
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT |
27 |
|
|
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY |
28 |
|
|
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF |
29 |
|
|
* SUCH DAMAGE. |
30 |
|
|
*/ |
31 |
|
|
|
32 |
|
|
#include <stdio.h> |
33 |
|
|
#include <stdlib.h> |
34 |
|
|
#include <ctype.h> |
35 |
|
|
#include <limits.h> |
36 |
|
|
#include <err.h> |
37 |
|
|
#include <errno.h> |
38 |
|
|
#include <unistd.h> |
39 |
|
|
|
40 |
|
|
static void usage(void); |
41 |
|
|
|
42 |
|
|
/* |
43 |
|
|
* head - give the first few lines of a stream or of each of a set of files |
44 |
|
|
* |
45 |
|
|
* Bill Joy UCB August 24, 1977 |
46 |
|
|
*/ |
47 |
|
|
|
48 |
|
|
int |
49 |
|
|
main(int argc, char *argv[]) |
50 |
|
|
{ |
51 |
|
|
FILE *fp; |
52 |
|
|
long cnt; |
53 |
|
|
int ch, firsttime; |
54 |
|
|
long linecnt = 10; |
55 |
|
|
char *p = NULL; |
56 |
|
|
int status = 0; |
57 |
|
|
|
58 |
✗✓ |
62 |
if (pledge("stdio rpath flock cpath wpath", NULL) == -1) |
59 |
|
|
err(1, "pledge"); |
60 |
|
|
|
61 |
|
|
/* handle obsolete -number syntax */ |
62 |
✓✗✓✗ ✓✓ |
93 |
if (argc > 1 && argv[1][0] == '-' && |
63 |
|
31 |
isdigit((unsigned char)argv[1][1])) { |
64 |
|
|
p = argv[1] + 1; |
65 |
|
16 |
argc--; |
66 |
|
16 |
argv++; |
67 |
|
16 |
} |
68 |
|
|
|
69 |
✓✓ |
46 |
while ((ch = getopt(argc, argv, "n:")) != -1) { |
70 |
✓✗ |
15 |
switch (ch) { |
71 |
|
|
case 'n': |
72 |
|
15 |
p = optarg; |
73 |
|
15 |
break; |
74 |
|
|
default: |
75 |
|
|
usage(); |
76 |
|
|
} |
77 |
|
|
} |
78 |
|
31 |
argc -= optind, argv += optind; |
79 |
|
|
|
80 |
✓✗ |
31 |
if (p) { |
81 |
|
31 |
const char *errstr; |
82 |
|
|
|
83 |
|
31 |
linecnt = strtonum(p, 1, LONG_MAX, &errstr); |
84 |
✗✓ |
31 |
if (errstr) |
85 |
|
|
errx(1, "line count %s: %s", errstr, p); |
86 |
|
31 |
} |
87 |
|
|
|
88 |
|
62 |
for (firsttime = 1; ; firsttime = 0) { |
89 |
✓✗ |
62 |
if (!*argv) { |
90 |
✓✓ |
62 |
if (!firsttime) |
91 |
|
|
exit(status); |
92 |
|
|
fp = stdin; |
93 |
✗✓ |
31 |
if (pledge("stdio flock rpath cpath wpath", NULL) == -1) |
94 |
|
|
err(1, "pledge"); |
95 |
|
|
} else { |
96 |
|
|
if ((fp = fopen(*argv, "r")) == NULL) { |
97 |
|
|
warn("%s", *argv++); |
98 |
|
|
status = 1; |
99 |
|
|
continue; |
100 |
|
|
} |
101 |
|
|
if (argc > 1) { |
102 |
|
|
if (!firsttime) |
103 |
|
|
putchar('\n'); |
104 |
|
|
printf("==> %s <==\n", *argv); |
105 |
|
|
} |
106 |
|
|
++argv; |
107 |
|
|
} |
108 |
✓✓✓✗ ✓✓ |
319 |
for (cnt = linecnt; cnt && !feof(fp); --cnt) |
109 |
✓✗✓✓ ✓✓ |
2192 |
while ((ch = getc(fp)) != EOF) |
110 |
✓✗✓✓
|
1617 |
if (putchar(ch) == '\n') |
111 |
|
|
break; |
112 |
|
31 |
fclose(fp); |
113 |
|
31 |
} |
114 |
|
|
/*NOTREACHED*/ |
115 |
|
|
} |
116 |
|
|
|
117 |
|
|
|
118 |
|
|
static void |
119 |
|
|
usage(void) |
120 |
|
|
{ |
121 |
|
|
fputs("usage: head [-count | -n count] [file ...]\n", stderr); |
122 |
|
|
exit(1); |
123 |
|
|
} |