Qv2ray/src/ui/NetSpeedBar/QvNetSpeedBar_linux.cpp

#ifdef __linux__
#include "QvNetSpeedPlugin.hpp"
#include "QvUtils.hpp"
#include <QLocalSocket>
#include <QLocalServer>
namespace Qv2ray
{
    namespace Utils
    {
        namespace NetSpeedPlugin
        {
            namespace _linux
            {
                static QThread *linuxWorkerThread;
                static QLocalServer *server;
                static QObjectMessageProxy *messageProxy;
                static bool isExiting = false;

                void qobject_proxy()
                {
                    QLocalSocket *socket = server->nextPendingConnection();

                    if (!socket->waitForConnected() || !socket->waitForReadyRead())
                        return;

                    try {
                        while (!isExiting && socket->isOpen() && socket->isValid() && socket->waitForReadyRead()) {
                            // CANNOT PROPERLY READ...
                            // Temp-ly fixed (but why and how?)
                            auto in = QString(socket->readAll());

                            if (!isExiting && !in.isEmpty()) {
                                auto out = GetAnswerToRequest(in);
                                //
                                socket->write(out.toUtf8());
                                socket->flush();
                            } else {
                                QThread::msleep(200);
                            }
                        }
                    }  catch (...) {
                        LOG(MODULE_PLUGIN, "Closing a broken socket.")
                    }
                }
                void DataMessageQThread()
                {
                    server = new QLocalServer();
                    // BUG Sometimes failed to listen due to improper close of last session.
                    bool listening = server->listen(QV2RAY_NETSPEED_PLUGIN_PIPE_NAME_LINUX);

                    while (!isExiting && !listening) {
                        QThread::msleep(500);
                        listening = server->listen(QV2RAY_NETSPEED_PLUGIN_PIPE_NAME_LINUX);
                    }

                    bool timeOut = false;
                    server->setSocketOptions(QLocalServer::WorldAccessOption);
                    messageProxy = new QObjectMessageProxy(&qobject_proxy);
                    QObject::connect(server, &QLocalServer::newConnection, messageProxy, &QObjectMessageProxy::processMessage);

                    while (!isExiting) {
                        bool result = server->waitForNewConnection(200, &timeOut);
                        LOG(MODULE_PLUGIN, "Plugin thread listening failed: " << server->errorString().toStdString())
                        LOG(MODULE_PLUGIN, "waitForNewConnection: " << (result ? "true" : "false") << ", " << (timeOut ? "true" : "false"))
                    }

                    server->close();
                }
                void StartMessageQThread()
                {
                    linuxWorkerThread = QThread::create(_linux::DataMessageQThread);
                    linuxWorkerThread->start();
                }

                void StopMessageQThread()
                {
                    isExiting = true;

                    if (linuxWorkerThread->isRunning()) {
                        LOG(MODULE_PLUGIN, "Waiting for linuxWorkerThread to stop.")
                        linuxWorkerThread->wait();
                    }

                    delete _linux::linuxWorkerThread;
                }

            }
        }
    }
}
#endif