You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
136 lines
2.5 KiB
136 lines
2.5 KiB
2 months ago
|
/*
|
||
|
|
* UCW Library -- Running of Commands
|
||
|
|
*
|
||
|
|
* (c) 2004 Martin Mares <mj@ucw.cz>
|
||
|
|
*
|
||
|
|
* This software may be freely distributed and used according to the terms
|
||
|
|
* of the GNU Lesser General Public License.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include <ucw/lib.h>
|
||
|
|
#include <ucw/process.h>
|
||
|
|
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <string.h>
|
||
|
|
#include <alloca.h>
|
||
|
|
#include <unistd.h>
|
||
|
|
#include <sys/wait.h>
|
||
|
|
|
||
|
|
void NONRET
|
||
|
|
exec_command_v(const char *cmd, va_list args)
|
||
|
|
{
|
||
|
|
va_list cargs;
|
||
|
|
va_copy(cargs, args);
|
||
|
|
int cnt = 2;
|
||
|
|
char *arg;
|
||
|
|
while (arg = va_arg(cargs, char *))
|
||
|
|
cnt++;
|
||
|
|
va_end(cargs);
|
||
|
|
char **argv = alloca(sizeof(char *) * cnt);
|
||
|
|
argv[0] = (char *)cmd;
|
||
|
|
cnt = 1;
|
||
|
|
va_copy(cargs, args);
|
||
|
|
while (arg = va_arg(cargs, char *))
|
||
|
|
argv[cnt++] = arg;
|
||
|
|
va_end(cargs);
|
||
|
|
argv[cnt] = NULL;
|
||
|
|
execv(cmd, argv);
|
||
|
|
char echo[256];
|
||
|
|
echo_command_v(echo, sizeof(echo), cmd, args);
|
||
|
|
msg(L_ERROR, "Cannot execute %s: %m", echo);
|
||
|
|
exit(255);
|
||
|
|
}
|
||
|
|
|
||
|
|
int
|
||
|
|
run_command_v(const char *cmd, va_list args)
|
||
|
|
{
|
||
|
|
pid_t p = fork();
|
||
|
|
if (p < 0)
|
||
|
|
{
|
||
|
|
msg(L_ERROR, "fork() failed: %m");
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
else if (!p)
|
||
|
|
exec_command_v(cmd, args);
|
||
|
|
else
|
||
|
|
{
|
||
|
|
int stat;
|
||
|
|
char status_msg[EXIT_STATUS_MSG_SIZE];
|
||
|
|
p = waitpid(p, &stat, 0);
|
||
|
|
if (p < 0)
|
||
|
|
die("waitpid() failed: %m");
|
||
|
|
if (format_exit_status(status_msg, stat))
|
||
|
|
{
|
||
|
|
char echo[256];
|
||
|
|
echo_command_v(echo, sizeof(echo), cmd, args);
|
||
|
|
msg(L_ERROR, "`%s' failed: %s", echo, status_msg);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
return 1;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void
|
||
|
|
echo_command_v(char *buf, int size, const char *cmd, va_list args)
|
||
|
|
{
|
||
|
|
char *limit = buf + size - 4;
|
||
|
|
char *p = buf;
|
||
|
|
const char *arg = cmd;
|
||
|
|
do
|
||
|
|
{
|
||
|
|
int l = strlen(arg);
|
||
|
|
if (p != buf && p < limit)
|
||
|
|
*p++ = ' ';
|
||
|
|
if (p+l > limit)
|
||
|
|
{
|
||
|
|
memcpy(p, arg, limit-p);
|
||
|
|
strcpy(limit, "...");
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
memcpy(p, arg, l);
|
||
|
|
p += l;
|
||
|
|
}
|
||
|
|
while (arg = va_arg(args, char *));
|
||
|
|
*p = 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
int
|
||
|
|
run_command(const char *cmd, ...)
|
||
|
|
{
|
||
|
|
va_list args;
|
||
|
|
va_start(args, cmd);
|
||
|
|
int e = run_command_v(cmd, args);
|
||
|
|
va_end(args);
|
||
|
|
return e;
|
||
|
|
}
|
||
|
|
|
||
|
|
void NONRET
|
||
|
|
exec_command(const char *cmd, ...)
|
||
|
|
{
|
||
|
|
va_list args;
|
||
|
|
va_start(args, cmd);
|
||
|
|
exec_command_v(cmd, args);
|
||
|
|
}
|
||
|
|
|
||
|
|
void
|
||
|
|
echo_command(char *buf, int len, const char *cmd, ...)
|
||
|
|
{
|
||
|
|
va_list args;
|
||
|
|
va_start(args, cmd);
|
||
|
|
echo_command_v(buf, len, cmd, args);
|
||
|
|
va_end(args);
|
||
|
|
}
|
||
|
|
|
||
|
|
#ifdef TEST
|
||
|
|
|
||
|
|
int main(void)
|
||
|
|
{
|
||
|
|
char msg[1024];
|
||
|
|
echo_command(msg, sizeof(msg), "/bin/echo", "datel", "strakapoud", NULL);
|
||
|
|
msg(L_INFO, "Running <%s>", msg);
|
||
|
|
run_command("/bin/echo", "datel", "strakapoud", NULL);
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
#endif
|