static bool rar_restart_solid(ar_archive *ar)
{
ar_archive_rar *rar = (ar_archive_rar *)ar;
off64_t current_offset = ar->entry_offset;
log("Restarting decompression for solid entry");
if (!ar_parse_entry_at(ar, ar->entry_offset_first)) {
ar_parse_entry_at(ar, current_offset);
return false;
}
while (ar->entry_offset < current_offset) {
size_t size = ar->entry_size_uncompressed;
rar->solid.restart = false;
while (size > 0) {
unsigned char buffer[1024];
size_t count = mins(size, sizeof(buffer));
if (!ar_entry_uncompress(ar, buffer, count)) {
ar_parse_entry_at(ar, current_offset);
return false;
}
size -= count;
}
if (!ar_parse_entry(ar)) {
ar_parse_entry_at(ar, current_offset);
return false;
}
}
rar->solid.restart = false;
return true;
}
static bool rar_restart_solid(ar_archive_rar *rar)
{
ar_archive *ar = &rar->super;
size_t current_offset = ar->entry_offset;
log("Restarting decompression for solid entry");
if (!ar_parse_entry_at(ar, 0)) {
ar_parse_entry_at(ar, current_offset);
return false;
}
while (ar->entry_offset != current_offset) {
size_t size = ar->entry_size_uncompressed;
rar->entry.restart_solid = false;
while (size > 0) {
unsigned char buffer[1024];
if (!ar_entry_uncompress(ar, buffer, min(size, sizeof(buffer)))) {
ar_parse_entry_at(ar, current_offset);
return false;
}
size -= min(size, sizeof(buffer));
}
if (!ar_parse_entry(ar)) {
ar_parse_entry_at(ar, current_offset);
return false;
}
}
rar->entry.restart_solid = false;
return true;
}
int main(int argc, char *argv[])
{
ar_stream *stream = NULL;
ar_archive *ar = NULL;
int entry_count = 1;
int entry_skips = 0;
int error_step = 1;
#if !defined(NDEBUG) && defined(_MSC_VER)
if (!IsDebuggerPresent()) {
_CrtSetReportMode(_CRT_WARN, _CRTDBG_MODE_FILE);
_CrtSetReportFile(_CRT_WARN, _CRTDBG_FILE_STDERR);
}
_CrtSetDbgFlag(_CRTDBG_ALLOC_MEM_DF | _CRTDBG_LEAK_CHECK_DF);
#endif
FailIf(argc != 2, "Syntax: %s <filename.ext>", argv[0]);
stream = ar_open_file(argv[1]);
FailIf(!stream, "Error: File \"%s\" not found!", argv[1]);
printf("Parsing \"%s\":\n", argv[1]);
ar = ar_open_any_archive(stream, strrchr(argv[1], '.'));
FailIf(!ar, "Error: No valid %s archive!", "RAR, ZIP, 7Z or TAR");
while (ar_parse_entry(ar)) {
size_t size = ar_entry_get_size(ar);
printf("%02d. %s (@%" PRIi64 ")\n", entry_count++, ar_entry_get_name(ar), ar_entry_get_offset(ar));
while (size > 0) {
unsigned char buffer[1024];
size_t count = size < sizeof(buffer) ? size : sizeof(buffer);
if (!ar_entry_uncompress(ar, buffer, count))
break;
size -= count;
}
if (size > 0) {
fprintf(stderr, "Warning: Failed to uncompress... skipping\n");
entry_skips++;
}
}
FailIf(!ar_at_eof(ar), "Error: Failed to parse entry %d!", entry_count);
error_step = entry_skips > 0 ? 1000 + entry_skips : 0;
CleanUp:
ar_close_archive(ar);
ar_close(stream);
return error_step;
}
ArchFile::ArchFile(ar_stream *data, ar_archive *(* openFormat)(ar_stream *)) : data(data), ar(nullptr)
{
if (data && openFormat)
ar = openFormat(data);
if (!ar)
return;
while (ar_parse_entry(ar)) {
const char *name = ar_entry_get_name(ar);
if (name)
filenames.Append(str::conv::FromUtf8(name));
else
filenames.Append(nullptr);
filepos.Append(ar_entry_get_offset(ar));
}
// extract (further) filenames with fallback in derived class constructor
// once GetFileFromFallback has been correctly set in the vtable
}
bool moveToNextFile(ARCHIVE * archive)
{
bool output;
//moveToNextFile discard the first call after rewinding, as we already are on the first file
if(archive->didJustRewind)
{
archive->didJustRewind = false;
output = true;
}
else
output = ar_parse_entry(archive->archive);
if(output)
output &= !isEOF(archive);
if(output && archive->currentEntryLength != UINT64_MAX)
archive->currentEntryLength = UINT64_MAX;
return output;
}
