git.maemo.org Git - busybox4maemo/blob - 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
  16  * must be 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 char *bb_get_chunk_from_file(FILE *file, int *end)
  20 {
  21         int ch;
  22         int idx = 0;
  23         char *linebuf = NULL;
  24         int linebufsz = 0;
  25
  26         while ((ch = getc(file)) != EOF) {
  27                 /* grow the line buffer as necessary */
  28                 if (idx >= linebufsz) {
  29                         linebufsz += 80;
  30                         linebuf = xrealloc(linebuf, linebufsz);
  31                 }
  32                 linebuf[idx++] = (char) ch;
  33                 if (!ch || (end && ch == '\n'))
  34                         break;
  35         }
  36         if (end)
  37                 *end = idx;
  38         if (linebuf) {
  39                 // huh, does fgets discard prior data on error like this?
  40                 // I don't think so....
  41                 //if (ferror(file)) {
  42                 //      free(linebuf);
  43                 //      return NULL;
  44                 //}
  45                 linebuf = xrealloc(linebuf, idx+1);
  46                 linebuf[idx] = '\0';
  47         }
  48         return linebuf;
  49 }
  50
  51 /* Get line, including trailing \n if any */
  52 char *xmalloc_fgets(FILE *file)
  53 {
  54         int i;
  55
  56         return bb_get_chunk_from_file(file, &i);
  57 }
  58
  59 /* Get line.  Remove trailing \n */
  60 char *xmalloc_getline(FILE *file)
  61 {
  62         int i;
  63         char *c = bb_get_chunk_from_file(file, &i);
  64
  65         if (i && c[--i] == '\n')
  66                 c[i] = '\0';
  67
  68         return c;
  69 }