void sinsp_chisel::set_args(vector<pair<string, string>> args)
{
#ifdef HAS_LUA_CHISELS
uint32_t j;
uint32_t n_required_args = get_n_required_args();
uint32_t n_optional_args = get_n_optional_args();
ASSERT(m_ls);
//
// Validate the arguments
//
if(args.size() < n_required_args)
{
throw sinsp_exception("wrong number of parameters for chisel " + m_filename +
", " + to_string((long long int)n_required_args) + " required, " +
to_string((long long int)args.size()) + " given");
}
else if(args.size() > n_optional_args + n_required_args)
{
throw sinsp_exception("too many parameters for chisel " + m_filename +
", " + to_string((long long int)(n_required_args)) + " required, " +
to_string((long long int)(n_optional_args)) + " optional, " +
to_string((long long int)args.size()) + " given");
}
//
// Push the arguments
//
for(j = 0; j < args.size(); j++)
{
lua_getglobal(m_ls, "on_set_arg");
if(!lua_isfunction(m_ls, -1))
{
lua_pop(m_ls, 1);
throw sinsp_exception("chisel " + m_filename + " misses a set_arg() function.");
}
lua_pushstring(m_ls, args[j].first.c_str());
lua_pushstring(m_ls, args[j].second.c_str());
//
// call get_info()
//
if(lua_pcall(m_ls, 2, 1, 0) != 0)
{
throw sinsp_exception(m_filename + " chisel error: " + lua_tostring(m_ls, -1));
}
if(!lua_isboolean(m_ls, -1))
{
throw sinsp_exception(m_filename + " chisel error: wrong set_arg() return value.");
}
int sares = lua_toboolean(m_ls, -1);
if(!sares)
{
throw sinsp_exception("set_arg() for chisel " + m_filename + " failed.");
}
lua_pop(m_ls, 1);
}
#endif
}
void sinsp_chisel::set_args(string args)
{
#ifdef HAS_LUA_CHISELS
uint32_t j;
uint32_t n_required_args = get_n_required_args();
uint32_t n_optional_args = get_n_optional_args();
ASSERT(m_ls);
//
// Split the argument string into tokens
//
uint32_t token_begin = 0;
bool inquotes = false;
uint32_t quote_correction = 0;
trim(args);
if(args.size() != 0)
{
for(j = 0; j < args.size(); j++)
{
if(args[j] == ' ' && !inquotes)
{
m_argvals.push_back(args.substr(token_begin, j - quote_correction - token_begin));
token_begin = j + 1;
quote_correction = 0;
}
else if(args[j] == '\'' || args[j] == '`')
{
if(inquotes)
{
quote_correction = 1;
inquotes = false;
}
else {
token_begin++;
inquotes = true;
}
}
}
if(inquotes)
{
throw sinsp_exception("corrupted parameters for chisel " + m_filename);
}
m_argvals.push_back(args.substr(token_begin, j - quote_correction - token_begin));
}
//
// Validate the arguments
//
if(m_argvals.size() < n_required_args)
{
throw sinsp_exception("wrong number of parameters for chisel " + m_filename +
", " + to_string((long long int)n_required_args) + " required, " +
to_string((long long int)m_argvals.size()) + " given");
}
else if(m_argvals.size() > n_optional_args + n_required_args)
{
throw sinsp_exception("too many parameters for chisel " + m_filename +
", " + to_string((long long int)(n_required_args)) + " required, " +
to_string((long long int)(n_optional_args)) + " optional, " +
to_string((long long int)m_argvals.size()) + " given");
}
//
// Push the arguments
//
for(j = 0; j < m_argvals.size(); j++)
{
lua_getglobal(m_ls, "on_set_arg");
if(!lua_isfunction(m_ls, -1))
{
lua_pop(m_ls, 1);
throw sinsp_exception("chisel " + m_filename + " misses a set_arg() function.");
}
lua_pushstring(m_ls, m_lua_script_info.m_args[j].m_name.c_str());
lua_pushstring(m_ls, m_argvals[j].c_str());
//
// call get_info()
//
if(lua_pcall(m_ls, 2, 1, 0) != 0)
{
throw sinsp_exception(m_filename + " chisel error: " + lua_tostring(m_ls, -1));
}
if(!lua_isboolean(m_ls, -1))
{
throw sinsp_exception(m_filename + " chisel error: wrong set_arg() return value.");
}
int sares = lua_toboolean(m_ls, -1);
if(!sares)
{
throw sinsp_exception("set_arg() for chisel " + m_filename + " failed.");
}
lua_pop(m_ls, 1);
}
#endif
}
void sinsp_chisel::set_args(string args)
{
#ifdef HAS_LUA_CHISELS
uint32_t j;
uint32_t n_required_args = get_n_required_args();
uint32_t n_optional_args = get_n_optional_args();
ASSERT(m_ls);
//
// Split the argument string into tokens
//
uint32_t token_begin = 0;
bool inquotes = false;
uint32_t quote_correction = 0;
trim(args);
if(args.size() != 0)
{
for(j = 0; j < args.size(); j++)
{
if(args[j] == ' ' && !inquotes)
{
m_argvals.push_back(args.substr(token_begin, j - quote_correction - token_begin));
token_begin = j + 1;
quote_correction = 0;
}
else if(args[j] == '\'' || args[j] == '`')
{
if(inquotes)
{
quote_correction = 1;
inquotes = false;
}
else {
token_begin++;
inquotes = true;
}
}
}
if(inquotes)
{
throw sinsp_exception("corrupted parameters for chisel " + m_filename);
}
m_argvals.push_back(args.substr(token_begin, j - quote_correction - token_begin));
}
//
// Validate the arguments
//
if(m_argvals.size() < n_required_args)
{
throw sinsp_exception("wrong number of parameters for chisel " + m_filename +
", " + to_string((long long int)n_required_args) + " required, " +
to_string((long long int)m_argvals.size()) + " given");
}
else if(m_argvals.size() > n_optional_args + n_required_args)
{
throw sinsp_exception("too many parameters for chisel " + m_filename +
", " + to_string((long long int)(n_required_args)) + " required, " +
to_string((long long int)(n_optional_args)) + " optional, " +
to_string((long long int)m_argvals.size()) + " given");
}
//
// Create the arguments vector
//
vector<pair<string, string>> vargs;
for(j = 0; j < m_argvals.size(); j++)
{
vargs.push_back(pair<string, string>(m_lua_script_info.m_args[j].m_name,
m_argvals[j]));
}
set_args(vargs);
#endif
}
