forked from dan/liminix
140 lines
4.5 KiB
Fennel
140 lines
4.5 KiB
Fennel
(local { : fork : execp : unlink } (require :posix.unistd))
|
|
(local { : wait } (require :posix.sys.wait))
|
|
(local { : mkstemp : setenv } (require :posix.stdlib))
|
|
(local { : fdopen } (require :posix.stdio))
|
|
|
|
(fn pad-file [name kb chr]
|
|
(let [(fd out) (mkstemp "run-vm-XXXXXX")
|
|
pad-string (string.rep (or chr "\0") 1024)]
|
|
(with-open [f (fdopen fd :w)]
|
|
(for [i 1 kb] (f:write pad-string))
|
|
(f:seek :set 0)
|
|
(with-open [input (assert (io.open name :rb))]
|
|
(f:write (input:read "*a")))
|
|
(f:seek :end 0))
|
|
out))
|
|
|
|
(fn spawn [command args]
|
|
(match (fork)
|
|
(nil msg) (error (.. "couldn't fork: " msg))
|
|
0 (execp command args)
|
|
pid (wait pid)))
|
|
|
|
(fn appendm [t2 t1]
|
|
(table.move t1 1 (# t1) (+ 1 (# t2)) t2)
|
|
t2)
|
|
|
|
(fn merge [table1 table2]
|
|
(collect [k v (pairs table2) &into table1]
|
|
k v))
|
|
|
|
(fn assoc [tbl k v]
|
|
(tset tbl k v)
|
|
tbl)
|
|
|
|
(fn parse-args [args]
|
|
(match args
|
|
["--background" dir & rest] (assoc (parse-args rest) :background dir)
|
|
["--u-boot" bin & rest]
|
|
(assoc (parse-args rest) :u-boot bin)
|
|
["--disk-image" image & rest ] (assoc (parse-args rest)
|
|
:disk-image (pad-file image 1024))
|
|
["--arch" arch & rest] (assoc (parse-args rest) :arch arch)
|
|
["--phram-address" addr & rest] (assoc (parse-args rest) :phram-address addr)
|
|
["--lan" spec & rest] (assoc (parse-args rest) :lan spec)
|
|
["--wan" spec & rest] (assoc (parse-args rest) :wan spec)
|
|
["--command-line" cmd & rest] (assoc (parse-args rest) :command-line cmd)
|
|
[kernel rootfsimg]
|
|
{ :kernel kernel :rootfs (pad-file rootfsimg (* 16 1024)) }
|
|
))
|
|
|
|
(fn pad-u-boot [options]
|
|
(if options.u-boot
|
|
(let [size (.
|
|
{
|
|
:mips (* 4 1024)
|
|
:aarch64 (* 64 1024)
|
|
:arm (* 64 1024)
|
|
}
|
|
options.arch)]
|
|
(assoc options
|
|
:u-boot
|
|
(pad-file options.u-boot size "\xff")))
|
|
options))
|
|
|
|
(local options
|
|
(assert
|
|
(pad-u-boot
|
|
(merge { :arch "mips" } (parse-args arg)))
|
|
(.. "Usage: " (. arg 0) " blah bah")))
|
|
|
|
(fn background [dir]
|
|
(let [pid (.. dir "/pid")
|
|
sock (.. dir "/console")
|
|
monitor (.. dir "/monitor")]
|
|
["--daemonize"
|
|
"--pidfile" pid
|
|
"-serial" (.. "unix:" sock ",server,nowait")
|
|
"-monitor" (.. "unix:" monitor ",server,nowait")]))
|
|
|
|
(fn access-net [override]
|
|
[
|
|
"-netdev" (.. (or override
|
|
"socket,mcast=230.0.0.1:1234,localaddr=127.0.0.1")
|
|
",id=access")
|
|
"-device" "virtio-net,disable-legacy=on,disable-modern=off,netdev=access,mac=ba:ad:1d:ea:21:02"
|
|
])
|
|
|
|
(fn local-net [override]
|
|
[
|
|
"-netdev" (.. (or override "socket,mcast=230.0.0.1:1235,localaddr=127.0.0.1")
|
|
",id=lan")
|
|
"-device" "virtio-net,disable-legacy=on,disable-modern=off,netdev=lan,mac=ba:ad:1d:ea:21:01"
|
|
])
|
|
|
|
|
|
(fn bootable [cmdline uboot disk]
|
|
(if uboot
|
|
["-drive" (.. "if=pflash,format=raw,file=" uboot )
|
|
"-drive" (.. "if=none,format=raw,id=hd0,file=" disk)
|
|
"-device" "virtio-blk-pci,drive=hd0"
|
|
]
|
|
(let [cmdline (.. cmdline " mem=256M liminix mtdparts=phram0:16M(rootfs) phram.phram=phram0," options.phram-address ",16Mi,65536 root=/dev/mtdblock0")]
|
|
["-kernel" options.kernel "-append" cmdline])))
|
|
|
|
(local bin {
|
|
:mips ["qemu-system-mips" "-M" "malta"]
|
|
:aarch64 ["qemu-system-aarch64" "-M" "virt"
|
|
"-cpu" "cortex-a72"]
|
|
:arm ["qemu-system-arm" "-M" "virt,highmem=off"
|
|
"-cpu" "cortex-a15"]
|
|
})
|
|
|
|
(local exec-args
|
|
(-> []
|
|
(appendm (. bin options.arch))
|
|
(appendm ["-m" "272"
|
|
"-echr" "16"
|
|
"-device"
|
|
(.. "loader,file=" options.rootfs ",addr=" options.phram-address)
|
|
])
|
|
(appendm
|
|
(if options.background
|
|
(background options.background)
|
|
["-serial" "mon:stdio"]))
|
|
(appendm (bootable (or options.command-line "")
|
|
options.u-boot options.disk-image))
|
|
(appendm (access-net options.wan))
|
|
(appendm (local-net options.lan))
|
|
(appendm ["-display" "none"])))
|
|
|
|
(each [n a (ipairs exec-args)]
|
|
(print (.. (if (> n 1) " " "") (string.format "%q" a))))
|
|
|
|
(match exec-args
|
|
[cmd & params] (print (spawn cmd params)))
|
|
|
|
(if options.rootfs (unlink options.rootfs))
|
|
(if options.u-boot (unlink options.u-boot))
|
|
(if options.disk-image (unlink options.disk-image))
|