command_line_parser.h

Go to the documentation of this file.

//
// Bareflank Hypervisor
//
// Copyright (C) 2015 Assured Information Security, Inc.
// Author: Rian Quinn        <quinnr@ainfosec.com>
// Author: Brendan Kerrigan  <kerriganb@ainfosec.com>
//
// This library is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation; either
// version 2.1 of the License, or (at your option) any later version.
//
// This library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
// Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public
// License along with this library; if not, write to the Free Software
// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

#ifndef COMMAND_LINE_PARSER_H
#define COMMAND_LINE_PARSER_H

#include <vector>
#include <string>

#include <file.h>
#include <ioctl.h>

namespace command_line_parser_command
{
enum type
{
    help = 1,
    load = 2,
    unload = 3,
    start = 4,
    stop = 5,
    dump = 6,
    status = 7,
    vmcall = 8
};
}

class command_line_parser
{
public:

    using registers_type = ioctl::registers_type;
    using arg_type = std::string;
    using arg_list_type = std::vector<arg_type>;
    using filename_type = file::filename_type;
    using cpuid_type = ioctl::cpuid_type;
    using vcpuid_type = ioctl::vcpuid_type;
    using command_type = command_line_parser_command::type;

    command_line_parser();

    virtual ~command_line_parser() = default;

    virtual void parse(const arg_list_type &args);

    virtual command_type cmd() const noexcept;

    virtual const filename_type &modules() const noexcept;

    virtual cpuid_type cpuid() const noexcept;

    virtual vcpuid_type vcpuid() const noexcept;

    virtual const registers_type &registers() const noexcept;

    virtual const filename_type &ifile() const noexcept;

    virtual const filename_type &ofile() const noexcept;

private:

    void reset() noexcept;

    void parse_load(arg_list_type &args);
    void parse_unload(arg_list_type &args);
    void parse_start(arg_list_type &args);
    void parse_stop(arg_list_type &args);
    void parse_dump(arg_list_type &args);
    void parse_status(arg_list_type &args);
    void parse_vmcall(arg_list_type &args);

    void parse_vmcall_version(arg_list_type &args);
    void parse_vmcall_registers(arg_list_type &args);
    void parse_vmcall_string(arg_list_type &args);
    void parse_vmcall_data(arg_list_type &args);
    void parse_vmcall_event(arg_list_type &args);
    void parse_vmcall_unittest(arg_list_type &args);

    void parse_vmcall_string_unformatted(arg_list_type &args);
    void parse_vmcall_string_json(arg_list_type &args);

    void parse_vmcall_data_unformatted(arg_list_type &args);

private:

    command_type m_cmd;
    filename_type m_modules;
    cpuid_type m_cpuid;
    vcpuid_type m_vcpuid;
    registers_type m_registers;
    filename_type m_ifile;
    filename_type m_ofile;
    arg_type m_string_data;
};

#endif