X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/c5f062cd6510b042c8fab3725f95a0588369b089..4fe07d7c76c7c9006780389a5babb524560b3ff0:/projects/stargazer/plugins/other/radius/config.cpp diff --git a/projects/stargazer/plugins/other/radius/config.cpp b/projects/stargazer/plugins/other/radius/config.cpp index 23339c02..108b61e6 100644 --- a/projects/stargazer/plugins/other/radius/config.cpp +++ b/projects/stargazer/plugins/other/radius/config.cpp @@ -1,208 +1,146 @@ -/* - * 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, write to the Free Software - * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA - */ - -/* - * Author : Maxim Mamontov - */ - #include "config.h" - +#include "radproto/error.h" #include "stg/common.h" +#include +#include #include -#include - -#include // strncasecmp +#include +#include +#include +#include using STG::Config; +using AttrValue = Config::AttrValue; +using ASection = Config::ASection; namespace { + std::string ShowRules(const std::vector>& rules) + { + std::string result; + for (const auto& at : rules) + { + if (!result.empty()) + result += ", "; -struct ParserError : public std::runtime_error -{ - ParserError(size_t pos, const std::string& message) - : runtime_error("Parsing error at position " + x2str(pos) + ". " + message), - position(pos), - error(message) - {} - virtual ~ParserError() throw() {} - - size_t position; - std::string error; -}; - -size_t skipSpaces(const std::string& value, size_t start) -{ - while (start < value.length() && std::isspace(value[start])) - ++start; - return start; + if (at.second.type == AttrValue::Type::PARAM_NAME) + result.append(at.first + " = " + at.second.value); + else + result.append(at.first + " = '" + at.second.value + "'"); + } + return result; + } } -size_t checkChar(const std::string& value, size_t start, char ch) +std::vector> Config::ParseRules(const std::string& value, const std::string& paramName) { - if (start >= value.length()) - throw ParserError(start, "Unexpected end of string. Expected '" + std::string(1, ch) + "'."); - if (value[start] != ch) - throw ParserError(start, "Expected '" + std::string(1, ch) + "', got '" + std::string(1, value[start]) + "'."); - return start + 1; -} + using tokenizer = boost::tokenizer>; + const boost::char_separator sep(","); -std::pair readString(const std::string& value, size_t start) -{ - std::string dest; - while (start < value.length() && !std::isspace(value[start]) && - value[start] != ',' && value[start] != '(' && value[start] != ')') - dest.push_back(value[start++]); - if (dest.empty()) { - if (start == value.length()) - throw ParserError(start, "Unexpected end of string. Expected string."); - else - throw ParserError(start, "Unexpected whitespace. Expected string."); - } - return std::make_pair(start, dest); -} + const tokenizer tokens(value, sep); -Config::Pairs toPairs(const std::vector& values) -{ - if (values.empty()) - return Config::Pairs(); - std::string value(values[0]); - Config::Pairs res; - size_t start = 0; - while (start < value.size()) { - Config::Pair pair; - start = skipSpaces(value, start); - if (!res.empty()) + std::vector> res; + + for (const auto& token : tokens) + { + std::vector keyValue; + + split(keyValue, boost::algorithm::trim_copy_if(token, boost::is_any_of(" \t")), boost::is_any_of(" ="), boost::token_compress_on); + + if (keyValue.size() != 2) + { + m_logger("The '%s' attribute specification has an incorrect format: '%s'.", paramName.c_str(), token.c_str()); + printfd(__FILE__, "The '%s' attribute specification has an incorrect format: '%s'.", paramName.c_str(), token.c_str()); + return {}; + } + + auto type = AttrValue::Type::PARAM_NAME; + std::string valueName = keyValue[1]; + if (valueName.front() == '\'' && valueName.back() == '\'') + { + type = AttrValue::Type::VALUE; + valueName.erase(0, 1); + valueName.erase(valueName.length() - 1, 1); + } + else if ((valueName.front() == '\'' && valueName.back() != '\'') || (valueName.front() != '\'' && valueName.back() == '\'')) { - start = checkChar(value, start, ','); - start = skipSpaces(value, start); + m_logger("Error ParseRules: '%s' attribute parameter value is invalid.\n", paramName.c_str()); + printfd(__FILE__, "Error ParseRules: '%s' attribute parameter value is invalid.\n", paramName.c_str()); + return {}; } - size_t pairStart = start; - start = checkChar(value, start, '('); - const std::pair key = readString(value, start); - start = key.first; - pair.first = key.second; - start = skipSpaces(value, start); - start = checkChar(value, start, ','); - start = skipSpaces(value, start); - const std::pair val = readString(value, start); - start = val.first; - pair.second = val.second; - start = skipSpaces(value, start); - start = checkChar(value, start, ')'); - if (res.find(pair.first) != res.end()) - throw ParserError(pairStart, "Duplicate field."); - res.insert(pair); + res.emplace_back(keyValue[0], AttrValue{valueName, type}); } return res; } -bool toBool(const std::vector& values) +ASection Config::parseASection(const std::vector& conf) { - if (values.empty()) - return false; - std::string value(values[0]); - return strncasecmp(value.c_str(), "yes", 3) == 0; -} + ASection res; + const auto mit = std::find(conf.begin(), conf.end(), ParamValue("match", {})); + if (mit != conf.end()) + res.match = ParseRules(mit->value[0], mit->param); -std::string toString(const std::vector& values) -{ - if (values.empty()) - return ""; - return values[0]; -} + const auto sit = std::find(conf.begin(), conf.end(), ParamValue("send", {})); + if (sit != conf.end()) + res.send = ParseRules(sit->value[0], sit->param); -template -T toInt(const std::vector& values) -{ - if (values.empty()) - return 0; - T res = 0; - if (str2x(values[0], res) == 0) - return res; - return 0; + return res; } -Config::Pairs parseVector(const std::string& paramName, const MODULE_SETTINGS& params) -{ - for (size_t i = 0; i < params.moduleParams.size(); ++i) - if (params.moduleParams[i].param == paramName) - return toPairs(params.moduleParams[i].value); - return Config::Pairs(); -} +Config::Config() + : m_port(1812), + m_dictionaries("/usr/share/freeradius/dictionary"), + m_logger(PluginLogger::get("radius")) +{} -bool parseBool(const std::string& paramName, const MODULE_SETTINGS& params) +int Config::ParseSettings(const ModuleSettings & s) { - for (size_t i = 0; i < params.moduleParams.size(); ++i) - if (params.moduleParams[i].param == paramName) - return toBool(params.moduleParams[i].value); - return false; -} + ParamValue pv; + int p; -std::string parseString(const std::string& paramName, const MODULE_SETTINGS& params) -{ - for (size_t i = 0; i < params.moduleParams.size(); ++i) - if (params.moduleParams[i].param == paramName) - return toString(params.moduleParams[i].value); - return ""; -} + pv.param = "Port"; + auto pvi = std::find(s.moduleParams.begin(), s.moduleParams.end(), pv); + if (pvi != s.moduleParams.end() && !pvi->value.empty()) + { + if (ParseIntInRange(pvi->value[0], 2, 65535, &p) != 0) + { + m_errorStr = "Cannot parse parameter \'Port\': " + m_errorStr; + printfd(__FILE__, "Cannot parse parameter 'Port'\n"); + return -1; + } + m_port = static_cast(p); + } -template -T parseInt(const std::string& paramName, const MODULE_SETTINGS& params) -{ - for (size_t i = 0; i < params.moduleParams.size(); ++i) - if (params.moduleParams[i].param == paramName) - return toInt(params.moduleParams[i].value); - return 0; -} + pv.param = "Secret"; + pvi = std::find(s.moduleParams.begin(), s.moduleParams.end(), pv); + if (pvi == s.moduleParams.end() || pvi->value.empty()) + { + m_errorStr = "Parameter \'Secret\' not found."; + printfd(__FILE__, "Parameter 'Secret' not found\n"); + return -1; + } + else + m_secret = pvi->value[0]; -std::string parseAddress(const std::string& address) -{ - size_t pos = address.find_first_of(':'); - if (pos == std::string::npos) - throw ParserError(0, "Connection type is not specified. Should be either 'unix' or 'tcp'."); - return address.substr(pos + 1); -} + pv.param = "Dictionaries"; + pvi = std::find(s.moduleParams.begin(), s.moduleParams.end(), pv); + if (pvi != s.moduleParams.end() && !pvi->value.empty()) + m_dictionaries = pvi->value[0]; -Config::Type parseConnectionType(const std::string& address) -{ - size_t pos = address.find_first_of(':'); - if (pos == std::string::npos) - throw ParserError(0, "Connection type is not specified. Should be either 'unix' or 'tcp'."); - std::string type = ToLower(address.substr(0, pos)); - if (type == "unix") - return Config::UNIX; - else if (type == "tcp") - return Config::TCP; - throw ParserError(0, "Invalid connection type. Should be either 'unix' or 'tcp', got '" + type + "'"); -} + const auto authIt = std::find(s.moduleParams.begin(), s.moduleParams.end(), ParamValue("auth", {})); + if (authIt != s.moduleParams.end()) + m_auth = parseASection(authIt->sections); -} // namespace anonymous - -Config::Config(const MODULE_SETTINGS& settings) - : match(parseVector("match", settings)), - modify(parseVector("modify", settings)), - reply(parseVector("reply", settings)), - verbose(parseBool("verbose", settings)), - address(parseString("bind_address", settings)), - bindAddress(parseAddress(address)), - connectionType(parseConnectionType(address)), - portStr(parseString("port", settings)), - port(parseInt("port", settings)), - key(parseString("key", settings)) -{ + const auto autzIt = std::find(s.moduleParams.begin(), s.moduleParams.end(), ParamValue("autz", {})); + if (autzIt != s.moduleParams.end()) + m_autz = parseASection(autzIt->sections); + + printfd(__FILE__, " auth.match = \"%s\"\n", ShowRules(m_auth.match).c_str()); + printfd(__FILE__, " auth.send = \"%s\"\n", ShowRules(m_auth.send).c_str()); + printfd(__FILE__, " autz.match = \"%s\"\n", ShowRules(m_autz.match).c_str()); + printfd(__FILE__, " autz.send = \"%s\"\n", ShowRules(m_autz.send).c_str()); + + return 0; } +