diff --git a/init/persistent_properties.cpp b/init/persistent_properties.cpp index 8efb72c7ea94..e89244f13be0 100644 --- a/init/persistent_properties.cpp +++ b/init/persistent_properties.cpp @@ -115,24 +115,6 @@ void RemoveLegacyPersistentPropertyFiles() { } } -PersistentProperties LoadPersistentPropertiesFromMemory() { - PersistentProperties persistent_properties; - __system_property_foreach( - [](const prop_info* pi, void* cookie) { - __system_property_read_callback( - pi, - [](void* cookie, const char* name, const char* value, unsigned serial) { - if (StartsWith(name, "persist.")) { - auto properties = reinterpret_cast(cookie); - AddPersistentProperty(name, value, properties); - } - }, - cookie); - }, - &persistent_properties); - return persistent_properties; -} - Result ReadPersistentPropertyFile() { const std::string temp_filename = persistent_property_filename + ".tmp"; if (access(temp_filename.c_str(), F_OK) == 0) { @@ -221,6 +203,24 @@ Result WritePersistentPropertyFile(const PersistentProperties& persistent_ return {}; } +PersistentProperties LoadPersistentPropertiesFromMemory() { + PersistentProperties persistent_properties; + __system_property_foreach( + [](const prop_info* pi, void* cookie) { + __system_property_read_callback( + pi, + [](void* cookie, const char* name, const char* value, unsigned serial) { + if (StartsWith(name, "persist.")) { + auto properties = reinterpret_cast(cookie); + AddPersistentProperty(name, value, properties); + } + }, + cookie); + }, + &persistent_properties); + return persistent_properties; +} + // Persistent properties are not written often, so we rather not keep any data in memory and read // then rewrite the persistent property file for each update. void WritePersistentProperty(const std::string& name, const std::string& value) { diff --git a/init/persistent_properties.h b/init/persistent_properties.h index a6f80e64f51d..7e9d4387a602 100644 --- a/init/persistent_properties.h +++ b/init/persistent_properties.h @@ -29,6 +29,7 @@ void AddPersistentProperty(const std::string& name, const std::string& value, PersistentProperties* persistent_properties); PersistentProperties LoadPersistentProperties(); void WritePersistentProperty(const std::string& name, const std::string& value); +PersistentProperties LoadPersistentPropertiesFromMemory(); // Exposed only for testing Result LoadPersistentPropertyFile(); diff --git a/init/property_service.cpp b/init/property_service.cpp index c97850eb9b34..64f979f24ee6 100644 --- a/init/property_service.cpp +++ b/init/property_service.cpp @@ -1419,8 +1419,6 @@ static void HandleInitSocket() { // Apply staged and persistent properties bool has_staged_prop = false; auto const staged_prefix = std::string_view("next_boot."); - auto const staged_persist_prefix = std::string_view("next_boot.persist."); - auto persist_props_map = std::unordered_map(); auto persistent_properties = LoadPersistentProperties(); for (const auto& property_record : persistent_properties.properties()) { @@ -1431,23 +1429,16 @@ static void HandleInitSocket() { has_staged_prop = true; auto actual_prop_name = prop_name.substr(staged_prefix.size()); InitPropertySet(actual_prop_name, prop_value); - if (StartsWith(prop_name, staged_persist_prefix)) { - persist_props_map[actual_prop_name] = prop_value; - } - } else if (!persist_props_map.count(prop_name)) { + } else { InitPropertySet(prop_name, prop_value); } } // Update persist prop file if there are staged props if (has_staged_prop) { - PersistentProperties updated_persist_props; - for (auto const& [prop_name, prop_value] : persist_props_map) { - AddPersistentProperty(prop_name, prop_value, &updated_persist_props); - } - + PersistentProperties props = LoadPersistentPropertiesFromMemory(); // write current updated persist prop file - auto result = WritePersistentPropertyFile(updated_persist_props); + auto result = WritePersistentPropertyFile(props); if (!result.ok()) { LOG(ERROR) << "Could not store persistent property: " << result.error(); }