mirror of
				https://git.zaroz.cloud/nintendo-back-up/yuzu/yuzu-mainline.git
				synced 2025-03-21 01:53:15 +00:00 
			
		
		
		
	key_manager: Add support for autogenerated keys
Stored in a separate file than manual keys.
This commit is contained in:
		
							parent
							
								
									f26fc64cb4
								
							
						
					
					
						commit
						61a5b56abd
					
				| @ -153,17 +153,17 @@ void KeyManager::LoadFromFile(const std::string& filename, bool is_title_keys) { | |||||||
|             u128 rights_id{}; |             u128 rights_id{}; | ||||||
|             std::memcpy(rights_id.data(), rights_id_raw.data(), rights_id_raw.size()); |             std::memcpy(rights_id.data(), rights_id_raw.data(), rights_id_raw.size()); | ||||||
|             Key128 key = Common::HexStringToArray<16>(out[1]); |             Key128 key = Common::HexStringToArray<16>(out[1]); | ||||||
|             SetKey(S128KeyType::Titlekey, key, rights_id[1], rights_id[0]); |             s128_keys[{S128KeyType::Titlekey, rights_id[1], rights_id[0]}] = key; | ||||||
|         } else { |         } else { | ||||||
|             std::transform(out[0].begin(), out[0].end(), out[0].begin(), ::tolower); |             std::transform(out[0].begin(), out[0].end(), out[0].begin(), ::tolower); | ||||||
|             if (s128_file_id.find(out[0]) != s128_file_id.end()) { |             if (s128_file_id.find(out[0]) != s128_file_id.end()) { | ||||||
|                 const auto index = s128_file_id.at(out[0]); |                 const auto index = s128_file_id.at(out[0]); | ||||||
|                 Key128 key = Common::HexStringToArray<16>(out[1]); |                 Key128 key = Common::HexStringToArray<16>(out[1]); | ||||||
|                 SetKey(index.type, key, index.field1, index.field2); |                 s128_keys[{index.type, index.field1, index.field2}] = key; | ||||||
|             } else if (s256_file_id.find(out[0]) != s256_file_id.end()) { |             } else if (s256_file_id.find(out[0]) != s256_file_id.end()) { | ||||||
|                 const auto index = s256_file_id.at(out[0]); |                 const auto index = s256_file_id.at(out[0]); | ||||||
|                 Key256 key = Common::HexStringToArray<32>(out[1]); |                 Key256 key = Common::HexStringToArray<32>(out[1]); | ||||||
|                 SetKey(index.type, key, index.field1, index.field2); |                 s256_keys[{index.type, index.field1, index.field2}] = key; | ||||||
|             } |             } | ||||||
|         } |         } | ||||||
|     } |     } | ||||||
| @ -197,11 +197,51 @@ Key256 KeyManager::GetKey(S256KeyType id, u64 field1, u64 field2) const { | |||||||
|     return s256_keys.at({id, field1, field2}); |     return s256_keys.at({id, field1, field2}); | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
|  | template <size_t Size> | ||||||
|  | void KeyManager::WriteKeyToFile(bool title_key, std::string_view keyname, | ||||||
|  |                                 std::array<u8, Size> key) { | ||||||
|  |     const std::string yuzu_keys_dir = FileUtil::GetUserPath(FileUtil::UserPath::KeysDir); | ||||||
|  |     std::string filename = "title.keys_autogenerated"; | ||||||
|  |     if (!title_key) | ||||||
|  |         filename = dev_mode ? "dev.keys_autogenerated" : "prod.keys_autogenerated"; | ||||||
|  |     const auto add_info_text = !FileUtil::Exists(yuzu_keys_dir + DIR_SEP + filename); | ||||||
|  |     std::ofstream file(yuzu_keys_dir + DIR_SEP + filename, std::ios::app); | ||||||
|  |     if (!file.is_open()) | ||||||
|  |         return; | ||||||
|  |     if (add_info_text) { | ||||||
|  |         file << "# This file is autogenerated by Yuzu" << std::endl | ||||||
|  |              << "# It serves to store keys that were automatically generated from the normal keys" | ||||||
|  |              << std::endl | ||||||
|  |              << "# If you are experiencing issues involving keys, it may help to delete this file" | ||||||
|  |              << std::endl; | ||||||
|  |     } | ||||||
|  | 
 | ||||||
|  |     file << std::endl | ||||||
|  |          << fmt::format("{} = {}", keyname, Common::HexArrayToString(key)) << std::endl; | ||||||
|  |     AttemptLoadKeyFile(yuzu_keys_dir, yuzu_keys_dir, filename, title_key); | ||||||
|  | } | ||||||
|  | 
 | ||||||
| void KeyManager::SetKey(S128KeyType id, Key128 key, u64 field1, u64 field2) { | void KeyManager::SetKey(S128KeyType id, Key128 key, u64 field1, u64 field2) { | ||||||
|  |     const auto iter = std::find_if( | ||||||
|  |         s128_file_id.begin(), s128_file_id.end(), | ||||||
|  |         [&id, &field1, &field2](const std::pair<std::string, KeyIndex<S128KeyType>> elem) { | ||||||
|  |             return std::tie(elem.second.type, elem.second.field1, elem.second.field2) == | ||||||
|  |                    std::tie(id, field1, field2); | ||||||
|  |         }); | ||||||
|  |     if (iter != s128_file_id.end()) | ||||||
|  |         WriteKeyToFile(id == S128KeyType::Titlekey, iter->first, key); | ||||||
|     s128_keys[{id, field1, field2}] = key; |     s128_keys[{id, field1, field2}] = key; | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
| void KeyManager::SetKey(S256KeyType id, Key256 key, u64 field1, u64 field2) { | void KeyManager::SetKey(S256KeyType id, Key256 key, u64 field1, u64 field2) { | ||||||
|  |     const auto iter = std::find_if( | ||||||
|  |         s256_file_id.begin(), s256_file_id.end(), | ||||||
|  |         [&id, &field1, &field2](const std::pair<std::string, KeyIndex<S256KeyType>> elem) { | ||||||
|  |             return std::tie(elem.second.type, elem.second.field1, elem.second.field2) == | ||||||
|  |                    std::tie(id, field1, field2); | ||||||
|  |         }); | ||||||
|  |     if (iter != s256_file_id.end()) | ||||||
|  |         WriteKeyToFile(false, iter->first, key); | ||||||
|     s256_keys[{id, field1, field2}] = key; |     s256_keys[{id, field1, field2}] = key; | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
|  | |||||||
| @ -106,6 +106,8 @@ private: | |||||||
|     void LoadFromFile(const std::string& filename, bool is_title_keys); |     void LoadFromFile(const std::string& filename, bool is_title_keys); | ||||||
|     void AttemptLoadKeyFile(const std::string& dir1, const std::string& dir2, |     void AttemptLoadKeyFile(const std::string& dir1, const std::string& dir2, | ||||||
|                             const std::string& filename, bool title); |                             const std::string& filename, bool title); | ||||||
|  |     template <size_t Size> | ||||||
|  |     void WriteKeyToFile(bool title_key, std::string_view keyname, std::array<u8, Size> key); | ||||||
| 
 | 
 | ||||||
|     static const boost::container::flat_map<std::string, KeyIndex<S128KeyType>> s128_file_id; |     static const boost::container::flat_map<std::string, KeyIndex<S128KeyType>> s128_file_id; | ||||||
|     static const boost::container::flat_map<std::string, KeyIndex<S256KeyType>> s256_file_id; |     static const boost::container::flat_map<std::string, KeyIndex<S256KeyType>> s256_file_id; | ||||||
|  | |||||||
		Loading…
	
		Reference in New Issue
	
	Block a user
	 Zach Hilman
						Zach Hilman