diff options
Diffstat (limited to 'scripts')
-rw-r--r-- | scripts/qemucommand.py | 118 | ||||
-rwxr-xr-x | scripts/run-qemu-ota | 118 |
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 @@ | |||
1 | from os.path import exists, join, realpath | ||
2 | from os import listdir | ||
3 | import random | ||
4 | import socket | ||
5 | |||
6 | EXTENSIONS = { | ||
7 | 'intel-corei7-64': 'wic', | ||
8 | 'qemux86-64': 'otaimg' | ||
9 | } | ||
10 | |||
11 | |||
12 | def 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 | |||
29 | def 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 | |||
39 | class 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 | ||
3 | from argparse import ArgumentParser | 3 | from argparse import ArgumentParser |
4 | from subprocess import Popen | 4 | from subprocess import Popen |
5 | from os.path import exists, join, realpath | 5 | from os.path import exists, join |
6 | from os import listdir | ||
7 | import random | ||
8 | import sys | 6 | import sys |
9 | import socket | 7 | from qemucommand import QemuCommand |
10 | 8 | ||
11 | DEFAULT_DIR = 'tmp/deploy/images' | 9 | DEFAULT_DIR = 'tmp/deploy/images' |
12 | 10 | ||
13 | EXTENSIONS = { | ||
14 | 'intel-corei7-64': 'wic', | ||
15 | 'qemux86-64': 'otaimg' | ||
16 | } | ||
17 | |||
18 | |||
19 | def 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 | |||
36 | def 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 | |||
46 | class 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 | ||
126 | def main(): | 12 | def main(): |
127 | parser = ArgumentParser(description='Run meta-updater image in qemu') | 13 | parser = ArgumentParser(description='Run meta-updater image in qemu') |