blob: b07f50a105caa961d4934f9206aa2d0a63508abe (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
|
// Copyright (c) 2013 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
// Makes a given program ("Google Chrome" by default) the default handler for
// some URL protocol ("http" by default) on Windows 8. These defaults can be
// overridden via the --program and --protocol command line switches.
#include <windows.h>
#include "base/at_exit.h"
#include "base/command_line.h"
#include "base/logging.h"
#include "base/strings/string16.h"
#include "base/strings/string_util.h"
#include "ui/base/win/atl_module.h"
#include "win8/test/open_with_dialog_controller.h"
namespace {
const char kSwitchProgram[] = "program";
const char kSwitchProtocol[] = "protocol";
const wchar_t kDefaultProgram[] = L"Google Chrome";
const wchar_t kDefaultProtocol[] = L"http";
} // namespace
extern "C"
int wmain(int argc, wchar_t* argv[]) {
// Initialize the commandline singleton from the environment.
CommandLine::Init(0, NULL);
// The exit manager is in charge of calling the dtors of singletons.
base::AtExitManager exit_manager;
logging::LoggingSettings settings;
settings.logging_dest = logging::LOG_TO_SYSTEM_DEBUG_LOG;
logging::InitLogging(settings);
logging::SetMinLogLevel(logging::LOG_VERBOSE);
ui::win::CreateATLModuleIfNeeded();
CommandLine* command_line = CommandLine::ForCurrentProcess();
base::string16 protocol(command_line->GetSwitchValueNative(kSwitchProtocol));
if (protocol.empty())
protocol = kDefaultProtocol;
base::string16 program(command_line->GetSwitchValueNative(kSwitchProgram));
if (program.empty())
program = kDefaultProgram;
std::vector<base::string16> choices;
HRESULT result = S_OK;
win8::OpenWithDialogController controller;
result = controller.RunSynchronously(NULL, protocol, program, &choices);
if (SUCCEEDED(result)) {
printf("success\n");
} else if (!choices.empty()) {
printf("failed to set program. possible choices: %ls\n",
JoinString(choices, L", ").c_str());
} else {
printf("failed with HRESULT: %0x08X\n", result);
}
return FAILED(result);
}
|
