native_messaging_host_manifest_unittest.cc revision 7dbb3d5cf0c15f500944d211057644d6a2f37371 
1// Copyright (c) 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_host_manifest.h"
6
7#include "base/file_util.h"
8#include "base/files/file_path.h"
9#include "base/files/scoped_temp_dir.h"
10#include "base/json/string_escape.h"
11#include "extensions/common/url_pattern_set.h"
12#include "testing/gtest/include/gtest/gtest.h"
13#include "url/gurl.h"
14
15namespace extensions {
16
17const char kTestHostName[] = "com.chrome.test.native_host";
18#if defined(OS_WIN)
19const char kTestHostPath[] = "C:\\ProgramFiles\\host.exe";
20#else
21const char kTestHostPath[] = "/usr/bin/host";
22#endif
23const char kTestOrigin[] =
24    "chrome-extension://knldjmfmopnpolahpmmgbagdohdnhkik/";
25
26class NativeMessagingHostManifestTest : public ::testing::Test {
27 public:
28  virtual void SetUp() OVERRIDE {
29    ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
30    manifest_path_ = temp_dir_.path().AppendASCII("test.json");
31  }
32
33 protected:
34  bool WriteManifest(const std::string& name,
35                     const std::string& path,
36                     const std::string& origin) {
37    std::string escaped_path;
38    base::JsonDoubleQuote(path, false, &escaped_path);
39    return WriteManifest("{"
40      "  \"name\": \"" + name + "\","
41      "  \"description\": \"Native Messaging Test\","
42      "  \"path\": \"" + escaped_path + "\","
43      "  \"type\": \"stdio\","
44      "  \"allowed_origins\": ["
45      "    \"" + origin + "\""
46      "  ]"
47      "}");
48  }
49
50  bool WriteManifest(const std::string& manifest_content) {
51    return file_util::WriteFile(
52        manifest_path_, manifest_content.data(), manifest_content.size());
53  }
54
55  base::ScopedTempDir temp_dir_;
56  base::FilePath manifest_path_;
57};
58
59TEST_F(NativeMessagingHostManifestTest, HostNameValidation) {
60  EXPECT_TRUE(NativeMessagingHostManifest::IsValidName("a"));
61  EXPECT_TRUE(NativeMessagingHostManifest::IsValidName("foo"));
62  EXPECT_TRUE(NativeMessagingHostManifest::IsValidName("foo132"));
63  EXPECT_TRUE(NativeMessagingHostManifest::IsValidName("foo.bar"));
64  EXPECT_TRUE(NativeMessagingHostManifest::IsValidName("foo.bar2"));
65  EXPECT_TRUE(NativeMessagingHostManifest::IsValidName("a._.c"));
66  EXPECT_TRUE(NativeMessagingHostManifest::IsValidName("a._.c"));
67  EXPECT_FALSE(NativeMessagingHostManifest::IsValidName("A.b"));
68  EXPECT_FALSE(NativeMessagingHostManifest::IsValidName("a..b"));
69  EXPECT_FALSE(NativeMessagingHostManifest::IsValidName(".a"));
70  EXPECT_FALSE(NativeMessagingHostManifest::IsValidName("b."));
71  EXPECT_FALSE(NativeMessagingHostManifest::IsValidName("a*"));
72}
73
74TEST_F(NativeMessagingHostManifestTest, LoadValid) {
75  ASSERT_TRUE(WriteManifest(kTestHostName, kTestHostPath, kTestOrigin));
76
77  std::string error_message;
78  scoped_ptr<NativeMessagingHostManifest> manifest =
79      NativeMessagingHostManifest::Load(manifest_path_, &error_message);
80  ASSERT_TRUE(manifest) << "Failed to load manifest: " << error_message;
81  EXPECT_TRUE(error_message.empty());
82
83  EXPECT_EQ(manifest->name(), "com.chrome.test.native_host");
84  EXPECT_EQ(manifest->description(), "Native Messaging Test");
85  EXPECT_EQ(manifest->interface(),
86            NativeMessagingHostManifest::HOST_INTERFACE_STDIO);
87  EXPECT_EQ(manifest->path(), base::FilePath::FromUTF8Unsafe(kTestHostPath));
88  EXPECT_TRUE(manifest->allowed_origins().MatchesSecurityOrigin(
89      GURL("chrome-extension://knldjmfmopnpolahpmmgbagdohdnhkik/")));
90  EXPECT_FALSE(manifest->allowed_origins().MatchesSecurityOrigin(
91      GURL("chrome-extension://jnldjmfmopnpolahpmmgbagdohdnhkik/")));
92}
93
94TEST_F(NativeMessagingHostManifestTest, InvalidName) {
95  ASSERT_TRUE(WriteManifest(".com.chrome.test.native_host",
96                            kTestHostPath, kTestOrigin));
97
98  std::string error_message;
99  scoped_ptr<NativeMessagingHostManifest> manifest =
100      NativeMessagingHostManifest::Load(manifest_path_, &error_message);
101  ASSERT_FALSE(manifest);
102  EXPECT_FALSE(error_message.empty());
103}
104
105TEST_F(NativeMessagingHostManifestTest, RelativePath) {
106  ASSERT_TRUE(WriteManifest(kTestHostName, "host.exe", kTestOrigin));
107
108  std::string error_message;
109  scoped_ptr<NativeMessagingHostManifest> manifest =
110      NativeMessagingHostManifest::Load(manifest_path_, &error_message);
111  ASSERT_FALSE(manifest);
112  EXPECT_FALSE(error_message.empty());
113}
114
115// Verify that match-all origins are rejected.
116TEST_F(NativeMessagingHostManifestTest, MatchAllOrigin) {
117  ASSERT_TRUE(WriteManifest(kTestHostName, kTestHostPath,
118                            "chrome-extension://*/"));
119
120  std::string error_message;
121  scoped_ptr<NativeMessagingHostManifest> manifest =
122      NativeMessagingHostManifest::Load(manifest_path_, &error_message);
123  ASSERT_FALSE(manifest);
124  EXPECT_FALSE(error_message.empty());
125}
126
127}  // namespace extensions
128