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
30namespace {
31
32const wchar_t kDelegateExecuteCLSID[] =
33    L"{FC0064A6-D1DE-4A83-92D2-5BB4EEBB70B5}";
34
35void 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.
64class 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
115TestDelegateExecuteModule _AtlModule;
116
117EXTERN_C const GUID CLSID_TestDefaultBrowserRegistrar;
118class 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
130extern "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
140OBJECT_ENTRY_AUTO(__uuidof(TestDefaultBrowserRegistrar),
141                  TestDefaultBrowserRegistrar);
142