mirror of
https://github.com/Xevion/easy7zip.git
synced 2025-12-06 05:15:00 -06:00
427 lines
10 KiB
C
Executable File
427 lines
10 KiB
C
Executable File
/* 7zMain.c - Test application for 7z Decoder
|
|
2008-08-05
|
|
Igor Pavlov
|
|
Public domain */
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
|
|
#ifdef _WIN32
|
|
#define USE_WINDOWS_FUNCTIONS
|
|
#endif
|
|
|
|
#ifdef USE_WINDOWS_FUNCTIONS
|
|
#include <windows.h>
|
|
#endif
|
|
|
|
#include "7zIn.h"
|
|
#include "7zExtract.h"
|
|
#include "7zAlloc.h"
|
|
|
|
#include "../../7zCrc.h"
|
|
|
|
|
|
#ifdef USE_WINDOWS_FUNCTIONS
|
|
typedef HANDLE MY_FILE_HANDLE;
|
|
#else
|
|
typedef FILE *MY_FILE_HANDLE;
|
|
#endif
|
|
|
|
void ConvertNumberToString(CFileSize value, char *s)
|
|
{
|
|
char temp[32];
|
|
int pos = 0;
|
|
do
|
|
{
|
|
temp[pos++] = (char)('0' + (int)(value % 10));
|
|
value /= 10;
|
|
}
|
|
while (value != 0);
|
|
do
|
|
*s++ = temp[--pos];
|
|
while(pos > 0);
|
|
*s = '\0';
|
|
}
|
|
|
|
#define PERIOD_4 (4 * 365 + 1)
|
|
#define PERIOD_100 (PERIOD_4 * 25 - 1)
|
|
#define PERIOD_400 (PERIOD_100 * 4 + 1)
|
|
|
|
void ConvertFileTimeToString(CNtfsFileTime *ft, char *s)
|
|
{
|
|
unsigned year, mon, day, hour, min, sec;
|
|
UInt64 v64 = ft->Low | ((UInt64)ft->High << 32);
|
|
Byte ms[] = { 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31 };
|
|
unsigned temp;
|
|
UInt32 v;
|
|
v64 /= 10000000;
|
|
sec = (unsigned)(v64 % 60);
|
|
v64 /= 60;
|
|
min = (unsigned)(v64 % 60);
|
|
v64 /= 60;
|
|
hour = (unsigned)(v64 % 24);
|
|
v64 /= 24;
|
|
|
|
v = (UInt32)v64;
|
|
|
|
year = (unsigned)(1601 + v / PERIOD_400 * 400);
|
|
v %= PERIOD_400;
|
|
|
|
temp = (unsigned)(v / PERIOD_100);
|
|
if (temp == 4)
|
|
temp = 3;
|
|
year += temp * 100;
|
|
v -= temp * PERIOD_100;
|
|
|
|
temp = v / PERIOD_4;
|
|
if (temp == 25)
|
|
temp = 24;
|
|
year += temp * 4;
|
|
v -= temp * PERIOD_4;
|
|
|
|
temp = v / 365;
|
|
if (temp == 4)
|
|
temp = 3;
|
|
year += temp;
|
|
v -= temp * 365;
|
|
|
|
if (year % 4 == 0 && (year % 100 != 0 || year % 400 == 0))
|
|
ms[1] = 29;
|
|
for (mon = 1; mon <= 12; mon++)
|
|
{
|
|
unsigned s = ms[mon - 1];
|
|
if (v < s)
|
|
break;
|
|
v -= s;
|
|
}
|
|
day = (unsigned)v + 1;
|
|
sprintf(s, "%04d-%02d-%02d %02d:%02d:%02d", year, mon, day, hour, min, sec);
|
|
}
|
|
|
|
|
|
#ifdef USE_WINDOWS_FUNCTIONS
|
|
/*
|
|
ReadFile and WriteFile functions in Windows have BUG:
|
|
If you Read or Write 64MB or more (probably min_failure_size = 64MB - 32KB + 1)
|
|
from/to Network file, it returns ERROR_NO_SYSTEM_RESOURCES
|
|
(Insufficient system resources exist to complete the requested service).
|
|
*/
|
|
#define kChunkSizeMax (1 << 24)
|
|
#endif
|
|
|
|
size_t MyReadFile(MY_FILE_HANDLE file, void *data, size_t size)
|
|
{
|
|
if (size == 0)
|
|
return 0;
|
|
#ifdef USE_WINDOWS_FUNCTIONS
|
|
{
|
|
size_t processedSize = 0;
|
|
do
|
|
{
|
|
DWORD curSize = (size > kChunkSizeMax) ? kChunkSizeMax : (DWORD)size;
|
|
DWORD processedLoc = 0;
|
|
BOOL res = ReadFile(file, data, curSize, &processedLoc, NULL);
|
|
data = (void *)((unsigned char *)data + processedLoc);
|
|
size -= processedLoc;
|
|
processedSize += processedLoc;
|
|
if (!res || processedLoc == 0)
|
|
break;
|
|
}
|
|
while (size > 0);
|
|
return processedSize;
|
|
}
|
|
#else
|
|
return fread(data, 1, size, file);
|
|
#endif
|
|
}
|
|
|
|
size_t MyWriteFile(MY_FILE_HANDLE file, void *data, size_t size)
|
|
{
|
|
if (size == 0)
|
|
return 0;
|
|
#ifdef USE_WINDOWS_FUNCTIONS
|
|
{
|
|
size_t processedSize = 0;
|
|
do
|
|
{
|
|
DWORD curSize = (size > kChunkSizeMax) ? kChunkSizeMax : (DWORD)size;
|
|
DWORD processedLoc = 0;
|
|
BOOL res = WriteFile(file, data, curSize, &processedLoc, NULL);
|
|
data = (void *)((unsigned char *)data + processedLoc);
|
|
size -= processedLoc;
|
|
processedSize += processedLoc;
|
|
if (!res)
|
|
break;
|
|
}
|
|
while (size > 0);
|
|
return processedSize;
|
|
}
|
|
#else
|
|
return fwrite(data, 1, size, file);
|
|
#endif
|
|
}
|
|
|
|
int MyCloseFile(MY_FILE_HANDLE file)
|
|
{
|
|
#ifdef USE_WINDOWS_FUNCTIONS
|
|
return (CloseHandle(file) != FALSE) ? 0 : 1;
|
|
#else
|
|
return fclose(file);
|
|
#endif
|
|
}
|
|
|
|
typedef struct _CFileInStream
|
|
{
|
|
ISzInStream InStream;
|
|
MY_FILE_HANDLE File;
|
|
} CFileInStream;
|
|
|
|
|
|
#define kBufferSize (1 << 12)
|
|
Byte g_Buffer[kBufferSize];
|
|
|
|
SRes SzFileReadImp(void *object, void **buffer, size_t *size)
|
|
{
|
|
CFileInStream *s = (CFileInStream *)object;
|
|
if (*size > kBufferSize)
|
|
*size = kBufferSize;
|
|
*size = MyReadFile(s->File, g_Buffer, *size);
|
|
*buffer = g_Buffer;
|
|
return SZ_OK;
|
|
}
|
|
|
|
SRes SzFileSeekImp(void *object, CFileSize pos, ESzSeek origin)
|
|
{
|
|
CFileInStream *s = (CFileInStream *)object;
|
|
|
|
#ifdef USE_WINDOWS_FUNCTIONS
|
|
{
|
|
LARGE_INTEGER value;
|
|
DWORD moveMethod;
|
|
value.LowPart = (DWORD)pos;
|
|
value.HighPart = (LONG)((UInt64)pos >> 32);
|
|
#ifdef _SZ_FILE_SIZE_32
|
|
/* VC 6.0 has bug with >> 32 shifts. */
|
|
value.HighPart = 0;
|
|
#endif
|
|
switch (origin)
|
|
{
|
|
case SZ_SEEK_SET: moveMethod = FILE_BEGIN; break;
|
|
case SZ_SEEK_CUR: moveMethod = FILE_CURRENT; break;
|
|
case SZ_SEEK_END: moveMethod = FILE_END; break;
|
|
default: return SZ_ERROR_PARAM;
|
|
}
|
|
value.LowPart = SetFilePointer(s->File, value.LowPart, &value.HighPart, moveMethod);
|
|
if (value.LowPart == 0xFFFFFFFF)
|
|
if (GetLastError() != NO_ERROR)
|
|
return SZ_ERROR_FAIL;
|
|
return SZ_OK;
|
|
}
|
|
#else
|
|
int moveMethod;
|
|
int res;
|
|
switch (origin)
|
|
{
|
|
case SZ_SEEK_SET: moveMethod = SEEK_SET; break;
|
|
case SZ_SEEK_CUR: moveMethod = SEEK_CUR; break;
|
|
case SZ_SEEK_END: moveMethod = SEEK_END; break;
|
|
default: return SZ_ERROR_PARAM;
|
|
}
|
|
res = fseek(s->File, (long)pos, moveMethod );
|
|
return (res == 0) ? SZ_OK : SZ_ERROR_FAIL;
|
|
#endif
|
|
}
|
|
|
|
void PrintError(char *sz)
|
|
{
|
|
printf("\nERROR: %s\n", sz);
|
|
}
|
|
|
|
int MY_CDECL main(int numargs, char *args[])
|
|
{
|
|
CFileInStream archiveStream;
|
|
CSzArEx db;
|
|
SRes res;
|
|
ISzAlloc allocImp;
|
|
ISzAlloc allocTempImp;
|
|
|
|
printf("\n7z ANSI-C Decoder 4.59 Copyright (c) 1999-2008 Igor Pavlov 2008-07-09\n");
|
|
if (numargs == 1)
|
|
{
|
|
printf(
|
|
"\nUsage: 7zDec <command> <archive_name>\n\n"
|
|
"<Commands>\n"
|
|
" e: Extract files from archive\n"
|
|
" l: List contents of archive\n"
|
|
" t: Test integrity of archive\n");
|
|
return 0;
|
|
}
|
|
if (numargs < 3)
|
|
{
|
|
PrintError("incorrect command");
|
|
return 1;
|
|
}
|
|
|
|
archiveStream.File =
|
|
#ifdef USE_WINDOWS_FUNCTIONS
|
|
CreateFileA(args[2], GENERIC_READ, FILE_SHARE_READ,
|
|
NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
|
|
if (archiveStream.File == INVALID_HANDLE_VALUE)
|
|
#else
|
|
archiveStream.File = fopen(args[2], "rb");
|
|
if (archiveStream.File == 0)
|
|
#endif
|
|
{
|
|
PrintError("can not open input file");
|
|
return 1;
|
|
}
|
|
|
|
archiveStream.InStream.Read = SzFileReadImp;
|
|
archiveStream.InStream.Seek = SzFileSeekImp;
|
|
|
|
allocImp.Alloc = SzAlloc;
|
|
allocImp.Free = SzFree;
|
|
|
|
allocTempImp.Alloc = SzAllocTemp;
|
|
allocTempImp.Free = SzFreeTemp;
|
|
|
|
CrcGenerateTable();
|
|
|
|
SzArEx_Init(&db);
|
|
res = SzArEx_Open(&db, &archiveStream.InStream, &allocImp, &allocTempImp);
|
|
if (res == SZ_OK)
|
|
{
|
|
char *command = args[1];
|
|
int listCommand = 0;
|
|
int testCommand = 0;
|
|
int extractCommand = 0;
|
|
if (strcmp(command, "l") == 0)
|
|
listCommand = 1;
|
|
if (strcmp(command, "t") == 0)
|
|
testCommand = 1;
|
|
else if (strcmp(command, "e") == 0)
|
|
extractCommand = 1;
|
|
|
|
if (listCommand)
|
|
{
|
|
UInt32 i;
|
|
for (i = 0; i < db.db.NumFiles; i++)
|
|
{
|
|
CSzFileItem *f = db.db.Files + i;
|
|
char s[32], t[32];
|
|
ConvertNumberToString(f->Size, s);
|
|
if (f->MTimeDefined)
|
|
ConvertFileTimeToString(&f->MTime, t);
|
|
else
|
|
strcpy(t, " ");
|
|
|
|
printf("%10s %s %s\n", s, t, f->Name);
|
|
}
|
|
}
|
|
else if (testCommand || extractCommand)
|
|
{
|
|
UInt32 i;
|
|
|
|
/*
|
|
if you need cache, use these 3 variables.
|
|
if you use external function, you can make these variable as static.
|
|
*/
|
|
UInt32 blockIndex = 0xFFFFFFFF; /* it can have any value before first call (if outBuffer = 0) */
|
|
Byte *outBuffer = 0; /* it must be 0 before first call for each new archive. */
|
|
size_t outBufferSize = 0; /* it can have any value before first call (if outBuffer = 0) */
|
|
|
|
printf("\n");
|
|
for (i = 0; i < db.db.NumFiles; i++)
|
|
{
|
|
size_t offset;
|
|
size_t outSizeProcessed;
|
|
CSzFileItem *f = db.db.Files + i;
|
|
if (f->IsDir)
|
|
printf("Directory ");
|
|
else
|
|
printf(testCommand ?
|
|
"Testing ":
|
|
"Extracting");
|
|
printf(" %s", f->Name);
|
|
if (f->IsDir)
|
|
{
|
|
printf("\n");
|
|
continue;
|
|
}
|
|
res = SzAr_Extract(&db, &archiveStream.InStream, i,
|
|
&blockIndex, &outBuffer, &outBufferSize,
|
|
&offset, &outSizeProcessed,
|
|
&allocImp, &allocTempImp);
|
|
if (res != SZ_OK)
|
|
break;
|
|
if (!testCommand)
|
|
{
|
|
MY_FILE_HANDLE outputHandle;
|
|
size_t processedSize;
|
|
char *fileName = f->Name;
|
|
size_t nameLen = strlen(f->Name);
|
|
for (; nameLen > 0; nameLen--)
|
|
if (f->Name[nameLen - 1] == '/')
|
|
{
|
|
fileName = f->Name + nameLen;
|
|
break;
|
|
}
|
|
|
|
outputHandle =
|
|
#ifdef USE_WINDOWS_FUNCTIONS
|
|
CreateFileA(fileName, GENERIC_WRITE, FILE_SHARE_READ,
|
|
NULL, CREATE_ALWAYS, FILE_ATTRIBUTE_NORMAL, NULL);
|
|
if (outputHandle == INVALID_HANDLE_VALUE)
|
|
#else
|
|
fopen(fileName, "wb+");
|
|
if (outputHandle == 0)
|
|
#endif
|
|
{
|
|
PrintError("can not open output file");
|
|
res = SZ_ERROR_FAIL;
|
|
break;
|
|
}
|
|
processedSize = MyWriteFile(outputHandle, outBuffer + offset, outSizeProcessed);
|
|
if (processedSize != outSizeProcessed)
|
|
{
|
|
PrintError("can not write output file");
|
|
res = SZ_ERROR_FAIL;
|
|
break;
|
|
}
|
|
if (MyCloseFile(outputHandle))
|
|
{
|
|
PrintError("can not close output file");
|
|
res = SZ_ERROR_FAIL;
|
|
break;
|
|
}
|
|
}
|
|
printf("\n");
|
|
}
|
|
IAlloc_Free(&allocImp, outBuffer);
|
|
}
|
|
else
|
|
{
|
|
PrintError("incorrect command");
|
|
res = SZ_ERROR_FAIL;
|
|
}
|
|
}
|
|
SzArEx_Free(&db, &allocImp);
|
|
|
|
MyCloseFile(archiveStream.File);
|
|
if (res == SZ_OK)
|
|
{
|
|
printf("\nEverything is Ok\n");
|
|
return 0;
|
|
}
|
|
if (res == SZ_ERROR_UNSUPPORTED)
|
|
PrintError("decoder doesn't support this archive");
|
|
else if (res == SZ_ERROR_MEM)
|
|
PrintError("can not allocate memory");
|
|
else if (res == SZ_ERROR_CRC)
|
|
PrintError("CRC error");
|
|
else
|
|
printf("\nERROR #%d\n", res);
|
|
return 1;
|
|
}
|