commit - 3b488fdeec85f8d497d8f35b86efc71f42c26342
commit + 9249e7e337edcfd344dc416d11ad4274c90a02f5
blob - 8a4452f74e19fbe55d28a95e4c184b9873ff920c
blob + 1c89bfa953de956bb2865877c48a43e95df37e3b
--- lib/deflate.c
+++ lib/deflate.c
return NULL;
}
-const struct got_error *
-got_deflate_read_mmap(struct got_deflate_buf *zb, uint8_t *map, size_t offset,
- size_t len, size_t *outlenp, size_t *consumed)
+static const struct got_error *
+deflate_read_mmap(struct got_deflate_buf *zb, uint8_t *map, size_t offset,
+ size_t len, size_t *outlenp, size_t *consumed, int flush_on_eof)
{
z_stream *z = &zb->z;
size_t last_total_out = z->total_out;
z->avail_in = len - *consumed;
if (z->avail_in == 0) {
/* EOF */
- ret = deflate(z, Z_FINISH);
+ if (flush_on_eof)
+ ret = deflate(z, Z_FINISH);
break;
}
}
}
*outlenp = z->total_out - last_total_out;
+ return NULL;
+}
+
+const struct got_error *
+got_deflate_read_mmap(struct got_deflate_buf *zb, uint8_t *map, size_t offset,
+ size_t len, size_t *outlenp, size_t *consumed)
+{
+ return deflate_read_mmap(zb, map, offset, len, outlenp, consumed, 1);
+}
+
+const struct got_error *
+got_deflate_flush(struct got_deflate_buf *zb, FILE *outfile,
+ struct got_deflate_checksum *csum, off_t *outlenp)
+{
+ int ret;
+ size_t n;
+ z_stream *z = &zb->z;
+
+ if (z->avail_in != 0)
+ return got_error_msg(GOT_ERR_COMPRESSION,
+ "cannot flush zb with pending input data");
+
+ do {
+ size_t avail, last_total_out = zb->z.total_out;
+
+ z->next_out = zb->outbuf;
+ z->avail_out = zb->outlen;
+
+ ret = deflate(z, Z_FINISH);
+ if (ret != Z_STREAM_END && ret != Z_OK)
+ return got_error(GOT_ERR_COMPRESSION);
+
+ avail = z->total_out - last_total_out;
+ if (avail > 0) {
+ n = fwrite(zb->outbuf, avail, 1, outfile);
+ if (n != 1)
+ return got_ferror(outfile, GOT_ERR_IO);
+ if (csum)
+ csum_output(csum, zb->outbuf, avail);
+ if (outlenp)
+ *outlenp += avail;
+ }
+ } while (ret != Z_STREAM_END);
+
+ zb->flags &= ~GOT_DEFLATE_F_HAVE_MORE;
return NULL;
}
*outlen += avail;
}
} while (zb.flags & GOT_DEFLATE_F_HAVE_MORE);
+
+done:
+ got_deflate_end(&zb);
+ return err;
+}
+
+const struct got_error *
+got_deflate_append_to_file_mmap(struct got_deflate_buf *zb, off_t *outlen,
+ uint8_t *map, size_t offset, size_t len, FILE *outfile,
+ struct got_deflate_checksum *csum)
+{
+ const struct got_error *err;
+ size_t avail, consumed;
+
+ do {
+ err = deflate_read_mmap(zb, map, offset, len, &avail,
+ &consumed, 0);
+ if (err)
+ break;
+ offset += consumed;
+ len -= consumed;
+ if (avail > 0) {
+ size_t n;
+ n = fwrite(zb->outbuf, avail, 1, outfile);
+ if (n != 1) {
+ err = got_ferror(outfile, GOT_ERR_IO);
+ break;
+ }
+ if (csum)
+ csum_output(csum, zb->outbuf, avail);
+ if (outlen)
+ *outlen += avail;
+ }
+ } while ((zb->flags & GOT_DEFLATE_F_HAVE_MORE) && len > 0);
+ return err;
+}
+
+const struct got_error *
+got_deflate_to_mem_mmap(uint8_t **outbuf, size_t *outlen,
+ size_t *consumed_total, struct got_deflate_checksum *csum, uint8_t *map,
+ size_t offset, size_t len)
+{
+ const struct got_error *err;
+ size_t avail, consumed;
+ struct got_deflate_buf zb;
+ void *newbuf;
+ size_t nbuf = 1;
+
+ if (outbuf) {
+ *outbuf = malloc(GOT_DEFLATE_BUFSIZE);
+ if (*outbuf == NULL)
+ return got_error_from_errno("malloc");
+ err = got_deflate_init(&zb, *outbuf, GOT_DEFLATE_BUFSIZE);
+ if (err) {
+ free(*outbuf);
+ *outbuf = NULL;
+ return err;
+ }
+ } else {
+ err = got_deflate_init(&zb, NULL, GOT_DEFLATE_BUFSIZE);
+ if (err)
+ return err;
+ }
+
+ *outlen = 0;
+ if (consumed_total)
+ *consumed_total = 0;
+ do {
+ err = got_deflate_read_mmap(&zb, map, offset, len, &avail,
+ &consumed);
+ if (err)
+ goto done;
+ offset += consumed;
+ if (consumed_total)
+ *consumed_total += consumed;
+ len -= consumed;
+ if (avail > 0 && csum)
+ csum_output(csum, zb.outbuf, avail);
+ *outlen += avail;
+ if ((zb.flags & GOT_DEFLATE_F_HAVE_MORE) && outbuf != NULL) {
+ newbuf = reallocarray(*outbuf, ++nbuf,
+ GOT_DEFLATE_BUFSIZE);
+ if (newbuf == NULL) {
+ err = got_error_from_errno("reallocarray");
+ free(*outbuf);
+ *outbuf = NULL;
+ *outlen = 0;
+ goto done;
+ }
+ *outbuf = newbuf;
+ zb.outbuf = newbuf + *outlen;
+ zb.outlen = (nbuf * GOT_DEFLATE_BUFSIZE) - *outlen;
+ }
+ } while (zb.flags & GOT_DEFLATE_F_HAVE_MORE);
done:
got_deflate_end(&zb);
return err;
blob - 1c429af85a27229451e05798e9511d0a8f474968
blob + 09a8755cf062db2ffd1f7e2c26deef470dbba250
--- lib/got_lib_deflate.h
+++ lib/got_lib_deflate.h
size_t);
const struct got_error *got_deflate_read(struct got_deflate_buf *, FILE *,
off_t, size_t *, off_t *);
+const struct got_error *got_deflate_read_mmap(struct got_deflate_buf *,
+ uint8_t *, size_t, size_t, size_t *, size_t *);
void got_deflate_end(struct got_deflate_buf *);
const struct got_error *got_deflate_to_file(off_t *, FILE *, off_t, FILE *,
struct got_deflate_checksum *);
const struct got_error *got_deflate_to_file_mmap(off_t *, uint8_t *,
size_t, size_t, FILE *, struct got_deflate_checksum *);
+const struct got_error *got_deflate_flush(struct got_deflate_buf *, FILE *,
+ struct got_deflate_checksum *, off_t *);
+const struct got_error *got_deflate_append_to_file_mmap(
+ struct got_deflate_buf *, off_t *, uint8_t *, size_t, size_t, FILE *,
+ struct got_deflate_checksum *);
+const struct got_error *got_deflate_to_mem_mmap(uint8_t **, size_t *, size_t *,
+ struct got_deflate_checksum *, uint8_t *, size_t, size_t);
blob - 6af8d574c7b345c52d3e0c19759bf3ae6bd62b20
blob + 4bbe44dda07c97ba6f0ef878da0caadbe6de9741
--- lib/got_lib_object.h
+++ lib/got_lib_object.h
struct got_object_id *, struct got_pack *, struct got_packidx *, int,
struct got_repository *);
const struct got_error *got_object_read_raw_delta(uint64_t *, uint64_t *,
- off_t *, off_t *, off_t *, struct got_object_id **, int,
+ off_t *, off_t *, off_t *, off_t *, struct got_object_id **, int,
struct got_packidx *, int, struct got_object_id *, struct got_repository *);
const struct got_error *got_object_read_header_privsep(struct got_object **,
struct got_object_id *, struct got_repository *, int);
blob - e8fb373e287ee80486d50ed07964d9d39924308d
blob + 6a3d3981c9afd96d48ef7746b2d0b1d78793a7ca
--- lib/got_lib_pack.h
+++ lib/got_lib_pack.h
const struct got_error *got_packfile_extract_object_to_mem(uint8_t **, size_t *,
struct got_object *, struct got_pack *);
const struct got_error *got_packfile_extract_raw_delta(uint8_t **, size_t *,
- off_t *, off_t *, struct got_object_id *, uint64_t *, uint64_t *,
+ size_t *, off_t *, off_t *, struct got_object_id *, uint64_t *, uint64_t *,
struct got_pack *, struct got_packidx *, int);
struct got_pack *got_repo_get_cached_pack(struct got_repository *,
const char *);
blob - 94dfb7f11c1e4edb26fbc644468d4a6943aeb167
blob + 8caa03b1bcf169d4123aaf0684d047db533c960d
--- lib/got_lib_privsep.h
+++ lib/got_lib_privsep.h
uint64_t base_size;
uint64_t result_size;
off_t delta_size;
+ off_t delta_compressed_size;
off_t delta_offset;
off_t delta_out_offset;
struct got_object_id *);
const struct got_error *got_privsep_send_raw_delta_outfd(struct imsgbuf *, int);
const struct got_error *got_privsep_send_raw_delta(struct imsgbuf *, uint64_t,
- uint64_t, off_t, off_t, off_t, struct got_object_id *);
+ uint64_t, off_t, off_t, off_t, off_t, struct got_object_id *);
const struct got_error *got_privsep_recv_raw_delta(uint64_t *, uint64_t *,
- off_t *, off_t *, off_t *, struct got_object_id **, struct imsgbuf *);
+ off_t *, off_t *, off_t *, off_t *, struct got_object_id **,
+ struct imsgbuf *);
void got_privsep_exec_child(int[2], const char *, const char *);
blob - f668ac5f7e715861d4f32a7ae5122addde9dd770
blob + 4abccb03e823857f8e0c3311ff95cf13c39c9a6c
--- lib/object.c
+++ lib/object.c
const struct got_error *
got_object_read_raw_delta(uint64_t *base_size, uint64_t *result_size,
- off_t *delta_size, off_t *delta_offset, off_t *delta_out_offset,
- struct got_object_id **base_id, int delta_cache_fd,
+ off_t *delta_size, off_t *delta_compressed_size, off_t *delta_offset,
+ off_t *delta_out_offset, struct got_object_id **base_id, int delta_cache_fd,
struct got_packidx *packidx, int obj_idx, struct got_object_id *id,
struct got_repository *repo)
{
*base_size = 0;
*result_size = 0;
*delta_size = 0;
+ *delta_compressed_size = 0;
*delta_offset = 0;
*delta_out_offset = 0;
return err;
return got_privsep_recv_raw_delta(base_size, result_size, delta_size,
- delta_offset, delta_out_offset, base_id, pack->privsep_child->ibuf);
+ delta_compressed_size, delta_offset, delta_out_offset, base_id,
+ pack->privsep_child->ibuf);
}
static const struct got_error *
blob - b7e2f83cb674abec61afb336d07774b71385fdab
blob + 17625d70bff605e978693087be58f33be9ae31e7
--- lib/pack.c
+++ lib/pack.c
static const struct got_error *
read_delta_data(uint8_t **delta_buf, size_t *delta_len,
- size_t delta_data_offset, struct got_pack *pack)
+ size_t *delta_compressed_len, size_t delta_data_offset,
+ struct got_pack *pack)
{
const struct got_error *err = NULL;
+ size_t consumed = 0;
if (pack->map) {
if (delta_data_offset >= pack->filesize)
return got_error(GOT_ERR_PACK_OFFSET);
err = got_inflate_to_mem_mmap(delta_buf, delta_len,
- NULL, NULL, pack->map, delta_data_offset,
+ &consumed, NULL, pack->map, delta_data_offset,
pack->filesize - delta_data_offset);
+ if (err)
+ return err;
} else {
if (lseek(pack->fd, delta_data_offset, SEEK_SET) == -1)
return got_error_from_errno("lseek");
- err = got_inflate_to_mem_fd(delta_buf, delta_len, NULL,
- NULL, 0, pack->fd);
+ err = got_inflate_to_mem_fd(delta_buf, delta_len,
+ &consumed, NULL, 0, pack->fd);
+ if (err)
+ return err;
}
- return err;
+
+ if (delta_compressed_len)
+ *delta_compressed_len = consumed;
+
+ return NULL;
}
static const struct got_error *
if (delta_buf == NULL) {
cached = 0;
err = read_delta_data(&delta_buf, &delta_len,
- delta->data_offset, pack);
+ NULL, delta->data_offset, pack);
if (err)
return err;
err = got_delta_cache_add(pack->delta_cache,
pack->delta_cache, delta->data_offset);
if (delta_buf == NULL) {
cached = 0;
- err = read_delta_data(&delta_buf, &delta_len,
+ err = read_delta_data(&delta_buf, &delta_len, NULL,
delta->data_offset, pack);
if (err)
goto done;
pack->delta_cache, delta->data_offset);
if (delta_buf == NULL) {
cached = 0;
- err = read_delta_data(&delta_buf, &delta_len,
+ err = read_delta_data(&delta_buf, &delta_len, NULL,
delta->data_offset, pack);
if (err)
goto done;
return err;
}
+static const struct got_error *
+read_raw_delta_data(uint8_t **delta_buf, size_t *delta_len,
+ size_t *delta_len_compressed, uint64_t *base_size, uint64_t *result_size,
+ off_t delta_data_offset, struct got_pack *pack, struct got_packidx *packidx)
+{
+ const struct got_error *err = NULL;
+
+ /* Validate decompression and obtain the decompressed size. */
+ err = read_delta_data(delta_buf, delta_len, delta_len_compressed,
+ delta_data_offset, pack);
+ if (err)
+ return err;
+
+ /* Read delta base/result sizes from head of delta stream. */
+ err = got_delta_get_sizes(base_size, result_size,
+ *delta_buf, *delta_len);
+ if (err)
+ goto done;
+
+ /* Discard decompressed delta and read it again in compressed form. */
+ free(*delta_buf);
+ *delta_buf = malloc(*delta_len_compressed);
+ if (*delta_buf == NULL) {
+ err = got_error_from_errno("malloc");
+ goto done;
+ }
+ if (pack->map) {
+ if (delta_data_offset >= pack->filesize)
+ err = got_error(GOT_ERR_PACK_OFFSET);
+ memcpy(*delta_buf, pack->map + delta_data_offset,
+ *delta_len_compressed);
+ } else {
+ ssize_t n;
+ if (lseek(pack->fd, delta_data_offset, SEEK_SET) == -1) {
+ err = got_error_from_errno("lseek");
+ goto done;
+ }
+ n = read(pack->fd, *delta_buf, *delta_len_compressed);
+ if (n < 0) {
+ err = got_error_from_errno("read");
+ goto done;
+ } else if (n != *delta_len_compressed) {
+ err = got_error(GOT_ERR_IO);
+ goto done;
+ }
+ }
+done:
+ if (err) {
+ free(*delta_buf);
+ *delta_buf = NULL;
+ *delta_len = 0;
+ *delta_len_compressed = 0;
+ *base_size = 0;
+ *result_size = 0;
+ }
+ return err;
+}
+
const struct got_error *
got_packfile_extract_raw_delta(uint8_t **delta_buf, size_t *delta_size,
- off_t *delta_offset, off_t *base_offset, struct got_object_id *base_id,
- uint64_t *base_size, uint64_t *result_size, struct got_pack *pack,
- struct got_packidx *packidx, int idx)
+ size_t *delta_compressed_size, off_t *delta_offset, off_t *base_offset,
+ struct got_object_id *base_id, uint64_t *base_size, uint64_t *result_size,
+ struct got_pack *pack, struct got_packidx *packidx, int idx)
{
const struct got_error *err = NULL;
off_t offset;
uint8_t type;
uint64_t size;
size_t tslen, delta_hdrlen;
+ off_t delta_data_offset;
*delta_buf = NULL;
*delta_size = 0;
+ *delta_compressed_size = 0;
*delta_offset = 0;
*base_offset = 0;
*base_size = 0;
offset + delta_hdrlen < delta_hdrlen)
return got_error(GOT_ERR_BAD_DELTA);
- err = read_delta_data(delta_buf, delta_size,
- offset + tslen + delta_hdrlen, pack);
+ delta_data_offset = offset + tslen + delta_hdrlen;
+ err = read_raw_delta_data(delta_buf, delta_size, delta_compressed_size,
+ base_size, result_size, delta_data_offset, pack, packidx);
if (err)
return err;
goto done;
}
- err = got_delta_get_sizes(base_size, result_size, *delta_buf, size);
- if (err)
- goto done;
-
*delta_offset = offset;
done:
if (err) {
free(*delta_buf);
*delta_buf = NULL;
+ *delta_size = 0;
+ *delta_compressed_size = 0;
+ *delta_offset = 0;
+ *base_offset = 0;
+ *base_size = 0;
+ *result_size = 0;
}
return err;
}
blob - ce750b37ec0e7263ed3b1588a807f07c815a249e
blob + 910146ecf61e05093ba959229b2b28e631416d26
--- lib/pack_create.c
+++ lib/pack_create.c
#include "got_lib_privsep.h"
#include "got_lib_repository.h"
#include "got_lib_ratelimit.h"
+#include "got_lib_inflate.h"
#ifndef MIN
#define MIN(_a,_b) ((_a) < (_b) ? (_a) : (_b))
/* The best delta we picked */
struct got_pack_meta *head;
struct got_pack_meta *prev;
- unsigned char *delta_buf; /* if not encoded in delta cache file */
- off_t delta_offset; /* offset in delta cache file */
+ unsigned char *delta_buf; /* if encoded in memory (compressed) */
+ off_t delta_offset; /* offset in delta cache file (compressed) */
off_t delta_len; /* encoded delta length */
+ off_t delta_compressed_len; /* encoded+compressed delta length */
int nchain;
int have_reused_delta;
const struct got_error *err;
unsigned char buf[16], *bp;
int i, j;
- size_t len = 0;
+ size_t len = 0, compressed_len;
+ off_t bufsize = delta_size;
off_t n;
struct got_delta_instruction *d;
+ uint8_t *delta_buf;
- m->delta_buf = malloc(delta_size);
- if (m->delta_buf == NULL)
- return got_error_from_errno("calloc");
+ delta_buf = malloc(bufsize);
+ if (delta_buf == NULL)
+ return got_error_from_errno("malloc");
/* base object size */
buf[0] = base_size & GOT_DELTA_SIZE_VAL_MASK;
buf[i] = n & GOT_DELTA_SIZE_VAL_MASK;
n >>= GOT_DELTA_SIZE_SHIFT;
}
- err = append(&m->delta_buf, &len, &delta_size, buf, i);
+ err = append(&delta_buf, &len, &bufsize, buf, i);
if (err)
- return err;
+ goto done;
/* target object size */
buf[0] = o->size & GOT_DELTA_SIZE_VAL_MASK;
buf[i] = n & GOT_DELTA_SIZE_VAL_MASK;
n >>= GOT_DELTA_SIZE_SHIFT;
}
- err = append(&m->delta_buf, &len, &delta_size, buf, i);
+ err = append(&delta_buf, &len, &bufsize, buf, i);
if (err)
- return err;
+ goto done;
for (j = 0; j < ndeltas; j++) {
d = &deltas[j];
n >>= 8;
}
}
- err = append(&m->delta_buf, &len, &delta_size,
+ err = append(&delta_buf, &len, &bufsize,
buf, bp - buf);
if (err)
- return err;
+ goto done;
} else if (o->f == NULL) {
n = 0;
while (n != d->len) {
buf[0] = (d->len - n < 127) ? d->len - n : 127;
- err = append(&m->delta_buf, &len, &delta_size,
+ err = append(&delta_buf, &len, &bufsize,
buf, 1);
if (err)
- return err;
- err = append(&m->delta_buf, &len, &delta_size,
+ goto done;
+ err = append(&delta_buf, &len, &bufsize,
o->data + o->hdrlen + d->offset + n,
buf[0]);
if (err)
- return err;
+ goto done;
n += buf[0];
}
} else {
char content[128];
size_t r;
- if (fseeko(o->f, o->hdrlen + d->offset, SEEK_SET) == -1)
- return got_error_from_errno("fseeko");
+ if (fseeko(o->f, o->hdrlen + d->offset, SEEK_SET) == -1) {
+ err = got_error_from_errno("fseeko");
+ goto done;
+ }
n = 0;
while (n != d->len) {
buf[0] = (d->len - n < 127) ? d->len - n : 127;
- err = append(&m->delta_buf, &len, &delta_size,
+ err = append(&delta_buf, &len, &bufsize,
buf, 1);
if (err)
- return err;
+ goto done;
r = fread(content, 1, buf[0], o->f);
- if (r != buf[0])
- return got_ferror(o->f, GOT_ERR_IO);
- err = append(&m->delta_buf, &len, &delta_size,
+ if (r != buf[0]) {
+ err = got_ferror(o->f, GOT_ERR_IO);
+ goto done;
+ }
+ err = append(&delta_buf, &len, &bufsize,
content, buf[0]);
if (err)
- return err;
+ goto done;
n += buf[0];
}
}
}
+ err = got_deflate_to_mem_mmap(&m->delta_buf, &compressed_len,
+ NULL, NULL, delta_buf, 0, len);
+ if (err)
+ goto done;
+
m->delta_len = len;
- return NULL;
+ m->delta_compressed_len = compressed_len;
+done:
+ free(delta_buf);
+ return err;
}
static const struct got_error *
struct got_delta_instruction *deltas, int ndeltas,
off_t base_size, FILE *f)
{
+ const struct got_error *err;
unsigned char buf[16], *bp;
int i, j;
off_t n;
- size_t w;
+ struct got_deflate_buf zb;
struct got_delta_instruction *d;
+ off_t delta_len = 0, compressed_len = 0;
+ err = got_deflate_init(&zb, NULL, GOT_DEFLATE_BUFSIZE);
+ if (err)
+ return err;
+
/* base object size */
buf[0] = base_size & GOT_DELTA_SIZE_VAL_MASK;
n = base_size >> GOT_DELTA_SIZE_SHIFT;
buf[i] = n & GOT_DELTA_SIZE_VAL_MASK;
n >>= GOT_DELTA_SIZE_SHIFT;
}
- w = fwrite(buf, 1, i, f);
- if (w != i)
- return got_ferror(f, GOT_ERR_IO);
+ err = got_deflate_append_to_file_mmap(&zb, &compressed_len,
+ buf, 0, i, f, NULL);
+ if (err)
+ goto done;
+ delta_len += i;
+
/* target object size */
buf[0] = o->size & GOT_DELTA_SIZE_VAL_MASK;
n = o->size >> GOT_DELTA_SIZE_SHIFT;
buf[i] = n & GOT_DELTA_SIZE_VAL_MASK;
n >>= GOT_DELTA_SIZE_SHIFT;
}
- w = fwrite(buf, 1, i, f);
- if (w != i)
- return got_ferror(f, GOT_ERR_IO);
+ err = got_deflate_append_to_file_mmap(&zb, &compressed_len,
+ buf, 0, i, f, NULL);
+ if (err)
+ goto done;
+ delta_len += i;
+
for (j = 0; j < ndeltas; j++) {
d = &deltas[j];
if (d->copy) {
if (n == 0)
break;
}
-
n = d->len;
if (n != GOT_DELTA_COPY_DEFAULT_LEN) {
/* DELTA_COPY_LEN1 ... DELTA_COPY_LEN3 */
n >>= 8;
}
}
- w = fwrite(buf, 1, bp - buf, f);
- if (w != bp - buf)
- return got_ferror(f, GOT_ERR_IO);
+ err = got_deflate_append_to_file_mmap(&zb,
+ &compressed_len, buf, 0, bp - buf, f, NULL);
+ if (err)
+ goto done;
+ delta_len += (bp - buf);
} else if (o->f == NULL) {
n = 0;
while (n != d->len) {
buf[0] = (d->len - n < 127) ? d->len - n : 127;
- w = fwrite(buf, 1, 1, f);
- if (w != 1)
- return got_ferror(f, GOT_ERR_IO);
- w = fwrite(o->data + o->hdrlen + d->offset + n,
- 1, buf[0], f);
- if (w != buf[0])
- return got_ferror(f, GOT_ERR_IO);
+ err = got_deflate_append_to_file_mmap(&zb,
+ &compressed_len, buf, 0, 1, f, NULL);
+ if (err)
+ goto done;
+ delta_len++;
+ err = got_deflate_append_to_file_mmap(&zb,
+ &compressed_len,
+ o->data + o->hdrlen + d->offset + n, 0,
+ buf[0], f, NULL);
+ if (err)
+ goto done;
+ delta_len += buf[0];
n += buf[0];
}
} else {
char content[128];
size_t r;
- if (fseeko(o->f, o->hdrlen + d->offset, SEEK_SET) == -1)
- return got_error_from_errno("fseeko");
+ if (fseeko(o->f, o->hdrlen + d->offset, SEEK_SET) == -1) {
+ err = got_error_from_errno("fseeko");
+ goto done;
+ }
n = 0;
while (n != d->len) {
buf[0] = (d->len - n < 127) ? d->len - n : 127;
- w = fwrite(buf, 1, 1, f);
- if (w != 1)
- return got_ferror(f, GOT_ERR_IO);
+ err = got_deflate_append_to_file_mmap(&zb,
+ &compressed_len, buf, 0, 1, f, NULL);
+ if (err)
+ goto done;
+ delta_len++;
r = fread(content, 1, buf[0], o->f);
- if (r != buf[0])
- return got_ferror(o->f, GOT_ERR_IO);
- w = fwrite(content, 1, buf[0], f);
- if (w != buf[0])
- return got_ferror(f, GOT_ERR_IO);
+ if (r != buf[0]) {
+ err = got_ferror(o->f, GOT_ERR_IO);
+ goto done;
+ }
+ err = got_deflate_append_to_file_mmap(&zb,
+ &compressed_len, content, 0, buf[0], f,
+ NULL);
+ if (err)
+ goto done;
+ delta_len += buf[0];
n += buf[0];
}
}
}
- m->delta_len = ftello(f) - m->delta_offset;
- return NULL;
+ err = got_deflate_flush(&zb, f, NULL, &compressed_len);
+ if (err)
+ goto done;
+
+ /* sanity check */
+ if (compressed_len != ftello(f) - m->delta_offset) {
+ err = got_error(GOT_ERR_COMPRESSION);
+ goto done;
+ }
+
+ m->delta_len = delta_len;
+ m->delta_compressed_len = compressed_len;
+done:
+ got_deflate_end(&zb);
+ return err;
}
static const struct got_error *
const struct got_error *err = NULL;
struct got_pack_meta *base = NULL;
struct got_object_id *base_obj_id = NULL;
- off_t delta_len = 0, delta_offset = 0, delta_cache_offset = 0;
+ off_t delta_len = 0, delta_compressed_len = 0;
+ off_t delta_offset = 0, delta_cache_offset = 0;
uint64_t base_size, result_size;
if (m->have_reused_delta)
return NULL;
err = got_object_read_raw_delta(&base_size, &result_size, &delta_len,
- &delta_offset, &delta_cache_offset, &base_obj_id, delta_cache_fd,
- packidx, idx, &m->id, repo);
+ &delta_compressed_len, &delta_offset, &delta_cache_offset,
+ &base_obj_id, delta_cache_fd, packidx, idx, &m->id, repo);
if (err)
return err;
goto done;
m->delta_len = delta_len;
+ m->delta_compressed_len = delta_compressed_len;
m->delta_offset = delta_cache_offset;
m->prev = base;
m->size = result_size;
best_ndeltas, best_size, m->prev->size);
} else {
m->delta_offset = ftello(delta_cache);
- /*
- * TODO:
- * Storing compressed delta data in the delta
- * cache file would probably be more efficient
- * than writing uncompressed delta data here
- * and compressing it while writing the pack
- * file. This would also allow for reusing
- * deltas in their compressed form.
- */
err = encode_delta(m, raw, best_deltas,
best_ndeltas, m->prev->size, delta_cache);
}
}
const struct got_error *
-hwrite(FILE *f, void *buf, int len, SHA1_CTX *ctx)
+hwrite(FILE *f, void *buf, off_t len, SHA1_CTX *ctx)
{
size_t n;
n = fwrite(buf, 1, len, f);
if (n != len)
return got_ferror(f, GOT_ERR_IO);
+ return NULL;
+}
+
+const struct got_error *
+hcopy(FILE *fsrc, FILE *fdst, off_t len, SHA1_CTX *ctx)
+{
+ unsigned char buf[65536];
+ off_t remain = len;
+ size_t n;
+
+ while (remain > 0) {
+ size_t copylen = MIN(sizeof(buf), remain);
+ n = fread(buf, 1, copylen, fsrc);
+ if (n != copylen)
+ return got_ferror(fsrc, GOT_ERR_IO);
+ SHA1Update(ctx, buf, copylen);
+ n = fwrite(buf, 1, copylen, fdst);
+ if (n != copylen)
+ return got_ferror(fdst, GOT_ERR_IO);
+ remain -= copylen;
+ }
+
return NULL;
}
err = deltahdr(packfile_size, ctx, packfile, m);
if (err)
goto done;
- err = got_deflate_to_file_mmap(&outlen,
- m->delta_buf, 0, m->delta_len, packfile, &csum);
+ err = hwrite(packfile, m->delta_buf,
+ m->delta_compressed_len, ctx);
if (err)
goto done;
- *packfile_size += outlen;
+ *packfile_size += m->delta_compressed_len;
free(m->delta_buf);
m->delta_buf = NULL;
} else {
err = deltahdr(packfile_size, ctx, packfile, m);
if (err)
goto done;
- err = got_deflate_to_file(&outlen, delta_cache,
- m->delta_len, packfile, &csum);
+ err = hcopy(delta_cache, packfile,
+ m->delta_compressed_len, ctx);
if (err)
goto done;
- *packfile_size += outlen;
+ *packfile_size += m->delta_compressed_len;
}
done:
if (raw)
progress_cb, progress_arg, &rl, cancel_cb, cancel_arg);
if (err)
goto done;
- if (fseeko(delta_cache, 0L, SEEK_SET) == -1) {
- err = got_error_from_errno("fseeko");
- goto done;
- }
}
+ if (fflush(delta_cache) == EOF) {
+ err = got_error_from_errno("fflush");
+ goto done;
+ }
err = genpack(packsha1, packfile, delta_cache, deltify.meta,
deltify.nmeta, reuse.meta, reuse.nmeta, ncolored, nfound, ntrees,
nours, repo, progress_cb, progress_arg, &rl,
blob - 6b04a1c0d62f06e7f81638f58bcb0509c7292087
blob + c2b79e435f47b4ae5ba22582b8c451cb7d5ddfd2
--- lib/privsep.c
+++ lib/privsep.c
const struct got_error *
got_privsep_send_raw_delta(struct imsgbuf *ibuf, uint64_t base_size,
- uint64_t result_size, off_t delta_size, off_t delta_offset,
- off_t delta_out_offset, struct got_object_id *base_id)
+ uint64_t result_size, off_t delta_size, off_t delta_compressed_size,
+ off_t delta_offset, off_t delta_out_offset, struct got_object_id *base_id)
{
struct got_imsg_raw_delta idelta;
int ret;
idelta.base_size = base_size;
idelta.result_size = result_size;
idelta.delta_size = delta_size;
+ idelta.delta_compressed_size = delta_compressed_size;
idelta.delta_offset = delta_offset;
idelta.delta_out_offset = delta_out_offset;
memcpy(idelta.base_id, base_id->sha1, SHA1_DIGEST_LENGTH);
const struct got_error *
got_privsep_recv_raw_delta(uint64_t *base_size, uint64_t *result_size,
- off_t *delta_size, off_t *delta_offset, off_t *delta_out_offset,
- struct got_object_id **base_id, struct imsgbuf *ibuf)
+ off_t *delta_size, off_t *delta_compressed_size, off_t *delta_offset,
+ off_t *delta_out_offset, struct got_object_id **base_id, struct imsgbuf *ibuf)
{
const struct got_error *err = NULL;
struct imsg imsg;
*base_size = 0;
*result_size = 0;
*delta_size = 0;
+ *delta_compressed_size = 0;
*delta_offset = 0;
*delta_out_offset = 0;
*base_id = NULL;
*base_size = delta->base_size;
*result_size = delta->result_size;
*delta_size = delta->delta_size;
+ *delta_compressed_size = delta->delta_compressed_size;
*delta_offset = delta->delta_offset;
*delta_out_offset = delta->delta_out_offset;
*base_id = calloc(1, sizeof(**base_id));
blob - bf79fbc1b585f4c34d13bcad9d48277a73d1b8a6
blob + cc827db1dcd1d77318526d65d157951d2cf457e0
--- libexec/got-read-pack/got-read-pack.c
+++ libexec/got-read-pack/got-read-pack.c
{
const struct got_error *err = NULL;
struct got_imsg_raw_delta_request req;
- size_t datalen, delta_size;
+ size_t datalen, delta_size, delta_compressed_size;
off_t delta_offset;
uint8_t *delta_buf = NULL;
struct got_object_id id, base_id;
imsg->fd = -1;
err = got_packfile_extract_raw_delta(&delta_buf, &delta_size,
- &delta_offset, &base_offset, &base_id, &base_size, &result_size,
- pack, packidx, req.idx);
+ &delta_compressed_size, &delta_offset, &base_offset, &base_id,
+ &base_size, &result_size, pack, packidx, req.idx);
if (err)
goto done;
}
delta_out_offset = ftello(delta_outfile);
- w = fwrite(delta_buf, 1, delta_size, delta_outfile);
- if (w != delta_size) {
+ w = fwrite(delta_buf, 1, delta_compressed_size, delta_outfile);
+ if (w != delta_compressed_size) {
err = got_ferror(delta_outfile, GOT_ERR_IO);
goto done;
}
}
err = got_privsep_send_raw_delta(ibuf, base_size, result_size,
- delta_size, delta_offset, delta_out_offset, &base_id);
+ delta_size, delta_compressed_size, delta_offset, delta_out_offset,
+ &base_id);
done:
free(delta_buf);
return err;
