iosystem.hpp

#pragma once

#include <map>
#include <memory>
#include <mutex>
#include <thread>

#include "dangerous/snmp/exception.hpp"

#include "bytestream.hpp"


namespace dangerous { namespace snmp {

class IoSystem {
public:
    IoSystem() throw( Exception );
    ~IoSystem();

    void registerInput( int handle, std::shared_ptr<ByteStream> buffer );
    void registerOutput( int handle, std::shared_ptr<ByteStream> buffer );
    
    void unregisterInput( int handle );
    void unregisterOutput( int handle );

    void main();

    void start();

    void stop();

    void wakeup();

protected:
    bool isStopped;

    std::thread thisThread;

    std::mutex lock;
    std::map< int, std::shared_ptr<ByteStream> > inMap;
    std::map< int, std::shared_ptr<ByteStream> > outMap;

    int localSockets[ 2 ];
};

} }