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