Persist broken and recently-broken alt-svcs to prefs in HttpServerPropertiesManager

net/http/http_server_properties_manager.cc

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "net/http/http_server_properties_manager.h"
 
 #include <utility>
 
 #include "base/bind.h"
 #include "base/memory/ptr_util.h"
@@ skipping 31 matching lines @@
 
 // Persist 200 MRU AlternateProtocolHostPortPairs.
 const int kMaxAlternateProtocolHostsToPersist = 200;
 
 // Persist 300 MRU SupportsSpdyServerHostPortPairs.
 const int kMaxSupportsSpdyServerHostsToPersist = 300;
 
 // Persist 200 ServerNetworkStats.
 const int kMaxServerNetworkStatsHostsToPersist = 200;
 
+// Persist 200 BrokenAlternativeServices
+const int kMaxBrokenAlternativeServicesToPersist = 200;
+
+// Persist 200 RecentlyBrokenAlternativeServices
+const int kMaxRecentlyBrokenAlternativeServicesToPersist = 200;
+
 const char kVersionKey[] = "version";
 const char kServersKey[] = "servers";
 const char kSupportsSpdyKey[] = "supports_spdy";
 const char kSupportsQuicKey[] = "supports_quic";
 const char kQuicServers[] = "quic_servers";
 const char kServerInfoKey[] = "server_info";
 const char kUsedQuicKey[] = "used_quic";
 const char kAddressKey[] = "address";
 const char kAlternativeServiceKey[] = "alternative_service";
 const char kProtocolKey[] = "protocol_str";
 const char kHostKey[] = "host";
 const char kPortKey[] = "port";
 const char kExpirationKey[] = "expiration";
 const char kNetworkStatsKey[] = "network_stats";
 const char kSrttKey[] = "srtt";
+const char kBrokenAlternativeServicesKey[] = "broken_alternative_services";
+const char kRecentlyBrokenAlternativeServicesKey[] =
+    "recently_broken_alternative_services";
 
 }  // namespace
 
 ////////////////////////////////////////////////////////////////////////////////
 //  HttpServerPropertiesManager
 
 HttpServerPropertiesManager::PrefDelegate::~PrefDelegate() {}
 
 HttpServerPropertiesManager::HttpServerPropertiesManager(
     PrefDelegate* pref_delegate,
     scoped_refptr<base::SingleThreadTaskRunner> pref_task_runner,
     scoped_refptr<base::SingleThreadTaskRunner> network_task_runner)
     : pref_task_runner_(std::move(pref_task_runner)),
       pref_delegate_(pref_delegate),
       setting_prefs_(false),
+      clock_(&default_clock_),
       is_initialized_(false),
       network_task_runner_(std::move(network_task_runner)) {
   DCHECK(pref_task_runner_->RunsTasksInCurrentSequence());
   DCHECK(pref_delegate_);
   pref_weak_ptr_factory_.reset(
       new base::WeakPtrFactory<HttpServerPropertiesManager>(this));
   pref_weak_ptr_ = pref_weak_ptr_factory_->GetWeakPtr();
   pref_cache_update_timer_.reset(new base::OneShotTimer);
   pref_cache_update_timer_->SetTaskRunner(pref_task_runner_);
   pref_delegate_->StartListeningForUpdates(
@@ skipping 415 matching lines @@
         detected_corrupted_prefs = true;
       }
     }
   }
 
   if (!AddToQuicServerInfoMap(http_server_properties_dict,
                               quic_server_info_map.get())) {
     detected_corrupted_prefs = true;
   }
 
+  std::unique_ptr<RecentlyBrokenAlternativeServices>
+      recently_broken_alternative_services;
+  const base::ListValue* recently_broken_alt_svc_list = nullptr;
+
+  std::unique_ptr<BrokenAlternativeServiceList> broken_alternative_service_list;
+  const base::ListValue* broken_alt_svc_list = nullptr;
+
+  // Read recently broken alternative services if they exist.
+  if (http_server_properties_dict.GetListWithoutPathExpansion(
+          kRecentlyBrokenAlternativeServicesKey,
+          &recently_broken_alt_svc_list)) {
+    recently_broken_alternative_services.reset(
+        new RecentlyBrokenAlternativeServices(
+            RecentlyBrokenAlternativeServices::NO_AUTO_EVICT));
+
+    for (auto it = recently_broken_alt_svc_list->begin();
+         it != recently_broken_alt_svc_list->end(); ++it) {
+      const base::DictionaryValue* entry_dict = nullptr;
+      if (!it->GetAsDictionary(&entry_dict) ||
+          !AddRecentlyBrokenAlternativeServiceData(
+              entry_dict, recently_broken_alternative_services.get())) {
+        DVLOG(1) << "Malformed http_server_properties for recently broken "
+                 << "alternative services list.";
+        detected_corrupted_prefs = true;
+      }
+    }
+
+    // Read broken alternative services if they exist
+    if (http_server_properties_dict.GetListWithoutPathExpansion(
+            kBrokenAlternativeServicesKey, &broken_alt_svc_list)) {
+      broken_alternative_service_list.reset(new BrokenAlternativeServiceList());
+
+      for (auto it = broken_alt_svc_list->begin();
+           it != broken_alt_svc_list->end(); ++it) {
+        const base::DictionaryValue* entry_dict = nullptr;
+        AlternativeService alternative_service;
+        base::TimeTicks expiration_time_ticks;
+        if (!it->GetAsDictionary(&entry_dict) ||
+            !ExtractBrokenAlternativeServiceData(
+                entry_dict, &alternative_service, &expiration_time_ticks)) {
+          DVLOG(1) << "Malformed http_server_properties for broken alternative "
+                   << "services list.";
+          detected_corrupted_prefs = true;
+          continue;
+        }
+        if (recently_broken_alternative_services->Peek(alternative_service) !=
+            recently_broken_alternative_services->end()) {
+          broken_alternative_service_list->push_back(
+              std::make_pair(alternative_service, expiration_time_ticks));
+        } else {
+          DVLOG(1) << "Broken alternative service does not have an entry in "
+                   << "map of recently broken alternative services. "
+                   << "Will not be added.";
+        }
+      }
+    }
+  }
+
   network_task_runner_->PostTask(
       FROM_HERE,
       base::Bind(
           &HttpServerPropertiesManager::UpdateCacheFromPrefsOnNetworkSequence,
           base::Unretained(this), base::Owned(spdy_servers.release()),
           base::Owned(alternative_service_map.release()), base::Owned(addr),
           base::Owned(server_network_stats_map.release()),
           base::Owned(quic_server_info_map.release()),
+          base::Passed(&broken_alternative_service_list),
+          base::Passed(&recently_broken_alternative_services),
           detected_corrupted_prefs));
 }
 
+bool HttpServerPropertiesManager::ExtractBrokenAlternativeServiceData(
+    const base::DictionaryValue* broken_alternative_service_dict,
+    AlternativeService* alternative_service,
+    base::TimeTicks* expiration_time_ticks) {
+  DCHECK(broken_alternative_service_dict);
+
+  if (broken_alternative_service_dict->size() != 1u)
+    return false;
+
+  base::DictionaryValue::Iterator it(*broken_alternative_service_dict);
+
+  const std::string& alternative_service_string = it.key();
+  if (!AlternativeService::FromString(alternative_service_string,
+                                      alternative_service))
+    return false;
+
+  int64_t expiration_int64;
+  std::string time_ticks_string;
+  if (!it.value().GetAsString(&time_ticks_string))
+    return false;
+  if (!base::StringToInt64(time_ticks_string, &expiration_int64))
+    return false;
+  time_t expiration_time_t = static_cast<time_t>(expiration_int64);
+  // Convert expiration from time_t to Time to TimeTicks
+  *expiration_time_ticks =
+      clock_->NowTicks() +
+      (base::Time::FromTimeT(expiration_time_t) - base::Time::Now());
+
+  return true;
+}
+
+bool HttpServerPropertiesManager::AddRecentlyBrokenAlternativeServiceData(
+    const base::DictionaryValue* recently_broken_alternative_service_dict,
+    RecentlyBrokenAlternativeServices* recently_broken_alternative_services) {
+  DCHECK(recently_broken_alternative_service_dict);
+
+  if (recently_broken_alternative_service_dict->size() != 1u)
+    return false;
+
+  base::DictionaryValue::Iterator it(*recently_broken_alternative_service_dict);
+
+  AlternativeService alt_service;
+  const std::string& alt_service_string = it.key();
+  if (!AlternativeService::FromString(alt_service_string, &alt_service))
+    return false;
+
+  int broken_count;
+  if (!it.value().GetAsInteger(&broken_count))
+    return false;
+
+  recently_broken_alternative_services->Put(alt_service, broken_count);
+  return true;
+}
+
 bool HttpServerPropertiesManager::AddServersData(
     const base::DictionaryValue& servers_dict,
     ServerList* spdy_servers,
     AlternativeServiceMap* alternative_service_map,
     ServerNetworkStatsMap* network_stats_map,
     int version) {
   for (base::DictionaryValue::Iterator it(servers_dict); !it.IsAtEnd();
        it.Advance()) {
     // Get server's scheme/host/pair.
     const std::string& server_str = it.key();
@@ skipping 235 matching lines @@
   }
   return !detected_corrupted_prefs;
 }
 
 void HttpServerPropertiesManager::UpdateCacheFromPrefsOnNetworkSequence(
     ServerList* spdy_servers,
     AlternativeServiceMap* alternative_service_map,
     IPAddress* last_quic_address,
     ServerNetworkStatsMap* server_network_stats_map,
     QuicServerInfoMap* quic_server_info_map,
+    std::unique_ptr<BrokenAlternativeServiceList>
+        broken_alternative_service_list,
+    std::unique_ptr<RecentlyBrokenAlternativeServices>
+        recently_broken_alternative_services,
     bool detected_corrupted_prefs) {
   // Preferences have the master data because admins might have pushed new
   // preferences. Update the cached data with new data from preferences.
   DCHECK(network_task_runner_->RunsTasksInCurrentSequence());
 
   UMA_HISTOGRAM_COUNTS("Net.CountOfSpdyServers", spdy_servers->size());
   http_server_properties_impl_->SetSpdyServers(spdy_servers, true);
 
   // Update the cached data and use the new alternative service list from
   // preferences.
   UMA_HISTOGRAM_COUNTS("Net.CountOfAlternateProtocolServers",
                        alternative_service_map->size());
   http_server_properties_impl_->SetAlternativeServiceServers(
       alternative_service_map);
 
   http_server_properties_impl_->SetSupportsQuic(last_quic_address);
 
   http_server_properties_impl_->SetServerNetworkStats(server_network_stats_map);
 
   UMA_HISTOGRAM_COUNTS_1000("Net.CountOfQuicServerInfos",
                             quic_server_info_map->size());
 
   http_server_properties_impl_->SetQuicServerInfoMap(quic_server_info_map);
 
+  if (recently_broken_alternative_services) {
+    DCHECK(broken_alternative_service_list);
+    http_server_properties_impl_->AddBrokenAndRecentlyBrokenAlternativeServices(
+        std::move(broken_alternative_service_list),
+        std::move(recently_broken_alternative_services));
+  }
+
   // Update the prefs with what we have read (delete all corrupted prefs).
   if (detected_corrupted_prefs)
     ScheduleUpdatePrefsOnNetworkSequence(DETECTED_CORRUPTED_PREFS);
 }
 
 //
 // Update Preferences with data from the cached data.
 //
 void HttpServerPropertiesManager::ScheduleUpdatePrefsOnNetworkSequence(
     Location location) {
@@ skipping 45 matching lines @@
         continue;
       }
       AlternativeService alternative_service(
           alternative_service_info.alternative_service);
       if (!IsAlternateProtocolValid(alternative_service.protocol)) {
         continue;
       }
       if (alternative_service.host.empty()) {
         alternative_service.host = server.host();
       }
-      if (IsAlternativeServiceBroken(alternative_service)) {
-        continue;
-      }
       notbroken_alternative_service_info_vector.push_back(
           alternative_service_info);
     }
     if (notbroken_alternative_service_info_vector.empty()) {
       continue;
     }
     const std::string* canonical_suffix =
         http_server_properties_impl_->GetCanonicalSuffix(server.host());
     if (canonical_suffix != nullptr) {
       if (persisted_map.find(*canonical_suffix) != persisted_map.end())
@@ skipping 23 matching lines @@
       http_server_properties_impl_->quic_server_info_map();
   if (main_quic_server_info_map.size() > 0) {
     quic_server_info_map =
         new QuicServerInfoMap(max_server_configs_stored_in_properties());
     for (const std::pair<const QuicServerId, std::string>& entry :
          main_quic_server_info_map) {
       quic_server_info_map->Put(entry.first, entry.second);
     }
   }
 
+  // Make copy of list of broken alternative services stored in
+  // |http_server_properties_impl_|
+  BrokenAlternativeServiceList* broken_alt_svc_list_copy = nullptr;
+  const BrokenAlternativeServiceList& broken_alternative_service_list =
+      http_server_properties_impl_->broken_alternative_service_list();
+  if (broken_alternative_service_list.size() > 0) {
+    broken_alt_svc_list_copy = new BrokenAlternativeServiceList();
+    size_t count = 0;
+    for (auto it = broken_alternative_service_list.begin();
+         it != broken_alternative_service_list.end() &&
+         count < kMaxBrokenAlternativeServicesToPersist;
+         ++it) {
+      broken_alt_svc_list_copy->push_back(*it);
+      ++count;
+    }
+  }
+
+  // Make copy of RecentlyBrokenAternativeServices stored in
+  // |http_server_properties_impl_|.
+  RecentlyBrokenAlternativeServices* recently_broken_alt_svc_copy = nullptr;
+  const RecentlyBrokenAlternativeServices& recently_broken_alt_services =
+      http_server_properties_impl_->recently_broken_alternative_services();
+  if (recently_broken_alt_services.size() > 0) {
+    recently_broken_alt_svc_copy = new RecentlyBrokenAlternativeServices(
+        kMaxRecentlyBrokenAlternativeServicesToPersist);
+    size_t count = 0;
+    for (auto it = recently_broken_alt_services.rbegin();
+         it != recently_broken_alt_services.rend() &&
+         count < kMaxRecentlyBrokenAlternativeServicesToPersist;
+         ++it) {
+      recently_broken_alt_svc_copy->Put(it->first, it->second);
+      ++count;
+    }
+  }
+
   IPAddress* last_quic_addr = new IPAddress;
   http_server_properties_impl_->GetSupportsQuic(last_quic_addr);
   // Update the preferences on the pref thread.
   pref_task_runner_->PostTask(
       FROM_HERE,
       base::Bind(
           &HttpServerPropertiesManager::UpdatePrefsOnPrefThread, pref_weak_ptr_,
           base::Owned(spdy_server_list), base::Owned(alternative_service_map),
           base::Owned(last_quic_addr), base::Owned(server_network_stats_map),
-          base::Owned(quic_server_info_map), completion));
+          base::Owned(quic_server_info_map),
+          base::Owned(broken_alt_svc_list_copy),
+          base::Owned(recently_broken_alt_svc_copy), completion));
 }
 
 // A local or temporary data structure to hold |supports_spdy|, SpdySettings,
 // AlternativeServiceInfoVector, and SupportsQuic preferences for a server. This
 // is used only in UpdatePrefsOnPrefThread.
 struct ServerPref {
   ServerPref()
       : supports_spdy(false),
         settings_map(nullptr),
         alternative_service_info_vector(nullptr),
@@ skipping 17 matching lines @@
   const ServerNetworkStats* server_network_stats;
 };
 
 // All maps and lists are in MRU order.
 void HttpServerPropertiesManager::UpdatePrefsOnPrefThread(
     base::ListValue* spdy_server_list,
     AlternativeServiceMap* alternative_service_map,
     IPAddress* last_quic_address,
     ServerNetworkStatsMap* server_network_stats_map,
     QuicServerInfoMap* quic_server_info_map,
+    BrokenAlternativeServiceList* broken_alternative_service_list,
+    RecentlyBrokenAlternativeServices* recently_broken_alternative_services,
     const base::Closure& completion) {
   typedef base::MRUCache<url::SchemeHostPort, ServerPref> ServerPrefMap;
   ServerPrefMap server_pref_map(ServerPrefMap::NO_AUTO_EVICT);
 
   DCHECK(pref_task_runner_->RunsTasksInCurrentSequence());
 
   // Add servers that support spdy to server_pref_map in the MRU order.
   for (size_t index = spdy_server_list->GetSize(); index > 0; --index) {
     std::string server_str;
     if (spdy_server_list->GetString(index - 1, &server_str)) {
@@ skipping 66 matching lines @@
 
   http_server_properties_dict.SetWithoutPathExpansion(kServersKey,
                                                       std::move(servers_list));
   SetVersion(&http_server_properties_dict, kVersionNumber);
 
   SaveSupportsQuicToPrefs(last_quic_address, &http_server_properties_dict);
 
   SaveQuicServerInfoMapToServerPrefs(quic_server_info_map,
                                      &http_server_properties_dict);
 
+  SaveBrokenAlternativeServiceListToPrefs(broken_alternative_service_list,
+                                          &http_server_properties_dict);
+
+  SaveRecentlyBrokenAlternativeServicesToPrefs(
+      recently_broken_alternative_services, &http_server_properties_dict);
+
   setting_prefs_ = true;
   pref_delegate_->SetServerProperties(http_server_properties_dict);
   setting_prefs_ = false;
 
   // Note that |completion| will be fired after we have written everything to
   // the Preferences, but likely before these changes are serialized to disk.
   // This is not a problem though, as JSONPrefStore guarantees that this will
   // happen, pretty soon, and even in the case we shut down immediately.
   if (!completion.is_null())
     completion.Run();
@@ skipping 76 matching lines @@
     auto quic_server_pref_dict = base::MakeUnique<base::DictionaryValue>();
     quic_server_pref_dict->SetStringWithoutPathExpansion(kServerInfoKey,
                                                          entry.second);
     quic_servers_dict->SetWithoutPathExpansion(
         server_id.ToString(), std::move(quic_server_pref_dict));
   }
   http_server_properties_dict->SetWithoutPathExpansion(
       kQuicServers, std::move(quic_servers_dict));
 }
 
+void HttpServerPropertiesManager::SaveBrokenAlternativeServiceListToPrefs(
+    const BrokenAlternativeServiceList* broken_alternative_service_list,
+    base::DictionaryValue* http_server_properties_dict) {
+  if (!broken_alternative_service_list)
+    return;
+
+  base::ListValue* broken_alt_services_list = new base::ListValue;
+  for (const auto& entry : *broken_alternative_service_list) {
+    const AlternativeService& alt_service = entry.first;
+    const base::TimeTicks& expiration_time_ticks = entry.second;
+    // Convert expiration from TimeTicks to Time to time_t
+    time_t expiration_time_t =
+        (base::Time::Now() + (expiration_time_ticks - clock_->NowTicks()))
+            .ToTimeT();
+    int64_t expiration_int64 = static_cast<int64_t>(expiration_time_t);
+    std::unique_ptr<base::DictionaryValue> broken_alt_service_entry_dict =
+        base::MakeUnique<base::DictionaryValue>();
+    broken_alt_service_entry_dict->SetStringWithoutPathExpansion(
+        alt_service.ToString(), base::Int64ToString(expiration_int64));
+    broken_alt_services_list->Append(std::move(broken_alt_service_entry_dict));
+  }
+  http_server_properties_dict->SetWithoutPathExpansion(
+      kBrokenAlternativeServicesKey,
+      std::unique_ptr<base::Value>(broken_alt_services_list));
+}
+
+void HttpServerPropertiesManager::SaveRecentlyBrokenAlternativeServicesToPrefs(
+    const RecentlyBrokenAlternativeServices*
+        recently_broken_alternative_services,
+    base::DictionaryValue* http_server_properties_dict) {
+  if (!recently_broken_alternative_services)
+    return;
+
+  // JSON list will have entries ordered from least recent to most recent.
+  base::ListValue* recently_broken_alt_services_list = new base::ListValue;
+  for (auto it = recently_broken_alternative_services->rbegin();
+       it != recently_broken_alternative_services->rend(); ++it) {
+    const AlternativeService& alt_service = it->first;
+    int broken_count = it->second;
+    std::unique_ptr<base::DictionaryValue> broken_alt_service_entry_dict =
+        base::MakeUnique<base::DictionaryValue>();
+    broken_alt_service_entry_dict->SetIntegerWithoutPathExpansion(
+        alt_service.ToString(), broken_count);
+    recently_broken_alt_services_list->Append(
+        std::move(broken_alt_service_entry_dict));
+  }
+  http_server_properties_dict->SetWithoutPathExpansion(
+      kRecentlyBrokenAlternativeServicesKey,
+      std::unique_ptr<base::Value>(recently_broken_alt_services_list));
+}
+
 void HttpServerPropertiesManager::OnHttpServerPropertiesChanged() {
   DCHECK(pref_task_runner_->RunsTasksInCurrentSequence());
   if (!setting_prefs_)
     ScheduleUpdateCacheOnPrefThread();
 }
 
 void HttpServerPropertiesManager::SetInitialized() {
   DCHECK(network_task_runner_->RunsTasksInCurrentSequence());
   is_initialized_ = true;
 }
 
+void HttpServerPropertiesManager::SetTickClockForTesting(
+    base::TickClock* clock) {
+  DCHECK(clock);
+  DCHECK(http_server_properties_impl_);
+  clock_ = clock;
+  http_server_properties_impl_->SetBrokenAlternativeServicesTickClockForTesting(
+      clock);
+}
+
 }  // namespace net