/*
* Read and interleave split parts from a ZIP file.
*/
xps_part *
xps_read_part(fz_context *ctx, xps_document *doc, char *partname)
{
fz_archive *zip = doc->zip;
fz_buffer *buf, *tmp;
char path[2048];
int count;
char *name;
int seen_last;
name = partname;
if (name[0] == '/')
name ++;
/* All in one piece */
if (fz_has_archive_entry(ctx, zip, name))
{
buf = fz_read_archive_entry(ctx, zip, name);
}
/* Assemble all the pieces */
else
{
buf = fz_new_buffer(ctx, 512);
seen_last = 0;
for (count = 0; !seen_last; ++count)
{
fz_snprintf(path, sizeof path, "%s/[%d].piece", name, count);
if (fz_has_archive_entry(ctx, zip, path))
{
tmp = fz_read_archive_entry(ctx, zip, path);
fz_append_buffer(ctx, buf, tmp);
fz_drop_buffer(ctx, tmp);
}
else
{
fz_snprintf(path, sizeof path, "%s/[%d].last.piece", name, count);
if (fz_has_archive_entry(ctx, zip, path))
{
tmp = fz_read_archive_entry(ctx, zip, path);
fz_append_buffer(ctx, buf, tmp);
fz_drop_buffer(ctx, tmp);
seen_last = 1;
}
else
{
fz_drop_buffer(ctx, buf);
fz_throw(ctx, FZ_ERROR_GENERIC, "cannot find all pieces for part '%s'", partname);
}
}
}
}
return xps_new_part(ctx, doc, partname, buf);
}
/*
* Read and interleave split parts from a ZIP file.
*/
xps_part *
xps_read_part(fz_context *ctx, xps_document *doc, char *partname)
{
fz_archive *zip = doc->zip;
fz_buffer *buf, *tmp;
char path[2048];
unsigned char *data;
int size;
int count;
char *name;
int seen_last;
name = partname;
if (name[0] == '/')
name ++;
/* All in one piece */
if (fz_has_archive_entry(ctx, zip, name))
{
buf = fz_read_archive_entry(ctx, zip, name);
}
/* Assemble all the pieces */
else
{
buf = fz_new_buffer(ctx, 512);
seen_last = 0;
for (count = 0; !seen_last; ++count)
{
sprintf(path, "%s/[%d].piece", name, count);
if (fz_has_archive_entry(ctx, zip, path))
{
tmp = fz_read_archive_entry(ctx, zip, path);
fz_append_buffer(ctx, buf, tmp);
fz_drop_buffer(ctx, tmp);
}
else
{
sprintf(path, "%s/[%d].last.piece", name, count);
if (fz_has_archive_entry(ctx, zip, path))
{
tmp = fz_read_archive_entry(ctx, zip, path);
fz_append_buffer(ctx, buf, tmp);
fz_drop_buffer(ctx, tmp);
seen_last = 1;
}
else
{
fz_drop_buffer(ctx, buf);
fz_throw(ctx, FZ_ERROR_GENERIC, "cannot find all pieces for part '%s'", partname);
}
}
}
}
fz_write_buffer_byte(ctx, buf, 0); /* zero-terminate */
/* take over the data */
data = buf->data;
/* size doesn't include the added zero-terminator */
size = buf->len - 1;
fz_free(ctx, buf);
return xps_new_part(ctx, doc, partname, data, size);
}
