#include <vserver.h>
#include <getopt.h>
#include <fcntl.h>
+#include <errno.h>
+#include <sys/socket.h>
+#include <sys/un.h>
+#define ENSC_WRAPPERS_PREFIX "vcontext: "
#define ENSC_WRAPPERS_UNISTD 1
#define ENSC_WRAPPERS_VSERVER 1
#define ENSC_WRAPPERS_FCNTL 1
+#define ENSC_WRAPPERS_SOCKET 1
+#define ENSC_WRAPPERS_IOSOCK 1
#include <wrappers.h>
#define CMD_HELP 0x1000
#define CMD_UID 0x4005
#define CMD_CHROOT 0x4006
#define CMD_SILENT 0x4007
+#define CMD_SYNCSOCK 0x4008
+#define CMD_SYNCMSG 0x4009
struct option const
CMDLINE_OPTIONS[] = {
{ "silent", no_argument, 0, CMD_SILENT },
{ "uid", no_argument, 0, CMD_UID },
{ "chroot", no_argument, 0, CMD_CHROOT },
+ { "syncsock", required_argument, 0, CMD_SYNCSOCK },
+ { "syncmsg", required_argument, 0, CMD_SYNCMSG },
{ 0,0,0,0 },
};
bool do_chroot;
uid_t uid;
xid_t xid;
+ char const * sync_sock;
+ char const * sync_msg;
};
int wrapper_exit_code = 255;
static void
showHelp(int fd, char const *cmd, int res)
{
- WRITE_MSG(fd, "Usage: ");
+ WRITE_MSG(fd, "Usage:\n ");
WRITE_STR(fd, cmd);
WRITE_MSG(fd,
- " [--xid <xid>] <opts>* [--] <program> <args>*\n"
- " --create [--migrate] [--xid <xid>] <opts>* [--] <program> <args>*\n"
- " --migrate --xid <xid> <opts>* [--] <program> <args>*\n"
+ " --create [--xid <xid>] <opts>* [--] <program> <args>*\n ");
+ WRITE_STR(fd, cmd);
+ WRITE_MSG(fd,
+ " --migrate --xid <xid> <opts>* [--] <program> <args>*\n"
"\n"
"<opts> can be:\n"
" --chroot ... chroot into current directory\n"
" --uid <uid> ... change uid\n"
" --fakeinit ... set current process as general process reaper\n"
" for ctx (possible for --migrate only)\n"
+ " --disconnect ... start program in background\n"
" --silent ... be silent\n"
+ " --syncsock <file-name>\n"
+ " ... before executing the program, send a message\n"
+ " to the socket and wait until it closes.\n"
+ " <file-name> must be a SOCK_STREAM unix socket\n"
+ " --syncmsg <message>\n"
+ " ... use <message> as synchronization message; by\n"
+ " default, 'ok' will be used\n"
+ "\n"
+ "'vcontext --create' exits with code 254 iff the context exists already.\n"
"\n"
"Please report bugs to " PACKAGE_BUGREPORT "\n");
WRITE_MSG(1,
"vcontext " VERSION " -- manages the creation of security contexts\n"
"This program is part of " PACKAGE_STRING "\n\n"
- "Copyright (C) 2003,2004 Enrico Scholz\n"
+ "Copyright (C) 2004 Enrico Scholz\n"
VERSION_COPYRIGHT_DISCLAIMER);
exit(0);
}
WRITE_MSG(2, "\n");
}
+static int
+connectExternalSync(char const *filename)
+{
+ int fd;
+ struct sockaddr_un addr;
+
+ if (filename==0) return -1;
+
+ ENSC_INIT_UNIX_SOCK(addr, filename);
+
+ fd = Esocket(PF_UNIX, SOCK_STREAM, 0);
+ Econnect(fd, &addr, sizeof(addr));
+
+ return fd;
+}
+
+static void
+doExternalSync(int fd, char const *msg)
+{
+ char c;
+
+ if (fd==-1) return;
+
+ if (msg) EsendAll(fd, msg, strlen(msg));
+ Eshutdown(fd, SHUT_WR);
+
+ if (TEMP_FAILURE_RETRY(recv(fd, &c, 1, MSG_NOSIGNAL))!=0) {
+ WRITE_MSG(2, "vcontext: unexpected external synchronization event\n");
+ exit(255);
+ }
+
+ Eclose(fd);
+}
+
+static inline ALWAYSINLINE int
+doit(struct Arguments const *args, char *argv[])
+{
+ int p[2];
+ pid_t pid = initSync(p, args->do_disconnect);
+
+ if (pid==0) {
+ xid_t xid;
+ int ext_sync_fd = connectExternalSync(args->sync_sock);
+
+ if (args->do_create) {
+ xid = vc_create_context(args->xid);
+ if (xid==VC_NOCTX) {
+ if (errno==EEXIST) return 254;
+ else {
+ perror(ENSC_WRAPPERS_PREFIX "vc_create_context()");
+ return wrapper_exit_code;
+ }
+ }
+ tellContext(xid, args->verbosity>=1);
+ }
+ else
+ xid = args->xid;
+
+ if (args->do_chroot)
+ Echroot(".");
+
+ if (args->uid!=(uid_t)(-1) && getuid()!=args->uid) {
+ Esetuid(args->uid);
+ if (getuid()!=args->uid) {
+ WRITE_MSG(2, "vcontext: Something went wrong while changing the UID\n");
+ exit(255);
+ }
+ }
+
+ if (args->is_fakeinit) {
+ struct vc_ctx_flags test_flags;
+ struct vc_ctx_flags flags = {
+ .flagword = S_CTX_INFO_INIT,
+ .mask = S_CTX_INFO_INIT
+ };
+
+ Evc_get_flags(xid, &test_flags);
+ if ((flags.mask & S_CTX_INFO_INIT)==0) {
+ WRITE_MSG(2, "vcontext: context has already a fakeinit-process\n");
+ exit(255);
+ }
+
+ Evc_set_flags(xid, &flags);
+ }
+
+ if (args->do_migrate)
+ Evc_migrate_context(xid);
+
+ doExternalSync(ext_sync_fd, args->sync_msg);
+ doSyncStage1(p, args->do_disconnect);
+ execvp (argv[optind],argv+optind);
+ doSyncStage2(p, args->do_disconnect);
+
+ PERROR_Q("chcontext: execvp", argv[optind]);
+ exit(255);
+ }
+
+ waitOnSync(pid, p);
+ return EXIT_SUCCESS;
+}
+
int main (int argc, char *argv[])
{
- pid_t pid;
- int p[2];
struct Arguments args = {
.do_create = false,
.do_migrate = false,
.verbosity = 1,
.uid = -1,
.xid = VC_DYNAMIC_XID,
+ .sync_msg = "ok",
};
while (1) {
case CMD_SILENT : --args.verbosity; break;
case CMD_XID : args.xid = atol(optarg); break;
case CMD_UID : args.uid = atol(optarg); break;
+ case CMD_SYNCSOCK : args.sync_sock = optarg; break;
+ case CMD_SYNCMSG : args.sync_msg = optarg; break;
default :
WRITE_MSG(2, "Try '");
}
}
- if (optind>=argc) {
- WRITE_MSG(2, "No command given; use '--help' for more information.\n");
- exit(255);
- }
-
if (!args.do_create && !args.do_migrate)
- args.do_create = args.do_migrate = true;
-
- if (!args.do_migrate && args.is_fakeinit) {
+ WRITE_MSG(2, "Neither '--create' nor '--migrate specified; try '--help' for more information\n");
+ else if (args.do_create && args.do_migrate)
+ WRITE_MSG(2, "Can not specify '--create' and '--migrate' at the same time; try '--help' for more information\n");
+ else if (!args.do_migrate && args.is_fakeinit)
WRITE_MSG(2, "'--fakeinit' is possible in combination with '--migrate' only\n");
- exit(255);
- }
-
- if (!args.do_create && args.xid==VC_DYNAMIC_XID) {
+ else if (!args.do_create && args.xid==VC_DYNAMIC_XID)
WRITE_MSG(2, "vcontext: Can not migrate to an unknown context\n");
- exit(255);
- }
-
- pid = initSync(p, args.do_disconnect);
- if (pid==0) {
- xid_t xid;
- if (args.do_create) {
- xid = Evc_create_context(args.xid);
- tellContext(xid, args.verbosity>=1);
- }
- else
- xid = args.xid;
-
- if (args.do_chroot)
- Echroot(".");
-
- if (args.uid!=(uid_t)(-1) && getuid()!=args.uid) {
- Esetuid(args.uid);
- if (getuid()!=args.uid) {
- WRITE_MSG(2, "vcontext: Something went wrong while changing the UID\n");
- exit(255);
- }
- }
-
- if (args.is_fakeinit) {
- struct vc_ctx_flags flags;
- Evc_get_flags(xid, &flags);
- #warning !!! IMPLEMENT ME !!!
- //if (flags.mask
- if (0) {
- WRITE_MSG(2, "vcontext: context has already a fakeinit-process\n");
- exit(255);
- }
- }
-
- if (args.do_migrate)
- Evc_migrate_context(xid);
-
- doSyncStage1(p, args.do_disconnect);
- execvp (argv[optind],argv+optind);
- doSyncStage2(p, args.do_disconnect);
-
- PERROR_Q("chcontext: execvp", argv[optind]);
- exit(255);
- }
+ else if (optind>=argc)
+ WRITE_MSG(2, "No command given; use '--help' for more information.\n");
+ else
+ return doit(&args, argv);
- waitOnSync(pid, p);
- return EXIT_SUCCESS;
+ return 255;
}