mirror of
https://github.com/danog/libtgvoip.git
synced 2024-12-11 08:39:49 +01:00
141 lines
3.6 KiB
C++
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> ¤tExtras, const int64_t &endpointId)
|
|
{
|
|
prepare(pm);
|
|
}
|
|
void Packet::prepare(PacketManager &pm, std::vector<UnacknowledgedExtraData> ¤tExtras, 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" |