Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit b3fe88bf authored by Abhishek Pandit-Subedi's avatar Abhishek Pandit-Subedi
Browse files

Add source_root parameter to bluetooth_packetgen 

In packetgen, add a source-root parameter that we use to rebase paths
from. This was necessary to get the same directory output structure on
Linux builds with GN.

Bug: 176847256
Tag: #floss
Test: atest --host bluetooth_test_gd
Change-Id: I75cdb1693655356ad1470d29a8a45440c775df8f
parent 423a2e3d

system/gd/packet/parser/main.cc

+41 −5
Original line number Diff line number Diff line
@@ -19,6 +19,7 @@ 
#include <cstdio>

#include <filesystem>

#include <fstream>

#include <iomanip>

#include <iostream>

#include <queue>

#include <regex>
@@ -102,39 +103,74 @@ extern "C" const char* __asan_default_options() { 
  return "detect_leaks=0";

}



void usage(const char* prog) {

  auto& ofs = std::cerr;



  ofs << "Usage: " << prog << " [OPTIONS] file1 file2..." << std::endl;



  ofs << std::setw(24) << "--out= ";

  ofs << "Root directory for generated output (relative to cwd)." << std::endl;



  ofs << std::setw(24) << "--include= ";

  ofs << "Generate namespaces relative to this path per file." << std::endl;



  ofs << std::setw(24) << "--root_namespace= ";

  ofs << "Change root namespace (default = bluetooth)." << std::endl;



  ofs << std::setw(24) << "--source_root= ";

  ofs << "Root path to the source directory. Find input files relative to this." << std::endl;



  ofs << std::setw(24) << "--num_shards= ";

  ofs << "Number of shards per output pybind11 cc file." << std::endl;

}



int main(int argc, const char** argv) {

  std::filesystem::path out_dir;

  std::filesystem::path include_dir;

  std::filesystem::path cwd = std::filesystem::current_path();

  std::filesystem::path source_root = cwd;

  std::string root_namespace = "bluetooth";

  // Number of shards per output pybind11 cc file

  size_t num_shards = 1;

  bool generate_rust = false;

  std::queue<std::filesystem::path> input_files;



  const std::string arg_out = "--out=";

  const std::string arg_include = "--include=";

  const std::string arg_namespace = "--root_namespace=";

  const std::string arg_num_shards = "--num_shards=";

  const std::string arg_rust = "--rust";

  const std::string arg_source_root = "--source_root=";



  // Parse the source root first (if it exists) since it will be used for other

  // paths.

  for (int i = 1; i < argc; i++) {

    std::string arg = argv[i];

    if (arg.find(arg_source_root) == 0) {

      source_root = std::filesystem::path(arg.substr(arg_source_root.size()));

    }

  }



  for (int i = 1; i < argc; i++) {

    std::string arg = argv[i];

    if (arg.find(arg_out) == 0) {

      out_dir = std::filesystem::current_path() / std::filesystem::path(arg.substr(arg_out.size()));

      out_dir = cwd / std::filesystem::path(arg.substr(arg_out.size()));

    } else if (arg.find(arg_include) == 0) {

      include_dir = std::filesystem::current_path() / std::filesystem::path(arg.substr(arg_include.size()));

      include_dir = source_root / std::filesystem::path(arg.substr(arg_include.size()));

    } else if (arg.find(arg_namespace) == 0) {

      root_namespace = arg.substr(arg_namespace.size());

    } else if (arg.find(arg_num_shards) == 0) {

      num_shards = std::stoul(arg.substr(arg_num_shards.size()));

    } else if (arg.find(arg_rust) == 0) {

      generate_rust = true;

    } else if (arg.find(arg_source_root) == 0) {

      // Do nothing (just don't treat it as input_files)

    } else {

      input_files.emplace(std::filesystem::current_path() / std::filesystem::path(arg));

      input_files.emplace(source_root / std::filesystem::path(arg));

    }

  }

  if (out_dir == std::filesystem::path() || include_dir == std::filesystem::path() || num_shards == 0) {

    std::cerr << "Usage: bt-packetgen --out=OUT --include=INCLUDE --root_namespace=NAMESPACE --num_shards=NUM_SHARDS "

              << "input_files..." << std::endl;

    usage(argv[0]);

    return 1;

  }