Rewrite SysConf handling

This rewrites the SysConf code for several reasons:

* Modernising the SysConf class. The naming was entirely cleaned up.
  constexpr for constants.

* Exposing less stuff in the header.

* Probably less efficient parsing and writing logic, but much simpler
  to understand and use in my opinion. No more hardcoded offsets.
  No more duplicated code for the initial SYSCONF generation.

* More flexibility. It is now possible to add and remove entries,
  since we rebuild the file. This allows us to stop spamming
  "section not found" panic alerts; we can now use and insert
  default entries.
This commit is contained in:
Léo Lam
2017-03-10 18:11:45 +01:00
parent 795157a9bc
commit d86f020e81
5 changed files with 377 additions and 604 deletions

View File

@ -2,193 +2,92 @@
// Licensed under GPLv2+
// Refer to the license.txt file included.
// Utilities to parse and modify a Wii SYSCONF file and its sections.
#pragma once
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <string>
#include <vector>
#include "Common/Assert.h"
#include "Common/CommonTypes.h"
#include "Common/Logging/Log.h"
#include "Common/MsgHandler.h"
#include "Common/NandPaths.h"
namespace File
{
class IOFile;
}
// This class is meant to edit the values in a given Wii SYSCONF file
// It currently does not add/remove/rearrange sections,
// instead only modifies exiting sections' data
#define SYSCONF_SIZE 0x4000
enum SysconfType
{
Type_BigArray = 1,
Type_SmallArray,
Type_Byte,
Type_Short,
Type_Long,
Type_LongLong,
Type_Bool
};
struct SSysConfHeader
{
char version[4];
u16 numEntries;
};
struct SSysConfEntry
{
u16 offset = 0;
SysconfType type;
u8 nameLength = 0;
char name[32] = {};
u16 dataLength = 0;
std::vector<u8> data;
template <class T>
T GetData() const
{
T extracted_data;
std::memcpy(&extracted_data, data.data(), sizeof(T));
return extracted_data;
}
bool GetArrayData(u8* dest, u16 destSize) const
{
if (dest && destSize >= dataLength)
{
memcpy(dest, data.data(), dataLength);
return true;
}
return false;
}
bool SetArrayData(const u8* buffer, u16 bufferSize)
{
if (buffer)
{
memcpy(data.data(), buffer, std::min<u16>(bufferSize, dataLength));
return true;
}
return false;
}
};
class SysConf
class SysConf final
{
public:
SysConf(Common::FromWhichRoot root_type);
~SysConf();
bool IsValid() const { return m_IsValid; }
template <class T>
T GetData(const char* sectionName) const
void Clear();
void Load();
bool Save() const;
struct Entry
{
if (!m_IsValid)
enum Type : u8
{
PanicAlertT("Trying to read from invalid SYSCONF");
return 0;
BigArray = 1,
SmallArray = 2,
Byte = 3,
Short = 4,
Long = 5,
LongLong = 6,
// Should really be named Bool, but this conflicts with some random macro. :/
ByteBool = 7,
};
Entry(Type type_, const std::string& name_);
Entry(Type type_, const std::string& name_, const std::vector<u8>& bytes_);
Entry(Type type_, const std::string& name_, std::vector<u8>&& bytes_);
// Intended for use with the non array types.
template <typename T>
T GetData(T default_value) const
{
if (bytes.size() != sizeof(T))
return default_value;
T value;
std::memcpy(&value, bytes.data(), bytes.size());
return value;
}
template <typename T>
void SetData(T value)
{
_assert_(sizeof(value) == bytes.size());
std::memcpy(bytes.data(), &value, bytes.size());
}
auto index = m_Entries.cbegin();
for (; index < m_Entries.cend() - 1; ++index)
{
if (strcmp(index->name, sectionName) == 0)
break;
}
if (index == m_Entries.cend() - 1)
{
PanicAlertT("Section %s not found in SYSCONF", sectionName);
return 0;
}
Type type;
std::string name;
std::vector<u8> bytes;
};
return index->GetData<T>();
}
void AddEntry(Entry&& entry);
Entry* GetEntry(const std::string& key);
const Entry* GetEntry(const std::string& key) const;
Entry* GetOrAddEntry(const std::string& key, Entry::Type type);
void RemoveEntry(const std::string& key);
bool GetArrayData(const char* sectionName, u8* dest, u16 destSize) const
// Intended for use with the non array types.
template <typename T>
T GetData(const std::string& key, T default_value) const
{
if (!m_IsValid)
{
PanicAlertT("Trying to read from invalid SYSCONF");
return false;
}
auto index = m_Entries.cbegin();
for (; index < m_Entries.cend() - 1; ++index)
{
if (strcmp(index->name, sectionName) == 0)
break;
}
if (index == m_Entries.cend() - 1)
{
PanicAlertT("Section %s not found in SYSCONF", sectionName);
return false;
}
return index->GetArrayData(dest, destSize);
const Entry* entry = GetEntry(key);
return entry ? entry->GetData(default_value) : default_value;
}
bool SetArrayData(const char* sectionName, const u8* buffer, u16 bufferSize)
template <typename T>
void SetData(const std::string& key, Entry::Type type, T value)
{
if (!m_IsValid)
return false;
auto index = m_Entries.begin();
for (; index < m_Entries.end() - 1; ++index)
{
if (strcmp(index->name, sectionName) == 0)
break;
}
if (index == m_Entries.end() - 1)
{
PanicAlertT("Section %s not found in SYSCONF", sectionName);
return false;
}
return index->SetArrayData(buffer, bufferSize);
GetOrAddEntry(key, type)->SetData(value);
}
template <class T>
bool SetData(const char* sectionName, T newValue)
{
if (!m_IsValid)
return false;
auto index = m_Entries.begin();
for (; index < m_Entries.end() - 1; ++index)
{
if (strcmp(index->name, sectionName) == 0)
break;
}
if (index == m_Entries.end() - 1)
{
PanicAlertT("Section %s not found in SYSCONF", sectionName);
return false;
}
std::memcpy(index->data.data(), &newValue, sizeof(T));
return true;
}
bool Save();
bool SaveToFile(const std::string& filename);
bool LoadFromFile(const std::string& filename);
bool Reload();
void UpdateLocation(Common::FromWhichRoot root_type);
private:
bool LoadFromFileInternal(File::IOFile&& file);
void GenerateSysConf();
void Clear();
void ApplySettingsFromMovie();
void InsertDefaultEntries();
bool LoadFromFile(const std::string& file_name);
std::string m_Filename;
std::string m_FilenameDefault;
std::vector<SSysConfEntry> m_Entries;
bool m_IsValid = false;
std::string m_file_name;
std::vector<Entry> m_entries;
};