registry.cpp

Go to the documentation of this file.

// Copyright (c) 2023 anabrid GmbH
// Contact: https://www.anabrid.com/licensing/
// SPDX-License-Identifier: MIT OR GPL-2.0-or-later
 
#include <proto/main.pb.h>
#include "protocol/registry.h"
 
#include "handlers/authenticate.h"
#include "handlers/loader_flasher.h"
#include "utils/logging.h"
 
#include <handlers/carrier.h>
#include <handlers/login_lock.h>
#include <handlers/ping.h>
#include <handlers/run_manager.h>
#include <handlers/sys.h>
#include <handlers/calibration.h>
 
void msg::handlers::DynamicRegistry::init(carrier::Carrier &c) {
  using namespace net::auth;
  using namespace msg::handlers;
 
  // Stateless protocol basics
  set(pb_Envelope_generic_tag, pb_GenericMessage_ping_command_tag, new PingRequestHandler{});
  //set("help", 200, new HelpHandler(), SecurityLevel::RequiresNothing);
 
  // Carrier and RunManager things
  set(pb_Envelope_message_v1_tag, pb_MessageV1_udp_data_streaming_command_tag, new UdpDataStreamingHandler());
  set(pb_Envelope_message_v1_tag, pb_MessageV1_stand_by_command_tag, new StandByRequestHandler(c));
  set(pb_Envelope_message_v1_tag, pb_MessageV1_reset_command_tag, new ResetRequestHandler(c));
  set(pb_Envelope_message_v1_tag, pb_MessageV1_config_command_tag, new ConfigMessageHandler(c));
  set(pb_Envelope_message_v1_tag, pb_MessageV1_extract_command_tag, new ExtractMessageHandler(c));
  set(pb_Envelope_message_v1_tag, pb_MessageV1_register_external_entities_command_tag,
      new RegisterExternalEntitiesRequestHandler());
  set(pb_Envelope_message_v1_tag, pb_MessageV1_start_run_command_tag, new StartRunRequestHandler());
  set(pb_Envelope_message_v1_tag, pb_MessageV1_stop_run_command_tag, new StopRunRequestHandler());
  set(pb_Envelope_message_v1_tag, pb_MessageV1_calibration_command_tag, new CalibrationHandler(c));
 
  set(pb_Envelope_message_v1_tag, pb_MessageV1_read_temperature_command_tag, new ReadTemperaturesHandler(c));
  set(pb_Envelope_message_v1_tag, pb_MessageV1_auth_request_tag, new AuthRequestHandler());
 
  auto& updater = loader::Updater::get();
  set(pb_Envelope_message_v1_tag, pb_MessageV1_update_command_tag, new UpdateHandler(updater));
}
 
static uint64_t kind2key(uint32_t envelope_kind, uint32_t message_kind) {
  return static_cast<uint64_t>(envelope_kind) << 32 | message_kind;
}
 
FASTRUN
msg::handlers::MessageHandler *msg::handlers::DynamicRegistry::lookup(uint32_t envelope_kind,
                                                                      uint32_t message_kind) {
  auto found = protobuf_entries.find(kind2key(envelope_kind, message_kind));
  if (found == protobuf_entries.end())
    return nullptr;
 
  return found->second.handler;
}
 
bool msg::handlers::DynamicRegistry::set(uint32_t envelope_kind, uint32_t message_kind,
                                         MessageHandler *handler) {
  protobuf_entries.emplace(
    kind2key(envelope_kind, message_kind),
    RegistryEntry{handler}
  );
  return true;
}