Initial commit: wowee native WoW 3.3.5a client

src/network/tcp_socket.cpp

@@ -0,0 +1,227 @@
+#include "network/tcp_socket.hpp"
+#include "network/packet.hpp"
+#include "core/logger.hpp"
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <arpa/inet.h>
+#include <unistd.h>
+#include <fcntl.h>
+#include <netdb.h>
+#include <cstring>
+
+namespace wowee {
+namespace network {
+
+TCPSocket::TCPSocket() = default;
+
+TCPSocket::~TCPSocket() {
+    disconnect();
+}
+
+bool TCPSocket::connect(const std::string& host, uint16_t port) {
+    LOG_INFO("Connecting to ", host, ":", port);
+
+    // Create socket
+    sockfd = socket(AF_INET, SOCK_STREAM, 0);
+    if (sockfd < 0) {
+        LOG_ERROR("Failed to create socket");
+        return false;
+    }
+
+    // Set non-blocking
+    int flags = fcntl(sockfd, F_GETFL, 0);
+    fcntl(sockfd, F_SETFL, flags | O_NONBLOCK);
+
+    // Resolve host
+    struct hostent* server = gethostbyname(host.c_str());
+    if (server == nullptr) {
+        LOG_ERROR("Failed to resolve host: ", host);
+        close(sockfd);
+        sockfd = -1;
+        return false;
+    }
+
+    // Connect
+    struct sockaddr_in serverAddr;
+    memset(&serverAddr, 0, sizeof(serverAddr));
+    serverAddr.sin_family = AF_INET;
+    memcpy(&serverAddr.sin_addr.s_addr, server->h_addr, server->h_length);
+    serverAddr.sin_port = htons(port);
+
+    int result = ::connect(sockfd, (struct sockaddr*)&serverAddr, sizeof(serverAddr));
+    if (result < 0 && errno != EINPROGRESS) {
+        LOG_ERROR("Failed to connect: ", strerror(errno));
+        close(sockfd);
+        sockfd = -1;
+        return false;
+    }
+
+    connected = true;
+    LOG_INFO("Connected to ", host, ":", port);
+    return true;
+}
+
+void TCPSocket::disconnect() {
+    if (sockfd >= 0) {
+        close(sockfd);
+        sockfd = -1;
+    }
+    connected = false;
+    receiveBuffer.clear();
+}
+
+void TCPSocket::send(const Packet& packet) {
+    if (!connected) return;
+
+    // Build complete packet with opcode
+    std::vector<uint8_t> sendData;
+
+    // Add opcode (1 byte) - always little-endian, but it's just 1 byte so doesn't matter
+    sendData.push_back(static_cast<uint8_t>(packet.getOpcode() & 0xFF));
+
+    // Add packet data
+    const auto& data = packet.getData();
+    sendData.insert(sendData.end(), data.begin(), data.end());
+
+    LOG_DEBUG("Sending packet: opcode=0x", std::hex, packet.getOpcode(), std::dec,
+              " size=", sendData.size(), " bytes");
+
+    // Send complete packet
+    ssize_t sent = ::send(sockfd, sendData.data(), sendData.size(), 0);
+    if (sent < 0) {
+        LOG_ERROR("Send failed: ", strerror(errno));
+    } else if (static_cast<size_t>(sent) != sendData.size()) {
+        LOG_WARNING("Partial send: ", sent, " of ", sendData.size(), " bytes");
+    }
+}
+
+void TCPSocket::update() {
+    if (!connected) return;
+
+    // Receive data into buffer
+    uint8_t buffer[4096];
+    ssize_t received = recv(sockfd, buffer, sizeof(buffer), 0);
+
+    if (received > 0) {
+        LOG_DEBUG("Received ", received, " bytes from server");
+        receiveBuffer.insert(receiveBuffer.end(), buffer, buffer + received);
+
+        // Try to parse complete packets from buffer
+        tryParsePackets();
+    }
+    else if (received == 0) {
+        LOG_INFO("Connection closed by server");
+        disconnect();
+    }
+    else if (errno != EAGAIN && errno != EWOULDBLOCK) {
+        LOG_ERROR("Receive failed: ", strerror(errno));
+        disconnect();
+    }
+}
+
+void TCPSocket::tryParsePackets() {
+    // For auth packets, we need at least 1 byte (opcode)
+    while (receiveBuffer.size() >= 1) {
+        uint8_t opcode = receiveBuffer[0];
+
+        // Determine expected packet size based on opcode
+        // This is specific to authentication protocol
+        size_t expectedSize = getExpectedPacketSize(opcode);
+
+        if (expectedSize == 0) {
+            // Unknown opcode or need more data to determine size
+            LOG_WARNING("Unknown opcode or indeterminate size: 0x", std::hex, (int)opcode, std::dec);
+            break;
+        }
+
+        if (receiveBuffer.size() < expectedSize) {
+            // Not enough data yet
+            LOG_DEBUG("Waiting for more data: have ", receiveBuffer.size(),
+                     " bytes, need ", expectedSize);
+            break;
+        }
+
+        // We have a complete packet!
+        LOG_DEBUG("Parsing packet: opcode=0x", std::hex, (int)opcode, std::dec,
+                 " size=", expectedSize, " bytes");
+
+        // Create packet from buffer data
+        std::vector<uint8_t> packetData(receiveBuffer.begin(),
+                                        receiveBuffer.begin() + expectedSize);
+
+        Packet packet(opcode, packetData);
+
+        // Remove parsed data from buffer
+        receiveBuffer.erase(receiveBuffer.begin(), receiveBuffer.begin() + expectedSize);
+
+        // Call callback if set
+        if (packetCallback) {
+            packetCallback(packet);
+        }
+    }
+}
+
+size_t TCPSocket::getExpectedPacketSize(uint8_t opcode) {
+    // Authentication packet sizes (WoW 3.3.5a)
+    // Note: These are minimum sizes. Some packets are variable length.
+
+    switch (opcode) {
+        case 0x00:  // LOGON_CHALLENGE response
+            // Need to read second byte to determine success/failure
+            if (receiveBuffer.size() >= 3) {
+                uint8_t status = receiveBuffer[2];
+                if (status == 0x00) {
+                    // Success - need to calculate full size
+                    // Minimum: opcode(1) + unknown(1) + status(1) + B(32) + glen(1) + g(1) + Nlen(1) + N(32) + salt(32) + unk(16) + flags(1)
+                    // With typical values: 1 + 1 + 1 + 32 + 1 + 1 + 1 + 32 + 32 + 16 + 1 = 119 bytes minimum
+                    // But N is usually 256 bytes, so more like: 1 + 1 + 1 + 32 + 1 + 1 + 1 + 256 + 32 + 16 + 1 = 343 bytes
+
+                    // For safety, let's parse dynamically:
+                    if (receiveBuffer.size() >= 36) {  // enough to read g_len
+                        uint8_t gLen = receiveBuffer[35];
+                        size_t minSize = 36 + gLen + 1;  // up to N_len
+                        if (receiveBuffer.size() >= minSize) {
+                            uint8_t nLen = receiveBuffer[36 + gLen];
+                            size_t totalSize = 36 + gLen + 1 + nLen + 32 + 16 + 1;
+                            return totalSize;
+                        }
+                    }
+                    return 0;  // Need more data
+                } else {
+                    // Failure - just opcode + unknown + status
+                    return 3;
+                }
+            }
+            return 0;  // Need more data to determine
+
+        case 0x01:  // LOGON_PROOF response
+            // opcode(1) + status(1) + M2(20) = 22 bytes on success
+            // opcode(1) + status(1) = 2 bytes on failure
+            if (receiveBuffer.size() >= 2) {
+                uint8_t status = receiveBuffer[1];
+                if (status == 0x00) {
+                    return 22;  // Success
+                } else {
+                    return 2;   // Failure
+                }
+            }
+            return 0;  // Need more data
+
+        case 0x10:  // REALM_LIST response
+            // Variable length - format: opcode(1) + size(2) + payload(size)
+            // Need to read size field (little-endian uint16 at offset 1-2)
+            if (receiveBuffer.size() >= 3) {
+                uint16_t size = receiveBuffer[1] | (receiveBuffer[2] << 8);
+                // Total packet size is: opcode(1) + size field(2) + payload(size)
+                return 1 + 2 + size;
+            }
+            return 0;  // Need more data to read size field
+
+        default:
+            LOG_WARNING("Unknown auth packet opcode: 0x", std::hex, (int)opcode, std::dec);
+            return 0;
+    }
+}
+
+} // namespace network
+} // namespace wowee