summaryrefslogtreecommitdiffstats
path: root/bitbake/lib/bb/ui/uievent.py
diff options
context:
space:
mode:
Diffstat (limited to 'bitbake/lib/bb/ui/uievent.py')
-rw-r--r--bitbake/lib/bb/ui/uievent.py150
1 files changed, 150 insertions, 0 deletions
diff --git a/bitbake/lib/bb/ui/uievent.py b/bitbake/lib/bb/ui/uievent.py
new file mode 100644
index 0000000000..c6b100c840
--- /dev/null
+++ b/bitbake/lib/bb/ui/uievent.py
@@ -0,0 +1,150 @@
1# ex:ts=4:sw=4:sts=4:et
2# -*- tab-width: 4; c-basic-offset: 4; indent-tabs-mode: nil -*-
3#
4# Copyright (C) 2006 - 2007 Michael 'Mickey' Lauer
5# Copyright (C) 2006 - 2007 Richard Purdie
6#
7# This program is free software; you can redistribute it and/or modify
8# it under the terms of the GNU General Public License version 2 as
9# published by the Free Software Foundation.
10#
11# This program is distributed in the hope that it will be useful,
12# but WITHOUT ANY WARRANTY; without even the implied warranty of
13# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14# GNU General Public License for more details.
15#
16# You should have received a copy of the GNU General Public License along
17# with this program; if not, write to the Free Software Foundation, Inc.,
18# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
19
20
21"""
22Use this class to fork off a thread to recieve event callbacks from the bitbake
23server and queue them for the UI to process. This process must be used to avoid
24client/server deadlocks.
25"""
26
27import socket, threading, pickle
28from SimpleXMLRPCServer import SimpleXMLRPCServer, SimpleXMLRPCRequestHandler
29
30class BBUIEventQueue:
31 def __init__(self, BBServer, clientinfo=("localhost, 0")):
32
33 self.eventQueue = []
34 self.eventQueueLock = threading.Lock()
35 self.eventQueueNotify = threading.Event()
36
37 self.BBServer = BBServer
38 self.clientinfo = clientinfo
39
40 server = UIXMLRPCServer(self.clientinfo)
41 self.host, self.port = server.socket.getsockname()
42
43 server.register_function( self.system_quit, "event.quit" )
44 server.register_function( self.send_event, "event.sendpickle" )
45 server.socket.settimeout(1)
46
47 self.EventHandler = None
48 count_tries = 0
49
50 # the event handler registration may fail here due to cooker being in invalid state
51 # this is a transient situation, and we should retry a couple of times before
52 # giving up
53
54 while self.EventHandler == None and count_tries < 5:
55 self.EventHandle = self.BBServer.registerEventHandler(self.host, self.port)
56
57 if (self.EventHandle != None):
58 break
59
60 bb.warn("Could not register UI event handler %s:%d, retry" % (self.host, self.port))
61 count_tries += 1
62 import time
63 time.sleep(1)
64
65
66 if self.EventHandle == None:
67 raise Exception("Could not register UI event handler")
68
69 self.server = server
70
71 self.t = threading.Thread()
72 self.t.setDaemon(True)
73 self.t.run = self.startCallbackHandler
74 self.t.start()
75
76 def getEvent(self):
77
78 self.eventQueueLock.acquire()
79
80 if len(self.eventQueue) == 0:
81 self.eventQueueLock.release()
82 return None
83
84 item = self.eventQueue.pop(0)
85
86 if len(self.eventQueue) == 0:
87 self.eventQueueNotify.clear()
88
89 self.eventQueueLock.release()
90 return item
91
92 def waitEvent(self, delay):
93 self.eventQueueNotify.wait(delay)
94 return self.getEvent()
95
96 def queue_event(self, event):
97 self.eventQueueLock.acquire()
98 self.eventQueue.append(event)
99 self.eventQueueNotify.set()
100 self.eventQueueLock.release()
101
102 def send_event(self, event):
103 self.queue_event(pickle.loads(event))
104
105 def startCallbackHandler(self):
106
107 self.server.timeout = 1
108 while not self.server.quit:
109 self.server.handle_request()
110 self.server.server_close()
111
112 def system_quit( self ):
113 """
114 Shut down the callback thread
115 """
116 try:
117 self.BBServer.unregisterEventHandler(self.EventHandle)
118 except:
119 pass
120 self.server.quit = True
121
122class UIXMLRPCServer (SimpleXMLRPCServer):
123
124 def __init__( self, interface ):
125 self.quit = False
126 SimpleXMLRPCServer.__init__( self,
127 interface,
128 requestHandler=SimpleXMLRPCRequestHandler,
129 logRequests=False, allow_none=True)
130
131 def get_request(self):
132 while not self.quit:
133 try:
134 sock, addr = self.socket.accept()
135 sock.settimeout(1)
136 return (sock, addr)
137 except socket.timeout:
138 pass
139 return (None, None)
140
141 def close_request(self, request):
142 if request is None:
143 return
144 SimpleXMLRPCServer.close_request(self, request)
145
146 def process_request(self, request, client_address):
147 if request is None:
148 return
149 SimpleXMLRPCServer.process_request(self, request, client_address)
150