• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 // ArchiveExports.cpp
2 
3 #include "StdAfx.h"
4 
5 #include "../../../C/7zVersion.h"
6 
7 #include "../../Common/ComTry.h"
8 
9 #include "../../Windows/PropVariant.h"
10 
11 #include "../Common/RegisterArc.h"
12 
13 static const unsigned kNumArcsMax = 64;
14 static unsigned g_NumArcs = 0;
15 static unsigned g_DefaultArcIndex = 0;
16 static const CArcInfo *g_Arcs[kNumArcsMax];
17 
RegisterArc(const CArcInfo * arcInfo)18 void RegisterArc(const CArcInfo *arcInfo) throw()
19 {
20   if (g_NumArcs < kNumArcsMax)
21   {
22     const char *p = arcInfo->Name;
23     if (p[0] == '7' && p[1] == 'z' && p[2] == 0)
24       g_DefaultArcIndex = g_NumArcs;
25     g_Arcs[g_NumArcs++] = arcInfo;
26   }
27 }
28 
29 DEFINE_GUID(CLSID_CArchiveHandler,
30     k_7zip_GUID_Data1,
31     k_7zip_GUID_Data2,
32     k_7zip_GUID_Data3_Common,
33     0x10, 0x00, 0x00, 0x01, 0x10, 0x00, 0x00, 0x00);
34 
35 #define CLS_ARC_ID_ITEM(cls) ((cls).Data4[5])
36 
SetPropStrFromBin(const char * s,unsigned size,PROPVARIANT * value)37 static inline HRESULT SetPropStrFromBin(const char *s, unsigned size, PROPVARIANT *value)
38 {
39   if ((value->bstrVal = ::SysAllocStringByteLen(s, size)) != 0)
40     value->vt = VT_BSTR;
41   return S_OK;
42 }
43 
SetPropGUID(const GUID & guid,PROPVARIANT * value)44 static inline HRESULT SetPropGUID(const GUID &guid, PROPVARIANT *value)
45 {
46   return SetPropStrFromBin((const char *)&guid, sizeof(guid), value);
47 }
48 
FindFormatCalssId(const GUID * clsid)49 int FindFormatCalssId(const GUID *clsid)
50 {
51   GUID cls = *clsid;
52   CLS_ARC_ID_ITEM(cls) = 0;
53   if (cls != CLSID_CArchiveHandler)
54     return -1;
55   Byte id = CLS_ARC_ID_ITEM(*clsid);
56   for (unsigned i = 0; i < g_NumArcs; i++)
57     if (g_Arcs[i]->Id == id)
58       return (int)i;
59   return -1;
60 }
61 
CreateArchiver(const GUID * clsid,const GUID * iid,void ** outObject)62 STDAPI CreateArchiver(const GUID *clsid, const GUID *iid, void **outObject)
63 {
64   COM_TRY_BEGIN
65   {
66     int needIn = (*iid == IID_IInArchive);
67     int needOut = (*iid == IID_IOutArchive);
68     if (!needIn && !needOut)
69       return E_NOINTERFACE;
70     int formatIndex = FindFormatCalssId(clsid);
71     if (formatIndex < 0)
72       return CLASS_E_CLASSNOTAVAILABLE;
73 
74     const CArcInfo &arc = *g_Arcs[formatIndex];
75     if (needIn)
76     {
77       *outObject = arc.CreateInArchive();
78       ((IInArchive *)*outObject)->AddRef();
79     }
80     else
81     {
82       if (!arc.CreateOutArchive)
83         return CLASS_E_CLASSNOTAVAILABLE;
84       *outObject = arc.CreateOutArchive();
85       ((IOutArchive *)*outObject)->AddRef();
86     }
87   }
88   COM_TRY_END
89   return S_OK;
90 }
91 
GetHandlerProperty2(UInt32 formatIndex,PROPID propID,PROPVARIANT * value)92 STDAPI GetHandlerProperty2(UInt32 formatIndex, PROPID propID, PROPVARIANT *value)
93 {
94   COM_TRY_BEGIN
95   NWindows::NCOM::PropVariant_Clear(value);
96   if (formatIndex >= g_NumArcs)
97     return E_INVALIDARG;
98   const CArcInfo &arc = *g_Arcs[formatIndex];
99   NWindows::NCOM::CPropVariant prop;
100   switch (propID)
101   {
102     case NArchive::NHandlerPropID::kName: prop = arc.Name; break;
103     case NArchive::NHandlerPropID::kClassID:
104     {
105       GUID clsId = CLSID_CArchiveHandler;
106       CLS_ARC_ID_ITEM(clsId) = arc.Id;
107       return SetPropGUID(clsId, value);
108     }
109     case NArchive::NHandlerPropID::kExtension: if (arc.Ext) prop = arc.Ext; break;
110     case NArchive::NHandlerPropID::kAddExtension: if (arc.AddExt) prop = arc.AddExt; break;
111     case NArchive::NHandlerPropID::kUpdate: prop = (bool)(arc.CreateOutArchive != NULL); break;
112     case NArchive::NHandlerPropID::kKeepName:   prop = ((arc.Flags & NArcInfoFlags::kKeepName) != 0); break;
113     case NArchive::NHandlerPropID::kAltStreams: prop = ((arc.Flags & NArcInfoFlags::kAltStreams) != 0); break;
114     case NArchive::NHandlerPropID::kNtSecure:   prop = ((arc.Flags & NArcInfoFlags::kNtSecure) != 0); break;
115     case NArchive::NHandlerPropID::kFlags: prop = (UInt32)arc.Flags; break;
116     case NArchive::NHandlerPropID::kSignatureOffset: prop = (UInt32)arc.SignatureOffset; break;
117     // case NArchive::NHandlerPropID::kVersion: prop = (UInt32)MY_VER_MIX; break;
118 
119     case NArchive::NHandlerPropID::kSignature:
120       if (arc.SignatureSize != 0 && !arc.IsMultiSignature())
121         return SetPropStrFromBin((const char *)arc.Signature, arc.SignatureSize, value);
122       break;
123     case NArchive::NHandlerPropID::kMultiSignature:
124       if (arc.SignatureSize != 0 && arc.IsMultiSignature())
125         return SetPropStrFromBin((const char *)arc.Signature, arc.SignatureSize, value);
126       break;
127   }
128   prop.Detach(value);
129   return S_OK;
130   COM_TRY_END
131 }
132 
GetHandlerProperty(PROPID propID,PROPVARIANT * value)133 STDAPI GetHandlerProperty(PROPID propID, PROPVARIANT *value)
134 {
135   return GetHandlerProperty2(g_DefaultArcIndex, propID, value);
136 }
137 
GetNumberOfFormats(UINT32 * numFormats)138 STDAPI GetNumberOfFormats(UINT32 *numFormats)
139 {
140   *numFormats = g_NumArcs;
141   return S_OK;
142 }
143 
GetIsArc(UInt32 formatIndex,Func_IsArc * isArc)144 STDAPI GetIsArc(UInt32 formatIndex, Func_IsArc *isArc)
145 {
146   *isArc = NULL;
147   if (formatIndex >= g_NumArcs)
148     return E_INVALIDARG;
149   *isArc = g_Arcs[formatIndex]->IsArc;
150   return S_OK;
151 }
152