390 lines
11 KiB
C++
390 lines
11 KiB
C++
/*
|
|
* Copyright (c) 2016 MariaDB Corporation Ab
|
|
*
|
|
* Use of this software is governed by the Business Source License included
|
|
* in the LICENSE.TXT file and at www.mariadb.com/bsl11.
|
|
*
|
|
* Change Date: 2022-01-01
|
|
*
|
|
* On the date above, in accordance with the Business Source License, use
|
|
* of this software will be governed by version 2 or later of the General
|
|
* Public License.
|
|
*/
|
|
|
|
#include "internal/externcmd.hh"
|
|
|
|
#include <ctype.h>
|
|
#include <errno.h>
|
|
#include <string.h>
|
|
#include <sys/wait.h>
|
|
#include <fcntl.h>
|
|
#include <thread>
|
|
|
|
#include <maxbase/assert.h>
|
|
#include <maxbase/alloc.h>
|
|
#include <maxscale/pcre2.h>
|
|
|
|
using std::string;
|
|
|
|
int ExternalCmd::tokenize_args(char* dest[], int dest_size)
|
|
{
|
|
bool quoted = false;
|
|
bool read = false;
|
|
bool escaped = false;
|
|
char qc = 0;
|
|
|
|
char args[m_subst_command.length() + 1];
|
|
strcpy(args, m_subst_command.c_str());
|
|
char* start = args;
|
|
char* ptr = start;
|
|
int i = 0;
|
|
|
|
while (*ptr != '\0' && i < dest_size)
|
|
{
|
|
if (escaped)
|
|
{
|
|
escaped = false;
|
|
}
|
|
else
|
|
{
|
|
if (*ptr == '\\')
|
|
{
|
|
escaped = true;
|
|
}
|
|
else if (quoted && !escaped && *ptr == qc) /** End of quoted string */
|
|
{
|
|
*ptr = '\0';
|
|
dest[i++] = MXS_STRDUP(start);
|
|
read = false;
|
|
quoted = false;
|
|
}
|
|
else if (!quoted)
|
|
{
|
|
if (isspace(*ptr))
|
|
{
|
|
*ptr = '\0';
|
|
if (read) /** New token */
|
|
{
|
|
dest[i++] = MXS_STRDUP(start);
|
|
read = false;
|
|
}
|
|
}
|
|
else if (*ptr == '\"' || *ptr == '\'')
|
|
{
|
|
/** New quoted token, strip quotes */
|
|
quoted = true;
|
|
qc = *ptr;
|
|
start = ptr + 1;
|
|
}
|
|
else if (!read)
|
|
{
|
|
start = ptr;
|
|
read = true;
|
|
}
|
|
}
|
|
}
|
|
ptr++;
|
|
}
|
|
if (read)
|
|
{
|
|
dest[i++] = MXS_STRDUP(start);
|
|
}
|
|
return i;
|
|
}
|
|
|
|
std::unique_ptr<ExternalCmd> ExternalCmd::create(const string& argstr, int timeout)
|
|
{
|
|
bool success = false;
|
|
std::unique_ptr<ExternalCmd> cmd(new ExternalCmd(argstr, timeout));
|
|
char* argvec[1] {}; // Parse just one argument for testing file existence and permissions.
|
|
if (cmd->tokenize_args(argvec, 1) > 0)
|
|
{
|
|
const char* cmdname = argvec[0];
|
|
if (access(cmdname, X_OK) != 0)
|
|
{
|
|
if (access(cmdname, F_OK) != 0)
|
|
{
|
|
MXS_ERROR("Cannot find file '%s'.", cmdname);
|
|
}
|
|
else
|
|
{
|
|
MXS_ERROR("Cannot execute file '%s'. Missing execution permissions.", cmdname);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
success = true;
|
|
}
|
|
MXS_FREE(argvec[0]);
|
|
}
|
|
else
|
|
{
|
|
MXS_ERROR("Failed to parse argument string '%s' for external command.", argstr.c_str());
|
|
}
|
|
|
|
if (!success)
|
|
{
|
|
cmd.reset();
|
|
}
|
|
return cmd;
|
|
}
|
|
|
|
ExternalCmd::ExternalCmd(const std::string& script, int timeout)
|
|
: m_orig_command(script)
|
|
, m_subst_command(script)
|
|
, m_timeout(timeout)
|
|
{
|
|
}
|
|
|
|
static const char* skip_whitespace(const char* ptr)
|
|
{
|
|
while (*ptr && isspace(*ptr))
|
|
{
|
|
ptr++;
|
|
}
|
|
|
|
return ptr;
|
|
}
|
|
|
|
static const char* skip_prefix(const char* str)
|
|
{
|
|
const char* ptr = strchr(str, ':');
|
|
mxb_assert(ptr);
|
|
|
|
ptr++;
|
|
return skip_whitespace(ptr);
|
|
}
|
|
|
|
static void log_output(const char* cmd, const std::string& str)
|
|
{
|
|
int err;
|
|
|
|
if (mxs_pcre2_simple_match("(?i)^[[:space:]]*alert[[:space:]]*[:]",
|
|
str.c_str(),
|
|
0,
|
|
&err) == MXS_PCRE2_MATCH)
|
|
{
|
|
MXS_ALERT("%s: %s", cmd, skip_prefix(str.c_str()));
|
|
}
|
|
else if (mxs_pcre2_simple_match("(?i)^[[:space:]]*error[[:space:]]*[:]",
|
|
str.c_str(),
|
|
0,
|
|
&err) == MXS_PCRE2_MATCH)
|
|
{
|
|
MXS_ERROR("%s: %s", cmd, skip_prefix(str.c_str()));
|
|
}
|
|
else if (mxs_pcre2_simple_match("(?i)^[[:space:]]*warning[[:space:]]*[:]",
|
|
str.c_str(),
|
|
0,
|
|
&err) == MXS_PCRE2_MATCH)
|
|
{
|
|
MXS_WARNING("%s: %s", cmd, skip_prefix(str.c_str()));
|
|
}
|
|
else if (mxs_pcre2_simple_match("(?i)^[[:space:]]*notice[[:space:]]*[:]",
|
|
str.c_str(),
|
|
0,
|
|
&err) == MXS_PCRE2_MATCH)
|
|
{
|
|
MXS_NOTICE("%s: %s", cmd, skip_prefix(str.c_str()));
|
|
}
|
|
else if (mxs_pcre2_simple_match("(?i)^[[:space:]]*(info|debug)[[:space:]]*[:]",
|
|
str.c_str(),
|
|
0,
|
|
&err) == MXS_PCRE2_MATCH)
|
|
{
|
|
MXS_INFO("%s: %s", cmd, skip_prefix(str.c_str()));
|
|
}
|
|
else
|
|
{
|
|
// No special format, log as notice level message
|
|
MXS_NOTICE("%s: %s", cmd, skip_whitespace(str.c_str()));
|
|
}
|
|
}
|
|
|
|
int ExternalCmd::externcmd_execute()
|
|
{
|
|
// Create a pipe where the command can print output
|
|
int fd[2];
|
|
if (pipe(fd) == -1)
|
|
{
|
|
MXS_ERROR("Failed to open pipe: [%d] %s", errno, mxs_strerror(errno));
|
|
return -1;
|
|
}
|
|
|
|
// "execvp" takes its arguments as an array of tokens where the first element is the command.
|
|
char* argvec[MAX_ARGS + 1] {};
|
|
tokenize_args(argvec, MAX_ARGS);
|
|
const char* cmdname = argvec[0];
|
|
|
|
int rval = 0;
|
|
// The SIGCHLD handler must be disabled before child process is forked,
|
|
// otherwise we'll get an error
|
|
pid_t pid = fork();
|
|
if (pid < 0)
|
|
{
|
|
close(fd[0]);
|
|
close(fd[1]);
|
|
MXS_ERROR("Failed to execute command '%s', fork failed: [%d] %s",
|
|
cmdname, errno, mxs_strerror(errno));
|
|
rval = -1;
|
|
}
|
|
else if (pid == 0)
|
|
{
|
|
// This is the child process. Close the read end of the pipe and redirect
|
|
// both stdout and stderr to the write end of the pipe
|
|
close(fd[0]);
|
|
dup2(fd[1], STDOUT_FILENO);
|
|
dup2(fd[1], STDERR_FILENO);
|
|
|
|
// Execute the command
|
|
execvp(cmdname, argvec);
|
|
|
|
// Close the write end of the pipe and exit
|
|
close(fd[1]);
|
|
_exit(1);
|
|
}
|
|
else
|
|
{
|
|
MXS_INFO("Executing command '%s' in process %d", cmdname, pid);
|
|
|
|
string output;
|
|
bool first_warning = true;
|
|
bool again = true;
|
|
uint64_t t = 0;
|
|
uint64_t t_max = m_timeout * 1000;
|
|
|
|
// Close the write end of the pipe and make the read end non-blocking
|
|
close(fd[1]);
|
|
fcntl(fd[0], F_SETFL, O_NONBLOCK);
|
|
|
|
while (again)
|
|
{
|
|
int exit_status;
|
|
|
|
switch (waitpid(pid, &exit_status, WNOHANG))
|
|
{
|
|
case -1:
|
|
MXS_ERROR("Failed to wait for child process: %d, %s", errno, mxs_strerror(errno));
|
|
again = false;
|
|
break;
|
|
|
|
case 0:
|
|
if (t++ > t_max)
|
|
{
|
|
// Command timed out
|
|
t = 0;
|
|
if (first_warning)
|
|
{
|
|
MXS_WARNING("Soft timeout for command '%s', sending SIGTERM", cmdname);
|
|
kill(pid, SIGTERM);
|
|
first_warning = false;
|
|
}
|
|
else
|
|
{
|
|
MXS_ERROR("Hard timeout for command '%s', sending SIGKILL", cmdname);
|
|
kill(pid, SIGKILL);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
// Sleep and try again
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(1));
|
|
}
|
|
break;
|
|
|
|
default:
|
|
again = false;
|
|
|
|
if (WIFEXITED(exit_status))
|
|
{
|
|
rval = WEXITSTATUS(exit_status);
|
|
}
|
|
else if (WIFSIGNALED(exit_status))
|
|
{
|
|
rval = WTERMSIG(exit_status);
|
|
}
|
|
else
|
|
{
|
|
rval = exit_status;
|
|
MXS_ERROR("Command '%s' did not exit normally. Exit status: %d", cmdname, exit_status);
|
|
}
|
|
break;
|
|
}
|
|
|
|
int n;
|
|
char buf[4096]; // This seems like enough space
|
|
|
|
while ((n = read(fd[0], buf, sizeof(buf))) > 0)
|
|
{
|
|
// Read all available output
|
|
output.append(buf, n);
|
|
|
|
for (size_t pos = output.find("\n"); pos != std::string::npos; pos = output.find("\n"))
|
|
{
|
|
if (pos == 0)
|
|
{
|
|
output.erase(0, 1);
|
|
}
|
|
else
|
|
{
|
|
std::string line = output.substr(0, pos);
|
|
output.erase(0, pos + 1);
|
|
log_output(cmdname, line);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if (!output.empty())
|
|
{
|
|
log_output(cmdname, output);
|
|
}
|
|
|
|
// Close the read end of the pipe and copy the data to the output parameter
|
|
close(fd[0]);
|
|
}
|
|
|
|
// Free the token array.
|
|
for (int i = 0; i < MAX_ARGS && argvec[i]; i++)
|
|
{
|
|
MXS_FREE(argvec[i]);
|
|
}
|
|
return rval;
|
|
}
|
|
|
|
void ExternalCmd::substitute_arg(const std::string& match, const std::string& replace)
|
|
{
|
|
// The match may be in the subject multiple times. Find all locations.
|
|
string::size_type next_search_begin = 0;
|
|
while (next_search_begin < m_subst_command.length())
|
|
{
|
|
auto position = m_subst_command.find(match, next_search_begin);
|
|
if (position == string::npos)
|
|
{
|
|
next_search_begin = m_subst_command.length();
|
|
}
|
|
else
|
|
{
|
|
m_subst_command.replace(position, match.length(), replace);
|
|
next_search_begin = position + replace.length();
|
|
}
|
|
}
|
|
}
|
|
|
|
void ExternalCmd::match_substitute(const string& keyword, const std::function<string(void)>& generator)
|
|
{
|
|
if (m_orig_command.find(keyword) != string::npos)
|
|
{
|
|
substitute_arg(keyword, generator());
|
|
}
|
|
}
|
|
|
|
void ExternalCmd::reset_substituted()
|
|
{
|
|
m_subst_command = m_orig_command;
|
|
}
|
|
|
|
const char* ExternalCmd::substituted() const
|
|
{
|
|
return m_subst_command.c_str();
|
|
} |