Home | History | Annotate | Download | only in chrome 
      1 // Copyright (c) 2012 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/test/chromedriver/chrome/chrome_finder.h"
      6 
      7 #include <string>
      8 #include <vector>
      9 
     10 #include "base/base_paths.h"
     11 #include "base/bind.h"
     12 #include "base/callback.h"
     13 #include "base/file_util.h"
     14 #include "base/files/file_path.h"
     15 #include "base/path_service.h"
     16 #include "build/build_config.h"
     17 
     18 #if defined(OS_WIN)
     19 #include "base/base_paths_win.h"
     20 #include "base/win/windows_version.h"
     21 #endif
     22 
     23 namespace {
     24 
     25 #if defined(OS_WIN)
     26 void GetApplicationDirs(std::vector<base::FilePath>* locations) {
     27   std::vector<base::FilePath> installation_locations;
     28   base::FilePath local_app_data, program_files, program_files_x86;
     29   if (PathService::Get(base::DIR_LOCAL_APP_DATA, &local_app_data))
     30     installation_locations.push_back(local_app_data);
     31   if (PathService::Get(base::DIR_PROGRAM_FILES, &program_files))
     32     installation_locations.push_back(program_files);
     33   if (PathService::Get(base::DIR_PROGRAM_FILESX86, &program_files_x86))
     34     installation_locations.push_back(program_files_x86);
     35 
     36   for (size_t i = 0; i < installation_locations.size(); ++i) {
     37     locations->push_back(
     38         installation_locations[i].Append(L"Google\\Chrome\\Application"));
     39   }
     40   for (size_t i = 0; i < installation_locations.size(); ++i) {
     41     locations->push_back(
     42         installation_locations[i].Append(L"Chromium\\Application"));
     43   }
     44 }
     45 #elif defined(OS_LINUX)
     46 void GetApplicationDirs(std::vector<base::FilePath>* locations) {
     47   locations->push_back(base::FilePath("/opt/google/chrome"));
     48   locations->push_back(base::FilePath("/usr/local/bin"));
     49   locations->push_back(base::FilePath("/usr/local/sbin"));
     50   locations->push_back(base::FilePath("/usr/bin"));
     51   locations->push_back(base::FilePath("/usr/sbin"));
     52   locations->push_back(base::FilePath("/bin"));
     53   locations->push_back(base::FilePath("/sbin"));
     54 }
     55 #endif
     56 
     57 }  // namespace
     58 
     59 namespace internal {
     60 
     61 bool FindExe(
     62     const base::Callback<bool(const base::FilePath&)>& exists_func,
     63     const std::vector<base::FilePath>& rel_paths,
     64     const std::vector<base::FilePath>& locations,
     65     base::FilePath* out_path) {
     66   for (size_t i = 0; i < rel_paths.size(); ++i) {
     67     for (size_t j = 0; j < locations.size(); ++j) {
     68       base::FilePath path = locations[j].Append(rel_paths[i]);
     69       if (exists_func.Run(path)) {
     70         *out_path = path;
     71         return true;
     72       }
     73     }
     74   }
     75   return false;
     76 }
     77 
     78 }  // namespace internal
     79 
     80 #if defined(OS_MACOSX)
     81 void GetApplicationDirs(std::vector<base::FilePath>* locations);
     82 #endif
     83 
     84 bool FindChrome(base::FilePath* browser_exe) {
     85 #if defined(OS_WIN)
     86   base::FilePath browser_exes_array[] = {
     87       base::FilePath(L"chrome.exe")
     88   };
     89 #elif defined(OS_MACOSX)
     90   base::FilePath browser_exes_array[] = {
     91       base::FilePath("Google Chrome.app/Contents/MacOS/Google Chrome"),
     92       base::FilePath("Chromium.app/Contents/MacOS/Chromium")
     93   };
     94 #elif defined(OS_LINUX)
     95   base::FilePath browser_exes_array[] = {
     96       base::FilePath("google-chrome"),
     97       base::FilePath("chrome"),
     98       base::FilePath("chromium"),
     99       base::FilePath("chromium-browser")
    100   };
    101 #endif
    102   std::vector<base::FilePath> browser_exes(
    103       browser_exes_array, browser_exes_array + arraysize(browser_exes_array));
    104   std::vector<base::FilePath> locations;
    105   base::FilePath module_dir;
    106   if (PathService::Get(base::DIR_MODULE, &module_dir))
    107     locations.push_back(module_dir);
    108   GetApplicationDirs(&locations);
    109   return internal::FindExe(
    110       base::Bind(&base::PathExists),
    111       browser_exes,
    112       locations,
    113       browser_exe);
    114 }
    115