flute-receiver.cpp File Reference

#include <cstdio>
#include <iostream>
#include <argp.h>
#include <cstdlib>
#include <fstream>
#include <string>
#include <filesystem>
#include <libconfig.h++>
#include <boost/asio.hpp>
#include "spdlog/async.h"
#include "spdlog/spdlog.h"
#include "spdlog/sinks/syslog_sink.h"
#include "Version.h"
#include "Receiver.h"
#include "File.h"

Include dependency graph for flute-receiver.cpp:

Go to the source code of this file.

Classes
struct	ft_arguments
	Holds all options passed on the command line. More...

Functions
static void	print_version (FILE *stream, struct argp_state *)
	Print the program version in MAJOR.MINOR.PATCH format. More...
static auto	parse_opt (int key, char *arg, struct argp_state *state) -> error_t
	Parses the command line options into the arguments struct. More...
auto	main (int argc, char **argv) -> int
	Main entry point for the program. More...

Variables
void(*	argp_program_version_hook )(FILE *, struct argp_state *) = print_version
const char *	argp_program_bug_address = "Austrian Broadcasting Services <obeca@ors.at>"
static char	doc [] = "FLUTE/ALC receiver demo"
static struct argp_option	options []
static struct argp	argp

Function Documentation

main()

auto main	(	int	argc,
		char **	argv
	)		-> int

Main entry point for the program.

Parameters

argc	Command line agument count
argv	Command line arguments

Returns

0 on clean exit, -1 on failure

Definition at line 129 of file flute-receiver.cpp.

                                        {
  struct ft_arguments arguments;
  /* Default values */
  arguments.mcast_target = "238.1.1.95";
  arguments.flute_interface= "0.0.0.0";
 
  // Parse the arguments
  argp_parse(&argp, argc, argv, 0, nullptr, &arguments);
 
  // Set up logging
  std::string ident = "flute-receiver";
  auto syslog_logger = spdlog::syslog_logger_mt("syslog", ident, LOG_PID | LOG_PERROR | LOG_CONS );
 
  spdlog::set_level(
      static_cast<spdlog::level::level_enum>(arguments.log_level));
  spdlog::set_pattern("[%H:%M:%S.%f %z] [%^%l%$] [thr %t] %v");
 
  spdlog::set_default_logger(syslog_logger);
  spdlog::info("FLUTE receiver demo starting up");
 
  try {
    // Create a Boost io_context
    boost::asio::io_context io;
 
    // Create the receiver
    LibFlute::Receiver receiver(
        arguments.flute_interface,
        arguments.mcast_target,
        (short)arguments.mcast_port,
        arguments.tsi,
        io);
 
    // Configure IPSEC, if enabled
    if (arguments.enable_ipsec)
    {
      receiver.enable_ipsec(1, arguments.aes_key);
    }
 
    receiver.register_completion_callback(
      [output_path = arguments.output_path](std::shared_ptr<LibFlute::File> file) { //NOLINT
        std::string out_file = file->meta().content_location;
        if (output_path && std::strlen(output_path) > 0) {
          out_file = (std::filesystem::path(output_path) / std::filesystem::path(out_file).filename()).string();
        }
 
        spdlog::info("{} (TOI {}) has been received",
                     out_file, file->meta().toi);
        FILE *fd = fopen(out_file.c_str(), "wb");
        fwrite(file->buffer(), 1, file->length(), fd);
        fclose(fd);
      });
 
    // Start the IO service
    io.run();
  } catch (std::exception ex ) {
    spdlog::error("Exiting on unhandled exception: %s", ex.what());
  }
 
exit:
  return 0;
}

parse_opt()

static auto parse_opt ( int  key, char *  arg, struct argp_state *  state  ) -> error_t

static

Parses the command line options into the arguments struct.

Definition at line 77 of file flute-receiver.cpp.

                                                                               {
  auto arguments = static_cast<struct ft_arguments *>(state->input);
  switch (key) {
    case 'm':
      arguments->mcast_target = arg;
      break;
    case 'i':
      arguments->flute_interface = arg;
      break;
    case 'k':
      arguments->aes_key = arg;
      arguments->enable_ipsec = true;
      break;
    case 'p':
      arguments->mcast_port = static_cast<unsigned short>(strtoul(arg, nullptr, 10));
      break;
    case 'l':
      arguments->log_level = static_cast<unsigned>(strtoul(arg, nullptr, 10));
      break;
    case 'T':
      arguments->tsi = static_cast<uint64_t>(strtoul(arg, nullptr, 10));
      break;
    case 'o':
      arguments->output_path = arg;
      break;
    default:
      return ARGP_ERR_UNKNOWN;
  }
  return 0;
}

print_version()

void print_version ( FILE *  stream, struct argp_state *  state  )

static

Print the program version in MAJOR.MINOR.PATCH format.

Definition at line 114 of file flute-receiver.cpp.

                                                        {
  fprintf(stream, "%s.%s.%s\n", std::to_string(VERSION_MAJOR).c_str(),
          std::to_string(VERSION_MINOR).c_str(),
          std::to_string(VERSION_PATCH).c_str());
}

Variable Documentation

argp

struct argp argp

static

Initial value:

= {options, parse_opt, nullptr, doc,
                           nullptr, nullptr,   nullptr}

Definition at line 77 of file flute-receiver.cpp.

argp_program_bug_address

const char* argp_program_bug_address = "Austrian Broadcasting Services <obeca@ors.at>"

Definition at line 43 of file flute-receiver.cpp.

argp_program_version_hook

void(* argp_program_version_hook) (FILE *, struct argp_state *)	(	FILE *	,
		struct argp_state *
	)		= print_version

Definition at line 42 of file flute-receiver.cpp.

doc

char doc[] = "FLUTE/ALC receiver demo"

static

Definition at line 44 of file flute-receiver.cpp.

options

struct argp_option options[]

static

Initial value:

= {  
    {"interface", 'i', "IF", 0, "IP address of the interface to bind flute receivers to (default: 0.0.0.0)", 0},
    {"target", 'm', "IP", 0, "Multicast address to receive on (default: 238.1.1.95)", 0},
    {"port", 'p', "PORT", 0, "Multicast port (default: 40085)", 0},
    {"ipsec-key", 'k', "KEY", 0, "To enable IPSec/ESP decryption of packets, provide a hex-encoded AES key here", 0},
    {"log-level", 'l', "LEVEL", 0,
     "Log verbosity: 0 = trace, 1 = debug, 2 = info, 3 = warn, 4 = error, 5 = "
     "critical, 6 = none. Default: 2.",
     0},
    {"tsi", 'T', "ID", 0, "The TSI to use for the FLUTE session (default: 16)", 0},
    {"output-path", 'o', "PATH", 0, "Directory to save received files", 0},
    {nullptr, 0, nullptr, 0, nullptr, 0}}

Definition at line 44 of file flute-receiver.cpp.