1b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch# Copyright 2012 the V8 project authors. All rights reserved. 2b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch# Redistribution and use in source and binary forms, with or without 3b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch# modification, are permitted provided that the following conditions are 4b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch# met: 5b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch# 6b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch# * Redistributions of source code must retain the above copyright 7b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch# notice, this list of conditions and the following disclaimer. 8b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch# * Redistributions in binary form must reproduce the above 9b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch# copyright notice, this list of conditions and the following 10b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch# disclaimer in the documentation and/or other materials provided 11b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch# with the distribution. 12b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch# * Neither the name of Google Inc. nor the names of its 13b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch# contributors may be used to endorse or promote products derived 14b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch# from this software without specific prior written permission. 15b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch# 16b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 17b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 18b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 19b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 20b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 21b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 22b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 23b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 24b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 25b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 26b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 27b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 28b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 29b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdochimport socket 30b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdochimport SocketServer 31b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdochimport StringIO 32b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 33b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdochfrom . import compression 34b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdochfrom . import constants 35b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 36b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 37b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdochdef LocalQuery(query): 38b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) 39b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch code = sock.connect_ex(("localhost", constants.CLIENT_PORT)) 40b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch if code != 0: return None 41b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch compression.Send(query, sock) 42b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch compression.Send(constants.END_OF_STREAM, sock) 43b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch rec = compression.Receiver(sock) 44b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch data = None 45b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch while not rec.IsDone(): 46b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch data = rec.Current() 47b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch assert data[0] == query[0] 48b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch data = data[1] 49b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch rec.Advance() 50b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch sock.close() 51b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch return data 52b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 53b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 54b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdochclass LocalHandler(SocketServer.BaseRequestHandler): 55b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch def handle(self): 56b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch rec = compression.Receiver(self.request) 57b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch while not rec.IsDone(): 58b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch data = rec.Current() 59b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch action = data[0] 60b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 61b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch if action == constants.REQUEST_PEERS: 62b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch with self.server.daemon.peer_list_lock: 63b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch response = [ p.Pack() for p in self.server.daemon.peers 64b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch if p.trusting_me ] 65b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch compression.Send([action, response], self.request) 66b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 67b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch elif action == constants.UNRESPONSIVE_PEER: 68b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch self.server.daemon.DeletePeer(data[1]) 69b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 70b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch elif action == constants.REQUEST_PUBKEY_FINGERPRINT: 71b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch compression.Send([action, self.server.daemon.pubkey_fingerprint], 72b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch self.request) 73b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 74b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch elif action == constants.REQUEST_STATUS: 75b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch compression.Send([action, self._GetStatusMessage()], self.request) 76b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 77b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch elif action == constants.ADD_TRUSTED: 78b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch fingerprint = self.server.daemon.CopyToTrusted(data[1]) 79b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch compression.Send([action, fingerprint], self.request) 80b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 81b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch elif action == constants.INFORM_DURATION: 82b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch test_key = data[1] 83b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch test_duration = data[2] 84b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch arch = data[3] 85b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch mode = data[4] 86b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch self.server.daemon.AddPerfData(test_key, test_duration, arch, mode) 87b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 88b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch elif action == constants.UPDATE_PERF: 89b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch address = data[1] 90b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch perf = data[2] 91b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch self.server.daemon.UpdatePeerPerformance(data[1], data[2]) 92b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 93b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch rec.Advance() 94b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch compression.Send(constants.END_OF_STREAM, self.request) 95b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 96b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch def _GetStatusMessage(self): 97b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch sio = StringIO.StringIO() 98b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch sio.write("Peers:\n") 99b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch with self.server.daemon.peer_list_lock: 100b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch for p in self.server.daemon.peers: 101b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch sio.write("%s\n" % p) 102b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch sio.write("My own jobs: %d, relative performance: %.2f\n" % 103b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch (self.server.daemon.jobs, self.server.daemon.relative_perf)) 104b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch # Low-priority TODO: Return more information. Ideas: 105b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch # - currently running anything, 106b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch # - time since last job, 107b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch # - time since last repository fetch 108b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch # - number of workpackets/testcases handled since startup 109b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch # - slowest test(s) 110b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch result = sio.getvalue() 111b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch sio.close() 112b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch return result 113b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 114b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 115b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdochclass LocalSocketServer(SocketServer.ThreadingMixIn, SocketServer.TCPServer): 116b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch def __init__(self, daemon): 117b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch SocketServer.TCPServer.__init__(self, ("localhost", constants.CLIENT_PORT), 118b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch LocalHandler) 119b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch self.daemon = daemon 120