mirror of
https://github.com/rustdesk/rustdesk.git
synced 2024-12-16 12:39:06 +08:00
127 lines
4.2 KiB
C++
127 lines
4.2 KiB
C++
#include <flutter/dart_project.h>
|
|
#include <flutter/flutter_view_controller.h>
|
|
#include <tchar.h>
|
|
#include <uni_links_desktop/uni_links_desktop_plugin.h>
|
|
#include <windows.h>
|
|
|
|
#include <algorithm>
|
|
#include <iostream>
|
|
|
|
#include "flutter_window.h"
|
|
#include "utils.h"
|
|
|
|
typedef char** (*FUNC_RUSTDESK_CORE_MAIN)(int*);
|
|
typedef void (*FUNC_RUSTDESK_FREE_ARGS)( char**, int);
|
|
const char* uniLinksPrefix = "rustdesk://";
|
|
/// Note: `--server`, `--service` are already handled in [core_main.rs].
|
|
const std::vector<std::string> parameters_white_list = {"--install", "--cm"};
|
|
|
|
int APIENTRY wWinMain(_In_ HINSTANCE instance, _In_opt_ HINSTANCE prev,
|
|
_In_ wchar_t *command_line, _In_ int show_command)
|
|
{
|
|
HINSTANCE hInstance = LoadLibraryA("librustdesk.dll");
|
|
if (!hInstance)
|
|
{
|
|
std::cout << "Failed to load librustdesk.dll" << std::endl;
|
|
return EXIT_FAILURE;
|
|
}
|
|
FUNC_RUSTDESK_CORE_MAIN rustdesk_core_main =
|
|
(FUNC_RUSTDESK_CORE_MAIN)GetProcAddress(hInstance, "rustdesk_core_main_args");
|
|
if (!rustdesk_core_main)
|
|
{
|
|
std::cout << "Failed to get rustdesk_core_main" << std::endl;
|
|
return EXIT_FAILURE;
|
|
}
|
|
FUNC_RUSTDESK_FREE_ARGS free_c_args =
|
|
(FUNC_RUSTDESK_FREE_ARGS)GetProcAddress(hInstance, "free_c_args");
|
|
if (!free_c_args)
|
|
{
|
|
std::cout << "Failed to get free_c_args" << std::endl;
|
|
return EXIT_FAILURE;
|
|
}
|
|
std::vector<std::string> command_line_arguments =
|
|
GetCommandLineArguments();
|
|
// Remove possible trailing whitespace from command line arguments
|
|
for (auto& argument : command_line_arguments) {
|
|
argument.erase(argument.find_last_not_of(" \n\r\t"));
|
|
}
|
|
|
|
int args_len = 0;
|
|
char** c_args = rustdesk_core_main(&args_len);
|
|
if (!c_args)
|
|
{
|
|
std::cout << "Rustdesk core returns false, exiting without launching Flutter app" << std::endl;
|
|
return EXIT_SUCCESS;
|
|
}
|
|
std::vector<std::string> rust_args(c_args, c_args + args_len);
|
|
free_c_args(c_args, args_len);
|
|
|
|
// Uri links dispatch
|
|
HWND hwnd = ::FindWindow(_T("FLUTTER_RUNNER_WIN32_WINDOW"), _T("RustDesk"));
|
|
if (hwnd != NULL) {
|
|
// Allow multiple flutter instances when being executed by parameters
|
|
// contained in whitelists.
|
|
bool allow_multiple_instances = false;
|
|
for (auto& whitelist_param : parameters_white_list) {
|
|
allow_multiple_instances =
|
|
allow_multiple_instances ||
|
|
std::find(command_line_arguments.begin(),
|
|
command_line_arguments.end(),
|
|
whitelist_param) != command_line_arguments.end();
|
|
}
|
|
if (!allow_multiple_instances) {
|
|
if (!command_line_arguments.empty()) {
|
|
// Dispatch command line arguments
|
|
DispatchToUniLinksDesktop(hwnd);
|
|
} else {
|
|
// Not called with arguments, or just open the app shortcut on desktop.
|
|
// So we just show the main window instead.
|
|
::ShowWindow(hwnd, SW_NORMAL);
|
|
::SetForegroundWindow(hwnd);
|
|
}
|
|
return EXIT_FAILURE;
|
|
}
|
|
}
|
|
|
|
// Attach to console when present (e.g., 'flutter run') or create a
|
|
// new console when running with a debugger.
|
|
if (!::AttachConsole(ATTACH_PARENT_PROCESS) && ::IsDebuggerPresent())
|
|
{
|
|
CreateAndAttachConsole();
|
|
}
|
|
|
|
// Initialize COM, so that it is available for use in the library and/or
|
|
// plugins.
|
|
::CoInitializeEx(nullptr, COINIT_APARTMENTTHREADED);
|
|
|
|
flutter::DartProject project(L"data");
|
|
// connection manager hide icon from taskbar
|
|
bool is_cm_page = false;
|
|
auto cmParam = std::string("--cm");
|
|
if (!command_line_arguments.empty() && command_line_arguments.front().compare(0, cmParam.size(), cmParam.c_str()) == 0) {
|
|
is_cm_page = true;
|
|
}
|
|
command_line_arguments.insert(command_line_arguments.end(), rust_args.begin(), rust_args.end());
|
|
project.set_dart_entrypoint_arguments(std::move(command_line_arguments));
|
|
|
|
FlutterWindow window(project);
|
|
Win32Window::Point origin(10, 10);
|
|
Win32Window::Size size(800, 600);
|
|
if (!window.CreateAndShow(
|
|
is_cm_page ? L"RustDesk - Connection Manager" : L"RustDesk", origin,
|
|
size, !is_cm_page)) {
|
|
return EXIT_FAILURE;
|
|
}
|
|
window.SetQuitOnClose(true);
|
|
|
|
::MSG msg;
|
|
while (::GetMessage(&msg, nullptr, 0, 0))
|
|
{
|
|
::TranslateMessage(&msg);
|
|
::DispatchMessage(&msg);
|
|
}
|
|
|
|
::CoUninitialize();
|
|
return EXIT_SUCCESS;
|
|
}
|