1 /*
2  * Copyright (c) 2018 Stefan Sperling <stsp@openbsd.org>
3  *
4  * Permission to use, copy, modify, and distribute this software for any
5  * purpose with or without fee is hereby granted, provided that the above
6  * copyright notice and this permission notice appear in all copies.
7  *
8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15  */
16 
17 struct got_inflate_checksum {
18 	/* If not NULL, mix input bytes into this CRC checksum. */
19 	uint32_t *input_crc;
20 
21 	/* If not NULL, mix input bytes into this SHA1 context. */
22 	SHA1_CTX *input_sha1;
23 
24 	/* If not NULL, mix input bytes into this SHA1 context. */
25 	SHA2_CTX *input_sha256;
26 
27 	/* If not NULL, mix output bytes into this CRC checksum. */
28 	uint32_t *output_crc;
29 
30 	/* If not NULL, mix output bytes into this SHA1 context. */
31 	SHA1_CTX *output_sha1;
32 
33 	/* If not NULL, mix output bytes into this SHA1 context. */
34 	SHA2_CTX *output_sha256;
35 };
36 
37 struct got_inflate_buf {
38 	z_stream z;
39 	uint8_t *inbuf;
40 	size_t inlen;
41 	uint8_t *outbuf;
42 	size_t outlen;
43 	int flags;
44 #define GOT_INFLATE_F_HAVE_MORE		0x01
45 #define GOT_INFLATE_F_OWN_OUTBUF	0x02
46 	struct got_inflate_checksum *csum;
47 };
48 
49 #define GOT_INFLATE_BUFSIZE		32768
50 
51 const struct got_error *got_inflate_init(struct got_inflate_buf *, uint8_t *,
52     size_t, struct got_inflate_checksum *);
53 const struct got_error *got_inflate_read(struct got_inflate_buf *, FILE *,
54     size_t *, size_t *);
55 const struct got_error *got_inflate_read_fd(struct got_inflate_buf *, int,
56     size_t *, size_t *);
57 const struct got_error *got_inflate_read_mmap(struct got_inflate_buf *,
58     uint8_t *, size_t, size_t, size_t *, size_t *);
59 void got_inflate_end(struct got_inflate_buf *);
60 const struct got_error *got_inflate_to_mem(uint8_t **, size_t *, size_t *,
61     struct got_inflate_checksum *, FILE *);
62 const struct got_error *got_inflate_to_mem_fd(uint8_t **, size_t *, size_t *,
63     struct got_inflate_checksum *, size_t, int);
64 const struct got_error *got_inflate_to_mem_mmap(uint8_t **, size_t *, size_t *,
65     struct got_inflate_checksum *, uint8_t *, size_t, size_t);
66 const struct got_error *got_inflate_to_file(size_t *, FILE *,
67     struct got_inflate_checksum *, FILE *);
68 const struct got_error *got_inflate_to_file_fd(size_t *, size_t *,
69     struct got_inflate_checksum *, int, FILE *);
70 const struct got_error *got_inflate_to_fd(size_t *, FILE *,
71     struct got_inflate_checksum *, int);
72 const struct got_error *got_inflate_to_file_mmap(size_t *, size_t *,
73     struct got_inflate_checksum *, uint8_t *, size_t, size_t, FILE *);