D:/opendownloadmanager/ODM-1.x/fsMirrorURLsMgr.cpp

Go to the documentation of this file.

/*
  Free Download Manager Copyright (c) 2003-2007 FreeDownloadManager.ORG
  Open Download Manager Copyright (c) 2008-2010 OpenDownloadManager.ORG
*/        

#include "stdafx.h"
#include "fsMirrorURLsMgr.h"
#include "inetutil.h"

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

fsMirrorURLsMgr::fsMirrorURLsMgr()
{
        m_dldr.SetEventFunc (_DldrEvents, this);
}

fsMirrorURLsMgr::~fsMirrorURLsMgr()
{

}  

void fsMirrorURLsMgr::Initialize(LPCSTR pszFileName, UINT64 uSize, LPCSTR pszBaseServer, fsInternetSession* pSession)
{
        LOG ("initializing MirrorURLsMgr..." << nl);
        LOG ("file: " << pszFileName << nl);
        LOG ("size: " << uSize << nl);
        LOG ("base server: " << pszBaseServer << nl);
        LOG ("session: " << LONG (pSession) << nl);

        m_strFile = pszFileName;
        m_uSize = uSize;
        m_strBaseServer = pszBaseServer;
        m_dldr.Initialize (pSession);

        LOG ("ok." << nl);
}  

void fsMirrorURLsMgr::Set_SearchURL(LPCSTR pszUrl)
{
        m_strSearchURL = pszUrl;
        LOG ("search url was set: " << pszUrl << nl);
}

int fsMirrorURLsMgr::Get_MirrorURLCount()
{
        return m_vMirrorURLs.size ();
}

LPCSTR fsMirrorURLsMgr::Get_MirrorURL(int iIndex)
{
        return m_vMirrorURLs [iIndex];
}

fsInternetResult fsMirrorURLsMgr::SearchForMirrors()
{
        fsInternetResult ir;

        LOG ("starting searching for mirrors..." << nl);

        CString strURL = m_strSearchURL;

        strURL.Replace ("%file%", m_strFile);

        if (m_uSize != _UI64_MAX)
        {
                CString strSize;
                strSize.Format ("%I64u", m_uSize);
                strURL.Replace ("%size%", strSize);
        }
        else
                strURL.Replace ("%size%", "");

        LOG ("resulting url for script is: " << strURL << nl);

        m_bAbort = FALSE;

        LOG ("calling script on server..." << nl);

        ir = m_dldr.Download (strURL);
        if (ir != IR_SUCCESS)
        {
                LOG ("error occured while retreiving search results: " << LONG (ir)
                         << "; GetLastError is: " << GetLastError () << nl);
                return ir;
        }

        LOG ("script called successfully" << nl);

        return OnSearchScriptResultsReceived ();
}

void fsMirrorURLsMgr::_DldrEvents(fsInternetURLFileDownloaderEvent ev, LPVOID lp)
{
        fsMirrorURLsMgr* pThis = (fsMirrorURLsMgr*) lp;

        switch (ev)
        {
                case UFDE_CONNECTING:
                        LOG ("connecting to search server..." << nl);
                        pThis->Event (MUME_CONNECTINGSEARCHSERVER);
                        break;

                case UFDE_DOWNLOADING:
                        LOG ("downloading search results..." << nl);
                        pThis->Event (MUME_RETREIVINGSEARCHRESULTS);
                        break;
        }
}

void fsMirrorURLsMgr::Set_EventFunc(fntMirrorURLsMgrEvents pfn, LPVOID lp)
{
        m_pfnEvents = pfn;
        m_lpEvParam = lp;
}

void fsMirrorURLsMgr::Event(fsMirrorURLsMgrEvent ev)
{
        if (m_pfnEvents)
                m_pfnEvents (ev, m_lpEvParam);
}

fsInternetResult fsMirrorURLsMgr::OnSearchScriptResultsReceived()
{
        fsHTMLParser parser;

        LOG ("processing search results..." << nl);

        parser.SetKillDupes (TRUE);

        LOG ("parsing HTML...");

        parser.ParseHTML (LPSTR (m_dldr.Get_FileBuffer ()));

        LOG ("ok." << nl);

        LOG ("initializing list of mirrors...");

        m_vMirrorURLs.clear ();

        LOG ("ok." << nl);
        LOG ("adding mirrors..." << nl);

        for (int i = 0; i < parser.GetUrlCount () && m_bAbort == FALSE; i++)
        {
                fsURL url;
                LPCSTR pszUrl = parser.GetUrl (i);

                if (IR_SUCCESS != url.Crack (pszUrl))
                        continue;

                if (fsIsServersEqual (url.GetHostName (), m_strBaseServer, TRUE) ||
                        fsIsServersEqual (m_strBaseServer, url.GetHostName (), TRUE)    )
                        continue;

                char szFileName [10000];
                fsFileNameFromUrlPath (url.GetPath (), url.GetInternetScheme () == INTERNET_SCHEME_FTP,
                        TRUE, szFileName, sizeof (szFileName));

                if (stricmp (szFileName, m_strFile) == 0)
                {
                        if (IsMirrorURLGood (pszUrl) == FALSE)
                                continue;

                        LOG ("adding mirror: " << pszUrl << nl);

                        m_vMirrorURLs.add (pszUrl);
                }
        }

        LOG ("done adding mirrors. freing resources...");

        m_dldr.Free_FileBuffer ();

        LOG ("ok." << nl);

        if (m_bAbort)
                return IR_S_FALSE;

        Event (MUME_DONE);
        return IR_SUCCESS;
}

BOOL fsMirrorURLsMgr::IsMirrorURLGood(LPCSTR )
{
        return TRUE;
}

void fsMirrorURLsMgr::Abort()
{
        m_bAbort = TRUE;
        m_dldr.Abort ();
}                

fsMirrorURLsMgr_FileMirrorsDotCom::fsMirrorURLsMgr_FileMirrorsDotCom()
{
        Set_SearchURL (0);
}

fsMirrorURLsMgr_FileMirrorsDotCom::~fsMirrorURLsMgr_FileMirrorsDotCom()
{

}

void fsMirrorURLsMgr_FileMirrorsDotCom::Set_SearchURL(int nURL)
{
        static LPCSTR _ppszURLs [] = {
                "http://www.filesearching.com/cgi-bin/s?q=%file%&w=a&t=f&e=on&m=20&o=n&s1=%size%&s2=%size%&d=&p=&p2=&x=28&y=14",
                "http://findfiles.com/list.php?string=%file%&db=Mirrors&size=%size%",
        };

        if (nURL < sizeof (_ppszURLs) / sizeof (LPCSTR))
                fsMirrorURLsMgr::Set_SearchURL (_ppszURLs [nURL]);
}

---