summaryrefslogtreecommitdiffstats
path: root/test/TEST-64-UDEV-STORAGE/nvme_basic.configure
blob: 28ddfa46c113683f261a21bf743aca4cc6df23b6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
#!/usr/bin/python3
# SPDX-License-Identifier: LGPL-2.1-or-later

import json
import os
import subprocess
import sys


config = json.load(sys.stdin)

qemu = f"qemu-system-{os.environ['QEMU_ARCHITECTURE']}"
result = subprocess.run([qemu, "-device", "help"], check=True, text=True, stdout=subprocess.PIPE)
if 'name "nvme"' not in result.stdout:
    print("nvme device driver is not available, skipping test...", file=sys.stderr)
    exit(77)

def add_drive(i: int, serial: str) -> None:
    global config
    id = f"nvme{i}"
    config["QemuDrives"] += [
        {
            "Id": id,
            "Size": "1M",
            "Options": "cache=unsafe",
        }
    ]
    config["QemuArgs"] += ["-device", f"nvme,drive={id},serial={serial},num_queues=8"]

for i in range(5):
    add_drive(i, serial=f"deadbeef{i}")
for i in range(5, 10):
    add_drive(i, serial=f"    deadbeef  {i}   ")
for i in range(10, 15):
    add_drive(i, serial=f"    dead/beef/{i}   ")
for i in range(15, 20):
    add_drive(i, serial=f"dead/../../beef/{i}")

json.dump(config, sys.stdout)