src/dash/network/socket.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2006 by David Cuadrado                                  *
   3  *   krawek@gmail.com                                                     *
   4  *                                                                         *
   5  *   This program is free software; you can redistribute it and/or modify  *
   6  *   it under the terms of the GNU General Public License as published by  *
   7  *   the Free Software Foundation; either version 2 of the License, or     *
   8  *   (at your option) any later version.                                   *
   9  *                                                                         *
  10  *   This program is distributed in the hope that it will be useful,       *
  11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  13  *   GNU General Public License for more details.                          *
  14  *                                                                         *
  15  *   You should have received a copy of the GNU General Public License     *
  16  *   along with this program; if not, write to the                         *
  17  *   Free Software Foundation, Inc.,                                       *
  18  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
  19  ***************************************************************************/
  20
  21 #include "socket.h"
  22
  23 #include <QTextStream>
  24 #include <QQueue>
  25
  26 #include <dcore/debug.h>
  27
  28
  29 // Dash
  30 #include "socketobserver.h"
  31
  32 namespace Dash {
  33 namespace Network {
  34
  35 struct Socket::Private
  36 {
  37         Private() : format(Text) {}
  38
  39         QQueue<QByteArray> queue;
  40         QList<SocketObserver *> observers;
  41
  42         Format format;
  43 };
  44
  45 Socket::Socket(QObject *parent)
  46  : /*QSslSocket*/QTcpSocket(parent), d(new Private)
  47 {
  48         connect(this, SIGNAL(readyRead ()), this, SLOT(readFromServer()) );
  49         connect(this, SIGNAL(connected()), this, SLOT(sendQueue()));
  50         connect(this, SIGNAL(disconnected()), this, SLOT(clearQueue()));
  51 }
  52
  53
  54 Socket::~Socket()
  55 {
  56         delete d;
  57 }
  58
  59 void Socket::setFormat(Format format)
  60 {
  61         d->format = format;
  62 }
  63
  64 Socket::Format Socket::format() const
  65 {
  66         return d->format;
  67 }
  68
  69 void Socket::addObserver(SocketObserver *observer)
  70 {
  71         d->observers << observer;
  72 }
  73
  74 void Socket::removeObserver(SocketObserver *observer)
  75 {
  76         d->observers.removeAll(observer);
  77 }
  78
  79 void Socket::sendQueue()
  80 {
  81         while( d->queue.count() > 0 )
  82         {
  83                 if ( state() == QAbstractSocket::ConnectedState )
  84                 {
  85                         send(d->queue.dequeue());
  86                 }
  87                 else break;
  88         }
  89 }
  90
  91 void Socket::clearQueue()
  92 {
  93         d->queue.clear();
  94 }
  95
  96 void Socket::send(const QByteArray &data)
  97 {
  98         if( state() == QAbstractSocket::ConnectedState )
  99         {
 100                 switch(d->format)
 101                 {
 102                         case Text:
 103                         {
 104                                 QTextStream stream(this);
 105                                 stream << data.toBase64() << "%%" << endl;
 106                         }
 107                         break;
 108                         case Binary:
 109                         {
 110                                 write(data);
 111                         }
 112                         break;
 113                 }
 114         }
 115         else
 116         {
 117                 d->queue.enqueue(data);
 118         }
 119 }
 120
 121 void Socket::send(const QDomDocument &doc)
 122 {
 123         send(doc.toString(0).toLocal8Bit());
 124 }
 125
 126 void Socket::readFromServer()
 127 {
 128         switch(d->format)
 129         {
 130                 case Text:
 131                 {
 132                         QString readed;
 133                         while(this->canReadLine())
 134                         {
 135                                 readed += this->readLine();
 136
 137                                 if ( readed.endsWith("%%\n") )
 138                                 {
 139                                         readed.remove(readed.lastIndexOf("%%"), 2);
 140                                         break;
 141                                 }
 142                         }
 143
 144                         if ( !readed.isEmpty() )
 145                         {
 146                                 readed = QByteArray::fromBase64(readed.toLocal8Bit());
 147
 148                                 foreach(SocketObserver *o, d->observers)
 149                                 {
 150                                         o->readed(readed);
 151                                 }
 152                         }
 153                 }
 154                 break;
 155                 case Binary:
 156                 {
 157                         while(bytesAvailable() > 0)
 158                         {
 159                                 QByteArray data = this->read(1024);
 160                                 foreach(SocketObserver *o, d->observers)
 161                                 {
 162                                         o->dataReaded(data);
 163                                 }
 164                         }
 165                 }
 166                 break;
 167         }
 168
 169         if (this->bytesAvailable() > 0) readFromServer();
 170 }
 171
 172 }
 173 }
 174