mirror of
https://github.com/kunkundi/crossdesk.git
synced 2026-09-01 06:12:01 +08:00
234 lines
11 KiB
C++
234 lines
11 KiB
C++
#include <cstring>
|
|
#include <cstdint>
|
|
#include <iostream>
|
|
#include <limits>
|
|
#include <string>
|
|
|
|
#include <display_stream_id.h>
|
|
#include <file_transfer_format.h>
|
|
#include <remote_action.h>
|
|
#include <stream_names.h>
|
|
|
|
namespace {
|
|
|
|
bool Expect(bool condition, const char* message) {
|
|
if (condition) return true;
|
|
std::cerr << message << '\n';
|
|
return false;
|
|
}
|
|
|
|
} // namespace
|
|
|
|
int main() {
|
|
bool ok = true;
|
|
|
|
ok &= Expect(static_cast<int>(crossdesk::ControlType::invalid) == -1 &&
|
|
static_cast<int>(crossdesk::ControlType::mouse) == 0 &&
|
|
static_cast<int>(crossdesk::ControlType::keyboard) == 1 &&
|
|
static_cast<int>(crossdesk::ControlType::audio_capture) == 2 &&
|
|
static_cast<int>(crossdesk::ControlType::host_infomation) == 3 &&
|
|
static_cast<int>(crossdesk::ControlType::display_id) == 4 &&
|
|
static_cast<int>(crossdesk::ControlType::service_status) == 5 &&
|
|
static_cast<int>(crossdesk::ControlType::service_command) == 6 &&
|
|
static_cast<int>(crossdesk::ControlType::keyboard_state) == 7 &&
|
|
static_cast<int>(crossdesk::ControlType::cursor_state) == 8,
|
|
"ControlType wire values changed");
|
|
ok &= Expect(static_cast<int>(crossdesk::RemoteCursorShape::default_cursor) ==
|
|
0 &&
|
|
static_cast<int>(crossdesk::RemoteCursorShape::pointer) == 3 &&
|
|
static_cast<int>(crossdesk::RemoteCursorShape::nwse_resize) ==
|
|
28,
|
|
"remote cursor wire values changed");
|
|
ok &= Expect(sizeof(crossdesk::FileChunkHeader) == 31 &&
|
|
sizeof(crossdesk::FileTransferAck) == 28,
|
|
"file transfer wire structure size changed");
|
|
|
|
crossdesk::RemoteAction unset_action{};
|
|
ok &= Expect(unset_action.type == crossdesk::ControlType::invalid,
|
|
"default RemoteAction should be invalid");
|
|
ok &= Expect(unset_action.to_json().empty(),
|
|
"invalid RemoteAction should not be serialized");
|
|
|
|
crossdesk::RemoteAction parsed_invalid{};
|
|
ok &= Expect(!parsed_invalid.from_json("{\"type\":-1}"),
|
|
"invalid RemoteAction JSON should be rejected");
|
|
|
|
crossdesk::RemoteAction unknown_action{};
|
|
unknown_action.type = static_cast<crossdesk::ControlType>(999);
|
|
ok &= Expect(unknown_action.to_json().empty(),
|
|
"unknown RemoteAction type should not be serialized");
|
|
|
|
crossdesk::RemoteAction mouse{};
|
|
mouse.type = crossdesk::ControlType::mouse;
|
|
mouse.m = {0.25f, 0.75f, -1, crossdesk::MouseFlag::wheel_vertical};
|
|
crossdesk::RemoteAction parsed_mouse{};
|
|
ok &= Expect(parsed_mouse.from_json(mouse.to_json()),
|
|
"mouse JSON should decode");
|
|
ok &= Expect(parsed_mouse.type == crossdesk::ControlType::mouse &&
|
|
parsed_mouse.m.x == mouse.m.x &&
|
|
parsed_mouse.m.y == mouse.m.y &&
|
|
parsed_mouse.m.s == mouse.m.s &&
|
|
parsed_mouse.m.flag == mouse.m.flag,
|
|
"mouse JSON round trip changed fields");
|
|
|
|
crossdesk::RemoteAction cursor{};
|
|
cursor.type = crossdesk::ControlType::cursor_state;
|
|
cursor.cs = {42, true, crossdesk::RemoteCursorShape::pointer,
|
|
true, 0.25f, 0.75f, 0.001f, -0.002f, 1, true};
|
|
crossdesk::RemoteAction parsed_cursor{};
|
|
ok &= Expect(parsed_cursor.from_json(cursor.to_json()) &&
|
|
parsed_cursor.type == crossdesk::ControlType::cursor_state &&
|
|
parsed_cursor.cs.seq == cursor.cs.seq &&
|
|
parsed_cursor.cs.visible == cursor.cs.visible &&
|
|
parsed_cursor.cs.shape == cursor.cs.shape &&
|
|
parsed_cursor.cs.position_valid &&
|
|
parsed_cursor.cs.x == cursor.cs.x &&
|
|
parsed_cursor.cs.y == cursor.cs.y &&
|
|
parsed_cursor.cs.visual_offset_x ==
|
|
cursor.cs.visual_offset_x &&
|
|
parsed_cursor.cs.visual_offset_y ==
|
|
cursor.cs.visual_offset_y &&
|
|
parsed_cursor.cs.display_id == cursor.cs.display_id &&
|
|
parsed_cursor.cs.position_update,
|
|
"cursor state JSON round trip changed position fields");
|
|
|
|
crossdesk::RemoteAction legacy_cursor{};
|
|
ok &= Expect(legacy_cursor.from_json(
|
|
"{\"type\":8,\"cursor_state\":{\"seq\":1,"
|
|
"\"visible\":true,\"shape\":0}}") &&
|
|
!legacy_cursor.cs.position_valid &&
|
|
legacy_cursor.cs.position_update &&
|
|
legacy_cursor.cs.display_id == -1,
|
|
"legacy cursor state JSON should remain compatible");
|
|
|
|
crossdesk::RemoteAction shape_only_cursor{};
|
|
shape_only_cursor.type = crossdesk::ControlType::cursor_state;
|
|
shape_only_cursor.cs = cursor.cs;
|
|
shape_only_cursor.cs.position_update = false;
|
|
crossdesk::RemoteAction parsed_shape_only_cursor{};
|
|
ok &= Expect(parsed_shape_only_cursor.from_json(
|
|
shape_only_cursor.to_json()) &&
|
|
!parsed_shape_only_cursor.cs.position_update &&
|
|
parsed_shape_only_cursor.cs.shape == cursor.cs.shape,
|
|
"shape-only cursor state should preserve update semantics");
|
|
|
|
crossdesk::RemoteAction command{};
|
|
command.type = crossdesk::ControlType::service_command;
|
|
command.c.flag = crossdesk::ServiceCommandFlag::send_sas;
|
|
crossdesk::RemoteAction parsed_command{};
|
|
ok &= Expect(parsed_command.from_json(command.to_json()) &&
|
|
parsed_command.type ==
|
|
crossdesk::ControlType::service_command &&
|
|
parsed_command.c.flag ==
|
|
crossdesk::ServiceCommandFlag::send_sas,
|
|
"service command JSON round trip failed");
|
|
|
|
char display_name[] = "Built-in Display";
|
|
char* display_names[] = {display_name};
|
|
int left[] = {0};
|
|
int top[] = {0};
|
|
int right[] = {2560};
|
|
int bottom[] = {1600};
|
|
crossdesk::RemoteAction host{};
|
|
host.type = crossdesk::ControlType::host_infomation;
|
|
std::strcpy(host.i.host_name, "Test Mac");
|
|
host.i.host_name_size = std::strlen(host.i.host_name);
|
|
host.i.display_list = display_names;
|
|
host.i.display_num = 1;
|
|
host.i.left = left;
|
|
host.i.top = top;
|
|
host.i.right = right;
|
|
host.i.bottom = bottom;
|
|
crossdesk::RemoteAction parsed_host{};
|
|
ok &= Expect(parsed_host.from_json(host.to_json()) &&
|
|
parsed_host.i.display_num == 1 &&
|
|
std::string(parsed_host.i.host_name) == "Test Mac" &&
|
|
std::string(parsed_host.i.display_list[0]) ==
|
|
"Built-in Display" &&
|
|
parsed_host.i.right[0] == 2560 &&
|
|
parsed_host.i.bottom[0] == 1600,
|
|
"host information JSON round trip failed");
|
|
crossdesk::FreeRemoteAction(parsed_host);
|
|
|
|
const std::string name = "example.txt";
|
|
const std::string payload = "CrossDesk wire";
|
|
const auto encoded = crossdesk::EncodeFileChunk(
|
|
17, 0, payload.size(), payload.data(),
|
|
static_cast<uint32_t>(payload.size()), &name, true, true);
|
|
crossdesk::FileChunkView chunk;
|
|
ok &= Expect(crossdesk::DecodeFileChunk(
|
|
encoded.data(), encoded.size(), &chunk),
|
|
"file chunk should decode");
|
|
ok &= Expect(chunk.header.file_id == 17 && chunk.file_name == name &&
|
|
chunk.payload_size == payload.size() &&
|
|
std::memcmp(chunk.payload, payload.data(), payload.size()) ==
|
|
0,
|
|
"file chunk round trip changed fields");
|
|
ok &= Expect(!crossdesk::DecodeFileChunk(
|
|
encoded.data(), encoded.size() - 1, &chunk),
|
|
"truncated file chunk should be rejected");
|
|
ok &= Expect(crossdesk::EncodeFileChunk(
|
|
17, payload.size() + 1, payload.size(), nullptr, 0, nullptr,
|
|
false, false)
|
|
.empty(),
|
|
"file chunk offset beyond total size should be rejected");
|
|
ok &= Expect(crossdesk::EncodeFileChunk(
|
|
17, payload.size(), payload.size(), payload.data(), 1,
|
|
nullptr, false, false)
|
|
.empty(),
|
|
"file chunk payload beyond remaining size should be rejected");
|
|
ok &= Expect(crossdesk::EncodeFileChunk(
|
|
17, 0, 1, nullptr, 1, nullptr, false, false)
|
|
.empty(),
|
|
"non-empty file chunk with null payload should be rejected");
|
|
const std::string oversized_name(
|
|
static_cast<std::size_t>(std::numeric_limits<uint16_t>::max()) + 1,
|
|
'x');
|
|
ok &= Expect(crossdesk::EncodeFileChunk(
|
|
17, 0, 0, nullptr, 0, &oversized_name, true, true)
|
|
.empty(),
|
|
"oversized file name should be rejected");
|
|
|
|
crossdesk::FileTransferAck ack{};
|
|
ack.magic = crossdesk::kFileAckMagic;
|
|
ack.file_id = 17;
|
|
ack.acked_offset = payload.size();
|
|
ack.total_size = payload.size();
|
|
ack.flags = 1;
|
|
const auto encoded_ack = crossdesk::EncodeFileTransferAck(ack);
|
|
crossdesk::FileTransferAck decoded_ack{};
|
|
ok &= Expect(crossdesk::DecodeFileTransferAck(
|
|
encoded_ack.data(), encoded_ack.size(), &decoded_ack) &&
|
|
decoded_ack.file_id == ack.file_id &&
|
|
decoded_ack.acked_offset == ack.acked_offset &&
|
|
decoded_ack.flags == ack.flags,
|
|
"file acknowledgement round trip failed");
|
|
ok &= Expect(!crossdesk::DecodeFileTransferAck(
|
|
encoded_ack.data(), encoded_ack.size() - 1, &decoded_ack),
|
|
"short file acknowledgement should be rejected");
|
|
auto invalid_ack = encoded_ack;
|
|
crossdesk::FileTransferAck invalid_ack_value = ack;
|
|
invalid_ack_value.acked_offset = invalid_ack_value.total_size + 1;
|
|
invalid_ack =
|
|
crossdesk::EncodeFileTransferAck(invalid_ack_value);
|
|
ok &= Expect(!crossdesk::DecodeFileTransferAck(
|
|
invalid_ack.data(), invalid_ack.size(), &decoded_ack),
|
|
"file acknowledgement beyond total size should be rejected");
|
|
|
|
ok &= Expect(std::string(crossdesk::kControlStream) ==
|
|
"control_data" &&
|
|
std::string(crossdesk::kFileFeedbackStream) ==
|
|
"file_feedback",
|
|
"stream names changed");
|
|
ok &= Expect(crossdesk::MakeDisplayStreamId(0) == "Display1" &&
|
|
crossdesk::MakeDisplayStreamId(2) == "Display3" &&
|
|
crossdesk::IsRegisteredDisplayStreamId("Display2", 2) &&
|
|
!crossdesk::IsRegisteredDisplayStreamId("Display3", 2),
|
|
"display stream identifiers changed");
|
|
ok &= Expect(crossdesk::ResolveDisplayStreamId(nullptr, 1) == "Display1" &&
|
|
crossdesk::ResolveDisplayStreamId("invalid", 3, 1) ==
|
|
"Display2",
|
|
"display stream fallback changed");
|
|
return ok ? 0 : 1;
|
|
}
|