Leaked source code of windows server 2003
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.
 
 
 
 
 
 

123 lines
2.8 KiB

/*****************************************************************************
*
* (C) COPYRIGHT MICROSOFT CORPORATION, 2001 - 2003
*
* TITLE: fusutils.cpp
*
* VERSION: 1.0
*
* DATE: 14-Feb-2001
*
* DESCRIPTION: Fusion utilities
*
*****************************************************************************/
#pragma once
#include "precomp.h"
#include "globals.h"
#include "fusutils.h"
// StrSafe.h needs to be included last
// to disallow bad string functions.
#include <STRSAFE.H>
#define MAX_LOOP 10
HANDLE GetMyActivationContext()
{
// Make sure we've created our activation context.
CreateMyActivationContext();
// Return the global.
return ghActCtx;
}
BOOL CreateMyActivationContext()
{
if(INVALID_HANDLE_VALUE != ghActCtx)
{
return TRUE;
}
ghActCtx = CreateActivationContextFromResource(ghInstance, MAKEINTRESOURCE(MANIFEST_RESOURCE));
return (INVALID_HANDLE_VALUE != ghActCtx);
}
HANDLE CreateActivationContextFromResource(HMODULE hModule, LPCTSTR pszResourceName)
{
DWORD dwSize = MAX_PATH;
DWORD dwUsed = 0;
DWORD dwLoop;
PTSTR pszModuleName = NULL;
ACTCTX act;
HANDLE hActCtx = INVALID_HANDLE_VALUE;
// Get the name for the module that contains the manifest resource
// to create the Activation Context from.
dwLoop = 0;
do
{
// May need to allocate or re-alloc buffer for module name.
if(NULL != pszModuleName)
{
// Need to re-alloc bigger buffer.
// First, delete old buffer.
delete[] pszModuleName;
// Second, increase buffer alloc size.
dwSize <<= 1;
}
pszModuleName = new TCHAR[dwSize];
if(NULL == pszModuleName)
{
goto Exit;
}
// Try to get the module name.
dwUsed = GetModuleFileName(hModule, pszModuleName, dwSize);
// Check to see if it failed.
if(0 == dwUsed)
{
goto Exit;
}
// If dwUsed is equla to dwSize or larger,
// the buffer passed in wasn't big enough.
} while ( (dwUsed >= dwSize) && (++dwLoop < MAX_LOOP) );
// Now let's try to create an activation context
// from manifest resource.
::ZeroMemory(&act, sizeof(act));
act.cbSize = sizeof(act);
act.dwFlags = ACTCTX_FLAG_RESOURCE_NAME_VALID;
act.lpResourceName = pszResourceName;
act.lpSource = pszModuleName;
hActCtx = CreateActCtx(&act);
Exit:
//
// Clean up.
//
if(NULL != pszModuleName)
{
delete[] pszModuleName;
}
return hActCtx;
}