1 a76a38d9 2018-03-11 stsp /*
  2 a76a38d9 2018-03-11 stsp  * Copyright (c) 2018 Stefan Sperling <stsp@openbsd.org>
  3 a76a38d9 2018-03-11 stsp  *
  4 a76a38d9 2018-03-11 stsp  * Permission to use, copy, modify, and distribute this software for any
  5 a76a38d9 2018-03-11 stsp  * purpose with or without fee is hereby granted, provided that the above
  6 a76a38d9 2018-03-11 stsp  * copyright notice and this permission notice appear in all copies.
  7 a76a38d9 2018-03-11 stsp  *
  8 a76a38d9 2018-03-11 stsp  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
  9 a76a38d9 2018-03-11 stsp  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 10 a76a38d9 2018-03-11 stsp  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 11 a76a38d9 2018-03-11 stsp  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 12 a76a38d9 2018-03-11 stsp  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 13 a76a38d9 2018-03-11 stsp  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 14 a76a38d9 2018-03-11 stsp  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 15 a76a38d9 2018-03-11 stsp  */
 16 a76a38d9 2018-03-11 stsp 
 17 a76a38d9 2018-03-11 stsp #include <sys/queue.h>
 18 a76a38d9 2018-03-11 stsp 
 19 a76a38d9 2018-03-11 stsp #include <stdio.h>
 20 a76a38d9 2018-03-11 stsp #include <stdlib.h>
 21 a76a38d9 2018-03-11 stsp #include <string.h>
 22 a76a38d9 2018-03-11 stsp #include <sha1.h>
 23 a76a38d9 2018-03-11 stsp #include <zlib.h>
 24 a76a38d9 2018-03-11 stsp 
 25 a76a38d9 2018-03-11 stsp #include "got_error.h"
 26 a76a38d9 2018-03-11 stsp #include "got_object.h"
 27 a76a38d9 2018-03-11 stsp 
 28 718b3ab0 2018-03-17 stsp #include "got_lib_path.h"
 29 718b3ab0 2018-03-17 stsp #include "got_lib_zbuf.h"
 30 a76a38d9 2018-03-11 stsp 
 31 a76a38d9 2018-03-11 stsp const struct got_error *
 32 19d747f7 2018-03-16 stsp got_inflate_init(struct got_zstream_buf *zb, uint8_t *outbuf, size_t bufsize)
 33 a76a38d9 2018-03-11 stsp {
 34 a76a38d9 2018-03-11 stsp 	const struct got_error *err = NULL;
 35 a76a38d9 2018-03-11 stsp 
 36 a76a38d9 2018-03-11 stsp 	memset(zb, 0, sizeof(*zb));
 37 a76a38d9 2018-03-11 stsp 
 38 a76a38d9 2018-03-11 stsp 	zb->z.zalloc = Z_NULL;
 39 a76a38d9 2018-03-11 stsp 	zb->z.zfree = Z_NULL;
 40 a76a38d9 2018-03-11 stsp 	if (inflateInit(&zb->z) != Z_OK) {
 41 a76a38d9 2018-03-11 stsp 		err = got_error(GOT_ERR_IO);
 42 a76a38d9 2018-03-11 stsp 		goto done;
 43 a76a38d9 2018-03-11 stsp 	}
 44 a76a38d9 2018-03-11 stsp 
 45 a76a38d9 2018-03-11 stsp 	zb->inlen = zb->outlen = bufsize;
 46 a76a38d9 2018-03-11 stsp 
 47 a76a38d9 2018-03-11 stsp 	zb->inbuf = calloc(1, zb->inlen);
 48 a76a38d9 2018-03-11 stsp 	if (zb->inbuf == NULL) {
 49 0a585a0d 2018-03-17 stsp 		err = got_error_from_errno();
 50 a76a38d9 2018-03-11 stsp 		goto done;
 51 a76a38d9 2018-03-11 stsp 	}
 52 a76a38d9 2018-03-11 stsp 
 53 19d747f7 2018-03-16 stsp 	if (outbuf == NULL) {
 54 19d747f7 2018-03-16 stsp 		zb->outbuf = calloc(1, zb->outlen);
 55 19d747f7 2018-03-16 stsp 		if (zb->outbuf == NULL) {
 56 0a585a0d 2018-03-17 stsp 			err = got_error_from_errno();
 57 19d747f7 2018-03-16 stsp 			goto done;
 58 19d747f7 2018-03-16 stsp 		}
 59 19d747f7 2018-03-16 stsp 		zb->flags |= GOT_ZSTREAM_F_OWN_OUTBUF;
 60 19d747f7 2018-03-16 stsp 	} else
 61 19d747f7 2018-03-16 stsp 		zb->outbuf = outbuf;
 62 a76a38d9 2018-03-11 stsp 
 63 a76a38d9 2018-03-11 stsp done:
 64 a76a38d9 2018-03-11 stsp 	if (err)
 65 a76a38d9 2018-03-11 stsp 		got_inflate_end(zb);
 66 a76a38d9 2018-03-11 stsp 	return err;
 67 a76a38d9 2018-03-11 stsp }
 68 a76a38d9 2018-03-11 stsp 
 69 a76a38d9 2018-03-11 stsp const struct got_error *
 70 a76a38d9 2018-03-11 stsp got_inflate_read(struct got_zstream_buf *zb, FILE *f, size_t *outlenp)
 71 a76a38d9 2018-03-11 stsp {
 72 a76a38d9 2018-03-11 stsp 	size_t last_total_out = zb->z.total_out;
 73 a76a38d9 2018-03-11 stsp 	z_stream *z = &zb->z;
 74 92088384 2018-04-01 stsp 	int ret = Z_ERRNO;
 75 a76a38d9 2018-03-11 stsp 
 76 a76a38d9 2018-03-11 stsp 	z->next_out = zb->outbuf;
 77 a76a38d9 2018-03-11 stsp 	z->avail_out = zb->outlen;
 78 a76a38d9 2018-03-11 stsp 
 79 a76a38d9 2018-03-11 stsp 	*outlenp = 0;
 80 a76a38d9 2018-03-11 stsp 	do {
 81 a76a38d9 2018-03-11 stsp 		if (z->avail_in == 0) {
 82 a76a38d9 2018-03-11 stsp 			size_t n = fread(zb->inbuf, 1, zb->inlen, f);
 83 a76a38d9 2018-03-11 stsp 			if (n == 0) {
 84 a76a38d9 2018-03-11 stsp 				if (ferror(f))
 85 a76a38d9 2018-03-11 stsp 					return got_ferror(f, GOT_ERR_IO);
 86 92088384 2018-04-01 stsp 				/* EOF */
 87 92088384 2018-04-01 stsp 				ret = Z_STREAM_END;
 88 92088384 2018-04-01 stsp 				break;
 89 a76a38d9 2018-03-11 stsp 			}
 90 a76a38d9 2018-03-11 stsp 			z->next_in = zb->inbuf;
 91 a76a38d9 2018-03-11 stsp 			z->avail_in = n;
 92 a76a38d9 2018-03-11 stsp 		}
 93 a76a38d9 2018-03-11 stsp 		ret = inflate(z, Z_SYNC_FLUSH);
 94 a76a38d9 2018-03-11 stsp 	} while (ret == Z_OK && z->avail_out > 0);
 95 a76a38d9 2018-03-11 stsp 
 96 a76a38d9 2018-03-11 stsp 	if (ret == Z_OK) {
 97 a76a38d9 2018-03-11 stsp 		zb->flags |= GOT_ZSTREAM_F_HAVE_MORE;
 98 a76a38d9 2018-03-11 stsp 	} else {
 99 a76a38d9 2018-03-11 stsp 		if (ret != Z_STREAM_END)
100 a76a38d9 2018-03-11 stsp 			return got_error(GOT_ERR_DECOMPRESSION);
101 a76a38d9 2018-03-11 stsp 		zb->flags &= ~GOT_ZSTREAM_F_HAVE_MORE;
102 a76a38d9 2018-03-11 stsp 	}
103 a76a38d9 2018-03-11 stsp 
104 a76a38d9 2018-03-11 stsp 	*outlenp = z->total_out - last_total_out;
105 a76a38d9 2018-03-11 stsp 	return NULL;
106 a76a38d9 2018-03-11 stsp }
107 a76a38d9 2018-03-11 stsp 
108 a76a38d9 2018-03-11 stsp void
109 a76a38d9 2018-03-11 stsp got_inflate_end(struct got_zstream_buf *zb)
110 a76a38d9 2018-03-11 stsp {
111 a76a38d9 2018-03-11 stsp 	free(zb->inbuf);
112 19d747f7 2018-03-16 stsp 	if (zb->flags & GOT_ZSTREAM_F_OWN_OUTBUF)
113 19d747f7 2018-03-16 stsp 		free(zb->outbuf);
114 a76a38d9 2018-03-11 stsp 	inflateEnd(&zb->z);
115 a76a38d9 2018-03-11 stsp }
116 a76a38d9 2018-03-11 stsp 
117 a76a38d9 2018-03-11 stsp const struct got_error *
118 a76a38d9 2018-03-11 stsp got_inflate_to_mem(uint8_t **outbuf, size_t *outlen, FILE *f)
119 a76a38d9 2018-03-11 stsp {
120 a76a38d9 2018-03-11 stsp 	const struct got_error *err;
121 a76a38d9 2018-03-11 stsp 	size_t avail;
122 a76a38d9 2018-03-11 stsp 	struct got_zstream_buf zb;
123 a76a38d9 2018-03-11 stsp 	void *newbuf;
124 a76a38d9 2018-03-11 stsp 
125 19d747f7 2018-03-16 stsp 	*outbuf = calloc(1, GOT_ZSTREAM_BUFSIZE);
126 19d747f7 2018-03-16 stsp 	if (*outbuf == NULL)
127 0a585a0d 2018-03-17 stsp 		return got_error_from_errno();
128 19d747f7 2018-03-16 stsp 	err = got_inflate_init(&zb, *outbuf, GOT_ZSTREAM_BUFSIZE);
129 a76a38d9 2018-03-11 stsp 	if (err)
130 a76a38d9 2018-03-11 stsp 		return err;
131 a76a38d9 2018-03-11 stsp 
132 a76a38d9 2018-03-11 stsp 	*outlen = 0;
133 a76a38d9 2018-03-11 stsp 
134 a76a38d9 2018-03-11 stsp 	do {
135 a76a38d9 2018-03-11 stsp 		err = got_inflate_read(&zb, f, &avail);
136 a76a38d9 2018-03-11 stsp 		if (err)
137 a76a38d9 2018-03-11 stsp 			return err;
138 19d747f7 2018-03-16 stsp 		*outlen += avail;
139 19d747f7 2018-03-16 stsp 		if (zb.flags & GOT_ZSTREAM_F_HAVE_MORE) {
140 19d747f7 2018-03-16 stsp 			newbuf = reallocarray(*outbuf, 1,
141 19d747f7 2018-03-16 stsp 			    *outlen + GOT_ZSTREAM_BUFSIZE);
142 a76a38d9 2018-03-11 stsp 			if (newbuf == NULL) {
143 0a585a0d 2018-03-17 stsp 				err = got_error_from_errno();
144 a76a38d9 2018-03-11 stsp 				free(*outbuf);
145 a76a38d9 2018-03-11 stsp 				*outbuf = NULL;
146 a76a38d9 2018-03-11 stsp 				*outlen = 0;
147 a76a38d9 2018-03-11 stsp 				goto done;
148 a76a38d9 2018-03-11 stsp 			}
149 a76a38d9 2018-03-11 stsp 			*outbuf = newbuf;
150 19d747f7 2018-03-16 stsp 			zb.outbuf = newbuf + *outlen;
151 19d747f7 2018-03-16 stsp 			zb.outlen = GOT_ZSTREAM_BUFSIZE;
152 a76a38d9 2018-03-11 stsp 		}
153 a76a38d9 2018-03-11 stsp 	} while (zb.flags & GOT_ZSTREAM_F_HAVE_MORE);
154 a76a38d9 2018-03-11 stsp 
155 a76a38d9 2018-03-11 stsp done:
156 a76a38d9 2018-03-11 stsp 	got_inflate_end(&zb);
157 a76a38d9 2018-03-11 stsp 	return err;
158 a76a38d9 2018-03-11 stsp }
159 a76a38d9 2018-03-11 stsp 
160 a76a38d9 2018-03-11 stsp const struct got_error *
161 a76a38d9 2018-03-11 stsp got_inflate_to_file(size_t *outlen, FILE *infile, FILE *outfile)
162 a76a38d9 2018-03-11 stsp {
163 a76a38d9 2018-03-11 stsp 	const struct got_error *err;
164 a76a38d9 2018-03-11 stsp 	size_t avail;
165 a76a38d9 2018-03-11 stsp 	struct got_zstream_buf zb;
166 a76a38d9 2018-03-11 stsp 
167 19d747f7 2018-03-16 stsp 	err = got_inflate_init(&zb, NULL, GOT_ZSTREAM_BUFSIZE);
168 a76a38d9 2018-03-11 stsp 	if (err)
169 a76a38d9 2018-03-11 stsp 		goto done;
170 a76a38d9 2018-03-11 stsp 
171 a76a38d9 2018-03-11 stsp 	*outlen = 0;
172 a76a38d9 2018-03-11 stsp 
173 a76a38d9 2018-03-11 stsp 	do {
174 a76a38d9 2018-03-11 stsp 		err = got_inflate_read(&zb, infile, &avail);
175 a76a38d9 2018-03-11 stsp 		if (err)
176 a76a38d9 2018-03-11 stsp 			return err;
177 a76a38d9 2018-03-11 stsp 		if (avail > 0) {
178 a76a38d9 2018-03-11 stsp 			size_t n;
179 a76a38d9 2018-03-11 stsp 			n = fwrite(zb.outbuf, avail, 1, outfile);
180 a76a38d9 2018-03-11 stsp 			if (n != 1) {
181 a76a38d9 2018-03-11 stsp 				err = got_ferror(outfile, GOT_ERR_IO);
182 a76a38d9 2018-03-11 stsp 				goto done;
183 a76a38d9 2018-03-11 stsp 			}
184 a76a38d9 2018-03-11 stsp 			*outlen += avail;
185 a76a38d9 2018-03-11 stsp 		}
186 a76a38d9 2018-03-11 stsp 	} while (zb.flags & GOT_ZSTREAM_F_HAVE_MORE);
187 a76a38d9 2018-03-11 stsp 
188 a76a38d9 2018-03-11 stsp done:
189 a76a38d9 2018-03-11 stsp 	if (err == NULL)
190 a76a38d9 2018-03-11 stsp 		rewind(outfile);
191 a76a38d9 2018-03-11 stsp 	got_inflate_end(&zb);
192 a76a38d9 2018-03-11 stsp 	return err;
193 a76a38d9 2018-03-11 stsp }