service/ptm: Add Get/SetPlayCoins functions

src/core/hle/service/ptm/ptm.cpp

@@ -136,12 +136,12 @@ void Module::Interface::CheckNew3DS(Kernel::HLERequestContext& ctx) {
     Service::PTM::CheckNew3DS(rb);
 }
 
-Module::Module() {
+static void WriteGameCoinData(GameCoin gamecoin_data) {
-    // Open the SharedExtSaveData archive 0xF000000B and create the gamecoin.dat file if it doesn't
-    // exist
     FileSys::Path archive_path(ptm_shared_extdata_id);
     auto archive_result =
         Service::FS::OpenArchive(Service::FS::ArchiveIdCode::SharedExtSaveData, archive_path);
+
+    FileSys::Path gamecoin_path("/gamecoin.dat");
     // If the archive didn't exist, create the files inside
     if (archive_result.Code() == FileSys::ERR_NOT_FORMATTED) {
         // Format the archive to create the directories
@@ -150,22 +150,74 @@ Module::Module() {
         // Open it again to get a valid archive now that the folder exists
         archive_result =
             Service::FS::OpenArchive(Service::FS::ArchiveIdCode::SharedExtSaveData, archive_path);
-        ASSERT_MSG(archive_result.Succeeded(), "Could not open the PTM SharedExtSaveData archive!");
+        // Create the game coin file
-
-        FileSys::Path gamecoin_path("/gamecoin.dat");
         Service::FS::CreateFileInArchive(*archive_result, gamecoin_path, sizeof(GameCoin));
-        FileSys::Mode open_mode = {};
+    } else {
-        open_mode.write_flag.Assign(1);
+        ASSERT_MSG(archive_result.Succeeded(), "Could not open the PTM SharedExtSaveData archive!");
-        // Open the file and write the default gamecoin information
-        auto gamecoin_result =
-            Service::FS::OpenFileFromArchive(*archive_result, gamecoin_path, open_mode);
-        if (gamecoin_result.Succeeded()) {
-            auto gamecoin = std::move(gamecoin_result).Unwrap();
-            gamecoin->backend->Write(0, sizeof(GameCoin), true,
-                                     reinterpret_cast<const u8*>(&default_game_coin));
-            gamecoin->backend->Close();
-        }
     }
+
+    FileSys::Mode open_mode = {};
+    open_mode.write_flag.Assign(1);
+    // Open the file and write the default gamecoin information
+    auto gamecoin_result =
+        Service::FS::OpenFileFromArchive(*archive_result, gamecoin_path, open_mode);
+    if (gamecoin_result.Succeeded()) {
+        auto gamecoin = std::move(gamecoin_result).Unwrap();
+        gamecoin->backend->Write(0, sizeof(GameCoin), true,
+                                 reinterpret_cast<const u8*>(&gamecoin_data));
+        gamecoin->backend->Close();
+    }
+}
+
+static GameCoin ReadGameCoinData() {
+    FileSys::Path archive_path(ptm_shared_extdata_id);
+    auto archive_result =
+        Service::FS::OpenArchive(Service::FS::ArchiveIdCode::SharedExtSaveData, archive_path);
+    if (!archive_result.Succeeded()) {
+        LOG_ERROR(Service_PTM, "Could not open the PTM SharedExtSaveData archive!");
+        return default_game_coin;
+    }
+
+    FileSys::Path gamecoin_path("/gamecoin.dat");
+    FileSys::Mode open_mode = {};
+    open_mode.read_flag.Assign(1);
+
+    auto gamecoin_result =
+        Service::FS::OpenFileFromArchive(*archive_result, gamecoin_path, open_mode);
+    if (!gamecoin_result.Succeeded()) {
+        LOG_ERROR(Service_PTM, "Could not open the game coin data file!");
+        return default_game_coin;
+    }
+    u16 result;
+    auto gamecoin = std::move(gamecoin_result).Unwrap();
+    GameCoin gamecoin_data;
+    gamecoin->backend->Read(0, sizeof(GameCoin), reinterpret_cast<u8*>(&gamecoin_data));
+    gamecoin->backend->Close();
+    return gamecoin_data;
+}
+
+Module::Module() {
+    // Open the SharedExtSaveData archive 0xF000000B and create the gamecoin.dat file if it doesn't
+    // exist
+    FileSys::Path archive_path(ptm_shared_extdata_id);
+    auto archive_result =
+        Service::FS::OpenArchive(Service::FS::ArchiveIdCode::SharedExtSaveData, archive_path);
+    // If the archive didn't exist, write the default game coin file
+    if (archive_result.Code() == FileSys::ERR_NOT_FORMATTED) {
+        WriteGameCoinData(default_game_coin);
+    }
+}
+
+u16 Module::GetPlayCoins() {
+    return ReadGameCoinData().total_coins;
+}
+
+void Module::SetPlayCoins(u16 play_coins) {
+    GameCoin game_coin = ReadGameCoinData();
+    game_coin.total_coins = play_coins;
+    // TODO: This may introduce potential race condition if the game is reading the
+    // game coin data at the same time
+    WriteGameCoinData(game_coin);
 }
 
 Module::Interface::Interface(std::shared_ptr<Module> ptm, const char* name, u32 max_session)

src/core/hle/service/ptm/ptm.h

@@ -42,6 +42,8 @@ void CheckNew3DS(IPC::RequestBuilder& rb);
 class Module final {
 public:
     Module();
+    static u16 GetPlayCoins();
+    static void SetPlayCoins(u16 play_coins);
 
     class Interface : public ServiceFramework<Interface> {
     public: