Filter out unnecessary files

src/Components/Modules/Download.cpp

@@ -116,8 +116,9 @@ namespace Components
  fileEntry.name = name;
  fileEntry.hash = hash;
  fileEntry.size = size;
+ fileEntry.isMap = download->isMap_;
 
- if (!fileEntry.name.empty())
+ if (!fileEntry.name.empty() && fileEntry.allowed())
  {
  download->files_.push_back(fileEntry);
  download->totalBytes_ += fileEntry.size;
@@ -139,6 +140,8 @@ namespace Components
 
  auto file = download->files_[index];
 
+ assert(file.allowed());
+
  auto path = download->mod_ + "/" + file.name;
  if (download->isMap_)
  {
@@ -884,4 +887,49 @@ namespace Components
  CLDownload.clear();
  }
  }
+
+ bool Download::ClientDownload::File::allowed() const
+ {
+ if (name.contains("..") || name.contains(":"))
+ {
+ return false;
+ }
+
+ if (name.contains("\\") || name.contains("/"))
+ {
+ return false;
+ }
+
+ if (isMap)
+ {
+ if (name.ends_with(".arena"))
+ {
+ return true;
+ }
+
+ if (name.ends_with(".iwd"))
+ {
+ return true;
+ }
+
+ if (name.ends_with(".ff"))
+ {
+ return true;
+ }
+ }
+ else
+ {
+ // Plain and simple
+ if (name == "mod.ff")
+ {
+ return true;
+ }
+ if (name.ends_with(".iwd"))
+ {
+ return true;
+ }
+ }
+
+ return false;
+ }
 }

src/Components/Modules/Download.hpp

@@ -55,6 +55,9 @@ namespace Components
  std::string name;
  std::string hash;
  std::size_t size;
+ bool isMap;
+
+ bool allowed() const;
  };
 
  std::vector<File> files_;