cqe_handler.hpp

Go to the documentation of this file.

 
#pragma once
 
#include "condy/concepts.hpp"
#include "condy/detail/context.hpp"
#include "condy/runtime.hpp"
#include <cassert>
#include <cerrno>
#include <cstdint>
#include <utility>
 
namespace condy {

struct SimpleCQEHandler {
    int32_t operator()(io_uring_cqe *cqe) noexcept { return cqe->res; }
};

template <typename Br>
    requires(requires(Br *br, io_uring_cqe *cqe) { br->handle_finish(cqe); })
class SelectBufferCQEHandler {
public:
    SelectBufferCQEHandler(Br *buffers) : buffers_(buffers) {}
 
    auto operator()(io_uring_cqe *cqe) noexcept {
        return std::make_pair(cqe->res, buffers_->handle_finish(cqe));
    }
 
private:
    Br *buffers_;
};

struct NVMePassthruCQEHandler {
    std::pair<int32_t, uint64_t> operator()(io_uring_cqe *cqe) noexcept {
        assert(
            detail::Context::current().runtime()->ring_internal().check_cqe32(
                cqe) &&
            "Expected big CQE for NVMe passthrough");
        return {cqe->res, cqe->big_cqe[0]};
    }
};

struct SCSIBsgResult {
    uint64_t res2;

    uint8_t device_status() noexcept { return res2 & 0xff; }

    uint8_t driver_status() noexcept { return (res2 >> 8) & 0xff; }

    uint8_t host_status() noexcept { return (res2 >> 16) & 0xff; }

    uint8_t sense_len() noexcept { return (res2 >> 24) & 0xff; }

    uint32_t resid_len() noexcept { return res2 >> 32; }
};

struct SCSIBsgPassthruCQEHandler {
    std::pair<int32_t, SCSIBsgResult> operator()(io_uring_cqe *cqe) noexcept {
        assert(
            detail::Context::current().runtime()->ring_internal().check_cqe32(
                cqe) &&
            "Expected big CQE for SCSI BSG passthrough");
        return {cqe->res, SCSIBsgResult{cqe->big_cqe[0]}};
    }
};
 
#if CONDY_URING_VERSION_GE(2, 12) // >= 2.12
struct TxTimestampResult {
    int tstype; // cqe->flags >> IORING_TIMESTAMP_TYPE_SHIFT

    bool hwts; // cqe->flags & IORING_CQE_F_TSTAMP_HW

    io_timespec ts; // *(io_timespec *)(cqe->big_cqe)
};

struct TxTimestampCQEHandler {
    std::pair<int32_t, TxTimestampResult>
    operator()(io_uring_cqe *cqe) noexcept {
        assert(
            detail::Context::current().runtime()->ring_internal().check_cqe32(
                cqe) &&
            "Expected big CQE for TX timestamp operations");
        TxTimestampResult result;
        result.tstype =
            static_cast<int>(cqe->flags >> IORING_TIMESTAMP_TYPE_SHIFT);
        result.hwts = cqe->flags & IORING_CQE_F_TSTAMP_HW;
        result.ts = reinterpret_cast<io_timespec &>(cqe->big_cqe);
        return {cqe->res, result};
    }
};
#endif
 
} // namespace condy