This source file includes following definitions.
- Set
- GetVar
- SetVar
- UnSetVar
- TEST
- TEST
- TEST
- TEST
- TEST
- TEST
- TEST
- TEST
- TEST
- TEST
#include "chrome/browser/shell_integration_linux.h"
#include <algorithm>
#include <cstdlib>
#include <map>
#include "base/base_paths.h"
#include "base/command_line.h"
#include "base/environment.h"
#include "base/file_util.h"
#include "base/files/file_path.h"
#include "base/files/scoped_temp_dir.h"
#include "base/message_loop/message_loop.h"
#include "base/stl_util.h"
#include "base/strings/string_util.h"
#include "base/strings/utf_string_conversions.h"
#include "base/test/scoped_path_override.h"
#include "chrome/browser/web_applications/web_app.h"
#include "chrome/common/chrome_constants.h"
#include "content/public/test/test_browser_thread.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "url/gurl.h"
#define FPL FILE_PATH_LITERAL
using content::BrowserThread;
using ::testing::ElementsAre;
namespace {
class MockEnvironment : public base::Environment {
public:
MockEnvironment() {}
void Set(const std::string& name, const std::string& value) {
variables_[name] = value;
}
virtual bool GetVar(const char* variable_name, std::string* result) OVERRIDE {
if (ContainsKey(variables_, variable_name)) {
*result = variables_[variable_name];
return true;
}
return false;
}
virtual bool SetVar(const char* variable_name,
const std::string& new_value) OVERRIDE {
ADD_FAILURE();
return false;
}
virtual bool UnSetVar(const char* variable_name) OVERRIDE {
ADD_FAILURE();
return false;
}
private:
std::map<std::string, std::string> variables_;
DISALLOW_COPY_AND_ASSIGN(MockEnvironment);
};
}
TEST(ShellIntegrationTest, GetDataWriteLocation) {
base::MessageLoop message_loop;
content::TestBrowserThread file_thread(BrowserThread::FILE, &message_loop);
{
MockEnvironment env;
env.Set("HOME", "/home/user");
env.Set("XDG_DATA_HOME", "/user/path");
base::FilePath path;
ASSERT_TRUE(ShellIntegrationLinux::GetDataWriteLocation(&env, &path));
EXPECT_EQ(base::FilePath("/user/path"), path);
}
{
MockEnvironment env;
env.Set("HOME", "/home/user");
base::FilePath path;
ASSERT_TRUE(ShellIntegrationLinux::GetDataWriteLocation(&env, &path));
EXPECT_EQ(base::FilePath("/home/user/.local/share"), path);
}
{
MockEnvironment env;
base::FilePath path;
ASSERT_FALSE(ShellIntegrationLinux::GetDataWriteLocation(&env, &path));
}
}
TEST(ShellIntegrationTest, GetDataSearchLocations) {
base::MessageLoop message_loop;
content::TestBrowserThread file_thread(BrowserThread::FILE, &message_loop);
{
MockEnvironment env;
env.Set("HOME", "/home/user");
env.Set("XDG_DATA_HOME", "/user/path");
env.Set("XDG_DATA_DIRS", "/system/path/1:/system/path/2");
EXPECT_THAT(
ShellIntegrationLinux::GetDataSearchLocations(&env),
ElementsAre(base::FilePath("/user/path"),
base::FilePath("/system/path/1"),
base::FilePath("/system/path/2")));
}
{
MockEnvironment env;
env.Set("HOME", "/home/user");
env.Set("XDG_DATA_DIRS", "/system/path/1:/system/path/2");
EXPECT_THAT(
ShellIntegrationLinux::GetDataSearchLocations(&env),
ElementsAre(base::FilePath("/home/user/.local/share"),
base::FilePath("/system/path/1"),
base::FilePath("/system/path/2")));
}
{
MockEnvironment env;
env.Set("XDG_DATA_DIRS", "/system/path/1:/system/path/2");
EXPECT_THAT(
ShellIntegrationLinux::GetDataSearchLocations(&env),
ElementsAre(base::FilePath("/system/path/1"),
base::FilePath("/system/path/2")));
}
{
MockEnvironment env;
env.Set("HOME", "/home/user");
env.Set("XDG_DATA_HOME", "/user/path");
EXPECT_THAT(
ShellIntegrationLinux::GetDataSearchLocations(&env),
ElementsAre(base::FilePath("/user/path"),
base::FilePath("/usr/local/share"),
base::FilePath("/usr/share")));
}
}
TEST(ShellIntegrationTest, GetExistingShortcutLocations) {
base::FilePath kProfilePath("Profile 1");
const char kExtensionId[] = "test_extension";
const char kTemplateFilename[] = "chrome-test_extension-Profile_1.desktop";
base::FilePath kTemplateFilepath(kTemplateFilename);
const char kNoDisplayDesktopFile[] = "[Desktop Entry]\nNoDisplay=true";
base::MessageLoop message_loop;
content::TestBrowserThread file_thread(BrowserThread::FILE, &message_loop);
{
MockEnvironment env;
ShellIntegration::ShortcutLocations result =
ShellIntegrationLinux::GetExistingShortcutLocations(
&env, kProfilePath, kExtensionId);
EXPECT_FALSE(result.on_desktop);
EXPECT_EQ(ShellIntegration::APP_MENU_LOCATION_NONE,
result.applications_menu_location);
EXPECT_FALSE(result.in_quick_launch_bar);
EXPECT_FALSE(result.hidden);
}
{
base::ScopedTempDir temp_dir;
ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
base::FilePath desktop_path = temp_dir.path();
MockEnvironment env;
ASSERT_TRUE(base::CreateDirectory(desktop_path));
ASSERT_FALSE(base::WriteFile(
desktop_path.AppendASCII(kTemplateFilename),
"", 0));
ShellIntegration::ShortcutLocations result =
ShellIntegrationLinux::GetExistingShortcutLocations(
&env, kProfilePath, kExtensionId, desktop_path);
EXPECT_TRUE(result.on_desktop);
EXPECT_EQ(ShellIntegration::APP_MENU_LOCATION_NONE,
result.applications_menu_location);
EXPECT_FALSE(result.in_quick_launch_bar);
EXPECT_FALSE(result.hidden);
}
{
base::ScopedTempDir temp_dir;
ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
base::FilePath apps_path = temp_dir.path().AppendASCII("applications");
MockEnvironment env;
env.Set("XDG_DATA_HOME", temp_dir.path().value());
ASSERT_TRUE(base::CreateDirectory(apps_path));
ASSERT_FALSE(base::WriteFile(
apps_path.AppendASCII(kTemplateFilename),
"", 0));
ShellIntegration::ShortcutLocations result =
ShellIntegrationLinux::GetExistingShortcutLocations(
&env, kProfilePath, kExtensionId);
EXPECT_FALSE(result.on_desktop);
EXPECT_EQ(ShellIntegration::APP_MENU_LOCATION_SUBDIR_CHROMEAPPS,
result.applications_menu_location);
EXPECT_FALSE(result.in_quick_launch_bar);
EXPECT_FALSE(result.hidden);
}
{
base::ScopedTempDir temp_dir;
ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
base::FilePath apps_path = temp_dir.path().AppendASCII("applications");
MockEnvironment env;
env.Set("XDG_DATA_HOME", temp_dir.path().value());
ASSERT_TRUE(base::CreateDirectory(apps_path));
ASSERT_TRUE(base::WriteFile(
apps_path.AppendASCII(kTemplateFilename),
kNoDisplayDesktopFile, strlen(kNoDisplayDesktopFile)));
ShellIntegration::ShortcutLocations result =
ShellIntegrationLinux::GetExistingShortcutLocations(
&env, kProfilePath, kExtensionId);
EXPECT_FALSE(result.on_desktop);
EXPECT_EQ(ShellIntegration::APP_MENU_LOCATION_NONE,
result.applications_menu_location);
EXPECT_FALSE(result.in_quick_launch_bar);
EXPECT_TRUE(result.hidden);
}
{
base::ScopedTempDir temp_dir1;
ASSERT_TRUE(temp_dir1.CreateUniqueTempDir());
base::FilePath desktop_path = temp_dir1.path();
base::ScopedTempDir temp_dir2;
ASSERT_TRUE(temp_dir2.CreateUniqueTempDir());
base::FilePath apps_path = temp_dir2.path().AppendASCII("applications");
MockEnvironment env;
ASSERT_TRUE(base::CreateDirectory(desktop_path));
ASSERT_FALSE(base::WriteFile(
desktop_path.AppendASCII(kTemplateFilename),
"", 0));
env.Set("XDG_DATA_HOME", temp_dir2.path().value());
ASSERT_TRUE(base::CreateDirectory(apps_path));
ASSERT_FALSE(base::WriteFile(
apps_path.AppendASCII(kTemplateFilename),
"", 0));
ShellIntegration::ShortcutLocations result =
ShellIntegrationLinux::GetExistingShortcutLocations(
&env, kProfilePath, kExtensionId, desktop_path);
EXPECT_TRUE(result.on_desktop);
EXPECT_EQ(ShellIntegration::APP_MENU_LOCATION_SUBDIR_CHROMEAPPS,
result.applications_menu_location);
EXPECT_FALSE(result.in_quick_launch_bar);
EXPECT_FALSE(result.hidden);
}
}
TEST(ShellIntegrationTest, GetExistingShortcutContents) {
const char kTemplateFilename[] = "shortcut-test.desktop";
base::FilePath kTemplateFilepath(kTemplateFilename);
const char kTestData1[] = "a magical testing string";
const char kTestData2[] = "a different testing string";
base::MessageLoop message_loop;
content::TestBrowserThread file_thread(BrowserThread::FILE, &message_loop);
{
base::ScopedTempDir temp_dir;
ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
MockEnvironment env;
env.Set("XDG_DATA_HOME", temp_dir.path().value());
ASSERT_TRUE(base::WriteFile(
temp_dir.path().AppendASCII(kTemplateFilename),
kTestData2, strlen(kTestData2)));
ASSERT_TRUE(base::CreateDirectory(
temp_dir.path().AppendASCII("applications")));
ASSERT_TRUE(base::WriteFile(
temp_dir.path().AppendASCII("applications")
.AppendASCII(kTemplateFilename),
kTestData1, strlen(kTestData1)));
std::string contents;
ASSERT_TRUE(
ShellIntegrationLinux::GetExistingShortcutContents(
&env, kTemplateFilepath, &contents));
EXPECT_EQ(kTestData1, contents);
}
{
base::ScopedTempDir temp_dir;
ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
MockEnvironment env;
env.Set("HOME", temp_dir.path().value());
ASSERT_TRUE(base::CreateDirectory(
temp_dir.path().AppendASCII(".local/share/applications")));
ASSERT_TRUE(base::WriteFile(
temp_dir.path().AppendASCII(".local/share/applications")
.AppendASCII(kTemplateFilename),
kTestData1, strlen(kTestData1)));
std::string contents;
ASSERT_TRUE(
ShellIntegrationLinux::GetExistingShortcutContents(
&env, kTemplateFilepath, &contents));
EXPECT_EQ(kTestData1, contents);
}
{
base::ScopedTempDir temp_dir;
ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
MockEnvironment env;
env.Set("XDG_DATA_DIRS", temp_dir.path().value());
ASSERT_TRUE(base::CreateDirectory(
temp_dir.path().AppendASCII("applications")));
ASSERT_TRUE(base::WriteFile(
temp_dir.path().AppendASCII("applications")
.AppendASCII(kTemplateFilename),
kTestData2, strlen(kTestData2)));
std::string contents;
ASSERT_TRUE(
ShellIntegrationLinux::GetExistingShortcutContents(
&env, kTemplateFilepath, &contents));
EXPECT_EQ(kTestData2, contents);
}
{
base::ScopedTempDir temp_dir1;
ASSERT_TRUE(temp_dir1.CreateUniqueTempDir());
base::ScopedTempDir temp_dir2;
ASSERT_TRUE(temp_dir2.CreateUniqueTempDir());
MockEnvironment env;
env.Set("XDG_DATA_DIRS", temp_dir1.path().value() + ":" +
temp_dir2.path().value());
ASSERT_TRUE(base::WriteFile(
temp_dir1.path().AppendASCII(kTemplateFilename),
kTestData1, strlen(kTestData1)));
ASSERT_TRUE(base::CreateDirectory(
temp_dir2.path().AppendASCII("applications")));
ASSERT_TRUE(base::WriteFile(
temp_dir2.path().AppendASCII("applications")
.AppendASCII(kTemplateFilename),
kTestData2, strlen(kTestData2)));
std::string contents;
ASSERT_TRUE(
ShellIntegrationLinux::GetExistingShortcutContents(
&env, kTemplateFilepath, &contents));
EXPECT_EQ(kTestData2, contents);
}
}
TEST(ShellIntegrationTest, GetExtensionShortcutFilename) {
base::FilePath kProfilePath("a/b/c/Profile Name?");
const char kExtensionId[] = "extensionid";
EXPECT_EQ(base::FilePath("chrome-extensionid-Profile_Name_.desktop"),
ShellIntegrationLinux::GetExtensionShortcutFilename(
kProfilePath, kExtensionId));
}
TEST(ShellIntegrationTest, GetExistingProfileShortcutFilenames) {
base::FilePath kProfilePath("a/b/c/Profile Name?");
const char kApp1Filename[] = "chrome-extension1-Profile_Name_.desktop";
const char kApp2Filename[] = "chrome-extension2-Profile_Name_.desktop";
const char kUnrelatedAppFilename[] = "chrome-extension-Other_Profile.desktop";
base::MessageLoop message_loop;
content::TestBrowserThread file_thread(BrowserThread::FILE, &message_loop);
base::ScopedTempDir temp_dir;
ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
ASSERT_EQ(0,
base::WriteFile(
temp_dir.path().AppendASCII(kApp1Filename), "", 0));
ASSERT_EQ(0,
base::WriteFile(
temp_dir.path().AppendASCII(kApp2Filename), "", 0));
ASSERT_EQ(0,
base::WriteFile(
temp_dir.path().AppendASCII(kUnrelatedAppFilename), "", 0));
std::vector<base::FilePath> paths =
ShellIntegrationLinux::GetExistingProfileShortcutFilenames(
kProfilePath, temp_dir.path());
std::sort(paths.begin(), paths.end());
EXPECT_THAT(paths,
ElementsAre(base::FilePath(kApp1Filename),
base::FilePath(kApp2Filename)));
}
TEST(ShellIntegrationTest, GetWebShortcutFilename) {
const struct {
const base::FilePath::CharType* path;
const char* url;
} test_cases[] = {
{ FPL("http___foo_.desktop"), "http://foo" },
{ FPL("http___foo_bar_.desktop"), "http://foo/bar/" },
{ FPL("http___foo_bar_a=b&c=d.desktop"), "http://foo/bar?a=b&c=d" },
{ FPL("http___foo_.desktop"), "http://foo/bar/baz/../../../../../" },
{ FPL("http___foo_.desktop"), "http://foo/bar/././../baz/././../" },
{ FPL("http___.._.desktop"), "http://../../../../" },
};
for (size_t i = 0; i < ARRAYSIZE_UNSAFE(test_cases); i++) {
EXPECT_EQ(std::string(chrome::kBrowserProcessExecutableName) + "-" +
test_cases[i].path,
ShellIntegrationLinux::GetWebShortcutFilename(
GURL(test_cases[i].url)).value()) <<
" while testing " << test_cases[i].url;
}
}
TEST(ShellIntegrationTest, GetDesktopFileContents) {
const base::FilePath kChromeExePath("/opt/google/chrome/google-chrome");
const struct {
const char* url;
const char* title;
const char* icon_name;
bool nodisplay;
const char* expected_output;
} test_cases[] = {
{ "http://gmail.com",
"GMail",
"chrome-http__gmail.com",
false,
"#!/usr/bin/env xdg-open\n"
"[Desktop Entry]\n"
"Version=1.0\n"
"Terminal=false\n"
"Type=Application\n"
"Name=GMail\n"
"Exec=/opt/google/chrome/google-chrome --app=http://gmail.com/\n"
"Icon=chrome-http__gmail.com\n"
"StartupWMClass=gmail.com\n"
},
{ "http://gmail.com",
"GMail",
"",
false,
"#!/usr/bin/env xdg-open\n"
"[Desktop Entry]\n"
"Version=1.0\n"
"Terminal=false\n"
"Type=Application\n"
"Name=GMail\n"
"Exec=/opt/google/chrome/google-chrome --app=http://gmail.com/\n"
#if defined(GOOGLE_CHROME_BUILD)
"Icon=google-chrome\n"
#else
"Icon=chromium-browser\n"
#endif
"StartupWMClass=gmail.com\n"
},
{ "http://gmail.com",
"GMail",
"chrome-http__gmail.com",
true,
"#!/usr/bin/env xdg-open\n"
"[Desktop Entry]\n"
"Version=1.0\n"
"Terminal=false\n"
"Type=Application\n"
"Name=GMail\n"
"Exec=/opt/google/chrome/google-chrome --app=http://gmail.com/\n"
"Icon=chrome-http__gmail.com\n"
"NoDisplay=true\n"
"StartupWMClass=gmail.com\n"
},
{ "http://evil.com/evil --join-the-b0tnet",
"Ownz0red\nExec=rm -rf /",
"chrome-http__evil.com_evil",
false,
"#!/usr/bin/env xdg-open\n"
"[Desktop Entry]\n"
"Version=1.0\n"
"Terminal=false\n"
"Type=Application\n"
"Name=http://evil.com/evil%20--join-the-b0tnet\n"
"Exec=/opt/google/chrome/google-chrome "
"--app=http://evil.com/evil%20--join-the-b0tnet\n"
"Icon=chrome-http__evil.com_evil\n"
"StartupWMClass=evil.com__evil%20--join-the-b0tnet\n"
},
{ "http://evil.com/evil; rm -rf /; \"; rm -rf $HOME >ownz0red",
"Innocent Title",
"chrome-http__evil.com_evil",
false,
"#!/usr/bin/env xdg-open\n"
"[Desktop Entry]\n"
"Version=1.0\n"
"Terminal=false\n"
"Type=Application\n"
"Name=Innocent Title\n"
"Exec=/opt/google/chrome/google-chrome "
"\"--app=http://evil.com/evil;%20rm%20-rf%20/;%20%22;%20rm%20"
"-rf%20\\\\$HOME%20%3Eownz0red\"\n"
"Icon=chrome-http__evil.com_evil\n"
"StartupWMClass=evil.com__evil;%20rm%20-rf%20_;%20%22;%20"
"rm%20-rf%20$HOME%20%3Eownz0red\n"
},
{ "http://evil.com/evil | cat `echo ownz0red` >/dev/null",
"Innocent Title",
"chrome-http__evil.com_evil",
false,
"#!/usr/bin/env xdg-open\n"
"[Desktop Entry]\n"
"Version=1.0\n"
"Terminal=false\n"
"Type=Application\n"
"Name=Innocent Title\n"
"Exec=/opt/google/chrome/google-chrome "
"--app=http://evil.com/evil%20%7C%20cat%20%60echo%20ownz0red"
"%60%20%3E/dev/null\n"
"Icon=chrome-http__evil.com_evil\n"
"StartupWMClass=evil.com__evil%20%7C%20cat%20%60echo%20ownz0red"
"%60%20%3E_dev_null\n"
},
};
for (size_t i = 0; i < ARRAYSIZE_UNSAFE(test_cases); i++) {
SCOPED_TRACE(i);
EXPECT_EQ(
test_cases[i].expected_output,
ShellIntegrationLinux::GetDesktopFileContents(
kChromeExePath,
web_app::GenerateApplicationNameFromURL(GURL(test_cases[i].url)),
GURL(test_cases[i].url),
std::string(),
base::ASCIIToUTF16(test_cases[i].title),
test_cases[i].icon_name,
base::FilePath(),
test_cases[i].nodisplay));
}
}
TEST(ShellIntegrationTest, GetDesktopFileContentsAppList) {
const base::FilePath kChromeExePath("/opt/google/chrome/google-chrome");
CommandLine command_line(kChromeExePath);
command_line.AppendSwitch("--show-app-list");
EXPECT_EQ(
"#!/usr/bin/env xdg-open\n"
"[Desktop Entry]\n"
"Version=1.0\n"
"Terminal=false\n"
"Type=Application\n"
"Name=Chrome App Launcher\n"
"Exec=/opt/google/chrome/google-chrome --show-app-list\n"
"Icon=chrome_app_list\n"
"StartupWMClass=chrome-app-list\n",
ShellIntegrationLinux::GetDesktopFileContentsForCommand(
command_line,
"chrome-app-list",
GURL(),
base::ASCIIToUTF16("Chrome App Launcher"),
"chrome_app_list",
false));
}
TEST(ShellIntegrationTest, GetDirectoryFileContents) {
const struct {
const char* title;
const char* icon_name;
const char* expected_output;
} test_cases[] = {
{ "Chrome Apps",
"chrome-apps",
"[Desktop Entry]\n"
"Version=1.0\n"
"Type=Directory\n"
"Name=Chrome Apps\n"
"Icon=chrome-apps\n"
},
{ "Chrome Apps",
"",
"[Desktop Entry]\n"
"Version=1.0\n"
"Type=Directory\n"
"Name=Chrome Apps\n"
#if defined(GOOGLE_CHROME_BUILD)
"Icon=google-chrome\n"
#else
"Icon=chromium-browser\n"
#endif
},
};
for (size_t i = 0; i < ARRAYSIZE_UNSAFE(test_cases); i++) {
SCOPED_TRACE(i);
EXPECT_EQ(
test_cases[i].expected_output,
ShellIntegrationLinux::GetDirectoryFileContents(
base::ASCIIToUTF16(test_cases[i].title),
test_cases[i].icon_name));
}
}