[SharedCache] Add parameter to SharedCache::InitializeHeader to check if m_exportInfos was modified

This probably makes more sense than the current solution of using execution of the callback parameter to determine if `m_exportInfos` was modified.

Author: WeiN76LQh

view/sharedcache/core/SharedCache.cpp

@@ -2808,18 +2808,24 @@ std::vector<Ref<Symbol>> SharedCache::ParseExportTrie(std::shared_ptr<MMappedFil
 }
 
 
-std::vector<std::pair<uint64_t, std::pair<BNSymbolType, std::string>>> SharedCache::GetExportListForHeader(SharedCacheMachOHeader header, std::function<std::shared_ptr<MMappedFileAccessor>()> provideLinkeditFile)
+std::vector<std::pair<uint64_t, std::pair<BNSymbolType, std::string>>> SharedCache::GetExportListForHeader(SharedCacheMachOHeader header, std::function<std::shared_ptr<MMappedFileAccessor>()> provideLinkeditFile, bool* didModifyExportList)
 {
 	if (auto it = m_state->exportInfos.find(header.textBase); it != m_state->exportInfos.end())
 	{
+		if (didModifyExportList)
+			*didModifyExportList = false;
 		return it->second;
 	}
 	else
 	{
 		// TODO does this have to be a functor? can't we just pass the accessor? if not, why?
 		std::shared_ptr<MMappedFileAccessor> linkeditFile = provideLinkeditFile();
 		if (!linkeditFile)
+		{
+			if (didModifyExportList)
+				*didModifyExportList = false;
 			return std::vector<std::pair<uint64_t, std::pair<BNSymbolType, std::string>>>();
+		}
 
 		auto exportList = SharedCache::ParseExportTrie(linkeditFile, header);
 		std::vector<std::pair<uint64_t, std::pair<BNSymbolType, std::string>>> exportMapping(exportList.size());
@@ -2828,6 +2834,8 @@ std::vector<std::pair<uint64_t, std::pair<BNSymbolType, std::string>>> SharedCac
 			exportMapping.push_back({sym->GetAddress(), {sym->GetType(), sym->GetRawName()}});
 		}
 		m_state->exportInfos[header.textBase] = exportMapping;
+		if (didModifyExportList)
+			*didModifyExportList = true;
 		return exportMapping;
 	}
 }
@@ -2858,15 +2866,14 @@ std::vector<std::pair<std::string, Ref<Symbol>>> SharedCache::LoadAllSymbolsAndW
 		auto exportList = GetExportListForHeader(*header, [&]() {
 			try {
 				auto mapping = MMappedFileAccessor::Open(m_dscView, m_dscView->GetFile()->GetSessionId(), header->exportTriePath)->lock();
-				doSave = true;
 				return mapping;
 			}
 			catch (...)
 			{
 				m_logger->LogWarn("Serious Error: Failed to open export trie %s for %s", header->exportTriePath.c_str(), header->installName.c_str());
 				return std::shared_ptr<MMappedFileAccessor>(nullptr);
 			}
-		});
+		}, &doSave);
 		for (const auto& sym : exportList)
 		{
 			symbols.push_back({img.installName, new Symbol(sym.second.first, sym.second.second, sym.first)});

view/sharedcache/core/SharedCache.h

@@ -643,7 +643,8 @@ namespace SharedCacheCore {
 			std::shared_ptr<MMappedFileAccessor> linkeditFile, const SharedCacheMachOHeader& header);
 
 		std::vector<std::pair<uint64_t, std::pair<BNSymbolType, std::string>>> GetExportListForHeader(
-			SharedCacheMachOHeader header, std::function<std::shared_ptr<MMappedFileAccessor>()> provideLinkeditFile);
+			SharedCacheMachOHeader header, std::function<std::shared_ptr<MMappedFileAccessor>()> provideLinkeditFile,
+			bool* didModifyExportList = nullptr);
 
 		Ref<TypeLibrary> TypeLibraryForImage(const std::string& installName);