win8/test/test_registrar.cc

   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   base::CommandLine& command_line = *base::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   HRESULT AddCommonRGSReplacements(IRegistrarBase* registrar) throw() override {
  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     base::CommandLine& command_line = *base::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     base::string16 exe_path(
  96         command_line.GetSwitchValueNative(win8::test::kTestExePath));
  97
  98     base::string16 exe_open_command(
  99         base::StringPrintf(L"\"%ls\" -- %%*", exe_path.c_str()));
 100     registrar->AddReplacement(L"EXE_OPEN_COMMAND", exe_open_command.c_str());
 101
 102     base::string16 exe_icon(base::StringPrintf(L"%ls,0", exe_path.c_str()));
 103     registrar->AddReplacement(L"EXE_ICON", exe_icon.c_str());
 104
 105     base::string16 prog_id_open_command(
 106         base::StringPrintf(L"\"%ls\" -- \"%%1\"", 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   base::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);