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 Murdoch 32b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdochfrom . import compression 33b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdochfrom . import constants 34b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 35b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 36b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdochdef _StatusQuery(peer, query): 37b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) 38b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch code = sock.connect_ex((peer.address, constants.STATUS_PORT)) 39b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch if code != 0: 40b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch # TODO(jkummerow): disconnect (after 3 failures?) 41b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch return 42b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch compression.Send(query, sock) 43b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch compression.Send(constants.END_OF_STREAM, sock) 44b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch rec = compression.Receiver(sock) 45b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch data = None 46b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch while not rec.IsDone(): 47b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch data = rec.Current() 48b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch assert data[0] == query[0] 49b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch data = data[1] 50b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch rec.Advance() 51b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch sock.close() 52b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch return data 53b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 54b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 55b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdochdef RequestTrustedPubkeys(peer, server): 56b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch pubkey_list = _StatusQuery(peer, [constants.LIST_TRUSTED_PUBKEYS]) 57b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch for pubkey in pubkey_list: 58b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch if server.IsTrusted(pubkey): continue 59b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch result = _StatusQuery(peer, [constants.GET_SIGNED_PUBKEY, pubkey]) 60b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch server.AcceptNewTrusted(result) 61b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 62b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 63b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdochdef NotifyNewTrusted(peer, data): 64b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch _StatusQuery(peer, [constants.NOTIFY_NEW_TRUSTED] + data) 65b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 66b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 67b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdochdef ITrustYouNow(peer): 68b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch _StatusQuery(peer, [constants.TRUST_YOU_NOW]) 69b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 70b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 71b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdochdef TryTransitiveTrust(peer, pubkey, server): 72b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch if _StatusQuery(peer, [constants.DO_YOU_TRUST, pubkey]): 73b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch result = _StatusQuery(peer, [constants.GET_SIGNED_PUBKEY, pubkey]) 74b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch server.AcceptNewTrusted(result) 75b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 76b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 77b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdochclass StatusHandler(SocketServer.BaseRequestHandler): 78b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch def handle(self): 79b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch rec = compression.Receiver(self.request) 80b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch while not rec.IsDone(): 81b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch data = rec.Current() 82b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch action = data[0] 83b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 84b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch if action == constants.LIST_TRUSTED_PUBKEYS: 85b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch response = self.server.daemon.ListTrusted() 86b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch compression.Send([action, response], self.request) 87b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 88b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch elif action == constants.GET_SIGNED_PUBKEY: 89b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch response = self.server.daemon.SignTrusted(data[1]) 90b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch compression.Send([action, response], self.request) 91b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 92b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch elif action == constants.NOTIFY_NEW_TRUSTED: 93b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch self.server.daemon.AcceptNewTrusted(data[1:]) 94b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch pass # No response. 95b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 96b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch elif action == constants.TRUST_YOU_NOW: 97b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch self.server.daemon.MarkPeerAsTrusting(self.client_address[0]) 98b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch pass # No response. 99b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 100b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch elif action == constants.DO_YOU_TRUST: 101b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch response = self.server.daemon.IsTrusted(data[1]) 102b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch compression.Send([action, response], self.request) 103b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 104b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch rec.Advance() 105b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch compression.Send(constants.END_OF_STREAM, self.request) 106b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 107b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 108b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdochclass StatusSocketServer(SocketServer.ThreadingMixIn, SocketServer.TCPServer): 109b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch def __init__(self, daemon): 110b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch address = (daemon.ip, constants.STATUS_PORT) 111b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch SocketServer.TCPServer.__init__(self, address, StatusHandler) 112b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch self.daemon = daemon 113