BodyWriter.cpp

Go to the documentation of this file.

/*

Copyright (C) 2005-2007 by Peter Dimov.

This file is part of Calitko (http://www.calitko.org).

Calitko 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) any later version.

Calitko 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 Calitko; if not, write to the Free Software
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

*/

#include "Qt.h"
#include "BodyWriter.h"
#include "Imports.cpp"

using namespace Http;

BodyWriter::BodyWriter (Connection *connection)
 :  p()
{
    Q_ASSERT (connection != 0);

    p.connection    = connection;
    p.file          = 0;
    p.bodySize      = 0;
    p.bytesWritten  = 0;
    p.bufferRead    = 0;
    p.bufferWritten = 0;

    QObject::connect (p.connection, SIGNAL  (bytesWritten (qint64)),
                      this,         SLOT    (bytesWritten (qint64)));
}

BodyWriter::~BodyWriter()
{
}

void BodyWriter::write (QIODevice *file, qint64 bodySize)
{
    p.file          = file;
    p.bodySize      = bodySize;
    p.bytesWritten  = 0;
    p.bufferRead    = 0;
    p.bufferWritten = 0;

    QMetaObject::invokeMethod (this, "bytesWritten", Qt::QueuedConnection,
                QGenericReturnArgument(), Q_ARG (qint64, 0));
}

void BodyWriter::bytesWritten (qint64 bytes)
{
    // Is there more to read?
    if (p.bytesWritten == p.bodySize)
        return;

    if (bytes > 0) {
        p.bytesWritten += bytes;
        if (p.bytesWritten == p.bodySize) {
            emit bodyWritten();
            return;
        }
    }

    qint64 written = 0;
    do {
        if (p.bufferWritten == p.bufferRead) {
            p.bufferWritten = 0;
            qint64 bytesLeft = p.bodySize - p.bytesWritten;
            if (bytesLeft > BufferSize)
                p.bufferRead = BufferSize;
            else
                p.bufferRead = bytesLeft;
            p.bufferRead = p.file->read (p.buffer, p.bufferRead);
            if (p.bufferRead == -1) {
                emit readError();
            }
        }

        written = p.connection->write (p.buffer + p.bufferWritten,
                                            p.bufferRead - p.bufferWritten);
        if (written == -1) {
            emit writeError(); // would be Error::DeviceError
        }
        if (written > 0) {
            p.bufferWritten += written;
        }
    } while (written > 0);
}

bool BodyWriter::canWrite() const
{
    return p.bytesWritten == p.bodySize;
}

Copyright © 2006 Peter Dimov.
Powered by Doxygen.