libbb/fgets_str.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * Utility routines.
   4  *
   5  * Copyright (C) many different people.
   6  * If you wrote this, please acknowledge your work.
   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 <stdio.h>
  25 #include <stdlib.h>
  26 #include <string.h>
  27
  28 #include "libbb.h"
  29
  30 /* Read up to (and including) TERMINATING_STRING from FILE and return it.
  31  * Return NULL on EOF.  */
  32
  33 char *fgets_str(FILE *file, const char *terminating_string)
  34 {
  35         char *linebuf = NULL;
  36         const int term_length = strlen(terminating_string);
  37         int end_string_offset;
  38         int linebufsz = 0;
  39         int idx = 0;
  40         int ch;
  41
  42         while (1) {
  43                 ch = fgetc(file);
  44                 if (ch == EOF) {
  45                         free(linebuf);
  46                         return NULL;
  47                 }
  48
  49                 /* grow the line buffer as necessary */
  50                 while (idx > linebufsz - 2) {
  51                         linebuf = xrealloc(linebuf, linebufsz += 1000);
  52                 }
  53
  54                 linebuf[idx] = ch;
  55                 idx++;
  56
  57                 /* Check for terminating string */
  58                 end_string_offset = idx - term_length;
  59                 if ((end_string_offset > 0) && (memcmp(&linebuf[end_string_offset], terminating_string, term_length) == 0)) {
  60                         idx -= term_length;
  61                         break;
  62                 }
  63         }
  64         linebuf[idx] = '\0';
  65         return(linebuf);
  66 }
  67