C / C++ / MFC

BOOL CWebGrab::GetFile ( LPCTSTR szURL, CString& strBuffer, LPCTSTR szAgentName /*=NULL*/, CWnd* pWnd /*=NULL*/ )<br />
{<br />
  //  TRACE1("URL is %s\n", szURL);<br />
	m_rawHeaders ="";<br />
	m_infoStatusCode=0;<br />
	strBuffer.Empty();<br />
<br />
    if (!m_pSession && !Initialise(szAgentName, pWnd))<br />
        return FALSE;<br />
<br />
    if (pWnd)<br />
        m_pSession->SetStatusWnd(pWnd);<br />
<br />
    //m_pSession->SetStatus("Downloading file...");<br />
<br />
    DWORD dwCount = 0;<br />
    CHttpFile* pFile = NULL;<br />
    try<br />
    {<br />
        pFile = (CHttpFile*) m_pSession->OpenURL(szURL, 1,<br />
                                                 INTERNET_FLAG_TRANSFER_BINARY | INTERNET_FLAG_DONT_CACHE<br />
                                                 //| INTERNET_OPEN_FLAG_USE_EXISTING_CONNECT |<br />
                                                 //| INTERNET_FLAG_RELOAD<br />
                                                 );<br />
    <br />
		<br />
	}<br />
    catch (CInternetException* e)<br />
    {<br />
        TCHAR   szCause[255];<br />
        e->GetErrorMessage(szCause, 255);<br />
        m_pSession->SetStatus(szCause);<br />
        // e->ReportError();<br />
        e->Delete();<br />
        delete pFile;<br />
        pFile = NULL;<br />
        return FALSE;<br />
    }<br />
    <br />
    COleDateTime startTime = COleDateTime::GetCurrentTime();<br />
    LPSTR pBuf = NULL;<br />
    if (pFile)<br />
    {<br />
        pBuf = (LPSTR) ::GlobalAlloc(GMEM_FIXED, BUFFER_SIZE+1);<br />
        if (!pBuf)<br />
        {<br />
            pFile->Close();<br />
            delete pFile;<br />
            return FALSE;<br />
        }<br />
<br />
        BYTE buffer[BUFFER_SIZE+1];<br />
        try {<br />
            UINT nRead = 0;<br />
            DWORD size;<br />
            dwCount = 0;<br />
            do<br />
            {<br />
                nRead = pFile->Read(buffer, BUFFER_SIZE); - LOCKS HERE waiting for the rest of the bytes.<br />
<br />
                if (nRead > 0)<br />
                {<br />
                    buffer[nRead] = 0;<br />
<br />
                    LPTSTR ptr = strBuffer.GetBufferSetLength(dwCount + nRead + 1);<br />
                    memcpy(ptr+dwCount, buffer, nRead);<br />
<br />
                    dwCount += nRead;<br />
                    strBuffer.ReleaseBuffer(dwCount+1);<br />
<br />
                    COleDateTimeSpan elapsed = COleDateTime::GetCurrentTime() - startTime;<br />
                    double dSecs = elapsed.GetTotalSeconds();<br />
                    if (dSecs > 0.0)<br />
					{<br />
                        m_transferRate = (double)dwCount / 1024.0 / dSecs;<br />
						m_pSession->SetStatus("Read %d bytes (%0.1f Kb/s)", <br />
                                             dwCount, m_transferRate );<br />
						<br />
					}<br />
					else<br />
					{<br />
                        m_pSession->SetStatus("Read %d bytes", dwCount);<br />
						m_transferRate = dwCount;<br />
					}<br />
                }<br />
            }<br />
            while (nRead > 0);<br />
        }<br />
        catch (CFileException *e)<br />
        {<br />
            TCHAR   szCause[255];<br />
            e->GetErrorMessage(szCause, 255);<br />
			m_ErrorMessage = szCause;<br />
            m_pSession->SetStatus(szCause);<br />
            //e->ReportError();<br />
            e->Delete();<br />
            delete pFile;<br />
            ::GlobalFree(pBuf);	// mem leak fix by Niek Albers <br />
            return FALSE;<br />
        }<br />
        pFile->QueryInfoStatusCode(m_infoStatusCode);       <br />
		pFile->QueryInfo(HTTP_QUERY_RAW_HEADERS ,m_rawHeaders);<br />
        pFile->Close();<br />
	  ::GlobalFree(pBuf);	// mem leak fix by Niek Albers <br />
        delete pFile;<br />
    }<br />
<br />
    m_pSession->SetStatus("");<br />
<br />
    return TRUE;<br />
}