From abc348e5e1545984dc752be26016ec2cfffdc8cb Mon Sep 17 00:00:00 2001
From: w1146869587 <wah1146869587@126.com>
Date: 星期四, 04 十一月 2021 09:57:46 +0800
Subject: [PATCH] daseam

---
 mbwatch/main.cpp |  149 ++++++++++++++++++++++++++++++++++++++++++++++++-
 1 files changed, 146 insertions(+), 3 deletions(-)

diff --git a/mbwatch/main.cpp b/mbwatch/main.cpp
index 470a030..ea8bd25 100644
--- a/mbwatch/main.cpp
+++ b/mbwatch/main.cpp
@@ -1,8 +1,151 @@
 #include <QCoreApplication>
+#include <QTcpServer>
+#include <QTcpSocket>
+#include <QTextStream>
+#include <QDateTime>
+#include <QStringList>
+#include <QDir>
+#include <QSettings>
 
-int main(int argc, char *argv[])
+#include "qtservice.h"
+
+// HttpDaemon is the the class that implements the simple HTTP server.
+class HttpDaemon : public QTcpServer
 {
-    QCoreApplication a(argc, argv);
+    Q_OBJECT
+public:
+    HttpDaemon(quint16 port, QObject* parent = 0)
+        : QTcpServer(parent), disabled(false)
+    {
+        listen(QHostAddress::Any, port);
+    }
 
-    return a.exec();
+    void incomingConnection(qintptr socket)
+    {
+        if (disabled)
+            return;
+
+        // When a new client connects, the server constructs a QTcpSocket and all
+        // communication with the client is done over this QTcpSocket. QTcpSocket
+        // works asynchronously, this means that all the communication is done
+        // in the two slots readClient() and discardClient().
+        QTcpSocket* s = new QTcpSocket(this);
+        connect(s, SIGNAL(readyRead()), this, SLOT(readClient()));
+        connect(s, SIGNAL(disconnected()), this, SLOT(discardClient()));
+        s->setSocketDescriptor(socket);
+
+        QtServiceBase::instance()->logMessage("New Connection");
+    }
+
+    void pause()
+    {
+        disabled = true;
+    }
+
+    void resume()
+    {
+        disabled = false;
+    }
+
+private slots:
+    void readClient()
+    {
+        if (disabled)
+            return;
+
+        // This slot is called when the client sent data to the server. The
+        // server looks if it was a get request and sends a very simple HTML
+        // document back.
+        QTcpSocket* socket = (QTcpSocket*)sender();
+        if (socket->canReadLine()) {
+            QStringList tokens = QString(socket->readLine()).split(QRegExp("[ \r\n][ \r\n]*"));
+            if (tokens[0] == "GET") {
+                QTextStream os(socket);
+                os.setAutoDetectUnicode(true);
+                os << "HTTP/1.0 200 Ok\r\n"
+                    "Content-Type: text/html; charset=\"utf-8\"\r\n"
+                    "\r\n"
+                    "<h1>Nothing to see here</h1>\n"
+                    << QDateTime::currentDateTime().toString() << "\n";
+                socket->close();
+
+                QtServiceBase::instance()->logMessage("Wrote to client");
+
+                if (socket->state() == QTcpSocket::UnconnectedState) {
+                    delete socket;
+                    QtServiceBase::instance()->logMessage("Connection closed");
+                }
+            }
+        }
+    }
+    void discardClient()
+    {
+        QTcpSocket* socket = (QTcpSocket*)sender();
+        socket->deleteLater();
+
+        QtServiceBase::instance()->logMessage("Connection closed");
+    }
+
+private:
+    bool disabled;
+};
+
+class HttpService : public QtService<QCoreApplication>
+{
+public:
+    HttpService(int argc, char **argv)
+    : QtService<QCoreApplication>(argc, argv, "Qt HTTP Daemon")
+    {
+        setServiceDescription("A dummy HTTP service implemented with Qt");
+        setServiceFlags(QtServiceBase::CanBeSuspended);
+    }
+
+protected:
+    void start()
+    {
+        QCoreApplication *app = application();
+
+#if QT_VERSION < 0x040100
+        quint16 port = (app->argc() > 1) ?
+                QString::fromLocal8Bit(app->argv()[1]).toUShort() : 8888;
+#else
+        const QStringList arguments = QCoreApplication::arguments();
+        quint16 port = (arguments.size() > 1) ?
+                arguments.at(1).toUShort() : 8888;
+#endif
+        port = 8888;
+        daemon = new HttpDaemon(port, app);
+
+        if (!daemon->isListening()) {
+            logMessage(QString("Failed to bind to port %1").arg(daemon->serverPort()), QtServiceBase::Error);
+            app->quit();
+        }
+    }
+
+    void pause()
+    {
+    daemon->pause();
+    }
+
+    void resume()
+    {
+    daemon->resume();
+    }
+
+private:
+    HttpDaemon *daemon;
+};
+
+#include "main.moc"
+
+int main(int argc, char **argv)
+{
+#if !defined(Q_OS_WIN)
+    // QtService stores service settings in SystemScope, which normally require root privileges.
+    // To allow testing this example as non-root, we change the directory of the SystemScope settings file.
+    QSettings::setPath(QSettings::NativeFormat, QSettings::SystemScope, QDir::tempPath());
+    qWarning("(Example uses dummy settings file: %s/QtSoftware.conf)", QDir::tempPath().toLatin1().constData());
+#endif
+    HttpService service(argc, argv);
+    return service.exec();
 }

--
Gitblit v1.9.1