IdentityContentLengthBodyReader.cpp

00001 /*
00002 
00003 Copyright (C) 2006-2007 by Peter Dimov.
00004 
00005 This file is part of Calitko (http://www.calitko.org).
00006 
00007 Calitko is free software; you can redistribute it and/or modify
00008 it under the terms of the GNU General Public License as published by
00009 the Free Software Foundation; either version 2 of the License, or
00010 (at your option) any later version.
00011 
00012 Calitko is distributed in the hope that it will be useful,
00013 but WITHOUT ANY WARRANTY; without even the implied warranty of
00014 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
00015 GNU General Public License for more details.
00016 
00017 You should have received a copy of the GNU General Public License
00018 along with Calitko; if not, write to the Free Software
00019 Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
00020 
00021 */
00022 
00023 #include "Qt.h"
00024 #include "IdentityContentLengthBodyReader.h"
00025 #include "BodyEnd.h"
00026 #include "Imports.cpp"
00027 
00029 IdentityContentLengthBodyReader::IdentityContentLengthBodyReader (
00030                                                             int contentLength)
00031  :  bytesLeft (contentLength)
00032 {
00033     Q_ASSERT (bytesLeft >= 0);
00034 }
00035 
00037 IdentityContentLengthBodyReader::~IdentityContentLengthBodyReader()
00038 {
00039 }
00040 
00042 
00049 Data IdentityContentLengthBodyReader::read (Transport *transport)
00050 {
00051     if (bytesLeft > 0) {
00052         QByteArray bytes = transport->readAtMost (bytesLeft);
00053         bytesLeft -= bytes.size();
00054         Q_ASSERT (bytesLeft >= 0);
00055         return RawData (bytes);
00056     } else if (bytesLeft == 0) {
00057         --bytesLeft;
00058         return BodyEnd();
00059     } else {
00060         // read() should not be called after BodyEnd() was send:
00061         Q_ASSERT (false);
00062         return Data();
00063     }
00064 }