NetBSD-5.0.2/dist/atf/tools/atf-compile.cpp

//
// Automated Testing Framework (atf)
//
// Copyright (c) 2007, 2008 The NetBSD Foundation, Inc.
// All rights reserved.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions
// are met:
// 1. Redistributions of source code must retain the above copyright
//    notice, this list of conditions and the following disclaimer.
// 2. Redistributions in binary form must reproduce the above copyright
//    notice, this list of conditions and the following disclaimer in the
//    documentation and/or other materials provided with the distribution.
//
// THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND
// CONTRIBUTORS ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
// INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
// MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
// IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS BE LIABLE FOR ANY
// DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
// DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
// GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
// IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
// OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
// IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
//

extern "C" {
#include <sys/types.h>
#include <sys/stat.h>
}

#include <cstdlib>
#include <fstream>
#include <iostream>

#include "atf-c++/application.hpp"
#include "atf-c++/config.hpp"
#include "atf-c++/sanity.hpp"

static
void
cat_file(std::ostream& os, const std::string& path)
{
    std::ifstream is(path.c_str());
    if (!is)
        throw std::runtime_error("Cannot open " + path);
    std::string line;
    while (std::getline(is, line))
        os << line << std::endl;
    is.close();
}

class atf_compile : public atf::application::app {
    static const char* m_description;

    std::string m_outfile;

    std::string specific_args(void) const;
    options_set specific_options(void) const;
    void process_option(int ch, const char*);

    void compile(std::ostream&);

public:
    atf_compile(void);

    int main(void);
};

const char* atf_compile::m_description =
    "atf-compile is a tool that ``compiles'' test cases written in the "
    "POSIX shell language, generating an executable.";

atf_compile::atf_compile(void) :
    app(m_description, "atf-compile(1)", "atf(7)")
{
}

std::string
atf_compile::specific_args(void)
    const
{
    return "<test-program>";
}

atf_compile::options_set
atf_compile::specific_options(void)
    const
{
    using atf::application::option;
    options_set opts;
    opts.insert(option('o', "out-file", "Name of the output file"));
    return opts;
}

void
atf_compile::process_option(int ch, const char* arg)
{
    switch (ch) {
    case 'o':
        m_outfile = arg;
        break;

    default:
        UNREACHABLE;
    }
}

void
atf_compile::compile(std::ostream& os)
{
    os << "#! " << atf::config::get("atf_shell") << std::endl;
    cat_file(os, atf::config::get("atf_pkgdatadir") + "/atf.init.subr");
    os << std::endl;
    os << ". ${Atf_Pkgdatadir}/atf.header.subr" << std::endl;
    os << std::endl;
    for (int i = 0; i < m_argc; i++) {
        cat_file(os, m_argv[i]);
        os << std::endl;
    }
    os << ". ${Atf_Pkgdatadir}/atf.footer.subr" << std::endl;
    os << std::endl;
    os << "main \"${@}\"" << std::endl;
}

int
atf_compile::main(void)
{
    if (m_argc < 1)
        throw atf::application::usage_error("No test program specified");

    if (m_outfile.empty())
        throw atf::application::usage_error("No output file specified");

    std::ofstream os(m_outfile.c_str());
    if (!os)
        throw std::runtime_error("Cannot open output file `" +
                                 m_outfile + "'");
    compile(os);
    os.close();

    mode_t umask = ::umask(S_IRWXU | S_IRWXG | S_IRWXO);
    ::umask(umask);

    if (::chmod(m_outfile.c_str(),
                (S_IRWXU | S_IRWXG | S_IRWXO) & ~umask) == -1)
        throw std::runtime_error("Cannot set executable permissions "
                                 "on `" + m_outfile + "'");

    return EXIT_SUCCESS;
}

int
main(int argc, char* const* argv)
{
    return atf_compile().run(argc, argv);
}