1455 lines
40 KiB
C++
1455 lines
40 KiB
C++
//RETURN
|
||
// Author: DebiM
|
||
// Date: January 97
|
||
//
|
||
//
|
||
// Class Store Administration Implementation
|
||
//
|
||
// This source file contains implementations for
|
||
// IClassAdmin interface.
|
||
//
|
||
// It uses ADs interfaces (over LDAP) to talk to an LDAP
|
||
// provider such as NTDS.
|
||
//
|
||
//---------------------------------------------------------------------
|
||
|
||
#include "csadm.hxx"
|
||
void GetCurrentUsn(CSUSN *pStoreUsn);
|
||
|
||
//----------------------------------------------------------
|
||
// Implementation for CClassContainer
|
||
//----------------------------------------------------------
|
||
//
|
||
CClassContainer::CClassContainer()
|
||
|
||
{
|
||
m_fOpen = FALSE;
|
||
m_ADsContainer = NULL;
|
||
m_pADsClassStore = NULL;
|
||
m_ADsClassContainer = NULL;
|
||
m_ADsPackageContainer = NULL;
|
||
m_ADsCategoryContainer = NULL;
|
||
|
||
m_uRefs = 1;
|
||
StartQuery(&m_pIDBCreateCommand);
|
||
}
|
||
|
||
|
||
//
|
||
// CClassContainer implementation
|
||
//
|
||
CClassContainer::CClassContainer(LPOLESTR szStoreName,
|
||
HRESULT *phr)
|
||
|
||
{
|
||
IADs *pADs = NULL;
|
||
LPOLESTR pszName = NULL;
|
||
DWORD dwStoreVersion = 0;
|
||
|
||
*phr = S_OK;
|
||
|
||
m_fOpen = FALSE;
|
||
m_ADsContainer = NULL;
|
||
m_pADsClassStore = NULL;
|
||
m_ADsClassContainer = NULL;
|
||
m_ADsPackageContainer = NULL;
|
||
m_ADsCategoryContainer = NULL;
|
||
|
||
//
|
||
// For every interface pointer, we create a separate Query session
|
||
//
|
||
StartQuery(&m_pIDBCreateCommand);
|
||
|
||
// Bind to a Class Store Container Object
|
||
// Cache the interface pointer
|
||
//
|
||
wcscpy (m_szContainerName, szStoreName);
|
||
|
||
*phr = ADsGetObject(
|
||
szStoreName,
|
||
IID_IADsContainer,
|
||
(void **)&m_ADsContainer
|
||
);
|
||
|
||
if (!SUCCEEDED(*phr))
|
||
return;
|
||
|
||
//
|
||
// Check the Schema Version of this container
|
||
//
|
||
|
||
*phr = m_ADsContainer->QueryInterface (IID_IADs, (void **)&m_pADsClassStore);
|
||
if (!SUCCEEDED(*phr))
|
||
return;
|
||
|
||
*phr = GetPropertyDW (m_pADsClassStore, STOREVERSION, &dwStoreVersion);
|
||
|
||
if ((!SUCCEEDED(*phr)) ||
|
||
(dwStoreVersion != SCHEMA_VERSION_NUMBER))
|
||
{
|
||
*phr = CS_E_INVALID_VERSION;
|
||
return;
|
||
}
|
||
|
||
//
|
||
// Bind to the class container Object
|
||
// Cache the interface pointer
|
||
//
|
||
|
||
m_ADsClassContainer = NULL;
|
||
|
||
*phr = m_ADsContainer->GetObject(
|
||
NULL,
|
||
CLASSCONTAINERNAME,
|
||
(IDispatch **)&pADs
|
||
);
|
||
|
||
if (!SUCCEEDED(*phr))
|
||
return;
|
||
|
||
pADs->QueryInterface(IID_IADsContainer,
|
||
(void **)&m_ADsClassContainer
|
||
);
|
||
|
||
*phr = pADs->get_ADsPath(&pszName);
|
||
wcscpy (m_szClassName, pszName);
|
||
SysFreeString(pszName);
|
||
|
||
pADs->Release();
|
||
pADs = NULL;
|
||
|
||
if (!SUCCEEDED(*phr))
|
||
return;
|
||
|
||
//
|
||
// Bind to the Package container Object
|
||
// Cache the interface pointer
|
||
//
|
||
|
||
m_ADsPackageContainer = NULL;
|
||
|
||
*phr = m_ADsContainer->GetObject(
|
||
NULL,
|
||
PACKAGECONTAINERNAME,
|
||
(IDispatch **)&pADs);
|
||
|
||
if (!SUCCEEDED(*phr))
|
||
return;
|
||
|
||
pADs->QueryInterface(IID_IADsContainer,
|
||
(void **)&m_ADsPackageContainer
|
||
);
|
||
|
||
*phr = pADs->get_ADsPath(&pszName);
|
||
wcscpy (m_szPackageName, pszName);
|
||
SysFreeString(pszName);
|
||
pADs->Release();
|
||
pADs = NULL;
|
||
|
||
if (!SUCCEEDED(*phr))
|
||
return;
|
||
|
||
//
|
||
// Bind to the category container Object
|
||
// Cache the interface pointer
|
||
//
|
||
m_ADsCategoryContainer = NULL;
|
||
|
||
*phr = m_ADsContainer->GetObject(
|
||
NULL,
|
||
CATEGORYCONTAINERNAME,
|
||
(IDispatch **)&pADs);
|
||
|
||
if (!SUCCEEDED(*phr))
|
||
return;
|
||
|
||
pADs->QueryInterface(IID_IADsContainer,
|
||
(void **)&m_ADsCategoryContainer
|
||
);
|
||
|
||
pADs->Release();
|
||
pADs = NULL;
|
||
|
||
m_fOpen = TRUE;
|
||
m_uRefs = 1;
|
||
return;
|
||
}
|
||
|
||
|
||
CClassContainer::~CClassContainer(void)
|
||
{
|
||
UINT i;
|
||
|
||
EndQuery(m_pIDBCreateCommand);
|
||
m_pIDBCreateCommand = NULL;
|
||
|
||
|
||
if (m_fOpen)
|
||
{
|
||
m_fOpen = FALSE;
|
||
}
|
||
|
||
if (m_ADsClassContainer)
|
||
{
|
||
m_ADsClassContainer->Release();
|
||
m_ADsClassContainer = NULL;
|
||
}
|
||
|
||
if (m_ADsPackageContainer)
|
||
{
|
||
m_ADsPackageContainer->Release();
|
||
m_ADsPackageContainer = NULL;
|
||
}
|
||
|
||
if (m_ADsCategoryContainer)
|
||
{
|
||
m_ADsCategoryContainer->Release();
|
||
m_ADsCategoryContainer = NULL;
|
||
}
|
||
|
||
if (m_ADsContainer)
|
||
{
|
||
m_ADsContainer->Release();
|
||
m_ADsContainer = NULL;
|
||
}
|
||
|
||
if (m_pADsClassStore)
|
||
{
|
||
m_pADsClassStore->Release();
|
||
m_pADsClassStore = NULL;
|
||
}
|
||
|
||
}
|
||
|
||
|
||
//
|
||
// Removing a class from the database
|
||
//
|
||
|
||
HRESULT CClassContainer::DeleteClass (LPOLESTR szClsid)
|
||
{
|
||
|
||
WCHAR szName[_MAX_PATH];
|
||
HRESULT hr = S_OK;
|
||
IADs *pADs = NULL;
|
||
DWORD refcount = 0;
|
||
|
||
if (!m_fOpen)
|
||
return E_FAIL;
|
||
|
||
wsprintf(szName, L"CN=%s", szClsid);
|
||
hr = m_ADsClassContainer->GetObject(NULL, szName, (IDispatch **)&pADs);
|
||
|
||
if (SUCCEEDED(hr))
|
||
hr = GetPropertyDW(pADs, CLASSREFCOUNTER, &refcount);
|
||
|
||
if (refcount <= 1)
|
||
hr = m_ADsClassContainer->Delete(CLASS_CS_CLASS, szName);
|
||
else {
|
||
refcount--;
|
||
hr = SetPropertyDW(pADs, CLASSREFCOUNTER, refcount);
|
||
|
||
if (SUCCEEDED(hr))
|
||
hr = StoreIt(pADs);
|
||
}
|
||
|
||
if (pADs)
|
||
pADs->Release();
|
||
return hr;
|
||
}
|
||
|
||
|
||
extern LPOLESTR szPackageInfoColumns;
|
||
|
||
HRESULT CClassContainer::EnumPackages(
|
||
LPOLESTR pszFileExt,
|
||
GUID *pCategory,
|
||
DWORD dwAppFlags,
|
||
DWORD *pdwLocale,
|
||
CSPLATFORM *pPlatform,
|
||
IEnumPackage **ppIEnumPackage
|
||
)
|
||
{
|
||
HRESULT hr = S_OK;
|
||
CEnumPackage *pEnum = NULL;
|
||
WCHAR szCommand[1000], szQry[1000];
|
||
|
||
if (pszFileExt && IsBadStringPtr(pszFileExt, _MAX_PATH))
|
||
return E_INVALIDARG;
|
||
|
||
if (pCategory && !IsValidReadPtrIn(pCategory, sizeof(GUID)))
|
||
return E_INVALIDARG;
|
||
|
||
if (!IsValidPtrOut(ppIEnumPackage, sizeof(IEnumPackage *)))
|
||
return E_INVALIDARG;
|
||
|
||
*ppIEnumPackage = NULL;
|
||
|
||
pEnum = new CEnumPackage;
|
||
if(NULL == pEnum)
|
||
return E_OUTOFMEMORY;
|
||
|
||
//
|
||
// Create a CommandText
|
||
//
|
||
|
||
wsprintf(szCommand, L"<%s>;(& (objectClass=packageRegistration) ", m_szPackageName);
|
||
|
||
if (pszFileExt)
|
||
{
|
||
wsprintf(szQry, L"(%s=%s*) ", PKGFILEEXTNLIST, pszFileExt);
|
||
wcscat(szCommand, szQry);
|
||
}
|
||
|
||
if (pCategory)
|
||
{
|
||
STRINGGUID szCat;
|
||
StringFromGUID (*pCategory, szCat);
|
||
wsprintf(szQry, L"(%s=%s) ", PKGCATEGORYLIST, szCat);
|
||
wcscat(szCommand, szQry);
|
||
}
|
||
|
||
wcscat(szCommand, L");");
|
||
|
||
wsprintf(szQry, L" %s", szPackageInfoColumns);
|
||
wcscat(szCommand, szQry);
|
||
|
||
hr = pEnum->Initialize(szCommand, dwAppFlags, pdwLocale, pPlatform);
|
||
|
||
if (FAILED(hr)) {
|
||
delete pEnum;
|
||
return hr;
|
||
}
|
||
|
||
hr = pEnum->QueryInterface(IID_IEnumPackage, (void**)ppIEnumPackage);
|
||
|
||
return hr;
|
||
}
|
||
|
||
// GetPackageDetails
|
||
// pszPackageName : name of the package to be got.
|
||
// pInstallInfo : InstallInfo to be filled in. ignored if NULL.
|
||
// pPlatformInfo : PlatformInfo to be filled in. ignored if NULL.
|
||
// both can be sent in as NULL to check whether package exists or not.
|
||
|
||
|
||
HRESULT CClassContainer::GetPackageDetails (
|
||
LPOLESTR pszPackageName,
|
||
PACKAGEDETAIL *pPackageDetail)
|
||
{
|
||
HRESULT hr = S_OK;
|
||
IADs *pPackageADs = NULL;
|
||
WCHAR szRdn [_MAX_PATH];
|
||
|
||
if ((!pszPackageName) || IsBadStringPtr(pszPackageName, _MAX_PATH))
|
||
return E_INVALIDARG;
|
||
|
||
wcscpy (szRdn, L"CN=");
|
||
wcscat (szRdn, pszPackageName);
|
||
|
||
hr = m_ADsPackageContainer->GetObject(NULL, szRdn, (IDispatch **)&pPackageADs);
|
||
if (!SUCCEEDED(hr))
|
||
return CS_E_PACKAGE_NOTFOUND;
|
||
|
||
hr = GetPackageDetail (pPackageADs, pPackageDetail);
|
||
return hr;
|
||
}
|
||
|
||
|
||
HRESULT CClassContainer::ChangePackageProperties(
|
||
LPOLESTR pszPackageName,
|
||
LPOLESTR pszNewName,
|
||
DWORD *pdwFlags,
|
||
LPOLESTR pszUrl,
|
||
LPOLESTR pszScriptPath,
|
||
DWORD *pInstallUiLevel
|
||
)
|
||
{
|
||
HRESULT hr = S_OK;
|
||
IADs *pPackageADs = NULL;
|
||
WCHAR szRdn [_MAX_PATH];
|
||
WCHAR szNewRdn [_MAX_PATH];
|
||
LPOLESTR pszPackageDN;
|
||
CSUSN Usn;
|
||
|
||
if ((!pszPackageName) || IsBadStringPtr(pszPackageName, _MAX_PATH))
|
||
return E_INVALIDARG;
|
||
|
||
wcscpy (szRdn, L"CN=");
|
||
wcscat (szRdn, pszPackageName);
|
||
|
||
if (pszNewName)
|
||
{
|
||
//
|
||
// rename package
|
||
//
|
||
|
||
if (IsBadStringPtr(pszNewName, _MAX_PATH))
|
||
return E_INVALIDARG;
|
||
|
||
//
|
||
// Check to see if any other change is requested.
|
||
//
|
||
|
||
if (pszScriptPath ||
|
||
pszUrl ||
|
||
pdwFlags)
|
||
return E_INVALIDARG;
|
||
|
||
BuildADsPathFromParent (m_szPackageName, szRdn, &pszPackageDN);
|
||
wcscpy (szNewRdn, L"CN=");
|
||
wcscat (szNewRdn, pszNewName);
|
||
hr = m_ADsPackageContainer->MoveHere(pszPackageDN, szNewRdn, (IDispatch **)&pPackageADs);
|
||
FreeADsMem(pszPackageDN);
|
||
|
||
if (SUCCEEDED(hr))
|
||
pPackageADs->Release();
|
||
else
|
||
CSDbgPrint(("Error in MoveHere. hr=0x%x\n", hr));
|
||
return hr;
|
||
}
|
||
|
||
//
|
||
// No rename.
|
||
// Just change some properties.
|
||
//
|
||
hr = m_ADsPackageContainer->GetObject(NULL, szRdn, (IDispatch **)&pPackageADs);
|
||
if (!SUCCEEDED(hr))
|
||
return CS_E_PACKAGE_NOTFOUND;
|
||
//
|
||
// Update the TimeStamp
|
||
//
|
||
GetCurrentUsn(&Usn);
|
||
hr = UsnUpd(pPackageADs, PKGUSN, &Usn);
|
||
ERROR_ON_FAILURE(hr);
|
||
|
||
//
|
||
// Change Package Flags
|
||
//
|
||
if (pdwFlags)
|
||
{
|
||
hr = SetPropertyDW (pPackageADs, PACKAGEFLAGS, *pdwFlags);
|
||
ERROR_ON_FAILURE(hr);
|
||
}
|
||
|
||
//
|
||
// Change Package Script
|
||
//
|
||
if (pszScriptPath)
|
||
{
|
||
hr = SetProperty(pPackageADs, SCRIPTPATH, pszScriptPath);
|
||
ERROR_ON_FAILURE(hr);
|
||
}
|
||
|
||
//
|
||
// Change Package Help URL
|
||
//
|
||
if (pszUrl)
|
||
{
|
||
hr = SetProperty(pPackageADs, HELPURL, pszUrl);
|
||
ERROR_ON_FAILURE(hr);
|
||
}
|
||
|
||
//
|
||
// Change UI Level
|
||
//
|
||
if (pInstallUiLevel)
|
||
{
|
||
hr = SetPropertyDW (pPackageADs, UILEVEL, *pInstallUiLevel);
|
||
ERROR_ON_FAILURE(hr);
|
||
}
|
||
|
||
hr = StoreIt(pPackageADs);
|
||
|
||
Error_Cleanup:
|
||
pPackageADs->Release();
|
||
return hr;
|
||
|
||
}
|
||
|
||
|
||
HRESULT CClassContainer::ChangePackageCategories(
|
||
LPOLESTR pszPackageName,
|
||
UINT cCategories,
|
||
GUID *rpCategory
|
||
)
|
||
{
|
||
//
|
||
// Does not change USN
|
||
//
|
||
HRESULT hr = S_OK;
|
||
IADs *pPackageADs = NULL;
|
||
WCHAR szRdn [_MAX_PATH];
|
||
LPOLESTR *pszGuid = NULL;
|
||
CSUSN Usn;
|
||
UINT count;
|
||
|
||
wcscpy (szRdn, L"CN=");
|
||
wcscat (szRdn, pszPackageName);
|
||
|
||
if ((!cCategories) ||
|
||
(!rpCategory) ||
|
||
(!IsValidReadPtrIn(rpCategory, sizeof(GUID) * cCategories)))
|
||
return E_INVALIDARG;
|
||
|
||
hr = m_ADsPackageContainer->GetObject(NULL, szRdn, (IDispatch **)&pPackageADs);
|
||
if (!SUCCEEDED(hr))
|
||
return CS_E_PACKAGE_NOTFOUND;
|
||
|
||
// fill in the categories
|
||
pszGuid = (LPOLESTR *)CoTaskMemAlloc(cCategories * sizeof(LPOLESTR));
|
||
if (!pszGuid)
|
||
{
|
||
hr = E_OUTOFMEMORY;
|
||
ERROR_ON_FAILURE(hr);
|
||
}
|
||
|
||
|
||
for (count = 0; (count < cCategories); count++)
|
||
{
|
||
pszGuid[count] = (LPOLESTR)CoTaskMemAlloc(STRINGGUIDLEN*sizeof(WCHAR));
|
||
{
|
||
hr = E_OUTOFMEMORY;
|
||
ERROR_ON_FAILURE(hr);
|
||
}
|
||
StringFromGUID(rpCategory[count], pszGuid[count]);
|
||
}
|
||
|
||
hr = SetPropertyList(pPackageADs, PKGCATEGORYLIST, cCategories,
|
||
pszGuid);
|
||
ERROR_ON_FAILURE(hr);
|
||
|
||
for (count = 0; (count < cCategories); count++)
|
||
CoTaskMemFree(pszGuid[count]);
|
||
|
||
CoTaskMemFree(pszGuid);
|
||
hr = StoreIt(pPackageADs);
|
||
|
||
Error_Cleanup:
|
||
pPackageADs->Release();
|
||
return hr;
|
||
}
|
||
|
||
|
||
|
||
HRESULT CClassContainer::SetPriorityByFileExt(
|
||
LPOLESTR pszPackageName,
|
||
LPOLESTR pszFileExt,
|
||
UINT Priority
|
||
)
|
||
{
|
||
//
|
||
// Does not change USN
|
||
//
|
||
HRESULT hr = S_OK;
|
||
IADs *pPackageADs = NULL;
|
||
WCHAR szRdn [_MAX_PATH];
|
||
LPOLESTR *prgFileExt = NULL;
|
||
ULONG cShellFileExt;
|
||
UINT i;
|
||
|
||
wcscpy (szRdn, L"CN=");
|
||
wcscat (szRdn, pszPackageName);
|
||
|
||
hr = m_ADsPackageContainer->GetObject(NULL, szRdn, (IDispatch **)&pPackageADs);
|
||
if (!SUCCEEDED(hr))
|
||
return CS_E_PACKAGE_NOTFOUND;
|
||
|
||
hr = GetPropertyListAlloc (pPackageADs, PKGFILEEXTNLIST,
|
||
&cShellFileExt,
|
||
&prgFileExt);
|
||
|
||
for (i=0; i < cShellFileExt; ++i)
|
||
{
|
||
if (wcsncmp(prgFileExt[i], pszFileExt, wcslen(pszFileExt)) == 0)
|
||
{
|
||
wsprintf(prgFileExt[i], L"%s:%2d",
|
||
pszFileExt,
|
||
Priority);
|
||
}
|
||
}
|
||
|
||
if (i == cShellFileExt)
|
||
{
|
||
ERROR_ON_FAILURE(CS_E_PACKAGE_NOTFOUND);
|
||
}
|
||
|
||
hr = SetPropertyList(pPackageADs, PKGFILEEXTNLIST, cShellFileExt, prgFileExt);
|
||
|
||
hr = StoreIt(pPackageADs);
|
||
|
||
Error_Cleanup:
|
||
pPackageADs->Release();
|
||
return hr;
|
||
}
|
||
|
||
|
||
|
||
extern LPOLESTR szAppCategoryColumns;
|
||
|
||
HRESULT CClassContainer::GetAppCategories (
|
||
LCID Locale,
|
||
APPCATEGORYINFOLIST *pAppCategoryList
|
||
)
|
||
{
|
||
HRESULT hr = S_OK;
|
||
WCHAR szCommand[1000], szQry[1000];
|
||
WCHAR szRootPath[_MAX_PATH],
|
||
szAppCategoryContainer[_MAX_PATH];
|
||
IRowset * pIRow = NULL;
|
||
HACCESSOR HAcc;
|
||
IAccessor * pIAccessor = NULL;
|
||
IDBCreateCommand * pIDBCreateCommand = NULL;
|
||
LPOLESTR ** ppszDesc = NULL;
|
||
DWORD cgot = 0;
|
||
|
||
if (!IsValidPtrOut(pAppCategoryList, sizeof(APPCATEGORYINFOLIST)))
|
||
return E_INVALIDARG;
|
||
|
||
hr = GetRootPath(szRootPath);
|
||
wsprintf(szAppCategoryContainer, L"%s%s%s", LDAPPREFIX,
|
||
APPCATEGORYCONTAINERNAME, szRootPath+wcslen(LDAPPREFIX));
|
||
|
||
wsprintf(szCommand, L"<%s>; (objectClass=categoryRegistration); %s", szAppCategoryContainer,
|
||
szAppCategoryColumns);
|
||
|
||
hr = StartQuery(&(pIDBCreateCommand));
|
||
RETURN_ON_FAILURE(hr);
|
||
|
||
hr = ExecuteQuery (pIDBCreateCommand,
|
||
szCommand,
|
||
APPCATEGORY_COLUMN_COUNT,
|
||
NULL,
|
||
&HAcc,
|
||
&pIAccessor,
|
||
&pIRow
|
||
);
|
||
|
||
RETURN_ON_FAILURE(hr);
|
||
|
||
pAppCategoryList->cCategory = 500;
|
||
// upper limit presently.
|
||
|
||
hr = FetchCategory(pIRow,
|
||
HAcc,
|
||
(pAppCategoryList->cCategory),
|
||
&cgot,
|
||
&(pAppCategoryList->pCategoryInfo),
|
||
Locale);
|
||
|
||
pAppCategoryList->cCategory = cgot;
|
||
|
||
CloseQuery(pIAccessor,
|
||
HAcc,
|
||
pIRow);
|
||
|
||
EndQuery(pIDBCreateCommand);
|
||
|
||
return hr;
|
||
}
|
||
|
||
|
||
|
||
HRESULT CClassContainer::RegisterAppCategory (
|
||
APPCATEGORYINFO *pAppCategory
|
||
)
|
||
{
|
||
WCHAR szRootPath[_MAX_PATH], *localedescription = NULL,
|
||
szAppCategoryContainer[_MAX_PATH], szRDN[_MAX_PATH];
|
||
HRESULT hr = S_OK;
|
||
IADsContainer *pADsContainer = NULL;
|
||
IADs *pADs = NULL;
|
||
IDispatch *pUnknown = NULL;
|
||
ULONG i, j, cdesc, posn;
|
||
LPOLESTR *pszDescExisting = NULL, *pszDesc = NULL;
|
||
|
||
|
||
if (!IsValidReadPtrIn(pAppCategory, sizeof(APPCATEGORYINFO)))
|
||
return E_INVALIDARG;
|
||
|
||
if (IsBadStringPtr(pAppCategory->pszDescription, _MAX_PATH))
|
||
return E_INVALIDARG;
|
||
|
||
if (IsNullGuid(pAppCategory->AppCategoryId))
|
||
return E_INVALIDARG;
|
||
|
||
hr = GetRootPath(szRootPath);
|
||
|
||
// Bind to a AppCategory container
|
||
wsprintf(szAppCategoryContainer, L"%s%s%s", LDAPPREFIX,
|
||
APPCATEGORYCONTAINERNAME, szRootPath+wcslen(LDAPPREFIX));
|
||
|
||
hr = ADsGetObject(
|
||
szAppCategoryContainer,
|
||
IID_IADsContainer,
|
||
(void **)&pADsContainer
|
||
);
|
||
RETURN_ON_FAILURE(hr);
|
||
|
||
RdnFromGUID(pAppCategory->AppCategoryId, szRDN);
|
||
|
||
localedescription = (LPOLESTR)CoTaskMemAlloc(sizeof(WCHAR)*(128+16));
|
||
|
||
wsprintf(localedescription, L"%x %s %s", pAppCategory->Locale, CATSEPERATOR,
|
||
pAppCategory->pszDescription);
|
||
|
||
hr = pADsContainer->GetObject(NULL, szRDN, (IDispatch **)&pADs);
|
||
|
||
if (SUCCEEDED(hr))
|
||
{
|
||
hr = GetPropertyListAlloc (pADs, LOCALEDESCRIPTION, &cdesc, &pszDescExisting);
|
||
ERROR_ON_FAILURE(hr);
|
||
// Existing list of descriptions
|
||
|
||
pszDesc = (LPOLESTR *)CoTaskMemAlloc(sizeof(LPOLESTR)*(cdesc+1));
|
||
if ((cdesc) && (!pszDesc)) {
|
||
hr = E_OUTOFMEMORY;
|
||
ERROR_ON_FAILURE(hr);
|
||
}
|
||
|
||
for (j = 0; j < cdesc; j++)
|
||
pszDesc[j] = pszDescExisting[j];
|
||
|
||
if (!(posn = FindDescription(pszDescExisting, cdesc, &(pAppCategory->Locale), NULL, 0)))
|
||
{
|
||
// if no description exists for the lcid.
|
||
pszDesc[cdesc] = localedescription;
|
||
cdesc++;
|
||
}
|
||
else
|
||
{ // overwrite the old value
|
||
CoTaskMemFree(pszDesc[posn-1]);
|
||
pszDesc[posn-1] = localedescription;
|
||
}
|
||
}
|
||
else
|
||
{
|
||
hr = pADsContainer->Create(
|
||
CLASS_CS_CATEGORY,
|
||
szRDN,
|
||
&pUnknown
|
||
);
|
||
ERROR_ON_FAILURE(hr);
|
||
|
||
pszDesc = (LPOLESTR *)CoTaskMemAlloc(sizeof(LPOLESTR));
|
||
if (!pszDesc) {
|
||
hr = E_OUTOFMEMORY;
|
||
ERROR_ON_FAILURE(hr);
|
||
}
|
||
|
||
cdesc = 1;
|
||
pszDesc[0] = localedescription;
|
||
|
||
hr = pUnknown->QueryInterface(IID_IADs, (void **)&pADs);
|
||
RETURN_ON_FAILURE(hr);
|
||
|
||
pUnknown->Release();
|
||
}
|
||
|
||
hr = SetPropertyGuid(pADs, CATEGORYCATID, pAppCategory->AppCategoryId);
|
||
ERROR_ON_FAILURE(hr);
|
||
|
||
hr = SetPropertyList(pADs, LOCALEDESCRIPTION, cdesc, pszDesc);
|
||
for (j = 0; j < cdesc; j++)
|
||
CoTaskMemFree(pszDesc[j]);
|
||
CoTaskMemFree(pszDesc);
|
||
ERROR_ON_FAILURE(hr);
|
||
|
||
hr = StoreIt(pADs);
|
||
ERROR_ON_FAILURE(hr);
|
||
|
||
Error_Cleanup:
|
||
|
||
if (pADs)
|
||
pADs->Release();
|
||
|
||
if (pADsContainer)
|
||
pADsContainer->Release();
|
||
|
||
// Add this category.
|
||
return hr;
|
||
}
|
||
|
||
|
||
HRESULT CClassContainer::UnregisterAppCategory (
|
||
GUID *pAppCategoryId
|
||
)
|
||
{
|
||
WCHAR szRootPath[_MAX_PATH], szRDN[_MAX_PATH],
|
||
szAppCategoryContainer[_MAX_PATH];
|
||
HRESULT hr = S_OK;
|
||
IADsContainer *pADsContainer = NULL;
|
||
|
||
if (!IsValidReadPtrIn(pAppCategoryId, sizeof(GUID)))
|
||
return E_INVALIDARG;
|
||
|
||
hr = GetRootPath(szRootPath);
|
||
// Bind to a AppCategory container
|
||
wsprintf(szAppCategoryContainer, L"%s%s%s", LDAPPREFIX,
|
||
APPCATEGORYCONTAINERNAME, szRootPath+wcslen(LDAPPREFIX));
|
||
|
||
hr = ADsGetObject(
|
||
szAppCategoryContainer,
|
||
IID_IADsContainer,
|
||
(void **)&pADsContainer
|
||
);
|
||
|
||
RETURN_ON_FAILURE(hr);
|
||
|
||
RdnFromGUID(*pAppCategoryId, szRDN);
|
||
|
||
hr = pADsContainer->Delete(CLASS_CS_CATEGORY, szRDN);
|
||
pADsContainer->Release();
|
||
|
||
// Delete this category
|
||
|
||
return hr;
|
||
}
|
||
|
||
|
||
//+
|
||
|
||
HRESULT CClassContainer::RemovePackage (LPOLESTR pszPackageName)
|
||
//
|
||
// Remove a Package and the associated Classes from class store
|
||
//
|
||
{
|
||
HRESULT hr = S_OK;
|
||
IADs *pADs = NULL;
|
||
DWORD cClasses = 0, count = 0;
|
||
WCHAR szRdn [_MAX_PATH];
|
||
LPOLESTR *szClasses;
|
||
|
||
if (IsBadStringPtr(pszPackageName, _MAX_PATH))
|
||
return E_INVALIDARG;
|
||
|
||
if ((pszPackageName == NULL) ||
|
||
(*(pszPackageName) == NULL))
|
||
return E_INVALIDARG;
|
||
|
||
wcscpy (szRdn, L"CN=");
|
||
wcscat (szRdn, pszPackageName);
|
||
|
||
if (!m_fOpen)
|
||
return E_FAIL;
|
||
|
||
hr = m_ADsPackageContainer->GetObject(NULL, szRdn, (IDispatch **)&pADs);
|
||
RETURN_ON_FAILURE(hr);
|
||
|
||
hr = GetPropertyListAlloc(pADs, PKGCLSIDLIST, &cClasses, &szClasses);
|
||
|
||
for (count = 0; count < cClasses; count++)
|
||
hr = DeleteClass(szClasses[count]);
|
||
// ignore errors
|
||
|
||
for (count = 0; count < cClasses; count++)
|
||
CoTaskMemFree(szClasses[count]);
|
||
CoTaskMemFree(szClasses);
|
||
|
||
pADs->Release();
|
||
|
||
hr = m_ADsPackageContainer->Delete(CLASS_CS_PACKAGE, szRdn);
|
||
return hr;
|
||
}
|
||
|
||
//+
|
||
|
||
HRESULT CClassContainer::NewClass (CLASSDETAIL *pClassDetail)
|
||
|
||
//
|
||
// Add a new class to the database
|
||
//
|
||
{
|
||
HRESULT hr = S_OK;
|
||
IADs * pADs = NULL;
|
||
IDispatch * pUnknown = NULL;
|
||
STRINGGUID szGUID;
|
||
WCHAR szRDN [_MAX_PATH];
|
||
|
||
if (!m_fOpen)
|
||
return E_FAIL;
|
||
|
||
//
|
||
// Cant be a NULL guid
|
||
//
|
||
if (IsNullGuid(pClassDetail->Clsid))
|
||
return E_INVALIDARG;
|
||
|
||
StringFromGUID(pClassDetail->Clsid, szGUID);
|
||
|
||
//
|
||
// Create the RDN for the Class Object
|
||
//
|
||
|
||
// BUGBUG:: attaching package name creates problems.
|
||
wsprintf(szRDN, L"CN=%s", szGUID);
|
||
|
||
hr = m_ADsClassContainer->Create(
|
||
CLASS_CS_CLASS,
|
||
szRDN,
|
||
&pUnknown
|
||
);
|
||
|
||
RETURN_ON_FAILURE(hr);
|
||
|
||
hr = pUnknown->QueryInterface(
|
||
IID_IADs,
|
||
(void **)&pADs
|
||
);
|
||
|
||
pUnknown->Release();
|
||
|
||
hr = SetProperty (pADs, CLASSCLSID, szGUID);
|
||
ERROR_ON_FAILURE(hr);
|
||
|
||
if (pClassDetail->cProgId)
|
||
{
|
||
hr = SetPropertyList(pADs, PROGIDLIST, pClassDetail->cProgId,
|
||
pClassDetail->prgProgId);
|
||
|
||
ERROR_ON_FAILURE(hr);
|
||
}
|
||
|
||
if (!IsNullGuid(pClassDetail->TreatAs))
|
||
{
|
||
StringFromGUID(pClassDetail->TreatAs, szGUID);
|
||
hr = SetProperty (pADs, TREATASCLSID, szGUID);
|
||
ERROR_ON_FAILURE(hr);
|
||
}
|
||
|
||
hr = SetPropertyDW(pADs, CLASSREFCOUNTER, 1);
|
||
ERROR_ON_FAILURE(hr);
|
||
|
||
hr = StoreIt (pADs);
|
||
|
||
// this does not return an error for an alreay existing entry.
|
||
|
||
if (hr == E_ADS_LDAP_ALREADY_EXISTS)
|
||
{
|
||
|
||
DWORD refcount = 0;
|
||
|
||
pADs->Release(); // release the interface pointer already got.
|
||
|
||
hr = m_ADsClassContainer->GetObject(NULL, // CLASS_CS_CLASS
|
||
szRDN,
|
||
(IDispatch **)&pADs);
|
||
RETURN_ON_FAILURE(hr);
|
||
|
||
if (pClassDetail->cProgId)
|
||
{
|
||
hr = SetPropertyListMerge(pADs, PROGIDLIST, pClassDetail->cProgId,
|
||
pClassDetail->prgProgId);
|
||
ERROR_ON_FAILURE(hr);
|
||
}
|
||
|
||
// increment reference counter.
|
||
hr = GetPropertyDW(pADs, CLASSREFCOUNTER, &refcount);
|
||
ERROR_ON_FAILURE(hr);
|
||
|
||
refcount++;
|
||
|
||
hr = SetPropertyDW(pADs, CLASSREFCOUNTER, refcount);
|
||
ERROR_ON_FAILURE(hr);
|
||
|
||
// No merging of the treatas.
|
||
|
||
hr = StoreIt(pADs);
|
||
}
|
||
|
||
Error_Cleanup:
|
||
|
||
pADs->Release();
|
||
return hr;
|
||
|
||
}
|
||
|
||
#define SCRIPT_IN_DIRECTORY 256
|
||
|
||
|
||
HRESULT CClassContainer::AddPackage(LPOLESTR pszPackageName,
|
||
PACKAGEDETAIL *pPackageDetail)
|
||
{
|
||
HRESULT hr;
|
||
IADs *pPackageADs = NULL;
|
||
IDispatch *pUnknown = NULL;
|
||
WCHAR szRDN [_MAX_PATH];
|
||
LPOLESTR *pszGuid, *pszProgId;
|
||
DWORD *pdwArch=NULL, count = 0, cPackProgId = 0;
|
||
CSUSN Usn;
|
||
|
||
if (!pszPackageName)
|
||
return E_INVALIDARG;
|
||
|
||
if (!pPackageDetail)
|
||
return E_INVALIDARG;
|
||
|
||
if (!IsValidReadPtrIn(pPackageDetail, sizeof(PACKAGEDETAIL)))
|
||
return E_INVALIDARG;
|
||
|
||
LPWSTR pName = pszPackageName;
|
||
while (*pName)
|
||
{
|
||
if ((*pName == L':') ||
|
||
(*pName == L',') ||
|
||
(*pName == L';') ||
|
||
(*pName == L'/') ||
|
||
(*pName == L'<') ||
|
||
(*pName == L'>') ||
|
||
(*pName == L'\\'))
|
||
return E_INVALIDARG;
|
||
++pName;
|
||
}
|
||
|
||
// Validating ActivationInfo
|
||
|
||
if (pPackageDetail->pActInfo)
|
||
{
|
||
if (!IsValidReadPtrIn(pPackageDetail->pActInfo, sizeof(ACTIVATIONINFO)))
|
||
return E_INVALIDARG;
|
||
|
||
if (!IsValidReadPtrIn(pPackageDetail->pActInfo->pClasses,
|
||
sizeof(CLASSDETAIL) * (pPackageDetail->pActInfo->cClasses)))
|
||
return E_INVALIDARG;
|
||
|
||
if (!IsValidReadPtrIn(pPackageDetail->pActInfo->prgShellFileExt,
|
||
sizeof(LPOLESTR) * (pPackageDetail->pActInfo->cShellFileExt)))
|
||
return E_INVALIDARG;
|
||
|
||
for (count = 0; count < (pPackageDetail->pActInfo->cShellFileExt); count++)
|
||
{
|
||
if (!pPackageDetail->pActInfo->prgShellFileExt[count])
|
||
return E_INVALIDARG;
|
||
}
|
||
|
||
if (!IsValidReadPtrIn(pPackageDetail->pActInfo->prgPriority,
|
||
sizeof(UINT) * (pPackageDetail->pActInfo->cShellFileExt)))
|
||
return E_INVALIDARG;
|
||
|
||
if (!IsValidReadPtrIn(pPackageDetail->pActInfo->prgInterfaceId,
|
||
sizeof(IID) * (pPackageDetail->pActInfo->cInterfaces)))
|
||
return E_INVALIDARG;
|
||
|
||
if (!IsValidReadPtrIn(pPackageDetail->pActInfo->prgTlbId,
|
||
sizeof(GUID) * (pPackageDetail->pActInfo->cTypeLib)))
|
||
return E_INVALIDARG;
|
||
}
|
||
|
||
// Validating InstallInfo
|
||
|
||
if ((pPackageDetail->pInstallInfo == NULL) ||
|
||
(!IsValidReadPtrIn(pPackageDetail->pInstallInfo, sizeof(INSTALLINFO)))
|
||
)
|
||
return E_INVALIDARG;
|
||
|
||
// Validating PlatformInfo
|
||
|
||
if ((pPackageDetail->pPlatformInfo == NULL) ||
|
||
(!IsValidReadPtrIn(pPackageDetail->pPlatformInfo, sizeof(PLATFORMINFO)))
|
||
)
|
||
return E_INVALIDARG;
|
||
|
||
if (!IsValidReadPtrIn(pPackageDetail->pPlatformInfo->prgPlatform,
|
||
sizeof(CSPLATFORM) * (pPackageDetail->pPlatformInfo->cPlatforms)))
|
||
return E_INVALIDARG;
|
||
|
||
if ((pPackageDetail->pPlatformInfo->cLocales == 0) ||
|
||
(pPackageDetail->pPlatformInfo->prgPlatform == 0))
|
||
return E_INVALIDARG;
|
||
|
||
if (!IsValidReadPtrIn(pPackageDetail->pPlatformInfo->prgLocale,
|
||
sizeof(LCID) * (pPackageDetail->pPlatformInfo->cLocales)))
|
||
return E_INVALIDARG;
|
||
|
||
// Validating other fields in PackageDetail structure
|
||
|
||
if ((pPackageDetail->pszSourceList == NULL) ||
|
||
(!IsValidReadPtrIn(pPackageDetail->pszSourceList,
|
||
sizeof(LPOLESTR) * (pPackageDetail->cSources)))
|
||
)
|
||
return E_INVALIDARG;
|
||
|
||
for (count = 0; count < (pPackageDetail->cSources); count++)
|
||
if (!pPackageDetail->pszSourceList[count])
|
||
return E_INVALIDARG;
|
||
|
||
|
||
if (pPackageDetail->rpCategory)
|
||
{
|
||
if (!IsValidReadPtrIn(pPackageDetail->rpCategory,
|
||
sizeof(GUID) * (pPackageDetail->cCategories)))
|
||
return E_INVALIDARG;
|
||
}
|
||
|
||
//
|
||
// Now we create the package
|
||
//
|
||
|
||
wcscpy (szRDN, L"CN=");
|
||
wcscat (szRDN, pszPackageName);
|
||
|
||
pUnknown = NULL;
|
||
hr = m_ADsPackageContainer->Create(
|
||
CLASS_CS_PACKAGE,
|
||
szRDN,
|
||
&pUnknown
|
||
);
|
||
|
||
RETURN_ON_FAILURE(hr);
|
||
|
||
hr = pUnknown->QueryInterface(
|
||
IID_IADs,
|
||
(void **)&pPackageADs
|
||
);
|
||
|
||
pUnknown->Release();
|
||
|
||
// fill in the activation info
|
||
|
||
if (pPackageDetail->pActInfo)
|
||
{
|
||
|
||
if ((pPackageDetail->pActInfo)->cClasses)
|
||
{
|
||
pszGuid = (LPOLESTR *)CoTaskMemAlloc((pPackageDetail->pActInfo->cClasses)*sizeof(LPOLESTR));
|
||
if (!pszGuid)
|
||
{
|
||
hr = E_OUTOFMEMORY;
|
||
ERROR_ON_FAILURE(hr);
|
||
}
|
||
|
||
for (count = 0; count < pPackageDetail->pActInfo->cClasses; count++)
|
||
{
|
||
pszGuid[count] = (LPOLESTR)CoTaskMemAlloc(sizeof(WCHAR)*STRINGGUIDLEN);
|
||
StringFromGUID(pPackageDetail->pActInfo->pClasses[count].Clsid, pszGuid[count]);
|
||
cPackProgId += pPackageDetail->pActInfo->pClasses[count].cProgId;
|
||
}
|
||
|
||
hr = SetPropertyList(pPackageADs, PKGCLSIDLIST, (pPackageDetail->pActInfo)->cClasses, pszGuid);
|
||
ERROR_ON_FAILURE(hr);
|
||
|
||
for (count = 0; (count < pPackageDetail->pActInfo->cClasses); count++)
|
||
CoTaskMemFree(pszGuid[count]);
|
||
CoTaskMemFree(pszGuid);
|
||
}
|
||
|
||
// collecting all the progids from the various clsids.
|
||
if (cPackProgId)
|
||
{
|
||
pszProgId = (LPOLESTR *)CoTaskMemAlloc(sizeof(LPOLESTR)*cPackProgId);
|
||
if (!pszProgId)
|
||
{
|
||
hr = E_OUTOFMEMORY;
|
||
ERROR_ON_FAILURE(hr);
|
||
}
|
||
|
||
for (count = 0, cPackProgId = 0; count < pPackageDetail->pActInfo->cClasses; count++)
|
||
{
|
||
DWORD cClassProgId = 0;
|
||
for (cClassProgId = 0; cClassProgId < pPackageDetail->pActInfo->pClasses[count].cProgId;
|
||
cClassProgId++)
|
||
{
|
||
pszProgId[cPackProgId++] =
|
||
pPackageDetail->pActInfo->pClasses[count].prgProgId[cClassProgId];
|
||
}
|
||
}
|
||
|
||
hr = SetPropertyList(pPackageADs, PROGIDLIST, cPackProgId, pszProgId);
|
||
|
||
CoTaskMemFree(pszProgId);
|
||
}
|
||
|
||
ERROR_ON_FAILURE(hr);
|
||
|
||
if (pPackageDetail->pActInfo->cShellFileExt)
|
||
{
|
||
//
|
||
// Store a tuple in the format <file ext>:<priority>
|
||
//
|
||
pszGuid = (LPOLESTR *)CoTaskMemAlloc((pPackageDetail->pActInfo->cShellFileExt) * sizeof(LPOLESTR));
|
||
if (!pszGuid)
|
||
{
|
||
hr = E_OUTOFMEMORY;
|
||
ERROR_ON_FAILURE(hr);
|
||
}
|
||
for (count = 0; count < pPackageDetail->pActInfo->cShellFileExt; count++)
|
||
{
|
||
pszGuid[count] = (LPOLESTR)CoTaskMemAlloc(sizeof(WCHAR) *
|
||
(wcslen(pPackageDetail->pActInfo->prgShellFileExt[count])+1+2+1));
|
||
// format is fileext+:+nn+NULL where nn = 2 digit priority
|
||
wsprintf(pszGuid[count], L"%s:%2d",
|
||
pPackageDetail->pActInfo->prgShellFileExt[count],
|
||
pPackageDetail->pActInfo->prgPriority[count]);
|
||
}
|
||
hr = SetPropertyList(pPackageADs, PKGFILEEXTNLIST, pPackageDetail->pActInfo->cShellFileExt, pszGuid);
|
||
for (count = 0; (count < pPackageDetail->pActInfo->cShellFileExt); count++)
|
||
CoTaskMemFree(pszGuid[count]);
|
||
CoTaskMemFree(pszGuid);
|
||
|
||
//
|
||
// Now IDS Workaround
|
||
// BUGBUG. Remove this when the DS bug is fixed. 130491 in NTDEV
|
||
//
|
||
hr = SetPropertyList(pPackageADs,
|
||
QRYFILEEXT,
|
||
pPackageDetail->pActInfo->cShellFileExt,
|
||
pPackageDetail->pActInfo->prgShellFileExt);
|
||
|
||
}
|
||
|
||
ERROR_ON_FAILURE(hr);
|
||
|
||
if (pPackageDetail->pActInfo->cInterfaces)
|
||
{
|
||
pszGuid = (LPOLESTR *)CoTaskMemAlloc((pPackageDetail->pActInfo->cInterfaces)*sizeof(LPOLESTR));
|
||
if (!pszGuid)
|
||
{
|
||
hr = E_OUTOFMEMORY;
|
||
ERROR_ON_FAILURE(hr);
|
||
}
|
||
|
||
for (count = 0; (count < (pPackageDetail->pActInfo->cInterfaces)); count++)
|
||
{
|
||
pszGuid[count] = (LPOLESTR)CoTaskMemAlloc(STRINGGUIDLEN*sizeof(WCHAR));
|
||
if (!pszGuid[count])
|
||
{
|
||
hr = E_OUTOFMEMORY;
|
||
ERROR_ON_FAILURE(hr);
|
||
}
|
||
|
||
StringFromGUID((pPackageDetail->pActInfo->prgInterfaceId)[count], pszGuid[count]);
|
||
}
|
||
|
||
hr = SetPropertyList(pPackageADs, PKGIIDLIST, pPackageDetail->pActInfo->cInterfaces,
|
||
pszGuid);
|
||
ERROR_ON_FAILURE(hr);
|
||
|
||
for (count = 0; (count < (pPackageDetail->pActInfo->cInterfaces)); count++)
|
||
CoTaskMemFree(pszGuid[count]);
|
||
CoTaskMemFree(pszGuid);
|
||
}
|
||
|
||
|
||
if (pPackageDetail->pActInfo->cTypeLib)
|
||
{
|
||
pszGuid = (LPOLESTR *)CoTaskMemAlloc((pPackageDetail->pActInfo->cTypeLib)*sizeof(LPOLESTR));
|
||
if (!pszGuid)
|
||
{
|
||
hr = E_OUTOFMEMORY;
|
||
ERROR_ON_FAILURE(hr);
|
||
}
|
||
|
||
for (count = 0; (count < (pPackageDetail->pActInfo)->cTypeLib); count++)
|
||
{
|
||
pszGuid[count] = (LPOLESTR)CoTaskMemAlloc(STRINGGUIDLEN*sizeof(WCHAR));
|
||
if (!pszGuid[count])
|
||
{
|
||
hr = E_OUTOFMEMORY;
|
||
ERROR_ON_FAILURE(hr);
|
||
}
|
||
|
||
StringFromGUID((pPackageDetail->pActInfo->prgTlbId)[count], pszGuid[count]);
|
||
}
|
||
|
||
hr = SetPropertyList(pPackageADs, PKGTLBIDLIST, pPackageDetail->pActInfo->cTypeLib,
|
||
pszGuid);
|
||
ERROR_ON_FAILURE(hr);
|
||
|
||
for (count = 0; (count < (pPackageDetail->pActInfo->cTypeLib)); count++)
|
||
CoTaskMemFree(pszGuid[count]);
|
||
CoTaskMemFree(pszGuid);
|
||
}
|
||
}
|
||
// fill in the platforminfo
|
||
|
||
// BUGBUG::***os version
|
||
if ((pPackageDetail->pPlatformInfo)->cPlatforms)
|
||
{
|
||
pdwArch = (DWORD *)CoTaskMemAlloc(sizeof(DWORD)*
|
||
((pPackageDetail->pPlatformInfo)->cPlatforms));
|
||
|
||
for (count = 0; (count < (pPackageDetail->pPlatformInfo)->cPlatforms); count++)
|
||
UnpackPlatform (pdwArch+count, ((pPackageDetail->pPlatformInfo)->prgPlatform)+count);
|
||
|
||
hr = SetPropertyListDW (pPackageADs, ARCHLIST, (pPackageDetail->pPlatformInfo)->cPlatforms, pdwArch);
|
||
ERROR_ON_FAILURE(hr);
|
||
|
||
CoTaskMemFree(pdwArch);
|
||
}
|
||
|
||
if ((pPackageDetail->pPlatformInfo)->cLocales)
|
||
{
|
||
hr = SetPropertyListDW (pPackageADs,
|
||
LOCALEID, (pPackageDetail->pPlatformInfo)->cLocales,
|
||
(pPackageDetail->pPlatformInfo)->prgLocale);
|
||
ERROR_ON_FAILURE(hr);
|
||
}
|
||
|
||
// fill in the installinfo
|
||
|
||
hr = SetProperty(pPackageADs, PACKAGENAME, pszPackageName);
|
||
ERROR_ON_FAILURE(hr);
|
||
|
||
hr = SetPropertyDW (pPackageADs, PACKAGETYPE, (DWORD)pPackageDetail->pInstallInfo->PathType);
|
||
ERROR_ON_FAILURE(hr);
|
||
|
||
if (pPackageDetail->pInstallInfo->pszScriptPath)
|
||
{
|
||
hr = SetProperty(pPackageADs, SCRIPTPATH, pPackageDetail->pInstallInfo->pszScriptPath);
|
||
ERROR_ON_FAILURE(hr);
|
||
}
|
||
|
||
if (pPackageDetail->pInstallInfo->pszSetupCommand)
|
||
{
|
||
hr = SetProperty(pPackageADs, SETUPCOMMAND, pPackageDetail->pInstallInfo->pszSetupCommand);
|
||
ERROR_ON_FAILURE(hr);
|
||
}
|
||
|
||
if (pPackageDetail->pInstallInfo->pszUrl)
|
||
{
|
||
hr = SetProperty(pPackageADs, HELPURL, pPackageDetail->pInstallInfo->pszUrl);
|
||
ERROR_ON_FAILURE(hr);
|
||
}
|
||
|
||
GetCurrentUsn(&Usn);
|
||
hr = UsnUpd(pPackageADs, PKGUSN, &Usn);
|
||
ERROR_ON_FAILURE(hr);
|
||
|
||
hr = SetPropertyDW (pPackageADs, PACKAGEFLAGS, pPackageDetail->pInstallInfo->dwActFlags);
|
||
ERROR_ON_FAILURE(hr);
|
||
|
||
hr = SetPropertyDW (pPackageADs, CLASSCTX, pPackageDetail->pInstallInfo->dwComClassContext);
|
||
ERROR_ON_FAILURE(hr);
|
||
|
||
hr = SetPropertyDW (pPackageADs, VERSIONHI, pPackageDetail->pInstallInfo->dwVersionHi);
|
||
ERROR_ON_FAILURE(hr);
|
||
|
||
hr = SetPropertyDW (pPackageADs, VERSIONLO, pPackageDetail->pInstallInfo->dwVersionLo);
|
||
ERROR_ON_FAILURE(hr);
|
||
|
||
hr = SetPropertyDW (pPackageADs, SCRIPTSIZE, pPackageDetail->pInstallInfo->cScriptLen);
|
||
ERROR_ON_FAILURE(hr);
|
||
|
||
hr = SetPropertyDW (pPackageADs, UILEVEL, (DWORD)pPackageDetail->pInstallInfo->InstallUiLevel);
|
||
ERROR_ON_FAILURE(hr);
|
||
|
||
|
||
if (pPackageDetail->pInstallInfo->cUpgrades)
|
||
{
|
||
LPOLESTR *rpszUpgrades;
|
||
rpszUpgrades = (LPOLESTR *)CoTaskMemAlloc(sizeof(LPOLESTR)*pPackageDetail->pInstallInfo->cUpgrades);
|
||
for (count = 0; (count < pPackageDetail->pInstallInfo->cUpgrades); count++)
|
||
{
|
||
UINT l = wcslen(pPackageDetail->pInstallInfo->prgUpgradeScript[count]);
|
||
rpszUpgrades[count] = (LPOLESTR)CoTaskMemAlloc(sizeof(WCHAR) *(4+l));
|
||
wsprintf(rpszUpgrades[count], L"%S:%1d",
|
||
pPackageDetail->pInstallInfo->prgUpgradeScript[count],
|
||
pPackageDetail->pInstallInfo->prgUpgradeFlag[count]);
|
||
}
|
||
|
||
hr = SetPropertyList(pPackageADs, UPGRADESCRIPTNAMES, pPackageDetail->pInstallInfo->cUpgrades,
|
||
rpszUpgrades);
|
||
|
||
ERROR_ON_FAILURE(hr);
|
||
for (count = 0; (count < pPackageDetail->pInstallInfo->cUpgrades); count++)
|
||
CoTaskMemFree(rpszUpgrades[count]);
|
||
CoTaskMemFree(rpszUpgrades);
|
||
}
|
||
|
||
|
||
// fill in the sources
|
||
if (pPackageDetail->cSources)
|
||
{
|
||
hr = SetPropertyList(pPackageADs, MSIFILELIST, pPackageDetail->cSources,
|
||
pPackageDetail->pszSourceList);
|
||
ERROR_ON_FAILURE(hr);
|
||
}
|
||
|
||
// fill in the categories
|
||
if (pPackageDetail->cCategories)
|
||
{
|
||
pszGuid = (LPOLESTR *)CoTaskMemAlloc((pPackageDetail->cCategories) * sizeof(LPOLESTR));
|
||
if (!pszGuid)
|
||
{
|
||
hr = E_OUTOFMEMORY;
|
||
ERROR_ON_FAILURE(hr);
|
||
}
|
||
|
||
for (count = 0; (count < pPackageDetail->cCategories); count++)
|
||
{
|
||
pszGuid[count] = (LPOLESTR)CoTaskMemAlloc(STRINGGUIDLEN*sizeof(WCHAR));
|
||
if (!pszGuid[count])
|
||
{
|
||
hr = E_OUTOFMEMORY;
|
||
ERROR_ON_FAILURE(hr);
|
||
}
|
||
|
||
StringFromGUID((pPackageDetail->rpCategory)[count], pszGuid[count]);
|
||
}
|
||
|
||
hr = SetPropertyList(pPackageADs, PKGCATEGORYLIST, pPackageDetail->cCategories,
|
||
pszGuid);
|
||
ERROR_ON_FAILURE(hr);
|
||
|
||
for (count = 0; (count < pPackageDetail->cCategories); count++)
|
||
CoTaskMemFree(pszGuid[count]);
|
||
CoTaskMemFree(pszGuid);
|
||
}
|
||
|
||
//
|
||
// Store the script in the directory
|
||
//
|
||
if ((pPackageDetail->pInstallInfo->dwActFlags & SCRIPT_IN_DIRECTORY) &&
|
||
(pPackageDetail->pInstallInfo->cScriptLen))
|
||
{
|
||
|
||
if ((pPackageDetail->pInstallInfo->cScriptLen) &&
|
||
(!IsValidReadPtrIn(pPackageDetail->pInstallInfo->pScript, pPackageDetail->pInstallInfo->cScriptLen)))
|
||
return E_INVALIDARG;
|
||
|
||
SAFEARRAYBOUND size; // Get rid of 16
|
||
SAFEARRAY FAR *psa;
|
||
CHAR HUGEP *pArray=NULL;
|
||
LONG dwSLBound = 0;
|
||
LONG dwSUBound = 0;
|
||
VARIANT vData;
|
||
|
||
VariantInit(&vData);
|
||
size.cElements = pPackageDetail->pInstallInfo->cScriptLen;
|
||
size.lLbound = 0;
|
||
|
||
psa = SafeArrayCreate(VT_UI1, 1, &size);
|
||
if (!psa) {
|
||
return(E_OUTOFMEMORY);
|
||
}
|
||
|
||
hr = SafeArrayAccessData( psa, (void HUGEP * FAR *) &pArray );
|
||
RETURN_ON_FAILURE(hr);
|
||
memcpy( pArray, pPackageDetail->pInstallInfo->pScript, size.cElements );
|
||
SafeArrayUnaccessData( psa );
|
||
|
||
V_VT(&vData) = VT_ARRAY | VT_UI1;
|
||
V_ARRAY(&vData) = psa;
|
||
hr = pPackageADs->Put(PKGSCRIPT, vData);
|
||
VariantClear(&vData);
|
||
ERROR_ON_FAILURE(hr);
|
||
}
|
||
|
||
hr = StoreIt(pPackageADs);
|
||
ERROR_ON_FAILURE(hr);
|
||
|
||
for (count = 0; count < pPackageDetail->pActInfo->cClasses; count++)
|
||
{
|
||
hr = NewClass((pPackageDetail->pActInfo->pClasses)+count);
|
||
ERROR_ON_FAILURE(hr);
|
||
}
|
||
|
||
|
||
Error_Cleanup:
|
||
pPackageADs->Release();
|
||
return hr;
|
||
}
|
||
|
||
|
||
|
||
|