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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
#!/usr/bin/env python
"""
Various helpers for spice2, they all get a qmp port and connect to it,
and do various helpful things.
"""
import qmp
import os
import atexit
import time
import copy
def kill_proc(pid):
os.kill(pid, 3)
time.sleep(0.01)
try:
os.kill(pid, 3)
print("sleeping 1 second before terminating %s" % pid)
time.sleep(1)
os.kill(pid, 15)
except OSError:
# already dead
pass
def fork_and_handle(qmp_address, handlers):
""" qemu doesn't like multiple clients to it's qmp handler, so just do with
one """
print("connecting to %s" % repr(qmp_address))
pid = os.fork()
if pid != 0:
# parent
atexit.register(kill_proc, pid)
return pid
q = qmp.QEMUMonitorProtocol(qmp_address)
q.connect()
while True:
for e in q.get_events():
for handler in handlers:
handler(q, e)
q.clear_events()
time.sleep(0.1)
def reset_on_shutdown():
"""
{u'seconds': 1316615017, u'microseconds': 145684}: "{u'event': u'SHUTDOWN'}"
{u'seconds': 1316615017, u'microseconds': 157869}: "{u'event': u'STOP'}"
"""
class ResetOnShutdownHandler(object):
def __init__(self):
self.next = self.wait_for_shutdown
def handler(self, q, event):
self.next(q, event)
def wait_for_shutdown(self, q, event):
if event['event'] == 'SHUTDOWN':
self.next = self.wait_for_stop
def wait_for_stop(self, q, event):
if event['event'] == 'STOP':
print("RESET_ON_SHUTDOWN: resetting and continuing machine")
q.cmd('system_reset')
q.cmd('cont')
return ResetOnShutdownHandler().handler
def print_events():
def handler(q, orig_event, d={}):
# we will mangle the event, so make a copy of it
event = copy.deepcopy(orig_event)
timestamp_sec, timestamp_usec = event['timestamp']['seconds'], event['timestamp']['microseconds']
if not 'base' in d:
d['base'] = (timestamp_sec, timestamp_usec)
event_type = event['event']
del event['timestamp']
del event['event']
data = {}
if 'data' in event:
data = event['data']
del event['data']
print("%4d.%04d: %s%s" % (timestamp_sec - d['base'][0], timestamp_usec / 1000, event_type,
' %r' % event if len(event) > 0 else ''))
for k, v in data.items():
print(" %12s: %r" % (k, v))
return handler
def test_print_events(qmp_address):
print_events(qmp_address)
try:
while True:
time.sleep(10)
except:
pass
if __name__ == '__main__':
import sys
test_print_events(sys.argv[1])
|