argdata_parser.cc

// Copyright (c) 2017 Nuxi (https://nuxi.nl/) and contributors.
//
// SPDX-License-Identifier: BSD-2-Clause

#include <cstdlib>

#include <argdata.h>
#include <arpc++/arpc++.h>

using namespace arpc;

ArgdataParser::ArgdataParser(argdata_reader_t* reader) : reader_(reader) {
}

ArgdataParser::~ArgdataParser() {
  // File descriptors that have been parsed are now owned by the
  // messages containing them. Allow the reader to close any file
  // descriptors attached to the message, except those that have been
  // handed out by us.
  if (reader_ != nullptr)
    for (const auto& file_descriptor : file_descriptors_)
      argdata_reader_release_fd(reader_, file_descriptor->get());
}

const argdata_t* ArgdataParser::ParseAnyFromMap(
    const argdata_map_iterator_t& it) {
  const argdata_t *key, *value;
  if (!argdata_map_get(&maps_.emplace_front(it), &key, &value))
    std::abort();
  return value;
}

std::shared_ptr<FileDescriptor> ArgdataParser::ParseFileDescriptor(
    const argdata_t& ad) {
  // Parse file descriptor object.
  int fd;
  if (argdata_get_fd(&ad, &fd) != 0)
    return nullptr;

  // Try to return an existing shared pointer instance. Create a new
  // instance if none exists.
  auto lookup = file_descriptors_.find(fd);
  if (lookup != file_descriptors_.end())
    return *lookup;
  return *file_descriptors_.insert(std::make_shared<FileDescriptor>(fd)).first;
}