#include <sys_defs.h>
#include <sys/wait.h>
#include <unistd.h>
#include <stdlib.h>
#include <errno.h>
#ifdef USE_PATHS_H
#include <paths.h>
#endif
#include <syslog.h>
#include <msg.h>
#include <exec_command.h>
#include <vstream.h>
#include <argv.h>
#include <set_ugid.h>
#include <clean_env.h>
#include <iostuff.h>
typedef struct VSTREAM_POPEN_ARGS {
char **argv;
char *command;
uid_t uid;
gid_t gid;
int privileged;
char **env;
char **export;
char *shell;
VSTREAM_WAITPID_FN waitpid_fn;
} VSTREAM_POPEN_ARGS;
static void vstream_parse_args(VSTREAM_POPEN_ARGS *args, va_list ap)
{
const char *myname = "vstream_parse_args";
int key;
args->argv = 0;
args->command = 0;
args->uid = 0;
args->gid = 0;
args->privileged = 0;
args->env = 0;
args->export = 0;
args->shell = 0;
args->waitpid_fn = 0;
while ((key = va_arg(ap, int)) != VSTREAM_POPEN_END) {
switch (key) {
case VSTREAM_POPEN_ARGV:
if (args->command != 0)
msg_panic("%s: got VSTREAM_POPEN_ARGV and VSTREAM_POPEN_COMMAND", myname);
args->argv = va_arg(ap, char **);
break;
case VSTREAM_POPEN_COMMAND:
if (args->argv != 0)
msg_panic("%s: got VSTREAM_POPEN_ARGV and VSTREAM_POPEN_COMMAND", myname);
args->command = va_arg(ap, char *);
break;
case VSTREAM_POPEN_UID:
args->privileged = 1;
args->uid = va_arg(ap, uid_t);
break;
case VSTREAM_POPEN_GID:
args->privileged = 1;
args->gid = va_arg(ap, gid_t);
break;
case VSTREAM_POPEN_ENV:
args->env = va_arg(ap, char **);
break;
case VSTREAM_POPEN_EXPORT:
args->export = va_arg(ap, char **);
break;
case VSTREAM_POPEN_SHELL:
args->shell = va_arg(ap, char *);
break;
case VSTREAM_POPEN_WAITPID_FN:
args->waitpid_fn = va_arg(ap, VSTREAM_WAITPID_FN);
break;
default:
msg_panic("%s: unknown key: %d", myname, key);
}
}
if (args->command == 0 && args->argv == 0)
msg_panic("%s: missing VSTREAM_POPEN_ARGV or VSTREAM_POPEN_COMMAND", myname);
if (args->privileged != 0 && args->uid == 0)
msg_panic("%s: privileged uid", myname);
if (args->privileged != 0 && args->gid == 0)
msg_panic("%s: privileged gid", myname);
}
VSTREAM *vstream_popen(int flags,...)
{
const char *myname = "vstream_popen";
VSTREAM_POPEN_ARGS args;
va_list ap;
VSTREAM *stream;
int sockfd[2];
int pid;
int fd;
ARGV *argv;
char **cpp;
va_start(ap, flags);
vstream_parse_args(&args, ap);
va_end(ap);
if (args.command == 0)
args.command = args.argv[0];
if (duplex_pipe(sockfd) < 0)
return (0);
switch (pid = fork()) {
case -1:
(void) close(sockfd[0]);
(void) close(sockfd[1]);
return (0);
case 0:
(void) msg_cleanup((MSG_CLEANUP_FN) 0);
if (close(sockfd[1]))
msg_warn("close: %m");
for (fd = 0; fd < 2; fd++)
if (sockfd[0] != fd)
if (DUP2(sockfd[0], fd) < 0)
msg_fatal("dup2: %m");
if (sockfd[0] >= 2 && close(sockfd[0]))
msg_warn("close: %m");
if (args.privileged)
set_ugid(args.uid, args.gid);
if (args.export)
clean_env(args.export);
if (setenv("PATH", _PATH_DEFPATH, 1))
msg_fatal("%s: setenv: %m", myname);
if (args.env)
for (cpp = args.env; *cpp; cpp += 2)
if (setenv(cpp[0], cpp[1], 1))
msg_fatal("setenv: %m");
closelog();
if (args.argv) {
execvp(args.argv[0], args.argv);
msg_fatal("%s: execvp %s: %m", myname, args.argv[0]);
} else if (args.shell && *args.shell) {
argv = argv_split(args.shell, " \t\r\n");
argv_add(argv, args.command, (char *) 0);
argv_terminate(argv);
execvp(argv->argv[0], argv->argv);
msg_fatal("%s: execvp %s: %m", myname, argv->argv[0]);
} else {
exec_command(args.command);
}
default:
if (close(sockfd[0]))
msg_warn("close: %m");
stream = vstream_fdopen(sockfd[1], flags);
stream->waitpid_fn = args.waitpid_fn;
stream->pid = pid;
return (stream);
}
}
int vstream_pclose(VSTREAM *stream)
{
pid_t saved_pid = stream->pid;
VSTREAM_WAITPID_FN saved_waitpid_fn = stream->waitpid_fn;
pid_t pid;
WAIT_STATUS_T wait_status;
if (saved_pid == 0)
msg_panic("vstream_pclose: stream has no process");
stream->pid = 0;
vstream_fclose(stream);
do {
if (saved_waitpid_fn != 0)
pid = saved_waitpid_fn(saved_pid, &wait_status, 0);
else
pid = waitpid(saved_pid, &wait_status, 0);
} while (pid == -1 && errno == EINTR);
return (pid == -1 ? -1 :
WIFSIGNALED(wait_status) ? WTERMSIG(wait_status) :
WEXITSTATUS(wait_status));
}
#ifdef TEST
#include <fcntl.h>
#include <vstring.h>
#include <vstring_vstream.h>
int main(int argc, char **argv)
{
VSTRING *buf = vstring_alloc(100);
VSTREAM *stream;
int status;
if (argc < 2)
msg_fatal("usage: %s 'command'", argv[0]);
if ((stream = vstream_popen(O_RDWR,
VSTREAM_POPEN_ARGV, argv + 1,
VSTREAM_POPEN_END)) == 0)
msg_fatal("vstream_popen: %m");
while (vstring_fgets(buf, stream) != 0) {
if (vstream_fwrite(VSTREAM_OUT, vstring_str(buf), VSTRING_LEN(buf))
!= VSTRING_LEN(buf))
msg_fatal("vstream_fwrite: %m");
if (vstream_fflush(VSTREAM_OUT) != 0)
msg_fatal("vstream_fflush: %m");
if (vstring_fgets(buf, VSTREAM_IN) == 0)
break;
if (vstream_fwrite(stream, vstring_str(buf), VSTRING_LEN(buf))
!= VSTRING_LEN(buf))
msg_fatal("vstream_fwrite: %m");
}
vstring_free(buf);
if ((status = vstream_pclose(stream)) != 0)
msg_warn("exit status: %d", status);
exit(status);
}
#endif