root/content/child/npapi/plugin_stream_url.cc

DEFINITIONS

1. id_
2. SetPluginURLFetcher
3. URLRedirectResponse
4. FetchRange
5. Close
6. AsResourceClient
7. CancelRequest
8. WillSendRequest
9. DidReceiveResponse
10. DidReceiveData
11. DidFinishLoading
12. DidFail
13. IsMultiByteResponseExpected
14. ResourceId
15. AddRangeRequestResourceId
16. SetDeferLoading
17. UpdateUrl

// 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 "content/child/npapi/plugin_stream_url.h"

#include <algorithm>

#include "base/stl_util.h"
#include "base/strings/string_util.h"
#include "content/child/npapi/plugin_host.h"
#include "content/child/npapi/plugin_instance.h"
#include "content/child/npapi/plugin_lib.h"
#include "content/child/npapi/plugin_url_fetcher.h"
#include "content/child/npapi/webplugin.h"
#include "net/http/http_response_headers.h"

namespace content {

PluginStreamUrl::PluginStreamUrl(
    unsigned long resource_id,
    const GURL &url,
    PluginInstance *instance,
    bool notify_needed,
    void *notify_data)
    : PluginStream(instance, url.spec().c_str(), notify_needed, notify_data),
      url_(url),
      id_(resource_id) {
}

void PluginStreamUrl::SetPluginURLFetcher(PluginURLFetcher* fetcher) {
  plugin_url_fetcher_.reset(fetcher);
}

void PluginStreamUrl::URLRedirectResponse(bool allow) {
  if (plugin_url_fetcher_.get()) {
    plugin_url_fetcher_->URLRedirectResponse(allow);
  } else {
    instance()->webplugin()->URLRedirectResponse(allow, id_);
  }

  if (allow)
    UpdateUrl(pending_redirect_url_.c_str());
}

void PluginStreamUrl::FetchRange(const std::string& range) {
  PluginURLFetcher* range_fetcher = new PluginURLFetcher(
      this, url_, plugin_url_fetcher_->first_party_for_cookies(), "GET", NULL,
      0, plugin_url_fetcher_->referrer(), range, false, false,
      plugin_url_fetcher_->origin_pid(),
      plugin_url_fetcher_->render_frame_id(),
      plugin_url_fetcher_->render_view_id(), id_,
      plugin_url_fetcher_->copy_stream_data());
  range_request_fetchers_.push_back(range_fetcher);
}

bool PluginStreamUrl::Close(NPReason reason) {
  // Protect the stream against it being destroyed or the whole plugin instance
  // being destroyed within the destroy stream handler.
  scoped_refptr<PluginStream> protect(this);
  CancelRequest();
  bool result = PluginStream::Close(reason);
  instance()->RemoveStream(this);
  return result;
}

WebPluginResourceClient* PluginStreamUrl::AsResourceClient() {
  return static_cast<WebPluginResourceClient*>(this);
}

void PluginStreamUrl::CancelRequest() {
  if (id_ > 0) {
    if (plugin_url_fetcher_.get()) {
      plugin_url_fetcher_->Cancel();
    } else {
      if (instance()->webplugin()) {
        instance()->webplugin()->CancelResource(id_);
      }
    }
    id_ = 0;
  }
  if (instance()->webplugin()) {
    for (size_t i = 0; i < range_requests_.size(); ++i)
      instance()->webplugin()->CancelResource(range_requests_[i]);
  }

  range_requests_.clear();

  STLDeleteElements(&range_request_fetchers_);
}

void PluginStreamUrl::WillSendRequest(const GURL& url, int http_status_code) {
  if (notify_needed()) {
    // If the plugin participates in HTTP url redirect handling then notify it.
    if (net::HttpResponseHeaders::IsRedirectResponseCode(http_status_code) &&
        instance()->handles_url_redirects()) {
      pending_redirect_url_ = url.spec();
      instance()->NPP_URLRedirectNotify(url.spec().c_str(), http_status_code,
          notify_data());
      return;
    }
  }
  url_ = url;
  UpdateUrl(url.spec().c_str());
}

void PluginStreamUrl::DidReceiveResponse(const std::string& mime_type,
                                         const std::string& headers,
                                         uint32 expected_length,
                                         uint32 last_modified,
                                         bool request_is_seekable) {
  // Protect the stream against it being destroyed or the whole plugin instance
  // being destroyed within the new stream handler.
  scoped_refptr<PluginStream> protect(this);

  bool opened = Open(mime_type,
                     headers,
                     expected_length,
                     last_modified,
                     request_is_seekable);
  if (!opened) {
    CancelRequest();
    instance()->RemoveStream(this);
  } else {
    SetDeferLoading(false);
  }
}

void PluginStreamUrl::DidReceiveData(const char* buffer, int length,
                                     int data_offset) {
  if (!open())
    return;

  // Protect the stream against it being destroyed or the whole plugin instance
  // being destroyed within the write handlers
  scoped_refptr<PluginStream> protect(this);

  if (length > 0) {
    // The PluginStreamUrl instance could get deleted if the plugin fails to
    // accept data in NPP_Write.
    if (Write(const_cast<char*>(buffer), length, data_offset) > 0) {
      SetDeferLoading(false);
    }
  }
}

void PluginStreamUrl::DidFinishLoading(unsigned long resource_id) {
  if (!seekable()) {
    Close(NPRES_DONE);
  } else {
    std::vector<unsigned long>::iterator it_resource = std::find(
        range_requests_.begin(),
        range_requests_.end(),
        resource_id);
    // Resource id must be known to us - either main resource id, or one
    // of the resources, created for range requests.
    DCHECK(resource_id == id_ || it_resource != range_requests_.end());
    // We should notify the plugin about failed/finished requests to ensure
    // that the number of active resource clients does not continue to grow.
    if (instance()->webplugin())
      instance()->webplugin()->CancelResource(resource_id);
    if (it_resource != range_requests_.end())
      range_requests_.erase(it_resource);
  }
}

void PluginStreamUrl::DidFail(unsigned long resource_id) {
  Close(NPRES_NETWORK_ERR);
}

bool PluginStreamUrl::IsMultiByteResponseExpected() {
  return seekable();
}

int PluginStreamUrl::ResourceId() {
  return id_;
}

PluginStreamUrl::~PluginStreamUrl() {
  if (!plugin_url_fetcher_.get() && instance() && instance()->webplugin()) {
    instance()->webplugin()->ResourceClientDeleted(AsResourceClient());
  }

  STLDeleteElements(&range_request_fetchers_);
}

void PluginStreamUrl::AddRangeRequestResourceId(unsigned long resource_id) {
  DCHECK_NE(resource_id, 0u);
  range_requests_.push_back(resource_id);
}

void PluginStreamUrl::SetDeferLoading(bool value) {
  // If we determined that the request had failed via the HTTP headers in the
  // response then we send out a failure notification to the plugin process, as
  // certain plugins don't handle HTTP failure codes correctly.
  if (plugin_url_fetcher_.get()) {
    if (!value && plugin_url_fetcher_->pending_failure_notification()) {
      // This object may be deleted now.
      DidFail(id_);
    }
    return;
  }
  if (id_ > 0)
    instance()->webplugin()->SetDeferResourceLoading(id_, value);
  for (size_t i = 0; i < range_requests_.size(); ++i)
    instance()->webplugin()->SetDeferResourceLoading(range_requests_[i],
                                                     value);
}

void PluginStreamUrl::UpdateUrl(const char* url) {
  DCHECK(!open());
  free(const_cast<char*>(stream()->url));
  stream()->url = base::strdup(url);
  pending_redirect_url_.clear();
}

}  // namespace content