88 lines
1.5 KiB
C
88 lines
1.5 KiB
C
/* head - print the first few lines of a file Author: Andy Tanenbaum */
|
|
|
|
#include <errno.h>
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
|
|
#define DEFAULT 10
|
|
|
|
_PROTOTYPE(int main, (int argc, char **argv));
|
|
_PROTOTYPE(void do_file, (int n, FILE *f));
|
|
_PROTOTYPE(void usage, (void));
|
|
|
|
int main(argc, argv)
|
|
int argc;
|
|
char *argv[];
|
|
{
|
|
FILE *f;
|
|
int legacy, n, k, nfiles;
|
|
char *ptr;
|
|
|
|
/* Check for flags. One can only specify how many lines to print. */
|
|
k = 1;
|
|
n = DEFAULT;
|
|
legacy = 0;
|
|
for (k = 1; k < argc && argv[k][0] == '-'; k++) {
|
|
ptr = &argv[k][1];
|
|
if (ptr[0] == 'n' && ptr[1] == 0) {
|
|
k++;
|
|
if (k >= argc) usage();
|
|
ptr = argv[k];
|
|
}
|
|
else if (ptr[0] == '-' && ptr[1] == 0) {
|
|
k++;
|
|
break;
|
|
}
|
|
else if (++legacy > 1) usage();
|
|
n = atoi(ptr);
|
|
if (n <= 0) usage();
|
|
}
|
|
nfiles = argc - k;
|
|
|
|
if (nfiles == 0) {
|
|
/* Print standard input only. */
|
|
do_file(n, stdin);
|
|
exit(0);
|
|
}
|
|
|
|
/* One or more files have been listed explicitly. */
|
|
while (k < argc) {
|
|
if (nfiles > 1) printf("==> %s <==\n", argv[k]);
|
|
if ((f = fopen(argv[k], "r")) == NULL)
|
|
fprintf(stderr, "%s: cannot open %s: %s\n",
|
|
argv[0], argv[k], strerror(errno));
|
|
else {
|
|
do_file(n, f);
|
|
fclose(f);
|
|
}
|
|
k++;
|
|
if (k < argc) printf("\n");
|
|
}
|
|
return(0);
|
|
}
|
|
|
|
|
|
|
|
void do_file(n, f)
|
|
int n;
|
|
FILE *f;
|
|
{
|
|
int c;
|
|
|
|
/* Print the first 'n' lines of a file. */
|
|
while (n) switch (c = getc(f)) {
|
|
case EOF:
|
|
return;
|
|
case '\n':
|
|
--n;
|
|
default: putc((char) c, stdout);
|
|
}
|
|
}
|
|
|
|
|
|
void usage()
|
|
{
|
|
fprintf(stderr, "Usage: head [-lines | -n lines] [file ...]\n");
|
|
exit(1);
|
|
}
|