blob: 63dc97689a317321e9ea87742f15102a0dc49d9c [file] [log] [blame]
// 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_frame/npapi_url_request.h"
#include "base/string_number_conversions.h"
#include "base/threading/platform_thread.h"
#include "chrome/common/automation_messages.h"
#include "chrome_frame/chrome_frame_npapi.h"
#include "chrome_frame/np_browser_functions.h"
#include "chrome_frame/np_utils.h"
#include "net/base/net_errors.h"
class NPAPIUrlRequest : public PluginUrlRequest {
public:
explicit NPAPIUrlRequest(NPP instance);
~NPAPIUrlRequest();
virtual bool Start();
virtual void Stop();
virtual bool Read(int bytes_to_read);
// Called from NPAPI
NPError OnStreamCreated(const char* mime_type, NPStream* stream);
NPError OnStreamDestroyed(NPReason reason);
int OnWriteReady();
int OnWrite(void* buffer, int len);
// Thread unsafe implementation of ref counting, since
// this will be called on the plugin UI thread only.
virtual unsigned long API_CALL AddRef();
virtual unsigned long API_CALL Release();
const net::URLRequestStatus& status() const {
return status_;
}
NPP instance() const {
return instance_;
}
private:
unsigned long ref_count_;
NPP instance_;
NPStream* stream_;
size_t pending_read_size_;
net::URLRequestStatus status_;
base::PlatformThreadId thread_;
static int instance_count_;
DISALLOW_COPY_AND_ASSIGN(NPAPIUrlRequest);
};
int NPAPIUrlRequest::instance_count_ = 0;
NPAPIUrlRequest::NPAPIUrlRequest(NPP instance)
: ref_count_(0), instance_(instance), stream_(NULL),
pending_read_size_(0),
status_(net::URLRequestStatus::FAILED, net::ERR_FAILED),
thread_(base::PlatformThread::CurrentId()) {
DVLOG(1) << "Created request. Count: " << ++instance_count_;
}
NPAPIUrlRequest::~NPAPIUrlRequest() {
DVLOG(1) << "Deleted request. Count: " << --instance_count_;
}
// NPAPIUrlRequest member defines.
bool NPAPIUrlRequest::Start() {
NPError result = NPERR_GENERIC_ERROR;
DVLOG(1) << "Starting URL request: " << url();
// Initialize the net::HostPortPair structure from the url
socket_address_ = net::HostPortPair::FromURL(GURL(url()));
if (LowerCaseEqualsASCII(method(), "get")) {
// TODO(joshia): if we have extra headers for HTTP GET, then implement
// it using XHR
result = npapi::GetURLNotify(instance_, url().c_str(), NULL, this);
} else if (LowerCaseEqualsASCII(method(), "post")) {
uint32 data_len = static_cast<uint32>(post_data_len());
std::string buffer;
if (extra_headers().length() > 0) {
buffer += extra_headers();
TrimWhitespace(buffer, TRIM_ALL, &buffer);
// Firefox looks specifically for "Content-length: \d+\r\n\r\n"
// to detect if extra headers are added to the message buffer.
buffer += "\r\nContent-length: ";
buffer += base::IntToString(data_len);
buffer += "\r\n\r\n";
}
std::string data;
data.resize(data_len);
uint32 bytes_read;
upload_data_->Read(&data[0], data_len,
reinterpret_cast<ULONG*>(&bytes_read));
DCHECK_EQ(data_len, bytes_read);
buffer += data;
result = npapi::PostURLNotify(instance_, url().c_str(), NULL,
buffer.length(), buffer.c_str(), false, this);
} else {
NOTREACHED() << "PluginUrlRequest only supports 'GET'/'POST'";
}
if (NPERR_NO_ERROR != result) {
int os_error = net::ERR_FAILED;
switch (result) {
case NPERR_INVALID_URL:
os_error = net::ERR_INVALID_URL;
break;
default:
break;
}
delegate_->OnResponseEnd(id(),
net::URLRequestStatus(net::URLRequestStatus::FAILED, os_error));
return false;
}
return true;
}
void NPAPIUrlRequest::Stop() {
DVLOG(1) << "Finished request: Url - " << url()
<< " result: " << static_cast<int>(status_.status());
status_.set_status(net::URLRequestStatus::CANCELED);
if (stream_) {
npapi::DestroyStream(instance_, stream_, NPRES_USER_BREAK);
stream_ = NULL;
}
}
bool NPAPIUrlRequest::Read(int bytes_to_read) {
pending_read_size_ = bytes_to_read;
return true;
}
NPError NPAPIUrlRequest::OnStreamCreated(const char* mime_type,
NPStream* stream) {
stream_ = stream;
status_.set_status(net::URLRequestStatus::IO_PENDING);
// TODO(iyengar)
// Add support for passing persistent cookies and information about any URL
// redirects to Chrome.
delegate_->OnResponseStarted(id(), mime_type, stream->headers, stream->end,
base::Time::FromTimeT(stream->lastmodified), std::string(), 0,
socket_address_);
return NPERR_NO_ERROR;
}
NPError NPAPIUrlRequest::OnStreamDestroyed(NPReason reason) {
// If the request has been aborted, then ignore the |reason| argument.
// Normally the execution flow is such than NPRES_USER_BREAK will be passed
// when the stream is aborted, but sometimes NPRES_NETWORK_ERROR is passed
// instead. To prevent Chrome from receiving a notification of a failed
// network connection, when Chrome actually canceled the request, we ignore
// the status here.
if (net::URLRequestStatus::CANCELED != status_.status()) {
switch (reason) {
case NPRES_DONE:
status_.set_status(net::URLRequestStatus::SUCCESS);
status_.set_os_error(0);
break;
case NPRES_USER_BREAK:
status_.set_status(net::URLRequestStatus::CANCELED);
status_.set_os_error(net::ERR_ABORTED);
break;
case NPRES_NETWORK_ERR:
default:
status_.set_status(net::URLRequestStatus::FAILED);
status_.set_os_error(net::ERR_CONNECTION_CLOSED);
break;
}
}
delegate_->OnResponseEnd(id(), status_);
return NPERR_NO_ERROR;
}
int NPAPIUrlRequest::OnWriteReady() {
return pending_read_size_;
}
int NPAPIUrlRequest::OnWrite(void* buffer, int len) {
pending_read_size_ = 0;
std::string data(reinterpret_cast<char*>(buffer), len);
delegate_->OnReadComplete(id(), data);
return len;
}
STDMETHODIMP_(ULONG) NPAPIUrlRequest::AddRef() {
DCHECK_EQ(base::PlatformThread::CurrentId(), thread_);
++ref_count_;
return ref_count_;
}
STDMETHODIMP_(ULONG) NPAPIUrlRequest::Release() {
DCHECK_EQ(base::PlatformThread::CurrentId(), thread_);
unsigned long ret = --ref_count_;
if (!ret)
delete this;
return ret;
}
NPAPIUrlRequestManager::NPAPIUrlRequestManager() : instance_(NULL) {
}
NPAPIUrlRequestManager::~NPAPIUrlRequestManager() {
StopAll();
}
// PluginUrlRequestManager implementation
PluginUrlRequestManager::ThreadSafeFlags
NPAPIUrlRequestManager::GetThreadSafeFlags() {
return PluginUrlRequestManager::NOT_THREADSAFE;
}
void NPAPIUrlRequestManager::StartRequest(int request_id,
const AutomationURLRequest& request_info) {
scoped_refptr<NPAPIUrlRequest> new_request(new NPAPIUrlRequest(instance_));
DCHECK(new_request);
if (new_request->Initialize(this, request_id, request_info.url,
request_info.method, request_info.referrer,
request_info.extra_request_headers,
request_info.upload_data,
static_cast<ResourceType::Type>(request_info.resource_type),
enable_frame_busting_,
request_info.load_flags)) {
DCHECK(request_map_.find(request_id) == request_map_.end());
if (new_request->Start()) {
request_map_[request_id] = new_request;
// Keep additional reference on request for NPSTREAM
// This will be released in NPP_UrlNotify
new_request->AddRef();
}
}
}
void NPAPIUrlRequestManager::ReadRequest(int request_id, int bytes_to_read) {
scoped_refptr<NPAPIUrlRequest> request = LookupRequest(request_id);
if (request)
request->Read(bytes_to_read);
}
void NPAPIUrlRequestManager::EndRequest(int request_id) {
scoped_refptr<NPAPIUrlRequest> request = LookupRequest(request_id);
if (request)
request->Stop();
}
void NPAPIUrlRequestManager::StopAll() {
std::vector<scoped_refptr<NPAPIUrlRequest> > request_list;
// We copy the pending requests into a temporary vector as the Stop
// function in the request could also try to delete the request from
// the request map and the iterator could end up being invalid.
for (RequestMap::iterator it = request_map_.begin();
it != request_map_.end(); ++it) {
DCHECK(it->second != NULL);
request_list.push_back(it->second);
}
for (std::vector<scoped_refptr<NPAPIUrlRequest> >::size_type index = 0;
index < request_list.size(); ++index) {
request_list[index]->Stop();
}
}
void NPAPIUrlRequestManager::SetCookiesForUrl(const GURL& url,
const std::string& cookie) {
if (npapi::VersionMinor() >= NPVERS_HAS_URL_AND_AUTH_INFO) {
npapi::SetValueForURL(instance_, NPNURLVCookie, url.spec().c_str(),
cookie.c_str(), cookie.length());
} else {
NOTREACHED() << "Unsupported version";
}
}
void NPAPIUrlRequestManager::GetCookiesForUrl(const GURL& url, int cookie_id) {
std::string cookie_string;
bool success = false;
if (npapi::VersionMinor() >= NPVERS_HAS_URL_AND_AUTH_INFO) {
char* cookies = NULL;
unsigned int cookie_length = 0;
NPError ret = npapi::GetValueForURL(instance_, NPNURLVCookie,
url.spec().c_str(), &cookies,
&cookie_length);
if (ret == NPERR_NO_ERROR) {
DVLOG(1) << "Obtained cookies:" << cookies << " from host";
cookie_string.append(cookies, cookie_length);
npapi::MemFree(cookies);
success = true;
}
} else {
NOTREACHED() << "Unsupported version";
}
if (!success)
DVLOG(1) << "Failed to return cookies for url:" << url.spec();
OnCookiesRetrieved(success, url, cookie_string, cookie_id);
}
// PluginRequestDelegate implementation.
// Callbacks from NPAPIUrlRequest. Simply forward to the delegate.
void NPAPIUrlRequestManager::OnResponseStarted(int request_id,
const char* mime_type, const char* headers, int size,
base::Time last_modified, const std::string& redirect_url,
int redirect_status, const net::HostPortPair& socket_address) {
delegate_->OnResponseStarted(request_id, mime_type, headers, size,
last_modified, redirect_url, redirect_status, socket_address);
}
void NPAPIUrlRequestManager::OnReadComplete(int request_id,
const std::string& data) {
delegate_->OnReadComplete(request_id, data);
}
void NPAPIUrlRequestManager::OnResponseEnd(
int request_id,
const net::URLRequestStatus& status) {
// Delete from map.
RequestMap::iterator it = request_map_.find(request_id);
DCHECK(request_map_.end() != it);
scoped_refptr<NPAPIUrlRequest> request = (*it).second;
request_map_.erase(it);
// Inform delegate unless canceled.
if (status.status() != net::URLRequestStatus::CANCELED)
delegate_->OnResponseEnd(request_id, status);
}
void NPAPIUrlRequestManager::OnCookiesRetrieved(bool success, const GURL& url,
const std::string& cookie_string, int cookie_id) {
delegate_->OnCookiesRetrieved(success, url, cookie_string, cookie_id);
}
// Notifications from browser. Find the NPAPIUrlRequest and forward to it.
NPError NPAPIUrlRequestManager::NewStream(NPMIMEType type,
NPStream* stream,
NPBool seekable,
uint16* stream_type) {
NPAPIUrlRequest* request = RequestFromNotifyData(stream->notifyData);
if (!request)
return NPERR_NO_ERROR;
// This stream is being constructed for a request that has already been
// canceled. Signal its immediate termination.
if (net::URLRequestStatus::CANCELED == request->status().status()) {
return npapi::DestroyStream(request->instance(),
stream, NPRES_USER_BREAK);
}
DCHECK(request_map_.find(request->id()) != request_map_.end());
// If the host browser does not support the NPAPI redirect notification
// spec, and if the request URL is implicitly redirected, we need to
// inform Chrome about the redirect and allow it to follow the redirect.
// We achieve this by comparing the URL requested with the URL received in
// the response and if they don't match we assume a redirect. This would have
// a sideffect that two GET requests would be sent out in this case.
if (!BrowserSupportsRedirectNotification()) {
if (GURL(request->url().c_str()) != GURL(stream->url)) {
DVLOG(1) << "Request URL:"
<< request->url()
<< " was redirected to:"
<< stream->url;
delegate_->OnResponseStarted(
request->id(), "", "", 0, base::Time(), stream->url, 302,
net::HostPortPair(net::HostPortPair::FromURL(GURL(stream->url))));
return NPERR_GENERIC_ERROR;
}
}
// We need to return the requested stream mode if we are returning a success
// code. If we don't do this it causes Opera to blow up.
*stream_type = NP_NORMAL;
return request->OnStreamCreated(type, stream);
}
int32 NPAPIUrlRequestManager::WriteReady(NPStream* stream) {
NPAPIUrlRequest* request = RequestFromNotifyData(stream->notifyData);
if (!request)
return 0x7FFFFFFF;
DCHECK(request_map_.find(request->id()) != request_map_.end());
return request->OnWriteReady();
}
int32 NPAPIUrlRequestManager::Write(NPStream* stream, int32 offset,
int32 len, void* buffer) {
NPAPIUrlRequest* request = RequestFromNotifyData(stream->notifyData);
if (!request)
return len;
DCHECK(request_map_.find(request->id()) != request_map_.end());
return request->OnWrite(buffer, len);
}
NPError NPAPIUrlRequestManager::DestroyStream(NPStream* stream,
NPReason reason) {
NPAPIUrlRequest* request = RequestFromNotifyData(stream->notifyData);
if (!request)
return NPERR_NO_ERROR;
// It is possible to receive notification of a destroyed stream for a
// unregistered request: EndRequest will unregister a request in response
// to an AutomationMsg_RequestEnd message. EndRequest will also invoke
// npapi::DestroyStream, which will call back to this function.
if (request_map_.find(request->id()) != request_map_.end())
return request->OnStreamDestroyed(reason);
return NPERR_NO_ERROR;
}
void NPAPIUrlRequestManager::UrlNotify(const char* url, NPReason reason,
void* notify_data) {
NPAPIUrlRequest* request = RequestFromNotifyData(notify_data);
DCHECK(request != NULL);
if (request) {
request->Stop();
request->Release();
}
}
void NPAPIUrlRequestManager::UrlRedirectNotify(const char* url, int status,
void* notify_data) {
NPAPIUrlRequest* request = RequestFromNotifyData(notify_data);
if (request) {
delegate_->OnResponseStarted(
request->id(), "", "", 0, base::Time(), url, status,
net::HostPortPair(net::HostPortPair::FromURL(GURL(url))));
} else {
NOTREACHED() << "Received unexpected redirect notification for url:"
<< url;
}
}
scoped_refptr<NPAPIUrlRequest> NPAPIUrlRequestManager::LookupRequest(
int request_id) {
RequestMap::iterator index = request_map_.find(request_id);
if (index != request_map_.end())
return index->second;
return NULL;
}