rate_limited_layer.hpp

Go to the documentation of this file.

#ifndef LIBFILEZILLA_RATE_LIMITED_LAYER_HEADER
#define LIBFILEZILLA_RATE_LIMITED_LAYER_HEADER

 
#include "rate_limiter.hpp"
#include "socket.hpp"
 
namespace fz {

class FZ_PUBLIC_SYMBOL rate_limited_layer final : public socket_layer, private bucket
{
public:
    rate_limited_layer(event_handler* handler, socket_interface& next_layer, rate_limiter * limiter = nullptr);
    virtual ~rate_limited_layer();
 
    virtual int read(void* buffer, unsigned int size, int& error) override;
    virtual int write(void const* buffer, unsigned int size, int& error) override;
 
    virtual void set_event_handler(event_handler* handler, socket_event_flag retrigger_block = socket_event_flag{}) override;
 
protected:
    virtual void wakeup(direction::type d) override;
};

class FZ_PUBLIC_SYMBOL compound_rate_limited_layer final : public socket_layer
{
public:
    compound_rate_limited_layer(event_handler* handler, socket_interface& next_layer);
    virtual ~compound_rate_limited_layer();
 
    void add_limiter(rate_limiter * limiter);
    void remove_limiter(rate_limiter * limiter);
 
    virtual int read(void* buffer, unsigned int size, int& error) override;
    virtual int write(void const* buffer, unsigned int size, int& error) override;
 
    virtual socket_state get_state() const override {
        return next_layer_.get_state();
    }
 
    virtual int connect(native_string const& host, unsigned int port, address_type family = address_type::unknown) override {
        return next_layer_.connect(host, port, family);
    }
 
    virtual int shutdown() override {
        return next_layer_.shutdown();
    }
 
    virtual void set_event_handler(event_handler* handler, fz::socket_event_flag retrigger_block = socket_event_flag{}) override;
 
protected:
    class crll_bucket;
    friend class crll_bucket;
    std::vector<std::unique_ptr<crll_bucket>> buckets_;
 
    fz::mutex mtx_{false};
};
 
 
}
 
#endif