Source code of Windows XP (NT5)
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

635 lines
18 KiB

/********************************************************************
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;
}