head.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Mini head implementation for busybox
   4  *
   5  *
   6  * Copyright (C) 1999,2000 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"
  31 #ifndef BB_FEATURE_TRIVIAL_HELP
  32         "\nPrint first 10 lines of each FILE to standard output.\n"
  33         "With more than one FILE, precede each with a header giving the\n"
  34         "file name. With no FILE, or when FILE is -, read standard input.\n\n"
  35
  36         "Options:\n" "\t-n NUM\t\tPrint first NUM lines instead of first 10\n"
  37 #endif
  38         ;
  39
  40 int head(int len, FILE * src)
  41 {
  42         int i;
  43         char buffer[1024];
  44
  45         for (i = 0; i < len; i++) {
  46                 fgets(buffer, 1024, src);
  47                 if (feof(src)) {
  48                         break;
  49                 }
  50                 fputs(buffer, stdout);
  51         }
  52         return 0;
  53 }
  54
  55 /* BusyBoxed head(1) */
  56 int head_main(int argc, char **argv)
  57 {
  58         char opt;
  59         int len = 10, tmplen, i;
  60
  61         /* parse argv[] */
  62         for (i = 1; i < argc; i++) {
  63                 if (argv[i][0] == '-') {
  64                         opt = argv[i][1];
  65                         switch (opt) {
  66                         case 'n':
  67                                 tmplen = 0;
  68                                 if (++i < argc)
  69                                         tmplen = atoi(argv[i]);
  70                                 if (tmplen < 1)
  71                                         usage(head_usage);
  72                                 len = tmplen;
  73                                 break;
  74                         case '-':
  75                         case 'h':
  76                                 usage(head_usage);
  77                         default:
  78                                 errorMsg("invalid option -- %c\n", opt);
  79                                 usage(head_usage);
  80                         }
  81                 } else {
  82                         break;
  83                 }
  84         }
  85
  86         /* get rest of argv[] or stdin if nothing's left */
  87         if (i >= argc) {
  88                 head(len, stdin);
  89
  90         } else {
  91                 int need_headers = ((argc - i) > 1);
  92
  93                 for (; i < argc; i++) {
  94                         FILE *src;
  95
  96                         src = fopen(argv[i], "r");
  97                         if (!src) {
  98                                 errorMsg("%s: %s\n", argv[i], strerror(errno));
  99                         } else {
 100                                 /* emulating GNU behaviour */
 101                                 if (need_headers) {
 102                                         fprintf(stdout, "==> %s <==\n", argv[i]);
 103                                 }
 104                                 head(len, src);
 105                                 if (i < argc - 1) {
 106                                         fprintf(stdout, "\n");
 107                                 }
 108                         }
 109                 }
 110         }
 111         return(0);
 112 }
 113
 114 /* $Id: head.c,v 1.12 2000/07/14 01:51:25 kraai Exp $ */