Home | History | Annotate | Download | only in test 
      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 // Win8 default browser registration utility.
      6 //
      7 // This tool can register and unregister a given exe as a potential default
      8 // metro browser on Win8. It does not make the exe become THE default browser,
      9 // for a mechnism to do this please see open_with_dialog_controller.h.
     10 //
     11 // TODO(robertshield): By default, this creates a run-time dependency on
     12 // chrome.exe since it's the only thing we have right now that works as a
     13 // default viewer process. Investigate extracting the metro init code and
     14 // building them into a standalone viewer process.
     15 
     16 #include <atlbase.h>
     17 #include <atlcom.h>
     18 #include <atlctl.h>
     19 #include <shellapi.h>
     20 
     21 #include "base/at_exit.h"
     22 #include "base/command_line.h"
     23 #include "base/files/file_path.h"
     24 #include "base/memory/scoped_ptr.h"
     25 #include "base/path_service.h"
     26 #include "base/strings/stringprintf.h"
     27 #include "win8/test/test_registrar_constants.h"
     28 #include "win8/test/test_registrar_resource.h"
     29 
     30 namespace {
     31 
     32 const wchar_t kDelegateExecuteCLSID[] =
     33     L"{FC0064A6-D1DE-4A83-92D2-5BB4EEBB70B5}";
     34 
     35 void InitializeCommandLineDefaultValues() {
     36   CommandLine& command_line = *CommandLine::ForCurrentProcess();
     37 
     38   if (!command_line.HasSwitch(win8::test::kTestAppUserModelId))
     39     command_line.AppendSwitchNative(win8::test::kTestAppUserModelId,
     40                                     win8::test::kDefaultTestAppUserModelId);
     41 
     42   if (!command_line.HasSwitch(win8::test::kTestExeName))
     43     command_line.AppendSwitchNative(win8::test::kTestExeName,
     44                                     win8::test::kDefaultTestExeName);
     45 
     46   if (!command_line.HasSwitch(win8::test::kTestExePath)) {
     47     base::FilePath exe_path;
     48     PathService::Get(base::DIR_EXE, &exe_path);
     49     exe_path = exe_path.Append(win8::test::kDefaultTestExePath);
     50 
     51     command_line.AppendSwitchNative(win8::test::kTestExePath,
     52                                     exe_path.value());
     53   }
     54 
     55   if (!command_line.HasSwitch(win8::test::kTestProgId))
     56     command_line.AppendSwitchNative(win8::test::kTestProgId,
     57                                     win8::test::kDefaultTestProgId);
     58 }
     59 
     60 }  // namespace
     61 
     62 // Implementation of an ATL module that provides the necessary replacement
     63 // values for the default browser .rgs script.
     64 class TestDelegateExecuteModule
     65     : public ATL::CAtlExeModuleT< TestDelegateExecuteModule > {
     66  public :
     67   typedef ATL::CAtlExeModuleT<TestDelegateExecuteModule> ParentClass;
     68 
     69   DECLARE_REGISTRY_RESOURCEID(IDR_TESTDELEGATEEXECUTE);
     70 
     71   HRESULT RegisterServer(BOOL reg_type_lib) {
     72     return ParentClass::RegisterServer(FALSE);
     73   }
     74 
     75   virtual HRESULT AddCommonRGSReplacements(IRegistrarBase* registrar) throw() {
     76     AtlTrace(L"In %hs\n", __FUNCTION__);
     77     HRESULT hr = ParentClass::AddCommonRGSReplacements(registrar);
     78     if (FAILED(hr))
     79       return hr;
     80 
     81     registrar->AddReplacement(L"DELEGATE_EXECUTE_CLSID", kDelegateExecuteCLSID);
     82 
     83     CommandLine& command_line = *CommandLine::ForCurrentProcess();
     84 
     85     registrar->AddReplacement(L"APP_USER_MODEL_ID",
     86                               command_line.GetSwitchValueNative(
     87                                   win8::test::kTestAppUserModelId).c_str());
     88     registrar->AddReplacement(L"EXE_NAME",
     89                               command_line.GetSwitchValueNative(
     90                                   win8::test::kTestExeName).c_str());
     91     registrar->AddReplacement(L"PROG_ID",
     92                               command_line.GetSwitchValueNative(
     93                                   win8::test::kTestProgId).c_str());
     94 
     95     string16 exe_path(
     96         command_line.GetSwitchValueNative(win8::test::kTestExePath));
     97 
     98     string16 exe_open_command(base::StringPrintf(L"\"%ls\" -- %%*",
     99                                                  exe_path.c_str()));
    100     registrar->AddReplacement(L"EXE_OPEN_COMMAND", exe_open_command.c_str());
    101 
    102     string16 exe_icon(base::StringPrintf(L"%ls,0", exe_path.c_str()));
    103     registrar->AddReplacement(L"EXE_ICON", exe_icon.c_str());
    104 
    105     string16 prog_id_open_command(base::StringPrintf(L"\"%ls\" -- \"%%1\"",
    106                                                      exe_path.c_str()));
    107     registrar->AddReplacement(L"PROG_ID_OPEN_COMMAND",
    108                               prog_id_open_command.c_str());
    109 
    110     ATLASSERT(SUCCEEDED(hr));
    111     return hr;
    112   }
    113 };
    114 
    115 TestDelegateExecuteModule _AtlModule;
    116 
    117 EXTERN_C const GUID CLSID_TestDefaultBrowserRegistrar;
    118 class ATL_NO_VTABLE DECLSPEC_UUID("FC0064A6-D1DE-4A83-92D2-5BB4EEBB70B5")
    119     TestDefaultBrowserRegistrar
    120     : public CComObjectRootEx<CComSingleThreadModel>,
    121       public CComCoClass<TestDefaultBrowserRegistrar,
    122                          &CLSID_TestDefaultBrowserRegistrar> {
    123  public:
    124   DECLARE_REGISTRY_RESOURCEID(IDR_TESTDELEGATEEXECUTE);
    125 
    126   BEGIN_COM_MAP(TestDefaultBrowserRegistrar)
    127   END_COM_MAP()
    128 };
    129 
    130 extern "C" int WINAPI _tWinMain(HINSTANCE, HINSTANCE, LPTSTR, int nShowCmd) {
    131   base::AtExitManager exit_manager;
    132   CommandLine::Init(0, NULL);
    133   InitializeCommandLineDefaultValues();
    134 
    135   HRESULT ret_code = _AtlModule.WinMain(nShowCmd);
    136 
    137   return ret_code;
    138 }
    139 
    140 OBJECT_ENTRY_AUTO(__uuidof(TestDefaultBrowserRegistrar),
    141                   TestDefaultBrowserRegistrar);
    142