/* * qemud.c: daemon start of day, guest process & i/o management * * Copyright (C) 2006, 2007 Red Hat, Inc. * Copyright (C) 2006 Daniel P. Berrange * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * This library 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 * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA * * Author: Daniel P. Berrange */ #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include "internal.h" #include "dispatch.h" #include "driver.h" #include "conf.h" static void reapchild(int sig ATTRIBUTE_UNUSED) { /* We explicitly waitpid the child later */ } static int qemudSetCloseExec(int fd) { int flags; if ((flags = fcntl(fd, F_GETFD)) < 0) { return -1; } flags |= FD_CLOEXEC; if ((fcntl(fd, F_SETFD, flags)) < 0) { return -1; } return 0; } static int qemudSetNonBlock(int fd) { int flags; if ((flags = fcntl(fd, F_GETFL)) < 0) { return -1; } flags |= O_NONBLOCK; if ((fcntl(fd, F_SETFL, flags)) < 0) { return -1; } return 0; } static int qemudGoDaemon(void) { int pid = fork(); switch (pid) { case 0: { int stdinfd = -1; int stdoutfd = -1; int i, open_max, nextpid; if ((stdinfd = open(_PATH_DEVNULL, O_RDONLY)) < 0) goto cleanup; if ((stdoutfd = open(_PATH_DEVNULL, O_WRONLY)) < 0) goto cleanup; if (dup2(stdinfd, STDIN_FILENO) != STDIN_FILENO) goto cleanup; if (dup2(stdoutfd, STDOUT_FILENO) != STDOUT_FILENO) goto cleanup; if (dup2(stdoutfd, STDERR_FILENO) != STDERR_FILENO) goto cleanup; if (close(stdinfd) < 0) goto cleanup; stdinfd = -1; if (close(stdoutfd) < 0) goto cleanup; stdoutfd = -1; open_max = sysconf (_SC_OPEN_MAX); for (i = 0; i < open_max; i++) if (i != STDIN_FILENO && i != STDOUT_FILENO && i != STDERR_FILENO) close(i); if (setsid() < 0) goto cleanup; nextpid = fork(); switch (nextpid) { case 0: return 0; case -1: return -1; default: return nextpid; } cleanup: if (stdoutfd != -1) close(stdoutfd); if (stdinfd != -1) close(stdinfd); return -1; } case -1: return -1; default: { int got, status = 0; /* We wait to make sure the next child forked successfully */ if ((got = waitpid(pid, &status, 0)) < 0 || got != pid || status != 0) { return -1; } return pid; } } } static int qemudListenUnix(struct qemud_server *server, const char *path, int readonly) { struct qemud_socket *sock = calloc(1, sizeof(struct qemud_socket)); struct sockaddr_un addr; mode_t oldmask; if (!sock) return -1; sock->readonly = readonly; sock->next = server->sockets; server->sockets = sock; server->nsockets++; if ((sock->fd = socket(PF_UNIX, SOCK_STREAM, 0)) < 0) return -1; if (qemudSetCloseExec(sock->fd) < 0) return -1; if (qemudSetNonBlock(sock->fd) < 0) return -1; memset(&addr, 0, sizeof(addr)); addr.sun_family = AF_UNIX; strncpy(addr.sun_path, path, sizeof(addr.sun_path)-1); if (addr.sun_path[0] == '@') addr.sun_path[0] = '\0'; if (readonly) oldmask = umask(~(S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH)); else oldmask = umask(~(S_IRUSR | S_IWUSR)); if (bind(sock->fd, (struct sockaddr *)&addr, sizeof(addr)) < 0) return -1; umask(oldmask); if (listen(sock->fd, 30) < 0) return -1; return 0; } static int qemudListen(struct qemud_server *server, int sys) { char sockname[PATH_MAX]; if (sys) { if (snprintf(sockname, sizeof(sockname), "%s/run/qemud/sock", LOCAL_STATE_DIR) >= (int)sizeof(sockname)) { return -1; } unlink(sockname); if (qemudListenUnix(server, sockname, 0) < 0) return -1; if (snprintf(sockname, sizeof(sockname), "%s/run/qemud/sock-ro", LOCAL_STATE_DIR) >= (int)sizeof(sockname)) { return -1; } unlink(sockname); if (qemudListenUnix(server, sockname, 1) < 0) return -1; } else { struct passwd *pw; int uid; if ((uid = geteuid()) < 0) { return -1; } if (!(pw = getpwuid(uid))) return -1; if (snprintf(sockname, sizeof(sockname), "@%s/.qemud/sock", pw->pw_dir) >= (int)sizeof(sockname)) { return -1; } if (qemudListenUnix(server, sockname, 0) < 0) return -1; } return 0; } static struct qemud_server *qemudInitialize(int sys) { struct qemud_server *server; if (!(server = calloc(1, sizeof(struct qemud_server)))) return NULL; /* XXX extract actual version */ server->qemuVersion = (0*1000000)+(8*1000)+(0); /* We don't have a dom-0, so start from 1 */ server->nextvmid = 1; if (sys) { if (snprintf(server->configDir, sizeof(server->configDir), "%s/qemud", SYSCONF_DIR) >= (int)sizeof(server->configDir)) { goto cleanup; } } else { struct passwd *pw; int uid; if ((uid = geteuid()) < 0) { goto cleanup; } if (!(pw = getpwuid(uid))) { goto cleanup; } if (snprintf(server->configDir, sizeof(server->configDir), "%s/.qemud", pw->pw_dir) >= (int)sizeof(server->configDir)) { goto cleanup; } } if (qemudListen(server, sys) < 0) { goto cleanup; } if (qemudScanConfigs(server) < 0) { goto cleanup; } return server; cleanup: if (server) { struct qemud_socket *sock = server->sockets; while (sock) { close(sock->fd); sock = sock->next; } free(server); } return NULL; } static int qemudDispatchServer(struct qemud_server *server, struct qemud_socket *sock) { int fd; struct sockaddr_storage addr; unsigned int addrlen = sizeof(addr); struct qemud_client *client; if ((fd = accept(sock->fd, (struct sockaddr *)&addr, &addrlen)) < 0) { if (errno == EAGAIN) return 0; return -1; } if (qemudSetCloseExec(fd) < 0) { close(fd); return -1; } if (qemudSetNonBlock(fd) < 0) { close(fd); return -1; } client = calloc(1, sizeof(struct qemud_client)); client->fd = fd; client->readonly = sock->readonly; client->next = server->clients; server->clients = client; server->nclients++; return 0; } static int qemudExec(struct qemud_server *server, char **argv, int *retpid, int *outfd, int *errfd) { int pid, null; int pipeout[2] = {-1,-1}; int pipeerr[2] = {-1,-1}; if ((null = open(_PATH_DEVNULL, O_RDONLY)) < 0) { qemudReportError(server, VIR_ERR_INTERNAL_ERROR, "cannot open %s : %s", _PATH_DEVNULL, strerror(errno)); goto cleanup; } if ((outfd != NULL && pipe(pipeout) < 0) || (errfd != NULL && pipe(pipeerr) < 0)) { qemudReportError(server, VIR_ERR_INTERNAL_ERROR, "cannot create pipe : %s", strerror(errno)); goto cleanup; } if ((pid = fork()) < 0) { qemudReportError(server, VIR_ERR_INTERNAL_ERROR, "cannot fork child process : %s", strerror(errno)); goto cleanup; } if (pid) { /* parent */ close(null); if (outfd) { close(pipeout[1]); qemudSetNonBlock(pipeout[0]); *outfd = pipeout[0]; } if (errfd) { close(pipeerr[1]); qemudSetNonBlock(pipeerr[0]); *errfd = pipeerr[0]; } *retpid = pid; return 0; } /* child */ if (pipeout[0] > 0 && close(pipeout[0]) < 0) _exit(1); if (pipeerr[0] > 0 && close(pipeerr[0]) < 0) _exit(1); if (dup2(null, STDIN_FILENO) < 0) _exit(1); if (dup2(pipeout[1] > 0 ? pipeout[1] : null, STDOUT_FILENO) < 0) _exit(1); if (dup2(pipeerr[1] > 0 ? pipeerr[1] : null, STDERR_FILENO) < 0) _exit(1); int i, open_max = sysconf (_SC_OPEN_MAX); for (i = 0; i < open_max; i++) if (i != STDOUT_FILENO && i != STDERR_FILENO && i != STDIN_FILENO) close(i); execvp(argv[0], argv); _exit(1); return 0; cleanup: if (pipeerr[0] > 0) close(pipeerr[0] > 0); if (pipeerr[1]) close(pipeerr[1] > 0); if (pipeout[0]) close(pipeout[0] > 0); if (pipeout[1]) close(pipeout[1] > 0); if (null > 0) close(null); return -1; } int qemudStartVMDaemon(struct qemud_server *server, struct qemud_vm *vm) { char **argv = NULL; int i, ret = -1; if (vm->def.vncPort < 0) vm->def.vncActivePort = 5900 + server->nextvmid; else vm->def.vncActivePort = vm->def.vncPort; if (qemudBuildCommandLine(server, vm, &argv) < 0) return -1; if (qemudExec(server, argv, &vm->pid, &vm->stdout, &vm->stderr) == 0) { vm->def.id = server->nextvmid++; ret = 0; } for (i = 0 ; argv[i] ; i++) free(argv[i]); free(argv); return ret; } static void qemudDispatchClientFailure(struct qemud_server *server, struct qemud_client *client) { struct qemud_client *tmp = server->clients; struct qemud_client *prev = NULL; while (tmp) { if (tmp == client) { if (prev == NULL) server->clients = client->next; else prev->next = client->next; server->nclients--; break; } prev = tmp; tmp = tmp->next; } close(client->fd); free(client); } static int qemudDispatchClientRequest(struct qemud_server *server, struct qemud_client *client) { if (qemudDispatch(server, client, &client->incoming, &client->outgoing) < 0) { return -1; } client->outgoingSent = 0; client->tx = 1; client->incomingReceived = 0; return 0; } static int qemudClientRead(struct qemud_server *server, struct qemud_client *client, char *buf, size_t want) { int ret; if ((ret = read(client->fd, buf, want)) <= 0) { QEMUD_DEBUG("Plain read error %d\n", ret); if (!ret || errno != EAGAIN) qemudDispatchClientFailure(server, client); return -1; } QEMUD_DEBUG("Plain data read %d\n", ret); return ret; } static void qemudDispatchClientRead(struct qemud_server *server, struct qemud_client *client) { char *data = (char *)&client->incoming; unsigned int got = client->incomingReceived; int want; int ret; restart: if (got >= sizeof(struct qemud_packet_header)) { want = sizeof(struct qemud_packet_header) + client->incoming.header.dataSize - got; } else { want = sizeof(struct qemud_packet_header) - got; } if ((ret = qemudClientRead(server, client, data+got, want)) < 0) { return; } got += ret; client->incomingReceived += ret; /* If we've finished header, move onto body */ if (client->incomingReceived == sizeof(struct qemud_packet_header)) { QEMUD_DEBUG("Type %d, data %d\n", client->incoming.header.type, client->incoming.header.dataSize); /* Client lied about dataSize */ if (client->incoming.header.dataSize > sizeof(union qemud_packet_data)) { QEMUD_DEBUG("Bogus data size %u\n", client->incoming.header.dataSize); qemudDispatchClientFailure(server, client); return; } if (client->incoming.header.dataSize) { QEMUD_DEBUG("- Restarting recv to process body (%d bytes)\n", client->incoming.header.dataSize); goto restart; } } /* If we've finished body, dispatch the request */ if (ret == want) { if (qemudDispatchClientRequest(server, client) < 0) qemudDispatchClientFailure(server, client); QEMUD_DEBUG("Dispatch\n"); } } static int qemudClientWrite(struct qemud_server *server, struct qemud_client *client, char *buf, size_t want) { int ret; if ((ret = write(client->fd, buf, want)) < 0) { QEMUD_DEBUG("Plain write error %d\n", ret); if (errno != EAGAIN) qemudDispatchClientFailure(server, client); return -1; } QEMUD_DEBUG("Plain data write %d\n", ret); return ret; } static void qemudDispatchClientWrite(struct qemud_server *server, struct qemud_client *client) { char *data = (char *)&client->outgoing; int sent = client->outgoingSent; int todo = sizeof(struct qemud_packet_header) + client->outgoing.header.dataSize - sent; int ret; if ((ret = qemudClientWrite(server, client, data+sent, todo)) < 0) { return; } client->outgoingSent += ret; QEMUD_DEBUG("Done %d %d\n", todo, ret); if (todo == ret) client->tx = 0; } static int qemudVMData(struct qemud_server *server ATTRIBUTE_UNUSED, struct qemud_vm *vm, int fd) { char buf[4096]; if (vm->pid < 0) return 0; for (;;) { int ret = read(fd, buf, sizeof(buf)-1); if (ret < 0) { if (errno == EAGAIN) return 0; return -1; } if (ret == 0) { return 0; } buf[ret] = '\0'; /* * XXX this is bad - we should wait for tty and open the * monitor when actually starting the guest, so we can * reliably trap startup failures */ if (vm->monitor == -1) { char monitor[20]; /* Fairly lame assuming we receive the data all in one chunk. This isn't guarenteed, but in practice it seems good enough. This will probably bite me in the future.... */ if (sscanf(buf, "char device redirected to %19s", monitor) == 1) { int monfd; if (!(monfd = open(monitor, O_RDWR))) { perror("cannot open monitor"); return -1; } if (qemudSetCloseExec(monfd) < 0) { close(monfd); return -1; } if (qemudSetNonBlock(monfd) < 0) { close(monfd); return -1; } /* Consume & discard the initial greeting */ /* XXX this is broken, we need to block until we see the initial prompt to ensure startup has completed */ for(;;) { char line[1024]; if (read(monfd, line, sizeof(line)) < 0) { if (errno == EAGAIN) { break; } close(monfd); return -1; } QEMUD_DEBUG("[%s]\n", line); } vm->monitor = monfd; } } QEMUD_DEBUG("[%s]\n", buf); } } int qemudShutdownVMDaemon(struct qemud_server *server, struct qemud_vm *vm) { struct qemud_vm *prev = NULL, *curr = server->activevms; /* Already cleaned-up */ if (vm->pid < 0) return 0; kill(vm->pid, SIGTERM); /* Move it to inactive vm list */ while (curr) { if (curr == vm) { if (prev) { prev->next = curr->next; } else { server->activevms = curr->next; } server->nactivevms--; curr->next = server->inactivevms; server->inactivevms = curr; server->ninactivevms++; break; } prev = curr; curr = curr->next; } if (!curr) { QEMUD_DEBUG("Could not find VM to shutdown\n"); return 0; } qemudVMData(server, vm, curr->stdout); qemudVMData(server, vm, curr->stderr); close(curr->stdout); close(curr->stderr); if (curr->monitor != -1) close(curr->monitor); curr->stdout = -1; curr->stderr = -1; curr->monitor = -1; server->nvmfds -= 2; if (waitpid(vm->pid, NULL, WNOHANG) != vm->pid) { kill(vm->pid, SIGKILL); if (waitpid(vm->pid, NULL, 0) != vm->pid) { QEMUD_DEBUG("Got unexpected pid, damn\n"); } } vm->pid = -1; vm->def.id = -1; return 0; } static int qemudDispatchVMLog(struct qemud_server *server, struct qemud_vm *vm, int fd) { if (qemudVMData(server, vm, fd) < 0) if (qemudShutdownVMDaemon(server, vm) < 0) return -1; return 0; } static int qemudDispatchVMFailure(struct qemud_server *server, struct qemud_vm *vm, int fd ATTRIBUTE_UNUSED) { if (qemudShutdownVMDaemon(server, vm) < 0) return -1; return 0; } static int qemudDispatchPoll(struct qemud_server *server, struct pollfd *fds) { struct qemud_socket *sock = server->sockets; struct qemud_client *client = server->clients; struct qemud_vm *vm = server->activevms; struct qemud_vm *tmp; int ret = 0; int fd = 0; while (sock) { struct qemud_socket *next = sock->next; if (fds[fd].revents) if (qemudDispatchServer(server, sock) < 0) return -1; fd++; sock = next; } while (client) { struct qemud_client *next = client->next; if (fds[fd].revents) { QEMUD_DEBUG("Poll data normal\n"); if (fds[fd].revents == POLLOUT) qemudDispatchClientWrite(server, client); else if (fds[fd].revents == POLLIN) qemudDispatchClientRead(server, client); else qemudDispatchClientFailure(server, client); } fd++; client = next; } while (vm) { struct qemud_vm *next = vm->next; int failed = 0, stdoutfd = vm->stdout, stderrfd = vm->stderr; if (stdoutfd != -1) { if (fds[fd].revents) { if (fds[fd].revents == POLLIN) { if (qemudDispatchVMLog(server, vm, fds[fd].fd) < 0) failed = 1; } else { if (qemudDispatchVMFailure(server, vm, fds[fd].fd) < 0) failed = 1; } } fd++; } if (stderrfd != -1) { if (!failed) { if (fds[fd].revents) { if (fds[fd].revents == POLLIN) { if (qemudDispatchVMLog(server, vm, fds[fd].fd) < 0) failed = 1; } else { if (qemudDispatchVMFailure(server, vm, fds[fd].fd) < 0) failed = 1; } } } fd++; } vm = next; if (failed) ret = -1; } /* Cleanup any VMs which shutdown & dont have an associated config file */ vm = server->inactivevms; tmp = NULL; while (vm) { if (!vm->configFile[0]) { struct qemud_vm *next = vm->next; if (tmp) { tmp->next = next; } else { server->inactivevms = next; } qemudFreeVM(vm); vm = next; } else { tmp = vm; vm = vm->next; } } return ret; } static void qemudPreparePoll(struct qemud_server *server, struct pollfd *fds) { int fd = 0; struct qemud_socket *sock; struct qemud_client *client; struct qemud_vm *vm; for (sock = server->sockets ; sock ; sock = sock->next) { fds[fd].fd = sock->fd; fds[fd].events = POLLIN; fd++; } for (client = server->clients ; client ; client = client->next) { fds[fd].fd = client->fd; /* Refuse to read more from client if tx is pending to rate limit */ if (client->tx) fds[fd].events = POLLOUT | POLLERR | POLLHUP; else fds[fd].events = POLLIN | POLLERR | POLLHUP; fd++; } for (vm = server->activevms ; vm ; vm = vm->next) { if (vm->stdout != -1) { fds[fd].fd = vm->stdout; fds[fd].events = POLLIN | POLLERR | POLLHUP; fd++; } if (vm->stderr != -1) { fds[fd].fd = vm->stderr; fds[fd].events = POLLIN | POLLERR | POLLHUP; fd++; } } } static int qemudOneLoop(struct qemud_server *server, int timeout) { int nfds = server->nsockets + server->nclients + server->nvmfds; struct pollfd fds[nfds]; int thistimeout = -1; int ret; /* If we have no clients or vms, then timeout after 30 seconds, letting daemon exit */ if (timeout > 0 && !server->nclients && !server->nactivevms) thistimeout = timeout; qemudPreparePoll(server, fds); retry: if ((ret = poll(fds, nfds, thistimeout * 1000)) < 0) { if (errno == EINTR) { goto retry; } return -1; } /* Must have timed out */ if (ret == 0) return -1; if (qemudDispatchPoll(server, fds) < 0) return -1; return 0; } static int qemudRunLoop(struct qemud_server *server, int timeout) { int ret; while ((ret = qemudOneLoop(server, timeout)) == 0) ; return ret == -1 ? -1 : 0; } static void qemudCleanup(struct qemud_server *server) { struct qemud_socket *sock = server->sockets; while (sock) { close(sock->fd); sock = sock->next; } free(server); } #define MAX_LISTEN 5 int main(int argc, char **argv) { int godaemon = 0; int verbose = 0; int sys = 0; int timeout = -1; struct qemud_server *server; struct option opts[] = { { "verbose", no_argument, &verbose, 1}, { "daemon", no_argument, &godaemon, 1}, { "system", no_argument, &sys, 1}, { "timeout", required_argument, 0, 't'}, {0, 0, 0, 0} }; while (1) { int optidx = 0; int c; char *tmp; c = getopt_long(argc, argv, "vsdt:", opts, &optidx); if (c == -1) { break; } switch (c) { case 0: /* Got one of the flags */ break; case 'v': verbose = 1; break; case 'd': godaemon = 1; break; case 's': sys = 1; break; case 't': timeout = strtol(optarg, &tmp, 10); if (!tmp) timeout = -1; if (timeout <= 0) timeout = -1; break; case '?': return 2; break; default: abort(); } } if (signal(SIGPIPE, SIG_IGN) == SIG_ERR) return 3; if (signal(SIGCHLD, reapchild) == SIG_ERR) return 3; if (godaemon) { int pid = qemudGoDaemon(); if (pid < 0) return 1; if (pid > 0) return 0; } if (!(server = qemudInitialize(sys))) return 2; qemudRunLoop(server, timeout); qemudCleanup(server); return 0; } /* * Local variables: * indent-tabs-mode: nil * c-indent-level: 4 * c-basic-offset: 4 * tab-width: 4 * End: */