HeaderReader.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 "HeaderReader.h"
#include "Header.h"
#include "Imports.cpp"

using namespace Http;

namespace Http {

enum Constants
{
    MaximalHeaderLength = 4096
};

} // namespace Http;

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

    p.connection = connection;
    p.header = 0;
    p.bytesRead = 0;
    p.done = true;

    connect (p.connection, SIGNAL (readyRead()), this, SLOT (readyRead()));
}

HeaderReader::~HeaderReader()
{
    if (p.header != 0)
        delete p.header;
}

void HeaderReader::startReading (Header *header)
{
    Q_ASSERT (header != 0); // We need an object of the rigth type!
    Q_ASSERT (p.done == true);

    p.handshakeBytes.resize (0);
    p.handshakeBytes.reserve (MaximalHeaderLength);
    p.handshakeBytes.append (char (0));
    p.bytesRead = 0;
    p.done = false;
    if (p.header)
        delete p.header;
    p.header = header;
    QMetaObject::invokeMethod (this, "readyRead", Qt::QueuedConnection);
}

void HeaderReader::readyRead ()
{
    if (p.done)
        return;

    bool ready = false;
    do {
        if (!p.connection->getChar (p.handshakeBytes.data() + p.bytesRead))
            return;

        p.handshakeBytes.append (char (0));
        ++p.bytesRead;
        if (p.bytesRead == MaximalHeaderLength) {
            p.done = true;
            emit readError(); // would be Error::InvalidHeader
            return;
        }
        if (p.handshakeBytes.lastIndexOf ("\r\n\r\n") != -1)
            ready = true;
        if (p.handshakeBytes.lastIndexOf ("\n\n") != -1)
            ready = true;
    } while (!ready);

    QString handshakeString = QString::fromLatin1 (p.handshakeBytes.data());

    if (p.header->parse (handshakeString)) {
        p.done = true;
        emit headerRead();
    } else {
        delete p.header;
        p.header = 0;
        p.done = true;
        emit readError(); // would be Error::InvalidHeader
    }
    return;
}

const Header * HeaderReader::header() const
{
    Q_ASSERT (p.done == true);

    return p.header;
}

Copyright © 2006 Peter Dimov.
Powered by Doxygen.