/* * The Qubes OS Project, http://www.qubes-os.org * * Copyright (C) 2010 Rafal Wojtczuk * * 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. * */ #include #include #include #include #include #include #include #include #include #include #include #include #include #include "libvchan.h" #ifndef CONFIG_STUBDOM #include "../u2mfn/u2mfnlib.h" #else #include #endif static int ring_init(struct libvchan *ctrl) { int mfn; struct vchan_interface *ring; #ifdef CONFIG_STUBDOM ring = (struct vchan_interface *) memalign(XC_PAGE_SIZE, sizeof(*ring)); if (!ring) return -1; mfn = virtual_to_mfn(ring); #else int u2mfn = open("/proc/u2mfn", O_RDONLY); ring = (struct vchan_interface *) u2mfn_alloc_kpage (); if (ring == MAP_FAILED) return -1; if (u2mfn_get_last_mfn (&mfn) < 0) return -1; close(u2mfn); #endif ctrl->ring = ring; ctrl->ring_ref = mfn; ring->cons_in = ring->prod_in = ring->cons_out = ring->prod_out = 0; ring->server_closed = ring->client_closed = 0; ring->debug = 0xaabbccdd; return 0; } /** creates event channel; creates "ring-ref" and "event-channel" xenstore entries; waits for connection to event channel from the peer */ static int server_interface_init(struct libvchan *ctrl, int devno) { int ret = -1; struct xs_handle *xs; char buf[64]; char ref[16]; /* XXX temp hack begin */ char *domid_s; int domid = 0; unsigned int len; /* XXX temp hack end */ #ifdef XENCTRL_HAS_XC_INTERFACE xc_evtchn *evfd; #else int evfd; #endif evtchn_port_or_error_t port; xs = xs_daemon_open(); if (!xs) { return ret; } #ifdef XENCTRL_HAS_XC_INTERFACE evfd = xc_evtchn_open(NULL, 0); if (!evfd) goto fail; #else evfd = xc_evtchn_open(); if (evfd < 0) goto fail; #endif ctrl->evfd = evfd; // the following hardcoded 0 is the peer domain id port = xc_evtchn_bind_unbound_port(evfd, 0); if (port < 0) goto fail2; ctrl->evport = port; ctrl->devno = devno; // stubdom debug HACK XXX domid_s = xs_read(xs, 0, "domid", &len); if (domid_s) domid = atoi(domid_s); snprintf(ref, sizeof ref, "%d", ctrl->ring_ref); snprintf(buf, sizeof buf, "device/vchan/%d/ring-ref", devno); if (!xs_write(xs, 0, buf, ref, strlen(ref))) #ifdef CONFIG_STUBDOM // TEMP HACK XXX FIXME goto fail2; fprintf(stderr, "xenstore-write /local/domain/%d/%s %s\n", domid, buf, ref); #else goto fail2; #endif snprintf(ref, sizeof ref, "%d", ctrl->evport); snprintf(buf, sizeof buf, "device/vchan/%d/event-channel", devno); if (!xs_write(xs, 0, buf, ref, strlen(ref))) #ifdef CONFIG_STUBDOM // TEMP HACK XXX FIXME goto fail2; fprintf(stderr, "xenstore-write /local/domain/%d/%s %s\n", domid, buf, ref); #else goto fail2; #endif // do not block in stubdom - libvchan_server_handle_connected will be // called on first input #ifndef CONFIG_STUBDOM // wait for the peer to arrive if (xc_evtchn_pending(evfd) == -1) goto fail2; xc_evtchn_unmask(ctrl->evfd, ctrl->evport); snprintf(buf, sizeof buf, "device/vchan/%d", devno); xs_rm(xs, 0, buf); #endif ret = 0; fail2: if (ret) #ifdef XENCTRL_HAS_XC_INTERFACE xc_evtchn_close(evfd); #else close(evfd); #endif fail: xs_daemon_close(xs); return ret; } #define dir_select(dir1, dir2) \ ctrl->wr_cons = &ctrl->ring->cons_##dir1; \ ctrl->wr_prod = &ctrl->ring->prod_##dir1; \ ctrl->rd_cons = &ctrl->ring->cons_##dir2; \ ctrl->rd_prod = &ctrl->ring->prod_##dir2; \ ctrl->wr_ring = ctrl->ring->buf_##dir1; \ ctrl->rd_ring = ctrl->ring->buf_##dir2; \ ctrl->wr_ring_size = sizeof(ctrl->ring->buf_##dir1); \ ctrl->rd_ring_size = sizeof(ctrl->ring->buf_##dir2) /** Run in AppVM (any domain). Sleeps until the connection is established. (unless in stubdom) \param devno something like a well-known port. \returns NULL on failure, handle on success */ struct libvchan *libvchan_server_init(int devno) { struct libvchan *ctrl = (struct libvchan *) malloc(sizeof(struct libvchan)); if (!ctrl) return 0; if (ring_init(ctrl)) return 0;; if (server_interface_init(ctrl, devno)) return 0; /* We want the same code for read/write functions, regardless whether we are client, or server. Thus, we do not access buf_in nor buf_out buffers directly. Instead, in *_init functions, the dir_select macro assigns proper values to wr* and rd* pointers, so that they point to correct one out of buf_in or buf_out related fields. */ dir_select(in, out); ctrl->is_server = 1; return ctrl; } int libvchan_server_handle_connected(struct libvchan *ctrl) { struct xs_handle *xs; char buf[64]; int ret = -1; int libvchan_fd; fd_set rfds; xs = xs_daemon_open(); if (!xs) { return ret; } // clear the pending flag xc_evtchn_pending(ctrl->evfd); snprintf(buf, sizeof buf, "device/vchan/%d", ctrl->devno); xs_rm(xs, 0, buf); ret = 0; #if 0 fail2: if (ret) #ifdef XENCTRL_HAS_XC_INTERFACE xc_evtchn_close(ctrl->evfd); #else close(ctrl->evfd); #endif #endif xs_daemon_close(xs); return ret; } /** retrieves ring-ref and event-channel numbers from xenstore (if they don't exist, return error, because nobody seems to listen); map the ring, connect the event channel */ static int client_interface_init(struct libvchan *ctrl, int domain, int devno) { int ret = -1; unsigned int len; struct xs_handle *xs; #ifdef XENCTRL_HAS_XC_INTERFACE xc_interface *xcfd; #else int xcfd; #endif char buf[64]; char *ref; #ifdef XENCTRL_HAS_XC_INTERFACE xc_evtchn *evfd; #else int evfd; #endif int remote_port; xs = xs_daemon_open(); if (!xs) { return ret; } snprintf(buf, sizeof buf, "/local/domain/%d/device/vchan/%d/ring-ref", domain, devno); ref = xs_read(xs, 0, buf, &len); if (!ref) goto fail; ctrl->ring_ref = atoi(ref); if (!ctrl->ring_ref) goto fail; free(ref); snprintf(buf, sizeof buf, "/local/domain/%d/device/vchan/%d/event-channel", domain, devno); ref = xs_read(xs, 0, buf, &len); if (!ref) goto fail; remote_port = atoi(ref); if (!remote_port) goto fail; free(ref); #ifdef XENCTRL_HAS_XC_INTERFACE xcfd = xc_interface_open(NULL, NULL, 0); if (!xcfd) goto fail; #else xcfd = xc_interface_open(); if (xcfd < 0) goto fail; #endif ctrl->ring = (struct vchan_interface *) xc_map_foreign_range(xcfd, domain, 4096, PROT_READ | PROT_WRITE, ctrl->ring_ref); #ifdef XENCTRL_HAS_XC_INTERFACE xc_interface_close(xcfd); #else close(xcfd); #endif if (ctrl->ring == 0 || ctrl->ring == MAP_FAILED) goto fail; #ifdef XENCTRL_HAS_XC_INTERFACE evfd = xc_evtchn_open(NULL, 0); if (!evfd) goto fail; #else evfd = xc_evtchn_open(); if (evfd < 0) goto fail; #endif ctrl->evfd = evfd; ctrl->evport = xc_evtchn_bind_interdomain(evfd, domain, remote_port); if (ctrl->evport < 0 || xc_evtchn_notify(evfd, ctrl->evport)) #ifdef XENCTRL_HAS_XC_INTERFACE xc_evtchn_close(evfd); #else close(evfd); #endif else ret = 0; fail: xs_daemon_close(xs); return ret; } /** Run on the client side of connection (currently, must be dom0). \returns NULL on failure (e.g. noone listening), handle on success */ struct libvchan *libvchan_client_init(int domain, int devno) { struct libvchan *ctrl = (struct libvchan *) malloc(sizeof(struct libvchan)); if (!ctrl) return 0; if (client_interface_init(ctrl, domain, devno)) return 0; // See comment in libvchan_server_init dir_select(out, in); ctrl->is_server = 0; return ctrl; }