udp.cpp

/*
    This file is part of RTBroker
    Made by Buzzing Light 2014-2015 — Buzzing Light

    Development:     Guillaume Jacquemin (https://www.buzzinglight.com)

    This file was written by Guillaume Jacquemin.

    HTTP2OSC is a 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 3 of the License, or
    any later version.

    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 "udp.h"

Udp::Udp(QWidget *_uiFeedback, QObject *parent) :
    QObject(parent), SenderInterface(_uiFeedback) {
    allowLog = true;
    socket = new QUdpSocket(this);
    connect(socket, SIGNAL(readyRead()), SLOT(parseOSC()));
}

void Udp::setPort(quint16 port) {
    if(port != socket->localPort()) {
        setUiFeedback(false);

        if((socket->isValid()) && (socket->localPort())) {
            qDebug("Fermeture de OSC sur %d (%d)", socket->localPort(), socket->isValid());
            socket->close();
        }
        if(socket->bind(port)) {
            setUiFeedback(true);
            qDebug("Ouverture OSC sur %d (%d)", port, socket->isValid());
        }
        else
            qDebug("Echec de l'ouverture OSC sur %d", port);
    }
    else
        qDebug("Port OSC déjà ouvert sur %d (%d)", socket->localPort(), socket->isValid());
}

void Udp::parseOSC() {
    //Parse all UDP datagram
    while(socket->hasPendingDatagrams()) {
        //Extract host, port & UDP datagram
        QHostAddress receivedHost;
        quint16 receivedPort;
        parsingBufferISize = socket->readDatagram((char*)parsingBufferI, 4096*32, &receivedHost, &receivedPort);
        qint64 indexBuffer = 0;

        //Parse UDP content
        while(indexBuffer < parsingBufferISize) {
            parsingIndexAddressBuffer = 0;
            parsingIndexArgumentsBuffer = 0;

            //Looking for '/'
            while((indexBuffer < parsingBufferISize) && (parsingBufferI[indexBuffer] != '/'))
                indexBuffer++;

            //Parse header
            if((parsingBufferI[indexBuffer] == '/') && (parsingBufferISize%4 == 0)) {
                //OSC Adress
                while((indexBuffer < parsingBufferISize) && (parsingBufferI[indexBuffer] != 0))
                    parsingAddressBuffer[parsingIndexAddressBuffer++] = parsingBufferI[indexBuffer++];
                parsingAddressBuffer[parsingIndexAddressBuffer] = 0;

                //Looking for ','
                while((indexBuffer < parsingBufferISize) && (parsingBufferI[indexBuffer++] != ',')) {}

                //OSC arguments type
                indexBuffer--;
                while((indexBuffer < parsingBufferISize) && (parsingBufferI[++indexBuffer] != 0))
                    parsingArgumentsBuffer[parsingIndexArgumentsBuffer++] = parsingBufferI[indexBuffer];
                parsingArgumentsBuffer[parsingIndexArgumentsBuffer] = 0;
                indexBuffer++;

                //Index modulo 4
                while((indexBuffer < parsingBufferISize) && ((indexBuffer++)%4 != 0)) {}
                indexBuffer--;

                //Parse content
                QString commandDestination = QString(parsingAddressBuffer);
                QString command = commandDestination + " ";
                QVariantList message;
                qint64 indexDataBuffer = 0;
                while((indexBuffer < parsingBufferISize) && (indexDataBuffer < parsingIndexArgumentsBuffer)) {
                    //Integer argument
                    if(parsingArgumentsBuffer[indexDataBuffer] == 'i') {
                        union { int i; char ch[4]; } u;
                        u.ch[3] = parsingBufferI[indexBuffer + 0];
                        u.ch[2] = parsingBufferI[indexBuffer + 1];
                        u.ch[1] = parsingBufferI[indexBuffer + 2];
                        u.ch[0] = parsingBufferI[indexBuffer + 3];
                        indexBuffer += 4;
                        QString commandValue = QString::number(u.i);
                        command += commandValue + " ";
                        message << u.i;
                    }
                    //Float argument
                    else if(parsingArgumentsBuffer[indexDataBuffer] == 'f') {
                        union { float f; char ch[4]; } u;
                        u.ch[3] = parsingBufferI[indexBuffer + 0];
                        u.ch[2] = parsingBufferI[indexBuffer + 1];
                        u.ch[1] = parsingBufferI[indexBuffer + 2];
                        u.ch[0] = parsingBufferI[indexBuffer + 3];
                        indexBuffer += 4;
                        QString commandValue = QString::number(u.f);
                        command += commandValue + " ";
                        message << u.f;
                    }
                    //String argument
                    else if(parsingArgumentsBuffer[indexDataBuffer] == 's') {
                        QString commandValue = "";
                        while((indexBuffer < parsingBufferISize) && (parsingBufferI[indexBuffer]) != 0)
                            commandValue += parsingBufferI[indexBuffer++];
                        indexBuffer++;
                        while(indexBuffer % 4 != 0)
                            indexBuffer++;
                        command += commandValue + " ";
                        message << commandValue;
                    }
                    else
                        indexBuffer += 4;
                    indexDataBuffer++;
                }

                //Dispatching
                if((commandDestination == "/ping") && (message.count()))
                    send(QVariantList(), receivedHost.toString(), message.first().toInt(), "/pong");
                else {
                    message.prepend(commandDestination);
                    MainWindowInterface::main->dispatch(message, QVariantList() << "/websockets");
                }
            }
        }
    }
}

QString Udp::send(const QVariantList &valeurs, const QString &ip, quint16 port, const QString &destination, void *) {
    new UdpDatagram(this, ip, port, destination, valeurs, false);
    QString retour = QString("%4: OSC sent on %3 (%1:%2): ").arg(ip).arg(port).arg(destination).arg(QDateTime::currentDateTime().toString("yyyy-MM-dd hh:mm:ss.zzz"));
    if(valeurs.count()) {
        foreach(const QVariant &valeur, valeurs)
            retour += valeur.toString() + " ";
    }
    else
        retour += "no args";
    return retour + "\n";
}
void Udp::send(UdpDatagram *datagram, quint16) {
    if(!datagram->host.isNull()) {
        qint64 sent = socket->writeDatagram(datagram->buffer, datagram->host, datagram->port);
        if(sent < 0)
            qDebug("%s : %d", qPrintable(socket->errorString()), datagram->buffer.size());
    }
}


UdpDatagram::UdpDatagram(Udp *_udp, const QString &_ip, quint16 _port, const QString &_destination, const QVariantList &_valeurs, bool _secure)
    : QObject(_udp) {
    ip = _ip;
    port = _port;
    destination = _destination;
    secure = _secure;
    valeurs = _valeurs;
    udp = _udp;
    counter = 0;

    host = QHostAddress(ip);
    if(!host.isNull()) {
        verboseMessage = "";
        address += destination;
        address += (char)0;
        pad(address);
        typetag += ',';

        foreach(const QVariant &valeur, valeurs) {
            if(valeur.type() == QVariant::String) {
                verboseMessage = verboseMessage + " " + qPrintable(valeur.toString());
                arguments += valeur.toString();
                arguments += (char)0;
                pad(arguments);
                typetag += 's';
            }
            else if(valeur.type() == QVariant::Int) {
                verboseMessage = verboseMessage + " " + QByteArray::number(valeur.toInt());
                union { int i; char ch[4]; } u;
                u.i = valeur.toInt();
                arguments += u.ch[3];
                arguments += u.ch[2];
                arguments += u.ch[1];
                arguments += u.ch[0];
                typetag += 'i';
            }
            else {
                verboseMessage = verboseMessage + " " + QByteArray::number(valeur.toDouble());
                union { float f; char ch[4]; } u;
                u.f = valeur.toDouble();
                arguments += u.ch[3];
                arguments += u.ch[2];
                arguments += u.ch[1];
                arguments += u.ch[0];
                typetag += 'f';
            }
        }

        buffer += address;
        buffer += typetag;
        buffer += (char)0;
        pad(buffer);
        buffer += arguments;

        udp->send(this, counter);
    }
    delete this;
}