coreutils/head.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Mini head implementation for busybox
   4  *
   5  *
   6  * Copyright (C) 1999 by Lineo, inc.
   7  * Written by John Beppu <beppu@lineo.com>
   8  *
   9  * This program is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License as published by
  11  * the Free Software Foundation; either version 2 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  17  * General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU General Public License
  20  * along with this program; if not, write to the Free Software
  21  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  22  *
  23  */
  24
  25 #include "internal.h"
  26 #include <errno.h>
  27 #include <stdio.h>
  28
  29 const char head_usage[] =
  30         "head [OPTION] [FILE]...\n\n"
  31         "Print first 10 lines of each FILE to standard output.\n"
  32         "With more than one FILE, precede each with a header giving the\n"
  33         "file name. With no FILE, or when FILE is -, read standard input.\n\n"
  34
  35         "Options:\n" "\t-n NUM\t\tPrint first NUM lines instead of first 10\n";
  36
  37 int head(int len, FILE * src)
  38 {
  39         int i;
  40         char buffer[1024];
  41
  42         for (i = 0; i < len; i++) {
  43                 fgets(buffer, 1024, src);
  44                 if (feof(src)) {
  45                         break;
  46                 }
  47                 fputs(buffer, stdout);
  48         }
  49         return 0;
  50 }
  51
  52 /* BusyBoxed head(1) */
  53 int head_main(int argc, char **argv)
  54 {
  55         char opt;
  56         int len = 10, tmplen, i;
  57
  58         /* parse argv[] */
  59         for (i = 1; i < argc; i++) {
  60                 if (argv[i][0] == '-') {
  61                         opt = argv[i][1];
  62                         switch (opt) {
  63                         case 'n':
  64                                 tmplen = 0;
  65                                 if (++i < argc)
  66                                         tmplen = atoi(argv[i]);
  67                                 if (tmplen < 1)
  68                                         usage(head_usage);
  69                                 len = tmplen;
  70                                 break;
  71                         case '-':
  72                         case 'h':
  73                                 usage(head_usage);
  74                         default:
  75                                 fprintf(stderr, "head: invalid option -- %c\n", opt);
  76                                 usage(head_usage);
  77                         }
  78                 } else {
  79                         break;
  80                 }
  81         }
  82
  83         /* get rest of argv[] or stdin if nothing's left */
  84         if (i >= argc) {
  85                 head(len, stdin);
  86
  87         } else {
  88                 int need_headers = ((argc - i) > 1);
  89
  90                 for (; i < argc; i++) {
  91                         FILE *src;
  92
  93                         src = fopen(argv[i], "r");
  94                         if (!src) {
  95                                 fprintf(stderr, "head: %s: %s\n", argv[i],
  96                                                 strerror(errno));
  97                         } else {
  98                                 /* emulating GNU behaviour */
  99                                 if (need_headers) {
 100                                         fprintf(stdout, "==> %s <==\n", argv[i]);
 101                                 }
 102                                 head(len, src);
 103                                 if (i < argc - 1) {
 104                                         fprintf(stdout, "\n");
 105                                 }
 106                         }
 107                 }
 108         }
 109         exit(0);
 110 }
 111
 112 /* $Id: head.c,v 1.8 2000/02/08 19:58:47 erik Exp $ */