blob: de35c4a5a589c5a1746f6de1e48f7fa4268e08eb [file] [log] [blame]
// Copyright 2010, Google Inc.
// All rights reserved.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are
// met:
//
// * Redistributions of source code must retain the above copyright
// notice, this list of conditions and the following disclaimer.
// * Redistributions in binary form must reproduce the above
// copyright notice, this list of conditions and the following disclaimer
// in the documentation and/or other materials provided with the
// distribution.
// * Neither the name of Google Inc. nor the names of its
// contributors may be used to endorse or promote products derived from
// this software without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
#include "chrome_frame/npapi_url_request.h"
#include "base/string_util.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();
private:
unsigned long ref_count_;
NPP instance_;
NPStream* stream_;
size_t pending_read_size_;
URLRequestStatus status_;
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_(URLRequestStatus::FAILED, net::ERR_FAILED),
thread_(PlatformThread::CurrentId()) {
DLOG(INFO) << "Created request. Count: " << ++instance_count_;
}
NPAPIUrlRequest::~NPAPIUrlRequest() {
DLOG(INFO) << "Deleted request. Count: " << --instance_count_;
}
// NPAPIUrlRequest member defines.
bool NPAPIUrlRequest::Start() {
NPError result = NPERR_GENERIC_ERROR;
DLOG(INFO) << "Starting URL request: " << 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")) {
result = npapi::PostURLNotify(instance_, url().c_str(), NULL,
extra_headers().length(), extra_headers().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(),
URLRequestStatus(URLRequestStatus::FAILED, os_error));
return false;
}
return true;
}
void NPAPIUrlRequest::Stop() {
DLOG(INFO) << "Finished request: Url - " << url() << " result: "
<< static_cast<int>(status_.status());
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(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);
return NPERR_NO_ERROR;
}
NPError NPAPIUrlRequest::OnStreamDestroyed(NPReason reason) {
URLRequestStatus::Status status = URLRequestStatus::FAILED;
switch (reason) {
case NPRES_DONE:
status_.set_status(URLRequestStatus::SUCCESS);
status_.set_os_error(0);
break;
case NPRES_USER_BREAK:
status_.set_status(URLRequestStatus::CANCELED);
status_.set_os_error(net::ERR_ABORTED);
break;
case NPRES_NETWORK_ERR:
default:
status_.set_status(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(PlatformThread::CurrentId(), thread_);
++ref_count_;
return ref_count_;
}
STDMETHODIMP_(ULONG) NPAPIUrlRequest::Release() {
DCHECK_EQ(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 IPC::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,
enable_frame_busting_)) {
// Add to map.
DCHECK(request_map_.find(request_id) == request_map_.end());
request_map_[request_id] = new_request;
if (new_request->Start()) {
// 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) {
// Use the newer NPAPI way if available
if (npapi::VersionMinor() >= NPVERS_HAS_URL_AND_AUTH_INFO) {
npapi::SetValueForURL(instance_, NPNURLVCookie, url.spec().c_str(),
cookie.c_str(), cookie.length());
} else {
DLOG(INFO) << "Host does not support NPVERS_HAS_URL_AND_AUTH_INFO.";
DLOG(INFO) << "Attempting to set cookie using XPCOM cookie service";
if (np_utils::SetCookiesUsingXPCOMCookieService(instance_, url.spec(),
cookie)) {
DLOG(INFO) << "Successfully set cookies using XPCOM cookie service";
DLOG(INFO) << cookie.c_str();
} else {
NOTREACHED() << "Failed to set cookies for host";
}
}
}
void NPAPIUrlRequestManager::GetCookiesForUrl(const GURL& url, int cookie_id) {
std::string cookie_string;
bool success = true;
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) {
DLOG(INFO) << "Obtained cookies:" << cookies << " from host";
cookie_string.append(cookies, cookie_length);
npapi::MemFree(cookies);
} else {
success = false;
}
} else {
DLOG(INFO) << "Host does not support NPVERS_HAS_URL_AND_AUTH_INFO.";
DLOG(INFO) << "Attempting to read cookie using XPCOM cookie service";
if (np_utils::GetCookiesUsingXPCOMCookieService(instance_, url.spec(),
&cookie_string)) {
DLOG(INFO) << "Successfully read cookies using XPCOM cookie service";
DLOG(INFO) << cookie_string.c_str();
} else {
success = false;
}
}
if (!success)
DLOG(INFO) << "Failed to return cookies for url:" << url.spec().c_str();
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) {
delegate_->OnResponseStarted(request_id, mime_type, headers, size,
last_modified, redirect_url, redirect_status);
}
void NPAPIUrlRequestManager::OnReadComplete(int request_id,
const std::string& data) {
delegate_->OnReadComplete(request_id, data);
}
void NPAPIUrlRequestManager::OnResponseEnd(int request_id,
const 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() != 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;
DCHECK(request_map_.find(request->id()) != request_map_.end());
// 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;
DCHECK(request_map_.find(request->id()) != request_map_.end());
return request->OnStreamDestroyed(reason);
}
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();
}
}
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;
}