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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
// Copyright 2015 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef HEADLESS_LIB_HEADLESS_CONTENT_MAIN_DELEGATE_H_
#define HEADLESS_LIB_HEADLESS_CONTENT_MAIN_DELEGATE_H_
#include <memory>
#include <optional>
#include <string>
#include <variant>
#include "build/build_config.h"
#include "content/public/app/content_main_delegate.h"
#include "content/public/browser/content_browser_client.h"
#include "content/public/renderer/content_renderer_client.h"
#include "headless/lib/headless_content_client.h"
#include "headless/public/headless_browser.h"
#include "headless/public/headless_export.h"
namespace base {
class CommandLine;
} // namespace base
namespace headless {
class HeadlessBrowserImpl;
// Exported for tests.
class HEADLESS_EXPORT HeadlessContentMainDelegate
: public content::ContentMainDelegate {
public:
explicit HeadlessContentMainDelegate(
std::unique_ptr<HeadlessBrowserImpl> browser);
HeadlessContentMainDelegate(const HeadlessContentMainDelegate&) = delete;
HeadlessContentMainDelegate& operator=(const HeadlessContentMainDelegate&) =
delete;
~HeadlessContentMainDelegate() override;
private:
// content::ContentMainDelegate implementation:
std::optional<int> BasicStartupComplete() override;
void PreSandboxStartup() override;
std::variant<int, content::MainFunctionParams> RunProcess(
const std::string& process_type,
content::MainFunctionParams main_function_params) override;
std::optional<int> PreBrowserMain() override;
#if BUILDFLAG(IS_WIN)
bool ShouldHandleConsoleControlEvents() override;
#endif
content::ContentClient* CreateContentClient() override;
content::ContentBrowserClient* CreateContentBrowserClient() override;
content::ContentUtilityClient* CreateContentUtilityClient() override;
content::ContentRendererClient* CreateContentRendererClient() override;
std::optional<int> PostEarlyInitialization(InvokedIn invoked_in) override;
#if defined(HEADLESS_SUPPORT_FIELD_TRIALS)
bool ShouldCreateFeatureList(InvokedIn invoked_in) override;
bool ShouldInitializeMojo(InvokedIn invoked_in) override;
#endif
#if BUILDFLAG(IS_MAC)
void PlatformPreBrowserMain();
#endif
// TODO(caseq): get rid of this method and GetInstance(), tests should get
// browser through other means.
// Note this is nullptr in processes other than the browser.
HeadlessBrowserImpl* browser() const { return browser_.get(); }
#if BUILDFLAG(IS_LINUX) || BUILDFLAG(IS_CHROMEOS)
void ZygoteForked() override;
#endif
friend class HeadlessBrowserTest;
static HeadlessContentMainDelegate* GetInstance();
void InitLogging(const base::CommandLine& command_line);
void InitCrashReporter(const base::CommandLine& command_line);
// Other clients may retain pointers to browser, so it should come
// first.
std::unique_ptr<HeadlessBrowserImpl> const browser_;
std::unique_ptr<content::ContentRendererClient> renderer_client_;
std::unique_ptr<content::ContentBrowserClient> browser_client_;
std::unique_ptr<content::ContentUtilityClient> utility_client_;
HeadlessContentClient content_client_;
};
} // namespace headless
#endif // HEADLESS_LIB_HEADLESS_CONTENT_MAIN_DELEGATE_H_
|
