wmbusmeters/shell.cc

76 wiersze
2.2 KiB
C++

/*
Copyright (C) 2017-2018 Fredrik Öhrström
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 3 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 "shell.h"
#include "util.h"
#include <memory.h>
#include <pthread.h>
#include <sys/types.h>
#include <wait.h>
#include <unistd.h>
void invokeShell(string program, vector<string> args, vector<string> envs)
{
vector<const char*> argv(args.size()+2);
argv[0] = program.c_str();
int i = 1;
debug("exec \"%s\"\n", program.c_str());
for (auto &a : args) {
argv[i] = a.c_str();
i++;
debug("arg \"%s\"\n", a.c_str());
}
argv[i] = NULL;
vector<const char*> env(envs.size()+1);
env[0] = program.c_str();
i = 0;
for (auto &e : envs) {
env[i] = e.c_str();
i++;
debug("env \"%s\"\n", e.c_str());
}
env[i] = NULL;
pid_t pid = fork();
int status;
if (pid == 0) {
// I am the child!
close(0); // Close stdin
execvpe(program.c_str(), (char*const*)&argv[0], (char*const*)&env[0]);
perror("Execvp failed:");
error("Invoking shell %s failed!\n", program.c_str());
} else {
if (pid == -1) {
error("Could not fork!\n");
}
debug("waiting for child %d.\n", pid);
// Wait for the child to finish!
waitpid(pid, &status, 0);
if (WIFEXITED(status)) {
// Child exited properly.
int rc = WEXITSTATUS(status);
debug("%s: return code %d\n", program.c_str(), rc);
if (rc != 0) {
warning("%s exited with non-zero return code: %d\n", program.c_str(), rc);
}
}
}
}