toaruos/apps/head.c

74 lines
1.3 KiB
C

/**
* @brief head - Print the first `n` lines of a file.
*
* @copyright
* This file is part of ToaruOS and is released under the terms
* of the NCSA / University of Illinois License - see LICENSE.md
* Copyright (C) 2018 K. Lange
*/
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <getopt.h>
#include <errno.h>
int main(int argc, char * argv[]) {
int n = 10;
int opt;
int print_names = 0;
int retval = 0;
while ((opt = getopt(argc, argv, "n:")) != -1) {
switch (opt) {
case 'n':
n = atoi(optarg);
break;
}
}
if (argc > optind + 1) {
/* Multiple files */
print_names = 1;
}
if (argc == optind) {
/* This is silly, but should work due to reasons. */
argv[optind] = "-";
argc++;
}
for (int i = optind; i < argc; ++i) {
FILE * f = (!strcmp(argv[i],"-")) ? stdin : fopen(argv[i],"r");
if (!f) {
fprintf(stderr, "%s: %s: %s\n", argv[0], argv[i], strerror(errno));
retval = 1;
continue;
}
if (print_names) {
fprintf(stdout, "==> %s <==\n", (f == stdin) ? "standard input" : argv[i]);
}
int line = 1;
while (!feof(f)) {
int c = fgetc(f);
if (c >= 0) {
fputc(c, stdout);
if (c == '\n') {
line++;
if (line > n) break;
}
}
}
if (f != stdin) {
fclose(f);
}
}
return retval;
}