Send policy blobs to session_manager

chrome/browser/chromeos/login/signed_settings_helper.cc

 // Copyright (c) 2011 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/chromeos/login/signed_settings_helper.h"
 
 #include <string>
 #include <vector>
 
 #include "base/lazy_instance.h"
 #include "base/logging.h"
 #include "base/memory/ref_counted.h"
 #include "chrome/browser/chromeos/login/signed_settings.h"
+#include "chrome/browser/policy/proto/device_management_backend.pb.h"
 #include "content/browser/browser_thread.h"
 
 namespace chromeos {
 
 namespace {
 
 class OpContext {
  public:
   class Delegate {
    public:
@@ skipping 23 matching lines @@
 
   // Cancels the callback.
   void CancelCallback() {
     callback_ = NULL;
   }
 
   // Cancels the callback and cancels the op if it is not executing.
   void Cancel() {
     CancelCallback();
 
-   if (!executing_)
+    if (!executing_)
       OnOpCompleted();
   }
 
   // Accessors.
   SignedSettings* op() const {
     return op_.get();
   }
 
   SignedSettingsHelper::Callback* callback() const {
     return callback_;
@@ skipping 42 matching lines @@
                      const std::string& email,
                      SignedSettingsHelper::Callback* callback,
                      Delegate* delegate)
       : OpContext(callback, delegate),
         type_(type),
         email_(email) {
   }
 
   // chromeos::SignedSettings::Delegate implementation
   virtual void OnSettingsOpCompleted(SignedSettings::ReturnCode code,
-                                     bool value) {
+                                     bool value) OVERRIDE {
     if (callback_) {
       switch (type_) {
         case CHECK:
           callback_->OnCheckWhitelistCompleted(code, email_);
           break;
         case ADD:
           callback_->OnWhitelistCompleted(code, email_);
           break;
         case REMOVE:
           callback_->OnUnwhitelistCompleted(code, email_);
           break;
         default:
           LOG(ERROR) << "Unknown WhitelistOpContext type " << type_;
           break;
       }
     }
     OnOpCompleted();
   }
 
  protected:
   // OpContext implemenetation
-  virtual void CreateOp() {
+  virtual void CreateOp() OVERRIDE {
     switch (type_) {
       case CHECK:
         op_ = SignedSettings::CreateCheckWhitelistOp(email_, this);
         break;
       case ADD:
         op_ = SignedSettings::CreateWhitelistOp(email_, true, this);
         break;
       case REMOVE:
         op_ = SignedSettings::CreateWhitelistOp(email_, false, this);
         break;
@@ skipping 17 matching lines @@
                          const std::string& value,
                          SignedSettingsHelper::Callback* callback,
                          Delegate* delegate)
       : OpContext(callback, delegate),
         name_(name),
         value_(value) {
   }
 
   // chromeos::SignedSettings::Delegate implementation
   virtual void OnSettingsOpCompleted(SignedSettings::ReturnCode code,
-                                     bool unused) {
+                                     bool unused) OVERRIDE {
     VLOG(2) << "OnSettingsOpCompleted, code = " << code;
     if (callback_)
       callback_->OnStorePropertyCompleted(code, name_, value_);
     OnOpCompleted();
   }
 
  protected:
   // OpContext implemenetation
-  virtual void CreateOp() {
+  virtual void CreateOp() OVERRIDE {
     op_ = SignedSettings::CreateStorePropertyOp(name_, value_, this);
   }
 
  private:
   std::string name_;
   std::string value_;
 
   DISALLOW_COPY_AND_ASSIGN(StorePropertyOpContext);
 };
 
 class RetrievePropertyOpContext
     : public SignedSettings::Delegate<std::string>,
       public OpContext {
  public:
   RetrievePropertyOpContext(const std::string& name,
                             SignedSettingsHelper::Callback* callback,
                             Delegate* delegate)
       : OpContext(callback, delegate),
         name_(name) {
   }
 
   // chromeos::SignedSettings::Delegate implementation
   virtual void OnSettingsOpCompleted(SignedSettings::ReturnCode code,
-                                     std::string value) {
+                                     std::string value) OVERRIDE {
     if (callback_)
       callback_->OnRetrievePropertyCompleted(code, name_, value);
 
     OnOpCompleted();
   }
 
  protected:
   // OpContext implemenetation
-  virtual void CreateOp() {
+  virtual void CreateOp() OVERRIDE {
     op_ = SignedSettings::CreateRetrievePropertyOp(name_, this);
   }
 
  private:
   std::string name_;
 
   DISALLOW_COPY_AND_ASSIGN(RetrievePropertyOpContext);
 };
 
+class StorePolicyOpContext : public SignedSettings::Delegate<bool>,
+                             public OpContext {
+ public:
+  StorePolicyOpContext(const em::PolicyFetchResponse& policy,
+                       SignedSettingsHelper::Callback* callback,
+                       Delegate* delegate)
+      : OpContext(callback, delegate),
+        policy_(policy) {
+  }
+
+  // chromeos::SignedSettings::Delegate implementation
+  virtual void OnSettingsOpCompleted(SignedSettings::ReturnCode code,
+                                     bool unused) OVERRIDE {
+    VLOG(2) << "OnSettingsOpCompleted, code = " << code;
+    if (callback_)
+      callback_->OnStorePolicyCompleted(code);
+    OnOpCompleted();
+  }
+
+ protected:
+  // OpContext implementation
+  virtual void CreateOp() OVERRIDE {
+    op_ = SignedSettings::CreateStorePolicyOp(&policy_, this);
+  }
+
+ private:
+  em::PolicyFetchResponse policy_;
+  DISALLOW_COPY_AND_ASSIGN(StorePolicyOpContext);
+};
+
+class RetrievePolicyOpContext
+    : public SignedSettings::Delegate<const em::PolicyFetchResponse&>,
+      public OpContext {
+ public:
+  RetrievePolicyOpContext(SignedSettingsHelper::Callback* callback,
+                          Delegate* delegate)
+      : OpContext(callback, delegate) {
+  }
+
+  // chromeos::SignedSettings::Delegate implementation
+  virtual void OnSettingsOpCompleted(
+      SignedSettings::ReturnCode code,
+      const em::PolicyFetchResponse& policy) OVERRIDE {
+    if (callback_)
+      callback_->OnRetrievePolicyCompleted(code, policy);
+    OnOpCompleted();
+  }
+
+ protected:
+  // OpContext implementation
+  virtual void CreateOp() OVERRIDE {
+    op_ = SignedSettings::CreateRetrievePolicyOp(this);
+  }
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(RetrievePolicyOpContext);
+};
+
 }  // namespace
 
 
 class SignedSettingsHelperImpl : public SignedSettingsHelper,
                                  public OpContext::Delegate {
  public:
   // SignedSettingsHelper implementation
   virtual void StartCheckWhitelistOp(const std::string& email,
-                                     Callback* callback);
+                                     Callback* callback) OVERRIDE;
   virtual void StartWhitelistOp(const std::string& email,
                                 bool add_to_whitelist,
-                                Callback* callback);
+                                Callback* callback) OVERRIDE;
   virtual void StartStorePropertyOp(const std::string& name,
                                     const std::string& value,
-                                    Callback* callback);
+                                    Callback* callback) OVERRIDE;
   virtual void StartRetrieveProperty(const std::string& name,
-                                     Callback* callback);
-  virtual void CancelCallback(Callback* callback);
+                                     Callback* callback) OVERRIDE;
+  virtual void StartStorePolicyOp(const em::PolicyFetchResponse& policy,
+                                  Callback* callback) OVERRIDE;
+  virtual void StartRetrievePolicyOp(Callback* callback) OVERRIDE;
+  virtual void CancelCallback(Callback* callback) OVERRIDE;
 
   // OpContext::Delegate implementation
   virtual void OnOpCreated(OpContext* context);
   virtual void OnOpStarted(OpContext* context);
   virtual void OnOpCompleted(OpContext* context);
 
  private:
   SignedSettingsHelperImpl();
   ~SignedSettingsHelperImpl();
 
@@ skipping 37 matching lines @@
   AddOpContext(new WhitelistOpContext(
       add_to_whitelist ? WhitelistOpContext::ADD : WhitelistOpContext::REMOVE,
       email,
       callback,
       this));
 }
 
 void SignedSettingsHelperImpl::StartStorePropertyOp(
     const std::string& name,
     const std::string& value,
-    SignedSettingsHelper::SignedSettingsHelper::Callback* callback) {
+    SignedSettingsHelper::Callback* callback) {
   AddOpContext(new StorePropertyOpContext(
       name,
       value,
       callback,
       this));
 }
 
 void SignedSettingsHelperImpl::StartRetrieveProperty(
     const std::string& name,
     SignedSettingsHelper::Callback* callback) {
   AddOpContext(new RetrievePropertyOpContext(
       name,
       callback,
       this));
 }
 
+void SignedSettingsHelperImpl::StartStorePolicyOp(
+    const em::PolicyFetchResponse& policy,
+    SignedSettingsHelper::Callback* callback) {
+  AddOpContext(new StorePolicyOpContext(policy, callback, this));
+}
+
+void SignedSettingsHelperImpl::StartRetrievePolicyOp(
+    SignedSettingsHelper::Callback* callback) {
+  AddOpContext(new RetrievePolicyOpContext(callback, this));
+}
+
 void SignedSettingsHelperImpl::CancelCallback(
     SignedSettingsHelper::Callback* callback) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
 
   for (size_t i = 0; i < pending_contexts_.size(); ++i) {
     if (pending_contexts_[i]->callback() == callback) {
       pending_contexts_[i]->CancelCallback();
     }
   }
 }
@@ skipping 37 matching lines @@
 
   if (test_delegate_)
     test_delegate_->OnOpCompleted(context->op());
 }
 
 SignedSettingsHelper* SignedSettingsHelper::Get() {
   return g_signed_settings_helper_impl.Pointer();
 }
 
 }  // namespace chromeos