]>
Commit | Line | Data |
---|---|---|
30a3378a DM |
1 | package PVE::QMPClient; |
2 | ||
3 | use strict; | |
990fc5e2 | 4 | use warnings; |
30a3378a DM |
5 | use PVE::QemuServer; |
6 | use IO::Multiplex; | |
2ae10d4e | 7 | use POSIX qw(EINTR EAGAIN); |
30a3378a | 8 | use JSON; |
2ae10d4e | 9 | use Time::HiRes qw(usleep gettimeofday tv_interval); |
b3ea07f7 | 10 | use Scalar::Util qw(weaken); |
558f1644 | 11 | use PVE::IPCC; |
2ae10d4e | 12 | |
30a3378a DM |
13 | use Data::Dumper; |
14 | ||
15 | # Qemu Monitor Protocol (QMP) client. | |
16 | # | |
17 | # This implementation uses IO::Multiplex (libio-multiplex-perl) and | |
7a6c2150 | 18 | # allows you to issue qmp and qga commands to different VMs in parallel. |
30a3378a | 19 | |
7a6c2150 | 20 | # Note: qemu can onyl handle 1 connection, so we close connections asap |
30a3378a DM |
21 | |
22 | sub new { | |
c5a07de5 | 23 | my ($class, $eventcb) = @_; |
30a3378a DM |
24 | |
25 | my $mux = new IO::Multiplex; | |
26 | ||
27 | my $self = bless { | |
28 | mux => $mux, | |
7a6c2150 DM |
29 | queue_lookup => {}, # $fh => $queue_info |
30 | queue_info => {}, | |
30a3378a DM |
31 | }, $class; |
32 | ||
33 | $self->{eventcb} = $eventcb if $eventcb; | |
34 | ||
35 | $mux->set_callback_object($self); | |
36 | ||
558f1644 | 37 | # make sure perl doesn't believe this is a circular reference as we |
b3ea07f7 DM |
38 | # delete mux in DESTROY |
39 | weaken($mux->{_object}); | |
40 | ||
30a3378a DM |
41 | return $self; |
42 | } | |
43 | ||
7a6c2150 DM |
44 | # Note: List of special QGA command. Those commands can close the connection |
45 | # without sending a response. | |
46 | ||
47 | my $qga_allow_close_cmds = { | |
48 | 'guest-shutdown' => 1, | |
49 | 'guest-suspend-ram' => 1, | |
50 | 'guest-suspend-disk' => 1, | |
51 | 'guest-suspend-hybrid' => 1, | |
52 | }; | |
53 | ||
54 | my $push_cmd_to_queue = sub { | |
55 | my ($self, $vmid, $cmd) = @_; | |
56 | ||
57 | my $execute = $cmd->{execute} || die "no command name specified"; | |
58 | ||
59 | my $qga = ($execute =~ /^guest\-+/) ? 1 : 0; | |
60 | ||
61 | my $sname = PVE::QemuServer::qmp_socket($vmid, $qga); | |
62 | ||
63 | $self->{queue_info}->{$sname} = { qga => $qga, vmid => $vmid, sname => $sname, cmds => [] } | |
64 | if !$self->{queue_info}->{$sname}; | |
65 | ||
66 | push @{$self->{queue_info}->{$sname}->{cmds}}, $cmd; | |
67 | ||
68 | return $self->{queue_info}->{$sname}; | |
69 | }; | |
70 | ||
558f1644 | 71 | # add a single command to the queue for later execution |
30a3378a DM |
72 | # with queue_execute() |
73 | sub queue_cmd { | |
74 | my ($self, $vmid, $callback, $execute, %params) = @_; | |
75 | ||
76 | my $cmd = {}; | |
77 | $cmd->{execute} = $execute; | |
78 | $cmd->{arguments} = \%params; | |
79 | $cmd->{callback} = $callback; | |
80 | ||
7a6c2150 DM |
81 | &$push_cmd_to_queue($self, $vmid, $cmd); |
82 | ||
83 | return undef; | |
30a3378a DM |
84 | } |
85 | ||
86 | # execute a single command | |
87 | sub cmd { | |
f0002f62 | 88 | my ($self, $vmid, $cmd, $timeout) = @_; |
30a3378a DM |
89 | |
90 | my $result; | |
91 | ||
92 | my $callback = sub { | |
93 | my ($vmid, $resp) = @_; | |
94 | $result = $resp->{'return'}; | |
95 | }; | |
96 | ||
7a6c2150 | 97 | die "no command specified" if !($cmd && $cmd->{execute}); |
f0002f62 | 98 | |
30a3378a DM |
99 | $cmd->{callback} = $callback; |
100 | $cmd->{arguments} = {} if !defined($cmd->{arguments}); | |
101 | ||
7a6c2150 | 102 | my $queue_info = &$push_cmd_to_queue($self, $vmid, $cmd); |
30a3378a | 103 | |
f0002f62 DM |
104 | if (!$timeout) { |
105 | # hack: monitor sometime blocks | |
106 | if ($cmd->{execute} eq 'query-migrate') { | |
107 | $timeout = 60*60; # 1 hour | |
108 | } elsif ($cmd->{execute} =~ m/^(eject|change)/) { | |
109 | $timeout = 60; # note: cdrom mount command is slow | |
9dcf4909 DM |
110 | } elsif ($cmd->{execute} eq 'savevm-start' || |
111 | $cmd->{execute} eq 'savevm-end' || | |
558f1644 | 112 | $cmd->{execute} eq 'query-backup' || |
4fca0153 | 113 | $cmd->{execute} eq 'query-block-jobs' || |
fc97ae27 | 114 | $cmd->{execute} eq 'backup-cancel' || |
9dcf4909 | 115 | $cmd->{execute} eq 'query-savevm' || |
1c0c1c17 WL |
116 | $cmd->{execute} eq 'delete-drive-snapshot' || |
117 | $cmd->{execute} eq 'guest-shutdown' || | |
9d689077 DM |
118 | $cmd->{execute} eq 'snapshot-drive' ) { |
119 | $timeout = 10*60; # 10 mins ? | |
14db5366 DM |
120 | } else { |
121 | $timeout = 3; # default | |
f0002f62 DM |
122 | } |
123 | } | |
124 | ||
c8125172 | 125 | $self->queue_execute($timeout, 2); |
7a6c2150 DM |
126 | |
127 | die "VM $vmid qmp command '$cmd->{execute}' failed - $queue_info->{error}" | |
128 | if defined($queue_info->{error}); | |
30a3378a DM |
129 | |
130 | return $result; | |
131 | }; | |
132 | ||
133 | my $cmdid_seq = 0; | |
c5a07de5 | 134 | my $cmdid_seq_qga = 0; |
7a6c2150 | 135 | |
30a3378a | 136 | my $next_cmdid = sub { |
c5a07de5 WL |
137 | my ($qga) = @_; |
138 | ||
139 | if($qga){ | |
140 | $cmdid_seq_qga++; | |
141 | return "$$"."0".$cmdid_seq_qga; | |
142 | } else { | |
143 | $cmdid_seq++; | |
144 | return "$$:$cmdid_seq"; | |
145 | } | |
30a3378a DM |
146 | }; |
147 | ||
7a6c2150 DM |
148 | my $lookup_queue_info = sub { |
149 | my ($self, $fh, $noerr) = @_; | |
558f1644 | 150 | |
7a6c2150 DM |
151 | my $queue_info = $self->{queue_lookup}->{$fh}; |
152 | if (!$queue_info) { | |
153 | warn "internal error - unable to lookup queue info" if !$noerr; | |
154 | return undef; | |
155 | } | |
156 | return $queue_info; | |
157 | }; | |
558f1644 | 158 | |
7a6c2150 DM |
159 | my $close_connection = sub { |
160 | my ($self, $queue_info) = @_; | |
30a3378a | 161 | |
7a6c2150 DM |
162 | if (my $fh = delete $queue_info->{fh}) { |
163 | delete $self->{queue_lookup}->{$fh}; | |
c8125172 | 164 | $self->{mux}->close($fh); |
7a6c2150 | 165 | } |
30a3378a DM |
166 | }; |
167 | ||
168 | my $open_connection = sub { | |
7a6c2150 DM |
169 | my ($self, $queue_info, $timeout) = @_; |
170 | ||
171 | die "duplicate call to open" if defined($queue_info->{fh}); | |
172 | ||
173 | my $vmid = $queue_info->{vmid}; | |
174 | my $qga = $queue_info->{qga}; | |
30a3378a | 175 | |
c5a07de5 | 176 | my $sname = PVE::QemuServer::qmp_socket($vmid, $qga); |
30a3378a | 177 | |
6d042176 DM |
178 | $timeout = 1 if !$timeout; |
179 | ||
2ae10d4e DM |
180 | my $fh; |
181 | my $starttime = [gettimeofday]; | |
182 | my $count = 0; | |
7a6c2150 DM |
183 | |
184 | my $sotype = $qga ? 'qga' : 'qmp'; | |
185 | ||
2ae10d4e DM |
186 | for (;;) { |
187 | $count++; | |
188 | $fh = IO::Socket::UNIX->new(Peer => $sname, Blocking => 0, Timeout => 1); | |
189 | last if $fh; | |
190 | if ($! != EINTR && $! != EAGAIN) { | |
7a6c2150 | 191 | die "unable to connect to VM $vmid $sotype socket - $!\n"; |
2ae10d4e DM |
192 | } |
193 | my $elapsed = tv_interval($starttime, [gettimeofday]); | |
6d042176 | 194 | if ($elapsed >= $timeout) { |
7a6c2150 | 195 | die "unable to connect to VM $vmid $sotype socket - timeout after $count retries\n"; |
2ae10d4e DM |
196 | } |
197 | usleep(100000); | |
198 | } | |
30a3378a | 199 | |
7a6c2150 DM |
200 | $queue_info->{fh} = $fh; |
201 | ||
202 | $self->{queue_lookup}->{$fh} = $queue_info; | |
203 | ||
30a3378a | 204 | $self->{mux}->add($fh); |
7a6c2150 | 205 | $self->{mux}->set_timeout($fh, $timeout); |
558f1644 | 206 | |
30a3378a DM |
207 | return $fh; |
208 | }; | |
209 | ||
210 | my $check_queue = sub { | |
211 | my ($self) = @_; | |
212 | ||
213 | my $running = 0; | |
558f1644 | 214 | |
7a6c2150 DM |
215 | foreach my $sname (keys %{$self->{queue_info}}) { |
216 | my $queue_info = $self->{queue_info}->{$sname}; | |
217 | my $fh = $queue_info->{fh}; | |
30a3378a DM |
218 | next if !$fh; |
219 | ||
7a6c2150 DM |
220 | my $qga = $queue_info->{qga}; |
221 | ||
222 | if ($queue_info->{error}) { | |
223 | &$close_connection($self, $queue_info); | |
30a3378a DM |
224 | next; |
225 | } | |
226 | ||
7a6c2150 | 227 | if ($queue_info->{current}) { # command running, waiting for response |
30a3378a DM |
228 | $running++; |
229 | next; | |
230 | } | |
231 | ||
7a6c2150 DM |
232 | if (!scalar(@{$queue_info->{cmds}})) { # no more commands |
233 | &$close_connection($self, $queue_info); | |
30a3378a DM |
234 | next; |
235 | } | |
236 | ||
237 | eval { | |
238 | ||
7a6c2150 DM |
239 | my $cmd = $queue_info->{current} = shift @{$queue_info->{cmds}}; |
240 | $cmd->{id} = &$next_cmdid($qga); | |
30a3378a | 241 | |
558f1644 | 242 | my $fd = -1; |
a0e7a5d0 | 243 | if ($cmd->{execute} eq 'add-fd' || $cmd->{execute} eq 'getfd') { |
558f1644 DM |
244 | $fd = $cmd->{arguments}->{fd}; |
245 | delete $cmd->{arguments}->{fd}; | |
246 | } | |
247 | ||
7a6c2150 | 248 | my $qmpcmd; |
a45a14fc | 249 | |
7a6c2150 | 250 | if ($qga) { |
a45a14fc | 251 | |
7a6c2150 DM |
252 | $qmpcmd = to_json({ execute => 'guest-sync', arguments => { id => int($cmd->{id})}}) . |
253 | to_json({ execute => $cmd->{execute}, arguments => $cmd->{arguments}}); | |
a45a14fc | 254 | |
7a6c2150 | 255 | } else { |
a45a14fc AD |
256 | |
257 | $qmpcmd = to_json({ | |
258 | execute => $cmd->{execute}, | |
259 | arguments => $cmd->{arguments}, | |
260 | id => $cmd->{id}}); | |
261 | } | |
30a3378a | 262 | |
558f1644 DM |
263 | if ($fd >= 0) { |
264 | my $ret = PVE::IPCC::sendfd(fileno($fh), $fd, $qmpcmd); | |
265 | die "sendfd failed" if $ret < 0; | |
266 | } else { | |
267 | $self->{mux}->write($fh, $qmpcmd); | |
268 | } | |
30a3378a DM |
269 | }; |
270 | if (my $err = $@) { | |
7a6c2150 | 271 | $queue_info->{error} = $err; |
30a3378a DM |
272 | } else { |
273 | $running++; | |
274 | } | |
275 | } | |
276 | ||
277 | $self->{mux}->endloop() if !$running; | |
278 | ||
279 | return $running; | |
280 | }; | |
281 | ||
282 | # execute all queued command | |
c8125172 | 283 | |
30a3378a | 284 | sub queue_execute { |
c8125172 | 285 | my ($self, $timeout, $noerr) = @_; |
30a3378a DM |
286 | |
287 | $timeout = 3 if !$timeout; | |
288 | ||
30a3378a | 289 | # open all necessary connections |
7a6c2150 DM |
290 | foreach my $sname (keys %{$self->{queue_info}}) { |
291 | my $queue_info = $self->{queue_info}->{$sname}; | |
292 | next if !scalar(@{$queue_info->{cmds}}); # no commands | |
293 | ||
294 | $queue_info->{error} = undef; | |
295 | $queue_info->{current} = undef; | |
c5a07de5 WL |
296 | |
297 | eval { | |
7a6c2150 | 298 | &$open_connection($self, $queue_info, $timeout); |
c6fb6a69 | 299 | |
7a6c2150 DM |
300 | if (!$queue_info->{qga}) { |
301 | my $cap_cmd = { execute => 'qmp_capabilities', arguments => {} }; | |
302 | unshift @{$queue_info->{cmds}}, $cap_cmd; | |
c6fb6a69 | 303 | } |
30a3378a DM |
304 | }; |
305 | if (my $err = $@) { | |
7a6c2150 | 306 | $queue_info->{error} = $err; |
30a3378a DM |
307 | } |
308 | } | |
309 | ||
310 | my $running; | |
311 | ||
312 | for (;;) { | |
313 | ||
314 | $running = &$check_queue($self); | |
315 | ||
316 | last if !$running; | |
317 | ||
318 | $self->{mux}->loop; | |
319 | } | |
320 | ||
321 | # make sure we close everything | |
c8125172 | 322 | my $errors = ''; |
7a6c2150 | 323 | foreach my $sname (keys %{$self->{queue_info}}) { |
c8125172 DM |
324 | my $queue_info = $self->{queue_info}->{$sname}; |
325 | &$close_connection($self, $queue_info); | |
326 | if ($queue_info->{error}) { | |
327 | if ($noerr) { | |
328 | warn $queue_info->{error} if $noerr < 2; | |
329 | } else { | |
330 | $errors .= $queue_info->{error} | |
331 | } | |
332 | } | |
30a3378a DM |
333 | } |
334 | ||
7a6c2150 | 335 | $self->{queue_info} = $self->{queue_lookup} = {}; |
c8125172 DM |
336 | |
337 | die $errors if $errors; | |
30a3378a DM |
338 | } |
339 | ||
b006e70b DM |
340 | sub mux_close { |
341 | my ($self, $mux, $fh) = @_; | |
342 | ||
7a6c2150 DM |
343 | my $queue_info = &$lookup_queue_info($self, $fh, 1); |
344 | return if !$queue_info; | |
b006e70b | 345 | |
7a6c2150 DM |
346 | $queue_info->{error} = "client closed connection\n" |
347 | if !$queue_info->{error}; | |
b006e70b DM |
348 | } |
349 | ||
c8125172 | 350 | # mux_input is called when input is available on one of the descriptors. |
30a3378a DM |
351 | sub mux_input { |
352 | my ($self, $mux, $fh, $input) = @_; | |
353 | ||
7a6c2150 DM |
354 | my $queue_info = &$lookup_queue_info($self, $fh); |
355 | return if !$queue_info; | |
356 | ||
357 | my $sname = $queue_info->{sname}; | |
358 | my $vmid = $queue_info->{vmid}; | |
359 | my $qga = $queue_info->{qga}; | |
c5a07de5 | 360 | |
7a6c2150 DM |
361 | my $curcmd = $queue_info->{current}; |
362 | die "unable to lookup current command for VM $vmid ($sname)\n" if !$curcmd; | |
363 | ||
bcfbc40b AD |
364 | my $raw; |
365 | ||
7a6c2150 | 366 | if ($qga) { |
edb52f14 | 367 | return if $$input !~ s/^([^\n]+}\r?\n[^\n]+})\r?\n(.*)$/$2/so; |
bcfbc40b | 368 | $raw = $1; |
c5a07de5 | 369 | } else { |
edb52f14 | 370 | return if $$input !~ s/^(.*})\r?\n(.*)$/$2/so; |
bcfbc40b AD |
371 | $raw = $1; |
372 | } | |
30a3378a | 373 | |
30a3378a DM |
374 | eval { |
375 | my @jsons = split("\n", $raw); | |
376 | ||
7a6c2150 | 377 | if ($qga) { |
bcfbc40b AD |
378 | |
379 | die "response is not complete" if @jsons != 2 ; | |
380 | ||
381 | my $obj = from_json($jsons[0]); | |
c8125172 | 382 | |
7a6c2150 | 383 | my $cmdid = $obj->{'return'}; |
bcfbc40b | 384 | die "received responsed without command id\n" if !$cmdid; |
7a6c2150 | 385 | |
bcfbc40b AD |
386 | if ($curcmd->{id} ne $cmdid) { |
387 | die "got wrong command id '$cmdid' (expected $curcmd->{id})\n"; | |
388 | } | |
389 | ||
c8125172 DM |
390 | delete $queue_info->{current}; |
391 | ||
bcfbc40b AD |
392 | $obj = from_json($jsons[1]); |
393 | ||
394 | if (my $callback = $curcmd->{callback}) { | |
395 | &$callback($vmid, $obj); | |
396 | } | |
397 | ||
398 | return; | |
399 | } | |
400 | ||
30a3378a DM |
401 | foreach my $json (@jsons) { |
402 | my $obj = from_json($json); | |
403 | next if defined($obj->{QMP}); # skip monitor greeting | |
404 | ||
405 | if (exists($obj->{error}->{desc})) { | |
406 | my $desc = $obj->{error}->{desc}; | |
407 | chomp $desc; | |
408 | die "$desc\n" if $desc !~ m/Connection can not be completed immediately/; | |
409 | next; | |
410 | } | |
411 | ||
30a3378a DM |
412 | if (defined($obj->{event})) { |
413 | if (my $eventcb = $self->{eventcb}) { | |
414 | &$eventcb($obj); | |
415 | } | |
416 | next; | |
417 | } | |
418 | ||
419 | my $cmdid = $obj->{id}; | |
420 | die "received responsed without command id\n" if !$cmdid; | |
421 | ||
30a3378a DM |
422 | if ($curcmd->{id} ne $cmdid) { |
423 | die "got wrong command id '$cmdid' (expected $curcmd->{id})\n"; | |
424 | } | |
425 | ||
c8125172 DM |
426 | delete $queue_info->{current}; |
427 | ||
30a3378a DM |
428 | if (my $callback = $curcmd->{callback}) { |
429 | &$callback($vmid, $obj); | |
430 | } | |
431 | } | |
432 | }; | |
433 | if (my $err = $@) { | |
7a6c2150 | 434 | $queue_info->{error} = $err; |
30a3378a DM |
435 | } |
436 | ||
437 | &$check_queue($self); | |
438 | } | |
439 | ||
440 | # This gets called every second to update player info, etc... | |
441 | sub mux_timeout { | |
442 | my ($self, $mux, $fh) = @_; | |
443 | ||
7a6c2150 DM |
444 | if (my $queue_info = &$lookup_queue_info($self, $fh)) { |
445 | $queue_info->{error} = "got timeout\n"; | |
c8125172 | 446 | $self->{mux}->inbuffer($fh, ''); # clear to avoid warnings |
30a3378a DM |
447 | } |
448 | ||
449 | &$check_queue($self); | |
450 | } | |
451 | ||
1c0c1c17 WL |
452 | sub mux_eof { |
453 | my ($self, $mux, $fh, $input) = @_; | |
c8125172 | 454 | |
7a6c2150 DM |
455 | my $queue_info = &$lookup_queue_info($self, $fh); |
456 | return if !$queue_info; | |
1c0c1c17 | 457 | |
7a6c2150 DM |
458 | my $sname = $queue_info->{sname}; |
459 | my $vmid = $queue_info->{vmid}; | |
460 | my $qga = $queue_info->{qga}; | |
461 | ||
462 | my $curcmd = $queue_info->{current}; | |
463 | die "unable to lookup current command for VM $vmid ($sname)\n" if !$curcmd; | |
1c0c1c17 | 464 | |
7a6c2150 | 465 | if ($qga && $qga_allow_close_cmds->{$curcmd->{execute}}) { |
1c0c1c17 | 466 | |
edb52f14 | 467 | return if $$input !~ s/^([^\n]+})\r?\n(.*)$/$2/so; |
1c0c1c17 | 468 | |
c8125172 DM |
469 | my $raw = $1; |
470 | ||
471 | eval { | |
472 | my $obj = from_json($raw); | |
1c0c1c17 | 473 | |
c8125172 DM |
474 | my $cmdid = $obj->{'return'}; |
475 | die "received responsed without command id\n" if !$cmdid; | |
7a6c2150 | 476 | |
c8125172 | 477 | delete $queue_info->{current}; |
7a6c2150 | 478 | |
c8125172 DM |
479 | if (my $callback = $curcmd->{callback}) { |
480 | &$callback($vmid, undef); | |
481 | } | |
482 | }; | |
483 | if (my $err = $@) { | |
484 | $queue_info->{error} = $err; | |
485 | } | |
1c0c1c17 | 486 | |
7a6c2150 | 487 | &$close_connection($self, $queue_info); |
c8125172 DM |
488 | |
489 | if (scalar(@{$queue_info->{cmds}}) && !$queue_info->{error}) { | |
490 | $queue_info->{error} = "Got EOF but command queue is not empty.\n"; | |
491 | } | |
1c0c1c17 | 492 | } |
1c0c1c17 WL |
493 | } |
494 | ||
26f11676 | 495 | 1; |