1
0
mirror of https://github.com/danog/libtgvoip.git synced 2024-12-11 08:39:49 +01:00
libtgvoip/controller/protocol/packets/PacketStructs.cpp

141 lines
3.6 KiB
C++

#include "PacketStructs.h"
#include "../../PrivateDefines.cpp"
#include "PacketManager.h"
using namespace tgvoip;
bool Packet::parse(const BufferInputStream &in, const VersionInfo &ver)
{
if (!ver.isNew())
{
return parseLegacy(in, ver);
}
uint16_t length;
uint8_t flags;
bool res = in.TryRead(seq) &&
in.TryRead(ackSeq) &&
in.TryRead(ackMask) &&
in.TryRead(flags);
if (!res)
return false;
streamId = flags & 3;
flags >>= 2;
if (streamId == StreamId::Extended && !in.TryRead(streamId))
return false;
if (!(flags & Flags::Len16 ? in.TryRead(length) : in.TryReadCompat<uint8_t>(length)))
return false;
eFlags = length >> 11;
length &= 0x7FF;
if (eFlags & EFlags::Fragmented)
{
if (!in.TryRead(fragmentCount))
return false;
if (!in.TryRead(fragmentIndex))
return false;
}
data = std::make_unique<Buffer>(length);
if (in.TryRead(*data))
return false;
if ((flags & Flags::RecvTS) && !in.TryRead(recvTS))
return false;
if ((flags & Flags::ExtraEC) && !in.TryRead(extraEC, ver))
return false;
if ((flags & Flags::ExtraSignaling) && !in.TryRead(extraSignaling, ver))
return false;
return true;
}
void Packet::serialize(BufferOutputStream &out, const VersionInfo &ver) const
{
uint8_t shortStreamId = streamId > StreamId::Extended ? StreamId::Extended : streamId;
uint8_t flags = 0;
if (data->Length() > 0xFF || eFlags)
flags |= Flags::Len16;
if (recvTS)
flags |= Flags::RecvTS;
if (extraEC)
flags |= Flags::ExtraEC;
if (extraSignaling)
flags |= Flags::ExtraSignaling;
out.WriteUInt32(seq);
out.WriteUInt32(ackSeq);
out.WriteUInt32(ackMask);
out.WriteByte(shortStreamId | (flags << 2));
if (shortStreamId == StreamId::Extended)
out.WriteByte(streamId);
if (flags & Flags::Len16)
out.WriteUInt16(data->Length() | (eFlags << 11));
else
out.WriteByte(data->Length());
out.WriteBytes(*data);
if (flags & Flags::RecvTS)
out.WriteUInt32(recvTS);
if (flags & Flags::ExtraEC)
out.Write(extraEC, ver);
if (flags & Flags::ExtraSignaling)
out.Write(extraSignaling, ver);
}
void Packet::prepare(PacketManager &pm)
{
if (!seq)
{
seq = pm.nextLocalSeq();
ackSeq = pm.getLastRemoteSeq();
ackMask = pm.getRemoteAckMask();
}
}
void Packet::prepare(PacketManager &pm, std::vector<UnacknowledgedExtraData> &currentExtras, const int64_t &endpointId)
{
prepare(pm);
}
void Packet::prepare(PacketManager &pm, std::vector<UnacknowledgedExtraData> &currentExtras, const int64_t &endpointId, PacketManager &legacyPm, const int peerVersion)
{
prepare(pm);
if (!legacySeq)
{
if (pm != legacyPm)
{
legacySeq = legacyPm.nextLocalSeq();
ackSeq = pm.getLastRemoteSeq();
ackMask = pm.getRemoteAckMask();
}
else
{
legacySeq = seq;
}
}
int tmpLegacySeq = legacySeq;
extraSignaling.v.clear();
for (auto &extra : currentExtras)
{
if (!endpointId || extra.endpointId == endpointId)
{
extraSignaling.v.push_back(extra.data);
if (extra.data.d->chooseType(peerVersion) == PKT_NOP)
{
extra.seqs.Add(seq);
}
else
{
extra.seqs.Add(tmpLegacySeq++);
}
}
}
}
#include "Legacy.cpp"