172 lines
3.7 KiB
C++
172 lines
3.7 KiB
C++
///////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// Copyright (c) 1998, Microsoft Corp. All rights reserved.
|
|
//
|
|
// FILE
|
|
//
|
|
// iasnap.cpp
|
|
//
|
|
// SYNOPSIS
|
|
//
|
|
// Implementation of DLL exports for an ATL in proc server.
|
|
//
|
|
// MODIFICATION HISTORY
|
|
//
|
|
// 02/05/1998 Original version.
|
|
// 02/03/2000 Added ProxyPolicyEnforcer & UserRestrictions.
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
#include <ias.h>
|
|
#include <iasutil.h>
|
|
#include <newop.cpp>
|
|
|
|
#include <enforcer.h>
|
|
#include <match.h>
|
|
#include <ntgroups.h>
|
|
#include <timeofday.h>
|
|
#include <userr.h>
|
|
|
|
CComModule _Module;
|
|
#include <atlimpl.cpp>
|
|
|
|
BEGIN_OBJECT_MAP(ObjectMap)
|
|
OBJECT_ENTRY(__uuidof(AttributeMatch), AttributeMatch)
|
|
OBJECT_ENTRY(__uuidof(NTGroups), NTGroups)
|
|
OBJECT_ENTRY(__uuidof(TimeOfDay), TimeOfDay)
|
|
OBJECT_ENTRY(__uuidof(PolicyEnforcer),
|
|
IASTL::IASRequestHandlerObject<PolicyEnforcer> )
|
|
OBJECT_ENTRY(__uuidof(ProxyPolicyEnforcer),
|
|
IASTL::IASRequestHandlerObject<ProxyPolicyEnforcer> )
|
|
OBJECT_ENTRY(__uuidof(URHandler),
|
|
IASTL::IASRequestHandlerObject<UserRestrictions> )
|
|
END_OBJECT_MAP()
|
|
|
|
|
|
//////////
|
|
// DLL Entry Point
|
|
//////////
|
|
BOOL
|
|
WINAPI
|
|
DllMain(
|
|
HINSTANCE hInstance,
|
|
DWORD dwReason,
|
|
LPVOID /*lpReserved*/
|
|
)
|
|
{
|
|
if (dwReason == DLL_PROCESS_ATTACH)
|
|
{
|
|
_Module.Init(ObjectMap, hInstance);
|
|
|
|
DisableThreadLibraryCalls(hInstance);
|
|
}
|
|
else if (dwReason == DLL_PROCESS_DETACH)
|
|
{
|
|
_Module.Term();
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
|
|
//////////
|
|
// Used to determine whether the DLL can be unloaded by OLE
|
|
//////////
|
|
STDAPI DllCanUnloadNow(void)
|
|
{
|
|
return (_Module.GetLockCount()==0) ? S_OK : S_FALSE;
|
|
}
|
|
|
|
|
|
//////////
|
|
// Returns a class factory to create an object of the requested type.
|
|
//////////
|
|
STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID* ppv)
|
|
{
|
|
return _Module.GetClassObject(rclsid, riid, ppv);
|
|
}
|
|
|
|
|
|
//////////
|
|
// DllRegisterServer - Adds entries to the system registry
|
|
//////////
|
|
STDAPI DllRegisterServer(void)
|
|
{
|
|
return _Module.RegisterServer(TRUE);
|
|
}
|
|
|
|
|
|
//////////
|
|
// DllUnregisterServer - Removes entries from the system registry
|
|
//////////
|
|
STDAPI DllUnregisterServer(void)
|
|
{
|
|
HRESULT hr = _Module.UnregisterServer();
|
|
|
|
if (FAILED(hr)) return hr;
|
|
|
|
hr = UnRegisterTypeLib(__uuidof(NetworkPolicy),
|
|
1,
|
|
0,
|
|
MAKELANGID(LANG_NEUTRAL, SUBLANG_NEUTRAL),
|
|
SYS_WIN32);
|
|
|
|
return hr;
|
|
}
|
|
|
|
#include <BuildTree.h>
|
|
|
|
#define IASNAPAPI
|
|
#include <xprparse.h>
|
|
|
|
//////////
|
|
// Apply an expression to a request.
|
|
//////////
|
|
HRESULT
|
|
WINAPI
|
|
IASEvaluateExpression(
|
|
IRequest* pRequest,
|
|
PCWSTR szExpression,
|
|
VARIANT_BOOL *pVal
|
|
)
|
|
{
|
|
if (pRequest == NULL || pVal == NULL) { return E_INVALIDARG; }
|
|
|
|
// Parse the expression into an array of tokens.
|
|
_variant_t v;
|
|
RETURN_ERROR(IASParseExpression(szExpression, &v));
|
|
|
|
// Build a logic tree from the tokens.
|
|
CComPtr<ICondition> condition;
|
|
RETURN_ERROR(IASBuildExpression(&v, &condition));
|
|
|
|
// Evaluate the expression.
|
|
return condition->IsTrue(pRequest, pVal);
|
|
}
|
|
|
|
//////////
|
|
// Evaluate a one-shot time of day constraint.
|
|
//////////
|
|
HRESULT
|
|
WINAPI
|
|
IASEvaluateTimeOfDay(
|
|
PCWSTR szTimeOfDay,
|
|
VARIANT_BOOL *pVal
|
|
)
|
|
{
|
|
if (pVal == NULL) { return E_INVALIDARG; }
|
|
|
|
// Convert the text to an hour map.
|
|
BYTE hourMap[IAS_HOUR_MAP_LENGTH];
|
|
DWORD dw = IASHourMapFromText(szTimeOfDay, hourMap);
|
|
if (dw != NO_ERROR) { return HRESULT_FROM_WIN32(dw); }
|
|
|
|
// Test the current hour.
|
|
*pVal = theTimeOfDayEvaluator.isCurrentHourSet(hourMap) ? VARIANT_TRUE
|
|
: VARIANT_FALSE;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
#include <xprparse.cpp>
|