newgrf_config.cpp

Go to the documentation of this file.

/*
 * This file is part of OpenTTD.
 * OpenTTD 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, version 2.
 * OpenTTD 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 OpenTTD. If not, see <http://www.gnu.org/licenses/>.
 */
 
#include "stdafx.h"
#include "debug.h"
#include "3rdparty/md5/md5.h"
#include "newgrf.h"
#include "network/network_func.h"
#include "gfx_func.h"
#include "newgrf_text.h"
#include "window_func.h"
#include "progress.h"
#include "video/video_driver.hpp"
#include "string_func.h"
#include "strings_func.h"
#include "textfile_gui.h"
#include "thread.h"
#include "newgrf_config.h"
#include "newgrf_text.h"
 
#include "fileio_func.h"
#include "fios.h"
 
#include "safeguards.h"
 
 
GRFConfig::GRFConfig(const std::string &filename) :
  filename(filename), num_valid_params(ClampTo<uint8_t>(GRFConfig::param.size()))
{
}
 
GRFConfig::GRFConfig(const GRFConfig &config) :
  ZeroedMemoryAllocator(),
  ident(config.ident),
  original_md5sum(config.original_md5sum),
  filename(config.filename),
  name(config.name),
  info(config.info),
  url(config.url),
  error(config.error),
  version(config.version),
  min_loadable_version(config.min_loadable_version),
  flags(config.flags & ~(1 << GCF_COPY)),
  status(config.status),
  grf_bugs(config.grf_bugs),
  param(config.param),
  num_params(config.num_params),
  num_valid_params(config.num_valid_params),
  palette(config.palette),
  param_info(config.param_info),
  has_param_defaults(config.has_param_defaults)
{
}
 
void GRFConfig::SetParams(const std::vector<uint32_t> &pars)
{
  this->num_params = static_cast<uint8_t>(std::min(this->param.size(), pars.size()));
  std::copy(pars.begin(), pars.begin() + this->num_params, this->param.begin());
}
 
bool GRFConfig::IsCompatible(uint32_t old_version) const
{
  return this->min_loadable_version <= old_version && old_version <= this->version;
}
 
void GRFConfig::CopyParams(const GRFConfig &src)
{
  this->num_params = src.num_params;
  this->param = src.param;
}
 
const char *GRFConfig::GetName() const
{
  const char *name = GetGRFStringFromGRFText(this->name);
  return StrEmpty(name) ? this->filename.c_str() : name;
}
 
const char *GRFConfig::GetDescription() const
{
  return GetGRFStringFromGRFText(this->info);
}
 
const char *GRFConfig::GetURL() const
{
  return GetGRFStringFromGRFText(this->url);
}
 
void GRFConfig::SetParameterDefaults()
{
  this->num_params = 0;
  this->param = {};
 
  if (!this->has_param_defaults) return;
 
  for (const auto &info : this->param_info) {
    if (!info.has_value()) continue;
    this->SetValue(info.value(), info->def_value);
  }
}
 
void GRFConfig::SetSuitablePalette()
{
  PaletteType pal;
  switch (this->palette & GRFP_GRF_MASK) {
    case GRFP_GRF_DOS:     pal = PAL_DOS;      break;
    case GRFP_GRF_WINDOWS: pal = PAL_WINDOWS;  break;
    default:               pal = _settings_client.gui.newgrf_default_palette == 1 ? PAL_WINDOWS : PAL_DOS; break;
  }
  SB(this->palette, GRFP_USE_BIT, 1, pal == PAL_WINDOWS ? GRFP_USE_WINDOWS : GRFP_USE_DOS);
}
 
void GRFConfig::FinalizeParameterInfo()
{
  for (auto &info : this->param_info) {
    if (!info.has_value()) continue;
    info->Finalize();
  }
}
 
GRFConfig *_all_grfs;
GRFConfig *_grfconfig;
GRFConfig *_grfconfig_newgame;
GRFConfig *_grfconfig_static;
uint _missing_extra_graphics = 0;
 
GRFError::GRFError(StringID severity, StringID message) : message(message), severity(severity)
{
}
 
uint32_t GRFConfig::GetValue(const GRFParameterInfo &info) const
{
  /* GB doesn't work correctly with nbits == 32, so handle that case here. */
  if (info.num_bit == 32) return this->param[info.param_nr];
 
  return GB(this->param[info.param_nr], info.first_bit, info.num_bit);
}
 
void GRFConfig::SetValue(const GRFParameterInfo &info, uint32_t value)
{
  value = Clamp(value, info.min_value, info.max_value);
 
  /* SB doesn't work correctly with nbits == 32, so handle that case here. */
  if (info.num_bit == 32) {
    this->param[info.param_nr] = value;
  } else {
    SB(this->param[info.param_nr], info.first_bit, info.num_bit, value);
  }
 
  this->num_params = std::max<uint>(this->num_params, info.param_nr + 1);
  SetWindowDirty(WC_GAME_OPTIONS, WN_GAME_OPTIONS_NEWGRF_STATE);
}
 
void GRFParameterInfo::Finalize()
{
  this->complete_labels = true;
  for (uint32_t value = this->min_value; value <= this->max_value; value++) {
    if (this->value_names.count(value) == 0) {
      this->complete_labels = false;
      break;
    }
  }
}
 
void UpdateNewGRFConfigPalette(int32_t)
{
  for (GRFConfig *c = _grfconfig_newgame; c != nullptr; c = c->next) c->SetSuitablePalette();
  for (GRFConfig *c = _grfconfig_static;  c != nullptr; c = c->next) c->SetSuitablePalette();
  for (GRFConfig *c = _all_grfs;          c != nullptr; c = c->next) c->SetSuitablePalette();
}
 
size_t GRFGetSizeOfDataSection(FileHandle &f)
{
  extern const uint8_t _grf_cont_v2_sig[];
  static const uint header_len = 14;
 
  uint8_t data[header_len];
  if (fread(data, 1, header_len, f) == header_len) {
    if (data[0] == 0 && data[1] == 0 && MemCmpT(data + 2, _grf_cont_v2_sig, 8) == 0) {
      /* Valid container version 2, get data section size. */
      size_t offset = (static_cast<size_t>(data[13]) << 24) | (static_cast<size_t>(data[12]) << 16) | (static_cast<size_t>(data[11]) << 8) | static_cast<size_t>(data[10]);
      if (offset >= 1 * 1024 * 1024 * 1024) {
        Debug(grf, 0, "Unexpectedly large offset for NewGRF");
        /* Having more than 1 GiB of data is very implausible. Mostly because then
         * all pools in OpenTTD are flooded already. Or it's just Action C all over.
         * In any case, the offsets to graphics will likely not work either. */
        return SIZE_MAX;
      }
      return header_len + offset;
    }
  }
 
  return SIZE_MAX;
}
 
static bool CalcGRFMD5Sum(GRFConfig *config, Subdirectory subdir)
{
  Md5 checksum;
  uint8_t buffer[1024];
  size_t len, size;
 
  /* open the file */
  auto f = FioFOpenFile(config->filename, "rb", subdir, &size);
  if (!f.has_value()) return false;
 
  long start = ftell(*f);
  size = std::min(size, GRFGetSizeOfDataSection(*f));
 
  if (start < 0 || fseek(*f, start, SEEK_SET) < 0) {
    return false;
  }
 
  /* calculate md5sum */
  while ((len = fread(buffer, 1, (size > sizeof(buffer)) ? sizeof(buffer) : size, *f)) != 0 && size != 0) {
    size -= len;
    checksum.Append(buffer, len);
  }
  checksum.Finish(config->ident.md5sum);
 
  return true;
}
 
 
bool FillGRFDetails(GRFConfig *config, bool is_static, Subdirectory subdir)
{
  if (!FioCheckFileExists(config->filename, subdir)) {
    config->status = GCS_NOT_FOUND;
    return false;
  }
 
  /* Find and load the Action 8 information */
  LoadNewGRFFile(config, GLS_FILESCAN, subdir, true);
  config->SetSuitablePalette();
  config->FinalizeParameterInfo();
 
  /* Skip if the grfid is 0 (not read) or if it is an internal GRF */
  if (config->ident.grfid == 0 || HasBit(config->flags, GCF_SYSTEM)) return false;
 
  if (is_static) {
    /* Perform a 'safety scan' for static GRFs */
    LoadNewGRFFile(config, GLS_SAFETYSCAN, subdir, true);
 
    /* GCF_UNSAFE is set if GLS_SAFETYSCAN finds unsafe actions */
    if (HasBit(config->flags, GCF_UNSAFE)) return false;
  }
 
  return CalcGRFMD5Sum(config, subdir);
}
 
 
void ClearGRFConfigList(GRFConfig **config)
{
  GRFConfig *c, *next;
  for (c = *config; c != nullptr; c = next) {
    next = c->next;
    delete c;
  }
  *config = nullptr;
}
 
 
GRFConfig **CopyGRFConfigList(GRFConfig **dst, const GRFConfig *src, bool init_only)
{
  /* Clear destination as it will be overwritten */
  ClearGRFConfigList(dst);
  for (; src != nullptr; src = src->next) {
    GRFConfig *c = new GRFConfig(*src);
 
    ClrBit(c->flags, GCF_INIT_ONLY);
    if (init_only) SetBit(c->flags, GCF_INIT_ONLY);
 
    *dst = c;
    dst = &c->next;
  }
 
  return dst;
}
 
static void RemoveDuplicatesFromGRFConfigList(GRFConfig *list)
{
  GRFConfig *prev;
  GRFConfig *cur;
 
  if (list == nullptr) return;
 
  for (prev = list, cur = list->next; cur != nullptr; prev = cur, cur = cur->next) {
    if (cur->ident.grfid != list->ident.grfid) continue;
 
    prev->next = cur->next;
    delete cur;
    cur = prev; // Just go back one so it continues as normal later on
  }
 
  RemoveDuplicatesFromGRFConfigList(list->next);
}
 
void AppendStaticGRFConfigs(GRFConfig **dst)
{
  GRFConfig **tail = dst;
  while (*tail != nullptr) tail = &(*tail)->next;
 
  CopyGRFConfigList(tail, _grfconfig_static, false);
  RemoveDuplicatesFromGRFConfigList(*dst);
}
 
void AppendToGRFConfigList(GRFConfig **dst, GRFConfig *el)
{
  GRFConfig **tail = dst;
  while (*tail != nullptr) tail = &(*tail)->next;
  *tail = el;
 
  RemoveDuplicatesFromGRFConfigList(*dst);
}
 
 
void ResetGRFConfig(bool defaults)
{
  CopyGRFConfigList(&_grfconfig, _grfconfig_newgame, !defaults);
  AppendStaticGRFConfigs(&_grfconfig);
}
 
 
GRFListCompatibility IsGoodGRFConfigList(GRFConfig *grfconfig)
{
  GRFListCompatibility res = GLC_ALL_GOOD;
 
  for (GRFConfig *c = grfconfig; c != nullptr; c = c->next) {
    const GRFConfig *f = FindGRFConfig(c->ident.grfid, FGCM_EXACT, &c->ident.md5sum);
    if (f == nullptr || HasBit(f->flags, GCF_INVALID)) {
      /* If we have not found the exactly matching GRF try to find one with the
       * same grfid, as it most likely is compatible */
      f = FindGRFConfig(c->ident.grfid, FGCM_COMPATIBLE, nullptr, c->version);
      if (f != nullptr) {
        Debug(grf, 1, "NewGRF {:08X} ({}) not found; checksum {}. Compatibility mode on", BSWAP32(c->ident.grfid), c->filename, FormatArrayAsHex(c->ident.md5sum));
        if (!HasBit(c->flags, GCF_COMPATIBLE)) {
          /* Preserve original_md5sum after it has been assigned */
          SetBit(c->flags, GCF_COMPATIBLE);
          c->original_md5sum = c->ident.md5sum;
        }
 
        /* Non-found has precedence over compatibility load */
        if (res != GLC_NOT_FOUND) res = GLC_COMPATIBLE;
        goto compatible_grf;
      }
 
      /* No compatible grf was found, mark it as disabled */
      Debug(grf, 0, "NewGRF {:08X} ({}) not found; checksum {}", BSWAP32(c->ident.grfid), c->filename, FormatArrayAsHex(c->ident.md5sum));
 
      c->status = GCS_NOT_FOUND;
      res = GLC_NOT_FOUND;
    } else {
compatible_grf:
      Debug(grf, 1, "Loading GRF {:08X} from {}", BSWAP32(f->ident.grfid), f->filename);
      /* The filename could be the filename as in the savegame. As we need
       * to load the GRF here, we need the correct filename, so overwrite that
       * in any case and set the name and info when it is not set already.
       * When the GCF_COPY flag is set, it is certain that the filename is
       * already a local one, so there is no need to replace it. */
      if (!HasBit(c->flags, GCF_COPY)) {
        c->filename = f->filename;
        c->ident.md5sum = f->ident.md5sum;
        c->name = f->name;
        c->info = f->name;
        c->error.reset();
        c->version = f->version;
        c->min_loadable_version = f->min_loadable_version;
        c->num_valid_params = f->num_valid_params;
        c->param_info = f->param_info;
        c->has_param_defaults = f->has_param_defaults;
      }
    }
  }
 
  return res;
}
 
 
int _skip_all_newgrf_scanning = 0;
 
class GRFFileScanner : FileScanner {
  std::chrono::steady_clock::time_point next_update; 
  uint num_scanned; 
 
public:
  GRFFileScanner() : num_scanned(0)
  {
    this->next_update = std::chrono::steady_clock::now();
  }
 
  bool AddFile(const std::string &filename, size_t basepath_length, const std::string &tar_filename) override;
 
  static uint DoScan()
  {
    if (_skip_all_newgrf_scanning > 0) {
      if (_skip_all_newgrf_scanning == 1) _skip_all_newgrf_scanning = 0;
      return 0;
    }
 
    GRFFileScanner fs;
    int ret = fs.Scan(".grf", NEWGRF_DIR);
    /* The number scanned and the number returned may not be the same;
     * duplicate NewGRFs and base sets are ignored in the return value. */
    _settings_client.gui.last_newgrf_count = fs.num_scanned;
    return ret;
  }
};
 
bool GRFFileScanner::AddFile(const std::string &filename, size_t basepath_length, const std::string &)
{
  /* Abort if the user stopped the game during a scan. */
  if (_exit_game) return false;
 
  GRFConfig *c = new GRFConfig(filename.c_str() + basepath_length);
 
  bool added = true;
  if (FillGRFDetails(c, false)) {
    if (_all_grfs == nullptr) {
      _all_grfs = c;
    } else {
      /* Insert file into list at a position determined by its
       * name, so the list is sorted as we go along */
      GRFConfig **pd, *d;
      bool stop = false;
      for (pd = &_all_grfs; (d = *pd) != nullptr; pd = &d->next) {
        if (c->ident.grfid == d->ident.grfid && c->ident.md5sum == d->ident.md5sum) added = false;
        /* Because there can be multiple grfs with the same name, make sure we checked all grfs with the same name,
         *  before inserting the entry. So insert a new grf at the end of all grfs with the same name, instead of
         *  just after the first with the same name. Avoids doubles in the list. */
        if (StrCompareIgnoreCase(c->GetName(), d->GetName()) <= 0) {
          stop = true;
        } else if (stop) {
          break;
        }
      }
      if (added) {
        c->next = d;
        *pd = c;
      }
    }
  } else {
    added = false;
  }
 
  this->num_scanned++;
 
  const char *name = nullptr;
  if (c->name != nullptr) name = GetGRFStringFromGRFText(c->name);
  if (name == nullptr) name = c->filename.c_str();
  UpdateNewGRFScanStatus(this->num_scanned, name);
  VideoDriver::GetInstance()->GameLoopPause();
 
  if (!added) {
    /* File couldn't be opened, or is either not a NewGRF or is a
     * 'system' NewGRF or it's already known, so forget about it. */
    delete c;
  }
 
  return added;
}
 
static bool GRFSorter(GRFConfig * const &c1, GRFConfig * const &c2)
{
  return StrNaturalCompare(c1->GetName(), c2->GetName()) < 0;
}
 
void DoScanNewGRFFiles(NewGRFScanCallback *callback)
{
  ClearGRFConfigList(&_all_grfs);
  TarScanner::DoScan(TarScanner::NEWGRF);
 
  Debug(grf, 1, "Scanning for NewGRFs");
  uint num = GRFFileScanner::DoScan();
 
  Debug(grf, 1, "Scan complete, found {} files", num);
  if (num != 0 && _all_grfs != nullptr) {
    /* Sort the linked list using quicksort.
     * For that we first have to make an array, then sort and
     * then remake the linked list. */
    std::vector<GRFConfig *> to_sort;
 
    uint i = 0;
    for (GRFConfig *p = _all_grfs; p != nullptr; p = p->next, i++) {
      to_sort.push_back(p);
    }
    /* Number of files is not necessarily right */
    num = i;
 
    std::sort(to_sort.begin(), to_sort.end(), GRFSorter);
 
    for (i = 1; i < num; i++) {
      to_sort[i - 1]->next = to_sort[i];
    }
    to_sort[num - 1]->next = nullptr;
    _all_grfs = to_sort[0];
 
    NetworkAfterNewGRFScan();
  }
 
  /* Yes... these are the NewGRF windows */
  InvalidateWindowClassesData(WC_SAVELOAD, 0, true);
  InvalidateWindowData(WC_GAME_OPTIONS, WN_GAME_OPTIONS_NEWGRF_STATE, GOID_NEWGRF_RESCANNED, true);
  if (!_exit_game && callback != nullptr) callback->OnNewGRFsScanned();
 
  CloseWindowByClass(WC_MODAL_PROGRESS);
  SetModalProgress(false);
  MarkWholeScreenDirty();
}
 
void ScanNewGRFFiles(NewGRFScanCallback *callback)
{
  /* First set the modal progress. This ensures that it will eventually let go of the paint mutex. */
  SetModalProgress(true);
  /* Only then can we really start, especially by marking the whole screen dirty. Get those other windows hidden!. */
  MarkWholeScreenDirty();
 
  DoScanNewGRFFiles(callback);
}
 
const GRFConfig *FindGRFConfig(uint32_t grfid, FindGRFConfigMode mode, const MD5Hash *md5sum, uint32_t desired_version)
{
  assert((mode == FGCM_EXACT) != (md5sum == nullptr));
  const GRFConfig *best = nullptr;
  for (const GRFConfig *c = _all_grfs; c != nullptr; c = c->next) {
    /* if md5sum is set, we look for an exact match and continue if not found */
    if (!c->ident.HasGrfIdentifier(grfid, md5sum)) continue;
    /* return it, if the exact same newgrf is found, or if we do not care about finding "the best" */
    if (md5sum != nullptr || mode == FGCM_ANY) return c;
    /* Skip incompatible stuff, unless explicitly allowed */
    if (mode != FGCM_NEWEST && HasBit(c->flags, GCF_INVALID)) continue;
    /* check version compatibility */
    if (mode == FGCM_COMPATIBLE && !c->IsCompatible(desired_version)) continue;
    /* remember the newest one as "the best" */
    if (best == nullptr || c->version > best->version) best = c;
  }
 
  return best;
}
 
GRFConfig *GetGRFConfig(uint32_t grfid, uint32_t mask)
{
  GRFConfig *c;
 
  for (c = _grfconfig; c != nullptr; c = c->next) {
    if ((c->ident.grfid & mask) == (grfid & mask)) return c;
  }
 
  return nullptr;
}
 
 
std::string GRFBuildParamList(const GRFConfig *c)
{
  std::string result;
  for (uint i = 0; i < c->num_params; i++) {
    if (!result.empty()) result += ' ';
    result += std::to_string(c->param[i]);
  }
  return result;
}
 
std::optional<std::string> GRFConfig::GetTextfile(TextfileType type) const
{
  return ::GetTextfile(type, NEWGRF_DIR, this->filename);
}