2 * Copyright (c) 1980, 1987, 1992, 1993
3 * The Regents of the University of California. All rights reserved.
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
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. Neither the name of the University nor the names of its contributors
14 * may be used to endorse or promote products derived from this software
15 * without specific prior written permission.
17 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
18 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
21 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29 * @(#)head.c 8.2 (Berkeley) 5/4/95
30 * $FreeBSD: src/usr.bin/head/head.c,v 1.10.2.1 2002/02/16 12:29:04 dwmalone Exp $
42 * head - give the first few lines of a stream or of each of a set of files
44 * Bill Joy UCB August 24, 1977
47 static void head(FILE *, size_t);
48 static void head_bytes(FILE *, size_t);
49 static void obsolete(char *[]);
50 static void usage(void) __dead2
;
53 main(int argc
, char **argv
)
58 bool qflag
= false, vflag
= false;
59 int linecnt
= -1, bytecnt
= -1, eval
= 0;
63 while ((ch
= getopt(argc
, argv
, "c:n:qv")) != -1) {
66 bytecnt
= strtol(optarg
, &ep
, 10);
67 if (*ep
!= 0 || bytecnt
<= 0)
68 errx(1, "illegal byte count -- %s", optarg
);
71 linecnt
= strtol(optarg
, &ep
, 10);
72 if (*ep
!= 0 || linecnt
<= 0)
73 errx(1, "illegal line count -- %s", optarg
);
90 if (linecnt
!= -1 && bytecnt
!= -1)
91 errx(1, "can't combine line and byte counts");
96 for (; *argv
; ++argv
) {
97 if ((fp
= fopen(*argv
, "r")) == NULL
) {
102 if (vflag
|| (!qflag
&& argc
> 1)) {
103 printf("%s==> %s <==\n",
104 first
? "" : "\n", *argv
);
110 head_bytes(fp
, bytecnt
);
113 } else if (bytecnt
== -1)
114 head(stdin
, linecnt
);
116 head_bytes(stdin
, bytecnt
);
122 head(FILE *fp
, size_t cnt
)
125 size_t error
, readlen
;
127 while (cnt
&& (cp
= fgetln(fp
, &readlen
)) != NULL
) {
128 error
= fwrite(cp
, sizeof(char), readlen
, stdout
);
129 if (error
!= readlen
)
136 head_bytes(FILE *fp
, size_t cnt
)
142 if (cnt
< sizeof(buf
))
145 readlen
= sizeof(buf
);
146 readlen
= fread(buf
, sizeof(char), readlen
, fp
);
149 if (fwrite(buf
, sizeof(char), readlen
, stdout
) != readlen
)
156 obsolete(char **argv
)
160 while ((ap
= *++argv
)) {
161 /* Return if "--" or not "-[0-9]*". */
162 if (ap
[0] != '-' || ap
[1] == '-' || !isdigit(ap
[1]))
164 if ((ap
= malloc(strlen(*argv
) + 2)) == NULL
)
165 err(1, "malloc failed");
168 strcpy(ap
+ 2, *argv
+ 1);
177 "usage: head [-q | -v] [-n lines | -c bytes] [file ...]\n");