//***************************************************************************
//
//  Copyright (c) 2000-2001 Microsoft Corporation
//
//  OPEXECQY.CPP
//
//  alanbos  07-Nov-00   Created.
//
//  WMI Exec Query operation implementation.  
//
//***************************************************************************

#include "precomp.h"

HRESULT WMIExecQueryOperation::BeginRequest (
	CComPtr<IWbemServices> & pIWbemServices
)
{
	HRESULT hr = WBEM_E_FAILED;
	LONG lFlags = WBEM_RETURN_IMMEDIATELY | WBEM_FLAG_FORWARD_ONLY;

	if (m_bUseAmendedQualifiers)
		lFlags |= WBEM_FLAG_USE_AMENDED_QUALIFIERS;

	if (m_bDirectRead)
		lFlags |= WBEM_FLAG_DIRECT_READ;

	if (m_bEnsureLocatable)
		lFlags |= WBEM_FLAG_ENSURE_LOCATABLE;

	if (m_bPrototype)
		lFlags |= WBEM_FLAG_PROTOTYPE;

	// Call the enumerator
	if (SUCCEEDED(hr = pIWbemServices->ExecQuery (
						m_bsQueryLanguage,
						m_bsQuery,
						lFlags, 
						GetContext(), 
						&m_pIEnumWbemClassObject)))
	{
		/*
		 * Get the first object - we do this as we are operating in semi-sync and some
		 * errors are only flagged on Next, not on ExecQuery.
		 */
		ULONG lReturned = 0;
		hr = m_pIEnumWbemClassObject->Next (WBEM_INFINITE, 1, &m_pFirstIWbemClassObject, &lReturned);

		if ((S_OK == hr) && (1 == lReturned))
			m_bIsEmptyResultsSet = false;
	}

	return hr;
}

HRESULT WMIExecQueryOperation::ProcessRequest (void)
{
	HRESULT hr = S_OK;
	
	// Start by checking if we have an initial object
	if (m_pFirstIWbemClassObject)
	{
		hr = EncodeAndSendObject (m_pFirstIWbemClassObject);
		m_pFirstIWbemClassObject.Release ();
	}

	// Now send the remainder
	if (SUCCEEDED(hr))
	{
		CComPtr<IWbemClassObject> pIWbemClassObject;
		ULONG lReturned = 0;

		while ((S_OK == (hr = m_pIEnumWbemClassObject->Next (WBEM_INFINITE, 1, &pIWbemClassObject, &lReturned)))
				&& (1 == lReturned))
		{
			if (FAILED(hr = EncodeAndSendObject (pIWbemClassObject)))
				break;

			lReturned = 0;
		}
	}

	return hr;
}

bool WMIExecQueryOperation::ProcessElement( 
      const wchar_t __RPC_FAR *pwchNamespaceUri,
      int cchNamespaceUri,
      const wchar_t __RPC_FAR *pwchLocalName,
      int cchLocalName,
      const wchar_t __RPC_FAR *pwchRawName,
      int cchRawName,
      ISAXAttributes __RPC_FAR *pAttributes)
{
	bool result = false;

	if (0 == wcscmp(WMI_EXECQUERY_PARAMETER_QUERYLANGUAGE, pwchLocalName))
	{
		// following content will be the value of the object name
		SetParseState (QueryLanguage);
		result = true;
	}
	else if (0 == wcscmp(WMI_EXECQUERY_PARAMETER_QUERY, pwchLocalName))
	{
		// following content will be the value of UseAmendedQualifiers
		SetParseState (Query);
		result = true;
	}
	else if (0 == wcscmp(WMI_GETOBJECT_PARAMETER_USEAQ, pwchLocalName))
	{
		// following content will be the value of UseAmendedQualifiers
		SetParseState (UseAmendedQualifiers);
		result = true;
	}
	else if (0 == wcscmp(WMI_EXECQUERY_PARAMETER_DIRECTREAD, pwchLocalName))
	{
		// following content will be the value of DirectRead
		SetParseState (DirectRead);
		result = true;
	}
	else if (0 == wcscmp(WMI_EXECQUERY_PARAMETER_ENSURELOC, pwchLocalName))
	{
		// following content will be the value of DirectRead
		SetParseState (EnsureLocatable);
		result = true;
	}
	else if (0 == wcscmp(WMI_EXECQUERY_PARAMETER_PROTOTYPE, pwchLocalName))
	{
		// following content will be the value of DirectRead
		SetParseState (Prototype);
		result = true;
	}

	return result;
}

bool WMIExecQueryOperation::ProcessContent (
        const unsigned short * pwchChars,
        int cchChars )
{
	bool result = true;

	switch (GetParseState ())
	{
		case QueryLanguage:
			m_bsQueryLanguage = SysAllocStringLen (pwchChars, cchChars);
			break;

		case Query:
			m_bsQuery = SysAllocStringLen (pwchChars, cchChars);
			break;

		case UseAmendedQualifiers:
			{
				if (0 == wcsncmp (pwchChars, L"true", cchChars))
					m_bUseAmendedQualifiers = true;
				else if (0 == wcsncmp (pwchChars, L"false", cchChars))
					m_bUseAmendedQualifiers = false;
				else
					result = false;
			}
			break;

		case DirectRead:
			{
				if (0 == wcsncmp (pwchChars, L"true", cchChars))
					m_bDirectRead = true;
				else if (0 == wcsncmp (pwchChars, L"false", cchChars))
					m_bDirectRead = false;
				else
					result = false;
			}
			break;

		case EnsureLocatable:
			{
				if (0 == wcsncmp (pwchChars, L"true", cchChars))
					m_bEnsureLocatable = true;
				else if (0 == wcsncmp (pwchChars, L"false", cchChars))
					m_bEnsureLocatable = false;
				else
					result = false;
			}
			break;

		case Prototype:
			{
				if (0 == wcsncmp (pwchChars, L"true", cchChars))
					m_bPrototype = true;
				else if (0 == wcsncmp (pwchChars, L"false", cchChars))
					m_bPrototype = false;
				else
					result = false;
			}
			break;
	}

	return result;
}