file_utils.hpp

/*
 * Copyright (c) 2025, NVIDIA CORPORATION.
 *
 * 根据 Apache 许可证 2.0 版（以下简称“许可证”）获得许可；
 * 除非遵守许可证的规定，否则您不得使用此文件。
 * 您可以在以下位置获取许可证的副本：
 *
 * https://apache.ac.cn/licenses/LICENSE-2.0
 *
 * 除非适用法律要求或书面同意，否则根据许可证分发的软件是
 * 按“原样”分发的，没有任何明示或暗示的保证或条件。
 * 有关特定语言的权限和
 * 许可证下的限制，请参阅许可证。
 */
/*
 */#pragma once

#include <optional>
#include <string>

#include <kvikio/shim/cufile_h_wrapper.hpp>

namespace kvikio {
class FileWrapper {
 private
 int _fd{-1};

 public
 FileWrapper(std::string const& file_path, std::string const& flags, bool o_direct, mode_t mode);

 FileWrapper() noexcept = default;

 ~FileWrapper() noexcept;
 FileWrapper(FileWrapper const&) = delete;
 FileWrapper& operator=(FileWrapper const&) = delete;
 FileWrapper(FileWrapper&& o) noexcept;
 FileWrapper& operator=(FileWrapper&& o) noexcept;

 void open(std::string const& file_path, std::string const& flags, bool o_direct, mode_t mode);

 bool opened() const noexcept;

 void close() noexcept;

 int fd() const noexcept;
};

class CUFileHandleWrapper {
 private
 CUfileHandle_t _handle{};
 bool _registered{false};

 public
 CUFileHandleWrapper() noexcept = default;
 ~CUFileHandleWrapper() noexcept;
 CUFileHandleWrapper(CUFileHandleWrapper const&) = delete;
 CUFileHandleWrapper& operator=(CUFileHandleWrapper const&) = delete;
 CUFileHandleWrapper(CUFileHandleWrapper&& o) noexcept;
 CUFileHandleWrapper& operator=(CUFileHandleWrapper&& o) noexcept;

 std::optional<CUfileError_t> register_handle(int fd) noexcept;

 bool registered() const noexcept;

 CUfileHandle_t handle() const noexcept;

 void unregister_handle() noexcept;
};

int open_fd_parse_flags(std::string const& flags, bool o_direct);

int open_fd(std::string const& file_path, std::string const& flags, bool o_direct, mode_t mode);

[[nodiscard]] int open_flags(int fd);

[[nodiscard]] std::size_t get_file_size(int file_descriptor);

} // namespace kvikio