Source code
Revision control
Copy as Markdown
Other Tools
/*
* Copyright (c) 2019 The WebRTC project authors. All Rights Reserved.
*
* Use of this source code is governed by a BSD-style license
* that can be found in the LICENSE file in the root of the source
* tree. An additional intellectual property rights grant can be found
* in the file PATENTS. All contributing project authors may
* be found in the AUTHORS file in the root of the source tree.
*/
#include "modules/desktop_capture/mac/full_screen_mac_application_handler.h"
#include <libproc.h>
#include <algorithm>
#include <functional>
#include <string>
#include "absl/strings/match.h"
#include "absl/strings/string_view.h"
#include "api/function_view.h"
#include "modules/desktop_capture/mac/window_list_utils.h"
namespace webrtc {
namespace {
static constexpr const char* kPowerPointSlideShowTitles[] = {
"PowerPoint-Bildschirmpräsentation",
"Προβολή παρουσίασης PowerPoint",
"PowerPoint スライド ショー",
"PowerPoint Slide Show",
"PowerPoint 幻灯片放映",
"Presentación de PowerPoint",
"PowerPoint-slideshow",
"Presentazione di PowerPoint",
"Prezentácia programu PowerPoint",
"Apresentação do PowerPoint",
"PowerPoint-bildspel",
"Prezentace v aplikaci PowerPoint",
"PowerPoint 슬라이드 쇼",
"PowerPoint-lysbildefremvisning",
"PowerPoint-vetítés",
"PowerPoint Slayt Gösterisi",
"Pokaz slajdów programu PowerPoint",
"PowerPoint 投影片放映",
"Демонстрация PowerPoint",
"Diaporama PowerPoint",
"PowerPoint-diaesitys",
"Peragaan Slide PowerPoint",
"PowerPoint-diavoorstelling",
"การนำเสนอสไลด์ PowerPoint",
"Apresentação de slides do PowerPoint",
"הצגת שקופיות של PowerPoint",
"عرض شرائح في PowerPoint"};
class FullScreenMacApplicationHandler : public FullScreenApplicationHandler {
public:
using TitlePredicate =
std::function<bool(absl::string_view, absl::string_view)>;
FullScreenMacApplicationHandler(DesktopCapturer::SourceId sourceId,
TitlePredicate title_predicate,
bool ignore_original_window)
: FullScreenApplicationHandler(sourceId),
title_predicate_(title_predicate),
owner_pid_(GetWindowOwnerPid(sourceId)),
ignore_original_window_(ignore_original_window) {}
protected:
using CachePredicate =
rtc::FunctionView<bool(const DesktopCapturer::Source&)>;
void InvalidateCacheIfNeeded(const DesktopCapturer::SourceList& source_list,
int64_t timestamp,
CachePredicate predicate) const {
if (timestamp != cache_timestamp_) {
cache_sources_.clear();
std::copy_if(source_list.begin(), source_list.end(),
std::back_inserter(cache_sources_), predicate);
cache_timestamp_ = timestamp;
}
}
WindowId FindFullScreenWindowWithSamePid(
const DesktopCapturer::SourceList& source_list,
int64_t timestamp) const {
InvalidateCacheIfNeeded(source_list, timestamp,
[&](const DesktopCapturer::Source& src) {
return src.id != GetSourceId() &&
GetWindowOwnerPid(src.id) == owner_pid_;
});
if (cache_sources_.empty())
return kCGNullWindowID;
const auto original_window = GetSourceId();
const std::string title = GetWindowTitle(original_window);
// We can ignore any windows with empty titles cause regardless type of
// application it's impossible to verify that full screen window and
// original window are related to the same document.
if (title.empty())
return kCGNullWindowID;
MacDesktopConfiguration desktop_config =
MacDesktopConfiguration::GetCurrent(
MacDesktopConfiguration::TopLeftOrigin);
const auto it = std::find_if(
cache_sources_.begin(), cache_sources_.end(),
[&](const DesktopCapturer::Source& src) {
const std::string window_title = GetWindowTitle(src.id);
if (window_title.empty())
return false;
if (title_predicate_ && !title_predicate_(title, window_title))
return false;
return IsWindowFullScreen(desktop_config, src.id);
});
return it != cache_sources_.end() ? it->id : 0;
}
DesktopCapturer::SourceId FindFullScreenWindow(
const DesktopCapturer::SourceList& source_list,
int64_t timestamp) const override {
return !ignore_original_window_ && IsWindowOnScreen(GetSourceId())
? 0
: FindFullScreenWindowWithSamePid(source_list, timestamp);
}
protected:
const TitlePredicate title_predicate_;
const int owner_pid_;
const bool ignore_original_window_;
mutable int64_t cache_timestamp_ = 0;
mutable DesktopCapturer::SourceList cache_sources_;
};
bool equal_title_predicate(absl::string_view original_title,
absl::string_view title) {
return original_title == title;
}
bool slide_show_title_predicate(absl::string_view original_title,
absl::string_view title) {
if (title.find(original_title) == absl::string_view::npos)
return false;
for (const char* pp_slide_title : kPowerPointSlideShowTitles) {
if (absl::StartsWith(title, pp_slide_title))
return true;
}
return false;
}
class OpenOfficeApplicationHandler : public FullScreenMacApplicationHandler {
public:
OpenOfficeApplicationHandler(DesktopCapturer::SourceId sourceId)
: FullScreenMacApplicationHandler(sourceId, nullptr, false) {}
DesktopCapturer::SourceId FindFullScreenWindow(
const DesktopCapturer::SourceList& source_list,
int64_t timestamp) const override {
InvalidateCacheIfNeeded(source_list, timestamp,
[&](const DesktopCapturer::Source& src) {
return GetWindowOwnerPid(src.id) == owner_pid_;
});
const auto original_window = GetSourceId();
const std::string original_title = GetWindowTitle(original_window);
// Check if we have only one document window, otherwise it's not possible
// to securely match a document window and a slide show window which has
// empty title.
if (std::any_of(cache_sources_.begin(), cache_sources_.end(),
[&original_title](const DesktopCapturer::Source& src) {
return src.title.length() && src.title != original_title;
})) {
return kCGNullWindowID;
}
MacDesktopConfiguration desktop_config =
MacDesktopConfiguration::GetCurrent(
MacDesktopConfiguration::TopLeftOrigin);
// Looking for slide show window,
// it must be a full screen window with empty title
const auto slide_show_window = std::find_if(
cache_sources_.begin(), cache_sources_.end(), [&](const auto& src) {
return src.title.empty() &&
IsWindowFullScreen(desktop_config, src.id);
});
if (slide_show_window == cache_sources_.end()) {
return kCGNullWindowID;
}
return slide_show_window->id;
}
};
} // namespace
std::unique_ptr<FullScreenApplicationHandler>
CreateFullScreenMacApplicationHandler(DesktopCapturer::SourceId sourceId) {
std::unique_ptr<FullScreenApplicationHandler> result;
int pid = GetWindowOwnerPid(sourceId);
char buffer[PROC_PIDPATHINFO_MAXSIZE];
int path_length = proc_pidpath(pid, buffer, sizeof(buffer));
if (path_length > 0) {
const char* last_slash = strrchr(buffer, '/');
const std::string name{last_slash ? last_slash + 1 : buffer};
const std::string owner_name = GetWindowOwnerName(sourceId);
FullScreenMacApplicationHandler::TitlePredicate predicate = nullptr;
bool ignore_original_window = false;
if (name.find("Google Chrome") == 0 || name == "Chromium") {
predicate = equal_title_predicate;
} else if (name == "Microsoft PowerPoint") {
predicate = slide_show_title_predicate;
ignore_original_window = true;
} else if (name == "Keynote") {
predicate = equal_title_predicate;
} else if (owner_name == "OpenOffice") {
return std::make_unique<OpenOfficeApplicationHandler>(sourceId);
}
if (predicate) {
result.reset(new FullScreenMacApplicationHandler(sourceId, predicate,
ignore_original_window));
}
}
return result;
}
} // namespace webrtc