File indexing completed on 2023-09-24 04:04:37

 /*  -*- C++ -*-
  *  Copyright (C) 2003-2005 Thiago Macieira <thiago@kde.org>
  *
  *
  *  Permission is hereby granted, free of charge, to any person obtaining
  *  a copy of this software and associated documentation files (the
  *  "Software"), to deal in the Software without restriction, including
  *  without limitation the rights to use, copy, modify, merge, publish,
  *  distribute, sublicense, and/or sell copies of the Software, and to
  *  permit persons to whom the Software is furnished to do so, subject to
  *  the following conditions:
  *
  *  The above copyright notice and this permission notice shall be included
  *  in all copies or substantial portions of the Software.
  *
  *  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  *  EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  *  MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  *  NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  *  LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  *  OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  *  WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  */
 
 #include "k3bufferedsocket.h"
 
 #include <config-network.h>
 
 #include <QMutex>
 #include <QTimer>
 
 #include "k3socketdevice.h"
 #include "k3socketaddress.h"
 #include "k3socketbuffer_p.h"
 
 using namespace KNetwork;
 using namespace KNetwork::Internal;
 
 class KNetwork::KBufferedSocketPrivate
 {
 public:
     mutable KSocketBuffer *input, *output;
 
     KBufferedSocketPrivate()
     {
         input = nullptr;
         output = nullptr;
     }
 };
 
 KBufferedSocket::KBufferedSocket(const QString &host, const QString &service,
                                  QObject *parent)
     : KStreamSocket(host, service, parent),
       d(new KBufferedSocketPrivate)
 {
     setInputBuffering(true);
     setOutputBuffering(true);
 }
 
 KBufferedSocket::~KBufferedSocket()
 {
     closeNow();
     delete d->input;
     delete d->output;
     delete d;
 }
 
 void KBufferedSocket::setSocketDevice(KSocketDevice *device)
 {
     KStreamSocket::setSocketDevice(device);
     device->setBlocking(false);
     KActiveSocketBase::open(openMode() & ~Unbuffered);
 }
 
 bool KBufferedSocket::setSocketOptions(int opts)
 {
     if (opts == Blocking) {
         return false;
     }
 
     opts &= ~Blocking;
     return KStreamSocket::setSocketOptions(opts);
 }
 
 void KBufferedSocket::close()
 {
     if (!d->output || d->output->isEmpty()) {
         closeNow();
     } else {
         setState(Closing);
         QSocketNotifier *n = socketDevice()->readNotifier();
         if (n) {
             n->setEnabled(false);
         }
         emit stateChanged(Closing);
     }
 }
 
 qint64 KBufferedSocket::bytesAvailable() const
 {
     if (!d->input) {
         return KStreamSocket::bytesAvailable();
     }
 
     return d->input->length();
 }
 
 qint64 KBufferedSocket::waitForMore(int msecs, bool *timeout)
 {
     qint64 retval = KStreamSocket::waitForMore(msecs, timeout);
     if (d->input) {
         resetError();
         slotReadActivity();
         return bytesAvailable();
     }
     return retval;
 }
 
 qint64 KBufferedSocket::readData(char *data, qint64 maxlen, KSocketAddress *from)
 {
     if (from) {
         *from = peerAddress();
     }
     if (d->input) {
         if (d->input->isEmpty()) {
             return 0;
         }
         resetError();
         return d->input->consumeBuffer(data, maxlen);
     }
     return KStreamSocket::readData(data, maxlen, nullptr);
 }
 
 qint64 KBufferedSocket::peekData(char *data, qint64 maxlen, KSocketAddress *from)
 {
     if (from) {
         *from = peerAddress();
     }
     if (d->input) {
         if (d->input->isEmpty()) {
             return 0;
         }
         resetError();
         return d->input->consumeBuffer(data, maxlen, false);
     }
     return KStreamSocket::peekData(data, maxlen, nullptr);
 }
 
 qint64 KBufferedSocket::writeData(const char *data, qint64 len,
                                   const KSocketAddress *)
 {
     // ignore the third parameter
     if (state() != Connected) {
         // cannot write now!
         setError(NotConnected);
         return -1;
     }
 
     if (d->output) {
         if (d->output->isFull()) {
             setError(WouldBlock);
             emit gotError(WouldBlock);
             return -1;
         }
         resetError();
 
         // enable notifier to send data
         QSocketNotifier *n = socketDevice()->writeNotifier();
         if (n) {
             n->setEnabled(true);
         }
 
         return d->output->feedBuffer(data, len);
     }
 
     return KStreamSocket::writeData(data, len, nullptr);
 }
 
 void KBufferedSocket::enableRead(bool enable)
 {
     KStreamSocket::enableRead(enable);
     if (!enable && d->input) {
         // reenable it
         QSocketNotifier *n = socketDevice()->readNotifier();
         if (n) {
             n->setEnabled(true);
         }
     }
 
     if (enable && state() != Connected && d->input && !d->input->isEmpty())
         // this means the buffer is still dirty
         // allow the signal to be emitted
     {
         QTimer::singleShot(0, this, SLOT(slotReadActivity()));
     }
 }
 
 void KBufferedSocket::enableWrite(bool enable)
 {
     KStreamSocket::enableWrite(enable);
     if (!enable && d->output && !d->output->isEmpty()) {
         // reenable it
         QSocketNotifier *n = socketDevice()->writeNotifier();
         if (n) {
             n->setEnabled(true);
         }
     }
 }
 
 void KBufferedSocket::stateChanging(SocketState newState)
 {
     if (newState == Connecting || newState == Connected) {
         // we're going to connect
         // make sure the buffers are clean
         if (d->input) {
             d->input->clear();
         }
         if (d->output) {
             d->output->clear();
         }
 
         // also, turn on notifiers
         enableRead(emitsReadyRead());
         enableWrite(emitsReadyWrite());
     }
     KStreamSocket::stateChanging(newState);
 }
 
 void KBufferedSocket::setInputBuffering(bool enable)
 {
     QMutexLocker locker(mutex());
     if (!enable) {
         delete d->input;
         d->input = nullptr;
     } else if (d->input == nullptr) {
         d->input = new KSocketBuffer;
     }
 }
 
 void KBufferedSocket::setOutputBuffering(bool enable)
 {
     QMutexLocker locker(mutex());
     if (!enable) {
         delete d->output;
         d->output = nullptr;
     } else if (d->output == nullptr) {
         d->output = new KSocketBuffer;
     }
 }
 
 qint64 KBufferedSocket::bytesToWrite() const
 {
     if (!d->output) {
         return 0;
     }
 
     return d->output->length();
 }
 
 void KBufferedSocket::closeNow()
 {
     KStreamSocket::close();
     if (d->output) {
         d->output->clear();
     }
 }
 
 bool KBufferedSocket::canReadLine() const
 {
     if (!d->input) {
         return false;
     }
 
     return d->input->canReadLine();
 }
 
 qint64 KBufferedSocket::readLineData(char *data, qint64 maxSize)
 {
     return d->input->readLine(data, maxSize);
 }
 
 void KBufferedSocket::waitForConnect()
 {
     if (state() != Connecting) {
         return;    // nothing to be waited on
     }
 
     KStreamSocket::setSocketOptions(socketOptions() | Blocking);
     connectionEvent();
     KStreamSocket::setSocketOptions(socketOptions() & ~Blocking);
 }
 
 void KBufferedSocket::slotReadActivity()
 {
     if (d->input && state() == Connected) {
         mutex()->lock();
         qint64 len = d->input->receiveFrom(socketDevice());
 
         if (len == -1) {
             if (socketDevice()->error() != WouldBlock) {
                 // nope, another error!
                 copyError();
                 mutex()->unlock();
                 emit gotError(error());
                 closeNow();   // emits closed
                 return;
             }
         } else if (len == 0) {
             // remotely closed
             setError(RemotelyDisconnected);
             mutex()->unlock();
             emit gotError(error());
             closeNow();       // emits closed
             return;
         }
 
         // no error
         mutex()->unlock();
     }
 
     if (state() == Connected) {
         KStreamSocket::slotReadActivity();    // this emits readyRead
     } else if (emitsReadyRead()) { // state() != Connected
         if (d->input && !d->input->isEmpty()) {
             // buffer isn't empty
             // keep emitting signals till it is
             QTimer::singleShot(0, this, SLOT(slotReadActivity()));
             emit readyRead();
         }
     }
 }
 
 void KBufferedSocket::slotWriteActivity()
 {
     if (d->output && !d->output->isEmpty() &&
             (state() == Connected || state() == Closing)) {
         mutex()->lock();
         qint64 len = d->output->sendTo(socketDevice());
 
         if (len == -1) {
             if (socketDevice()->error() != WouldBlock) {
                 // nope, another error!
                 copyError();
                 mutex()->unlock();
                 emit gotError(error());
                 closeNow();
                 return;
             }
         } else if (len == 0) {
             // remotely closed
             setError(RemotelyDisconnected);
             mutex()->unlock();
             emit gotError(error());
             closeNow();
             return;
         }
 
         if (d->output->isEmpty())
             // deactivate the notifier until we have something to send
             // writeNotifier can't return NULL here
         {
             socketDevice()->writeNotifier()->setEnabled(false);
         }
 
         mutex()->unlock();
         emit bytesWritten(len);
     }
 
     if (state() != Closing) {
         KStreamSocket::slotWriteActivity();
     } else if (d->output && d->output->isEmpty() && state() == Closing) {
         KStreamSocket::close();   // finished sending data
     }
 }
 
 #include "moc_k3bufferedsocket.cpp"