Initial commit of Command & Conquer Generals and Command & Conquer Generals Zero Hour source code.

GeneralsMD/Code/Tools/PATCHGET/DownloadManager.cpp

@@ -0,0 +1,210 @@
+/*
+**	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;
+}