438 lines
9.6 KiB
C
438 lines
9.6 KiB
C
/* 7zMain.c - Test application for 7z Decoder
|
||
2010-10-28 : Igor Pavlov : Public domain */
|
||
|
||
#include <stdio.h>
|
||
#include <string.h>
|
||
|
||
#include "7z/7z.h"
|
||
#include "7z/7zAlloc.h"
|
||
#include "7z/7zCrc.h"
|
||
#include "7z/7zFile.h"
|
||
#include "7z/7zVersion.h"
|
||
|
||
/*
|
||
#ifndef UNICODE
|
||
#define UNICODE
|
||
#endif
|
||
*/
|
||
#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, int fileMode)
|
||
{
|
||
int len = 0;
|
||
for (len = 0; s[len] != '\0'; len++);
|
||
|
||
#ifdef _WIN32
|
||
{
|
||
int size = len * 3 + 100;
|
||
if (!Buf_EnsureSize(buf, size))
|
||
return SZ_ERROR_MEM;
|
||
{
|
||
char defaultChar = '_';
|
||
BOOL defUsed;
|
||
int numChars = WideCharToMultiByte(fileMode ?
|
||
(
|
||
#ifdef UNDER_CE
|
||
CP_ACP
|
||
#else
|
||
AreFileApisANSI() ? CP_ACP : CP_OEMCP
|
||
#endif
|
||
) : CP_OEMCP,
|
||
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
|
||
fileMode = fileMode;
|
||
return Utf16_To_Utf8Buf(buf, s, len);
|
||
#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, 1));
|
||
|
||
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, 1));
|
||
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, 0);
|
||
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;
|
||
}
|
||
|
||
#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 *ft, char *s)
|
||
{
|
||
unsigned year, mon, day, hour, min, sec;
|
||
UInt64 v64 = (ft->Low | ((UInt64)ft->High << 32)) / 10000000;
|
||
Byte ms[] = { 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31 };
|
||
unsigned t;
|
||
UInt32 v;
|
||
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 = 1; mon <= 12; mon++)
|
||
{
|
||
unsigned s = ms[mon - 1];
|
||
if (v < s)
|
||
break;
|
||
v -= s;
|
||
}
|
||
day = (unsigned)v + 1;
|
||
s = UIntToStr(s, year, 4); *s++ = '-';
|
||
s = UIntToStr(s, mon, 2); *s++ = '-';
|
||
s = UIntToStr(s, day, 2); *s++ = ' ';
|
||
s = UIntToStr(s, hour, 2); *s++ = ':';
|
||
s = UIntToStr(s, min, 2); *s++ = ':';
|
||
s = UIntToStr(s, sec, 2);
|
||
}
|
||
|
||
void PrintError(char *sz)
|
||
{
|
||
printf("\nERROR: %s\n", sz);
|
||
}
|
||
|
||
#ifdef USE_WINDOWS_FILE
|
||
#define kEmptyAttribChar '.'
|
||
static void GetAttribString(UInt32 wa, Bool isDir, char *s)
|
||
{
|
||
s[0] = (char)(((wa & FILE_ATTRIBUTE_DIRECTORY) != 0 || isDir) ? 'D' : kEmptyAttribChar);
|
||
s[1] = (char)(((wa & FILE_ATTRIBUTE_READONLY) != 0) ? 'R': kEmptyAttribChar);
|
||
s[2] = (char)(((wa & FILE_ATTRIBUTE_HIDDEN) != 0) ? 'H': kEmptyAttribChar);
|
||
s[3] = (char)(((wa & FILE_ATTRIBUTE_SYSTEM) != 0) ? 'S': kEmptyAttribChar);
|
||
s[4] = (char)(((wa & FILE_ATTRIBUTE_ARCHIVE) != 0) ? 'A': kEmptyAttribChar);
|
||
s[5] = '\0';
|
||
}
|
||
#else
|
||
static void GetAttribString(UInt32, Bool, char *s)
|
||
{
|
||
s[0] = '\0';
|
||
}
|
||
#endif
|
||
|
||
int SevenZipUnCompress( char *fname, unsigned char ** ppBuf,size_t * lpdwSize)
|
||
{
|
||
CFileInStream archiveStream;
|
||
CLookToRead lookStream;
|
||
CSzArEx db;
|
||
SRes res;
|
||
ISzAlloc allocImp;
|
||
ISzAlloc allocTempImp;
|
||
UInt16 *temp = NULL;
|
||
size_t tempSize = 0;
|
||
|
||
int bHasRomFile = 0;
|
||
int bExtractRomOK = 0;
|
||
//ö<><C3B6><EFBFBD>ļ<EFBFBD><C4BC><EFBFBD><EFBFBD>ҵ<EFBFBD>NES<45><53><EFBFBD><EFBFBD>UNF<4E>ļ<EFBFBD><C4BC><EFBFBD><EFBFBD>Ҳ<EFBFBD><D2B2><EFBFBD>ֱ<EFBFBD>ӷ<EFBFBD><D3B7><EFBFBD>
|
||
allocImp.Alloc = SzAlloc;
|
||
allocImp.Free = SzFree;
|
||
|
||
allocTempImp.Alloc = SzAllocTemp;
|
||
allocTempImp.Free = SzFreeTemp;
|
||
|
||
if (InFile_Open(&archiveStream.file, fname))
|
||
{
|
||
return FALSE;
|
||
}
|
||
|
||
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)
|
||
{
|
||
UInt32 i;
|
||
|
||
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. */
|
||
|
||
for (i = 0; i < db.db.NumFiles; i++)
|
||
{
|
||
size_t offset = 0;
|
||
size_t outSizeProcessed = 0;
|
||
const CSzFileItem *f = db.db.Files + i;
|
||
size_t len;
|
||
|
||
len = SzArEx_GetFileNameUtf16(&db, i, NULL);
|
||
|
||
if (len > tempSize)
|
||
{
|
||
SzFree(NULL, temp);
|
||
tempSize = len;
|
||
temp = (UInt16 *)SzAlloc(NULL, tempSize * sizeof(temp[0]));
|
||
if (temp == 0)
|
||
{
|
||
res = SZ_ERROR_MEM;
|
||
break;
|
||
}
|
||
}
|
||
|
||
SzArEx_GetFileNameUtf16(&db, i, temp);
|
||
//<2F><>ȡ<EFBFBD>ļ<EFBFBD><C4BC><EFBFBD>Ϣ
|
||
{
|
||
char attr[8];
|
||
GetAttribString(f->AttribDefined ? f->Attrib : 0, f->IsDir, attr);// f->IsDir <20>ļ<EFBFBD><C4BC>Ƿ<EFBFBD><C7B7><EFBFBD>Ŀ¼
|
||
|
||
//UInt64ToStr(f->Size, s);//f->Size <20>ļ<EFBFBD><C4BC><EFBFBD>С
|
||
if(f->IsDir!=1)
|
||
{
|
||
//<2F><><EFBFBD><EFBFBD><EFBFBD><EFBFBD>չ<EFBFBD><D5B9>
|
||
if(len>5)
|
||
{
|
||
//"*.nes",
|
||
//"*.unf",
|
||
//"*.fds",
|
||
//"*.nsf",
|
||
if(wcsicmp(L".nes",&temp[len-5])==0)
|
||
{
|
||
bHasRomFile =1;
|
||
}
|
||
else if(wcsicmp(L".unf",&temp[len-5])==0)
|
||
{
|
||
bHasRomFile =1;
|
||
}
|
||
else if(wcsicmp(L".fds",&temp[len-5])==0)
|
||
{
|
||
bHasRomFile =1;
|
||
}
|
||
else if(wcsicmp(L".nsf",&temp[len-5])==0)
|
||
{
|
||
bHasRomFile =1;
|
||
}
|
||
}
|
||
//if( (f->Size>16000)&&(f->Size<10000000) )
|
||
//bHasRomFile =1;
|
||
}
|
||
|
||
if(bHasRomFile==0)
|
||
continue;
|
||
}
|
||
|
||
if (bHasRomFile)
|
||
{
|
||
res = SzArEx_Extract(&db, &lookStream.s, i,&blockIndex, ppBuf, lpdwSize,&offset, &outSizeProcessed,&allocImp, &allocTempImp);
|
||
if (res != SZ_OK)
|
||
continue;
|
||
else
|
||
{
|
||
bExtractRomOK = 1;
|
||
if(offset!=0)
|
||
{//<2F><><EFBFBD><EFBFBD>
|
||
void *pRom = malloc(f->Size);
|
||
memcpy(pRom,*ppBuf+offset,f->Size);
|
||
*lpdwSize = f->Size;
|
||
free(*ppBuf);
|
||
*ppBuf = pRom;
|
||
}
|
||
}
|
||
}
|
||
if(bExtractRomOK)
|
||
break;
|
||
}//<2F><><EFBFBD><EFBFBD><EFBFBD>ļ<EFBFBD>ö<EFBFBD><C3B6>ѭ<EFBFBD><D1AD>
|
||
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;
|
||
}*/
|
||
return bExtractRomOK;
|
||
}
|
||
|
||
/*
|
||
|
||
int MY_CDECL main(int c, char *v[])
|
||
{
|
||
unsigned char *pBuf = 0;
|
||
size_t fileLen;
|
||
int i = SevenZipUnCompress(v[1],&pBuf,&fileLen);
|
||
free(pBuf); pBuf = 0;
|
||
printf("%d",i);
|
||
}
|
||
*/ |