/*

7zMain.c

Test application for 7z Decoder

LZMA SDK 4.43 Copyright (c) 1999-2006 Igor Pavlov (2006-06-04)

*/

#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 "../../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(CArchiveFileTime *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;

#ifdef _LZMA_IN_CB

#define kBufferSize (1 << 12)

Byte g_Buffer[kBufferSize];

SZ_RESULT SzFileReadImp(void *object, void **buffer, size_t maxRequiredSize, size_t *processedSize)

{

CFileInStream *s = (CFileInStream *)object;

size_t processedSizeLoc;

if (maxRequiredSize > kBufferSize)

maxRequiredSize = kBufferSize;

processedSizeLoc = MyReadFile(s->File, g_Buffer, maxRequiredSize);

*buffer = g_Buffer;

if (processedSize != 0)

*processedSize = processedSizeLoc;

return SZ_OK;

}

#else

SZ_RESULT SzFileReadImp(void *object, void *buffer, size_t size, size_t *processedSize)

{

CFileInStream *s = (CFileInStream *)object;

size_t processedSizeLoc = MyReadFile(s->File, buffer, size);

if (processedSize != 0)

*processedSize = processedSizeLoc;

return SZ_OK;

}

#endif

SZ_RESULT SzFileSeekImp(void *object, CFileSize pos)

{

CFileInStream *s = (CFileInStream *)object;

#ifdef USE_WINDOWS_FUNCTIONS

{

LARGE_INTEGER value;

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

value.LowPart = SetFilePointer(s->File, value.LowPart, &value.HighPart, FILE_BEGIN);

if (value.LowPart == 0xFFFFFFFF)

if(GetLastError() != NO_ERROR)

return SZE_FAIL;

return SZ_OK;

}

#else

int res = fseek(s->File, (long)pos, SEEK_SET);

if (res == 0)

return SZ_OK;

return SZE_FAIL;

#endif

}

void PrintError(char *sz)

{

printf("\nERROR: %s\n", sz);

}

int main(int numargs, char *args[])

{

CFileInStream archiveStream;

CArchiveDatabaseEx db;

SZ_RESULT res;

ISzAlloc allocImp;

ISzAlloc allocTempImp;

printf("\n7z ANSI-C Decoder 4.48 Copyright (c) 1999-2007 Igor Pavlov 2007-06-21\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

CreateFile(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();

SzArDbExInit(&db);

res = SzArchiveOpen(&archiveStream.InStream, &db, &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.Database.NumFiles; i++)

{

CFileItem *f = db.Database.Files + i;

char s[32], t[32];

ConvertNumberToString(f->Size, s);

if (f->IsLastWriteTimeDefined)

ConvertFileTimeToString(&f->LastWriteTime, 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.Database.NumFiles; i++)

{

size_t offset;

size_t outSizeProcessed;

CFileItem *f = db.Database.Files + i;

if (f->IsDirectory)

printf("Directory ");

else

printf(testCommand ?

"Testing ":

"Extracting");

printf(" %s", f->Name);

if (f->IsDirectory)

{

printf("\n");

continue;

}

res = SzExtract(&archiveStream.InStream, &db, 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

CreateFile(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 = SZE_FAIL;

break;

}

processedSize = MyWriteFile(outputHandle, outBuffer + offset, outSizeProcessed);

if (processedSize != outSizeProcessed)

{

PrintError("can not write output file");

res = SZE_FAIL;

break;

}

if (MyCloseFile(outputHandle))

{

PrintError("can not close output file");

res = SZE_FAIL;

break;

}

}

printf("\n");

}

allocImp.Free(outBuffer);

}

else

{

PrintError("incorrect command");

res = SZE_FAIL;

}

}

SzArDbExFree(&db, allocImp.Free);

MyCloseFile(archiveStream.File);

if (res == SZ_OK)

{

printf("\nEverything is Ok\n");

return 0;

}

if (res == (SZ_RESULT)SZE_NOTIMPL)

PrintError("decoder doesn't support this archive");

else if (res == (SZ_RESULT)SZE_OUTOFMEMORY)

PrintError("can not allocate memory");

else if (res == (SZ_RESULT)SZE_CRC_ERROR)

PrintError("CRC error");

else

printf("\nERROR #%d\n", res);

return 1;

}