mirror of
https://github.com/kunkundi/crossdesk.git
synced 2026-03-25 01:27:25 +08:00
2426 lines
77 KiB
C++
2426 lines
77 KiB
C++
#include "render.h"
|
|
|
|
#include <libyuv.h>
|
|
|
|
#if defined(__linux__) && !defined(__APPLE__)
|
|
#include <X11/Xatom.h>
|
|
#include <X11/Xlib.h>
|
|
#endif
|
|
|
|
#include <cstring>
|
|
#include <filesystem>
|
|
#include <fstream>
|
|
#include <iostream>
|
|
#include <string>
|
|
#include <thread>
|
|
|
|
#include "OPPOSans_Regular.h"
|
|
#include "clipboard.h"
|
|
#include "device_controller_factory.h"
|
|
#include "fa_regular_400.h"
|
|
#include "fa_solid_900.h"
|
|
#include "file_transfer.h"
|
|
#include "layout_relative.h"
|
|
#include "localization.h"
|
|
#include "platform.h"
|
|
#include "rd_log.h"
|
|
#include "screen_capturer_factory.h"
|
|
#include "version_checker.h"
|
|
|
|
#if defined(__APPLE__)
|
|
#include "window_util_mac.h"
|
|
#endif
|
|
|
|
#define NV12_BUFFER_SIZE 1280 * 720 * 3 / 2
|
|
|
|
namespace crossdesk {
|
|
|
|
namespace {
|
|
#if defined(__linux__) && !defined(__APPLE__)
|
|
inline bool X11GetDisplayAndWindow(SDL_Window* window, Display** display_out,
|
|
::Window* x11_window_out) {
|
|
if (!window || !display_out || !x11_window_out) {
|
|
return false;
|
|
}
|
|
|
|
#if !defined(SDL_PROP_WINDOW_X11_DISPLAY_POINTER) || \
|
|
!defined(SDL_PROP_WINDOW_X11_WINDOW_NUMBER)
|
|
// SDL build does not expose X11 window properties.
|
|
return false;
|
|
#else
|
|
SDL_PropertiesID props = SDL_GetWindowProperties(window);
|
|
Display* display = (Display*)SDL_GetPointerProperty(
|
|
props, SDL_PROP_WINDOW_X11_DISPLAY_POINTER, NULL);
|
|
const Sint64 x11_window_num =
|
|
SDL_GetNumberProperty(props, SDL_PROP_WINDOW_X11_WINDOW_NUMBER, 0);
|
|
const ::Window x11_window = (::Window)x11_window_num;
|
|
|
|
if (!display || !x11_window) {
|
|
return false;
|
|
}
|
|
|
|
*display_out = display;
|
|
*x11_window_out = x11_window;
|
|
return true;
|
|
#endif
|
|
}
|
|
|
|
inline void X11SendNetWmState(Display* display, ::Window x11_window,
|
|
long action, Atom state1, Atom state2 = 0) {
|
|
if (!display || !x11_window) {
|
|
return;
|
|
}
|
|
|
|
const Atom wm_state = XInternAtom(display, "_NET_WM_STATE", False);
|
|
|
|
XEvent event;
|
|
memset(&event, 0, sizeof(event));
|
|
event.xclient.type = ClientMessage;
|
|
event.xclient.serial = 0;
|
|
event.xclient.send_event = True;
|
|
event.xclient.message_type = wm_state;
|
|
event.xclient.window = x11_window;
|
|
event.xclient.format = 32;
|
|
event.xclient.data.l[0] = action;
|
|
event.xclient.data.l[1] = (long)state1;
|
|
event.xclient.data.l[2] = (long)state2;
|
|
event.xclient.data.l[3] = 1; // normal source indication
|
|
event.xclient.data.l[4] = 0;
|
|
|
|
XSendEvent(display, DefaultRootWindow(display), False,
|
|
SubstructureRedirectMask | SubstructureNotifyMask, &event);
|
|
}
|
|
|
|
inline void X11SetWindowTypeUtility(Display* display, ::Window x11_window) {
|
|
if (!display || !x11_window) {
|
|
return;
|
|
}
|
|
|
|
const Atom wm_window_type =
|
|
XInternAtom(display, "_NET_WM_WINDOW_TYPE", False);
|
|
const Atom wm_window_type_utility =
|
|
XInternAtom(display, "_NET_WM_WINDOW_TYPE_UTILITY", False);
|
|
|
|
XChangeProperty(display, x11_window, wm_window_type, XA_ATOM, 32,
|
|
PropModeReplace, (unsigned char*)&wm_window_type_utility, 1);
|
|
}
|
|
|
|
inline void X11SetWindowAlwaysOnTop(SDL_Window* window) {
|
|
Display* display = nullptr;
|
|
::Window x11_window = 0;
|
|
if (!X11GetDisplayAndWindow(window, &display, &x11_window)) {
|
|
return;
|
|
}
|
|
|
|
const Atom state_above = XInternAtom(display, "_NET_WM_STATE_ABOVE", False);
|
|
const Atom state_stays_on_top =
|
|
XInternAtom(display, "_NET_WM_STATE_STAYS_ON_TOP", False);
|
|
|
|
// Request _NET_WM_STATE_ADD for ABOVE + STAYS_ON_TOP.
|
|
X11SendNetWmState(display, x11_window, 1, state_above, state_stays_on_top);
|
|
XFlush(display);
|
|
}
|
|
|
|
inline void X11SetWindowSkipTaskbar(SDL_Window* window) {
|
|
Display* display = nullptr;
|
|
::Window x11_window = 0;
|
|
if (!X11GetDisplayAndWindow(window, &display, &x11_window)) {
|
|
return;
|
|
}
|
|
|
|
const Atom skip_taskbar =
|
|
XInternAtom(display, "_NET_WM_STATE_SKIP_TASKBAR", False);
|
|
const Atom skip_pager =
|
|
XInternAtom(display, "_NET_WM_STATE_SKIP_PAGER", False);
|
|
|
|
// Request _NET_WM_STATE_ADD for SKIP_TASKBAR + SKIP_PAGER.
|
|
X11SendNetWmState(display, x11_window, 1, skip_taskbar, skip_pager);
|
|
|
|
// Hint the WM that this is an auxiliary/utility window.
|
|
X11SetWindowTypeUtility(display, x11_window);
|
|
|
|
XFlush(display);
|
|
}
|
|
#endif
|
|
} // namespace
|
|
|
|
std::vector<char> Render::SerializeRemoteAction(const RemoteAction& action) {
|
|
std::vector<char> buffer;
|
|
buffer.push_back(static_cast<char>(action.type));
|
|
|
|
auto insert_bytes = [&](const void* ptr, size_t len) {
|
|
buffer.insert(buffer.end(), (const char*)ptr, (const char*)ptr + len);
|
|
};
|
|
|
|
if (action.type == ControlType::host_infomation) {
|
|
insert_bytes(&action.i.host_name_size, sizeof(size_t));
|
|
insert_bytes(action.i.host_name, action.i.host_name_size);
|
|
|
|
size_t num = action.i.display_num;
|
|
insert_bytes(&num, sizeof(size_t));
|
|
|
|
for (size_t i = 0; i < num; ++i) {
|
|
size_t len = strlen(action.i.display_list[i]);
|
|
insert_bytes(&len, sizeof(size_t));
|
|
insert_bytes(action.i.display_list[i], len);
|
|
}
|
|
|
|
insert_bytes(action.i.left, sizeof(int) * num);
|
|
insert_bytes(action.i.top, sizeof(int) * num);
|
|
insert_bytes(action.i.right, sizeof(int) * num);
|
|
insert_bytes(action.i.bottom, sizeof(int) * num);
|
|
}
|
|
|
|
return buffer;
|
|
}
|
|
|
|
bool Render::DeserializeRemoteAction(const char* data, size_t size,
|
|
RemoteAction& out) {
|
|
size_t offset = 0;
|
|
auto read = [&](void* dst, size_t len) -> bool {
|
|
if (offset + len > size) return false;
|
|
memcpy(dst, data + offset, len);
|
|
offset += len;
|
|
return true;
|
|
};
|
|
|
|
if (size < 1) return false;
|
|
out.type = static_cast<ControlType>(data[offset++]);
|
|
|
|
if (out.type == ControlType::host_infomation) {
|
|
size_t name_len;
|
|
if (!read(&name_len, sizeof(size_t)) || name_len >= sizeof(out.i.host_name))
|
|
return false;
|
|
if (!read(out.i.host_name, name_len)) return false;
|
|
out.i.host_name[name_len] = '\0';
|
|
out.i.host_name_size = name_len;
|
|
|
|
size_t num;
|
|
if (!read(&num, sizeof(size_t))) return false;
|
|
out.i.display_num = num;
|
|
|
|
out.i.display_list = (char**)malloc(num * sizeof(char*));
|
|
for (size_t i = 0; i < num; ++i) {
|
|
size_t len;
|
|
if (!read(&len, sizeof(size_t))) return false;
|
|
if (offset + len > size) return false;
|
|
out.i.display_list[i] = (char*)malloc(len + 1);
|
|
memcpy(out.i.display_list[i], data + offset, len);
|
|
out.i.display_list[i][len] = '\0';
|
|
offset += len;
|
|
}
|
|
|
|
auto alloc_int_array = [&](int*& arr) {
|
|
arr = (int*)malloc(num * sizeof(int));
|
|
return read(arr, num * sizeof(int));
|
|
};
|
|
|
|
return alloc_int_array(out.i.left) && alloc_int_array(out.i.top) &&
|
|
alloc_int_array(out.i.right) && alloc_int_array(out.i.bottom);
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
void Render::FreeRemoteAction(RemoteAction& action) {
|
|
if (action.type == ControlType::host_infomation) {
|
|
for (size_t i = 0; i < action.i.display_num; ++i) {
|
|
free(action.i.display_list[i]);
|
|
}
|
|
free(action.i.display_list);
|
|
free(action.i.left);
|
|
free(action.i.top);
|
|
free(action.i.right);
|
|
free(action.i.bottom);
|
|
|
|
action.i.display_list = nullptr;
|
|
action.i.left = action.i.top = action.i.right = action.i.bottom = nullptr;
|
|
action.i.display_num = 0;
|
|
}
|
|
}
|
|
|
|
SDL_HitTestResult Render::HitTestCallback(SDL_Window* window,
|
|
const SDL_Point* area, void* data) {
|
|
Render* render = (Render*)data;
|
|
if (!render) {
|
|
return SDL_HITTEST_NORMAL;
|
|
}
|
|
|
|
if (render->fullscreen_button_pressed_) {
|
|
return SDL_HITTEST_NORMAL;
|
|
}
|
|
|
|
// Server window: OS-level dragging for the title bar, but keep the left-side
|
|
// collapse/expand button clickable.
|
|
if (render->server_window_ && window == render->server_window_) {
|
|
const float title_h = render->server_window_title_bar_height_;
|
|
const float button_w = title_h;
|
|
if (area->y >= 0 && area->y < title_h) {
|
|
if (area->x >= 0 && area->x < button_w) {
|
|
return SDL_HITTEST_NORMAL;
|
|
}
|
|
return SDL_HITTEST_DRAGGABLE;
|
|
}
|
|
return SDL_HITTEST_NORMAL;
|
|
}
|
|
|
|
int window_width, window_height;
|
|
SDL_GetWindowSize(window, &window_width, &window_height);
|
|
|
|
// check if curosor is in tab bar
|
|
if (render->stream_window_inited_ && render->stream_window_created_ &&
|
|
!render->fullscreen_button_pressed_ && render->stream_ctx_) {
|
|
ImGuiContext* prev_ctx = ImGui::GetCurrentContext();
|
|
ImGui::SetCurrentContext(render->stream_ctx_);
|
|
|
|
ImGuiWindow* tab_bar_window = ImGui::FindWindowByName("TabBar");
|
|
if (tab_bar_window && tab_bar_window->Active) {
|
|
ImGuiIO& io = ImGui::GetIO();
|
|
float scale_x = io.DisplayFramebufferScale.x;
|
|
float scale_y = io.DisplayFramebufferScale.y;
|
|
|
|
float tab_bar_x = tab_bar_window->Pos.x * scale_x;
|
|
float tab_bar_y = tab_bar_window->Pos.y * scale_y;
|
|
float tab_bar_width = tab_bar_window->Size.x * scale_x;
|
|
float tab_bar_height = tab_bar_window->Size.y * scale_y;
|
|
|
|
ImGui::SetCurrentContext(prev_ctx);
|
|
|
|
if (area->x >= tab_bar_x && area->x <= tab_bar_x + tab_bar_width &&
|
|
area->y >= tab_bar_y && area->y <= tab_bar_y + tab_bar_height) {
|
|
return SDL_HITTEST_NORMAL;
|
|
}
|
|
} else {
|
|
ImGui::SetCurrentContext(prev_ctx);
|
|
}
|
|
}
|
|
|
|
float mouse_grab_padding = render->title_bar_button_width_ * 0.16f;
|
|
if (area->y < render->title_bar_button_width_ &&
|
|
area->y > mouse_grab_padding &&
|
|
area->x < window_width - render->title_bar_button_width_ * 4.0f &&
|
|
area->x > mouse_grab_padding) {
|
|
return SDL_HITTEST_DRAGGABLE;
|
|
}
|
|
|
|
// if (!render->streaming_) {
|
|
// return SDL_HITTEST_NORMAL;
|
|
// }
|
|
|
|
if (area->y < mouse_grab_padding) {
|
|
if (area->x < mouse_grab_padding) {
|
|
return SDL_HITTEST_RESIZE_TOPLEFT;
|
|
} else if (area->x > window_width - mouse_grab_padding) {
|
|
return SDL_HITTEST_RESIZE_TOPRIGHT;
|
|
} else {
|
|
return SDL_HITTEST_RESIZE_TOP;
|
|
}
|
|
} else if (area->y > window_height - mouse_grab_padding) {
|
|
if (area->x < mouse_grab_padding) {
|
|
return SDL_HITTEST_RESIZE_BOTTOMLEFT;
|
|
} else if (area->x > window_width - mouse_grab_padding) {
|
|
return SDL_HITTEST_RESIZE_BOTTOMRIGHT;
|
|
} else {
|
|
return SDL_HITTEST_RESIZE_BOTTOM;
|
|
}
|
|
} else if (area->x < mouse_grab_padding) {
|
|
return SDL_HITTEST_RESIZE_LEFT;
|
|
} else if (area->x > window_width - mouse_grab_padding) {
|
|
return SDL_HITTEST_RESIZE_RIGHT;
|
|
}
|
|
|
|
return SDL_HITTEST_NORMAL;
|
|
}
|
|
|
|
Render::Render() : last_rejoin_check_time_(std::chrono::steady_clock::now()) {}
|
|
|
|
Render::~Render() {}
|
|
|
|
int Render::SaveSettingsIntoCacheFile() {
|
|
cd_cache_mutex_.lock();
|
|
|
|
std::ofstream cd_cache_v2_file(cache_path_ + "/secure_cache_v2.enc",
|
|
std::ios::binary);
|
|
if (!cd_cache_v2_file) {
|
|
cd_cache_mutex_.unlock();
|
|
return -1;
|
|
}
|
|
|
|
memset(&cd_cache_v2_.client_id_with_password, 0,
|
|
sizeof(cd_cache_v2_.client_id_with_password));
|
|
memcpy(cd_cache_v2_.client_id_with_password, client_id_with_password_,
|
|
sizeof(client_id_with_password_));
|
|
memcpy(&cd_cache_v2_.key, &aes128_key_, sizeof(aes128_key_));
|
|
memcpy(&cd_cache_v2_.iv, &aes128_iv_, sizeof(aes128_iv_));
|
|
|
|
memset(&cd_cache_v2_.self_hosted_id, 0, sizeof(cd_cache_v2_.self_hosted_id));
|
|
memcpy(cd_cache_v2_.self_hosted_id, self_hosted_id_, sizeof(self_hosted_id_));
|
|
|
|
cd_cache_v2_file.write(reinterpret_cast<char*>(&cd_cache_v2_),
|
|
sizeof(CDCacheV2));
|
|
cd_cache_v2_file.close();
|
|
|
|
std::ofstream cd_cache_file(cache_path_ + "/secure_cache.enc",
|
|
std::ios::binary);
|
|
if (cd_cache_file) {
|
|
memset(&cd_cache_.client_id_with_password, 0,
|
|
sizeof(cd_cache_.client_id_with_password));
|
|
memcpy(cd_cache_.client_id_with_password, client_id_with_password_,
|
|
sizeof(client_id_with_password_));
|
|
memcpy(&cd_cache_.key, &aes128_key_, sizeof(aes128_key_));
|
|
memcpy(&cd_cache_.iv, &aes128_iv_, sizeof(aes128_iv_));
|
|
cd_cache_file.write(reinterpret_cast<char*>(&cd_cache_), sizeof(CDCache));
|
|
cd_cache_file.close();
|
|
}
|
|
|
|
cd_cache_mutex_.unlock();
|
|
|
|
return 0;
|
|
}
|
|
|
|
int Render::LoadSettingsFromCacheFile() {
|
|
cd_cache_mutex_.lock();
|
|
|
|
std::ifstream cd_cache_v2_file(cache_path_ + "/secure_cache_v2.enc",
|
|
std::ios::binary);
|
|
bool v2_file_exists = cd_cache_v2_file.good();
|
|
|
|
if (v2_file_exists) {
|
|
cd_cache_v2_file.read(reinterpret_cast<char*>(&cd_cache_v2_),
|
|
sizeof(CDCacheV2));
|
|
cd_cache_v2_file.close();
|
|
|
|
memset(&client_id_with_password_, 0, sizeof(client_id_with_password_));
|
|
memcpy(client_id_with_password_, cd_cache_v2_.client_id_with_password,
|
|
sizeof(client_id_with_password_));
|
|
|
|
memset(&self_hosted_id_, 0, sizeof(self_hosted_id_));
|
|
memcpy(self_hosted_id_, cd_cache_v2_.self_hosted_id,
|
|
sizeof(self_hosted_id_));
|
|
|
|
memcpy(aes128_key_, cd_cache_v2_.key, sizeof(cd_cache_v2_.key));
|
|
memcpy(aes128_iv_, cd_cache_v2_.iv, sizeof(cd_cache_v2_.iv));
|
|
|
|
LOG_INFO("Load settings from v2 cache file");
|
|
} else {
|
|
std::ifstream cd_cache_file(cache_path_ + "/secure_cache.enc",
|
|
std::ios::binary);
|
|
if (!cd_cache_file) {
|
|
cd_cache_mutex_.unlock();
|
|
|
|
memset(password_saved_, 0, sizeof(password_saved_));
|
|
memset(aes128_key_, 0, sizeof(aes128_key_));
|
|
memset(aes128_iv_, 0, sizeof(aes128_iv_));
|
|
memset(self_hosted_id_, 0, sizeof(self_hosted_id_));
|
|
|
|
thumbnail_.reset();
|
|
thumbnail_ = std::make_shared<Thumbnail>(cache_path_ + "/thumbnails/");
|
|
thumbnail_->GetKeyAndIv(aes128_key_, aes128_iv_);
|
|
thumbnail_->DeleteAllFilesInDirectory();
|
|
|
|
SaveSettingsIntoCacheFile();
|
|
|
|
return -1;
|
|
}
|
|
|
|
cd_cache_file.read(reinterpret_cast<char*>(&cd_cache_), sizeof(CDCache));
|
|
cd_cache_file.close();
|
|
|
|
memset(&cd_cache_v2_.client_id_with_password, 0,
|
|
sizeof(cd_cache_v2_.client_id_with_password));
|
|
memcpy(cd_cache_v2_.client_id_with_password,
|
|
cd_cache_.client_id_with_password,
|
|
sizeof(cd_cache_.client_id_with_password));
|
|
memcpy(&cd_cache_v2_.key, &cd_cache_.key, sizeof(cd_cache_.key));
|
|
memcpy(&cd_cache_v2_.iv, &cd_cache_.iv, sizeof(cd_cache_.iv));
|
|
|
|
memset(&cd_cache_v2_.self_hosted_id, 0,
|
|
sizeof(cd_cache_v2_.self_hosted_id));
|
|
|
|
memset(&client_id_with_password_, 0, sizeof(client_id_with_password_));
|
|
memcpy(client_id_with_password_, cd_cache_.client_id_with_password,
|
|
sizeof(client_id_with_password_));
|
|
|
|
memset(&self_hosted_id_, 0, sizeof(self_hosted_id_));
|
|
|
|
memcpy(aes128_key_, cd_cache_.key, sizeof(cd_cache_.key));
|
|
memcpy(aes128_iv_, cd_cache_.iv, sizeof(cd_cache_.iv));
|
|
|
|
cd_cache_mutex_.unlock();
|
|
SaveSettingsIntoCacheFile();
|
|
cd_cache_mutex_.lock();
|
|
|
|
LOG_INFO("Migrated settings from v1 to v2 cache file");
|
|
}
|
|
|
|
cd_cache_mutex_.unlock();
|
|
|
|
if (strchr(client_id_with_password_, '@') != nullptr) {
|
|
std::string id, password;
|
|
const char* at_pos = strchr(client_id_with_password_, '@');
|
|
if (at_pos == nullptr) {
|
|
id = client_id_with_password_;
|
|
password.clear();
|
|
} else {
|
|
id.assign(client_id_with_password_, at_pos - client_id_with_password_);
|
|
password = at_pos + 1;
|
|
}
|
|
|
|
memset(&client_id_, 0, sizeof(client_id_));
|
|
strncpy(client_id_, id.c_str(), sizeof(client_id_) - 1);
|
|
client_id_[sizeof(client_id_) - 1] = '\0';
|
|
|
|
memset(&password_saved_, 0, sizeof(password_saved_));
|
|
strncpy(password_saved_, password.c_str(), sizeof(password_saved_) - 1);
|
|
password_saved_[sizeof(password_saved_) - 1] = '\0';
|
|
}
|
|
|
|
thumbnail_.reset();
|
|
thumbnail_ = std::make_shared<Thumbnail>(cache_path_ + "/thumbnails/",
|
|
aes128_key_, aes128_iv_);
|
|
|
|
language_button_value_ = (int)config_center_->GetLanguage();
|
|
video_quality_button_value_ = (int)config_center_->GetVideoQuality();
|
|
video_frame_rate_button_value_ = (int)config_center_->GetVideoFrameRate();
|
|
video_encode_format_button_value_ =
|
|
(int)config_center_->GetVideoEncodeFormat();
|
|
enable_hardware_video_codec_ = config_center_->IsHardwareVideoCodec();
|
|
enable_turn_ = config_center_->IsEnableTurn();
|
|
enable_srtp_ = config_center_->IsEnableSrtp();
|
|
enable_self_hosted_ = config_center_->IsSelfHosted();
|
|
enable_autostart_ = config_center_->IsEnableAutostart();
|
|
enable_daemon_ = config_center_->IsEnableDaemon();
|
|
enable_minimize_to_tray_ = config_center_->IsMinimizeToTray();
|
|
|
|
// File transfer save path
|
|
{
|
|
std::string saved_path = config_center_->GetFileTransferSavePath();
|
|
strncpy(file_transfer_save_path_buf_, saved_path.c_str(),
|
|
sizeof(file_transfer_save_path_buf_) - 1);
|
|
file_transfer_save_path_buf_[sizeof(file_transfer_save_path_buf_) - 1] =
|
|
'\0';
|
|
file_transfer_save_path_last_ = saved_path;
|
|
}
|
|
|
|
language_button_value_last_ = language_button_value_;
|
|
video_quality_button_value_last_ = video_quality_button_value_;
|
|
video_encode_format_button_value_last_ = video_encode_format_button_value_;
|
|
enable_hardware_video_codec_last_ = enable_hardware_video_codec_;
|
|
enable_turn_last_ = enable_turn_;
|
|
enable_srtp_last_ = enable_srtp_;
|
|
enable_self_hosted_last_ = enable_self_hosted_;
|
|
enable_autostart_last_ = enable_autostart_;
|
|
enable_minimize_to_tray_last_ = enable_minimize_to_tray_;
|
|
|
|
LOG_INFO("Load settings from cache file");
|
|
|
|
return 0;
|
|
}
|
|
|
|
int Render::ScreenCapturerInit() {
|
|
if (!screen_capturer_) {
|
|
screen_capturer_ = (ScreenCapturer*)screen_capturer_factory_->Create();
|
|
}
|
|
|
|
last_frame_time_ = std::chrono::duration_cast<std::chrono::milliseconds>(
|
|
std::chrono::steady_clock::now().time_since_epoch())
|
|
.count();
|
|
int fps = config_center_->GetVideoFrameRate() ==
|
|
ConfigCenter::VIDEO_FRAME_RATE::FPS_30
|
|
? 30
|
|
: 60;
|
|
LOG_INFO("Init screen capturer with {} fps", fps);
|
|
|
|
int screen_capturer_init_ret = screen_capturer_->Init(
|
|
fps,
|
|
[this, fps](unsigned char* data, int size, int width, int height,
|
|
const char* display_name) -> void {
|
|
auto now_time = std::chrono::duration_cast<std::chrono::milliseconds>(
|
|
std::chrono::steady_clock::now().time_since_epoch())
|
|
.count();
|
|
auto duration = now_time - last_frame_time_;
|
|
if (duration * fps >= 1000) { // ~60 FPS
|
|
XVideoFrame frame;
|
|
frame.data = (const char*)data;
|
|
frame.size = size;
|
|
frame.width = width;
|
|
frame.height = height;
|
|
frame.captured_timestamp = GetSystemTimeMicros(peer_);
|
|
SendVideoFrame(peer_, &frame, display_name);
|
|
last_frame_time_ = now_time;
|
|
}
|
|
});
|
|
|
|
if (0 == screen_capturer_init_ret) {
|
|
LOG_INFO("Init screen capturer success");
|
|
if (display_info_list_.empty()) {
|
|
display_info_list_ = screen_capturer_->GetDisplayInfoList();
|
|
}
|
|
return 0;
|
|
} else {
|
|
LOG_ERROR("Init screen capturer failed");
|
|
screen_capturer_->Destroy();
|
|
delete screen_capturer_;
|
|
screen_capturer_ = nullptr;
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
int Render::StartScreenCapturer() {
|
|
if (screen_capturer_) {
|
|
LOG_INFO("Start screen capturer, show cursor: {}", show_cursor_);
|
|
|
|
screen_capturer_->Start(show_cursor_);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int Render::StopScreenCapturer() {
|
|
if (screen_capturer_) {
|
|
LOG_INFO("Stop screen capturer");
|
|
screen_capturer_->Stop();
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int Render::StartSpeakerCapturer() {
|
|
if (!speaker_capturer_) {
|
|
speaker_capturer_ = (SpeakerCapturer*)speaker_capturer_factory_->Create();
|
|
int speaker_capturer_init_ret =
|
|
speaker_capturer_->Init([this](unsigned char* data, size_t size,
|
|
const char* audio_name) -> void {
|
|
SendAudioFrame(peer_, (const char*)data, size, audio_label_.c_str());
|
|
});
|
|
|
|
if (0 != speaker_capturer_init_ret) {
|
|
speaker_capturer_->Destroy();
|
|
delete speaker_capturer_;
|
|
speaker_capturer_ = nullptr;
|
|
}
|
|
}
|
|
|
|
if (speaker_capturer_) {
|
|
speaker_capturer_->Start();
|
|
start_speaker_capturer_ = true;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int Render::StopSpeakerCapturer() {
|
|
if (speaker_capturer_) {
|
|
speaker_capturer_->Stop();
|
|
start_speaker_capturer_ = false;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int Render::StartMouseController() {
|
|
if (!device_controller_factory_) {
|
|
LOG_INFO("Device controller factory is nullptr");
|
|
return -1;
|
|
}
|
|
mouse_controller_ = (MouseController*)device_controller_factory_->Create(
|
|
DeviceControllerFactory::Device::Mouse);
|
|
|
|
int mouse_controller_init_ret = mouse_controller_->Init(display_info_list_);
|
|
if (0 != mouse_controller_init_ret) {
|
|
LOG_INFO("Destroy mouse controller");
|
|
mouse_controller_->Destroy();
|
|
mouse_controller_ = nullptr;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int Render::StopMouseController() {
|
|
if (mouse_controller_) {
|
|
mouse_controller_->Destroy();
|
|
delete mouse_controller_;
|
|
mouse_controller_ = nullptr;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int Render::StartKeyboardCapturer() {
|
|
if (!keyboard_capturer_) {
|
|
LOG_INFO("keyboard capturer is nullptr");
|
|
return -1;
|
|
}
|
|
|
|
int keyboard_capturer_init_ret = keyboard_capturer_->Hook(
|
|
[](int key_code, bool is_down, void* user_ptr) {
|
|
if (user_ptr) {
|
|
Render* render = (Render*)user_ptr;
|
|
render->SendKeyCommand(key_code, is_down);
|
|
}
|
|
},
|
|
this);
|
|
if (0 != keyboard_capturer_init_ret) {
|
|
LOG_ERROR("Start keyboard capturer failed");
|
|
} else {
|
|
LOG_INFO("Start keyboard capturer");
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int Render::StopKeyboardCapturer() {
|
|
if (keyboard_capturer_) {
|
|
keyboard_capturer_->Unhook();
|
|
LOG_INFO("Stop keyboard capturer");
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int Render::CreateConnectionPeer() {
|
|
params_.use_cfg_file = false;
|
|
|
|
std::string signal_server_ip;
|
|
int signal_server_port;
|
|
int coturn_server_port;
|
|
|
|
if (config_center_->IsSelfHosted()) {
|
|
signal_server_ip = config_center_->GetSignalServerHost();
|
|
signal_server_port = config_center_->GetSignalServerPort();
|
|
coturn_server_port = config_center_->GetCoturnServerPort();
|
|
|
|
std::string current_self_hosted_ip = config_center_->GetSignalServerHost();
|
|
bool use_cached_id = false;
|
|
|
|
// Check secure_cache_v2.enc exists or not
|
|
std::ifstream v2_file(cache_path_ + "/secure_cache_v2.enc",
|
|
std::ios::binary);
|
|
if (v2_file.good()) {
|
|
CDCacheV2 temp_cache;
|
|
v2_file.read(reinterpret_cast<char*>(&temp_cache), sizeof(CDCacheV2));
|
|
v2_file.close();
|
|
|
|
if (strlen(temp_cache.self_hosted_id) > 0) {
|
|
memset(&self_hosted_id_, 0, sizeof(self_hosted_id_));
|
|
strncpy(self_hosted_id_, temp_cache.self_hosted_id,
|
|
sizeof(self_hosted_id_) - 1);
|
|
self_hosted_id_[sizeof(self_hosted_id_) - 1] = '\0';
|
|
use_cached_id = true;
|
|
|
|
std::string id, password;
|
|
const char* at_pos = strchr(self_hosted_id_, '@');
|
|
if (at_pos == nullptr) {
|
|
id = self_hosted_id_;
|
|
password.clear();
|
|
} else {
|
|
id.assign(self_hosted_id_, at_pos - self_hosted_id_);
|
|
password = at_pos + 1;
|
|
}
|
|
|
|
memset(&client_id_, 0, sizeof(client_id_));
|
|
strncpy(client_id_, id.c_str(), sizeof(client_id_) - 1);
|
|
client_id_[sizeof(client_id_) - 1] = '\0';
|
|
|
|
memset(&password_saved_, 0, sizeof(password_saved_));
|
|
strncpy(password_saved_, password.c_str(), sizeof(password_saved_) - 1);
|
|
password_saved_[sizeof(password_saved_) - 1] = '\0';
|
|
}
|
|
} else {
|
|
memset(&self_hosted_id_, 0, sizeof(self_hosted_id_));
|
|
LOG_INFO(
|
|
"secure_cache_v2.enc not found, will use empty id to get new id from "
|
|
"server");
|
|
}
|
|
|
|
if (use_cached_id && strlen(self_hosted_id_) > 0) {
|
|
memset(&self_hosted_user_id_, 0, sizeof(self_hosted_user_id_));
|
|
strncpy(self_hosted_user_id_, self_hosted_id_,
|
|
sizeof(self_hosted_user_id_) - 1);
|
|
self_hosted_user_id_[sizeof(self_hosted_user_id_) - 1] = '\0';
|
|
params_.user_id = self_hosted_user_id_;
|
|
} else {
|
|
memset(&self_hosted_user_id_, 0, sizeof(self_hosted_user_id_));
|
|
params_.user_id = self_hosted_user_id_;
|
|
LOG_INFO(
|
|
"Using empty id for self-hosted server, server will assign new id");
|
|
}
|
|
} else {
|
|
signal_server_ip = config_center_->GetDefaultServerHost();
|
|
signal_server_port = config_center_->GetDefaultSignalServerPort();
|
|
coturn_server_port = config_center_->GetDefaultCoturnServerPort();
|
|
params_.user_id = client_id_with_password_;
|
|
}
|
|
|
|
// self hosted server config
|
|
strncpy(signal_server_ip_self_, config_center_->GetSignalServerHost().c_str(),
|
|
sizeof(signal_server_ip_self_) - 1);
|
|
signal_server_ip_self_[sizeof(signal_server_ip_self_) - 1] = '\0';
|
|
int signal_port = config_center_->GetSignalServerPort();
|
|
if (signal_port > 0) {
|
|
strncpy(signal_server_port_self_, std::to_string(signal_port).c_str(),
|
|
sizeof(signal_server_port_self_) - 1);
|
|
signal_server_port_self_[sizeof(signal_server_port_self_) - 1] = '\0';
|
|
} else {
|
|
signal_server_port_self_[0] = '\0';
|
|
}
|
|
int coturn_port = config_center_->GetCoturnServerPort();
|
|
if (coturn_port > 0) {
|
|
strncpy(coturn_server_port_self_, std::to_string(coturn_port).c_str(),
|
|
sizeof(coturn_server_port_self_) - 1);
|
|
coturn_server_port_self_[sizeof(coturn_server_port_self_) - 1] = '\0';
|
|
} else {
|
|
coturn_server_port_self_[0] = '\0';
|
|
}
|
|
|
|
// peer config
|
|
strncpy((char*)params_.signal_server_ip, signal_server_ip.c_str(),
|
|
sizeof(params_.signal_server_ip) - 1);
|
|
params_.signal_server_ip[sizeof(params_.signal_server_ip) - 1] = '\0';
|
|
params_.signal_server_port = signal_server_port;
|
|
strncpy((char*)params_.stun_server_ip, signal_server_ip.c_str(),
|
|
sizeof(params_.stun_server_ip) - 1);
|
|
params_.stun_server_ip[sizeof(params_.stun_server_ip) - 1] = '\0';
|
|
params_.stun_server_port = coturn_server_port;
|
|
strncpy((char*)params_.turn_server_ip, signal_server_ip.c_str(),
|
|
sizeof(params_.turn_server_ip) - 1);
|
|
params_.turn_server_ip[sizeof(params_.turn_server_ip) - 1] = '\0';
|
|
params_.turn_server_port = coturn_server_port;
|
|
strncpy((char*)params_.turn_server_username, "crossdesk",
|
|
sizeof(params_.turn_server_username) - 1);
|
|
params_.turn_server_username[sizeof(params_.turn_server_username) - 1] = '\0';
|
|
strncpy((char*)params_.turn_server_password, "crossdeskpw",
|
|
sizeof(params_.turn_server_password) - 1);
|
|
params_.turn_server_password[sizeof(params_.turn_server_password) - 1] = '\0';
|
|
|
|
strncpy(params_.log_path, dll_log_path_.c_str(),
|
|
sizeof(params_.log_path) - 1);
|
|
params_.log_path[sizeof(params_.log_path) - 1] = '\0';
|
|
params_.hardware_acceleration = config_center_->IsHardwareVideoCodec();
|
|
params_.av1_encoding = config_center_->GetVideoEncodeFormat() ==
|
|
ConfigCenter::VIDEO_ENCODE_FORMAT::AV1
|
|
? true
|
|
: false;
|
|
params_.enable_turn = config_center_->IsEnableTurn();
|
|
params_.enable_srtp = config_center_->IsEnableSrtp();
|
|
params_.video_quality =
|
|
static_cast<VideoQuality>(config_center_->GetVideoQuality());
|
|
params_.on_receive_video_buffer = nullptr;
|
|
params_.on_receive_audio_buffer = OnReceiveAudioBufferCb;
|
|
params_.on_receive_data_buffer = OnReceiveDataBufferCb;
|
|
|
|
params_.on_receive_video_frame = OnReceiveVideoBufferCb;
|
|
|
|
params_.on_signal_status = OnSignalStatusCb;
|
|
params_.on_connection_status = OnConnectionStatusCb;
|
|
params_.on_net_status_report = OnNetStatusReport;
|
|
|
|
params_.user_data = this;
|
|
|
|
peer_ = CreatePeer(¶ms_);
|
|
if (peer_) {
|
|
LOG_INFO("Create peer instance [{}] successful", client_id_);
|
|
Init(peer_);
|
|
LOG_INFO("Peer [{}] init finish", client_id_);
|
|
} else {
|
|
LOG_INFO("Create peer [{}] instance failed", client_id_);
|
|
}
|
|
|
|
if (0 == ScreenCapturerInit()) {
|
|
for (auto& display_info : display_info_list_) {
|
|
AddVideoStream(peer_, display_info.name.c_str());
|
|
}
|
|
|
|
AddAudioStream(peer_, audio_label_.c_str());
|
|
AddDataStream(peer_, data_label_.c_str(), false);
|
|
AddDataStream(peer_, control_data_label_.c_str(), true);
|
|
AddDataStream(peer_, file_label_.c_str(), true);
|
|
AddDataStream(peer_, file_feedback_label_.c_str(), true);
|
|
AddDataStream(peer_, clipboard_label_.c_str(), true);
|
|
return 0;
|
|
} else {
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
int Render::AudioDeviceInit() {
|
|
SDL_AudioSpec desired_out{};
|
|
desired_out.freq = 48000;
|
|
desired_out.format = SDL_AUDIO_S16;
|
|
desired_out.channels = 1;
|
|
|
|
output_stream_ = SDL_OpenAudioDeviceStream(SDL_AUDIO_DEVICE_DEFAULT_PLAYBACK,
|
|
&desired_out, nullptr, nullptr);
|
|
if (!output_stream_) {
|
|
LOG_ERROR("Failed to open output stream: {}", SDL_GetError());
|
|
return -1;
|
|
}
|
|
|
|
SDL_ResumeAudioDevice(SDL_GetAudioStreamDevice(output_stream_));
|
|
|
|
return 0;
|
|
}
|
|
|
|
int Render::AudioDeviceDestroy() {
|
|
if (output_stream_) {
|
|
SDL_CloseAudioDevice(SDL_GetAudioStreamDevice(output_stream_));
|
|
SDL_DestroyAudioStream(output_stream_);
|
|
output_stream_ = nullptr;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
void Render::UpdateInteractions() {
|
|
if (start_screen_capturer_ && !screen_capturer_is_started_) {
|
|
StartScreenCapturer();
|
|
screen_capturer_is_started_ = true;
|
|
} else if (!start_screen_capturer_ && screen_capturer_is_started_) {
|
|
StopScreenCapturer();
|
|
screen_capturer_is_started_ = false;
|
|
}
|
|
|
|
if (start_speaker_capturer_ && !speaker_capturer_is_started_) {
|
|
StartSpeakerCapturer();
|
|
speaker_capturer_is_started_ = true;
|
|
} else if (!start_speaker_capturer_ && speaker_capturer_is_started_) {
|
|
StopSpeakerCapturer();
|
|
speaker_capturer_is_started_ = false;
|
|
}
|
|
|
|
if (start_mouse_controller_ && !mouse_controller_is_started_) {
|
|
StartMouseController();
|
|
mouse_controller_is_started_ = true;
|
|
} else if (!start_mouse_controller_ && mouse_controller_is_started_) {
|
|
StopMouseController();
|
|
mouse_controller_is_started_ = false;
|
|
}
|
|
|
|
if (start_keyboard_capturer_ && foucs_on_stream_window_) {
|
|
if (!keyboard_capturer_is_started_) {
|
|
StartKeyboardCapturer();
|
|
keyboard_capturer_is_started_ = true;
|
|
}
|
|
} else if (keyboard_capturer_is_started_) {
|
|
StopKeyboardCapturer();
|
|
keyboard_capturer_is_started_ = false;
|
|
}
|
|
}
|
|
|
|
int Render::CreateMainWindow() {
|
|
main_ctx_ = ImGui::CreateContext();
|
|
if (!main_ctx_) {
|
|
LOG_ERROR("Main context is null");
|
|
return -1;
|
|
}
|
|
|
|
ImGui::SetCurrentContext(main_ctx_);
|
|
|
|
if (!SDL_CreateWindowAndRenderer(
|
|
"Remote Desk", (int)main_window_width_, (int)main_window_height_,
|
|
SDL_WINDOW_HIGH_PIXEL_DENSITY | SDL_WINDOW_BORDERLESS |
|
|
SDL_WINDOW_HIDDEN,
|
|
&main_window_, &main_renderer_)) {
|
|
LOG_ERROR("Error creating MainWindow and MainRenderer: {}", SDL_GetError());
|
|
return -1;
|
|
}
|
|
|
|
float dpi_scale = SDL_GetDisplayContentScale(SDL_GetPrimaryDisplay());
|
|
if (std::abs(dpi_scale_ - dpi_scale) > 0.01f) {
|
|
dpi_scale_ = dpi_scale;
|
|
|
|
main_window_width_ = (int)(main_window_width_default_ * dpi_scale_);
|
|
main_window_height_ = (int)(main_window_height_default_ * dpi_scale_);
|
|
stream_window_width_ = (int)(stream_window_width_default_ * dpi_scale_);
|
|
stream_window_height_ = (int)(stream_window_height_default_ * dpi_scale_);
|
|
server_window_width_ = (int)(server_window_width_default_ * dpi_scale_);
|
|
server_window_height_ = (int)(server_window_height_default_ * dpi_scale_);
|
|
server_window_normal_width_ =
|
|
(int)(server_window_width_default_ * dpi_scale_);
|
|
server_window_normal_height_ =
|
|
(int)(server_window_height_default_ * dpi_scale_);
|
|
|
|
SDL_SetWindowSize(main_window_, (int)main_window_width_,
|
|
(int)main_window_height_);
|
|
}
|
|
|
|
SDL_SetWindowResizable(main_window_, false);
|
|
|
|
// for window region action
|
|
SDL_SetWindowHitTest(main_window_, HitTestCallback, this);
|
|
|
|
SetupFontAndStyle(&main_windows_system_chinese_font_);
|
|
|
|
ImGuiStyle& style = ImGui::GetStyle();
|
|
style.ScaleAllSizes(dpi_scale_);
|
|
style.FontScaleDpi = dpi_scale_;
|
|
|
|
#if _WIN32
|
|
SDL_PropertiesID props = SDL_GetWindowProperties(main_window_);
|
|
HWND main_hwnd = (HWND)SDL_GetPointerProperty(
|
|
props, SDL_PROP_WINDOW_WIN32_HWND_POINTER, NULL);
|
|
|
|
HICON tray_icon = (HICON)LoadImageW(NULL, L"crossdesk.ico", IMAGE_ICON, 0, 0,
|
|
LR_LOADFROMFILE | LR_DEFAULTSIZE);
|
|
tray_ = std::make_unique<WinTray>(main_hwnd, tray_icon, L"CrossDesk",
|
|
localization_language_index_);
|
|
#endif
|
|
|
|
ImGui_ImplSDL3_InitForSDLRenderer(main_window_, main_renderer_);
|
|
ImGui_ImplSDLRenderer3_Init(main_renderer_);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int Render::DestroyMainWindow() {
|
|
if (main_ctx_) {
|
|
ImGui::SetCurrentContext(main_ctx_);
|
|
}
|
|
|
|
if (main_renderer_) {
|
|
SDL_DestroyRenderer(main_renderer_);
|
|
}
|
|
|
|
if (main_window_) {
|
|
SDL_DestroyWindow(main_window_);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int Render::CreateStreamWindow() {
|
|
if (stream_window_created_) {
|
|
return 0;
|
|
}
|
|
|
|
stream_window_width_ = (int)(stream_window_width_default_ * dpi_scale_);
|
|
stream_window_height_ = (int)(stream_window_height_default_ * dpi_scale_);
|
|
|
|
stream_ctx_ = ImGui::CreateContext();
|
|
if (!stream_ctx_) {
|
|
LOG_ERROR("Stream context is null");
|
|
return -1;
|
|
}
|
|
|
|
ImGui::SetCurrentContext(stream_ctx_);
|
|
|
|
if (!SDL_CreateWindowAndRenderer(
|
|
"Stream window", (int)stream_window_width_,
|
|
(int)stream_window_height_,
|
|
SDL_WINDOW_HIGH_PIXEL_DENSITY | SDL_WINDOW_BORDERLESS,
|
|
&stream_window_, &stream_renderer_)) {
|
|
LOG_ERROR("Error creating stream_window_ and stream_renderer_: {}",
|
|
SDL_GetError());
|
|
return -1;
|
|
}
|
|
|
|
stream_pixformat_ = SDL_PIXELFORMAT_NV12;
|
|
|
|
SDL_SetWindowResizable(stream_window_, true);
|
|
|
|
// for window region action
|
|
SDL_SetWindowHitTest(stream_window_, HitTestCallback, this);
|
|
|
|
SetupFontAndStyle(&stream_windows_system_chinese_font_);
|
|
|
|
ImGuiStyle& style = ImGui::GetStyle();
|
|
style.ScaleAllSizes(dpi_scale_);
|
|
style.FontScaleDpi = dpi_scale_;
|
|
|
|
ImGui_ImplSDL3_InitForSDLRenderer(stream_window_, stream_renderer_);
|
|
ImGui_ImplSDLRenderer3_Init(stream_renderer_);
|
|
|
|
// change props->stream_render_rect_
|
|
SDL_Event event;
|
|
event.type = SDL_EVENT_WINDOW_PIXEL_SIZE_CHANGED;
|
|
event.window.windowID = SDL_GetWindowID(stream_window_);
|
|
SDL_PushEvent(&event);
|
|
|
|
stream_window_created_ = true;
|
|
just_created_ = true;
|
|
|
|
stream_window_inited_ = true;
|
|
LOG_INFO("Stream window inited");
|
|
|
|
return 0;
|
|
}
|
|
|
|
int Render::DestroyStreamWindow() {
|
|
stream_window_width_ = (float)stream_window_width_default_;
|
|
stream_window_height_ = (float)stream_window_height_default_;
|
|
|
|
if (stream_ctx_) {
|
|
ImGui::SetCurrentContext(stream_ctx_);
|
|
}
|
|
|
|
if (stream_renderer_) {
|
|
SDL_DestroyRenderer(stream_renderer_);
|
|
}
|
|
|
|
if (stream_window_) {
|
|
SDL_DestroyWindow(stream_window_);
|
|
}
|
|
|
|
stream_window_created_ = false;
|
|
|
|
return 0;
|
|
}
|
|
|
|
int Render::CreateServerWindow() {
|
|
if (server_window_created_) {
|
|
return 0;
|
|
}
|
|
server_ctx_ = ImGui::CreateContext();
|
|
if (!server_ctx_) {
|
|
LOG_ERROR("Server context is null");
|
|
return -1;
|
|
}
|
|
ImGui::SetCurrentContext(server_ctx_);
|
|
if (!SDL_CreateWindowAndRenderer("Server window", (int)server_window_width_,
|
|
(int)server_window_height_,
|
|
SDL_WINDOW_HIGH_PIXEL_DENSITY |
|
|
SDL_WINDOW_BORDERLESS |
|
|
SDL_WINDOW_TRANSPARENT,
|
|
&server_window_, &server_renderer_)) {
|
|
LOG_ERROR("Error creating server_window_ and server_renderer_: {}",
|
|
SDL_GetError());
|
|
return -1;
|
|
}
|
|
|
|
#if _WIN32
|
|
// Hide server window from the taskbar by making it a tool window.
|
|
{
|
|
SDL_PropertiesID server_props = SDL_GetWindowProperties(server_window_);
|
|
HWND server_hwnd = (HWND)SDL_GetPointerProperty(
|
|
server_props, SDL_PROP_WINDOW_WIN32_HWND_POINTER, NULL);
|
|
|
|
if (server_hwnd) {
|
|
LONG_PTR ex_style = GetWindowLongPtr(server_hwnd, GWL_EXSTYLE);
|
|
ex_style |= WS_EX_TOOLWINDOW;
|
|
ex_style &= ~WS_EX_APPWINDOW;
|
|
SetWindowLongPtr(server_hwnd, GWL_EXSTYLE, ex_style);
|
|
|
|
// Keep the server window above normal windows.
|
|
SetWindowPos(server_hwnd, HWND_TOPMOST, 0, 0, 0, 0,
|
|
SWP_NOMOVE | SWP_NOSIZE | SWP_FRAMECHANGED | SWP_NOACTIVATE);
|
|
}
|
|
}
|
|
#endif
|
|
|
|
#if defined(__linux__) && !defined(__APPLE__)
|
|
// Best-effort keep above other windows on X11.
|
|
X11SetWindowAlwaysOnTop(server_window_);
|
|
// Best-effort hide from taskbar on X11.
|
|
X11SetWindowSkipTaskbar(server_window_);
|
|
#endif
|
|
|
|
#if defined(__APPLE__)
|
|
// Best-effort keep above other windows on macOS.
|
|
MacSetWindowAlwaysOnTop(server_window_, true);
|
|
// Best-effort exclude from Window menu / window cycling.
|
|
MacSetWindowExcludedFromWindowMenu(server_window_, true);
|
|
#endif
|
|
|
|
// Set window position to bottom-right corner
|
|
SDL_Rect display_bounds;
|
|
if (SDL_GetDisplayUsableBounds(SDL_GetDisplayForWindow(server_window_),
|
|
&display_bounds)) {
|
|
int window_x =
|
|
display_bounds.x + display_bounds.w - (int)server_window_width_;
|
|
int window_y =
|
|
display_bounds.y + display_bounds.h - (int)server_window_height_;
|
|
SDL_SetWindowPosition(server_window_, window_x, window_y);
|
|
}
|
|
|
|
SDL_SetWindowResizable(server_window_, false);
|
|
|
|
SDL_SetRenderDrawBlendMode(server_renderer_, SDL_BLENDMODE_BLEND);
|
|
|
|
// for window region action
|
|
SDL_SetWindowHitTest(server_window_, HitTestCallback, this);
|
|
|
|
SetupFontAndStyle(&server_windows_system_chinese_font_);
|
|
|
|
ImGuiStyle& style = ImGui::GetStyle();
|
|
style.ScaleAllSizes(dpi_scale_);
|
|
style.FontScaleDpi = dpi_scale_;
|
|
|
|
ImGui_ImplSDL3_InitForSDLRenderer(server_window_, server_renderer_);
|
|
ImGui_ImplSDLRenderer3_Init(server_renderer_);
|
|
|
|
server_window_created_ = true;
|
|
server_window_inited_ = true;
|
|
|
|
LOG_INFO("Server window inited");
|
|
|
|
return 0;
|
|
}
|
|
|
|
int Render::DestroyServerWindow() {
|
|
if (server_ctx_) {
|
|
ImGui::SetCurrentContext(server_ctx_);
|
|
}
|
|
|
|
if (server_renderer_) {
|
|
SDL_DestroyRenderer(server_renderer_);
|
|
}
|
|
|
|
if (server_window_) {
|
|
SDL_DestroyWindow(server_window_);
|
|
}
|
|
|
|
server_window_created_ = false;
|
|
|
|
return 0;
|
|
}
|
|
|
|
int Render::SetupFontAndStyle(ImFont** system_chinese_font_out) {
|
|
float font_size = 32.0f;
|
|
|
|
// Setup Dear ImGui style
|
|
ImGuiIO& io = ImGui::GetIO();
|
|
|
|
io.IniFilename = NULL; // disable imgui.ini
|
|
|
|
// Load Fonts
|
|
ImFontConfig config;
|
|
config.FontDataOwnedByAtlas = false;
|
|
io.Fonts->AddFontFromMemoryTTF(OPPOSans_Regular_ttf, OPPOSans_Regular_ttf_len,
|
|
font_size, &config,
|
|
io.Fonts->GetGlyphRangesChineseFull());
|
|
config.MergeMode = true;
|
|
static const ImWchar icon_ranges[] = {ICON_MIN_FA, ICON_MAX_FA, 0};
|
|
io.Fonts->AddFontFromMemoryTTF(fa_solid_900_ttf, fa_solid_900_ttf_len, 30.0f,
|
|
&config, icon_ranges);
|
|
|
|
// Load system Chinese font as fallback
|
|
config.MergeMode = false;
|
|
config.FontDataOwnedByAtlas = false;
|
|
if (system_chinese_font_out) {
|
|
*system_chinese_font_out = nullptr;
|
|
}
|
|
|
|
#if defined(_WIN32)
|
|
// Windows: Try Microsoft YaHei (微软雅黑) first, then SimSun (宋体)
|
|
const char* font_paths[] = {"C:/Windows/Fonts/msyh.ttc",
|
|
"C:/Windows/Fonts/msyhbd.ttc",
|
|
"C:/Windows/Fonts/simsun.ttc", nullptr};
|
|
#elif defined(__APPLE__)
|
|
// macOS: Try PingFang SC first, then STHeiti
|
|
const char* font_paths[] = {"/System/Library/Fonts/PingFang.ttc",
|
|
"/System/Library/Fonts/STHeiti Light.ttc",
|
|
"/System/Library/Fonts/STHeiti Medium.ttc",
|
|
nullptr};
|
|
#else
|
|
// Linux: Try common Chinese fonts
|
|
const char* font_paths[] = {
|
|
"/usr/share/fonts/truetype/wqy/wqy-microhei.ttc",
|
|
"/usr/share/fonts/truetype/wqy/wqy-zenhei.ttc",
|
|
"/usr/share/fonts/truetype/arphic/uming.ttc",
|
|
"/usr/share/fonts/truetype/noto/NotoSansCJK-Regular.ttc", nullptr};
|
|
#endif
|
|
|
|
for (int i = 0; font_paths[i] != nullptr; i++) {
|
|
std::ifstream font_file(font_paths[i], std::ios::binary);
|
|
if (font_file.good()) {
|
|
font_file.close();
|
|
if (!system_chinese_font_out) {
|
|
break;
|
|
}
|
|
|
|
*system_chinese_font_out =
|
|
io.Fonts->AddFontFromFileTTF(font_paths[i], font_size, &config,
|
|
io.Fonts->GetGlyphRangesChineseFull());
|
|
if (*system_chinese_font_out != nullptr) {
|
|
// Merge FontAwesome icons into the Chinese font
|
|
config.MergeMode = true;
|
|
static const ImWchar icon_ranges[] = {ICON_MIN_FA, ICON_MAX_FA, 0};
|
|
io.Fonts->AddFontFromMemoryTTF(fa_solid_900_ttf, fa_solid_900_ttf_len,
|
|
font_size, &config, icon_ranges);
|
|
config.MergeMode = false;
|
|
LOG_INFO("Loaded system Chinese font with icons: {}", font_paths[i]);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
// If no system font found, use default font
|
|
if (system_chinese_font_out && *system_chinese_font_out == nullptr) {
|
|
*system_chinese_font_out = io.Fonts->AddFontDefault(&config);
|
|
// Merge FontAwesome icons into the default font
|
|
config.MergeMode = true;
|
|
static const ImWchar icon_ranges[] = {ICON_MIN_FA, ICON_MAX_FA, 0};
|
|
io.Fonts->AddFontFromMemoryTTF(fa_solid_900_ttf, fa_solid_900_ttf_len,
|
|
font_size, &config, icon_ranges);
|
|
config.MergeMode = false;
|
|
LOG_WARN("System Chinese font not found, using default font with icons");
|
|
}
|
|
|
|
ImGui::StyleColorsLight();
|
|
|
|
return 0;
|
|
}
|
|
|
|
int Render::DestroyMainWindowContext() {
|
|
ImGui::SetCurrentContext(main_ctx_);
|
|
ImGui_ImplSDLRenderer3_Shutdown();
|
|
ImGui_ImplSDL3_Shutdown();
|
|
ImGui::DestroyContext(main_ctx_);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int Render::DestroyStreamWindowContext() {
|
|
stream_window_inited_ = false;
|
|
ImGui::SetCurrentContext(stream_ctx_);
|
|
ImGui_ImplSDLRenderer3_Shutdown();
|
|
ImGui_ImplSDL3_Shutdown();
|
|
ImGui::DestroyContext(stream_ctx_);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int Render::DrawMainWindow() {
|
|
if (!main_ctx_) {
|
|
LOG_ERROR("Main context is null");
|
|
return -1;
|
|
}
|
|
|
|
ImGui::SetCurrentContext(main_ctx_);
|
|
ImGui_ImplSDLRenderer3_NewFrame();
|
|
ImGui_ImplSDL3_NewFrame();
|
|
ImGui::NewFrame();
|
|
|
|
ImGuiIO& io = ImGui::GetIO();
|
|
ImGui::PushStyleColor(ImGuiCol_WindowBg, ImVec4(1.0f, 1.0f, 1.0f, 1.0f));
|
|
ImGui::SetNextWindowPos(ImVec2(0, 0), ImGuiCond_Always);
|
|
ImGui::SetNextWindowSize(ImVec2(io.DisplaySize.x, io.DisplaySize.y),
|
|
ImGuiCond_Always);
|
|
ImGui::Begin("MainRender", nullptr,
|
|
ImGuiWindowFlags_NoMove | ImGuiWindowFlags_NoDecoration |
|
|
ImGuiWindowFlags_NoBringToFrontOnFocus |
|
|
ImGuiWindowFlags_NoDocking);
|
|
ImGui::PopStyleColor();
|
|
|
|
TitleBar(true);
|
|
|
|
MainWindow();
|
|
|
|
UpdateNotificationWindow();
|
|
|
|
#ifdef __APPLE__
|
|
if (show_request_permission_window_) {
|
|
RequestPermissionWindow();
|
|
}
|
|
#endif
|
|
|
|
ImGui::End();
|
|
|
|
// Rendering
|
|
(void)io;
|
|
ImGui::Render();
|
|
SDL_SetRenderScale(main_renderer_, io.DisplayFramebufferScale.x,
|
|
io.DisplayFramebufferScale.y);
|
|
SDL_RenderClear(main_renderer_);
|
|
ImGui_ImplSDLRenderer3_RenderDrawData(ImGui::GetDrawData(), main_renderer_);
|
|
SDL_RenderPresent(main_renderer_);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int Render::DrawStreamWindow() {
|
|
if (!stream_ctx_) {
|
|
LOG_ERROR("Stream context is null");
|
|
return -1;
|
|
}
|
|
|
|
ImGui::SetCurrentContext(stream_ctx_);
|
|
ImGui_ImplSDLRenderer3_NewFrame();
|
|
ImGui_ImplSDL3_NewFrame();
|
|
ImGui::NewFrame();
|
|
|
|
StreamWindow();
|
|
|
|
ImGuiIO& io = ImGui::GetIO();
|
|
float stream_title_window_height =
|
|
fullscreen_button_pressed_ ? 0 : title_bar_height_;
|
|
|
|
ImGui::PushStyleColor(ImGuiCol_WindowBg, ImVec4(1.0f, 1.0f, 1.0f, 1.0f));
|
|
// Set minimum window size to 0 to allow exact height control
|
|
ImGui::PushStyleVar(ImGuiStyleVar_WindowMinSize, ImVec2(0, 0));
|
|
ImGui::PushStyleVar(ImGuiStyleVar_WindowPadding, ImVec2(0, 0));
|
|
ImGui::SetNextWindowPos(ImVec2(0, 0), ImGuiCond_Always);
|
|
ImGui::SetNextWindowSize(ImVec2(io.DisplaySize.x, stream_title_window_height),
|
|
ImGuiCond_Always);
|
|
ImGui::Begin("StreamTitleWindow", nullptr,
|
|
ImGuiWindowFlags_NoMove | ImGuiWindowFlags_NoDecoration |
|
|
ImGuiWindowFlags_NoBringToFrontOnFocus |
|
|
ImGuiWindowFlags_NoDocking);
|
|
ImGui::PopStyleVar(2);
|
|
ImGui::PopStyleColor();
|
|
|
|
if (!fullscreen_button_pressed_) {
|
|
TitleBar(false);
|
|
}
|
|
|
|
ImGui::End();
|
|
|
|
// Rendering
|
|
(void)io;
|
|
ImGui::Render();
|
|
SDL_SetRenderScale(stream_renderer_, io.DisplayFramebufferScale.x,
|
|
io.DisplayFramebufferScale.y);
|
|
SDL_RenderClear(stream_renderer_);
|
|
|
|
// std::shared_lock lock(client_properties_mutex_);
|
|
for (auto& it : client_properties_) {
|
|
auto props = it.second;
|
|
if (props->tab_selected_) {
|
|
SDL_FRect render_rect_f = {
|
|
static_cast<float>(props->stream_render_rect_.x),
|
|
static_cast<float>(props->stream_render_rect_.y),
|
|
static_cast<float>(props->stream_render_rect_.w),
|
|
static_cast<float>(props->stream_render_rect_.h)};
|
|
SDL_RenderTexture(stream_renderer_, props->stream_texture_, NULL,
|
|
&render_rect_f);
|
|
}
|
|
}
|
|
ImGui_ImplSDLRenderer3_RenderDrawData(ImGui::GetDrawData(), stream_renderer_);
|
|
SDL_RenderPresent(stream_renderer_);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int Render::DrawServerWindow() {
|
|
if (!server_ctx_) {
|
|
LOG_ERROR("Server context is null");
|
|
return -1;
|
|
}
|
|
|
|
if (server_window_) {
|
|
int w = 0;
|
|
int h = 0;
|
|
SDL_GetWindowSize(server_window_, &w, &h);
|
|
if (w > 0 && h > 0) {
|
|
server_window_width_ = (float)w;
|
|
server_window_height_ = (float)h;
|
|
}
|
|
}
|
|
|
|
ImGui::SetCurrentContext(server_ctx_);
|
|
ImGui_ImplSDLRenderer3_NewFrame();
|
|
ImGui_ImplSDL3_NewFrame();
|
|
ImGui::NewFrame();
|
|
ServerWindow();
|
|
ImGui::Render();
|
|
SDL_SetRenderDrawColor(server_renderer_, 255, 255, 255, 255);
|
|
SDL_RenderClear(server_renderer_);
|
|
ImGui_ImplSDLRenderer3_RenderDrawData(ImGui::GetDrawData(), server_renderer_);
|
|
SDL_RenderPresent(server_renderer_);
|
|
return 0;
|
|
}
|
|
|
|
int Render::Run() {
|
|
latest_version_info_ = CheckUpdate();
|
|
if (!latest_version_info_.empty() &&
|
|
latest_version_info_.contains("version") &&
|
|
latest_version_info_["version"].is_string()) {
|
|
latest_version_ = latest_version_info_["version"];
|
|
if (latest_version_info_.contains("releaseNotes") &&
|
|
latest_version_info_["releaseNotes"].is_string()) {
|
|
release_notes_ = latest_version_info_["releaseNotes"];
|
|
} else {
|
|
release_notes_ = "";
|
|
}
|
|
update_available_ = IsNewerVersion(CROSSDESK_VERSION, latest_version_);
|
|
if (update_available_) {
|
|
show_update_notification_window_ = true;
|
|
}
|
|
} else {
|
|
latest_version_ = "";
|
|
update_available_ = false;
|
|
}
|
|
|
|
path_manager_ = std::make_unique<PathManager>("CrossDesk");
|
|
if (path_manager_) {
|
|
exec_log_path_ = path_manager_->GetLogPath().string();
|
|
dll_log_path_ = path_manager_->GetLogPath().string();
|
|
cache_path_ = path_manager_->GetCachePath().string();
|
|
config_center_ =
|
|
std::make_unique<ConfigCenter>(cache_path_ + "/config.ini");
|
|
strncpy(signal_server_ip_self_,
|
|
config_center_->GetSignalServerHost().c_str(),
|
|
sizeof(signal_server_ip_self_) - 1);
|
|
signal_server_ip_self_[sizeof(signal_server_ip_self_) - 1] = '\0';
|
|
int signal_port_init = config_center_->GetSignalServerPort();
|
|
if (signal_port_init > 0) {
|
|
strncpy(signal_server_port_self_,
|
|
std::to_string(signal_port_init).c_str(),
|
|
sizeof(signal_server_port_self_) - 1);
|
|
signal_server_port_self_[sizeof(signal_server_port_self_) - 1] = '\0';
|
|
} else {
|
|
signal_server_port_self_[0] = '\0';
|
|
}
|
|
} else {
|
|
std::cerr << "Failed to create PathManager" << std::endl;
|
|
return -1;
|
|
}
|
|
|
|
InitializeLogger();
|
|
LOG_INFO("CrossDesk version: {}", CROSSDESK_VERSION);
|
|
|
|
InitializeSettings();
|
|
InitializeSDL();
|
|
InitializeModules();
|
|
InitializeMainWindow();
|
|
|
|
const int scaled_video_width_ = 160;
|
|
const int scaled_video_height_ = 90;
|
|
|
|
MainLoop();
|
|
|
|
Cleanup();
|
|
|
|
return 0;
|
|
}
|
|
|
|
void Render::InitializeLogger() { InitLogger(exec_log_path_); }
|
|
|
|
void Render::InitializeSettings() {
|
|
LoadSettingsFromCacheFile();
|
|
|
|
localization_language_ = (ConfigCenter::LANGUAGE)language_button_value_;
|
|
localization_language_index_ = language_button_value_;
|
|
if (localization_language_index_ != 0 && localization_language_index_ != 1) {
|
|
localization_language_index_ = 0;
|
|
LOG_ERROR("Invalid language index: [{}], use [0] by default",
|
|
localization_language_index_);
|
|
}
|
|
}
|
|
|
|
void Render::InitializeSDL() {
|
|
if (!SDL_Init(SDL_INIT_VIDEO | SDL_INIT_AUDIO)) {
|
|
LOG_ERROR("Error: {}", SDL_GetError());
|
|
return;
|
|
}
|
|
|
|
const SDL_DisplayMode* dm = SDL_GetCurrentDisplayMode(0);
|
|
if (dm) {
|
|
screen_width_ = dm->w;
|
|
screen_height_ = dm->h;
|
|
}
|
|
|
|
STREAM_REFRESH_EVENT = SDL_RegisterEvents(1);
|
|
if (STREAM_REFRESH_EVENT == (uint32_t)-1) {
|
|
LOG_ERROR("Failed to register custom SDL event");
|
|
}
|
|
|
|
LOG_INFO("Screen resolution: [{}x{}]", screen_width_, screen_height_);
|
|
}
|
|
|
|
void Render::InitializeModules() {
|
|
if (!modules_inited_) {
|
|
AudioDeviceInit();
|
|
screen_capturer_factory_ = new ScreenCapturerFactory();
|
|
speaker_capturer_factory_ = new SpeakerCapturerFactory();
|
|
device_controller_factory_ = new DeviceControllerFactory();
|
|
keyboard_capturer_ = (KeyboardCapturer*)device_controller_factory_->Create(
|
|
DeviceControllerFactory::Device::Keyboard);
|
|
CreateConnectionPeer();
|
|
|
|
// start clipboard monitoring with callback to send data to peers
|
|
Clipboard::StartMonitoring(
|
|
100, [this](const char* data, size_t size) -> int {
|
|
// send clipboard data to all connected peers
|
|
std::shared_lock lock(client_properties_mutex_);
|
|
int ret = -1;
|
|
for (const auto& [remote_id, props] : client_properties_) {
|
|
if (props && props->peer_ && props->connection_established_) {
|
|
ret = SendReliableDataFrame(props->peer_, data, size,
|
|
props->clipboard_label_.c_str());
|
|
if (ret != 0) {
|
|
LOG_WARN("Failed to send clipboard data to peer [{}], ret={}",
|
|
remote_id.c_str(), ret);
|
|
return ret;
|
|
}
|
|
}
|
|
}
|
|
|
|
ret = SendReliableDataFrame(peer_, data, size,
|
|
clipboard_label_.c_str());
|
|
if (ret != 0) {
|
|
LOG_WARN("Failed to send clipboard data to peer [{}], ret={}",
|
|
remote_id_display_, ret);
|
|
return ret;
|
|
}
|
|
|
|
return 0;
|
|
});
|
|
|
|
modules_inited_ = true;
|
|
}
|
|
}
|
|
|
|
void Render::InitializeMainWindow() {
|
|
CreateMainWindow();
|
|
if (SDL_WINDOW_HIDDEN & SDL_GetWindowFlags(main_window_)) {
|
|
SDL_ShowWindow(main_window_);
|
|
}
|
|
}
|
|
|
|
void Render::MainLoop() {
|
|
while (!exit_) {
|
|
if (!peer_) {
|
|
CreateConnectionPeer();
|
|
}
|
|
|
|
SDL_Event event;
|
|
if (SDL_WaitEventTimeout(&event, sdl_refresh_ms_)) {
|
|
ProcessSdlEvent(event);
|
|
}
|
|
|
|
#if _WIN32
|
|
MSG msg;
|
|
while (PeekMessage(&msg, nullptr, 0, 0, PM_REMOVE)) {
|
|
TranslateMessage(&msg);
|
|
DispatchMessage(&msg);
|
|
}
|
|
#endif
|
|
|
|
UpdateLabels();
|
|
HandleRecentConnections();
|
|
HandleStreamWindow();
|
|
HandleServerWindow();
|
|
|
|
DrawMainWindow();
|
|
if (stream_window_inited_) {
|
|
DrawStreamWindow();
|
|
}
|
|
|
|
if (is_server_mode_) {
|
|
DrawServerWindow();
|
|
}
|
|
|
|
UpdateInteractions();
|
|
}
|
|
}
|
|
|
|
void Render::UpdateLabels() {
|
|
if (!label_inited_ ||
|
|
localization_language_index_last_ != localization_language_index_) {
|
|
connect_button_label_ =
|
|
connect_button_pressed_
|
|
? localization::disconnect[localization_language_index_]
|
|
: localization::connect[localization_language_index_];
|
|
label_inited_ = true;
|
|
localization_language_index_last_ = localization_language_index_;
|
|
}
|
|
}
|
|
|
|
void Render::HandleRecentConnections() {
|
|
if (reload_recent_connections_ && main_renderer_) {
|
|
uint32_t now_time = SDL_GetTicks();
|
|
if (now_time - recent_connection_image_save_time_ >= 50) {
|
|
int ret = thumbnail_->LoadThumbnail(main_renderer_, recent_connections_,
|
|
&recent_connection_image_width_,
|
|
&recent_connection_image_height_);
|
|
if (!ret) {
|
|
LOG_INFO("Load recent connection thumbnails");
|
|
}
|
|
reload_recent_connections_ = false;
|
|
}
|
|
}
|
|
}
|
|
|
|
void Render::HandleStreamWindow() {
|
|
if (need_to_create_stream_window_) {
|
|
CreateStreamWindow();
|
|
need_to_create_stream_window_ = false;
|
|
}
|
|
|
|
if (stream_window_inited_) {
|
|
if (!stream_window_grabbed_ && control_mouse_) {
|
|
SDL_SetWindowMouseGrab(stream_window_, true);
|
|
stream_window_grabbed_ = true;
|
|
} else if (stream_window_grabbed_ && !control_mouse_) {
|
|
SDL_SetWindowMouseGrab(stream_window_, false);
|
|
stream_window_grabbed_ = false;
|
|
}
|
|
}
|
|
}
|
|
|
|
void Render::HandleServerWindow() {
|
|
if (need_to_create_server_window_) {
|
|
CreateServerWindow();
|
|
need_to_create_server_window_ = false;
|
|
}
|
|
|
|
if (need_to_destroy_server_window_) {
|
|
DestroyServerWindow();
|
|
need_to_destroy_server_window_ = false;
|
|
}
|
|
}
|
|
|
|
void Render::Cleanup() {
|
|
Clipboard::StopMonitoring();
|
|
|
|
if (screen_capturer_) {
|
|
screen_capturer_->Destroy();
|
|
delete screen_capturer_;
|
|
screen_capturer_ = nullptr;
|
|
}
|
|
|
|
if (speaker_capturer_) {
|
|
speaker_capturer_->Destroy();
|
|
delete speaker_capturer_;
|
|
speaker_capturer_ = nullptr;
|
|
}
|
|
|
|
if (mouse_controller_) {
|
|
mouse_controller_->Destroy();
|
|
delete mouse_controller_;
|
|
mouse_controller_ = nullptr;
|
|
}
|
|
|
|
if (keyboard_capturer_) {
|
|
delete keyboard_capturer_;
|
|
keyboard_capturer_ = nullptr;
|
|
}
|
|
|
|
CleanupFactories();
|
|
CleanupPeers();
|
|
|
|
WaitForThumbnailSaveTasks();
|
|
|
|
AudioDeviceDestroy();
|
|
DestroyMainWindowContext();
|
|
DestroyMainWindow();
|
|
SDL_Quit();
|
|
}
|
|
|
|
void Render::CleanupFactories() {
|
|
if (screen_capturer_factory_) {
|
|
delete screen_capturer_factory_;
|
|
screen_capturer_factory_ = nullptr;
|
|
}
|
|
|
|
if (speaker_capturer_factory_) {
|
|
delete speaker_capturer_factory_;
|
|
speaker_capturer_factory_ = nullptr;
|
|
}
|
|
|
|
if (device_controller_factory_) {
|
|
delete device_controller_factory_;
|
|
device_controller_factory_ = nullptr;
|
|
}
|
|
}
|
|
|
|
void Render::CleanupPeer(std::shared_ptr<SubStreamWindowProperties> props) {
|
|
SDL_FlushEvent(STREAM_REFRESH_EVENT);
|
|
|
|
std::shared_ptr<std::vector<unsigned char>> frame_snapshot;
|
|
int video_width = 0;
|
|
int video_height = 0;
|
|
{
|
|
std::lock_guard<std::mutex> lock(props->video_frame_mutex_);
|
|
frame_snapshot = props->front_frame_;
|
|
video_width = props->video_width_;
|
|
video_height = props->video_height_;
|
|
}
|
|
|
|
if (frame_snapshot && !frame_snapshot->empty() && video_width > 0 &&
|
|
video_height > 0) {
|
|
std::vector<unsigned char> buffer_copy(*frame_snapshot);
|
|
std::string remote_id = props->remote_id_;
|
|
std::string remote_host_name = props->remote_host_name_;
|
|
std::string password =
|
|
props->remember_password_ ? props->remote_password_ : "";
|
|
|
|
std::thread save_thread([buffer_copy, video_width, video_height, remote_id,
|
|
remote_host_name, password,
|
|
thumbnail = thumbnail_]() {
|
|
thumbnail->SaveToThumbnail((char*)buffer_copy.data(), video_width,
|
|
video_height, remote_id, remote_host_name,
|
|
password);
|
|
});
|
|
|
|
{
|
|
std::lock_guard<std::mutex> lock(thumbnail_save_threads_mutex_);
|
|
thumbnail_save_threads_.emplace_back(std::move(save_thread));
|
|
}
|
|
}
|
|
|
|
if (props->peer_) {
|
|
LOG_INFO("[{}] Leave connection [{}]", props->local_id_, props->remote_id_);
|
|
LeaveConnection(props->peer_, props->remote_id_.c_str());
|
|
LOG_INFO("Destroy peer [{}]", props->local_id_);
|
|
DestroyPeer(&props->peer_);
|
|
}
|
|
}
|
|
|
|
void Render::CleanupPeers() {
|
|
if (peer_) {
|
|
LOG_INFO("[{}] Leave connection [{}]", client_id_, client_id_);
|
|
LeaveConnection(peer_, client_id_);
|
|
is_client_mode_ = false;
|
|
StopScreenCapturer();
|
|
StopSpeakerCapturer();
|
|
StopMouseController();
|
|
StopKeyboardCapturer();
|
|
LOG_INFO("Destroy peer [{}]", client_id_);
|
|
DestroyPeer(&peer_);
|
|
}
|
|
|
|
{
|
|
// std::shared_lock lock(client_properties_mutex_);
|
|
for (auto& it : client_properties_) {
|
|
auto props = it.second;
|
|
CleanupPeer(props);
|
|
}
|
|
}
|
|
|
|
{
|
|
// std::unique_lock lock(client_properties_mutex_);
|
|
client_properties_.clear();
|
|
}
|
|
}
|
|
|
|
void Render::WaitForThumbnailSaveTasks() {
|
|
std::vector<std::thread> threads_to_join;
|
|
|
|
{
|
|
std::lock_guard<std::mutex> lock(thumbnail_save_threads_mutex_);
|
|
threads_to_join.swap(thumbnail_save_threads_);
|
|
}
|
|
|
|
if (threads_to_join.empty()) {
|
|
return;
|
|
}
|
|
|
|
for (auto& thread : threads_to_join) {
|
|
if (thread.joinable()) {
|
|
thread.join();
|
|
}
|
|
}
|
|
}
|
|
|
|
void Render::CleanSubStreamWindowProperties(
|
|
std::shared_ptr<SubStreamWindowProperties> props) {
|
|
if (props->stream_texture_) {
|
|
SDL_DestroyTexture(props->stream_texture_);
|
|
props->stream_texture_ = nullptr;
|
|
}
|
|
|
|
{
|
|
std::lock_guard<std::mutex> lock(props->video_frame_mutex_);
|
|
props->front_frame_.reset();
|
|
props->back_frame_.reset();
|
|
props->video_width_ = 0;
|
|
props->video_height_ = 0;
|
|
props->video_size_ = 0;
|
|
props->render_rect_dirty_ = true;
|
|
props->stream_cleanup_pending_ = false;
|
|
}
|
|
}
|
|
|
|
std::shared_ptr<Render::SubStreamWindowProperties>
|
|
Render::GetSubStreamWindowPropertiesByRemoteId(const std::string& remote_id) {
|
|
if (remote_id.empty()) {
|
|
return nullptr;
|
|
}
|
|
|
|
std::shared_lock lock(client_properties_mutex_);
|
|
auto it = client_properties_.find(remote_id);
|
|
if (it == client_properties_.end()) {
|
|
return nullptr;
|
|
}
|
|
return it->second;
|
|
}
|
|
|
|
void Render::StartFileTransfer(std::shared_ptr<SubStreamWindowProperties> props,
|
|
const std::filesystem::path& file_path,
|
|
const std::string& file_label,
|
|
const std::string& remote_id) {
|
|
const bool is_global = (props == nullptr);
|
|
PeerPtr* peer = is_global ? peer_ : props->peer_;
|
|
if (!peer) {
|
|
LOG_ERROR("StartFileTransfer: invalid peer");
|
|
return;
|
|
}
|
|
|
|
bool expected = false;
|
|
if (!(is_global ? file_transfer_.file_sending_
|
|
: props->file_transfer_.file_sending_)
|
|
.compare_exchange_strong(expected, true)) {
|
|
// Already sending, this should not happen if called correctly
|
|
LOG_WARN(
|
|
"StartFileTransfer called but file_sending_ is already true, "
|
|
"file should have been queued: {}",
|
|
file_path.filename().string().c_str());
|
|
return;
|
|
}
|
|
|
|
auto props_weak = std::weak_ptr<SubStreamWindowProperties>(props);
|
|
Render* render_ptr = this;
|
|
|
|
std::thread([peer, file_path, file_label, props_weak, render_ptr, remote_id,
|
|
is_global]() {
|
|
auto props_locked = props_weak.lock();
|
|
|
|
FileTransferState* state = nullptr;
|
|
if (props_locked) {
|
|
state = &props_locked->file_transfer_;
|
|
} else if (is_global) {
|
|
state = &render_ptr->file_transfer_;
|
|
} else {
|
|
return;
|
|
}
|
|
|
|
std::error_code ec;
|
|
uint64_t total_size = std::filesystem::file_size(file_path, ec);
|
|
if (ec) {
|
|
LOG_ERROR("Failed to get file size: {}", ec.message().c_str());
|
|
state->file_sending_ = false;
|
|
return;
|
|
}
|
|
|
|
state->file_sent_bytes_ = 0;
|
|
state->file_total_bytes_ = total_size;
|
|
state->file_send_rate_bps_ = 0;
|
|
state->file_transfer_window_visible_ = true;
|
|
{
|
|
std::lock_guard<std::mutex> lock(state->file_transfer_mutex_);
|
|
state->file_send_start_time_ = std::chrono::steady_clock::now();
|
|
state->file_send_last_update_time_ = state->file_send_start_time_;
|
|
state->file_send_last_bytes_ = 0;
|
|
}
|
|
|
|
LOG_INFO(
|
|
"File transfer started: {} ({} bytes), file_sending_={}, "
|
|
"total_bytes_={}",
|
|
file_path.filename().string(), total_size, state->file_sending_.load(),
|
|
state->file_total_bytes_.load());
|
|
|
|
FileSender sender;
|
|
uint32_t file_id = FileSender::NextFileId();
|
|
|
|
if (props_locked) {
|
|
std::lock_guard<std::shared_mutex> lock(
|
|
render_ptr->file_id_to_props_mutex_);
|
|
render_ptr->file_id_to_props_[file_id] = props_weak;
|
|
} else {
|
|
std::lock_guard<std::shared_mutex> lock(
|
|
render_ptr->file_id_to_transfer_state_mutex_);
|
|
render_ptr->file_id_to_transfer_state_[file_id] = state;
|
|
}
|
|
|
|
state->current_file_id_ = file_id;
|
|
|
|
// Update file transfer list: mark as sending
|
|
// Find the queued file that matches the exact file path
|
|
{
|
|
std::lock_guard<std::mutex> lock(state->file_transfer_list_mutex_);
|
|
for (auto& info : state->file_transfer_list_) {
|
|
if (info.file_path == file_path &&
|
|
info.status == FileTransferState::FileTransferStatus::Queued) {
|
|
info.status = FileTransferState::FileTransferStatus::Sending;
|
|
info.file_id = file_id;
|
|
info.file_size = total_size;
|
|
info.sent_bytes = 0;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
state->file_transfer_window_visible_ = true;
|
|
|
|
// Progress will be updated via ACK from receiver
|
|
int ret = sender.SendFile(
|
|
file_path, file_path.filename().string(),
|
|
[peer, file_label, remote_id](const char* buf, size_t sz) -> int {
|
|
if (remote_id.empty()) {
|
|
return SendReliableDataFrame(peer, buf, sz, file_label.c_str());
|
|
} else {
|
|
return SendReliableDataFrameToPeer(
|
|
peer, buf, sz, file_label.c_str(), remote_id.c_str(),
|
|
remote_id.size());
|
|
}
|
|
},
|
|
64 * 1024, file_id);
|
|
|
|
// file_sending_ should remain true until we receive the final ACK from
|
|
// receiver
|
|
// On error, set file_sending_ to false immediately to allow next file
|
|
if (ret != 0) {
|
|
state->file_sending_ = false;
|
|
state->file_transfer_window_visible_ = false;
|
|
state->file_sent_bytes_ = 0;
|
|
state->file_total_bytes_ = 0;
|
|
state->file_send_rate_bps_ = 0;
|
|
state->current_file_id_ = 0;
|
|
|
|
// Unregister file_id mapping on error
|
|
if (props_locked) {
|
|
std::lock_guard<std::shared_mutex> lock(
|
|
render_ptr->file_id_to_props_mutex_);
|
|
render_ptr->file_id_to_props_.erase(file_id);
|
|
} else {
|
|
std::lock_guard<std::shared_mutex> lock(
|
|
render_ptr->file_id_to_transfer_state_mutex_);
|
|
render_ptr->file_id_to_transfer_state_.erase(file_id);
|
|
}
|
|
|
|
// Update file transfer list: mark as failed
|
|
{
|
|
std::lock_guard<std::mutex> lock(state->file_transfer_list_mutex_);
|
|
for (auto& info : state->file_transfer_list_) {
|
|
if (info.file_id == file_id) {
|
|
info.status = FileTransferState::FileTransferStatus::Failed;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
LOG_ERROR("FileSender::SendFile failed for [{}], ret={}",
|
|
file_path.string().c_str(), ret);
|
|
|
|
render_ptr->ProcessFileQueue(props_locked);
|
|
}
|
|
}).detach();
|
|
}
|
|
|
|
void Render::ProcessFileQueue(
|
|
std::shared_ptr<SubStreamWindowProperties> props) {
|
|
FileTransferState* state = props ? &props->file_transfer_ : &file_transfer_;
|
|
if (!state) {
|
|
return;
|
|
}
|
|
|
|
if (state->file_sending_.load()) {
|
|
return;
|
|
}
|
|
|
|
FileTransferState::QueuedFile queued_file;
|
|
{
|
|
std::lock_guard<std::mutex> lock(state->file_queue_mutex_);
|
|
if (state->file_send_queue_.empty()) {
|
|
return;
|
|
}
|
|
queued_file = state->file_send_queue_.front();
|
|
state->file_send_queue_.pop();
|
|
}
|
|
|
|
LOG_INFO("Processing next file in queue: {}",
|
|
queued_file.file_path.string().c_str());
|
|
StartFileTransfer(props, queued_file.file_path, queued_file.file_label,
|
|
queued_file.remote_id);
|
|
}
|
|
|
|
void Render::UpdateRenderRect() {
|
|
// std::shared_lock lock(client_properties_mutex_);
|
|
for (auto& [_, props] : client_properties_) {
|
|
if (!props->reset_control_bar_pos_) {
|
|
props->mouse_diff_control_bar_pos_x_ = 0;
|
|
props->mouse_diff_control_bar_pos_y_ = 0;
|
|
}
|
|
|
|
if (!just_created_) {
|
|
props->reset_control_bar_pos_ = true;
|
|
}
|
|
|
|
int stream_window_width, stream_window_height;
|
|
SDL_GetWindowSize(stream_window_, &stream_window_width,
|
|
&stream_window_height);
|
|
stream_window_width_ = (float)stream_window_width;
|
|
stream_window_height_ = (float)stream_window_height;
|
|
|
|
float video_ratio =
|
|
(float)props->video_width_ / (float)props->video_height_;
|
|
float video_ratio_reverse =
|
|
(float)props->video_height_ / (float)props->video_width_;
|
|
|
|
float render_area_width = props->render_window_width_;
|
|
float render_area_height = props->render_window_height_;
|
|
|
|
props->stream_render_rect_last_ = props->stream_render_rect_;
|
|
if (render_area_width < render_area_height * video_ratio) {
|
|
props->stream_render_rect_ = {
|
|
(int)props->render_window_x_,
|
|
(int)(abs(render_area_height -
|
|
render_area_width * video_ratio_reverse) /
|
|
2 +
|
|
(int)props->render_window_y_),
|
|
(int)render_area_width,
|
|
(int)(render_area_width * video_ratio_reverse)};
|
|
} else if (render_area_width > render_area_height * video_ratio) {
|
|
props->stream_render_rect_ = {
|
|
(int)abs(render_area_width - render_area_height * video_ratio) / 2,
|
|
(int)props->render_window_y_, (int)(render_area_height * video_ratio),
|
|
(int)render_area_height};
|
|
} else {
|
|
props->stream_render_rect_ = {
|
|
(int)props->render_window_x_, (int)props->render_window_y_,
|
|
(int)render_area_width, (int)render_area_height};
|
|
}
|
|
}
|
|
}
|
|
|
|
void Render::ProcessSdlEvent(const SDL_Event& event) {
|
|
if (main_ctx_) {
|
|
ImGui::SetCurrentContext(main_ctx_);
|
|
ImGui_ImplSDL3_ProcessEvent(&event);
|
|
} else {
|
|
LOG_ERROR("Main context is null");
|
|
return;
|
|
}
|
|
|
|
if (stream_window_inited_) {
|
|
if (stream_ctx_) {
|
|
ImGui::SetCurrentContext(stream_ctx_);
|
|
ImGui_ImplSDL3_ProcessEvent(&event);
|
|
} else {
|
|
LOG_ERROR("Stream context is null");
|
|
return;
|
|
}
|
|
}
|
|
|
|
if (server_window_inited_) {
|
|
if (server_ctx_) {
|
|
ImGui::SetCurrentContext(server_ctx_);
|
|
ImGui_ImplSDL3_ProcessEvent(&event);
|
|
} else {
|
|
LOG_ERROR("Server context is null");
|
|
return;
|
|
}
|
|
}
|
|
|
|
switch (event.type) {
|
|
case SDL_EVENT_QUIT:
|
|
if (stream_window_inited_) {
|
|
LOG_INFO("Destroy stream window");
|
|
SDL_SetWindowMouseGrab(stream_window_, false);
|
|
DestroyStreamWindow();
|
|
DestroyStreamWindowContext();
|
|
|
|
{
|
|
// std::shared_lock lock(client_properties_mutex_);
|
|
for (auto& [host_name, props] : client_properties_) {
|
|
std::shared_ptr<std::vector<unsigned char>> frame_snapshot;
|
|
int video_width = 0;
|
|
int video_height = 0;
|
|
{
|
|
std::lock_guard<std::mutex> lock(props->video_frame_mutex_);
|
|
frame_snapshot = props->front_frame_;
|
|
video_width = props->video_width_;
|
|
video_height = props->video_height_;
|
|
}
|
|
if (frame_snapshot && !frame_snapshot->empty() && video_width > 0 &&
|
|
video_height > 0) {
|
|
thumbnail_->SaveToThumbnail(
|
|
(char*)frame_snapshot->data(), video_width, video_height,
|
|
host_name, props->remote_host_name_,
|
|
props->remember_password_ ? props->remote_password_ : "");
|
|
}
|
|
|
|
if (props->peer_) {
|
|
std::string client_id = (host_name == client_id_)
|
|
? "C-" + std::string(client_id_)
|
|
: client_id_;
|
|
LOG_INFO("[{}] Leave connection [{}]", client_id, host_name);
|
|
LeaveConnection(props->peer_, host_name.c_str());
|
|
LOG_INFO("Destroy peer [{}]", client_id);
|
|
DestroyPeer(&props->peer_);
|
|
}
|
|
|
|
props->streaming_ = false;
|
|
props->remember_password_ = false;
|
|
props->connection_established_ = false;
|
|
props->audio_capture_button_pressed_ = false;
|
|
|
|
memset(&props->net_traffic_stats_, 0,
|
|
sizeof(props->net_traffic_stats_));
|
|
SDL_SetWindowFullscreen(main_window_, false);
|
|
SDL_FlushEvents(STREAM_REFRESH_EVENT, STREAM_REFRESH_EVENT);
|
|
memset(audio_buffer_, 0, 720);
|
|
}
|
|
}
|
|
|
|
{
|
|
// std::unique_lock lock(client_properties_mutex_);
|
|
client_properties_.clear();
|
|
}
|
|
|
|
rejoin_ = false;
|
|
is_client_mode_ = false;
|
|
reload_recent_connections_ = true;
|
|
fullscreen_button_pressed_ = false;
|
|
start_keyboard_capturer_ = false;
|
|
just_created_ = false;
|
|
recent_connection_image_save_time_ = SDL_GetTicks();
|
|
} else {
|
|
LOG_INFO("Quit program");
|
|
exit_ = true;
|
|
}
|
|
break;
|
|
|
|
case SDL_EVENT_WINDOW_CLOSE_REQUESTED:
|
|
if (event.window.windowID != SDL_GetWindowID(stream_window_)) {
|
|
exit_ = true;
|
|
}
|
|
break;
|
|
|
|
case SDL_EVENT_WINDOW_PIXEL_SIZE_CHANGED:
|
|
if (stream_window_created_ &&
|
|
event.window.windowID == SDL_GetWindowID(stream_window_)) {
|
|
UpdateRenderRect();
|
|
}
|
|
break;
|
|
|
|
case SDL_EVENT_WINDOW_FOCUS_GAINED:
|
|
if (stream_window_ &&
|
|
SDL_GetWindowID(stream_window_) == event.window.windowID) {
|
|
foucs_on_stream_window_ = true;
|
|
} else if (main_window_ &&
|
|
SDL_GetWindowID(main_window_) == event.window.windowID) {
|
|
foucs_on_main_window_ = true;
|
|
}
|
|
break;
|
|
|
|
case SDL_EVENT_WINDOW_FOCUS_LOST:
|
|
if (stream_window_ &&
|
|
SDL_GetWindowID(stream_window_) == event.window.windowID) {
|
|
foucs_on_stream_window_ = false;
|
|
} else if (main_window_ &&
|
|
SDL_GetWindowID(main_window_) == event.window.windowID) {
|
|
foucs_on_main_window_ = false;
|
|
}
|
|
break;
|
|
case SDL_EVENT_DROP_FILE:
|
|
ProcessFileDropEvent(event);
|
|
break;
|
|
|
|
case SDL_EVENT_MOUSE_MOTION:
|
|
case SDL_EVENT_MOUSE_BUTTON_DOWN:
|
|
case SDL_EVENT_MOUSE_BUTTON_UP:
|
|
case SDL_EVENT_MOUSE_WHEEL:
|
|
if (foucs_on_stream_window_) {
|
|
ProcessMouseEvent(event);
|
|
}
|
|
break;
|
|
|
|
default:
|
|
if (event.type == STREAM_REFRESH_EVENT) {
|
|
auto* props = static_cast<SubStreamWindowProperties*>(event.user.data1);
|
|
if (!props) {
|
|
break;
|
|
}
|
|
std::shared_ptr<std::vector<unsigned char>> frame_snapshot;
|
|
int video_width = 0;
|
|
int video_height = 0;
|
|
bool render_rect_dirty = false;
|
|
bool cleanup_pending = false;
|
|
{
|
|
std::lock_guard<std::mutex> lock(props->video_frame_mutex_);
|
|
cleanup_pending = props->stream_cleanup_pending_;
|
|
if (!cleanup_pending) {
|
|
frame_snapshot = props->front_frame_;
|
|
video_width = props->video_width_;
|
|
video_height = props->video_height_;
|
|
}
|
|
render_rect_dirty = props->render_rect_dirty_;
|
|
}
|
|
|
|
if (cleanup_pending) {
|
|
if (props->stream_texture_) {
|
|
SDL_DestroyTexture(props->stream_texture_);
|
|
props->stream_texture_ = nullptr;
|
|
}
|
|
{
|
|
std::lock_guard<std::mutex> lock(props->video_frame_mutex_);
|
|
props->stream_cleanup_pending_ = false;
|
|
}
|
|
|
|
if (render_rect_dirty) {
|
|
UpdateRenderRect();
|
|
std::lock_guard<std::mutex> lock(props->video_frame_mutex_);
|
|
props->render_rect_dirty_ = false;
|
|
}
|
|
break;
|
|
}
|
|
|
|
if (video_width <= 0 || video_height <= 0) {
|
|
break;
|
|
}
|
|
if (!frame_snapshot || frame_snapshot->empty()) {
|
|
break;
|
|
}
|
|
|
|
if (props->stream_texture_) {
|
|
if (video_width != props->texture_width_ ||
|
|
video_height != props->texture_height_) {
|
|
props->texture_width_ = video_width;
|
|
props->texture_height_ = video_height;
|
|
|
|
SDL_DestroyTexture(props->stream_texture_);
|
|
// props->stream_texture_ = SDL_CreateTexture(
|
|
// stream_renderer_, stream_pixformat_,
|
|
// SDL_TEXTUREACCESS_STREAMING, props->texture_width_,
|
|
// props->texture_height_);
|
|
|
|
SDL_PropertiesID nvProps = SDL_CreateProperties();
|
|
SDL_SetNumberProperty(nvProps, SDL_PROP_TEXTURE_CREATE_WIDTH_NUMBER,
|
|
props->texture_width_);
|
|
SDL_SetNumberProperty(nvProps,
|
|
SDL_PROP_TEXTURE_CREATE_HEIGHT_NUMBER,
|
|
props->texture_height_);
|
|
SDL_SetNumberProperty(nvProps,
|
|
SDL_PROP_TEXTURE_CREATE_FORMAT_NUMBER,
|
|
SDL_PIXELFORMAT_NV12);
|
|
SDL_SetNumberProperty(nvProps,
|
|
SDL_PROP_TEXTURE_CREATE_COLORSPACE_NUMBER,
|
|
SDL_COLORSPACE_BT601_LIMITED);
|
|
props->stream_texture_ =
|
|
SDL_CreateTextureWithProperties(stream_renderer_, nvProps);
|
|
SDL_DestroyProperties(nvProps);
|
|
}
|
|
} else {
|
|
props->texture_width_ = video_width;
|
|
props->texture_height_ = video_height;
|
|
// props->stream_texture_ = SDL_CreateTexture(
|
|
// stream_renderer_, stream_pixformat_,
|
|
// SDL_TEXTUREACCESS_STREAMING, props->texture_width_,
|
|
// props->texture_height_);
|
|
|
|
SDL_PropertiesID nvProps = SDL_CreateProperties();
|
|
SDL_SetNumberProperty(nvProps, SDL_PROP_TEXTURE_CREATE_WIDTH_NUMBER,
|
|
props->texture_width_);
|
|
SDL_SetNumberProperty(nvProps, SDL_PROP_TEXTURE_CREATE_HEIGHT_NUMBER,
|
|
props->texture_height_);
|
|
SDL_SetNumberProperty(nvProps, SDL_PROP_TEXTURE_CREATE_FORMAT_NUMBER,
|
|
SDL_PIXELFORMAT_NV12);
|
|
SDL_SetNumberProperty(nvProps,
|
|
SDL_PROP_TEXTURE_CREATE_COLORSPACE_NUMBER,
|
|
SDL_COLORSPACE_BT601_LIMITED);
|
|
props->stream_texture_ =
|
|
SDL_CreateTextureWithProperties(stream_renderer_, nvProps);
|
|
SDL_DestroyProperties(nvProps);
|
|
}
|
|
|
|
SDL_UpdateTexture(props->stream_texture_, NULL, frame_snapshot->data(),
|
|
props->texture_width_);
|
|
|
|
if (render_rect_dirty) {
|
|
UpdateRenderRect();
|
|
std::lock_guard<std::mutex> lock(props->video_frame_mutex_);
|
|
props->render_rect_dirty_ = false;
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
|
|
void Render::ProcessFileDropEvent(const SDL_Event& event) {
|
|
if (!((stream_window_ &&
|
|
SDL_GetWindowID(stream_window_) == event.window.windowID) ||
|
|
(server_window_ &&
|
|
SDL_GetWindowID(server_window_) == event.window.windowID))) {
|
|
return;
|
|
}
|
|
|
|
if (event.type != SDL_EVENT_DROP_FILE) {
|
|
return;
|
|
}
|
|
|
|
if (SDL_GetWindowID(stream_window_) == event.window.windowID) {
|
|
if (!stream_window_inited_) {
|
|
return;
|
|
}
|
|
|
|
std::shared_lock lock(client_properties_mutex_);
|
|
for (auto& [_, props] : client_properties_) {
|
|
if (props->tab_selected_) {
|
|
if (event.drop.data == nullptr) {
|
|
LOG_ERROR("ProcessFileDropEvent: drop event data is null");
|
|
break;
|
|
}
|
|
|
|
if (!props || !props->peer_) {
|
|
LOG_ERROR("ProcessFileDropEvent: invalid props or peer");
|
|
break;
|
|
}
|
|
|
|
std::string utf8_path = static_cast<const char*>(event.drop.data);
|
|
std::filesystem::path file_path = std::filesystem::u8path(utf8_path);
|
|
|
|
// Check if file exists
|
|
std::error_code ec;
|
|
if (!std::filesystem::exists(file_path, ec)) {
|
|
LOG_ERROR("ProcessFileDropEvent: file does not exist: {}",
|
|
file_path.string().c_str());
|
|
break;
|
|
}
|
|
|
|
// Check if it's a regular file
|
|
if (!std::filesystem::is_regular_file(file_path, ec)) {
|
|
LOG_ERROR("ProcessFileDropEvent: path is not a regular file: {}",
|
|
file_path.string().c_str());
|
|
break;
|
|
}
|
|
|
|
// Get file size
|
|
uint64_t file_size = std::filesystem::file_size(file_path, ec);
|
|
if (ec) {
|
|
LOG_ERROR("ProcessFileDropEvent: failed to get file size: {}",
|
|
ec.message().c_str());
|
|
break;
|
|
}
|
|
|
|
LOG_INFO("Drop file [{}] to send (size: {} bytes)", event.drop.data,
|
|
file_size);
|
|
|
|
// Use ProcessSelectedFile to handle the file processing
|
|
ProcessSelectedFile(utf8_path, props, props->file_label_);
|
|
|
|
break;
|
|
}
|
|
}
|
|
} else if (SDL_GetWindowID(server_window_) == event.window.windowID) {
|
|
if (!server_window_inited_) {
|
|
return;
|
|
}
|
|
|
|
if (event.drop.data == nullptr) {
|
|
LOG_ERROR("ProcessFileDropEvent: drop event data is null");
|
|
return;
|
|
}
|
|
|
|
std::string utf8_path = static_cast<const char*>(event.drop.data);
|
|
std::filesystem::path file_path = std::filesystem::u8path(utf8_path);
|
|
|
|
// Check if file exists
|
|
std::error_code ec;
|
|
if (!std::filesystem::exists(file_path, ec)) {
|
|
LOG_ERROR("ProcessFileDropEvent: file does not exist: {}",
|
|
file_path.string().c_str());
|
|
return;
|
|
}
|
|
|
|
// Check if it's a regular file
|
|
if (!std::filesystem::is_regular_file(file_path, ec)) {
|
|
LOG_ERROR("ProcessFileDropEvent: path is not a regular file: {}",
|
|
file_path.string().c_str());
|
|
return;
|
|
}
|
|
|
|
// Get file size
|
|
uint64_t file_size = std::filesystem::file_size(file_path, ec);
|
|
if (ec) {
|
|
LOG_ERROR("ProcessFileDropEvent: failed to get file size: {}",
|
|
ec.message().c_str());
|
|
return;
|
|
}
|
|
|
|
LOG_INFO("Drop file [{}] on server window (size: {} bytes)",
|
|
event.drop.data, file_size);
|
|
|
|
// Handle the dropped file on server window as needed
|
|
}
|
|
}
|
|
} // namespace crossdesk
|