141 lines
3.4 KiB
C++
141 lines
3.4 KiB
C++
|
///////////////////////////////////////////////////////////////////////////////
|
||
|
|
//
|
||
|
|
// Copyright (c) 1998, Microsoft Corp. All rights reserved.
|
||
|
|
//
|
||
|
|
// FILE
|
||
|
|
//
|
||
|
|
// extension.cpp
|
||
|
|
//
|
||
|
|
// SYNOPSIS
|
||
|
|
//
|
||
|
|
// Defines the class BaseCampExtension.
|
||
|
|
//
|
||
|
|
// MODIFICATION HISTORY
|
||
|
|
//
|
||
|
|
// 12/01/1998 Original version.
|
||
|
|
// 04/01/1999 Add new entry points.
|
||
|
|
//
|
||
|
|
///////////////////////////////////////////////////////////////////////////////
|
||
|
|
|
||
|
|
#include <ias.h>
|
||
|
|
#include <new>
|
||
|
|
#include <extension.h>
|
||
|
|
|
||
|
|
BaseCampExtension::BaseCampExtension() throw ()
|
||
|
|
: hModule(NULL),
|
||
|
|
name(NULL),
|
||
|
|
initFn(NULL),
|
||
|
|
termFn(NULL),
|
||
|
|
processFn(NULL),
|
||
|
|
processExFn(NULL),
|
||
|
|
freeAttrsFn(NULL)
|
||
|
|
|
||
|
|
{ }
|
||
|
|
|
||
|
|
BaseCampExtension::~BaseCampExtension() throw ()
|
||
|
|
{
|
||
|
|
if (termFn) { termFn(); }
|
||
|
|
FreeLibrary(hModule);
|
||
|
|
delete[] name;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Loads the extension DLL.
|
||
|
|
DWORD BaseCampExtension::load(PCWSTR dllPath) throw ()
|
||
|
|
{
|
||
|
|
// Load the extension DLL.
|
||
|
|
hModule = LoadLibraryW(dllPath);
|
||
|
|
if (!hModule)
|
||
|
|
{
|
||
|
|
DWORD error = GetLastError();
|
||
|
|
IASTraceFailure("LoadLibraryW", error);
|
||
|
|
return error;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Look-up the entry points.
|
||
|
|
initFn = (PRADIUS_EXTENSION_INIT)
|
||
|
|
GetProcAddress(
|
||
|
|
hModule,
|
||
|
|
RADIUS_EXTENSION_INIT
|
||
|
|
);
|
||
|
|
processFn = (PRADIUS_EXTENSION_PROCESS)
|
||
|
|
GetProcAddress(
|
||
|
|
hModule,
|
||
|
|
RADIUS_EXTENSION_PROCESS
|
||
|
|
);
|
||
|
|
processExFn = (PRADIUS_EXTENSION_PROCESS_EX)
|
||
|
|
GetProcAddress(
|
||
|
|
hModule,
|
||
|
|
RADIUS_EXTENSION_PROCESS_EX
|
||
|
|
);
|
||
|
|
freeAttrsFn = (PRADIUS_EXTENSION_FREE_ATTRIBUTES)
|
||
|
|
GetProcAddress(
|
||
|
|
hModule,
|
||
|
|
RADIUS_EXTENSION_FREE_ATTRIBUTES
|
||
|
|
);
|
||
|
|
|
||
|
|
// Validate the entry points.
|
||
|
|
if (!processFn && !processExFn)
|
||
|
|
{
|
||
|
|
IASTraceString("Either RadiusExtensionProcess or "
|
||
|
|
"RadiusExtensionProcessEx must be defined.");
|
||
|
|
return ERROR_PROC_NOT_FOUND;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (processExFn && !freeAttrsFn)
|
||
|
|
{
|
||
|
|
IASTraceString("RadiusExtensionFreeAttributes must be defined if "
|
||
|
|
"RadiusExtensionProcessEx is defined.");
|
||
|
|
return ERROR_PROC_NOT_FOUND;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Initialize the DLL.
|
||
|
|
if (initFn)
|
||
|
|
{
|
||
|
|
DWORD error = initFn();
|
||
|
|
if (error)
|
||
|
|
{
|
||
|
|
IASTraceFailure("RadiusExtensionInit", error);
|
||
|
|
return error;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// We look up the term function last, so that it won't get invoked in
|
||
|
|
// the destructor unless everything else succeeded.
|
||
|
|
termFn = (PRADIUS_EXTENSION_TERM)
|
||
|
|
GetProcAddress(
|
||
|
|
hModule,
|
||
|
|
RADIUS_EXTENSION_TERM
|
||
|
|
);
|
||
|
|
|
||
|
|
// Strip everything before the last backslash.
|
||
|
|
const WCHAR* basename = wcsrchr(dllPath, L'\\');
|
||
|
|
if (basename == NULL)
|
||
|
|
{
|
||
|
|
basename = dllPath;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
++basename;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Save the basename.
|
||
|
|
if (name = new (std::nothrow) WCHAR[wcslen(basename) + 1])
|
||
|
|
{
|
||
|
|
wcscpy(name, basename);
|
||
|
|
}
|
||
|
|
|
||
|
|
return NO_ERROR;
|
||
|
|
}
|
||
|
|
|
||
|
|
// Process a request.
|
||
|
|
DWORD BaseCampExtension::process(
|
||
|
|
IN CONST RADIUS_ATTRIBUTE* pInAttrs,
|
||
|
|
OUT PRADIUS_ATTRIBUTE* pOutAttrs,
|
||
|
|
OUT PRADIUS_ACTION pfAction
|
||
|
|
) throw ()
|
||
|
|
{
|
||
|
|
// Prefer the new entry point.
|
||
|
|
return processExFn ? processExFn(pInAttrs, pOutAttrs, pfAction)
|
||
|
|
: processFn(pInAttrs, pfAction);
|
||
|
|
}
|