amnezia-client/client/core/local-proxy/xraycontroller.cpp

#include "xraycontroller.h"
#include "proxylogger.h"
#include <QDir>
#include <QFileInfo>
#include <QCoreApplication>
#include <QDebug>

XrayController::XrayController(QObject *parent)
    : QObject(parent), m_process(nullptr)
{
    ProxyLogger::getInstance().debug("XrayController initialized");
}

XrayController::~XrayController()
{
    stop();
}

bool XrayController::start(const QString &configPath)
{
    if (isXrayRunning())
    {
        ProxyLogger::getInstance().info("Xray process is already running");
        return true;
    }

    QString xrayPath = getXrayExecutablePath();
    ProxyLogger::getInstance().debug(QString("Xray executable path: %1").arg(xrayPath));

    if (!QFile::exists(xrayPath))
    {
        ProxyLogger::getInstance().error(QString("Xray binary not found at: %1").arg(xrayPath));
        return false;
    }

    if (!QFile::exists(configPath))
    {
        ProxyLogger::getInstance().error(QString("Config file not found at: %1").arg(configPath));
        return false;
    }

    ProxyLogger::getInstance().info("Starting Xray process");
    m_process.reset(new QProcess(this));
    m_process->setWorkingDirectory(QFileInfo(xrayPath).dir().absolutePath());
    m_process->setProgram(xrayPath);
    m_process->setArguments(getXrayArguments(configPath));

    m_process->start();
    if (!m_process->waitForStarted())
    {
        ProxyLogger::getInstance().error(QString("Failed to start Xray process: %1").arg(m_process->errorString()));
        m_process.reset();
        return false;
    }

    ProxyLogger::getInstance().info(QString("Xray process started successfully (PID: %1)").arg(m_process->processId()));
    return true;
}

void XrayController::stop()
{
    if (!m_process.isNull())
    {
        if (m_process->state() == QProcess::Running)
        {
            ProxyLogger::getInstance().info(QString("Killing Xray process (PID: %1)").arg(m_process->processId()));
            m_process->kill();
            m_process->waitForFinished(3000);
        }
        m_process.reset();
        ProxyLogger::getInstance().info("Xray process stopped");
    }
}

bool XrayController::isXrayRunning() const
{
    return !m_process.isNull() && m_process->state() == QProcess::Running;
}

qint64 XrayController::getProcessId() const
{
    return m_process ? m_process->processId() : -1;
}

QString XrayController::getError() const
{
    if (m_process && m_process->error() != QProcess::UnknownError)
    {
        QString error = m_process->errorString();
        ProxyLogger::getInstance().error(QString("Xray process error: %1").arg(error));
        return error;
    }
    return QString();
}

QString XrayController::getXrayExecutablePath() const
{
    QString xrayDir = QCoreApplication::applicationDirPath() + "/bin";
    QString xrayPath;

#if defined(Q_OS_WIN)
    xrayPath = QDir(xrayDir).filePath("xray.exe");
#else
    xrayPath = QDir(xrayDir).filePath("xray");
#endif

    ProxyLogger::getInstance().debug(QString("Resolved Xray executable path: %1").arg(xrayPath));
    return xrayPath;
}

QStringList XrayController::getXrayArguments(const QString &configPath) const
{
    QStringList args = QStringList() << "-c" << configPath << "-format=json";
    ProxyLogger::getInstance().debug(QString("Xray arguments: %1").arg(args.join(' ')));
    return args;
}