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 Murdochfrom . import context 30b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdochfrom . import testcase 31b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 32b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdochclass WorkPacket(object): 33b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch def __init__(self, peer=None, context=None, tests=None, binaries=None, 34b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch base_revision=None, patch=None, pubkey=None): 35b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch self.peer = peer 36b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch self.context = context 37b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch self.tests = tests 38b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch self.binaries = binaries 39b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch self.base_revision = base_revision 40b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch self.patch = patch 41b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch self.pubkey_fingerprint = pubkey 42b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 43b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch def Pack(self, binaries_dict): 44b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch """ 45b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch Creates a JSON serializable object containing the data of this 46b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch work packet. 47b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch """ 48b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch need_libv8 = False 49b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch binaries = [] 50b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch for shell in self.peer.shells: 51b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch prefetched_binary = binaries_dict[shell] 52b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch binaries.append({"name": shell, 53b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch "blob": prefetched_binary[0], 54b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch "sign": prefetched_binary[1]}) 55b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch if prefetched_binary[2]: 56b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch need_libv8 = True 57b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch if need_libv8: 58b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch libv8 = binaries_dict["libv8.so"] 59b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch binaries.append({"name": "libv8.so", 60b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch "blob": libv8[0], 61b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch "sign": libv8[1]}) 62b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch tests = [] 63b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch test_map = {} 64b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch for t in self.peer.tests: 65b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch test_map[t.id] = t 66b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch tests.append(t.PackTask()) 67b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch result = { 68b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch "binaries": binaries, 69b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch "pubkey": self.pubkey_fingerprint, 70b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch "context": self.context.Pack(), 71b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch "base_revision": self.base_revision, 72b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch "patch": self.patch, 73b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch "tests": tests 74b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch } 75b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch return result, test_map 76b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch 77b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch @staticmethod 78b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch def Unpack(packed): 79b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch """ 80b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch Creates a WorkPacket object from the given packed representation. 81b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch """ 82b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch binaries = packed["binaries"] 83b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch pubkey_fingerprint = packed["pubkey"] 84b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch ctx = context.Context.Unpack(packed["context"]) 85b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch base_revision = packed["base_revision"] 86b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch patch = packed["patch"] 87b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch tests = [ testcase.TestCase.UnpackTask(t) for t in packed["tests"] ] 88b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch return WorkPacket(context=ctx, tests=tests, binaries=binaries, 89b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch base_revision=base_revision, patch=patch, 90b8a8cc1952d61a2f3a2568848933943a543b5d3eBen Murdoch pubkey=pubkey_fingerprint) 91