lib/input/string.cc

Go to the documentation of this file.

//
// nis-util - NIS Administration Utilities
// Copyright (C) 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/string.h>

#include <lib/input/null.h>
#include <lib/input/string.h>


input_string::~input_string()
{
}


input_string::input_string(const rcstring &a_filename, const rcstring &a_text) :
    filename(a_filename),
    line_number(1),
    text(a_text),
    tpos(0)
{
}


input::pointer
input_string::create(const rcstring &a_filename, const rcstring &a_text)
{
    if (a_text.empty())
        return input_null::create();
    return pointer(new input_string(a_filename, a_text));
}


input_string::pointer
input_string::create(const rcstring &a_filename, const char *a_text)
{
    if (!a_text || !*a_text)
        return input_null::create();
    return pointer(new input_string(a_filename, a_text));
}


input_string::pointer
input_string::create_env(const rcstring &environment_variable_name)
{
    if (environment_variable_name.empty())
        return input_null::create();
    const char *a_text = getenv(environment_variable_name.c_str());
    if (!a_text || !*a_text)
        return input_null::create();
    rcstring a_filename = "$" + environment_variable_name;
    return create(a_filename, a_text);
}


size_t
input_string::read_inner(source_location &data_locn, void *data,
    size_t data_size)
{
    data_locn = source_location(filename, line_number);
    if (tpos >= text.size())
        return 0;
    size_t nbytes = text.size() - tpos;
    if (nbytes > data_size)
        nbytes = data_size;
    memcpy(data, text.c_str() + tpos, nbytes);
    tpos += nbytes;
    line_number += count_newlines(data, nbytes);
    return nbytes;
}


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