--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/realtimenetprots/sipfw/SIP/ServerResolver/src/CSIPQueryNaptr.cpp	Tue Feb 02 01:03:15 2010 +0200
@@ -0,0 +1,361 @@
+// 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          : CSIPQueryNaptr.cpp
+// Part of       : ServerResolver
+// Version       : SIP/4.0 
+//
+
+
+
+#include <stringpool.h>
+#include "sipstrings.h"
+#include "sipstrconsts.h"
+#include "CSIPQueryNaptr.h"
+#include "MSIPHostResolver.h"
+
+_LIT8(Ks, "s");
+_LIT8(KServiceSIP, "SIP+*");
+_LIT8(KServiceSIPS, "SIPS+*");
+_LIT8(KStarPlusD2T, "*+D2T");
+_LIT8(KStarPlusD2TPlusStar, "*+D2T+*");
+_LIT8(KStarPlusD2U, "*+D2U");
+_LIT8(KStarPlusD2UPlusStar, "*+D2U+*");
+
+// ----------------------------------------------------------------------------
+// CSIPQueryNaptr::NewL
+// ----------------------------------------------------------------------------
+//
+CSIPQueryNaptr* CSIPQueryNaptr::NewL( const TDesC8& aTarget, TBool aSipsUri )
+	{
+	CSIPQueryNaptr* self = new ( ELeave ) CSIPQueryNaptr(aSipsUri);
+	CleanupStack::PushL( self );
+	self->ConstructL( aTarget );
+	CleanupStack::Pop( self );
+	return self;
+	}
+
+// ----------------------------------------------------------------------------
+// CSIPQueryNaptr::QueryBuf
+// ----------------------------------------------------------------------------
+//
+const TDesC8& CSIPQueryNaptr::QueryBuf()
+	{
+	TDnsQuery query( *iTarget, KDnsRRTypeNAPTR );
+	TDnsQueryBuf querybuf( query );
+	return iQueryBuf = querybuf;
+	}
+
+// ----------------------------------------------------------------------------
+// CSIPQueryNaptr::QueryResultBuf
+// ----------------------------------------------------------------------------
+//
+TDes8& CSIPQueryNaptr::QueryResultBuf()
+	{
+	return iResult; 
+	}
+
+// ----------------------------------------------------------------------------
+// CSIPQueryNaptr::QueryResult
+// ----------------------------------------------------------------------------
+//
+TDnsRespNAPTR CSIPQueryNaptr::QueryResult()
+	{
+	return iResult();
+	}
+
+// ----------------------------------------------------------------------------
+// CSIPQueryNaptr::~CSIPQueryNaptr
+// ----------------------------------------------------------------------------
+//
+CSIPQueryNaptr::~CSIPQueryNaptr()
+	{
+	delete iTarget;
+	if ( iRecordArray )
+		{
+		iRecordArray->ResetAndDestroy();
+		delete iRecordArray;
+		}
+	}
+
+// ----------------------------------------------------------------------------
+// CSIPQueryNaptr::AddL
+// ----------------------------------------------------------------------------
+//
+void CSIPQueryNaptr::AddL( TDnsRespNAPTR* aNaptr )
+	{
+	TLinearOrder<TDnsRespNAPTR> order( CSIPQueryNaptr::Compare );
+	User::LeaveIfError( iRecordArray->InsertInOrderAllowRepeats( aNaptr,
+																 order ) );
+	}
+
+// ----------------------------------------------------------------------------
+// CSIPQueryNaptr::Compare
+// ----------------------------------------------------------------------------
+//
+TInt CSIPQueryNaptr::Compare( const TDnsRespNAPTR& aFirst, 
+							  const TDnsRespNAPTR& aSecond )
+	{
+	if ( aFirst.Order() < aSecond.Order() )
+		{
+		return -1;
+		}
+	if ( aFirst.Order() > aSecond.Order() )
+		{
+		return 1;
+		}
+	if ( aFirst.Order() == aSecond.Order() )
+		{
+		if ( aFirst.Pref() == aSecond.Pref() )
+			{
+			if ( NaptrProtocol( aFirst ) ==
+				 SIPStrings::StringF( SipStrConsts::EUDP ) )
+				{
+				return -1;
+				}
+			else if ( NaptrProtocol( aSecond ) ==
+					  SIPStrings::StringF( SipStrConsts::EUDP ) )
+				{
+				return 1;
+				}
+				else
+					{
+					return 0;
+					}
+			}
+		if ( aFirst.Pref() < aSecond.Pref() )
+			{
+			return -1;
+			}
+		if ( aFirst.Pref() > aSecond.Pref() )
+			{
+			return 1;
+			}
+		}
+	return 0;
+	}
+
+// ----------------------------------------------------------------------------
+// CSIPQueryNaptr::NAPTRRecordL
+// ----------------------------------------------------------------------------
+//
+TDnsRespNAPTR& CSIPQueryNaptr::NAPTRRecordL( TInt aIndex )
+	{
+	__ASSERT_ALWAYS ( aIndex < iRecordArray->Count(), User::Leave( KErrArgument ) );
+	return ( *( *iRecordArray )[aIndex] );
+	}
+
+// ----------------------------------------------------------------------------
+// CSIPQueryNaptr::RemoveElementL
+// ----------------------------------------------------------------------------
+//
+void CSIPQueryNaptr::RemoveElementL( TInt aIndex )
+	{
+	if ( aIndex < iRecordArray->Count() && aIndex >= 0 )
+		{
+		TDnsRespNAPTR* naptr = ( *iRecordArray )[aIndex];
+		delete naptr;
+		naptr = NULL;
+		iRecordArray->Remove( aIndex );
+		}
+	}
+
+// ----------------------------------------------------------------------------
+// CSIPQueryNaptr::MatchToRulesL
+// ----------------------------------------------------------------------------
+//
+TBool CSIPQueryNaptr::MatchToRulesL( const TDnsRespNAPTR& aNaptr )
+	{	
+	if ( aNaptr.Flags().CompareF( Ks ) !=0 )
+		{
+		return EFalse;
+		}
+		
+	if ( aNaptr.Regexp().Length() != 0 )
+		{
+		return EFalse;
+		}
+		
+	HBufC8* temp = aNaptr.Replacement().AllocL();
+	TPtr8 ptr = temp->Des();
+	ptr.Trim();
+	if ( temp->Length() == 0 )
+		{
+		delete temp;
+		return EFalse;
+		}
+	delete temp;
+	
+	if ( aNaptr.Service().Match( KServiceSIPS ) != KErrNotFound )
+		{
+		if ( NaptrProtocol( aNaptr ) ==
+			 SIPStrings::StringF( SipStrConsts::ETLS ) )
+			{
+			return ETrue;
+			}
+		}
+	
+	if (!iSipsUri)
+		{
+		if ( aNaptr.Service().Match( KServiceSIP ) != KErrNotFound )
+			{
+			if ( NaptrProtocol( aNaptr ) !=
+				 SIPStrings::StringF( SipStrConsts::EEmpty ) )
+				{
+				return ETrue;
+				}
+			}
+		}
+	
+	return EFalse;
+	}
+
+// ----------------------------------------------------------------------------
+// CSIPQueryNaptr::RecordArray
+// ----------------------------------------------------------------------------
+//
+RPointerArray<TDnsRespNAPTR>& CSIPQueryNaptr::RecordArray()
+	{
+	return *iRecordArray;
+	}
+
+// ----------------------------------------------------------------------------
+// CSIPQueryNaptr::NaptrProtocol
+// ----------------------------------------------------------------------------
+//
+RStringF CSIPQueryNaptr::NaptrProtocol( const TDnsRespNAPTR& aNaptr )
+	{	
+	if ( aNaptr.Service().Match( KStarPlusD2T ) != KErrNotFound ||
+		aNaptr.Service().Match( KStarPlusD2TPlusStar ) != KErrNotFound )
+		{
+		if (aNaptr.Service().Match( KServiceSIPS ) != KErrNotFound)
+			{
+			return SIPStrings::StringF( SipStrConsts::ETLS );
+			}
+		return SIPStrings::StringF( SipStrConsts::ETCP );
+		}
+	
+	if( aNaptr.Service().Match( KStarPlusD2U ) != KErrNotFound ||
+		aNaptr.Service().Match( KStarPlusD2UPlusStar ) != KErrNotFound )
+		{
+		return SIPStrings::StringF( SipStrConsts::EUDP );
+		}
+	
+	return SIPStrings::StringF( SipStrConsts::EEmpty );
+	}
+
+
+// ----------------------------------------------------------------------------
+// CSIPQueryNaptr::Query
+// ----------------------------------------------------------------------------
+//
+void CSIPQueryNaptr::Query( MSIPHostResolver& aResolver )
+	{
+	aResolver.Resolver().Query( QueryBuf(),
+							    QueryResultBuf(),
+								aResolver.RequestStatus() );	
+	}
+
+// ----------------------------------------------------------------------------
+// CSIPQueryNaptr::ConstructL
+// ----------------------------------------------------------------------------
+//
+void CSIPQueryNaptr::ConstructL( const TDesC8& aTarget )
+	{
+	// KQueryDataMaxLength is same as TDnsQryData's max length in dns_qry.h
+	const TInt KQueryDataMaxLength = 255; 	
+ 
+	__ASSERT_ALWAYS ( aTarget.Length() < KQueryDataMaxLength, 
+					  User::Leave( KErrArgument ) );
+					  
+	__ASSERT_ALWAYS ( aTarget.Length() > 0, User::Leave( KErrArgument ) );
+
+
+	iTarget = aTarget.AllocL();
+	iRecordArray = new ( ELeave ) RPointerArray<TDnsRespNAPTR>; 
+	}
+
+// ----------------------------------------------------------------------------
+// CSIPQueryNaptr::CSIPQueryNaptr
+// ----------------------------------------------------------------------------
+//
+CSIPQueryNaptr::CSIPQueryNaptr(TBool aSipsUri)
+:CSIPQueryBase(), iSipsUri(aSipsUri)
+	{
+	}
+
+// ----------------------------------------------------------------------------
+// CSIPQueryNaptr::HandleQueryResultL
+// ----------------------------------------------------------------------------
+//
+void CSIPQueryNaptr::HandleQueryResultL( MSIPHostResolver& aResolver )
+	{
+	TDnsRespNAPTR* naptr = NULL;
+	if ( MatchToRulesL( QueryResult() ) )
+		{
+		naptr = new ( ELeave ) TDnsRespNAPTR();
+		CleanupStack::PushL( naptr );
+		naptr->SetFlags( iResult().Flags() );
+		naptr->SetOrder( iResult().Order() );
+		naptr->SetPref( iResult().Pref() );
+		naptr->SetRegexp( iResult().Regexp() );
+		naptr->SetReplacement( iResult().Replacement() );
+		naptr->SetService( iResult().Service() );
+		AddL( naptr );
+		CleanupStack::Pop( naptr );
+		}
+	while ( aResolver.Resolver().QueryGetNext( QueryResultBuf() ) 
+												== KErrNone )
+		{
+		if ( MatchToRulesL( QueryResult() ) )
+			{
+			naptr = new ( ELeave ) TDnsRespNAPTR();
+			CleanupStack::PushL( naptr );
+			naptr->SetFlags( iResult().Flags() );
+			naptr->SetOrder( iResult().Order() );
+			naptr->SetPref( iResult().Pref() );
+			naptr->SetRegexp( iResult().Regexp() );
+			naptr->SetReplacement( iResult().Replacement() );
+			naptr->SetService( iResult().Service() );
+			AddL( naptr );
+			CleanupStack::Pop( naptr );
+			}
+		}
+	}
+
+// ----------------------------------------------------------------------------
+// CSIPQueryNaptr::ResultTargetL
+// ----------------------------------------------------------------------------
+//
+const TDesC8& CSIPQueryNaptr::ResultTargetL()
+	{
+	return NAPTRRecordL( 0 ).Replacement();
+	}
+
+// ----------------------------------------------------------------------------
+// CSIPQueryNaptr::ResultProtocolL
+// ----------------------------------------------------------------------------
+//
+RStringF CSIPQueryNaptr::ResultProtocolL() 
+	{
+	return NaptrProtocol( NAPTRRecordL( 0 ) );
+	}
+	
+// ----------------------------------------------------------------------------
+// CSIPQueryNaptr::ArrayCount
+// ----------------------------------------------------------------------------
+//
+TInt CSIPQueryNaptr::ArrayCount() 
+	{
+	return iRecordArray->Count();
+	}