123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882 |
- /*
- * Bael'Zharon's Respite
- * Copyright (C) 2014 Daniel Skorupski
- *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation; either version 2 of the License, or
- * (at your option) any later version.
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License along
- * with this program; if not, write to the Free Software Foundation, Inc.,
- * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
- */
- #include "net/Session.h"
- #include "net/SessionManager.h"
- #include "net/Socket.h"
- #include "BinReader.h"
- #include "BinWriter.h"
- #include "Core.h"
- #include "Log.h"
- #include <algorithm>
- #ifndef _WIN32
- #include <arpa/inet.h>
- #endif
- enum OptionalHeaderFlags
- {
- kDisposable = 0x00000001, // this header may be removed from a retransmission
- kExclusive = 0x00000002, // a packet with this header has its own sequence number
- kNotConn = 0x00000004, // this header is sent before connect request/reply handshake completes
- kTimeSensitive = 0x00000008,
- kShouldPiggyBack = 0x00000010, // this header should ride along in a packet with others headers and content
- kHighPriority = 0x00000020,
- kCountsAsTouch = 0x00000040,
- kEncrypted = 0x20000000, // a packet with this header has its checksum encrypted
- kSigned = 0x40000000
- };
- enum PacketFlags
- {
- kRetransmission = 0x00000001,
- kEncryptedChecksum = 0x00000002,
- kBlobFragments = 0x00000004,
- kServerSwitch = 0x00000100, // CServerSwitchStruct (60, kHighPriority|kCountsAsTouch)
- kUnknown1 = 0x00000200, // CLogonRouteHeader (sockaddr_in) (7, kDisposable|kExclusive|kNotConn)
- kUnknown2 = 0x00000400, // EmptyHeader (7, kDisposable|kExclusive|kNotConn)
- kReferral = 0x00000800, // CReferralStruct (40000062, kExclusive|kHighPriority|kCountsAsTouch|kSigned)
- kRequestRetransmit = 0x00001000, // SeqIDList (33, kDisposable|kExclusive|kShouldPiggyBack|kHighPriority)
- kRejectRetransmit = 0x00002000, // SeqIDList (33, kDisposable|kExclusive|kShouldPiggyBack|kHighPriority)
- kAckSequence = 0x00004000, // CPakHeader (unsigned long) (1, kDisposable)
- kDisconnect = 0x00008000, // EmptyHeader (3, kDisposable|kExclusive)
- kLogon = 0x00010000, // CLogonHeader (?)
- kReferred = 0x00020000, // uint64_t (7, kDisposable|kExclusive|kNotConn)
- kConnectRequest = 0x00040000, // CConnectHeader (?)
- kConnectResponse = 0x00080000, // uint64_t (20000007, kDisposable|kExclusive|kNotConn|kEncrypted)
- kNetError1 = 0x00100000, // PackObjHeader<NetError> (7, kDisposable|kExclusive|kNotConn)
- kNetError2 = 0x00200000, // PackObjHeader<NetError> (2, kExclusive)
- kCICMDCommand = 0x00400000, // CICMDCommandStruct (7, kDisposable|kExclusive|kNotConn)
- kTimeSync = 0x01000000, // CTimeSyncHeader (?)
- kEchoRequest = 0x02000000, // CEchoRequestHeader (?)
- kEchoResponse = 0x04000000, // CEchoResponseHeader (?)
- kFlow = 0x08000000 // CFlowStruct (10, kShouldPiggyBack)
- };
- enum class ServerSwitchType
- {
- WorldSwitch,
- LogonSwitch,
- };
- static const chrono::milliseconds kLogonPacketDelay(300);
- static const chrono::milliseconds kReferredPacketDelay(300);
- static const chrono::milliseconds kConnectResponsePacketDelay(300);
- static const chrono::milliseconds kPingPacketDelay(2000);
- static const chrono::milliseconds kMissingPacketDelay(300);
- static const size_t kMaxPeriodicSent = 10;
- static ostream& operator<<(ostream& os, const PacketHeader& header)
- {
- os << "sequence=" << hexn(header.sequence)
- << " flags=" << hexn(header.flags)
- << " checksum=" << hexn(header.checksum)
- << " netId=" << hexn(header.netId)
- << " time=" << hexn(header.time)
- << " size=" << hexn(header.size)
- << " iteration=" << hexn(header.iteration);
- return os;
- }
- static ostream& operator<<(ostream& os, const FragmentHeader& header)
- {
- os << "id=" << hexn(header.id)
- << " count=" << hexn(header.count)
- << " size=" << hexn(header.size)
- << " index=" << hexn(header.index)
- << " queueId=" << hexn(header.queueId);
- return os;
- }
- static uint32_t checksum(const void* data, size_t size)
- {
- uint32_t result = static_cast<uint32_t>(size) << 16;
- for(size_t i = 0; i < size / 4; i++)
- {
- result += static_cast<const uint32_t*>(data)[i];
- }
- int shift = 24;
- for(size_t i = (size / 4) * 4; i < size; i++)
- {
- result += static_cast<const uint8_t*>(data)[i] << shift;
- shift -= 8;
- }
- return result;
- }
- static uint32_t checksumHeader(const PacketHeader& header)
- {
- PacketHeader& header_nc = const_cast<PacketHeader&>(header);
- uint32_t origChecksum = header.checksum;
- header_nc.checksum = 0xBADD70DD;
- uint32_t result = checksum(&header, sizeof(PacketHeader));
- header_nc.checksum = origChecksum;
- return result;
- }
- static uint32_t checksumContent(const PacketHeader& header, const void* data)
- {
- if(header.flags & kBlobFragments)
- {
- BinReader reader(data, header.size);
- if(header.flags & kServerSwitch)
- {
- reader.readRaw(8);
- }
- if(header.flags & kRequestRetransmit)
- {
- uint32_t nseq = reader.readInt();
- reader.readRaw(nseq * sizeof(uint32_t));
- }
- if(header.flags & kRejectRetransmit)
- {
- uint32_t nseq = reader.readInt();
- reader.readRaw(nseq * sizeof(uint32_t));
- }
- if(header.flags & kAckSequence)
- {
- reader.readRaw(4);
- }
- if(header.flags & kCICMDCommand)
- {
- reader.readRaw(8);
- }
- if(header.flags & kTimeSync)
- {
- reader.readRaw(8);
- }
- if(header.flags & kEchoRequest)
- {
- reader.readRaw(4);
- }
- if(header.flags & kEchoResponse)
- {
- reader.readRaw(8);
- }
- if(header.flags & kFlow)
- {
- reader.readRaw(6);
- }
- uint32_t result = checksum(data, reader.position());
- while(reader.remaining() != 0)
- {
- const FragmentHeader* fragment = reader.readPointer<FragmentHeader>();
- reader.readRaw(fragment->size - sizeof(FragmentHeader));
- result += checksum(fragment, fragment->size);
- }
- assert(reader.remaining() == 0);
- return result;
- }
- return checksum(data, header.size);
- }
- static uint32_t checksumPacket(const Packet& packet, ChecksumXorGenerator& xorGen)
- {
- uint32_t xorVal = (packet.header.flags & kEncryptedChecksum) ? xorGen.get(packet.header.sequence) : 0;
- return checksumHeader(packet.header) + (checksumContent(packet.header, packet.payload) ^ xorVal);
- }
- Session::Session(SessionManager& manager,
- Address address,
- string accountName,
- string accountKey) :
- manager_(manager),
- address_(address),
- state_(State::kLogon),
- sessionBegin_(net_clock::now()),
- nextPeriodic_(sessionBegin_),
- numPeriodicSent_(0),
- accountName_(move(accountName)),
- accountKey_(move(accountKey)),
- blobId_(0)
- {
- LOG(Net, Info) << address_ << " logon session created\n";
- }
- Session::Session(SessionManager& manager,
- Address address,
- uint64_t cookie) :
- manager_(manager),
- address_(address),
- state_(State::kReferred),
- sessionBegin_(net_clock::now()),
- nextPeriodic_(sessionBegin_),
- numPeriodicSent_(0),
- cookie_(cookie),
- blobId_(0)
- {
- LOG(Net, Info) << address_ << " referred session created\n";
- }
- Session::~Session()
- {
- LOG(Net, Info) << address_ << " destroyed\n";
- }
- void Session::handle(const Packet& packet)
- {
- uint32_t calcChecksum = checksumPacket(packet, serverXorGen_);
- if(packet.header.checksum != calcChecksum)
- {
- LOG(Net, Warn) << address_ << " dropping packet, calc checksum is " << hexn(calcChecksum) << ", packet has " << hexn(packet.header.checksum) << "\n";
- return;
- }
- LOG(Net, Debug) << address_ << " received packet, " << packet.header << "\n";
- BinReader reader(packet.payload, packet.header.size);
- if(state_ == State::kConnectResponse)
- {
- state_ = State::kConnected;
- nextPeriodic_ = net_clock::now() + kPingPacketDelay;
- numPeriodicSent_ = 0;
- LOG(Net, Info) << address_ << " transitioned to connected\n";
- }
- if(state_ == State::kLogon || state_ == State::kReferred)
- {
- // TODO Handle kNetError1 here
- if(packet.header.flags != kConnectRequest)
- {
- LOG(Net, Warn) << address_ << " dropping packet, flags should be " << hexn(static_cast<uint32_t>(kConnectRequest)) << ", packet has " << hexn(packet.header.flags) << "\n";
- return;
- }
- handleConnect(reader, packet.header);
- state_ = State::kConnectResponse;
- numPeriodicSent_ = 0;
- sendConnectResponse();
- }
- else if(state_ == State::kConnected)
- {
- if(packet.header.flags != kAckSequence) // AckSequence alone does not get it's own sequence
- {
- if(packet.header.sequence <= serverSequence_ || serverPackets_.find(packet.header.sequence) != serverPackets_.end())
- {
- LOG(Net, Warn) << address_ << " dropping packet, already received sequence " << packet.header.sequence << "\n";
- return;
- }
- }
- if(packet.header.netId != serverNetId_)
- {
- LOG(Net, Warn) << address_ << " dropping packet, server net id is " << serverNetId_ << ", packet has " << packet.header.netId << "\n";
- return;
- }
- if(packet.header.iteration != iteration_)
- {
- LOG(Net, Warn) << address_ << " dropping packet, iteration is " << iteration_ << ", packet has " << packet.header.iteration << "\n";
- return;
- }
- // record that we've received the packet
- serverPackets_.insert(packet.header.sequence);
- advanceServerSequence();
- // record that we've received some bytes
- lastFlowBytes_ += static_cast<uint32_t>(sizeof(PacketHeader) + packet.header.size);
- uint32_t flags = packet.header.flags & ~(kRetransmission | kEncryptedChecksum);
- if(flags == kServerSwitch)
- {
- handleServerSwitch(reader);
- }
- else if(flags == kReferral)
- {
- handleReferral(reader);
- }
- else
- {
- if(flags & kRequestRetransmit)
- {
- handleRequestRetransmit(reader);
- flags &= ~kRequestRetransmit;
- }
- if(flags & kRejectRetransmit)
- {
- handleRejectRetransmit(reader);
- flags &= ~kRejectRetransmit;
- }
- if(flags & kAckSequence)
- {
- handleAckSequence(reader);
- flags &= ~kAckSequence;
- }
- if(flags & kTimeSync)
- {
- handleTimeSync(reader);
- flags &= ~kTimeSync;
- }
- if(flags & kEchoResponse)
- {
- handleEchoResponse(reader);
- flags &= ~kEchoResponse;
- }
- if(flags & kFlow)
- {
- handleFlow(reader);
- flags &= ~kFlow;
- }
- if(flags & kBlobFragments)
- {
- handleBlobFragments(reader);
- flags &= ~kBlobFragments;
- }
- if(flags != 0)
- {
- LOG(Net, Error) << address_ << " unhandled flags " << hexn(flags) << " in packet\n";
- throw runtime_error("unhandled flags");
- }
- }
- }
- else
- {
- throw logic_error("bad state");
- }
- assert(reader.remaining() == 0);
- }
- void Session::tick(net_time_point now)
- {
- if(numPeriodicSent_ > kMaxPeriodicSent)
- {
- throw runtime_error("session timed out");
- }
- if(state_ == State::kLogon)
- {
- if(now > nextPeriodic_)
- {
- sendLogon();
- }
- }
- else if(state_ == State::kReferred)
- {
- if(now > nextPeriodic_)
- {
- sendReferred();
- }
- }
- else if(state_ == State::kConnectResponse)
- {
- if(now > nextPeriodic_)
- {
- sendConnectResponse();
- }
- }
- else if(state_ == State::kConnected)
- {
- Packet packet;
- packet.address = sendAddress();
- BinWriter writer(packet.payload, sizeof(packet.payload));
- uint16_t time = chrono::duration_cast<SessionDuration>(now - sessionBegin_).count();
- uint32_t flags = 0;
- /*
- if(now > nextRequestMissing_)
- {
- flags |= kRequestRetransmit;
- vector<uint32_t> missingPackets;
- uint32_t previousSequence = serverLeadingSequence_;
- for(uint32_t sequence : serverPackets_)
- {
- for(uint32_t missingSequence = previousSequence + 1; missingSequence < sequence; missingSequence++)
- {
- missingPackets.push_back(missingSequence);
- }
- previousSequence = sequence;
- }
- writer.writeInt(static_cast<uint32_t>(missingPackets.size()));
- for(uint32_t sequence : missingPackets)
- {
- writer.writeInt(sequence);
- }
- nextRequestMissing_ = now + kMissingPacketDelay;
- }
- */
- /*
- if(serverLeadingSequence_ > serverSequence_)
- {
- flags |= kAckSequence;
- writer.writeInt(serverLeadingSequence_);
- serverSequence_ = serverLeadingSequence_;
- LOG(Net, Debug) << address_ << " sending ack sequence " << serverLeadingSequence_ << "\n";
- }
- */
- if(now > nextPeriodic_)
- {
- flags |= kTimeSync;
- writer.writeDouble(beginTime_ + chrono::duration<double>(now - beginLocalTime_).count());
- flags |= kEchoRequest;
- writer.writeFloat(chrono::duration<float>(now - manager_.getClientBegin()).count());
- if(lastFlowBytes_ != 0)
- {
- flags |= kFlow;
- writer.writeInt(lastFlowBytes_);
- writer.writeShort(lastFlowTime_);
- lastFlowBytes_ = 0;
- lastFlowTime_ = time;
- }
- nextPeriodic_ = now + kPingPacketDelay;
- LOG(Net, Debug) << address_ << " sending time sync, echo request, flow\n";
- }
- if(flags != 0)
- {
- packet.header.sequence = clientLeadingSequence_ + 1;
- packet.header.flags = flags | kEncryptedChecksum;
- packet.header.netId = clientNetId_;
- packet.header.time = time;
- packet.header.size = static_cast<uint16_t>(writer.position());
- packet.header.iteration = iteration_;
- packet.header.checksum = checksumPacket(packet, clientXorGen_); // must be done last
- LOG(Net, Debug) << address_ << " sending packet " << packet.header << "\n";
- manager_.send(packet);
- packet.header.flags |= kRetransmission;
- packet.header.checksum = checksumPacket(packet, clientXorGen_); // must be done last
- clientPackets_[packet.header.sequence].reset(new Packet(packet));
- clientLeadingSequence_++;
- // since we already generated the retransmission checksum, we can purge right away
- clientXorGen_.purge(clientLeadingSequence_);
- }
- }
- else
- {
- throw logic_error("bad state");
- }
- }
- void Session::sendBlob(BlobPtr blob)
- {
- Packet packet;
- packet.address = sendAddress();
- packet.header.sequence = clientLeadingSequence_ + 1;
- packet.header.flags = kEncryptedChecksum | kBlobFragments;
- packet.header.netId = clientNetId_;
- packet.header.time = chrono::duration_cast<SessionDuration>(net_clock::now() - sessionBegin_).count();
- packet.header.iteration = iteration_;
- BinWriter writer(packet.payload, sizeof(packet.payload));
- // HACK
- //packet.header.flags |= kAckSequence;
- //writer.writeInt(2);
- FragmentHeader fragment;
- fragment.id = blobId_++;
- fragment.count = 1;
- fragment.size = static_cast<uint16_t>(sizeof(FragmentHeader) + blob->size);
- fragment.index = 0;
- fragment.queueId = blob->queueId;
- writer.writeRaw(&fragment, sizeof(fragment));
- writer.writeRaw(blob.get() + 1, blob->size);
- packet.header.size = static_cast<uint16_t>(writer.position());
- packet.header.checksum = checksumPacket(packet, clientXorGen_);
- manager_.send(packet);
- LOG(Net, Debug) << address_ << " sending " << packet.header << "\n";
- LOG(Net, Debug) << " fragment " << fragment << "\n";
- packet.header.flags |= kRetransmission;
- packet.header.checksum = checksumPacket(packet, clientXorGen_);
- clientPackets_[packet.header.sequence].reset(new Packet(packet));
- clientLeadingSequence_++;
- clientXorGen_.purge(clientLeadingSequence_);
- }
- Address Session::baseAddress() const
- {
- return address_;
- }
- Address Session::sendAddress() const
- {
- if(state_ == State::kConnectResponse)
- {
- return Address(address_.ip(), address_.port() + 1);
- }
- else
- {
- return address_;
- }
- }
- Address Session::recvAddress() const
- {
- if(state_ == State::kLogon || state_ == State::kReferred)
- {
- return address_;
- }
- else
- {
- return Address(address_.ip(), address_.port() + 1);
- }
- }
- net_time_point Session::nextTick() const
- {
- return min(nextPeriodic_, nextRequestMissing_);
- }
- BlobAssembler& Session::blobAssembler()
- {
- return blobAssembler_;
- }
- void Session::sendLogon()
- {
- Packet packet;
- packet.address = sendAddress();
- memset(&packet.header, 0, sizeof(packet.header));
- packet.header.flags = kLogon;
- BinWriter writer(packet.payload, sizeof(packet.payload));
- writer.writeString("1802"); // NetVersion
- size_t authDataLenOff = writer.position();
- writer.skip(sizeof(uint32_t));
- writer.writeInt(0x40000002); // GLSUserNameTicket_NetAuthType
- writer.writeInt(0);
- writer.writeInt(static_cast<uint32_t>(time(NULL)));
- writer.writeString(accountName_);
- writer.writeInt(0);
- writer.writeInt(static_cast<uint32_t>(sizeof(uint16_t) + accountKey_.size()));
- writer.writeShort(0xF480);
- writer.writeRaw(accountKey_.data(), accountKey_.size());
- packet.header.size = static_cast<uint16_t>(writer.position());
- writer.seek(authDataLenOff);
- writer.writeInt(static_cast<uint32_t>(packet.header.size - (authDataLenOff + sizeof(uint32_t))));
-
- packet.header.checksum = checksumPacket(packet, clientXorGen_); // must be done last
- manager_.send(packet);
- nextPeriodic_ = net_clock::now() + kLogonPacketDelay;
- numPeriodicSent_++;
- LOG(Net, Info) << address_ << " sent logon\n";
- }
- void Session::sendReferred()
- {
- Packet packet;
- packet.address = sendAddress();
- memset(&packet.header, 0, sizeof(packet.header));
- packet.header.flags = kReferred;
- packet.header.size = sizeof(uint64_t);
- BinWriter writer(packet.payload, sizeof(packet.payload));
- writer.writeLong(cookie_);
- packet.header.checksum = checksumPacket(packet, clientXorGen_); // must be done last
- manager_.send(packet);
- nextPeriodic_ = net_clock::now() + kReferredPacketDelay;
- numPeriodicSent_++;
- LOG(Net, Info) << address_ << " sent referred\n";
- }
- void Session::sendConnectResponse()
- {
- Packet packet;
- packet.address = sendAddress();
- memset(&packet.header, 0, sizeof(packet.header));
- packet.header.flags = kConnectResponse;
- packet.header.netId = clientNetId_;
- packet.header.size = sizeof(uint64_t);
- packet.header.iteration = iteration_;
- BinWriter writer(packet.payload, sizeof(packet.payload));
- writer.writeLong(cookie_);
- packet.header.checksum = checksumPacket(packet, clientXorGen_); // must be done last
- manager_.send(packet);
- nextPeriodic_ = net_clock::now() + kConnectResponsePacketDelay;
- numPeriodicSent_++;
- LOG(Net, Info) << address_ << " sent connect response\n";
- }
- void Session::handleBlobFragments(BinReader& reader)
- {
- while(reader.remaining() != 0)
- {
- const FragmentHeader* fragment = reader.readPointer<FragmentHeader>();
- reader.readRaw(fragment->size - sizeof(FragmentHeader));
- blobAssembler_.addFragment(fragment);
- }
- LOG(Net, Info) << address_ << " received blob fragments\n";
- }
- void Session::handleServerSwitch(BinReader& reader)
- {
- /*sequence*/ reader.readInt();
- /*type*/ reader.readInt();
- manager_.setPrimary(this);
- LOG(Net, Info) << address_ << " received server switch\n";
- }
- void Session::handleRequestRetransmit(BinReader& reader)
- {
- uint32_t numSequence = reader.readInt();
- for(uint32_t i = 0; i < numSequence; i++)
- {
- uint32_t sequence = reader.readInt();
- auto it = clientPackets_.find(sequence);
- if(it == clientPackets_.end())
- {
- LOG(Net, Warn) << address_ << " ignoring retransmit request of " << sequence << "\n";
- continue;
- }
- manager_.send(*it->second);
- }
- LOG(Net, Debug) << address_ << " received request retransmit for " << numSequence << " packets\n";
- }
- void Session::handleRejectRetransmit(BinReader& reader)
- {
- uint32_t numSequence = reader.readInt();
- for(uint32_t i = 0; i < numSequence; i++)
- {
- uint32_t sequence = reader.readInt();
- if(sequence > serverSequence_)
- {
- serverPackets_.insert(sequence);
- }
- }
- advanceServerSequence();
- LOG(Net, Debug) << address_ << " received reject retransmit for " << numSequence << " packets\n";
- }
- void Session::handleAckSequence(BinReader& reader)
- {
- clientSequence_ = max(clientSequence_, reader.readInt());
- for(auto it = clientPackets_.begin(); it != clientPackets_.end(); /**/)
- {
- if(it->first > clientSequence_)
- {
- break;
- }
- it = clientPackets_.erase(it);
- }
- LOG(Net, Debug) << address_ << " received ack sequence " << clientSequence_ << "\n";
- }
- void Session::handleReferral(BinReader& reader)
- {
- uint64_t cookie = reader.readLong();
- /*family*/ reader.readShort();
- uint16_t port = htons(reader.readShort());
- uint32_t ip = htonl(reader.readInt());
- /*zero*/ reader.readRaw(8);
- /*serverId*/ reader.readShort();
- /*padding*/ reader.readRaw(6);
- Address referral(ip, port);
- if(manager_.exists(referral))
- {
- LOG(Net, Warn) << address_ << " received referral for existing session " << referral << "\n";
- return;
- }
- unique_ptr<Session> session(new Session(manager_, referral, cookie));
- manager_.add(move(session));
- LOG(Net, Info) << address_ << " received referral to " << referral << "\n";
- }
- void Session::handleConnect(BinReader& reader, const PacketHeader& header)
- {
- double beginTime = reader.readDouble();
- uint64_t cookie = reader.readLong();
- uint16_t clientNetId = static_cast<uint16_t>(reader.readInt());
- uint32_t serverSeed = reader.readInt();
- uint32_t clientSeed = reader.readInt();
- reader.readInt(); // padding
- cookie_ = cookie;
- serverSequence_ = 1;
- serverLeadingSequence_ = 1;
- clientSequence_ = 1;
- clientLeadingSequence_ = 1;
- serverNetId_ = header.netId;
- clientNetId_ = clientNetId;
- iteration_ = header.iteration;
- serverXorGen_.init(serverSeed);
- clientXorGen_.init(clientSeed);
- beginTime_ = beginTime;
- beginLocalTime_ = net_clock::now();
- serverPackets_.clear();
- clientPackets_.clear();
- nextRequestMissing_ = net_time_point::max();
- lastFlowTime_ = 0;
- lastFlowBytes_ = 0;
- LOG(Net, Info) << address_ << " received connect\n";
- }
- void Session::handleTimeSync(BinReader& reader)
- {
- beginTime_ = reader.readDouble();
- beginLocalTime_ = net_clock::now();
- LOG(Net, Debug) << address_ << " received time sync " << beginTime_ << "\n";
- }
- void Session::handleEchoResponse(BinReader& reader)
- {
- numPeriodicSent_ = 0;
- // Ignored for now
- /*pingTime*/ reader.readFloat();
- /*pingResult*/ reader.readFloat();
- LOG(Net, Debug) << address_ << " received echo response\n";
- }
- void Session::handleFlow(BinReader& reader)
- {
- // Ignored for now
- /*numBytes*/ reader.readInt();
- /*time*/ reader.readShort();
- LOG(Net, Debug) << address_ << " received flow\n";
- }
- void Session::advanceServerSequence()
- {
- auto it = serverPackets_.begin();
- while(it != serverPackets_.end())
- {
- if(*it != serverLeadingSequence_ + 1)
- {
- break;
- }
- it = serverPackets_.erase(it);
- serverLeadingSequence_++;
- }
- serverXorGen_.purge(serverLeadingSequence_);
- if(!serverPackets_.empty())
- {
- if(nextRequestMissing_ == net_time_point::max())
- {
- nextRequestMissing_ = net_clock::now() + kMissingPacketDelay;
- }
- }
- else
- {
- if(nextRequestMissing_ != net_time_point::max())
- {
- nextRequestMissing_ = net_time_point::max();
- }
- }
- }
|