///////////////////////////////////////////////////////////////////////////
//
// Microsoft WMIOLE DB Provider
//
// (C) Copyright 1999 Microsoft Corporation. All Rights Reserved.
//
//  IBindResource.CPP CImplIBindResource interface implementation
//
///////////////////////////////////////////////////////////////////////////

#include "headers.h"

////////////////////////////////////////////////////////////////////////////////
// Method of the IBindResource which binds the requested URL
// Returns one of the following values:
//			S_OK						Bind succeeded
//			DB_S_ERRORSOCCURRED			Bind succeeded, but some bind flags 
//										or properties were not satisfied
//			DB_E_NOAGGREGATION			Aggregation not supported by the 
//										object requested
//			DB_E_NOTFOUND				Object requested as from URL not found
//			DB_E_OBJECTMISMATCH			The object requested and the URL passed
//										does not match
//			DB_SEC_E_PERMISSIONDENIED	User does not have permission for the
//										object requested
//			E_FAIL						Other error ( WMI specifice errors)
//			E_INVALIDARG				one or more arguments are not valid
//			E_NOINTERFACE				The interface requested is not supported
//			E_UNEXPECTED				unexpected error
////////////////////////////////////////////////////////////////////////////////
STDMETHODIMP CImplIBindResource::Bind( IUnknown *            pUnkOuter,
									   LPCOLESTR             pwszURL,
									   DBBINDURLFLAG         dwBindURLFlags,
									   REFGUID               rguid,
									   REFIID                riid,
									   IAuthenticate *       pAuthenticate,
									   DBIMPLICITSESSION *   pImplSession,
									   DBBINDURLSTATUS *     pdwBindStatus,
									   IUnknown **           ppUnk)
{ 
	HRESULT hr = E_FAIL;
	BSTR strUrl;

    CSetStructuredExceptionHandler seh;

	TRY_BLOCK;

	// Serialize the object
	CAutoBlock	cab(BINDER->GetCriticalSection());
	g_pCError->ClearErrorInfo();

	// If URL is NULL return Invalid Argument
	if(pwszURL == NULL)
	{
		hr = E_INVALIDARG;
	}
	else
	{
		// Allocate the string
		strUrl = Wmioledb_SysAllocString(pwszURL);

		// Check if the flags passed are valid for the object requested
		if(!CheckBindURLFlags(dwBindURLFlags,rguid))
		{
			hr = E_INVALIDARG;
		}
		else
		if(!CheckIfProperURL(strUrl,rguid,pdwBindStatus))
		{
			hr = DB_E_OBJECTMISMATCH;
		}
		else
		if( pUnkOuter != NULL && riid != IID_IUnknown)
		{
			hr = DB_E_NOAGGREGATION;
		}
		else
		{
			// Calling this to bind the URL to the appropriate object
			hr = BindURL(pUnkOuter,strUrl,dwBindURLFlags,rguid,riid,pImplSession,pdwBindStatus,ppUnk);
		}

		// Free the string
		SysFreeString(strUrl);
	}

	hr = hr == S_OK ? hr :g_pCError->PostHResult(hr,&IID_IBindResource);

	CATCH_BLOCK_HRESULT(hr,L"IBindResource::Bind");
	return hr;
}


////////////////////////////////////////////////////////////////////////////////////////////
// Function which checks if the URL flags matches the requested object
//  This is as per the OLEDB specs
///////////////////////////////////////////////////////////////////////////////////////////
BOOL CImplIBindResource::CheckBindURLFlags(DBBINDURLFLAG dwBindURLFlags , REFGUID rguid)
{
	BOOL bFlag = FALSE;
	LONG lTemp = 0;

	if( DBGUID_DSO == rguid)
	{
		lTemp = DBBINDURLFLAG_ASYNCHRONOUS | DBBINDURLFLAG_READ | DBBINDURLFLAG_WAITFORINIT;
		// Flags can have only these values
		if((dwBindURLFlags & ~lTemp) == 0)
			bFlag = TRUE;

	}

	if( DBGUID_SESSION == rguid)
	{
		if( dwBindURLFlags == DBBINDURLFLAG_READ) // Flags can have only these values
			bFlag = TRUE;
	}

	if( DBGUID_COMMAND == rguid)
	{
		lTemp = DBBINDURLFLAG_READ | DBBINDURLFLAG_WAITFORINIT;
		if((dwBindURLFlags & ~lTemp) == 0)
			bFlag = TRUE;
	}

	if( DBGUID_ROW == rguid)
	{
		bFlag = TRUE;
	}

	if( DBGUID_ROWSET == rguid)
	{
		if(!((dwBindURLFlags & DBBINDURLFLAG_DELAYFETCHCOLUMNS) ||			// Flags cannot have any of these two values
			(dwBindURLFlags & DBBINDURLFLAG_DELAYFETCHSTREAM)))
			bFlag = TRUE;

	}

	if( DBGUID_STREAM == rguid)
	{
	}


	return bFlag;
}


////////////////////////////////////////////////////////////////////////////////////////////
// Function which checks if the URL is valid for the requested object
///////////////////////////////////////////////////////////////////////////////////////////
BOOL CImplIBindResource::CheckIfProperURL(BSTR & strUrl,REFGUID rguid,DBBINDURLSTATUS * pdwBindStatus)
{
	BOOL bRet = FALSE;
	LONG lUrlType = -1;
	
	m_pObj->m_pUrlParser->ClearParser();
	//==================================================================
	// Set the URL string of the URL parser utility class
	//==================================================================
	if(SUCCEEDED(m_pObj->m_pUrlParser->SetURL(strUrl)))
	{
		bRet = TRUE;
		//==================================================================
		// If the url is a valid URL
		//==================================================================
		if((lUrlType = m_pObj->m_pUrlParser->GetURLType()) != -1)
		{
			switch(lUrlType)
			{
				case URL_ROW:
				case URL_EMBEDEDCLASS:
					bRet = TRUE;

					//==================================================================
					// if the url is of type row or rowset and if the requested object is
					// datasource or session or command , then set the status 
					//==================================================================
					if(rguid == DBGUID_DSO || rguid == DBGUID_SESSION || rguid == DBGUID_COMMAND)
						*pdwBindStatus = DBBINDURLSTATUS_S_REDIRECTED;					

					break;

				case URL_DATASOURCE:
					//==================================================================
					// check if the url is of type datasource and the requested object is
					// row or rowset or stream 
					//==================================================================
					if(!(rguid == DBGUID_ROW || rguid == DBGUID_ROWSET || rguid == DBGUID_STREAM))
						bRet = TRUE;

					break;

				case URL_ROWSET:

					//==================================================================
					// if the url is of type rowset and if the requested object is
					// datasource or session or command , then set the status 
					//==================================================================
					if(rguid == DBGUID_DSO || rguid == DBGUID_SESSION || rguid == DBGUID_COMMAND)
						*pdwBindStatus = DBBINDURLSTATUS_S_REDIRECTED;
	
					bRet = TRUE;
					
					if(rguid == DBGUID_ROW)
						bRet = FALSE;

					break;
			};
		}
	}

	return bRet;
	
}



///////////////////////////////////////////////////////////////////////////
// Function to bind the requested URL
///////////////////////////////////////////////////////////////////////////
HRESULT CImplIBindResource::BindURL(IUnknown *            pUnkOuter,
									LPCOLESTR             pwszURL,
									DBBINDURLFLAG         dwBindURLFlags,
									REFGUID               rguid,
									REFIID                riid,
									DBIMPLICITSESSION *   pImplSession,
									DBBINDURLSTATUS *     pdwBindStatus,
									IUnknown **           ppUnk)
{

	HRESULT hr = E_FAIL;
	IUnknown *pTempUnk = NULL;
	LONG lInitFlags  = 0;
	LONG lBindFlags	 = 0;
	REFGUID guidTemp = GUID_NULL;
	IUnknown* pReqestedPtr = NULL;



	GetInitAndBindFlagsFromBindFlags(dwBindURLFlags,lInitFlags,lBindFlags);
	
	
	//=========================================================================================
	// If the requested object is DSO then create the object aggregating with the outer object
	//=========================================================================================
	if( rguid == DBGUID_DSO)
	{
		if(DBBINDURLFLAG_WAITFORINIT & dwBindURLFlags)
			lInitFlags |= DBBINDURLFLAG_WAITFORINIT;

		hr = m_pObj->CreateDSO(pUnkOuter,lInitFlags,riid,&pReqestedPtr);

	}
	else
	{
		hr = m_pObj->CreateDSO(NULL,lInitFlags,guidTemp,NULL);
	}

	if(SUCCEEDED(hr))
	{

		if(rguid != DBGUID_DSO)
		{
			//=========================================================================================
			// If the requested object is DSO then create the object aggregating with the outer object
			//=========================================================================================
			pReqestedPtr = NULL;
			pTempUnk	 = NULL;
			if( rguid == DBGUID_SESSION)
			{
				hr = m_pObj->CreateSession(pUnkOuter,riid,&pReqestedPtr);
			}
			else
			{
				if(pImplSession != NULL)
				{
					pTempUnk = pImplSession->pUnkOuter;
					
					// if the session is to be created with aggregation 
					// if the requested pointer is on IUnknown then throw error
					if( pTempUnk != NULL && *pImplSession->piid != IID_IUnknown)
					{
						hr = DB_E_NOAGGREGATION;
					}
				}
				if(SUCCEEDED(hr))
				{
					hr = m_pObj->CreateSession(pTempUnk,guidTemp,&pReqestedPtr);
				}
			}
		}


		if( SUCCEEDED(hr))
		{

			//=========================================================================================
			// If requested object is command then call function to 
			// to create a command
			//=========================================================================================
			if(rguid == DBGUID_COMMAND)
			{
				pReqestedPtr = NULL;
				hr = m_pObj->CreateCommand(pUnkOuter,riid,&pReqestedPtr);
			}

			//=========================================================================================
			// If requested object is row then call function to 
			// to create a row
			//=========================================================================================
			if( rguid == DBGUID_ROW)
			{
				pReqestedPtr = NULL;
				hr = m_pObj->CreateRow(pUnkOuter,riid,&pReqestedPtr);
			}

			//=========================================================================================
			// If requested object is rowset then call function to 
			// to create a rowset
			//=========================================================================================
			if( rguid == DBGUID_ROWSET)
			{
				pReqestedPtr = NULL;
				hr = m_pObj->CreateRowset(pUnkOuter,riid,&pReqestedPtr);
			}

		}
		
	}

	//=========================================================================================
	// If failed due to some reason then release all the
	// object created for binding
	//=========================================================================================
	if( hr != S_OK)
	{
		m_pObj->ReleaseAllObjects();
	}

	if( SUCCEEDED(hr))
	{
		*ppUnk = pReqestedPtr;
	}

	return hr ;
}