rumpclient.c revision 1.10 1 1.10 pooka /* $NetBSD: rumpclient.c,v 1.10 2010/12/16 17:05:44 pooka Exp $ */
2 1.1 pooka
3 1.1 pooka /*
4 1.1 pooka * Copyright (c) 2010 Antti Kantee. All Rights Reserved.
5 1.1 pooka *
6 1.1 pooka * Redistribution and use in source and binary forms, with or without
7 1.1 pooka * modification, are permitted provided that the following conditions
8 1.1 pooka * are met:
9 1.1 pooka * 1. Redistributions of source code must retain the above copyright
10 1.1 pooka * notice, this list of conditions and the following disclaimer.
11 1.1 pooka * 2. Redistributions in binary form must reproduce the above copyright
12 1.1 pooka * notice, this list of conditions and the following disclaimer in the
13 1.1 pooka * documentation and/or other materials provided with the distribution.
14 1.1 pooka *
15 1.1 pooka * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS
16 1.1 pooka * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
17 1.1 pooka * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
18 1.1 pooka * DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
19 1.1 pooka * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20 1.1 pooka * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
21 1.1 pooka * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22 1.1 pooka * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23 1.1 pooka * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24 1.1 pooka * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
25 1.1 pooka * SUCH DAMAGE.
26 1.1 pooka */
27 1.1 pooka
28 1.1 pooka /*
29 1.1 pooka * Client side routines for rump syscall proxy.
30 1.1 pooka */
31 1.1 pooka
32 1.1 pooka #include <sys/cdefs.h>
33 1.1 pooka __RCSID("$NetBSD");
34 1.1 pooka
35 1.5 pooka #include <sys/param.h>
36 1.1 pooka #include <sys/mman.h>
37 1.1 pooka #include <sys/socket.h>
38 1.1 pooka
39 1.1 pooka #include <arpa/inet.h>
40 1.1 pooka #include <netinet/in.h>
41 1.1 pooka #include <netinet/tcp.h>
42 1.1 pooka
43 1.1 pooka #include <assert.h>
44 1.1 pooka #include <errno.h>
45 1.1 pooka #include <fcntl.h>
46 1.1 pooka #include <poll.h>
47 1.1 pooka #include <pthread.h>
48 1.1 pooka #include <stdarg.h>
49 1.1 pooka #include <stdio.h>
50 1.1 pooka #include <stdlib.h>
51 1.1 pooka #include <string.h>
52 1.1 pooka #include <unistd.h>
53 1.1 pooka
54 1.1 pooka #include <rump/rumpclient.h>
55 1.1 pooka
56 1.1 pooka #include "sp_common.c"
57 1.1 pooka
58 1.1 pooka static struct spclient clispc;
59 1.1 pooka
60 1.1 pooka static int
61 1.3 pooka syscall_req(struct spclient *spc, int sysnum,
62 1.3 pooka const void *data, size_t dlen, void **resp)
63 1.1 pooka {
64 1.1 pooka struct rsp_hdr rhdr;
65 1.3 pooka struct respwait rw;
66 1.3 pooka int rv;
67 1.1 pooka
68 1.1 pooka rhdr.rsp_len = sizeof(rhdr) + dlen;
69 1.3 pooka rhdr.rsp_class = RUMPSP_REQ;
70 1.3 pooka rhdr.rsp_type = RUMPSP_SYSCALL;
71 1.1 pooka rhdr.rsp_sysnum = sysnum;
72 1.1 pooka
73 1.6 pooka do {
74 1.6 pooka putwait(spc, &rw, &rhdr);
75 1.6 pooka rv = dosend(spc, &rhdr, sizeof(rhdr));
76 1.6 pooka rv = dosend(spc, data, dlen);
77 1.6 pooka if (rv) {
78 1.6 pooka unputwait(spc, &rw);
79 1.6 pooka return rv;
80 1.6 pooka }
81 1.6 pooka
82 1.6 pooka rv = waitresp(spc, &rw);
83 1.6 pooka } while (rv == EAGAIN);
84 1.3 pooka
85 1.3 pooka *resp = rw.rw_data;
86 1.3 pooka return rv;
87 1.1 pooka }
88 1.1 pooka
89 1.1 pooka static int
90 1.10 pooka handshake_req(struct spclient *spc)
91 1.10 pooka {
92 1.10 pooka struct rsp_hdr rhdr;
93 1.10 pooka struct respwait rw;
94 1.10 pooka int rv;
95 1.10 pooka
96 1.10 pooka /* performs server handshake */
97 1.10 pooka rhdr.rsp_len = sizeof(rhdr);
98 1.10 pooka rhdr.rsp_class = RUMPSP_REQ;
99 1.10 pooka rhdr.rsp_type = RUMPSP_HANDSHAKE;
100 1.10 pooka rhdr.rsp_handshake = HANDSHAKE_GUEST;
101 1.10 pooka
102 1.10 pooka putwait(spc, &rw, &rhdr);
103 1.10 pooka rv = dosend(spc, &rhdr, sizeof(rhdr));
104 1.10 pooka if (rv != 0) {
105 1.10 pooka unputwait(spc, &rw);
106 1.10 pooka return rv;
107 1.10 pooka }
108 1.10 pooka
109 1.10 pooka rv = waitresp(spc, &rw);
110 1.10 pooka if (rv)
111 1.10 pooka return rv;
112 1.10 pooka
113 1.10 pooka rv = *(int *)rw.rw_data;
114 1.10 pooka free(rw.rw_data);
115 1.10 pooka
116 1.10 pooka return rv;
117 1.10 pooka }
118 1.10 pooka
119 1.10 pooka static int
120 1.5 pooka send_copyin_resp(struct spclient *spc, uint64_t reqno, void *data, size_t dlen,
121 1.5 pooka int wantstr)
122 1.1 pooka {
123 1.1 pooka struct rsp_hdr rhdr;
124 1.3 pooka int rv;
125 1.1 pooka
126 1.5 pooka if (wantstr)
127 1.5 pooka dlen = MIN(dlen, strlen(data)+1);
128 1.5 pooka
129 1.1 pooka rhdr.rsp_len = sizeof(rhdr) + dlen;
130 1.1 pooka rhdr.rsp_reqno = reqno;
131 1.3 pooka rhdr.rsp_class = RUMPSP_RESP;
132 1.3 pooka rhdr.rsp_type = RUMPSP_COPYIN;
133 1.1 pooka rhdr.rsp_sysnum = 0;
134 1.1 pooka
135 1.3 pooka sendlock(spc);
136 1.3 pooka rv = dosend(spc, &rhdr, sizeof(rhdr));
137 1.3 pooka rv = dosend(spc, data, dlen);
138 1.3 pooka sendunlock(spc);
139 1.1 pooka
140 1.3 pooka return rv;
141 1.1 pooka }
142 1.1 pooka
143 1.1 pooka static int
144 1.1 pooka send_anonmmap_resp(struct spclient *spc, uint64_t reqno, void *addr)
145 1.1 pooka {
146 1.1 pooka struct rsp_hdr rhdr;
147 1.3 pooka int rv;
148 1.1 pooka
149 1.1 pooka rhdr.rsp_len = sizeof(rhdr) + sizeof(addr);
150 1.1 pooka rhdr.rsp_reqno = reqno;
151 1.3 pooka rhdr.rsp_class = RUMPSP_RESP;
152 1.3 pooka rhdr.rsp_type = RUMPSP_ANONMMAP;
153 1.1 pooka rhdr.rsp_sysnum = 0;
154 1.1 pooka
155 1.3 pooka sendlock(spc);
156 1.3 pooka rv = dosend(spc, &rhdr, sizeof(rhdr));
157 1.3 pooka rv = dosend(spc, &addr, sizeof(addr));
158 1.3 pooka sendunlock(spc);
159 1.1 pooka
160 1.3 pooka return rv;
161 1.1 pooka }
162 1.1 pooka
163 1.1 pooka int
164 1.1 pooka rumpclient_syscall(int sysnum, const void *data, size_t dlen,
165 1.1 pooka register_t *retval)
166 1.1 pooka {
167 1.1 pooka struct rsp_sysresp *resp;
168 1.3 pooka void *rdata;
169 1.3 pooka int rv;
170 1.3 pooka
171 1.3 pooka DPRINTF(("rumpsp syscall_req: syscall %d with %p/%zu\n",
172 1.3 pooka sysnum, data, dlen));
173 1.3 pooka
174 1.3 pooka rv = syscall_req(&clispc, sysnum, data, dlen, &rdata);
175 1.3 pooka if (rv)
176 1.3 pooka return rv;
177 1.3 pooka
178 1.3 pooka resp = rdata;
179 1.3 pooka DPRINTF(("rumpsp syscall_resp: syscall %d error %d, rv: %d/%d\n",
180 1.3 pooka sysnum, rv, resp->rsys_retval[0], resp->rsys_retval[1]));
181 1.1 pooka
182 1.3 pooka memcpy(retval, &resp->rsys_retval, sizeof(resp->rsys_retval));
183 1.3 pooka rv = resp->rsys_error;
184 1.3 pooka free(rdata);
185 1.1 pooka
186 1.3 pooka return rv;
187 1.3 pooka }
188 1.1 pooka
189 1.3 pooka static void
190 1.3 pooka handlereq(struct spclient *spc)
191 1.3 pooka {
192 1.3 pooka struct rsp_copydata *copydata;
193 1.3 pooka void *mapaddr;
194 1.3 pooka size_t maplen;
195 1.5 pooka int reqtype = spc->spc_hdr.rsp_type;
196 1.1 pooka
197 1.5 pooka switch (reqtype) {
198 1.3 pooka case RUMPSP_COPYIN:
199 1.5 pooka case RUMPSP_COPYINSTR:
200 1.3 pooka /*LINTED*/
201 1.3 pooka copydata = (struct rsp_copydata *)spc->spc_buf;
202 1.3 pooka DPRINTF(("rump_sp handlereq: copyin request: %p/%zu\n",
203 1.3 pooka copydata->rcp_addr, copydata->rcp_len));
204 1.3 pooka send_copyin_resp(spc, spc->spc_hdr.rsp_reqno,
205 1.5 pooka copydata->rcp_addr, copydata->rcp_len,
206 1.5 pooka reqtype == RUMPSP_COPYINSTR);
207 1.3 pooka break;
208 1.3 pooka case RUMPSP_COPYOUT:
209 1.5 pooka case RUMPSP_COPYOUTSTR:
210 1.3 pooka /*LINTED*/
211 1.3 pooka copydata = (struct rsp_copydata *)spc->spc_buf;
212 1.3 pooka DPRINTF(("rump_sp handlereq: copyout request: %p/%zu\n",
213 1.3 pooka copydata->rcp_addr, copydata->rcp_len));
214 1.3 pooka /*LINTED*/
215 1.3 pooka memcpy(copydata->rcp_addr, copydata->rcp_data,
216 1.3 pooka copydata->rcp_len);
217 1.3 pooka break;
218 1.3 pooka case RUMPSP_ANONMMAP:
219 1.3 pooka /*LINTED*/
220 1.3 pooka maplen = *(size_t *)spc->spc_buf;
221 1.3 pooka mapaddr = mmap(NULL, maplen, PROT_READ|PROT_WRITE,
222 1.3 pooka MAP_ANON, -1, 0);
223 1.3 pooka if (mapaddr == MAP_FAILED)
224 1.3 pooka mapaddr = NULL;
225 1.3 pooka DPRINTF(("rump_sp handlereq: anonmmap: %p\n", mapaddr));
226 1.3 pooka send_anonmmap_resp(spc, spc->spc_hdr.rsp_reqno, mapaddr);
227 1.3 pooka break;
228 1.3 pooka default:
229 1.3 pooka printf("PANIC: INVALID TYPE\n");
230 1.3 pooka abort();
231 1.3 pooka break;
232 1.1 pooka }
233 1.1 pooka
234 1.6 pooka spcfreebuf(spc);
235 1.1 pooka }
236 1.1 pooka
237 1.1 pooka int
238 1.1 pooka rumpclient_init()
239 1.1 pooka {
240 1.9 pooka char banner[MAXBANNER];
241 1.1 pooka struct sockaddr *sap;
242 1.1 pooka char *p;
243 1.1 pooka unsigned idx;
244 1.9 pooka ssize_t n;
245 1.1 pooka int error, s;
246 1.1 pooka
247 1.7 pooka if ((p = getenv("RUMP_SERVER")) == NULL) {
248 1.2 pooka errno = ENOENT;
249 1.2 pooka return -1;
250 1.2 pooka }
251 1.1 pooka
252 1.2 pooka if ((error = parseurl(p, &sap, &idx, 0)) != 0) {
253 1.2 pooka errno = error;
254 1.2 pooka return -1;
255 1.2 pooka }
256 1.1 pooka
257 1.1 pooka s = socket(parsetab[idx].domain, SOCK_STREAM, 0);
258 1.1 pooka if (s == -1)
259 1.2 pooka return -1;
260 1.1 pooka
261 1.8 pooka if (connect(s, sap, (socklen_t)sap->sa_len) == -1) {
262 1.2 pooka error = errno;
263 1.1 pooka fprintf(stderr, "rump_sp: client connect failed\n");
264 1.2 pooka errno = error;
265 1.2 pooka return -1;
266 1.1 pooka }
267 1.4 pooka
268 1.10 pooka if ((n = read(s, banner, sizeof(banner)-1)) < 0) {
269 1.2 pooka error = errno;
270 1.10 pooka fprintf(stderr, "rump_sp: failed to read banner\n");
271 1.2 pooka errno = error;
272 1.2 pooka return -1;
273 1.1 pooka }
274 1.9 pooka
275 1.9 pooka if (banner[n-1] != '\n') {
276 1.9 pooka fprintf(stderr, "rump_sp: invalid banner\n");
277 1.9 pooka errno = EINVAL;
278 1.9 pooka return -1;
279 1.9 pooka }
280 1.9 pooka banner[n] = '\0';
281 1.9 pooka
282 1.9 pooka /* parse the banner some day */
283 1.9 pooka
284 1.10 pooka if ((error = parsetab[idx].connhook(s)) != 0) {
285 1.10 pooka error = errno;
286 1.10 pooka fprintf(stderr, "rump_sp: connect hook failed\n");
287 1.10 pooka errno = error;
288 1.10 pooka return -1;
289 1.10 pooka }
290 1.10 pooka
291 1.10 pooka pthread_mutex_init(&clispc.spc_mtx, NULL);
292 1.10 pooka pthread_cond_init(&clispc.spc_cv, NULL);
293 1.10 pooka clispc.spc_fd = s;
294 1.10 pooka TAILQ_INIT(&clispc.spc_respwait);
295 1.10 pooka
296 1.10 pooka error = handshake_req(&clispc);
297 1.10 pooka if (error) {
298 1.10 pooka pthread_mutex_destroy(&clispc.spc_mtx);
299 1.10 pooka pthread_cond_destroy(&clispc.spc_cv);
300 1.10 pooka close(s);
301 1.10 pooka }
302 1.10 pooka return error;
303 1.1 pooka }
304