123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847 |
- // Copyright (c) 2015 GitHub, Inc.
- // Use of this source code is governed by the MIT license that can be
- // found in the LICENSE file.
- #include "atom/browser/api/atom_api_session.h"
- #include <map>
- #include <memory>
- #include <string>
- #include <utility>
- #include <vector>
- #include "atom/browser/api/atom_api_cookies.h"
- #include "atom/browser/api/atom_api_download_item.h"
- #include "atom/browser/api/atom_api_net_log.h"
- #include "atom/browser/api/atom_api_protocol.h"
- #include "atom/browser/api/atom_api_web_request.h"
- #include "atom/browser/atom_browser_context.h"
- #include "atom/browser/atom_browser_main_parts.h"
- #include "atom/browser/atom_permission_manager.h"
- #include "atom/browser/browser.h"
- #include "atom/browser/media/media_device_id_salt.h"
- #include "atom/browser/net/atom_cert_verifier.h"
- #include "atom/browser/session_preferences.h"
- #include "atom/common/native_mate_converters/callback.h"
- #include "atom/common/native_mate_converters/content_converter.h"
- #include "atom/common/native_mate_converters/file_path_converter.h"
- #include "atom/common/native_mate_converters/gurl_converter.h"
- #include "atom/common/native_mate_converters/net_converter.h"
- #include "atom/common/native_mate_converters/value_converter.h"
- #include "atom/common/node_includes.h"
- #include "base/files/file_path.h"
- #include "base/guid.h"
- #include "base/strings/string_number_conversions.h"
- #include "base/strings/string_util.h"
- #include "base/task/post_task.h"
- #include "chrome/browser/browser_process.h"
- #include "chrome/common/pref_names.h"
- #include "components/download/public/common/download_danger_type.h"
- #include "components/prefs/pref_service.h"
- #include "components/prefs/value_map_pref_store.h"
- #include "components/proxy_config/proxy_config_dictionary.h"
- #include "components/proxy_config/proxy_config_pref_names.h"
- #include "content/public/browser/browser_task_traits.h"
- #include "content/public/browser/browser_thread.h"
- #include "content/public/browser/download_item_utils.h"
- #include "content/public/browser/download_manager_delegate.h"
- #include "content/public/browser/storage_partition.h"
- #include "native_mate/dictionary.h"
- #include "native_mate/object_template_builder.h"
- #include "net/base/completion_repeating_callback.h"
- #include "net/base/load_flags.h"
- #include "net/http/http_auth_handler_factory.h"
- #include "net/http/http_auth_preferences.h"
- #include "net/http/http_cache.h"
- #include "net/http/http_transaction_factory.h"
- #include "net/url_request/static_http_user_agent_settings.h"
- #include "net/url_request/url_request_context.h"
- #include "net/url_request/url_request_context_getter.h"
- #include "ui/base/l10n/l10n_util.h"
- using content::BrowserThread;
- using content::StoragePartition;
- namespace {
- struct ClearStorageDataOptions {
- GURL origin;
- uint32_t storage_types = StoragePartition::REMOVE_DATA_MASK_ALL;
- uint32_t quota_types = StoragePartition::QUOTA_MANAGED_STORAGE_MASK_ALL;
- };
- struct ClearAuthCacheOptions {
- std::string type;
- GURL origin;
- std::string realm;
- base::string16 username;
- base::string16 password;
- net::HttpAuth::Scheme auth_scheme;
- };
- uint32_t GetStorageMask(const std::vector<std::string>& storage_types) {
- uint32_t storage_mask = 0;
- for (const auto& it : storage_types) {
- auto type = base::ToLowerASCII(it);
- if (type == "appcache")
- storage_mask |= StoragePartition::REMOVE_DATA_MASK_APPCACHE;
- else if (type == "cookies")
- storage_mask |= StoragePartition::REMOVE_DATA_MASK_COOKIES;
- else if (type == "filesystem")
- storage_mask |= StoragePartition::REMOVE_DATA_MASK_FILE_SYSTEMS;
- else if (type == "indexdb")
- storage_mask |= StoragePartition::REMOVE_DATA_MASK_INDEXEDDB;
- else if (type == "localstorage")
- storage_mask |= StoragePartition::REMOVE_DATA_MASK_LOCAL_STORAGE;
- else if (type == "shadercache")
- storage_mask |= StoragePartition::REMOVE_DATA_MASK_SHADER_CACHE;
- else if (type == "websql")
- storage_mask |= StoragePartition::REMOVE_DATA_MASK_WEBSQL;
- else if (type == "serviceworkers")
- storage_mask |= StoragePartition::REMOVE_DATA_MASK_SERVICE_WORKERS;
- else if (type == "cachestorage")
- storage_mask |= StoragePartition::REMOVE_DATA_MASK_CACHE_STORAGE;
- }
- return storage_mask;
- }
- uint32_t GetQuotaMask(const std::vector<std::string>& quota_types) {
- uint32_t quota_mask = 0;
- for (const auto& it : quota_types) {
- auto type = base::ToLowerASCII(it);
- if (type == "temporary")
- quota_mask |= StoragePartition::QUOTA_MANAGED_STORAGE_MASK_TEMPORARY;
- else if (type == "persistent")
- quota_mask |= StoragePartition::QUOTA_MANAGED_STORAGE_MASK_PERSISTENT;
- else if (type == "syncable")
- quota_mask |= StoragePartition::QUOTA_MANAGED_STORAGE_MASK_SYNCABLE;
- }
- return quota_mask;
- }
- net::HttpAuth::Scheme GetAuthSchemeFromString(const std::string& scheme) {
- if (scheme == "basic")
- return net::HttpAuth::AUTH_SCHEME_BASIC;
- if (scheme == "digest")
- return net::HttpAuth::AUTH_SCHEME_DIGEST;
- if (scheme == "ntlm")
- return net::HttpAuth::AUTH_SCHEME_NTLM;
- if (scheme == "negotiate")
- return net::HttpAuth::AUTH_SCHEME_NEGOTIATE;
- return net::HttpAuth::AUTH_SCHEME_MAX;
- }
- void SetUserAgentInIO(scoped_refptr<net::URLRequestContextGetter> getter,
- const std::string& accept_lang,
- const std::string& user_agent) {
- getter->GetURLRequestContext()->set_http_user_agent_settings(
- new net::StaticHttpUserAgentSettings(
- net::HttpUtil::GenerateAcceptLanguageHeader(accept_lang),
- user_agent));
- }
- } // namespace
- namespace mate {
- template <>
- struct Converter<ClearStorageDataOptions> {
- static bool FromV8(v8::Isolate* isolate,
- v8::Local<v8::Value> val,
- ClearStorageDataOptions* out) {
- mate::Dictionary options;
- if (!ConvertFromV8(isolate, val, &options))
- return false;
- options.Get("origin", &out->origin);
- std::vector<std::string> types;
- if (options.Get("storages", &types))
- out->storage_types = GetStorageMask(types);
- if (options.Get("quotas", &types))
- out->quota_types = GetQuotaMask(types);
- return true;
- }
- };
- template <>
- struct Converter<ClearAuthCacheOptions> {
- static bool FromV8(v8::Isolate* isolate,
- v8::Local<v8::Value> val,
- ClearAuthCacheOptions* out) {
- mate::Dictionary options;
- if (!ConvertFromV8(isolate, val, &options))
- return false;
- options.Get("type", &out->type);
- options.Get("origin", &out->origin);
- options.Get("realm", &out->realm);
- options.Get("username", &out->username);
- options.Get("password", &out->password);
- std::string scheme;
- if (options.Get("scheme", &scheme))
- out->auth_scheme = GetAuthSchemeFromString(scheme);
- return true;
- }
- };
- template <>
- struct Converter<atom::VerifyRequestParams> {
- static v8::Local<v8::Value> ToV8(v8::Isolate* isolate,
- atom::VerifyRequestParams val) {
- mate::Dictionary dict = mate::Dictionary::CreateEmpty(isolate);
- dict.Set("hostname", val.hostname);
- dict.Set("certificate", val.certificate);
- dict.Set("verificationResult", val.default_result);
- dict.Set("errorCode", val.error_code);
- return dict.GetHandle();
- }
- };
- } // namespace mate
- namespace atom {
- namespace api {
- namespace {
- const char kPersistPrefix[] = "persist:";
- // Referenced session objects.
- std::map<uint32_t, v8::Global<v8::Object>> g_sessions;
- void SetCertVerifyProcInIO(
- const scoped_refptr<net::URLRequestContextGetter>& context_getter,
- const AtomCertVerifier::VerifyProc& proc) {
- auto* request_context = context_getter->GetURLRequestContext();
- static_cast<AtomCertVerifier*>(request_context->cert_verifier())
- ->SetVerifyProc(proc);
- }
- void ClearHostResolverCacheInIO(
- const scoped_refptr<net::URLRequestContextGetter>& context_getter,
- util::Promise promise) {
- auto* request_context = context_getter->GetURLRequestContext();
- auto* cache = request_context->host_resolver()->GetHostCache();
- if (cache) {
- cache->clear();
- DCHECK_EQ(0u, cache->size());
- base::PostTaskWithTraits(
- FROM_HERE, {BrowserThread::UI},
- base::BindOnce(util::Promise::ResolveEmptyPromise, std::move(promise)));
- }
- }
- void ClearAuthCacheInIO(
- const scoped_refptr<net::URLRequestContextGetter>& context_getter,
- const ClearAuthCacheOptions& options,
- util::Promise promise) {
- auto* request_context = context_getter->GetURLRequestContext();
- auto* network_session =
- request_context->http_transaction_factory()->GetSession();
- if (network_session) {
- if (options.type == "password") {
- auto* auth_cache = network_session->http_auth_cache();
- if (!options.origin.is_empty()) {
- auth_cache->Remove(
- options.origin, options.realm, options.auth_scheme,
- net::AuthCredentials(options.username, options.password));
- } else {
- auth_cache->ClearAllEntries();
- }
- } else if (options.type == "clientCertificate") {
- auto* client_auth_cache = network_session->ssl_client_auth_cache();
- client_auth_cache->Remove(net::HostPortPair::FromURL(options.origin));
- }
- network_session->CloseAllConnections();
- }
- base::PostTaskWithTraits(
- FROM_HERE, {BrowserThread::UI},
- base::BindOnce(util::Promise::ResolveEmptyPromise, std::move(promise)));
- }
- void AllowNTLMCredentialsForDomainsInIO(
- const scoped_refptr<net::URLRequestContextGetter>& context_getter,
- const std::string& domains) {
- auto* request_context = context_getter->GetURLRequestContext();
- auto* auth_handler = request_context->http_auth_handler_factory();
- if (auth_handler) {
- auto* auth_preferences = const_cast<net::HttpAuthPreferences*>(
- auth_handler->http_auth_preferences());
- if (auth_preferences)
- auth_preferences->SetServerWhitelist(domains);
- }
- }
- void DownloadIdCallback(content::DownloadManager* download_manager,
- const base::FilePath& path,
- const std::vector<GURL>& url_chain,
- const std::string& mime_type,
- int64_t offset,
- int64_t length,
- const std::string& last_modified,
- const std::string& etag,
- const base::Time& start_time,
- uint32_t id) {
- download_manager->CreateDownloadItem(
- base::GenerateGUID(), id, path, path, url_chain, GURL(), GURL(), GURL(),
- GURL(), base::nullopt, mime_type, mime_type, start_time, base::Time(),
- etag, last_modified, offset, length, std::string(),
- download::DownloadItem::INTERRUPTED,
- download::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS,
- download::DOWNLOAD_INTERRUPT_REASON_NETWORK_TIMEOUT, false, base::Time(),
- false, std::vector<download::DownloadItem::ReceivedSlice>());
- }
- void DestroyGlobalHandle(v8::Isolate* isolate,
- const v8::Global<v8::Value>& global_handle) {
- v8::Locker locker(isolate);
- v8::HandleScope handle_scope(isolate);
- if (!global_handle.IsEmpty()) {
- v8::Local<v8::Value> local_handle = global_handle.Get(isolate);
- v8::Local<v8::Object> object;
- if (local_handle->IsObject() &&
- local_handle->ToObject(isolate->GetCurrentContext()).ToLocal(&object)) {
- void* ptr = object->GetAlignedPointerFromInternalField(0);
- if (!ptr)
- return;
- delete static_cast<mate::WrappableBase*>(ptr);
- object->SetAlignedPointerInInternalField(0, nullptr);
- }
- }
- }
- } // namespace
- Session::Session(v8::Isolate* isolate, AtomBrowserContext* browser_context)
- : network_emulation_token_(base::UnguessableToken::Create()),
- browser_context_(browser_context) {
- // Observe DownloadManager to get download notifications.
- content::BrowserContext::GetDownloadManager(browser_context)
- ->AddObserver(this);
- new SessionPreferences(browser_context);
- Init(isolate);
- AttachAsUserData(browser_context);
- }
- Session::~Session() {
- content::BrowserContext::GetDownloadManager(browser_context())
- ->RemoveObserver(this);
- DestroyGlobalHandle(isolate(), cookies_);
- DestroyGlobalHandle(isolate(), web_request_);
- DestroyGlobalHandle(isolate(), protocol_);
- DestroyGlobalHandle(isolate(), net_log_);
- g_sessions.erase(weak_map_id());
- }
- void Session::OnDownloadCreated(content::DownloadManager* manager,
- download::DownloadItem* item) {
- if (item->IsSavePackageDownload())
- return;
- v8::Locker locker(isolate());
- v8::HandleScope handle_scope(isolate());
- auto handle = DownloadItem::Create(isolate(), item);
- if (item->GetState() == download::DownloadItem::INTERRUPTED)
- handle->SetSavePath(item->GetTargetFilePath());
- content::WebContents* web_contents =
- content::DownloadItemUtils::GetWebContents(item);
- bool prevent_default = Emit("will-download", handle, web_contents);
- if (prevent_default) {
- item->Cancel(true);
- item->Remove();
- }
- }
- v8::Local<v8::Promise> Session::ResolveProxy(mate::Arguments* args) {
- v8::Isolate* isolate = args->isolate();
- util::Promise promise(isolate);
- v8::Local<v8::Promise> handle = promise.GetHandle();
- GURL url;
- args->GetNext(&url);
- browser_context_->GetResolveProxyHelper()->ResolveProxy(
- url,
- base::Bind(util::CopyablePromise::ResolveCopyablePromise<std::string>,
- util::CopyablePromise(promise)));
- return handle;
- }
- v8::Local<v8::Promise> Session::GetCacheSize() {
- auto* isolate = v8::Isolate::GetCurrent();
- auto promise = util::Promise(isolate);
- auto handle = promise.GetHandle();
- content::BrowserContext::GetDefaultStoragePartition(browser_context_.get())
- ->GetNetworkContext()
- ->ComputeHttpCacheSize(base::Time(), base::Time::Max(),
- base::BindOnce(
- [](util::Promise promise, bool is_upper_bound,
- int64_t size_or_error) {
- if (size_or_error < 0) {
- promise.RejectWithErrorMessage(
- net::ErrorToString(size_or_error));
- } else {
- promise.Resolve(size_or_error);
- }
- },
- std::move(promise)));
- return handle;
- }
- v8::Local<v8::Promise> Session::ClearCache() {
- auto* isolate = v8::Isolate::GetCurrent();
- auto promise = util::Promise(isolate);
- auto handle = promise.GetHandle();
- content::BrowserContext::GetDefaultStoragePartition(browser_context_.get())
- ->GetNetworkContext()
- ->ClearHttpCache(base::Time(), base::Time::Max(), nullptr,
- base::BindOnce(util::Promise::ResolveEmptyPromise,
- std::move(promise)));
- return handle;
- }
- v8::Local<v8::Promise> Session::ClearStorageData(mate::Arguments* args) {
- v8::Isolate* isolate = args->isolate();
- util::Promise promise(isolate);
- v8::Local<v8::Promise> handle = promise.GetHandle();
- ClearStorageDataOptions options;
- args->GetNext(&options);
- auto* storage_partition =
- content::BrowserContext::GetStoragePartition(browser_context(), nullptr);
- if (options.storage_types & StoragePartition::REMOVE_DATA_MASK_COOKIES) {
- // Reset media device id salt when cookies are cleared.
- // https://w3c.github.io/mediacapture-main/#dom-mediadeviceinfo-deviceid
- MediaDeviceIDSalt::Reset(browser_context()->prefs());
- }
- storage_partition->ClearData(
- options.storage_types, options.quota_types, options.origin, base::Time(),
- base::Time::Max(),
- base::Bind(util::CopyablePromise::ResolveEmptyCopyablePromise,
- util::CopyablePromise(promise)));
- return handle;
- }
- void Session::FlushStorageData() {
- auto* storage_partition =
- content::BrowserContext::GetStoragePartition(browser_context(), nullptr);
- storage_partition->Flush();
- }
- v8::Local<v8::Promise> Session::SetProxy(mate::Arguments* args) {
- v8::Isolate* isolate = args->isolate();
- util::Promise promise(isolate);
- v8::Local<v8::Promise> handle = promise.GetHandle();
- mate::Dictionary options;
- args->GetNext(&options);
- if (!browser_context_->in_memory_pref_store()) {
- promise.Resolve();
- return handle;
- }
- std::string proxy_rules, bypass_list, pac_url;
- options.Get("pacScript", &pac_url);
- options.Get("proxyRules", &proxy_rules);
- options.Get("proxyBypassRules", &bypass_list);
- // pacScript takes precedence over proxyRules.
- if (!pac_url.empty()) {
- browser_context_->in_memory_pref_store()->SetValue(
- proxy_config::prefs::kProxy,
- std::make_unique<base::Value>(ProxyConfigDictionary::CreatePacScript(
- pac_url, true /* pac_mandatory */)),
- WriteablePrefStore::DEFAULT_PREF_WRITE_FLAGS);
- } else {
- browser_context_->in_memory_pref_store()->SetValue(
- proxy_config::prefs::kProxy,
- std::make_unique<base::Value>(ProxyConfigDictionary::CreateFixedServers(
- proxy_rules, bypass_list)),
- WriteablePrefStore::DEFAULT_PREF_WRITE_FLAGS);
- }
- base::ThreadTaskRunnerHandle::Get()->PostTask(
- FROM_HERE,
- base::BindOnce(util::Promise::ResolveEmptyPromise, std::move(promise)));
- return handle;
- }
- void Session::SetDownloadPath(const base::FilePath& path) {
- browser_context_->prefs()->SetFilePath(prefs::kDownloadDefaultDirectory,
- path);
- }
- void Session::EnableNetworkEmulation(const mate::Dictionary& options) {
- auto conditions = network::mojom::NetworkConditions::New();
- options.Get("offline", &conditions->offline);
- options.Get("downloadThroughput", &conditions->download_throughput);
- options.Get("uploadThroughput", &conditions->upload_throughput);
- double latency = 0.0;
- if (options.Get("latency", &latency) && latency) {
- conditions->latency = base::TimeDelta::FromMillisecondsD(latency);
- }
- auto* network_context = content::BrowserContext::GetDefaultStoragePartition(
- browser_context_.get())
- ->GetNetworkContext();
- network_context->SetNetworkConditions(network_emulation_token_,
- std::move(conditions));
- }
- void Session::DisableNetworkEmulation() {
- auto* network_context = content::BrowserContext::GetDefaultStoragePartition(
- browser_context_.get())
- ->GetNetworkContext();
- network_context->SetNetworkConditions(
- network_emulation_token_, network::mojom::NetworkConditions::New());
- }
- void WrapVerifyProc(base::Callback<void(const VerifyRequestParams& request,
- base::Callback<void(int)>)> proc,
- const VerifyRequestParams& request,
- base::OnceCallback<void(int)> cb) {
- if (proc.is_null()) {
- LOG(ERROR) << "WrapVerifyProc (proc=null)";
- return;
- }
- proc.Run(request, base::AdaptCallbackForRepeating(std::move(cb)));
- }
- void Session::SetCertVerifyProc(v8::Local<v8::Value> val,
- mate::Arguments* args) {
- base::Callback<void(const VerifyRequestParams& request,
- base::Callback<void(int)>)>
- proc;
- if (!(val->IsNull() || mate::ConvertFromV8(args->isolate(), val, &proc))) {
- args->ThrowError("Must pass null or function");
- return;
- }
- base::PostTaskWithTraits(
- FROM_HERE, {BrowserThread::IO},
- base::BindOnce(&SetCertVerifyProcInIO,
- WrapRefCounted(browser_context_->GetRequestContext()),
- base::Bind(&WrapVerifyProc, proc)));
- }
- void Session::SetPermissionRequestHandler(v8::Local<v8::Value> val,
- mate::Arguments* args) {
- auto* permission_manager = static_cast<AtomPermissionManager*>(
- browser_context()->GetPermissionControllerDelegate());
- if (val->IsNull()) {
- permission_manager->SetPermissionRequestHandler(
- AtomPermissionManager::RequestHandler());
- return;
- }
- using StatusCallback =
- base::RepeatingCallback<void(blink::mojom::PermissionStatus)>;
- using RequestHandler =
- base::Callback<void(content::WebContents*, content::PermissionType,
- StatusCallback, const base::Value&)>;
- auto handler = std::make_unique<RequestHandler>();
- if (!mate::ConvertFromV8(args->isolate(), val, handler.get())) {
- args->ThrowError("Must pass null or function");
- return;
- }
- permission_manager->SetPermissionRequestHandler(base::BindRepeating(
- [](RequestHandler* handler, content::WebContents* web_contents,
- content::PermissionType permission_type,
- AtomPermissionManager::StatusCallback callback,
- const base::Value& details) {
- handler->Run(web_contents, permission_type,
- base::AdaptCallbackForRepeating(std::move(callback)),
- details);
- },
- base::Owned(std::move(handler))));
- }
- void Session::SetPermissionCheckHandler(v8::Local<v8::Value> val,
- mate::Arguments* args) {
- AtomPermissionManager::CheckHandler handler;
- if (!(val->IsNull() || mate::ConvertFromV8(args->isolate(), val, &handler))) {
- args->ThrowError("Must pass null or function");
- return;
- }
- auto* permission_manager = static_cast<AtomPermissionManager*>(
- browser_context()->GetPermissionControllerDelegate());
- permission_manager->SetPermissionCheckHandler(handler);
- }
- v8::Local<v8::Promise> Session::ClearHostResolverCache(mate::Arguments* args) {
- v8::Isolate* isolate = args->isolate();
- util::Promise promise(isolate);
- v8::Local<v8::Promise> handle = promise.GetHandle();
- base::PostTaskWithTraits(
- FROM_HERE, {BrowserThread::IO},
- base::BindOnce(&ClearHostResolverCacheInIO,
- WrapRefCounted(browser_context_->GetRequestContext()),
- std::move(promise)));
- return handle;
- }
- v8::Local<v8::Promise> Session::ClearAuthCache(mate::Arguments* args) {
- v8::Isolate* isolate = args->isolate();
- util::Promise promise(isolate);
- v8::Local<v8::Promise> handle = promise.GetHandle();
- ClearAuthCacheOptions options;
- if (!args->GetNext(&options)) {
- promise.RejectWithErrorMessage("Must specify options object");
- return handle;
- }
- base::PostTaskWithTraits(
- FROM_HERE, {BrowserThread::IO},
- base::BindOnce(&ClearAuthCacheInIO,
- WrapRefCounted(browser_context_->GetRequestContext()),
- options, std::move(promise)));
- return handle;
- }
- void Session::AllowNTLMCredentialsForDomains(const std::string& domains) {
- base::PostTaskWithTraits(
- FROM_HERE, {BrowserThread::IO},
- base::BindOnce(&AllowNTLMCredentialsForDomainsInIO,
- WrapRefCounted(browser_context_->GetRequestContext()),
- domains));
- }
- void Session::SetUserAgent(const std::string& user_agent,
- mate::Arguments* args) {
- browser_context_->SetUserAgent(user_agent);
- std::string accept_lang = g_browser_process->GetApplicationLocale();
- args->GetNext(&accept_lang);
- scoped_refptr<net::URLRequestContextGetter> getter(
- browser_context_->GetRequestContext());
- getter->GetNetworkTaskRunner()->PostTask(
- FROM_HERE,
- base::BindOnce(&SetUserAgentInIO, getter, accept_lang, user_agent));
- }
- std::string Session::GetUserAgent() {
- return browser_context_->GetUserAgent();
- }
- v8::Local<v8::Promise> Session::GetBlobData(v8::Isolate* isolate,
- const std::string& uuid) {
- util::Promise promise(isolate);
- v8::Local<v8::Promise> handle = promise.GetHandle();
- AtomBlobReader* blob_reader = browser_context()->GetBlobReader();
- base::PostTaskWithTraits(
- FROM_HERE, {BrowserThread::IO},
- base::BindOnce(&AtomBlobReader::StartReading,
- base::Unretained(blob_reader), uuid, std::move(promise)));
- return handle;
- }
- void Session::CreateInterruptedDownload(const mate::Dictionary& options) {
- int64_t offset = 0, length = 0;
- double start_time = 0.0;
- std::string mime_type, last_modified, etag;
- base::FilePath path;
- std::vector<GURL> url_chain;
- options.Get("path", &path);
- options.Get("urlChain", &url_chain);
- options.Get("mimeType", &mime_type);
- options.Get("offset", &offset);
- options.Get("length", &length);
- options.Get("lastModified", &last_modified);
- options.Get("eTag", &etag);
- options.Get("startTime", &start_time);
- if (path.empty() || url_chain.empty() || length == 0) {
- isolate()->ThrowException(v8::Exception::Error(mate::StringToV8(
- isolate(), "Must pass non-empty path, urlChain and length.")));
- return;
- }
- if (offset >= length) {
- isolate()->ThrowException(v8::Exception::Error(mate::StringToV8(
- isolate(), "Must pass an offset value less than length.")));
- return;
- }
- auto* download_manager =
- content::BrowserContext::GetDownloadManager(browser_context());
- download_manager->GetDelegate()->GetNextId(base::Bind(
- &DownloadIdCallback, download_manager, path, url_chain, mime_type, offset,
- length, last_modified, etag, base::Time::FromDoubleT(start_time)));
- }
- void Session::SetPreloads(
- const std::vector<base::FilePath::StringType>& preloads) {
- auto* prefs = SessionPreferences::FromBrowserContext(browser_context());
- DCHECK(prefs);
- prefs->set_preloads(preloads);
- }
- std::vector<base::FilePath::StringType> Session::GetPreloads() const {
- auto* prefs = SessionPreferences::FromBrowserContext(browser_context());
- DCHECK(prefs);
- return prefs->preloads();
- }
- v8::Local<v8::Value> Session::Cookies(v8::Isolate* isolate) {
- if (cookies_.IsEmpty()) {
- auto handle = Cookies::Create(isolate, browser_context());
- cookies_.Reset(isolate, handle.ToV8());
- }
- return v8::Local<v8::Value>::New(isolate, cookies_);
- }
- v8::Local<v8::Value> Session::Protocol(v8::Isolate* isolate) {
- if (protocol_.IsEmpty()) {
- auto handle = atom::api::Protocol::Create(isolate, browser_context());
- protocol_.Reset(isolate, handle.ToV8());
- }
- return v8::Local<v8::Value>::New(isolate, protocol_);
- }
- v8::Local<v8::Value> Session::WebRequest(v8::Isolate* isolate) {
- if (web_request_.IsEmpty()) {
- auto handle = atom::api::WebRequest::Create(isolate, browser_context());
- web_request_.Reset(isolate, handle.ToV8());
- }
- return v8::Local<v8::Value>::New(isolate, web_request_);
- }
- v8::Local<v8::Value> Session::NetLog(v8::Isolate* isolate) {
- if (net_log_.IsEmpty()) {
- auto handle = atom::api::NetLog::Create(isolate, browser_context());
- net_log_.Reset(isolate, handle.ToV8());
- }
- return v8::Local<v8::Value>::New(isolate, net_log_);
- }
- // static
- mate::Handle<Session> Session::CreateFrom(v8::Isolate* isolate,
- AtomBrowserContext* browser_context) {
- auto* existing = TrackableObject::FromWrappedClass(isolate, browser_context);
- if (existing)
- return mate::CreateHandle(isolate, static_cast<Session*>(existing));
- auto handle =
- mate::CreateHandle(isolate, new Session(isolate, browser_context));
- // The Sessions should never be garbage collected, since the common pattern is
- // to use partition strings, instead of using the Session object directly.
- g_sessions[handle->weak_map_id()] =
- v8::Global<v8::Object>(isolate, handle.ToV8());
- return handle;
- }
- // static
- mate::Handle<Session> Session::FromPartition(
- v8::Isolate* isolate,
- const std::string& partition,
- const base::DictionaryValue& options) {
- scoped_refptr<AtomBrowserContext> browser_context;
- if (partition.empty()) {
- browser_context = AtomBrowserContext::From("", false, options);
- } else if (base::StartsWith(partition, kPersistPrefix,
- base::CompareCase::SENSITIVE)) {
- std::string name = partition.substr(8);
- browser_context = AtomBrowserContext::From(name, false, options);
- } else {
- browser_context = AtomBrowserContext::From(partition, true, options);
- }
- return CreateFrom(isolate, browser_context.get());
- }
- // static
- void Session::BuildPrototype(v8::Isolate* isolate,
- v8::Local<v8::FunctionTemplate> prototype) {
- prototype->SetClassName(mate::StringToV8(isolate, "Session"));
- mate::ObjectTemplateBuilder(isolate, prototype->PrototypeTemplate())
- .MakeDestroyable()
- .SetMethod("resolveProxy", &Session::ResolveProxy)
- .SetMethod("getCacheSize", &Session::GetCacheSize)
- .SetMethod("clearCache", &Session::ClearCache)
- .SetMethod("clearStorageData", &Session::ClearStorageData)
- .SetMethod("flushStorageData", &Session::FlushStorageData)
- .SetMethod("setProxy", &Session::SetProxy)
- .SetMethod("setDownloadPath", &Session::SetDownloadPath)
- .SetMethod("enableNetworkEmulation", &Session::EnableNetworkEmulation)
- .SetMethod("disableNetworkEmulation", &Session::DisableNetworkEmulation)
- .SetMethod("setCertificateVerifyProc", &Session::SetCertVerifyProc)
- .SetMethod("setPermissionRequestHandler",
- &Session::SetPermissionRequestHandler)
- .SetMethod("setPermissionCheckHandler",
- &Session::SetPermissionCheckHandler)
- .SetMethod("clearHostResolverCache", &Session::ClearHostResolverCache)
- .SetMethod("clearAuthCache", &Session::ClearAuthCache)
- .SetMethod("allowNTLMCredentialsForDomains",
- &Session::AllowNTLMCredentialsForDomains)
- .SetMethod("setUserAgent", &Session::SetUserAgent)
- .SetMethod("getUserAgent", &Session::GetUserAgent)
- .SetMethod("getBlobData", &Session::GetBlobData)
- .SetMethod("createInterruptedDownload",
- &Session::CreateInterruptedDownload)
- .SetMethod("setPreloads", &Session::SetPreloads)
- .SetMethod("getPreloads", &Session::GetPreloads)
- .SetProperty("cookies", &Session::Cookies)
- .SetProperty("netLog", &Session::NetLog)
- .SetProperty("protocol", &Session::Protocol)
- .SetProperty("webRequest", &Session::WebRequest);
- }
- } // namespace api
- } // namespace atom
- namespace {
- using atom::api::Cookies;
- using atom::api::NetLog;
- using atom::api::Protocol;
- using atom::api::Session;
- v8::Local<v8::Value> FromPartition(const std::string& partition,
- mate::Arguments* args) {
- if (!atom::Browser::Get()->is_ready()) {
- args->ThrowError("Session can only be received when app is ready");
- return v8::Null(args->isolate());
- }
- base::DictionaryValue options;
- args->GetNext(&options);
- return Session::FromPartition(args->isolate(), partition, options).ToV8();
- }
- void Initialize(v8::Local<v8::Object> exports,
- v8::Local<v8::Value> unused,
- v8::Local<v8::Context> context,
- void* priv) {
- v8::Isolate* isolate = context->GetIsolate();
- mate::Dictionary dict(isolate, exports);
- dict.Set(
- "Session",
- Session::GetConstructor(isolate)->GetFunction(context).ToLocalChecked());
- dict.Set(
- "Cookies",
- Cookies::GetConstructor(isolate)->GetFunction(context).ToLocalChecked());
- dict.Set(
- "NetLog",
- NetLog::GetConstructor(isolate)->GetFunction(context).ToLocalChecked());
- dict.Set(
- "Protocol",
- Protocol::GetConstructor(isolate)->GetFunction(context).ToLocalChecked());
- dict.SetMethod("fromPartition", &FromPartition);
- }
- } // namespace
- NODE_LINKED_MODULE_CONTEXT_AWARE(atom_browser_session, Initialize)
|