mirror of
https://bitbucket.org/smil3y/kde-playground.git
synced 2025-02-23 18:32:51 +00:00
kdesktopspy: new application
Signed-off-by: Ivailo Monev <xakepa10@gmail.com>
This commit is contained in:
Ivailo Monev
parent
68c846e439
commit
64cea12813
2 changed files with 210 additions and 1 deletions
|
|
@ -22,4 +22,5 @@ macro_optional_add_subdirectory (kcachegrind)
|
|||
# macro_optional_add_subdirectory (kdepim-runtime)
|
||||
# macro_optional_add_subdirectory (kdepim)
|
||||
macro_optional_add_subdirectory (kman)
|
||||
macro_optional_add_subdirectory (zeroconf-ioslave)
|
||||
macro_optional_add_subdirectory (zeroconf-ioslave)
|
||||
macro_optional_add_subdirectory (kdestopspy)
|
||||
|
|
|
|||
208
kdesktopspy/kdesktopspy.cpp
Normal file
208
kdesktopspy/kdesktopspy.cpp
Normal file
|
|
@ -0,0 +1,208 @@
|
|||
/* This file is part of KMan
|
||||
Copyright (C) 2022 Ivailo Monev <xakepa10@gmail.com>
|
||||
|
||||
This library is free software; you can redistribute it and/or
|
||||
modify it under the terms of the GNU Library General Public
|
||||
License version 2, as published by the Free Software Foundation.
|
||||
|
||||
This library is distributed in the hope that it will be useful,
|
||||
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
||||
Library General Public License for more details.
|
||||
|
||||
You should have received a copy of the GNU Library General Public License
|
||||
along with this library; see the file COPYING.LIB. If not, write to
|
||||
the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
|
||||
Boston, MA 02110-1301, USA.
|
||||
*/
|
||||
|
||||
|
||||
#include <KAboutData>
|
||||
#include <KCmdLineArgs>
|
||||
#include <KApplication>
|
||||
#include <KUrl>
|
||||
#include <KDebug>
|
||||
#include <QTcpServer>
|
||||
#include <QTcpSocket>
|
||||
#include <QBuffer>
|
||||
#include <QDesktopWidget>
|
||||
|
||||
|
||||
class HttpHeaderParser
|
||||
{
|
||||
public:
|
||||
void parseHeader(const QByteArray &header);
|
||||
|
||||
QString path() const { return m_path; }
|
||||
|
||||
private:
|
||||
QString m_path;
|
||||
};
|
||||
|
||||
void HttpHeaderParser::parseHeader(const QByteArray &header)
|
||||
{
|
||||
bool firstline = true;
|
||||
foreach (const QByteArray &line, header.split('\n')) {
|
||||
if (line.isEmpty()) {
|
||||
firstline = false;
|
||||
continue;
|
||||
}
|
||||
if (firstline) {
|
||||
const QList<QByteArray> splitline = line.split(' ');
|
||||
if (splitline.size() == 3) {
|
||||
m_path = splitline.at(1).trimmed();
|
||||
}
|
||||
}
|
||||
firstline = false;
|
||||
}
|
||||
// qDebug() << Q_FUNC_INFO << m_path;
|
||||
}
|
||||
|
||||
|
||||
class KDesktopSpy : public QObject
|
||||
{
|
||||
Q_OBJECT
|
||||
public:
|
||||
KDesktopSpy(QObject *parent = nullptr);
|
||||
~KDesktopSpy();
|
||||
|
||||
bool start(const QString &host, int port);
|
||||
QString errorString() const;
|
||||
|
||||
public Q_SLOTS:
|
||||
void handleRequest();
|
||||
|
||||
private:
|
||||
QTcpServer m_tcpserver;
|
||||
};
|
||||
|
||||
KDesktopSpy::KDesktopSpy(QObject *parent)
|
||||
: QObject(parent)
|
||||
{
|
||||
connect(&m_tcpserver, SIGNAL(newConnection()), this, SLOT(handleRequest()));
|
||||
}
|
||||
|
||||
KDesktopSpy::~KDesktopSpy()
|
||||
{
|
||||
}
|
||||
|
||||
bool KDesktopSpy::start(const QString &host, int port)
|
||||
{
|
||||
QHostAddress address;
|
||||
address.setAddress(host);
|
||||
return m_tcpserver.listen(address, port);
|
||||
}
|
||||
|
||||
|
||||
QString KDesktopSpy::errorString() const
|
||||
{
|
||||
return m_tcpserver.errorString();
|
||||
}
|
||||
|
||||
void KDesktopSpy::handleRequest()
|
||||
{
|
||||
QTcpSocket *clientConnection = m_tcpserver.nextPendingConnection();
|
||||
|
||||
connect(clientConnection, SIGNAL(disconnected()),
|
||||
clientConnection, SLOT(deleteLater()));
|
||||
|
||||
if (!clientConnection->waitForReadyRead()) {
|
||||
clientConnection->disconnectFromHost();
|
||||
}
|
||||
|
||||
QByteArray request(clientConnection->readAll());
|
||||
// qDebug() << Q_FUNC_INFO << "request" << request;
|
||||
|
||||
HttpHeaderParser headerparser;
|
||||
headerparser.parseHeader(request);
|
||||
|
||||
QDesktopWidget* desktopwidget = qApp->desktop();
|
||||
|
||||
QByteArray block;
|
||||
if (desktopwidget && headerparser.path() == "/") {
|
||||
block.append("HTTP/1.1 200 OK\r\n");
|
||||
block.append(QString("Date: %1 GMT\r\n").arg(QDateTime(QDateTime::currentDateTime())
|
||||
.toString("ddd, dd MMM yyyy hh:mm:ss")).toAscii());
|
||||
block.append("Server: KDestopSpy\r\n");
|
||||
|
||||
QByteArray data;
|
||||
data.append(QString("<html><img src=\"image.png\" width=\"%1\" height=\"%2\"><meta http-equiv=\"refresh\" content=\"1\"></html>").arg(desktopwidget->width()).arg(desktopwidget->height()).toAscii());
|
||||
|
||||
block.append("Content-Type: text/html; charset=UTF-8\r\n");
|
||||
block.append(QString("Content-Length: %1\r\n\r\n").arg(data.length()).toAscii());
|
||||
block.append(data);
|
||||
} else if (desktopwidget && headerparser.path() == "/image.png") {
|
||||
block.append("HTTP/1.1 200 OK\r\n");
|
||||
block.append(QString("Date: %1 GMT\r\n").arg(QDateTime(QDateTime::currentDateTime())
|
||||
.toString("ddd, dd MMM yyyy hh:mm:ss")).toAscii());
|
||||
block.append("Server: KDestopSpy\r\n");
|
||||
|
||||
// FIXME: grabbing QDesktopWidget does not work?
|
||||
// const QPixmap desktoppixmap = QPixmap::grabWidget((QWidget*)desktopwidget);
|
||||
const QPixmap desktoppixmap = QPixmap::fromX11Pixmap(QX11Info::appRootWindow());
|
||||
QBuffer desktopbuffer;
|
||||
desktopbuffer.open(QBuffer::ReadWrite);
|
||||
// desktoppixmap.save("/tmp/image.png", "PNG");
|
||||
if (!desktoppixmap.save(&desktopbuffer, "PNG")) {
|
||||
kWarning() << "could not save image";
|
||||
}
|
||||
const QByteArray data = desktopbuffer.data();
|
||||
|
||||
block.append("Content-Type: image/png\r\n");
|
||||
block.append(QString("Content-Length: %1\r\n\r\n").arg(data.length()).toAscii());
|
||||
block.append(data);
|
||||
} else {
|
||||
block.append("HTTP/1.1 404 Not Found\r\n");
|
||||
block.append(QString("Date: %1 GMT\r\n").arg(QDateTime(QDateTime::currentDateTime())
|
||||
.toString("ddd, dd MMM yyyy hh:mm:ss")).toAscii());
|
||||
block.append("Server: KDestopSpy\r\n");
|
||||
|
||||
QByteArray data("<html>404 Not Found</html>");
|
||||
|
||||
block.append("Content-Type: text/html; charset=UTF-8\r\n");
|
||||
block.append(QString("Content-Length: %1\r\n\r\n").arg(data.length()).toAscii());
|
||||
block.append("\r\n");
|
||||
block.append(data);
|
||||
}
|
||||
|
||||
clientConnection->write(block);
|
||||
clientConnection->disconnectFromHost();
|
||||
|
||||
}
|
||||
|
||||
int main(int argc, char** argv)
|
||||
{
|
||||
KAboutData aboutData("kdesktopspy", 0, ki18n("KDesktopSpy"),
|
||||
"1.0.0", ki18n("Spy desktop over network."),
|
||||
KAboutData::License_GPL_V2,
|
||||
ki18n("(c) 2022 Ivailo Monev"),
|
||||
KLocalizedString(),
|
||||
"http://github.com/fluxer/katana"
|
||||
);
|
||||
|
||||
aboutData.addAuthor(ki18n("Ivailo Monev"),
|
||||
ki18n("Maintainer"),
|
||||
"xakepa10@gmail.com");
|
||||
aboutData.setProgramIconName(QLatin1String("krfb"));
|
||||
|
||||
KCmdLineArgs::init(argc, argv, &aboutData);
|
||||
KCmdLineOptions option;
|
||||
option.add("address <address>", ki18n("Address to use, default value"), QHostAddress(QHostAddress::Any).toString().toAscii());
|
||||
option.add("port <port>", ki18n("Port to use, default value"), "8080");
|
||||
KCmdLineArgs::addCmdLineOptions(option);
|
||||
|
||||
KCmdLineArgs *args = KCmdLineArgs::parsedArgs();
|
||||
|
||||
KApplication *kdestopspyapp = new KApplication();
|
||||
|
||||
KDesktopSpy kdesktopspy(kdestopspyapp);
|
||||
if (!kdesktopspy.start(args->getOption("address"), args->getOption("port").toInt())) {
|
||||
kWarning() << kdesktopspy.errorString();
|
||||
return 1;
|
||||
}
|
||||
qDebug() << "Server running on" << args->getOption("address") << args->getOption("port");
|
||||
|
||||
return kdestopspyapp->exec();
|
||||
}
|
||||
|
||||
#include "kdesktopspy.moc"
|
||||
Loading…
Add table
Reference in a new issue