ppapi/proxy/file_system_resource.cc

   1 // Copyright (c) 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "ppapi/proxy/file_system_resource.h"
   6
   7 #include "base/bind.h"
   8 #include "base/stl_util.h"
   9 #include "ipc/ipc_message.h"
  10 #include "ppapi/c/pp_errors.h"
  11 #include "ppapi/proxy/ppapi_messages.h"
  12 #include "ppapi/shared_impl/file_growth.h"
  13 #include "ppapi/shared_impl/tracked_callback.h"
  14 #include "ppapi/thunk/enter.h"
  15 #include "ppapi/thunk/ppb_file_io_api.h"
  16
  17 using ppapi::thunk::EnterResourceNoLock;
  18 using ppapi::thunk::PPB_FileIO_API;
  19 using ppapi::thunk::PPB_FileSystem_API;
  20
  21 namespace ppapi {
  22 namespace proxy {
  23
  24 FileSystemResource::QuotaRequest::QuotaRequest(
  25     int64_t amount_arg,
  26     const RequestQuotaCallback& callback_arg)
  27     : amount(amount_arg),
  28       callback(callback_arg) {
  29 }
  30
  31 FileSystemResource::QuotaRequest::~QuotaRequest() {
  32 }
  33
  34 FileSystemResource::FileSystemResource(Connection connection,
  35                                        PP_Instance instance,
  36                                        PP_FileSystemType type)
  37     : PluginResource(connection, instance),
  38       type_(type),
  39       called_open_(false),
  40       callback_count_(0),
  41       callback_result_(PP_OK),
  42       reserved_quota_(0),
  43       reserving_quota_(false) {
  44   DCHECK(type_ != PP_FILESYSTEMTYPE_INVALID);
  45   SendCreate(RENDERER, PpapiHostMsg_FileSystem_Create(type_));
  46   SendCreate(BROWSER, PpapiHostMsg_FileSystem_Create(type_));
  47 }
  48
  49 FileSystemResource::FileSystemResource(Connection connection,
  50                                        PP_Instance instance,
  51                                        int pending_renderer_id,
  52                                        int pending_browser_id,
  53                                        PP_FileSystemType type)
  54     : PluginResource(connection, instance),
  55       type_(type),
  56       called_open_(true),
  57       callback_count_(0),
  58       callback_result_(PP_OK),
  59       reserved_quota_(0),
  60       reserving_quota_(false) {
  61   DCHECK(type_ != PP_FILESYSTEMTYPE_INVALID);
  62   AttachToPendingHost(RENDERER, pending_renderer_id);
  63   AttachToPendingHost(BROWSER, pending_browser_id);
  64 }
  65
  66 FileSystemResource::~FileSystemResource() {
  67 }
  68
  69 PPB_FileSystem_API* FileSystemResource::AsPPB_FileSystem_API() {
  70   return this;
  71 }
  72
  73 int32_t FileSystemResource::Open(int64_t expected_size,
  74                                  scoped_refptr<TrackedCallback> callback) {
  75   DCHECK(type_ != PP_FILESYSTEMTYPE_ISOLATED);
  76   if (called_open_)
  77     return PP_ERROR_FAILED;
  78   called_open_ = true;
  79
  80   Call<PpapiPluginMsg_FileSystem_OpenReply>(RENDERER,
  81       PpapiHostMsg_FileSystem_Open(expected_size),
  82       base::Bind(&FileSystemResource::OpenComplete,
  83                  this,
  84                  callback));
  85   Call<PpapiPluginMsg_FileSystem_OpenReply>(BROWSER,
  86       PpapiHostMsg_FileSystem_Open(expected_size),
  87       base::Bind(&FileSystemResource::OpenComplete,
  88                  this,
  89                  callback));
  90   return PP_OK_COMPLETIONPENDING;
  91 }
  92
  93 PP_FileSystemType FileSystemResource::GetType() {
  94   return type_;
  95 }
  96
  97 void FileSystemResource::OpenQuotaFile(PP_Resource file_io) {
  98   DCHECK(!ContainsKey(files_, file_io));
  99   files_.insert(file_io);
 100 }
 101
 102 void FileSystemResource::CloseQuotaFile(PP_Resource file_io) {
 103   DCHECK(ContainsKey(files_, file_io));
 104   files_.erase(file_io);
 105 }
 106
 107 int64_t FileSystemResource::RequestQuota(
 108     int64_t amount,
 109     const RequestQuotaCallback& callback) {
 110   DCHECK(amount >= 0);
 111   if (!reserving_quota_ && reserved_quota_ >= amount) {
 112     reserved_quota_ -= amount;
 113     return amount;
 114   }
 115
 116   // Queue up a pending quota request.
 117   pending_quota_requests_.push(QuotaRequest(amount, callback));
 118
 119   // Reserve more quota if we haven't already.
 120   if (!reserving_quota_)
 121     ReserveQuota(amount);
 122
 123   return PP_OK_COMPLETIONPENDING;
 124 }
 125
 126 int32_t FileSystemResource::InitIsolatedFileSystem(
 127     const std::string& fsid,
 128     PP_IsolatedFileSystemType_Private type,
 129     const base::Callback<void(int32_t)>& callback) {
 130   // This call is mutually exclusive with Open() above, so we can reuse the
 131   // called_open state.
 132   DCHECK(type_ == PP_FILESYSTEMTYPE_ISOLATED);
 133   if (called_open_)
 134     return PP_ERROR_FAILED;
 135   called_open_ = true;
 136
 137   Call<PpapiPluginMsg_FileSystem_InitIsolatedFileSystemReply>(RENDERER,
 138       PpapiHostMsg_FileSystem_InitIsolatedFileSystem(fsid, type),
 139       base::Bind(&FileSystemResource::InitIsolatedFileSystemComplete,
 140       this,
 141       callback));
 142   Call<PpapiPluginMsg_FileSystem_InitIsolatedFileSystemReply>(BROWSER,
 143       PpapiHostMsg_FileSystem_InitIsolatedFileSystem(fsid, type),
 144       base::Bind(&FileSystemResource::InitIsolatedFileSystemComplete,
 145       this,
 146       callback));
 147   return PP_OK_COMPLETIONPENDING;
 148 }
 149
 150 void FileSystemResource::OpenComplete(
 151     scoped_refptr<TrackedCallback> callback,
 152     const ResourceMessageReplyParams& params) {
 153   ++callback_count_;
 154   // Prioritize worse result since only one status can be returned.
 155   if (params.result() != PP_OK)
 156     callback_result_ = params.result();
 157   // Received callback from browser and renderer.
 158   if (callback_count_ == 2)
 159     callback->Run(callback_result_);
 160 }
 161
 162 void FileSystemResource::InitIsolatedFileSystemComplete(
 163     const base::Callback<void(int32_t)>& callback,
 164     const ResourceMessageReplyParams& params) {
 165   ++callback_count_;
 166   // Prioritize worse result since only one status can be returned.
 167   if (params.result() != PP_OK)
 168     callback_result_ = params.result();
 169   // Received callback from browser and renderer.
 170   if (callback_count_ == 2)
 171     callback.Run(callback_result_);
 172 }
 173
 174 void FileSystemResource::ReserveQuota(int64_t amount) {
 175   DCHECK(!reserving_quota_);
 176   reserving_quota_ = true;
 177
 178   FileGrowthMap file_growths;
 179   for (std::set<PP_Resource>::iterator it = files_.begin();
 180        it != files_.end(); ++it) {
 181     EnterResourceNoLock<PPB_FileIO_API> enter(*it, true);
 182     if (enter.failed()) {
 183       NOTREACHED();
 184       continue;
 185     }
 186     PPB_FileIO_API* file_io_api = enter.object();
 187     file_growths[*it] = FileGrowth(
 188         file_io_api->GetMaxWrittenOffset(),
 189         file_io_api->GetAppendModeWriteAmount());
 190   }
 191   Call<PpapiPluginMsg_FileSystem_ReserveQuotaReply>(BROWSER,
 192       PpapiHostMsg_FileSystem_ReserveQuota(amount, file_growths),
 193       base::Bind(&FileSystemResource::ReserveQuotaComplete,
 194                  this));
 195 }
 196
 197 void FileSystemResource::ReserveQuotaComplete(
 198     const ResourceMessageReplyParams& params,
 199     int64_t amount,
 200     const FileSizeMap& file_sizes) {
 201   DCHECK(reserving_quota_);
 202   reserving_quota_ = false;
 203   reserved_quota_ = amount;
 204
 205   for (FileSizeMap::const_iterator it = file_sizes.begin();
 206        it != file_sizes.end(); ++it) {
 207     EnterResourceNoLock<PPB_FileIO_API> enter(it->first, true);
 208
 209     // It is possible that the host has sent an offset for a file that has been
 210     // destroyed in the plugin. Ignore it.
 211     if (enter.failed())
 212       continue;
 213     PPB_FileIO_API* file_io_api = enter.object();
 214     file_io_api->SetMaxWrittenOffset(it->second);
 215     file_io_api->SetAppendModeWriteAmount(0);
 216   }
 217
 218   DCHECK(!pending_quota_requests_.empty());
 219   // If we can't grant the first request after refreshing reserved_quota_, then
 220   // fail all pending quota requests to avoid an infinite refresh/fail loop.
 221   bool fail_all = reserved_quota_ < pending_quota_requests_.front().amount;
 222   while (!pending_quota_requests_.empty()) {
 223     QuotaRequest& request = pending_quota_requests_.front();
 224     if (fail_all) {
 225       request.callback.Run(0);
 226       pending_quota_requests_.pop();
 227     } else if (reserved_quota_ >= request.amount) {
 228       reserved_quota_ -= request.amount;
 229       request.callback.Run(request.amount);
 230       pending_quota_requests_.pop();
 231     } else {
 232       // Refresh the quota reservation for the first pending request that we
 233       // can't satisfy.
 234       ReserveQuota(request.amount);
 235       break;
 236     }
 237   }
 238 }
 239
 240 }  // namespace proxy
 241 }  // namespace ppapi