Data reduction proxy sessions last no more than 24 hours

components/data_reduction_proxy/browser/data_reduction_proxy_auth_request_handler.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 "components/data_reduction_proxy/browser/data_reduction_proxy_auth_request_handler.h"
 
 #include "base/bind.h"
 #include "base/command_line.h"
 #include "base/single_thread_task_runner.h"
 #include "base/strings/stringprintf.h"
@@ skipping 81 matching lines @@
     return;
   if (data_reduction_proxy_params_ &&
       data_reduction_proxy_params_->IsDataReductionProxy(
           proxy_server.host_port_pair(), NULL)) {
     AddAuthorizationHeader(request_headers);
   }
 }
 
 void DataReductionProxyAuthRequestHandler::AddAuthorizationHeader(
     net::HttpRequestHeaders* headers) {
+  base::Time now = Now();
+  if (now - last_update_time_ > base::TimeDelta::FromHours(24)) {
+    last_update_time_ = now;
+    ComputeCredentials(last_update_time_, &session_, &credentials_);
+  }
   const char kChromeProxyHeader[] = "Chrome-Proxy";
   std::string header_value;
   if (headers->HasHeader(kChromeProxyHeader)) {
     headers->GetHeader(kChromeProxyHeader, &header_value);
     headers->RemoveHeader(kChromeProxyHeader);
     header_value += ", ";
   }
   header_value +=
       "ps=" + session_ + ", sid=" + credentials_ +  ", v=" + version_;
   if (!client_.empty())
     header_value += ", c=" + client_;
   headers->SetHeader(kChromeProxyHeader, header_value);
 }
 
 void DataReductionProxyAuthRequestHandler::InitAuthenticationOnUI(
     const std::string& key) {
-  key_ = key;
+  network_task_runner_->PostTask(FROM_HERE, base::Bind(
+      &DataReductionProxyAuthRequestHandler::InitAuthentication,
+      base::Unretained(this),
+      key));
+}
+
+void DataReductionProxyAuthRequestHandler::ComputeCredentials(
+    const base::Time& now,
+    std::string* session,
+    std::string* credentials) {
+  DCHECK(session);
+  DCHECK(credentials);
   int64 timestamp =
-      (Now() - base::Time::UnixEpoch()).InMilliseconds() / 1000;
+      (now - base::Time::UnixEpoch()).InMilliseconds() / 1000;
 
   int32 rand[3];
   RandBytes(rand, 3 * sizeof(rand[0]));
-  std::string session = base::StringPrintf("%lld-%u-%u-%u",
-                                           static_cast<long long>(timestamp),
-                                           rand[0],
-                                           rand[1],
-                                           rand[2]);
-  std::string credentials = base::UTF16ToUTF8(AuthHashForSalt(timestamp, key_));
+  *session = base::StringPrintf("%lld-%u-%u-%u",
+                                static_cast<long long>(timestamp),
+                                rand[0],
+                                rand[1],
+                                rand[2]);
+  *credentials = base::UTF16ToUTF8(AuthHashForSalt(timestamp, key_));
 
-  DVLOG(1) << "session: [" << session << "] "
-           << "password: [" << credentials  << "]";
-  network_task_runner_->PostTask(FROM_HERE, base::Bind(
-      &DataReductionProxyAuthRequestHandler::InitAuthentication,
-      base::Unretained(this),
-      session,
-      credentials));
+  DVLOG(1) << "session: [" << *session << "] "
+           << "password: [" << *credentials  << "]";
 }
 
 void DataReductionProxyAuthRequestHandler::InitAuthentication(
-    const std::string& session,
-    const std::string& credentials) {
+    const std::string& key) {
   DCHECK(network_task_runner_->BelongsToCurrentThread());
-  session_ = session;
-  credentials_ = credentials;
+  key_ = key;
+  last_update_time_ = Now();
+  ComputeCredentials(last_update_time_, &session_, &credentials_);
 }
 
 void DataReductionProxyAuthRequestHandler::SetKeyOnUI(const std::string& key) {
   if (!key.empty())
     InitAuthenticationOnUI(key);
 }
 
 
 std::string DataReductionProxyAuthRequestHandler::GetDefaultKey() const {
   const CommandLine& command_line = *CommandLine::ForCurrentProcess();
   std::string key =
     command_line.GetSwitchValueASCII(switches::kDataReductionProxyKey);
 #if defined(SPDY_PROXY_AUTH_VALUE)
   if (key.empty())
     key = SPDY_PROXY_AUTH_VALUE;
 #endif
   return key;
 }
 
 }  // namespace data_reduction_proxy