/* Copyright (C) 2022 The Qt Company Ltd.
 *
 * SPDX-License-Identifier: GPL-3.0-only WITH Qt-GPL-exception-1.0
*/

#include "tcpclient.h"

namespace QLicenseService {

TcpClient::TcpClient(const std::string &connAddr, uint16_t port)
{
#ifdef _WIN32
    // Initialize Winsock
    WSADATA wsaData;
    int iResult = 0;
    iResult = WSAStartup(MAKEWORD(2, 2), &wsaData);
    if (iResult != NO_ERROR) {
        wprintf(L"Error at WSAStartup()\n");
        exit(EXIT_FAILURE);
    }
    const int inetSuccess = inet_pton(AF_INET, connAddr.c_str(), &m_server.sin_addr.s_addr);
#else
    const int inetSuccess = inet_aton(connAddr.c_str(), &m_server.sin_addr);
#endif
    if (!inetSuccess) {
        // Try to resolve hostname
        struct hostent *host;
        struct in_addr **addrList;
        if ( (host = gethostbyname(connAddr.c_str()) ) == nullptr){
            std::cout << "Failed to resolve hostname\n";
            exit(EXIT_FAILURE);
        }
        addrList = (struct in_addr **) host->h_addr_list;
        m_server.sin_addr = *addrList[0];
    }
    m_server.sin_port = htons(port);
}

int TcpClient::sendAndReceive(const std::string &message, std::string &reply)
{
    m_server.sin_family = AF_INET;
    m_socketFD = (socket(AF_INET , SOCK_STREAM, 0));
    if (m_socketFD < 0) {
        doCloseSocket();
        return e_tcp_fail_socket;
    }

    // Connect to daemon
    int connectR = connect(m_socketFD, (struct sockaddr *)&m_server, sizeof(m_server));
    if (connectR == -1) {
        doCloseSocket();
        return e_tcp_error_conn;
    }
    const size_t bytesSent = send(m_socketFD, message.c_str(), (int)message.length(), 0);

    if (bytesSent < 0 ) { // send failed
        doCloseSocket();
        return e_tcp_error_send;
    }

    std::string resp(BUFFER_SIZE, ' ');
    size_t bytes_recv;
    try {
        bytes_recv = recv(m_socketFD, &resp.front(), (int)resp.size(), 0);
    }
    catch (...) {
        std::cout << "Error receiving" << std::endl;
        doCloseSocket();
        return e_tcp_error_recv;
    }
    if (bytes_recv == -1) {
        doCloseSocket();
        return e_tcp_error_recv;
    }
    doCloseSocket();
    reply = resp.substr(0, bytes_recv); // Had to cut it like this for windows
    return e_tcp_success;
}

} // namespace QLicenseService