1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
|
// Copyright 2013 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/browser/extensions/api/messaging/native_messaging_test_util.h"
#include <memory>
#include <utility>
#include "base/files/file_path.h"
#include "base/json/json_file_value_serializer.h"
#include "base/path_service.h"
#include "base/strings/stringprintf.h"
#include "base/strings/utf_string_conversions.h"
#include "base/threading/thread_restrictions.h"
#include "base/values.h"
#include "build/build_config.h"
#include "chrome/common/chrome_paths.h"
#include "testing/gtest/include/gtest/gtest.h"
#if defined(OS_WIN)
#include "base/win/registry.h"
#endif
namespace extensions {
namespace {
void WriteTestNativeHostManifest(const base::FilePath& target_dir,
const std::string& host_name,
const base::FilePath& host_path,
bool user_level) {
std::unique_ptr<base::DictionaryValue> manifest(new base::DictionaryValue());
manifest->SetString("name", host_name);
manifest->SetString("description", "Native Messaging Echo Test");
manifest->SetString("type", "stdio");
manifest->SetString("path", host_path.AsUTF8Unsafe());
std::unique_ptr<base::ListValue> origins(new base::ListValue());
origins->AppendString(base::StringPrintf(
"chrome-extension://%s/", ScopedTestNativeMessagingHost::kExtensionId));
manifest->Set("allowed_origins", std::move(origins));
base::FilePath manifest_path = target_dir.AppendASCII(host_name + ".json");
JSONFileValueSerializer serializer(manifest_path);
ASSERT_TRUE(serializer.Serialize(*manifest));
#if defined(OS_WIN)
HKEY root_key = user_level ? HKEY_CURRENT_USER : HKEY_LOCAL_MACHINE;
base::string16 key = L"SOFTWARE\\Google\\Chrome\\NativeMessagingHosts\\" +
base::UTF8ToUTF16(host_name);
base::win::RegKey manifest_key(
root_key, key.c_str(),
KEY_SET_VALUE | KEY_CREATE_SUB_KEY | KEY_CREATE_LINK);
ASSERT_EQ(ERROR_SUCCESS,
manifest_key.WriteValue(NULL, manifest_path.value().c_str()));
#endif
}
} // namespace
const char ScopedTestNativeMessagingHost::kHostName[] =
"com.google.chrome.test.echo";
const char ScopedTestNativeMessagingHost::kBinaryMissingHostName[] =
"com.google.chrome.test.host_binary_missing";
const char ScopedTestNativeMessagingHost::kExtensionId[] =
"knldjmfmopnpolahpmmgbagdohdnhkik";
ScopedTestNativeMessagingHost::ScopedTestNativeMessagingHost() {}
void ScopedTestNativeMessagingHost::RegisterTestHost(bool user_level) {
base::ScopedAllowBlockingForTesting allow_blocking;
ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
ScopedTestNativeMessagingHost test_host;
base::FilePath test_user_data_dir;
ASSERT_TRUE(
base::PathService::Get(chrome::DIR_TEST_DATA, &test_user_data_dir));
test_user_data_dir = test_user_data_dir.AppendASCII("native_messaging")
.AppendASCII("native_hosts");
#if defined(OS_WIN)
HKEY root_key = user_level ? HKEY_CURRENT_USER : HKEY_LOCAL_MACHINE;
ASSERT_NO_FATAL_FAILURE(registry_override_.OverrideRegistry(root_key));
#else
path_override_.reset(new base::ScopedPathOverride(
user_level ? chrome::DIR_USER_NATIVE_MESSAGING
: chrome::DIR_NATIVE_MESSAGING,
temp_dir_.GetPath()));
#endif
#if defined(OS_POSIX)
base::FilePath host_path = test_user_data_dir.AppendASCII("echo.py");
#else
base::FilePath host_path = test_user_data_dir.AppendASCII("echo.bat");
#endif
ASSERT_NO_FATAL_FAILURE(WriteTestNativeHostManifest(
temp_dir_.GetPath(), kHostName, host_path, user_level));
ASSERT_NO_FATAL_FAILURE(WriteTestNativeHostManifest(
temp_dir_.GetPath(), kBinaryMissingHostName,
test_user_data_dir.AppendASCII("missing_nm_binary.exe"), user_level));
}
ScopedTestNativeMessagingHost::~ScopedTestNativeMessagingHost() {
base::ScopedAllowBlockingForTesting allow_blocking;
ignore_result(temp_dir_.Delete());
}
} // namespace extensions
|
