service.h

Go to the documentation of this file.

#ifndef SNIFF_SERVICE
#define SNIFF_SERVICE
 
#include "decrypter.h"
#include "proto/service.grpc.pb.h"
#include "proto/service.pb.h"
#include "sniffer.h"
#include "uuid.h"
#include <filesystem>
#include <grpcpp/server_context.h>
#include <grpcpp/support/sync_stream.h>
#include <optional>
#include <tins/sniffer.h>
#include <unordered_map>
 
#include "database.h"
 
namespace yarilo {
 
class Service : public proto::Sniffer::Service {
public:
  struct config {
    bool save_on_shutdown;
    std::filesystem::path saves_path;
    std::filesystem::path db_file;
    std::filesystem::path oid_file;
    std::filesystem::path battery_file_path;
    std::vector<MACAddress> ignored_bssids;
  };
 
  Service(const config &cfg);
 
  std::optional<uuid::UUIDv4>
  add_file_sniffer(const std::filesystem::path &file);
  std::optional<uuid::UUIDv4> add_iface_sniffer(const std::string &iface_name);
  void shutdown();
  void clean_save_dir();
 
  grpc::Status SnifferCreate(grpc::ServerContext *context,
                             const proto::SnifferCreateRequest *request,
                             proto::SnifferID *reply) override;
 
  grpc::Status SnifferDestroy(grpc::ServerContext *context,
                              const proto::SnifferID *request,
                              proto::Empty *reply) override;
 
  grpc::Status SnifferList(grpc::ServerContext *context,
                           const proto::Empty *request,
                           proto::SnifferListResponse *reply) override;
 
  grpc::Status AccessPointList(grpc::ServerContext *context,
                               const proto::SnifferID *request,
                               proto::APListResponse *reply) override;
 
  grpc::Status AccessPointGet(grpc::ServerContext *context,
                              const proto::APGetRequest *request,
                              proto::APGetResponse *reply) override;
 
  grpc::Status
  AccessPointProvidePassword(grpc::ServerContext *context,
                             const proto::APProvidePasswordRequest *request,
                             proto::APProvidePasswordResponse *reply) override;
 
  grpc::Status AccessPointGetDeryptedStream(
      grpc::ServerContext *context,
      const proto::APGetDeryptedStreamRequest *request,
      grpc::ServerWriter<proto::Packet> *writer) override;
 
  grpc::Status AccessPointDeauth(grpc::ServerContext *context,
                                 const proto::APDeauthRequest *request,
                                 proto::Empty *reply) override;
 
  grpc::Status
  AccessPointDeauthClient(grpc::ServerContext *context,
                          const proto::APDeauthClientRequest *request,
                          proto::Empty *reply) override;
 
  grpc::Status AccessPointGetHash(grpc::ServerContext *context,
                                  const proto::APGetHashRequest *request,
                                  proto::APGetHashResponse *reply) override;
 
  grpc::Status AccessPointIgnore(grpc::ServerContext *context,
                                 const proto::APIgnoreRequest *request,
                                 proto::Empty *reply) override;
 
  grpc::Status AccessPointListIgnored(grpc::ServerContext *context,
                                      const proto::SnifferID *request,
                                      proto::APListResponse *reply) override;
 
  grpc::Status
  AccessPointCreateRecording(grpc::ServerContext *context,
                             const proto::APCreateRecordingRequest *request,
                             proto::APCreateRecordingResponse *reply) override;
 
  grpc::Status FocusStart(grpc::ServerContext *context,
                          const proto::FocusStartRequest *request,
                          proto::FocusStartResponse *reply) override;
 
  grpc::Status FocusGetActive(grpc::ServerContext *context,
                              const proto::SnifferID *request,
                              proto::FocusGetActiveResponse *reply) override;
 
  grpc::Status FocusStop(grpc::ServerContext *context,
                         const proto::SnifferID *request,
                         proto::Empty *reply) override;
 
  grpc::Status RecordingCreate(grpc::ServerContext *context,
                               const proto::RecordingCreateRequest *request,
                               proto::RecordingCreateResponse *reply) override;
 
  grpc::Status RecordingList(grpc::ServerContext *context,
                             const proto::RecordingListRequest *request,
                             proto::RecordingListResponse *reply) override;
 
  grpc::Status
  RecordingLoadDecrypted(grpc::ServerContext *context,
                         const proto::RecordingLoadDecryptedRequest *request,
                         grpc::ServerWriter<proto::Packet> *writer) override;
 
  grpc::Status
  NetworkInterfaceList(grpc::ServerContext *context,
                       const proto::Empty *request,
                       proto::NetworkInterfaceListResponse *reply) override;
 
  grpc::Status
  LogGetStream(grpc::ServerContext *context, const proto::Empty *request,
               grpc::ServerWriter<proto::LogEntry> *writer) override;
 
  grpc::Status BatteryGetLevel(grpc::ServerContext *context,
                               const proto::Empty *request,
                               proto::BatteryGetLevelResponse *reply) override;
 
private:
  std::optional<std::unique_ptr<PacketChannel>>
  get_recording_stream(const uuid::UUIDv4 &uuid);
 
  std::unordered_map<uuid::UUIDv4, std::unique_ptr<Sniffer>> sniffers;
  std::unordered_map<uuid::UUIDv4, std::unique_ptr<Sniffer>>
      erased_sniffers; // Kept for shutdown logic
  std::shared_ptr<spdlog::logger> logger;
  const config cfg;
  Database db;
};
 
} // namespace yarilo
 
#endif // SNIFF_SERVICE