// Copyright (c) 2004-2009 Nokia Corporation and/or its subsidiary(-ies).
// All rights reserved.
// This component and the accompanying materials are made available
// under the terms of "Eclipse Public License v1.0"
// which accompanies this distribution, and is available
// at the URL "http://www.eclipse.org/legal/epl-v10.html".
//
// Initial Contributors:
// Nokia Corporation - initial contribution.
//
// Contributors:
//
// Description:
// Name          : CSIPNaptrOrigDomain.cpp
// Part of       : ServerResolver
// Version       : SIP/4.0 
//



#include "sipstrings.h"
#include "sipstrconsts.h"
#include "CSIPNaptrOrigDomain.h"
#include "CSIPQueryBase.h"
#include "CSIPQueryNaptr.h"
#include "CSIPQuerySrv.h"
#include "CSIPSRVOrigDomainUDP.h"
#include "CSIPSRVNaptrDomain.h"
#include "MSIPServerQuery.h"
#include "CSIPQueryData.h"

// ----------------------------------------------------------------------------
// CSIPNaptrOrigDomain::CSIPNaptrOrigDomain
// ----------------------------------------------------------------------------
//
CSIPNaptrOrigDomain::CSIPNaptrOrigDomain( MSIPServerQuery& aServerQuery )
:CSIPQueryConditionBase(), iServerQuery ( aServerQuery )
	{
	}

// ----------------------------------------------------------------------------
// CSIPNaptrOrigDomain::ConstructL
// ----------------------------------------------------------------------------
//
void CSIPNaptrOrigDomain::ConstructL()
	{
	iQueryBase = 
		     CSIPQueryNaptr::NewL( iServerQuery.QueryData().OriginalTarget(),
		     					   iServerQuery.QueryData().IsSipsUri() );
	}

// ----------------------------------------------------------------------------
// CSIPNaptrOrigDomain::NewL
// ----------------------------------------------------------------------------
//
CSIPNaptrOrigDomain* CSIPNaptrOrigDomain::NewL( MSIPServerQuery& aServerQuery )
	{
	CSIPNaptrOrigDomain* self =
						   new ( ELeave ) CSIPNaptrOrigDomain( aServerQuery );
	CleanupStack::PushL( self );
	self->ConstructL();
	CleanupStack::Pop( self );
	return self;
	}

// ----------------------------------------------------------------------------
// CSIPNaptrOrigDomain::~CSIPNaptrOrigDomain
// ----------------------------------------------------------------------------
//
CSIPNaptrOrigDomain::~CSIPNaptrOrigDomain()
	{
	delete iQueryBase;
	}

// ----------------------------------------------------------------------------
// CSIPNaptrOrigDomain::QueryBase
// ----------------------------------------------------------------------------
//	
CSIPQueryBase& CSIPNaptrOrigDomain::QueryBase() 
	{
	return *iQueryBase;
	}

// ----------------------------------------------------------------------------
// CSIPNaptrOrigDomain::HandleQueryResultL
// ----------------------------------------------------------------------------
//
CSIPQueryConditionBase* CSIPNaptrOrigDomain::HandleQueryResultL( TInt aStatus ) 
	{
	if ( aStatus == KErrNone )
		{		
		CheckIfOnlyUDPResultsL();

		iQueryBase->HandleQueryResultL( iServerQuery.QueryData().Resolver() );
		if ( iQueryBase->ArrayCount() > 0 )
			{
			iServerQuery.QueryData().SetProtocol(
											iQueryBase->ResultProtocolL() );
			iServerQuery.QueryData().SetTargetL( iQueryBase->ResultTargetL() );
			CSIPQueryConditionBase* conditionbase = 
					     CSIPSRVNaptrDomain::NewL( iServerQuery, iQueryBase );
			iQueryBase = NULL;
			return conditionbase;
			}
		}
	iServerQuery.QueryData().SetPort( 0 );
	if (iServerQuery.QueryData().IsSipsUri())
		{
		iServerQuery.QueryData().SetProtocol(SIPStrings::StringF(
											SipStrConsts::ETLS) );
		}
	else
		{
		iServerQuery.QueryData().SetProtocol(SIPStrings::StringF(
											SipStrConsts::EUDP) );
		}
	
	HBufC8* temp = CSIPQuerySrv::SetTargetProtocolL(
									iServerQuery.QueryData().OriginalTarget(),
					 				iServerQuery.QueryData().Protocol(),
					 				iServerQuery.QueryData().IsSipsUri());
	CleanupStack::PushL( temp );
	iServerQuery.QueryData().SetTargetL( *temp );
	CleanupStack::PopAndDestroy( temp );
	return ( CSIPSRVOrigDomainUDP::NewL( iServerQuery ) );	
	}

// ----------------------------------------------------------------------------
// CSIPNaptrOrigDomain::CheckIfOnlyUDPResultsL
// If the result array has only records where protocol is UDP, the "force UDP"
// flag is set to guarantee the protocol won't be changed by later stages in
// the SIP stack.
// Since the "force UDP" can be set only if CSIPNaptrOrigDomain query returned
// at least one result, the further resolving phases do not change the
// protocol.
// ----------------------------------------------------------------------------
//
void CSIPNaptrOrigDomain::CheckIfOnlyUDPResultsL() const
	{
	__ASSERT_ALWAYS( iQueryBase, User::Leave( KErrNotFound ) );

	//CSIPNaptrOrigDomain::iQueryBase is always CSIPQueryNaptr.
	CSIPQueryNaptr& queryNaptr = static_cast<CSIPQueryNaptr&>( *iQueryBase );
	TBool udpOnly = ( queryNaptr.ArrayCount() > 0 );

	for ( TInt i = 0; i < queryNaptr.ArrayCount() && udpOnly; ++i )
		{
		TDnsRespNAPTR& dnsResp = queryNaptr.NAPTRRecordL( i );
		udpOnly = ( queryNaptr.NaptrProtocol( dnsResp ) ==
			 		SIPStrings::StringF( SipStrConsts::EUDP ) );
		}

	//There is only one query data instance for a single GetByUriL request
	if ( udpOnly )
		{
		iServerQuery.QueryData().SetForceUDP();
		}
	}