nis-util-services/main.cc

Go to the documentation of this file.

//
// nis-util - NIS Administration Utilities
// Copyright (C) 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/ac/assert.h>
#include <libexplain/output.h>

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


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

        case arglex_services::token_check:
            cmdline.get_filename_or_stdin(by_check);
            break;

        case arglex_services::token_by_name:
            cmdline.get_filename_or_stdin(by_name);
            break;

        case arglex_services::token_by_servicename:
            cmdline.get_filename_or_stdin(by_servicename);
            break;

        case arglex_services::token_output:
            if (!output.empty())
            {
                explain_output_error("too many output files specified");
                cmdline.usage();
            }
            output = cmdline.get_filename_or_stdout();
            break;

        case arglex_services::token_protocols:
            cmdline.get_filename_or_stdin(etc_protocols);
            break;
        }
        cmdline.token_next();
    }
    if (!by_name.empty() + !by_servicename.empty() + !by_check.empty() != 1)
    {
        explain_output_error
        (
            "exactly one of %s or %s or %s must be specified",
            cmdline.token_name(arglex_services::token_check),
            cmdline.token_name(arglex_services::token_by_name),
            cmdline.token_name(arglex_services::token_by_servicename)
        );
        cmdline.usage();
    }
    if (!by_name.empty())
        map_by_name(by_name, output);
    else if (!by_servicename.empty())
        map_by_servicename(by_servicename, output);
    else
    {
        assert(!by_check.empty());
        if (etc_protocols.empty())
            explain_output_error_and_die("no protocols file specified");
        check(by_check, etc_protocols);
    }
    return 0;
}


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