parent
3b183854ff
commit
2d3c65beca
31 changed files with 200 additions and 152 deletions
|
@ -48,7 +48,7 @@ namespace electron {
|
|||
|
||||
namespace {
|
||||
|
||||
bool GetProcessExecPath(std::u16string* exe) {
|
||||
bool GetProcessExecPath(std::wstring* exe) {
|
||||
base::FilePath path;
|
||||
if (!base::PathService::Get(base::FILE_EXE, &path)) {
|
||||
return false;
|
||||
|
@ -57,13 +57,13 @@ bool GetProcessExecPath(std::u16string* exe) {
|
|||
return true;
|
||||
}
|
||||
|
||||
bool GetProtocolLaunchPath(gin::Arguments* args, std::u16string* exe) {
|
||||
bool GetProtocolLaunchPath(gin::Arguments* args, std::wstring* exe) {
|
||||
if (!args->GetNext(exe) && !GetProcessExecPath(exe)) {
|
||||
return false;
|
||||
}
|
||||
|
||||
// Read in optional args arg
|
||||
std::vector<std::u16string> launch_args;
|
||||
std::vector<std::wstring> launch_args;
|
||||
if (args->GetNext(&launch_args) && !launch_args.empty())
|
||||
*exe = base::StringPrintf(L"\"%ls\" %ls \"%%1\"", exe->c_str(),
|
||||
base::JoinString(launch_args, L" ").c_str());
|
||||
|
@ -75,7 +75,7 @@ bool GetProtocolLaunchPath(gin::Arguments* args, std::u16string* exe) {
|
|||
// Windows treats a given scheme as an Internet scheme only if its registry
|
||||
// entry has a "URL Protocol" key. Check this, otherwise we allow ProgIDs to be
|
||||
// used as custom protocols which leads to security bugs.
|
||||
bool IsValidCustomProtocol(const std::u16string& scheme) {
|
||||
bool IsValidCustomProtocol(const std::wstring& scheme) {
|
||||
if (scheme.empty())
|
||||
return false;
|
||||
base::win::RegKey cmd_key(HKEY_CLASSES_ROOT, scheme.c_str(), KEY_QUERY_VALUE);
|
||||
|
@ -90,11 +90,10 @@ bool IsValidCustomProtocol(const std::u16string& scheme) {
|
|||
// Windows 8 introduced a new protocol->executable binding system which cannot
|
||||
// be retrieved in the HKCR registry subkey method implemented below. We call
|
||||
// AssocQueryString with the new Win8-only flag ASSOCF_IS_PROTOCOL instead.
|
||||
std::u16string GetAppInfoHelperForProtocol(ASSOCSTR assoc_str,
|
||||
const GURL& url) {
|
||||
const std::u16string url_scheme = base::ASCIIToUTF16(url.scheme());
|
||||
std::wstring GetAppInfoHelperForProtocol(ASSOCSTR assoc_str, const GURL& url) {
|
||||
const std::wstring url_scheme = base::ASCIIToWide(url.scheme());
|
||||
if (!IsValidCustomProtocol(url_scheme))
|
||||
return std::u16string();
|
||||
return std::wstring();
|
||||
|
||||
wchar_t out_buffer[1024];
|
||||
DWORD buffer_size = base::size(out_buffer);
|
||||
|
@ -103,13 +102,13 @@ std::u16string GetAppInfoHelperForProtocol(ASSOCSTR assoc_str,
|
|||
out_buffer, &buffer_size);
|
||||
if (FAILED(hr)) {
|
||||
DLOG(WARNING) << "AssocQueryString failed!";
|
||||
return std::u16string();
|
||||
return std::wstring();
|
||||
}
|
||||
return std::u16string(out_buffer);
|
||||
return std::wstring(out_buffer);
|
||||
}
|
||||
|
||||
void OnIconDataAvailable(const base::FilePath& app_path,
|
||||
const std::u16string& app_display_name,
|
||||
const std::wstring& app_display_name,
|
||||
gin_helper::Promise<gin_helper::Dictionary> promise,
|
||||
gfx::Image icon) {
|
||||
if (!icon.IsEmpty()) {
|
||||
|
@ -126,21 +125,21 @@ void OnIconDataAvailable(const base::FilePath& app_path,
|
|||
}
|
||||
}
|
||||
|
||||
std::u16string GetAppDisplayNameForProtocol(const GURL& url) {
|
||||
std::wstring GetAppDisplayNameForProtocol(const GURL& url) {
|
||||
return GetAppInfoHelperForProtocol(ASSOCSTR_FRIENDLYAPPNAME, url);
|
||||
}
|
||||
|
||||
std::u16string GetAppPathForProtocol(const GURL& url) {
|
||||
std::wstring GetAppPathForProtocol(const GURL& url) {
|
||||
return GetAppInfoHelperForProtocol(ASSOCSTR_EXECUTABLE, url);
|
||||
}
|
||||
|
||||
std::u16string GetAppForProtocolUsingRegistry(const GURL& url) {
|
||||
const std::u16string url_scheme = base::ASCIIToUTF16(url.scheme());
|
||||
std::wstring GetAppForProtocolUsingRegistry(const GURL& url) {
|
||||
const std::wstring url_scheme = base::ASCIIToWide(url.scheme());
|
||||
if (!IsValidCustomProtocol(url_scheme))
|
||||
return std::u16string();
|
||||
return std::wstring();
|
||||
|
||||
// First, try and extract the application's display name.
|
||||
std::u16string command_to_launch;
|
||||
std::wstring command_to_launch;
|
||||
base::win::RegKey cmd_key_name(HKEY_CLASSES_ROOT, url_scheme.c_str(),
|
||||
KEY_READ);
|
||||
if (cmd_key_name.ReadValue(NULL, &command_to_launch) == ERROR_SUCCESS &&
|
||||
|
@ -150,7 +149,7 @@ std::u16string GetAppForProtocolUsingRegistry(const GURL& url) {
|
|||
|
||||
// Otherwise, parse the command line in the registry, and return the basename
|
||||
// of the program path if it exists.
|
||||
const std::u16string cmd_key_path = url_scheme + L"\\shell\\open\\command";
|
||||
const std::wstring cmd_key_path = url_scheme + L"\\shell\\open\\command";
|
||||
base::win::RegKey cmd_key_exe(HKEY_CLASSES_ROOT, cmd_key_path.c_str(),
|
||||
KEY_READ);
|
||||
if (cmd_key_exe.ReadValue(NULL, &command_to_launch) == ERROR_SUCCESS) {
|
||||
|
@ -159,18 +158,20 @@ std::u16string GetAppForProtocolUsingRegistry(const GURL& url) {
|
|||
return command_line.GetProgram().BaseName().value();
|
||||
}
|
||||
|
||||
return std::u16string();
|
||||
return std::wstring();
|
||||
}
|
||||
|
||||
bool FormatCommandLineString(std::u16string* exe,
|
||||
bool FormatCommandLineString(std::wstring* exe,
|
||||
const std::vector<std::u16string>& launch_args) {
|
||||
if (exe->empty() && !GetProcessExecPath(exe)) {
|
||||
return false;
|
||||
}
|
||||
|
||||
if (!launch_args.empty()) {
|
||||
std::u16string joined_launch_args =
|
||||
base::JoinString(launch_args, base::UTF8ToUTF16(" "));
|
||||
*exe = base::StringPrintf(L"%ls %ls", exe->c_str(),
|
||||
base::JoinString(launch_args, L" ").c_str());
|
||||
base::UTF16ToWide(joined_launch_args).c_str());
|
||||
}
|
||||
|
||||
return true;
|
||||
|
@ -184,18 +185,20 @@ bool FormatCommandLineString(std::u16string* exe,
|
|||
std::vector<Browser::LaunchItem> GetLoginItemSettingsHelper(
|
||||
base::win::RegistryValueIterator* it,
|
||||
boolean* executable_will_launch_at_login,
|
||||
std::u16string scope,
|
||||
std::wstring scope,
|
||||
const Browser::LoginItemSettings& options) {
|
||||
std::vector<Browser::LaunchItem> launch_items;
|
||||
|
||||
base::FilePath lookup_exe_path;
|
||||
if (options.path.empty()) {
|
||||
std::u16string process_exe_path;
|
||||
std::wstring process_exe_path;
|
||||
GetProcessExecPath(&process_exe_path);
|
||||
lookup_exe_path =
|
||||
base::CommandLine::FromString(process_exe_path).GetProgram();
|
||||
} else {
|
||||
lookup_exe_path = base::CommandLine::FromString(options.path).GetProgram();
|
||||
lookup_exe_path =
|
||||
base::CommandLine::FromString(base::UTF16ToWide(options.path))
|
||||
.GetProgram();
|
||||
}
|
||||
|
||||
if (!lookup_exe_path.empty()) {
|
||||
|
@ -287,7 +290,7 @@ Browser::UserTask::~UserTask() = default;
|
|||
void GetFileIcon(const base::FilePath& path,
|
||||
v8::Isolate* isolate,
|
||||
base::CancelableTaskTracker* cancelable_task_tracker_,
|
||||
const std::u16string app_display_name,
|
||||
const std::wstring app_display_name,
|
||||
gin_helper::Promise<gin_helper::Dictionary> promise) {
|
||||
base::FilePath normalized_path = path.NormalizePathSeparators();
|
||||
IconLoader::IconSize icon_size = IconLoader::IconSize::LARGE;
|
||||
|
@ -316,13 +319,13 @@ void GetApplicationInfoForProtocolUsingRegistry(
|
|||
base::CancelableTaskTracker* cancelable_task_tracker_) {
|
||||
base::FilePath app_path;
|
||||
|
||||
const std::u16string url_scheme = base::ASCIIToUTF16(url.scheme());
|
||||
const std::wstring url_scheme = base::ASCIIToWide(url.scheme());
|
||||
if (!IsValidCustomProtocol(url_scheme)) {
|
||||
promise.RejectWithErrorMessage("invalid url_scheme");
|
||||
return;
|
||||
}
|
||||
std::u16string command_to_launch;
|
||||
const std::u16string cmd_key_path = url_scheme + L"\\shell\\open\\command";
|
||||
std::wstring command_to_launch;
|
||||
const std::wstring cmd_key_path = url_scheme + L"\\shell\\open\\command";
|
||||
base::win::RegKey cmd_key_exe(HKEY_CLASSES_ROOT, cmd_key_path.c_str(),
|
||||
KEY_READ);
|
||||
if (cmd_key_exe.ReadValue(NULL, &command_to_launch) == ERROR_SUCCESS) {
|
||||
|
@ -334,7 +337,7 @@ void GetApplicationInfoForProtocolUsingRegistry(
|
|||
"Unable to retrieve installation path to app");
|
||||
return;
|
||||
}
|
||||
const std::u16string app_display_name = GetAppForProtocolUsingRegistry(url);
|
||||
const std::wstring app_display_name = GetAppForProtocolUsingRegistry(url);
|
||||
|
||||
if (app_display_name.empty()) {
|
||||
promise.RejectWithErrorMessage(
|
||||
|
@ -354,7 +357,7 @@ void GetApplicationInfoForProtocolUsingAssocQuery(
|
|||
const GURL& url,
|
||||
gin_helper::Promise<gin_helper::Dictionary> promise,
|
||||
base::CancelableTaskTracker* cancelable_task_tracker_) {
|
||||
std::u16string app_path = GetAppPathForProtocol(url);
|
||||
std::wstring app_path = GetAppPathForProtocol(url);
|
||||
|
||||
if (app_path.empty()) {
|
||||
promise.RejectWithErrorMessage(
|
||||
|
@ -362,7 +365,7 @@ void GetApplicationInfoForProtocolUsingAssocQuery(
|
|||
return;
|
||||
}
|
||||
|
||||
std::u16string app_display_name = GetAppDisplayNameForProtocol(url);
|
||||
std::wstring app_display_name = GetAppDisplayNameForProtocol(url);
|
||||
|
||||
if (app_display_name.empty()) {
|
||||
promise.RejectWithErrorMessage("Unable to retrieve display name of app");
|
||||
|
@ -426,12 +429,12 @@ bool Browser::RemoveAsDefaultProtocolClient(const std::string& protocol,
|
|||
|
||||
// Main Registry Key
|
||||
HKEY root = HKEY_CURRENT_USER;
|
||||
std::u16string keyPath = L"Software\\Classes\\";
|
||||
std::wstring keyPath = L"Software\\Classes\\";
|
||||
|
||||
// Command Key
|
||||
std::u16string wprotocol = base::UTF8ToUTF16(protocol);
|
||||
std::u16string shellPath = wprotocol + L"\\shell";
|
||||
std::u16string cmdPath = keyPath + shellPath + L"\\open\\command";
|
||||
std::wstring wprotocol = base::UTF8ToWide(protocol);
|
||||
std::wstring shellPath = wprotocol + L"\\shell";
|
||||
std::wstring cmdPath = keyPath + shellPath + L"\\open\\command";
|
||||
|
||||
base::win::RegKey classesKey;
|
||||
base::win::RegKey commandKey;
|
||||
|
@ -445,12 +448,12 @@ bool Browser::RemoveAsDefaultProtocolClient(const std::string& protocol,
|
|||
// Key doesn't even exist, we can confirm that it is not set
|
||||
return true;
|
||||
|
||||
std::u16string keyVal;
|
||||
std::wstring keyVal;
|
||||
if (FAILED(commandKey.ReadValue(L"", &keyVal)))
|
||||
// Default value not set, we can confirm that it is not set
|
||||
return true;
|
||||
|
||||
std::u16string exe;
|
||||
std::wstring exe;
|
||||
if (!GetProtocolLaunchPath(args, &exe))
|
||||
return false;
|
||||
|
||||
|
@ -461,7 +464,7 @@ bool Browser::RemoveAsDefaultProtocolClient(const std::string& protocol,
|
|||
|
||||
// Let's clean up after ourselves
|
||||
base::win::RegKey protocolKey;
|
||||
std::u16string protocolPath = keyPath + wprotocol;
|
||||
std::wstring protocolPath = keyPath + wprotocol;
|
||||
|
||||
if (SUCCEEDED(
|
||||
protocolKey.Open(root, protocolPath.c_str(), KEY_ALL_ACCESS))) {
|
||||
|
@ -500,17 +503,17 @@ bool Browser::SetAsDefaultProtocolClient(const std::string& protocol,
|
|||
if (protocol.empty())
|
||||
return false;
|
||||
|
||||
std::u16string exe;
|
||||
std::wstring exe;
|
||||
if (!GetProtocolLaunchPath(args, &exe))
|
||||
return false;
|
||||
|
||||
// Main Registry Key
|
||||
HKEY root = HKEY_CURRENT_USER;
|
||||
std::u16string keyPath = base::UTF8ToUTF16("Software\\Classes\\" + protocol);
|
||||
std::u16string urlDecl = base::UTF8ToUTF16("URL:" + protocol);
|
||||
std::wstring keyPath = base::UTF8ToWide("Software\\Classes\\" + protocol);
|
||||
std::wstring urlDecl = base::UTF8ToWide("URL:" + protocol);
|
||||
|
||||
// Command Key
|
||||
std::u16string cmdPath = keyPath + L"\\shell\\open\\command";
|
||||
std::wstring cmdPath = keyPath + L"\\shell\\open\\command";
|
||||
|
||||
// Write information to registry
|
||||
base::win::RegKey key(root, keyPath.c_str(), KEY_ALL_ACCESS);
|
||||
|
@ -530,16 +533,16 @@ bool Browser::IsDefaultProtocolClient(const std::string& protocol,
|
|||
if (protocol.empty())
|
||||
return false;
|
||||
|
||||
std::u16string exe;
|
||||
std::wstring exe;
|
||||
if (!GetProtocolLaunchPath(args, &exe))
|
||||
return false;
|
||||
|
||||
// Main Registry Key
|
||||
HKEY root = HKEY_CURRENT_USER;
|
||||
std::u16string keyPath = base::UTF8ToUTF16("Software\\Classes\\" + protocol);
|
||||
std::wstring keyPath = base::UTF8ToWide("Software\\Classes\\" + protocol);
|
||||
|
||||
// Command Key
|
||||
std::u16string cmdPath = keyPath + L"\\shell\\open\\command";
|
||||
std::wstring cmdPath = keyPath + L"\\shell\\open\\command";
|
||||
|
||||
base::win::RegKey key;
|
||||
base::win::RegKey commandKey;
|
||||
|
@ -551,7 +554,7 @@ bool Browser::IsDefaultProtocolClient(const std::string& protocol,
|
|||
// Key doesn't exist, we can confirm that it is not set
|
||||
return false;
|
||||
|
||||
std::u16string keyVal;
|
||||
std::wstring keyVal;
|
||||
if (FAILED(commandKey.ReadValue(L"", &keyVal)))
|
||||
// Default value not set, we can confirm that it is not set
|
||||
return false;
|
||||
|
@ -563,12 +566,12 @@ bool Browser::IsDefaultProtocolClient(const std::string& protocol,
|
|||
std::u16string Browser::GetApplicationNameForProtocol(const GURL& url) {
|
||||
// Windows 8 or above has a new protocol association query.
|
||||
if (base::win::GetVersion() >= base::win::Version::WIN8) {
|
||||
std::u16string application_name = GetAppDisplayNameForProtocol(url);
|
||||
std::wstring application_name = GetAppDisplayNameForProtocol(url);
|
||||
if (!application_name.empty())
|
||||
return application_name;
|
||||
return base::WideToUTF16(application_name);
|
||||
}
|
||||
|
||||
return GetAppForProtocolUsingRegistry(url);
|
||||
return base::WideToUTF16(GetAppForProtocolUsingRegistry(url));
|
||||
}
|
||||
|
||||
v8::Local<v8::Promise> Browser::GetApplicationInfoForProtocol(
|
||||
|
@ -686,11 +689,10 @@ void Browser::UpdateBadgeContents(
|
|||
}
|
||||
|
||||
void Browser::SetLoginItemSettings(LoginItemSettings settings) {
|
||||
std::u16string key_path =
|
||||
L"Software\\Microsoft\\Windows\\CurrentVersion\\Run";
|
||||
std::wstring key_path = L"Software\\Microsoft\\Windows\\CurrentVersion\\Run";
|
||||
base::win::RegKey key(HKEY_CURRENT_USER, key_path.c_str(), KEY_ALL_ACCESS);
|
||||
|
||||
std::u16string startup_approved_key_path =
|
||||
std::wstring startup_approved_key_path =
|
||||
L"Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\StartupApproved"
|
||||
L"\\Run";
|
||||
base::win::RegKey startup_approved_key(
|
||||
|
@ -699,7 +701,7 @@ void Browser::SetLoginItemSettings(LoginItemSettings settings) {
|
|||
!settings.name.empty() ? settings.name.c_str() : GetAppUserModelID();
|
||||
|
||||
if (settings.open_at_login) {
|
||||
std::u16string exe = settings.path;
|
||||
std::wstring exe = base::UTF16ToWide(settings.path);
|
||||
if (FormatCommandLineString(&exe, settings.args)) {
|
||||
key.WriteValue(key_name, exe.c_str());
|
||||
|
||||
|
@ -732,13 +734,13 @@ void Browser::SetLoginItemSettings(LoginItemSettings settings) {
|
|||
Browser::LoginItemSettings Browser::GetLoginItemSettings(
|
||||
const LoginItemSettings& options) {
|
||||
LoginItemSettings settings;
|
||||
std::u16string keyPath = L"Software\\Microsoft\\Windows\\CurrentVersion\\Run";
|
||||
std::wstring keyPath = L"Software\\Microsoft\\Windows\\CurrentVersion\\Run";
|
||||
base::win::RegKey key(HKEY_CURRENT_USER, keyPath.c_str(), KEY_ALL_ACCESS);
|
||||
std::u16string keyVal;
|
||||
std::wstring keyVal;
|
||||
|
||||
// keep old openAtLogin behaviour
|
||||
if (!FAILED(key.ReadValue(GetAppUserModelID(), &keyVal))) {
|
||||
std::u16string exe = options.path;
|
||||
std::wstring exe = base::UTF16ToWide(options.path);
|
||||
if (FormatCommandLineString(&exe, options.args)) {
|
||||
settings.open_at_login = keyVal == exe;
|
||||
}
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue