1
2
3
4
5
6
7
8
9
10
11
12
13#include "qemu/osdep.h"
14#include "libqtest.h"
15#include "libqos/virtio.h"
16#include "qapi/qmp/qdict.h"
17
18
19#define qmp_discard_response(q, ...) qobject_unref(qtest_qmp(q, __VA_ARGS__))
20
21static void drive_add(QTestState *qts)
22{
23 char *resp = qtest_hmp(qts, "drive_add 0 if=none,id=drive0");
24
25 g_assert_cmpstr(resp, ==, "OK\r\n");
26 g_free(resp);
27}
28
29static void drive_del(QTestState *qts)
30{
31 char *resp = qtest_hmp(qts, "drive_del drive0");
32
33 g_assert_cmpstr(resp, ==, "");
34 g_free(resp);
35}
36
37static void device_del(QTestState *qts)
38{
39 QDict *response;
40
41
42 qmp_discard_response(qts, "{'execute': 'device_del',"
43 " 'arguments': { 'id': 'dev0' } }");
44 response = qtest_qmp_receive(qts);
45 g_assert(response);
46 g_assert(qdict_haskey(response, "return"));
47 qobject_unref(response);
48}
49
50static void test_drive_without_dev(void)
51{
52 QTestState *qts;
53
54
55 qts = qtest_init("-drive if=none,id=drive0");
56
57
58 drive_del(qts);
59
60
61
62
63 drive_add(qts);
64
65 qtest_quit(qts);
66}
67
68
69
70
71
72
73static const char *qvirtio_get_dev_type(void)
74{
75 const char *arch = qtest_get_arch();
76
77 if (g_str_equal(arch, "arm") || g_str_equal(arch, "aarch64")) {
78 return "device";
79 } else if (g_str_equal(arch, "s390x")) {
80 return "ccw";
81 } else {
82 return "pci";
83 }
84}
85
86static void test_after_failed_device_add(void)
87{
88 char driver[32];
89 QDict *response;
90 QTestState *qts;
91
92 snprintf(driver, sizeof(driver), "virtio-blk-%s",
93 qvirtio_get_dev_type());
94
95 qts = qtest_init("-drive if=none,id=drive0");
96
97
98
99
100 response = qtest_qmp(qts, "{'execute': 'device_add',"
101 " 'arguments': {"
102 " 'driver': %s,"
103 " 'drive': 'drive0'"
104 "}}", driver);
105 g_assert(response);
106 qmp_assert_error_class(response, "GenericError");
107
108
109 drive_del(qts);
110
111
112
113
114 drive_add(qts);
115
116 qtest_quit(qts);
117}
118
119static void test_drive_del_device_del(void)
120{
121 QTestState *qts;
122
123
124 qts = qtest_initf("-drive if=none,id=drive0,file=null-co://,format=raw"
125 " -device virtio-scsi-%s"
126 " -device scsi-hd,drive=drive0,id=dev0",
127 qvirtio_get_dev_type());
128
129
130
131
132
133 drive_del(qts);
134 device_del(qts);
135
136 qtest_quit(qts);
137}
138
139int main(int argc, char **argv)
140{
141 g_test_init(&argc, &argv, NULL);
142
143 qtest_add_func("/drive_del/without-dev", test_drive_without_dev);
144
145 if (qvirtio_get_dev_type() != NULL) {
146 qtest_add_func("/drive_del/after_failed_device_add",
147 test_after_failed_device_add);
148 qtest_add_func("/blockdev/drive_del_device_del",
149 test_drive_del_device_del);
150 }
151
152 return g_test_run();
153}
154