SourceURL.cpp

Go to the documentation of this file.

//
// SourceURL.cpp
//
// Copyright (c) Shareaza Development Team, 2002-2005.
// This file is part of SHAREAZA (www.shareaza.com)
//
// Shareaza is free software; you can redistribute it
// and/or modify it under the terms of the GNU General Public License
// as published by the Free Software Foundation; either version 2 of
// the License, or (at your option) any later version.
//
// Shareaza is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with Shareaza; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
//

#include "StdAfx.h"
#include "Shareaza.h"
#include "Settings.h"
#include "Network.h"
#include "SourceURL.h"
#include "SHA.h"
#include "ED2K.h"

#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif


// CSourceURL construction

CSourceURL::CSourceURL(LPCTSTR pszURL)
{
        Clear();
        if ( pszURL != NULL ) Parse( pszURL );
}

// CSourceURL clear

void CSourceURL::Clear()
{
        m_sURL.Empty();
        m_nProtocol = PROTOCOL_NULL;
        m_sAddress.Empty();
        m_pAddress.S_un.S_addr = 0;
        m_nPort = 0;
        m_pServerAddress.S_un.S_addr = 0;
        m_nServerPort = 0;
        m_sPath.Empty();
        m_bSHA1 = FALSE;
        m_bED2K = FALSE;
        m_bBTH  = FALSE;
        m_bBTC  = FALSE;
        m_bSize = FALSE;
        m_sLogin.Empty ();
        m_sPassword.Empty ();
}

// CSourceURL root parse

BOOL CSourceURL::Parse(LPCTSTR pszURL, BOOL bResolve)
{
        Clear();
        if ( ParseHTTP( pszURL, bResolve ) ) return TRUE;
        Clear();
        if ( ParseFTP( pszURL, bResolve ) ) return TRUE;
        Clear();
        if ( ParseED2KFTP( pszURL, bResolve ) ) return TRUE;
        Clear();
        if ( ParseBTC( pszURL, bResolve ) ) return TRUE;
        Clear();
        return FALSE;
}

// CSourceURL HTTP

BOOL CSourceURL::ParseHTTP(LPCTSTR pszURL, BOOL bResolve)
{
        if ( _tcsncmp( pszURL, _T("http://"), 7 ) != 0 ) return FALSE;
        
        CString strURL = pszURL + 7;
        
        int nSlash = strURL.Find( _T('/') );
        
        if ( nSlash >= 0 )
        {
                m_sAddress      = strURL.Left( nSlash );
                m_sPath         = strURL.Mid( nSlash );
        }
        else
        {
                m_sAddress = strURL;
                m_sPath = _T("/");
        }
        
        int nAt = m_sAddress.Find( _T('@') );
        if ( nAt >= 0 ) m_sAddress = m_sAddress.Mid( nAt + 1 );
        
        if ( m_sAddress.IsEmpty() ) return FALSE;
        
        if ( _tcsnicmp( m_sPath, _T("/uri-res/N2R?urn:sha1:"), 22 ) == 0 )
        {
                m_bSHA1 = CSHA::HashFromURN( m_sPath.Mid( 13 ), &m_pSHA1 );
        }
        else if ( _tcsnicmp( m_sPath, _T("/uri-res/N2R?urn:bitprint:"), 26 ) == 0 )
        {
                m_bSHA1 = CSHA::HashFromURN( m_sPath.Mid( 13 ), &m_pSHA1 );
        }
        
        SOCKADDR_IN saHost;
        
        BOOL bResult = Network.Resolve( m_sAddress, INTERNET_DEFAULT_HTTP_PORT, &saHost, bResolve );
        
        m_pAddress      = saHost.sin_addr;
        m_nPort         = htons( saHost.sin_port );
        
        m_sURL          = pszURL;
        m_nProtocol     = PROTOCOL_HTTP;
        
        return bResult;
}

// CSourceURL FTP

BOOL CSourceURL::ParseFTP(LPCTSTR pszURL, BOOL bResolve)
{
        // URI format
        // ftp://[user[:password]@]host[:port][/path]

        if ( _tcsncmp( pszURL, _T("ftp://"), 6 ) != 0 ) return FALSE;
        
        CString strURL ( pszURL + 6 );
        
        int nSlash = strURL.Find( _T('/') );
        
        if ( nSlash >= 0 )
        {
                m_sAddress      = strURL.Left( nSlash );
                m_sPath         = strURL.Mid( nSlash );
        }
        else
        {
                m_sAddress = strURL;
                m_sPath = _T("/");
        }
        
        int nAt = m_sAddress.Find( _T('@') );
        if ( nAt >= 0 )
        {
                m_sLogin = m_sAddress.Left( nAt );
                m_sAddress = m_sAddress.Mid( nAt + 1 );
        
                int nColon = m_sLogin.Find( _T(':') );
                if ( nColon >= 0 )
                {
            m_sPassword = m_sLogin.Mid( nColon + 1 );                   
            m_sLogin = m_sLogin.Left( nColon );                 
                }
        }
        else
        {
                m_sLogin = _T("anonymous");
                m_sPassword = _T("[email protected]");
        }

        if ( m_sAddress.IsEmpty() || m_sLogin.IsEmpty() )
                return FALSE;
        
        SOCKADDR_IN saHost;
        
        BOOL bResult = Network.Resolve( m_sAddress, INTERNET_DEFAULT_FTP_PORT, &saHost, bResolve );
        
        m_pAddress      = saHost.sin_addr;
        m_nPort         = htons( saHost.sin_port );
        
        m_sURL          = pszURL;
        m_nProtocol     = PROTOCOL_FTP;
        
        return bResult;
}

// CSourceURL ED2KFTP

BOOL CSourceURL::ParseED2KFTP(LPCTSTR pszURL, BOOL bResolve)
{
        if ( _tcsnicmp( pszURL, _T("ed2kftp://"), 10 ) != 0 ) return FALSE;
        
        CString strURL = pszURL + 10;
        BOOL bPush = FALSE;
        
        int nSlash = strURL.Find( _T('/') );
        if ( nSlash < 7 ) return FALSE;

        m_sAddress      = strURL.Left( nSlash );
        strURL          = strURL.Mid( nSlash + 1 );
        
        nSlash = strURL.Find( _T('/') );
        if ( nSlash != 32 ) return FALSE;
        
        CString strHash = strURL.Left( 32 );
        strURL                  = strURL.Mid( 33 );
        
        if ( ! CED2K::HashFromString( strHash, &m_pED2K ) ) return FALSE;
        
        m_bSize = _stscanf( strURL, _T("%I64i"), &m_nSize ) == 1;
        if ( ! m_bSize ) return FALSE;
        
        nSlash = m_sAddress.Find( _T('@') );
        
        if ( nSlash > 0 )
        {
                strHash = m_sAddress.Left( nSlash );
                m_sAddress = m_sAddress.Mid( nSlash + 1 );
                if ( _stscanf( strHash, _T("%lu"), &m_pAddress.S_un.S_addr ) != 1 ) return FALSE;
                bPush = TRUE;
        }
        
        SOCKADDR_IN saHost;
        BOOL bResult = Network.Resolve( m_sAddress, ED2K_DEFAULT_PORT, &saHost, bResolve );
        
        if ( bPush )
        {
                m_pServerAddress        = saHost.sin_addr;
                m_nServerPort           = htons( saHost.sin_port );
                m_nPort                         = 0;
        }
        else
        {
                m_pAddress      = saHost.sin_addr;
                m_nPort         = htons( saHost.sin_port );
        }
        
        m_sURL          = pszURL;
        m_nProtocol     = PROTOCOL_ED2K;
        
        return bResult;
}

// CSourceURL BTC

BOOL CSourceURL::ParseBTC(LPCTSTR pszURL, BOOL bResolve)
{
        if ( _tcsnicmp( pszURL, _T("btc://"), 6 ) != 0 ) return FALSE;
        
        CString strURL = pszURL + 6;
        BOOL bPush = FALSE;
        
        int nSlash = strURL.Find( _T('/') );
        if ( nSlash < 7 ) return FALSE;

        m_sAddress      = strURL.Left( nSlash );
        strURL          = strURL.Mid( nSlash + 1 );
        
        nSlash = strURL.Find( _T('/') );
        m_bBTC = FALSE;
        
        if ( nSlash == 32 )
        {
                CString strGUID = strURL.Left( 32 );
                m_bBTC = CSHA::HashFromString( strGUID, &m_pBTC );
        }
        else if ( nSlash < 0 ) return FALSE;
        
        strURL = strURL.Mid( nSlash + 1 );
        
        if ( ! CSHA::HashFromString( strURL, &m_pBTH ) ) return FALSE;
        
        SOCKADDR_IN saHost;
        BOOL bResult = Network.Resolve( m_sAddress, ED2K_DEFAULT_PORT, &saHost, bResolve );
        
        m_pAddress      = saHost.sin_addr;
        m_nPort         = htons( saHost.sin_port );
        
        m_sURL          = pszURL;
        m_nProtocol     = PROTOCOL_BT;
        
        return bResult;
}

---