db.h

Go to the documentation of this file.

/// \file db.h
/// \brief Обёртка SQLite-базы данных и утилиты парсинга CIDR.
/// \details Предоставляет RAII-обёртку над SQLite3 (BlockerDB),
///          парсер CIDR-нотации и конвертер IP-адресов.
 
#ifndef DB_H
#define DB_H
 
#include <cstdint>
#include <sqlite3.h>
#include <string>
#include <vector>
 
/// Одна запись (IP/CIDR + комментарий) для whitelist или blacklist.
struct BlockEntry {
  std::string ip; ///< IP-адрес или CIDR-нотация, например "10.0.0.0/8".
  std::string comment; ///< Человекочитаемый комментарий.
};
 
/// Распарсенное представление CIDR-подсети.
struct CIDR {
  uint32_t ip; ///< IP-адрес в network byte order с применённой маской.
  uint32_t prefixlen; ///< Длина префикса (0-32). 32 означает одиночный хост.
};
 
/// Парсит CIDR-строку (например "192.168.1.0/24") в структуру \ref CIDR.
/// @param[in] str  Входная строка в формате "IP" или "IP/длина_префикса".
/// @return         Распарсенный CIDR (ip в network byte order, с маской).
/// @throws std::invalid_argument при некорректном IP или длине префикса.
CIDR ParseCIDR(const std::string &str);
 
/// Конвертирует 32-битный IP (network byte order) в dotted-decimal строку.
/// @param[in] ip  IP-адрес в network byte order.
/// @return        Строка вида "192.168.1.1".
std::string IPToString(uint32_t ip);
 
/// RAII-обёртка над SQLite3 для хранения белого и чёрного списков.
///
/// Управляет подключением к базе данных, создаёт таблицы при первом запуске
/// и предоставляет методы для чтения и обновления записей.
class BlockerDB {
public:
  /// Открывает (или создаёт) базу данных по указанному пути.
  /// @param[in] path  Путь к файлу SQLite-базы данных.
  /// @throws std::runtime_error если не удалось открыть БД.
  explicit BlockerDB(const std::string &path);
  ~BlockerDB();
 
  BlockerDB(const BlockerDB &) = delete;
  BlockerDB &operator=(const BlockerDB &) = delete;
  BlockerDB(BlockerDB &&) = delete;
  BlockerDB &operator=(BlockerDB &&) = delete;
 
  /// Создаёт таблицы, если они не существуют; применяет миграции схемы.
  ///
  /// Включает WAL-режим журнала. Идемпотентна.
  /// @throws std::runtime_error если SQL миграции завершился ошибкой.
  void Migrate();
 
  /// Возвращает все записи белого списка из базы данных.
  /// @return Вектор записей BlockEntry.
  std::vector<BlockEntry> GetWhitelist();
  /// Возвращает все записи чёрного списка из базы данных.
  /// @return Вектор записей BlockEntry.
  std::vector<BlockEntry> GetBlacklist();
 
  /// Обновляет last_blocked_at текущим временем для указанного IP.
  ///
  /// Также обновляет любую CIDR-запись, покрывающую этот IP.
  /// @param[in] ip  IP-адрес источника в network byte order.
  void UpdateLastBlockedAt(uint32_t ip);
 
private:
  sqlite3 *db_; ///< Указатель на дескриптор SQLite3.
  /// Обобщённый SELECT ip, comment из указанной таблицы.
  /// @param[in] table  Имя таблицы ("whitelist_ips" или "blocked_ips").
  /// @return           Вектор записей BlockEntry.
  std::vector<BlockEntry> GetEntries(const std::string &table);
};
 
#endif