scanner.h

Go to the documentation of this file.

#ifndef SRC_SCANNER_LIB_SCANNER_H_
#define SRC_SCANNER_LIB_SCANNER_H_
 
#include <cstdint>
 
#include <atomic>
#include <filesystem>
#include <future>
 
#include "scanner/interfaces.h"
#include "src/scanner_lib/thread_pool.h"
 
namespace scanner {
 
class Scanner final : public IScanner {
public:
  explicit Scanner(IHashDatabase& db, ILogger& logger, IFileHasher& hasher,
                   std::size_t num_threads);
 
  ScanResult Scan(const std::filesystem::path& scan_path) override;
 
private:
  void ProducerTask(const std::filesystem::path& scan_path, ThreadPool& pool,
                    std::promise<void>& producer_promise);
 
  void ConsumerTask(const std::filesystem::path& path);
 
  IHashDatabase& db_;
  ILogger& logger_;
  IFileHasher& hasher_;
  std::size_t num_threads_;
 
  std::atomic<std::uint64_t> total_files_processed_{0};
  std::atomic<std::uint64_t> malicious_files_detected_{0};
  std::atomic<std::uint64_t> errors_{0};
};
 
}  // namespace scanner
 
#endif  // SRC_SCANNER_LIB_SCANNER_H_