libbb/get_line_from_file.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Utility routines.
   4  *
   5  * Copyright (C) 2005, 2006 Rob Landley <rob@landley.net>
   6  * Copyright (C) 2004 Erik Andersen <andersen@codepoet.org>
   7  * Copyright (C) 2001 Matt Krai
   8  *
   9  * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
  10  */
  11
  12 #include "libbb.h"
  13
  14 /* This function reads an entire line from a text file, up to a newline
  15  * or NUL byte, inclusive.  It returns a malloc'ed char * which must be
  16  * stored and free'ed by the caller.  If end is NULL '\n' isn't considered
  17  * end of line.  If end isn't NULL, length of the chunk read is stored in it.
  18  * Return NULL if EOF/error */
  19
  20 char *bb_get_chunk_from_file(FILE *file, int *end)
  21 {
  22         int ch;
  23         int idx = 0;
  24         char *linebuf = NULL;
  25         int linebufsz = 0;
  26
  27         while ((ch = getc(file)) != EOF) {
  28                 /* grow the line buffer as necessary */
  29                 if (idx >= linebufsz) {
  30                         linebufsz += 80;
  31                         linebuf = xrealloc(linebuf, linebufsz);
  32                 }
  33                 linebuf[idx++] = (char) ch;
  34                 if (!ch || (end && ch == '\n'))
  35                         break;
  36         }
  37         if (end)
  38                 *end = idx;
  39         if (linebuf) {
  40                 // huh, does fgets discard prior data on error like this?
  41                 // I don't think so....
  42                 //if (ferror(file)) {
  43                 //      free(linebuf);
  44                 //      return NULL;
  45                 //}
  46                 linebuf = xrealloc(linebuf, idx+1);
  47                 linebuf[idx] = '\0';
  48         }
  49         return linebuf;
  50 }
  51
  52 /* Get line, including trailing \n if any */
  53 char *xmalloc_fgets(FILE *file)
  54 {
  55         int i;
  56
  57         return bb_get_chunk_from_file(file, &i);
  58 }
  59
  60 /* Get line.  Remove trailing \n */
  61 char *xmalloc_getline(FILE *file)
  62 {
  63         int i;
  64         char *c = bb_get_chunk_from_file(file, &i);
  65
  66         if (i && c[--i] == '\n')
  67                 c[i] = '\0';
  68
  69         return c;
  70 }