121 lines
3.4 KiB
C
121 lines
3.4 KiB
C
/*
|
|
* The Qubes OS Project, http://www.qubes-os.org
|
|
*
|
|
* Copyright (C) 2010 Rafal Wojtczuk <rafal@invisiblethingslab.com>
|
|
*
|
|
* 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 2
|
|
* 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, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
*
|
|
*/
|
|
#define _GNU_SOURCE
|
|
#include <sys/socket.h>
|
|
#include <sys/un.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
#include <fcntl.h>
|
|
#include <string.h>
|
|
#include "qrexec.h"
|
|
int connect_unix_socket()
|
|
{
|
|
int s, len;
|
|
struct sockaddr_un remote;
|
|
|
|
if ((s = socket(AF_UNIX, SOCK_STREAM, 0)) == -1) {
|
|
perror("socket");
|
|
return -1;
|
|
}
|
|
|
|
remote.sun_family = AF_UNIX;
|
|
strncpy(remote.sun_path, QREXEC_AGENT_FDPASS_PATH,
|
|
sizeof(remote.sun_path));
|
|
len = strlen(remote.sun_path) + sizeof(remote.sun_family);
|
|
if (connect(s, (struct sockaddr *) &remote, len) == -1) {
|
|
perror("connect");
|
|
exit(1);
|
|
}
|
|
return s;
|
|
}
|
|
|
|
char *get_program_name(char *prog)
|
|
{
|
|
char *basename = rindex(prog, '/');
|
|
if (basename)
|
|
return basename + 1;
|
|
else
|
|
return prog;
|
|
}
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
int trigger_fd;
|
|
struct trigger_connect_params params;
|
|
int local_fd[3], remote_fd[3];
|
|
int i;
|
|
char *abs_exec_path;
|
|
|
|
if (argc < 4) {
|
|
fprintf(stderr,
|
|
"usage: %s target_vmname program_ident local_program [local program arguments]\n",
|
|
argv[0]);
|
|
exit(1);
|
|
}
|
|
|
|
trigger_fd = open(QREXEC_AGENT_TRIGGER_PATH, O_WRONLY);
|
|
if (trigger_fd < 0) {
|
|
perror("open " QREXEC_AGENT_TRIGGER_PATH);
|
|
exit(1);
|
|
}
|
|
|
|
for (i = 0; i < 3; i++) {
|
|
local_fd[i] = connect_unix_socket();
|
|
if (read(local_fd[i], &remote_fd[i], sizeof(remote_fd[i])) != sizeof(remote_fd[i])) {
|
|
perror("read client fd");
|
|
exit(1);
|
|
}
|
|
if (i != 2 || getenv("PASS_LOCAL_STDERR")) {
|
|
char *env;
|
|
if (asprintf(&env, "SAVED_FD_%d=%d", i, dup(i)) < 0) {
|
|
perror("prepare SAVED_FD_");
|
|
exit(1);
|
|
}
|
|
putenv(env);
|
|
dup2(local_fd[i], i);
|
|
close(local_fd[i]);
|
|
}
|
|
}
|
|
|
|
memset(¶ms, 0, sizeof(params));
|
|
strncpy(params.exec_index, argv[2], sizeof(params.exec_index));
|
|
strncpy(params.target_vmname, argv[1],
|
|
sizeof(params.target_vmname));
|
|
snprintf(params.process_fds.ident,
|
|
sizeof(params.process_fds.ident), "%d %d %d",
|
|
remote_fd[0], remote_fd[1], remote_fd[2]);
|
|
|
|
if (write(trigger_fd, ¶ms, sizeof(params)) < 0) {
|
|
if (!getenv("PASS_LOCAL_STDERR"))
|
|
perror("write to agent");
|
|
exit(1);
|
|
}
|
|
|
|
close(trigger_fd);
|
|
|
|
abs_exec_path = strdup(argv[3]);
|
|
argv[3] = get_program_name(argv[3]);
|
|
execv(abs_exec_path, argv + 3);
|
|
perror("execv");
|
|
return 1;
|
|
}
|