summaryrefslogtreecommitdiffstats
path: root/scripts
diff options
context:
space:
mode:
Diffstat (limited to 'scripts')
-rw-r--r--scripts/qemucommand.py118
-rwxr-xr-xscripts/run-qemu-ota118
2 files changed, 120 insertions, 116 deletions
diff --git a/scripts/qemucommand.py b/scripts/qemucommand.py
new file mode 100644
index 0000000..ed14d9b
--- /dev/null
+++ b/scripts/qemucommand.py
@@ -0,0 +1,118 @@
1from os.path import exists, join, realpath
2from os import listdir
3import random
4import socket
5
6EXTENSIONS = {
7 'intel-corei7-64': 'wic',
8 'qemux86-64': 'otaimg'
9}
10
11
12def find_local_port(start_port):
13 """"
14 Find the next free TCP port after 'start_port'.
15 """
16
17 for port in range(start_port, start_port + 10):
18 try:
19 s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
20 s.bind(('', port))
21 return port
22 except socket.error:
23 print("Skipping port %d" % port)
24 finally:
25 s.close()
26 raise Exception("Could not find a free TCP port")
27
28
29def random_mac():
30 """Return a random Ethernet MAC address
31 @link https://www.iana.org/assignments/ethernet-numbers/ethernet-numbers.xhtml#ethernet-numbers-2
32 """
33 head = "ca:fe:"
34 hex_digits = '0123456789abcdef'
35 tail = ':'.join([random.choice(hex_digits) + random.choice(hex_digits) for _ in range(4)])
36 return head + tail
37
38
39class QemuCommand(object):
40 def __init__(self, args):
41 if args.machine:
42 self.machine = args.machine
43 else:
44 machines = listdir(args.dir)
45 if len(machines) == 1:
46 self.machine = machines[0]
47 else:
48 raise ValueError("Could not autodetect machine type from %s" % args.dir)
49 if args.efi:
50 self.bios = 'OVMF.fd'
51 else:
52 uboot = join(args.dir, self.machine, 'u-boot-qemux86-64.rom')
53 if not exists(uboot):
54 raise ValueError("U-Boot image %s does not exist" % uboot)
55 self.bios = uboot
56 if exists(args.imagename):
57 image = args.imagename
58 else:
59 ext = EXTENSIONS.get(self.machine, 'wic')
60 image = join(args.dir, self.machine, '%s-%s.%s' % (args.imagename, self.machine, ext))
61 self.image = realpath(image)
62 if not exists(self.image):
63 raise ValueError("OS image %s does not exist" % self.image)
64 if args.mac:
65 self.mac_address = args.mac
66 else:
67 self.mac_address = random_mac()
68 self.serial_port = find_local_port(8990)
69 self.ssh_port = find_local_port(2222)
70 self.kvm = not args.no_kvm
71 self.gui = not args.no_gui
72 self.gdb = args.gdb
73 self.pcap = args.pcap
74 self.overlay = args.overlay
75
76 def command_line(self):
77 netuser = 'user,hostfwd=tcp:0.0.0.0:%d-:22,restrict=off' % self.ssh_port
78 if self.gdb:
79 netuser += ',hostfwd=tcp:0.0.0.0:2159-:2159'
80 cmdline = [
81 "qemu-system-x86_64",
82 "-bios", self.bios
83 ]
84 if not self.overlay:
85 cmdline += ["-drive", "file=%s,if=ide,format=raw,snapshot=on" % self.image]
86 cmdline += [
87 "-serial", "tcp:127.0.0.1:%d,server,nowait" % self.serial_port,
88 "-m", "1G",
89 "-usb",
90 "-usbdevice", "tablet",
91 "-show-cursor",
92 "-vga", "std",
93 "-net", netuser,
94 "-net", "nic,macaddr=%s" % self.mac_address
95 ]
96 if self.pcap:
97 cmdline += ['-net', 'dump,file=' + self.pcap]
98 if self.gui:
99 cmdline += ["-serial", "stdio"]
100 else:
101 cmdline.append('-nographic')
102 if self.kvm:
103 cmdline.append('-enable-kvm')
104 else:
105 cmdline += ['-cpu', 'Haswell']
106 if self.overlay:
107 cmdline.append(self.overlay)
108 return cmdline
109
110 def img_command_line(self):
111 cmdline = [
112 "qemu-img", "create",
113 "-o", "backing_file=%s" % self.image,
114 "-f", "qcow2",
115 self.overlay]
116 return cmdline
117
118
diff --git a/scripts/run-qemu-ota b/scripts/run-qemu-ota
index 641296c..5f9cebe 100755
--- a/scripts/run-qemu-ota
+++ b/scripts/run-qemu-ota
@@ -2,126 +2,12 @@
2 2
3from argparse import ArgumentParser 3from argparse import ArgumentParser
4from subprocess import Popen 4from subprocess import Popen
5from os.path import exists, join, realpath 5from os.path import exists, join
6from os import listdir
7import random
8import sys 6import sys
9import socket 7from qemucommand import QemuCommand
10 8
11DEFAULT_DIR = 'tmp/deploy/images' 9DEFAULT_DIR = 'tmp/deploy/images'
12 10
13EXTENSIONS = {
14 'intel-corei7-64': 'wic',
15 'qemux86-64': 'otaimg'
16}
17
18
19def find_local_port(start_port):
20 """"
21 Find the next free TCP port after 'start_port'.
22 """
23
24 for port in range(start_port, start_port + 10):
25 try:
26 s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
27 s.bind(('', port))
28 return port
29 except socket.error:
30 print("Skipping port %d" % port)
31 finally:
32 s.close()
33 raise Exception("Could not find a free TCP port")
34
35
36def random_mac():
37 """Return a random Ethernet MAC address
38 @link https://www.iana.org/assignments/ethernet-numbers/ethernet-numbers.xhtml#ethernet-numbers-2
39 """
40 head = "ca:fe:"
41 hex_digits = '0123456789abcdef'
42 tail = ':'.join([random.choice(hex_digits) + random.choice(hex_digits) for _ in range(4)])
43 return head + tail
44
45
46class QemuCommand(object):
47 def __init__(self, args):
48 if args.machine:
49 self.machine = args.machine
50 else:
51 machines = listdir(args.dir)
52 if len(machines) == 1:
53 self.machine = machines[0]
54 else:
55 raise ValueError("Could not autodetect machine type from %s" % args.dir)
56 if args.efi:
57 self.bios = 'OVMF.fd'
58 else:
59 uboot = join(args.dir, self.machine, 'u-boot-qemux86-64.rom')
60 if not exists(uboot):
61 raise ValueError("U-Boot image %s does not exist" % uboot)
62 self.bios = uboot
63 if exists(args.imagename):
64 image = args.imagename
65 else:
66 ext = EXTENSIONS.get(self.machine, 'wic')
67 image = join(args.dir, self.machine, '%s-%s.%s' % (args.imagename, self.machine, ext))
68 self.image = realpath(image)
69 if not exists(self.image):
70 raise ValueError("OS image %s does not exist" % self.image)
71 if args.mac:
72 self.mac_address = args.mac
73 else:
74 self.mac_address = random_mac()
75 self.serial_port = find_local_port(8990)
76 self.ssh_port = find_local_port(2222)
77 self.kvm = not args.no_kvm
78 self.gui = not args.no_gui
79 self.gdb = args.gdb
80 self.pcap = args.pcap
81 self.overlay = args.overlay
82
83 def command_line(self):
84 netuser = 'user,hostfwd=tcp:0.0.0.0:%d-:22,restrict=off' % self.ssh_port
85 if self.gdb:
86 netuser += ',hostfwd=tcp:0.0.0.0:2159-:2159'
87 cmdline = [
88 "qemu-system-x86_64",
89 "-bios", self.bios
90 ]
91 if not self.overlay:
92 cmdline += ["-drive", "file=%s,if=ide,format=raw,snapshot=on" % self.image]
93 cmdline += [
94 "-serial", "tcp:127.0.0.1:%d,server,nowait" % self.serial_port,
95 "-m", "1G",
96 "-usb",
97 "-usbdevice", "tablet",
98 "-show-cursor",
99 "-vga", "std",
100 "-net", netuser,
101 "-net", "nic,macaddr=%s" % self.mac_address
102 ]
103 if self.pcap:
104 cmdline += ['-net', 'dump,file=' + self.pcap]
105 if self.gui:
106 cmdline += ["-serial", "stdio"]
107 else:
108 cmdline.append('-nographic')
109 if self.kvm:
110 cmdline.append('-enable-kvm')
111 else:
112 cmdline += ['-cpu', 'Haswell']
113 if self.overlay:
114 cmdline.append(self.overlay)
115 return cmdline
116
117 def img_command_line(self):
118 cmdline = [
119 "qemu-img", "create",
120 "-o", "backing_file=%s" % self.image,
121 "-f", "qcow2",
122 self.overlay]
123 return cmdline
124
125 11
126def main(): 12def main():
127 parser = ArgumentParser(description='Run meta-updater image in qemu') 13 parser = ArgumentParser(description='Run meta-updater image in qemu')