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 <stdio.h>
  13 #include <stdlib.h>
  14 #include "libbb.h"
  15
  16 /* get_line_from_file() - This function reads an entire line from a text file,
  17  * up to a newline or NUL byte.  It returns a malloc'ed char * which must be
  18  * stored and free'ed  by the caller.  If end is null '\n' isn't considered
  19  * and of line.  If end isn't null, length of the chunk read is stored in it. */
  20
  21 char *bb_get_chunk_from_file(FILE *file, int *end)
  22 {
  23         int ch;
  24         int idx = 0;
  25         char *linebuf = NULL;
  26         int linebufsz = 0;
  27
  28         while ((ch = getc(file)) != EOF) {
  29                 /* grow the line buffer as necessary */
  30                 if (idx > linebufsz - 2) {
  31                         linebuf = xrealloc(linebuf, linebufsz += 80);
  32                 }
  33                 linebuf[idx++] = (char)ch;
  34                 if (!ch || (end && ch == '\n')) break;
  35         }
  36         if (end) *end = idx;
  37         if (linebuf) {
  38                 if (ferror(file)) {
  39                         free(linebuf);
  40                         return NULL;
  41                 }
  42                 linebuf[idx] = 0;
  43         }
  44         return linebuf;
  45 }
  46
  47 /* Get line, including trailing /n if any */
  48 char *bb_get_line_from_file(FILE *file)
  49 {
  50         int i;
  51         return bb_get_chunk_from_file(file, &i);
  52 }
  53
  54 /* Get line.  Remove trailing /n */
  55 char *bb_get_chomped_line_from_file(FILE *file)
  56 {
  57         int i;
  58         char *c=bb_get_chunk_from_file(file, &i);
  59         if(i) c[--i]=0;
  60
  61         return c;
  62 }