diff -r 000000000000 -r 307788aac0a8 realtimenetprots/sipfw/SIP/ServerResolver/src/CSIPNaptrOrigDomain.cpp
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/realtimenetprots/sipfw/SIP/ServerResolver/src/CSIPNaptrOrigDomain.cpp	Tue Feb 02 01:03:15 2010 +0200
@@ -0,0 +1,158 @@
+// 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();
+		}
+	}