1ca987d46SWarner Losh /*-
2ca987d46SWarner Losh * Copyright (c) 2011 Google, Inc.
3ca987d46SWarner Losh * All rights reserved.
4ca987d46SWarner Losh *
5ca987d46SWarner Losh * Redistribution and use in source and binary forms, with or without
6ca987d46SWarner Losh * modification, are permitted provided that the following conditions
7ca987d46SWarner Losh * are met:
8ca987d46SWarner Losh * 1. Redistributions of source code must retain the above copyright
9ca987d46SWarner Losh * notice, this list of conditions and the following disclaimer.
10ca987d46SWarner Losh * 2. Redistributions in binary form must reproduce the above copyright
11ca987d46SWarner Losh * notice, this list of conditions and the following disclaimer in the
12ca987d46SWarner Losh * documentation and/or other materials provided with the distribution.
13ca987d46SWarner Losh *
14ca987d46SWarner Losh * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15ca987d46SWarner Losh * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16ca987d46SWarner Losh * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17ca987d46SWarner Losh * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18ca987d46SWarner Losh * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19ca987d46SWarner Losh * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20ca987d46SWarner Losh * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21ca987d46SWarner Losh * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22ca987d46SWarner Losh * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23ca987d46SWarner Losh * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24ca987d46SWarner Losh * SUCH DAMAGE.
25ca987d46SWarner Losh */
26ca987d46SWarner Losh
27ca987d46SWarner Losh #include <sys/types.h>
28ca987d46SWarner Losh #include <sys/disk.h>
29ca987d46SWarner Losh #include <sys/ioctl.h>
30ca987d46SWarner Losh #include <sys/stat.h>
31ca987d46SWarner Losh #include <dirent.h>
32ca987d46SWarner Losh #include <dlfcn.h>
33ca987d46SWarner Losh #include <err.h>
34ca987d46SWarner Losh #include <errno.h>
35ca987d46SWarner Losh #include <fcntl.h>
36ca987d46SWarner Losh #include <getopt.h>
37ca987d46SWarner Losh #include <inttypes.h>
38ca987d46SWarner Losh #include <limits.h>
39ca987d46SWarner Losh #include <stdio.h>
40ca987d46SWarner Losh #include <stdlib.h>
41ca987d46SWarner Losh #include <string.h>
42ca987d46SWarner Losh #include <termios.h>
43ca987d46SWarner Losh #include <unistd.h>
44ca987d46SWarner Losh
45ca987d46SWarner Losh #include <userboot.h>
46ca987d46SWarner Losh
47ca987d46SWarner Losh char *host_base = NULL;
48ca987d46SWarner Losh struct termios term, oldterm;
49ca987d46SWarner Losh char *image;
50ca987d46SWarner Losh size_t image_size;
51ca987d46SWarner Losh
52ca987d46SWarner Losh uint64_t regs[16];
53ca987d46SWarner Losh uint64_t pc;
54bcbb1e60SToomas Soome int *disk_fd;
55bcbb1e60SToomas Soome int disk_index = -1;
56ca987d46SWarner Losh
57ca987d46SWarner Losh void test_exit(void *arg, int v);
58ca987d46SWarner Losh
59ca987d46SWarner Losh /*
60ca987d46SWarner Losh * Console i/o
61ca987d46SWarner Losh */
62ca987d46SWarner Losh
63ca987d46SWarner Losh void
test_putc(void * arg,int ch)64ca987d46SWarner Losh test_putc(void *arg, int ch)
65ca987d46SWarner Losh {
66ca987d46SWarner Losh char c = ch;
67ca987d46SWarner Losh
68ca987d46SWarner Losh write(1, &c, 1);
69ca987d46SWarner Losh }
70ca987d46SWarner Losh
71ca987d46SWarner Losh int
test_getc(void * arg)72ca987d46SWarner Losh test_getc(void *arg)
73ca987d46SWarner Losh {
74ca987d46SWarner Losh char c;
75ca987d46SWarner Losh
76ca987d46SWarner Losh if (read(0, &c, 1) == 1)
77ca987d46SWarner Losh return c;
78ca987d46SWarner Losh return -1;
79ca987d46SWarner Losh }
80ca987d46SWarner Losh
81ca987d46SWarner Losh int
test_poll(void * arg)82ca987d46SWarner Losh test_poll(void *arg)
83ca987d46SWarner Losh {
84ca987d46SWarner Losh int n;
85ca987d46SWarner Losh
86ca987d46SWarner Losh if (ioctl(0, FIONREAD, &n) >= 0)
87ca987d46SWarner Losh return (n > 0);
88ca987d46SWarner Losh return (0);
89ca987d46SWarner Losh }
90ca987d46SWarner Losh
91ca987d46SWarner Losh /*
92ca987d46SWarner Losh * Host filesystem i/o
93ca987d46SWarner Losh */
94ca987d46SWarner Losh
95ca987d46SWarner Losh struct test_file {
96ca987d46SWarner Losh int tf_isdir;
97ca987d46SWarner Losh size_t tf_size;
98ca987d46SWarner Losh struct stat tf_stat;
99ca987d46SWarner Losh union {
100ca987d46SWarner Losh int fd;
101ca987d46SWarner Losh DIR *dir;
102ca987d46SWarner Losh } tf_u;
103ca987d46SWarner Losh };
104ca987d46SWarner Losh
105ca987d46SWarner Losh int
test_open(void * arg,const char * filename,void ** h_return)106ca987d46SWarner Losh test_open(void *arg, const char *filename, void **h_return)
107ca987d46SWarner Losh {
108ca987d46SWarner Losh struct stat st;
109ca987d46SWarner Losh struct test_file *tf;
110ca987d46SWarner Losh char path[PATH_MAX];
111ca987d46SWarner Losh
112ca987d46SWarner Losh if (!host_base)
113ca987d46SWarner Losh return (ENOENT);
114ca987d46SWarner Losh
115ca987d46SWarner Losh strlcpy(path, host_base, PATH_MAX);
116ca987d46SWarner Losh if (path[strlen(path) - 1] == '/')
117ca987d46SWarner Losh path[strlen(path) - 1] = 0;
118ca987d46SWarner Losh strlcat(path, filename, PATH_MAX);
119ca987d46SWarner Losh tf = malloc(sizeof(struct test_file));
120ca987d46SWarner Losh if (stat(path, &tf->tf_stat) < 0) {
121ca987d46SWarner Losh free(tf);
122ca987d46SWarner Losh return (errno);
123ca987d46SWarner Losh }
124ca987d46SWarner Losh
125ca987d46SWarner Losh tf->tf_size = st.st_size;
126ca987d46SWarner Losh if (S_ISDIR(tf->tf_stat.st_mode)) {
127ca987d46SWarner Losh tf->tf_isdir = 1;
128ca987d46SWarner Losh tf->tf_u.dir = opendir(path);
129ca987d46SWarner Losh if (!tf->tf_u.dir)
130ca987d46SWarner Losh goto out;
131ca987d46SWarner Losh *h_return = tf;
132ca987d46SWarner Losh return (0);
133ca987d46SWarner Losh }
134ca987d46SWarner Losh if (S_ISREG(tf->tf_stat.st_mode)) {
135ca987d46SWarner Losh tf->tf_isdir = 0;
136ca987d46SWarner Losh tf->tf_u.fd = open(path, O_RDONLY);
137ca987d46SWarner Losh if (tf->tf_u.fd < 0)
138ca987d46SWarner Losh goto out;
139ca987d46SWarner Losh *h_return = tf;
140ca987d46SWarner Losh return (0);
141ca987d46SWarner Losh }
142ca987d46SWarner Losh
143ca987d46SWarner Losh out:
144ca987d46SWarner Losh free(tf);
145ca987d46SWarner Losh return (EINVAL);
146ca987d46SWarner Losh }
147ca987d46SWarner Losh
148ca987d46SWarner Losh int
test_close(void * arg,void * h)149ca987d46SWarner Losh test_close(void *arg, void *h)
150ca987d46SWarner Losh {
151ca987d46SWarner Losh struct test_file *tf = h;
152ca987d46SWarner Losh
153ca987d46SWarner Losh if (tf->tf_isdir)
154ca987d46SWarner Losh closedir(tf->tf_u.dir);
155ca987d46SWarner Losh else
156ca987d46SWarner Losh close(tf->tf_u.fd);
157ca987d46SWarner Losh free(tf);
158ca987d46SWarner Losh
159ca987d46SWarner Losh return (0);
160ca987d46SWarner Losh }
161ca987d46SWarner Losh
162ca987d46SWarner Losh int
test_isdir(void * arg,void * h)163ca987d46SWarner Losh test_isdir(void *arg, void *h)
164ca987d46SWarner Losh {
165ca987d46SWarner Losh struct test_file *tf = h;
166ca987d46SWarner Losh
167ca987d46SWarner Losh return (tf->tf_isdir);
168ca987d46SWarner Losh }
169ca987d46SWarner Losh
170ca987d46SWarner Losh int
test_read(void * arg,void * h,void * dst,size_t size,size_t * resid_return)171ca987d46SWarner Losh test_read(void *arg, void *h, void *dst, size_t size, size_t *resid_return)
172ca987d46SWarner Losh {
173ca987d46SWarner Losh struct test_file *tf = h;
174ca987d46SWarner Losh ssize_t sz;
175ca987d46SWarner Losh
176ca987d46SWarner Losh if (tf->tf_isdir)
177ca987d46SWarner Losh return (EINVAL);
178ca987d46SWarner Losh sz = read(tf->tf_u.fd, dst, size);
179ca987d46SWarner Losh if (sz < 0)
180ca987d46SWarner Losh return (EINVAL);
181ca987d46SWarner Losh *resid_return = size - sz;
182ca987d46SWarner Losh return (0);
183ca987d46SWarner Losh }
184ca987d46SWarner Losh
185ca987d46SWarner Losh int
test_readdir(void * arg,void * h,uint32_t * fileno_return,uint8_t * type_return,size_t * namelen_return,char * name)186ca987d46SWarner Losh test_readdir(void *arg, void *h, uint32_t *fileno_return, uint8_t *type_return,
187ca987d46SWarner Losh size_t *namelen_return, char *name)
188ca987d46SWarner Losh {
189ca987d46SWarner Losh struct test_file *tf = h;
190ca987d46SWarner Losh struct dirent *dp;
191ca987d46SWarner Losh
192ca987d46SWarner Losh if (!tf->tf_isdir)
193ca987d46SWarner Losh return (EINVAL);
194ca987d46SWarner Losh
195ca987d46SWarner Losh dp = readdir(tf->tf_u.dir);
196ca987d46SWarner Losh if (!dp)
197ca987d46SWarner Losh return (ENOENT);
198ca987d46SWarner Losh
199ca987d46SWarner Losh /*
200ca987d46SWarner Losh * Note: d_namlen is in the range 0..255 and therefore less
201ca987d46SWarner Losh * than PATH_MAX so we don't need to test before copying.
202ca987d46SWarner Losh */
203ca987d46SWarner Losh *fileno_return = dp->d_fileno;
204ca987d46SWarner Losh *type_return = dp->d_type;
205ca987d46SWarner Losh *namelen_return = dp->d_namlen;
206ca987d46SWarner Losh memcpy(name, dp->d_name, dp->d_namlen);
207ca987d46SWarner Losh name[dp->d_namlen] = 0;
208ca987d46SWarner Losh
209ca987d46SWarner Losh return (0);
210ca987d46SWarner Losh }
211ca987d46SWarner Losh
212ca987d46SWarner Losh int
test_seek(void * arg,void * h,uint64_t offset,int whence)213ca987d46SWarner Losh test_seek(void *arg, void *h, uint64_t offset, int whence)
214ca987d46SWarner Losh {
215ca987d46SWarner Losh struct test_file *tf = h;
216ca987d46SWarner Losh
217ca987d46SWarner Losh if (tf->tf_isdir)
218ca987d46SWarner Losh return (EINVAL);
219ca987d46SWarner Losh if (lseek(tf->tf_u.fd, offset, whence) < 0)
220ca987d46SWarner Losh return (errno);
221ca987d46SWarner Losh return (0);
222ca987d46SWarner Losh }
223ca987d46SWarner Losh
224ca987d46SWarner Losh int
test_stat(void * arg,void * h,struct stat * stp)22553f151f9SSimon J. Gerraty test_stat(void *arg, void *h, struct stat *stp)
226ca987d46SWarner Losh {
227ca987d46SWarner Losh struct test_file *tf = h;
228ca987d46SWarner Losh
22953f151f9SSimon J. Gerraty if (!stp)
23053f151f9SSimon J. Gerraty return (-1);
23153f151f9SSimon J. Gerraty memset(stp, 0, sizeof(struct stat));
23253f151f9SSimon J. Gerraty stp->st_mode = tf->tf_stat.st_mode;
23353f151f9SSimon J. Gerraty stp->st_uid = tf->tf_stat.st_uid;
23453f151f9SSimon J. Gerraty stp->st_gid = tf->tf_stat.st_gid;
23553f151f9SSimon J. Gerraty stp->st_size = tf->tf_stat.st_size;
23653f151f9SSimon J. Gerraty stp->st_ino = tf->tf_stat.st_ino;
23753f151f9SSimon J. Gerraty stp->st_dev = tf->tf_stat.st_dev;
23853f151f9SSimon J. Gerraty stp->st_mtime = tf->tf_stat.st_mtime;
239ca987d46SWarner Losh return (0);
240ca987d46SWarner Losh }
241ca987d46SWarner Losh
242ca987d46SWarner Losh /*
243ca987d46SWarner Losh * Disk image i/o
244ca987d46SWarner Losh */
245ca987d46SWarner Losh
246ca987d46SWarner Losh int
test_diskread(void * arg,int unit,uint64_t offset,void * dst,size_t size,size_t * resid_return)247ca987d46SWarner Losh test_diskread(void *arg, int unit, uint64_t offset, void *dst, size_t size,
248ca987d46SWarner Losh size_t *resid_return)
249ca987d46SWarner Losh {
250ca987d46SWarner Losh ssize_t n;
251ca987d46SWarner Losh
252bcbb1e60SToomas Soome if (unit > disk_index || disk_fd[unit] == -1)
253ca987d46SWarner Losh return (EIO);
254bcbb1e60SToomas Soome n = pread(disk_fd[unit], dst, size, offset);
255942e52f7SToomas Soome if (n == 0) {
256942e52f7SToomas Soome printf("%s: end of disk (%ju)\n", __func__, (intmax_t)offset);
257942e52f7SToomas Soome return (EIO);
258942e52f7SToomas Soome }
259942e52f7SToomas Soome
260ca987d46SWarner Losh if (n < 0)
261ca987d46SWarner Losh return (errno);
262ca987d46SWarner Losh *resid_return = size - n;
263ca987d46SWarner Losh return (0);
264ca987d46SWarner Losh }
265ca987d46SWarner Losh
266ca987d46SWarner Losh int
test_diskwrite(void * arg,int unit,uint64_t offset,void * src,size_t size,size_t * resid_return)267e307eb94SToomas Soome test_diskwrite(void *arg, int unit, uint64_t offset, void *src, size_t size,
268e307eb94SToomas Soome size_t *resid_return)
269e307eb94SToomas Soome {
270e307eb94SToomas Soome ssize_t n;
271e307eb94SToomas Soome
272e307eb94SToomas Soome if (unit > disk_index || disk_fd[unit] == -1)
273e307eb94SToomas Soome return (EIO);
274e307eb94SToomas Soome n = pwrite(disk_fd[unit], src, size, offset);
275e307eb94SToomas Soome if (n < 0)
276e307eb94SToomas Soome return (errno);
277e307eb94SToomas Soome *resid_return = size - n;
278e307eb94SToomas Soome return (0);
279e307eb94SToomas Soome }
280e307eb94SToomas Soome
281e307eb94SToomas Soome int
test_diskioctl(void * arg,int unit,u_long cmd,void * data)282ca987d46SWarner Losh test_diskioctl(void *arg, int unit, u_long cmd, void *data)
283ca987d46SWarner Losh {
284ca987d46SWarner Losh struct stat sb;
285ca987d46SWarner Losh
286bcbb1e60SToomas Soome if (unit > disk_index || disk_fd[unit] == -1)
287ca987d46SWarner Losh return (EBADF);
288ca987d46SWarner Losh switch (cmd) {
289ca987d46SWarner Losh case DIOCGSECTORSIZE:
290ca987d46SWarner Losh *(u_int *)data = 512;
291ca987d46SWarner Losh break;
292ca987d46SWarner Losh case DIOCGMEDIASIZE:
293bcbb1e60SToomas Soome if (fstat(disk_fd[unit], &sb) == 0)
294ca987d46SWarner Losh *(off_t *)data = sb.st_size;
295ca987d46SWarner Losh else
296ca987d46SWarner Losh return (ENOTTY);
297ca987d46SWarner Losh break;
298ca987d46SWarner Losh default:
299ca987d46SWarner Losh return (ENOTTY);
300ca987d46SWarner Losh }
301ca987d46SWarner Losh return (0);
302ca987d46SWarner Losh }
303ca987d46SWarner Losh
304ca987d46SWarner Losh /*
305ca987d46SWarner Losh * Guest virtual machine i/o
306ca987d46SWarner Losh *
307ca987d46SWarner Losh * Note: guest addresses are kernel virtual
308ca987d46SWarner Losh */
309ca987d46SWarner Losh
310ca987d46SWarner Losh int
test_copyin(void * arg,const void * from,uint64_t to,size_t size)311ca987d46SWarner Losh test_copyin(void *arg, const void *from, uint64_t to, size_t size)
312ca987d46SWarner Losh {
313ca987d46SWarner Losh
314ca987d46SWarner Losh to &= 0x7fffffff;
315ca987d46SWarner Losh if (to > image_size)
316ca987d46SWarner Losh return (EFAULT);
317ca987d46SWarner Losh if (to + size > image_size)
318ca987d46SWarner Losh size = image_size - to;
319ca987d46SWarner Losh memcpy(&image[to], from, size);
320ca987d46SWarner Losh return(0);
321ca987d46SWarner Losh }
322ca987d46SWarner Losh
323ca987d46SWarner Losh int
test_copyout(void * arg,uint64_t from,void * to,size_t size)324ca987d46SWarner Losh test_copyout(void *arg, uint64_t from, void *to, size_t size)
325ca987d46SWarner Losh {
326ca987d46SWarner Losh
327ca987d46SWarner Losh from &= 0x7fffffff;
328ca987d46SWarner Losh if (from > image_size)
329ca987d46SWarner Losh return (EFAULT);
330ca987d46SWarner Losh if (from + size > image_size)
331ca987d46SWarner Losh size = image_size - from;
332ca987d46SWarner Losh memcpy(to, &image[from], size);
333ca987d46SWarner Losh return(0);
334ca987d46SWarner Losh }
335ca987d46SWarner Losh
336ca987d46SWarner Losh void
test_setreg(void * arg,int r,uint64_t v)337ca987d46SWarner Losh test_setreg(void *arg, int r, uint64_t v)
338ca987d46SWarner Losh {
339ca987d46SWarner Losh
340ca987d46SWarner Losh if (r < 0 || r >= 16)
341ca987d46SWarner Losh return;
342ca987d46SWarner Losh regs[r] = v;
343ca987d46SWarner Losh }
344ca987d46SWarner Losh
345ca987d46SWarner Losh void
test_setmsr(void * arg,int r,uint64_t v)346ca987d46SWarner Losh test_setmsr(void *arg, int r, uint64_t v)
347ca987d46SWarner Losh {
348ca987d46SWarner Losh }
349ca987d46SWarner Losh
350ca987d46SWarner Losh void
test_setcr(void * arg,int r,uint64_t v)351ca987d46SWarner Losh test_setcr(void *arg, int r, uint64_t v)
352ca987d46SWarner Losh {
353ca987d46SWarner Losh }
354ca987d46SWarner Losh
355ca987d46SWarner Losh void
test_setgdt(void * arg,uint64_t v,size_t sz)356ca987d46SWarner Losh test_setgdt(void *arg, uint64_t v, size_t sz)
357ca987d46SWarner Losh {
358ca987d46SWarner Losh }
359ca987d46SWarner Losh
360ca987d46SWarner Losh void
test_exec(void * arg,uint64_t pc)361ca987d46SWarner Losh test_exec(void *arg, uint64_t pc)
362ca987d46SWarner Losh {
36396b2f9c9SToomas Soome printf("Execute at 0x%"PRIx64"\n", pc);
364ca987d46SWarner Losh test_exit(arg, 0);
365ca987d46SWarner Losh }
366ca987d46SWarner Losh
367ca987d46SWarner Losh /*
368ca987d46SWarner Losh * Misc
369ca987d46SWarner Losh */
370ca987d46SWarner Losh
371ca987d46SWarner Losh void
test_delay(void * arg,int usec)372ca987d46SWarner Losh test_delay(void *arg, int usec)
373ca987d46SWarner Losh {
374ca987d46SWarner Losh
375ca987d46SWarner Losh usleep(usec);
376ca987d46SWarner Losh }
377ca987d46SWarner Losh
378ca987d46SWarner Losh void
test_exit(void * arg,int v)379ca987d46SWarner Losh test_exit(void *arg, int v)
380ca987d46SWarner Losh {
381ca987d46SWarner Losh
382ca987d46SWarner Losh tcsetattr(0, TCSAFLUSH, &oldterm);
383ca987d46SWarner Losh exit(v);
384ca987d46SWarner Losh }
385ca987d46SWarner Losh
386ca987d46SWarner Losh void
test_getmem(void * arg,uint64_t * lowmem,uint64_t * highmem)387ca987d46SWarner Losh test_getmem(void *arg, uint64_t *lowmem, uint64_t *highmem)
388ca987d46SWarner Losh {
389ca987d46SWarner Losh
390ca987d46SWarner Losh *lowmem = 128*1024*1024;
391ca987d46SWarner Losh *highmem = 0;
392ca987d46SWarner Losh }
393ca987d46SWarner Losh
394e8e6a5f9SWarner Losh char *
test_getenv(void * arg,int idx)395ca987d46SWarner Losh test_getenv(void *arg, int idx)
396ca987d46SWarner Losh {
397e8e6a5f9SWarner Losh static char *vars[] = {
398ca987d46SWarner Losh "foo=bar",
399ca987d46SWarner Losh "bar=barbar",
400ca987d46SWarner Losh NULL
401ca987d46SWarner Losh };
402ca987d46SWarner Losh
403ca987d46SWarner Losh return (vars[idx]);
404ca987d46SWarner Losh }
405ca987d46SWarner Losh
406ca987d46SWarner Losh struct loader_callbacks cb = {
407ca987d46SWarner Losh .putc = test_putc,
408ca987d46SWarner Losh .getc = test_getc,
409ca987d46SWarner Losh .poll = test_poll,
410ca987d46SWarner Losh
411ca987d46SWarner Losh .open = test_open,
412ca987d46SWarner Losh .close = test_close,
413ca987d46SWarner Losh .isdir = test_isdir,
414ca987d46SWarner Losh .read = test_read,
415ca987d46SWarner Losh .readdir = test_readdir,
416ca987d46SWarner Losh .seek = test_seek,
417ca987d46SWarner Losh .stat = test_stat,
418ca987d46SWarner Losh
419ca987d46SWarner Losh .diskread = test_diskread,
420e307eb94SToomas Soome .diskwrite = test_diskwrite,
421ca987d46SWarner Losh .diskioctl = test_diskioctl,
422ca987d46SWarner Losh
423ca987d46SWarner Losh .copyin = test_copyin,
424ca987d46SWarner Losh .copyout = test_copyout,
425ca987d46SWarner Losh .setreg = test_setreg,
426ca987d46SWarner Losh .setmsr = test_setmsr,
427ca987d46SWarner Losh .setcr = test_setcr,
428ca987d46SWarner Losh .setgdt = test_setgdt,
429ca987d46SWarner Losh .exec = test_exec,
430ca987d46SWarner Losh
431ca987d46SWarner Losh .delay = test_delay,
432ca987d46SWarner Losh .exit = test_exit,
433ca987d46SWarner Losh .getmem = test_getmem,
434ca987d46SWarner Losh
435ca987d46SWarner Losh .getenv = test_getenv,
436ca987d46SWarner Losh };
437ca987d46SWarner Losh
438ca987d46SWarner Losh void
usage()439ca987d46SWarner Losh usage()
440ca987d46SWarner Losh {
441ca987d46SWarner Losh
442ca987d46SWarner Losh printf("usage: [-b <userboot shared object>] [-d <disk image path>] [-h <host filesystem path>\n");
443ca987d46SWarner Losh exit(1);
444ca987d46SWarner Losh }
445ca987d46SWarner Losh
446ca987d46SWarner Losh int
main(int argc,char ** argv)447ca987d46SWarner Losh main(int argc, char** argv)
448ca987d46SWarner Losh {
449ca987d46SWarner Losh void *h;
450d927d443SWarner Losh void (*func)(struct loader_callbacks *, void *, int, int) __dead2;
451ca987d46SWarner Losh int opt;
452ca987d46SWarner Losh const char *userboot_obj = "/boot/userboot.so";
453e307eb94SToomas Soome int oflag = O_RDONLY;
454ca987d46SWarner Losh
455e307eb94SToomas Soome while ((opt = getopt(argc, argv, "wb:d:h:")) != -1) {
456ca987d46SWarner Losh switch (opt) {
457ca987d46SWarner Losh case 'b':
458ca987d46SWarner Losh userboot_obj = optarg;
459ca987d46SWarner Losh break;
460ca987d46SWarner Losh
461ca987d46SWarner Losh case 'd':
462bcbb1e60SToomas Soome disk_index++;
463bcbb1e60SToomas Soome disk_fd = reallocarray(disk_fd, disk_index + 1,
464bcbb1e60SToomas Soome sizeof (int));
465e307eb94SToomas Soome disk_fd[disk_index] = open(optarg, oflag);
466bcbb1e60SToomas Soome if (disk_fd[disk_index] < 0)
467bcbb1e60SToomas Soome err(1, "Can't open disk image '%s'", optarg);
468ca987d46SWarner Losh break;
469ca987d46SWarner Losh
470ca987d46SWarner Losh case 'h':
471ca987d46SWarner Losh host_base = optarg;
472ca987d46SWarner Losh break;
473ca987d46SWarner Losh
474e307eb94SToomas Soome case 'w':
475e307eb94SToomas Soome oflag = O_RDWR;
476e307eb94SToomas Soome break;
477e307eb94SToomas Soome
478ca987d46SWarner Losh case '?':
479ca987d46SWarner Losh usage();
480ca987d46SWarner Losh }
481ca987d46SWarner Losh }
482ca987d46SWarner Losh
483ca987d46SWarner Losh h = dlopen(userboot_obj, RTLD_LOCAL);
484ca987d46SWarner Losh if (!h) {
485ca987d46SWarner Losh printf("%s\n", dlerror());
486ca987d46SWarner Losh return (1);
487ca987d46SWarner Losh }
488ca987d46SWarner Losh func = dlsym(h, "loader_main");
489ca987d46SWarner Losh if (!func) {
490ca987d46SWarner Losh printf("%s\n", dlerror());
491ca987d46SWarner Losh return (1);
492ca987d46SWarner Losh }
493ca987d46SWarner Losh
494ca987d46SWarner Losh image_size = 128*1024*1024;
495ca987d46SWarner Losh image = malloc(image_size);
496ca987d46SWarner Losh
497ca987d46SWarner Losh tcgetattr(0, &term);
498ca987d46SWarner Losh oldterm = term;
499ca987d46SWarner Losh term.c_iflag &= ~(ICRNL);
500ca987d46SWarner Losh term.c_lflag &= ~(ICANON|ECHO);
501ca987d46SWarner Losh tcsetattr(0, TCSAFLUSH, &term);
502ca987d46SWarner Losh
503bcbb1e60SToomas Soome func(&cb, NULL, USERBOOT_VERSION_3, disk_index + 1);
504ca987d46SWarner Losh }
505