Skip to content
Snippets Groups Projects
Commit 2ff4059b authored by Timo Teräs's avatar Timo Teräs
Browse files

io: use mmap to write archive entries to disk

avoids some copying and system calls.
parent 7b05eef6
No related branches found
No related tags found
No related merge requests found
......@@ -114,15 +114,25 @@ size_t apk_istream_splice(void *stream, int fd, size_t size,
{
struct apk_istream *is = (struct apk_istream *) stream;
unsigned char *buf;
size_t bufsz, done = 0, r, togo;
size_t bufsz, done = 0, r, togo, mmapped = 0;
bufsz = size;
if (bufsz > 256*1024)
bufsz = 256*1024;
if (size > 256 * 1024) {
buf = mmap(NULL, size, PROT_WRITE, 0, fd, 0);
if (buf != NULL) {
mmapped = 1;
if (bufsz > 2*1024*1024)
bufsz = 2*1024*1024;
}
}
if (!mmapped) {
if (bufsz > 256*1024)
bufsz = 256*1024;
buf = malloc(bufsz);
if (buf == NULL)
return -ENOMEM;
buf = malloc(bufsz);
if (buf == NULL)
return -ENOMEM;
}
while (done < size) {
if (done != 0 && cb != NULL)
......@@ -134,18 +144,25 @@ size_t apk_istream_splice(void *stream, int fd, size_t size,
r = is->read(is, buf, togo);
if (r < 0)
goto err;
if (write(fd, buf, r) != r) {
if (r < 0)
r = -errno;
goto err;
if (!mmapped) {
if (write(fd, buf, r) != r) {
if (r < 0)
r = -errno;
goto err;
}
}
done += r;
if (r != togo)
break;
}
r = done;
err:
free(buf);
if (mmapped)
munmap(buf, size);
else
free(buf);
return r;
}
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment