sbase/head.c
Hiltjo Posthuma fab4b384e7 use agetline instead of agets
also use agetline where fgets with a static buffer was used previously.

Signed-off-by: Hiltjo Posthuma <hiltjo@codemadness.org>
2014-06-01 18:03:10 +01:00

67 lines
1.0 KiB
C

/* See LICENSE file for copyright and license details. */
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include "text.h"
#include "util.h"
static void head(FILE *, const char *, long);
static void
usage(void)
{
eprintf("usage: %s [-n] [FILE...]\n", argv0);
}
int
main(int argc, char *argv[])
{
long n = 10;
FILE *fp;
ARGBEGIN {
case 'n':
n = estrtol(EARGF(usage()), 0);
break;
ARGNUM:
n = ARGNUMF(0);
break;
default:
usage();
} ARGEND;
if(argc == 0) {
head(stdin, "<stdin>", n);
} else {
for(; argc > 0; argc--, argv++) {
if(!(fp = fopen(argv[0], "r"))) {
weprintf("fopen %s:", argv[0]);
continue;
}
head(fp, argv[0], n);
fclose(fp);
}
}
return EXIT_SUCCESS;
}
static void
head(FILE *fp, const char *str, long n)
{
char *buf = NULL;
size_t size = 0;
ssize_t len;
unsigned long i = 0;
while(i < n && ((len = agetline(&buf, &size, fp)) != -1)) {
fputs(buf, stdout);
if(buf[len - 1] == '\n')
i++;
}
free(buf);
if(ferror(fp))
eprintf("%s: read error:", str);
}