Finalizing the voice connection cleanup

2020-07-29 22:53:40 +02:00
parent 4c91a7a3bf
commit 7a974677fb
25 changed files with 864 additions and 566 deletions
@@ -0,0 +1,84 @@
+#include <tommath.h>
+#include <misc/endianness.h>
+#include <log/LogUtils.h>
+#include "../web/WebClient.h"
+#include "VoiceClient.h"
+
+using namespace std;
+using namespace std::chrono;
+using namespace ts::connection;
+using namespace ts::protocol;
+
+void VoiceClientConnection::handlePacketPong(const ts::protocol::ClientPacketParser &packet) {
+    if(packet.payload_length() < 2) return;
+
+    this->ping_handler_.received_pong(be2le16((char*) packet.payload().data_ptr()));
+}
+
+void VoiceClientConnection::handlePacketPing(const protocol::ClientPacketParser& packet) {
+#ifdef PKT_LOG_PING
+    logMessage(this->getServerId(), "{}[Ping] Sending pong for client requested ping {}", CLIENT_STR_LOG_PREFIX, packet->packetId());
+#endif
+    char buffer[2];
+    le2be16(packet.packet_id(), buffer);
+    this->send_packet(PacketType::PONG, PacketFlag::Unencrypted, buffer, 2);
+}
+
+void VoiceClientConnection::handlePacketVoice(const protocol::ClientPacketParser& packet) {
+    auto client = this->getCurrentClient();
+    if(!client) return;
+
+    client->handlePacketVoice(packet.payload(), (packet.flags() & PacketFlag::Compressed) > 0, (packet.flags() & PacketFlag::Fragmented) > 0);
+}
+
+void VoiceClientConnection::handlePacketVoiceWhisper(const ts::protocol::ClientPacketParser &packet) {
+    auto client = this->getCurrentClient();
+    if(!client) return;
+
+    client->handlePacketVoiceWhisper(packet.payload(), (packet.flags() & PacketFlag::NewProtocol) > 0);
+}
+
+void VoiceClientConnection::handlePacketAck(const protocol::ClientPacketParser& packet) {
+    if(packet.payload_length() < 2) return;
+    uint16_t target_id{be2le16(packet.payload().data_ptr<char>())};
+
+    this->ping_handler_.received_command_acknowledged();
+    this->packet_statistics().received_acknowledge((protocol::PacketType) packet.type(), target_id | (uint32_t) (packet.estimated_generation() << 16U));
+
+    string error{};
+    if(!this->packet_encoder().acknowledge_manager().process_acknowledge(packet.type(), target_id, error))
+        debugMessage(this->virtual_server_id_, "{} Failed to handle acknowledge: {}", this->log_prefix(), error);
+}
+
+void VoiceClientConnection::handlePacketAckLow(const ts::protocol::ClientPacketParser &packet) {
+    this->handlePacketAck(packet);
+}
+
+void VoiceClientConnection::handlePacketCommand(ReassembledCommand* command) {
+    {
+        using CommandHandleResult = CryptSetupHandler::CommandHandleResult ;
+
+        auto result = this->crypt_setup_handler_.handle_command(command->command_view());
+        switch (result) {
+            case CommandHandleResult::PASS_THROUGH:
+                break;
+
+            case CommandHandleResult::CONSUME_COMMAND:
+                return;
+
+            case CommandHandleResult::CLOSE_CONNECTION:
+                auto client = this->getCurrentClient();
+                assert(client); /* FIXME! */
+                client->close_connection(std::chrono::system_clock::time_point{});
+                return;
+        }
+    }
+
+    auto client = this->getCurrentClient();
+    if(!client) {
+        /* TODO! */
+        return;
+    }
+
+    client->server_command_executor().enqueue_command_execution(command);
+}