518 lines
14 KiB
C++
518 lines
14 KiB
C++
|
//#--------------------------------------------------------------
|
||
|
|
//
|
||
|
|
// File: sendtopipe.cpp
|
||
|
|
//
|
||
|
|
// Synopsis: Implementation of CSendToPipe class methods
|
||
|
|
//
|
||
|
|
//
|
||
|
|
// History: 11/22/97 MKarki Created
|
||
|
|
// 06/12/98 SBens Changed put_Response to SetResponse.
|
||
|
|
//
|
||
|
|
// Copyright (C) 1997-98 Microsoft Corporation
|
||
|
|
// All rights reserved.
|
||
|
|
//
|
||
|
|
//----------------------------------------------------------------
|
||
|
|
#include "radcommon.h"
|
||
|
|
#include "sendtopipe.h"
|
||
|
|
|
||
|
|
//+++-------------------------------------------------------------
|
||
|
|
//
|
||
|
|
// Function: CSendToPipe
|
||
|
|
//
|
||
|
|
// Synopsis: This is the constructor of the CSendToPipe
|
||
|
|
// class method
|
||
|
|
//
|
||
|
|
// Arguments: NONE
|
||
|
|
//
|
||
|
|
// Returns: NONE
|
||
|
|
//
|
||
|
|
// History: MKarki Created 11/22/97
|
||
|
|
//
|
||
|
|
//----------------------------------------------------------------
|
||
|
|
CSendToPipe::CSendToPipe()
|
||
|
|
: m_pIRequestHandler (NULL),
|
||
|
|
m_pIRequestSource (NULL),
|
||
|
|
m_pIClassFactory (NULL),
|
||
|
|
m_pCVSAFilter (NULL),
|
||
|
|
m_pCReportEvent (NULL)
|
||
|
|
{
|
||
|
|
} // end of CSendToPipe constructor
|
||
|
|
|
||
|
|
//++--------------------------------------------------------------
|
||
|
|
//
|
||
|
|
// Function: ~CSendToPipe
|
||
|
|
//
|
||
|
|
// Synopsis: This is the destructor of the CSendToPipe
|
||
|
|
// class method
|
||
|
|
//
|
||
|
|
// Arguments: NONE
|
||
|
|
//
|
||
|
|
// Returns: NONE
|
||
|
|
//
|
||
|
|
// History: MKarki Created 11/22/97
|
||
|
|
//
|
||
|
|
//----------------------------------------------------------------
|
||
|
|
CSendToPipe::~CSendToPipe()
|
||
|
|
{
|
||
|
|
if (m_pIClassFactory) { m_pIClassFactory->Release(); }
|
||
|
|
|
||
|
|
} // end of CSendToPipe destructor
|
||
|
|
|
||
|
|
//+++-------------------------------------------------------------
|
||
|
|
//
|
||
|
|
// Function: Process
|
||
|
|
//
|
||
|
|
// Synopsis: This is the public method of the CSendToPipe class
|
||
|
|
// that gets hold of a IRequestRaw interface, puts the
|
||
|
|
// data in it and sends it on its way.
|
||
|
|
//
|
||
|
|
// Arguments:
|
||
|
|
// [in] CPacketRadius*
|
||
|
|
//
|
||
|
|
// Returns: HRESULT - status
|
||
|
|
//
|
||
|
|
// History: MKarki Created 11/22/97
|
||
|
|
//
|
||
|
|
//----------------------------------------------------------------
|
||
|
|
HRESULT
|
||
|
|
CSendToPipe::Process(
|
||
|
|
CPacketRadius *pCPacketRadius
|
||
|
|
)
|
||
|
|
{
|
||
|
|
BOOL bRetVal = FALSE;
|
||
|
|
BOOL bStatus = FALSE;
|
||
|
|
HRESULT hr = S_OK;
|
||
|
|
IRequest *pIRequest = NULL;
|
||
|
|
IAttributesRaw *pIAttributesRaw = NULL;
|
||
|
|
PATTRIBUTEPOSITION pIasAttribPos = NULL;
|
||
|
|
DWORD dwCount = 0;
|
||
|
|
DWORD dwRetVal = 0;
|
||
|
|
PACKETTYPE ePacketType;
|
||
|
|
IRequestHandler *pIRequestHandler = m_pIRequestHandler;
|
||
|
|
|
||
|
|
|
||
|
|
_ASSERT (pCPacketRadius);
|
||
|
|
|
||
|
|
__try
|
||
|
|
{
|
||
|
|
//
|
||
|
|
// check if the pipeline is present to process our
|
||
|
|
// request
|
||
|
|
//
|
||
|
|
if (NULL != pIRequestHandler)
|
||
|
|
{
|
||
|
|
pIRequestHandler->AddRef ();
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
//
|
||
|
|
// should never reach here
|
||
|
|
//
|
||
|
|
_ASSERT (0);
|
||
|
|
IASTracePrintf (
|
||
|
|
"Unable to send request to backend as request handler "
|
||
|
|
"unavailable"
|
||
|
|
);
|
||
|
|
hr = E_FAIL;
|
||
|
|
__leave;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// create the Request COM object here
|
||
|
|
//
|
||
|
|
hr = m_pIClassFactory->CreateInstance (
|
||
|
|
NULL,
|
||
|
|
__uuidof (IRequest),
|
||
|
|
reinterpret_cast <PVOID*> (&pIRequest)
|
||
|
|
);
|
||
|
|
if (FAILED (hr))
|
||
|
|
{
|
||
|
|
IASTracePrintf (
|
||
|
|
"Unable to create a Request object from class factory "
|
||
|
|
"before sending request to backend"
|
||
|
|
);
|
||
|
|
__leave;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// get the packettype
|
||
|
|
//
|
||
|
|
ePacketType = pCPacketRadius->GetInCode ();
|
||
|
|
|
||
|
|
//
|
||
|
|
// get the attributes collection
|
||
|
|
// get the total attributes in the collection
|
||
|
|
// get as many IASATTRIBUTE structs
|
||
|
|
// fill in the IAS attribute structs with the appropriate
|
||
|
|
// values
|
||
|
|
//
|
||
|
|
dwCount = pCPacketRadius->GetInAttributeCount();
|
||
|
|
if (dwCount > 0)
|
||
|
|
{
|
||
|
|
//
|
||
|
|
// get the attributes collection now
|
||
|
|
//
|
||
|
|
pIasAttribPos = pCPacketRadius->GetInAttributes ();
|
||
|
|
|
||
|
|
//
|
||
|
|
// if the attribute count is greater than 0 there
|
||
|
|
// should always be attributes around
|
||
|
|
//
|
||
|
|
_ASSERT (pIasAttribPos);
|
||
|
|
|
||
|
|
//
|
||
|
|
// get IAttributesRaw interface
|
||
|
|
//
|
||
|
|
hr = pIRequest->QueryInterface (
|
||
|
|
__uuidof (IAttributesRaw),
|
||
|
|
reinterpret_cast <PVOID*> (&pIAttributesRaw)
|
||
|
|
);
|
||
|
|
if (FAILED (hr))
|
||
|
|
{
|
||
|
|
IASTracePrintf (
|
||
|
|
"Unable to obtain Attributes interface in request object "
|
||
|
|
"before sending request to backend"
|
||
|
|
);
|
||
|
|
__leave;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// put the attributes collection into the request
|
||
|
|
//
|
||
|
|
hr = pIAttributesRaw->AddAttributes (dwCount, pIasAttribPos);
|
||
|
|
if (FAILED (hr))
|
||
|
|
{
|
||
|
|
IASTracePrintf (
|
||
|
|
"Unable to add Attributes to request object "
|
||
|
|
"before sending request to backend"
|
||
|
|
);
|
||
|
|
__leave;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// set IRequestRaw interface properties
|
||
|
|
//
|
||
|
|
hr = SetRequestProperties (
|
||
|
|
pIRequest,
|
||
|
|
pCPacketRadius,
|
||
|
|
ePacketType
|
||
|
|
);
|
||
|
|
if (FAILED (hr)) { __leave; }
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// convert the VSA attributes into IAS format
|
||
|
|
//
|
||
|
|
hr = m_pCVSAFilter->radiusToIAS (pIAttributesRaw);
|
||
|
|
if (FAILED (hr))
|
||
|
|
{
|
||
|
|
IASTracePrintf (
|
||
|
|
"Unable to convert Radius VSAs to IAS attributes in "
|
||
|
|
"request object before sending it to backend"
|
||
|
|
);
|
||
|
|
__leave;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
//
|
||
|
|
// now the packet is ready for sending out
|
||
|
|
//
|
||
|
|
hr = pIRequestHandler->OnRequest (pIRequest);
|
||
|
|
if (FAILED (hr))
|
||
|
|
{
|
||
|
|
IASTracePrintf ("Unable to send request object to backend");
|
||
|
|
__leave;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// success
|
||
|
|
//
|
||
|
|
}
|
||
|
|
__finally
|
||
|
|
{
|
||
|
|
if (pIRequestHandler) {pIRequestHandler->Release ();}
|
||
|
|
if (pIAttributesRaw) {pIAttributesRaw->Release();}
|
||
|
|
if (pIRequest) {pIRequest->Release();}
|
||
|
|
}
|
||
|
|
|
||
|
|
return (hr);
|
||
|
|
|
||
|
|
} // end of CSendToPipe::Process method
|
||
|
|
|
||
|
|
//+++-------------------------------------------------------------
|
||
|
|
//
|
||
|
|
// Function: Init
|
||
|
|
//
|
||
|
|
// Synopsis: This is the CSendToPipe class public method which
|
||
|
|
// initializes the class object
|
||
|
|
//
|
||
|
|
// Arguments:
|
||
|
|
// [in] IRequestSource*
|
||
|
|
//
|
||
|
|
// Returns: BOOL - status
|
||
|
|
//
|
||
|
|
// History: MKarki Created 11/22/97
|
||
|
|
//
|
||
|
|
//----------------------------------------------------------------
|
||
|
|
BOOL CSendToPipe::Init (
|
||
|
|
IRequestSource *pIRequestSource,
|
||
|
|
VSAFilter *pCVSAFilter,
|
||
|
|
CReportEvent *pCReportEvent
|
||
|
|
)
|
||
|
|
{
|
||
|
|
BOOL bStatus = TRUE;
|
||
|
|
HRESULT hr = S_OK;
|
||
|
|
|
||
|
|
_ASSERT (pIRequestSource && pCReportEvent && pCVSAFilter);
|
||
|
|
|
||
|
|
m_pCReportEvent = pCReportEvent;
|
||
|
|
|
||
|
|
m_pCVSAFilter = pCVSAFilter;
|
||
|
|
|
||
|
|
//
|
||
|
|
// get the IClassFactory interface to be used to create
|
||
|
|
// the Request COM object
|
||
|
|
// TODO - replace CLSID with __uuidof
|
||
|
|
//
|
||
|
|
hr = ::CoGetClassObject (
|
||
|
|
CLSID_Request,
|
||
|
|
CLSCTX_INPROC_SERVER,
|
||
|
|
NULL,
|
||
|
|
IID_IClassFactory,
|
||
|
|
reinterpret_cast <PVOID*> (&m_pIClassFactory)
|
||
|
|
);
|
||
|
|
if (FAILED (hr))
|
||
|
|
{
|
||
|
|
IASTracePrintf (
|
||
|
|
"Unable to obtain Request object class factory"
|
||
|
|
);
|
||
|
|
bStatus = FALSE;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
m_pIRequestSource = pIRequestSource;
|
||
|
|
}
|
||
|
|
|
||
|
|
return (bStatus);
|
||
|
|
|
||
|
|
} // end of CSendToPipe::Init method
|
||
|
|
|
||
|
|
//+++-------------------------------------------------------------
|
||
|
|
//
|
||
|
|
// Function: StartProcessing
|
||
|
|
//
|
||
|
|
// Synopsis: This is the CSendToPipe class public method which
|
||
|
|
// gets object ready to send data to the PipeLine
|
||
|
|
//
|
||
|
|
// Arguments:
|
||
|
|
// [in] IRequestHandler*
|
||
|
|
//
|
||
|
|
// Returns: BOOL - status
|
||
|
|
//
|
||
|
|
// History: MKarki Created 11/22/97
|
||
|
|
//
|
||
|
|
//----------------------------------------------------------------
|
||
|
|
BOOL
|
||
|
|
CSendToPipe::StartProcessing (
|
||
|
|
IRequestHandler *pIRequestHandler
|
||
|
|
)
|
||
|
|
{
|
||
|
|
_ASSERT (pIRequestHandler);
|
||
|
|
|
||
|
|
//
|
||
|
|
// set the value of the handler
|
||
|
|
//
|
||
|
|
m_pIRequestHandler = pIRequestHandler;
|
||
|
|
|
||
|
|
return (TRUE);
|
||
|
|
|
||
|
|
} // end of CSendToPipe::StartProcessing method
|
||
|
|
|
||
|
|
//+++-------------------------------------------------------------
|
||
|
|
//
|
||
|
|
// Function: StopProcessing
|
||
|
|
//
|
||
|
|
// Synopsis: This is the CSendToPipe class public method which
|
||
|
|
// gets object to stop sending data to the PipeLine
|
||
|
|
//
|
||
|
|
// Arguments: none
|
||
|
|
//
|
||
|
|
// Returns: BOOL - status
|
||
|
|
//
|
||
|
|
// History: MKarki Created 11/22/97
|
||
|
|
//
|
||
|
|
//----------------------------------------------------------------
|
||
|
|
BOOL
|
||
|
|
CSendToPipe::StopProcessing (
|
||
|
|
VOID
|
||
|
|
)
|
||
|
|
{
|
||
|
|
//
|
||
|
|
// set the value of the handlers
|
||
|
|
//
|
||
|
|
m_pIRequestHandler = NULL;
|
||
|
|
|
||
|
|
return (TRUE);
|
||
|
|
|
||
|
|
} // end of CSendToPipe::StartProcessing method
|
||
|
|
|
||
|
|
//+++-------------------------------------------------------------
|
||
|
|
//
|
||
|
|
// Function: SetRequestProperties
|
||
|
|
//
|
||
|
|
// Synopsis: This is the CSendToPipe class public method which
|
||
|
|
// set the properties in the IRequestRaw object
|
||
|
|
//
|
||
|
|
// Arguments:
|
||
|
|
// [in] IRequesetRaw*
|
||
|
|
// [in] CPacketRadius*
|
||
|
|
// [in] PACKETTYPE
|
||
|
|
//
|
||
|
|
// Returns: HRESULT - status
|
||
|
|
//
|
||
|
|
// History: MKarki Created 11/22/97
|
||
|
|
//
|
||
|
|
//----------------------------------------------------------------
|
||
|
|
HRESULT
|
||
|
|
CSendToPipe::SetRequestProperties (
|
||
|
|
IRequest *pIRequest,
|
||
|
|
CPacketRadius *pCPacketRadius,
|
||
|
|
PACKETTYPE ePacketType
|
||
|
|
)
|
||
|
|
{
|
||
|
|
IRequestState *pIRequestState = NULL;
|
||
|
|
IASREQUEST eRequest;
|
||
|
|
IASRESPONSE eResponse;
|
||
|
|
HRESULT hr = S_OK;
|
||
|
|
|
||
|
|
_ASSERT (pIRequest && pCPacketRadius);
|
||
|
|
|
||
|
|
__try
|
||
|
|
{
|
||
|
|
//
|
||
|
|
// decide the Request and Response Type
|
||
|
|
//
|
||
|
|
switch (ePacketType)
|
||
|
|
{
|
||
|
|
case ACCESS_REQUEST:
|
||
|
|
eRequest = IAS_REQUEST_ACCESS_REQUEST;
|
||
|
|
eResponse = IAS_RESPONSE_ACCESS_ACCEPT;
|
||
|
|
break;
|
||
|
|
case ACCOUNTING_REQUEST:
|
||
|
|
eRequest = IAS_REQUEST_ACCOUNTING;
|
||
|
|
eResponse = IAS_RESPONSE_ACCOUNTING;
|
||
|
|
break;
|
||
|
|
case ACCESS_ACCEPT:
|
||
|
|
eRequest = IAS_REQUEST_PROXY_PACKET;
|
||
|
|
eResponse = IAS_RESPONSE_ACCESS_ACCEPT;
|
||
|
|
break;
|
||
|
|
case ACCOUNTING_RESPONSE:
|
||
|
|
eRequest = IAS_REQUEST_PROXY_PACKET;
|
||
|
|
eResponse = IAS_RESPONSE_ACCOUNTING;
|
||
|
|
break;
|
||
|
|
case ACCESS_CHALLENGE:
|
||
|
|
eRequest = IAS_REQUEST_PROXY_PACKET;
|
||
|
|
eResponse = IAS_RESPONSE_ACCESS_CHALLENGE;
|
||
|
|
break;
|
||
|
|
default:
|
||
|
|
//
|
||
|
|
// should never reach here
|
||
|
|
//
|
||
|
|
_ASSERT (0);
|
||
|
|
IASTracePrintf (
|
||
|
|
"Packet of unsupported type:%d, before sending request to "
|
||
|
|
"backend",
|
||
|
|
static_cast <DWORD> (ePacketType)
|
||
|
|
);
|
||
|
|
|
||
|
|
hr = E_FAIL;
|
||
|
|
__leave;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// set the request type now
|
||
|
|
//
|
||
|
|
hr = pIRequest->put_Request (eRequest);
|
||
|
|
if (FAILED (hr))
|
||
|
|
{
|
||
|
|
IASTracePrintf (
|
||
|
|
"Unable to set request type in request before sending "
|
||
|
|
"it to the backend"
|
||
|
|
);
|
||
|
|
__leave;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// set the protocol now
|
||
|
|
//
|
||
|
|
hr = pIRequest->put_Protocol (IAS_PROTOCOL_RADIUS);
|
||
|
|
if (FAILED (hr))
|
||
|
|
{
|
||
|
|
IASTracePrintf (
|
||
|
|
"Unable to set protocol type in request before sending "
|
||
|
|
"it to the backend"
|
||
|
|
);
|
||
|
|
__leave;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// Set source callback
|
||
|
|
//
|
||
|
|
hr = pIRequest->put_Source (m_pIRequestSource);
|
||
|
|
if (FAILED (hr))
|
||
|
|
{
|
||
|
|
IASTracePrintf (
|
||
|
|
"Unable to set request source type in request before sending "
|
||
|
|
"it to the backend"
|
||
|
|
);
|
||
|
|
__leave;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// get the request state interface to put in our state now
|
||
|
|
//
|
||
|
|
//
|
||
|
|
hr = pIRequest->QueryInterface (
|
||
|
|
__uuidof (IRequestState),
|
||
|
|
reinterpret_cast <PVOID*> (&pIRequestState)
|
||
|
|
);
|
||
|
|
if (FAILED (hr))
|
||
|
|
{
|
||
|
|
IASTracePrintf (
|
||
|
|
"Unable to get RequestState interface from request object "
|
||
|
|
"before sending it to the backend"
|
||
|
|
);
|
||
|
|
__leave;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// put in the request state now
|
||
|
|
//
|
||
|
|
hr = pIRequestState->Push (
|
||
|
|
reinterpret_cast <unsigned hyper> (pCPacketRadius)
|
||
|
|
);
|
||
|
|
if (FAILED (hr))
|
||
|
|
{
|
||
|
|
IASTracePrintf (
|
||
|
|
"Unable to set information in request state "
|
||
|
|
"before sending request to backend"
|
||
|
|
);
|
||
|
|
__leave;
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// success
|
||
|
|
//
|
||
|
|
}
|
||
|
|
__finally
|
||
|
|
{
|
||
|
|
if (pIRequestState) { pIRequestState->Release (); }
|
||
|
|
}
|
||
|
|
|
||
|
|
return (hr);
|
||
|
|
|
||
|
|
} // end of CSendToPipe::SetRequestProperties method
|
||
|
|
|