211 lines
5.3 KiB
C++

/*
** Command & Conquer Generals Zero Hour(tm)
** Copyright 2025 Electronic Arts Inc.
**
** This program 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 3 of the License, or
** (at your option) any later version.
**
** This program 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 this program. If not, see <http://www.gnu.org/licenses/>.
*/
// FILE: DownloadManager.cpp //////////////////////////////////////////////////////
// Generals download manager code
// Author: Matthew D. Campbell, July 2002
#include "debug.h"
#include "chatapi.h"
#include "DownloadManager.h"
#include "resource.h"
DownloadManager *TheDownloadManager = NULL;
DownloadManager::DownloadManager()
{
m_download = new CDownload(this);
m_wasError = m_sawEnd = false;
m_statusString = Fetch_String(FTP_StatusIdle);
// ----- Initialize Winsock -----
m_winsockInit = true;
WORD verReq = MAKEWORD(2, 2);
WSADATA wsadata;
int err = WSAStartup(verReq, &wsadata);
if (err != 0)
{
m_winsockInit = false;
}
else
{
if ((LOBYTE(wsadata.wVersion) != 2) || (HIBYTE(wsadata.wVersion) !=2))
{
WSACleanup();
m_winsockInit = false;
}
}
}
DownloadManager::~DownloadManager()
{
delete m_download;
if (m_winsockInit)
{
WSACleanup();
m_winsockInit = false;
}
}
void DownloadManager::init( void )
{
}
void DownloadManager::reset( void )
{
}
HRESULT DownloadManager::update( void )
{
return m_download->PumpMessages();
}
HRESULT DownloadManager::downloadFile( std::string server, std::string username, std::string password, std::string file, std::string localfile, std::string regkey, bool tryResume )
{
return m_download->DownloadFile( server.c_str(), username.c_str(), password.c_str(), file.c_str(), localfile.c_str(), regkey.c_str(), tryResume );
}
void DownloadManager::queueFileForDownload( std::string server, std::string username, std::string password, std::string file, std::string localfile, std::string regkey, bool tryResume )
{
QueuedDownload q;
q.file = file;
q.localFile = localfile;
q.password = password;
q.regKey = regkey;
q.server = server;
q.tryResume = tryResume;
q.userName = username;
m_queuedDownloads.push_back(q);
}
HRESULT DownloadManager::downloadNextQueuedFile( void )
{
QueuedDownload q;
std::list<QueuedDownload>::iterator it = m_queuedDownloads.begin();
if (it != m_queuedDownloads.end())
{
q = *it;
m_queuedDownloads.pop_front();
m_wasError = m_sawEnd = false;
return downloadFile( q.server, q.userName, q.password, q.file, q.localFile, q.regKey, q.tryResume );
}
else
{
DEBUG_CRASH(("Starting non-existent download!"));
return S_OK;
}
}
std::string DownloadManager::getLastLocalFile( void )
{
char buf[256] = "";
m_download->GetLastLocalFile(buf, 256);
return buf;
}
HRESULT DownloadManager::OnError( int error )
{
m_wasError = true;
std::string s = Fetch_String(FTP_UnknownError);
switch (error)
{
case DOWNLOADEVENT_NOSUCHSERVER:
s = Fetch_String(FTP_NoSuchServer);
break;
case DOWNLOADEVENT_COULDNOTCONNECT:
s = Fetch_String(FTP_CouldNotConnect);
break;
case DOWNLOADEVENT_LOGINFAILED:
s = Fetch_String(FTP_LoginFailed);
break;
case DOWNLOADEVENT_NOSUCHFILE:
s = Fetch_String(FTP_NoSuchFile);
break;
case DOWNLOADEVENT_LOCALFILEOPENFAILED:
s = Fetch_String(FTP_LocalFileOpenFailed);
break;
case DOWNLOADEVENT_TCPERROR:
s = Fetch_String(FTP_TCPError);
break;
case DOWNLOADEVENT_DISCONNECTERROR:
s = Fetch_String(FTP_DisconnectError);
break;
}
m_errorString = s;
DEBUG_LOG(("DownloadManager::OnError(): %s(%d)\n", s.c_str(), error));
return S_OK;
}
HRESULT DownloadManager::OnEnd()
{
m_sawEnd = true;
DEBUG_LOG(("DownloadManager::OnEnd()\n"));
return S_OK;
}
HRESULT DownloadManager::OnQueryResume()
{
DEBUG_LOG(("DownloadManager::OnQueryResume()\n"));
//return DOWNLOADEVENT_DONOTRESUME;
return DOWNLOADEVENT_RESUME;
}
HRESULT DownloadManager::OnProgressUpdate( int bytesread, int totalsize, int timetaken, int timeleft )
{
DEBUG_LOG(("DownloadManager::OnProgressUpdate(): %d/%d %d/%d\n", bytesread, totalsize, timetaken, timeleft));
return S_OK;
}
HRESULT DownloadManager::OnStatusUpdate( int status )
{
std::string s = Fetch_String(FTP_StatusNone);
switch (status)
{
case DOWNLOADSTATUS_CONNECTING:
s = Fetch_String(FTP_StatusConnecting);
break;
case DOWNLOADSTATUS_LOGGINGIN:
s = Fetch_String(FTP_StatusLoggingIn);
break;
case DOWNLOADSTATUS_FINDINGFILE:
s = Fetch_String(FTP_StatusFindingFile);
break;
case DOWNLOADSTATUS_QUERYINGRESUME:
s = Fetch_String(FTP_StatusQueryingResume);
break;
case DOWNLOADSTATUS_DOWNLOADING:
s = Fetch_String(FTP_StatusDownloading);
break;
case DOWNLOADSTATUS_DISCONNECTING:
s = Fetch_String(FTP_StatusDisconnecting);
break;
case DOWNLOADSTATUS_FINISHING:
s = Fetch_String(FTP_StatusFinishing);
break;
case DOWNLOADSTATUS_DONE:
s = Fetch_String(FTP_StatusDone);
break;
}
m_statusString = s;
DEBUG_LOG(("DownloadManager::OnStatusUpdate(): %s(%d)\n", s.c_str(), status));
return S_OK;
}