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());
 
  set(pb_Envelope_message_v1_tag, pb_MessageV1_flash_init_command_tag, new FlasherInitHandler());
  set(pb_Envelope_message_v1_tag, pb_MessageV1_flash_data_command_tag, new FlasherDataHandler());
  set(pb_Envelope_message_v1_tag, pb_MessageV1_flash_complete_command_tag, new FlasherCompleteHandler());
 
  //TODO: remove or implement with protobuf
  // manual hardware access
  //set("manual_mode", 900, new ManualControlHandler(), SecurityLevel::RequiresNothing);
  //set("overload_status", 1000, new GetOverloadStatusHandler(c), SecurityLevel::RequiresLogin);
//
  //set("net_get", 6000, new GetNetworkSettingsHandler(), SecurityLevel::RequiresAdmin);
  //set("net_set", 6100, new SetNetworkSettingsHandler(), SecurityLevel::RequiresAdmin);
  //set("net_reset", 6200, new ResetNetworkSettingsHandler(), SecurityLevel::RequiresAdmin);
  //set("net_status", 6300, new NetworkStatusHandler(), SecurityLevel::RequiresNothing);
  // ^ this net_status won't contain sensitive information...
 
  // set("status",            3000, new GetSystemStatus(), SecurityLevel::RequiresNothing);
  //set("login", 3100, new LoginHandler(), SecurityLevel::RequiresNothing);
  //set("lock_acquire", 3200, new LockAcquire(), SecurityLevel::RequiresLogin);
  //set("lock_release", 3300, new LockRelease(), SecurityLevel::RequiresLogin);
 
  //set("sys_ident", 3400, new GetSystemIdent(), SecurityLevel::RequiresNothing);
  //set("sys_reboot", 3500, new RebootHandler(), SecurityLevel::RequiresAdmin);
  //set("sys_log", 3600, new SyslogHandler(), SecurityLevel::RequiresLogin);
  //set("sys_stats", 3700, new SystemStats(), SecurityLevel::RequiresLogin);
  //
//
  //set("load_plugin", 4100, new LoadPluginHandler(), SecurityLevel::RequiresAdmin);
  //set("unload_plugin", 4200, new UnloadPluginHandler(), SecurityLevel::RequiresAdmin);
//
  //#ifdef ANABRID_WRITE_EEPROM
  //set("sys_read_permanent",  4300, new ReadSystemIdent(), SecurityLevel::RequiresAdmin);
  //set("sys_reset_permanent", 4400, new ResetSystemIdent(), SecurityLevel::RequiresAdmin);
  //set("sys_write_permanent", 4500, new WriteSystemIdent(), SecurityLevel::RequiresAdmin);
  //#endif
//
  //set("ota_update_status", 5000, new FlasherStatusHandler(), SecurityLevel::RequiresAdmin);
  //set("ota_update_init", 5000, new FlasherInitHandler(), SecurityLevel::RequiresAdmin);
  //set("ota_update_stream", 5100, new FlasherDataHandler(), SecurityLevel::RequiresAdmin);
  //set("ota_update_abort", 5200, new FlasherAbortHandler(), SecurityLevel::RequiresAdmin);
  //set("ota_update_complete", 5300, new FlasherCompleteHandler(), SecurityLevel::RequiresAdmin);
}
 
static uint64_t kind2key(uint32_t envelope_kind, uint32_t message_kind) {
  return static_cast<uint64_t>(envelope_kind) << 32 | message_kind;
}
 
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;
}