mirror/naiveproxy
1
0
Fork 0
mirror of https://github.com/klzgrad/naiveproxy.git synced 2024-11-24 14:26:09 +03:00
Code Issues Projects Releases Wiki Activity
c728759132
naiveproxy/net/proxy_resolution/proxy_resolver_v8.cc
klzgrad c677cbe429 Import chromium-70.0.3521.2
2018-08-14 22:19:20 +00:00

914 lines
33 KiB
C++
Raw Blame History

// 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 "net/proxy_resolution/proxy_resolver_v8.h"
#include <algorithm>
#include <cstdio>
#include <utility>
#include "base/auto_reset.h"
#include "base/compiler_specific.h"
#include "base/debug/leak_annotations.h"
#include "base/lazy_instance.h"
#include "base/logging.h"
#include "base/macros.h"
#include "base/strings/string_tokenizer.h"
#include "base/strings/string_util.h"
#include "base/strings/utf_string_conversions.h"
#include "base/synchronization/lock.h"
#include "base/threading/thread_task_runner_handle.h"
#include "gin/array_buffer.h"
#include "gin/public/isolate_holder.h"
#include "gin/v8_initializer.h"
#include "net/base/ip_address.h"
#include "net/base/net_errors.h"
#include "net/proxy_resolution/pac_file_data.h"
#include "net/proxy_resolution/pac_js_library.h"
#include "net/proxy_resolution/proxy_info.h"
#include "url/gurl.h"
#include "url/url_canon.h"
#include "v8/include/v8.h"
// Notes on the javascript environment:
//
// For the majority of the PAC utility functions, we use the same code
// as Firefox. See the javascript library that pac_js_library.h pulls in.
//
// In addition, we implement a subset of Microsoft's extensions to PAC.
// - myIpAddressEx()
// - dnsResolveEx()
// - isResolvableEx()
// - isInNetEx()
// - sortIpAddressList()
//
// It is worth noting that the original PAC specification does not describe
// the return values on failure. Consequently, there are compatibility
// differences between browsers on what to return on failure, which are
// illustrated below:
//
// --------------------+-------------+-------------------+--------------
// | Firefox3 | InternetExplorer8 | --> Us <---
// --------------------+-------------+-------------------+--------------
// myIpAddress() | "127.0.0.1" | ??? | "127.0.0.1"
// dnsResolve() | null | false | null
// myIpAddressEx() | N/A | "" | ""
// sortIpAddressList() | N/A | false | false
// dnsResolveEx() | N/A | "" | ""
// isInNetEx() | N/A | false | false
// --------------------+-------------+-------------------+--------------
//
// TODO(eroman): The cell above reading ??? means I didn't test it.
//
// Another difference is in how dnsResolve() and myIpAddress() are
// implemented -- whether they should restrict to IPv4 results, or
// include both IPv4 and IPv6. The following table illustrates the
// differences:
//
// --------------------+-------------+-------------------+--------------
// | Firefox3 | InternetExplorer8 | --> Us <---
// --------------------+-------------+-------------------+--------------
// myIpAddress() | IPv4/IPv6 | IPv4 | IPv4
// dnsResolve() | IPv4/IPv6 | IPv4 | IPv4
// isResolvable() | IPv4/IPv6 | IPv4 | IPv4
// myIpAddressEx() | N/A | IPv4/IPv6 | IPv4/IPv6
// dnsResolveEx() | N/A | IPv4/IPv6 | IPv4/IPv6
// sortIpAddressList() | N/A | IPv4/IPv6 | IPv4/IPv6
// isResolvableEx() | N/A | IPv4/IPv6 | IPv4/IPv6
// isInNetEx() | N/A | IPv4/IPv6 | IPv4/IPv6
// -----------------+-------------+-------------------+--------------
namespace net {
namespace {
// Pseudo-name for the PAC script.
const char kPacResourceName[] = "proxy-pac-script.js";
// Pseudo-name for the PAC utility script.
const char kPacUtilityResourceName[] = "proxy-pac-utility-script.js";
// External string wrapper so V8 can access the UTF16 string wrapped by
// PacFileData.
class V8ExternalStringFromScriptData
: public v8::String::ExternalStringResource {
public:
explicit V8ExternalStringFromScriptData(
const scoped_refptr<PacFileData>& script_data)
: script_data_(script_data) {}
const uint16_t* data() const override {
return reinterpret_cast<const uint16_t*>(script_data_->utf16().data());
}
size_t length() const override { return script_data_->utf16().size(); }
private:
const scoped_refptr<PacFileData> script_data_;
DISALLOW_COPY_AND_ASSIGN(V8ExternalStringFromScriptData);
};
// External string wrapper so V8 can access a string literal.
class V8ExternalASCIILiteral
: public v8::String::ExternalOneByteStringResource {
public:
// |ascii| must be a NULL-terminated C string, and must remain valid
// throughout this object's lifetime.
V8ExternalASCIILiteral(const char* ascii, size_t length)
: ascii_(ascii), length_(length) {
DCHECK(base::IsStringASCII(ascii));
}
const char* data() const override { return ascii_; }
size_t length() const override { return length_; }
private:
const char* ascii_;
size_t length_;
DISALLOW_COPY_AND_ASSIGN(V8ExternalASCIILiteral);
};
// When creating a v8::String from a C++ string we have two choices: create
// a copy, or create a wrapper that shares the same underlying storage.
// For small strings it is better to just make a copy, whereas for large
// strings there are savings by sharing the storage. This number identifies
// the cutoff length for when to start wrapping rather than creating copies.
const size_t kMaxStringBytesForCopy = 256;
// Converts a V8 String to a UTF8 std::string.
std::string V8StringToUTF8(v8::Isolate* isolate, v8::Local<v8::String> s) {
int len = s->Length();
std::string result;
if (len > 0)
s->WriteUtf8(isolate, base::WriteInto(&result, len + 1));
return result;
}
// Converts a V8 String to a UTF16 base::string16.
base::string16 V8StringToUTF16(v8::Isolate* isolate, v8::Local<v8::String> s) {
int len = s->Length();
base::string16 result;
// Note that the reinterpret cast is because on Windows string16 is an alias
// to wstring, and hence has character type wchar_t not uint16_t.
if (len > 0) {
s->Write(isolate,
reinterpret_cast<uint16_t*>(base::WriteInto(&result, len + 1)), 0,
len);
}
return result;
}
// Converts an ASCII std::string to a V8 string.
v8::Local<v8::String> ASCIIStringToV8String(v8::Isolate* isolate,
const std::string& s) {
DCHECK(base::IsStringASCII(s));
return v8::String::NewFromUtf8(isolate, s.data(), v8::NewStringType::kNormal,
s.size()).ToLocalChecked();
}
// Converts a UTF16 base::string16 (wrapped by a PacFileData) to a
// V8 string.
v8::Local<v8::String> ScriptDataToV8String(
v8::Isolate* isolate,
const scoped_refptr<PacFileData>& s) {
if (s->utf16().size() * 2 <= kMaxStringBytesForCopy) {
return v8::String::NewFromTwoByte(
isolate, reinterpret_cast<const uint16_t*>(s->utf16().data()),
v8::NewStringType::kNormal, s->utf16().size()).ToLocalChecked();
}
return v8::String::NewExternalTwoByte(
isolate, new V8ExternalStringFromScriptData(s)).ToLocalChecked();
}
// Converts an ASCII string literal to a V8 string.
v8::Local<v8::String> ASCIILiteralToV8String(v8::Isolate* isolate,
const char* ascii) {
DCHECK(base::IsStringASCII(ascii));
size_t length = strlen(ascii);
if (length <= kMaxStringBytesForCopy)
return v8::String::NewFromUtf8(isolate, ascii, v8::NewStringType::kNormal,
length).ToLocalChecked();
return v8::String::NewExternalOneByte(
isolate, new V8ExternalASCIILiteral(ascii, length))
.ToLocalChecked();
}
// Stringizes a V8 object by calling its toString() method. Returns true
// on success. This may fail if the toString() throws an exception.
bool V8ObjectToUTF16String(v8::Local<v8::Value> object,
base::string16* utf16_result,
v8::Isolate* isolate) {
if (object.IsEmpty())
return false;
v8::HandleScope scope(isolate);
v8::Local<v8::String> str_object;
if (!object->ToString(isolate->GetCurrentContext()).ToLocal(&str_object))
return false;
*utf16_result = V8StringToUTF16(isolate, str_object);
return true;
}
// Extracts an hostname argument from |args|. On success returns true
// and fills |*hostname| with the result.
bool GetHostnameArgument(const v8::FunctionCallbackInfo<v8::Value>& args,
std::string* hostname) {
// The first argument should be a string.
if (args.Length() == 0 || args[0].IsEmpty() || !args[0]->IsString())
return false;
const base::string16 hostname_utf16 =
V8StringToUTF16(args.GetIsolate(), v8::Local<v8::String>::Cast(args[0]));
// If the hostname is already in ASCII, simply return it as is.
if (base::IsStringASCII(hostname_utf16)) {
*hostname = base::UTF16ToASCII(hostname_utf16);
return true;
}
// Otherwise try to convert it from IDN to punycode.
const int kInitialBufferSize = 256;
url::RawCanonOutputT<base::char16, kInitialBufferSize> punycode_output;
if (!url::IDNToASCII(hostname_utf16.data(), hostname_utf16.length(),
&punycode_output)) {
return false;
}
// |punycode_output| should now be ASCII; convert it to a std::string.
// (We could use UTF16ToASCII() instead, but that requires an extra string
// copy. Since ASCII is a subset of UTF8 the following is equivalent).
bool success = base::UTF16ToUTF8(punycode_output.data(),
punycode_output.length(),
hostname);
DCHECK(success);
DCHECK(base::IsStringASCII(*hostname));
return success;
}
// Wrapper around an IP address that stores the original string as well as a
// corresponding parsed IPAddress.
// This struct is used as a helper for sorting IP address strings - the IP
// literal is parsed just once and used as the sorting key, while also
// preserving the original IP literal string.
struct IPAddressSortingEntry {
IPAddressSortingEntry(const std::string& ip_string,
const IPAddress& ip_address)
: string_value(ip_string), ip_address(ip_address) {}
// Used for sorting IP addresses in ascending order in SortIpAddressList().
// IPv6 addresses are placed ahead of IPv4 addresses.
bool operator<(const IPAddressSortingEntry& rhs) const {
const IPAddress& ip1 = this->ip_address;
const IPAddress& ip2 = rhs.ip_address;
if (ip1.size() != ip2.size())
return ip1.size() > ip2.size(); // IPv6 before IPv4.
return ip1 < ip2; // Ascending order.
}
std::string string_value;
IPAddress ip_address;
};
// Handler for "sortIpAddressList(IpAddressList)". |ip_address_list| is a
// semi-colon delimited string containing IP addresses.
// |sorted_ip_address_list| is the resulting list of sorted semi-colon delimited
// IP addresses or an empty string if unable to sort the IP address list.
// Returns 'true' if the sorting was successful, and 'false' if the input was an
// empty string, a string of separators (";" in this case), or if any of the IP
// addresses in the input list failed to parse.
bool SortIpAddressList(const std::string& ip_address_list,
std::string* sorted_ip_address_list) {
sorted_ip_address_list->clear();
// Strip all whitespace (mimics IE behavior).
std::string cleaned_ip_address_list;
base::RemoveChars(ip_address_list, " \t", &cleaned_ip_address_list);
if (cleaned_ip_address_list.empty())
return false;
// Split-up IP addresses and store them in a vector.
std::vector<IPAddressSortingEntry> ip_vector;
IPAddress ip_address;
base::StringTokenizer str_tok(cleaned_ip_address_list, ";");
while (str_tok.GetNext()) {
if (!ip_address.AssignFromIPLiteral(str_tok.token()))
return false;
ip_vector.push_back(IPAddressSortingEntry(str_tok.token(), ip_address));
}
if (ip_vector.empty()) // Can happen if we have something like
return false; // sortIpAddressList(";") or sortIpAddressList("; ;")
DCHECK(!ip_vector.empty());
// Sort lists according to ascending numeric value.
if (ip_vector.size() > 1)
std::stable_sort(ip_vector.begin(), ip_vector.end());
// Return a semi-colon delimited list of sorted addresses (IPv6 followed by
// IPv4).
for (size_t i = 0; i < ip_vector.size(); ++i) {
if (i > 0)
*sorted_ip_address_list += ";";
*sorted_ip_address_list += ip_vector[i].string_value;
}
return true;
}
// Handler for "isInNetEx(ip_address, ip_prefix)". |ip_address| is a string
// containing an IPv4/IPv6 address, and |ip_prefix| is a string containg a
// slash-delimited IP prefix with the top 'n' bits specified in the bit
// field. This returns 'true' if the address is in the same subnet, and
// 'false' otherwise. Also returns 'false' if the prefix is in an incorrect
// format, or if an address and prefix of different types are used (e.g. IPv6
// address and IPv4 prefix).
bool IsInNetEx(const std::string& ip_address, const std::string& ip_prefix) {
IPAddress address;
if (!address.AssignFromIPLiteral(ip_address))
return false;
IPAddress prefix;
size_t prefix_length_in_bits;
if (!ParseCIDRBlock(ip_prefix, &prefix, &prefix_length_in_bits))
return false;
// Both |address| and |prefix| must be of the same type (IPv4 or IPv6).
if (address.size() != prefix.size())
return false;
DCHECK((address.IsIPv4() && prefix.IsIPv4()) ||
(address.IsIPv6() && prefix.IsIPv6()));
return IPAddressMatchesPrefix(address, prefix, prefix_length_in_bits);
}
// Consider only single component domains like 'foo' as plain host names.
bool IsPlainHostName(const std::string& hostname_utf8) {
if (hostname_utf8.find('.') != std::string::npos)
return false;
// IPv6 literals might not contain any periods, however are not considered
// plain host names.
IPAddress unused;
return !unused.AssignFromIPLiteral(hostname_utf8);
}
// All instances of ProxyResolverV8 share the same v8::Isolate. This isolate is
// created lazily the first time it is needed and lives until process shutdown.
// This creation might happen from any thread, as ProxyResolverV8 is typically
// run in a threadpool.
//
// TODO(eroman): The lazily created isolate is never freed. Instead it should be
// disposed once there are no longer any ProxyResolverV8 referencing it.
class SharedIsolateFactory {
public:
SharedIsolateFactory() : has_initialized_v8_(false) {}
// Lazily creates a v8::Isolate, or returns the already created instance.
v8::Isolate* GetSharedIsolate() {
base::AutoLock lock(lock_);
if (!holder_) {
// Do one-time initialization for V8.
if (!has_initialized_v8_) {
#ifdef V8_USE_EXTERNAL_STARTUP_DATA
gin::V8Initializer::LoadV8Snapshot();
gin::V8Initializer::LoadV8Natives();
#endif
// The performance of the proxy resolver is limited by DNS resolution,
// and not V8, so tune down V8 to use as little memory as possible.
static const char kOptimizeForSize[] = "--optimize_for_size";
v8::V8::SetFlagsFromString(kOptimizeForSize, strlen(kOptimizeForSize));
static const char kNoOpt[] = "--noopt";
v8::V8::SetFlagsFromString(kNoOpt, strlen(kNoOpt));
gin::IsolateHolder::Initialize(
gin::IsolateHolder::kNonStrictMode,
gin::IsolateHolder::kStableV8Extras,
gin::ArrayBufferAllocator::SharedInstance());
has_initialized_v8_ = true;
}
holder_.reset(new gin::IsolateHolder(base::ThreadTaskRunnerHandle::Get(),
gin::IsolateHolder::kUseLocker));
}
return holder_->isolate();
}
v8::Isolate* GetSharedIsolateWithoutCreating() {
base::AutoLock lock(lock_);
return holder_ ? holder_->isolate() : NULL;
}
private:
base::Lock lock_;
std::unique_ptr<gin::IsolateHolder> holder_;
bool has_initialized_v8_;
DISALLOW_COPY_AND_ASSIGN(SharedIsolateFactory);
};
base::LazyInstance<SharedIsolateFactory>::Leaky g_isolate_factory =
LAZY_INSTANCE_INITIALIZER;
} // namespace
// ProxyResolverV8::Context ---------------------------------------------------
class ProxyResolverV8::Context {
public:
explicit Context(v8::Isolate* isolate)
: js_bindings_(nullptr), isolate_(isolate) {
DCHECK(isolate);
}
~Context() {
v8::Locker locked(isolate_);
v8::Isolate::Scope isolate_scope(isolate_);
v8_this_.Reset();
v8_context_.Reset();
}
JSBindings* js_bindings() { return js_bindings_; }
int ResolveProxy(const GURL& query_url,
ProxyInfo* results,
JSBindings* bindings) {
DCHECK(bindings);
base::AutoReset<JSBindings*> bindings_reset(&js_bindings_, bindings);
v8::Locker locked(isolate_);
v8::Isolate::Scope isolate_scope(isolate_);
v8::Isolate::SafeForTerminationScope safe_for_termination(isolate_);
v8::HandleScope scope(isolate_);
v8::Local<v8::Context> context =
v8::Local<v8::Context>::New(isolate_, v8_context_);
v8::Context::Scope function_scope(context);
v8::Local<v8::Value> function;
int rv = GetFindProxyForURL(&function);
if (rv != OK)
return rv;
v8::Local<v8::Value> argv[] = {
ASCIIStringToV8String(isolate_, query_url.spec()),
ASCIIStringToV8String(isolate_, query_url.HostNoBrackets()),
};
v8::TryCatch try_catch(isolate_);
v8::Local<v8::Value> ret;
if (!v8::Function::Cast(*function)
->Call(context, context->Global(), arraysize(argv), argv)
.ToLocal(&ret)) {
DCHECK(try_catch.HasCaught());
HandleError(try_catch.Message());
return ERR_PAC_SCRIPT_FAILED;
}
if (!ret->IsString()) {
js_bindings()->OnError(
-1, base::ASCIIToUTF16("FindProxyForURL() did not return a string."));
return ERR_PAC_SCRIPT_FAILED;
}
base::string16 ret_str =
V8StringToUTF16(isolate_, v8::Local<v8::String>::Cast(ret));
if (!base::IsStringASCII(ret_str)) {
// TODO(eroman): Rather than failing when a wide string is returned, we
// could extend the parsing to handle IDNA hostnames by
// converting them to ASCII punycode.
// crbug.com/47234
base::string16 error_message =
base::ASCIIToUTF16("FindProxyForURL() returned a non-ASCII string "
"(crbug.com/47234): ") + ret_str;
js_bindings()->OnError(-1, error_message);
return ERR_PAC_SCRIPT_FAILED;
}
results->UsePacString(base::UTF16ToASCII(ret_str));
return OK;
}
int InitV8(const scoped_refptr<PacFileData>& pac_script,
JSBindings* bindings) {
base::AutoReset<JSBindings*> bindings_reset(&js_bindings_, bindings);
v8::Locker locked(isolate_);
v8::Isolate::Scope isolate_scope(isolate_);
v8::HandleScope scope(isolate_);
v8_this_.Reset(isolate_, v8::External::New(isolate_, this));
v8::Local<v8::External> v8_this =
v8::Local<v8::External>::New(isolate_, v8_this_);
v8::Local<v8::ObjectTemplate> global_template =
v8::ObjectTemplate::New(isolate_);
// Attach the javascript bindings.
v8::Local<v8::FunctionTemplate> alert_template =
v8::FunctionTemplate::New(isolate_, &AlertCallback, v8_this);
alert_template->RemovePrototype();
global_template->Set(ASCIILiteralToV8String(isolate_, "alert"),
alert_template);
v8::Local<v8::FunctionTemplate> my_ip_address_template =
v8::FunctionTemplate::New(isolate_, &MyIpAddressCallback, v8_this);
my_ip_address_template->RemovePrototype();
global_template->Set(ASCIILiteralToV8String(isolate_, "myIpAddress"),
my_ip_address_template);
v8::Local<v8::FunctionTemplate> dns_resolve_template =
v8::FunctionTemplate::New(isolate_, &DnsResolveCallback, v8_this);
dns_resolve_template->RemovePrototype();
global_template->Set(ASCIILiteralToV8String(isolate_, "dnsResolve"),
dns_resolve_template);
v8::Local<v8::FunctionTemplate> is_plain_host_name_template =
v8::FunctionTemplate::New(isolate_, &IsPlainHostNameCallback, v8_this);
is_plain_host_name_template->RemovePrototype();
global_template->Set(ASCIILiteralToV8String(isolate_, "isPlainHostName"),
is_plain_host_name_template);
// Microsoft's PAC extensions:
v8::Local<v8::FunctionTemplate> dns_resolve_ex_template =
v8::FunctionTemplate::New(isolate_, &DnsResolveExCallback, v8_this);
dns_resolve_ex_template->RemovePrototype();
global_template->Set(ASCIILiteralToV8String(isolate_, "dnsResolveEx"),
dns_resolve_ex_template);
v8::Local<v8::FunctionTemplate> my_ip_address_ex_template =
v8::FunctionTemplate::New(isolate_, &MyIpAddressExCallback, v8_this);
my_ip_address_ex_template->RemovePrototype();
global_template->Set(ASCIILiteralToV8String(isolate_, "myIpAddressEx"),
my_ip_address_ex_template);
v8::Local<v8::FunctionTemplate> sort_ip_address_list_template =
v8::FunctionTemplate::New(isolate_,
&SortIpAddressListCallback,
v8_this);
sort_ip_address_list_template->RemovePrototype();
global_template->Set(ASCIILiteralToV8String(isolate_, "sortIpAddressList"),
sort_ip_address_list_template);
v8::Local<v8::FunctionTemplate> is_in_net_ex_template =
v8::FunctionTemplate::New(isolate_, &IsInNetExCallback, v8_this);
is_in_net_ex_template->RemovePrototype();
global_template->Set(ASCIILiteralToV8String(isolate_, "isInNetEx"),
is_in_net_ex_template);
v8_context_.Reset(
isolate_, v8::Context::New(isolate_, NULL, global_template));
v8::Local<v8::Context> context =
v8::Local<v8::Context>::New(isolate_, v8_context_);
v8::Context::Scope ctx(context);
// Add the PAC utility functions to the environment.
// (This script should never fail, as it is a string literal!)
// Note that the two string literals are concatenated.
int rv = RunScript(
ASCIILiteralToV8String(isolate_, PAC_JS_LIBRARY PAC_JS_LIBRARY_EX),
kPacUtilityResourceName);
if (rv != OK) {
NOTREACHED();
return rv;
}
// Add the user's PAC code to the environment.
rv =
RunScript(ScriptDataToV8String(isolate_, pac_script), kPacResourceName);
if (rv != OK)
return rv;
// At a minimum, the FindProxyForURL() function must be defined for this
// to be a legitimiate PAC script.
v8::Local<v8::Value> function;
return GetFindProxyForURL(&function);
}
private:
int GetFindProxyForURL(v8::Local<v8::Value>* function) {
v8::Local<v8::Context> context =
v8::Local<v8::Context>::New(isolate_, v8_context_);
v8::TryCatch try_catch(isolate_);
if (!context->Global()
->Get(context, ASCIILiteralToV8String(isolate_, "FindProxyForURL"))
.ToLocal(function)) {
DCHECK(try_catch.HasCaught());
HandleError(try_catch.Message());
}
// The value should only be empty if an exception was thrown. Code
// defensively just in case.
DCHECK_EQ(function->IsEmpty(), try_catch.HasCaught());
if (function->IsEmpty() || try_catch.HasCaught()) {
js_bindings()->OnError(
-1,
base::ASCIIToUTF16("Accessing FindProxyForURL threw an exception."));
return ERR_PAC_SCRIPT_FAILED;
}
if (!(*function)->IsFunction()) {
js_bindings()->OnError(
-1, base::ASCIIToUTF16(
"FindProxyForURL is undefined or not a function."));
return ERR_PAC_SCRIPT_FAILED;
}
return OK;
}
// Handle an exception thrown by V8.
void HandleError(v8::Local<v8::Message> message) {
v8::Local<v8::Context> context =
v8::Local<v8::Context>::New(isolate_, v8_context_);
base::string16 error_message;
int line_number = -1;
if (!message.IsEmpty()) {
auto maybe = message->GetLineNumber(context);
if (maybe.IsJust())
line_number = maybe.FromJust();
V8ObjectToUTF16String(message->Get(), &error_message, isolate_);
}
js_bindings()->OnError(line_number, error_message);
}
// Compiles and runs |script| in the current V8 context.
// Returns OK on success, otherwise an error code.
int RunScript(v8::Local<v8::String> script, const char* script_name) {
v8::Local<v8::Context> context =
v8::Local<v8::Context>::New(isolate_, v8_context_);
v8::TryCatch try_catch(isolate_);
// Compile the script.
v8::ScriptOrigin origin =
v8::ScriptOrigin(ASCIILiteralToV8String(isolate_, script_name));
v8::ScriptCompiler::Source script_source(script, origin);
v8::Local<v8::Script> code;
if (!v8::ScriptCompiler::Compile(
context, &script_source, v8::ScriptCompiler::kNoCompileOptions,
v8::ScriptCompiler::NoCacheReason::kNoCacheBecausePacScript)
.ToLocal(&code)) {
DCHECK(try_catch.HasCaught());
HandleError(try_catch.Message());
return ERR_PAC_SCRIPT_FAILED;
}
// Execute.
auto result = code->Run(context);
if (result.IsEmpty()) {
DCHECK(try_catch.HasCaught());
HandleError(try_catch.Message());
return ERR_PAC_SCRIPT_FAILED;
}
return OK;
}
// V8 callback for when "alert()" is invoked by the PAC script.
static void AlertCallback(const v8::FunctionCallbackInfo<v8::Value>& args) {
Context* context =
static_cast<Context*>(v8::External::Cast(*args.Data())->Value());
// Like firefox we assume "undefined" if no argument was specified, and
// disregard any arguments beyond the first.
base::string16 message;
if (args.Length() == 0) {
message = base::ASCIIToUTF16("undefined");
} else {
if (!V8ObjectToUTF16String(args[0], &message, args.GetIsolate()))
return; // toString() threw an exception.
}
context->js_bindings()->Alert(message);
}
// V8 callback for when "myIpAddress()" is invoked by the PAC script.
static void MyIpAddressCallback(
const v8::FunctionCallbackInfo<v8::Value>& args) {
DnsResolveCallbackHelper(args, JSBindings::MY_IP_ADDRESS);
}
// V8 callback for when "myIpAddressEx()" is invoked by the PAC script.
static void MyIpAddressExCallback(
const v8::FunctionCallbackInfo<v8::Value>& args) {
DnsResolveCallbackHelper(args, JSBindings::MY_IP_ADDRESS_EX);
}
// V8 callback for when "dnsResolve()" is invoked by the PAC script.
static void DnsResolveCallback(
const v8::FunctionCallbackInfo<v8::Value>& args) {
DnsResolveCallbackHelper(args, JSBindings::DNS_RESOLVE);
}
// V8 callback for when "dnsResolveEx()" is invoked by the PAC script.
static void DnsResolveExCallback(
const v8::FunctionCallbackInfo<v8::Value>& args) {
DnsResolveCallbackHelper(args, JSBindings::DNS_RESOLVE_EX);
}
// Shared code for implementing:
// - myIpAddress(), myIpAddressEx(), dnsResolve(), dnsResolveEx().
static void DnsResolveCallbackHelper(
const v8::FunctionCallbackInfo<v8::Value>& args,
JSBindings::ResolveDnsOperation op) {
Context* context =
static_cast<Context*>(v8::External::Cast(*args.Data())->Value());
std::string hostname;
// dnsResolve() and dnsResolveEx() need at least 1 argument.
if (op == JSBindings::DNS_RESOLVE || op == JSBindings::DNS_RESOLVE_EX) {
if (!GetHostnameArgument(args, &hostname)) {
if (op == JSBindings::DNS_RESOLVE)
args.GetReturnValue().SetNull();
return;
}
}
std::string result;
bool success;
bool terminate = false;
{
v8::Unlocker unlocker(args.GetIsolate());
success = context->js_bindings()->ResolveDns(
hostname, op, &result, &terminate);
}
if (terminate)
args.GetIsolate()->TerminateExecution();
if (success) {
args.GetReturnValue().Set(
ASCIIStringToV8String(args.GetIsolate(), result));
return;
}
// Each function handles resolution errors differently.
switch (op) {
case JSBindings::DNS_RESOLVE:
args.GetReturnValue().SetNull();
return;
case JSBindings::DNS_RESOLVE_EX:
args.GetReturnValue().SetEmptyString();
return;
case JSBindings::MY_IP_ADDRESS:
args.GetReturnValue().Set(
ASCIILiteralToV8String(args.GetIsolate(), "127.0.0.1"));
return;
case JSBindings::MY_IP_ADDRESS_EX:
args.GetReturnValue().SetEmptyString();
return;
}
NOTREACHED();
}
// V8 callback for when "sortIpAddressList()" is invoked by the PAC script.
static void SortIpAddressListCallback(
const v8::FunctionCallbackInfo<v8::Value>& args) {
// We need at least one string argument.
if (args.Length() == 0 || args[0].IsEmpty() || !args[0]->IsString()) {
args.GetReturnValue().SetNull();
return;
}
std::string ip_address_list =
V8StringToUTF8(args.GetIsolate(), v8::Local<v8::String>::Cast(args[0]));
if (!base::IsStringASCII(ip_address_list)) {
args.GetReturnValue().SetNull();
return;
}
std::string sorted_ip_address_list;
bool success = SortIpAddressList(ip_address_list, &sorted_ip_address_list);
if (!success) {
args.GetReturnValue().Set(false);
return;
}
args.GetReturnValue().Set(
ASCIIStringToV8String(args.GetIsolate(), sorted_ip_address_list));
}
// V8 callback for when "isInNetEx()" is invoked by the PAC script.
static void IsInNetExCallback(
const v8::FunctionCallbackInfo<v8::Value>& args) {
// We need at least 2 string arguments.
if (args.Length() < 2 || args[0].IsEmpty() || !args[0]->IsString() ||
args[1].IsEmpty() || !args[1]->IsString()) {
args.GetReturnValue().SetNull();
return;
}
std::string ip_address =
V8StringToUTF8(args.GetIsolate(), v8::Local<v8::String>::Cast(args[0]));
if (!base::IsStringASCII(ip_address)) {
args.GetReturnValue().Set(false);
return;
}
std::string ip_prefix =
V8StringToUTF8(args.GetIsolate(), v8::Local<v8::String>::Cast(args[1]));
if (!base::IsStringASCII(ip_prefix)) {
args.GetReturnValue().Set(false);
return;
}
args.GetReturnValue().Set(IsInNetEx(ip_address, ip_prefix));
}
// V8 callback for when "isPlainHostName()" is invoked by the PAC script.
static void IsPlainHostNameCallback(
const v8::FunctionCallbackInfo<v8::Value>& args) {
// Need at least 1 string arguments.
if (args.Length() < 1 || args[0].IsEmpty() || !args[0]->IsString()) {
args.GetIsolate()->ThrowException(
v8::Exception::TypeError(ASCIIStringToV8String(
args.GetIsolate(), "Requires 1 string parameter")));
return;
}
std::string hostname_utf8 =
V8StringToUTF8(args.GetIsolate(), v8::Local<v8::String>::Cast(args[0]));
args.GetReturnValue().Set(IsPlainHostName(hostname_utf8));
}
mutable base::Lock lock_;
ProxyResolverV8::JSBindings* js_bindings_;
v8::Isolate* isolate_;
v8::Persistent<v8::External> v8_this_;
v8::Persistent<v8::Context> v8_context_;
};
// ProxyResolverV8 ------------------------------------------------------------
ProxyResolverV8::ProxyResolverV8(std::unique_ptr<Context> context)
: context_(std::move(context)) {
DCHECK(context_);
}
ProxyResolverV8::~ProxyResolverV8() = default;
int ProxyResolverV8::GetProxyForURL(const GURL& query_url,
ProxyInfo* results,
ProxyResolverV8::JSBindings* bindings) {
return context_->ResolveProxy(query_url, results, bindings);
}
// static
int ProxyResolverV8::Create(const scoped_refptr<PacFileData>& script_data,
ProxyResolverV8::JSBindings* js_bindings,
std::unique_ptr<ProxyResolverV8>* resolver) {
DCHECK(script_data.get());
DCHECK(js_bindings);
if (script_data->utf16().empty())
return ERR_PAC_SCRIPT_FAILED;
// Try parsing the PAC script.
std::unique_ptr<Context> context(
new Context(g_isolate_factory.Get().GetSharedIsolate()));
int rv = context->InitV8(script_data, js_bindings);
if (rv == OK)
resolver->reset(new ProxyResolverV8(std::move(context)));
return rv;
}
// static
size_t ProxyResolverV8::GetTotalHeapSize() {
v8::Isolate* isolate =
g_isolate_factory.Get().GetSharedIsolateWithoutCreating();
if (!isolate)
return 0;
v8::Locker locked(isolate);
v8::Isolate::Scope isolate_scope(isolate);
v8::HeapStatistics heap_statistics;
isolate->GetHeapStatistics(&heap_statistics);
return heap_statistics.total_heap_size();
}
// static
size_t ProxyResolverV8::GetUsedHeapSize() {
v8::Isolate* isolate =
g_isolate_factory.Get().GetSharedIsolateWithoutCreating();
if (!isolate)
return 0;
v8::Locker locked(isolate);
v8::Isolate::Scope isolate_scope(isolate);
v8::HeapStatistics heap_statistics;
isolate->GetHeapStatistics(&heap_statistics);
return heap_statistics.used_heap_size();
}
} // namespace net
Reference in New Issue View Git Blame Copy Permalink