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 int head(int len, FILE * src)
  30 {
  31         int i;
  32         char buffer[1024];
  33
  34         for (i = 0; i < len; i++) {
  35                 fgets(buffer, 1024, src);
  36                 if (feof(src)) {
  37                         break;
  38                 }
  39                 fputs(buffer, stdout);
  40         }
  41         return 0;
  42 }
  43
  44 /* BusyBoxed head(1) */
  45 int head_main(int argc, char **argv)
  46 {
  47         char opt;
  48         int len = 10, tmplen, i;
  49
  50         /* parse argv[] */
  51         for (i = 1; i < argc; i++) {
  52                 if (argv[i][0] == '-') {
  53                         opt = argv[i][1];
  54                         switch (opt) {
  55                         case 'n':
  56                                 tmplen = 0;
  57                                 if (++i < argc)
  58                                         tmplen = atoi(argv[i]);
  59                                 if (tmplen < 1)
  60                                         usage(head_usage);
  61                                 len = tmplen;
  62                                 break;
  63                         case '-':
  64                         case 'h':
  65                                 usage(head_usage);
  66                         default:
  67                                 errorMsg("invalid option -- %c\n", opt);
  68                                 usage(head_usage);
  69                         }
  70                 } else {
  71                         break;
  72                 }
  73         }
  74
  75         /* get rest of argv[] or stdin if nothing's left */
  76         if (i >= argc) {
  77                 head(len, stdin);
  78
  79         } else {
  80                 int need_headers = ((argc - i) > 1);
  81
  82                 for (; i < argc; i++) {
  83                         FILE *src;
  84
  85                         src = fopen(argv[i], "r");
  86                         if (!src) {
  87                                 errorMsg("%s: %s\n", argv[i], strerror(errno));
  88                         } else {
  89                                 /* emulating GNU behaviour */
  90                                 if (need_headers) {
  91                                         fprintf(stdout, "==> %s <==\n", argv[i]);
  92                                 }
  93                                 head(len, src);
  94                                 if (i < argc - 1) {
  95                                         fprintf(stdout, "\n");
  96                                 }
  97                         }
  98                 }
  99         }
 100         return(0);
 101 }
 102
 103 /* $Id: head.c,v 1.13 2000/07/16 20:57:15 kraai Exp $ */