diff options
Diffstat (limited to 'bitbake/lib/bb/ui/uievent.py')
-rw-r--r-- | bitbake/lib/bb/ui/uievent.py | 125 |
1 files changed, 125 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..36302f4da7 --- /dev/null +++ b/bitbake/lib/bb/ui/uievent.py | |||
@@ -0,0 +1,125 @@ | |||
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 | """ | ||
22 | Use this class to fork off a thread to recieve event callbacks from the bitbake | ||
23 | server and queue them for the UI to process. This process must be used to avoid | ||
24 | client/server deadlocks. | ||
25 | """ | ||
26 | |||
27 | import socket, threading, pickle | ||
28 | from SimpleXMLRPCServer import SimpleXMLRPCServer, SimpleXMLRPCRequestHandler | ||
29 | |||
30 | class BBUIEventQueue: | ||
31 | def __init__(self, BBServer): | ||
32 | |||
33 | self.eventQueue = [] | ||
34 | self.eventQueueLock = threading.Lock() | ||
35 | self.eventQueueNotify = threading.Event() | ||
36 | |||
37 | self.BBServer = BBServer | ||
38 | |||
39 | self.t = threading.Thread() | ||
40 | self.t.setDaemon(True) | ||
41 | self.t.run = self.startCallbackHandler | ||
42 | self.t.start() | ||
43 | |||
44 | def getEvent(self): | ||
45 | |||
46 | self.eventQueueLock.acquire() | ||
47 | |||
48 | if len(self.eventQueue) == 0: | ||
49 | self.eventQueueLock.release() | ||
50 | return None | ||
51 | |||
52 | item = self.eventQueue.pop(0) | ||
53 | |||
54 | if len(self.eventQueue) == 0: | ||
55 | self.eventQueueNotify.clear() | ||
56 | |||
57 | self.eventQueueLock.release() | ||
58 | return item | ||
59 | |||
60 | def waitEvent(self, delay): | ||
61 | self.eventQueueNotify.wait(delay) | ||
62 | return self.getEvent() | ||
63 | |||
64 | def queue_event(self, event): | ||
65 | self.eventQueueLock.acquire() | ||
66 | self.eventQueue.append(pickle.loads(event)) | ||
67 | self.eventQueueNotify.set() | ||
68 | self.eventQueueLock.release() | ||
69 | |||
70 | def startCallbackHandler(self): | ||
71 | |||
72 | server = UIXMLRPCServer() | ||
73 | self.host, self.port = server.socket.getsockname() | ||
74 | |||
75 | server.register_function( self.system_quit, "event.quit" ) | ||
76 | server.register_function( self.queue_event, "event.send" ) | ||
77 | server.socket.settimeout(1) | ||
78 | |||
79 | self.EventHandle = self.BBServer.registerEventHandler(self.host, self.port) | ||
80 | |||
81 | self.server = server | ||
82 | while not server.quit: | ||
83 | server.handle_request() | ||
84 | server.server_close() | ||
85 | |||
86 | def system_quit( self ): | ||
87 | """ | ||
88 | Shut down the callback thread | ||
89 | """ | ||
90 | try: | ||
91 | self.BBServer.unregisterEventHandler(self.EventHandle) | ||
92 | except: | ||
93 | pass | ||
94 | self.server.quit = True | ||
95 | |||
96 | class UIXMLRPCServer (SimpleXMLRPCServer): | ||
97 | |||
98 | def __init__( self, interface = ("localhost", 0) ): | ||
99 | self.quit = False | ||
100 | SimpleXMLRPCServer.__init__( self, | ||
101 | interface, | ||
102 | requestHandler=SimpleXMLRPCRequestHandler, | ||
103 | logRequests=False, allow_none=True) | ||
104 | |||
105 | def get_request(self): | ||
106 | while not self.quit: | ||
107 | try: | ||
108 | sock, addr = self.socket.accept() | ||
109 | sock.settimeout(1) | ||
110 | return (sock, addr) | ||
111 | except socket.timeout: | ||
112 | pass | ||
113 | return (None,None) | ||
114 | |||
115 | def close_request(self, request): | ||
116 | if request is None: | ||
117 | return | ||
118 | SimpleXMLRPCServer.close_request(self, request) | ||
119 | |||
120 | def process_request(self, request, client_address): | ||
121 | if request is None: | ||
122 | return | ||
123 | SimpleXMLRPCServer.process_request(self, request, client_address) | ||
124 | |||
125 | |||