kdeconnect-kde/kded/backends/lan/landevicelink.cpp

/**
 * Copyright 2013 Albert Vaca <albertvaka@gmail.com>
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License as
 * published by the Free Software Foundation; either version 2 of
 * the License or (at your option) version 3 or any later version
 * accepted by the membership of KDE e.V. (or its successor approved
 * by the membership of KDE e.V.), which shall act as a proxy
 * defined in Section 14 of version 3 of the license.
 *
 * This program 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "landevicelink.h"

#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/tcp.h>
#include <netdb.h>

#include "../linkprovider.h"
#include "uploadjob.h"
#include "downloadjob.h"

LanDeviceLink::LanDeviceLink(const QString& d, LinkProvider* a, QTcpSocket* socket)
    : DeviceLink(d, a)
{
    mSocket = socket;

    int fd = socket->socketDescriptor();
    int enableKeepAlive = 1;
    setsockopt(fd, SOL_SOCKET, SO_KEEPALIVE, &enableKeepAlive, sizeof(enableKeepAlive));

    int maxIdle = 60; /* seconds */
    setsockopt(fd, IPPROTO_TCP, TCP_KEEPIDLE, &maxIdle, sizeof(maxIdle));

    int count = 3;  // send up to 3 keepalive packets out, then disconnect if no response
    setsockopt(fd, getprotobyname("TCP")->p_proto, TCP_KEEPCNT, &count, sizeof(count));

    int interval = 5;   // send a keepalive packet out every 2 seconds (after the 5 second idle period)
    setsockopt(fd, getprotobyname("TCP")->p_proto, TCP_KEEPINTVL, &interval, sizeof(interval));

    connect(mSocket, SIGNAL(disconnected()),
            this, SLOT(deleteLater()));
    connect(mSocket, SIGNAL(readyRead()),
            this, SLOT(dataReceived()));
}

bool LanDeviceLink::sendPackageEncrypted(QCA::PublicKey& key, NetworkPackage& np)
{
    if (np.hasPayload()) {
         UploadJob* job = new UploadJob(np.payload());
         job->start();
         np.setPayloadTransferInfo(job->getTransferInfo());
    }

    np.encrypt(key);

    int written = mSocket->write(np.serialize());
    return (written != -1);
}

bool LanDeviceLink::sendPackage(NetworkPackage& np)
{
    if (np.hasPayload()) {
         UploadJob* job = new UploadJob(np.payload());
         job->start();
         np.setPayloadTransferInfo(job->getTransferInfo());
    }

    int written = mSocket->write(np.serialize());
    return (written != -1);
}

void LanDeviceLink::dataReceived()
{
    QByteArray data = mSocket->readAll();

    //qDebug() << "LanDeviceLink dataReceived" << data;

    QList<QByteArray> packages = data.split('\n');
    Q_FOREACH(const QByteArray& package, packages) {

        if (package.length() < 3) {
            continue;
        }

        NetworkPackage unserialized(QString::null);
        NetworkPackage::unserialize(package, &unserialized);
        if (unserialized.isEncrypted()) {

            //mPrivateKey should always be set when device link is added to device, no null-checking done here
            NetworkPackage decrypted(QString::null);
            unserialized.decrypt(mPrivateKey, &decrypted);

            if (decrypted.hasPayloadTransferInfo()) {
                qDebug() << "HasPayloadTransferInfo";
                DownloadJob* job = new DownloadJob(mSocket->peerAddress(), decrypted.payloadTransferInfo());
                job->start();
                decrypted.setPayload(job->getPayload(), decrypted.payloadSize());
            }

            Q_EMIT receivedPackage(decrypted);

        } else {

            if (unserialized.hasPayloadTransferInfo()) {
                qWarning() << "Ignoring unencrypted payload";
                continue;
            }

            Q_EMIT receivedPackage(unserialized);

        }

    }

    //qDebug() << "MOAR BYTES" << mSocket->bytesAvailable() << packages.length();

    if (mSocket->bytesAvailable() > 0) {
        QMetaObject::invokeMethod(this, "dataReceived", Qt::QueuedConnection);
    }

}

void LanDeviceLink::readyRead()
{

}