1 /* head -- output first part of file(s)
2 Copyright (C) 89, 90, 91, 95, 1996 Free Software Foundation, Inc.
4 This program is free software; you can redistribute it and/or modify
5 it under the terms of the GNU General Public License as published by
6 the Free Software Foundation; either version 2, or (at your option)
9 This program is distributed in the hope that it will be useful,
10 but WITHOUT ANY WARRANTY; without even the implied warranty of
11 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 GNU General Public License for more details.
14 You should have received a copy of the GNU General Public License
15 along with this program; if not, write to the Free Software Foundation,
16 Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. */
18 /* Options: (see usage)
19 Reads from standard input if no files are given or when a filename of
21 By default, filename headers are printed only if more than one file
23 By default, prints the first 10 lines (head -n 10).
25 David MacKenzie <djm@gnu.ai.mit.edu> */
31 #include <sys/types.h>
35 /* Number of lines/chars/blocks to head. */
36 #define DEFAULT_NUMBER 10
38 /* Size of atomic reads. */
39 #define BUFSIZE (512 * 8)
41 /* Number of bytes per item we are printing.
42 If 0, head in lines. */
45 /* If nonzero, print filename headers. */
46 static int print_headers;
48 /* When to print the filename banners. */
51 multiple_files, always, never
56 /* The name this program was run with. */
59 /* Have we ever read standard input? */
60 static int have_read_stdin;
62 /* If nonzero, display usage information and exit. */
65 /* If nonzero, print the version on standard output then exit. */
66 static int show_version;
68 static struct option const long_options[] =
70 {"bytes", required_argument, NULL, 'c'},
71 {"lines", required_argument, NULL, 'n'},
72 {"quiet", no_argument, NULL, 'q'},
73 {"silent", no_argument, NULL, 'q'},
74 {"verbose", no_argument, NULL, 'v'},
75 {"help", no_argument, &show_help, 1},
76 {"version", no_argument, &show_version, 1},
84 fprintf (stderr, _("Try `%s --help' for more information.\n"),
89 Usage: %s [OPTION]... [FILE]...\n\
93 Print first 10 lines of each FILE to standard output.\n\
94 With more than one FILE, precede each with a header giving the file name.\n\
95 With no FILE, or when FILE is -, read standard input.\n\
97 -c, --bytes=SIZE print first SIZE bytes\n\
98 -n, --lines=NUMBER print first NUMBER lines instead of first 10\n\
99 -q, --quiet, --silent never print headers giving file names\n\
100 -v, --verbose always print headers giving file names\n\
101 --help display this help and exit\n\
102 --version output version information and exit\n\
104 SIZE may have a multiplier suffix: b for 512, k for 1K, m for 1 Meg.\n\
105 If -VALUE is used as first OPTION, read -c VALUE when one of\n\
106 multipliers bkm follows concatenated, else read -n VALUE.\n\
108 puts (_("\nReport bugs to textutils-bugs@gnu.ai.mit.edu"));
110 exit (status == 0 ? EXIT_SUCCESS : EXIT_FAILURE);
113 /* Convert STR, a string of ASCII digits, into an unsigned integer.
114 Return -1 if STR does not represent a valid unsigned integer. */
117 atou (const char *str)
121 for (value = 0; ISDIGIT (*str); ++str)
122 value = value * 10 + *str - '0';
123 return *str ? -1 : value;
127 parse_unit (char *str)
129 int arglen = strlen (str);
134 switch (str[arglen - 1])
138 str[arglen - 1] = '\0';
142 str[arglen - 1] = '\0';
146 str[arglen - 1] = '\0';
152 write_header (const char *filename)
154 static int first_file = 1;
156 printf ("%s==> %s <==\n", (first_file ? "" : "\n"), filename);
161 head_bytes (const char *filename, int fd, long int bytes_to_write)
163 char buffer[BUFSIZE];
166 while (bytes_to_write)
168 bytes_read = safe_read (fd, buffer, BUFSIZE);
171 error (0, errno, "%s", filename);
176 if (bytes_read > bytes_to_write)
177 bytes_read = bytes_to_write;
178 if (fwrite (buffer, 1, bytes_read, stdout) == 0)
179 error (EXIT_FAILURE, errno, _("write error"));
180 bytes_to_write -= bytes_read;
186 head_lines (const char *filename, int fd, long int lines_to_write)
188 char buffer[BUFSIZE];
192 while (lines_to_write)
194 bytes_read = safe_read (fd, buffer, BUFSIZE);
197 error (0, errno, "%s", filename);
203 while (bytes_to_write < bytes_read)
204 if (buffer[bytes_to_write++] == '\n' && --lines_to_write == 0)
206 if (fwrite (buffer, 1, bytes_to_write, stdout) == 0)
207 error (EXIT_FAILURE, errno, _("write error"));
213 head (const char *filename, int fd, long int number)
216 return head_bytes (filename, fd, number);
218 return head_lines (filename, fd, number);
222 head_file (const char *filename, long int number)
226 if (!strcmp (filename, "-"))
229 filename = _("standard input");
231 write_header (filename);
232 return head (filename, 0, number);
236 fd = open (filename, O_RDONLY);
242 write_header (filename);
243 errors = head (filename, fd, number);
247 error (0, errno, "%s", filename);
253 main (int argc, char **argv)
255 enum header_mode header_mode = multiple_files;
257 long number = -1; /* Number of items to print (-1 if undef.). */
258 int c; /* Option character. */
260 program_name = argv[0];
261 setlocale (LC_ALL, "");
262 bindtextdomain (PACKAGE, LOCALEDIR);
263 textdomain (PACKAGE);
269 if (argc > 1 && argv[1][0] == '-' && ISDIGIT (argv[1][1]))
271 /* Old option syntax; a dash, one or more digits, and one or
272 more option letters. Move past the number. */
273 for (number = 0, ++argv[1]; ISDIGIT (*argv[1]); ++argv[1])
274 number = number * 10 + *argv[1] - '0';
275 /* Parse any appended option letters. */
305 header_mode = always;
309 error (0, 0, _("unrecognized option `-%c'"), *argv[1]);
314 /* Make the options we just parsed invisible to getopt. */
320 while ((c = getopt_long (argc, argv, "c:n:qv", long_options, NULL)) != -1)
334 /* FIXME: use xstrtoul instead. */
335 number = atou (optarg);
337 error (EXIT_FAILURE, 0, _("invalid number `%s'"), optarg);
345 header_mode = always;
355 printf ("head (%s) %s\n", GNU_PACKAGE, VERSION);
363 number = DEFAULT_NUMBER;
368 if (header_mode == always
369 || (header_mode == multiple_files && optind < argc - 1))
373 exit_status |= head_file ("-", number);
375 for (; optind < argc; ++optind)
376 exit_status |= head_file (argv[optind], number);
378 if (have_read_stdin && close (0) < 0)
379 error (EXIT_FAILURE, errno, "-");
380 if (fclose (stdout) == EOF)
381 error (EXIT_FAILURE, errno, _("write error"));
383 exit (exit_status == 0 ? EXIT_SUCCESS : EXIT_FAILURE);