273 lines
5.8 KiB
C++
273 lines
5.8 KiB
C++
// Copyright 2019 Dolphin Emulator Project
|
|
// Licensed under GPLv2+
|
|
// Refer to the license.txt file included.
|
|
|
|
#pragma once
|
|
|
|
#include <optional>
|
|
|
|
#include <zlib.h>
|
|
|
|
#include "Common/CommonTypes.h"
|
|
|
|
namespace ciface::DualShockUDPClient::Proto
|
|
{
|
|
// CemuHook DualShockUDP protocol implementation using UdpServer.cs from
|
|
// https://github.com/Ryochan7/DS4Windows as documentation.
|
|
//
|
|
// WARNING: Little endian host assumed
|
|
|
|
static constexpr u16 CEMUHOOK_PROTOCOL_VERSION = 1001;
|
|
|
|
static constexpr int PORT_COUNT = 4;
|
|
|
|
static constexpr char CLIENT[] = "DSUC";
|
|
static constexpr char SERVER[] = "DSUS";
|
|
|
|
#pragma pack(push, 1)
|
|
|
|
enum class DsState : u8
|
|
{
|
|
Disconnected = 0x00,
|
|
Reserved = 0x01,
|
|
Connected = 0x02
|
|
};
|
|
|
|
enum class DsConnection : u8
|
|
{
|
|
None = 0x00,
|
|
Usb = 0x01,
|
|
Bluetooth = 0x02
|
|
};
|
|
|
|
enum class DsModel : u8
|
|
{
|
|
None = 0,
|
|
DS3 = 1,
|
|
DS4 = 2,
|
|
Generic = 3
|
|
};
|
|
|
|
enum class DsBattery : u8
|
|
{
|
|
None = 0x00,
|
|
Dying = 0x01,
|
|
Low = 0x02,
|
|
Medium = 0x03,
|
|
High = 0x04,
|
|
Full = 0x05,
|
|
Charging = 0xEE,
|
|
Charged = 0xEF
|
|
};
|
|
|
|
enum RegisterFlags : u8
|
|
{
|
|
AllPads = 0x00,
|
|
PadID = 0x01,
|
|
PadMACAdddress = 0x02
|
|
};
|
|
|
|
struct MessageHeader
|
|
{
|
|
u8 source[4];
|
|
u16 protocol_version;
|
|
u16 message_length; // actually message size minus header size
|
|
u32 crc32;
|
|
u32 source_uid;
|
|
};
|
|
|
|
struct Touch
|
|
{
|
|
u8 active;
|
|
u8 id;
|
|
s16 x;
|
|
s16 y;
|
|
};
|
|
|
|
namespace MessageType
|
|
{
|
|
struct VersionRequest
|
|
{
|
|
static constexpr auto FROM = CLIENT;
|
|
static constexpr auto TYPE = 0x100000U;
|
|
MessageHeader header;
|
|
u32 message_type;
|
|
};
|
|
|
|
struct VersionResponse
|
|
{
|
|
static constexpr auto FROM = SERVER;
|
|
static constexpr auto TYPE = 0x100000U;
|
|
MessageHeader header;
|
|
u32 message_type;
|
|
u16 max_protocol_version;
|
|
u8 padding[2];
|
|
};
|
|
|
|
struct ListPorts
|
|
{
|
|
static constexpr auto FROM = CLIENT;
|
|
static constexpr auto TYPE = 0x100001U;
|
|
MessageHeader header;
|
|
u32 message_type;
|
|
u32 pad_request_count;
|
|
u8 pad_id[4];
|
|
};
|
|
|
|
struct PortInfo
|
|
{
|
|
static constexpr auto FROM = SERVER;
|
|
static constexpr auto TYPE = 0x100001U;
|
|
MessageHeader header;
|
|
u32 message_type;
|
|
u8 pad_id;
|
|
DsState pad_state;
|
|
DsModel model;
|
|
DsConnection connection_type;
|
|
u8 pad_mac_address[6];
|
|
DsBattery battery_status;
|
|
u8 padding;
|
|
};
|
|
|
|
struct PadDataRequest
|
|
{
|
|
static constexpr auto FROM = CLIENT;
|
|
static constexpr auto TYPE = 0x100002U;
|
|
MessageHeader header;
|
|
u32 message_type;
|
|
RegisterFlags register_flags;
|
|
u8 pad_id_to_register;
|
|
u8 mac_address_to_register[6];
|
|
};
|
|
|
|
struct PadDataResponse
|
|
{
|
|
static constexpr auto FROM = SERVER;
|
|
static constexpr auto TYPE = 0x100002U;
|
|
MessageHeader header;
|
|
u32 message_type;
|
|
u8 pad_id;
|
|
DsState pad_state;
|
|
DsModel model;
|
|
DsConnection connection_type;
|
|
u8 pad_mac_address[6];
|
|
DsBattery battery_status;
|
|
u8 active;
|
|
u32 hid_packet_counter;
|
|
u8 button_states1;
|
|
u8 button_states2;
|
|
u8 button_ps;
|
|
u8 button_touch;
|
|
u8 left_stick_x;
|
|
u8 left_stick_y_inverted;
|
|
u8 right_stick_x;
|
|
u8 right_stick_y_inverted;
|
|
u8 button_dpad_left_analog;
|
|
u8 button_dpad_down_analog;
|
|
u8 button_dpad_right_analog;
|
|
u8 button_dpad_up_analog;
|
|
u8 button_square_analog;
|
|
u8 button_cross_analog;
|
|
u8 button_circle_analog;
|
|
u8 button_triangle_analog;
|
|
u8 button_r1_analog;
|
|
u8 button_l1_analog;
|
|
u8 trigger_r2;
|
|
u8 trigger_l2;
|
|
Touch touch1;
|
|
Touch touch2;
|
|
u64 timestamp_us;
|
|
float accelerometer_x_g;
|
|
float accelerometer_y_g;
|
|
float accelerometer_z_inverted_g;
|
|
float gyro_pitch_deg_s;
|
|
float gyro_yaw_deg_s;
|
|
float gyro_roll_deg_s;
|
|
};
|
|
|
|
struct FromServer
|
|
{
|
|
union
|
|
{
|
|
struct
|
|
{
|
|
MessageHeader header;
|
|
u32 message_type;
|
|
};
|
|
MessageType::VersionResponse version_response;
|
|
MessageType::PortInfo port_info;
|
|
MessageType::PadDataResponse pad_data_response;
|
|
};
|
|
};
|
|
|
|
struct FromClient
|
|
{
|
|
union
|
|
{
|
|
struct
|
|
{
|
|
MessageHeader header;
|
|
u32 message_type;
|
|
};
|
|
MessageType::VersionRequest version_request;
|
|
MessageType::ListPorts list_ports;
|
|
MessageType::PadDataRequest pad_data_request;
|
|
};
|
|
};
|
|
} // namespace MessageType
|
|
|
|
static inline u32 CRC32(const void* buffer, unsigned length)
|
|
{
|
|
return crc32(crc32(0L, Z_NULL, 0), static_cast<const Bytef*>(buffer), length);
|
|
}
|
|
|
|
template <typename MsgType>
|
|
struct Message
|
|
{
|
|
Message() : m_message{} {}
|
|
|
|
explicit Message(u32 source_uid) : m_message{}
|
|
{
|
|
memcpy((char*)m_message.header.source, MsgType::FROM, sizeof(m_message.header.source));
|
|
m_message.header.protocol_version = CEMUHOOK_PROTOCOL_VERSION;
|
|
m_message.header.message_length = sizeof(*this) - sizeof(m_message.header);
|
|
m_message.header.source_uid = source_uid;
|
|
m_message.message_type = MsgType::TYPE;
|
|
}
|
|
|
|
void Finish() { m_message.header.crc32 = CRC32(&m_message, sizeof(m_message)); }
|
|
|
|
template <class ToMsgType>
|
|
std::optional<ToMsgType> CheckAndCastTo()
|
|
{
|
|
u32 crc32_in_header = m_message.header.crc32;
|
|
// zero out the crc32 in the packet once we got it since that's whats needed for calculation
|
|
m_message.header.crc32 = 0;
|
|
u32 crc32_calculated = CRC32(&m_message, sizeof(ToMsgType));
|
|
if (crc32_in_header != crc32_calculated)
|
|
{
|
|
NOTICE_LOG(SERIALINTERFACE,
|
|
"DualShockUDPClient Received message with bad CRC in header: got %u, expected %u",
|
|
crc32_in_header, crc32_calculated);
|
|
return std::nullopt;
|
|
}
|
|
if (m_message.header.protocol_version > CEMUHOOK_PROTOCOL_VERSION)
|
|
return std::nullopt;
|
|
if (memcmp(m_message.header.source, ToMsgType::FROM, sizeof(m_message.header.source)))
|
|
return std::nullopt;
|
|
if (m_message.message_type != ToMsgType::TYPE)
|
|
return std::nullopt;
|
|
if (m_message.header.message_length + sizeof(m_message.header) > sizeof(ToMsgType))
|
|
return std::nullopt;
|
|
|
|
ToMsgType tomsg;
|
|
memcpy(&tomsg, &m_message, sizeof(tomsg));
|
|
return tomsg;
|
|
}
|
|
|
|
MsgType m_message;
|
|
};
|
|
|
|
#pragma pack(pop)
|
|
} // namespace ciface::DualShockUDPClient::Proto
|