Implement Pepper proxy for PPB_DirectoryReader

content/renderer/pepper/pepper_directory_reader_host.cc

+// Copyright (c) 2013 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 "content/renderer/pepper/pepper_directory_reader_host.h"
+
+#include "content/public/renderer/renderer_ppapi_host.h"
+#include "ppapi/c/pp_completion_callback.h"
+#include "ppapi/c/pp_errors.h"
+#include "ppapi/host/dispatch_host_message.h"
+#include "ppapi/host/host_message_context.h"
+#include "ppapi/host/ppapi_host.h"
+#include "ppapi/proxy/enter_proxy.h"
+#include "ppapi/proxy/ppapi_messages.h"
+#include "ppapi/proxy/ppb_file_ref_proxy.h"
+#include "ppapi/shared_impl/ppb_file_ref_shared.h"
+
+using ppapi::thunk::EnterResource;
+using ppapi::thunk::EnterResourceCreation;
+using ppapi::thunk::PPB_DirectoryReader_API;
+using ppapi::thunk::PPB_FileRef_API;
+
+namespace content {
+
+PepperDirectoryReaderHost::PepperDirectoryReaderHost(
+    RendererPpapiHost* host,
+    PP_Instance instance,
+    PP_Resource resource)
+    : ResourceHost(host->GetPpapiHost(), instance, resource),
+      instance_(instance),
+      directory_reader_(0) {
+  entry_.file_ref = 0;
+}
+
+PepperDirectoryReaderHost::~PepperDirectoryReaderHost() {
+}
+
+int32_t PepperDirectoryReaderHost::OnResourceMessageReceived(
+    const IPC::Message& msg,
+    ppapi::host::HostMessageContext* context) {
+  IPC_BEGIN_MESSAGE_MAP(PepperDirectoryReaderHost, msg)
+    PPAPI_DISPATCH_HOST_RESOURCE_CALL(
+        PpapiHostMsg_DirectoryReader_GetNextEntry, OnGetNextEntry)
+  IPC_END_MESSAGE_MAP()
+  return PP_ERROR_FAILED;
+}
+
+// static
+void PepperDirectoryReaderHost::OnGetNextEntryCallback(
+    void* data,
+    int32_t result) {
+  PepperDirectoryReaderHost* self =
+      static_cast<PepperDirectoryReaderHost*>(data);
+  if (result == PP_OK && self->entry_.file_ref) {
+    self->PushEntry(self->entry_);
+    self->GetNextEntryInternal();
+    return;
+  }
+  self->SendGetNextEntryReply(result);
+}
+
+int32_t PepperDirectoryReaderHost::OnGetNextEntry(
+  ppapi::host::HostMessageContext* host_context,
+  ppapi::HostResource resource) {
+  reply_context_ = host_context->MakeReplyMessageContext();
+  host_resource_ = resource;
+
+  ppapi::thunk::EnterResourceCreation enter(pp_instance());
+  if (enter.failed())
+    return PP_ERROR_FAILED;
+
+  if (!directory_reader_)
+    directory_reader_ = enter.functions()->CreateDirectoryReader(
+        pp_instance(),resource.host_resource());
+  DCHECK(directory_reader_);
+
+  return GetNextEntryInternal();
+}
+
+int32_t PepperDirectoryReaderHost::GetNextEntryInternal() {
+  EnterResource<PPB_DirectoryReader_API> enter(directory_reader_, false);
+  if (enter.failed())
+    return PP_ERROR_FAILED;
+
+  ppapi::ResourceTracker* tracker =
+      ppapi::PpapiGlobals::Get()->GetResourceTracker();
+  ppapi::TrackedCallback* callback = new ppapi::TrackedCallback(
+      tracker->GetResource(directory_reader_),
+      PP_MakeCompletionCallback(
+          &PepperDirectoryReaderHost::OnGetNextEntryCallback,
+          AsWeakPtr()));
+
+  int32_t result = enter.object()->GetNextEntry(&entry_, callback);
+  if (result == PP_OK_COMPLETIONPENDING)
+    return result;
+
+  while (result == PP_OK && entry_.file_ref) {
+    PushEntry(entry_);
+    result = enter.object()->GetNextEntry(&entry_, callback);
+  }
+  SendGetNextEntryReply(result);
+  return result;
+}
+
+void PepperDirectoryReaderHost::SendGetNextEntryReply(int32_t result) {
+  reply_context_.params.set_result(result);
+  host()->SendReply(
+      reply_context_,
+      PpapiPluginMsg_DirectoryReader_GetNextEntryReply(host_resources_,
+                                                       file_types_));
+}
+
+void PepperDirectoryReaderHost::PushEntry(const PP_DirectoryEntry_Dev& entry) {
+  ppapi::PPB_FileRef_CreateInfo info;
+  ppapi::proxy::PPB_FileRef_Proxy::SerializeFileRef(entry.file_ref, &info);
+  host_resources_.push_back(info);
+  file_types_.push_back(entry.file_type);
+}
+
+}  // namespace content