Added retry support to AttestationPolicyObserver.

chrome/browser/chromeos/attestation/attestation_policy_observer.h

 // 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.
 
 #ifndef CHROME_BROWSER_CHROMEOS_ATTESTATION_ATTESTATION_POLICY_OBSERVER_H_
 #define CHROME_BROWSER_CHROMEOS_ATTESTATION_ATTESTATION_POLICY_OBSERVER_H_
 
 #include <string>
 
 #include "base/basictypes.h"
@@ skipping 32 matching lines @@
                             CryptohomeClient* cryptohome_client,
                             AttestationFlow* attestation_flow);
 
   virtual ~AttestationPolicyObserver();
 
   // content::NotificationObserver:
   virtual void Observe(int type,
                        const content::NotificationSource& source,
                        const content::NotificationDetails& details) OVERRIDE;
 
+  // Sets the retry delay in seconds; useful in testing.
+  void set_retry_delay(int retry_delay) {
+    retry_delay_ = retry_delay;
+  }
+
  private:
   // Checks attestation policy and starts any necessary work.
   void Start();
 
   // Gets a new certificate for the Enterprise Machine Key (EMK).
   void GetNewCertificate();
 
   // Gets the existing EMK certificate and sends it to CheckCertificateExpiry.
   void GetExistingCertificate();
 
@@ skipping 10 matching lines @@
   // Gets the payload associated with the EMK and sends it to |callback|.
   void GetKeyPayload(base::Callback<void(const std::string&)> callback);
 
   // Called when a certificate upload operation completes.  On success, |status|
   // will be true.
   void OnUploadComplete(bool status);
 
   // Marks a key as uploaded in the payload proto.
   void MarkAsUploaded(const std::string& key_payload);
 
+  // Reschedules a policy check (i.e. a call to Start) for a later time.
+  // TODO(dkrahn): A better solution would be to wait for a dbus signal which
+  // indicates the system is ready to process this task. See crbug.com/256845.
+  void Reschedule();
+
   CrosSettings* cros_settings_;
   policy::CloudPolicyClient* policy_client_;
   CryptohomeClient* cryptohome_client_;
   AttestationFlow* attestation_flow_;
   scoped_ptr<AttestationFlow> default_attestation_flow_;
+  int num_retries_;
+  int retry_delay_;
 
   // Note: This should remain the last member so it'll be destroyed and
   // invalidate the weak pointers before any other members are destroyed.
   base::WeakPtrFactory<AttestationPolicyObserver> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(AttestationPolicyObserver);
 };
 
 }  // namespace attestation
 }  // namespace chromeos
 
 #endif  // CHROME_BROWSER_CHROMEOS_ATTESTATION_ATTESTATION_POLICY_OBSERVER_H_