coreutils/head.c

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