mirror of
https://github.com/kunkundi/crossdesk.git
synced 2026-03-24 17:23:17 +08:00
Compare commits
9 Commits
latest
...
file-trans
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
eee6c588bd | ||
|
|
eca68f6c7a | ||
|
|
f4e28d8774 | ||
|
|
21b179e01c | ||
|
|
83cacf6f51 | ||
|
|
13c37f01b1 | ||
|
|
511831ced3 | ||
|
|
518e1afa58 | ||
|
|
43d03ac081 |
@@ -42,7 +42,9 @@ Description: $DESCRIPTION
|
|||||||
Depends: libc6 (>= 2.29), libstdc++6 (>= 9), libx11-6, libxcb1,
|
Depends: libc6 (>= 2.29), libstdc++6 (>= 9), libx11-6, libxcb1,
|
||||||
libxcb-randr0, libxcb-xtest0, libxcb-xinerama0, libxcb-shape0,
|
libxcb-randr0, libxcb-xtest0, libxcb-xinerama0, libxcb-shape0,
|
||||||
libxcb-xkb1, libxcb-xfixes0, libxv1, libxtst6, libasound2,
|
libxcb-xkb1, libxcb-xfixes0, libxv1, libxtst6, libasound2,
|
||||||
libsndio7.0, libxcb-shm0, libpulse0
|
libsndio7.0, libxcb-shm0, libpulse0, libdrm2, libdbus-1-3,
|
||||||
|
libpipewire-0.3-0, xdg-desktop-portal,
|
||||||
|
xdg-desktop-portal-gtk | xdg-desktop-portal-kde | xdg-desktop-portal-wlr
|
||||||
Recommends: nvidia-cuda-toolkit
|
Recommends: nvidia-cuda-toolkit
|
||||||
Priority: optional
|
Priority: optional
|
||||||
Section: utils
|
Section: utils
|
||||||
@@ -93,4 +95,4 @@ mv "$DEB_DIR.deb" "$OUTPUT_FILE"
|
|||||||
|
|
||||||
rm -rf "$DEB_DIR"
|
rm -rf "$DEB_DIR"
|
||||||
|
|
||||||
echo "✅ Deb package created: $OUTPUT_FILE"
|
echo "✅ Deb package created: $OUTPUT_FILE"
|
||||||
|
|||||||
@@ -42,7 +42,9 @@ Description: $DESCRIPTION
|
|||||||
Depends: libc6 (>= 2.29), libstdc++6 (>= 9), libx11-6, libxcb1,
|
Depends: libc6 (>= 2.29), libstdc++6 (>= 9), libx11-6, libxcb1,
|
||||||
libxcb-randr0, libxcb-xtest0, libxcb-xinerama0, libxcb-shape0,
|
libxcb-randr0, libxcb-xtest0, libxcb-xinerama0, libxcb-shape0,
|
||||||
libxcb-xkb1, libxcb-xfixes0, libxv1, libxtst6, libasound2,
|
libxcb-xkb1, libxcb-xfixes0, libxv1, libxtst6, libasound2,
|
||||||
libsndio7.0, libxcb-shm0, libpulse0
|
libsndio7.0, libxcb-shm0, libpulse0, libdrm2, libdbus-1-3,
|
||||||
|
libpipewire-0.3-0, xdg-desktop-portal,
|
||||||
|
xdg-desktop-portal-gtk | xdg-desktop-portal-kde | xdg-desktop-portal-wlr
|
||||||
Priority: optional
|
Priority: optional
|
||||||
Section: utils
|
Section: utils
|
||||||
EOF
|
EOF
|
||||||
@@ -92,4 +94,4 @@ mv "$DEB_DIR.deb" "$OUTPUT_FILE"
|
|||||||
|
|
||||||
rm -rf "$DEB_DIR"
|
rm -rf "$DEB_DIR"
|
||||||
|
|
||||||
echo "✅ Deb package created: $OUTPUT_FILE"
|
echo "✅ Deb package created: $OUTPUT_FILE"
|
||||||
|
|||||||
@@ -1,5 +1,8 @@
|
|||||||
#include "platform.h"
|
#include "platform.h"
|
||||||
|
|
||||||
|
#include <cstdlib>
|
||||||
|
#include <cstring>
|
||||||
|
|
||||||
#include "rd_log.h"
|
#include "rd_log.h"
|
||||||
|
|
||||||
#ifdef _WIN32
|
#ifdef _WIN32
|
||||||
@@ -125,4 +128,25 @@ std::string GetHostName() {
|
|||||||
#endif
|
#endif
|
||||||
return hostname;
|
return hostname;
|
||||||
}
|
}
|
||||||
} // namespace crossdesk
|
|
||||||
|
bool IsWaylandSession() {
|
||||||
|
#if defined(__linux__) && !defined(__APPLE__)
|
||||||
|
const char* session_type = std::getenv("XDG_SESSION_TYPE");
|
||||||
|
if (session_type) {
|
||||||
|
if (std::strcmp(session_type, "wayland") == 0 ||
|
||||||
|
std::strcmp(session_type, "Wayland") == 0) {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
if (std::strcmp(session_type, "x11") == 0 ||
|
||||||
|
std::strcmp(session_type, "X11") == 0) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const char* wayland_display = std::getenv("WAYLAND_DISPLAY");
|
||||||
|
return wayland_display && wayland_display[0] != '\0';
|
||||||
|
#else
|
||||||
|
return false;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
} // namespace crossdesk
|
||||||
|
|||||||
@@ -13,6 +13,7 @@ namespace crossdesk {
|
|||||||
|
|
||||||
std::string GetMac();
|
std::string GetMac();
|
||||||
std::string GetHostName();
|
std::string GetHostName();
|
||||||
|
bool IsWaylandSession();
|
||||||
|
|
||||||
} // namespace crossdesk
|
} // namespace crossdesk
|
||||||
#endif
|
#endif
|
||||||
|
|||||||
279
src/common/wayland_portal_shared.cpp
Normal file
279
src/common/wayland_portal_shared.cpp
Normal file
@@ -0,0 +1,279 @@
|
|||||||
|
#include "wayland_portal_shared.h"
|
||||||
|
|
||||||
|
#include <chrono>
|
||||||
|
#include <mutex>
|
||||||
|
|
||||||
|
#if defined(CROSSDESK_HAS_WAYLAND_CAPTURER) && CROSSDESK_HAS_WAYLAND_CAPTURER
|
||||||
|
#include <dbus/dbus.h>
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#include "rd_log.h"
|
||||||
|
|
||||||
|
namespace crossdesk {
|
||||||
|
|
||||||
|
namespace {
|
||||||
|
|
||||||
|
std::mutex& SharedSessionMutex() {
|
||||||
|
static std::mutex mutex;
|
||||||
|
return mutex;
|
||||||
|
}
|
||||||
|
|
||||||
|
SharedWaylandPortalSessionInfo& SharedSessionInfo() {
|
||||||
|
static SharedWaylandPortalSessionInfo info;
|
||||||
|
return info;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool& SharedSessionActive() {
|
||||||
|
static bool active = false;
|
||||||
|
return active;
|
||||||
|
}
|
||||||
|
|
||||||
|
int& SharedSessionRefs() {
|
||||||
|
static int refs = 0;
|
||||||
|
return refs;
|
||||||
|
}
|
||||||
|
|
||||||
|
#if defined(CROSSDESK_HAS_WAYLAND_CAPTURER) && CROSSDESK_HAS_WAYLAND_CAPTURER
|
||||||
|
constexpr const char* kPortalBusName = "org.freedesktop.portal.Desktop";
|
||||||
|
constexpr const char* kPortalSessionInterface =
|
||||||
|
"org.freedesktop.portal.Session";
|
||||||
|
constexpr int kPortalCloseWaitMs = 100;
|
||||||
|
|
||||||
|
void LogCloseDbusError(const char* action, DBusError* error) {
|
||||||
|
if (error && dbus_error_is_set(error)) {
|
||||||
|
LOG_ERROR("{} failed: {} ({})", action,
|
||||||
|
error->message ? error->message : "unknown",
|
||||||
|
error->name ? error->name : "unknown");
|
||||||
|
} else {
|
||||||
|
LOG_ERROR("{} failed", action);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
struct SessionClosedState {
|
||||||
|
std::string session_handle;
|
||||||
|
bool received = false;
|
||||||
|
};
|
||||||
|
|
||||||
|
DBusHandlerResult HandleSessionClosedSignal(DBusConnection* connection,
|
||||||
|
DBusMessage* message,
|
||||||
|
void* user_data) {
|
||||||
|
(void)connection;
|
||||||
|
auto* state = static_cast<SessionClosedState*>(user_data);
|
||||||
|
if (!state || !message) {
|
||||||
|
return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!dbus_message_is_signal(message, kPortalSessionInterface, "Closed")) {
|
||||||
|
return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
|
||||||
|
}
|
||||||
|
|
||||||
|
const char* path = dbus_message_get_path(message);
|
||||||
|
if (!path || state->session_handle != path) {
|
||||||
|
return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
|
||||||
|
}
|
||||||
|
|
||||||
|
state->received = true;
|
||||||
|
return DBUS_HANDLER_RESULT_HANDLED;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool BeginSessionClosedWatch(DBusConnection* connection,
|
||||||
|
const std::string& session_handle,
|
||||||
|
SessionClosedState* state,
|
||||||
|
std::string* match_rule_out) {
|
||||||
|
if (!connection || session_handle.empty() || !state || !match_rule_out) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
state->session_handle = session_handle;
|
||||||
|
state->received = false;
|
||||||
|
DBusError error;
|
||||||
|
dbus_error_init(&error);
|
||||||
|
const std::string match_rule =
|
||||||
|
"type='signal',interface='" + std::string(kPortalSessionInterface) +
|
||||||
|
"',member='Closed',path='" + session_handle + "'";
|
||||||
|
dbus_bus_add_match(connection, match_rule.c_str(), &error);
|
||||||
|
if (dbus_error_is_set(&error)) {
|
||||||
|
LogCloseDbusError("dbus_bus_add_match(Session.Closed)", &error);
|
||||||
|
dbus_error_free(&error);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
dbus_connection_add_filter(connection, HandleSessionClosedSignal, state,
|
||||||
|
nullptr);
|
||||||
|
*match_rule_out = match_rule;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
void EndSessionClosedWatch(DBusConnection* connection, SessionClosedState* state,
|
||||||
|
const std::string& match_rule) {
|
||||||
|
if (!connection || !state || match_rule.empty()) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
dbus_connection_remove_filter(connection, HandleSessionClosedSignal, state);
|
||||||
|
|
||||||
|
DBusError remove_error;
|
||||||
|
dbus_error_init(&remove_error);
|
||||||
|
dbus_bus_remove_match(connection, match_rule.c_str(), &remove_error);
|
||||||
|
if (dbus_error_is_set(&remove_error)) {
|
||||||
|
dbus_error_free(&remove_error);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void WaitForSessionClosed(DBusConnection* connection, SessionClosedState* state,
|
||||||
|
int timeout_ms = kPortalCloseWaitMs) {
|
||||||
|
if (!connection || !state) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const auto deadline =
|
||||||
|
std::chrono::steady_clock::now() + std::chrono::milliseconds(timeout_ms);
|
||||||
|
while (!state->received && std::chrono::steady_clock::now() < deadline) {
|
||||||
|
dbus_connection_read_write(connection, 100);
|
||||||
|
while (dbus_connection_dispatch(connection) == DBUS_DISPATCH_DATA_REMAINS) {
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
} // namespace
|
||||||
|
|
||||||
|
bool PublishSharedWaylandPortalSession(
|
||||||
|
const SharedWaylandPortalSessionInfo& info) {
|
||||||
|
if (!info.connection || info.session_handle.empty() || info.stream_id == 0) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::lock_guard<std::mutex> lock(SharedSessionMutex());
|
||||||
|
if (SharedSessionActive()) {
|
||||||
|
const auto& active_info = SharedSessionInfo();
|
||||||
|
if (active_info.session_handle != info.session_handle &&
|
||||||
|
SharedSessionRefs() > 0) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const bool same_session =
|
||||||
|
SharedSessionActive() &&
|
||||||
|
SharedSessionInfo().session_handle == info.session_handle;
|
||||||
|
SharedSessionInfo() = info;
|
||||||
|
SharedSessionActive() = true;
|
||||||
|
if (!same_session || SharedSessionRefs() <= 0) {
|
||||||
|
SharedSessionRefs() = 1;
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool AcquireSharedWaylandPortalSession(bool require_pointer,
|
||||||
|
SharedWaylandPortalSessionInfo* out) {
|
||||||
|
if (!out) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::lock_guard<std::mutex> lock(SharedSessionMutex());
|
||||||
|
if (!SharedSessionActive()) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
const auto& info = SharedSessionInfo();
|
||||||
|
if (require_pointer && !info.pointer_granted) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
++SharedSessionRefs();
|
||||||
|
*out = info;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ReleaseSharedWaylandPortalSession(DBusConnection** connection_out,
|
||||||
|
std::string* session_handle_out) {
|
||||||
|
if (connection_out) {
|
||||||
|
*connection_out = nullptr;
|
||||||
|
}
|
||||||
|
if (session_handle_out) {
|
||||||
|
session_handle_out->clear();
|
||||||
|
}
|
||||||
|
|
||||||
|
std::lock_guard<std::mutex> lock(SharedSessionMutex());
|
||||||
|
if (!SharedSessionActive()) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (SharedSessionRefs() > 0) {
|
||||||
|
--SharedSessionRefs();
|
||||||
|
}
|
||||||
|
|
||||||
|
if (SharedSessionRefs() > 0) {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (connection_out) {
|
||||||
|
*connection_out = SharedSessionInfo().connection;
|
||||||
|
}
|
||||||
|
if (session_handle_out) {
|
||||||
|
*session_handle_out = SharedSessionInfo().session_handle;
|
||||||
|
}
|
||||||
|
|
||||||
|
SharedSessionInfo() = SharedWaylandPortalSessionInfo{};
|
||||||
|
SharedSessionActive() = false;
|
||||||
|
SharedSessionRefs() = 0;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
void CloseWaylandPortalSessionAndConnection(DBusConnection* connection,
|
||||||
|
const std::string& session_handle,
|
||||||
|
const char* close_action) {
|
||||||
|
#if defined(CROSSDESK_HAS_WAYLAND_CAPTURER) && CROSSDESK_HAS_WAYLAND_CAPTURER
|
||||||
|
if (!connection) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!session_handle.empty()) {
|
||||||
|
SessionClosedState close_state;
|
||||||
|
std::string close_match_rule;
|
||||||
|
const bool watching_closed = BeginSessionClosedWatch(
|
||||||
|
connection, session_handle, &close_state, &close_match_rule);
|
||||||
|
|
||||||
|
DBusMessage* message = dbus_message_new_method_call(
|
||||||
|
kPortalBusName, session_handle.c_str(), kPortalSessionInterface,
|
||||||
|
"Close");
|
||||||
|
if (message) {
|
||||||
|
DBusError error;
|
||||||
|
dbus_error_init(&error);
|
||||||
|
DBusMessage* reply = dbus_connection_send_with_reply_and_block(
|
||||||
|
connection, message, 1000, &error);
|
||||||
|
if (!reply && dbus_error_is_set(&error)) {
|
||||||
|
LogCloseDbusError(close_action, &error);
|
||||||
|
dbus_error_free(&error);
|
||||||
|
}
|
||||||
|
if (reply) {
|
||||||
|
dbus_message_unref(reply);
|
||||||
|
}
|
||||||
|
dbus_message_unref(message);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (watching_closed) {
|
||||||
|
WaitForSessionClosed(connection, &close_state);
|
||||||
|
if (!close_state.received) {
|
||||||
|
LOG_WARN("Timed out waiting for portal session to close: {}",
|
||||||
|
session_handle);
|
||||||
|
LOG_WARN("Forcing local teardown without waiting for Session.Closed: {}",
|
||||||
|
session_handle);
|
||||||
|
EndSessionClosedWatch(connection, &close_state, close_match_rule);
|
||||||
|
} else {
|
||||||
|
EndSessionClosedWatch(connection, &close_state, close_match_rule);
|
||||||
|
LOG_INFO("Portal session closed: {}", session_handle);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
dbus_connection_close(connection);
|
||||||
|
dbus_connection_unref(connection);
|
||||||
|
#else
|
||||||
|
(void)connection;
|
||||||
|
(void)session_handle;
|
||||||
|
(void)close_action;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace crossdesk
|
||||||
37
src/common/wayland_portal_shared.h
Normal file
37
src/common/wayland_portal_shared.h
Normal file
@@ -0,0 +1,37 @@
|
|||||||
|
/*
|
||||||
|
* Shared Wayland portal session state used by the Linux Wayland capturer and
|
||||||
|
* mouse controller so they can reuse one RemoteDesktop session.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#ifndef _WAYLAND_PORTAL_SHARED_H_
|
||||||
|
#define _WAYLAND_PORTAL_SHARED_H_
|
||||||
|
|
||||||
|
#include <cstdint>
|
||||||
|
#include <string>
|
||||||
|
|
||||||
|
struct DBusConnection;
|
||||||
|
|
||||||
|
namespace crossdesk {
|
||||||
|
|
||||||
|
struct SharedWaylandPortalSessionInfo {
|
||||||
|
DBusConnection* connection = nullptr;
|
||||||
|
std::string session_handle;
|
||||||
|
uint32_t stream_id = 0;
|
||||||
|
int width = 0;
|
||||||
|
int height = 0;
|
||||||
|
bool pointer_granted = false;
|
||||||
|
};
|
||||||
|
|
||||||
|
bool PublishSharedWaylandPortalSession(
|
||||||
|
const SharedWaylandPortalSessionInfo& info);
|
||||||
|
bool AcquireSharedWaylandPortalSession(bool require_pointer,
|
||||||
|
SharedWaylandPortalSessionInfo* out);
|
||||||
|
bool ReleaseSharedWaylandPortalSession(DBusConnection** connection_out,
|
||||||
|
std::string* session_handle_out);
|
||||||
|
void CloseWaylandPortalSessionAndConnection(DBusConnection* connection,
|
||||||
|
const std::string& session_handle,
|
||||||
|
const char* close_action);
|
||||||
|
|
||||||
|
} // namespace crossdesk
|
||||||
|
|
||||||
|
#endif
|
||||||
@@ -4,6 +4,7 @@
|
|||||||
#include <poll.h>
|
#include <poll.h>
|
||||||
|
|
||||||
#include "keyboard_converter.h"
|
#include "keyboard_converter.h"
|
||||||
|
#include "platform.h"
|
||||||
#include "rd_log.h"
|
#include "rd_log.h"
|
||||||
|
|
||||||
namespace crossdesk {
|
namespace crossdesk {
|
||||||
@@ -11,10 +12,28 @@ namespace crossdesk {
|
|||||||
static OnKeyAction g_on_key_action = nullptr;
|
static OnKeyAction g_on_key_action = nullptr;
|
||||||
static void* g_user_ptr = nullptr;
|
static void* g_user_ptr = nullptr;
|
||||||
|
|
||||||
|
static KeySym NormalizeKeySym(KeySym key_sym) {
|
||||||
|
if (key_sym >= XK_a && key_sym <= XK_z) {
|
||||||
|
return key_sym - XK_a + XK_A;
|
||||||
|
}
|
||||||
|
return key_sym;
|
||||||
|
}
|
||||||
|
|
||||||
static int KeyboardEventHandler(Display* display, XEvent* event) {
|
static int KeyboardEventHandler(Display* display, XEvent* event) {
|
||||||
|
(void)display;
|
||||||
if (event->xkey.type == KeyPress || event->xkey.type == KeyRelease) {
|
if (event->xkey.type == KeyPress || event->xkey.type == KeyRelease) {
|
||||||
KeySym keySym = XLookupKeysym(&event->xkey, 0);
|
KeySym key_sym = NormalizeKeySym(XLookupKeysym(&event->xkey, 0));
|
||||||
int key_code = XKeysymToKeycode(display, keySym);
|
auto key_it = x11KeySymToVkCode.find(static_cast<int>(key_sym));
|
||||||
|
if (key_it == x11KeySymToVkCode.end()) {
|
||||||
|
key_sym = NormalizeKeySym(XLookupKeysym(&event->xkey, 1));
|
||||||
|
key_it = x11KeySymToVkCode.find(static_cast<int>(key_sym));
|
||||||
|
}
|
||||||
|
|
||||||
|
if (key_it == x11KeySymToVkCode.end()) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int key_code = key_it->second;
|
||||||
bool is_key_down = (event->xkey.type == KeyPress);
|
bool is_key_down = (event->xkey.type == KeyPress);
|
||||||
|
|
||||||
if (g_on_key_action) {
|
if (g_on_key_action) {
|
||||||
@@ -25,7 +44,12 @@ static int KeyboardEventHandler(Display* display, XEvent* event) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
KeyboardCapturer::KeyboardCapturer()
|
KeyboardCapturer::KeyboardCapturer()
|
||||||
: display_(nullptr), root_(0), running_(false) {
|
: display_(nullptr),
|
||||||
|
root_(0),
|
||||||
|
running_(false),
|
||||||
|
use_wayland_portal_(false),
|
||||||
|
wayland_init_attempted_(false),
|
||||||
|
dbus_connection_(nullptr) {
|
||||||
XInitThreads();
|
XInitThreads();
|
||||||
display_ = XOpenDisplay(nullptr);
|
display_ = XOpenDisplay(nullptr);
|
||||||
if (!display_) {
|
if (!display_) {
|
||||||
@@ -35,6 +59,7 @@ KeyboardCapturer::KeyboardCapturer()
|
|||||||
|
|
||||||
KeyboardCapturer::~KeyboardCapturer() {
|
KeyboardCapturer::~KeyboardCapturer() {
|
||||||
Unhook();
|
Unhook();
|
||||||
|
CleanupWaylandPortal();
|
||||||
|
|
||||||
if (display_) {
|
if (display_) {
|
||||||
XCloseDisplay(display_);
|
XCloseDisplay(display_);
|
||||||
@@ -122,6 +147,22 @@ int KeyboardCapturer::Unhook() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
int KeyboardCapturer::SendKeyboardCommand(int key_code, bool is_down) {
|
int KeyboardCapturer::SendKeyboardCommand(int key_code, bool is_down) {
|
||||||
|
if (IsWaylandSession()) {
|
||||||
|
if (!use_wayland_portal_ && !wayland_init_attempted_) {
|
||||||
|
wayland_init_attempted_ = true;
|
||||||
|
if (InitWaylandPortal()) {
|
||||||
|
use_wayland_portal_ = true;
|
||||||
|
LOG_INFO("Keyboard controller initialized with Wayland portal backend");
|
||||||
|
} else {
|
||||||
|
LOG_WARN("Wayland keyboard control init failed, falling back to X11/XTest backend");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (use_wayland_portal_) {
|
||||||
|
return SendWaylandKeyboardCommand(key_code, is_down);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
if (!display_) {
|
if (!display_) {
|
||||||
LOG_ERROR("Display not initialized.");
|
LOG_ERROR("Display not initialized.");
|
||||||
return -1;
|
return -1;
|
||||||
@@ -135,4 +176,4 @@ int KeyboardCapturer::SendKeyboardCommand(int key_code, bool is_down) {
|
|||||||
}
|
}
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
} // namespace crossdesk
|
} // namespace crossdesk
|
||||||
|
|||||||
@@ -12,10 +12,16 @@
|
|||||||
#include <X11/keysym.h>
|
#include <X11/keysym.h>
|
||||||
|
|
||||||
#include <atomic>
|
#include <atomic>
|
||||||
|
#include <cstdint>
|
||||||
|
#include <functional>
|
||||||
|
#include <string>
|
||||||
#include <thread>
|
#include <thread>
|
||||||
|
|
||||||
#include "device_controller.h"
|
#include "device_controller.h"
|
||||||
|
|
||||||
|
struct DBusConnection;
|
||||||
|
struct DBusMessageIter;
|
||||||
|
|
||||||
namespace crossdesk {
|
namespace crossdesk {
|
||||||
|
|
||||||
class KeyboardCapturer : public DeviceController {
|
class KeyboardCapturer : public DeviceController {
|
||||||
@@ -28,11 +34,25 @@ class KeyboardCapturer : public DeviceController {
|
|||||||
virtual int Unhook();
|
virtual int Unhook();
|
||||||
virtual int SendKeyboardCommand(int key_code, bool is_down);
|
virtual int SendKeyboardCommand(int key_code, bool is_down);
|
||||||
|
|
||||||
|
private:
|
||||||
|
bool InitWaylandPortal();
|
||||||
|
void CleanupWaylandPortal();
|
||||||
|
int SendWaylandKeyboardCommand(int key_code, bool is_down);
|
||||||
|
bool NotifyWaylandKeyboardKeysym(int keysym, uint32_t state);
|
||||||
|
bool NotifyWaylandKeyboardKeycode(int keycode, uint32_t state);
|
||||||
|
bool SendWaylandPortalVoidCall(const char* method_name,
|
||||||
|
const std::function<void(DBusMessageIter*)>&
|
||||||
|
append_args);
|
||||||
|
|
||||||
private:
|
private:
|
||||||
Display* display_;
|
Display* display_;
|
||||||
Window root_;
|
Window root_;
|
||||||
std::atomic<bool> running_;
|
std::atomic<bool> running_;
|
||||||
std::thread event_thread_;
|
std::thread event_thread_;
|
||||||
|
bool use_wayland_portal_ = false;
|
||||||
|
bool wayland_init_attempted_ = false;
|
||||||
|
DBusConnection* dbus_connection_ = nullptr;
|
||||||
|
std::string wayland_session_handle_;
|
||||||
};
|
};
|
||||||
} // namespace crossdesk
|
} // namespace crossdesk
|
||||||
#endif
|
#endif
|
||||||
|
|||||||
@@ -0,0 +1,711 @@
|
|||||||
|
#include "keyboard_capturer.h"
|
||||||
|
|
||||||
|
#include <chrono>
|
||||||
|
#include <cstring>
|
||||||
|
#include <map>
|
||||||
|
|
||||||
|
#if defined(CROSSDESK_HAS_WAYLAND_CAPTURER) && CROSSDESK_HAS_WAYLAND_CAPTURER
|
||||||
|
#include <dbus/dbus.h>
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#include "rd_log.h"
|
||||||
|
#include "wayland_portal_shared.h"
|
||||||
|
|
||||||
|
namespace crossdesk {
|
||||||
|
|
||||||
|
extern std::map<int, int> vkCodeToX11KeySym;
|
||||||
|
|
||||||
|
#if defined(CROSSDESK_HAS_WAYLAND_CAPTURER) && CROSSDESK_HAS_WAYLAND_CAPTURER
|
||||||
|
namespace {
|
||||||
|
|
||||||
|
constexpr const char* kPortalBusName = "org.freedesktop.portal.Desktop";
|
||||||
|
constexpr const char* kPortalObjectPath = "/org/freedesktop/portal/desktop";
|
||||||
|
constexpr const char* kPortalRemoteDesktopInterface =
|
||||||
|
"org.freedesktop.portal.RemoteDesktop";
|
||||||
|
constexpr const char* kPortalRequestInterface =
|
||||||
|
"org.freedesktop.portal.Request";
|
||||||
|
constexpr const char* kPortalRequestPathPrefix =
|
||||||
|
"/org/freedesktop/portal/desktop/request/";
|
||||||
|
constexpr const char* kPortalSessionPathPrefix =
|
||||||
|
"/org/freedesktop/portal/desktop/session/";
|
||||||
|
|
||||||
|
constexpr uint32_t kRemoteDesktopDeviceKeyboard = 1u;
|
||||||
|
constexpr uint32_t kKeyboardReleased = 0u;
|
||||||
|
constexpr uint32_t kKeyboardPressed = 1u;
|
||||||
|
|
||||||
|
int NormalizeFallbackKeysym(int keysym) {
|
||||||
|
if (keysym >= XK_A && keysym <= XK_Z) {
|
||||||
|
return keysym - XK_A + XK_a;
|
||||||
|
}
|
||||||
|
return keysym;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string MakeToken(const char* prefix) {
|
||||||
|
const auto now = std::chrono::steady_clock::now().time_since_epoch().count();
|
||||||
|
return std::string(prefix) + "_" + std::to_string(now);
|
||||||
|
}
|
||||||
|
|
||||||
|
void LogDbusError(const char* action, DBusError* error) {
|
||||||
|
if (error && dbus_error_is_set(error)) {
|
||||||
|
LOG_ERROR("{} failed: {} ({})", action,
|
||||||
|
error->message ? error->message : "unknown",
|
||||||
|
error->name ? error->name : "unknown");
|
||||||
|
} else {
|
||||||
|
LOG_ERROR("{} failed", action);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void AppendDictEntryString(DBusMessageIter* dict, const char* key,
|
||||||
|
const std::string& value) {
|
||||||
|
DBusMessageIter entry;
|
||||||
|
DBusMessageIter variant;
|
||||||
|
const char* key_cstr = key;
|
||||||
|
const char* value_cstr = value.c_str();
|
||||||
|
|
||||||
|
dbus_message_iter_open_container(dict, DBUS_TYPE_DICT_ENTRY, nullptr, &entry);
|
||||||
|
dbus_message_iter_append_basic(&entry, DBUS_TYPE_STRING, &key_cstr);
|
||||||
|
dbus_message_iter_open_container(&entry, DBUS_TYPE_VARIANT, "s", &variant);
|
||||||
|
dbus_message_iter_append_basic(&variant, DBUS_TYPE_STRING, &value_cstr);
|
||||||
|
dbus_message_iter_close_container(&entry, &variant);
|
||||||
|
dbus_message_iter_close_container(dict, &entry);
|
||||||
|
}
|
||||||
|
|
||||||
|
void AppendDictEntryUint32(DBusMessageIter* dict, const char* key,
|
||||||
|
uint32_t value) {
|
||||||
|
DBusMessageIter entry;
|
||||||
|
DBusMessageIter variant;
|
||||||
|
const char* key_cstr = key;
|
||||||
|
|
||||||
|
dbus_message_iter_open_container(dict, DBUS_TYPE_DICT_ENTRY, nullptr, &entry);
|
||||||
|
dbus_message_iter_append_basic(&entry, DBUS_TYPE_STRING, &key_cstr);
|
||||||
|
dbus_message_iter_open_container(&entry, DBUS_TYPE_VARIANT, "u", &variant);
|
||||||
|
dbus_message_iter_append_basic(&variant, DBUS_TYPE_UINT32, &value);
|
||||||
|
dbus_message_iter_close_container(&entry, &variant);
|
||||||
|
dbus_message_iter_close_container(dict, &entry);
|
||||||
|
}
|
||||||
|
|
||||||
|
void AppendEmptyOptionsDict(DBusMessageIter* iter) {
|
||||||
|
DBusMessageIter options;
|
||||||
|
dbus_message_iter_open_container(iter, DBUS_TYPE_ARRAY, "{sv}", &options);
|
||||||
|
dbus_message_iter_close_container(iter, &options);
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ReadPathLikeVariant(DBusMessageIter* variant, std::string* value) {
|
||||||
|
if (!variant || !value) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
const int type = dbus_message_iter_get_arg_type(variant);
|
||||||
|
if (type == DBUS_TYPE_OBJECT_PATH || type == DBUS_TYPE_STRING) {
|
||||||
|
const char* temp = nullptr;
|
||||||
|
dbus_message_iter_get_basic(variant, &temp);
|
||||||
|
if (temp && temp[0] != '\0') {
|
||||||
|
*value = temp;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ReadUint32Like(DBusMessageIter* iter, uint32_t* value) {
|
||||||
|
if (!iter || !value) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
const int type = dbus_message_iter_get_arg_type(iter);
|
||||||
|
if (type == DBUS_TYPE_UINT32) {
|
||||||
|
uint32_t temp = 0;
|
||||||
|
dbus_message_iter_get_basic(iter, &temp);
|
||||||
|
*value = temp;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (type == DBUS_TYPE_INT32) {
|
||||||
|
int32_t temp = 0;
|
||||||
|
dbus_message_iter_get_basic(iter, &temp);
|
||||||
|
if (temp < 0) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
*value = static_cast<uint32_t>(temp);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string BuildSessionHandleFromRequestPath(
|
||||||
|
const std::string& request_path, const std::string& session_handle_token) {
|
||||||
|
if (request_path.rfind(kPortalRequestPathPrefix, 0) != 0 ||
|
||||||
|
session_handle_token.empty()) {
|
||||||
|
return "";
|
||||||
|
}
|
||||||
|
|
||||||
|
const size_t sender_start = strlen(kPortalRequestPathPrefix);
|
||||||
|
const size_t token_sep = request_path.find('/', sender_start);
|
||||||
|
if (token_sep == std::string::npos || token_sep <= sender_start) {
|
||||||
|
return "";
|
||||||
|
}
|
||||||
|
|
||||||
|
const std::string sender =
|
||||||
|
request_path.substr(sender_start, token_sep - sender_start);
|
||||||
|
if (sender.empty()) {
|
||||||
|
return "";
|
||||||
|
}
|
||||||
|
|
||||||
|
return std::string(kPortalSessionPathPrefix) + sender + "/" +
|
||||||
|
session_handle_token;
|
||||||
|
}
|
||||||
|
|
||||||
|
struct PortalResponseState {
|
||||||
|
std::string request_path;
|
||||||
|
bool received = false;
|
||||||
|
DBusMessage* message = nullptr;
|
||||||
|
};
|
||||||
|
|
||||||
|
DBusHandlerResult HandlePortalResponseSignal(DBusConnection* connection,
|
||||||
|
DBusMessage* message,
|
||||||
|
void* user_data) {
|
||||||
|
(void)connection;
|
||||||
|
auto* state = static_cast<PortalResponseState*>(user_data);
|
||||||
|
if (!state || !message) {
|
||||||
|
return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!dbus_message_is_signal(message, kPortalRequestInterface, "Response")) {
|
||||||
|
return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
|
||||||
|
}
|
||||||
|
|
||||||
|
const char* path = dbus_message_get_path(message);
|
||||||
|
if (!path || state->request_path != path) {
|
||||||
|
return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (state->message) {
|
||||||
|
dbus_message_unref(state->message);
|
||||||
|
state->message = nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
state->message = dbus_message_ref(message);
|
||||||
|
state->received = true;
|
||||||
|
return DBUS_HANDLER_RESULT_HANDLED;
|
||||||
|
}
|
||||||
|
|
||||||
|
DBusMessage* WaitForPortalResponse(DBusConnection* connection,
|
||||||
|
const std::string& request_path,
|
||||||
|
int timeout_ms = 120000) {
|
||||||
|
if (!connection || request_path.empty()) {
|
||||||
|
return nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
PortalResponseState state;
|
||||||
|
state.request_path = request_path;
|
||||||
|
|
||||||
|
DBusError error;
|
||||||
|
dbus_error_init(&error);
|
||||||
|
|
||||||
|
const std::string match_rule =
|
||||||
|
"type='signal',interface='" + std::string(kPortalRequestInterface) +
|
||||||
|
"',member='Response',path='" + request_path + "'";
|
||||||
|
dbus_bus_add_match(connection, match_rule.c_str(), &error);
|
||||||
|
if (dbus_error_is_set(&error)) {
|
||||||
|
LogDbusError("dbus_bus_add_match", &error);
|
||||||
|
dbus_error_free(&error);
|
||||||
|
return nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
dbus_connection_add_filter(connection, HandlePortalResponseSignal, &state,
|
||||||
|
nullptr);
|
||||||
|
|
||||||
|
auto deadline =
|
||||||
|
std::chrono::steady_clock::now() + std::chrono::milliseconds(timeout_ms);
|
||||||
|
while (!state.received && std::chrono::steady_clock::now() < deadline) {
|
||||||
|
dbus_connection_read_write(connection, 100);
|
||||||
|
while (dbus_connection_dispatch(connection) == DBUS_DISPATCH_DATA_REMAINS) {
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
dbus_connection_remove_filter(connection, HandlePortalResponseSignal, &state);
|
||||||
|
|
||||||
|
DBusError remove_error;
|
||||||
|
dbus_error_init(&remove_error);
|
||||||
|
dbus_bus_remove_match(connection, match_rule.c_str(), &remove_error);
|
||||||
|
if (dbus_error_is_set(&remove_error)) {
|
||||||
|
dbus_error_free(&remove_error);
|
||||||
|
}
|
||||||
|
|
||||||
|
return state.message;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ExtractRequestPath(DBusMessage* reply, std::string* request_path) {
|
||||||
|
if (!reply || !request_path) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
const char* path = nullptr;
|
||||||
|
DBusError error;
|
||||||
|
dbus_error_init(&error);
|
||||||
|
const dbus_bool_t ok = dbus_message_get_args(
|
||||||
|
reply, &error, DBUS_TYPE_OBJECT_PATH, &path, DBUS_TYPE_INVALID);
|
||||||
|
if (!ok || !path) {
|
||||||
|
LogDbusError("dbus_message_get_args(request_path)", &error);
|
||||||
|
dbus_error_free(&error);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
*request_path = path;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ExtractPortalResponse(DBusMessage* message, uint32_t* response_code,
|
||||||
|
DBusMessageIter* results_array) {
|
||||||
|
if (!message || !response_code || !results_array) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
DBusMessageIter iter;
|
||||||
|
if (!dbus_message_iter_init(message, &iter) ||
|
||||||
|
dbus_message_iter_get_arg_type(&iter) != DBUS_TYPE_UINT32) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
dbus_message_iter_get_basic(&iter, response_code);
|
||||||
|
if (!dbus_message_iter_next(&iter) ||
|
||||||
|
dbus_message_iter_get_arg_type(&iter) != DBUS_TYPE_ARRAY) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
*results_array = iter;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool SendPortalRequestAndHandleResponse(
|
||||||
|
DBusConnection* connection, const char* interface_name,
|
||||||
|
const char* method_name, const char* action_name,
|
||||||
|
const std::function<bool(DBusMessage*)>& append_message_args,
|
||||||
|
const std::function<bool(uint32_t, DBusMessageIter*)>& handle_results,
|
||||||
|
std::string* request_path_out = nullptr) {
|
||||||
|
if (!connection || !interface_name || interface_name[0] == '\0' ||
|
||||||
|
!method_name || method_name[0] == '\0') {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
DBusMessage* message = dbus_message_new_method_call(
|
||||||
|
kPortalBusName, kPortalObjectPath, interface_name, method_name);
|
||||||
|
if (!message) {
|
||||||
|
LOG_ERROR("Failed to allocate {} message", method_name);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (append_message_args && !append_message_args(message)) {
|
||||||
|
dbus_message_unref(message);
|
||||||
|
LOG_ERROR("{} arguments are malformed", method_name);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
DBusError error;
|
||||||
|
dbus_error_init(&error);
|
||||||
|
DBusMessage* reply =
|
||||||
|
dbus_connection_send_with_reply_and_block(connection, message, -1, &error);
|
||||||
|
dbus_message_unref(message);
|
||||||
|
if (!reply) {
|
||||||
|
LogDbusError(action_name ? action_name : method_name, &error);
|
||||||
|
dbus_error_free(&error);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string request_path;
|
||||||
|
const bool got_request_path = ExtractRequestPath(reply, &request_path);
|
||||||
|
dbus_message_unref(reply);
|
||||||
|
if (!got_request_path) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
if (request_path_out) {
|
||||||
|
*request_path_out = request_path;
|
||||||
|
}
|
||||||
|
|
||||||
|
DBusMessage* response = WaitForPortalResponse(connection, request_path);
|
||||||
|
if (!response) {
|
||||||
|
LOG_ERROR("Timed out waiting for {} response", method_name);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
uint32_t response_code = 1;
|
||||||
|
DBusMessageIter results;
|
||||||
|
const bool parsed = ExtractPortalResponse(response, &response_code, &results);
|
||||||
|
if (!parsed) {
|
||||||
|
dbus_message_unref(response);
|
||||||
|
LOG_ERROR("{} response was malformed", method_name);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
const bool ok = handle_results ? handle_results(response_code, &results)
|
||||||
|
: (response_code == 0);
|
||||||
|
dbus_message_unref(response);
|
||||||
|
return ok;
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace
|
||||||
|
#endif
|
||||||
|
|
||||||
|
bool KeyboardCapturer::InitWaylandPortal() {
|
||||||
|
#if defined(CROSSDESK_HAS_WAYLAND_CAPTURER) && CROSSDESK_HAS_WAYLAND_CAPTURER
|
||||||
|
CleanupWaylandPortal();
|
||||||
|
|
||||||
|
DBusError error;
|
||||||
|
dbus_error_init(&error);
|
||||||
|
|
||||||
|
DBusConnection* check_connection = dbus_bus_get(DBUS_BUS_SESSION, &error);
|
||||||
|
if (!check_connection) {
|
||||||
|
LogDbusError("dbus_bus_get", &error);
|
||||||
|
dbus_error_free(&error);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
const dbus_bool_t has_owner =
|
||||||
|
dbus_bus_name_has_owner(check_connection, kPortalBusName, &error);
|
||||||
|
if (dbus_error_is_set(&error)) {
|
||||||
|
LogDbusError("dbus_bus_name_has_owner", &error);
|
||||||
|
dbus_error_free(&error);
|
||||||
|
dbus_connection_unref(check_connection);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
dbus_connection_unref(check_connection);
|
||||||
|
|
||||||
|
if (!has_owner) {
|
||||||
|
LOG_ERROR("xdg-desktop-portal is not available on session bus");
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
dbus_connection_ = dbus_bus_get_private(DBUS_BUS_SESSION, &error);
|
||||||
|
if (!dbus_connection_) {
|
||||||
|
LogDbusError("dbus_bus_get_private", &error);
|
||||||
|
dbus_error_free(&error);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
dbus_connection_set_exit_on_disconnect(dbus_connection_, FALSE);
|
||||||
|
|
||||||
|
const std::string session_handle_token =
|
||||||
|
MakeToken("crossdesk_keyboard_session");
|
||||||
|
std::string request_path;
|
||||||
|
const bool create_ok = SendPortalRequestAndHandleResponse(
|
||||||
|
dbus_connection_, kPortalRemoteDesktopInterface, "CreateSession",
|
||||||
|
"CreateSession",
|
||||||
|
[&](DBusMessage* message) {
|
||||||
|
DBusMessageIter iter;
|
||||||
|
DBusMessageIter options;
|
||||||
|
dbus_message_iter_init_append(message, &iter);
|
||||||
|
dbus_message_iter_open_container(&iter, DBUS_TYPE_ARRAY, "{sv}",
|
||||||
|
&options);
|
||||||
|
AppendDictEntryString(&options, "session_handle_token",
|
||||||
|
session_handle_token);
|
||||||
|
AppendDictEntryString(&options, "handle_token",
|
||||||
|
MakeToken("crossdesk_keyboard_req"));
|
||||||
|
dbus_message_iter_close_container(&iter, &options);
|
||||||
|
return true;
|
||||||
|
},
|
||||||
|
[&](uint32_t response_code, DBusMessageIter* results) {
|
||||||
|
if (response_code != 0) {
|
||||||
|
LOG_ERROR("RemoteDesktop.CreateSession denied, response={}",
|
||||||
|
response_code);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
DBusMessageIter dict;
|
||||||
|
dbus_message_iter_recurse(results, &dict);
|
||||||
|
while (dbus_message_iter_get_arg_type(&dict) != DBUS_TYPE_INVALID) {
|
||||||
|
if (dbus_message_iter_get_arg_type(&dict) == DBUS_TYPE_DICT_ENTRY) {
|
||||||
|
DBusMessageIter entry;
|
||||||
|
dbus_message_iter_recurse(&dict, &entry);
|
||||||
|
|
||||||
|
const char* key = nullptr;
|
||||||
|
dbus_message_iter_get_basic(&entry, &key);
|
||||||
|
if (key && dbus_message_iter_next(&entry) &&
|
||||||
|
dbus_message_iter_get_arg_type(&entry) == DBUS_TYPE_VARIANT &&
|
||||||
|
strcmp(key, "session_handle") == 0) {
|
||||||
|
DBusMessageIter variant;
|
||||||
|
std::string parsed_handle;
|
||||||
|
dbus_message_iter_recurse(&entry, &variant);
|
||||||
|
if (ReadPathLikeVariant(&variant, &parsed_handle) &&
|
||||||
|
!parsed_handle.empty()) {
|
||||||
|
wayland_session_handle_ = parsed_handle;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
dbus_message_iter_next(&dict);
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
},
|
||||||
|
&request_path);
|
||||||
|
|
||||||
|
if (!create_ok) {
|
||||||
|
CleanupWaylandPortal();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (wayland_session_handle_.empty()) {
|
||||||
|
wayland_session_handle_ =
|
||||||
|
BuildSessionHandleFromRequestPath(request_path, session_handle_token);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (wayland_session_handle_.empty()) {
|
||||||
|
LOG_ERROR("RemoteDesktop.CreateSession did not return session handle");
|
||||||
|
CleanupWaylandPortal();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
const char* session_handle = wayland_session_handle_.c_str();
|
||||||
|
const bool select_ok = SendPortalRequestAndHandleResponse(
|
||||||
|
dbus_connection_, kPortalRemoteDesktopInterface, "SelectDevices",
|
||||||
|
"SelectDevices",
|
||||||
|
[&](DBusMessage* message) {
|
||||||
|
DBusMessageIter iter;
|
||||||
|
DBusMessageIter options;
|
||||||
|
dbus_message_iter_init_append(message, &iter);
|
||||||
|
dbus_message_iter_append_basic(&iter, DBUS_TYPE_OBJECT_PATH,
|
||||||
|
&session_handle);
|
||||||
|
dbus_message_iter_open_container(&iter, DBUS_TYPE_ARRAY, "{sv}",
|
||||||
|
&options);
|
||||||
|
AppendDictEntryUint32(&options, "types", kRemoteDesktopDeviceKeyboard);
|
||||||
|
AppendDictEntryString(&options, "handle_token",
|
||||||
|
MakeToken("crossdesk_keyboard_req"));
|
||||||
|
dbus_message_iter_close_container(&iter, &options);
|
||||||
|
return true;
|
||||||
|
},
|
||||||
|
[](uint32_t response_code, DBusMessageIter*) {
|
||||||
|
if (response_code != 0) {
|
||||||
|
LOG_ERROR("RemoteDesktop.SelectDevices denied, response={}",
|
||||||
|
response_code);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
});
|
||||||
|
|
||||||
|
if (!select_ok) {
|
||||||
|
CleanupWaylandPortal();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
const char* parent_window = "";
|
||||||
|
bool keyboard_granted = false;
|
||||||
|
const bool start_ok = SendPortalRequestAndHandleResponse(
|
||||||
|
dbus_connection_, kPortalRemoteDesktopInterface, "Start", "Start",
|
||||||
|
[&](DBusMessage* message) {
|
||||||
|
DBusMessageIter iter;
|
||||||
|
DBusMessageIter options;
|
||||||
|
dbus_message_iter_init_append(message, &iter);
|
||||||
|
dbus_message_iter_append_basic(&iter, DBUS_TYPE_OBJECT_PATH,
|
||||||
|
&session_handle);
|
||||||
|
dbus_message_iter_append_basic(&iter, DBUS_TYPE_STRING, &parent_window);
|
||||||
|
dbus_message_iter_open_container(&iter, DBUS_TYPE_ARRAY, "{sv}",
|
||||||
|
&options);
|
||||||
|
AppendDictEntryString(&options, "handle_token",
|
||||||
|
MakeToken("crossdesk_keyboard_req"));
|
||||||
|
dbus_message_iter_close_container(&iter, &options);
|
||||||
|
return true;
|
||||||
|
},
|
||||||
|
[&](uint32_t response_code, DBusMessageIter* results) {
|
||||||
|
if (response_code != 0) {
|
||||||
|
LOG_ERROR("RemoteDesktop.Start denied, response={}", response_code);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
uint32_t granted_devices = 0;
|
||||||
|
DBusMessageIter dict;
|
||||||
|
dbus_message_iter_recurse(results, &dict);
|
||||||
|
while (dbus_message_iter_get_arg_type(&dict) != DBUS_TYPE_INVALID) {
|
||||||
|
if (dbus_message_iter_get_arg_type(&dict) == DBUS_TYPE_DICT_ENTRY) {
|
||||||
|
DBusMessageIter entry;
|
||||||
|
dbus_message_iter_recurse(&dict, &entry);
|
||||||
|
|
||||||
|
const char* key = nullptr;
|
||||||
|
dbus_message_iter_get_basic(&entry, &key);
|
||||||
|
if (key && dbus_message_iter_next(&entry) &&
|
||||||
|
dbus_message_iter_get_arg_type(&entry) == DBUS_TYPE_VARIANT) {
|
||||||
|
DBusMessageIter variant;
|
||||||
|
dbus_message_iter_recurse(&entry, &variant);
|
||||||
|
if (strcmp(key, "devices") == 0) {
|
||||||
|
ReadUint32Like(&variant, &granted_devices);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
dbus_message_iter_next(&dict);
|
||||||
|
}
|
||||||
|
|
||||||
|
keyboard_granted =
|
||||||
|
(granted_devices & kRemoteDesktopDeviceKeyboard) != 0;
|
||||||
|
if (!keyboard_granted) {
|
||||||
|
LOG_ERROR(
|
||||||
|
"RemoteDesktop.Start granted devices mask={}, keyboard not allowed",
|
||||||
|
granted_devices);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
});
|
||||||
|
|
||||||
|
if (!start_ok) {
|
||||||
|
CleanupWaylandPortal();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!keyboard_granted) {
|
||||||
|
LOG_ERROR("RemoteDesktop session started without keyboard permission");
|
||||||
|
CleanupWaylandPortal();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
return true;
|
||||||
|
#else
|
||||||
|
return false;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
void KeyboardCapturer::CleanupWaylandPortal() {
|
||||||
|
#if defined(CROSSDESK_HAS_WAYLAND_CAPTURER) && CROSSDESK_HAS_WAYLAND_CAPTURER
|
||||||
|
if (dbus_connection_) {
|
||||||
|
CloseWaylandPortalSessionAndConnection(dbus_connection_,
|
||||||
|
wayland_session_handle_,
|
||||||
|
"RemoteDesktop.Session.Close");
|
||||||
|
dbus_connection_ = nullptr;
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
use_wayland_portal_ = false;
|
||||||
|
wayland_session_handle_.clear();
|
||||||
|
}
|
||||||
|
|
||||||
|
int KeyboardCapturer::SendWaylandKeyboardCommand(int key_code, bool is_down) {
|
||||||
|
#if defined(CROSSDESK_HAS_WAYLAND_CAPTURER) && CROSSDESK_HAS_WAYLAND_CAPTURER
|
||||||
|
if (!dbus_connection_ || wayland_session_handle_.empty()) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
const auto key_it = vkCodeToX11KeySym.find(key_code);
|
||||||
|
if (key_it == vkCodeToX11KeySym.end()) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
const uint32_t key_state = is_down ? kKeyboardPressed : kKeyboardReleased;
|
||||||
|
const int keysym = key_it->second;
|
||||||
|
|
||||||
|
// Prefer keycode injection to preserve physical-key semantics and avoid
|
||||||
|
// implicit Shift interpretation for uppercase keysyms.
|
||||||
|
if (display_) {
|
||||||
|
const KeyCode x11_keycode =
|
||||||
|
XKeysymToKeycode(display_, static_cast<KeySym>(keysym));
|
||||||
|
if (x11_keycode > 8) {
|
||||||
|
const int evdev_keycode = static_cast<int>(x11_keycode) - 8;
|
||||||
|
if (NotifyWaylandKeyboardKeycode(evdev_keycode, key_state)) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const int fallback_keysym = NormalizeFallbackKeysym(keysym);
|
||||||
|
if (NotifyWaylandKeyboardKeysym(fallback_keysym, key_state)) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
LOG_ERROR("Failed to send Wayland keyboard event, vk_code={}, is_down={}",
|
||||||
|
key_code, is_down);
|
||||||
|
return -3;
|
||||||
|
#else
|
||||||
|
(void)key_code;
|
||||||
|
(void)is_down;
|
||||||
|
return -1;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
bool KeyboardCapturer::NotifyWaylandKeyboardKeysym(int keysym, uint32_t state) {
|
||||||
|
#if defined(CROSSDESK_HAS_WAYLAND_CAPTURER) && CROSSDESK_HAS_WAYLAND_CAPTURER
|
||||||
|
return SendWaylandPortalVoidCall(
|
||||||
|
"NotifyKeyboardKeysym", [&](DBusMessageIter* iter) {
|
||||||
|
const char* session_handle = wayland_session_handle_.c_str();
|
||||||
|
int32_t key_sym = keysym;
|
||||||
|
uint32_t key_state = state;
|
||||||
|
dbus_message_iter_append_basic(iter, DBUS_TYPE_OBJECT_PATH,
|
||||||
|
&session_handle);
|
||||||
|
AppendEmptyOptionsDict(iter);
|
||||||
|
dbus_message_iter_append_basic(iter, DBUS_TYPE_INT32, &key_sym);
|
||||||
|
dbus_message_iter_append_basic(iter, DBUS_TYPE_UINT32, &key_state);
|
||||||
|
});
|
||||||
|
#else
|
||||||
|
(void)keysym;
|
||||||
|
(void)state;
|
||||||
|
return false;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
bool KeyboardCapturer::NotifyWaylandKeyboardKeycode(int keycode,
|
||||||
|
uint32_t state) {
|
||||||
|
#if defined(CROSSDESK_HAS_WAYLAND_CAPTURER) && CROSSDESK_HAS_WAYLAND_CAPTURER
|
||||||
|
return SendWaylandPortalVoidCall(
|
||||||
|
"NotifyKeyboardKeycode", [&](DBusMessageIter* iter) {
|
||||||
|
const char* session_handle = wayland_session_handle_.c_str();
|
||||||
|
int32_t key_code = keycode;
|
||||||
|
uint32_t key_state = state;
|
||||||
|
dbus_message_iter_append_basic(iter, DBUS_TYPE_OBJECT_PATH,
|
||||||
|
&session_handle);
|
||||||
|
AppendEmptyOptionsDict(iter);
|
||||||
|
dbus_message_iter_append_basic(iter, DBUS_TYPE_INT32, &key_code);
|
||||||
|
dbus_message_iter_append_basic(iter, DBUS_TYPE_UINT32, &key_state);
|
||||||
|
});
|
||||||
|
#else
|
||||||
|
(void)keycode;
|
||||||
|
(void)state;
|
||||||
|
return false;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
bool KeyboardCapturer::SendWaylandPortalVoidCall(
|
||||||
|
const char* method_name,
|
||||||
|
const std::function<void(DBusMessageIter*)>& append_args) {
|
||||||
|
#if defined(CROSSDESK_HAS_WAYLAND_CAPTURER) && CROSSDESK_HAS_WAYLAND_CAPTURER
|
||||||
|
if (!dbus_connection_ || !method_name || method_name[0] == '\0') {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
DBusMessage* message = dbus_message_new_method_call(
|
||||||
|
kPortalBusName, kPortalObjectPath, kPortalRemoteDesktopInterface,
|
||||||
|
method_name);
|
||||||
|
if (!message) {
|
||||||
|
LOG_ERROR("Failed to allocate {} message", method_name);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
DBusMessageIter iter;
|
||||||
|
dbus_message_iter_init_append(message, &iter);
|
||||||
|
if (append_args) {
|
||||||
|
append_args(&iter);
|
||||||
|
}
|
||||||
|
|
||||||
|
DBusError error;
|
||||||
|
dbus_error_init(&error);
|
||||||
|
DBusMessage* reply = dbus_connection_send_with_reply_and_block(
|
||||||
|
dbus_connection_, message, 5000, &error);
|
||||||
|
dbus_message_unref(message);
|
||||||
|
if (!reply) {
|
||||||
|
LogDbusError(method_name, &error);
|
||||||
|
dbus_error_free(&error);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (dbus_message_get_type(reply) == DBUS_MESSAGE_TYPE_ERROR) {
|
||||||
|
const char* error_name = dbus_message_get_error_name(reply);
|
||||||
|
LOG_ERROR("{} returned DBus error: {}", method_name,
|
||||||
|
error_name ? error_name : "unknown");
|
||||||
|
dbus_message_unref(reply);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
dbus_message_unref(reply);
|
||||||
|
return true;
|
||||||
|
#else
|
||||||
|
(void)method_name;
|
||||||
|
(void)append_args;
|
||||||
|
return false;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace crossdesk
|
||||||
@@ -1,5 +1,7 @@
|
|||||||
#include "keyboard_capturer.h"
|
#include "keyboard_capturer.h"
|
||||||
|
|
||||||
|
#include <unordered_map>
|
||||||
|
|
||||||
#include "keyboard_converter.h"
|
#include "keyboard_converter.h"
|
||||||
#include "rd_log.h"
|
#include "rd_log.h"
|
||||||
|
|
||||||
@@ -7,9 +9,100 @@ namespace crossdesk {
|
|||||||
|
|
||||||
static OnKeyAction g_on_key_action = nullptr;
|
static OnKeyAction g_on_key_action = nullptr;
|
||||||
static void* g_user_ptr = nullptr;
|
static void* g_user_ptr = nullptr;
|
||||||
|
static std::unordered_map<int, int> g_unmapped_keycode_to_vk;
|
||||||
|
|
||||||
|
static int VkCodeFromUnicode(UniChar ch) {
|
||||||
|
if (ch >= 'a' && ch <= 'z') {
|
||||||
|
return static_cast<int>(ch - 'a' + 'A');
|
||||||
|
}
|
||||||
|
if (ch >= 'A' && ch <= 'Z') {
|
||||||
|
return static_cast<int>(ch);
|
||||||
|
}
|
||||||
|
if (ch >= '0' && ch <= '9') {
|
||||||
|
return static_cast<int>(ch);
|
||||||
|
}
|
||||||
|
|
||||||
|
switch (ch) {
|
||||||
|
case ' ':
|
||||||
|
return 0x20; // VK_SPACE
|
||||||
|
case '-':
|
||||||
|
case '_':
|
||||||
|
return 0xBD; // VK_OEM_MINUS
|
||||||
|
case '=':
|
||||||
|
case '+':
|
||||||
|
return 0xBB; // VK_OEM_PLUS
|
||||||
|
case '[':
|
||||||
|
case '{':
|
||||||
|
return 0xDB; // VK_OEM_4
|
||||||
|
case ']':
|
||||||
|
case '}':
|
||||||
|
return 0xDD; // VK_OEM_6
|
||||||
|
case '\\':
|
||||||
|
case '|':
|
||||||
|
return 0xDC; // VK_OEM_5
|
||||||
|
case ';':
|
||||||
|
case ':':
|
||||||
|
return 0xBA; // VK_OEM_1
|
||||||
|
case '\'':
|
||||||
|
case '"':
|
||||||
|
return 0xDE; // VK_OEM_7
|
||||||
|
case ',':
|
||||||
|
case '<':
|
||||||
|
return 0xBC; // VK_OEM_COMMA
|
||||||
|
case '.':
|
||||||
|
case '>':
|
||||||
|
return 0xBE; // VK_OEM_PERIOD
|
||||||
|
case '/':
|
||||||
|
case '?':
|
||||||
|
return 0xBF; // VK_OEM_2
|
||||||
|
case '`':
|
||||||
|
case '~':
|
||||||
|
return 0xC0; // VK_OEM_3
|
||||||
|
default:
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static int ResolveVkCodeFromMacEvent(CGEventRef event, CGKeyCode key_code,
|
||||||
|
bool is_key_down) {
|
||||||
|
auto key_it = CGKeyCodeToVkCode.find(key_code);
|
||||||
|
if (key_it != CGKeyCodeToVkCode.end()) {
|
||||||
|
if (is_key_down) {
|
||||||
|
g_unmapped_keycode_to_vk.erase(static_cast<int>(key_code));
|
||||||
|
}
|
||||||
|
return key_it->second;
|
||||||
|
}
|
||||||
|
|
||||||
|
int vk_code = -1;
|
||||||
|
UniChar chars[4] = {0};
|
||||||
|
UniCharCount char_count = 0;
|
||||||
|
CGEventKeyboardGetUnicodeString(event, 4, &char_count, chars);
|
||||||
|
if (char_count > 0) {
|
||||||
|
vk_code = VkCodeFromUnicode(chars[0]);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (vk_code < 0) {
|
||||||
|
auto fallback_it =
|
||||||
|
g_unmapped_keycode_to_vk.find(static_cast<int>(key_code));
|
||||||
|
if (fallback_it != g_unmapped_keycode_to_vk.end()) {
|
||||||
|
vk_code = fallback_it->second;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (vk_code >= 0) {
|
||||||
|
if (is_key_down) {
|
||||||
|
g_unmapped_keycode_to_vk[static_cast<int>(key_code)] = vk_code;
|
||||||
|
} else {
|
||||||
|
g_unmapped_keycode_to_vk.erase(static_cast<int>(key_code));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return vk_code;
|
||||||
|
}
|
||||||
|
|
||||||
CGEventRef eventCallback(CGEventTapProxy proxy, CGEventType type,
|
CGEventRef eventCallback(CGEventTapProxy proxy, CGEventType type,
|
||||||
CGEventRef event, void* userInfo) {
|
CGEventRef event, void* userInfo) {
|
||||||
|
(void)proxy;
|
||||||
if (!g_on_key_action) {
|
if (!g_on_key_action) {
|
||||||
return event;
|
return event;
|
||||||
}
|
}
|
||||||
@@ -20,84 +113,74 @@ CGEventRef eventCallback(CGEventTapProxy proxy, CGEventType type,
|
|||||||
return event;
|
return event;
|
||||||
}
|
}
|
||||||
|
|
||||||
int vk_code = 0;
|
|
||||||
|
|
||||||
if (type == kCGEventKeyDown || type == kCGEventKeyUp) {
|
if (type == kCGEventKeyDown || type == kCGEventKeyUp) {
|
||||||
|
const bool is_key_down = (type == kCGEventKeyDown);
|
||||||
CGKeyCode key_code = static_cast<CGKeyCode>(
|
CGKeyCode key_code = static_cast<CGKeyCode>(
|
||||||
CGEventGetIntegerValueField(event, kCGKeyboardEventKeycode));
|
CGEventGetIntegerValueField(event, kCGKeyboardEventKeycode));
|
||||||
if (CGKeyCodeToVkCode.find(key_code) != CGKeyCodeToVkCode.end()) {
|
int vk_code = ResolveVkCodeFromMacEvent(event, key_code, is_key_down);
|
||||||
g_on_key_action(CGKeyCodeToVkCode[key_code], type == kCGEventKeyDown,
|
if (vk_code >= 0) {
|
||||||
g_user_ptr);
|
g_on_key_action(vk_code, is_key_down, g_user_ptr);
|
||||||
}
|
}
|
||||||
} else if (type == kCGEventFlagsChanged) {
|
} else if (type == kCGEventFlagsChanged) {
|
||||||
CGEventFlags current_flags = CGEventGetFlags(event);
|
CGEventFlags current_flags = CGEventGetFlags(event);
|
||||||
CGKeyCode key_code = static_cast<CGKeyCode>(
|
CGKeyCode key_code = static_cast<CGKeyCode>(
|
||||||
CGEventGetIntegerValueField(event, kCGKeyboardEventKeycode));
|
CGEventGetIntegerValueField(event, kCGKeyboardEventKeycode));
|
||||||
|
auto key_it = CGKeyCodeToVkCode.find(key_code);
|
||||||
|
if (key_it == CGKeyCodeToVkCode.end()) {
|
||||||
|
return nullptr;
|
||||||
|
}
|
||||||
|
const int vk_code = key_it->second;
|
||||||
|
|
||||||
// caps lock
|
// caps lock
|
||||||
bool caps_lock_state = (current_flags & kCGEventFlagMaskAlphaShift) != 0;
|
bool caps_lock_state = (current_flags & kCGEventFlagMaskAlphaShift) != 0;
|
||||||
if (caps_lock_state != keyboard_capturer->caps_lock_flag_) {
|
if (caps_lock_state != keyboard_capturer->caps_lock_flag_) {
|
||||||
keyboard_capturer->caps_lock_flag_ = caps_lock_state;
|
keyboard_capturer->caps_lock_flag_ = caps_lock_state;
|
||||||
if (keyboard_capturer->caps_lock_flag_) {
|
g_on_key_action(vk_code, keyboard_capturer->caps_lock_flag_, g_user_ptr);
|
||||||
g_on_key_action(CGKeyCodeToVkCode[key_code], true, g_user_ptr);
|
|
||||||
} else {
|
|
||||||
g_on_key_action(CGKeyCodeToVkCode[key_code], false, g_user_ptr);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// shift
|
// shift
|
||||||
bool shift_state = (current_flags & kCGEventFlagMaskShift) != 0;
|
bool shift_state = (current_flags & kCGEventFlagMaskShift) != 0;
|
||||||
if (shift_state != keyboard_capturer->shift_flag_) {
|
if (shift_state != keyboard_capturer->shift_flag_) {
|
||||||
keyboard_capturer->shift_flag_ = shift_state;
|
keyboard_capturer->shift_flag_ = shift_state;
|
||||||
if (keyboard_capturer->shift_flag_) {
|
g_on_key_action(vk_code, keyboard_capturer->shift_flag_, g_user_ptr);
|
||||||
g_on_key_action(CGKeyCodeToVkCode[key_code], true, g_user_ptr);
|
|
||||||
} else {
|
|
||||||
g_on_key_action(CGKeyCodeToVkCode[key_code], false, g_user_ptr);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// control
|
// control
|
||||||
bool control_state = (current_flags & kCGEventFlagMaskControl) != 0;
|
bool control_state = (current_flags & kCGEventFlagMaskControl) != 0;
|
||||||
if (control_state != keyboard_capturer->control_flag_) {
|
if (control_state != keyboard_capturer->control_flag_) {
|
||||||
keyboard_capturer->control_flag_ = control_state;
|
keyboard_capturer->control_flag_ = control_state;
|
||||||
if (keyboard_capturer->control_flag_) {
|
g_on_key_action(vk_code, keyboard_capturer->control_flag_, g_user_ptr);
|
||||||
g_on_key_action(CGKeyCodeToVkCode[key_code], true, g_user_ptr);
|
|
||||||
} else {
|
|
||||||
g_on_key_action(CGKeyCodeToVkCode[key_code], false, g_user_ptr);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// option
|
// option
|
||||||
bool option_state = (current_flags & kCGEventFlagMaskAlternate) != 0;
|
bool option_state = (current_flags & kCGEventFlagMaskAlternate) != 0;
|
||||||
if (option_state != keyboard_capturer->option_flag_) {
|
if (option_state != keyboard_capturer->option_flag_) {
|
||||||
keyboard_capturer->option_flag_ = option_state;
|
keyboard_capturer->option_flag_ = option_state;
|
||||||
if (keyboard_capturer->option_flag_) {
|
g_on_key_action(vk_code, keyboard_capturer->option_flag_, g_user_ptr);
|
||||||
g_on_key_action(CGKeyCodeToVkCode[key_code], true, g_user_ptr);
|
|
||||||
} else {
|
|
||||||
g_on_key_action(CGKeyCodeToVkCode[key_code], false, g_user_ptr);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// command
|
// command
|
||||||
bool command_state = (current_flags & kCGEventFlagMaskCommand) != 0;
|
bool command_state = (current_flags & kCGEventFlagMaskCommand) != 0;
|
||||||
if (command_state != keyboard_capturer->command_flag_) {
|
if (command_state != keyboard_capturer->command_flag_) {
|
||||||
keyboard_capturer->command_flag_ = command_state;
|
keyboard_capturer->command_flag_ = command_state;
|
||||||
if (keyboard_capturer->command_flag_) {
|
g_on_key_action(vk_code, keyboard_capturer->command_flag_, g_user_ptr);
|
||||||
g_on_key_action(CGKeyCodeToVkCode[key_code], true, g_user_ptr);
|
|
||||||
} else {
|
|
||||||
g_on_key_action(CGKeyCodeToVkCode[key_code], false, g_user_ptr);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return nullptr;
|
return nullptr;
|
||||||
}
|
}
|
||||||
|
|
||||||
KeyboardCapturer::KeyboardCapturer() {}
|
KeyboardCapturer::KeyboardCapturer()
|
||||||
|
: event_tap_(nullptr), run_loop_source_(nullptr) {}
|
||||||
|
|
||||||
KeyboardCapturer::~KeyboardCapturer() {}
|
KeyboardCapturer::~KeyboardCapturer() { Unhook(); }
|
||||||
|
|
||||||
int KeyboardCapturer::Hook(OnKeyAction on_key_action, void* user_ptr) {
|
int KeyboardCapturer::Hook(OnKeyAction on_key_action, void* user_ptr) {
|
||||||
|
if (event_tap_) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
g_unmapped_keycode_to_vk.clear();
|
||||||
g_on_key_action = on_key_action;
|
g_on_key_action = on_key_action;
|
||||||
g_user_ptr = user_ptr;
|
g_user_ptr = user_ptr;
|
||||||
|
|
||||||
@@ -115,15 +198,30 @@ int KeyboardCapturer::Hook(OnKeyAction on_key_action, void* user_ptr) {
|
|||||||
|
|
||||||
run_loop_source_ =
|
run_loop_source_ =
|
||||||
CFMachPortCreateRunLoopSource(kCFAllocatorDefault, event_tap_, 0);
|
CFMachPortCreateRunLoopSource(kCFAllocatorDefault, event_tap_, 0);
|
||||||
|
if (!run_loop_source_) {
|
||||||
|
LOG_ERROR("CFMachPortCreateRunLoopSource failed");
|
||||||
|
CFRelease(event_tap_);
|
||||||
|
event_tap_ = nullptr;
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
CFRunLoopAddSource(CFRunLoopGetCurrent(), run_loop_source_,
|
CFRunLoopAddSource(CFRunLoopGetCurrent(), run_loop_source_,
|
||||||
kCFRunLoopCommonModes);
|
kCFRunLoopCommonModes);
|
||||||
|
|
||||||
|
const CGEventFlags current_flags =
|
||||||
|
CGEventSourceFlagsState(kCGEventSourceStateCombinedSessionState);
|
||||||
|
caps_lock_flag_ = (current_flags & kCGEventFlagMaskAlphaShift) != 0;
|
||||||
|
shift_flag_ = (current_flags & kCGEventFlagMaskShift) != 0;
|
||||||
|
control_flag_ = (current_flags & kCGEventFlagMaskControl) != 0;
|
||||||
|
option_flag_ = (current_flags & kCGEventFlagMaskAlternate) != 0;
|
||||||
|
command_flag_ = (current_flags & kCGEventFlagMaskCommand) != 0;
|
||||||
|
|
||||||
CGEventTapEnable(event_tap_, true);
|
CGEventTapEnable(event_tap_, true);
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
int KeyboardCapturer::Unhook() {
|
int KeyboardCapturer::Unhook() {
|
||||||
|
g_unmapped_keycode_to_vk.clear();
|
||||||
g_on_key_action = nullptr;
|
g_on_key_action = nullptr;
|
||||||
g_user_ptr = nullptr;
|
g_user_ptr = nullptr;
|
||||||
|
|
||||||
@@ -170,9 +268,12 @@ int KeyboardCapturer::SendKeyboardCommand(int key_code, bool is_down) {
|
|||||||
if (vkCodeToCGKeyCode.find(key_code) != vkCodeToCGKeyCode.end()) {
|
if (vkCodeToCGKeyCode.find(key_code) != vkCodeToCGKeyCode.end()) {
|
||||||
CGKeyCode cg_key_code = vkCodeToCGKeyCode[key_code];
|
CGKeyCode cg_key_code = vkCodeToCGKeyCode[key_code];
|
||||||
CGEventRef event = CGEventCreateKeyboardEvent(NULL, cg_key_code, is_down);
|
CGEventRef event = CGEventCreateKeyboardEvent(NULL, cg_key_code, is_down);
|
||||||
CGEventRef clearFlags =
|
if (!event) {
|
||||||
CGEventCreateKeyboardEvent(NULL, (CGKeyCode)0, true);
|
LOG_ERROR("CGEventCreateKeyboardEvent failed");
|
||||||
CGEventSetFlags(clearFlags, 0);
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
CGEventSetFlags(event, 0);
|
||||||
CGEventPost(kCGHIDEventTap, event);
|
CGEventPost(kCGHIDEventTap, event);
|
||||||
CFRelease(event);
|
CFRelease(event);
|
||||||
|
|
||||||
@@ -188,4 +289,4 @@ int KeyboardCapturer::SendKeyboardCommand(int key_code, bool is_down) {
|
|||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
} // namespace crossdesk
|
} // namespace crossdesk
|
||||||
|
|||||||
@@ -24,8 +24,8 @@ class KeyboardCapturer : public DeviceController {
|
|||||||
virtual int SendKeyboardCommand(int key_code, bool is_down);
|
virtual int SendKeyboardCommand(int key_code, bool is_down);
|
||||||
|
|
||||||
private:
|
private:
|
||||||
CFMachPortRef event_tap_;
|
CFMachPortRef event_tap_ = nullptr;
|
||||||
CFRunLoopSourceRef run_loop_source_;
|
CFRunLoopSourceRef run_loop_source_ = nullptr;
|
||||||
|
|
||||||
public:
|
public:
|
||||||
bool caps_lock_flag_ = false;
|
bool caps_lock_flag_ = false;
|
||||||
@@ -36,4 +36,4 @@ class KeyboardCapturer : public DeviceController {
|
|||||||
int fn_key_code_ = 0x3F;
|
int fn_key_code_ = 0x3F;
|
||||||
};
|
};
|
||||||
} // namespace crossdesk
|
} // namespace crossdesk
|
||||||
#endif
|
#endif
|
||||||
|
|||||||
@@ -54,11 +54,28 @@ int KeyboardCapturer::SendKeyboardCommand(int key_code, bool is_down) {
|
|||||||
input.type = INPUT_KEYBOARD;
|
input.type = INPUT_KEYBOARD;
|
||||||
input.ki.wVk = (WORD)key_code;
|
input.ki.wVk = (WORD)key_code;
|
||||||
|
|
||||||
if (!is_down) {
|
const UINT scan_code =
|
||||||
input.ki.dwFlags = KEYEVENTF_KEYUP;
|
MapVirtualKeyW(static_cast<UINT>(key_code), MAPVK_VK_TO_VSC_EX);
|
||||||
|
if (scan_code != 0) {
|
||||||
|
input.ki.wVk = 0;
|
||||||
|
input.ki.wScan = static_cast<WORD>(scan_code & 0xFF);
|
||||||
|
input.ki.dwFlags |= KEYEVENTF_SCANCODE;
|
||||||
|
if ((scan_code & 0xFF00) != 0) {
|
||||||
|
input.ki.dwFlags |= KEYEVENTF_EXTENDEDKEY;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!is_down) {
|
||||||
|
input.ki.dwFlags |= KEYEVENTF_KEYUP;
|
||||||
|
}
|
||||||
|
|
||||||
|
UINT sent = SendInput(1, &input, sizeof(INPUT));
|
||||||
|
if (sent != 1) {
|
||||||
|
LOG_WARN("SendInput failed for key_code={}, is_down={}, err={}", key_code,
|
||||||
|
is_down, GetLastError());
|
||||||
|
return -1;
|
||||||
}
|
}
|
||||||
SendInput(1, &input, sizeof(INPUT));
|
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
} // namespace crossdesk
|
} // namespace crossdesk
|
||||||
|
|||||||
@@ -73,13 +73,13 @@ std::map<int, int> vkCodeToCGKeyCode = {
|
|||||||
{0x20, 0x31}, // Space
|
{0x20, 0x31}, // Space
|
||||||
{0x08, 0x33}, // Backspace
|
{0x08, 0x33}, // Backspace
|
||||||
{0x09, 0x30}, // Tab
|
{0x09, 0x30}, // Tab
|
||||||
{0x2C, 0x74}, // Print Screen
|
{0x2C, 0x69}, // Print Screen(F13)
|
||||||
{0x2D, 0x72}, // Insert
|
{0x2D, 0x72}, // Insert
|
||||||
{0x2E, 0x75}, // Delete
|
{0x2E, 0x75}, // Delete
|
||||||
{0x24, 0x73}, // Home
|
{0x24, 0x73}, // Home
|
||||||
{0x23, 0x77}, // End
|
{0x23, 0x77}, // End
|
||||||
{0x21, 0x79}, // Page Up
|
{0x21, 0x74}, // Page Up
|
||||||
{0x22, 0x7A}, // Page Down
|
{0x22, 0x79}, // Page Down
|
||||||
|
|
||||||
// arrow keys
|
// arrow keys
|
||||||
{0x25, 0x7B}, // Left Arrow
|
{0x25, 0x7B}, // Left Arrow
|
||||||
@@ -191,13 +191,13 @@ std::map<int, int> CGKeyCodeToVkCode = {
|
|||||||
{0x31, 0x20}, // Space
|
{0x31, 0x20}, // Space
|
||||||
{0x33, 0x08}, // Backspace
|
{0x33, 0x08}, // Backspace
|
||||||
{0x30, 0x09}, // Tab
|
{0x30, 0x09}, // Tab
|
||||||
{0x74, 0x2C}, // Print Screen
|
{0x69, 0x2C}, // Print Screen(F13)
|
||||||
{0x72, 0x2D}, // Insert
|
{0x72, 0x2D}, // Insert
|
||||||
{0x75, 0x2E}, // Delete
|
{0x75, 0x2E}, // Delete
|
||||||
{0x73, 0x24}, // Home
|
{0x73, 0x24}, // Home
|
||||||
{0x77, 0x23}, // End
|
{0x77, 0x23}, // End
|
||||||
{0x79, 0x21}, // Page Up
|
{0x74, 0x21}, // Page Up
|
||||||
{0x7A, 0x22}, // Page Down
|
{0x79, 0x22}, // Page Down
|
||||||
|
|
||||||
// arrow keys
|
// arrow keys
|
||||||
{0x7B, 0x25}, // Left Arrow
|
{0x7B, 0x25}, // Left Arrow
|
||||||
@@ -326,21 +326,21 @@ std::map<int, int> vkCodeToX11KeySym = {
|
|||||||
{0x28, 0xFF54}, // Down Arrow
|
{0x28, 0xFF54}, // Down Arrow
|
||||||
|
|
||||||
// numpad
|
// numpad
|
||||||
{0x60, 0x0030}, // Numpad 0
|
{0x60, 0xFFB0}, // Numpad 0
|
||||||
{0x61, 0x0031}, // Numpad 1
|
{0x61, 0xFFB1}, // Numpad 1
|
||||||
{0x62, 0x0032}, // Numpad 2
|
{0x62, 0xFFB2}, // Numpad 2
|
||||||
{0x63, 0x0033}, // Numpad 3
|
{0x63, 0xFFB3}, // Numpad 3
|
||||||
{0x64, 0x0034}, // Numpad 4
|
{0x64, 0xFFB4}, // Numpad 4
|
||||||
{0x65, 0x0035}, // Numpad 5
|
{0x65, 0xFFB5}, // Numpad 5
|
||||||
{0x66, 0x0036}, // Numpad 6
|
{0x66, 0xFFB6}, // Numpad 6
|
||||||
{0x67, 0x0037}, // Numpad 7
|
{0x67, 0xFFB7}, // Numpad 7
|
||||||
{0x68, 0x0038}, // Numpad 8
|
{0x68, 0xFFB8}, // Numpad 8
|
||||||
{0x69, 0x0039}, // Numpad 9
|
{0x69, 0xFFB9}, // Numpad 9
|
||||||
{0x6E, 0x003A}, // Numpad .
|
{0x6E, 0xFFAE}, // Numpad .
|
||||||
{0x6F, 0x002F}, // Numpad /
|
{0x6F, 0xFFAF}, // Numpad /
|
||||||
{0x6A, 0x002A}, // Numpad *
|
{0x6A, 0xFFAA}, // Numpad *
|
||||||
{0x6D, 0x002D}, // Numpad -
|
{0x6D, 0xFFAD}, // Numpad -
|
||||||
{0x6B, 0x002B}, // Numpad +
|
{0x6B, 0xFFAB}, // Numpad +
|
||||||
|
|
||||||
// symbol keys
|
// symbol keys
|
||||||
{0xBA, 0x003B}, // ; (Semicolon)
|
{0xBA, 0x003B}, // ; (Semicolon)
|
||||||
@@ -454,21 +454,21 @@ std::map<int, int> x11KeySymToVkCode = {
|
|||||||
{0xFF54, 0x28}, // Down Arrow
|
{0xFF54, 0x28}, // Down Arrow
|
||||||
|
|
||||||
// numpad
|
// numpad
|
||||||
{0x0030, 0x60}, // Numpad 0
|
{0xFFB0, 0x60}, // Numpad 0
|
||||||
{0x0031, 0x61}, // Numpad 1
|
{0xFFB1, 0x61}, // Numpad 1
|
||||||
{0x0032, 0x62}, // Numpad 2
|
{0xFFB2, 0x62}, // Numpad 2
|
||||||
{0x0033, 0x63}, // Numpad 3
|
{0xFFB3, 0x63}, // Numpad 3
|
||||||
{0x0034, 0x64}, // Numpad 4
|
{0xFFB4, 0x64}, // Numpad 4
|
||||||
{0x0035, 0x65}, // Numpad 5
|
{0xFFB5, 0x65}, // Numpad 5
|
||||||
{0x0036, 0x66}, // Numpad 6
|
{0xFFB6, 0x66}, // Numpad 6
|
||||||
{0x0037, 0x67}, // Numpad 7
|
{0xFFB7, 0x67}, // Numpad 7
|
||||||
{0x0038, 0x68}, // Numpad 8
|
{0xFFB8, 0x68}, // Numpad 8
|
||||||
{0x0039, 0x69}, // Numpad 9
|
{0xFFB9, 0x69}, // Numpad 9
|
||||||
{0x003A, 0x6E}, // Numpad .
|
{0xFFAE, 0x6E}, // Numpad .
|
||||||
{0x002F, 0x6F}, // Numpad /
|
{0xFFAF, 0x6F}, // Numpad /
|
||||||
{0x002A, 0x6A}, // Numpad *
|
{0xFFAA, 0x6A}, // Numpad *
|
||||||
{0x002D, 0x6D}, // Numpad -
|
{0xFFAD, 0x6D}, // Numpad -
|
||||||
{0x002B, 0x6B}, // Numpad +
|
{0xFFAB, 0x6B}, // Numpad +
|
||||||
|
|
||||||
// symbol keys
|
// symbol keys
|
||||||
{0x003B, 0xBA}, // ; (Semicolon)
|
{0x003B, 0xBA}, // ; (Semicolon)
|
||||||
@@ -557,13 +557,13 @@ std::map<int, int> cgKeyCodeToX11KeySym = {
|
|||||||
{0x31, 0x0020}, // Space
|
{0x31, 0x0020}, // Space
|
||||||
{0x33, 0xFF08}, // Backspace
|
{0x33, 0xFF08}, // Backspace
|
||||||
{0x30, 0xFF09}, // Tab
|
{0x30, 0xFF09}, // Tab
|
||||||
{0x74, 0xFF15}, // Print Screen
|
{0x69, 0xFF15}, // Print Screen(F13)
|
||||||
{0x72, 0xFF63}, // Insert
|
{0x72, 0xFF63}, // Insert
|
||||||
{0x75, 0xFFFF}, // Delete
|
{0x75, 0xFFFF}, // Delete
|
||||||
{0x73, 0xFF50}, // Home
|
{0x73, 0xFF50}, // Home
|
||||||
{0x77, 0xFF57}, // End
|
{0x77, 0xFF57}, // End
|
||||||
{0x79, 0xFF55}, // Page Up
|
{0x74, 0xFF55}, // Page Up
|
||||||
{0x7A, 0xFF56}, // Page Down
|
{0x79, 0xFF56}, // Page Down
|
||||||
|
|
||||||
// arrow keys
|
// arrow keys
|
||||||
{0x7B, 0xFF51}, // Left Arrow
|
{0x7B, 0xFF51}, // Left Arrow
|
||||||
@@ -572,21 +572,21 @@ std::map<int, int> cgKeyCodeToX11KeySym = {
|
|||||||
{0x7D, 0xFF54}, // Down Arrow
|
{0x7D, 0xFF54}, // Down Arrow
|
||||||
|
|
||||||
// numpad
|
// numpad
|
||||||
{0x52, 0x0030}, // Numpad 0
|
{0x52, 0xFFB0}, // Numpad 0
|
||||||
{0x53, 0x0031}, // Numpad 1
|
{0x53, 0xFFB1}, // Numpad 1
|
||||||
{0x54, 0x0032}, // Numpad 2
|
{0x54, 0xFFB2}, // Numpad 2
|
||||||
{0x55, 0x0033}, // Numpad 3
|
{0x55, 0xFFB3}, // Numpad 3
|
||||||
{0x56, 0x0034}, // Numpad 4
|
{0x56, 0xFFB4}, // Numpad 4
|
||||||
{0x57, 0x0035}, // Numpad 5
|
{0x57, 0xFFB5}, // Numpad 5
|
||||||
{0x58, 0x0036}, // Numpad 6
|
{0x58, 0xFFB6}, // Numpad 6
|
||||||
{0x59, 0x0037}, // Numpad 7
|
{0x59, 0xFFB7}, // Numpad 7
|
||||||
{0x5B, 0x0038}, // Numpad 8
|
{0x5B, 0xFFB8}, // Numpad 8
|
||||||
{0x5C, 0x0039}, // Numpad 9
|
{0x5C, 0xFFB9}, // Numpad 9
|
||||||
{0x41, 0x003A}, // Numpad .
|
{0x41, 0xFFAE}, // Numpad .
|
||||||
{0x4B, 0x002F}, // Numpad /
|
{0x4B, 0xFFAF}, // Numpad /
|
||||||
{0x43, 0x002A}, // Numpad *
|
{0x43, 0xFFAA}, // Numpad *
|
||||||
{0x4E, 0x002D}, // Numpad -
|
{0x4E, 0xFFAD}, // Numpad -
|
||||||
{0x45, 0x002B}, // Numpad +
|
{0x45, 0xFFAB}, // Numpad +
|
||||||
|
|
||||||
// symbol keys
|
// symbol keys
|
||||||
{0x29, 0x003B}, // ; (Semicolon)
|
{0x29, 0x003B}, // ; (Semicolon)
|
||||||
@@ -683,13 +683,13 @@ std::map<int, int> x11KeySymToCgKeyCode = {
|
|||||||
{0x0020, 0x31}, // Space
|
{0x0020, 0x31}, // Space
|
||||||
{0xFF08, 0x33}, // Backspace
|
{0xFF08, 0x33}, // Backspace
|
||||||
{0xFF09, 0x30}, // Tab
|
{0xFF09, 0x30}, // Tab
|
||||||
{0xFF15, 0x74}, // Print Screen
|
{0xFF15, 0x69}, // Print Screen(F13)
|
||||||
{0xFF63, 0x72}, // Insert
|
{0xFF63, 0x72}, // Insert
|
||||||
{0xFFFF, 0x75}, // Delete
|
{0xFFFF, 0x75}, // Delete
|
||||||
{0xFF50, 0x73}, // Home
|
{0xFF50, 0x73}, // Home
|
||||||
{0xFF57, 0x77}, // End
|
{0xFF57, 0x77}, // End
|
||||||
{0xFF55, 0x79}, // Page Up
|
{0xFF55, 0x74}, // Page Up
|
||||||
{0xFF56, 0x7A}, // Page Down
|
{0xFF56, 0x79}, // Page Down
|
||||||
|
|
||||||
// arrow keys
|
// arrow keys
|
||||||
{0xFF51, 0x7B}, // Left Arrow
|
{0xFF51, 0x7B}, // Left Arrow
|
||||||
@@ -698,21 +698,21 @@ std::map<int, int> x11KeySymToCgKeyCode = {
|
|||||||
{0xFF54, 0x7D}, // Down Arrow
|
{0xFF54, 0x7D}, // Down Arrow
|
||||||
|
|
||||||
// numpad
|
// numpad
|
||||||
{0x0030, 0x52}, // Numpad 0
|
{0xFFB0, 0x52}, // Numpad 0
|
||||||
{0x0031, 0x53}, // Numpad 1
|
{0xFFB1, 0x53}, // Numpad 1
|
||||||
{0x0032, 0x54}, // Numpad 2
|
{0xFFB2, 0x54}, // Numpad 2
|
||||||
{0x0033, 0x55}, // Numpad 3
|
{0xFFB3, 0x55}, // Numpad 3
|
||||||
{0x0034, 0x56}, // Numpad 4
|
{0xFFB4, 0x56}, // Numpad 4
|
||||||
{0x0035, 0x57}, // Numpad 5
|
{0xFFB5, 0x57}, // Numpad 5
|
||||||
{0x0036, 0x58}, // Numpad 6
|
{0xFFB6, 0x58}, // Numpad 6
|
||||||
{0x0037, 0x59}, // Numpad 7
|
{0xFFB7, 0x59}, // Numpad 7
|
||||||
{0x0038, 0x5B}, // Numpad 8
|
{0xFFB8, 0x5B}, // Numpad 8
|
||||||
{0x0039, 0x5C}, // Numpad 9
|
{0xFFB9, 0x5C}, // Numpad 9
|
||||||
{0x003A, 0x41}, // Numpad .
|
{0xFFAE, 0x41}, // Numpad .
|
||||||
{0x002F, 0x4B}, // Numpad /
|
{0xFFAF, 0x4B}, // Numpad /
|
||||||
{0x002A, 0x43}, // Numpad *
|
{0xFFAA, 0x43}, // Numpad *
|
||||||
{0x002D, 0x4E}, // Numpad -
|
{0xFFAD, 0x4E}, // Numpad -
|
||||||
{0x002B, 0x45}, // Numpad +
|
{0xFFAB, 0x45}, // Numpad +
|
||||||
|
|
||||||
// symbol keys
|
// symbol keys
|
||||||
{0x003B, 0x29}, // ; (Semicolon)
|
{0x003B, 0x29}, // ; (Semicolon)
|
||||||
@@ -739,4 +739,4 @@ std::map<int, int> x11KeySymToCgKeyCode = {
|
|||||||
{0xFFEC, 0x36}, // Right Command
|
{0xFFEC, 0x36}, // Right Command
|
||||||
};
|
};
|
||||||
} // namespace crossdesk
|
} // namespace crossdesk
|
||||||
#endif
|
#endif
|
||||||
|
|||||||
@@ -2,6 +2,7 @@
|
|||||||
|
|
||||||
#include <X11/extensions/XTest.h>
|
#include <X11/extensions/XTest.h>
|
||||||
|
|
||||||
|
#include "platform.h"
|
||||||
#include "rd_log.h"
|
#include "rd_log.h"
|
||||||
|
|
||||||
namespace crossdesk {
|
namespace crossdesk {
|
||||||
@@ -12,6 +13,17 @@ MouseController::~MouseController() { Destroy(); }
|
|||||||
|
|
||||||
int MouseController::Init(std::vector<DisplayInfo> display_info_list) {
|
int MouseController::Init(std::vector<DisplayInfo> display_info_list) {
|
||||||
display_info_list_ = display_info_list;
|
display_info_list_ = display_info_list;
|
||||||
|
|
||||||
|
if (IsWaylandSession()) {
|
||||||
|
if (InitWaylandPortal()) {
|
||||||
|
use_wayland_portal_ = true;
|
||||||
|
LOG_INFO("Mouse controller initialized with Wayland portal backend");
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
LOG_WARN(
|
||||||
|
"Wayland mouse control init failed, falling back to X11/XTest backend");
|
||||||
|
}
|
||||||
|
|
||||||
display_ = XOpenDisplay(NULL);
|
display_ = XOpenDisplay(NULL);
|
||||||
if (!display_) {
|
if (!display_) {
|
||||||
LOG_ERROR("Cannot connect to X server");
|
LOG_ERROR("Cannot connect to X server");
|
||||||
@@ -25,26 +37,68 @@ int MouseController::Init(std::vector<DisplayInfo> display_info_list) {
|
|||||||
&minor_version)) {
|
&minor_version)) {
|
||||||
LOG_ERROR("XTest extension not available");
|
LOG_ERROR("XTest extension not available");
|
||||||
XCloseDisplay(display_);
|
XCloseDisplay(display_);
|
||||||
|
display_ = nullptr;
|
||||||
return -2;
|
return -2;
|
||||||
}
|
}
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void MouseController::UpdateDisplayInfoList(
|
||||||
|
const std::vector<DisplayInfo>& display_info_list) {
|
||||||
|
if (display_info_list.empty()) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
display_info_list_ = display_info_list;
|
||||||
|
if (use_wayland_portal_) {
|
||||||
|
OnWaylandDisplayInfoListUpdated();
|
||||||
|
}
|
||||||
|
|
||||||
|
if (last_display_index_ < 0 ||
|
||||||
|
last_display_index_ >= static_cast<int>(display_info_list_.size())) {
|
||||||
|
last_display_index_ = -1;
|
||||||
|
last_norm_x_ = -1.0;
|
||||||
|
last_norm_y_ = -1.0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
int MouseController::Destroy() {
|
int MouseController::Destroy() {
|
||||||
|
CleanupWaylandPortal();
|
||||||
|
|
||||||
if (display_) {
|
if (display_) {
|
||||||
XCloseDisplay(display_);
|
XCloseDisplay(display_);
|
||||||
display_ = nullptr;
|
display_ = nullptr;
|
||||||
}
|
}
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
int MouseController::SendMouseCommand(RemoteAction remote_action,
|
int MouseController::SendMouseCommand(RemoteAction remote_action,
|
||||||
int display_index) {
|
int display_index) {
|
||||||
|
if (remote_action.type != ControlType::mouse) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (use_wayland_portal_) {
|
||||||
|
return SendWaylandMouseCommand(remote_action, display_index);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!display_) {
|
||||||
|
LOG_ERROR("X11 display not initialized");
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
switch (remote_action.type) {
|
switch (remote_action.type) {
|
||||||
case mouse:
|
case mouse:
|
||||||
switch (remote_action.m.flag) {
|
switch (remote_action.m.flag) {
|
||||||
case MouseFlag::move:
|
case MouseFlag::move: {
|
||||||
|
if (display_index < 0 ||
|
||||||
|
display_index >= static_cast<int>(display_info_list_.size())) {
|
||||||
|
LOG_ERROR("Invalid display index: {}", display_index);
|
||||||
|
return -2;
|
||||||
|
}
|
||||||
|
|
||||||
SetMousePosition(
|
SetMousePosition(
|
||||||
static_cast<int>(remote_action.m.x *
|
static_cast<int>(remote_action.m.x *
|
||||||
display_info_list_[display_index].width +
|
display_info_list_[display_index].width +
|
||||||
@@ -53,6 +107,7 @@ int MouseController::SendMouseCommand(RemoteAction remote_action,
|
|||||||
display_info_list_[display_index].height +
|
display_info_list_[display_index].height +
|
||||||
display_info_list_[display_index].top));
|
display_info_list_[display_index].top));
|
||||||
break;
|
break;
|
||||||
|
}
|
||||||
case MouseFlag::left_down:
|
case MouseFlag::left_down:
|
||||||
XTestFakeButtonEvent(display_, 1, True, CurrentTime);
|
XTestFakeButtonEvent(display_, 1, True, CurrentTime);
|
||||||
XFlush(display_);
|
XFlush(display_);
|
||||||
@@ -103,25 +158,39 @@ int MouseController::SendMouseCommand(RemoteAction remote_action,
|
|||||||
}
|
}
|
||||||
|
|
||||||
void MouseController::SetMousePosition(int x, int y) {
|
void MouseController::SetMousePosition(int x, int y) {
|
||||||
|
if (!display_) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
XWarpPointer(display_, None, root_, 0, 0, 0, 0, x, y);
|
XWarpPointer(display_, None, root_, 0, 0, 0, 0, x, y);
|
||||||
XFlush(display_);
|
XFlush(display_);
|
||||||
}
|
}
|
||||||
|
|
||||||
void MouseController::SimulateKeyDown(int kval) {
|
void MouseController::SimulateKeyDown(int kval) {
|
||||||
|
if (!display_) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
XTestFakeKeyEvent(display_, kval, True, CurrentTime);
|
XTestFakeKeyEvent(display_, kval, True, CurrentTime);
|
||||||
XFlush(display_);
|
XFlush(display_);
|
||||||
}
|
}
|
||||||
|
|
||||||
void MouseController::SimulateKeyUp(int kval) {
|
void MouseController::SimulateKeyUp(int kval) {
|
||||||
|
if (!display_) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
XTestFakeKeyEvent(display_, kval, False, CurrentTime);
|
XTestFakeKeyEvent(display_, kval, False, CurrentTime);
|
||||||
XFlush(display_);
|
XFlush(display_);
|
||||||
}
|
}
|
||||||
|
|
||||||
void MouseController::SimulateMouseWheel(int direction_button, int count) {
|
void MouseController::SimulateMouseWheel(int direction_button, int count) {
|
||||||
|
if (!display_) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
for (int i = 0; i < count; ++i) {
|
for (int i = 0; i < count; ++i) {
|
||||||
XTestFakeButtonEvent(display_, direction_button, True, CurrentTime);
|
XTestFakeButtonEvent(display_, direction_button, True, CurrentTime);
|
||||||
XTestFakeButtonEvent(display_, direction_button, False, CurrentTime);
|
XTestFakeButtonEvent(display_, direction_button, False, CurrentTime);
|
||||||
}
|
}
|
||||||
XFlush(display_);
|
XFlush(display_);
|
||||||
}
|
}
|
||||||
} // namespace crossdesk
|
|
||||||
|
} // namespace crossdesk
|
||||||
|
|||||||
@@ -11,10 +11,16 @@
|
|||||||
#include <X11/Xutil.h>
|
#include <X11/Xutil.h>
|
||||||
#include <unistd.h>
|
#include <unistd.h>
|
||||||
|
|
||||||
|
#include <functional>
|
||||||
|
#include <cstdint>
|
||||||
|
#include <string>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
|
|
||||||
#include "device_controller.h"
|
#include "device_controller.h"
|
||||||
|
|
||||||
|
struct DBusConnection;
|
||||||
|
struct DBusMessageIter;
|
||||||
|
|
||||||
namespace crossdesk {
|
namespace crossdesk {
|
||||||
|
|
||||||
class MouseController : public DeviceController {
|
class MouseController : public DeviceController {
|
||||||
@@ -26,18 +32,47 @@ class MouseController : public DeviceController {
|
|||||||
virtual int Init(std::vector<DisplayInfo> display_info_list);
|
virtual int Init(std::vector<DisplayInfo> display_info_list);
|
||||||
virtual int Destroy();
|
virtual int Destroy();
|
||||||
virtual int SendMouseCommand(RemoteAction remote_action, int display_index);
|
virtual int SendMouseCommand(RemoteAction remote_action, int display_index);
|
||||||
|
void UpdateDisplayInfoList(const std::vector<DisplayInfo>& display_info_list);
|
||||||
|
|
||||||
private:
|
private:
|
||||||
void SimulateKeyDown(int kval);
|
void SimulateKeyDown(int kval);
|
||||||
void SimulateKeyUp(int kval);
|
void SimulateKeyUp(int kval);
|
||||||
void SetMousePosition(int x, int y);
|
void SetMousePosition(int x, int y);
|
||||||
void SimulateMouseWheel(int direction_button, int count);
|
void SimulateMouseWheel(int direction_button, int count);
|
||||||
|
bool InitWaylandPortal();
|
||||||
|
void CleanupWaylandPortal();
|
||||||
|
int SendWaylandMouseCommand(RemoteAction remote_action, int display_index);
|
||||||
|
void OnWaylandDisplayInfoListUpdated();
|
||||||
|
bool NotifyWaylandPointerMotion(double dx, double dy);
|
||||||
|
bool NotifyWaylandPointerMotionAbsolute(uint32_t stream, double x, double y);
|
||||||
|
bool NotifyWaylandPointerButton(int button, uint32_t state);
|
||||||
|
bool NotifyWaylandPointerAxisDiscrete(uint32_t axis, int32_t steps);
|
||||||
|
bool SendWaylandPortalVoidCall(const char* method_name,
|
||||||
|
const std::function<void(DBusMessageIter*)>&
|
||||||
|
append_args);
|
||||||
|
|
||||||
|
enum class WaylandAbsoluteMode { kUnknown, kPixels, kNormalized, kDisabled };
|
||||||
|
|
||||||
Display* display_ = nullptr;
|
Display* display_ = nullptr;
|
||||||
Window root_ = 0;
|
Window root_ = 0;
|
||||||
std::vector<DisplayInfo> display_info_list_;
|
std::vector<DisplayInfo> display_info_list_;
|
||||||
int screen_width_ = 0;
|
int screen_width_ = 0;
|
||||||
int screen_height_ = 0;
|
int screen_height_ = 0;
|
||||||
|
bool use_wayland_portal_ = false;
|
||||||
|
|
||||||
|
DBusConnection* dbus_connection_ = nullptr;
|
||||||
|
std::string wayland_session_handle_;
|
||||||
|
int last_display_index_ = -1;
|
||||||
|
double last_norm_x_ = -1.0;
|
||||||
|
double last_norm_y_ = -1.0;
|
||||||
|
bool logged_wayland_display_info_ = false;
|
||||||
|
uintptr_t last_logged_wayland_stream_ = 0;
|
||||||
|
int last_logged_wayland_width_ = 0;
|
||||||
|
int last_logged_wayland_height_ = 0;
|
||||||
|
WaylandAbsoluteMode wayland_absolute_mode_ = WaylandAbsoluteMode::kUnknown;
|
||||||
|
bool wayland_absolute_disabled_logged_ = false;
|
||||||
|
uint32_t wayland_absolute_stream_id_ = 0;
|
||||||
|
bool using_shared_wayland_session_ = false;
|
||||||
};
|
};
|
||||||
} // namespace crossdesk
|
} // namespace crossdesk
|
||||||
#endif
|
#endif
|
||||||
|
|||||||
1075
src/device_controller/mouse/linux/mouse_controller_wayland.cpp
Normal file
1075
src/device_controller/mouse/linux/mouse_controller_wayland.cpp
Normal file
File diff suppressed because it is too large
Load Diff
@@ -224,6 +224,8 @@ int Render::ConnectTo(const std::string& remote_id, const char* password,
|
|||||||
}
|
}
|
||||||
AddAudioStream(props->peer_, props->audio_label_.c_str());
|
AddAudioStream(props->peer_, props->audio_label_.c_str());
|
||||||
AddDataStream(props->peer_, props->data_label_.c_str(), false);
|
AddDataStream(props->peer_, props->data_label_.c_str(), false);
|
||||||
|
AddDataStream(props->peer_, props->mouse_label_.c_str(), false);
|
||||||
|
AddDataStream(props->peer_, props->keyboard_label_.c_str(), true);
|
||||||
AddDataStream(props->peer_, props->control_data_label_.c_str(), true);
|
AddDataStream(props->peer_, props->control_data_label_.c_str(), true);
|
||||||
AddDataStream(props->peer_, props->file_label_.c_str(), true);
|
AddDataStream(props->peer_, props->file_label_.c_str(), true);
|
||||||
AddDataStream(props->peer_, props->file_feedback_label_.c_str(), true);
|
AddDataStream(props->peer_, props->file_feedback_label_.c_str(), true);
|
||||||
|
|||||||
@@ -8,6 +8,7 @@
|
|||||||
#endif
|
#endif
|
||||||
|
|
||||||
#include <cstdlib>
|
#include <cstdlib>
|
||||||
|
#include <cmath>
|
||||||
#include <cstring>
|
#include <cstring>
|
||||||
#include <filesystem>
|
#include <filesystem>
|
||||||
#include <fstream>
|
#include <fstream>
|
||||||
@@ -580,8 +581,9 @@ int Render::ScreenCapturerInit() {
|
|||||||
|
|
||||||
if (0 == screen_capturer_init_ret) {
|
if (0 == screen_capturer_init_ret) {
|
||||||
LOG_INFO("Init screen capturer success");
|
LOG_INFO("Init screen capturer success");
|
||||||
if (display_info_list_.empty()) {
|
const auto latest_display_info = screen_capturer_->GetDisplayInfoList();
|
||||||
display_info_list_ = screen_capturer_->GetDisplayInfoList();
|
if (!latest_display_info.empty()) {
|
||||||
|
display_info_list_ = latest_display_info;
|
||||||
}
|
}
|
||||||
return 0;
|
return 0;
|
||||||
} else {
|
} else {
|
||||||
@@ -594,10 +596,22 @@ int Render::ScreenCapturerInit() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
int Render::StartScreenCapturer() {
|
int Render::StartScreenCapturer() {
|
||||||
|
if (!screen_capturer_) {
|
||||||
|
LOG_INFO("Screen capturer instance missing, recreating before start");
|
||||||
|
if (0 != ScreenCapturerInit()) {
|
||||||
|
LOG_ERROR("Recreate screen capturer failed");
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
if (screen_capturer_) {
|
if (screen_capturer_) {
|
||||||
LOG_INFO("Start screen capturer, show cursor: {}", show_cursor_);
|
LOG_INFO("Start screen capturer, show cursor: {}", show_cursor_);
|
||||||
|
|
||||||
screen_capturer_->Start(show_cursor_);
|
const int ret = screen_capturer_->Start(show_cursor_);
|
||||||
|
if (ret != 0) {
|
||||||
|
LOG_ERROR("Start screen capturer failed: {}", ret);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
@@ -650,14 +664,42 @@ int Render::StartMouseController() {
|
|||||||
LOG_INFO("Device controller factory is nullptr");
|
LOG_INFO("Device controller factory is nullptr");
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#if defined(__linux__) && !defined(__APPLE__)
|
||||||
|
if (IsWaylandSession()) {
|
||||||
|
if (!screen_capturer_) {
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
const auto latest_display_info = screen_capturer_->GetDisplayInfoList();
|
||||||
|
if (latest_display_info.empty() ||
|
||||||
|
latest_display_info[0].handle == nullptr) {
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (screen_capturer_) {
|
||||||
|
const auto latest_display_info = screen_capturer_->GetDisplayInfoList();
|
||||||
|
if (!latest_display_info.empty()) {
|
||||||
|
display_info_list_ = latest_display_info;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
mouse_controller_ = (MouseController*)device_controller_factory_->Create(
|
mouse_controller_ = (MouseController*)device_controller_factory_->Create(
|
||||||
DeviceControllerFactory::Device::Mouse);
|
DeviceControllerFactory::Device::Mouse);
|
||||||
|
if (!mouse_controller_) {
|
||||||
|
LOG_ERROR("Create mouse controller failed");
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
int mouse_controller_init_ret = mouse_controller_->Init(display_info_list_);
|
int mouse_controller_init_ret = mouse_controller_->Init(display_info_list_);
|
||||||
if (0 != mouse_controller_init_ret) {
|
if (0 != mouse_controller_init_ret) {
|
||||||
LOG_INFO("Destroy mouse controller");
|
LOG_INFO("Destroy mouse controller");
|
||||||
mouse_controller_->Destroy();
|
mouse_controller_->Destroy();
|
||||||
|
delete mouse_controller_;
|
||||||
mouse_controller_ = nullptr;
|
mouse_controller_ = nullptr;
|
||||||
|
return mouse_controller_init_ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
@@ -692,7 +734,7 @@ int Render::StartKeyboardCapturer() {
|
|||||||
LOG_INFO("Start keyboard capturer");
|
LOG_INFO("Start keyboard capturer");
|
||||||
}
|
}
|
||||||
|
|
||||||
return 0;
|
return keyboard_capturer_init_ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
int Render::StopKeyboardCapturer() {
|
int Render::StopKeyboardCapturer() {
|
||||||
@@ -859,6 +901,8 @@ int Render::CreateConnectionPeer() {
|
|||||||
|
|
||||||
AddAudioStream(peer_, audio_label_.c_str());
|
AddAudioStream(peer_, audio_label_.c_str());
|
||||||
AddDataStream(peer_, data_label_.c_str(), false);
|
AddDataStream(peer_, data_label_.c_str(), false);
|
||||||
|
AddDataStream(peer_, mouse_label_.c_str(), false);
|
||||||
|
AddDataStream(peer_, keyboard_label_.c_str(), true);
|
||||||
AddDataStream(peer_, control_data_label_.c_str(), true);
|
AddDataStream(peer_, control_data_label_.c_str(), true);
|
||||||
AddDataStream(peer_, file_label_.c_str(), true);
|
AddDataStream(peer_, file_label_.c_str(), true);
|
||||||
AddDataStream(peer_, file_feedback_label_.c_str(), true);
|
AddDataStream(peer_, file_feedback_label_.c_str(), true);
|
||||||
@@ -924,9 +968,24 @@ int Render::AudioDeviceDestroy() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
void Render::UpdateInteractions() {
|
void Render::UpdateInteractions() {
|
||||||
|
#if defined(__linux__) && !defined(__APPLE__)
|
||||||
|
const bool is_wayland_session = IsWaylandSession();
|
||||||
|
const bool stop_wayland_mouse_before_screen =
|
||||||
|
is_wayland_session && !start_screen_capturer_ &&
|
||||||
|
screen_capturer_is_started_ && !start_mouse_controller_ &&
|
||||||
|
mouse_controller_is_started_;
|
||||||
|
if (stop_wayland_mouse_before_screen) {
|
||||||
|
LOG_INFO("Stopping Wayland mouse controller before screen capturer to "
|
||||||
|
"cleanly release the shared portal session");
|
||||||
|
StopMouseController();
|
||||||
|
mouse_controller_is_started_ = false;
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
if (start_screen_capturer_ && !screen_capturer_is_started_) {
|
if (start_screen_capturer_ && !screen_capturer_is_started_) {
|
||||||
StartScreenCapturer();
|
if (0 == StartScreenCapturer()) {
|
||||||
screen_capturer_is_started_ = true;
|
screen_capturer_is_started_ = true;
|
||||||
|
}
|
||||||
} else if (!start_screen_capturer_ && screen_capturer_is_started_) {
|
} else if (!start_screen_capturer_ && screen_capturer_is_started_) {
|
||||||
StopScreenCapturer();
|
StopScreenCapturer();
|
||||||
screen_capturer_is_started_ = false;
|
screen_capturer_is_started_ = false;
|
||||||
@@ -941,17 +1000,29 @@ void Render::UpdateInteractions() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
if (start_mouse_controller_ && !mouse_controller_is_started_) {
|
if (start_mouse_controller_ && !mouse_controller_is_started_) {
|
||||||
StartMouseController();
|
if (0 == StartMouseController()) {
|
||||||
mouse_controller_is_started_ = true;
|
mouse_controller_is_started_ = true;
|
||||||
|
}
|
||||||
} else if (!start_mouse_controller_ && mouse_controller_is_started_) {
|
} else if (!start_mouse_controller_ && mouse_controller_is_started_) {
|
||||||
StopMouseController();
|
StopMouseController();
|
||||||
mouse_controller_is_started_ = false;
|
mouse_controller_is_started_ = false;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#if defined(__linux__) && !defined(__APPLE__)
|
||||||
|
if (screen_capturer_is_started_ && screen_capturer_ && mouse_controller_) {
|
||||||
|
const auto latest_display_info = screen_capturer_->GetDisplayInfoList();
|
||||||
|
if (!latest_display_info.empty()) {
|
||||||
|
display_info_list_ = latest_display_info;
|
||||||
|
mouse_controller_->UpdateDisplayInfoList(display_info_list_);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
if (start_keyboard_capturer_ && focus_on_stream_window_) {
|
if (start_keyboard_capturer_ && focus_on_stream_window_) {
|
||||||
if (!keyboard_capturer_is_started_) {
|
if (!keyboard_capturer_is_started_) {
|
||||||
StartKeyboardCapturer();
|
if (StartKeyboardCapturer() == 0) {
|
||||||
keyboard_capturer_is_started_ = true;
|
keyboard_capturer_is_started_ = true;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
} else if (keyboard_capturer_is_started_) {
|
} else if (keyboard_capturer_is_started_) {
|
||||||
StopKeyboardCapturer();
|
StopKeyboardCapturer();
|
||||||
@@ -1276,9 +1347,9 @@ int Render::SetupFontAndStyle(ImFont** system_chinese_font_out) {
|
|||||||
const char* base_font_paths[] = {
|
const char* base_font_paths[] = {
|
||||||
"/usr/share/fonts/opentype/noto/NotoSansCJK-Regular.ttc",
|
"/usr/share/fonts/opentype/noto/NotoSansCJK-Regular.ttc",
|
||||||
"/usr/share/fonts/opentype/noto/NotoSans-Regular.ttf",
|
"/usr/share/fonts/opentype/noto/NotoSans-Regular.ttf",
|
||||||
"/usr/share/fonts/opentype/wqy/wqy-microhei.ttc",
|
"/usr/share/fonts/truetype/wqy/wqy-microhei.ttc",
|
||||||
"/usr/share/fonts/opentype/wqy/wqy-zenhei.ttc",
|
"/usr/share/fonts/truetype/wqy/wqy-zenhei.ttc",
|
||||||
"/usr/share/fonts/opentype/dejavu/DejaVuSans.ttf",
|
"/usr/share/fonts/truetype/dejavu/DejaVuSans.ttf",
|
||||||
nullptr};
|
nullptr};
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
@@ -1439,10 +1510,8 @@ int Render::DrawStreamWindow() {
|
|||||||
auto props = it.second;
|
auto props = it.second;
|
||||||
if (props->tab_selected_) {
|
if (props->tab_selected_) {
|
||||||
SDL_FRect render_rect_f = {
|
SDL_FRect render_rect_f = {
|
||||||
static_cast<float>(props->stream_render_rect_.x),
|
props->stream_render_rect_f_.x, props->stream_render_rect_f_.y,
|
||||||
static_cast<float>(props->stream_render_rect_.y),
|
props->stream_render_rect_f_.w, props->stream_render_rect_f_.h};
|
||||||
static_cast<float>(props->stream_render_rect_.w),
|
|
||||||
static_cast<float>(props->stream_render_rect_.h)};
|
|
||||||
SDL_RenderTexture(stream_renderer_, props->stream_texture_, NULL,
|
SDL_RenderTexture(stream_renderer_, props->stream_texture_, NULL,
|
||||||
&render_rect_f);
|
&render_rect_f);
|
||||||
}
|
}
|
||||||
@@ -1850,6 +1919,12 @@ void Render::HandleServerWindow() {
|
|||||||
void Render::Cleanup() {
|
void Render::Cleanup() {
|
||||||
Clipboard::StopMonitoring();
|
Clipboard::StopMonitoring();
|
||||||
|
|
||||||
|
if (mouse_controller_) {
|
||||||
|
mouse_controller_->Destroy();
|
||||||
|
delete mouse_controller_;
|
||||||
|
mouse_controller_ = nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
if (screen_capturer_) {
|
if (screen_capturer_) {
|
||||||
screen_capturer_->Destroy();
|
screen_capturer_->Destroy();
|
||||||
delete screen_capturer_;
|
delete screen_capturer_;
|
||||||
@@ -1862,12 +1937,6 @@ void Render::Cleanup() {
|
|||||||
speaker_capturer_ = nullptr;
|
speaker_capturer_ = nullptr;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (mouse_controller_) {
|
|
||||||
mouse_controller_->Destroy();
|
|
||||||
delete mouse_controller_;
|
|
||||||
mouse_controller_ = nullptr;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (keyboard_capturer_) {
|
if (keyboard_capturer_) {
|
||||||
delete keyboard_capturer_;
|
delete keyboard_capturer_;
|
||||||
keyboard_capturer_ = nullptr;
|
keyboard_capturer_ = nullptr;
|
||||||
@@ -1949,9 +2018,9 @@ void Render::CleanupPeers() {
|
|||||||
LOG_INFO("[{}] Leave connection [{}]", client_id_, client_id_);
|
LOG_INFO("[{}] Leave connection [{}]", client_id_, client_id_);
|
||||||
LeaveConnection(peer_, client_id_);
|
LeaveConnection(peer_, client_id_);
|
||||||
is_client_mode_ = false;
|
is_client_mode_ = false;
|
||||||
|
StopMouseController();
|
||||||
StopScreenCapturer();
|
StopScreenCapturer();
|
||||||
StopSpeakerCapturer();
|
StopSpeakerCapturer();
|
||||||
StopMouseController();
|
|
||||||
StopKeyboardCapturer();
|
StopKeyboardCapturer();
|
||||||
LOG_INFO("Destroy peer [{}]", client_id_);
|
LOG_INFO("Destroy peer [{}]", client_id_);
|
||||||
DestroyPeer(&peer_);
|
DestroyPeer(&peer_);
|
||||||
@@ -2229,26 +2298,36 @@ void Render::UpdateRenderRect() {
|
|||||||
float render_area_height = props->render_window_height_;
|
float render_area_height = props->render_window_height_;
|
||||||
|
|
||||||
props->stream_render_rect_last_ = props->stream_render_rect_;
|
props->stream_render_rect_last_ = props->stream_render_rect_;
|
||||||
|
|
||||||
|
SDL_FRect rect_f{props->render_window_x_, props->render_window_y_,
|
||||||
|
render_area_width, render_area_height};
|
||||||
if (render_area_width < render_area_height * video_ratio) {
|
if (render_area_width < render_area_height * video_ratio) {
|
||||||
props->stream_render_rect_ = {
|
rect_f.x = props->render_window_x_;
|
||||||
(int)props->render_window_x_,
|
rect_f.y = std::abs(render_area_height -
|
||||||
(int)(abs(render_area_height -
|
render_area_width * video_ratio_reverse) /
|
||||||
render_area_width * video_ratio_reverse) /
|
2.0f +
|
||||||
2 +
|
props->render_window_y_;
|
||||||
(int)props->render_window_y_),
|
rect_f.w = render_area_width;
|
||||||
(int)render_area_width,
|
rect_f.h = render_area_width * video_ratio_reverse;
|
||||||
(int)(render_area_width * video_ratio_reverse)};
|
|
||||||
} else if (render_area_width > render_area_height * video_ratio) {
|
} else if (render_area_width > render_area_height * video_ratio) {
|
||||||
props->stream_render_rect_ = {
|
rect_f.x =
|
||||||
(int)abs(render_area_width - render_area_height * video_ratio) / 2 +
|
std::abs(render_area_width - render_area_height * video_ratio) / 2.0f +
|
||||||
(int)props->render_window_x_,
|
props->render_window_x_;
|
||||||
(int)props->render_window_y_, (int)(render_area_height * video_ratio),
|
rect_f.y = props->render_window_y_;
|
||||||
(int)render_area_height};
|
rect_f.w = render_area_height * video_ratio;
|
||||||
|
rect_f.h = render_area_height;
|
||||||
} else {
|
} else {
|
||||||
props->stream_render_rect_ = {
|
rect_f.x = props->render_window_x_;
|
||||||
(int)props->render_window_x_, (int)props->render_window_y_,
|
rect_f.y = props->render_window_y_;
|
||||||
(int)render_area_width, (int)render_area_height};
|
rect_f.w = render_area_width;
|
||||||
|
rect_f.h = render_area_height;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
props->stream_render_rect_f_ = rect_f;
|
||||||
|
props->stream_render_rect_ = {static_cast<int>(std::lround(rect_f.x)),
|
||||||
|
static_cast<int>(std::lround(rect_f.y)),
|
||||||
|
static_cast<int>(std::lround(rect_f.w)),
|
||||||
|
static_cast<int>(std::lround(rect_f.h))};
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -2376,6 +2455,7 @@ void Render::ProcessSdlEvent(const SDL_Event& event) {
|
|||||||
case SDL_EVENT_WINDOW_FOCUS_LOST:
|
case SDL_EVENT_WINDOW_FOCUS_LOST:
|
||||||
if (stream_window_ &&
|
if (stream_window_ &&
|
||||||
SDL_GetWindowID(stream_window_) == event.window.windowID) {
|
SDL_GetWindowID(stream_window_) == event.window.windowID) {
|
||||||
|
ForceReleasePressedModifiers();
|
||||||
focus_on_stream_window_ = false;
|
focus_on_stream_window_ = false;
|
||||||
} else if (main_window_ &&
|
} else if (main_window_ &&
|
||||||
SDL_GetWindowID(main_window_) == event.window.windowID) {
|
SDL_GetWindowID(main_window_) == event.window.windowID) {
|
||||||
@@ -2389,12 +2469,23 @@ void Render::ProcessSdlEvent(const SDL_Event& event) {
|
|||||||
case SDL_EVENT_MOUSE_MOTION:
|
case SDL_EVENT_MOUSE_MOTION:
|
||||||
case SDL_EVENT_MOUSE_BUTTON_DOWN:
|
case SDL_EVENT_MOUSE_BUTTON_DOWN:
|
||||||
case SDL_EVENT_MOUSE_BUTTON_UP:
|
case SDL_EVENT_MOUSE_BUTTON_UP:
|
||||||
case SDL_EVENT_MOUSE_WHEEL:
|
case SDL_EVENT_MOUSE_WHEEL: {
|
||||||
|
Uint32 mouse_window_id = 0;
|
||||||
|
if (event.type == SDL_EVENT_MOUSE_MOTION) {
|
||||||
|
mouse_window_id = event.motion.windowID;
|
||||||
|
} else if (event.type == SDL_EVENT_MOUSE_BUTTON_DOWN ||
|
||||||
|
event.type == SDL_EVENT_MOUSE_BUTTON_UP) {
|
||||||
|
mouse_window_id = event.button.windowID;
|
||||||
|
} else if (event.type == SDL_EVENT_MOUSE_WHEEL) {
|
||||||
|
mouse_window_id = event.wheel.windowID;
|
||||||
|
}
|
||||||
|
|
||||||
if (focus_on_stream_window_ && stream_window_ &&
|
if (focus_on_stream_window_ && stream_window_ &&
|
||||||
SDL_GetWindowID(stream_window_) == event.motion.windowID) {
|
SDL_GetWindowID(stream_window_) == mouse_window_id) {
|
||||||
ProcessMouseEvent(event);
|
ProcessMouseEvent(event);
|
||||||
}
|
}
|
||||||
break;
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
default:
|
default:
|
||||||
if (event.type == STREAM_REFRESH_EVENT) {
|
if (event.type == STREAM_REFRESH_EVENT) {
|
||||||
|
|||||||
@@ -20,6 +20,7 @@
|
|||||||
#include <shared_mutex>
|
#include <shared_mutex>
|
||||||
#include <string>
|
#include <string>
|
||||||
#include <unordered_map>
|
#include <unordered_map>
|
||||||
|
#include <unordered_set>
|
||||||
#include <vector>
|
#include <vector>
|
||||||
|
|
||||||
#include "IconsFontAwesome6.h"
|
#include "IconsFontAwesome6.h"
|
||||||
@@ -83,6 +84,8 @@ class Render {
|
|||||||
PeerPtr* peer_ = nullptr;
|
PeerPtr* peer_ = nullptr;
|
||||||
std::string audio_label_ = "control_audio";
|
std::string audio_label_ = "control_audio";
|
||||||
std::string data_label_ = "data";
|
std::string data_label_ = "data";
|
||||||
|
std::string mouse_label_ = "mouse";
|
||||||
|
std::string keyboard_label_ = "keyboard";
|
||||||
std::string file_label_ = "file";
|
std::string file_label_ = "file";
|
||||||
std::string control_data_label_ = "control_data";
|
std::string control_data_label_ = "control_data";
|
||||||
std::string file_feedback_label_ = "file_feedback";
|
std::string file_feedback_label_ = "file_feedback";
|
||||||
@@ -160,6 +163,7 @@ class Render {
|
|||||||
SDL_Texture* stream_texture_ = nullptr;
|
SDL_Texture* stream_texture_ = nullptr;
|
||||||
uint8_t* argb_buffer_ = nullptr;
|
uint8_t* argb_buffer_ = nullptr;
|
||||||
int argb_buffer_size_ = 0;
|
int argb_buffer_size_ = 0;
|
||||||
|
SDL_FRect stream_render_rect_f_ = {0.0f, 0.0f, 0.0f, 0.0f};
|
||||||
SDL_Rect stream_render_rect_;
|
SDL_Rect stream_render_rect_;
|
||||||
SDL_Rect stream_render_rect_last_;
|
SDL_Rect stream_render_rect_last_;
|
||||||
ImVec2 control_window_pos_;
|
ImVec2 control_window_pos_;
|
||||||
@@ -319,6 +323,9 @@ class Render {
|
|||||||
|
|
||||||
private:
|
private:
|
||||||
int SendKeyCommand(int key_code, bool is_down);
|
int SendKeyCommand(int key_code, bool is_down);
|
||||||
|
static bool IsModifierVkKey(int key_code);
|
||||||
|
void UpdatePressedModifierState(int key_code, bool is_down);
|
||||||
|
void ForceReleasePressedModifiers();
|
||||||
int ProcessMouseEvent(const SDL_Event& event);
|
int ProcessMouseEvent(const SDL_Event& event);
|
||||||
|
|
||||||
static void SdlCaptureAudioIn(void* userdata, Uint8* stream, int len);
|
static void SdlCaptureAudioIn(void* userdata, Uint8* stream, int len);
|
||||||
@@ -503,6 +510,8 @@ class Render {
|
|||||||
std::string controlled_remote_id_ = "";
|
std::string controlled_remote_id_ = "";
|
||||||
std::string focused_remote_id_ = "";
|
std::string focused_remote_id_ = "";
|
||||||
std::string remote_client_id_ = "";
|
std::string remote_client_id_ = "";
|
||||||
|
std::unordered_set<int> pressed_modifier_keys_;
|
||||||
|
std::mutex pressed_modifier_keys_mutex_;
|
||||||
SDL_Event last_mouse_event;
|
SDL_Event last_mouse_event;
|
||||||
SDL_AudioStream* output_stream_;
|
SDL_AudioStream* output_stream_;
|
||||||
uint32_t STREAM_REFRESH_EVENT = 0;
|
uint32_t STREAM_REFRESH_EVENT = 0;
|
||||||
@@ -602,6 +611,8 @@ class Render {
|
|||||||
std::string video_secondary_label_ = "secondary_display";
|
std::string video_secondary_label_ = "secondary_display";
|
||||||
std::string audio_label_ = "audio";
|
std::string audio_label_ = "audio";
|
||||||
std::string data_label_ = "data";
|
std::string data_label_ = "data";
|
||||||
|
std::string mouse_label_ = "mouse";
|
||||||
|
std::string keyboard_label_ = "keyboard";
|
||||||
std::string info_label_ = "info";
|
std::string info_label_ = "info";
|
||||||
std::string control_data_label_ = "control_data";
|
std::string control_data_label_ = "control_data";
|
||||||
std::string file_label_ = "file";
|
std::string file_label_ = "file";
|
||||||
|
|||||||
@@ -1,3 +1,4 @@
|
|||||||
|
#include <algorithm>
|
||||||
#include <chrono>
|
#include <chrono>
|
||||||
#include <cmath>
|
#include <cmath>
|
||||||
#include <cstdlib>
|
#include <cstdlib>
|
||||||
@@ -80,6 +81,55 @@ void Render::OnSignalMessageCb(const char* message, size_t size,
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
bool Render::IsModifierVkKey(int key_code) {
|
||||||
|
switch (key_code) {
|
||||||
|
case 0x10: // VK_SHIFT
|
||||||
|
case 0x11: // VK_CONTROL
|
||||||
|
case 0x12: // VK_MENU(ALT)
|
||||||
|
case 0x5B: // VK_LWIN
|
||||||
|
case 0x5C: // VK_RWIN
|
||||||
|
case 0xA0: // VK_LSHIFT
|
||||||
|
case 0xA1: // VK_RSHIFT
|
||||||
|
case 0xA2: // VK_LCONTROL
|
||||||
|
case 0xA3: // VK_RCONTROL
|
||||||
|
case 0xA4: // VK_LMENU
|
||||||
|
case 0xA5: // VK_RMENU
|
||||||
|
return true;
|
||||||
|
default:
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void Render::UpdatePressedModifierState(int key_code, bool is_down) {
|
||||||
|
if (!IsModifierVkKey(key_code)) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::lock_guard<std::mutex> lock(pressed_modifier_keys_mutex_);
|
||||||
|
if (is_down) {
|
||||||
|
pressed_modifier_keys_.insert(key_code);
|
||||||
|
} else {
|
||||||
|
pressed_modifier_keys_.erase(key_code);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void Render::ForceReleasePressedModifiers() {
|
||||||
|
std::vector<int> pressed_keys;
|
||||||
|
{
|
||||||
|
std::lock_guard<std::mutex> lock(pressed_modifier_keys_mutex_);
|
||||||
|
if (pressed_modifier_keys_.empty()) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
pressed_keys.assign(pressed_modifier_keys_.begin(),
|
||||||
|
pressed_modifier_keys_.end());
|
||||||
|
pressed_modifier_keys_.clear();
|
||||||
|
}
|
||||||
|
|
||||||
|
for (int key_code : pressed_keys) {
|
||||||
|
SendKeyCommand(key_code, false);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
int Render::SendKeyCommand(int key_code, bool is_down) {
|
int Render::SendKeyCommand(int key_code, bool is_down) {
|
||||||
RemoteAction remote_action;
|
RemoteAction remote_action;
|
||||||
remote_action.type = ControlType::keyboard;
|
remote_action.type = ControlType::keyboard;
|
||||||
@@ -98,21 +148,84 @@ int Render::SendKeyCommand(int key_code, bool is_down) {
|
|||||||
if (props->connection_status_ == ConnectionStatus::Connected &&
|
if (props->connection_status_ == ConnectionStatus::Connected &&
|
||||||
props->peer_) {
|
props->peer_) {
|
||||||
std::string msg = remote_action.to_json();
|
std::string msg = remote_action.to_json();
|
||||||
SendDataFrame(props->peer_, msg.c_str(), msg.size(),
|
int ret = SendReliableDataFrame(props->peer_, msg.c_str(), msg.size(),
|
||||||
props->data_label_.c_str());
|
props->keyboard_label_.c_str());
|
||||||
|
if (ret != 0) {
|
||||||
|
LOG_WARN("Send keyboard command failed, remote_id={}, ret={}",
|
||||||
|
target_id, ret);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
UpdatePressedModifierState(key_code, is_down);
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
int Render::ProcessMouseEvent(const SDL_Event& event) {
|
int Render::ProcessMouseEvent(const SDL_Event& event) {
|
||||||
controlled_remote_id_ = "";
|
controlled_remote_id_ = "";
|
||||||
int video_width, video_height = 0;
|
|
||||||
int render_width, render_height = 0;
|
|
||||||
float ratio_x, ratio_y = 0;
|
|
||||||
RemoteAction remote_action;
|
RemoteAction remote_action;
|
||||||
|
float cursor_x = last_mouse_event.motion.x;
|
||||||
|
float cursor_y = last_mouse_event.motion.y;
|
||||||
|
|
||||||
|
auto normalize_cursor_to_window_space = [&](float* x, float* y) {
|
||||||
|
if (!x || !y || !stream_window_) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
int window_width = 0;
|
||||||
|
int window_height = 0;
|
||||||
|
int pixel_width = 0;
|
||||||
|
int pixel_height = 0;
|
||||||
|
SDL_GetWindowSize(stream_window_, &window_width, &window_height);
|
||||||
|
SDL_GetWindowSizeInPixels(stream_window_, &pixel_width, &pixel_height);
|
||||||
|
|
||||||
|
if (window_width <= 0 || window_height <= 0 || pixel_width <= 0 ||
|
||||||
|
pixel_height <= 0) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if ((window_width != pixel_width || window_height != pixel_height) &&
|
||||||
|
(*x > static_cast<float>(window_width) + 1.0f ||
|
||||||
|
*y > static_cast<float>(window_height) + 1.0f)) {
|
||||||
|
const float scale_x =
|
||||||
|
static_cast<float>(window_width) / static_cast<float>(pixel_width);
|
||||||
|
const float scale_y =
|
||||||
|
static_cast<float>(window_height) / static_cast<float>(pixel_height);
|
||||||
|
*x *= scale_x;
|
||||||
|
*y *= scale_y;
|
||||||
|
|
||||||
|
static bool logged_pixel_to_window_conversion = false;
|
||||||
|
if (!logged_pixel_to_window_conversion) {
|
||||||
|
LOG_INFO(
|
||||||
|
"Mouse coordinate space converted from pixels to window units: "
|
||||||
|
"window={}x{}, pixels={}x{}, scale=({:.4f},{:.4f})",
|
||||||
|
window_width, window_height, pixel_width, pixel_height, scale_x,
|
||||||
|
scale_y);
|
||||||
|
logged_pixel_to_window_conversion = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
if (event.type == SDL_EVENT_MOUSE_MOTION) {
|
||||||
|
cursor_x = event.motion.x;
|
||||||
|
cursor_y = event.motion.y;
|
||||||
|
normalize_cursor_to_window_space(&cursor_x, &cursor_y);
|
||||||
|
} else if (event.type == SDL_EVENT_MOUSE_BUTTON_DOWN ||
|
||||||
|
event.type == SDL_EVENT_MOUSE_BUTTON_UP) {
|
||||||
|
cursor_x = event.button.x;
|
||||||
|
cursor_y = event.button.y;
|
||||||
|
normalize_cursor_to_window_space(&cursor_x, &cursor_y);
|
||||||
|
} else if (event.type == SDL_EVENT_MOUSE_WHEEL) {
|
||||||
|
cursor_x = last_mouse_event.motion.x;
|
||||||
|
cursor_y = last_mouse_event.motion.y;
|
||||||
|
}
|
||||||
|
|
||||||
|
const bool is_pointer_position_event =
|
||||||
|
(event.type == SDL_EVENT_MOUSE_MOTION ||
|
||||||
|
event.type == SDL_EVENT_MOUSE_BUTTON_DOWN ||
|
||||||
|
event.type == SDL_EVENT_MOUSE_BUTTON_UP);
|
||||||
|
|
||||||
// std::shared_lock lock(client_properties_mutex_);
|
// std::shared_lock lock(client_properties_mutex_);
|
||||||
for (auto& it : client_properties_) {
|
for (auto& it : client_properties_) {
|
||||||
@@ -121,23 +234,24 @@ int Render::ProcessMouseEvent(const SDL_Event& event) {
|
|||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (event.button.x >= props->stream_render_rect_.x &&
|
const SDL_FRect render_rect = props->stream_render_rect_f_;
|
||||||
event.button.x <=
|
if (render_rect.w <= 1.0f || render_rect.h <= 1.0f) {
|
||||||
props->stream_render_rect_.x + props->stream_render_rect_.w &&
|
continue;
|
||||||
event.button.y >= props->stream_render_rect_.y &&
|
}
|
||||||
event.button.y <=
|
|
||||||
props->stream_render_rect_.y + props->stream_render_rect_.h) {
|
|
||||||
controlled_remote_id_ = it.first;
|
|
||||||
render_width = props->stream_render_rect_.w;
|
|
||||||
render_height = props->stream_render_rect_.h;
|
|
||||||
last_mouse_event.button.x = event.button.x;
|
|
||||||
last_mouse_event.button.y = event.button.y;
|
|
||||||
|
|
||||||
remote_action.m.x =
|
if (is_pointer_position_event && cursor_x >= render_rect.x &&
|
||||||
(float)(event.button.x - props->stream_render_rect_.x) / render_width;
|
cursor_x <= render_rect.x + render_rect.w && cursor_y >= render_rect.y &&
|
||||||
remote_action.m.y =
|
cursor_y <= render_rect.y + render_rect.h) {
|
||||||
(float)(event.button.y - props->stream_render_rect_.y) /
|
controlled_remote_id_ = it.first;
|
||||||
render_height;
|
last_mouse_event.motion.x = cursor_x;
|
||||||
|
last_mouse_event.motion.y = cursor_y;
|
||||||
|
last_mouse_event.button.x = cursor_x;
|
||||||
|
last_mouse_event.button.y = cursor_y;
|
||||||
|
|
||||||
|
remote_action.m.x = (cursor_x - render_rect.x) / render_rect.w;
|
||||||
|
remote_action.m.y = (cursor_y - render_rect.y) / render_rect.h;
|
||||||
|
remote_action.m.x = std::clamp(remote_action.m.x, 0.0f, 1.0f);
|
||||||
|
remote_action.m.y = std::clamp(remote_action.m.y, 0.0f, 1.0f);
|
||||||
|
|
||||||
if (SDL_EVENT_MOUSE_BUTTON_DOWN == event.type) {
|
if (SDL_EVENT_MOUSE_BUTTON_DOWN == event.type) {
|
||||||
remote_action.type = ControlType::mouse;
|
remote_action.type = ControlType::mouse;
|
||||||
@@ -168,15 +282,13 @@ int Render::ProcessMouseEvent(const SDL_Event& event) {
|
|||||||
if (props->peer_) {
|
if (props->peer_) {
|
||||||
std::string msg = remote_action.to_json();
|
std::string msg = remote_action.to_json();
|
||||||
SendDataFrame(props->peer_, msg.c_str(), msg.size(),
|
SendDataFrame(props->peer_, msg.c_str(), msg.size(),
|
||||||
props->data_label_.c_str());
|
props->mouse_label_.c_str());
|
||||||
}
|
}
|
||||||
} else if (SDL_EVENT_MOUSE_WHEEL == event.type &&
|
} else if (SDL_EVENT_MOUSE_WHEEL == event.type &&
|
||||||
last_mouse_event.button.x >= props->stream_render_rect_.x &&
|
last_mouse_event.button.x >= render_rect.x &&
|
||||||
last_mouse_event.button.x <= props->stream_render_rect_.x +
|
last_mouse_event.button.x <= render_rect.x + render_rect.w &&
|
||||||
props->stream_render_rect_.w &&
|
last_mouse_event.button.y >= render_rect.y &&
|
||||||
last_mouse_event.button.y >= props->stream_render_rect_.y &&
|
last_mouse_event.button.y <= render_rect.y + render_rect.h) {
|
||||||
last_mouse_event.button.y <= props->stream_render_rect_.y +
|
|
||||||
props->stream_render_rect_.h) {
|
|
||||||
float scroll_x = event.wheel.x;
|
float scroll_x = event.wheel.x;
|
||||||
float scroll_y = event.wheel.y;
|
float scroll_y = event.wheel.y;
|
||||||
if (event.wheel.direction == SDL_MOUSEWHEEL_FLIPPED) {
|
if (event.wheel.direction == SDL_MOUSEWHEEL_FLIPPED) {
|
||||||
@@ -203,14 +315,12 @@ int Render::ProcessMouseEvent(const SDL_Event& event) {
|
|||||||
remote_action.m.s = roundUp(scroll_x);
|
remote_action.m.s = roundUp(scroll_x);
|
||||||
}
|
}
|
||||||
|
|
||||||
render_width = props->stream_render_rect_.w;
|
remote_action.m.x = (last_mouse_event.button.x - render_rect.x) /
|
||||||
render_height = props->stream_render_rect_.h;
|
(std::max)(render_rect.w, 1.0f);
|
||||||
remote_action.m.x =
|
remote_action.m.y = (last_mouse_event.button.y - render_rect.y) /
|
||||||
(float)(last_mouse_event.button.x - props->stream_render_rect_.x) /
|
(std::max)(render_rect.h, 1.0f);
|
||||||
render_width;
|
remote_action.m.x = std::clamp(remote_action.m.x, 0.0f, 1.0f);
|
||||||
remote_action.m.y =
|
remote_action.m.y = std::clamp(remote_action.m.y, 0.0f, 1.0f);
|
||||||
(float)(last_mouse_event.button.y - props->stream_render_rect_.y) /
|
|
||||||
render_height;
|
|
||||||
|
|
||||||
if (props->control_bar_hovered_) {
|
if (props->control_bar_hovered_) {
|
||||||
continue;
|
continue;
|
||||||
@@ -218,7 +328,7 @@ int Render::ProcessMouseEvent(const SDL_Event& event) {
|
|||||||
if (props->peer_) {
|
if (props->peer_) {
|
||||||
std::string msg = remote_action.to_json();
|
std::string msg = remote_action.to_json();
|
||||||
SendDataFrame(props->peer_, msg.c_str(), msg.size(),
|
SendDataFrame(props->peer_, msg.c_str(), msg.size(),
|
||||||
props->data_label_.c_str());
|
props->mouse_label_.c_str());
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -783,6 +893,9 @@ void Render::OnConnectionStatusCb(ConnectionStatus status, const char* user_id,
|
|||||||
0, (int)render->title_bar_height_,
|
0, (int)render->title_bar_height_,
|
||||||
(int)render->stream_window_width_,
|
(int)render->stream_window_width_,
|
||||||
(int)(render->stream_window_height_ - render->title_bar_height_)};
|
(int)(render->stream_window_height_ - render->title_bar_height_)};
|
||||||
|
props->stream_render_rect_f_ = {
|
||||||
|
0.0f, render->title_bar_height_, render->stream_window_width_,
|
||||||
|
render->stream_window_height_ - render->title_bar_height_};
|
||||||
render->start_keyboard_capturer_ = true;
|
render->start_keyboard_capturer_ = true;
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
@@ -910,7 +1023,19 @@ void Render::OnConnectionStatusCb(ConnectionStatus status, const char* user_id,
|
|||||||
})) {
|
})) {
|
||||||
render->need_to_destroy_server_window_ = true;
|
render->need_to_destroy_server_window_ = true;
|
||||||
render->is_server_mode_ = false;
|
render->is_server_mode_ = false;
|
||||||
|
#if defined(__linux__) && !defined(__APPLE__)
|
||||||
|
if (IsWaylandSession()) {
|
||||||
|
// Keep Wayland capture session warm to avoid black screen on
|
||||||
|
// subsequent reconnects.
|
||||||
|
render->start_screen_capturer_ = true;
|
||||||
|
LOG_INFO("Keeping Wayland screen capturer running after "
|
||||||
|
"disconnect to preserve reconnect stability");
|
||||||
|
} else {
|
||||||
|
render->start_screen_capturer_ = false;
|
||||||
|
}
|
||||||
|
#else
|
||||||
render->start_screen_capturer_ = false;
|
render->start_screen_capturer_ = false;
|
||||||
|
#endif
|
||||||
render->start_speaker_capturer_ = false;
|
render->start_speaker_capturer_ = false;
|
||||||
render->start_mouse_controller_ = false;
|
render->start_mouse_controller_ = false;
|
||||||
render->start_keyboard_capturer_ = false;
|
render->start_keyboard_capturer_ = false;
|
||||||
@@ -1074,4 +1199,4 @@ void Render::OnNetStatusReport(const char* client_id, size_t client_id_size,
|
|||||||
props->net_traffic_stats_ = *net_traffic_stats;
|
props->net_traffic_stats_ = *net_traffic_stats;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
} // namespace crossdesk
|
} // namespace crossdesk
|
||||||
|
|||||||
573
src/screen_capturer/linux/screen_capturer_drm.cpp
Normal file
573
src/screen_capturer/linux/screen_capturer_drm.cpp
Normal file
@@ -0,0 +1,573 @@
|
|||||||
|
#include "screen_capturer_drm.h"
|
||||||
|
|
||||||
|
#if defined(CROSSDESK_HAS_DRM) && CROSSDESK_HAS_DRM && \
|
||||||
|
defined(__has_include) && __has_include(<xf86drm.h>) && \
|
||||||
|
__has_include(<xf86drmMode.h>)
|
||||||
|
#define CROSSDESK_DRM_BUILD_ENABLED 1
|
||||||
|
#include <xf86drm.h>
|
||||||
|
#include <xf86drmMode.h>
|
||||||
|
#elif defined(CROSSDESK_HAS_DRM) && CROSSDESK_HAS_DRM && \
|
||||||
|
defined(__has_include) && __has_include(<libdrm/xf86drm.h>) && \
|
||||||
|
__has_include(<libdrm/xf86drmMode.h>)
|
||||||
|
#define CROSSDESK_DRM_BUILD_ENABLED 1
|
||||||
|
#include <libdrm/xf86drm.h>
|
||||||
|
#include <libdrm/xf86drmMode.h>
|
||||||
|
#else
|
||||||
|
#define CROSSDESK_DRM_BUILD_ENABLED 0
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#if CROSSDESK_DRM_BUILD_ENABLED
|
||||||
|
|
||||||
|
#include <fcntl.h>
|
||||||
|
#include <sys/mman.h>
|
||||||
|
#include <unistd.h>
|
||||||
|
|
||||||
|
#include <algorithm>
|
||||||
|
#include <chrono>
|
||||||
|
#include <thread>
|
||||||
|
|
||||||
|
#include "libyuv.h"
|
||||||
|
#include "rd_log.h"
|
||||||
|
|
||||||
|
namespace crossdesk {
|
||||||
|
|
||||||
|
namespace {
|
||||||
|
|
||||||
|
constexpr int kMaxDrmCards = 16;
|
||||||
|
|
||||||
|
const char* ConnectorTypeName(uint32_t type) {
|
||||||
|
switch (type) {
|
||||||
|
case DRM_MODE_CONNECTOR_VGA:
|
||||||
|
return "VGA";
|
||||||
|
case DRM_MODE_CONNECTOR_DVII:
|
||||||
|
return "DVI-I";
|
||||||
|
case DRM_MODE_CONNECTOR_DVID:
|
||||||
|
return "DVI-D";
|
||||||
|
case DRM_MODE_CONNECTOR_DVIA:
|
||||||
|
return "DVI-A";
|
||||||
|
case DRM_MODE_CONNECTOR_HDMIA:
|
||||||
|
return "HDMI-A";
|
||||||
|
case DRM_MODE_CONNECTOR_HDMIB:
|
||||||
|
return "HDMI-B";
|
||||||
|
case DRM_MODE_CONNECTOR_DisplayPort:
|
||||||
|
return "DP";
|
||||||
|
case DRM_MODE_CONNECTOR_eDP:
|
||||||
|
return "eDP";
|
||||||
|
case DRM_MODE_CONNECTOR_LVDS:
|
||||||
|
return "LVDS";
|
||||||
|
#ifdef DRM_MODE_CONNECTOR_VIRTUAL
|
||||||
|
case DRM_MODE_CONNECTOR_VIRTUAL:
|
||||||
|
return "Virtual";
|
||||||
|
#endif
|
||||||
|
default:
|
||||||
|
return "Display";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace
|
||||||
|
|
||||||
|
ScreenCapturerDrm::ScreenCapturerDrm() {}
|
||||||
|
|
||||||
|
ScreenCapturerDrm::~ScreenCapturerDrm() { Destroy(); }
|
||||||
|
|
||||||
|
int ScreenCapturerDrm::Init(const int fps, cb_desktop_data cb) {
|
||||||
|
Destroy();
|
||||||
|
|
||||||
|
if (!cb) {
|
||||||
|
LOG_ERROR("DRM screen capturer callback is null");
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
fps_ = std::max(1, fps);
|
||||||
|
callback_ = cb;
|
||||||
|
monitor_index_ = 0;
|
||||||
|
initial_monitor_index_ = 0;
|
||||||
|
consecutive_failures_ = 0;
|
||||||
|
display_info_list_.clear();
|
||||||
|
outputs_.clear();
|
||||||
|
y_plane_.clear();
|
||||||
|
uv_plane_.clear();
|
||||||
|
|
||||||
|
if (!DiscoverOutputs()) {
|
||||||
|
LOG_ERROR("DRM screen capturer could not find active outputs");
|
||||||
|
callback_ = nullptr;
|
||||||
|
CloseDevices();
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int ScreenCapturerDrm::Destroy() {
|
||||||
|
Stop();
|
||||||
|
callback_ = nullptr;
|
||||||
|
display_info_list_.clear();
|
||||||
|
outputs_.clear();
|
||||||
|
y_plane_.clear();
|
||||||
|
uv_plane_.clear();
|
||||||
|
CloseDevices();
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int ScreenCapturerDrm::Start(bool show_cursor) {
|
||||||
|
if (running_) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (outputs_.empty()) {
|
||||||
|
LOG_ERROR("DRM screen capturer has no output to capture");
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
show_cursor_ = show_cursor;
|
||||||
|
paused_ = false;
|
||||||
|
|
||||||
|
int probe_index = monitor_index_.load();
|
||||||
|
if (probe_index < 0 || probe_index >= static_cast<int>(outputs_.size())) {
|
||||||
|
probe_index = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!CaptureOutputFrame(outputs_[probe_index], false)) {
|
||||||
|
LOG_ERROR("DRM start probe failed on output {}", outputs_[probe_index].name);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
running_ = true;
|
||||||
|
thread_ = std::thread([this]() { CaptureLoop(); });
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int ScreenCapturerDrm::Stop() {
|
||||||
|
if (!running_) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
running_ = false;
|
||||||
|
if (thread_.joinable()) {
|
||||||
|
thread_.join();
|
||||||
|
}
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int ScreenCapturerDrm::Pause([[maybe_unused]] int monitor_index) {
|
||||||
|
paused_ = true;
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int ScreenCapturerDrm::Resume([[maybe_unused]] int monitor_index) {
|
||||||
|
paused_ = false;
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int ScreenCapturerDrm::SwitchTo(int monitor_index) {
|
||||||
|
if (monitor_index < 0 ||
|
||||||
|
monitor_index >= static_cast<int>(display_info_list_.size())) {
|
||||||
|
LOG_ERROR("Invalid DRM monitor index: {}", monitor_index);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
monitor_index_ = monitor_index;
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int ScreenCapturerDrm::ResetToInitialMonitor() {
|
||||||
|
monitor_index_ = initial_monitor_index_;
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::vector<DisplayInfo> ScreenCapturerDrm::GetDisplayInfoList() {
|
||||||
|
return display_info_list_;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ScreenCapturerDrm::DiscoverOutputs() {
|
||||||
|
for (int card_index = 0; card_index < kMaxDrmCards; ++card_index) {
|
||||||
|
const std::string card_path = "/dev/dri/card" + std::to_string(card_index);
|
||||||
|
const int fd = open(card_path.c_str(), O_RDWR | O_CLOEXEC);
|
||||||
|
if (fd < 0) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
drmModeRes* resources = drmModeGetResources(fd);
|
||||||
|
if (!resources) {
|
||||||
|
close(fd);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
DrmDevice device;
|
||||||
|
device.fd = fd;
|
||||||
|
device.path = card_path;
|
||||||
|
devices_.push_back(device);
|
||||||
|
const int device_slot = static_cast<int>(devices_.size()) - 1;
|
||||||
|
const size_t output_count_before = outputs_.size();
|
||||||
|
|
||||||
|
for (int i = 0; i < resources->count_connectors; ++i) {
|
||||||
|
drmModeConnector* connector =
|
||||||
|
drmModeGetConnector(fd, resources->connectors[i]);
|
||||||
|
if (!connector) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (connector->connection != DRM_MODE_CONNECTED ||
|
||||||
|
connector->count_modes <= 0) {
|
||||||
|
drmModeFreeConnector(connector);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
uint32_t crtc_id = 0;
|
||||||
|
if (connector->encoder_id != 0) {
|
||||||
|
drmModeEncoder* encoder = drmModeGetEncoder(fd, connector->encoder_id);
|
||||||
|
if (encoder) {
|
||||||
|
crtc_id = encoder->crtc_id;
|
||||||
|
drmModeFreeEncoder(encoder);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (crtc_id == 0) {
|
||||||
|
for (int enc_idx = 0; enc_idx < connector->count_encoders; ++enc_idx) {
|
||||||
|
drmModeEncoder* encoder =
|
||||||
|
drmModeGetEncoder(fd, connector->encoders[enc_idx]);
|
||||||
|
if (!encoder) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
if (encoder->crtc_id != 0) {
|
||||||
|
crtc_id = encoder->crtc_id;
|
||||||
|
drmModeFreeEncoder(encoder);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
drmModeFreeEncoder(encoder);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (crtc_id == 0) {
|
||||||
|
drmModeFreeConnector(connector);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
drmModeCrtc* crtc = drmModeGetCrtc(fd, crtc_id);
|
||||||
|
if (!crtc || !crtc->mode_valid || crtc->width <= 0 || crtc->height <= 0) {
|
||||||
|
if (crtc) {
|
||||||
|
drmModeFreeCrtc(crtc);
|
||||||
|
}
|
||||||
|
drmModeFreeConnector(connector);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
DrmOutput output;
|
||||||
|
output.device_index = device_slot;
|
||||||
|
output.connector_id = connector->connector_id;
|
||||||
|
output.crtc_id = crtc_id;
|
||||||
|
output.left = crtc->x;
|
||||||
|
output.top = crtc->y;
|
||||||
|
output.width = static_cast<int>(crtc->width);
|
||||||
|
output.height = static_cast<int>(crtc->height);
|
||||||
|
output.name = std::string(ConnectorTypeName(connector->connector_type)) +
|
||||||
|
std::to_string(connector->connector_type_id);
|
||||||
|
|
||||||
|
outputs_.push_back(output);
|
||||||
|
display_info_list_.push_back(
|
||||||
|
DisplayInfo(output.name, output.left, output.top,
|
||||||
|
output.left + output.width, output.top + output.height));
|
||||||
|
|
||||||
|
LOG_INFO("DRM output found: {} on {}, {}x{} @ ({}, {})", output.name,
|
||||||
|
card_path, output.width, output.height, output.left, output.top);
|
||||||
|
|
||||||
|
drmModeFreeCrtc(crtc);
|
||||||
|
drmModeFreeConnector(connector);
|
||||||
|
}
|
||||||
|
|
||||||
|
drmModeFreeResources(resources);
|
||||||
|
|
||||||
|
if (outputs_.size() == output_count_before) {
|
||||||
|
close(fd);
|
||||||
|
devices_.pop_back();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (outputs_.empty()) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
LOG_INFO("DRM screen capturer discovered {} output(s)", outputs_.size());
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
void ScreenCapturerDrm::CloseDevices() {
|
||||||
|
for (auto& device : devices_) {
|
||||||
|
if (device.fd >= 0) {
|
||||||
|
close(device.fd);
|
||||||
|
device.fd = -1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
devices_.clear();
|
||||||
|
}
|
||||||
|
|
||||||
|
void ScreenCapturerDrm::CaptureLoop() {
|
||||||
|
using clock = std::chrono::steady_clock;
|
||||||
|
const auto frame_interval =
|
||||||
|
std::chrono::milliseconds(std::max(1, 1000 / std::max(1, fps_)));
|
||||||
|
|
||||||
|
while (running_) {
|
||||||
|
const auto frame_start = clock::now();
|
||||||
|
if (!paused_) {
|
||||||
|
int index = monitor_index_.load();
|
||||||
|
if (index >= 0 && index < static_cast<int>(outputs_.size())) {
|
||||||
|
const bool ok = CaptureOutputFrame(outputs_[index], true);
|
||||||
|
if (!ok) {
|
||||||
|
++consecutive_failures_;
|
||||||
|
if (consecutive_failures_ == 1 || consecutive_failures_ % 60 == 0) {
|
||||||
|
LOG_WARN("DRM capture failed (consecutive={})",
|
||||||
|
consecutive_failures_);
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
consecutive_failures_ = 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const auto elapsed = std::chrono::duration_cast<std::chrono::milliseconds>(
|
||||||
|
clock::now() - frame_start);
|
||||||
|
if (elapsed < frame_interval) {
|
||||||
|
std::this_thread::sleep_for(frame_interval - elapsed);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ScreenCapturerDrm::CaptureOutputFrame(const DrmOutput& output,
|
||||||
|
bool emit_callback) {
|
||||||
|
if (output.device_index < 0 ||
|
||||||
|
output.device_index >= static_cast<int>(devices_.size())) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
const int fd = devices_[output.device_index].fd;
|
||||||
|
if (fd < 0) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
drmModeCrtc* crtc = drmModeGetCrtc(fd, output.crtc_id);
|
||||||
|
if (!crtc) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
const uint32_t fb_id = crtc->buffer_id;
|
||||||
|
drmModeFreeCrtc(crtc);
|
||||||
|
if (fb_id == 0) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
drmModeFB* fb = drmModeGetFB(fd, fb_id);
|
||||||
|
if (!fb) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
const uint32_t handle = fb->handle;
|
||||||
|
const uint32_t pitch = fb->pitch;
|
||||||
|
const int src_width = static_cast<int>(fb->width);
|
||||||
|
const int src_height = static_cast<int>(fb->height);
|
||||||
|
const int bpp = static_cast<int>(fb->bpp);
|
||||||
|
drmModeFreeFB(fb);
|
||||||
|
|
||||||
|
if (handle == 0 || pitch == 0 || src_width <= 1 || src_height <= 1) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (bpp != 32) {
|
||||||
|
LOG_WARN("DRM capture unsupported bpp: {}", bpp);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
const size_t map_size =
|
||||||
|
static_cast<size_t>(pitch) * static_cast<size_t>(src_height);
|
||||||
|
uint8_t* mapped_ptr = nullptr;
|
||||||
|
size_t mapped_size = 0;
|
||||||
|
int prime_fd = -1;
|
||||||
|
if (!MapFramebuffer(fd, handle, map_size, &mapped_ptr, &mapped_size,
|
||||||
|
&prime_fd)) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
int capture_width = std::min(src_width, output.width);
|
||||||
|
int capture_height = std::min(src_height, output.height);
|
||||||
|
if (capture_width <= 0 || capture_height <= 0) {
|
||||||
|
capture_width = src_width;
|
||||||
|
capture_height = src_height;
|
||||||
|
}
|
||||||
|
|
||||||
|
capture_width &= ~1;
|
||||||
|
capture_height &= ~1;
|
||||||
|
if (capture_width <= 1 || capture_height <= 1) {
|
||||||
|
UnmapFramebuffer(mapped_ptr, mapped_size, prime_fd);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
const size_t y_size =
|
||||||
|
static_cast<size_t>(capture_width) * static_cast<size_t>(capture_height);
|
||||||
|
const size_t uv_size = y_size / 2;
|
||||||
|
if (y_plane_.size() != y_size) {
|
||||||
|
y_plane_.resize(y_size);
|
||||||
|
}
|
||||||
|
if (uv_plane_.size() != uv_size) {
|
||||||
|
uv_plane_.resize(uv_size);
|
||||||
|
}
|
||||||
|
|
||||||
|
const int convert_ret =
|
||||||
|
libyuv::ARGBToNV12(mapped_ptr, static_cast<int>(pitch), y_plane_.data(),
|
||||||
|
capture_width, uv_plane_.data(), capture_width,
|
||||||
|
capture_width, capture_height);
|
||||||
|
|
||||||
|
if (convert_ret != 0) {
|
||||||
|
UnmapFramebuffer(mapped_ptr, mapped_size, prime_fd);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::vector<uint8_t> nv12;
|
||||||
|
nv12.reserve(y_plane_.size() + uv_plane_.size());
|
||||||
|
nv12.insert(nv12.end(), y_plane_.begin(), y_plane_.end());
|
||||||
|
nv12.insert(nv12.end(), uv_plane_.begin(), uv_plane_.end());
|
||||||
|
|
||||||
|
if (emit_callback && callback_) {
|
||||||
|
callback_(nv12.data(), static_cast<int>(nv12.size()), capture_width,
|
||||||
|
capture_height, output.name.c_str());
|
||||||
|
}
|
||||||
|
|
||||||
|
UnmapFramebuffer(mapped_ptr, mapped_size, prime_fd);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ScreenCapturerDrm::MapFramebuffer(int fd, uint32_t handle, size_t map_size,
|
||||||
|
uint8_t** mapped_ptr,
|
||||||
|
size_t* mapped_size,
|
||||||
|
int* prime_fd) const {
|
||||||
|
if (!mapped_ptr || !mapped_size || !prime_fd || map_size == 0) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
*mapped_ptr = nullptr;
|
||||||
|
*mapped_size = 0;
|
||||||
|
*prime_fd = -1;
|
||||||
|
|
||||||
|
drm_mode_map_dumb map_arg{};
|
||||||
|
map_arg.handle = handle;
|
||||||
|
if (drmIoctl(fd, DRM_IOCTL_MODE_MAP_DUMB, &map_arg) == 0) {
|
||||||
|
void* mapped = mmap(nullptr, map_size, PROT_READ, MAP_SHARED, fd,
|
||||||
|
static_cast<off_t>(map_arg.offset));
|
||||||
|
if (mapped != MAP_FAILED) {
|
||||||
|
*mapped_ptr = static_cast<uint8_t*>(mapped);
|
||||||
|
*mapped_size = map_size;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
int dma_fd = -1;
|
||||||
|
if (drmPrimeHandleToFD(fd, handle, DRM_CLOEXEC, &dma_fd) == 0) {
|
||||||
|
size_t dma_map_size = map_size;
|
||||||
|
const off_t fd_size = lseek(dma_fd, 0, SEEK_END);
|
||||||
|
if (fd_size > 0) {
|
||||||
|
dma_map_size = std::min(map_size, static_cast<size_t>(fd_size));
|
||||||
|
}
|
||||||
|
|
||||||
|
void* mapped =
|
||||||
|
mmap(nullptr, dma_map_size, PROT_READ, MAP_SHARED, dma_fd, 0);
|
||||||
|
if (mapped != MAP_FAILED) {
|
||||||
|
*mapped_ptr = static_cast<uint8_t*>(mapped);
|
||||||
|
*mapped_size = dma_map_size;
|
||||||
|
*prime_fd = dma_fd;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
close(dma_fd);
|
||||||
|
}
|
||||||
|
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
void ScreenCapturerDrm::UnmapFramebuffer(uint8_t* mapped_ptr, size_t mapped_size,
|
||||||
|
int prime_fd) const {
|
||||||
|
if (mapped_ptr && mapped_size > 0) {
|
||||||
|
munmap(mapped_ptr, mapped_size);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (prime_fd >= 0) {
|
||||||
|
close(prime_fd);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace crossdesk
|
||||||
|
|
||||||
|
#else
|
||||||
|
|
||||||
|
#include "rd_log.h"
|
||||||
|
|
||||||
|
namespace crossdesk {
|
||||||
|
|
||||||
|
ScreenCapturerDrm::ScreenCapturerDrm() {}
|
||||||
|
|
||||||
|
ScreenCapturerDrm::~ScreenCapturerDrm() { Destroy(); }
|
||||||
|
|
||||||
|
int ScreenCapturerDrm::Init([[maybe_unused]] const int fps, cb_desktop_data cb) {
|
||||||
|
Destroy();
|
||||||
|
callback_ = cb;
|
||||||
|
LOG_WARN("DRM screen capturer disabled: libdrm headers not available");
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
int ScreenCapturerDrm::Destroy() {
|
||||||
|
Stop();
|
||||||
|
callback_ = nullptr;
|
||||||
|
display_info_list_.clear();
|
||||||
|
outputs_.clear();
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int ScreenCapturerDrm::Start([[maybe_unused]] bool show_cursor) { return -1; }
|
||||||
|
|
||||||
|
int ScreenCapturerDrm::Stop() {
|
||||||
|
running_ = false;
|
||||||
|
if (thread_.joinable()) {
|
||||||
|
thread_.join();
|
||||||
|
}
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int ScreenCapturerDrm::Pause([[maybe_unused]] int monitor_index) { return 0; }
|
||||||
|
|
||||||
|
int ScreenCapturerDrm::Resume([[maybe_unused]] int monitor_index) { return 0; }
|
||||||
|
|
||||||
|
int ScreenCapturerDrm::SwitchTo([[maybe_unused]] int monitor_index) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
int ScreenCapturerDrm::ResetToInitialMonitor() { return 0; }
|
||||||
|
|
||||||
|
std::vector<DisplayInfo> ScreenCapturerDrm::GetDisplayInfoList() {
|
||||||
|
return display_info_list_;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ScreenCapturerDrm::DiscoverOutputs() { return false; }
|
||||||
|
|
||||||
|
void ScreenCapturerDrm::CloseDevices() {}
|
||||||
|
|
||||||
|
void ScreenCapturerDrm::CaptureLoop() {}
|
||||||
|
|
||||||
|
bool ScreenCapturerDrm::CaptureOutputFrame(
|
||||||
|
[[maybe_unused]] const DrmOutput& output,
|
||||||
|
[[maybe_unused]] bool emit_callback) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ScreenCapturerDrm::MapFramebuffer([[maybe_unused]] int fd,
|
||||||
|
[[maybe_unused]] uint32_t handle,
|
||||||
|
[[maybe_unused]] size_t map_size,
|
||||||
|
[[maybe_unused]] uint8_t** mapped_ptr,
|
||||||
|
[[maybe_unused]] size_t* mapped_size,
|
||||||
|
[[maybe_unused]] int* prime_fd) const {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
void ScreenCapturerDrm::UnmapFramebuffer([[maybe_unused]] uint8_t* mapped_ptr,
|
||||||
|
[[maybe_unused]] size_t mapped_size,
|
||||||
|
[[maybe_unused]] int prime_fd) const {}
|
||||||
|
|
||||||
|
} // namespace crossdesk
|
||||||
|
|
||||||
|
#endif
|
||||||
87
src/screen_capturer/linux/screen_capturer_drm.h
Normal file
87
src/screen_capturer/linux/screen_capturer_drm.h
Normal file
@@ -0,0 +1,87 @@
|
|||||||
|
/*
|
||||||
|
* @Author: DI JUNKUN
|
||||||
|
* @Date: 2026-03-22
|
||||||
|
* Copyright (c) 2026 by DI JUNKUN, All Rights Reserved.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#ifndef _SCREEN_CAPTURER_DRM_H_
|
||||||
|
#define _SCREEN_CAPTURER_DRM_H_
|
||||||
|
|
||||||
|
#include <atomic>
|
||||||
|
#include <cstdint>
|
||||||
|
#include <string>
|
||||||
|
#include <thread>
|
||||||
|
#include <vector>
|
||||||
|
|
||||||
|
#include "screen_capturer.h"
|
||||||
|
|
||||||
|
namespace crossdesk {
|
||||||
|
|
||||||
|
class ScreenCapturerDrm : public ScreenCapturer {
|
||||||
|
public:
|
||||||
|
ScreenCapturerDrm();
|
||||||
|
~ScreenCapturerDrm();
|
||||||
|
|
||||||
|
public:
|
||||||
|
int Init(const int fps, cb_desktop_data cb) override;
|
||||||
|
int Destroy() override;
|
||||||
|
int Start(bool show_cursor) override;
|
||||||
|
int Stop() override;
|
||||||
|
|
||||||
|
int Pause(int monitor_index) override;
|
||||||
|
int Resume(int monitor_index) override;
|
||||||
|
|
||||||
|
int SwitchTo(int monitor_index) override;
|
||||||
|
int ResetToInitialMonitor() override;
|
||||||
|
|
||||||
|
std::vector<DisplayInfo> GetDisplayInfoList() override;
|
||||||
|
|
||||||
|
private:
|
||||||
|
struct DrmDevice {
|
||||||
|
int fd = -1;
|
||||||
|
std::string path;
|
||||||
|
};
|
||||||
|
|
||||||
|
struct DrmOutput {
|
||||||
|
int device_index = -1;
|
||||||
|
uint32_t connector_id = 0;
|
||||||
|
uint32_t crtc_id = 0;
|
||||||
|
std::string name;
|
||||||
|
int left = 0;
|
||||||
|
int top = 0;
|
||||||
|
int width = 0;
|
||||||
|
int height = 0;
|
||||||
|
};
|
||||||
|
|
||||||
|
private:
|
||||||
|
bool DiscoverOutputs();
|
||||||
|
void CloseDevices();
|
||||||
|
void CaptureLoop();
|
||||||
|
bool CaptureOutputFrame(const DrmOutput& output, bool emit_callback = true);
|
||||||
|
bool MapFramebuffer(int fd, uint32_t handle, size_t map_size,
|
||||||
|
uint8_t** mapped_ptr, size_t* mapped_size,
|
||||||
|
int* prime_fd) const;
|
||||||
|
void UnmapFramebuffer(uint8_t* mapped_ptr, size_t mapped_size,
|
||||||
|
int prime_fd) const;
|
||||||
|
|
||||||
|
private:
|
||||||
|
std::vector<DrmDevice> devices_;
|
||||||
|
std::vector<DrmOutput> outputs_;
|
||||||
|
std::vector<DisplayInfo> display_info_list_;
|
||||||
|
std::thread thread_;
|
||||||
|
std::atomic<bool> running_{false};
|
||||||
|
std::atomic<bool> paused_{false};
|
||||||
|
std::atomic<int> monitor_index_{0};
|
||||||
|
int initial_monitor_index_ = 0;
|
||||||
|
std::atomic<bool> show_cursor_{true};
|
||||||
|
int fps_ = 60;
|
||||||
|
cb_desktop_data callback_;
|
||||||
|
int consecutive_failures_ = 0;
|
||||||
|
|
||||||
|
std::vector<uint8_t> y_plane_;
|
||||||
|
std::vector<uint8_t> uv_plane_;
|
||||||
|
};
|
||||||
|
|
||||||
|
} // namespace crossdesk
|
||||||
|
|
||||||
|
#endif
|
||||||
506
src/screen_capturer/linux/screen_capturer_linux.cpp
Normal file
506
src/screen_capturer/linux/screen_capturer_linux.cpp
Normal file
@@ -0,0 +1,506 @@
|
|||||||
|
#include "screen_capturer_linux.h"
|
||||||
|
|
||||||
|
#include <cstdlib>
|
||||||
|
#include <cstring>
|
||||||
|
#include <memory>
|
||||||
|
#include <string>
|
||||||
|
#include <utility>
|
||||||
|
|
||||||
|
#include "platform.h"
|
||||||
|
#include "rd_log.h"
|
||||||
|
#if defined(CROSSDESK_HAS_DRM) && CROSSDESK_HAS_DRM
|
||||||
|
#include "screen_capturer_drm.h"
|
||||||
|
#endif
|
||||||
|
#if defined(CROSSDESK_HAS_WAYLAND_CAPTURER) && CROSSDESK_HAS_WAYLAND_CAPTURER
|
||||||
|
#include "screen_capturer_wayland.h"
|
||||||
|
#endif
|
||||||
|
#include "screen_capturer_x11.h"
|
||||||
|
|
||||||
|
namespace crossdesk {
|
||||||
|
|
||||||
|
namespace {
|
||||||
|
|
||||||
|
#if defined(CROSSDESK_HAS_DRM) && CROSSDESK_HAS_DRM
|
||||||
|
constexpr bool kDrmBuildEnabled = true;
|
||||||
|
#else
|
||||||
|
constexpr bool kDrmBuildEnabled = false;
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#if defined(CROSSDESK_HAS_WAYLAND_CAPTURER) && CROSSDESK_HAS_WAYLAND_CAPTURER
|
||||||
|
constexpr bool kWaylandBuildEnabled = true;
|
||||||
|
#else
|
||||||
|
constexpr bool kWaylandBuildEnabled = false;
|
||||||
|
#endif
|
||||||
|
|
||||||
|
} // namespace
|
||||||
|
|
||||||
|
ScreenCapturerLinux::ScreenCapturerLinux() {}
|
||||||
|
|
||||||
|
ScreenCapturerLinux::~ScreenCapturerLinux() { Destroy(); }
|
||||||
|
|
||||||
|
int ScreenCapturerLinux::Init(const int fps, cb_desktop_data cb) {
|
||||||
|
Destroy();
|
||||||
|
|
||||||
|
if (!cb) {
|
||||||
|
LOG_ERROR("Linux screen capturer callback is null");
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
fps_ = fps;
|
||||||
|
callback_orig_ = std::move(cb);
|
||||||
|
callback_ = [this](unsigned char* data, int size, int width, int height,
|
||||||
|
const char* display_name) {
|
||||||
|
const std::string mapped_name = MapDisplayName(display_name);
|
||||||
|
if (callback_orig_) {
|
||||||
|
callback_orig_(data, size, width, height, mapped_name.c_str());
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
const char* force_backend = getenv("CROSSDESK_SCREEN_BACKEND");
|
||||||
|
if (force_backend && force_backend[0] != '\0') {
|
||||||
|
if (strcmp(force_backend, "drm") == 0) {
|
||||||
|
#if defined(CROSSDESK_HAS_DRM) && CROSSDESK_HAS_DRM
|
||||||
|
LOG_INFO("Linux screen capturer forced backend: DRM");
|
||||||
|
return InitDrm();
|
||||||
|
#else
|
||||||
|
LOG_ERROR(
|
||||||
|
"Linux screen capturer forced backend DRM is disabled at build time");
|
||||||
|
return -1;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
if (strcmp(force_backend, "x11") == 0) {
|
||||||
|
LOG_INFO("Linux screen capturer forced backend: X11");
|
||||||
|
return InitX11();
|
||||||
|
}
|
||||||
|
if (strcmp(force_backend, "wayland") == 0) {
|
||||||
|
#if defined(CROSSDESK_HAS_WAYLAND_CAPTURER) && CROSSDESK_HAS_WAYLAND_CAPTURER
|
||||||
|
LOG_INFO("Linux screen capturer forced backend: Wayland");
|
||||||
|
return InitWayland();
|
||||||
|
#else
|
||||||
|
LOG_ERROR(
|
||||||
|
"Linux screen capturer forced backend Wayland is disabled at build "
|
||||||
|
"time");
|
||||||
|
return -1;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
LOG_WARN("Unknown CROSSDESK_SCREEN_BACKEND={}, using auto strategy",
|
||||||
|
force_backend);
|
||||||
|
}
|
||||||
|
|
||||||
|
const bool wayland_session = IsWaylandSession();
|
||||||
|
if (wayland_session) {
|
||||||
|
if (kDrmBuildEnabled) {
|
||||||
|
LOG_INFO("Wayland session detected, prefer DRM -> X11 -> Wayland");
|
||||||
|
if (InitDrm() == 0) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
LOG_INFO("Wayland session detected, DRM disabled, prefer X11 -> Wayland");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (InitX11() == 0) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (kDrmBuildEnabled) {
|
||||||
|
LOG_WARN(
|
||||||
|
"DRM and X11 init failed in Wayland session, trying Wayland portal");
|
||||||
|
} else {
|
||||||
|
LOG_WARN("X11 init failed in Wayland session, trying Wayland portal");
|
||||||
|
}
|
||||||
|
if (kWaylandBuildEnabled) {
|
||||||
|
return InitWayland();
|
||||||
|
}
|
||||||
|
LOG_ERROR("Wayland session detected but Wayland backend is disabled");
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (InitX11() == 0) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (kDrmBuildEnabled) {
|
||||||
|
LOG_WARN("X11 init failed, trying DRM fallback");
|
||||||
|
return InitDrm();
|
||||||
|
}
|
||||||
|
|
||||||
|
LOG_ERROR("X11 init failed and DRM backend is disabled");
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
int ScreenCapturerLinux::Destroy() {
|
||||||
|
if (impl_) {
|
||||||
|
impl_->Destroy();
|
||||||
|
impl_.reset();
|
||||||
|
}
|
||||||
|
|
||||||
|
backend_ = BackendType::kNone;
|
||||||
|
callback_ = nullptr;
|
||||||
|
callback_orig_ = nullptr;
|
||||||
|
{
|
||||||
|
std::lock_guard<std::mutex> lock(alias_mutex_);
|
||||||
|
canonical_displays_.clear();
|
||||||
|
label_alias_.clear();
|
||||||
|
}
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int ScreenCapturerLinux::Start(bool show_cursor) {
|
||||||
|
if (!impl_) {
|
||||||
|
LOG_ERROR("Linux screen capturer backend is not initialized");
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
#if defined(CROSSDESK_HAS_WAYLAND_CAPTURER) && CROSSDESK_HAS_WAYLAND_CAPTURER
|
||||||
|
if (backend_ == BackendType::kWayland) {
|
||||||
|
const int refresh_ret = RefreshWaylandBackend();
|
||||||
|
if (refresh_ret != 0) {
|
||||||
|
LOG_WARN("Linux screen capturer Wayland backend refresh failed: {}",
|
||||||
|
refresh_ret);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
const int ret = impl_->Start(show_cursor);
|
||||||
|
if (ret == 0) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
const char* backend_name = "None";
|
||||||
|
if (backend_ == BackendType::kX11) {
|
||||||
|
backend_name = "X11";
|
||||||
|
} else if (backend_ == BackendType::kDrm) {
|
||||||
|
backend_name = "DRM";
|
||||||
|
} else if (backend_ == BackendType::kWayland) {
|
||||||
|
backend_name = "Wayland";
|
||||||
|
}
|
||||||
|
|
||||||
|
LOG_WARN("Linux screen capturer backend {} start failed: {}",
|
||||||
|
backend_name, ret);
|
||||||
|
|
||||||
|
if (backend_ == BackendType::kX11 && kDrmBuildEnabled &&
|
||||||
|
TryFallbackToDrm(show_cursor)) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
if (backend_ == BackendType::kX11 && kWaylandBuildEnabled &&
|
||||||
|
TryFallbackToWayland(show_cursor)) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
if (backend_ == BackendType::kDrm && kDrmBuildEnabled) {
|
||||||
|
if (TryFallbackToX11(show_cursor)) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
if (kWaylandBuildEnabled && TryFallbackToWayland(show_cursor)) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (backend_ == BackendType::kWayland && kWaylandBuildEnabled) {
|
||||||
|
if (kDrmBuildEnabled && TryFallbackToDrm(show_cursor)) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
if (TryFallbackToX11(show_cursor)) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
int ScreenCapturerLinux::Stop() {
|
||||||
|
if (!impl_) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
const int ret = impl_->Stop();
|
||||||
|
UpdateAliasesFromBackend(impl_.get());
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
int ScreenCapturerLinux::Pause(int monitor_index) {
|
||||||
|
if (!impl_) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
return impl_->Pause(monitor_index);
|
||||||
|
}
|
||||||
|
|
||||||
|
int ScreenCapturerLinux::Resume(int monitor_index) {
|
||||||
|
if (!impl_) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
return impl_->Resume(monitor_index);
|
||||||
|
}
|
||||||
|
|
||||||
|
int ScreenCapturerLinux::SwitchTo(int monitor_index) {
|
||||||
|
if (!impl_) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
return impl_->SwitchTo(monitor_index);
|
||||||
|
}
|
||||||
|
|
||||||
|
int ScreenCapturerLinux::ResetToInitialMonitor() {
|
||||||
|
if (!impl_) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
return impl_->ResetToInitialMonitor();
|
||||||
|
}
|
||||||
|
|
||||||
|
std::vector<DisplayInfo> ScreenCapturerLinux::GetDisplayInfoList() {
|
||||||
|
if (!impl_) {
|
||||||
|
return std::vector<DisplayInfo>();
|
||||||
|
}
|
||||||
|
|
||||||
|
// Wayland backend may update display geometry/stream handle asynchronously
|
||||||
|
// after Start(). Refresh aliases every time to keep canonical displays fresh.
|
||||||
|
UpdateAliasesFromBackend(impl_.get());
|
||||||
|
|
||||||
|
std::lock_guard<std::mutex> lock(alias_mutex_);
|
||||||
|
if (!canonical_displays_.empty()) {
|
||||||
|
return canonical_displays_;
|
||||||
|
}
|
||||||
|
|
||||||
|
return impl_->GetDisplayInfoList();
|
||||||
|
}
|
||||||
|
|
||||||
|
int ScreenCapturerLinux::InitX11() {
|
||||||
|
auto backend = std::make_unique<ScreenCapturerX11>();
|
||||||
|
const int ret = backend->Init(fps_, callback_);
|
||||||
|
if (ret != 0) {
|
||||||
|
backend->Destroy();
|
||||||
|
LOG_WARN("Linux screen capturer X11 init failed: {}", ret);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
UpdateAliasesFromBackend(backend.get());
|
||||||
|
impl_ = std::move(backend);
|
||||||
|
backend_ = BackendType::kX11;
|
||||||
|
LOG_INFO("Linux screen capturer backend selected: X11");
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int ScreenCapturerLinux::InitDrm() {
|
||||||
|
#if defined(CROSSDESK_HAS_DRM) && CROSSDESK_HAS_DRM
|
||||||
|
auto backend = std::make_unique<ScreenCapturerDrm>();
|
||||||
|
const int ret = backend->Init(fps_, callback_);
|
||||||
|
if (ret != 0) {
|
||||||
|
backend->Destroy();
|
||||||
|
LOG_WARN("Linux screen capturer DRM init failed: {}", ret);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
UpdateAliasesFromBackend(backend.get());
|
||||||
|
impl_ = std::move(backend);
|
||||||
|
backend_ = BackendType::kDrm;
|
||||||
|
LOG_INFO("Linux screen capturer backend selected: DRM");
|
||||||
|
return 0;
|
||||||
|
#else
|
||||||
|
LOG_WARN("Linux screen capturer DRM backend is disabled at build time");
|
||||||
|
return -1;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
int ScreenCapturerLinux::InitWayland() {
|
||||||
|
#if defined(CROSSDESK_HAS_WAYLAND_CAPTURER) && CROSSDESK_HAS_WAYLAND_CAPTURER
|
||||||
|
auto backend = std::make_unique<ScreenCapturerWayland>();
|
||||||
|
const int ret = backend->Init(fps_, callback_);
|
||||||
|
if (ret != 0) {
|
||||||
|
backend->Destroy();
|
||||||
|
LOG_WARN("Linux screen capturer Wayland init failed: {}", ret);
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
UpdateAliasesFromBackend(backend.get());
|
||||||
|
impl_ = std::move(backend);
|
||||||
|
backend_ = BackendType::kWayland;
|
||||||
|
LOG_INFO("Linux screen capturer backend selected: Wayland");
|
||||||
|
return 0;
|
||||||
|
#else
|
||||||
|
LOG_WARN("Linux screen capturer Wayland backend is disabled at build time");
|
||||||
|
return -1;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
int ScreenCapturerLinux::RefreshWaylandBackend() {
|
||||||
|
#if defined(CROSSDESK_HAS_WAYLAND_CAPTURER) && CROSSDESK_HAS_WAYLAND_CAPTURER
|
||||||
|
auto backend = std::make_unique<ScreenCapturerWayland>();
|
||||||
|
const int ret = backend->Init(fps_, callback_);
|
||||||
|
if (ret != 0) {
|
||||||
|
backend->Destroy();
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (impl_) {
|
||||||
|
impl_->Destroy();
|
||||||
|
}
|
||||||
|
|
||||||
|
UpdateAliasesFromBackend(backend.get());
|
||||||
|
impl_ = std::move(backend);
|
||||||
|
backend_ = BackendType::kWayland;
|
||||||
|
LOG_INFO("Linux screen capturer Wayland backend refreshed before start");
|
||||||
|
return 0;
|
||||||
|
#else
|
||||||
|
return -1;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ScreenCapturerLinux::TryFallbackToDrm(bool show_cursor) {
|
||||||
|
#if defined(CROSSDESK_HAS_DRM) && CROSSDESK_HAS_DRM
|
||||||
|
auto drm_backend = std::make_unique<ScreenCapturerDrm>();
|
||||||
|
int ret = drm_backend->Init(fps_, callback_);
|
||||||
|
if (ret != 0) {
|
||||||
|
LOG_ERROR("Linux screen capturer fallback DRM init failed: {}", ret);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
UpdateAliasesFromBackend(drm_backend.get());
|
||||||
|
ret = drm_backend->Start(show_cursor);
|
||||||
|
if (ret != 0) {
|
||||||
|
drm_backend->Destroy();
|
||||||
|
LOG_ERROR("Linux screen capturer fallback DRM start failed: {}", ret);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (impl_) {
|
||||||
|
impl_->Stop();
|
||||||
|
impl_->Destroy();
|
||||||
|
}
|
||||||
|
|
||||||
|
impl_ = std::move(drm_backend);
|
||||||
|
backend_ = BackendType::kDrm;
|
||||||
|
LOG_INFO("Linux screen capturer fallback switched to DRM");
|
||||||
|
return true;
|
||||||
|
#else
|
||||||
|
(void)show_cursor;
|
||||||
|
LOG_WARN("Linux screen capturer DRM fallback is disabled at build time");
|
||||||
|
return false;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ScreenCapturerLinux::TryFallbackToX11(bool show_cursor) {
|
||||||
|
auto x11_backend = std::make_unique<ScreenCapturerX11>();
|
||||||
|
int ret = x11_backend->Init(fps_, callback_);
|
||||||
|
if (ret != 0) {
|
||||||
|
LOG_ERROR("Linux screen capturer fallback X11 init failed: {}", ret);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
UpdateAliasesFromBackend(x11_backend.get());
|
||||||
|
ret = x11_backend->Start(show_cursor);
|
||||||
|
if (ret != 0) {
|
||||||
|
x11_backend->Destroy();
|
||||||
|
LOG_ERROR("Linux screen capturer fallback X11 start failed: {}", ret);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (impl_) {
|
||||||
|
impl_->Stop();
|
||||||
|
impl_->Destroy();
|
||||||
|
}
|
||||||
|
|
||||||
|
impl_ = std::move(x11_backend);
|
||||||
|
backend_ = BackendType::kX11;
|
||||||
|
LOG_INFO("Linux screen capturer fallback switched to X11");
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ScreenCapturerLinux::TryFallbackToWayland(bool show_cursor) {
|
||||||
|
#if defined(CROSSDESK_HAS_WAYLAND_CAPTURER) && CROSSDESK_HAS_WAYLAND_CAPTURER
|
||||||
|
auto wayland_backend = std::make_unique<ScreenCapturerWayland>();
|
||||||
|
int ret = wayland_backend->Init(fps_, callback_);
|
||||||
|
if (ret != 0) {
|
||||||
|
LOG_ERROR("Linux screen capturer fallback Wayland init failed: {}", ret);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
UpdateAliasesFromBackend(wayland_backend.get());
|
||||||
|
ret = wayland_backend->Start(show_cursor);
|
||||||
|
if (ret != 0) {
|
||||||
|
wayland_backend->Destroy();
|
||||||
|
LOG_ERROR("Linux screen capturer fallback Wayland start failed: {}", ret);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (impl_) {
|
||||||
|
impl_->Stop();
|
||||||
|
impl_->Destroy();
|
||||||
|
}
|
||||||
|
|
||||||
|
impl_ = std::move(wayland_backend);
|
||||||
|
backend_ = BackendType::kWayland;
|
||||||
|
LOG_INFO("Linux screen capturer fallback switched to Wayland");
|
||||||
|
return true;
|
||||||
|
#else
|
||||||
|
(void)show_cursor;
|
||||||
|
LOG_WARN("Linux screen capturer Wayland fallback is disabled at build time");
|
||||||
|
return false;
|
||||||
|
#endif
|
||||||
|
}
|
||||||
|
|
||||||
|
void ScreenCapturerLinux::UpdateAliasesFromBackend(ScreenCapturer* backend) {
|
||||||
|
if (!backend) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const auto backend_displays = backend->GetDisplayInfoList();
|
||||||
|
if (backend_displays.empty()) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::lock_guard<std::mutex> lock(alias_mutex_);
|
||||||
|
label_alias_.clear();
|
||||||
|
|
||||||
|
if (canonical_displays_.empty()) {
|
||||||
|
canonical_displays_ = backend_displays;
|
||||||
|
for (const auto& display : backend_displays) {
|
||||||
|
label_alias_[display.name] = display.name;
|
||||||
|
}
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (canonical_displays_.size() < backend_displays.size()) {
|
||||||
|
for (size_t i = canonical_displays_.size(); i < backend_displays.size();
|
||||||
|
++i) {
|
||||||
|
canonical_displays_.push_back(backend_displays[i]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for (size_t i = 0; i < backend_displays.size(); ++i) {
|
||||||
|
const std::string mapped_name = i < canonical_displays_.size()
|
||||||
|
? canonical_displays_[i].name
|
||||||
|
: backend_displays[i].name;
|
||||||
|
label_alias_[backend_displays[i].name] = mapped_name;
|
||||||
|
|
||||||
|
if (i < canonical_displays_.size()) {
|
||||||
|
// Keep original stable names, but refresh geometry from active backend.
|
||||||
|
canonical_displays_[i].handle = backend_displays[i].handle;
|
||||||
|
canonical_displays_[i].is_primary = backend_displays[i].is_primary;
|
||||||
|
canonical_displays_[i].left = backend_displays[i].left;
|
||||||
|
canonical_displays_[i].top = backend_displays[i].top;
|
||||||
|
canonical_displays_[i].right = backend_displays[i].right;
|
||||||
|
canonical_displays_[i].bottom = backend_displays[i].bottom;
|
||||||
|
canonical_displays_[i].width = backend_displays[i].width;
|
||||||
|
canonical_displays_[i].height = backend_displays[i].height;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string ScreenCapturerLinux::MapDisplayName(const char* display_name) const {
|
||||||
|
std::string input_name = display_name ? display_name : "";
|
||||||
|
if (input_name.empty()) {
|
||||||
|
return input_name;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::lock_guard<std::mutex> lock(alias_mutex_);
|
||||||
|
auto it = label_alias_.find(input_name);
|
||||||
|
if (it != label_alias_.end()) {
|
||||||
|
return it->second;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (canonical_displays_.size() == 1) {
|
||||||
|
return canonical_displays_[0].name;
|
||||||
|
}
|
||||||
|
|
||||||
|
return input_name;
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace crossdesk
|
||||||
66
src/screen_capturer/linux/screen_capturer_linux.h
Normal file
66
src/screen_capturer/linux/screen_capturer_linux.h
Normal file
@@ -0,0 +1,66 @@
|
|||||||
|
/*
|
||||||
|
* @Author: DI JUNKUN
|
||||||
|
* @Date: 2026-03-22
|
||||||
|
* Copyright (c) 2026 by DI JUNKUN, All Rights Reserved.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#ifndef _SCREEN_CAPTURER_LINUX_H_
|
||||||
|
#define _SCREEN_CAPTURER_LINUX_H_
|
||||||
|
|
||||||
|
#include <memory>
|
||||||
|
#include <mutex>
|
||||||
|
#include <string>
|
||||||
|
#include <unordered_map>
|
||||||
|
#include <vector>
|
||||||
|
|
||||||
|
#include "screen_capturer.h"
|
||||||
|
|
||||||
|
namespace crossdesk {
|
||||||
|
|
||||||
|
class ScreenCapturerLinux : public ScreenCapturer {
|
||||||
|
public:
|
||||||
|
ScreenCapturerLinux();
|
||||||
|
~ScreenCapturerLinux();
|
||||||
|
|
||||||
|
public:
|
||||||
|
int Init(const int fps, cb_desktop_data cb) override;
|
||||||
|
int Destroy() override;
|
||||||
|
int Start(bool show_cursor) override;
|
||||||
|
int Stop() override;
|
||||||
|
|
||||||
|
int Pause(int monitor_index) override;
|
||||||
|
int Resume(int monitor_index) override;
|
||||||
|
|
||||||
|
int SwitchTo(int monitor_index) override;
|
||||||
|
int ResetToInitialMonitor() override;
|
||||||
|
|
||||||
|
std::vector<DisplayInfo> GetDisplayInfoList() override;
|
||||||
|
|
||||||
|
private:
|
||||||
|
enum class BackendType { kNone, kX11, kDrm, kWayland };
|
||||||
|
|
||||||
|
private:
|
||||||
|
int InitX11();
|
||||||
|
int InitDrm();
|
||||||
|
int InitWayland();
|
||||||
|
int RefreshWaylandBackend();
|
||||||
|
bool TryFallbackToDrm(bool show_cursor);
|
||||||
|
bool TryFallbackToX11(bool show_cursor);
|
||||||
|
bool TryFallbackToWayland(bool show_cursor);
|
||||||
|
void UpdateAliasesFromBackend(ScreenCapturer* backend);
|
||||||
|
std::string MapDisplayName(const char* display_name) const;
|
||||||
|
|
||||||
|
private:
|
||||||
|
std::unique_ptr<ScreenCapturer> impl_;
|
||||||
|
BackendType backend_ = BackendType::kNone;
|
||||||
|
int fps_ = 60;
|
||||||
|
cb_desktop_data callback_;
|
||||||
|
cb_desktop_data callback_orig_;
|
||||||
|
std::vector<DisplayInfo> canonical_displays_;
|
||||||
|
mutable std::mutex alias_mutex_;
|
||||||
|
std::unordered_map<std::string, std::string> label_alias_;
|
||||||
|
};
|
||||||
|
|
||||||
|
} // namespace crossdesk
|
||||||
|
|
||||||
|
#endif
|
||||||
242
src/screen_capturer/linux/screen_capturer_wayland.cpp
Normal file
242
src/screen_capturer/linux/screen_capturer_wayland.cpp
Normal file
@@ -0,0 +1,242 @@
|
|||||||
|
#include "screen_capturer_wayland.h"
|
||||||
|
|
||||||
|
#include "screen_capturer_wayland_build.h"
|
||||||
|
|
||||||
|
#if !CROSSDESK_WAYLAND_BUILD_ENABLED
|
||||||
|
#error "Wayland capturer requires USE_WAYLAND=true and Wayland development headers"
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#include <cstdlib>
|
||||||
|
#include <cstring>
|
||||||
|
|
||||||
|
#include <chrono>
|
||||||
|
#include <thread>
|
||||||
|
|
||||||
|
#include "platform.h"
|
||||||
|
#include "rd_log.h"
|
||||||
|
#include "wayland_portal_shared.h"
|
||||||
|
|
||||||
|
namespace crossdesk {
|
||||||
|
|
||||||
|
namespace {
|
||||||
|
|
||||||
|
int64_t NowMs() {
|
||||||
|
return std::chrono::duration_cast<std::chrono::milliseconds>(
|
||||||
|
std::chrono::steady_clock::now().time_since_epoch())
|
||||||
|
.count();
|
||||||
|
}
|
||||||
|
|
||||||
|
struct PipeWireRecoveryConfig {
|
||||||
|
ScreenCapturerWayland::PipeWireConnectMode mode;
|
||||||
|
bool relaxed_connect = false;
|
||||||
|
};
|
||||||
|
|
||||||
|
constexpr auto kPipeWireCloseSettleDelay = std::chrono::milliseconds(200);
|
||||||
|
|
||||||
|
} // namespace
|
||||||
|
|
||||||
|
ScreenCapturerWayland::ScreenCapturerWayland() {}
|
||||||
|
|
||||||
|
ScreenCapturerWayland::~ScreenCapturerWayland() { Destroy(); }
|
||||||
|
|
||||||
|
int ScreenCapturerWayland::Init(const int fps, cb_desktop_data cb) {
|
||||||
|
Destroy();
|
||||||
|
|
||||||
|
if (!IsWaylandSession()) {
|
||||||
|
LOG_ERROR("Wayland screen capturer requires a Wayland session");
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!cb) {
|
||||||
|
LOG_ERROR("Wayland screen capturer callback is null");
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!CheckPortalAvailability()) {
|
||||||
|
LOG_ERROR("xdg-desktop-portal screencast service is unavailable");
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
fps_ = fps;
|
||||||
|
callback_ = cb;
|
||||||
|
pointer_granted_ = false;
|
||||||
|
shared_session_registered_ = false;
|
||||||
|
display_info_list_.clear();
|
||||||
|
display_info_list_.push_back(
|
||||||
|
DisplayInfo(display_name_, 0, 0, kFallbackWidth, kFallbackHeight));
|
||||||
|
monitor_index_ = 0;
|
||||||
|
initial_monitor_index_ = 0;
|
||||||
|
frame_width_ = kFallbackWidth;
|
||||||
|
frame_height_ = kFallbackHeight;
|
||||||
|
frame_stride_ = kFallbackWidth * 4;
|
||||||
|
logical_width_ = kFallbackWidth;
|
||||||
|
logical_height_ = kFallbackHeight;
|
||||||
|
y_plane_.resize(kFallbackWidth * kFallbackHeight);
|
||||||
|
uv_plane_.resize((kFallbackWidth / 2) * (kFallbackHeight / 2) * 2);
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int ScreenCapturerWayland::Destroy() {
|
||||||
|
Stop();
|
||||||
|
y_plane_.clear();
|
||||||
|
uv_plane_.clear();
|
||||||
|
display_info_list_.clear();
|
||||||
|
callback_ = nullptr;
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int ScreenCapturerWayland::Start(bool show_cursor) {
|
||||||
|
if (running_) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
show_cursor_ = show_cursor;
|
||||||
|
paused_ = false;
|
||||||
|
pipewire_node_id_ = 0;
|
||||||
|
UpdateDisplayGeometry(logical_width_ > 0 ? logical_width_ : kFallbackWidth,
|
||||||
|
logical_height_ > 0 ? logical_height_
|
||||||
|
: kFallbackHeight);
|
||||||
|
pipewire_format_ready_.store(false);
|
||||||
|
pipewire_stream_start_ms_.store(0);
|
||||||
|
pipewire_last_frame_ms_.store(0);
|
||||||
|
running_ = true;
|
||||||
|
thread_ = std::thread([this]() { Run(); });
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int ScreenCapturerWayland::Stop() {
|
||||||
|
running_ = false;
|
||||||
|
if (thread_.joinable()) {
|
||||||
|
thread_.join();
|
||||||
|
}
|
||||||
|
pipewire_node_id_ = 0;
|
||||||
|
UpdateDisplayGeometry(logical_width_ > 0 ? logical_width_ : kFallbackWidth,
|
||||||
|
logical_height_ > 0 ? logical_height_
|
||||||
|
: kFallbackHeight);
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int ScreenCapturerWayland::Pause([[maybe_unused]] int monitor_index) {
|
||||||
|
paused_ = true;
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int ScreenCapturerWayland::Resume([[maybe_unused]] int monitor_index) {
|
||||||
|
paused_ = false;
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int ScreenCapturerWayland::SwitchTo(int monitor_index) {
|
||||||
|
if (monitor_index != 0) {
|
||||||
|
LOG_WARN("Wayland screencast currently supports one logical display");
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
monitor_index_ = 0;
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int ScreenCapturerWayland::ResetToInitialMonitor() {
|
||||||
|
monitor_index_ = initial_monitor_index_;
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::vector<DisplayInfo> ScreenCapturerWayland::GetDisplayInfoList() {
|
||||||
|
return display_info_list_;
|
||||||
|
}
|
||||||
|
|
||||||
|
void ScreenCapturerWayland::Run() {
|
||||||
|
static constexpr PipeWireRecoveryConfig kRecoveryConfigs[] = {
|
||||||
|
{PipeWireConnectMode::kTargetObject, false},
|
||||||
|
{PipeWireConnectMode::kAny, true},
|
||||||
|
{PipeWireConnectMode::kNodeId, false},
|
||||||
|
{PipeWireConnectMode::kNodeId, true},
|
||||||
|
};
|
||||||
|
|
||||||
|
int recovery_index = 0;
|
||||||
|
auto setup_pipewire = [this, &recovery_index]() -> bool {
|
||||||
|
const auto& config = kRecoveryConfigs[recovery_index];
|
||||||
|
return OpenPipeWireRemote() &&
|
||||||
|
SetupPipeWireStream(config.relaxed_connect, config.mode);
|
||||||
|
};
|
||||||
|
auto setup_pipeline = [this, &setup_pipewire]() -> bool {
|
||||||
|
return ConnectSessionBus() && CreatePortalSession() &&
|
||||||
|
SelectPortalDevices() && SelectPortalSource() &&
|
||||||
|
StartPortalSession() && setup_pipewire();
|
||||||
|
};
|
||||||
|
|
||||||
|
if (!setup_pipeline()) {
|
||||||
|
running_ = false;
|
||||||
|
CleanupPipeWire();
|
||||||
|
ClosePortalSession();
|
||||||
|
CleanupDbus();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
while (running_) {
|
||||||
|
if (!paused_) {
|
||||||
|
const int64_t now = NowMs();
|
||||||
|
const int64_t stream_start = pipewire_stream_start_ms_.load();
|
||||||
|
const int64_t last_frame = pipewire_last_frame_ms_.load();
|
||||||
|
const bool format_ready = pipewire_format_ready_.load();
|
||||||
|
|
||||||
|
const bool format_timeout =
|
||||||
|
stream_start > 0 && !format_ready && (now - stream_start) > 1200;
|
||||||
|
const bool first_frame_timeout =
|
||||||
|
stream_start > 0 && format_ready && last_frame == 0 &&
|
||||||
|
(now - stream_start) > 4000;
|
||||||
|
const bool frame_stall = last_frame > 0 && (now - last_frame) > 5000;
|
||||||
|
|
||||||
|
if (format_timeout || first_frame_timeout || frame_stall) {
|
||||||
|
if (recovery_index + 1 >=
|
||||||
|
static_cast<int>(sizeof(kRecoveryConfigs) /
|
||||||
|
sizeof(kRecoveryConfigs[0]))) {
|
||||||
|
LOG_ERROR(
|
||||||
|
"Wayland capture stalled and recovery limit reached, "
|
||||||
|
"format_ready={}, stream_start={}, last_frame={}, attempts={}",
|
||||||
|
format_ready, stream_start, last_frame, recovery_index);
|
||||||
|
running_ = false;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
++recovery_index;
|
||||||
|
const char* reason = format_timeout
|
||||||
|
? "format-timeout"
|
||||||
|
: (first_frame_timeout ? "first-frame-timeout"
|
||||||
|
: "frame-stall");
|
||||||
|
const auto& config = kRecoveryConfigs[recovery_index];
|
||||||
|
LOG_WARN(
|
||||||
|
"Wayland capture stalled ({}) - retrying PipeWire only, "
|
||||||
|
"attempt {}/{}, mode={}, relaxed_connect={}",
|
||||||
|
reason, recovery_index,
|
||||||
|
static_cast<int>(sizeof(kRecoveryConfigs) /
|
||||||
|
sizeof(kRecoveryConfigs[0])) -
|
||||||
|
1,
|
||||||
|
config.mode == PipeWireConnectMode::kTargetObject
|
||||||
|
? "target-object"
|
||||||
|
: (config.mode == PipeWireConnectMode::kNodeId ? "node-id"
|
||||||
|
: "any"),
|
||||||
|
config.relaxed_connect);
|
||||||
|
|
||||||
|
CleanupPipeWire();
|
||||||
|
if (!setup_pipewire()) {
|
||||||
|
LOG_ERROR("Wayland PipeWire-only recovery failed at attempt {}",
|
||||||
|
recovery_index);
|
||||||
|
running_ = false;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
std::this_thread::sleep_for(std::chrono::milliseconds(50));
|
||||||
|
}
|
||||||
|
|
||||||
|
CleanupPipeWire();
|
||||||
|
if (!session_handle_.empty()) {
|
||||||
|
std::this_thread::sleep_for(kPipeWireCloseSettleDelay);
|
||||||
|
}
|
||||||
|
ClosePortalSession();
|
||||||
|
CleanupDbus();
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace crossdesk
|
||||||
110
src/screen_capturer/linux/screen_capturer_wayland.h
Normal file
110
src/screen_capturer/linux/screen_capturer_wayland.h
Normal file
@@ -0,0 +1,110 @@
|
|||||||
|
/*
|
||||||
|
* @Author: DI JUNKUN
|
||||||
|
* @Date: 2026-03-22
|
||||||
|
* Copyright (c) 2026 by DI JUNKUN, All Rights Reserved.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#ifndef _SCREEN_CAPTURER_WAYLAND_H_
|
||||||
|
#define _SCREEN_CAPTURER_WAYLAND_H_
|
||||||
|
|
||||||
|
struct DBusConnection;
|
||||||
|
struct pw_context;
|
||||||
|
struct pw_core;
|
||||||
|
struct pw_stream;
|
||||||
|
struct pw_thread_loop;
|
||||||
|
|
||||||
|
#include <atomic>
|
||||||
|
#include <cstdint>
|
||||||
|
#include <string>
|
||||||
|
#include <thread>
|
||||||
|
#include <vector>
|
||||||
|
|
||||||
|
#include "screen_capturer.h"
|
||||||
|
|
||||||
|
namespace crossdesk {
|
||||||
|
|
||||||
|
class ScreenCapturerWayland : public ScreenCapturer {
|
||||||
|
public:
|
||||||
|
enum class PipeWireConnectMode { kTargetObject, kNodeId, kAny };
|
||||||
|
|
||||||
|
public:
|
||||||
|
ScreenCapturerWayland();
|
||||||
|
~ScreenCapturerWayland();
|
||||||
|
|
||||||
|
public:
|
||||||
|
int Init(const int fps, cb_desktop_data cb) override;
|
||||||
|
int Destroy() override;
|
||||||
|
int Start(bool show_cursor) override;
|
||||||
|
int Stop() override;
|
||||||
|
|
||||||
|
int Pause(int monitor_index) override;
|
||||||
|
int Resume(int monitor_index) override;
|
||||||
|
|
||||||
|
int SwitchTo(int monitor_index) override;
|
||||||
|
int ResetToInitialMonitor() override;
|
||||||
|
|
||||||
|
std::vector<DisplayInfo> GetDisplayInfoList() override;
|
||||||
|
|
||||||
|
private:
|
||||||
|
bool CheckPortalAvailability() const;
|
||||||
|
bool ConnectSessionBus();
|
||||||
|
bool CreatePortalSession();
|
||||||
|
bool SelectPortalDevices();
|
||||||
|
bool SelectPortalSource();
|
||||||
|
bool StartPortalSession();
|
||||||
|
bool OpenPipeWireRemote();
|
||||||
|
bool SetupPipeWireStream(bool relaxed_connect, PipeWireConnectMode mode);
|
||||||
|
|
||||||
|
void Run();
|
||||||
|
void CleanupPipeWire();
|
||||||
|
void CleanupDbus();
|
||||||
|
void ClosePortalSession();
|
||||||
|
void HandlePipeWireBuffer();
|
||||||
|
void UpdateDisplayGeometry(int width, int height);
|
||||||
|
|
||||||
|
private:
|
||||||
|
static constexpr int kFallbackWidth = 1920;
|
||||||
|
static constexpr int kFallbackHeight = 1080;
|
||||||
|
|
||||||
|
std::thread thread_;
|
||||||
|
std::atomic<bool> running_{false};
|
||||||
|
std::atomic<bool> paused_{false};
|
||||||
|
std::atomic<int> monitor_index_{0};
|
||||||
|
std::atomic<bool> pipewire_format_ready_{false};
|
||||||
|
std::atomic<int64_t> pipewire_stream_start_ms_{0};
|
||||||
|
std::atomic<int64_t> pipewire_last_frame_ms_{0};
|
||||||
|
int initial_monitor_index_ = 0;
|
||||||
|
std::atomic<bool> show_cursor_{true};
|
||||||
|
int fps_ = 60;
|
||||||
|
cb_desktop_data callback_ = nullptr;
|
||||||
|
std::vector<DisplayInfo> display_info_list_;
|
||||||
|
|
||||||
|
DBusConnection* dbus_connection_ = nullptr;
|
||||||
|
std::string session_handle_;
|
||||||
|
std::string display_name_ = "WAYLAND0";
|
||||||
|
uint32_t pipewire_node_id_ = 0;
|
||||||
|
int pipewire_fd_ = -1;
|
||||||
|
|
||||||
|
pw_thread_loop* pw_thread_loop_ = nullptr;
|
||||||
|
pw_context* pw_context_ = nullptr;
|
||||||
|
pw_core* pw_core_ = nullptr;
|
||||||
|
pw_stream* pw_stream_ = nullptr;
|
||||||
|
void* stream_listener_ = nullptr;
|
||||||
|
bool pipewire_initialized_ = false;
|
||||||
|
bool pipewire_thread_loop_started_ = false;
|
||||||
|
bool pointer_granted_ = false;
|
||||||
|
bool shared_session_registered_ = false;
|
||||||
|
uint32_t spa_video_format_ = 0;
|
||||||
|
int frame_width_ = 0;
|
||||||
|
int frame_height_ = 0;
|
||||||
|
int frame_stride_ = 0;
|
||||||
|
int logical_width_ = 0;
|
||||||
|
int logical_height_ = 0;
|
||||||
|
|
||||||
|
std::vector<uint8_t> y_plane_;
|
||||||
|
std::vector<uint8_t> uv_plane_;
|
||||||
|
};
|
||||||
|
|
||||||
|
} // namespace crossdesk
|
||||||
|
|
||||||
|
#endif
|
||||||
46
src/screen_capturer/linux/screen_capturer_wayland_build.h
Normal file
46
src/screen_capturer/linux/screen_capturer_wayland_build.h
Normal file
@@ -0,0 +1,46 @@
|
|||||||
|
/*
|
||||||
|
* @Author: DI JUNKUN
|
||||||
|
* @Date: 2026-03-22
|
||||||
|
* Copyright (c) 2026 by DI JUNKUN, All Rights Reserved.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#ifndef _SCREEN_CAPTURER_WAYLAND_BUILD_H_
|
||||||
|
#define _SCREEN_CAPTURER_WAYLAND_BUILD_H_
|
||||||
|
|
||||||
|
#if defined(CROSSDESK_HAS_WAYLAND_CAPTURER) && CROSSDESK_HAS_WAYLAND_CAPTURER
|
||||||
|
|
||||||
|
#define CROSSDESK_WAYLAND_BUILD_ENABLED 1
|
||||||
|
|
||||||
|
#include <dbus/dbus.h>
|
||||||
|
#include <pipewire/keys.h>
|
||||||
|
#include <pipewire/pipewire.h>
|
||||||
|
#include <pipewire/stream.h>
|
||||||
|
#include <pipewire/thread-loop.h>
|
||||||
|
#include <spa/param/param.h>
|
||||||
|
#include <spa/param/format-utils.h>
|
||||||
|
#include <spa/param/video/format-utils.h>
|
||||||
|
#include <spa/param/video/raw.h>
|
||||||
|
#include <spa/buffer/meta.h>
|
||||||
|
#include <spa/utils/result.h>
|
||||||
|
|
||||||
|
#if defined(__has_include)
|
||||||
|
#if __has_include(<spa/param/buffers.h>)
|
||||||
|
#include <spa/param/buffers.h>
|
||||||
|
#endif
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#define CROSSDESK_SPA_PARAM_BUFFERS_BUFFERS 1u
|
||||||
|
#define CROSSDESK_SPA_PARAM_BUFFERS_BLOCKS 2u
|
||||||
|
#define CROSSDESK_SPA_PARAM_BUFFERS_SIZE 3u
|
||||||
|
#define CROSSDESK_SPA_PARAM_BUFFERS_STRIDE 4u
|
||||||
|
|
||||||
|
#define CROSSDESK_SPA_PARAM_META_TYPE 1u
|
||||||
|
#define CROSSDESK_SPA_PARAM_META_SIZE 2u
|
||||||
|
|
||||||
|
#else
|
||||||
|
|
||||||
|
#define CROSSDESK_WAYLAND_BUILD_ENABLED 0
|
||||||
|
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#endif
|
||||||
630
src/screen_capturer/linux/screen_capturer_wayland_pipewire.cpp
Normal file
630
src/screen_capturer/linux/screen_capturer_wayland_pipewire.cpp
Normal file
@@ -0,0 +1,630 @@
|
|||||||
|
#include "screen_capturer_wayland.h"
|
||||||
|
|
||||||
|
#include "screen_capturer_wayland_build.h"
|
||||||
|
|
||||||
|
#if CROSSDESK_WAYLAND_BUILD_ENABLED
|
||||||
|
|
||||||
|
#include <chrono>
|
||||||
|
#include <cstdint>
|
||||||
|
#include <thread>
|
||||||
|
#include <unistd.h>
|
||||||
|
|
||||||
|
#include <vector>
|
||||||
|
|
||||||
|
#include "libyuv.h"
|
||||||
|
#include "rd_log.h"
|
||||||
|
|
||||||
|
namespace crossdesk {
|
||||||
|
|
||||||
|
namespace {
|
||||||
|
|
||||||
|
const char* PipeWireFormatName(uint32_t spa_format) {
|
||||||
|
switch (spa_format) {
|
||||||
|
case SPA_VIDEO_FORMAT_BGRx:
|
||||||
|
return "BGRx";
|
||||||
|
case SPA_VIDEO_FORMAT_BGRA:
|
||||||
|
return "BGRA";
|
||||||
|
#ifdef SPA_VIDEO_FORMAT_RGBx
|
||||||
|
case SPA_VIDEO_FORMAT_RGBx:
|
||||||
|
return "RGBx";
|
||||||
|
#endif
|
||||||
|
#ifdef SPA_VIDEO_FORMAT_RGBA
|
||||||
|
case SPA_VIDEO_FORMAT_RGBA:
|
||||||
|
return "RGBA";
|
||||||
|
#endif
|
||||||
|
default:
|
||||||
|
return "unsupported";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const char* PipeWireConnectModeName(
|
||||||
|
ScreenCapturerWayland::PipeWireConnectMode mode) {
|
||||||
|
switch (mode) {
|
||||||
|
case ScreenCapturerWayland::PipeWireConnectMode::kTargetObject:
|
||||||
|
return "target-object";
|
||||||
|
case ScreenCapturerWayland::PipeWireConnectMode::kNodeId:
|
||||||
|
return "node-id";
|
||||||
|
case ScreenCapturerWayland::PipeWireConnectMode::kAny:
|
||||||
|
return "any";
|
||||||
|
default:
|
||||||
|
return "unknown";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
int64_t NowMs() {
|
||||||
|
return std::chrono::duration_cast<std::chrono::milliseconds>(
|
||||||
|
std::chrono::steady_clock::now().time_since_epoch())
|
||||||
|
.count();
|
||||||
|
}
|
||||||
|
|
||||||
|
struct PipeWireTargetLookupState {
|
||||||
|
pw_thread_loop* loop = nullptr;
|
||||||
|
uint32_t target_node_id = 0;
|
||||||
|
int sync_seq = -1;
|
||||||
|
bool done = false;
|
||||||
|
bool found = false;
|
||||||
|
std::string object_serial;
|
||||||
|
};
|
||||||
|
|
||||||
|
std::string LookupPipeWireTargetObjectSerial(pw_core* core,
|
||||||
|
pw_thread_loop* loop,
|
||||||
|
uint32_t node_id) {
|
||||||
|
if (!core || !loop || node_id == 0) {
|
||||||
|
return "";
|
||||||
|
}
|
||||||
|
|
||||||
|
PipeWireTargetLookupState state;
|
||||||
|
state.loop = loop;
|
||||||
|
state.target_node_id = node_id;
|
||||||
|
|
||||||
|
pw_registry* registry = pw_core_get_registry(core, PW_VERSION_REGISTRY, 0);
|
||||||
|
if (!registry) {
|
||||||
|
return "";
|
||||||
|
}
|
||||||
|
|
||||||
|
spa_hook registry_listener{};
|
||||||
|
spa_hook core_listener{};
|
||||||
|
|
||||||
|
pw_registry_events registry_events{};
|
||||||
|
registry_events.version = PW_VERSION_REGISTRY_EVENTS;
|
||||||
|
registry_events.global =
|
||||||
|
[](void* userdata, uint32_t id, uint32_t permissions, const char* type,
|
||||||
|
uint32_t version, const spa_dict* props) {
|
||||||
|
(void)permissions;
|
||||||
|
(void)version;
|
||||||
|
auto* state = static_cast<PipeWireTargetLookupState*>(userdata);
|
||||||
|
if (!state || !props || id != state->target_node_id || !type) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (strcmp(type, PW_TYPE_INTERFACE_Node) != 0) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const char* object_serial = spa_dict_lookup(props, PW_KEY_OBJECT_SERIAL);
|
||||||
|
if (!object_serial || object_serial[0] == '\0') {
|
||||||
|
object_serial = spa_dict_lookup(props, "object.serial");
|
||||||
|
}
|
||||||
|
if (!object_serial || object_serial[0] == '\0') {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
state->object_serial = object_serial;
|
||||||
|
state->found = true;
|
||||||
|
};
|
||||||
|
|
||||||
|
pw_core_events core_events{};
|
||||||
|
core_events.version = PW_VERSION_CORE_EVENTS;
|
||||||
|
core_events.done = [](void* userdata, uint32_t id, int seq) {
|
||||||
|
auto* state = static_cast<PipeWireTargetLookupState*>(userdata);
|
||||||
|
if (!state || id != PW_ID_CORE || seq != state->sync_seq) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
state->done = true;
|
||||||
|
pw_thread_loop_signal(state->loop, false);
|
||||||
|
};
|
||||||
|
core_events.error = [](void* userdata, uint32_t id, int seq, int res,
|
||||||
|
const char* message) {
|
||||||
|
(void)id;
|
||||||
|
(void)seq;
|
||||||
|
(void)res;
|
||||||
|
auto* state = static_cast<PipeWireTargetLookupState*>(userdata);
|
||||||
|
if (!state) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
LOG_WARN("PipeWire registry lookup error: {}",
|
||||||
|
message ? message : "unknown");
|
||||||
|
state->done = true;
|
||||||
|
pw_thread_loop_signal(state->loop, false);
|
||||||
|
};
|
||||||
|
|
||||||
|
pw_registry_add_listener(registry, ®istry_listener, ®istry_events,
|
||||||
|
&state);
|
||||||
|
pw_core_add_listener(core, &core_listener, &core_events, &state);
|
||||||
|
state.sync_seq = pw_core_sync(core, PW_ID_CORE, 0);
|
||||||
|
|
||||||
|
while (!state.done) {
|
||||||
|
pw_thread_loop_wait(loop);
|
||||||
|
}
|
||||||
|
|
||||||
|
spa_hook_remove(®istry_listener);
|
||||||
|
spa_hook_remove(&core_listener);
|
||||||
|
pw_proxy_destroy(reinterpret_cast<pw_proxy*>(registry));
|
||||||
|
return state.found ? state.object_serial : "";
|
||||||
|
}
|
||||||
|
|
||||||
|
int BytesPerPixel(uint32_t spa_format) {
|
||||||
|
switch (spa_format) {
|
||||||
|
case SPA_VIDEO_FORMAT_BGRx:
|
||||||
|
case SPA_VIDEO_FORMAT_BGRA:
|
||||||
|
#ifdef SPA_VIDEO_FORMAT_RGBx
|
||||||
|
case SPA_VIDEO_FORMAT_RGBx:
|
||||||
|
#endif
|
||||||
|
#ifdef SPA_VIDEO_FORMAT_RGBA
|
||||||
|
case SPA_VIDEO_FORMAT_RGBA:
|
||||||
|
#endif
|
||||||
|
return 4;
|
||||||
|
default:
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace
|
||||||
|
|
||||||
|
bool ScreenCapturerWayland::SetupPipeWireStream(bool relaxed_connect,
|
||||||
|
PipeWireConnectMode mode) {
|
||||||
|
if (pipewire_fd_ < 0 || pipewire_node_id_ == 0) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!pipewire_initialized_) {
|
||||||
|
pw_init(nullptr, nullptr);
|
||||||
|
pipewire_initialized_ = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
pw_thread_loop_ = pw_thread_loop_new("crossdesk-wayland-capture", nullptr);
|
||||||
|
if (!pw_thread_loop_) {
|
||||||
|
LOG_ERROR("Failed to create PipeWire thread loop");
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (pw_thread_loop_start(pw_thread_loop_) < 0) {
|
||||||
|
LOG_ERROR("Failed to start PipeWire thread loop");
|
||||||
|
CleanupPipeWire();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
pipewire_thread_loop_started_ = true;
|
||||||
|
|
||||||
|
pw_thread_loop_lock(pw_thread_loop_);
|
||||||
|
|
||||||
|
pw_context_ =
|
||||||
|
pw_context_new(pw_thread_loop_get_loop(pw_thread_loop_), nullptr, 0);
|
||||||
|
if (!pw_context_) {
|
||||||
|
LOG_ERROR("Failed to create PipeWire context");
|
||||||
|
pw_thread_loop_unlock(pw_thread_loop_);
|
||||||
|
CleanupPipeWire();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
pw_core_ = pw_context_connect_fd(pw_context_, pipewire_fd_, nullptr, 0);
|
||||||
|
if (!pw_core_) {
|
||||||
|
LOG_ERROR("Failed to connect to PipeWire remote");
|
||||||
|
pw_thread_loop_unlock(pw_thread_loop_);
|
||||||
|
CleanupPipeWire();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
pipewire_fd_ = -1;
|
||||||
|
|
||||||
|
pw_properties* stream_props =
|
||||||
|
pw_properties_new(PW_KEY_MEDIA_TYPE, "Video", PW_KEY_MEDIA_CATEGORY,
|
||||||
|
"Capture", PW_KEY_MEDIA_ROLE, "Screen", nullptr);
|
||||||
|
if (!stream_props) {
|
||||||
|
LOG_ERROR("Failed to allocate PipeWire stream properties");
|
||||||
|
pw_thread_loop_unlock(pw_thread_loop_);
|
||||||
|
CleanupPipeWire();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string target_object_serial;
|
||||||
|
if (mode == PipeWireConnectMode::kTargetObject) {
|
||||||
|
target_object_serial =
|
||||||
|
LookupPipeWireTargetObjectSerial(pw_core_, pw_thread_loop_,
|
||||||
|
pipewire_node_id_);
|
||||||
|
if (!target_object_serial.empty()) {
|
||||||
|
pw_properties_set(stream_props, PW_KEY_TARGET_OBJECT,
|
||||||
|
target_object_serial.c_str());
|
||||||
|
LOG_INFO("PipeWire target object serial for node {} is {}",
|
||||||
|
pipewire_node_id_, target_object_serial);
|
||||||
|
} else {
|
||||||
|
LOG_WARN("PipeWire target object serial lookup failed for node {}, "
|
||||||
|
"falling back to direct target id in target-object mode",
|
||||||
|
pipewire_node_id_);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pw_stream_ = pw_stream_new(pw_core_, "CrossDesk Wayland Capture",
|
||||||
|
stream_props);
|
||||||
|
if (!pw_stream_) {
|
||||||
|
LOG_ERROR("Failed to create PipeWire stream");
|
||||||
|
pw_thread_loop_unlock(pw_thread_loop_);
|
||||||
|
CleanupPipeWire();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
auto* listener = new spa_hook();
|
||||||
|
stream_listener_ = listener;
|
||||||
|
|
||||||
|
static const pw_stream_events stream_events = [] {
|
||||||
|
pw_stream_events events{};
|
||||||
|
events.version = PW_VERSION_STREAM_EVENTS;
|
||||||
|
events.state_changed =
|
||||||
|
[](void* userdata, enum pw_stream_state old_state,
|
||||||
|
enum pw_stream_state state, const char* error_message) {
|
||||||
|
auto* self = static_cast<ScreenCapturerWayland*>(userdata);
|
||||||
|
if (!self) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (state == PW_STREAM_STATE_ERROR) {
|
||||||
|
LOG_ERROR("PipeWire stream error: {}",
|
||||||
|
error_message ? error_message : "unknown");
|
||||||
|
self->running_ = false;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
LOG_INFO("PipeWire stream state: {} -> {}",
|
||||||
|
pw_stream_state_as_string(old_state),
|
||||||
|
pw_stream_state_as_string(state));
|
||||||
|
|
||||||
|
};
|
||||||
|
events.param_changed =
|
||||||
|
[](void* userdata, uint32_t id, const struct spa_pod* param) {
|
||||||
|
auto* self = static_cast<ScreenCapturerWayland*>(userdata);
|
||||||
|
if (!self || id != SPA_PARAM_Format || !param) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
spa_video_info_raw info{};
|
||||||
|
if (spa_format_video_raw_parse(param, &info) < 0) {
|
||||||
|
LOG_ERROR("Failed to parse PipeWire video format");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
self->spa_video_format_ = info.format;
|
||||||
|
self->frame_width_ = static_cast<int>(info.size.width);
|
||||||
|
self->frame_height_ = static_cast<int>(info.size.height);
|
||||||
|
self->frame_stride_ = static_cast<int>(info.size.width) * 4;
|
||||||
|
|
||||||
|
bool supported_format =
|
||||||
|
(self->spa_video_format_ == SPA_VIDEO_FORMAT_BGRx) ||
|
||||||
|
(self->spa_video_format_ == SPA_VIDEO_FORMAT_BGRA);
|
||||||
|
#ifdef SPA_VIDEO_FORMAT_RGBx
|
||||||
|
supported_format =
|
||||||
|
supported_format ||
|
||||||
|
(self->spa_video_format_ == SPA_VIDEO_FORMAT_RGBx);
|
||||||
|
#endif
|
||||||
|
#ifdef SPA_VIDEO_FORMAT_RGBA
|
||||||
|
supported_format =
|
||||||
|
supported_format ||
|
||||||
|
(self->spa_video_format_ == SPA_VIDEO_FORMAT_RGBA);
|
||||||
|
#endif
|
||||||
|
if (!supported_format) {
|
||||||
|
LOG_ERROR("Unsupported PipeWire pixel format: {}",
|
||||||
|
PipeWireFormatName(self->spa_video_format_));
|
||||||
|
self->running_ = false;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const int bytes_per_pixel = BytesPerPixel(self->spa_video_format_);
|
||||||
|
if (bytes_per_pixel <= 0 || self->frame_width_ <= 0 ||
|
||||||
|
self->frame_height_ <= 0) {
|
||||||
|
LOG_ERROR("Invalid PipeWire frame layout: format={}, size={}x{}",
|
||||||
|
PipeWireFormatName(self->spa_video_format_),
|
||||||
|
self->frame_width_, self->frame_height_);
|
||||||
|
self->running_ = false;
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
self->frame_stride_ = self->frame_width_ * bytes_per_pixel;
|
||||||
|
|
||||||
|
uint8_t buffer[1024];
|
||||||
|
spa_pod_builder builder = SPA_POD_BUILDER_INIT(buffer, sizeof(buffer));
|
||||||
|
const spa_pod* params[2];
|
||||||
|
uint32_t param_count = 0;
|
||||||
|
|
||||||
|
params[param_count++] = reinterpret_cast<const spa_pod*>(
|
||||||
|
spa_pod_builder_add_object(
|
||||||
|
&builder, SPA_TYPE_OBJECT_ParamBuffers, SPA_PARAM_Buffers,
|
||||||
|
CROSSDESK_SPA_PARAM_BUFFERS_BUFFERS,
|
||||||
|
SPA_POD_CHOICE_RANGE_Int(8, 4, 16),
|
||||||
|
CROSSDESK_SPA_PARAM_BUFFERS_BLOCKS, SPA_POD_Int(1),
|
||||||
|
CROSSDESK_SPA_PARAM_BUFFERS_SIZE,
|
||||||
|
SPA_POD_CHOICE_RANGE_Int(self->frame_stride_ *
|
||||||
|
self->frame_height_,
|
||||||
|
self->frame_stride_ *
|
||||||
|
self->frame_height_,
|
||||||
|
self->frame_stride_ *
|
||||||
|
self->frame_height_),
|
||||||
|
CROSSDESK_SPA_PARAM_BUFFERS_STRIDE,
|
||||||
|
SPA_POD_CHOICE_RANGE_Int(self->frame_stride_,
|
||||||
|
self->frame_stride_,
|
||||||
|
self->frame_stride_)));
|
||||||
|
|
||||||
|
params[param_count++] = reinterpret_cast<const spa_pod*>(
|
||||||
|
spa_pod_builder_add_object(
|
||||||
|
&builder, SPA_TYPE_OBJECT_ParamMeta, SPA_PARAM_Meta,
|
||||||
|
CROSSDESK_SPA_PARAM_META_TYPE, SPA_POD_Id(SPA_META_Header),
|
||||||
|
CROSSDESK_SPA_PARAM_META_SIZE,
|
||||||
|
SPA_POD_Int(sizeof(struct spa_meta_header))));
|
||||||
|
|
||||||
|
if (self->pw_stream_) {
|
||||||
|
pw_stream_update_params(self->pw_stream_, params, param_count);
|
||||||
|
}
|
||||||
|
self->pipewire_format_ready_.store(true);
|
||||||
|
|
||||||
|
const int pointer_width =
|
||||||
|
self->logical_width_ > 0 ? self->logical_width_ : self->frame_width_;
|
||||||
|
const int pointer_height = self->logical_height_ > 0
|
||||||
|
? self->logical_height_
|
||||||
|
: self->frame_height_;
|
||||||
|
self->UpdateDisplayGeometry(pointer_width, pointer_height);
|
||||||
|
LOG_INFO(
|
||||||
|
"PipeWire video format: {}, {}x{} stride={} (pointer space {}x{})",
|
||||||
|
PipeWireFormatName(self->spa_video_format_),
|
||||||
|
self->frame_width_, self->frame_height_, self->frame_stride_,
|
||||||
|
pointer_width, pointer_height);
|
||||||
|
};
|
||||||
|
events.process = [](void* userdata) {
|
||||||
|
auto* self = static_cast<ScreenCapturerWayland*>(userdata);
|
||||||
|
if (self) {
|
||||||
|
self->HandlePipeWireBuffer();
|
||||||
|
}
|
||||||
|
};
|
||||||
|
return events;
|
||||||
|
}();
|
||||||
|
|
||||||
|
pw_stream_add_listener(pw_stream_, listener, &stream_events, this);
|
||||||
|
pipewire_format_ready_.store(false);
|
||||||
|
pipewire_stream_start_ms_.store(NowMs());
|
||||||
|
pipewire_last_frame_ms_.store(0);
|
||||||
|
|
||||||
|
uint8_t buffer[4096];
|
||||||
|
spa_pod_builder builder = SPA_POD_BUILDER_INIT(buffer, sizeof(buffer));
|
||||||
|
const spa_pod* params[8];
|
||||||
|
int param_count = 0;
|
||||||
|
const spa_rectangle fixed_size{
|
||||||
|
static_cast<uint32_t>(logical_width_ > 0 ? logical_width_ : kFallbackWidth),
|
||||||
|
static_cast<uint32_t>(logical_height_ > 0 ? logical_height_
|
||||||
|
: kFallbackHeight)};
|
||||||
|
const spa_rectangle min_size{1u, 1u};
|
||||||
|
const spa_rectangle max_size{16384u, 16384u};
|
||||||
|
|
||||||
|
if (!relaxed_connect) {
|
||||||
|
auto add_format_param = [&](uint32_t spa_format) {
|
||||||
|
if (param_count >= static_cast<int>(sizeof(params) / sizeof(params[0]))) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
params[param_count++] =
|
||||||
|
reinterpret_cast<const spa_pod*>(spa_pod_builder_add_object(
|
||||||
|
&builder, SPA_TYPE_OBJECT_Format, SPA_PARAM_EnumFormat,
|
||||||
|
SPA_FORMAT_mediaType, SPA_POD_Id(SPA_MEDIA_TYPE_video),
|
||||||
|
SPA_FORMAT_mediaSubtype, SPA_POD_Id(SPA_MEDIA_SUBTYPE_raw),
|
||||||
|
SPA_FORMAT_VIDEO_format, SPA_POD_Id(spa_format),
|
||||||
|
SPA_FORMAT_VIDEO_size,
|
||||||
|
SPA_POD_CHOICE_RANGE_Rectangle(&fixed_size, &min_size,
|
||||||
|
&max_size)));
|
||||||
|
};
|
||||||
|
|
||||||
|
add_format_param(SPA_VIDEO_FORMAT_BGRx);
|
||||||
|
add_format_param(SPA_VIDEO_FORMAT_BGRA);
|
||||||
|
#ifdef SPA_VIDEO_FORMAT_RGBx
|
||||||
|
add_format_param(SPA_VIDEO_FORMAT_RGBx);
|
||||||
|
#endif
|
||||||
|
#ifdef SPA_VIDEO_FORMAT_RGBA
|
||||||
|
add_format_param(SPA_VIDEO_FORMAT_RGBA);
|
||||||
|
#endif
|
||||||
|
|
||||||
|
if (param_count == 0) {
|
||||||
|
LOG_ERROR("No valid PipeWire format params were built");
|
||||||
|
pw_thread_loop_unlock(pw_thread_loop_);
|
||||||
|
CleanupPipeWire();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
LOG_INFO("PipeWire stream using relaxed format negotiation");
|
||||||
|
}
|
||||||
|
|
||||||
|
uint32_t target_id = PW_ID_ANY;
|
||||||
|
if (mode == PipeWireConnectMode::kNodeId ||
|
||||||
|
(mode == PipeWireConnectMode::kTargetObject &&
|
||||||
|
target_object_serial.empty())) {
|
||||||
|
target_id = pipewire_node_id_;
|
||||||
|
}
|
||||||
|
LOG_INFO(
|
||||||
|
"PipeWire connecting stream: mode={}, node_id={}, target_id={}, "
|
||||||
|
"target_object_serial={}, relaxed_connect={}, param_count={}, "
|
||||||
|
"requested_size={}x{}",
|
||||||
|
PipeWireConnectModeName(mode), pipewire_node_id_, target_id,
|
||||||
|
target_object_serial.empty() ? "none" : target_object_serial.c_str(),
|
||||||
|
relaxed_connect, param_count, fixed_size.width, fixed_size.height);
|
||||||
|
const int ret = pw_stream_connect(
|
||||||
|
pw_stream_, PW_DIRECTION_INPUT, target_id,
|
||||||
|
static_cast<pw_stream_flags>(PW_STREAM_FLAG_AUTOCONNECT |
|
||||||
|
PW_STREAM_FLAG_MAP_BUFFERS),
|
||||||
|
param_count > 0 ? params : nullptr, static_cast<uint32_t>(param_count));
|
||||||
|
pw_thread_loop_unlock(pw_thread_loop_);
|
||||||
|
|
||||||
|
if (ret < 0) {
|
||||||
|
LOG_ERROR("pw_stream_connect failed: {}", spa_strerror(ret));
|
||||||
|
CleanupPipeWire();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
void ScreenCapturerWayland::CleanupPipeWire() {
|
||||||
|
const bool need_lock = pw_thread_loop_ &&
|
||||||
|
(pw_stream_ != nullptr || pw_core_ != nullptr ||
|
||||||
|
pw_context_ != nullptr);
|
||||||
|
if (need_lock) {
|
||||||
|
pw_thread_loop_lock(pw_thread_loop_);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (pw_stream_) {
|
||||||
|
pw_stream_set_active(pw_stream_, false);
|
||||||
|
pw_stream_disconnect(pw_stream_);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (stream_listener_) {
|
||||||
|
spa_hook_remove(static_cast<spa_hook*>(stream_listener_));
|
||||||
|
delete static_cast<spa_hook*>(stream_listener_);
|
||||||
|
stream_listener_ = nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (pw_stream_) {
|
||||||
|
pw_stream_destroy(pw_stream_);
|
||||||
|
pw_stream_ = nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (pw_core_) {
|
||||||
|
pw_core_disconnect(pw_core_);
|
||||||
|
pw_core_ = nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (pw_context_) {
|
||||||
|
pw_context_destroy(pw_context_);
|
||||||
|
pw_context_ = nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (need_lock) {
|
||||||
|
pw_thread_loop_unlock(pw_thread_loop_);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (pw_thread_loop_) {
|
||||||
|
if (pipewire_thread_loop_started_) {
|
||||||
|
pw_thread_loop_stop(pw_thread_loop_);
|
||||||
|
pipewire_thread_loop_started_ = false;
|
||||||
|
}
|
||||||
|
pw_thread_loop_destroy(pw_thread_loop_);
|
||||||
|
pw_thread_loop_ = nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (pipewire_fd_ >= 0) {
|
||||||
|
close(pipewire_fd_);
|
||||||
|
pipewire_fd_ = -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
pipewire_format_ready_.store(false);
|
||||||
|
pipewire_stream_start_ms_.store(0);
|
||||||
|
pipewire_last_frame_ms_.store(0);
|
||||||
|
|
||||||
|
if (pipewire_initialized_) {
|
||||||
|
pw_deinit();
|
||||||
|
pipewire_initialized_ = false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void ScreenCapturerWayland::HandlePipeWireBuffer() {
|
||||||
|
if (!pw_stream_) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
pw_buffer* buffer = pw_stream_dequeue_buffer(pw_stream_);
|
||||||
|
if (!buffer) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
auto requeue = [&]() { pw_stream_queue_buffer(pw_stream_, buffer); };
|
||||||
|
|
||||||
|
if (paused_) {
|
||||||
|
requeue();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
spa_buffer* spa_buffer = buffer->buffer;
|
||||||
|
if (!spa_buffer || spa_buffer->n_datas == 0 || !spa_buffer->datas[0].data) {
|
||||||
|
requeue();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const spa_data& data = spa_buffer->datas[0];
|
||||||
|
if (!data.chunk) {
|
||||||
|
requeue();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (frame_width_ <= 1 || frame_height_ <= 1) {
|
||||||
|
requeue();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
uint8_t* src = static_cast<uint8_t*>(data.data);
|
||||||
|
src += data.chunk->offset;
|
||||||
|
|
||||||
|
int stride = frame_stride_;
|
||||||
|
if (data.chunk->stride > 0) {
|
||||||
|
stride = data.chunk->stride;
|
||||||
|
} else if (stride <= 0) {
|
||||||
|
stride = frame_width_ * 4;
|
||||||
|
}
|
||||||
|
|
||||||
|
int even_width = frame_width_ & ~1;
|
||||||
|
int even_height = frame_height_ & ~1;
|
||||||
|
if (even_width <= 0 || even_height <= 0) {
|
||||||
|
requeue();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const size_t y_size = static_cast<size_t>(even_width) * even_height;
|
||||||
|
const size_t uv_size = y_size / 2;
|
||||||
|
if (y_plane_.size() != y_size) {
|
||||||
|
y_plane_.resize(y_size);
|
||||||
|
}
|
||||||
|
if (uv_plane_.size() != uv_size) {
|
||||||
|
uv_plane_.resize(uv_size);
|
||||||
|
}
|
||||||
|
|
||||||
|
libyuv::ARGBToNV12(src, stride, y_plane_.data(), even_width,
|
||||||
|
uv_plane_.data(), even_width, even_width, even_height);
|
||||||
|
|
||||||
|
std::vector<uint8_t> nv12;
|
||||||
|
nv12.reserve(y_plane_.size() + uv_plane_.size());
|
||||||
|
nv12.insert(nv12.end(), y_plane_.begin(), y_plane_.end());
|
||||||
|
nv12.insert(nv12.end(), uv_plane_.begin(), uv_plane_.end());
|
||||||
|
|
||||||
|
if (callback_) {
|
||||||
|
callback_(nv12.data(), static_cast<int>(nv12.size()), even_width,
|
||||||
|
even_height, display_name_.c_str());
|
||||||
|
}
|
||||||
|
pipewire_last_frame_ms_.store(NowMs());
|
||||||
|
|
||||||
|
requeue();
|
||||||
|
}
|
||||||
|
|
||||||
|
void ScreenCapturerWayland::UpdateDisplayGeometry(int width, int height) {
|
||||||
|
if (width <= 0 || height <= 0) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
void* stream_handle =
|
||||||
|
reinterpret_cast<void*>(static_cast<uintptr_t>(pipewire_node_id_));
|
||||||
|
|
||||||
|
if (display_info_list_.empty()) {
|
||||||
|
display_info_list_.push_back(
|
||||||
|
DisplayInfo(stream_handle, display_name_, true, 0, 0, width, height));
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
auto& display = display_info_list_[0];
|
||||||
|
display.handle = stream_handle;
|
||||||
|
display.left = 0;
|
||||||
|
display.top = 0;
|
||||||
|
display.right = width;
|
||||||
|
display.bottom = height;
|
||||||
|
display.width = width;
|
||||||
|
display.height = height;
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace crossdesk
|
||||||
|
|
||||||
|
#endif
|
||||||
816
src/screen_capturer/linux/screen_capturer_wayland_portal.cpp
Normal file
816
src/screen_capturer/linux/screen_capturer_wayland_portal.cpp
Normal file
@@ -0,0 +1,816 @@
|
|||||||
|
#include "screen_capturer_wayland.h"
|
||||||
|
|
||||||
|
#include "screen_capturer_wayland_build.h"
|
||||||
|
#include "wayland_portal_shared.h"
|
||||||
|
|
||||||
|
#if CROSSDESK_WAYLAND_BUILD_ENABLED
|
||||||
|
|
||||||
|
#include <chrono>
|
||||||
|
#include <cstring>
|
||||||
|
#include <functional>
|
||||||
|
#include <string>
|
||||||
|
#include <unistd.h>
|
||||||
|
|
||||||
|
#include "rd_log.h"
|
||||||
|
|
||||||
|
namespace crossdesk {
|
||||||
|
|
||||||
|
namespace {
|
||||||
|
|
||||||
|
constexpr const char* kPortalBusName = "org.freedesktop.portal.Desktop";
|
||||||
|
constexpr const char* kPortalObjectPath = "/org/freedesktop/portal/desktop";
|
||||||
|
constexpr const char* kPortalRemoteDesktopInterface =
|
||||||
|
"org.freedesktop.portal.RemoteDesktop";
|
||||||
|
constexpr const char* kPortalScreenCastInterface =
|
||||||
|
"org.freedesktop.portal.ScreenCast";
|
||||||
|
constexpr const char* kPortalRequestInterface =
|
||||||
|
"org.freedesktop.portal.Request";
|
||||||
|
constexpr const char* kPortalSessionInterface =
|
||||||
|
"org.freedesktop.portal.Session";
|
||||||
|
constexpr const char* kPortalRequestPathPrefix =
|
||||||
|
"/org/freedesktop/portal/desktop/request/";
|
||||||
|
constexpr const char* kPortalSessionPathPrefix =
|
||||||
|
"/org/freedesktop/portal/desktop/session/";
|
||||||
|
|
||||||
|
constexpr uint32_t kScreenCastSourceMonitor = 1u;
|
||||||
|
constexpr uint32_t kCursorModeHidden = 1u;
|
||||||
|
constexpr uint32_t kCursorModeEmbedded = 2u;
|
||||||
|
constexpr uint32_t kRemoteDesktopDevicePointer = 2u;
|
||||||
|
|
||||||
|
std::string MakeToken(const char* prefix) {
|
||||||
|
const auto now = std::chrono::steady_clock::now().time_since_epoch().count();
|
||||||
|
return std::string(prefix) + "_" + std::to_string(now);
|
||||||
|
}
|
||||||
|
|
||||||
|
void LogDbusError(const char* action, DBusError* error) {
|
||||||
|
if (error && dbus_error_is_set(error)) {
|
||||||
|
LOG_ERROR("{} failed: {} ({})", action,
|
||||||
|
error->message ? error->message : "unknown",
|
||||||
|
error->name ? error->name : "unknown");
|
||||||
|
} else {
|
||||||
|
LOG_ERROR("{} failed", action);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void AppendDictEntryString(DBusMessageIter* dict, const char* key,
|
||||||
|
const std::string& value) {
|
||||||
|
DBusMessageIter entry;
|
||||||
|
DBusMessageIter variant;
|
||||||
|
const char* key_cstr = key;
|
||||||
|
const char* value_cstr = value.c_str();
|
||||||
|
|
||||||
|
dbus_message_iter_open_container(dict, DBUS_TYPE_DICT_ENTRY, nullptr, &entry);
|
||||||
|
dbus_message_iter_append_basic(&entry, DBUS_TYPE_STRING, &key_cstr);
|
||||||
|
dbus_message_iter_open_container(&entry, DBUS_TYPE_VARIANT, "s", &variant);
|
||||||
|
dbus_message_iter_append_basic(&variant, DBUS_TYPE_STRING, &value_cstr);
|
||||||
|
dbus_message_iter_close_container(&entry, &variant);
|
||||||
|
dbus_message_iter_close_container(dict, &entry);
|
||||||
|
}
|
||||||
|
|
||||||
|
void AppendDictEntryUint32(DBusMessageIter* dict, const char* key,
|
||||||
|
uint32_t value) {
|
||||||
|
DBusMessageIter entry;
|
||||||
|
DBusMessageIter variant;
|
||||||
|
const char* key_cstr = key;
|
||||||
|
|
||||||
|
dbus_message_iter_open_container(dict, DBUS_TYPE_DICT_ENTRY, nullptr, &entry);
|
||||||
|
dbus_message_iter_append_basic(&entry, DBUS_TYPE_STRING, &key_cstr);
|
||||||
|
dbus_message_iter_open_container(&entry, DBUS_TYPE_VARIANT, "u", &variant);
|
||||||
|
dbus_message_iter_append_basic(&variant, DBUS_TYPE_UINT32, &value);
|
||||||
|
dbus_message_iter_close_container(&entry, &variant);
|
||||||
|
dbus_message_iter_close_container(dict, &entry);
|
||||||
|
}
|
||||||
|
|
||||||
|
void AppendDictEntryBool(DBusMessageIter* dict, const char* key, bool value) {
|
||||||
|
DBusMessageIter entry;
|
||||||
|
DBusMessageIter variant;
|
||||||
|
const char* key_cstr = key;
|
||||||
|
dbus_bool_t bool_value = value ? TRUE : FALSE;
|
||||||
|
|
||||||
|
dbus_message_iter_open_container(dict, DBUS_TYPE_DICT_ENTRY, nullptr, &entry);
|
||||||
|
dbus_message_iter_append_basic(&entry, DBUS_TYPE_STRING, &key_cstr);
|
||||||
|
dbus_message_iter_open_container(&entry, DBUS_TYPE_VARIANT, "b", &variant);
|
||||||
|
dbus_message_iter_append_basic(&variant, DBUS_TYPE_BOOLEAN, &bool_value);
|
||||||
|
dbus_message_iter_close_container(&entry, &variant);
|
||||||
|
dbus_message_iter_close_container(dict, &entry);
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ReadIntLike(DBusMessageIter* iter, int* value) {
|
||||||
|
if (!iter || !value) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
const int type = dbus_message_iter_get_arg_type(iter);
|
||||||
|
if (type == DBUS_TYPE_INT32) {
|
||||||
|
int32_t temp = 0;
|
||||||
|
dbus_message_iter_get_basic(iter, &temp);
|
||||||
|
*value = static_cast<int>(temp);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (type == DBUS_TYPE_UINT32) {
|
||||||
|
uint32_t temp = 0;
|
||||||
|
dbus_message_iter_get_basic(iter, &temp);
|
||||||
|
*value = static_cast<int>(temp);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ReadPathLikeVariant(DBusMessageIter* variant, std::string* value) {
|
||||||
|
if (!variant || !value) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
const int type = dbus_message_iter_get_arg_type(variant);
|
||||||
|
if (type == DBUS_TYPE_OBJECT_PATH || type == DBUS_TYPE_STRING) {
|
||||||
|
const char* temp = nullptr;
|
||||||
|
dbus_message_iter_get_basic(variant, &temp);
|
||||||
|
if (temp && temp[0] != '\0') {
|
||||||
|
*value = temp;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string BuildSessionHandleFromRequestPath(
|
||||||
|
const std::string& request_path, const std::string& session_handle_token) {
|
||||||
|
if (request_path.rfind(kPortalRequestPathPrefix, 0) != 0 ||
|
||||||
|
session_handle_token.empty()) {
|
||||||
|
return "";
|
||||||
|
}
|
||||||
|
|
||||||
|
const size_t sender_start = strlen(kPortalRequestPathPrefix);
|
||||||
|
const size_t token_sep = request_path.find('/', sender_start);
|
||||||
|
if (token_sep == std::string::npos || token_sep <= sender_start) {
|
||||||
|
return "";
|
||||||
|
}
|
||||||
|
|
||||||
|
const std::string sender = request_path.substr(sender_start,
|
||||||
|
token_sep - sender_start);
|
||||||
|
if (sender.empty()) {
|
||||||
|
return "";
|
||||||
|
}
|
||||||
|
|
||||||
|
return std::string(kPortalSessionPathPrefix) + sender + "/" +
|
||||||
|
session_handle_token;
|
||||||
|
}
|
||||||
|
|
||||||
|
struct PortalResponseState {
|
||||||
|
std::string request_path;
|
||||||
|
bool received = false;
|
||||||
|
DBusMessage* message = nullptr;
|
||||||
|
};
|
||||||
|
|
||||||
|
DBusHandlerResult HandlePortalResponseSignal(DBusConnection* connection,
|
||||||
|
DBusMessage* message,
|
||||||
|
void* user_data) {
|
||||||
|
auto* state = static_cast<PortalResponseState*>(user_data);
|
||||||
|
if (!state || !message) {
|
||||||
|
return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!dbus_message_is_signal(message, kPortalRequestInterface, "Response")) {
|
||||||
|
return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
|
||||||
|
}
|
||||||
|
|
||||||
|
const char* path = dbus_message_get_path(message);
|
||||||
|
if (!path || state->request_path != path) {
|
||||||
|
return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (state->message) {
|
||||||
|
dbus_message_unref(state->message);
|
||||||
|
state->message = nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
state->message = dbus_message_ref(message);
|
||||||
|
state->received = true;
|
||||||
|
return DBUS_HANDLER_RESULT_HANDLED;
|
||||||
|
}
|
||||||
|
|
||||||
|
DBusMessage* WaitForPortalResponse(DBusConnection* connection,
|
||||||
|
const std::string& request_path,
|
||||||
|
const std::atomic<bool>& running,
|
||||||
|
int timeout_ms = 120000) {
|
||||||
|
if (!connection || request_path.empty()) {
|
||||||
|
return nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
PortalResponseState state;
|
||||||
|
state.request_path = request_path;
|
||||||
|
|
||||||
|
DBusError error;
|
||||||
|
dbus_error_init(&error);
|
||||||
|
|
||||||
|
const std::string match_rule =
|
||||||
|
"type='signal',interface='" + std::string(kPortalRequestInterface) +
|
||||||
|
"',member='Response',path='" + request_path + "'";
|
||||||
|
dbus_bus_add_match(connection, match_rule.c_str(), &error);
|
||||||
|
if (dbus_error_is_set(&error)) {
|
||||||
|
LogDbusError("dbus_bus_add_match", &error);
|
||||||
|
dbus_error_free(&error);
|
||||||
|
return nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
dbus_connection_add_filter(connection, HandlePortalResponseSignal, &state,
|
||||||
|
nullptr);
|
||||||
|
|
||||||
|
auto deadline =
|
||||||
|
std::chrono::steady_clock::now() + std::chrono::milliseconds(timeout_ms);
|
||||||
|
while (running.load() && !state.received &&
|
||||||
|
std::chrono::steady_clock::now() < deadline) {
|
||||||
|
dbus_connection_read_write(connection, 100);
|
||||||
|
while (dbus_connection_dispatch(connection) == DBUS_DISPATCH_DATA_REMAINS) {
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
dbus_connection_remove_filter(connection, HandlePortalResponseSignal, &state);
|
||||||
|
|
||||||
|
DBusError remove_error;
|
||||||
|
dbus_error_init(&remove_error);
|
||||||
|
dbus_bus_remove_match(connection, match_rule.c_str(), &remove_error);
|
||||||
|
if (dbus_error_is_set(&remove_error)) {
|
||||||
|
dbus_error_free(&remove_error);
|
||||||
|
}
|
||||||
|
|
||||||
|
return state.message;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ExtractRequestPath(DBusMessage* reply, std::string* request_path) {
|
||||||
|
if (!reply || !request_path) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
const char* path = nullptr;
|
||||||
|
DBusError error;
|
||||||
|
dbus_error_init(&error);
|
||||||
|
const dbus_bool_t ok = dbus_message_get_args(
|
||||||
|
reply, &error, DBUS_TYPE_OBJECT_PATH, &path, DBUS_TYPE_INVALID);
|
||||||
|
if (!ok || !path) {
|
||||||
|
LogDbusError("dbus_message_get_args(request_path)", &error);
|
||||||
|
dbus_error_free(&error);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
*request_path = path;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ExtractPortalResponse(DBusMessage* message, uint32_t* response_code,
|
||||||
|
DBusMessageIter* results_array) {
|
||||||
|
if (!message || !response_code || !results_array) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
DBusMessageIter iter;
|
||||||
|
if (!dbus_message_iter_init(message, &iter) ||
|
||||||
|
dbus_message_iter_get_arg_type(&iter) != DBUS_TYPE_UINT32) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
dbus_message_iter_get_basic(&iter, response_code);
|
||||||
|
if (!dbus_message_iter_next(&iter) ||
|
||||||
|
dbus_message_iter_get_arg_type(&iter) != DBUS_TYPE_ARRAY) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
*results_array = iter;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool SendPortalRequestAndHandleResponse(
|
||||||
|
DBusConnection* connection, const char* interface_name,
|
||||||
|
const char* method_name,
|
||||||
|
const char* action_name,
|
||||||
|
const std::function<bool(DBusMessage*)>& append_message_args,
|
||||||
|
const std::atomic<bool>& running,
|
||||||
|
const std::function<bool(uint32_t, DBusMessageIter*)>& handle_results,
|
||||||
|
std::string* request_path_out = nullptr) {
|
||||||
|
if (!connection || !interface_name || interface_name[0] == '\0' ||
|
||||||
|
!method_name || method_name[0] == '\0') {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
DBusMessage* message =
|
||||||
|
dbus_message_new_method_call(kPortalBusName, kPortalObjectPath,
|
||||||
|
interface_name, method_name);
|
||||||
|
if (!message) {
|
||||||
|
LOG_ERROR("Failed to allocate {} message", method_name);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (append_message_args && !append_message_args(message)) {
|
||||||
|
dbus_message_unref(message);
|
||||||
|
LOG_ERROR("{} arguments are malformed", method_name);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
DBusError error;
|
||||||
|
dbus_error_init(&error);
|
||||||
|
DBusMessage* reply =
|
||||||
|
dbus_connection_send_with_reply_and_block(connection, message, -1, &error);
|
||||||
|
dbus_message_unref(message);
|
||||||
|
if (!reply) {
|
||||||
|
LogDbusError(action_name ? action_name : method_name, &error);
|
||||||
|
dbus_error_free(&error);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
std::string request_path;
|
||||||
|
const bool got_request_path = ExtractRequestPath(reply, &request_path);
|
||||||
|
dbus_message_unref(reply);
|
||||||
|
if (!got_request_path) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
if (request_path_out) {
|
||||||
|
*request_path_out = request_path;
|
||||||
|
}
|
||||||
|
|
||||||
|
DBusMessage* response =
|
||||||
|
WaitForPortalResponse(connection, request_path, running);
|
||||||
|
if (!response) {
|
||||||
|
LOG_ERROR("Timed out waiting for {} response", method_name);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
uint32_t response_code = 1;
|
||||||
|
DBusMessageIter results;
|
||||||
|
const bool parsed = ExtractPortalResponse(response, &response_code, &results);
|
||||||
|
if (!parsed) {
|
||||||
|
dbus_message_unref(response);
|
||||||
|
LOG_ERROR("{} response was malformed", method_name);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
const bool ok = handle_results ? handle_results(response_code, &results)
|
||||||
|
: (response_code == 0);
|
||||||
|
dbus_message_unref(response);
|
||||||
|
return ok;
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace
|
||||||
|
|
||||||
|
bool ScreenCapturerWayland::CheckPortalAvailability() const {
|
||||||
|
DBusError error;
|
||||||
|
dbus_error_init(&error);
|
||||||
|
|
||||||
|
DBusConnection* connection = dbus_bus_get(DBUS_BUS_SESSION, &error);
|
||||||
|
if (!connection) {
|
||||||
|
LogDbusError("dbus_bus_get", &error);
|
||||||
|
dbus_error_free(&error);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
const dbus_bool_t has_owner = dbus_bus_name_has_owner(
|
||||||
|
connection, kPortalBusName, &error);
|
||||||
|
if (dbus_error_is_set(&error)) {
|
||||||
|
LogDbusError("dbus_bus_name_has_owner", &error);
|
||||||
|
dbus_error_free(&error);
|
||||||
|
dbus_connection_unref(connection);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
dbus_connection_unref(connection);
|
||||||
|
return has_owner == TRUE;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ScreenCapturerWayland::ConnectSessionBus() {
|
||||||
|
if (dbus_connection_) {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
DBusError error;
|
||||||
|
dbus_error_init(&error);
|
||||||
|
|
||||||
|
dbus_connection_ = dbus_bus_get_private(DBUS_BUS_SESSION, &error);
|
||||||
|
if (!dbus_connection_) {
|
||||||
|
LogDbusError("dbus_bus_get_private", &error);
|
||||||
|
dbus_error_free(&error);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
dbus_connection_set_exit_on_disconnect(dbus_connection_, FALSE);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ScreenCapturerWayland::CreatePortalSession() {
|
||||||
|
if (!dbus_connection_) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
const std::string session_handle_token = MakeToken("crossdesk_session");
|
||||||
|
std::string request_path;
|
||||||
|
const bool ok = SendPortalRequestAndHandleResponse(
|
||||||
|
dbus_connection_, kPortalRemoteDesktopInterface, "CreateSession",
|
||||||
|
"CreateSession",
|
||||||
|
[&](DBusMessage* message) {
|
||||||
|
DBusMessageIter iter;
|
||||||
|
DBusMessageIter options;
|
||||||
|
dbus_message_iter_init_append(message, &iter);
|
||||||
|
dbus_message_iter_open_container(&iter, DBUS_TYPE_ARRAY, "{sv}",
|
||||||
|
&options);
|
||||||
|
AppendDictEntryString(&options, "session_handle_token",
|
||||||
|
session_handle_token);
|
||||||
|
AppendDictEntryString(&options, "handle_token", MakeToken("crossdesk_req"));
|
||||||
|
dbus_message_iter_close_container(&iter, &options);
|
||||||
|
return true;
|
||||||
|
},
|
||||||
|
running_,
|
||||||
|
[&](uint32_t response_code, DBusMessageIter* results) {
|
||||||
|
if (response_code != 0) {
|
||||||
|
LOG_ERROR("CreateSession was denied or malformed, response={}",
|
||||||
|
response_code);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
DBusMessageIter dict;
|
||||||
|
dbus_message_iter_recurse(results, &dict);
|
||||||
|
while (dbus_message_iter_get_arg_type(&dict) != DBUS_TYPE_INVALID) {
|
||||||
|
if (dbus_message_iter_get_arg_type(&dict) == DBUS_TYPE_DICT_ENTRY) {
|
||||||
|
DBusMessageIter entry;
|
||||||
|
dbus_message_iter_recurse(&dict, &entry);
|
||||||
|
|
||||||
|
const char* key = nullptr;
|
||||||
|
dbus_message_iter_get_basic(&entry, &key);
|
||||||
|
if (key && dbus_message_iter_next(&entry) &&
|
||||||
|
dbus_message_iter_get_arg_type(&entry) == DBUS_TYPE_VARIANT &&
|
||||||
|
strcmp(key, "session_handle") == 0) {
|
||||||
|
DBusMessageIter variant;
|
||||||
|
std::string parsed_handle;
|
||||||
|
dbus_message_iter_recurse(&entry, &variant);
|
||||||
|
if (ReadPathLikeVariant(&variant, &parsed_handle) &&
|
||||||
|
!parsed_handle.empty()) {
|
||||||
|
session_handle_ = parsed_handle;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
dbus_message_iter_next(&dict);
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
},
|
||||||
|
&request_path);
|
||||||
|
if (!ok) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (session_handle_.empty()) {
|
||||||
|
const std::string fallback_handle = BuildSessionHandleFromRequestPath(
|
||||||
|
request_path, session_handle_token);
|
||||||
|
if (!fallback_handle.empty()) {
|
||||||
|
LOG_WARN(
|
||||||
|
"CreateSession response missing session_handle, using derived handle "
|
||||||
|
"{}",
|
||||||
|
fallback_handle);
|
||||||
|
session_handle_ = fallback_handle;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (session_handle_.empty()) {
|
||||||
|
LOG_ERROR("CreateSession response did not include a session handle");
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ScreenCapturerWayland::SelectPortalSource() {
|
||||||
|
if (!dbus_connection_ || session_handle_.empty()) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
const char* session_handle = session_handle_.c_str();
|
||||||
|
return SendPortalRequestAndHandleResponse(
|
||||||
|
dbus_connection_, kPortalScreenCastInterface, "SelectSources",
|
||||||
|
"SelectSources",
|
||||||
|
[&](DBusMessage* message) {
|
||||||
|
DBusMessageIter iter;
|
||||||
|
DBusMessageIter options;
|
||||||
|
dbus_message_iter_init_append(message, &iter);
|
||||||
|
dbus_message_iter_append_basic(&iter, DBUS_TYPE_OBJECT_PATH,
|
||||||
|
&session_handle);
|
||||||
|
dbus_message_iter_open_container(&iter, DBUS_TYPE_ARRAY, "{sv}",
|
||||||
|
&options);
|
||||||
|
AppendDictEntryUint32(&options, "types", kScreenCastSourceMonitor);
|
||||||
|
AppendDictEntryBool(&options, "multiple", false);
|
||||||
|
AppendDictEntryUint32(
|
||||||
|
&options, "cursor_mode",
|
||||||
|
show_cursor_ ? kCursorModeEmbedded : kCursorModeHidden);
|
||||||
|
AppendDictEntryString(&options, "handle_token",
|
||||||
|
MakeToken("crossdesk_req"));
|
||||||
|
dbus_message_iter_close_container(&iter, &options);
|
||||||
|
return true;
|
||||||
|
},
|
||||||
|
running_, [](uint32_t response_code, DBusMessageIter*) {
|
||||||
|
if (response_code != 0) {
|
||||||
|
LOG_ERROR("SelectSources was denied or malformed, response={}",
|
||||||
|
response_code);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ScreenCapturerWayland::SelectPortalDevices() {
|
||||||
|
if (!dbus_connection_ || session_handle_.empty()) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
const char* session_handle = session_handle_.c_str();
|
||||||
|
return SendPortalRequestAndHandleResponse(
|
||||||
|
dbus_connection_, kPortalRemoteDesktopInterface, "SelectDevices",
|
||||||
|
"SelectDevices",
|
||||||
|
[&](DBusMessage* message) {
|
||||||
|
DBusMessageIter iter;
|
||||||
|
DBusMessageIter options;
|
||||||
|
dbus_message_iter_init_append(message, &iter);
|
||||||
|
dbus_message_iter_append_basic(&iter, DBUS_TYPE_OBJECT_PATH,
|
||||||
|
&session_handle);
|
||||||
|
dbus_message_iter_open_container(&iter, DBUS_TYPE_ARRAY, "{sv}",
|
||||||
|
&options);
|
||||||
|
AppendDictEntryUint32(&options, "types", kRemoteDesktopDevicePointer);
|
||||||
|
AppendDictEntryString(&options, "handle_token",
|
||||||
|
MakeToken("crossdesk_req"));
|
||||||
|
dbus_message_iter_close_container(&iter, &options);
|
||||||
|
return true;
|
||||||
|
},
|
||||||
|
running_, [](uint32_t response_code, DBusMessageIter*) {
|
||||||
|
if (response_code != 0) {
|
||||||
|
LOG_ERROR("SelectDevices was denied or malformed, response={}",
|
||||||
|
response_code);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ScreenCapturerWayland::StartPortalSession() {
|
||||||
|
if (!dbus_connection_ || session_handle_.empty()) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
const char* session_handle = session_handle_.c_str();
|
||||||
|
const char* parent_window = "";
|
||||||
|
pointer_granted_ = false;
|
||||||
|
const bool ok = SendPortalRequestAndHandleResponse(
|
||||||
|
dbus_connection_, kPortalRemoteDesktopInterface, "Start", "Start",
|
||||||
|
[&](DBusMessage* message) {
|
||||||
|
DBusMessageIter iter;
|
||||||
|
DBusMessageIter options;
|
||||||
|
dbus_message_iter_init_append(message, &iter);
|
||||||
|
dbus_message_iter_append_basic(&iter, DBUS_TYPE_OBJECT_PATH,
|
||||||
|
&session_handle);
|
||||||
|
dbus_message_iter_append_basic(&iter, DBUS_TYPE_STRING, &parent_window);
|
||||||
|
dbus_message_iter_open_container(&iter, DBUS_TYPE_ARRAY, "{sv}",
|
||||||
|
&options);
|
||||||
|
AppendDictEntryString(&options, "handle_token", MakeToken("crossdesk_req"));
|
||||||
|
dbus_message_iter_close_container(&iter, &options);
|
||||||
|
return true;
|
||||||
|
},
|
||||||
|
running_,
|
||||||
|
[&](uint32_t response_code, DBusMessageIter* results) {
|
||||||
|
if (response_code != 0) {
|
||||||
|
LOG_ERROR("Start was denied or malformed, response={}", response_code);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
uint32_t granted_devices = 0;
|
||||||
|
DBusMessageIter dict;
|
||||||
|
dbus_message_iter_recurse(results, &dict);
|
||||||
|
while (dbus_message_iter_get_arg_type(&dict) != DBUS_TYPE_INVALID) {
|
||||||
|
if (dbus_message_iter_get_arg_type(&dict) == DBUS_TYPE_DICT_ENTRY) {
|
||||||
|
DBusMessageIter entry;
|
||||||
|
dbus_message_iter_recurse(&dict, &entry);
|
||||||
|
|
||||||
|
const char* key = nullptr;
|
||||||
|
dbus_message_iter_get_basic(&entry, &key);
|
||||||
|
if (key && dbus_message_iter_next(&entry) &&
|
||||||
|
dbus_message_iter_get_arg_type(&entry) == DBUS_TYPE_VARIANT) {
|
||||||
|
DBusMessageIter variant;
|
||||||
|
dbus_message_iter_recurse(&entry, &variant);
|
||||||
|
if (strcmp(key, "devices") == 0) {
|
||||||
|
int granted_devices_int = 0;
|
||||||
|
if (ReadIntLike(&variant, &granted_devices_int) &&
|
||||||
|
granted_devices_int >= 0) {
|
||||||
|
granted_devices = static_cast<uint32_t>(granted_devices_int);
|
||||||
|
}
|
||||||
|
} else if (strcmp(key, "streams") == 0) {
|
||||||
|
DBusMessageIter streams;
|
||||||
|
dbus_message_iter_recurse(&variant, &streams);
|
||||||
|
|
||||||
|
if (dbus_message_iter_get_arg_type(&streams) == DBUS_TYPE_STRUCT) {
|
||||||
|
DBusMessageIter stream;
|
||||||
|
dbus_message_iter_recurse(&streams, &stream);
|
||||||
|
|
||||||
|
if (dbus_message_iter_get_arg_type(&stream) == DBUS_TYPE_UINT32) {
|
||||||
|
dbus_message_iter_get_basic(&stream, &pipewire_node_id_);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (dbus_message_iter_next(&stream) &&
|
||||||
|
dbus_message_iter_get_arg_type(&stream) == DBUS_TYPE_ARRAY) {
|
||||||
|
DBusMessageIter props;
|
||||||
|
int stream_width = 0;
|
||||||
|
int stream_height = 0;
|
||||||
|
int logical_width = 0;
|
||||||
|
int logical_height = 0;
|
||||||
|
dbus_message_iter_recurse(&stream, &props);
|
||||||
|
while (dbus_message_iter_get_arg_type(&props) !=
|
||||||
|
DBUS_TYPE_INVALID) {
|
||||||
|
if (dbus_message_iter_get_arg_type(&props) ==
|
||||||
|
DBUS_TYPE_DICT_ENTRY) {
|
||||||
|
DBusMessageIter prop_entry;
|
||||||
|
dbus_message_iter_recurse(&props, &prop_entry);
|
||||||
|
|
||||||
|
const char* prop_key = nullptr;
|
||||||
|
dbus_message_iter_get_basic(&prop_entry, &prop_key);
|
||||||
|
if (prop_key && dbus_message_iter_next(&prop_entry) &&
|
||||||
|
dbus_message_iter_get_arg_type(&prop_entry) ==
|
||||||
|
DBUS_TYPE_VARIANT) {
|
||||||
|
DBusMessageIter prop_variant;
|
||||||
|
dbus_message_iter_recurse(&prop_entry, &prop_variant);
|
||||||
|
if (dbus_message_iter_get_arg_type(&prop_variant) ==
|
||||||
|
DBUS_TYPE_STRUCT) {
|
||||||
|
DBusMessageIter size_iter;
|
||||||
|
int width = 0;
|
||||||
|
int height = 0;
|
||||||
|
dbus_message_iter_recurse(&prop_variant, &size_iter);
|
||||||
|
if (ReadIntLike(&size_iter, &width) &&
|
||||||
|
dbus_message_iter_next(&size_iter) &&
|
||||||
|
ReadIntLike(&size_iter, &height)) {
|
||||||
|
if (strcmp(prop_key, "logical_size") == 0) {
|
||||||
|
logical_width = width;
|
||||||
|
logical_height = height;
|
||||||
|
} else if (strcmp(prop_key, "size") == 0) {
|
||||||
|
stream_width = width;
|
||||||
|
stream_height = height;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
dbus_message_iter_next(&props);
|
||||||
|
}
|
||||||
|
|
||||||
|
const int picked_width =
|
||||||
|
logical_width > 0 ? logical_width : stream_width;
|
||||||
|
const int picked_height =
|
||||||
|
logical_height > 0 ? logical_height : stream_height;
|
||||||
|
LOG_INFO(
|
||||||
|
"Wayland portal stream geometry: stream_size={}x{}, "
|
||||||
|
"logical_size={}x{}, pointer_space={}x{}",
|
||||||
|
stream_width, stream_height, logical_width,
|
||||||
|
logical_height, picked_width, picked_height);
|
||||||
|
|
||||||
|
if (logical_width > 0 && logical_height > 0) {
|
||||||
|
logical_width_ = logical_width;
|
||||||
|
logical_height_ = logical_height;
|
||||||
|
UpdateDisplayGeometry(logical_width_, logical_height_);
|
||||||
|
} else if (stream_width > 0 && stream_height > 0) {
|
||||||
|
logical_width_ = stream_width;
|
||||||
|
logical_height_ = stream_height;
|
||||||
|
UpdateDisplayGeometry(logical_width_, logical_height_);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
dbus_message_iter_next(&dict);
|
||||||
|
}
|
||||||
|
pointer_granted_ =
|
||||||
|
(granted_devices & kRemoteDesktopDevicePointer) != 0;
|
||||||
|
return true;
|
||||||
|
});
|
||||||
|
if (!ok) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (pipewire_node_id_ == 0) {
|
||||||
|
LOG_ERROR("Start response did not include a PipeWire node id");
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
if (!pointer_granted_) {
|
||||||
|
LOG_ERROR("Start response did not grant pointer control");
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
shared_session_registered_ = PublishSharedWaylandPortalSession(
|
||||||
|
SharedWaylandPortalSessionInfo{
|
||||||
|
dbus_connection_, session_handle_, pipewire_node_id_, logical_width_,
|
||||||
|
logical_height_, pointer_granted_});
|
||||||
|
if (!shared_session_registered_) {
|
||||||
|
LOG_WARN("Failed to publish shared Wayland portal session");
|
||||||
|
}
|
||||||
|
|
||||||
|
LOG_INFO("Wayland screencast ready, node_id={}", pipewire_node_id_);
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool ScreenCapturerWayland::OpenPipeWireRemote() {
|
||||||
|
if (!dbus_connection_ || session_handle_.empty()) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
DBusMessage* message = dbus_message_new_method_call(
|
||||||
|
kPortalBusName, kPortalObjectPath, kPortalScreenCastInterface,
|
||||||
|
"OpenPipeWireRemote");
|
||||||
|
if (!message) {
|
||||||
|
LOG_ERROR("Failed to allocate OpenPipeWireRemote message");
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
DBusMessageIter iter;
|
||||||
|
DBusMessageIter options;
|
||||||
|
const char* session_handle = session_handle_.c_str();
|
||||||
|
dbus_message_iter_init_append(message, &iter);
|
||||||
|
dbus_message_iter_append_basic(&iter, DBUS_TYPE_OBJECT_PATH,
|
||||||
|
&session_handle);
|
||||||
|
dbus_message_iter_open_container(&iter, DBUS_TYPE_ARRAY, "{sv}", &options);
|
||||||
|
dbus_message_iter_close_container(&iter, &options);
|
||||||
|
|
||||||
|
DBusError error;
|
||||||
|
dbus_error_init(&error);
|
||||||
|
DBusMessage* reply =
|
||||||
|
dbus_connection_send_with_reply_and_block(dbus_connection_, message, -1,
|
||||||
|
&error);
|
||||||
|
dbus_message_unref(message);
|
||||||
|
if (!reply) {
|
||||||
|
LogDbusError("OpenPipeWireRemote", &error);
|
||||||
|
dbus_error_free(&error);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
DBusMessageIter reply_iter;
|
||||||
|
if (!dbus_message_iter_init(reply, &reply_iter) ||
|
||||||
|
dbus_message_iter_get_arg_type(&reply_iter) != DBUS_TYPE_UNIX_FD) {
|
||||||
|
LOG_ERROR("OpenPipeWireRemote returned an unexpected payload");
|
||||||
|
dbus_message_unref(reply);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
int received_fd = -1;
|
||||||
|
dbus_message_iter_get_basic(&reply_iter, &received_fd);
|
||||||
|
dbus_message_unref(reply);
|
||||||
|
|
||||||
|
if (received_fd < 0) {
|
||||||
|
LOG_ERROR("OpenPipeWireRemote returned an invalid fd");
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
pipewire_fd_ = dup(received_fd);
|
||||||
|
if (pipewire_fd_ < 0) {
|
||||||
|
LOG_ERROR("Failed to duplicate PipeWire remote fd");
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
void ScreenCapturerWayland::CleanupDbus() {
|
||||||
|
if (!dbus_connection_) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (shared_session_registered_) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
dbus_connection_close(dbus_connection_);
|
||||||
|
dbus_connection_unref(dbus_connection_);
|
||||||
|
dbus_connection_ = nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
void ScreenCapturerWayland::ClosePortalSession() {
|
||||||
|
if (shared_session_registered_) {
|
||||||
|
DBusConnection* close_connection = nullptr;
|
||||||
|
std::string close_session_handle;
|
||||||
|
ReleaseSharedWaylandPortalSession(&close_connection, &close_session_handle);
|
||||||
|
shared_session_registered_ = false;
|
||||||
|
if (close_connection) {
|
||||||
|
CloseWaylandPortalSessionAndConnection(close_connection,
|
||||||
|
close_session_handle,
|
||||||
|
"Session.Close");
|
||||||
|
}
|
||||||
|
dbus_connection_ = nullptr;
|
||||||
|
} else if (dbus_connection_ && !session_handle_.empty()) {
|
||||||
|
CloseWaylandPortalSessionAndConnection(dbus_connection_, session_handle_,
|
||||||
|
"Session.Close");
|
||||||
|
dbus_connection_ = nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
session_handle_.clear();
|
||||||
|
pipewire_node_id_ = 0;
|
||||||
|
UpdateDisplayGeometry(logical_width_ > 0 ? logical_width_ : kFallbackWidth,
|
||||||
|
logical_height_ > 0 ? logical_height_
|
||||||
|
: kFallbackHeight);
|
||||||
|
pointer_granted_ = false;
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace crossdesk
|
||||||
|
|
||||||
|
#endif
|
||||||
@@ -5,7 +5,9 @@
|
|||||||
#include <X11/extensions/Xfixes.h>
|
#include <X11/extensions/Xfixes.h>
|
||||||
#include <X11/extensions/Xrandr.h>
|
#include <X11/extensions/Xrandr.h>
|
||||||
|
|
||||||
|
#include <algorithm>
|
||||||
#include <chrono>
|
#include <chrono>
|
||||||
|
#include <mutex>
|
||||||
#include <thread>
|
#include <thread>
|
||||||
|
|
||||||
#include "libyuv.h"
|
#include "libyuv.h"
|
||||||
@@ -13,11 +15,58 @@
|
|||||||
|
|
||||||
namespace crossdesk {
|
namespace crossdesk {
|
||||||
|
|
||||||
|
namespace {
|
||||||
|
|
||||||
|
std::atomic<int> g_x11_last_error_code{0};
|
||||||
|
std::mutex g_x11_error_handler_mutex;
|
||||||
|
|
||||||
|
int CaptureX11ErrorHandler([[maybe_unused]] Display* display,
|
||||||
|
XErrorEvent* error_event) {
|
||||||
|
if (error_event) {
|
||||||
|
g_x11_last_error_code.store(error_event->error_code);
|
||||||
|
} else {
|
||||||
|
g_x11_last_error_code.store(-1);
|
||||||
|
}
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
class ScopedX11ErrorTrap {
|
||||||
|
public:
|
||||||
|
explicit ScopedX11ErrorTrap(Display* display)
|
||||||
|
: display_(display), lock_(g_x11_error_handler_mutex) {
|
||||||
|
g_x11_last_error_code.store(0);
|
||||||
|
previous_handler_ = XSetErrorHandler(CaptureX11ErrorHandler);
|
||||||
|
}
|
||||||
|
|
||||||
|
~ScopedX11ErrorTrap() {
|
||||||
|
if (display_) {
|
||||||
|
XSync(display_, False);
|
||||||
|
}
|
||||||
|
XSetErrorHandler(previous_handler_);
|
||||||
|
}
|
||||||
|
|
||||||
|
int SyncAndGetError() const {
|
||||||
|
if (display_) {
|
||||||
|
XSync(display_, False);
|
||||||
|
}
|
||||||
|
return g_x11_last_error_code.load();
|
||||||
|
}
|
||||||
|
|
||||||
|
private:
|
||||||
|
Display* display_ = nullptr;
|
||||||
|
int (*previous_handler_)(Display*, XErrorEvent*) = nullptr;
|
||||||
|
std::unique_lock<std::mutex> lock_;
|
||||||
|
};
|
||||||
|
|
||||||
|
} // namespace
|
||||||
|
|
||||||
ScreenCapturerX11::ScreenCapturerX11() {}
|
ScreenCapturerX11::ScreenCapturerX11() {}
|
||||||
|
|
||||||
ScreenCapturerX11::~ScreenCapturerX11() { Destroy(); }
|
ScreenCapturerX11::~ScreenCapturerX11() { Destroy(); }
|
||||||
|
|
||||||
int ScreenCapturerX11::Init(const int fps, cb_desktop_data cb) {
|
int ScreenCapturerX11::Init(const int fps, cb_desktop_data cb) {
|
||||||
|
Destroy();
|
||||||
|
|
||||||
display_ = XOpenDisplay(nullptr);
|
display_ = XOpenDisplay(nullptr);
|
||||||
if (!display_) {
|
if (!display_) {
|
||||||
LOG_ERROR("Cannot connect to X server");
|
LOG_ERROR("Cannot connect to X server");
|
||||||
@@ -29,6 +78,7 @@ int ScreenCapturerX11::Init(const int fps, cb_desktop_data cb) {
|
|||||||
if (!screen_res_) {
|
if (!screen_res_) {
|
||||||
LOG_ERROR("Failed to get screen resources");
|
LOG_ERROR("Failed to get screen resources");
|
||||||
XCloseDisplay(display_);
|
XCloseDisplay(display_);
|
||||||
|
display_ = nullptr;
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -71,8 +121,16 @@ int ScreenCapturerX11::Init(const int fps, cb_desktop_data cb) {
|
|||||||
width_ = attr.width;
|
width_ = attr.width;
|
||||||
height_ = attr.height;
|
height_ = attr.height;
|
||||||
|
|
||||||
if (width_ % 2 != 0 || height_ % 2 != 0) {
|
if ((width_ & 1) != 0 || (height_ & 1) != 0) {
|
||||||
LOG_ERROR("Width and height must be even numbers");
|
LOG_WARN(
|
||||||
|
"X11 root size {}x{} is not even, aligning down to {}x{} for NV12",
|
||||||
|
width_, height_, width_ & ~1, height_ & ~1);
|
||||||
|
width_ &= ~1;
|
||||||
|
height_ &= ~1;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (width_ <= 1 || height_ <= 1) {
|
||||||
|
LOG_ERROR("Invalid capture size after alignment: {}x{}", width_, height_);
|
||||||
return -2;
|
return -2;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -82,6 +140,11 @@ int ScreenCapturerX11::Init(const int fps, cb_desktop_data cb) {
|
|||||||
y_plane_.resize(width_ * height_);
|
y_plane_.resize(width_ * height_);
|
||||||
uv_plane_.resize((width_ / 2) * (height_ / 2) * 2);
|
uv_plane_.resize((width_ / 2) * (height_ / 2) * 2);
|
||||||
|
|
||||||
|
if (!ProbeCapture()) {
|
||||||
|
LOG_ERROR("X11 backend probe failed, XGetImage is not usable");
|
||||||
|
return -3;
|
||||||
|
}
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -108,9 +171,23 @@ int ScreenCapturerX11::Start(bool show_cursor) {
|
|||||||
show_cursor_ = show_cursor;
|
show_cursor_ = show_cursor;
|
||||||
running_ = true;
|
running_ = true;
|
||||||
paused_ = false;
|
paused_ = false;
|
||||||
|
capture_error_count_ = 0;
|
||||||
thread_ = std::thread([this]() {
|
thread_ = std::thread([this]() {
|
||||||
|
using clock = std::chrono::steady_clock;
|
||||||
|
const auto frame_interval =
|
||||||
|
std::chrono::milliseconds(std::max(1, 1000 / std::max(1, fps_)));
|
||||||
|
|
||||||
while (running_) {
|
while (running_) {
|
||||||
if (!paused_) OnFrame();
|
const auto frame_start = clock::now();
|
||||||
|
if (!paused_) {
|
||||||
|
OnFrame();
|
||||||
|
}
|
||||||
|
|
||||||
|
const auto elapsed = std::chrono::duration_cast<std::chrono::milliseconds>(
|
||||||
|
clock::now() - frame_start);
|
||||||
|
if (elapsed < frame_interval) {
|
||||||
|
std::this_thread::sleep_for(frame_interval - elapsed);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
return 0;
|
return 0;
|
||||||
@@ -152,19 +229,44 @@ void ScreenCapturerX11::OnFrame() {
|
|||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (monitor_index_ < 0 || monitor_index_ >= display_info_list_.size()) {
|
const int monitor_index = monitor_index_.load();
|
||||||
LOG_ERROR("Invalid monitor index: {}", monitor_index_.load());
|
if (monitor_index < 0 ||
|
||||||
|
monitor_index >= static_cast<int>(display_info_list_.size())) {
|
||||||
|
LOG_ERROR("Invalid monitor index: {}", monitor_index);
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
left_ = display_info_list_[monitor_index_].left;
|
left_ = display_info_list_[monitor_index].left;
|
||||||
top_ = display_info_list_[monitor_index_].top;
|
top_ = display_info_list_[monitor_index].top;
|
||||||
width_ = display_info_list_[monitor_index_].width;
|
width_ = display_info_list_[monitor_index].width & ~1;
|
||||||
height_ = display_info_list_[monitor_index_].height;
|
height_ = display_info_list_[monitor_index].height & ~1;
|
||||||
|
|
||||||
XImage* image = XGetImage(display_, root_, left_, top_, width_, height_,
|
if (width_ <= 1 || height_ <= 1) {
|
||||||
AllPlanes, ZPixmap);
|
LOG_ERROR("Invalid capture size: {}x{}", width_, height_);
|
||||||
if (!image) return;
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
XImage* image = nullptr;
|
||||||
|
int x11_error = 0;
|
||||||
|
{
|
||||||
|
ScopedX11ErrorTrap trap(display_);
|
||||||
|
image = XGetImage(display_, root_, left_, top_, width_, height_, AllPlanes,
|
||||||
|
ZPixmap);
|
||||||
|
x11_error = trap.SyncAndGetError();
|
||||||
|
}
|
||||||
|
|
||||||
|
if (x11_error != 0 || !image) {
|
||||||
|
if (image) {
|
||||||
|
XDestroyImage(image);
|
||||||
|
}
|
||||||
|
++capture_error_count_;
|
||||||
|
if (capture_error_count_ == 1 || capture_error_count_ % 120 == 0) {
|
||||||
|
LOG_WARN("X11 capture failed: x11_error={}, image={}, consecutive={}",
|
||||||
|
x11_error, image ? "valid" : "null", capture_error_count_);
|
||||||
|
}
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
capture_error_count_ = 0;
|
||||||
|
|
||||||
// if enable show cursor, draw cursor
|
// if enable show cursor, draw cursor
|
||||||
if (show_cursor_) {
|
if (show_cursor_) {
|
||||||
@@ -195,6 +297,16 @@ void ScreenCapturerX11::OnFrame() {
|
|||||||
src_argb = reinterpret_cast<uint8_t*>(image->data);
|
src_argb = reinterpret_cast<uint8_t*>(image->data);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const size_t y_size =
|
||||||
|
static_cast<size_t>(width_) * static_cast<size_t>(height_);
|
||||||
|
const size_t uv_size = y_size / 2;
|
||||||
|
if (y_plane_.size() != y_size) {
|
||||||
|
y_plane_.resize(y_size);
|
||||||
|
}
|
||||||
|
if (uv_plane_.size() != uv_size) {
|
||||||
|
uv_plane_.resize(uv_size);
|
||||||
|
}
|
||||||
|
|
||||||
libyuv::ARGBToNV12(src_argb, width_ * 4, y_plane_.data(), width_,
|
libyuv::ARGBToNV12(src_argb, width_ * 4, y_plane_.data(), width_,
|
||||||
uv_plane_.data(), width_, width_, height_);
|
uv_plane_.data(), width_, width_, height_);
|
||||||
|
|
||||||
@@ -205,7 +317,7 @@ void ScreenCapturerX11::OnFrame() {
|
|||||||
|
|
||||||
if (callback_) {
|
if (callback_) {
|
||||||
callback_(nv12.data(), width_ * height_ * 3 / 2, width_, height_,
|
callback_(nv12.data(), width_ * height_ * 3 / 2, width_, height_,
|
||||||
display_info_list_[monitor_index_].name.c_str());
|
display_info_list_[monitor_index].name.c_str());
|
||||||
}
|
}
|
||||||
|
|
||||||
XDestroyImage(image);
|
XDestroyImage(image);
|
||||||
@@ -288,4 +400,32 @@ void ScreenCapturerX11::DrawCursor(XImage* image, int x, int y) {
|
|||||||
|
|
||||||
XFree(cursor_image);
|
XFree(cursor_image);
|
||||||
}
|
}
|
||||||
} // namespace crossdesk
|
|
||||||
|
bool ScreenCapturerX11::ProbeCapture() {
|
||||||
|
if (!display_ || display_info_list_.empty()) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
const auto& first_display = display_info_list_[0];
|
||||||
|
XImage* probe_image = nullptr;
|
||||||
|
int x11_error = 0;
|
||||||
|
{
|
||||||
|
ScopedX11ErrorTrap trap(display_);
|
||||||
|
probe_image = XGetImage(display_, root_, first_display.left,
|
||||||
|
first_display.top, 1, 1, AllPlanes, ZPixmap);
|
||||||
|
x11_error = trap.SyncAndGetError();
|
||||||
|
}
|
||||||
|
|
||||||
|
if (probe_image) {
|
||||||
|
XDestroyImage(probe_image);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (x11_error != 0 || !probe_image) {
|
||||||
|
LOG_WARN("X11 probe XGetImage failed: x11_error={}, image={}", x11_error,
|
||||||
|
probe_image ? "valid" : "null");
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
} // namespace crossdesk
|
||||||
|
|||||||
@@ -17,6 +17,7 @@ struct _XImage;
|
|||||||
typedef struct _XImage XImage;
|
typedef struct _XImage XImage;
|
||||||
|
|
||||||
#include <atomic>
|
#include <atomic>
|
||||||
|
#include <cctype>
|
||||||
#include <cstring>
|
#include <cstring>
|
||||||
#include <functional>
|
#include <functional>
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
@@ -50,6 +51,7 @@ class ScreenCapturerX11 : public ScreenCapturer {
|
|||||||
|
|
||||||
private:
|
private:
|
||||||
void DrawCursor(XImage* image, int x, int y);
|
void DrawCursor(XImage* image, int x, int y);
|
||||||
|
bool ProbeCapture();
|
||||||
|
|
||||||
private:
|
private:
|
||||||
Display* display_ = nullptr;
|
Display* display_ = nullptr;
|
||||||
@@ -68,9 +70,10 @@ class ScreenCapturerX11 : public ScreenCapturer {
|
|||||||
int fps_ = 60;
|
int fps_ = 60;
|
||||||
cb_desktop_data callback_;
|
cb_desktop_data callback_;
|
||||||
std::vector<DisplayInfo> display_info_list_;
|
std::vector<DisplayInfo> display_info_list_;
|
||||||
|
int capture_error_count_ = 0;
|
||||||
|
|
||||||
std::vector<uint8_t> y_plane_;
|
std::vector<uint8_t> y_plane_;
|
||||||
std::vector<uint8_t> uv_plane_;
|
std::vector<uint8_t> uv_plane_;
|
||||||
};
|
};
|
||||||
} // namespace crossdesk
|
} // namespace crossdesk
|
||||||
#endif
|
#endif
|
||||||
|
|||||||
@@ -10,7 +10,7 @@
|
|||||||
#ifdef _WIN32
|
#ifdef _WIN32
|
||||||
#include "screen_capturer_win.h"
|
#include "screen_capturer_win.h"
|
||||||
#elif __linux__
|
#elif __linux__
|
||||||
#include "screen_capturer_x11.h"
|
#include "screen_capturer_linux.h"
|
||||||
#elif __APPLE__
|
#elif __APPLE__
|
||||||
// #include "screen_capturer_avf.h"
|
// #include "screen_capturer_avf.h"
|
||||||
#include "screen_capturer_sck.h"
|
#include "screen_capturer_sck.h"
|
||||||
@@ -27,7 +27,7 @@ class ScreenCapturerFactory {
|
|||||||
#ifdef _WIN32
|
#ifdef _WIN32
|
||||||
return new ScreenCapturerWin();
|
return new ScreenCapturerWin();
|
||||||
#elif __linux__
|
#elif __linux__
|
||||||
return new ScreenCapturerX11();
|
return new ScreenCapturerLinux();
|
||||||
#elif __APPLE__
|
#elif __APPLE__
|
||||||
// return new ScreenCapturerAvf();
|
// return new ScreenCapturerAvf();
|
||||||
return new ScreenCapturerSck();
|
return new ScreenCapturerSck();
|
||||||
|
|||||||
229
xmake.lua
229
xmake.lua
@@ -1,227 +1,10 @@
|
|||||||
set_project("crossdesk")
|
set_project("crossdesk")
|
||||||
set_license("LGPL-3.0")
|
set_license("LGPL-3.0")
|
||||||
|
|
||||||
option("CROSSDESK_VERSION")
|
includes("xmake/options.lua")
|
||||||
set_default("0.0.0")
|
includes("xmake/platform.lua")
|
||||||
set_showmenu(true)
|
includes("xmake/targets.lua")
|
||||||
set_description("Set CROSSDESK_VERSION for build")
|
|
||||||
option_end()
|
|
||||||
|
|
||||||
option("USE_CUDA")
|
setup_options_and_dependencies()
|
||||||
set_default(false)
|
setup_platform_settings()
|
||||||
set_showmenu(true)
|
setup_targets()
|
||||||
set_description("Use CUDA for hardware codec acceleration")
|
|
||||||
option_end()
|
|
||||||
|
|
||||||
add_rules("mode.release", "mode.debug")
|
|
||||||
set_languages("c++17")
|
|
||||||
set_encodings("utf-8")
|
|
||||||
|
|
||||||
-- set_policy("build.warning", true)
|
|
||||||
-- set_warnings("all", "extra")
|
|
||||||
-- add_cxxflags("/W4", "/WX")
|
|
||||||
|
|
||||||
add_defines("UNICODE")
|
|
||||||
add_defines("USE_CUDA=" .. (is_config("USE_CUDA", true) and "1" or "0"))
|
|
||||||
|
|
||||||
if is_mode("debug") then
|
|
||||||
add_defines("CROSSDESK_DEBUG")
|
|
||||||
end
|
|
||||||
|
|
||||||
add_requireconfs("*.python", {version = "3.12", override = true, configs = {pgo = false}})
|
|
||||||
add_requires("spdlog 1.14.1", {system = false})
|
|
||||||
add_requires("imgui v1.92.1-docking", {configs = {sdl3 = true, sdl3_renderer = true}})
|
|
||||||
add_requires("openssl3 3.3.2", {system = false})
|
|
||||||
add_requires("nlohmann_json 3.11.3")
|
|
||||||
add_requires("cpp-httplib v0.26.0", {configs = {ssl = true}})
|
|
||||||
add_requires("tinyfiledialogs 3.15.1")
|
|
||||||
|
|
||||||
if is_os("windows") then
|
|
||||||
add_requires("libyuv", "miniaudio 0.11.21")
|
|
||||||
add_links("Shell32", "dwmapi", "User32", "kernel32",
|
|
||||||
"SDL3-static", "gdi32", "winmm", "setupapi", "version",
|
|
||||||
"Imm32", "iphlpapi", "d3d11", "dxgi")
|
|
||||||
add_cxflags("/WX")
|
|
||||||
set_runtimes("MT")
|
|
||||||
elseif is_os("linux") then
|
|
||||||
add_links("pulse-simple", "pulse")
|
|
||||||
add_requires("libyuv")
|
|
||||||
add_syslinks("pthread", "dl")
|
|
||||||
add_links("SDL3", "asound", "X11", "Xtst", "Xrandr", "Xfixes")
|
|
||||||
add_cxflags("-Wno-unused-variable")
|
|
||||||
elseif is_os("macosx") then
|
|
||||||
add_links("SDL3")
|
|
||||||
add_ldflags("-Wl,-ld_classic")
|
|
||||||
add_cxflags("-Wno-unused-variable")
|
|
||||||
add_frameworks("OpenGL", "IOSurface", "ScreenCaptureKit", "AVFoundation",
|
|
||||||
"CoreMedia", "CoreVideo", "CoreAudio", "AudioToolbox")
|
|
||||||
end
|
|
||||||
|
|
||||||
add_packages("spdlog", "imgui", "nlohmann_json")
|
|
||||||
|
|
||||||
includes("submodules", "thirdparty")
|
|
||||||
|
|
||||||
target("rd_log")
|
|
||||||
set_kind("object")
|
|
||||||
add_packages("spdlog")
|
|
||||||
add_files("src/log/rd_log.cpp")
|
|
||||||
add_includedirs("src/log", {public = true})
|
|
||||||
|
|
||||||
target("common")
|
|
||||||
set_kind("object")
|
|
||||||
add_deps("rd_log")
|
|
||||||
add_files("src/common/*.cpp")
|
|
||||||
if is_os("macosx") then
|
|
||||||
add_files("src/common/*.mm")
|
|
||||||
end
|
|
||||||
add_includedirs("src/common", {public = true})
|
|
||||||
|
|
||||||
target("path_manager")
|
|
||||||
set_kind("object")
|
|
||||||
add_deps("rd_log")
|
|
||||||
add_includedirs("src/path_manager", {public = true})
|
|
||||||
add_files("src/path_manager/*.cpp")
|
|
||||||
add_includedirs("src/path_manager", {public = true})
|
|
||||||
|
|
||||||
target("screen_capturer")
|
|
||||||
set_kind("object")
|
|
||||||
add_deps("rd_log", "common")
|
|
||||||
add_includedirs("src/screen_capturer", {public = true})
|
|
||||||
if is_os("windows") then
|
|
||||||
add_packages("libyuv")
|
|
||||||
add_files("src/screen_capturer/windows/screen_capturer_dxgi.cpp",
|
|
||||||
"src/screen_capturer/windows/screen_capturer_gdi.cpp",
|
|
||||||
"src/screen_capturer/windows/screen_capturer_win.cpp")
|
|
||||||
add_includedirs("src/screen_capturer/windows", {public = true})
|
|
||||||
elseif is_os("macosx") then
|
|
||||||
add_files("src/screen_capturer/macosx/*.cpp",
|
|
||||||
"src/screen_capturer/macosx/*.mm")
|
|
||||||
add_includedirs("src/screen_capturer/macosx", {public = true})
|
|
||||||
elseif is_os("linux") then
|
|
||||||
add_packages("libyuv")
|
|
||||||
add_files("src/screen_capturer/linux/*.cpp")
|
|
||||||
add_includedirs("src/screen_capturer/linux", {public = true})
|
|
||||||
end
|
|
||||||
|
|
||||||
target("speaker_capturer")
|
|
||||||
set_kind("object")
|
|
||||||
add_deps("rd_log")
|
|
||||||
add_includedirs("src/speaker_capturer", {public = true})
|
|
||||||
if is_os("windows") then
|
|
||||||
add_packages("miniaudio")
|
|
||||||
add_files("src/speaker_capturer/windows/*.cpp")
|
|
||||||
add_includedirs("src/speaker_capturer/windows", {public = true})
|
|
||||||
elseif is_os("macosx") then
|
|
||||||
add_files("src/speaker_capturer/macosx/*.cpp",
|
|
||||||
"src/speaker_capturer/macosx/*.mm")
|
|
||||||
add_includedirs("src/speaker_capturer/macosx", {public = true})
|
|
||||||
elseif is_os("linux") then
|
|
||||||
add_files("src/speaker_capturer/linux/*.cpp")
|
|
||||||
add_includedirs("src/speaker_capturer/linux", {public = true})
|
|
||||||
end
|
|
||||||
|
|
||||||
target("device_controller")
|
|
||||||
set_kind("object")
|
|
||||||
add_deps("rd_log", "common")
|
|
||||||
add_includedirs("src/device_controller", {public = true})
|
|
||||||
if is_os("windows") then
|
|
||||||
add_files("src/device_controller/mouse/windows/*.cpp",
|
|
||||||
"src/device_controller/keyboard/windows/*.cpp")
|
|
||||||
add_includedirs("src/device_controller/mouse/windows",
|
|
||||||
"src/device_controller/keyboard/windows", {public = true})
|
|
||||||
elseif is_os("macosx") then
|
|
||||||
add_files("src/device_controller/mouse/mac/*.cpp",
|
|
||||||
"src/device_controller/keyboard/mac/*.cpp")
|
|
||||||
add_includedirs("src/device_controller/mouse/mac",
|
|
||||||
"src/device_controller/keyboard/mac", {public = true})
|
|
||||||
elseif is_os("linux") then
|
|
||||||
add_files("src/device_controller/mouse/linux/*.cpp",
|
|
||||||
"src/device_controller/keyboard/linux/*.cpp")
|
|
||||||
add_includedirs("src/device_controller/mouse/linux",
|
|
||||||
"src/device_controller/keyboard/linux", {public = true})
|
|
||||||
end
|
|
||||||
|
|
||||||
target("thumbnail")
|
|
||||||
set_kind("object")
|
|
||||||
add_packages("libyuv", "openssl3")
|
|
||||||
add_deps("rd_log", "common")
|
|
||||||
add_files("src/thumbnail/*.cpp")
|
|
||||||
add_includedirs("src/thumbnail", {public = true})
|
|
||||||
|
|
||||||
target("autostart")
|
|
||||||
set_kind("object")
|
|
||||||
add_deps("rd_log")
|
|
||||||
add_files("src/autostart/*.cpp")
|
|
||||||
add_includedirs("src/autostart", {public = true})
|
|
||||||
|
|
||||||
target("config_center")
|
|
||||||
set_kind("object")
|
|
||||||
add_deps("rd_log", "autostart")
|
|
||||||
add_files("src/config_center/*.cpp")
|
|
||||||
add_includedirs("src/config_center", {public = true})
|
|
||||||
|
|
||||||
target("assets")
|
|
||||||
set_kind("headeronly")
|
|
||||||
add_includedirs("src/gui/assets/localization",
|
|
||||||
"src/gui/assets/fonts",
|
|
||||||
"src/gui/assets/icons",
|
|
||||||
"src/gui/assets/layouts", {public = true})
|
|
||||||
|
|
||||||
target("version_checker")
|
|
||||||
set_kind("object")
|
|
||||||
add_packages("cpp-httplib")
|
|
||||||
add_defines("CROSSDESK_VERSION=\"" .. (get_config("CROSSDESK_VERSION") or "Unknown") .. "\"")
|
|
||||||
add_deps("rd_log")
|
|
||||||
add_files("src/version_checker/*.cpp")
|
|
||||||
add_includedirs("src/version_checker", {public = true})
|
|
||||||
|
|
||||||
target("tools")
|
|
||||||
set_kind("object")
|
|
||||||
add_deps("rd_log")
|
|
||||||
add_files("src/tools/*.cpp")
|
|
||||||
if is_os("macosx") then
|
|
||||||
add_files("src/tools/*.mm")
|
|
||||||
end
|
|
||||||
add_includedirs("src/tools", {public = true})
|
|
||||||
|
|
||||||
target("gui")
|
|
||||||
set_kind("object")
|
|
||||||
add_packages("libyuv", "tinyfiledialogs")
|
|
||||||
add_defines("CROSSDESK_VERSION=\"" .. (get_config("CROSSDESK_VERSION") or "Unknown") .. "\"")
|
|
||||||
add_deps("rd_log", "common", "assets", "config_center", "minirtc",
|
|
||||||
"path_manager", "screen_capturer", "speaker_capturer",
|
|
||||||
"device_controller", "thumbnail", "version_checker", "tools")
|
|
||||||
add_files("src/gui/*.cpp", "src/gui/panels/*.cpp", "src/gui/toolbars/*.cpp",
|
|
||||||
"src/gui/windows/*.cpp")
|
|
||||||
add_includedirs("src/gui", "src/gui/panels", "src/gui/toolbars",
|
|
||||||
"src/gui/windows", {public = true})
|
|
||||||
if is_os("windows") then
|
|
||||||
add_files("src/gui/tray/*.cpp")
|
|
||||||
add_includedirs("src/gui/tray", {public = true})
|
|
||||||
elseif is_os("macosx") then
|
|
||||||
add_files("src/gui/windows/*.mm")
|
|
||||||
end
|
|
||||||
|
|
||||||
if is_os("windows") then
|
|
||||||
target("wgc_plugin")
|
|
||||||
set_kind("shared")
|
|
||||||
add_packages("libyuv")
|
|
||||||
add_deps("rd_log")
|
|
||||||
add_defines("CROSSDESK_WGC_PLUGIN_BUILD=1")
|
|
||||||
add_links("windowsapp")
|
|
||||||
add_files("src/screen_capturer/windows/screen_capturer_wgc.cpp",
|
|
||||||
"src/screen_capturer/windows/wgc_session_impl.cpp",
|
|
||||||
"src/screen_capturer/windows/wgc_plugin_entry.cpp")
|
|
||||||
add_includedirs("src/common", "src/screen_capturer",
|
|
||||||
"src/screen_capturer/windows")
|
|
||||||
end
|
|
||||||
|
|
||||||
target("crossdesk")
|
|
||||||
set_kind("binary")
|
|
||||||
add_deps("rd_log", "common", "gui")
|
|
||||||
add_files("src/app/*.cpp")
|
|
||||||
add_includedirs("src/app", {public = true})
|
|
||||||
if is_os("windows") then
|
|
||||||
add_deps("wgc_plugin")
|
|
||||||
add_files("scripts/windows/crossdesk.rc")
|
|
||||||
end
|
|
||||||
|
|||||||
50
xmake/options.lua
Normal file
50
xmake/options.lua
Normal file
@@ -0,0 +1,50 @@
|
|||||||
|
function setup_options_and_dependencies()
|
||||||
|
option("CROSSDESK_VERSION")
|
||||||
|
set_default("0.0.0")
|
||||||
|
set_showmenu(true)
|
||||||
|
set_description("Set CROSSDESK_VERSION for build")
|
||||||
|
option_end()
|
||||||
|
|
||||||
|
option("USE_CUDA")
|
||||||
|
set_default(false)
|
||||||
|
set_showmenu(true)
|
||||||
|
set_description("Use CUDA for hardware codec acceleration")
|
||||||
|
option_end()
|
||||||
|
|
||||||
|
option("USE_WAYLAND")
|
||||||
|
set_default(false)
|
||||||
|
set_showmenu(true)
|
||||||
|
set_description("Enable Wayland capture on Linux (assumes dependencies are installed)")
|
||||||
|
option_end()
|
||||||
|
|
||||||
|
option("USE_DRM")
|
||||||
|
set_default(false)
|
||||||
|
set_showmenu(true)
|
||||||
|
set_description("Enable DRM capture on Linux (assumes dependencies are installed)")
|
||||||
|
option_end()
|
||||||
|
|
||||||
|
add_rules("mode.release", "mode.debug")
|
||||||
|
set_languages("c++17")
|
||||||
|
set_encodings("utf-8")
|
||||||
|
|
||||||
|
-- set_policy("build.warning", true)
|
||||||
|
-- set_warnings("all", "extra")
|
||||||
|
-- add_cxxflags("/W4", "/WX")
|
||||||
|
|
||||||
|
add_defines("UNICODE")
|
||||||
|
add_defines("USE_CUDA=" .. (is_config("USE_CUDA", true) and "1" or "0"))
|
||||||
|
add_defines("USE_WAYLAND=" .. (is_config("USE_WAYLAND", true) and "1" or "0"))
|
||||||
|
add_defines("USE_DRM=" .. (is_config("USE_DRM", true) and "1" or "0"))
|
||||||
|
|
||||||
|
if is_mode("debug") then
|
||||||
|
add_defines("CROSSDESK_DEBUG")
|
||||||
|
end
|
||||||
|
|
||||||
|
add_requireconfs("*.python", {version = "3.12", override = true, configs = {pgo = false}})
|
||||||
|
add_requires("spdlog 1.14.1", {system = false})
|
||||||
|
add_requires("imgui v1.92.1-docking", {configs = {sdl3 = true, sdl3_renderer = true}})
|
||||||
|
add_requires("openssl3 3.3.2", {system = false})
|
||||||
|
add_requires("nlohmann_json 3.11.3")
|
||||||
|
add_requires("cpp-httplib v0.26.0", {configs = {ssl = true}})
|
||||||
|
add_requires("tinyfiledialogs 3.15.1")
|
||||||
|
end
|
||||||
81
xmake/platform.lua
Normal file
81
xmake/platform.lua
Normal file
@@ -0,0 +1,81 @@
|
|||||||
|
local function add_existing_include_dirs(paths, opts)
|
||||||
|
for _, dir in ipairs(paths) do
|
||||||
|
if os.isdir(dir) then
|
||||||
|
add_includedirs(dir, opts)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
local function collect_dbus_arch_include_dirs()
|
||||||
|
local include_dirs = {}
|
||||||
|
for _, pattern in ipairs({
|
||||||
|
"/usr/lib/*/dbus-1.0/include",
|
||||||
|
"/usr/lib64/dbus-1.0/include",
|
||||||
|
"/usr/lib/dbus-1.0/include",
|
||||||
|
"/lib/*/dbus-1.0/include",
|
||||||
|
"/lib64/dbus-1.0/include",
|
||||||
|
"/lib/dbus-1.0/include"
|
||||||
|
}) do
|
||||||
|
for _, include_dir in ipairs(os.dirs(pattern)) do
|
||||||
|
table.insert(include_dirs, include_dir)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
return include_dirs
|
||||||
|
end
|
||||||
|
|
||||||
|
function setup_platform_settings()
|
||||||
|
if is_os("windows") then
|
||||||
|
add_requires("libyuv", "miniaudio 0.11.21")
|
||||||
|
add_links("Shell32", "dwmapi", "User32", "kernel32",
|
||||||
|
"SDL3-static", "gdi32", "winmm", "setupapi", "version",
|
||||||
|
"Imm32", "iphlpapi", "d3d11", "dxgi")
|
||||||
|
add_cxflags("/WX")
|
||||||
|
set_runtimes("MT")
|
||||||
|
elseif is_os("linux") then
|
||||||
|
add_links("pulse-simple", "pulse")
|
||||||
|
add_requires("libyuv")
|
||||||
|
add_syslinks("pthread", "dl")
|
||||||
|
add_links("SDL3", "asound", "X11", "Xtst", "Xrandr", "Xfixes")
|
||||||
|
|
||||||
|
if is_config("USE_DRM", true) then
|
||||||
|
add_links("drm")
|
||||||
|
add_defines("CROSSDESK_HAS_DRM=1")
|
||||||
|
add_existing_include_dirs({
|
||||||
|
"/usr/include/libdrm",
|
||||||
|
"/usr/local/include/libdrm"
|
||||||
|
}, {system = true})
|
||||||
|
else
|
||||||
|
add_defines("CROSSDESK_HAS_DRM=0")
|
||||||
|
end
|
||||||
|
|
||||||
|
if is_config("USE_WAYLAND", true) then
|
||||||
|
add_links("dbus-1", "pipewire-0.3")
|
||||||
|
add_defines("CROSSDESK_HAS_WAYLAND_CAPTURER=1")
|
||||||
|
add_existing_include_dirs({
|
||||||
|
"/usr/include/dbus-1.0",
|
||||||
|
"/usr/local/include/dbus-1.0",
|
||||||
|
"/usr/include/pipewire-0.3",
|
||||||
|
"/usr/local/include/pipewire-0.3",
|
||||||
|
"/usr/include/pipewire",
|
||||||
|
"/usr/local/include/pipewire",
|
||||||
|
"/usr/include/spa-0.2",
|
||||||
|
"/usr/local/include/spa-0.2",
|
||||||
|
"/usr/include/spa",
|
||||||
|
"/usr/local/include/spa"
|
||||||
|
}, {system = true})
|
||||||
|
for _, include_dir in ipairs(collect_dbus_arch_include_dirs()) do
|
||||||
|
add_includedirs(include_dir, {system = true})
|
||||||
|
end
|
||||||
|
else
|
||||||
|
add_defines("CROSSDESK_HAS_WAYLAND_CAPTURER=0")
|
||||||
|
end
|
||||||
|
|
||||||
|
add_cxflags("-Wno-unused-variable")
|
||||||
|
elseif is_os("macosx") then
|
||||||
|
add_links("SDL3")
|
||||||
|
add_ldflags("-Wl,-ld_classic")
|
||||||
|
add_cxflags("-Wno-unused-variable")
|
||||||
|
add_frameworks("OpenGL", "IOSurface", "ScreenCaptureKit", "AVFoundation",
|
||||||
|
"CoreMedia", "CoreVideo", "CoreAudio", "AudioToolbox")
|
||||||
|
end
|
||||||
|
end
|
||||||
177
xmake/targets.lua
Normal file
177
xmake/targets.lua
Normal file
@@ -0,0 +1,177 @@
|
|||||||
|
function setup_targets()
|
||||||
|
add_packages("spdlog", "imgui", "nlohmann_json")
|
||||||
|
|
||||||
|
includes("submodules", "thirdparty")
|
||||||
|
|
||||||
|
target("rd_log")
|
||||||
|
set_kind("object")
|
||||||
|
add_packages("spdlog")
|
||||||
|
add_files("src/log/rd_log.cpp")
|
||||||
|
add_includedirs("src/log", {public = true})
|
||||||
|
|
||||||
|
target("common")
|
||||||
|
set_kind("object")
|
||||||
|
add_deps("rd_log")
|
||||||
|
add_files("src/common/*.cpp")
|
||||||
|
if is_os("macosx") then
|
||||||
|
add_files("src/common/*.mm")
|
||||||
|
end
|
||||||
|
add_includedirs("src/common", {public = true})
|
||||||
|
|
||||||
|
target("path_manager")
|
||||||
|
set_kind("object")
|
||||||
|
add_deps("rd_log")
|
||||||
|
add_includedirs("src/path_manager", {public = true})
|
||||||
|
add_files("src/path_manager/*.cpp")
|
||||||
|
add_includedirs("src/path_manager", {public = true})
|
||||||
|
|
||||||
|
target("screen_capturer")
|
||||||
|
set_kind("object")
|
||||||
|
add_deps("rd_log", "common")
|
||||||
|
add_includedirs("src/screen_capturer", {public = true})
|
||||||
|
if is_os("windows") then
|
||||||
|
add_packages("libyuv")
|
||||||
|
add_files("src/screen_capturer/windows/screen_capturer_dxgi.cpp",
|
||||||
|
"src/screen_capturer/windows/screen_capturer_gdi.cpp",
|
||||||
|
"src/screen_capturer/windows/screen_capturer_win.cpp")
|
||||||
|
add_includedirs("src/screen_capturer/windows", {public = true})
|
||||||
|
elseif is_os("macosx") then
|
||||||
|
add_files("src/screen_capturer/macosx/*.cpp",
|
||||||
|
"src/screen_capturer/macosx/*.mm")
|
||||||
|
add_includedirs("src/screen_capturer/macosx", {public = true})
|
||||||
|
elseif is_os("linux") then
|
||||||
|
add_packages("libyuv")
|
||||||
|
add_files("src/screen_capturer/linux/screen_capturer_linux.cpp")
|
||||||
|
add_files("src/screen_capturer/linux/screen_capturer_x11.cpp")
|
||||||
|
add_files("src/screen_capturer/linux/screen_capturer_drm.cpp")
|
||||||
|
if is_config("USE_WAYLAND", true) then
|
||||||
|
add_files("src/screen_capturer/linux/screen_capturer_wayland.cpp")
|
||||||
|
add_files("src/screen_capturer/linux/screen_capturer_wayland_portal.cpp")
|
||||||
|
add_files("src/screen_capturer/linux/screen_capturer_wayland_pipewire.cpp")
|
||||||
|
end
|
||||||
|
add_includedirs("src/screen_capturer/linux", {public = true})
|
||||||
|
end
|
||||||
|
|
||||||
|
target("speaker_capturer")
|
||||||
|
set_kind("object")
|
||||||
|
add_deps("rd_log")
|
||||||
|
add_includedirs("src/speaker_capturer", {public = true})
|
||||||
|
if is_os("windows") then
|
||||||
|
add_packages("miniaudio")
|
||||||
|
add_files("src/speaker_capturer/windows/*.cpp")
|
||||||
|
add_includedirs("src/speaker_capturer/windows", {public = true})
|
||||||
|
elseif is_os("macosx") then
|
||||||
|
add_files("src/speaker_capturer/macosx/*.cpp",
|
||||||
|
"src/speaker_capturer/macosx/*.mm")
|
||||||
|
add_includedirs("src/speaker_capturer/macosx", {public = true})
|
||||||
|
elseif is_os("linux") then
|
||||||
|
add_files("src/speaker_capturer/linux/*.cpp")
|
||||||
|
add_includedirs("src/speaker_capturer/linux", {public = true})
|
||||||
|
end
|
||||||
|
|
||||||
|
target("device_controller")
|
||||||
|
set_kind("object")
|
||||||
|
add_deps("rd_log", "common")
|
||||||
|
add_includedirs("src/device_controller", {public = true})
|
||||||
|
if is_os("windows") then
|
||||||
|
add_files("src/device_controller/mouse/windows/*.cpp",
|
||||||
|
"src/device_controller/keyboard/windows/*.cpp")
|
||||||
|
add_includedirs("src/device_controller/mouse/windows",
|
||||||
|
"src/device_controller/keyboard/windows", {public = true})
|
||||||
|
elseif is_os("macosx") then
|
||||||
|
add_files("src/device_controller/mouse/mac/*.cpp",
|
||||||
|
"src/device_controller/keyboard/mac/*.cpp")
|
||||||
|
add_includedirs("src/device_controller/mouse/mac",
|
||||||
|
"src/device_controller/keyboard/mac", {public = true})
|
||||||
|
elseif is_os("linux") then
|
||||||
|
add_files("src/device_controller/mouse/linux/*.cpp",
|
||||||
|
"src/device_controller/keyboard/linux/*.cpp")
|
||||||
|
add_includedirs("src/device_controller/mouse/linux",
|
||||||
|
"src/device_controller/keyboard/linux", {public = true})
|
||||||
|
end
|
||||||
|
|
||||||
|
target("thumbnail")
|
||||||
|
set_kind("object")
|
||||||
|
add_packages("libyuv", "openssl3")
|
||||||
|
add_deps("rd_log", "common")
|
||||||
|
add_files("src/thumbnail/*.cpp")
|
||||||
|
add_includedirs("src/thumbnail", {public = true})
|
||||||
|
|
||||||
|
target("autostart")
|
||||||
|
set_kind("object")
|
||||||
|
add_deps("rd_log")
|
||||||
|
add_files("src/autostart/*.cpp")
|
||||||
|
add_includedirs("src/autostart", {public = true})
|
||||||
|
|
||||||
|
target("config_center")
|
||||||
|
set_kind("object")
|
||||||
|
add_deps("rd_log", "autostart")
|
||||||
|
add_files("src/config_center/*.cpp")
|
||||||
|
add_includedirs("src/config_center", {public = true})
|
||||||
|
|
||||||
|
target("assets")
|
||||||
|
set_kind("headeronly")
|
||||||
|
add_includedirs("src/gui/assets/localization",
|
||||||
|
"src/gui/assets/fonts",
|
||||||
|
"src/gui/assets/icons",
|
||||||
|
"src/gui/assets/layouts", {public = true})
|
||||||
|
|
||||||
|
target("version_checker")
|
||||||
|
set_kind("object")
|
||||||
|
add_packages("cpp-httplib")
|
||||||
|
add_defines("CROSSDESK_VERSION=\"" .. (get_config("CROSSDESK_VERSION") or "Unknown") .. "\"")
|
||||||
|
add_deps("rd_log")
|
||||||
|
add_files("src/version_checker/*.cpp")
|
||||||
|
add_includedirs("src/version_checker", {public = true})
|
||||||
|
|
||||||
|
target("tools")
|
||||||
|
set_kind("object")
|
||||||
|
add_deps("rd_log")
|
||||||
|
add_files("src/tools/*.cpp")
|
||||||
|
if is_os("macosx") then
|
||||||
|
add_files("src/tools/*.mm")
|
||||||
|
end
|
||||||
|
add_includedirs("src/tools", {public = true})
|
||||||
|
|
||||||
|
target("gui")
|
||||||
|
set_kind("object")
|
||||||
|
add_packages("libyuv", "tinyfiledialogs")
|
||||||
|
add_defines("CROSSDESK_VERSION=\"" .. (get_config("CROSSDESK_VERSION") or "Unknown") .. "\"")
|
||||||
|
add_deps("rd_log", "common", "assets", "config_center", "minirtc",
|
||||||
|
"path_manager", "screen_capturer", "speaker_capturer",
|
||||||
|
"device_controller", "thumbnail", "version_checker", "tools")
|
||||||
|
add_files("src/gui/*.cpp", "src/gui/panels/*.cpp", "src/gui/toolbars/*.cpp",
|
||||||
|
"src/gui/windows/*.cpp")
|
||||||
|
add_includedirs("src/gui", "src/gui/panels", "src/gui/toolbars",
|
||||||
|
"src/gui/windows", {public = true})
|
||||||
|
if is_os("windows") then
|
||||||
|
add_files("src/gui/tray/*.cpp")
|
||||||
|
add_includedirs("src/gui/tray", {public = true})
|
||||||
|
elseif is_os("macosx") then
|
||||||
|
add_files("src/gui/windows/*.mm")
|
||||||
|
end
|
||||||
|
|
||||||
|
if is_os("windows") then
|
||||||
|
target("wgc_plugin")
|
||||||
|
set_kind("shared")
|
||||||
|
add_packages("libyuv")
|
||||||
|
add_deps("rd_log")
|
||||||
|
add_defines("CROSSDESK_WGC_PLUGIN_BUILD=1")
|
||||||
|
add_links("windowsapp")
|
||||||
|
add_files("src/screen_capturer/windows/screen_capturer_wgc.cpp",
|
||||||
|
"src/screen_capturer/windows/wgc_session_impl.cpp",
|
||||||
|
"src/screen_capturer/windows/wgc_plugin_entry.cpp")
|
||||||
|
add_includedirs("src/common", "src/screen_capturer",
|
||||||
|
"src/screen_capturer/windows")
|
||||||
|
end
|
||||||
|
|
||||||
|
target("crossdesk")
|
||||||
|
set_kind("binary")
|
||||||
|
add_deps("rd_log", "common", "gui")
|
||||||
|
add_files("src/app/*.cpp")
|
||||||
|
add_includedirs("src/app", {public = true})
|
||||||
|
if is_os("windows") then
|
||||||
|
add_deps("wgc_plugin")
|
||||||
|
add_files("scripts/windows/crossdesk.rc")
|
||||||
|
end
|
||||||
|
end
|
||||||
Reference in New Issue
Block a user