Lukas Stockner
2b170f16d6
Instead of handling mmap, compression etc. all directly in readfile.c, refactor the code to use a generic FileReader. This makes it easier to add new compression methods or similar, and allows to reuse the logic in other places (e.g. thumbnail reading). Reviewed By: campbellbarton, brecht, mont29 Differential Revision: https://developer.blender.org/D5799
109 lines
2.7 KiB
C
109 lines
2.7 KiB
C
/*
|
|
* 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.
|
|
*
|
|
* The Original Code is Copyright (C) 2004-2021 Blender Foundation
|
|
* All rights reserved.
|
|
*/
|
|
|
|
/** \file
|
|
* \ingroup bli
|
|
*/
|
|
|
|
#include <zlib.h>
|
|
|
|
#include "BLI_blenlib.h"
|
|
#include "BLI_filereader.h"
|
|
|
|
#include "MEM_guardedalloc.h"
|
|
|
|
typedef struct {
|
|
FileReader reader;
|
|
|
|
FileReader *base;
|
|
|
|
z_stream strm;
|
|
|
|
void *in_buf;
|
|
size_t in_size;
|
|
} GzipReader;
|
|
|
|
static ssize_t gzip_read(FileReader *reader, void *buffer, size_t size)
|
|
{
|
|
GzipReader *gzip = (GzipReader *)reader;
|
|
|
|
gzip->strm.avail_out = size;
|
|
gzip->strm.next_out = buffer;
|
|
|
|
while (gzip->strm.avail_out > 0) {
|
|
if (gzip->strm.avail_in == 0) {
|
|
/* Ran out of buffered input data, read some more. */
|
|
size_t readsize = gzip->base->read(gzip->base, gzip->in_buf, gzip->in_size);
|
|
|
|
if (readsize > 0) {
|
|
/* We got some data, so mark the buffer as refilled. */
|
|
gzip->strm.avail_in = readsize;
|
|
gzip->strm.next_in = gzip->in_buf;
|
|
}
|
|
else {
|
|
/* The underlying file is EOF, so return as much as we can. */
|
|
break;
|
|
}
|
|
}
|
|
|
|
int ret = inflate(&gzip->strm, Z_NO_FLUSH);
|
|
|
|
if (ret != Z_OK && ret != Z_BUF_ERROR) {
|
|
break;
|
|
}
|
|
}
|
|
|
|
ssize_t read_len = size - gzip->strm.avail_out;
|
|
gzip->reader.offset += read_len;
|
|
return read_len;
|
|
}
|
|
|
|
static void gzip_close(FileReader *reader)
|
|
{
|
|
GzipReader *gzip = (GzipReader *)reader;
|
|
|
|
if (inflateEnd(&gzip->strm) != Z_OK) {
|
|
printf("close gzip stream error\n");
|
|
}
|
|
MEM_freeN((void *)gzip->in_buf);
|
|
|
|
gzip->base->close(gzip->base);
|
|
MEM_freeN(gzip);
|
|
}
|
|
|
|
FileReader *BLI_filereader_new_gzip(FileReader *base)
|
|
{
|
|
GzipReader *gzip = MEM_callocN(sizeof(GzipReader), __func__);
|
|
gzip->base = base;
|
|
|
|
if (inflateInit2(&gzip->strm, 16 + MAX_WBITS) != Z_OK) {
|
|
MEM_freeN(gzip);
|
|
return NULL;
|
|
}
|
|
|
|
gzip->in_size = 256 * 2014;
|
|
gzip->in_buf = MEM_mallocN(gzip->in_size, "gzip in buf");
|
|
|
|
gzip->reader.read = gzip_read;
|
|
gzip->reader.seek = NULL;
|
|
gzip->reader.close = gzip_close;
|
|
|
|
return (FileReader *)gzip;
|
|
}
|