mirror of
https://github.com/Xevion/easy7zip.git
synced 2026-01-31 20:24:05 -06:00
4.59 beta
This commit is contained in:
Igor Pavlov
committed by
Kornel Lesiński
Kornel Lesiński
parent
3901bf0ab8
commit
173c07e166
Executable
+303
@@ -0,0 +1,303 @@
|
||||
// RpmHandler.cpp
|
||||
|
||||
#include "StdAfx.h"
|
||||
|
||||
#include "../../../C/CpuArch.h"
|
||||
|
||||
#include "Common/ComTry.h"
|
||||
#include "Common/MyString.h"
|
||||
|
||||
#include "Windows/PropVariant.h"
|
||||
|
||||
#include "../Common/ProgressUtils.h"
|
||||
#include "../Common/RegisterArc.h"
|
||||
#include "../Common/StreamUtils.h"
|
||||
|
||||
#include "../Compress/Copy/CopyCoder.h"
|
||||
|
||||
using namespace NWindows;
|
||||
|
||||
#define Get16(p) GetBe16(p)
|
||||
#define Get32(p) GetBe32(p)
|
||||
|
||||
namespace NArchive {
|
||||
namespace NRpm {
|
||||
|
||||
/* Reference: lib/signature.h of rpm package */
|
||||
#define RPMSIG_NONE 0 /* Do not change! */
|
||||
/* The following types are no longer generated */
|
||||
#define RPMSIG_PGP262_1024 1 /* No longer generated */ /* 256 byte */
|
||||
/* These are the new-style signatures. They are Header structures. */
|
||||
/* Inside them we can put any number of any type of signature we like. */
|
||||
|
||||
#define RPMSIG_HEADERSIG 5 /* New Header style signature */
|
||||
|
||||
const UInt32 kLeadSize = 96;
|
||||
struct CLead
|
||||
{
|
||||
unsigned char Magic[4];
|
||||
unsigned char Major; // not supported ver1, only support 2,3 and lator
|
||||
unsigned char Minor;
|
||||
UInt16 Type;
|
||||
UInt16 ArchNum;
|
||||
char Name[66];
|
||||
UInt16 OSNum;
|
||||
UInt16 SignatureType;
|
||||
char Reserved[16]; // pad to 96 bytes -- 8 byte aligned
|
||||
bool MagicCheck() const
|
||||
{ return Magic[0] == 0xed && Magic[1] == 0xab && Magic[2] == 0xee && Magic[3] == 0xdb; };
|
||||
};
|
||||
|
||||
const UInt32 kEntryInfoSize = 16;
|
||||
/*
|
||||
struct CEntryInfo
|
||||
{
|
||||
int Tag;
|
||||
int Type;
|
||||
int Offset; // Offset from beginning of data segment, only defined on disk
|
||||
int Count;
|
||||
};
|
||||
*/
|
||||
|
||||
// case: SignatureType == RPMSIG_HEADERSIG
|
||||
const UInt32 kCSigHeaderSigSize = 16;
|
||||
struct CSigHeaderSig
|
||||
{
|
||||
unsigned char Magic[4];
|
||||
UInt32 Reserved;
|
||||
UInt32 IndexLen; // count of index entries
|
||||
UInt32 DataLen; // number of bytes
|
||||
bool MagicCheck()
|
||||
{ return Magic[0] == 0x8e && Magic[1] == 0xad && Magic[2] == 0xe8 && Magic[3] == 0x01; };
|
||||
UInt32 GetLostHeaderLen()
|
||||
{ return IndexLen * kEntryInfoSize + DataLen; };
|
||||
};
|
||||
|
||||
static HRESULT RedSigHeaderSig(IInStream *inStream, CSigHeaderSig &h)
|
||||
{
|
||||
char dat[kCSigHeaderSigSize];
|
||||
char *cur = dat;
|
||||
RINOK(ReadStream_FALSE(inStream, dat, kCSigHeaderSigSize));
|
||||
memmove(h.Magic, cur, 4);
|
||||
cur += 4;
|
||||
cur += 4;
|
||||
h.IndexLen = Get32(cur);
|
||||
cur += 4;
|
||||
h.DataLen = Get32(cur);
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
HRESULT OpenArchive(IInStream *inStream)
|
||||
{
|
||||
UInt64 pos;
|
||||
char leadData[kLeadSize];
|
||||
char *cur = leadData;
|
||||
CLead lead;
|
||||
RINOK(ReadStream_FALSE(inStream, leadData, kLeadSize));
|
||||
memmove(lead.Magic, cur, 4);
|
||||
cur += 4;
|
||||
lead.Major = *cur++;
|
||||
lead.Minor = *cur++;
|
||||
lead.Type = Get16(cur);
|
||||
cur += 2;
|
||||
lead.ArchNum = Get16(cur);
|
||||
cur += 2;
|
||||
memmove(lead.Name, cur, sizeof(lead.Name));
|
||||
cur += sizeof(lead.Name);
|
||||
lead.OSNum = Get16(cur);
|
||||
cur += 2;
|
||||
lead.SignatureType = Get16(cur);
|
||||
cur += 2;
|
||||
|
||||
if (!lead.MagicCheck() || lead.Major < 3)
|
||||
return S_FALSE;
|
||||
|
||||
CSigHeaderSig sigHeader, header;
|
||||
if(lead.SignatureType == RPMSIG_NONE)
|
||||
{
|
||||
;
|
||||
}
|
||||
else if(lead.SignatureType == RPMSIG_PGP262_1024)
|
||||
{
|
||||
UInt64 pos;
|
||||
RINOK(inStream->Seek(256, STREAM_SEEK_CUR, &pos));
|
||||
}
|
||||
else if(lead.SignatureType == RPMSIG_HEADERSIG)
|
||||
{
|
||||
RINOK(RedSigHeaderSig(inStream, sigHeader));
|
||||
if(!sigHeader.MagicCheck())
|
||||
return S_FALSE;
|
||||
UInt32 len = sigHeader.GetLostHeaderLen();
|
||||
RINOK(inStream->Seek(len, STREAM_SEEK_CUR, &pos));
|
||||
if((pos % 8) != 0)
|
||||
{
|
||||
RINOK(inStream->Seek((pos / 8 + 1) * 8 - pos,
|
||||
STREAM_SEEK_CUR, &pos));
|
||||
}
|
||||
}
|
||||
else
|
||||
return S_FALSE;
|
||||
|
||||
RINOK(RedSigHeaderSig(inStream, header));
|
||||
if(!header.MagicCheck())
|
||||
return S_FALSE;
|
||||
int headerLen = header.GetLostHeaderLen();
|
||||
if(headerLen == -1)
|
||||
return S_FALSE;
|
||||
RINOK(inStream->Seek(headerLen, STREAM_SEEK_CUR, &pos));
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
class CHandler:
|
||||
public IInArchive,
|
||||
public CMyUnknownImp
|
||||
{
|
||||
public:
|
||||
MY_UNKNOWN_IMP1(IInArchive)
|
||||
|
||||
INTERFACE_IInArchive(;)
|
||||
|
||||
private:
|
||||
CMyComPtr<IInStream> m_InStream;
|
||||
UInt64 m_Pos;
|
||||
UInt64 m_Size;
|
||||
Byte _sig[4];
|
||||
};
|
||||
|
||||
STATPROPSTG kProps[] =
|
||||
{
|
||||
{ NULL, kpidSize, VT_UI8},
|
||||
{ NULL, kpidPackSize, VT_UI8}
|
||||
};
|
||||
|
||||
IMP_IInArchive_Props
|
||||
IMP_IInArchive_ArcProps_NO
|
||||
|
||||
STDMETHODIMP CHandler::Open(IInStream *inStream,
|
||||
const UInt64 * /* maxCheckStartPosition */,
|
||||
IArchiveOpenCallback * /* openArchiveCallback */)
|
||||
{
|
||||
COM_TRY_BEGIN
|
||||
try
|
||||
{
|
||||
if(OpenArchive(inStream) != S_OK)
|
||||
return S_FALSE;
|
||||
RINOK(inStream->Seek(0, STREAM_SEEK_CUR, &m_Pos));
|
||||
UInt64 endPosition;
|
||||
RINOK(inStream->Seek(0, STREAM_SEEK_END, &endPosition));
|
||||
m_Size = endPosition - m_Pos;
|
||||
|
||||
RINOK(inStream->Seek(m_Pos, STREAM_SEEK_SET, NULL));
|
||||
RINOK(ReadStream_FALSE(inStream, _sig, sizeof(_sig) / sizeof(_sig[0])));
|
||||
|
||||
m_InStream = inStream;
|
||||
return S_OK;
|
||||
}
|
||||
catch(...)
|
||||
{
|
||||
return S_FALSE;
|
||||
}
|
||||
COM_TRY_END
|
||||
}
|
||||
|
||||
STDMETHODIMP CHandler::Close()
|
||||
{
|
||||
m_InStream.Release();
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
STDMETHODIMP CHandler::GetNumberOfItems(UInt32 *numItems)
|
||||
{
|
||||
*numItems = 1;
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
STDMETHODIMP CHandler::GetProperty(UInt32 /* index */, PROPID propID, PROPVARIANT *value)
|
||||
{
|
||||
NWindows::NCOM::CPropVariant prop;
|
||||
switch(propID)
|
||||
{
|
||||
case kpidSize:
|
||||
case kpidPackSize:
|
||||
prop = m_Size;
|
||||
break;
|
||||
case kpidExtension:
|
||||
{
|
||||
wchar_t s[32];
|
||||
MyStringCopy(s, L"cpio.");
|
||||
const wchar_t *ext;
|
||||
if (_sig[0] == 0x1F && _sig[1] == 0x8B)
|
||||
ext = L"gz";
|
||||
else if (_sig[0] == 'B' && _sig[1] == 'Z' && _sig[2] == 'h')
|
||||
ext = L"bz2";
|
||||
else
|
||||
ext = L"lzma";
|
||||
MyStringCopy(s + MyStringLen(s), ext);
|
||||
prop = s;
|
||||
break;
|
||||
}
|
||||
}
|
||||
prop.Detach(value);
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
STDMETHODIMP CHandler::Extract(const UInt32* indices, UInt32 numItems,
|
||||
Int32 _aTestMode, IArchiveExtractCallback *extractCallback)
|
||||
{
|
||||
COM_TRY_BEGIN
|
||||
bool allFilesMode = (numItems == UInt32(-1));
|
||||
if (allFilesMode)
|
||||
numItems = 1;
|
||||
if(numItems == 0)
|
||||
return S_OK;
|
||||
if(numItems != 1)
|
||||
return E_FAIL;
|
||||
if (indices[0] != 0)
|
||||
return E_FAIL;
|
||||
|
||||
bool testMode = (_aTestMode != 0);
|
||||
|
||||
UInt64 currentTotalSize = 0;
|
||||
RINOK(extractCallback->SetTotal(m_Size));
|
||||
RINOK(extractCallback->SetCompleted(¤tTotalSize));
|
||||
CMyComPtr<ISequentialOutStream> realOutStream;
|
||||
Int32 askMode = testMode ?
|
||||
NArchive::NExtract::NAskMode::kTest :
|
||||
NArchive::NExtract::NAskMode::kExtract;
|
||||
Int32 index = 0;
|
||||
|
||||
RINOK(extractCallback->GetStream(index, &realOutStream, askMode));
|
||||
|
||||
if(!testMode && (!realOutStream))
|
||||
return S_OK;
|
||||
|
||||
RINOK(extractCallback->PrepareOperation(askMode));
|
||||
|
||||
if (testMode)
|
||||
{
|
||||
RINOK(extractCallback->SetOperationResult(NArchive::NExtract::NOperationResult::kOK));
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
RINOK(m_InStream->Seek(m_Pos, STREAM_SEEK_SET, NULL));
|
||||
|
||||
CMyComPtr<ICompressCoder> copyCoder = new NCompress::CCopyCoder;
|
||||
|
||||
CLocalProgress *lps = new CLocalProgress;
|
||||
CMyComPtr<ICompressProgressInfo> progress = lps;
|
||||
lps->Init(extractCallback, false);
|
||||
|
||||
RINOK(copyCoder->Code(m_InStream, realOutStream, NULL, NULL, progress));
|
||||
realOutStream.Release();
|
||||
return extractCallback->SetOperationResult(NArchive::NExtract::NOperationResult::kOK);
|
||||
COM_TRY_END
|
||||
}
|
||||
|
||||
static IInArchive *CreateArc() { return new NArchive::NRpm::CHandler; }
|
||||
|
||||
static CArcInfo g_ArcInfo =
|
||||
{ L"Rpm", L"rpm", 0, 0xEB, { 0}, 0, false, CreateArc, 0 };
|
||||
|
||||
REGISTER_ARC(Rpm)
|
||||
|
||||
}}
|
||||
Reference in New Issue
Block a user