// Copyright © 2019 Intel Corporation
//
// SPDX-License-Identifier: Apache-2.0
//

extern crate vmm;

#[macro_use(crate_version, crate_authors)]
extern crate clap;

use clap::{App, Arg};

use std::path::PathBuf;

use vmm::vm::*;

fn main() {
    let cmd_arguments = App::new("cloud-hypervisor")
        .version(crate_version!())
        .author(crate_authors!())
        .about("Launch a cloud-hypervisor VMM.")
        .arg(
            Arg::with_name("kernel")
                .long("kernel")
                .help("Path to kernel image (vmlinux)")
                .takes_value(true),
        )
        .arg(
            Arg::with_name("cmdline")
                .long("cmdline")
                .help("Kernel command line")
                .takes_value(true),
        )
        .arg(
            Arg::with_name("disk")
                .long("disk")
                .help("Path to VM disk image")
                .takes_value(true),
        )
        .arg(
            Arg::with_name("net")
                .long("net")
                .help("Network parameters \"tap=<if_name>,ip=<ip_addr>,mask=<net_mask>,mac=<mac_addr>\"")
                .takes_value(true),
        )
        .arg(
            Arg::with_name("cpus")
                .long("cpus")
                .help("Number of virtual CPUs")
                .takes_value(true),
        )
        .arg(
            Arg::with_name("memory")
                .long("memory")
                .help("Amount of RAM (in MB)")
                .takes_value(true),
        )
        .arg(
            Arg::with_name("rng")
                .long("rng")
                .help("Path to entropy source")
                .default_value("/dev/urandom"),
        )
        .get_matches();

    let kernel_arg = cmd_arguments
        .value_of("kernel")
        .map(PathBuf::from)
        .expect("Missing argument: kernel");
    let kernel_path = kernel_arg.as_path();

    let disk_arg = cmd_arguments
        .value_of("disk")
        .map(PathBuf::from)
        .expect("Missing argument: disk");
    let disk_path = disk_arg.as_path();

    let cmdline = cmd_arguments
        .value_of("cmdline")
        .map(std::string::ToString::to_string)
        .unwrap_or_else(String::new);

    let net_params = cmd_arguments
        .value_of("net")
        .map(std::string::ToString::to_string);

    let rng_path = match cmd_arguments.occurrences_of("rng") {
        0 => None,
        _ => Some(cmd_arguments.value_of("rng").unwrap().to_string()),
    };

    let vcpus = cmd_arguments
        .value_of("cpus")
        .and_then(|c| c.parse::<u8>().ok())
        .unwrap_or(DEFAULT_VCPUS);

    let memory = cmd_arguments
        .value_of("memory")
        .and_then(|m| m.parse::<u64>().ok())
        .unwrap_or(DEFAULT_MEMORY);

    println!(
        "Cloud Hypervisor Guest\n\tvCPUs: {}\n\tMemory: {} MB\n\tKernel: {:?}\n\tKernel cmdline: {}\n\tDisk: {:?}",
        vcpus, memory, kernel_path, cmdline, disk_path
    );

    let vm_config = VmConfig::new(
        kernel_path,
        disk_path,
        rng_path,
        cmdline,
        net_params,
        vcpus,
        memory,
    )
    .unwrap();

    if let Err(e) = vmm::boot_kernel(vm_config) {
        println!("Guest boot failed: {}", e);
    }
}