nis-util-group-members/main.cc

Go to the documentation of this file.

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

#include <lib/colon/group/slurp.h>
#include <lib/colon/passwd/functor/group_closure.h>
#include <lib/colon/passwd/slurp.h>
#include <lib/output/file.h>

#include <nis-util-group-members/arglex/group-members.h>


int
main(int argc, char **argv)
{
    arglex_group_members cmdline(argc, argv);
    cmdline.token_first();
    typedef std::list<rcstring> group_names_t;
    group_names_t group_names;
    rcstring passwd_file_name;
    rcstring group_file_name;
    rcstring output_filename;
    while (cmdline.token_cur() != arglex::token_eoln)
    {
        switch (cmdline.token_cur())
        {
        default:
            cmdline.generic_argument();
            break;

        case arglex_group_members::token_passwd:
            cmdline.get_filename_or_stdin(passwd_file_name);
            break;

        case arglex_group_members::token_group:
            cmdline.get_filename_or_stdin(group_file_name);
            break;

        case arglex::token_string:
            group_names.push_back(cmdline.get_string_value());
            break;

        case arglex_group_members::token_output:
            cmdline.get_filename_or_stdout(output_filename);
            break;
        }
        cmdline.token_next();
    }

    //
    // Read the password file.
    //
    if (passwd_file_name.empty())
        explain_output_error_and_die("no passwd file specified");
    colon_passwd_slurp::pointer psw =
        colon_passwd_slurp::create(passwd_file_name);
    psw->read_and_process();

    //
    // Read the group file.
    //
    if (group_file_name.empty())
        explain_output_error_and_die("no group file specified");
    colon_group_slurp::pointer grp = colon_group_slurp::create(group_file_name);
    grp->read_and_process();

    // For each user, add that user to the group correponding to the
    // user's default gid.
    {
        colon_passwd_functor_group_closure w(grp);
        psw->walk(w);
    }

    output::pointer op = output_file::create(output_filename);

    //
    // for each group named, emit its members
    //
    if (group_names.empty())
    {
        explain_output_error("no group names specified");
        cmdline.usage();
    }
    int number_of_errors = 0;
    for
    (
        group_names_t::const_iterator git = group_names.begin();
        git != group_names.end();
        ++git
    )
    {
        rcstring name = *git;
        colon_group::record::pointer gp = grp->query_by_name(name);
        if (!gp)
        {
            explain_output_error
            (
                "%s: group %s unknown",
                group_file_name.c_str(),
                name.quote_c().c_str()
            );
            ++number_of_errors;
        }

        //
        // emit the names
        //
        op->put(name);
        op->put(':');
        const colon_group::record::members_t &members = gp->get_members();
        for
        (
            colon_group::record::members_t::const_iterator it =
                members.begin();
            it != members.end();
            ++it
        )
        {
            if (it != members.begin())
                op->put(',');
            op->put(' ');
            op->put(*it);
        }
        op->put('\n');
    }
    op.reset();

    if (number_of_errors > 0)
    {
        explain_output_error_and_die
        (
            "found %d fatal error%s",
            number_of_errors,
            (number_of_errors == 1 ? "" : "s")
        );
    }
    return 0;
}


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