#include "Python.h"

#include "windows.h"

#ifdef Py_ENABLE_SHARED

#ifdef MS_DLL_ID

char dllVersionBuffer[16] = MS_DLL_ID;

char dllVersionBuffer[16] = ""; // a private buffer

HMODULE PyWin_DLLhModule = NULL;

const char *PyWin_DLLVersionString = dllVersionBuffer;

#if HAVE_SXS

typedef BOOL (WINAPI * PFN_GETCURRENTACTCTX)(HANDLE *);

typedef BOOL (WINAPI * PFN_ACTIVATEACTCTX)(HANDLE, ULONG_PTR *);

typedef BOOL (WINAPI * PFN_DEACTIVATEACTCTX)(DWORD, ULONG_PTR);

typedef BOOL (WINAPI * PFN_ADDREFACTCTX)(HANDLE);

typedef BOOL (WINAPI * PFN_RELEASEACTCTX)(HANDLE);

static HANDLE PyWin_DLLhActivationContext = NULL; // one day it might be public

static PFN_GETCURRENTACTCTX pfnGetCurrentActCtx = NULL;

static PFN_ACTIVATEACTCTX pfnActivateActCtx = NULL;

static PFN_DEACTIVATEACTCTX pfnDeactivateActCtx = NULL;

static PFN_ADDREFACTCTX pfnAddRefActCtx = NULL;

static PFN_RELEASEACTCTX pfnReleaseActCtx = NULL;

void _LoadActCtxPointers()

{

HINSTANCE hKernel32 = GetModuleHandleW(L"kernel32.dll");

if (hKernel32)

pfnGetCurrentActCtx = (PFN_GETCURRENTACTCTX) GetProcAddress(hKernel32, "GetCurrentActCtx");

// If we can't load GetCurrentActCtx (ie, pre XP) , don't bother with the rest.

if (pfnGetCurrentActCtx) {

pfnActivateActCtx = (PFN_ACTIVATEACTCTX) GetProcAddress(hKernel32, "ActivateActCtx");

pfnDeactivateActCtx = (PFN_DEACTIVATEACTCTX) GetProcAddress(hKernel32, "DeactivateActCtx");

pfnAddRefActCtx = (PFN_ADDREFACTCTX) GetProcAddress(hKernel32, "AddRefActCtx");

pfnReleaseActCtx = (PFN_RELEASEACTCTX) GetProcAddress(hKernel32, "ReleaseActCtx");

}

}

ULONG_PTR _Py_ActivateActCtx()

{

ULONG_PTR ret = 0;

if (PyWin_DLLhActivationContext && pfnActivateActCtx)

if (!(*pfnActivateActCtx)(PyWin_DLLhActivationContext, &ret)) {

OutputDebugString("Python failed to activate the activation context before loading a DLL\n");

ret = 0; // no promise the failing function didn't change it!

}

return ret;

}

void _Py_DeactivateActCtx(ULONG_PTR cookie)

{

if (cookie && pfnDeactivateActCtx)

if (!(*pfnDeactivateActCtx)(0, cookie))

OutputDebugString("Python failed to de-activate the activation context\n");

}

#endif /* HAVE_SXS */

BOOL WINAPI DllMain (HANDLE hInst,

ULONG ul_reason_for_call,

LPVOID lpReserved)

{

switch (ul_reason_for_call)

{

case DLL_PROCESS_ATTACH:

PyWin_DLLhModule = hInst;

#ifndef MS_DLL_ID

// If we have MS_DLL_ID, we don't need to load the string.

// 1000 is a magic number I picked out of the air. Could do with a #define, I spose...

LoadString(hInst, 1000, dllVersionBuffer, sizeof(dllVersionBuffer));

#if HAVE_SXS

// and capture our activation context for use when loading extensions.

_LoadActCtxPointers();

if (pfnGetCurrentActCtx && pfnAddRefActCtx)

if ((*pfnGetCurrentActCtx)(&PyWin_DLLhActivationContext))

if (!(*pfnAddRefActCtx)(PyWin_DLLhActivationContext))

OutputDebugString("Python failed to load the default activation context\n");

break;

case DLL_PROCESS_DETACH:

#if HAVE_SXS

if (pfnReleaseActCtx)

(*pfnReleaseActCtx)(PyWin_DLLhActivationContext);

break;

}

return TRUE;

}

#endif /* Py_ENABLE_SHARED */