ring_settings.hpp

Go to the documentation of this file.

 
#pragma once
 
#include "condy/condy_uring.hpp"
#include "condy/utils.hpp"
#include <cassert>
#include <cerrno>
#include <cstddef>
#include <cstring>
 
namespace condy {

class FdTable {
public:
    FdTable(io_uring &ring) : ring_(ring) {}
 
    CONDY_DELETE_COPY_MOVE(FdTable);
 
public:
    int init(size_t capacity) noexcept {
        return io_uring_register_files_sparse(&ring_, capacity);
    }

    int init(const int *fds, unsigned nr_fds) {
        return io_uring_register_files(&ring_, fds, nr_fds);
    }

    int destroy() noexcept { return io_uring_unregister_files(&ring_); }

    int update(unsigned index_base, const int *fds, unsigned nr_fds) noexcept {
        return io_uring_register_files_update(&ring_, index_base, fds, nr_fds);
    }

    int set_file_alloc_range(unsigned offset, unsigned size) noexcept {
        return io_uring_register_file_alloc_range(&ring_, offset, size);
    }
 
private:
    io_uring &ring_;
};

class BufferTable {
public:
    BufferTable(io_uring &ring) : ring_(ring) {}
 
    CONDY_DELETE_COPY_MOVE(BufferTable);
 
public:
    int init(size_t capacity) noexcept {
        return io_uring_register_buffers_sparse(&ring_, capacity);
    }

    int init(const iovec *vecs, unsigned nr_vecs) {
        return io_uring_register_buffers(&ring_, vecs, nr_vecs);
    }

    int destroy() noexcept { return io_uring_unregister_buffers(&ring_); }

    int update(unsigned index_base, const iovec *vecs,
               unsigned nr_vecs) noexcept {
        return io_uring_register_buffers_update_tag(&ring_, index_base, vecs,
                                                    nullptr, nr_vecs);
    }
 
#if !IO_URING_CHECK_VERSION(2, 10) // >= 2.10

    int clone_buffers(BufferTable &src, unsigned int dst_off = 0,
                      unsigned int src_off = 0, unsigned int nr = 0) noexcept {
        auto *src_ring = &src.ring_;
        auto *dst_ring = &ring_;
        return __io_uring_clone_buffers_offset(dst_ring, src_ring, dst_off,
                                               src_off, nr,
                                               IORING_REGISTER_DST_REPLACE);
    }
#endif
 
private:
    io_uring &ring_;
};

class RingSettings {
public:
    RingSettings(io_uring &ring) : ring_(ring) {}
 
    ~RingSettings() {
        if (probe_) {
            io_uring_free_probe(probe_);
            probe_ = nullptr;
        }
    }
 
    CONDY_DELETE_COPY_MOVE(RingSettings);
 
public:
    int apply_iowq_aff(size_t cpusz, const cpu_set_t *mask) noexcept {
        return io_uring_register_iowq_aff(&ring_, cpusz, mask);
    }

    int remove_iowq_aff() noexcept {
        return io_uring_unregister_iowq_aff(&ring_);
    }

    int set_iowq_max_workers(unsigned int *values) noexcept {
        return io_uring_register_iowq_max_workers(&ring_, values);
    }

    io_uring_probe *get_probe() noexcept {
        if (probe_) {
            return probe_;
        }
        probe_ = io_uring_get_probe_ring(&ring_);
        return probe_;
    }

    uint32_t get_features() const noexcept { return ring_.features; }
 
#if !IO_URING_CHECK_VERSION(2, 6) // >= 2.6
    int apply_napi(io_uring_napi *napi) noexcept {
        return io_uring_register_napi(&ring_, napi);
    }

    int remove_napi(io_uring_napi *napi = nullptr) noexcept {
        return io_uring_unregister_napi(&ring_, napi);
    }
#endif
 
#if !IO_URING_CHECK_VERSION(2, 8) // >= 2.8
    int set_clock(io_uring_clock_register *clock_reg) noexcept {
        return io_uring_register_clock(&ring_, clock_reg);
    }
#endif
 
#if !IO_URING_CHECK_VERSION(2, 9) // >= 2.9
    int set_rings_size(io_uring_params *params) noexcept {
        return io_uring_resize_rings(&ring_, params);
    }
#endif
 
#if !IO_URING_CHECK_VERSION(2, 10) // >= 2.10
    int set_iowait(bool enable_iowait) noexcept {
        return io_uring_set_iowait(&ring_, enable_iowait);
    }
#endif
 
private:
    io_uring &ring_;
    io_uring_probe *probe_ = nullptr;
};
 
} // namespace condy