1f9349d42SDag-Erling Smørgrav /*-
2f9349d42SDag-Erling Smørgrav * SPDX-License-Identifier: BSD-2-Clause
3f9349d42SDag-Erling Smørgrav *
4f9349d42SDag-Erling Smørgrav * Copyright (c) 2022 Klara, Inc.
5f9349d42SDag-Erling Smørgrav *
6f9349d42SDag-Erling Smørgrav * Redistribution and use in source and binary forms, with or without
7f9349d42SDag-Erling Smørgrav * modification, are permitted provided that the following conditions
8f9349d42SDag-Erling Smørgrav * are met:
9f9349d42SDag-Erling Smørgrav * 1. Redistributions of source code must retain the above copyright
10f9349d42SDag-Erling Smørgrav * notice, this list of conditions and the following disclaimer.
11f9349d42SDag-Erling Smørgrav * 2. Redistributions in binary form must reproduce the above copyright
12f9349d42SDag-Erling Smørgrav * notice, this list of conditions and the following disclaimer in the
13f9349d42SDag-Erling Smørgrav * documentation and/or other materials provided with the distribution.
14f9349d42SDag-Erling Smørgrav *
15f9349d42SDag-Erling Smørgrav * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
16f9349d42SDag-Erling Smørgrav * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17f9349d42SDag-Erling Smørgrav * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18f9349d42SDag-Erling Smørgrav * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
19f9349d42SDag-Erling Smørgrav * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20f9349d42SDag-Erling Smørgrav * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
21f9349d42SDag-Erling Smørgrav * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22f9349d42SDag-Erling Smørgrav * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23f9349d42SDag-Erling Smørgrav * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24f9349d42SDag-Erling Smørgrav * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
25f9349d42SDag-Erling Smørgrav * SUCH DAMAGE.
26f9349d42SDag-Erling Smørgrav */
27f9349d42SDag-Erling Smørgrav
28f9349d42SDag-Erling Smørgrav /* This file is #included by gzip.c */
29f9349d42SDag-Erling Smørgrav
30f9349d42SDag-Erling Smørgrav static off_t
unzstd(int in,int out,char * pre,size_t prelen,off_t * bytes_in)31f9349d42SDag-Erling Smørgrav unzstd(int in, int out, char *pre, size_t prelen, off_t *bytes_in)
32f9349d42SDag-Erling Smørgrav {
33f9349d42SDag-Erling Smørgrav static char *ibuf, *obuf;
34f9349d42SDag-Erling Smørgrav ZSTD_inBuffer zib;
35f9349d42SDag-Erling Smørgrav ZSTD_outBuffer zob;
36f9349d42SDag-Erling Smørgrav ZSTD_DCtx *zds;
37f9349d42SDag-Erling Smørgrav ssize_t res;
38f9349d42SDag-Erling Smørgrav size_t zres;
39f9349d42SDag-Erling Smørgrav size_t bytes_out = 0;
40f9349d42SDag-Erling Smørgrav int eof = 0;
41f9349d42SDag-Erling Smørgrav
42f9349d42SDag-Erling Smørgrav if (ibuf == NULL)
43f9349d42SDag-Erling Smørgrav ibuf = malloc(BUFLEN);
44f9349d42SDag-Erling Smørgrav if (obuf == NULL)
45f9349d42SDag-Erling Smørgrav obuf = malloc(BUFLEN);
46f9349d42SDag-Erling Smørgrav if (ibuf == NULL || obuf == NULL)
47f9349d42SDag-Erling Smørgrav maybe_err("malloc");
48f9349d42SDag-Erling Smørgrav
49f9349d42SDag-Erling Smørgrav zds = ZSTD_createDStream();
50f9349d42SDag-Erling Smørgrav ZSTD_initDStream(zds);
51f9349d42SDag-Erling Smørgrav
52f9349d42SDag-Erling Smørgrav zib.src = pre;
53f9349d42SDag-Erling Smørgrav zib.size = prelen;
54f9349d42SDag-Erling Smørgrav zib.pos = 0;
55f9349d42SDag-Erling Smørgrav if (bytes_in != NULL)
56f9349d42SDag-Erling Smørgrav *bytes_in = prelen;
57f9349d42SDag-Erling Smørgrav zob.dst = obuf;
58f9349d42SDag-Erling Smørgrav zob.size = BUFLEN;
59f9349d42SDag-Erling Smørgrav zob.pos = 0;
60f9349d42SDag-Erling Smørgrav
61f9349d42SDag-Erling Smørgrav while (!eof) {
62f9349d42SDag-Erling Smørgrav if (zib.pos >= zib.size) {
63f9349d42SDag-Erling Smørgrav res = read(in, ibuf, BUFLEN);
64f9349d42SDag-Erling Smørgrav if (res < 0)
65f9349d42SDag-Erling Smørgrav maybe_err("read");
66f9349d42SDag-Erling Smørgrav if (res == 0)
67f9349d42SDag-Erling Smørgrav eof = 1;
68f9349d42SDag-Erling Smørgrav infile_newdata(res);
69f9349d42SDag-Erling Smørgrav zib.src = ibuf;
70f9349d42SDag-Erling Smørgrav zib.size = res;
71f9349d42SDag-Erling Smørgrav zib.pos = 0;
72f9349d42SDag-Erling Smørgrav if (bytes_in != NULL)
73f9349d42SDag-Erling Smørgrav *bytes_in += res;
74f9349d42SDag-Erling Smørgrav }
75f9349d42SDag-Erling Smørgrav zres = ZSTD_decompressStream(zds, &zob, &zib);
76f9349d42SDag-Erling Smørgrav if (ZSTD_isError(zres)) {
77f9349d42SDag-Erling Smørgrav maybe_errx("%s", ZSTD_getErrorName(zres));
78f9349d42SDag-Erling Smørgrav }
79f9349d42SDag-Erling Smørgrav if (zob.pos > 0) {
80f9349d42SDag-Erling Smørgrav res = write(out, obuf, zob.pos);
81f9349d42SDag-Erling Smørgrav if (res < 0)
82f9349d42SDag-Erling Smørgrav maybe_err("write");
83f9349d42SDag-Erling Smørgrav zob.pos = 0;
84f9349d42SDag-Erling Smørgrav bytes_out += res;
85f9349d42SDag-Erling Smørgrav }
86f9349d42SDag-Erling Smørgrav }
87f9349d42SDag-Erling Smørgrav ZSTD_freeDStream(zds);
88f9349d42SDag-Erling Smørgrav return (bytes_out);
89f9349d42SDag-Erling Smørgrav }
90