bin/guest/cli/main.cc

// Copyright 2018 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <iostream>

#include <lib/async-loop/cpp/loop.h>
#include <lib/fit/function.h>

#include "garnet/bin/guest/cli/balloon.h"
#include "garnet/bin/guest/cli/launch.h"
#include "garnet/bin/guest/cli/list.h"
#include "garnet/bin/guest/cli/serial.h"
#include "garnet/bin/guest/cli/socat.h"
#include "lib/component/cpp/startup_context.h"
#include "lib/fxl/strings/string_number_conversions.h"

static void usage() {
  std::cerr << "Usage: guest <command> <package> <command-args>...\n"
            << "\n"
            << "Commands:\n"
            << "  balloon       <env_id> <cid> <num-pages>\n"
            << "  balloon-stats <env_id> <cid>\n"
            << "  launch        <package> <vmm-args>...\n"
            << "  list\n"
            << "  serial        <env_id> <cid>\n"
            << "  socat         <env_id> <cid> <port>\n"
            << "  socat-listen  <env_id> <host-port>\n";
}

template <class T>
static bool parse_number(const char* arg, const char* name, T* value,
                         fxl::Base base = fxl::Base::k10) {
  fxl::StringView arg_view(arg);
  if (!fxl::StringToNumberWithError(arg_view, value, base)) {
    std::cerr << "Invalid " << name << ": " << arg_view << "\n";
    return false;
  }
  return true;
}

static bool parse_args(int argc, const char** argv, async::Loop* loop,
                       component::StartupContext* context, fit::closure* func) {
  if (argc < 2) {
    return false;
  }
  fxl::StringView cmd_view(argv[1]);
  if (cmd_view == "balloon" && argc == 5) {
    uint32_t env_id, cid, num_pages;
    if (!parse_number(argv[2], "environment ID", &env_id)) {
      return false;
    } else if (!parse_number(argv[3], "context ID", &cid)) {
      return false;
    } else if (!parse_number(argv[4], "number of pages", &num_pages)) {
      return false;
    }
    *func = [env_id, cid, num_pages, context]() {
      handle_balloon(env_id, cid, num_pages, context);
    };
  } else if (cmd_view == "balloon-stats" && argc == 4) {
    uint32_t env_id, cid;
    if (!parse_number(argv[2], "environment ID", &env_id)) {
      return false;
    } else if (!parse_number(argv[3], "context ID", &cid)) {
      return false;
    }
    *func = [env_id, cid, context]() {
      handle_balloon_stats(env_id, cid, context);
    };
  } else if (cmd_view == "launch" && argc >= 3) {
    *func = [argc, argv, loop, context]() {
      handle_launch(argc - 2, argv + 2, loop, context);
    };
  } else if (cmd_view == "list") {
    *func = [context]() { handle_list(context); };
  } else if (cmd_view == "serial" && argc == 4) {
    uint32_t env_id, cid;
    if (!parse_number(argv[2], "environment ID", &env_id)) {
      return false;
    } else if (!parse_number(argv[3], "context ID", &cid)) {
      return false;
    }
    *func = [env_id, cid, loop, context]() {
      handle_serial(env_id, cid, loop, context);
    };
  } else if (cmd_view == "socat" && argc == 5) {
    uint32_t env_id, cid, port;
    if (!parse_number(argv[2], "environment ID", &env_id)) {
      return false;
    } else if (!parse_number(argv[3], "context ID", &cid)) {
      return false;
    } else if (!parse_number(argv[4], "port", &port)) {
      return false;
    }
    *func = [env_id, cid, port, loop, context]() {
      handle_socat_connect(env_id, cid, port, loop, context);
    };
  } else if (cmd_view == "socat-listen" && argc == 4) {
    uint32_t env_id, host_port;
    if (!parse_number(argv[2], "environment ID", &env_id)) {
      return false;
    } else if (!parse_number(argv[3], "host port", &host_port)) {
      return false;
    }
    *func = [env_id, host_port, loop, context]() {
      handle_socat_listen(env_id, host_port, loop, context);
    };
  } else {
    return false;
  }
  return true;
}

int main(int argc, const char** argv) {
  fit::closure func;
  async::Loop loop(&kAsyncLoopConfigAttachToThread);
  auto context = component::StartupContext::CreateFromStartupInfo();
  if (!parse_args(argc, argv, &loop, context.get(), &func)) {
    usage();
    return ZX_ERR_INVALID_ARGS;
  }

  func();
  return 0;
}