libcore/plugins/tcpsst/ASIOReadBuffer.hpp

Go to the documentation of this file.

/*  Sirikata Network Utilities
 *  ASIOReadBuffer.hpp
 *
 *  Copyright (c) 2009, Daniel Reiter Horn
 *  All rights reserved.
 *
 *  Redistribution and use in source and binary forms, with or without
 *  modification, are permitted provided that the following conditions are
 *  met:
 *  * Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 *  * Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in
 *    the documentation and/or other materials provided with the
 *    distribution.
 *  * Neither the name of Sirikata nor the names of its contributors may
 *    be used to endorse or promote products derived from this software
 *    without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
 * IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
 * PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
 * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include "MultiplexedSocket.hpp"
#include <sirikata/core/network/IOStrand.hpp>

namespace Sirikata {
namespace Network {
class ASIOReadBuffer;
struct ASIOReadBufferUtil;
ASIOReadBuffer* MakeASIOReadBuffer(const MultiplexedSocketPtr &parentSocket,unsigned int whichSocket, const MemoryReference &strayBytesAfterHeader, TCPStream::StreamType streamType);
class ASIOReadBuffer {
    friend ASIOReadBuffer* MakeASIOReadBuffer(const MultiplexedSocketPtr &parentSocket,unsigned int whichSocket, const MemoryReference &strayBytesAfterHeader, TCPStream::StreamType streamType);
public:
    enum {
        sBufferLength=64*1024,
        sLowWaterMark=sBufferLength
    };
private:
    friend struct ASIOReadBufferUtil;
    enum ReceivedResponse {
        StreamNotPaused,
        PausedStream
    };
    enum ReadStatus{
        PAUSED_FIXED_BUFFER=0x0,
        READING_FIXED_BUFFER=0x1,
        PAUSED_NEW_CHUNK=0x2,
        READING_NEW_CHUNK=0x3,
    }mReadStatus;
    uint8 mBuffer[sBufferLength];
    unsigned int mFixedBufferPos;
    unsigned int mChunkBufferPos;
    std::vector<uint8> mPartialStreamId;
    bool mFirstFrame; 
    bool mLastFrame; 

    unsigned int mWhichBuffer;
    Chunk mNewChunk;
    Chunk *mCachedRejectedChunk;
    Stream::StreamID mNewChunkID;
    uint8 mDataMask[4];
    TCPStream::StreamType mStreamType;
    std::tr1::weak_ptr<MultiplexedSocket> mParentSocket;
    typedef boost::system::error_code ErrorCode;
    std::tr1::function<void(const ErrorCode&,std::size_t)> mAsioReadIntoFixedBuffer;
    std::tr1::function<void(const ErrorCode&,std::size_t)> mAsioReadIntoChunk;

    void bindFunctions(IOStrand* strand);
    void processError(MultiplexedSocket*parentSocket, const ErrorCode &error);
    ReceivedResponse processFullChunk(const MultiplexedSocketPtr &parentSocket,
        unsigned int whichSocket,
        const Stream::StreamID& sid,
        Chunk&newChunk,
        const Stream::PauseReceiveCallback& pauseReceive);

    void readIntoFixedBuffer(const MultiplexedSocketPtr &parentSocket);
    void readIntoChunk(const MultiplexedSocketPtr &parentSocket);

    void processPartialChunk(uint8* dataBuffer, uint32 packetLength, uint32 &bufferReceived, Chunk&retval);

    void translateFixedBuffer(const MultiplexedSocketPtr &thus);

    void asioReadIntoChunk(const ErrorCode&error,std::size_t bytes_read);

    void asioReadIntoFixedBuffer(const ErrorCode&error,std::size_t bytes_read);

    ASIOReadBuffer(const MultiplexedSocketPtr &parentSocket,unsigned int whichSocket, TCPStream::StreamType streamType);
    ASIOReadBuffer(const ASIOReadBuffer&);
public:
    void ioReactorThreadResumeRead(MultiplexedSocketPtr&thus);
    ~ASIOReadBuffer();
};



} // namespace Network
} // namespace Sirikata