Fix cloud policy duplicate registrations issue.

chrome/browser/policy/cloud_policy_client.cc

 // Copyright (c) 2012 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 "chrome/browser/policy/cloud_policy_client.h"
 
 #include "base/bind.h"
 #include "base/guid.h"
 #include "base/logging.h"
 #include "base/stl_util.h"
@@ skipping 92 matching lines @@
   if (!client_id.empty())
     request->set_reregister(true);
   request->set_type(type);
   if (!machine_id_.empty())
     request->set_machine_id(machine_id_);
   if (!machine_model_.empty())
     request->set_machine_model(machine_model_);
   if (is_auto_enrollement)
     request->set_auto_enrolled(true);
 
+  request_job_->SetRetryCallback(
+      base::Bind(&CloudPolicyClient::OnRetryRegister, base::Unretained(this)));
+
   request_job_->Start(base::Bind(&CloudPolicyClient::OnRegisterCompleted,
                                  base::Unretained(this)));
 }
 
 void CloudPolicyClient::FetchPolicy() {
   CHECK(is_registered());
   CHECK(!namespaces_to_fetch_.empty());
 
   request_job_.reset(
       service_->CreateJob(DeviceManagementRequestJob::TYPE_POLICY_FETCH));
@@ skipping 77 matching lines @@
 void CloudPolicyClient::RemoveNamespaceToFetch(const PolicyNamespaceKey& key) {
   namespaces_to_fetch_.erase(key);
 }
 
 const em::PolicyFetchResponse* CloudPolicyClient::GetPolicyFor(
     const PolicyNamespaceKey& key) const {
   ResponseMap::const_iterator it = responses_.find(key);
   return it == responses_.end() ? NULL : it->second;
 }
 
+void CloudPolicyClient::OnRetryRegister(DeviceManagementRequestJob* job) {
+  DCHECK_EQ(request_job_.get(), job);
+  // If the initial request managed to get to the server but the response didn't
+  // arrive at the client then retrying with the same client ID will fail.
+  // Set the re-registration flag so that the server accepts it.
+  // If the server hasn't seen the client ID before then it will also accept
+  // the re-registration.
+  job->GetRequest()->mutable_register_request()->set_reregister(true);
+}
+
 void CloudPolicyClient::OnRegisterCompleted(
     DeviceManagementStatus status,
     const em::DeviceManagementResponse& response) {
   if (status == DM_STATUS_SUCCESS &&
       (!response.has_register_response() ||
        !response.register_response().has_device_management_token())) {
     LOG(WARNING) << "Invalid registration response.";
     status = DM_STATUS_RESPONSE_DECODING_ERROR;
   }
 
@@ skipping 84 matching lines @@
 
 void CloudPolicyClient::NotifyRegistrationStateChanged() {
   FOR_EACH_OBSERVER(Observer, observers_, OnRegistrationStateChanged(this));
 }
 
 void CloudPolicyClient::NotifyClientError() {
   FOR_EACH_OBSERVER(Observer, observers_, OnClientError(this));
 }
 
 }  // namespace policy