blob: 9503f59109ffc958e84816fe6cf8364a9c60e06d (
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
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
|
// Copyright (C) 2023 The Qt Company Ltd.
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR LGPL-3.0-only OR GPL-2.0-only OR GPL-3.0-only
// Copyright 2021 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.Chromium file.
#include "pdf_util_qt.h"
#include <QtGlobal>
#include "base/check.h"
#include "chrome/common/webui_url_constants.h"
#include "content/public/browser/render_frame_host.h"
#include "content/public/browser/render_process_host.h"
#include "content/public/browser/web_contents.h"
#include "extensions/buildflags/buildflags.h"
#include "url/gurl.h"
#include "url/origin.h"
#if BUILDFLAG(ENABLE_EXTENSIONS)
#include "extensions/browser/guest_view/mime_handler_view/mime_handler_view_guest.h"
#include "extensions/common/constants.h"
#endif // BUILDFLAG(ENABLE_EXTENSIONS)
namespace QtWebEngineCore {
bool IsPdfExtensionOrigin(const url::Origin &origin)
{
#if BUILDFLAG(ENABLE_EXTENSIONS)
return origin.scheme() == extensions::kExtensionScheme
&& origin.host() == extension_misc::kPdfExtensionId;
#else
Q_UNUSED(origin);
return false;
#endif
}
bool IsPdfInternalPluginAllowedOrigin(const url::Origin &origin)
{
if (IsPdfExtensionOrigin(origin))
return true;
// Allow embedding the internal PDF plugin in chrome://print.
if (origin == url::Origin::Create(GURL(chrome::kChromeUIPrintURL)))
return true;
// Only allow the PDF plugin in the known, trustworthy origins that are
// allowlisted above. See also https://crbug.com/520422 and
// https://crbug.com/1027173.
return false;
}
content::RenderFrameHost *GetFullPagePlugin(content::WebContents *contents)
{
content::RenderFrameHost *full_page_plugin = nullptr;
#if BUILDFLAG(ENABLE_EXTENSIONS)
contents->ForEachRenderFrameHostWithAction([&full_page_plugin](content::RenderFrameHost *rfh) {
auto* guest_view = extensions::MimeHandlerViewGuest::FromRenderFrameHost(rfh);
if (guest_view && guest_view->is_full_page_plugin()) {
DCHECK_EQ(guest_view->GetGuestMainFrame(), rfh);
full_page_plugin = rfh;
return content::RenderFrameHost::FrameIterationAction::kStop;
}
return content::RenderFrameHost::FrameIterationAction::kContinue;
});
#endif // BUILDFLAG(ENABLE_EXTENSIONS)
return full_page_plugin;
}
content::RenderFrameHost *FindPdfChildFrame(content::RenderFrameHost *rfh)
{
if (!rfh)
return nullptr;
if (!IsPdfExtensionOrigin(rfh->GetLastCommittedOrigin()))
return nullptr;
content::RenderFrameHost *pdf_rfh = nullptr;
rfh->ForEachRenderFrameHost([&pdf_rfh](content::RenderFrameHost *rfh) {
if (!rfh->GetProcess()->IsPdf())
return;
DCHECK(IsPdfExtensionOrigin(rfh->GetParent()->GetLastCommittedOrigin()));
DCHECK(!pdf_rfh);
pdf_rfh = rfh;
});
return pdf_rfh;
}
} // namespace QtWebEngineCore
|
