Adding support to retrieve remaining and total space on disk/file shelf.

chrome/browser/chromeos/cros/mount_library.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/cros/mount_library.h"
 
 #include <set>
+#include <sys/statvfs.h>
 #include <vector>
 
 #include "base/message_loop.h"
 #include "base/string_util.h"
 #include "chrome/browser/chromeos/cros/cros_library.h"
 #include "content/browser/browser_thread.h"
 
 const char* kLibraryNotLoaded = "Cros Library not loaded";
 const char* kDeviceNotFound = "Device could not be found";
 
@@ skipping 177 matching lines @@
     libcros_proxy_->CallMountPath(source_path, type, options,
                                   &MountCompletedHandler, this);
   }
 
   virtual void UnmountPath(const char* mount_path) OVERRIDE {
     CHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
     libcros_proxy_->CallUnmountPath(mount_path, &UnmountMountPointCallback,
                                     this);
   }
 
+  virtual void GetSizeStatsOnFileThread(const char* mount_path,
+      size_t* total_size_kb, size_t* remaining_size_kb) OVERRIDE {
+    CHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE));
+
+    uint64_t total_size_uint64 = 0;
+    uint64_t remaining_size_uint64 = 0;
+
+    struct statvfs stat;
+    if (statvfs(mount_path, &stat) == 0) {
+      total_size_uint64 =
+          static_cast<uint64_t>(stat.f_blocks) * stat.f_frsize;
+      remaining_size_uint64 =
+          static_cast<uint64_t>(stat.f_bfree) * stat.f_frsize;
+    }
+
+    *total_size_kb = static_cast<size_t>(total_size_uint64 / 1024);
+    *remaining_size_kb = static_cast<size_t>(remaining_size_uint64 / 1024);
+  }
+
   virtual void FormatUnmountedDevice(const char* file_path) OVERRIDE {
     DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
     for (MountLibrary::DiskMap::iterator it = disks_.begin();
         it != disks_.end(); ++it) {
       if (it->second->file_path().compare(file_path) == 0 &&
           !it->second->mount_path().empty()) {
         OnFormatDevice(file_path,
                        false,
                        MOUNT_METHOD_ERROR_LOCAL,
                        "Device is still mounted.");
@@ skipping 567 matching lines @@
   virtual void AddObserver(Observer* observer) OVERRIDE {}
   virtual void RemoveObserver(Observer* observer) OVERRIDE {}
   virtual const DiskMap& disks() const OVERRIDE { return disks_; }
   virtual const MountPointMap& mount_points() const OVERRIDE {
     return mount_points_;
   }
   virtual void RequestMountInfoRefresh() OVERRIDE {}
   virtual void MountPath(const char* source_path, MountType type,
                          const MountPathOptions& options) OVERRIDE {}
   virtual void UnmountPath(const char* mount_path) OVERRIDE {}
+  virtual void GetSizeStatsOnFileThread(const char* mount_path,
+      size_t* total_size_kb, size_t* remaining_size_kb) OVERRIDE {}
   virtual void FormatUnmountedDevice(const char* device_path) OVERRIDE {}
   virtual void FormatMountedDevice(const char* mount_path) OVERRIDE {}
   virtual void UnmountDeviceRecursive(const char* device_path,
       UnmountDeviceRecursiveCallbackType callback, void* user_data)
       OVERRIDE {}
  private:
   // The list of disks found.
   DiskMap disks_;
   MountPointMap mount_points_;
 
   DISALLOW_COPY_AND_ASSIGN(MountLibraryStubImpl);
 };
 
 // static
 MountLibrary* MountLibrary::GetImpl(bool stub) {
   MountLibrary* impl;
   if (stub)
     impl = new MountLibraryStubImpl();
   else
     impl = new MountLibraryImpl();
   impl->Init();
   return impl;
 }
 
 }  // namespace chromeos