aboutsummaryrefslogtreecommitdiffstats
path: root/tools/xen/lib/xend/XendConsole.py
blob: 4420c388f237213ec4461900d9f0edbdf09348b4 (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
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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
# Copyright (C) 2004 Mike Wray <mike.wray@hp.com>

import socket
import xen.ext.xc
xc = xen.ext.xc.new()

import sxp
import XendRoot
xroot = XendRoot.instance()
import XendDB

import EventServer
eserver = EventServer.instance()

from xen.xend.server import SrvDaemon
daemon = SrvDaemon.instance()

class XendConsoleInfo:
    """Console information record.
    """

    def __init__(self, console, dom1, port1, dom2, port2, conn=None):
        self.console = console
        self.dom1  = int(dom1)
        self.port1 = int(port1)
        self.dom2  = int(dom2)
        self.port2 = int(port2)
        self.conn  = conn
        #self.id = "%d.%d-%d.%d" % (self.dom1, self.port1, self.dom2, self.port2)
        self.id = str(port1)

    def __str__(self):
        s = "console"
        s += " id=%s" % self.id
        s += " src=%d.%d" % (self.dom1, self.port1)
        s += " dst=%d.%d" % (self.dom2, self.port2)
        s += " port=%s" % self.console
        if self.conn:
            s += " conn=%s:%s" % (self.conn[0], self.conn[1])
        return s

    def sxpr(self):
        sxpr = ['console',
                ['id', self.id],
                ['src', self.dom1, self.port1],
                ['dst', self.dom2, self.port2],
                ['port', self.console],
                ]
        if self.conn:
            sxpr.append(['connected', self.conn[0], self.conn[1]])
        return sxpr

    def connection(self):
        return self.conn

    def update(self, consinfo):
        conn = sxp.child(consinfo, 'connected')
        if conn:
            self.conn = conn[1:]
        else:
            self.conn = None

    def uri(self):
        """Get the uri to use to connect to the console.
        This will be a telnet: uri.

        return uri
        """
        host = socket.gethostname()
        return "telnet://%s:%s" % (host, self.console)

class XendConsole:

    dbpath = "console"

    def  __init__(self):
        self.db = XendDB.XendDB(self.dbpath)
        self.console = {}
        self.console_db = self.db.fetchall("")
        if xroot.get_rebooted():
            print 'XendConsole> rebooted: removing all console info'
            self.rm_all()
        eserver.subscribe('xend.domain.died', self.onDomainDied)
        eserver.subscribe('xend.domain.destroy', self.onDomainDied)

    def rm_all(self):
        """Remove all console info. Used after reboot.
        """
        for (k, v) in self.console_db.items():
            self._delete_console(k)

    def refresh(self):
        consoles = daemon.consoles()
        cons = {}
        for consinfo in consoles:
            id = str(sxp.child_value(consinfo, 'id'))
            cons[id] = consinfo
            if id not in self.console:
                self._new_console(consinfo)
        for c in self.console.values():
            consinfo = cons.get(c.id)
            if consinfo:
                c.update(consinfo)
            else:
                self._delete_console(c.id)

    def onDomainDied(self, event, val):
        dom = int(val)
        #print 'XendConsole>onDomainDied', 'event', event, "dom=", dom
        for c in self.consoles():
            #print 'onDomainDied', "dom=", dom, "dom1=", c.dom1, "dom2=", c.dom2
            if (c.dom1 == dom) or (c.dom2 == dom):
                'XendConsole>onDomainDied', 'delete console dom=', dom
                ctrl = daemon.get_domain_console(dom)
                if ctrl:
                    ctrl.close()
                self._delete_console(c.id)

    def sync(self):
        self.db.saveall("", self.console_db)

    def sync_console(self, id):
        self.db.save(id, self.console_db[id])

    def _new_console(self, consinfo):
        # todo: xen needs a call to get current domain id.
        dom1 = 0
        port1 = sxp.child_value(consinfo, 'local_port')
        dom2 = sxp.child_value(consinfo, 'domain')
        port2 = sxp.child_value(consinfo, 'remote_port')
        console = sxp.child_value(consinfo, 'console_port')
        info = XendConsoleInfo(console, dom1, int(port1), int(dom2), int(port2))
        info.update(consinfo)
        self._add_console(info.id, info)
        return info

    def _add_console(self, id, info):
        self.console[id] = info
        self.console_db[id] = info.sxpr()
        self.sync_console(id)

    def _delete_console(self, id):
        if id in self.console:
            del self.console[id]
        if id in self.console_db:
            del self.console_db[id]
            self.db.delete(id)

    def console_ls(self):
        self.refresh()
        return self.console.keys()

    def consoles(self):
        self.refresh()
        return self.console.values()
    
    def console_create(self, dom):
        consinfo = daemon.console_create(dom)
        info = self._new_console(consinfo)
        return info
    
    def console_get(self, id):
        self.refresh()
        return self.console.get(id)

    def console_delete(self, id):
        self._delete_console(id)

    def console_disconnect(self, id):
        id = int(id)
        daemon.console_disconnect(id)

def instance():
    global inst
    try:
        inst
    except:
        inst = XendConsole()
    return inst