memstream.c

   1 /*
   2  * This file is part of ltrace.
   3  * Copyright (C) 2012 Petr Machata, Red Hat Inc.
   4  *
   5  * This program is free software; you can redistribute it and/or
   6  * modify it under the terms of the GNU General Public License as
   7  * published by the Free Software Foundation; either version 2 of the
   8  * License, or (at your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful, but
  11  * WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License
  16  * along with this program; if not, write to the Free Software
  17  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
  18  * 02110-1301 USA
  19  */
  20
  21 /* _GNU_SOURCE may be necessary for open_memstream visibility (see
  22  * configure.ac), and there's no harm defining it just in case.  */
  23 #define _GNU_SOURCE
  24
  25 #include <stdio.h>
  26 #include <stdlib.h>
  27
  28 #include "config.h"
  29 #include "memstream.h"
  30
  31 int
  32 memstream_init(struct memstream *memstream)
  33 {
  34 #if HAVE_OPEN_MEMSTREAM
  35         memstream->stream = open_memstream(&memstream->buf,
  36                                            &memstream->size);
  37 #else
  38         memstream->stream = tmpfile();
  39         memstream->buf = NULL;
  40 #endif
  41         return memstream->stream != NULL ? 0 : -1;
  42 }
  43
  44 int
  45 memstream_close(struct memstream *memstream)
  46 {
  47 #if !defined(HAVE_OPEN_MEMSTREAM) || !HAVE_OPEN_MEMSTREAM
  48         if (fseek(memstream->stream, 0, SEEK_END) < 0) {
  49         fail:
  50                 fclose(memstream->stream);
  51                 return -1;
  52         }
  53         memstream->size = ftell(memstream->stream);
  54         if (memstream->size == (size_t)-1)
  55                 goto fail;
  56         memstream->buf = malloc(memstream->size);
  57         if (memstream->buf == NULL)
  58                 goto fail;
  59
  60         rewind(memstream->stream);
  61         if (fread(memstream->buf, 1, memstream->size, memstream->stream)
  62             < memstream->size)
  63                 goto fail;
  64 #endif
  65
  66         return fclose(memstream->stream) == 0 ? 0 : -1;
  67 }
  68
  69 void
  70 memstream_destroy(struct memstream *memstream)
  71 {
  72         free(memstream->buf);
  73 }