/* io_gunzip.c - Alpine Package Keeper (APK) * * Copyright (C) 2008-2011 Timo Teräs * All rights reserved. * * SPDX-License-Identifier: GPL-2.0-only */ #include #include #include #include #include #include #include "apk_defines.h" #include "apk_io.h" struct apk_gzip_istream { struct apk_istream is; struct apk_istream *zis; z_stream zs; apk_multipart_cb cb; void *cbctx; void *cbprev; apk_blob_t cbarg; }; static void gzi_get_meta(struct apk_istream *is, struct apk_file_meta *meta) { struct apk_gzip_istream *gis = container_of(is, struct apk_gzip_istream, is); apk_istream_get_meta(gis->zis, meta); } static int gzi_boundary_change(struct apk_gzip_istream *gis) { int r; if (!gis->cb) return 0; r = gis->cb(gis->cbctx, gis->is.err ? APK_MPART_END : APK_MPART_BOUNDARY, gis->cbarg); if (r > 0) r = -ECANCELED; if (r != 0) gis->is.err = r; return r; } static ssize_t gzi_read(struct apk_istream *is, void *ptr, size_t size) { struct apk_gzip_istream *gis = container_of(is, struct apk_gzip_istream, is); int r; gis->zs.avail_out = size; gis->zs.next_out = ptr; while (gis->zs.avail_out != 0 && gis->is.err == 0) { if (!APK_BLOB_IS_NULL(gis->cbarg)) { if (gzi_boundary_change(gis)) goto ret; gis->cbarg = APK_BLOB_NULL; } if (gis->zs.avail_in == 0) { apk_blob_t blob; if (gis->cb != NULL && gis->cbprev != NULL && gis->cbprev != gis->zs.next_in) { gis->cb(gis->cbctx, APK_MPART_DATA, APK_BLOB_PTR_LEN(gis->cbprev, (void *)gis->zs.next_in - gis->cbprev)); } r = apk_istream_get_all(gis->zis, &blob); gis->cbprev = blob.ptr; gis->zs.avail_in = blob.len; gis->zs.next_in = (void *) gis->cbprev; if (r < 0) { if (r == -APKE_EOF) { gis->is.err = 1; gis->cbarg = APK_BLOB_NULL; gzi_boundary_change(gis); } else { gis->is.err = r; } goto ret; } } r = inflate(&gis->zs, Z_NO_FLUSH); switch (r) { case Z_STREAM_END: /* Digest the inflated bytes */ if (gis->zis->err && gis->zs.avail_in == 0) gis->is.err = gis->zis->err; if (gis->cb != NULL) { gis->cbarg = APK_BLOB_PTR_LEN(gis->cbprev, (void *) gis->zs.next_in - gis->cbprev); gis->cbprev = gis->zs.next_in; } /* If we hit end of the bitstream (not end * of just this gzip), we need to do the * callback here, as we won't be called again. * For boundaries it should be postponed to not * be called until next gzip read is started. */ if (gis->is.err) { gzi_boundary_change(gis); goto ret; } inflateEnd(&gis->zs); if (inflateInit2(&gis->zs, 15+32) != Z_OK) return -ENOMEM; if (gis->cb && gis->zs.avail_out != size) goto ret; break; case Z_OK: break; default: gis->is.err = -EIO; break; } } ret: return size - gis->zs.avail_out; } static int gzi_close(struct apk_istream *is) { int r; struct apk_gzip_istream *gis = container_of(is, struct apk_gzip_istream, is); inflateEnd(&gis->zs); r = apk_istream_close_error(gis->zis, gis->is.err); free(gis); return r; } static const struct apk_istream_ops gunzip_istream_ops = { .get_meta = gzi_get_meta, .read = gzi_read, .close = gzi_close, }; static int window_bits(int window_bits, int raw) { if (raw) return -window_bits; // raw mode return window_bits | 16; // gzip mode } struct apk_istream *apk_istream_zlib(struct apk_istream *is, int raw, apk_multipart_cb cb, void *ctx) { struct apk_gzip_istream *gis; if (IS_ERR_OR_NULL(is)) return ERR_CAST(is); gis = malloc(sizeof(*gis) + apk_io_bufsize); if (!gis) goto err; *gis = (struct apk_gzip_istream) { .is.ops = &gunzip_istream_ops, .is.buf = (uint8_t*)(gis + 1), .is.buf_size = apk_io_bufsize, .zis = is, .cb = cb, .cbctx = ctx, }; if (inflateInit2(&gis->zs, window_bits(15, raw)) != Z_OK) { free(gis); goto err; } return &gis->is; err: return ERR_PTR(apk_istream_close_error(is, -ENOMEM)); } struct apk_gzip_ostream { struct apk_ostream os; struct apk_ostream *output; z_stream zs; }; static int gzo_write(struct apk_ostream *os, const void *ptr, size_t size) { struct apk_gzip_ostream *gos = container_of(os, struct apk_gzip_ostream, os); unsigned char buffer[1024]; ssize_t have, r; gos->zs.avail_in = size; gos->zs.next_in = (void *) ptr; while (gos->zs.avail_in) { gos->zs.avail_out = sizeof(buffer); gos->zs.next_out = buffer; r = deflate(&gos->zs, Z_NO_FLUSH); if (r == Z_STREAM_ERROR) return apk_ostream_cancel(gos->output, -EIO); have = sizeof(buffer) - gos->zs.avail_out; if (have != 0) { r = apk_ostream_write(gos->output, buffer, have); if (r < 0) return r; } } return 0; } static int gzo_close(struct apk_ostream *os) { struct apk_gzip_ostream *gos = container_of(os, struct apk_gzip_ostream, os); unsigned char buffer[1024]; size_t have; int r, rc = os->rc; do { gos->zs.avail_out = sizeof(buffer); gos->zs.next_out = buffer; r = deflate(&gos->zs, Z_FINISH); have = sizeof(buffer) - gos->zs.avail_out; if (apk_ostream_write(gos->output, buffer, have) < 0) break; } while (r == Z_OK); r = apk_ostream_close(gos->output); deflateEnd(&gos->zs); free(gos); return rc ?: r; } static const struct apk_ostream_ops gzip_ostream_ops = { .write = gzo_write, .close = gzo_close, }; struct apk_ostream *apk_ostream_zlib(struct apk_ostream *output, int raw) { struct apk_gzip_ostream *gos; if (IS_ERR_OR_NULL(output)) return ERR_CAST(output); gos = malloc(sizeof(struct apk_gzip_ostream)); if (gos == NULL) goto err; *gos = (struct apk_gzip_ostream) { .os.ops = &gzip_ostream_ops, .output = output, }; if (deflateInit2(&gos->zs, 9, Z_DEFLATED, window_bits(15, raw), 8, Z_DEFAULT_STRATEGY) != Z_OK) { free(gos); goto err; } return &gos->os; err: apk_ostream_close(output); return ERR_PTR(-ENOMEM); }