Storage manager: Show the disk usage of other users.

chromeos/cryptohome/homedir_methods.cc

 // Copyright 2014 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/cryptohome/homedir_methods.h"
 
 #include <stddef.h>
 #include <stdint.h>
 
 #include "base/bind.h"
@@ skipping 268 matching lines @@
   }
 
   void RenameCryptohome(const Identification& id_from,
                         const Identification& id_to,
                         const Callback& callback) override {
     DBusThreadManager::Get()->GetCryptohomeClient()->RenameCryptohome(
         id_from, id_to, base::Bind(&HomedirMethodsImpl::OnBaseReplyCallback,
                                    weak_ptr_factory_.GetWeakPtr(), callback));
   }
 
+  void GetAccountDiskUsage(
+      const Identification& id,
+      const GetAccountDiskUsageCallback& callback) override {
+    DBusThreadManager::Get()->GetCryptohomeClient()->GetAccountDiskUsage(
+        id, base::Bind(&HomedirMethodsImpl::OnGetAccountDiskUsageCallback,
+                       weak_ptr_factory_.GetWeakPtr(), callback));
+  }
+
  private:
   void OnGetKeyDataExCallback(const GetKeyDataCallback& callback,
                               chromeos::DBusMethodCallStatus call_status,
                               bool result,
                               const BaseReply& reply) {
     if (call_status != chromeos::DBUS_METHOD_CALL_SUCCESS) {
       callback.Run(false, MOUNT_ERROR_FATAL, std::vector<KeyDefinition>());
       return;
     }
     if (reply.has_error()) {
@@ skipping 100 matching lines @@
     if (!reply.HasExtension(MountReply::reply)) {
       callback.Run(false, MOUNT_ERROR_FATAL, std::string());
       return;
     }
 
     std::string mount_hash;
     mount_hash = reply.GetExtension(MountReply::reply).sanitized_username();
     callback.Run(true, MOUNT_ERROR_NONE, mount_hash);
   }
 
+  void OnGetAccountDiskUsageCallback(
+      const GetAccountDiskUsageCallback& callback,
+      chromeos::DBusMethodCallStatus call_status,
+      bool result,
+      const BaseReply& reply) {
+    if (call_status != chromeos::DBUS_METHOD_CALL_SUCCESS) {
+      callback.Run(false, -1);
+      return;
+    }
+    if (reply.has_error()) {
+      if (reply.error() != CRYPTOHOME_ERROR_NOT_SET) {
+        callback.Run(false, -1);
+        return;
+      }
+    }
+    if (!reply.HasExtension(GetAccountDiskUsageReply::reply)) {
+      callback.Run(false, -1);
+      return;
+    }
+
+    int64_t size = reply.GetExtension(GetAccountDiskUsageReply::reply).size();
+    callback.Run(true, size);
+  }
+
   void OnBaseReplyCallback(const Callback& callback,
                            chromeos::DBusMethodCallStatus call_status,
                            bool result,
                            const BaseReply& reply) {
     if (call_status != chromeos::DBUS_METHOD_CALL_SUCCESS) {
       callback.Run(false, MOUNT_ERROR_FATAL);
       return;
     }
     if (reply.has_error()) {
       if (reply.error() != CRYPTOHOME_ERROR_NOT_SET) {
@@ skipping 39 matching lines @@
   }
   delete g_homedir_methods;
   g_homedir_methods = NULL;
   VLOG(1) << "HomedirMethods Shutdown completed";
 }
 
 // static
 HomedirMethods* HomedirMethods::GetInstance() { return g_homedir_methods; }
 
 }  // namespace cryptohome