/* This file is part of Flowee
*
* Copyright (C) 2017 Nathan Osman
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Library General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* 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.
*
* For the full copy of the License see
*/
#include
#include
#include
#include
#include "filesystemhandler.h"
#include "qiodevicecopier.h"
#include "range.h"
#include "socket.h"
#include "filesystemhandler_p.h"
using namespace HttpEngine;
// Template for listing directory contents
const QString ListTemplate =
""
""
""
""
"%1"
""
""
"%1
"
"Directory listing:
"
""
"
"
"Flowee HttpEngine %3
"
""
"";
FilesystemHandlerPrivate::FilesystemHandlerPrivate(FilesystemHandler *handler)
: QObject(handler)
{
}
bool FilesystemHandlerPrivate::absolutePath(const QString &path, QString &absolutePath)
{
// Resolve the path according to the document root
absolutePath = documentRoot.absoluteFilePath(path);
// Perhaps not the most efficient way of doing things, but one way to
// determine if path is within the document root is to convert it to a
// relative path and check to see if it begins with "../" (it shouldn't)
return documentRoot.exists(absolutePath) && !documentRoot.relativeFilePath(path).startsWith("../");
}
QByteArray FilesystemHandlerPrivate::mimeType(const QString &absolutePath)
{
// Query the MIME database based on the filename and its contents
return database.mimeTypeForFile(absolutePath).name().toUtf8();
}
void FilesystemHandlerPrivate::processFile(Socket *socket, const QString &absolutePath)
{
// Attempt to open the file for reading
QFile *file = new QFile(absolutePath);
if (!file->open(QIODevice::ReadOnly)) {
delete file;
socket->writeError(Socket::Forbidden);
return;
}
// Create a QIODeviceCopier to copy the file contents to the socket
QIODeviceCopier *copier = new QIODeviceCopier(file, socket);
connect(copier, &QIODeviceCopier::finished, copier, &QIODeviceCopier::deleteLater);
connect(copier, &QIODeviceCopier::finished, file, &QFile::deleteLater);
connect(copier, &QIODeviceCopier::finished, [socket]() {
socket->close();
});
// Stop the copier if the socket is disconnected
connect(socket, &Socket::disconnected, copier, &QIODeviceCopier::stop);
qint64 fileSize = file->size();
// Checking for partial content request
QByteArray rangeHeader = socket->headers().value("Range");
Range range;
if (!rangeHeader.isEmpty() && rangeHeader.startsWith("bytes=")) {
// Skiping 'bytes=' - first 6 chars and spliting ranges by comma
QList rangeList = rangeHeader.mid(6).split(',');
// Taking only first range, as multiple ranges require multipart
// reply support
range = Range(QString(rangeList.at(0)), fileSize);
}
// If range is valid, send partial content
if (range.isValid()) {
socket->setStatusCode(Socket::PartialContent);
socket->setHeader("Content-Length", QByteArray::number(range.length()));
socket->setHeader("Content-Range", QByteArray("bytes ") + range.contentRange().toLatin1());
copier->setRange(range.from(), range.to());
} else {
// If range is invalid or if it is not a partial content request,
// send full file
socket->setHeader("Content-Length", QByteArray::number(fileSize));
}
// Set the mimetype and content length
socket->setHeader("Content-Type", mimeType(absolutePath));
socket->writeHeaders();
// Start the copy
copier->start();
}
void FilesystemHandlerPrivate::processDirectory(Socket *socket, const QString &path, const QString &absolutePath)
{
// Add entries for each of the files
QString listing;
foreach (QFileInfo info, QDir(absolutePath).entryInfoList(QDir::NoFilter, QDir::Name | QDir::DirsFirst | QDir::IgnoreCase)) {
listing.append(QString("%1%2")
.arg(info.fileName().toHtmlEscaped())
.arg(info.isDir() ? "/" : ""));
}
// Build the response and convert the string to UTF-8
QByteArray data = ListTemplate
.arg("/" + path.toHtmlEscaped())
.arg(listing)
.arg(HTTPENGINE_VERSION)
.toUtf8();
// Set the headers and write the content
socket->setHeader("Content-Type", "text/html");
socket->setHeader("Content-Length", QByteArray::number(data.length()));
socket->write(data);
socket->close();
}
FilesystemHandler::FilesystemHandler(QObject *parent)
: Handler(parent),
d(new FilesystemHandlerPrivate(this))
{
}
FilesystemHandler::FilesystemHandler(const QString &documentRoot, QObject *parent)
: Handler(parent),
d(new FilesystemHandlerPrivate(this))
{
setDocumentRoot(documentRoot);
}
void FilesystemHandler::setDocumentRoot(const QString &documentRoot)
{
d->documentRoot.setPath(documentRoot);
}
void FilesystemHandler::process(Socket *socket, const QString &path)
{
// If a document root is not set, an error has occurred
if (d->documentRoot.path().isNull()) {
socket->writeError(Socket::InternalServerError);
return;
}
// URL-decode the path
QString decodedPath = QUrl::fromPercentEncoding(path.toUtf8());
// Attempt to retrieve the absolute path
QString absolutePath;
if (!d->absolutePath(decodedPath, absolutePath)) {
socket->writeError(Socket::NotFound);
return;
}
if (QFileInfo(absolutePath).isDir()) {
d->processDirectory(socket, decodedPath, absolutePath);
} else {
d->processFile(socket, absolutePath);
}
}