mirror of
https://github.com/Xevion/easy7zip.git
synced 2025-12-06 13:14:59 -06:00
9.34
This commit is contained in:
Igor Pavlov
committed by
Kornel Lesiński
Kornel Lesiński
parent
83f8ddcc5b
commit
f08f4dcc3c
179
CPP/7zip/Compress/CodecExports.cpp
Executable file → Normal file
179
CPP/7zip/Compress/CodecExports.cpp
Executable file → Normal file
@@ -2,7 +2,10 @@
|
||||
|
||||
#include "StdAfx.h"
|
||||
|
||||
#include "../../../C/CpuArch.h"
|
||||
|
||||
#include "../../Common/ComTry.h"
|
||||
#include "../../Common/MyCom.h"
|
||||
|
||||
#include "../../Windows/PropVariant.h"
|
||||
|
||||
@@ -13,10 +16,15 @@
|
||||
extern unsigned int g_NumCodecs;
|
||||
extern const CCodecInfo *g_Codecs[];
|
||||
|
||||
extern unsigned int g_NumHashers;
|
||||
extern const CHasherInfo *g_Hashers[];
|
||||
|
||||
static const UInt16 kDecodeId = 0x2790;
|
||||
static const UInt16 kEncodeId = 0x2791;
|
||||
static const UInt16 kHasherId = 0x2792;
|
||||
|
||||
DEFINE_GUID(CLSID_CCodec,
|
||||
0x23170F69, 0x40C1, kDecodeId, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00);
|
||||
0x23170F69, 0x40C1, kDecodeId, 0, 0, 0, 0, 0, 0, 0, 0);
|
||||
|
||||
static inline HRESULT SetPropString(const char *s, unsigned int size, PROPVARIANT *value)
|
||||
{
|
||||
@@ -30,13 +38,13 @@ static inline HRESULT SetPropGUID(const GUID &guid, PROPVARIANT *value)
|
||||
return SetPropString((const char *)&guid, sizeof(GUID), value);
|
||||
}
|
||||
|
||||
static HRESULT SetClassID(CMethodId id, bool encode, PROPVARIANT *value)
|
||||
static HRESULT SetClassID(CMethodId id, UInt16 typeId, PROPVARIANT *value)
|
||||
{
|
||||
GUID clsId = CLSID_CCodec;
|
||||
for (int i = 0; i < sizeof(id); i++, id >>= 8)
|
||||
clsId.Data4[i] = (Byte)(id & 0xFF);
|
||||
if (encode)
|
||||
clsId.Data3++;
|
||||
GUID clsId;
|
||||
clsId.Data1 = CLSID_CCodec.Data1;
|
||||
clsId.Data2 = CLSID_CCodec.Data2;
|
||||
clsId.Data3 = typeId;
|
||||
SetUi64(clsId.Data4, id);
|
||||
return SetPropGUID(clsId, value);
|
||||
}
|
||||
|
||||
@@ -44,13 +52,14 @@ static HRESULT FindCodecClassId(const GUID *clsID, UInt32 isCoder2, bool isFilte
|
||||
{
|
||||
index = -1;
|
||||
if (clsID->Data1 != CLSID_CCodec.Data1 ||
|
||||
clsID->Data2 != CLSID_CCodec.Data2 ||
|
||||
(clsID->Data3 & ~1) != kDecodeId)
|
||||
clsID->Data2 != CLSID_CCodec.Data2)
|
||||
return S_OK;
|
||||
encode = (clsID->Data3 != kDecodeId);
|
||||
UInt64 id = 0;
|
||||
for (int j = 0; j < 8; j++)
|
||||
id |= ((UInt64)clsID->Data4[j]) << (8 * j);
|
||||
encode = true;
|
||||
if (clsID->Data3 == kDecodeId)
|
||||
encode = false;
|
||||
else if (clsID->Data3 != kEncodeId)
|
||||
return S_OK;
|
||||
UInt64 id = GetUi64(clsID->Data4);
|
||||
for (unsigned i = 0; i < g_NumCodecs; i++)
|
||||
{
|
||||
const CCodecInfo &codec = *g_Codecs[i];
|
||||
@@ -65,7 +74,7 @@ static HRESULT FindCodecClassId(const GUID *clsID, UInt32 isCoder2, bool isFilte
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
STDAPI CreateCoder2(bool encode, UInt32 index, const GUID *iid, void **outObject)
|
||||
STDAPI CreateCoder2(bool encode, int index, const GUID *iid, void **outObject)
|
||||
{
|
||||
COM_TRY_BEGIN
|
||||
*outObject = 0;
|
||||
@@ -88,18 +97,22 @@ STDAPI CreateCoder2(bool encode, UInt32 index, const GUID *iid, void **outObject
|
||||
return CLASS_E_CLASSNOTAVAILABLE;
|
||||
*outObject = codec.CreateDecoder();
|
||||
}
|
||||
if (isCoder)
|
||||
((ICompressCoder *)*outObject)->AddRef();
|
||||
else if (isCoder2)
|
||||
((ICompressCoder2 *)*outObject)->AddRef();
|
||||
else
|
||||
((ICompressFilter *)*outObject)->AddRef();
|
||||
if (*outObject)
|
||||
{
|
||||
if (isCoder)
|
||||
((ICompressCoder *)*outObject)->AddRef();
|
||||
else if (isCoder2)
|
||||
((ICompressCoder2 *)*outObject)->AddRef();
|
||||
else
|
||||
((ICompressFilter *)*outObject)->AddRef();
|
||||
}
|
||||
return S_OK;
|
||||
COM_TRY_END
|
||||
}
|
||||
|
||||
STDAPI CreateCoder(const GUID *clsid, const GUID *iid, void **outObject)
|
||||
{
|
||||
COM_TRY_BEGIN
|
||||
*outObject = 0;
|
||||
bool isCoder = (*iid == IID_ICompressCoder) != 0;
|
||||
bool isCoder2 = (*iid == IID_ICompressCoder2) != 0;
|
||||
@@ -113,42 +126,54 @@ STDAPI CreateCoder(const GUID *clsid, const GUID *iid, void **outObject)
|
||||
return res;
|
||||
if (codecIndex < 0)
|
||||
return CLASS_E_CLASSNOTAVAILABLE;
|
||||
return CreateCoder2(encode, codecIndex, iid, outObject);
|
||||
|
||||
const CCodecInfo &codec = *g_Codecs[codecIndex];
|
||||
if (encode)
|
||||
*outObject = codec.CreateEncoder();
|
||||
else
|
||||
*outObject = codec.CreateDecoder();
|
||||
if (*outObject)
|
||||
{
|
||||
if (isCoder)
|
||||
((ICompressCoder *)*outObject)->AddRef();
|
||||
else if (isCoder2)
|
||||
((ICompressCoder2 *)*outObject)->AddRef();
|
||||
else
|
||||
((ICompressFilter *)*outObject)->AddRef();
|
||||
}
|
||||
return S_OK;
|
||||
COM_TRY_END
|
||||
}
|
||||
|
||||
STDAPI GetMethodProperty(UInt32 codecIndex, PROPID propID, PROPVARIANT *value)
|
||||
{
|
||||
::VariantClear((VARIANTARG *)value);
|
||||
const CCodecInfo &codec = *g_Codecs[codecIndex];
|
||||
switch(propID)
|
||||
switch (propID)
|
||||
{
|
||||
case NMethodPropID::kID:
|
||||
{
|
||||
value->uhVal.QuadPart = (UInt64)codec.Id;
|
||||
value->vt = VT_UI8;
|
||||
break;
|
||||
}
|
||||
case NMethodPropID::kName:
|
||||
if ((value->bstrVal = ::SysAllocString(codec.Name)) != 0)
|
||||
value->vt = VT_BSTR;
|
||||
break;
|
||||
case NMethodPropID::kDecoder:
|
||||
if (codec.CreateDecoder)
|
||||
return SetClassID(codec.Id, false, value);
|
||||
return SetClassID(codec.Id, kDecodeId, value);
|
||||
break;
|
||||
case NMethodPropID::kEncoder:
|
||||
if (codec.CreateEncoder)
|
||||
return SetClassID(codec.Id, true, value);
|
||||
return SetClassID(codec.Id, kEncodeId, value);
|
||||
break;
|
||||
case NMethodPropID::kInStreams:
|
||||
{
|
||||
if (codec.NumInStreams != 1)
|
||||
{
|
||||
value->vt = VT_UI4;
|
||||
value->ulVal = (ULONG)codec.NumInStreams;
|
||||
}
|
||||
break;
|
||||
}
|
||||
}
|
||||
return S_OK;
|
||||
}
|
||||
@@ -158,3 +183,101 @@ STDAPI GetNumberOfMethods(UINT32 *numCodecs)
|
||||
*numCodecs = g_NumCodecs;
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
|
||||
static int FindHasherClassId(const GUID *clsID)
|
||||
{
|
||||
if (clsID->Data1 != CLSID_CCodec.Data1 ||
|
||||
clsID->Data2 != CLSID_CCodec.Data2 ||
|
||||
clsID->Data3 != kHasherId)
|
||||
return -1;
|
||||
UInt64 id = GetUi64(clsID->Data4);
|
||||
for (unsigned i = 0; i < g_NumCodecs; i++)
|
||||
if (id == g_Hashers[i]->Id)
|
||||
return i;
|
||||
return -1;
|
||||
}
|
||||
|
||||
static HRESULT CreateHasher2(UInt32 index, IHasher **hasher)
|
||||
{
|
||||
COM_TRY_BEGIN
|
||||
*hasher = g_Hashers[index]->CreateHasher();
|
||||
if (*hasher)
|
||||
(*hasher)->AddRef();
|
||||
return S_OK;
|
||||
COM_TRY_END
|
||||
}
|
||||
|
||||
STDAPI CreateHasher(const GUID *clsid, IHasher **outObject)
|
||||
{
|
||||
COM_TRY_BEGIN
|
||||
*outObject = 0;
|
||||
int index = FindHasherClassId(clsid);
|
||||
if (index < 0)
|
||||
return CLASS_E_CLASSNOTAVAILABLE;
|
||||
return CreateHasher2(index, outObject);
|
||||
COM_TRY_END
|
||||
}
|
||||
|
||||
STDAPI GetHasherProp(UInt32 codecIndex, PROPID propID, PROPVARIANT *value)
|
||||
{
|
||||
::VariantClear((VARIANTARG *)value);
|
||||
const CHasherInfo &codec = *g_Hashers[codecIndex];
|
||||
switch (propID)
|
||||
{
|
||||
case NMethodPropID::kID:
|
||||
value->uhVal.QuadPart = (UInt64)codec.Id;
|
||||
value->vt = VT_UI8;
|
||||
break;
|
||||
case NMethodPropID::kName:
|
||||
if ((value->bstrVal = ::SysAllocString(codec.Name)) != 0)
|
||||
value->vt = VT_BSTR;
|
||||
break;
|
||||
case NMethodPropID::kEncoder:
|
||||
if (codec.CreateHasher)
|
||||
return SetClassID(codec.Id, kHasherId, value);
|
||||
break;
|
||||
case NMethodPropID::kDigestSize:
|
||||
value->ulVal = (ULONG)codec.DigestSize;
|
||||
value->vt = VT_UI4;
|
||||
break;
|
||||
}
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
class CHashers:
|
||||
public IHashers,
|
||||
public CMyUnknownImp
|
||||
{
|
||||
public:
|
||||
MY_UNKNOWN_IMP1(IHashers)
|
||||
|
||||
STDMETHOD_(UInt32, GetNumHashers)();
|
||||
STDMETHOD(GetHasherProp)(UInt32 index, PROPID propID, PROPVARIANT *value);
|
||||
STDMETHOD(CreateHasher)(UInt32 index, IHasher **hasher);
|
||||
};
|
||||
|
||||
STDAPI GetHashers(IHashers **hashers)
|
||||
{
|
||||
COM_TRY_BEGIN
|
||||
*hashers = new CHashers;
|
||||
if (*hashers)
|
||||
(*hashers)->AddRef();
|
||||
return S_OK;
|
||||
COM_TRY_END
|
||||
}
|
||||
|
||||
STDMETHODIMP_(UInt32) CHashers::GetNumHashers()
|
||||
{
|
||||
return g_NumHashers;
|
||||
}
|
||||
|
||||
STDMETHODIMP CHashers::GetHasherProp(UInt32 index, PROPID propID, PROPVARIANT *value)
|
||||
{
|
||||
return ::GetHasherProp(index, propID, value);
|
||||
}
|
||||
|
||||
STDMETHODIMP CHashers::CreateHasher(UInt32 index, IHasher **hasher)
|
||||
{
|
||||
return ::CreateHasher2(index, hasher);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user