Add signal handler for low disk notification

chromeos/dbus/cryptohome_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 "chromeos/dbus/cryptohome_client.h"
 
 #include <stddef.h>
 #include <stdint.h>
 
 #include "base/bind.h"
@@ skipping 45 matching lines @@
     async_call_status_data_handler_ = data_handler;
   }
 
   // CryptohomeClient override.
   void ResetAsyncCallStatusHandlers() override {
     async_call_status_handler_.Reset();
     async_call_status_data_handler_.Reset();
   }
 
   // CryptohomeClient override.
+  void SetLowDiskSpaceHandler(const LowDiskSpaceHandler& handler) override {
+    low_disk_space_handler_ = handler;
+  }
+
+  // CryptohomeClient override.
   void WaitForServiceToBeAvailable(
       const WaitForServiceToBeAvailableCallback& callback) override {
     proxy_->WaitForServiceToBeAvailable(callback);
   }
 
   // CryptohomeClient override.
   void IsMounted(const BoolDBusMethodCallback& callback) override {
     dbus::MethodCall method_call(cryptohome::kCryptohomeInterface,
                                  cryptohome::kCryptohomeIsMounted);
     CallBoolMethod(&method_call, callback);
@@ skipping 845 matching lines @@
                                        weak_ptr_factory_.GetWeakPtr()),
                             base::Bind(&CryptohomeClientImpl::OnSignalConnected,
                                        weak_ptr_factory_.GetWeakPtr()));
     proxy_->ConnectToSignal(
         cryptohome::kCryptohomeInterface,
         cryptohome::kSignalAsyncCallStatusWithData,
         base::Bind(&CryptohomeClientImpl::OnAsyncCallStatusWithData,
                    weak_ptr_factory_.GetWeakPtr()),
         base::Bind(&CryptohomeClientImpl::OnSignalConnected,
                    weak_ptr_factory_.GetWeakPtr()));
+    proxy_->ConnectToSignal(cryptohome::kCryptohomeInterface,
+                            cryptohome::kSignalLowDiskSpace,
+                            base::Bind(&CryptohomeClientImpl::OnLowDiskSpace,
+                                       weak_ptr_factory_.GetWeakPtr()),
+                            base::Bind(&CryptohomeClientImpl::OnSignalConnected,
+                                       weak_ptr_factory_.GetWeakPtr()));
   }
 
  private:
   // Handles the result of AsyncXXX methods.
   void OnAsyncMethodCall(const AsyncMethodCallback& callback,
                          dbus::Response* response) {
     if (!response) {
       callback.Run(kNotReadyAsyncId);
       return;
     }
@@ skipping 201 matching lines @@
       LOG(ERROR) << "Invalid signal: " << signal->ToString();
       return;
     }
     if (!async_call_status_data_handler_.is_null()) {
       std::string return_data(reinterpret_cast<const char*>(return_data_buffer),
                               return_data_length);
       async_call_status_data_handler_.Run(async_id, return_status, return_data);
     }
   }
 
+  // Handles LowDiskSpace signal.
+  void OnLowDiskSpace(dbus::Signal* signal) {
+    dbus::MessageReader reader(signal);
+    uint64_t disk_free_bytes = 0;
+    if (!reader.PopUint64(&disk_free_bytes)) {
+      LOG(ERROR) << "Invalid signal: " << signal->ToString();
+      return;
+    }
+    if (!low_disk_space_handler_.is_null())
+      low_disk_space_handler_.Run(disk_free_bytes);
+  }
+
   // Handles the result of signal connection setup.
   void OnSignalConnected(const std::string& interface,
                          const std::string& signal,
                          bool succeeded) {
     LOG_IF(ERROR, !succeeded) << "Connect to " << interface << " " <<
         signal << " failed.";
   }
 
   dbus::ObjectProxy* proxy_;
   std::unique_ptr<BlockingMethodCaller> blocking_method_caller_;
   AsyncCallStatusHandler async_call_status_handler_;
   AsyncCallStatusWithDataHandler async_call_status_data_handler_;
+  LowDiskSpaceHandler low_disk_space_handler_;
 
   // Note: This should remain the last member so it'll be destroyed and
   // invalidate its weak pointers before any other members are destroyed.
   base::WeakPtrFactory<CryptohomeClientImpl> weak_ptr_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(CryptohomeClientImpl);
 };
 
 }  // namespace
 
 ////////////////////////////////////////////////////////////////////////////////
 // CryptohomeClient
 
 CryptohomeClient::CryptohomeClient() {}
 
 CryptohomeClient::~CryptohomeClient() {}
 
 // static
 CryptohomeClient* CryptohomeClient::Create() {
   return new CryptohomeClientImpl();
 }
 
 // static
 std::string CryptohomeClient::GetStubSanitizedUsername(
     const cryptohome::Identification& cryptohome_id) {
   return cryptohome_id.id() + kUserIdStubHashSuffix;
 }
 
 }  // namespace chromeos