src/lib/stdio_helpers.c - master - githubmirror/libreport

src/lib/stdio_helpers.c
95 строк2 KB
Michal Fabik
Use g_clear_pointer to free and NULL
22 сен 2020, 15:11
22 сен 2020, 15:11943690a
/*
 * Utility routines.
 *
 * Copyright (C) 2001 Matt Krai
 * Copyright (C) 2004 Erik Andersen <andersen@codepoet.org>
 * Copyright (C) 2005, 2006 Rob Landley <rob@landley.net>
 * Copyright (C) 2010 ABRT Team
 *
 * Copyright (C) 2010  ABRT team
 * Copyright (C) 2010  RedHat Inc
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License along
 * with this program; if not, write to the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 */
#include "internal_libreport.h"

//TODO: add sanitizing upper limit (e.g 64K, 1M, or configurable).
//This is why we don't use GNU's getline: it doesn't have
//any upper sanity bound on line size.

static char *xmalloc_fgets_internal(FILE *file, int *sizep)
{
	unsigned idx = 0;
	char *linebuf = NULL;

	while (1) {
		char *r;

		linebuf = g_realloc(linebuf, idx + 0x100);
		r = fgets(&linebuf[idx], 0x100, file);
		if (!r) {
			/* need to terminate the line */
			linebuf[idx] = '\0';
			break;
		}

		/* stupid. fgets knows the len, it should report it somehow */
		unsigned len = strlen(&linebuf[idx]);

		idx += len;
		if (len < 0xff || linebuf[idx - 1] == '\n')
			break; /* we found \n or EOF */
	}

	*sizep = idx;

	if (!idx) {
		/* The very first fgets returned NULL. It's EOF (or error) */
        g_clear_pointer(&linebuf, g_free);
	}
	return linebuf;
}

char *libreport_xmalloc_fgets(FILE *file)
{
	int sz;
	char *r = xmalloc_fgets_internal(file, &sz);
	if (!r)
		return r;
	return g_realloc(r, sz + 1);
}

char *libreport_xmalloc_fgetline(FILE *file)
{
	int sz;
	char *r = xmalloc_fgets_internal(file, &sz);
	if (!r)
		return r;
	if (r[sz - 1] == '\n')
		r[--sz] = '\0';
	return g_realloc(r, sz + 1);
}

char *libreport_xmalloc_fopen_fgetline_fclose(const char *filename)
{
    char *s = NULL;
    FILE *fp = fopen(filename, "r");
    if (fp)
    {
        s = libreport_xmalloc_fgetline(fp);
        fclose(fp);
    }
    return s;
}