VFS_tcp_export.cpp

Go to the documentation of this file.

 
#include <QJsonDocument>
#include <QJsonObject>
 
#include "VFS_tcp_export.h"
#include "VFS_tcp_export_client.h"
#include "VFS_base/VFS_icons.h"
#include "utilities/rutils.h"
 
#include "VFS.h"
 
VFS_tcp_export::VFS_tcp_export(QString exportRoot, quint16 port, QHostAddress address,  bool ssl, QString sslCertPath, QString sslKeyPath, QByteArray icon, QString broadcastName, QString broadcastAddress, QString multicastAddress, quint16 multicastPort, QJsonArray tags)
: VFS_tcp_server(port,address,ssl,sslCertPath,sslKeyPath)
, _exportRoot(exportRoot)
, _broadcastid(-1)
, _broadcastName(broadcastName)
, _broadcastAddress(broadcastAddress)
, _multicastPort(multicastPort)
, _multicastAddress(multicastAddress)
, _tags(tags)
, _icon(icon)
{
        _exportRoot = rutils::cleanPath(_exportRoot);
 
        if (_multicastPort)
        {
                if (_broadcastAddress == "")
                {
                        VFS::ERROR("A multicast port was specified, but a broadcastAddress was not.  Will not broadcast.",0,className());
                        _multicastPort = 0;
                }
                else
                        _broadcastSocket.bind(QHostAddress(QHostAddress::AnyIPv4), 0);
        }
}
 
VFS_tcp_export::~VFS_tcp_export()
{
}
 
QByteArray VFS_tcp_export::icon()
{
        return VFS_icons::get("unknown");
}
 
 
QString VFS_tcp_export::reportDetails()
{
        return "export: "+_exportRoot+"\n"+VFS_tcp_server::reportDetails();
}
 
 
VFS_tcp_client *VFS_tcp_export::newClient(QTcpSocket *socket, bool ssl, QSslConfiguration sslConfiguration)
{
        VFS_tcp_client *client = new VFS_tcp_export_client(socket, _exportRoot, ssl, sslConfiguration, _icon );
 
        if (_broadcastid != -1)
        {
                killTimer(_broadcastid);
                _broadcastid = -1;
                VFS::LOG( QString("Mount received.  Stopping broadcast on '%1'...").arg(_exportRoot), 9, className() );
 
                connect(client,SIGNAL(finished()),this,SLOT(startBroadcast()));
        }
 
        return client;
}
 
 
void VFS_tcp_export::clientMessage(QByteArray message)
{
        Q_UNUSED(message)
 
        //VFS::WARN("clientMessage() should be implemented in VFS_tcp_export subclasses");
}
 
 
bool VFS_tcp_export::listen()
{
        bool l = VFS_tcp_server::listen();
 
        if (l) startBroadcast();
 
        return l;
}
 
 
void VFS_tcp_export::timerEvent(QTimerEvent *event)
{
        if (event->timerId() == _broadcastid)
                sendBroadcast();
        else
                VFS::WARN("Unknown timer event.",0,className());
}
 
 
void VFS_tcp_export::startBroadcast()
{
        if (_multicastPort && _broadcastid==-1)
        {
                //https://docs.oracle.com/cd/E23824_01/html/821-1602/sockets-137.html
                _broadcastSocket.setSocketOption(QAbstractSocket::MulticastTtlOption, 35);
 
                _broadcastid = startTimer(3000);
                sendBroadcast();
        }
}
 
 
void VFS_tcp_export::sendBroadcast()
{
        VFS::LOG( QString("Broadcasting availability of '%1' on UDP %2:%3...").arg(_exportRoot).arg(_multicastAddress.toString()).arg(_multicastPort), 9, className() );
 
        QJsonObject b;
        b["name"] = _broadcastName;
        b["address"] = _broadcastAddress;
        b["port"] = _port;
        b["ssl"] = _ssl;
        b["tags"] = _tags;
 
        QByteArray datagram = QJsonDocument(b).toJson();
        _broadcastSocket.writeDatagram(datagram, _multicastAddress, _multicastPort);
}