[Download] Implement map download

This commit is contained in:
momo5502 2017-04-06 22:22:47 +02:00
parent 8bb3c1deca
commit 67defdf5a3
4 changed files with 150 additions and 37 deletions

View File

@ -7,7 +7,12 @@ namespace Components
#pragma region Client
void Download::InitiateClientDownload(std::string mod)
void Download::InitiateMapDownload(std::string map)
{
Download::InitiateClientDownload(map, true);
}
void Download::InitiateClientDownload(std::string mod, bool map)
{
if (Download::CLDownload.running) return;
@ -21,6 +26,7 @@ namespace Components
Command::Execute("openmenu mod_download_popmenu", false);
Download::CLDownload.running = true;
Download::CLDownload.isMap = map;
Download::CLDownload.mod = mod;
Download::CLDownload.terminateThread = false;
Download::CLDownload.totalBytes = 0;
@ -39,8 +45,11 @@ namespace Components
std::string error;
json11::Json listData = json11::Json::parse(list, error);
if (!error.empty() || !listData.is_array()) return false;
if (!error.empty() || !listData.is_array())
{
Logger::Print("Error: %s\n", error.data());
return false;
}
download->totalBytes = 0;
@ -163,6 +172,11 @@ namespace Components
auto file = download->files[index];
std::string path = download->mod + "/" + file.name;
if (download->isMap)
{
path = "usermaps/" + path;
}
if (Utils::IO::FileExists(path))
{
std::string data = Utils::IO::ReadFile(path);
@ -174,7 +188,7 @@ namespace Components
}
}
std::string url = "http://" + download->target.getString() + "/file/" + file.name;
std::string url = "http://" + download->target.getString() + "/file/" + (download->isMap ? "map/" : "") + file.name;
Download::FileDownload fDownload;
fDownload.file = file;
@ -202,7 +216,7 @@ namespace Components
return false;
}
Utils::IO::CreateDir(download->mod);
Utils::IO::CreateDir("usermaps/" + download->mod);
Utils::IO::WriteFile(path, fDownload.buffer);
return true;
@ -213,8 +227,8 @@ namespace Components
if (!download) download = &Download::CLDownload;
std::string host = "http://" + download->target.getString();
std::string list = Utils::WebIO("IW4x", host + "/list").setTimeout(5000)->get();
std::string list = Utils::WebIO("IW4x", host + (download->isMap ? "/map" : "/list")).setTimeout(5000)->get();
if (list.empty())
{
if (download->terminateThread) return;
@ -222,8 +236,9 @@ namespace Components
download->thread.detach();
download->clear();
QuickPatch::Once([] ()
QuickPatch::Once([]()
{
Command::Execute("closemenu mod_download_popmenu");
Party::ConnectError("Failed to download the modlist!");
});
@ -239,8 +254,9 @@ namespace Components
download->thread.detach();
download->clear();
QuickPatch::Once([] ()
QuickPatch::Once([]()
{
Command::Execute("closemenu mod_download_popmenu");
Party::ConnectError("Failed to parse the modlist!");
});
@ -264,7 +280,7 @@ namespace Components
download->thread.detach();
download->clear();
QuickPatch::Once([] ()
QuickPatch::Once([]()
{
Dvar::Var("partyend_reason").set(mod);
mod.clear();
@ -282,23 +298,33 @@ namespace Components
download->thread.detach();
download->clear();
// Run this on the main thread
QuickPatch::Once([] ()
if(download->isMap)
{
auto fsGame = Dvar::Var("fs_game");
fsGame.set(mod);
fsGame.get<Game::dvar_t*>()->modified = true;
mod.clear();
Command::Execute("closemenu mod_download_popmenu", false);
if (Dvar::Var("cl_modVidRestart").get<bool>())
QuickPatch::Once([]()
{
Command::Execute("vid_restart", false);
}
Command::Execute("reconnect", false);
});
}
else
{
// Run this on the main thread
QuickPatch::Once([]()
{
auto fsGame = Dvar::Var("fs_game");
fsGame.set(mod);
fsGame.get<Game::dvar_t*>()->modified = true;
mod.clear();
Command::Execute("reconnect", false);
});
Command::Execute("closemenu mod_download_popmenu", false);
if (Dvar::Var("cl_modVidRestart").get<bool>())
{
Command::Execute("vid_restart", false);
}
Command::Execute("reconnect", false);
});
}
}
#pragma endregion
@ -341,6 +367,58 @@ namespace Components
nc->flags |= MG_F_SEND_AND_CLOSE;
}
void Download::MapHandler(mg_connection *nc, int ev, void* /*ev_data*/)
{
// Only handle http requests
if (ev != MG_EV_HTTP_REQUEST) return;
static std::string mapnamePre;
static json11::Json jsonList;
std::string mapname = Maps::GetUserMap()->getName();
if(!Maps::GetUserMap()->isValid())
{
mapnamePre.clear();
jsonList = std::vector<json11::Json>();
}
else if (!mapname.empty() && mapname != mapnamePre)
{
std::vector<json11::Json> fileList;
mapnamePre = mapname;
std::string path = Dvar::Var("fs_basepath").get<std::string>() + "\\usermaps\\" + mapname;
std::vector<std::string> list = { mapname + ".ff", mapname + "_load.ff", mapname + ".iwd" };
for (auto i = list.begin(); i != list.end(); ++i)
{
std::string filename = path + "\\" + *i;
if (strstr(i->data(), "_svr_") == nullptr && Utils::IO::FileExists(filename))
{
std::map<std::string, json11::Json> file;
std::string fileBuffer = Utils::IO::ReadFile(filename);
file["name"] = *i;
file["size"] = static_cast<int>(fileBuffer.size());
file["hash"] = Utils::Cryptography::SHA256::Compute(fileBuffer, true);
fileList.push_back(file);
}
}
jsonList = fileList;
}
mg_printf(nc,
"HTTP/1.1 200 OK\r\n"
"Content-Type: application/json\r\n"
"Connection: close\r\n"
"\r\n"
"%s", jsonList.dump().data());
nc->flags |= MG_F_SEND_AND_CLOSE;
}
void Download::ListHandler(mg_connection* nc, int ev, void* /*ev_data*/)
{
// Only handle http requests
@ -374,7 +452,7 @@ namespace Components
if (strstr(i->data(), "_svr_") == nullptr && Utils::IO::FileExists(filename))
{
std::map<std::string, json11::Json> file;
std::string fileBuffer = Utils::IO::ReadFile(path + "\\" + *i);
std::string fileBuffer = Utils::IO::ReadFile(filename);
file["name"] = *i;
file["size"] = static_cast<int>(fileBuffer.size());
@ -413,7 +491,7 @@ namespace Components
{
std::string url(message->uri.p, message->uri.len);
Utils::String::Replace(url, "\\", "/");
if (url.size() >= 6)
{
url = url.substr(6);
@ -421,17 +499,35 @@ namespace Components
Utils::String::Replace(url, "%20", " ");
if (url.find_first_of("/") != std::string::npos || (!Utils::String::EndsWith(url, ".iwd") && url != "mod.ff") || strstr(url.data(), "_svr_") != nullptr)
bool isMap = false;
if (Utils::String::StartsWith(url, "map/"))
{
Download::Forbid(nc);
return;
isMap = true;
url = url.substr(4);
std::string mapname = Maps::GetUserMap()->getName();
if(!Maps::GetUserMap()->isValid() || (url != (mapname + ".ff") && url != (mapname + ".iwd") && url != (mapname + "_load.ff")))
{
Download::Forbid(nc);
return;
}
url = Utils::String::VA("usermaps\\%s\\%s", mapname.data(), url.data());
}
else
{
if (url.find_first_of("/") != std::string::npos || (!Utils::String::EndsWith(url, ".iwd") && url != "mod.ff") || strstr(url.data(), "_svr_") != nullptr)
{
Download::Forbid(nc);
return;
}
}
std::string file;
std::string fsGame = Dvar::Var("fs_game").get<std::string>();
std::string path = Dvar::Var("fs_basepath").get<std::string>() + "\\" + fsGame + "\\" + url;
std::string path = Dvar::Var("fs_basepath").get<std::string>() + "\\" + (isMap ? "" : fsGame + "\\") + url;
if (fsGame.empty() || !Utils::IO::ReadFile(path, &file))
if ((!isMap && fsGame.empty()) || !Utils::IO::ReadFile(path, &file))
{
mg_printf(nc,
"HTTP/1.1 404 Not Found\r\n"
@ -606,6 +702,7 @@ namespace Components
// Handle special requests
mg_register_http_endpoint(nc, "/info", Download::InfoHandler);
mg_register_http_endpoint(nc, "/list", Download::ListHandler);
mg_register_http_endpoint(nc, "/map", Download::MapHandler);
mg_register_http_endpoint(nc, "/file/", Download::FileHandler);
mg_set_protocol_http_websocket(nc);

View File

@ -14,18 +14,20 @@ namespace Components
void preDestroy() override;
static void InitiateClientDownload(std::string mod);
static void InitiateClientDownload(std::string mod, bool map = false);
static void InitiateMapDownload(std::string map);
private:
class ClientDownload
{
public:
ClientDownload() : running(false), valid(false), terminateThread(false), totalBytes(0), downBytes(0), lastTimeStamp(0), timeStampBytes(0) {}
ClientDownload(bool _isMap = false) : running(false), valid(false), terminateThread(false), isMap(_isMap), totalBytes(0), downBytes(0), lastTimeStamp(0), timeStampBytes(0) {}
~ClientDownload() { this->clear(); }
bool running;
bool valid;
bool terminateThread;
bool isMap;
mg_mgr mgr;
Network::Address target;
std::string mod;
@ -86,6 +88,7 @@ namespace Components
static void EventHandler(mg_connection *nc, int ev, void *ev_data);
static void ListHandler(mg_connection *nc, int ev, void *ev_data);
static void MapHandler(mg_connection *nc, int ev, void *ev_data);
static void FileHandler(mg_connection *nc, int ev, void *ev_data);
static void InfoHandler(mg_connection *nc, int ev, void *ev_data);
static void DownloadHandler(mg_connection *nc, int ev, void *ev_data);

View File

@ -325,6 +325,11 @@ namespace Components
info.set("securityLevel", Utils::String::VA("%i", Dvar::Var("sv_securityLevel").get<int>()));
info.set("sv_running", (Dvar::Var("sv_running").get<bool>() ? "1" : "0"));
if (Maps::GetUserMap()->isValid())
{
info.set("usermaphash", Utils::String::VA("%i", Maps::GetUserMap()->getHash()));
}
if (Dedicated::IsEnabled())
{
info.set("sv_motd", Dvar::Var("sv_motd").get<std::string>());
@ -372,6 +377,8 @@ namespace Components
Party::Container.matchType = atoi(info.get("matchtype").data());
uint32_t securityLevel = static_cast<uint32_t>(atoi(info.get("securityLevel").data()));
bool isUsermap = !info.get("usermaphash").empty();
unsigned int usermapHash = atoi(info.get("usermaphash").data());
std::string mod = Dvar::Var("fs_game").get<std::string>();
@ -393,6 +400,15 @@ namespace Components
{
Party::ConnectError("Invalid join response: Unknown matchtype");
}
else if (Party::Container.info.get("mapname").empty() || Party::Container.info.get("gametype").empty())
{
Party::ConnectError("Invalid map or gametype.");
}
else if(isUsermap && usermapHash != Maps::GetUsermapHash(info.get("mapname")))
{
Command::Execute("closemenu popup_reconnectingtoparty");
Download::InitiateMapDownload(info.get("mapname"));
}
else if(!info.get("fs_game").empty() && Utils::String::ToLower(mod) != Utils::String::ToLower(info.get("fs_game")))
{
Command::Execute("closemenu popup_reconnectingtoparty");
@ -411,6 +427,8 @@ namespace Components
}
else
{
if (!Maps::CheckMapInstalled(Party::Container.info.get("mapname").data(), true)) return;
Party::Container.motd = info.get("sv_motd");
if (Party::Container.matchType == 1) // Party
@ -433,10 +451,6 @@ namespace Components
{
Party::ConnectError("@EXE_SERVERISFULL");
}
if (Party::Container.info.get("mapname") == "" || Party::Container.info.get("gametype") == "")
{
Party::ConnectError("Invalid map or gametype.");
}
else
{
Dvar::Var("xblive_privateserver").set(true);

View File

@ -780,7 +780,6 @@ namespace Components
if (info)
{
if (!Maps::CheckMapInstalled(info->mapname.data(), true)) return;
Party::Connect(info->addr);
}
});