aos/network/message_bridge_client.cc

#include "aos/events/shm_event_loop.h"
#include "aos/init.h"
#include "aos/logging/dynamic_logging.h"
#include "aos/network/message_bridge_client_lib.h"
#include "aos/network/sctp_lib.h"
#include "aos/sha256.h"
#include "aos/util/file.h"

DEFINE_string(config, "aos_config.json", "Path to the config.");
DEFINE_int32(rt_priority, -1, "If > 0, run as this RT priority");

#if HAS_SCTP_AUTH
DEFINE_string(sctp_auth_key_file, "",
              "When set, use the provided key for SCTP authentication as "
              "defined in RFC 4895. The file should be binary-encoded");
#endif

namespace aos {
namespace message_bridge {

using ::aos::util::ReadFileToVecOrDie;

int Main() {
  aos::FlatbufferDetachedBuffer<aos::Configuration> config =
      aos::configuration::ReadConfig(FLAGS_config);

  aos::ShmEventLoop event_loop(&config.message());
  if (FLAGS_rt_priority > 0) {
    event_loop.SetRuntimeRealtimePriority(FLAGS_rt_priority);
  }

  std::vector<uint8_t> auth_key;
#if HAS_SCTP_AUTH
  if (!FLAGS_sctp_auth_key_file.empty()) {
    auth_key = ReadFileToVecOrDie(FLAGS_sctp_auth_key_file);
  }
#endif
  MessageBridgeClient app(&event_loop, Sha256(config.span()),
                          std::move(auth_key));

  logging::DynamicLogging dynamic_logging(&event_loop);
  // TODO(austin): Save messages into a vector to be logged.  One file per
  // channel?  Need to sort out ordering.
  //
  // TODO(austin): Low priority, "reliable" logging channel.

  event_loop.Run();

  return EXIT_SUCCESS;
}

}  // namespace message_bridge
}  // namespace aos

int main(int argc, char **argv) {
  aos::InitGoogle(&argc, &argv);

  return aos::message_bridge::Main();
}