Add a header when fetching pages under the DMServer URL.

components/policy/core/browser/policy_header_io_helper.cc

+// Copyright (c) 2013 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/policy/core/browser/policy_header_io_helper.h"
+
+#include "base/bind.h"
+#include "base/location.h"
+#include "base/sequenced_task_runner.h"
+#include "net/url_request/url_request.h"
+
+namespace {
+
+// The name of the header containing the policy information.
+const char kChromePolicyHeader[] = "Chrome-Policy-Posture";
+
+}  // namespace
+
+namespace policy {
+
+PolicyHeaderIOHelper::PolicyHeaderIOHelper(
+    const std::string& server_url,
+    const std::string& initial_header_value,
+    scoped_refptr<base::SequencedTaskRunner> task_runner)
+    : server_url_(server_url),
+      task_runner_(task_runner),
+      policy_header_(initial_header_value),
+      weak_factory_(this) {
+}
+
+PolicyHeaderIOHelper::~PolicyHeaderIOHelper() {
+}
+
+// Sets any necessary policy headers on the passed request.
+void PolicyHeaderIOHelper::AddPolicyHeaders(net::URLRequest* request) const {
+  DCHECK(task_runner_->RunsTasksOnCurrentThread());
+  const GURL& url = request->url();
+  if (!policy_header_.empty() &&
+      url.spec().compare(0, server_url_.size(), server_url_) == 0) {

[Joao da Silva, 2013/12/09 14:46:21]

I guess this works for our purposes. I wonder if this will correctly match
"https://m.google.com:443/devicemanagement/api" though?

[Andrew T Wilson (Slow), 2013/12/10 07:15:56]

I don't think we care - I'm OK with requiring specific URL formats for this
mechanism to work. If you configure your extension and/or SAML endpoint to try
to contact m.google.com:443 or directly via the IP address or via some other
method, Chrome won't report policy, and that's OK.

+    request->SetExtraRequestHeaderByName(kChromePolicyHeader,
+                                         policy_header_,
+                                         true /* overwrite */);
+  }
+}
+
+void PolicyHeaderIOHelper::UpdateHeaderFromUI(const std::string& new_header) {
+  // Post a task to the IO thread to modify this.
+  task_runner_->PostTask(FROM_HERE,
+                        base::Bind(&PolicyHeaderIOHelper::UpdateHeaderOnIO,
+                                   weak_factory_.GetWeakPtr(), new_header));
+}
+
+void PolicyHeaderIOHelper::UpdateHeaderOnIO(const std::string& new_header) {
+  DCHECK(task_runner_->RunsTasksOnCurrentThread());
+  policy_header_ = new_header;
+}
+
+}  // namespace policy