636 lines
18 KiB
C++
636 lines
18 KiB
C++
|
/********************************************************************
|
||
|
|
|
||
|
|
Copyright (c) 1999-2001 Microsoft Corporation
|
||
|
|
|
||
|
|
Module Name:
|
||
|
|
Package.cpp
|
||
|
|
|
||
|
|
Abstract:
|
||
|
|
Cryptography stuff for hcupdate packages
|
||
|
|
|
||
|
|
Revision History:
|
||
|
|
|
||
|
|
Ghim-Sim Chua (gschua) 07/07/99
|
||
|
|
- created
|
||
|
|
|
||
|
|
Davide Massarenti (dmassare) 03/24/2001
|
||
|
|
- moved into database code.
|
||
|
|
|
||
|
|
********************************************************************/
|
||
|
|
|
||
|
|
#include "stdafx.h"
|
||
|
|
|
||
|
|
#include <wintrust.h>
|
||
|
|
#include <wincrypt.h>
|
||
|
|
#include <sipbase.h>
|
||
|
|
#include <softpub.h>
|
||
|
|
|
||
|
|
////////////////////////////////////////////////////////////////////////////////
|
||
|
|
|
||
|
|
static void local_GetDNInfo( PCCERT_CONTEXT pCC ,
|
||
|
|
LPCSTR field ,
|
||
|
|
LPCWSTR fieldName ,
|
||
|
|
MPC::wstring& strBuf ,
|
||
|
|
MPC::wstring* strPlainName = NULL)
|
||
|
|
{
|
||
|
|
WCHAR rgTmp[512];
|
||
|
|
|
||
|
|
if(strBuf.length()) strBuf += L",";
|
||
|
|
|
||
|
|
strBuf += fieldName;
|
||
|
|
|
||
|
|
::CertGetNameStringW( pCC, CERT_NAME_ATTR_TYPE, 0, (void*)field, rgTmp, MAXSTRLEN(rgTmp) ); rgTmp[MAXSTRLEN(rgTmp)] = 0;
|
||
|
|
|
||
|
|
if(strPlainName) *strPlainName = rgTmp;
|
||
|
|
|
||
|
|
strBuf += rgTmp;
|
||
|
|
}
|
||
|
|
|
||
|
|
////////////////////////////////////////////////////////////////////////////////
|
||
|
|
|
||
|
|
Taxonomy::Package::Package()
|
||
|
|
{
|
||
|
|
// MPC::wstring m_strFileName;
|
||
|
|
m_fTemporary = false; // bool m_fTemporary;
|
||
|
|
m_lSequence = 0; // long m_lSequence;
|
||
|
|
m_dwCRC = 0; // DWORD m_dwCRC;
|
||
|
|
//
|
||
|
|
// MPC::wstring m_strSKU;
|
||
|
|
// MPC::wstring m_strLanguage;
|
||
|
|
// MPC::wstring m_strVendorID;
|
||
|
|
// MPC::wstring m_strVendorName;
|
||
|
|
// MPC::wstring m_strProductID;
|
||
|
|
// MPC::wstring m_strVersion;
|
||
|
|
//
|
||
|
|
m_fMicrosoft = false; // bool m_fMicrosoft;
|
||
|
|
m_fBuiltin = false; // bool m_fBuiltin;
|
||
|
|
}
|
||
|
|
|
||
|
|
Taxonomy::Package::~Package()
|
||
|
|
{
|
||
|
|
if(m_fTemporary) Remove( Logger() );
|
||
|
|
}
|
||
|
|
|
||
|
|
HRESULT Taxonomy::operator>>( /*[in]*/ MPC::Serializer& stream, /*[out]*/ Taxonomy::Package& val )
|
||
|
|
{
|
||
|
|
HRESULT hr;
|
||
|
|
|
||
|
|
if(SUCCEEDED(hr = (stream >> val.m_lSequence )) &&
|
||
|
|
SUCCEEDED(hr = (stream >> val.m_dwCRC )) &&
|
||
|
|
|
||
|
|
SUCCEEDED(hr = (stream >> val.m_strSKU )) &&
|
||
|
|
SUCCEEDED(hr = (stream >> val.m_strLanguage )) &&
|
||
|
|
SUCCEEDED(hr = (stream >> val.m_strVendorID )) &&
|
||
|
|
SUCCEEDED(hr = (stream >> val.m_strVendorName)) &&
|
||
|
|
SUCCEEDED(hr = (stream >> val.m_strProductID )) &&
|
||
|
|
SUCCEEDED(hr = (stream >> val.m_strVersion )) &&
|
||
|
|
|
||
|
|
SUCCEEDED(hr = (stream >> val.m_fMicrosoft )) &&
|
||
|
|
SUCCEEDED(hr = (stream >> val.m_fBuiltin )) )
|
||
|
|
{
|
||
|
|
hr = S_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
HRESULT Taxonomy::operator<<( /*[in]*/ MPC::Serializer& stream, /*[in] */ const Taxonomy::Package& val )
|
||
|
|
{
|
||
|
|
HRESULT hr;
|
||
|
|
|
||
|
|
if(SUCCEEDED(hr = (stream << val.m_lSequence )) &&
|
||
|
|
SUCCEEDED(hr = (stream << val.m_dwCRC )) &&
|
||
|
|
|
||
|
|
SUCCEEDED(hr = (stream << val.m_strSKU )) &&
|
||
|
|
SUCCEEDED(hr = (stream << val.m_strLanguage )) &&
|
||
|
|
SUCCEEDED(hr = (stream << val.m_strVendorID )) &&
|
||
|
|
SUCCEEDED(hr = (stream << val.m_strVendorName)) &&
|
||
|
|
SUCCEEDED(hr = (stream << val.m_strProductID )) &&
|
||
|
|
SUCCEEDED(hr = (stream << val.m_strVersion )) &&
|
||
|
|
|
||
|
|
SUCCEEDED(hr = (stream << val.m_fMicrosoft )) &&
|
||
|
|
SUCCEEDED(hr = (stream << val.m_fBuiltin )) )
|
||
|
|
{
|
||
|
|
hr = S_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
////////////////////
|
||
|
|
|
||
|
|
static int local_nCompareVersion( /*[in]*/ DWORD dwVer1, /*[in]*/ DWORD dwBuild1 ,
|
||
|
|
/*[in]*/ DWORD dwVer2, /*[in]*/ DWORD dwBuild2 )
|
||
|
|
{
|
||
|
|
if(dwVer1 > dwVer2) return 1;
|
||
|
|
if(dwVer1 < dwVer2) return -1;
|
||
|
|
|
||
|
|
// dwVer1 == dwVer2
|
||
|
|
if(dwBuild1 > dwBuild2) return 1;
|
||
|
|
if(dwBuild1 < dwBuild2) return -1;
|
||
|
|
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
static bool local_fConvertDotVersionStrToDwords( /*[in ]*/ const MPC::wstring& strVer ,
|
||
|
|
/*[out]*/ DWORD& dwVer ,
|
||
|
|
/*[out]*/ DWORD& dwBuild )
|
||
|
|
{
|
||
|
|
unsigned int iVerFields[4];
|
||
|
|
|
||
|
|
if(swscanf( strVer.c_str(), L"%u.%u.%u.%u", &iVerFields[0], &iVerFields[1], &iVerFields[2], &iVerFields[3] ) == 4)
|
||
|
|
{
|
||
|
|
dwVer = (iVerFields[0] << 16) + iVerFields[1];
|
||
|
|
dwBuild = (iVerFields[2] << 16) + iVerFields[3];
|
||
|
|
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
int Taxonomy::Package::Compare( /*[in]*/ const Package& pkg, /*[in]*/ DWORD dwMode ) const
|
||
|
|
{
|
||
|
|
int iCmp = 0;
|
||
|
|
DWORD dwVer1;
|
||
|
|
DWORD dwVer2;
|
||
|
|
DWORD dwBuild1;
|
||
|
|
DWORD dwBuild2;
|
||
|
|
|
||
|
|
if(dwMode & c_Cmp_SKU)
|
||
|
|
{
|
||
|
|
iCmp = MPC::StrICmp( m_strSKU , pkg.m_strSKU ); if(iCmp) return iCmp;
|
||
|
|
iCmp = MPC::StrICmp( m_strLanguage , pkg.m_strLanguage ); if(iCmp) return iCmp;
|
||
|
|
}
|
||
|
|
|
||
|
|
if(dwMode & c_Cmp_ID)
|
||
|
|
{
|
||
|
|
iCmp = MPC::StrICmp( m_strVendorID , pkg.m_strVendorID ); if(iCmp) return iCmp;
|
||
|
|
iCmp = MPC::StrICmp( m_strProductID, pkg.m_strProductID ); if(iCmp) return iCmp;
|
||
|
|
}
|
||
|
|
|
||
|
|
if(dwMode & c_Cmp_VERSION)
|
||
|
|
{
|
||
|
|
if(local_fConvertDotVersionStrToDwords( m_strVersion, dwVer1, dwBuild1 ) &&
|
||
|
|
local_fConvertDotVersionStrToDwords( pkg.m_strVersion, dwVer2, dwBuild2 ) )
|
||
|
|
{
|
||
|
|
iCmp = local_nCompareVersion( dwVer1, dwBuild1, dwVer2, dwBuild2 );
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
iCmp = MPC::StrICmp( m_strVersion, pkg.m_strVersion );
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return iCmp;
|
||
|
|
}
|
||
|
|
|
||
|
|
////////////////////////////////////////////////////////////////////////////////
|
||
|
|
|
||
|
|
HRESULT Taxonomy::Package::GenerateFileName()
|
||
|
|
{
|
||
|
|
HRESULT hr;
|
||
|
|
|
||
|
|
if(m_strFileName.size() == 0)
|
||
|
|
{
|
||
|
|
WCHAR rgBuf[MAX_PATH]; _snwprintf( rgBuf, MAXSTRLEN(rgBuf), L"%s\\package_%ld.cab", HC_ROOT_HELPSVC_PKGSTORE, m_lSequence ); rgBuf[MAXSTRLEN(rgBuf)] = 0;
|
||
|
|
|
||
|
|
if(FAILED(hr = MPC::SubstituteEnvVariables( m_strFileName = rgBuf ))) return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
return S_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
HRESULT Taxonomy::Package::Import( /*[in]*/ Logger& log ,
|
||
|
|
/*[in]*/ LPCWSTR szFile ,
|
||
|
|
/*[in]*/ long lSequence ,
|
||
|
|
/*[in]*/ MPC::Impersonation* imp )
|
||
|
|
{
|
||
|
|
__HCP_FUNC_ENTRY( "Taxonomy::Package::Import" );
|
||
|
|
|
||
|
|
HRESULT hr;
|
||
|
|
|
||
|
|
|
||
|
|
m_fTemporary = true;
|
||
|
|
m_lSequence = lSequence;
|
||
|
|
|
||
|
|
__MPC_EXIT_IF_METHOD_FAILS(hr, GenerateFileName());
|
||
|
|
|
||
|
|
__MPC_EXIT_IF_METHOD_FAILS(hr, MPC::DeleteFile( m_strFileName ));
|
||
|
|
|
||
|
|
|
||
|
|
if(imp)
|
||
|
|
{
|
||
|
|
__MPC_EXIT_IF_METHOD_FAILS(hr, SVC::CopyFileWhileImpersonating( szFile, m_strFileName.c_str(), *imp, /*fImpersonateForSource*/true ));
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
__MPC_EXIT_IF_METHOD_FAILS(hr, MPC::CopyFile( szFile, m_strFileName.c_str() ));
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
__MPC_EXIT_IF_METHOD_FAILS(hr, MPC::ComputeCRC( m_dwCRC, m_strFileName.c_str() ));
|
||
|
|
|
||
|
|
::SetFileAttributesW( m_strFileName.c_str(), FILE_ATTRIBUTE_READONLY | FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_SYSTEM );
|
||
|
|
|
||
|
|
log.WriteLog( -1, L"\nImported package %s into package store\n\n", szFile );
|
||
|
|
|
||
|
|
hr = S_OK;
|
||
|
|
|
||
|
|
|
||
|
|
__HCP_FUNC_CLEANUP;
|
||
|
|
|
||
|
|
if(FAILED(hr))
|
||
|
|
{
|
||
|
|
log.WriteLog( hr, L"Failed to copy %s into package store", szFile );
|
||
|
|
}
|
||
|
|
|
||
|
|
__HCP_FUNC_EXIT(hr);
|
||
|
|
}
|
||
|
|
|
||
|
|
////////////////////////////////////////////////////////////////////////////////
|
||
|
|
|
||
|
|
static HRESULT local_GetSignerInfo( /*[in]*/ HCRYPTMSG hMsg, /*[in]*/ DWORD index, /*[in/out]*/ BYTE*& pSignerInfo )
|
||
|
|
{
|
||
|
|
__HCP_FUNC_ENTRY( "local_GetSignerInfo" );
|
||
|
|
|
||
|
|
HRESULT hr;
|
||
|
|
BYTE* pbEncodedSigner = NULL;
|
||
|
|
DWORD cbEncodedSigner = 0;
|
||
|
|
DWORD cbSignerInfo = 0;
|
||
|
|
|
||
|
|
|
||
|
|
delete [] pSignerInfo; pSignerInfo = NULL;
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// get the encoded signer BLOB
|
||
|
|
//
|
||
|
|
::CryptMsgGetParam( hMsg, CMSG_ENCODED_SIGNER, index, NULL, &cbEncodedSigner );
|
||
|
|
|
||
|
|
__MPC_EXIT_IF_ALLOC_FAILS(hr, pbEncodedSigner, new BYTE[cbEncodedSigner]);
|
||
|
|
|
||
|
|
__MPC_EXIT_IF_CALL_RETURNS_FALSE(hr, ::CryptMsgGetParam( hMsg, CMSG_ENCODED_SIGNER, index, pbEncodedSigner, &cbEncodedSigner ));
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// decode the EncodedSigner info
|
||
|
|
//
|
||
|
|
::CryptDecodeObject( PKCS_7_ASN_ENCODING|CRYPT_ASN_ENCODING, PKCS7_SIGNER_INFO, pbEncodedSigner, cbEncodedSigner, 0, NULL, &cbSignerInfo );
|
||
|
|
|
||
|
|
__MPC_EXIT_IF_ALLOC_FAILS(hr, pSignerInfo, new BYTE[cbSignerInfo]);
|
||
|
|
|
||
|
|
__MPC_EXIT_IF_CALL_RETURNS_FALSE(hr, ::CryptDecodeObject( PKCS_7_ASN_ENCODING|CRYPT_ASN_ENCODING, PKCS7_SIGNER_INFO,
|
||
|
|
pbEncodedSigner, cbEncodedSigner, 0, pSignerInfo, &cbSignerInfo ));
|
||
|
|
|
||
|
|
|
||
|
|
hr = S_OK;
|
||
|
|
|
||
|
|
|
||
|
|
__HCP_FUNC_CLEANUP;
|
||
|
|
|
||
|
|
delete [] pbEncodedSigner;
|
||
|
|
|
||
|
|
__HCP_FUNC_EXIT(hr);
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
HRESULT Taxonomy::Package::Authenticate( /*[in]*/ Logger& log )
|
||
|
|
{
|
||
|
|
__HCP_FUNC_ENTRY( "Taxonomy::Package::Authenticate" );
|
||
|
|
|
||
|
|
HRESULT hr;
|
||
|
|
|
||
|
|
if(m_fBuiltin == false)
|
||
|
|
{
|
||
|
|
__MPC_EXIT_IF_METHOD_FAILS(hr, GenerateFileName());
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// First, verify the signature is valid.
|
||
|
|
//
|
||
|
|
{
|
||
|
|
WINTRUST_DATA wtdWinTrust;
|
||
|
|
WINTRUST_FILE_INFO wtfWinTrustFile;
|
||
|
|
GUID c_guidPubSoftwareTrustProv = WINTRUST_ACTION_GENERIC_VERIFY_V2;
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// set up wintrust file info struct
|
||
|
|
//
|
||
|
|
::ZeroMemory( &wtfWinTrustFile, sizeof(wtfWinTrustFile) );
|
||
|
|
|
||
|
|
wtfWinTrustFile.cbStruct = sizeof(wtfWinTrustFile);
|
||
|
|
wtfWinTrustFile.pcwszFilePath = m_strFileName.c_str();
|
||
|
|
wtfWinTrustFile.hFile = NULL;
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// set up wintrust data struct
|
||
|
|
//
|
||
|
|
::ZeroMemory( &wtdWinTrust, sizeof(wtdWinTrust) );
|
||
|
|
|
||
|
|
wtdWinTrust.cbStruct = sizeof(wtdWinTrust);
|
||
|
|
wtdWinTrust.dwUnionChoice = WTD_CHOICE_FILE;
|
||
|
|
wtdWinTrust.pFile = &wtfWinTrustFile;
|
||
|
|
wtdWinTrust.dwUIChoice = WTD_UI_NONE; // Whistler special : must always be silent otherwise won't come back
|
||
|
|
|
||
|
|
// Verify the trust of the help package
|
||
|
|
if(FAILED(hr = ::WinVerifyTrust( 0, &c_guidPubSoftwareTrustProv, &wtdWinTrust )))
|
||
|
|
{
|
||
|
|
LPCWSTR szError;
|
||
|
|
|
||
|
|
switch(hr)
|
||
|
|
{
|
||
|
|
case TRUST_E_SUBJECT_NOT_TRUSTED : szError = L"subject not trusted" ; break;
|
||
|
|
case TRUST_E_PROVIDER_UNKNOWN : szError = L"provider unknown" ; break;
|
||
|
|
case TRUST_E_ACTION_UNKNOWN : szError = L"action unknown" ; break;
|
||
|
|
case TRUST_E_SUBJECT_FORM_UNKNOWN: szError = L"subject form unknown"; break;
|
||
|
|
default: szError = L"unknown" ; break;
|
||
|
|
}
|
||
|
|
|
||
|
|
__MPC_SET_ERROR_AND_EXIT(hr, log.WriteLog( hr, L"Help package cannot be trusted\nError WinVerifyTrust %s", szError ));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Then open the certificate and extract the DN.
|
||
|
|
//
|
||
|
|
{
|
||
|
|
HCRYPTMSG hMsg = NULL;
|
||
|
|
|
||
|
|
m_strVendorID.erase();
|
||
|
|
|
||
|
|
if(::CryptQueryObject( CERT_QUERY_OBJECT_FILE,
|
||
|
|
m_strFileName.c_str(),
|
||
|
|
CERT_QUERY_CONTENT_FLAG_PKCS7_SIGNED_EMBED,
|
||
|
|
CERT_QUERY_FORMAT_FLAG_BINARY ,
|
||
|
|
0, NULL, NULL, NULL, NULL, &hMsg, NULL ))
|
||
|
|
{
|
||
|
|
HCERTSTORE hCertStore;
|
||
|
|
|
||
|
|
hCertStore = ::CertOpenStore( CERT_STORE_PROV_MSG, X509_ASN_ENCODING | PKCS_7_ASN_ENCODING, NULL, NULL, (const void *)hMsg );
|
||
|
|
if(hCertStore)
|
||
|
|
{
|
||
|
|
BYTE* rgSignerInfo = NULL;
|
||
|
|
|
||
|
|
if(SUCCEEDED(local_GetSignerInfo( hMsg, 0, rgSignerInfo )))
|
||
|
|
{
|
||
|
|
PCMSG_SIGNER_INFO pSignerInfo = (PCMSG_SIGNER_INFO)rgSignerInfo;
|
||
|
|
CERT_INFO certInfo; ::ZeroMemory( &certInfo, sizeof(certInfo) );
|
||
|
|
PCCERT_CONTEXT pCertContext;
|
||
|
|
|
||
|
|
certInfo.SerialNumber = pSignerInfo->SerialNumber;
|
||
|
|
certInfo.Issuer = pSignerInfo->Issuer;
|
||
|
|
|
||
|
|
pCertContext = ::CertGetSubjectCertificateFromStore( hCertStore, X509_ASN_ENCODING, &certInfo );
|
||
|
|
if(pCertContext)
|
||
|
|
{
|
||
|
|
local_GetDNInfo( pCertContext, szOID_COMMON_NAME , L"CN=", m_strVendorID, &m_strVendorName );
|
||
|
|
local_GetDNInfo( pCertContext, szOID_LOCALITY_NAME , L"L=" , m_strVendorID );
|
||
|
|
local_GetDNInfo( pCertContext, szOID_STATE_OR_PROVINCE_NAME, L"S=" , m_strVendorID );
|
||
|
|
local_GetDNInfo( pCertContext, szOID_COUNTRY_NAME , L"C=" , m_strVendorID );
|
||
|
|
|
||
|
|
::CertFreeCertificateContext( pCertContext );
|
||
|
|
}
|
||
|
|
|
||
|
|
delete [] rgSignerInfo;
|
||
|
|
}
|
||
|
|
|
||
|
|
::CertCloseStore( hCertStore, 0 );
|
||
|
|
}
|
||
|
|
|
||
|
|
::CryptMsgClose( hMsg );
|
||
|
|
}
|
||
|
|
|
||
|
|
if(m_strVendorID.size() == 0)
|
||
|
|
{
|
||
|
|
__MPC_SET_ERROR_AND_EXIT(hr, log.WriteLog( HRESULT_FROM_WIN32(ERROR_ACCESS_DENIED), L"Help package cannot be trusted: unable to extract signer\n" ));
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Escape unsafe character in the CertID.
|
||
|
|
//
|
||
|
|
{
|
||
|
|
LPWSTR szCertID = (LPWSTR)m_strVendorID.c_str();
|
||
|
|
|
||
|
|
while(szCertID[0])
|
||
|
|
{
|
||
|
|
switch(szCertID[0])
|
||
|
|
{
|
||
|
|
case L'\\':
|
||
|
|
case L'/':
|
||
|
|
case L':':
|
||
|
|
case L'*':
|
||
|
|
case L'?':
|
||
|
|
case L'"':
|
||
|
|
case L'<':
|
||
|
|
case L'>':
|
||
|
|
case L'|':
|
||
|
|
szCertID[0] = L'_';
|
||
|
|
}
|
||
|
|
|
||
|
|
szCertID++;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
m_strVendorID = HC_MICROSOFT_DN;
|
||
|
|
|
||
|
|
MPC::LocalizeString( IDS_HELPSVC_SEMGR_OWNER, m_strVendorName );
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
m_fMicrosoft = (MPC::StrICmp( m_strVendorID, HC_MICROSOFT_DN ) == 0);
|
||
|
|
|
||
|
|
//
|
||
|
|
// Extract package info from packagedescription.xml
|
||
|
|
//
|
||
|
|
{
|
||
|
|
MPC::XmlUtil xml;
|
||
|
|
bool fFound;
|
||
|
|
|
||
|
|
__MPC_EXIT_IF_METHOD_FAILS(hr, ExtractPkgDesc( log, xml ));
|
||
|
|
|
||
|
|
__MPC_EXIT_IF_METHOD_FAILS(hr, xml.GetAttribute( L"PRODUCT", L"ID", m_strProductID, fFound ));
|
||
|
|
if(fFound == false) // Set some default.
|
||
|
|
{
|
||
|
|
m_strProductID = L"<default>";
|
||
|
|
}
|
||
|
|
|
||
|
|
__MPC_EXIT_IF_METHOD_FAILS(hr, xml.GetAttribute( L"VERSION", L"VALUE", m_strVersion, fFound ));
|
||
|
|
if(fFound == false) // Set some default.
|
||
|
|
{
|
||
|
|
m_strVersion = L"1.0.0.0";
|
||
|
|
}
|
||
|
|
|
||
|
|
if(m_fBuiltin == false) // System packages inherit SKU/Language from the database.
|
||
|
|
{
|
||
|
|
__MPC_EXIT_IF_METHOD_FAILS(hr, xml.GetAttribute( L"SKU", L"VALUE", m_strSKU, fFound ));
|
||
|
|
if(fFound == false) // Not a valid package!!
|
||
|
|
{
|
||
|
|
__MPC_SET_ERROR_AND_EXIT(hr, log.WriteLog( E_INVALIDARG, L"Missing SKU element" ));
|
||
|
|
}
|
||
|
|
|
||
|
|
__MPC_EXIT_IF_METHOD_FAILS(hr, xml.GetAttribute( L"LANGUAGE", L"VALUE", m_strLanguage, fFound ));
|
||
|
|
if(fFound == false) // Not a valid package!!
|
||
|
|
{
|
||
|
|
__MPC_SET_ERROR_AND_EXIT(hr, log.WriteLog( E_INVALIDARG, L"Missing LANGAUGE element" ));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if(m_lSequence >= 0)
|
||
|
|
{
|
||
|
|
m_fTemporary = false;
|
||
|
|
}
|
||
|
|
|
||
|
|
hr = S_OK;
|
||
|
|
|
||
|
|
|
||
|
|
__HCP_FUNC_CLEANUP;
|
||
|
|
|
||
|
|
__HCP_FUNC_EXIT(hr);
|
||
|
|
}
|
||
|
|
|
||
|
|
HRESULT Taxonomy::Package::Remove( /*[in]*/ Logger& log )
|
||
|
|
{
|
||
|
|
__HCP_FUNC_ENTRY( "Taxonomy::Package::Remove" );
|
||
|
|
|
||
|
|
HRESULT hr;
|
||
|
|
|
||
|
|
//
|
||
|
|
// There's no private file for this package, nothing to do.
|
||
|
|
//
|
||
|
|
if(m_lSequence == -1)
|
||
|
|
{
|
||
|
|
__MPC_SET_ERROR_AND_EXIT(hr, S_OK);
|
||
|
|
}
|
||
|
|
|
||
|
|
if(m_lSequence)
|
||
|
|
{
|
||
|
|
if(SUCCEEDED(GenerateFileName()))
|
||
|
|
{
|
||
|
|
__MPC_EXIT_IF_METHOD_FAILS(hr, MPC::DeleteFile( m_strFileName ));
|
||
|
|
|
||
|
|
log.WriteLog( -1, L"\nRemove package %s [%s] (Vendor: %s) from package store\n\n", m_strProductID.c_str(), m_strVersion.c_str(), m_strVendorID.c_str() );
|
||
|
|
|
||
|
|
m_strFileName.erase();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
hr = S_OK;
|
||
|
|
|
||
|
|
|
||
|
|
__HCP_FUNC_CLEANUP;
|
||
|
|
|
||
|
|
__HCP_FUNC_EXIT(hr);
|
||
|
|
}
|
||
|
|
|
||
|
|
HRESULT Taxonomy::Package::ExtractFile( /*[in]*/ Logger& log ,
|
||
|
|
/*[in]*/ LPCWSTR szFileDestination ,
|
||
|
|
/*[in]*/ LPCWSTR szNameInCabinet )
|
||
|
|
{
|
||
|
|
__HCP_FUNC_ENTRY( "Taxonomy::Package::ExtractFile" );
|
||
|
|
|
||
|
|
HRESULT hr;
|
||
|
|
|
||
|
|
|
||
|
|
__MPC_EXIT_IF_METHOD_FAILS(hr, GenerateFileName());
|
||
|
|
|
||
|
|
|
||
|
|
if(FAILED(hr = MPC::DecompressFromCabinet( m_strFileName.c_str(), szFileDestination, szNameInCabinet )))
|
||
|
|
{
|
||
|
|
__MPC_SET_ERROR_AND_EXIT(hr, log.WriteLog( hr, L"Error extracting %s from help package", szNameInCabinet ));
|
||
|
|
}
|
||
|
|
|
||
|
|
log.WriteLog( S_OK, L"Extracted %s from help package", szNameInCabinet );
|
||
|
|
|
||
|
|
hr = S_OK;
|
||
|
|
|
||
|
|
|
||
|
|
__HCP_FUNC_CLEANUP;
|
||
|
|
|
||
|
|
__HCP_FUNC_EXIT(hr);
|
||
|
|
}
|
||
|
|
|
||
|
|
HRESULT Taxonomy::Package::ExtractXMLFile( /*[in]*/ Logger& log ,
|
||
|
|
/*[in]*/ MPC::XmlUtil& xml ,
|
||
|
|
/*[in]*/ LPCWSTR szTag ,
|
||
|
|
/*[in]*/ LPCWSTR szNameInCabinet )
|
||
|
|
{
|
||
|
|
__HCP_FUNC_ENTRY( "Taxonomy::Package::ExtractXMLFile" );
|
||
|
|
|
||
|
|
HRESULT hr;
|
||
|
|
MPC::wstring strTmp;
|
||
|
|
bool fLoaded;
|
||
|
|
|
||
|
|
|
||
|
|
__MPC_EXIT_IF_METHOD_FAILS(hr, MPC::GetTemporaryFileName( strTmp ));
|
||
|
|
|
||
|
|
__MPC_EXIT_IF_METHOD_FAILS(hr, ExtractFile( log, strTmp.c_str(), szNameInCabinet ));
|
||
|
|
|
||
|
|
// load the XML with the root tag
|
||
|
|
__MPC_EXIT_IF_METHOD_FAILS(hr, xml.Load( strTmp.c_str(), szTag, fLoaded ));
|
||
|
|
if(fLoaded == false)
|
||
|
|
{
|
||
|
|
__MPC_SET_ERROR_AND_EXIT(hr, log.WriteLog( HRESULT_FROM_WIN32(ERROR_ACCESS_DENIED), L"Error invalid XML in %s", szNameInCabinet ));
|
||
|
|
}
|
||
|
|
|
||
|
|
hr = S_OK;
|
||
|
|
|
||
|
|
|
||
|
|
__HCP_FUNC_CLEANUP;
|
||
|
|
|
||
|
|
(void)MPC::RemoveTemporaryFile( strTmp );
|
||
|
|
|
||
|
|
__HCP_FUNC_EXIT(hr);
|
||
|
|
}
|
||
|
|
|
||
|
|
HRESULT Taxonomy::Package::ExtractPkgDesc( /*[in]*/ Logger& log ,
|
||
|
|
/*[in]*/ MPC::XmlUtil& xml )
|
||
|
|
{
|
||
|
|
return ExtractXMLFile( log, xml, Taxonomy::Strings::s_tag_root_PackageDescription, Taxonomy::Strings::s_file_PackageDescription );
|
||
|
|
}
|
||
|
|
|
||
|
|
////////////////////////////////////////////////////////////////////////////////
|
||
|
|
////////////////////////////////////////////////////////////////////////////////
|
||
|
|
|
||
|
|
Taxonomy::ProcessedPackage::ProcessedPackage()
|
||
|
|
{
|
||
|
|
m_lSequence = 0; // long m_lSequence;
|
||
|
|
m_fProcessed = false; // bool m_fProcessed;
|
||
|
|
m_fDisabled = false; // bool m_fDisabled;
|
||
|
|
}
|
||
|
|
|
||
|
|
HRESULT Taxonomy::operator>>( /*[in]*/ MPC::Serializer& stream, /*[out]*/ Taxonomy::ProcessedPackage& val )
|
||
|
|
{
|
||
|
|
HRESULT hr;
|
||
|
|
|
||
|
|
if(SUCCEEDED(hr = (stream >> val.m_lSequence )) &&
|
||
|
|
SUCCEEDED(hr = (stream >> val.m_fProcessed)) &&
|
||
|
|
SUCCEEDED(hr = (stream >> val.m_fDisabled )) )
|
||
|
|
{
|
||
|
|
hr = S_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
HRESULT Taxonomy::operator<<( /*[in]*/ MPC::Serializer& stream, /*[in] */ const Taxonomy::ProcessedPackage& val )
|
||
|
|
{
|
||
|
|
HRESULT hr;
|
||
|
|
|
||
|
|
if(SUCCEEDED(hr = (stream << val.m_lSequence )) &&
|
||
|
|
SUCCEEDED(hr = (stream << val.m_fProcessed)) &&
|
||
|
|
SUCCEEDED(hr = (stream << val.m_fDisabled )) )
|
||
|
|
{
|
||
|
|
hr = S_OK;
|
||
|
|
}
|
||
|
|
|
||
|
|
return hr;
|
||
|
|
}
|