1
0
mirror of https://github.com/NekoX-Dev/NekoX.git synced 2024-12-14 05:29:33 +01:00
NekoX/TMessagesProj/jni/voip/tgcalls/Message.cpp
2020-10-01 04:59:32 +03:00

381 lines
11 KiB
C++

#include "Message.h"
#include "rtc_base/byte_buffer.h"
#include "api/jsep_ice_candidate.h"
namespace tgcalls {
namespace {
constexpr auto kMaxStringLength = 65536;
void Serialize(rtc::ByteBufferWriter &to, const std::string &from) {
assert(from.size() < kMaxStringLength);
to.WriteUInt32(uint32_t(from.size()));
to.WriteString(from);
}
bool Deserialize(std::string &to, rtc::ByteBufferReader &from) {
uint32_t length = 0;
if (!from.ReadUInt32(&length)) {
RTC_LOG(LS_ERROR) << "Could not read string length.";
return false;
} else if (length >= kMaxStringLength) {
RTC_LOG(LS_ERROR) << "Invalid string length: " << length;
return false;
} else if (!from.ReadString(&to, length)) {
RTC_LOG(LS_ERROR) << "Could not read string data.";
return false;
}
return true;
}
void Serialize(rtc::ByteBufferWriter &to, const webrtc::SdpVideoFormat &from) {
assert(from.parameters.size() < std::numeric_limits<uint8_t>::max());
Serialize(to, from.name);
to.WriteUInt8(uint8_t(from.parameters.size()));
for (const auto &pair : from.parameters) {
Serialize(to, pair.first);
Serialize(to, pair.second);
}
}
bool Deserialize(webrtc::SdpVideoFormat &to, rtc::ByteBufferReader &from) {
if (!Deserialize(to.name, from)) {
RTC_LOG(LS_ERROR) << "Could not read video format name.";
return false;
}
auto count = uint8_t();
if (!from.ReadUInt8(&count)) {
RTC_LOG(LS_ERROR) << "Could not read video format parameters count.";
return false;
}
for (uint32_t i = 0; i != count; ++i) {
auto key = std::string();
auto value = std::string();
if (!Deserialize(key, from)) {
RTC_LOG(LS_ERROR) << "Could not read video format parameter key.";
return false;
} else if (!Deserialize(value, from)) {
RTC_LOG(LS_ERROR) << "Could not read video format parameter value.";
return false;
}
to.parameters.emplace(key, value);
}
return true;
}
void Serialize(rtc::ByteBufferWriter &to, const cricket::Candidate &from) {
webrtc::JsepIceCandidate iceCandidate{ std::string(), 0 };
iceCandidate.SetCandidate(from);
std::string serialized;
const auto success = iceCandidate.ToString(&serialized);
assert(success);
Serialize(to, serialized);
}
bool Deserialize(cricket::Candidate &to, rtc::ByteBufferReader &from) {
std::string candidate;
if (!Deserialize(candidate, from)) {
RTC_LOG(LS_ERROR) << "Could not read candidate string.";
return false;
}
webrtc::JsepIceCandidate parseCandidate{ std::string(), 0 };
if (!parseCandidate.Initialize(candidate, nullptr)) {
RTC_LOG(LS_ERROR) << "Could not parse candidate: " << candidate;
return false;
}
to = parseCandidate.candidate();
return true;
}
void Serialize(rtc::ByteBufferWriter &to, const RequestVideoMessage &from, bool singleMessagePacket) {
}
bool Deserialize(RequestVideoMessage &to, rtc::ByteBufferReader &reader, bool singleMessagePacket) {
return true;
}
void Serialize(rtc::ByteBufferWriter &to, const RemoteMediaStateMessage &from, bool singleMessagePacket) {
uint8_t state = (uint8_t(from.video) << 1) | uint8_t(from.audio);
to.WriteUInt8(state);
}
bool Deserialize(RemoteMediaStateMessage &to, rtc::ByteBufferReader &reader, bool singleMessagePacket) {
uint8_t state = 0;
if (!reader.ReadUInt8(&state)) {
RTC_LOG(LS_ERROR) << "Could not read remote media state.";
return false;
}
to.audio = AudioState(state & 0x01);
to.video = VideoState((state >> 1) & 0x03);
if (to.video == VideoState(0x03)) {
RTC_LOG(LS_ERROR) << "Invalid value for remote video state.";
return false;
}
return true;
}
void Serialize(rtc::ByteBufferWriter &to, const CandidatesListMessage &from, bool singleMessagePacket) {
assert(from.candidates.size() < std::numeric_limits<uint8_t>::max());
to.WriteUInt8(uint8_t(from.candidates.size()));
for (const auto &candidate : from.candidates) {
Serialize(to, candidate);
}
Serialize(to, from.iceParameters.ufrag);
Serialize(to, from.iceParameters.pwd);
}
bool Deserialize(CandidatesListMessage &to, rtc::ByteBufferReader &reader, bool singleMessagePacket) {
auto count = uint8_t();
if (!reader.ReadUInt8(&count)) {
RTC_LOG(LS_ERROR) << "Could not read candidates count.";
return false;
}
for (uint32_t i = 0; i != count; ++i) {
auto candidate = cricket::Candidate();
if (!Deserialize(candidate, reader)) {
RTC_LOG(LS_ERROR) << "Could not read candidate.";
return false;
}
to.candidates.push_back(std::move(candidate));
}
if (!Deserialize(to.iceParameters.ufrag, reader)) {
return false;
}
if (!Deserialize(to.iceParameters.pwd, reader)) {
return false;
}
return true;
}
void Serialize(rtc::ByteBufferWriter &to, const VideoFormatsMessage &from, bool singleMessagePacket) {
assert(from.formats.size() < std::numeric_limits<uint8_t>::max());
assert(from.encodersCount <= from.formats.size());
to.WriteUInt8(uint8_t(from.formats.size()));
for (const auto &format : from.formats) {
Serialize(to, format);
}
to.WriteUInt8(uint8_t(from.encodersCount));
}
bool Deserialize(VideoFormatsMessage &to, rtc::ByteBufferReader &from, bool singleMessagePacket) {
auto count = uint8_t();
if (!from.ReadUInt8(&count)) {
RTC_LOG(LS_ERROR) << "Could not read video formats count.";
return false;
}
for (uint32_t i = 0; i != count; ++i) {
auto format = webrtc::SdpVideoFormat(std::string());
if (!Deserialize(format, from)) {
RTC_LOG(LS_ERROR) << "Could not read video format.";
return false;
}
to.formats.push_back(std::move(format));
}
auto encoders = uint8_t();
if (!from.ReadUInt8(&encoders)) {
RTC_LOG(LS_ERROR) << "Could not read encoders count.";
return false;
} else if (encoders > to.formats.size()) {
RTC_LOG(LS_ERROR) << "Invalid encoders count: " << encoders << ", full formats count: " << to.formats.size();
return false;
}
to.encodersCount = encoders;
return true;
}
void Serialize(rtc::ByteBufferWriter &to, const rtc::CopyOnWriteBuffer &from, bool singleMessagePacket) {
if (!singleMessagePacket) {
assert(from.size() <= UINT16_MAX);
to.WriteUInt16(from.size());
}
to.WriteBytes(reinterpret_cast<const char*>(from.cdata()), from.size());
}
bool Deserialize(rtc::CopyOnWriteBuffer &to, rtc::ByteBufferReader &from, bool singleMessagePacket) {
auto length = uint16_t(from.Length());
if (!singleMessagePacket) {
if (!from.ReadUInt16(&length)) {
RTC_LOG(LS_ERROR) << "Could not read buffer length.";
return false;
} else if (from.Length() < length) {
RTC_LOG(LS_ERROR) << "Invalid buffer length: " << length << ", available: " << from.Length();
return false;
}
}
to.AppendData(from.Data(), length);
from.Consume(length);
return true;
}
void Serialize(rtc::ByteBufferWriter &to, const AudioDataMessage &from, bool singleMessagePacket) {
Serialize(to, from.data, singleMessagePacket);
}
bool Deserialize(AudioDataMessage &to, rtc::ByteBufferReader &from, bool singleMessagePacket) {
return Deserialize(to.data, from, singleMessagePacket);
}
void Serialize(rtc::ByteBufferWriter &to, const VideoDataMessage &from, bool singleMessagePacket) {
Serialize(to, from.data, singleMessagePacket);
}
bool Deserialize(VideoDataMessage &to, rtc::ByteBufferReader &from, bool singleMessagePacket) {
return Deserialize(to.data, from, singleMessagePacket);
}
void Serialize(rtc::ByteBufferWriter &to, const UnstructuredDataMessage &from, bool singleMessagePacket) {
Serialize(to, from.data, singleMessagePacket);
}
bool Deserialize(UnstructuredDataMessage &to, rtc::ByteBufferReader &from, bool singleMessagePacket) {
return Deserialize(to.data, from, singleMessagePacket);
}
void Serialize(rtc::ByteBufferWriter &to, const VideoParametersMessage &from, bool singleMessagePacket) {
to.WriteUInt32(from.aspectRatio);
}
bool Deserialize(VideoParametersMessage &to, rtc::ByteBufferReader &from, bool singleMessagePacket) {
uint32_t aspectRatio = 0;
if (!from.ReadUInt32(&aspectRatio)) {
return false;
}
to.aspectRatio = aspectRatio;
return true;
}
void Serialize(rtc::ByteBufferWriter &to, const RemoteBatteryLevelIsLowMessage &from, bool singleMessagePacket) {
to.WriteUInt8(from.batteryLow ? 1 : 0);
}
bool Deserialize(RemoteBatteryLevelIsLowMessage &to, rtc::ByteBufferReader &reader, bool singleMessagePacket) {
uint8_t value = 0;
if (!reader.ReadUInt8(&value)) {
RTC_LOG(LS_ERROR) << "Could not read batteryLevelIsLow.";
return false;
}
to.batteryLow = (value != 0);
return true;
}
void Serialize(rtc::ByteBufferWriter &to, const RemoteNetworkStatusMessage &from, bool singleMessagePacket) {
to.WriteUInt8(from.isLowCost ? 1 : 0);
to.WriteUInt8(from.isLowDataRequested ? 1 : 0);
}
bool Deserialize(RemoteNetworkStatusMessage &to, rtc::ByteBufferReader &reader, bool singleMessagePacket) {
uint8_t value = 0;
if (!reader.ReadUInt8(&value)) {
RTC_LOG(LS_ERROR) << "Could not read isLowCost.";
return false;
}
to.isLowCost = (value != 0);
if (reader.ReadUInt8(&value)) {
to.isLowDataRequested = (value != 0);
}
return true;
}
enum class TryResult : uint8_t {
Success,
TryNext,
Abort,
};
template <typename T>
TryResult TryDeserialize(
absl::optional<Message> &to,
rtc::ByteBufferReader &reader,
bool singleMessagePacket) {
assert(reader.Length() != 0);
constexpr auto id = T::kId;
if (uint8_t(*reader.Data()) != id) {
return TryResult::TryNext;
}
reader.Consume(1);
auto parsed = T();
if (!Deserialize(parsed, reader, singleMessagePacket)) {
RTC_LOG(LS_ERROR) << "Could not read message with kId: " << id;
return TryResult::Abort;
}
to = Message{ parsed };
return TryResult::Success;
}
template <typename ...Types>
struct TryDeserializeNext;
template <>
struct TryDeserializeNext<> {
static bool Call(
absl::optional<Message> &to,
rtc::ByteBufferReader &reader,
bool singleMessagePacket) {
return false;
}
};
template <typename T, typename ...Other>
struct TryDeserializeNext<T, Other...> {
static bool Call(
absl::optional<Message> &to,
rtc::ByteBufferReader &reader,
bool singleMessagePacket) {
const auto result = TryDeserialize<T>(to, reader, singleMessagePacket);
return (result == TryResult::TryNext)
? TryDeserializeNext<Other...>::Call(to, reader, singleMessagePacket)
: (result == TryResult::Success);
}
};
template <typename ...Types>
bool TryDeserializeRecursive(
absl::optional<Message> &to,
rtc::ByteBufferReader &reader,
bool singleMessagePacket,
absl::variant<Types...> *) {
return TryDeserializeNext<Types...>::Call(to, reader, singleMessagePacket);
}
} // namespace
rtc::CopyOnWriteBuffer SerializeMessageWithSeq(
const Message &message,
uint32_t seq,
bool singleMessagePacket) {
rtc::ByteBufferWriter writer;
writer.WriteUInt32(seq);
absl::visit([&](const auto &data) {
writer.WriteUInt8(std::decay_t<decltype(data)>::kId);
Serialize(writer, data, singleMessagePacket);
}, message.data);
auto result = rtc::CopyOnWriteBuffer();
result.AppendData(writer.Data(), writer.Length());
return result;
}
absl::optional<Message> DeserializeMessage(
rtc::ByteBufferReader &reader,
bool singleMessagePacket) {
if (!reader.Length()) {
return absl::nullopt;
}
using Variant = decltype(std::declval<Message>().data);
auto result = absl::make_optional<Message>();
return TryDeserializeRecursive(result, reader, singleMessagePacket, (Variant*)nullptr)
? result
: absl::nullopt;
}
} // namespace tgcalls