mirror of
https://bitbucket.org/smil3y/kde-playground.git
synced 2025-02-24 02:42:51 +00:00
301 lines
10 KiB
C++
301 lines
10 KiB
C++
/* This file is part of KHTTPD
|
|
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 <KIcon>
|
|
#include <KMimeType>
|
|
#include <KDebug>
|
|
#include <QBuffer>
|
|
#include <QDir>
|
|
#include <QTcpServer>
|
|
#include <QTcpSocket>
|
|
#include <QHostInfo>
|
|
#include <DNSSD/PublicService>
|
|
|
|
static QByteArray contentForDirectory(const QString &path, const QString &basedir)
|
|
{
|
|
QByteArray data;
|
|
data.append("<html>");
|
|
data.append("<table>");
|
|
data.append(" <tr>");
|
|
data.append(" <th></th>"); // icon
|
|
data.append(" <th>Filename</th>");
|
|
data.append(" <th>MIME</th>");
|
|
data.append(" <th>Size</th>");
|
|
data.append(" </tr>");
|
|
QDir::Filters dirfilters = (QDir::Files | QDir::AllDirs | QDir::NoDot);
|
|
if (QDir::cleanPath(path) == QDir::cleanPath(basedir)) {
|
|
dirfilters = (QDir::Files | QDir::AllDirs | QDir::NoDotAndDotDot);
|
|
}
|
|
QDir::SortFlags dirsortflags = (QDir::Name | QDir::DirsFirst);
|
|
QDir dir(path);
|
|
foreach (const QFileInfo &fileinfo, dir.entryInfoList(dirfilters, dirsortflags)) {
|
|
const QString fullpath = path.toLocal8Bit() + QLatin1Char('/') + fileinfo.fileName();
|
|
// chromium does weird stuff if the link starts with two slashes - removes, the host and
|
|
// port part of the link and converts the first directory to lower-case
|
|
const QString cleanpath = QDir::cleanPath(fullpath.mid(basedir.size()));
|
|
|
|
data.append(" <tr>");
|
|
|
|
const bool isdotdot = (fileinfo.fileName() == QLatin1String(".."));
|
|
if (isdotdot) {
|
|
const QString fileicon = QString::fromLatin1("<img src=\"/khttpd_icons/go-previous\" width=\"20\" height=\"20\">");
|
|
data.append("<td>");
|
|
data.append(fileicon.toAscii());
|
|
data.append("</td>");
|
|
} else {
|
|
const QString fileicon = QString::fromLatin1("<img src=\"/khttpd_icons/%1\" width=\"20\" height=\"20\">").arg(KMimeType::iconNameForUrl(KUrl(fullpath)));
|
|
data.append("<td>");
|
|
data.append(fileicon.toAscii());
|
|
data.append("</td>");
|
|
}
|
|
|
|
// qDebug() << Q_FUNC_INFO << fullpath << basedir << cleanpath;
|
|
data.append("<td><a href=\"");
|
|
data.append(cleanpath.toLocal8Bit());
|
|
data.append("\">");
|
|
data.append(fileinfo.fileName().toLocal8Bit());
|
|
data.append("</a><br></td>");
|
|
|
|
data.append("<td>");
|
|
if (!isdotdot) {
|
|
const QString filemime = KMimeType::findByPath(fullpath)->name();
|
|
data.append(filemime.toAscii());
|
|
}
|
|
data.append("</td>");
|
|
|
|
data.append("<td>");
|
|
if (fileinfo.isFile()) {
|
|
const QString filesize = KGlobal::locale()->formatByteSize(fileinfo.size(), 1);
|
|
data.append(filesize.toAscii());
|
|
}
|
|
data.append("</td>");
|
|
|
|
data.append(" </tr>");
|
|
}
|
|
data.append("</table>");
|
|
data.append("</html>");
|
|
return data;
|
|
}
|
|
|
|
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 KHTTPD : public QObject
|
|
{
|
|
Q_OBJECT
|
|
public:
|
|
KHTTPD(QObject *parent = nullptr);
|
|
~KHTTPD();
|
|
|
|
bool start(const QString &host, int port, const QString &directory);
|
|
QString errorString() const;
|
|
|
|
public Q_SLOTS:
|
|
void handleRequest();
|
|
|
|
private:
|
|
QTcpServer m_tcpserver;
|
|
QString m_directory;
|
|
DNSSD::PublicService *m_publicservice;
|
|
};
|
|
|
|
KHTTPD::KHTTPD(QObject *parent)
|
|
: QObject(parent),
|
|
m_publicservice(nullptr)
|
|
{
|
|
connect(&m_tcpserver, SIGNAL(newConnection()), this, SLOT(handleRequest()));
|
|
}
|
|
|
|
KHTTPD::~KHTTPD()
|
|
{
|
|
if (m_publicservice) {
|
|
m_publicservice->stop();
|
|
delete m_publicservice;
|
|
}
|
|
}
|
|
|
|
bool KHTTPD::start(const QString &host, int port, const QString &directory)
|
|
{
|
|
m_publicservice = new DNSSD::PublicService(
|
|
i18n("KHTTPD@%1", QHostInfo::localHostName()),
|
|
"_http._tcp", port
|
|
);
|
|
m_publicservice->publishAsync();
|
|
m_directory = directory;
|
|
QHostAddress address;
|
|
address.setAddress(host);
|
|
return m_tcpserver.listen(address, port);
|
|
}
|
|
|
|
|
|
QString KHTTPD::errorString() const
|
|
{
|
|
return m_tcpserver.errorString();
|
|
}
|
|
|
|
void KHTTPD::handleRequest()
|
|
{
|
|
QTcpSocket *clientConnection = m_tcpserver.nextPendingConnection();
|
|
|
|
connect(clientConnection, SIGNAL(disconnected()),
|
|
clientConnection, SLOT(deleteLater()));
|
|
|
|
if (!clientConnection->waitForReadyRead()) {
|
|
clientConnection->disconnectFromHost();
|
|
}
|
|
|
|
const QByteArray request(clientConnection->readAll());
|
|
// qDebug() << Q_FUNC_INFO << "request" << request;
|
|
|
|
HttpHeaderParser headerparser;
|
|
headerparser.parseHeader(request);
|
|
|
|
QFileInfo pathinfo(m_directory + QLatin1Char('/') + headerparser.path());
|
|
// qDebug() << Q_FUNC_INFO << headerparser.path() << pathinfo.filePath();
|
|
const bool isdirectory = pathinfo.isDir();
|
|
const bool isfile = pathinfo.isFile();
|
|
QByteArray block;
|
|
if (headerparser.path().startsWith(QLatin1String("/khttpd_icons/"))) {
|
|
block.append("HTTP/1.1 200 OK\r\n");
|
|
block.append(QString::fromLatin1("Date: %1 GMT\r\n").arg(QDateTime(QDateTime::currentDateTime())
|
|
.toString("ddd, dd MMM yyyy hh:mm:ss")).toAscii());
|
|
block.append("Server: KHTTPD\r\n");
|
|
|
|
const QPixmap iconpixmap = KIcon(headerparser.path().mid(14)).pixmap(20);
|
|
QBuffer iconbuffer;
|
|
iconbuffer.open(QBuffer::ReadWrite);
|
|
if (!iconpixmap.save(&iconbuffer, "PNG")) {
|
|
kWarning() << "could not save image";
|
|
}
|
|
const QByteArray data = iconbuffer.data();
|
|
|
|
block.append("Content-Type: image/png\r\n");
|
|
block.append(QString::fromLatin1("Content-Length: %1\r\n\r\n").arg(data.length()).toAscii());
|
|
block.append(data);
|
|
} else if (isdirectory) {
|
|
block.append("HTTP/1.1 200 OK\r\n");
|
|
block.append(QString::fromLatin1("Date: %1 GMT\r\n").arg(QDateTime(QDateTime::currentDateTime())
|
|
.toString("ddd, dd MMM yyyy hh:mm:ss")).toAscii());
|
|
block.append("Server: KHTTPD\r\n");
|
|
|
|
QByteArray data = contentForDirectory(pathinfo.filePath(), m_directory);
|
|
|
|
block.append("Content-Type: text/html; charset=UTF-8\r\n");
|
|
block.append(QString::fromLatin1("Content-Length: %1\r\n\r\n").arg(data.length()).toAscii());
|
|
block.append(data);
|
|
} else if (isfile) {
|
|
block.append("HTTP/1.1 200 OK\r\n");
|
|
block.append(QString::fromLatin1("Date: %1 GMT\r\n").arg(QDateTime(QDateTime::currentDateTime())
|
|
.toString("ddd, dd MMM yyyy hh:mm:ss")).toAscii());
|
|
block.append("Server: KHTTPD\r\n");
|
|
|
|
QFile datafile(pathinfo.filePath());
|
|
datafile.open(QFile::ReadOnly);
|
|
const QByteArray data = datafile.readAll();
|
|
|
|
const QString filemime = KMimeType::findByPath(pathinfo.filePath())->name();
|
|
block.append(QString::fromLatin1("Content-Type: %1; charset=UTF-8\r\n").arg(filemime).toAscii());
|
|
block.append(QString::fromLatin1("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::fromLatin1("Date: %1 GMT\r\n").arg(QDateTime(QDateTime::currentDateTime())
|
|
.toString("ddd, dd MMM yyyy hh:mm:ss")).toAscii());
|
|
block.append("Server: KHTTPD\r\n");
|
|
|
|
const QByteArray data("<html>404 Not Found</html>");
|
|
|
|
block.append("Content-Type: text/html; charset=UTF-8\r\n");
|
|
block.append(QString::fromLatin1("Content-Length: %1\r\n\r\n").arg(data.length()).toAscii());
|
|
block.append(data);
|
|
}
|
|
|
|
clientConnection->write(block);
|
|
clientConnection->disconnectFromHost();
|
|
|
|
}
|
|
|
|
int main(int argc, char** argv)
|
|
{
|
|
KAboutData aboutData(
|
|
"khttpd", 0, ki18n("KHTTPD"),
|
|
"1.0.0", ki18n("Serve directory and publish it on service discovery."),
|
|
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("network-server"));
|
|
|
|
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");
|
|
option.add("directory <directory>", ki18n("Directory to use, default value"), QDir::currentPath().toAscii());
|
|
KCmdLineArgs::addCmdLineOptions(option);
|
|
|
|
KCmdLineArgs *args = KCmdLineArgs::parsedArgs();
|
|
|
|
KApplication *khttpdapp = new KApplication();
|
|
|
|
KHTTPD khttpd(khttpdapp);
|
|
if (!khttpd.start(args->getOption("address"), args->getOption("port").toInt(), args->getOption("directory"))) {
|
|
kWarning() << khttpd.errorString();
|
|
return 1;
|
|
}
|
|
qDebug() << "Server running on" << args->getOption("address") << args->getOption("port") << args->getOption("directory");
|
|
|
|
return khttpdapp->exec();
|
|
}
|
|
|
|
#include "khttpd.moc"
|