mirror of https://github.com/NekoX-Dev/NekoX.git
115 lines
3.6 KiB
C++
115 lines
3.6 KiB
C++
/*
|
|
* This is the source code of tgnet library v. 1.1
|
|
* It is licensed under GNU GPL v. 2 or later.
|
|
* You should have received a copy of the license in this archive (see LICENSE).
|
|
*
|
|
* Copyright Nikolai Kudashov, 2015-2018.
|
|
*/
|
|
|
|
#include <algorithm>
|
|
#include <openssl/rand.h>
|
|
#include "ConnectionSession.h"
|
|
#include "MTProtoScheme.h"
|
|
#include "ConnectionsManager.h"
|
|
#include "NativeByteBuffer.h"
|
|
|
|
ConnectionSession::ConnectionSession(int32_t instance) {
|
|
instanceNum = instance;
|
|
}
|
|
|
|
void ConnectionSession::recreateSession() {
|
|
processedMessageIds.clear();
|
|
messagesIdsForConfirmation.clear();
|
|
processedSessionChanges.clear();
|
|
nextSeqNo = 0;
|
|
|
|
genereateNewSessionId();
|
|
}
|
|
|
|
void ConnectionSession::genereateNewSessionId() {
|
|
int64_t newSessionId;
|
|
RAND_bytes((uint8_t *) &newSessionId, 8);
|
|
#if USE_DEBUG_SESSION
|
|
sessionId = (0xabcd000000000000L | (newSessionId & 0x0000ffffffffffffL));
|
|
#else
|
|
sessionId = newSessionId;
|
|
#endif
|
|
}
|
|
|
|
void ConnectionSession::setSessionId(int64_t id) {
|
|
sessionId = id;
|
|
}
|
|
|
|
int64_t ConnectionSession::getSessionId() {
|
|
return sessionId;
|
|
}
|
|
|
|
uint32_t ConnectionSession::generateMessageSeqNo(bool increment) {
|
|
uint32_t value = nextSeqNo;
|
|
if (increment) {
|
|
nextSeqNo++;
|
|
}
|
|
return value * 2 + (increment ? 1 : 0);
|
|
}
|
|
|
|
int32_t ConnectionSession::isMessageIdProcessed(int64_t messageId) {
|
|
if (!(messageId & 1)) {
|
|
return 1;
|
|
}
|
|
if (minProcessedMessageId != 0 && messageId < minProcessedMessageId) {
|
|
return 2;
|
|
}
|
|
if (std::find(processedMessageIds.begin(), processedMessageIds.end(), messageId) != processedMessageIds.end()) {
|
|
return 1;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
void ConnectionSession::addProcessedMessageId(int64_t messageId) {
|
|
if (processedMessageIds.size() > 300) {
|
|
std::sort(processedMessageIds.begin(), processedMessageIds.end());
|
|
processedMessageIds.erase(processedMessageIds.begin(), processedMessageIds.begin() + 100);
|
|
minProcessedMessageId = *(processedMessageIds.begin());
|
|
}
|
|
processedMessageIds.push_back(messageId);
|
|
}
|
|
|
|
bool ConnectionSession::hasMessagesToConfirm() {
|
|
return !messagesIdsForConfirmation.empty();
|
|
}
|
|
|
|
void ConnectionSession::addMessageToConfirm(int64_t messageId) {
|
|
if (std::find(messagesIdsForConfirmation.begin(), messagesIdsForConfirmation.end(), messageId) != messagesIdsForConfirmation.end()) {
|
|
return;
|
|
}
|
|
messagesIdsForConfirmation.push_back(messageId);
|
|
}
|
|
|
|
NetworkMessage *ConnectionSession::generateConfirmationRequest() {
|
|
NetworkMessage *networkMessage = nullptr;
|
|
|
|
if (!messagesIdsForConfirmation.empty()) {
|
|
TL_msgs_ack *msgAck = new TL_msgs_ack();
|
|
msgAck->msg_ids.insert(msgAck->msg_ids.begin(), messagesIdsForConfirmation.begin(), messagesIdsForConfirmation.end());
|
|
NativeByteBuffer *os = new NativeByteBuffer(true);
|
|
msgAck->serializeToStream(os);
|
|
networkMessage = new NetworkMessage();
|
|
networkMessage->message = std::unique_ptr<TL_message>(new TL_message);
|
|
networkMessage->message->msg_id = ConnectionsManager::getInstance(instanceNum).generateMessageId();
|
|
networkMessage->message->seqno = generateMessageSeqNo(false);
|
|
networkMessage->message->bytes = os->capacity();
|
|
networkMessage->message->body = std::unique_ptr<TLObject>(msgAck);
|
|
messagesIdsForConfirmation.clear();
|
|
}
|
|
|
|
return networkMessage;
|
|
}
|
|
|
|
bool ConnectionSession::isSessionProcessed(int64_t sessionId) {
|
|
return std::find(processedSessionChanges.begin(), processedSessionChanges.end(), sessionId) != processedSessionChanges.end();
|
|
}
|
|
|
|
void ConnectionSession::addProcessedSession(int64_t sessionId) {
|
|
processedSessionChanges.push_back(sessionId);
|
|
}
|