nis-util-hosts/main.cc

Go to the documentation of this file.

//
// nis-util - NIS Administration Utilities
// Copyright (C) 2001, 2002, 2008, 2009, 2011, 2012 Peter Miller
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 of the License, or (at
// your option) any later version.
//
// This program 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
// General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
//

#include <lib/config.h>
#include <libexplain/output.h>

#include <nis-util-hosts/arglex/hosts.h>
#include <nis-util-hosts/check.h>
#include <nis-util-hosts/map.h>


int
main(int argc, char **argv)
{
    arglex_hosts cmdline(argc, argv);
    cmdline.token_first();
    rcstring hosts;
    rcstring by_name;
    rcstring by_addr;
    rcstring output;
    while (cmdline.token_cur() != arglex::token_eoln)
    {
        switch (cmdline.token_cur())
        {
        default:
            cmdline.generic_argument();
            break;

        case arglex_hosts::token_check:
            if (!hosts.empty())
            {
                explain_output_error("too many hosts files specified");
                cmdline.usage();
            }
            hosts = cmdline.get_filename_or_stdin();
            break;

        case arglex_hosts::token_by_name:
            if (!by_name.empty())
            {
                explain_output_error("too many by_name files specified");
                cmdline.usage();
            }
            by_name = cmdline.get_filename_or_stdin();
            break;

        case arglex_hosts::token_by_address:
            if (!by_addr.empty())
            {
                explain_output_error("too many by_addr files specified");
                cmdline.usage();
            }
            by_addr = cmdline.get_filename_or_stdin();
            break;

        case arglex_hosts::token_output:
            if (!output.empty())
            {
                explain_output_error("too many output files specified");
                cmdline.usage();
            }
            output = cmdline.get_filename_or_stdout();
            break;
        }
        cmdline.token_next();
    }
    if (!by_name.empty())
        map_by_name(by_name, output);
    else if (!by_addr.empty())
        map_by_address(by_addr, output);
    else
    {
        if (hosts.empty())
            explain_output_error_and_die("no hosts file specified");
        check(hosts);
    }
    return 0;
}


// vim: set ts=8 sw=4 et :