mirror of
https://github.com/nillerusr/source-engine.git
synced 2024-12-24 15:16:49 +00:00
547 lines
13 KiB
C
547 lines
13 KiB
C
/* 7zMain.c - Test application for 7z Decoder
|
|
2015-01-02 : Igor Pavlov : Public domain */
|
|
|
|
#include "Precomp.h"
|
|
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
|
|
#include "../../7z.h"
|
|
#include "../../7zAlloc.h"
|
|
#include "../../7zBuf.h"
|
|
#include "../../7zCrc.h"
|
|
#include "../../7zFile.h"
|
|
#include "../../7zVersion.h"
|
|
|
|
#ifndef USE_WINDOWS_FILE
|
|
/* for mkdir */
|
|
#ifdef _WIN32
|
|
#include <direct.h>
|
|
#else
|
|
#include <sys/stat.h>
|
|
#include <errno.h>
|
|
#endif
|
|
#endif
|
|
|
|
static ISzAlloc g_Alloc = { SzAlloc, SzFree };
|
|
|
|
static int Buf_EnsureSize(CBuf *dest, size_t size)
|
|
{
|
|
if (dest->size >= size)
|
|
return 1;
|
|
Buf_Free(dest, &g_Alloc);
|
|
return Buf_Create(dest, size, &g_Alloc);
|
|
}
|
|
|
|
#ifndef _WIN32
|
|
|
|
static Byte kUtf8Limits[5] = { 0xC0, 0xE0, 0xF0, 0xF8, 0xFC };
|
|
|
|
static Bool Utf16_To_Utf8(Byte *dest, size_t *destLen, const UInt16 *src, size_t srcLen)
|
|
{
|
|
size_t destPos = 0, srcPos = 0;
|
|
for (;;)
|
|
{
|
|
unsigned numAdds;
|
|
UInt32 value;
|
|
if (srcPos == srcLen)
|
|
{
|
|
*destLen = destPos;
|
|
return True;
|
|
}
|
|
value = src[srcPos++];
|
|
if (value < 0x80)
|
|
{
|
|
if (dest)
|
|
dest[destPos] = (char)value;
|
|
destPos++;
|
|
continue;
|
|
}
|
|
if (value >= 0xD800 && value < 0xE000)
|
|
{
|
|
UInt32 c2;
|
|
if (value >= 0xDC00 || srcPos == srcLen)
|
|
break;
|
|
c2 = src[srcPos++];
|
|
if (c2 < 0xDC00 || c2 >= 0xE000)
|
|
break;
|
|
value = (((value - 0xD800) << 10) | (c2 - 0xDC00)) + 0x10000;
|
|
}
|
|
for (numAdds = 1; numAdds < 5; numAdds++)
|
|
if (value < (((UInt32)1) << (numAdds * 5 + 6)))
|
|
break;
|
|
if (dest)
|
|
dest[destPos] = (char)(kUtf8Limits[numAdds - 1] + (value >> (6 * numAdds)));
|
|
destPos++;
|
|
do
|
|
{
|
|
numAdds--;
|
|
if (dest)
|
|
dest[destPos] = (char)(0x80 + ((value >> (6 * numAdds)) & 0x3F));
|
|
destPos++;
|
|
}
|
|
while (numAdds != 0);
|
|
}
|
|
*destLen = destPos;
|
|
return False;
|
|
}
|
|
|
|
static SRes Utf16_To_Utf8Buf(CBuf *dest, const UInt16 *src, size_t srcLen)
|
|
{
|
|
size_t destLen = 0;
|
|
Bool res;
|
|
Utf16_To_Utf8(NULL, &destLen, src, srcLen);
|
|
destLen += 1;
|
|
if (!Buf_EnsureSize(dest, destLen))
|
|
return SZ_ERROR_MEM;
|
|
res = Utf16_To_Utf8(dest->data, &destLen, src, srcLen);
|
|
dest->data[destLen] = 0;
|
|
return res ? SZ_OK : SZ_ERROR_FAIL;
|
|
}
|
|
|
|
#endif
|
|
|
|
static SRes Utf16_To_Char(CBuf *buf, const UInt16 *s
|
|
#ifdef _WIN32
|
|
, UINT codePage
|
|
#endif
|
|
)
|
|
{
|
|
unsigned len = 0;
|
|
for (len = 0; s[len] != 0; len++);
|
|
|
|
#ifdef _WIN32
|
|
{
|
|
unsigned size = len * 3 + 100;
|
|
if (!Buf_EnsureSize(buf, size))
|
|
return SZ_ERROR_MEM;
|
|
{
|
|
buf->data[0] = 0;
|
|
if (len != 0)
|
|
{
|
|
char defaultChar = '_';
|
|
BOOL defUsed;
|
|
unsigned numChars = 0;
|
|
numChars = WideCharToMultiByte(codePage, 0, s, len, (char *)buf->data, size, &defaultChar, &defUsed);
|
|
if (numChars == 0 || numChars >= size)
|
|
return SZ_ERROR_FAIL;
|
|
buf->data[numChars] = 0;
|
|
}
|
|
return SZ_OK;
|
|
}
|
|
}
|
|
#else
|
|
return Utf16_To_Utf8Buf(buf, s, len);
|
|
#endif
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
#ifndef USE_WINDOWS_FILE
|
|
static UINT g_FileCodePage = CP_ACP;
|
|
#endif
|
|
#define MY_FILE_CODE_PAGE_PARAM ,g_FileCodePage
|
|
#else
|
|
#define MY_FILE_CODE_PAGE_PARAM
|
|
#endif
|
|
|
|
static WRes MyCreateDir(const UInt16 *name)
|
|
{
|
|
#ifdef USE_WINDOWS_FILE
|
|
|
|
return CreateDirectoryW(name, NULL) ? 0 : GetLastError();
|
|
|
|
#else
|
|
|
|
CBuf buf;
|
|
WRes res;
|
|
Buf_Init(&buf);
|
|
RINOK(Utf16_To_Char(&buf, name MY_FILE_CODE_PAGE_PARAM));
|
|
|
|
res =
|
|
#ifdef _WIN32
|
|
_mkdir((const char *)buf.data)
|
|
#else
|
|
mkdir((const char *)buf.data, 0777)
|
|
#endif
|
|
== 0 ? 0 : errno;
|
|
Buf_Free(&buf, &g_Alloc);
|
|
return res;
|
|
|
|
#endif
|
|
}
|
|
|
|
static WRes OutFile_OpenUtf16(CSzFile *p, const UInt16 *name)
|
|
{
|
|
#ifdef USE_WINDOWS_FILE
|
|
return OutFile_OpenW(p, name);
|
|
#else
|
|
CBuf buf;
|
|
WRes res;
|
|
Buf_Init(&buf);
|
|
RINOK(Utf16_To_Char(&buf, name MY_FILE_CODE_PAGE_PARAM));
|
|
res = OutFile_Open(p, (const char *)buf.data);
|
|
Buf_Free(&buf, &g_Alloc);
|
|
return res;
|
|
#endif
|
|
}
|
|
|
|
static SRes PrintString(const UInt16 *s)
|
|
{
|
|
CBuf buf;
|
|
SRes res;
|
|
Buf_Init(&buf);
|
|
res = Utf16_To_Char(&buf, s
|
|
#ifdef _WIN32
|
|
, CP_OEMCP
|
|
#endif
|
|
);
|
|
if (res == SZ_OK)
|
|
fputs((const char *)buf.data, stdout);
|
|
Buf_Free(&buf, &g_Alloc);
|
|
return res;
|
|
}
|
|
|
|
static void UInt64ToStr(UInt64 value, char *s)
|
|
{
|
|
char temp[32];
|
|
int pos = 0;
|
|
do
|
|
{
|
|
temp[pos++] = (char)('0' + (unsigned)(value % 10));
|
|
value /= 10;
|
|
}
|
|
while (value != 0);
|
|
do
|
|
*s++ = temp[--pos];
|
|
while (pos);
|
|
*s = '\0';
|
|
}
|
|
|
|
static char *UIntToStr(char *s, unsigned value, int numDigits)
|
|
{
|
|
char temp[16];
|
|
int pos = 0;
|
|
do
|
|
temp[pos++] = (char)('0' + (value % 10));
|
|
while (value /= 10);
|
|
for (numDigits -= pos; numDigits > 0; numDigits--)
|
|
*s++ = '0';
|
|
do
|
|
*s++ = temp[--pos];
|
|
while (pos);
|
|
*s = '\0';
|
|
return s;
|
|
}
|
|
|
|
static void UIntToStr_2(char *s, unsigned value)
|
|
{
|
|
s[0] = (char)('0' + (value / 10));
|
|
s[1] = (char)('0' + (value % 10));
|
|
}
|
|
|
|
#define PERIOD_4 (4 * 365 + 1)
|
|
#define PERIOD_100 (PERIOD_4 * 25 - 1)
|
|
#define PERIOD_400 (PERIOD_100 * 4 + 1)
|
|
|
|
static void ConvertFileTimeToString(const CNtfsFileTime *nt, char *s)
|
|
{
|
|
unsigned year, mon, hour, min, sec;
|
|
Byte ms[] = { 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31 };
|
|
unsigned t;
|
|
UInt32 v;
|
|
UInt64 v64 = nt->Low | ((UInt64)nt->High << 32);
|
|
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;
|
|
|
|
t = v / PERIOD_100; if (t == 4) t = 3; year += t * 100; v -= t * PERIOD_100;
|
|
t = v / PERIOD_4; if (t == 25) t = 24; year += t * 4; v -= t * PERIOD_4;
|
|
t = v / 365; if (t == 4) t = 3; year += t; v -= t * 365;
|
|
|
|
if (year % 4 == 0 && (year % 100 != 0 || year % 400 == 0))
|
|
ms[1] = 29;
|
|
for (mon = 0;; mon++)
|
|
{
|
|
unsigned s = ms[mon];
|
|
if (v < s)
|
|
break;
|
|
v -= s;
|
|
}
|
|
s = UIntToStr(s, year, 4); *s++ = '-';
|
|
UIntToStr_2(s, mon + 1); s[2] = '-'; s += 3;
|
|
UIntToStr_2(s, (unsigned)v + 1); s[2] = ' '; s += 3;
|
|
UIntToStr_2(s, hour); s[2] = ':'; s += 3;
|
|
UIntToStr_2(s, min); s[2] = ':'; s += 3;
|
|
UIntToStr_2(s, sec); s[2] = 0;
|
|
}
|
|
|
|
void PrintError(const char *sz)
|
|
{
|
|
printf("\nERROR: %s\n", sz);
|
|
}
|
|
|
|
#ifdef USE_WINDOWS_FILE
|
|
static void GetAttribString(UInt32 wa, Bool isDir, char *s)
|
|
{
|
|
s[0] = (char)(((wa & FILE_ATTRIBUTE_DIRECTORY) != 0 || isDir) ? 'D' : '.');
|
|
s[1] = (char)(((wa & FILE_ATTRIBUTE_READONLY ) != 0) ? 'R': '.');
|
|
s[2] = (char)(((wa & FILE_ATTRIBUTE_HIDDEN ) != 0) ? 'H': '.');
|
|
s[3] = (char)(((wa & FILE_ATTRIBUTE_SYSTEM ) != 0) ? 'S': '.');
|
|
s[4] = (char)(((wa & FILE_ATTRIBUTE_ARCHIVE ) != 0) ? 'A': '.');
|
|
s[5] = '\0';
|
|
}
|
|
#else
|
|
static void GetAttribString(UInt32, Bool, char *s)
|
|
{
|
|
s[0] = '\0';
|
|
}
|
|
#endif
|
|
|
|
// #define NUM_PARENTS_MAX 128
|
|
|
|
int MY_CDECL main(int numargs, char *args[])
|
|
{
|
|
CFileInStream archiveStream;
|
|
CLookToRead lookStream;
|
|
CSzArEx db;
|
|
SRes res;
|
|
ISzAlloc allocImp;
|
|
ISzAlloc allocTempImp;
|
|
UInt16 *temp = NULL;
|
|
size_t tempSize = 0;
|
|
// UInt32 parents[NUM_PARENTS_MAX];
|
|
|
|
printf("\n7z ANSI-C Decoder " MY_VERSION_COPYRIGHT_DATE "\n\n");
|
|
if (numargs == 1)
|
|
{
|
|
printf(
|
|
"Usage: 7zDec <command> <archive_name>\n\n"
|
|
"<Commands>\n"
|
|
" e: Extract files from archive (without using directory names)\n"
|
|
" l: List contents of archive\n"
|
|
" t: Test integrity of archive\n"
|
|
" x: eXtract files with full paths\n");
|
|
return 0;
|
|
}
|
|
if (numargs < 3)
|
|
{
|
|
PrintError("incorrect command");
|
|
return 1;
|
|
}
|
|
|
|
#if defined(_WIN32) && !defined(USE_WINDOWS_FILE) && !defined(UNDER_CE)
|
|
g_FileCodePage = AreFileApisANSI() ? CP_ACP : CP_OEMCP;
|
|
#endif
|
|
|
|
allocImp.Alloc = SzAlloc;
|
|
allocImp.Free = SzFree;
|
|
|
|
allocTempImp.Alloc = SzAllocTemp;
|
|
allocTempImp.Free = SzFreeTemp;
|
|
|
|
#ifdef UNDER_CE
|
|
if (InFile_OpenW(&archiveStream.file, L"\test.7z"))
|
|
#else
|
|
if (InFile_Open(&archiveStream.file, args[2]))
|
|
#endif
|
|
{
|
|
PrintError("can not open input file");
|
|
return 1;
|
|
}
|
|
|
|
FileInStream_CreateVTable(&archiveStream);
|
|
LookToRead_CreateVTable(&lookStream, False);
|
|
|
|
lookStream.realStream = &archiveStream.s;
|
|
LookToRead_Init(&lookStream);
|
|
|
|
CrcGenerateTable();
|
|
|
|
SzArEx_Init(&db);
|
|
res = SzArEx_Open(&db, &lookStream.s, &allocImp, &allocTempImp);
|
|
if (res == SZ_OK)
|
|
{
|
|
char *command = args[1];
|
|
int listCommand = 0, testCommand = 0, fullPaths = 0;
|
|
if (strcmp(command, "l") == 0) listCommand = 1;
|
|
else if (strcmp(command, "t") == 0) testCommand = 1;
|
|
else if (strcmp(command, "e") == 0) { }
|
|
else if (strcmp(command, "x") == 0) { fullPaths = 1; }
|
|
else
|
|
{
|
|
PrintError("incorrect command");
|
|
res = SZ_ERROR_FAIL;
|
|
}
|
|
|
|
if (res == SZ_OK)
|
|
{
|
|
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) */
|
|
|
|
for (i = 0; i < db.NumFiles; i++)
|
|
{
|
|
size_t offset = 0;
|
|
size_t outSizeProcessed = 0;
|
|
// const CSzFileItem *f = db.Files + i;
|
|
size_t len;
|
|
int isDir = SzArEx_IsDir(&db, i);
|
|
if (listCommand == 0 && isDir && !fullPaths)
|
|
continue;
|
|
len = SzArEx_GetFileNameUtf16(&db, i, NULL);
|
|
// len = SzArEx_GetFullNameLen(&db, i);
|
|
|
|
if (len > tempSize)
|
|
{
|
|
SzFree(NULL, temp);
|
|
tempSize = len;
|
|
temp = (UInt16 *)SzAlloc(NULL, tempSize * sizeof(temp[0]));
|
|
if (!temp)
|
|
{
|
|
res = SZ_ERROR_MEM;
|
|
break;
|
|
}
|
|
}
|
|
|
|
SzArEx_GetFileNameUtf16(&db, i, temp);
|
|
/*
|
|
if (SzArEx_GetFullNameUtf16_Back(&db, i, temp + len) != temp)
|
|
{
|
|
res = SZ_ERROR_FAIL;
|
|
break;
|
|
}
|
|
*/
|
|
|
|
if (listCommand)
|
|
{
|
|
char attr[8], s[32], t[32];
|
|
UInt64 fileSize;
|
|
|
|
GetAttribString(SzBitWithVals_Check(&db.Attribs, i) ? db.Attribs.Vals[i] : 0, isDir, attr);
|
|
|
|
fileSize = SzArEx_GetFileSize(&db, i);
|
|
UInt64ToStr(fileSize, s);
|
|
if (SzBitWithVals_Check(&db.MTime, i))
|
|
ConvertFileTimeToString(&db.MTime.Vals[i], t);
|
|
else
|
|
{
|
|
size_t j;
|
|
for (j = 0; j < 19; j++)
|
|
t[j] = ' ';
|
|
t[j] = '\0';
|
|
}
|
|
|
|
printf("%s %s %10s ", t, attr, s);
|
|
res = PrintString(temp);
|
|
if (res != SZ_OK)
|
|
break;
|
|
if (isDir)
|
|
printf("/");
|
|
printf("\n");
|
|
continue;
|
|
}
|
|
fputs(testCommand ?
|
|
"Testing ":
|
|
"Extracting ",
|
|
stdout);
|
|
res = PrintString(temp);
|
|
if (res != SZ_OK)
|
|
break;
|
|
if (isDir)
|
|
printf("/");
|
|
else
|
|
{
|
|
res = SzArEx_Extract(&db, &lookStream.s, i,
|
|
&blockIndex, &outBuffer, &outBufferSize,
|
|
&offset, &outSizeProcessed,
|
|
&allocImp, &allocTempImp);
|
|
if (res != SZ_OK)
|
|
break;
|
|
}
|
|
if (!testCommand)
|
|
{
|
|
CSzFile outFile;
|
|
size_t processedSize;
|
|
size_t j;
|
|
UInt16 *name = (UInt16 *)temp;
|
|
const UInt16 *destPath = (const UInt16 *)name;
|
|
for (j = 0; name[j] != 0; j++)
|
|
if (name[j] == '/')
|
|
{
|
|
if (fullPaths)
|
|
{
|
|
name[j] = 0;
|
|
MyCreateDir(name);
|
|
name[j] = CHAR_PATH_SEPARATOR;
|
|
}
|
|
else
|
|
destPath = name + j + 1;
|
|
}
|
|
|
|
if (isDir)
|
|
{
|
|
MyCreateDir(destPath);
|
|
printf("\n");
|
|
continue;
|
|
}
|
|
else if (OutFile_OpenUtf16(&outFile, destPath))
|
|
{
|
|
PrintError("can not open output file");
|
|
res = SZ_ERROR_FAIL;
|
|
break;
|
|
}
|
|
processedSize = outSizeProcessed;
|
|
if (File_Write(&outFile, outBuffer + offset, &processedSize) != 0 || processedSize != outSizeProcessed)
|
|
{
|
|
PrintError("can not write output file");
|
|
res = SZ_ERROR_FAIL;
|
|
break;
|
|
}
|
|
if (File_Close(&outFile))
|
|
{
|
|
PrintError("can not close output file");
|
|
res = SZ_ERROR_FAIL;
|
|
break;
|
|
}
|
|
#ifdef USE_WINDOWS_FILE
|
|
if (SzBitWithVals_Check(&db.Attribs, i))
|
|
SetFileAttributesW(destPath, db.Attribs.Vals[i]);
|
|
#endif
|
|
}
|
|
printf("\n");
|
|
}
|
|
IAlloc_Free(&allocImp, outBuffer);
|
|
}
|
|
}
|
|
SzArEx_Free(&db, &allocImp);
|
|
SzFree(NULL, temp);
|
|
|
|
File_Close(&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;
|
|
}
|