summaryrefslogtreecommitdiff
path: root/utils/gen-ipa-pub-key.py
blob: a4a1f7b79337d6a2ad93dc113a648fe444445e19 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
#!/usr/bin/env python3
# SPDX-License-Identifier: GPL-2.0-or-later
# Copyright (C) 2020, Google Inc.
#
# Author: Laurent Pinchart <laurent.pinchart@ideasonboard.com>
#
# ipa-gen-key.py - Generate the IPA module signing public key

import string
import subprocess
import sys


def main(argv):
    if len(argv) != 4:
        print('Usage: %s priv-key template output' % argv[0])
        return 1

    priv_key = argv[1]
    template = argv[2]
    output = argv[3]

    try:
        ret = subprocess.run(['openssl', 'rsa', '-pubout', '-in', priv_key,
                              '-outform', 'DER'],
                             stdout=subprocess.PIPE)
    except FileNotFoundError:
        print('Please install openssl to sign IPA modules')
        return 1

    ipa_key = ['0x%02x' % c for c in ret.stdout]
    ipa_key = [', '.join(ipa_key[bound:bound + 8]) for bound in range(0, len(ipa_key), 8)]
    ipa_key = ',\n\t'.join(ipa_key)
    data = {'ipa_key': ipa_key}

    template = open(template, 'rb').read()
    template = template.decode('utf-8')
    template = string.Template(template)

    f = open(output, 'wb')
    f.write(template.substitute(data).encode('utf-8'))
    f.close()

    return 0


if __name__ == '__main__':
    sys.exit(main(sys.argv))
* number of buffers. */ int testSequential(V4L2BufferCache *cache, const std::vector<std::unique_ptr<FrameBuffer>> &buffers) { for (unsigned int i = 0; i < buffers.size() * 100; i++) { int nBuffer = i % buffers.size(); int index = cache->get(*buffers[nBuffer].get()); if (index != nBuffer) { std::cout << "Expected index " << nBuffer << " got " << index << std::endl; return TestFail; } cache->put(index); } return TestPass; } /* * Test that randomly putting buffers to the cache always results in a * valid index. */ int testRandom(V4L2BufferCache *cache, const std::vector<std::unique_ptr<FrameBuffer>> &buffers) { std::uniform_int_distribution<> dist(0, buffers.size() - 1); for (unsigned int i = 0; i < buffers.size() * 100; i++) { int nBuffer = dist(generator_); int index = cache->get(*buffers[nBuffer].get()); if (index < 0) { std::cout << "Failed lookup from cache" << std::endl; return TestFail; } cache->put(index); } return TestPass; } /* * Test that using a buffer more frequently keeps it hot in the cache at * all times. */ int testHot(V4L2BufferCache *cache, const std::vector<std::unique_ptr<FrameBuffer>> &buffers, unsigned int hotFrequency) { /* Run the random test on the cache to make it messy. */ if (testRandom(cache, buffers) != TestPass) return TestFail; std::uniform_int_distribution<> dist(0, buffers.size() - 1); /* Pick a hot buffer at random and store its index. */ int hotBuffer = dist(generator_); int hotIndex = cache->get(*buffers[hotBuffer].get()); cache->put(hotIndex); /* * Queue hot buffer at the requested frequency and make sure * it stays hot. */ for (unsigned int i = 0; i < buffers.size() * 100; i++) { int nBuffer, index; bool hotQueue = i % hotFrequency == 0; if (hotQueue) nBuffer = hotBuffer; else nBuffer = dist(generator_); index = cache->get(*buffers[nBuffer].get()); if (index < 0) { std::cout << "Failed lookup from cache" << std::endl; return TestFail; } if (hotQueue && index != hotIndex) { std::cout << "Hot buffer got cold" << std::endl; return TestFail; } cache->put(index); } return TestPass; } int testIsEmpty(const std::vector<std::unique_ptr<FrameBuffer>> &buffers) { V4L2BufferCache cache(buffers.size()); if (!cache.isEmpty()) return TestFail; for (auto const &buffer : buffers) { FrameBuffer &b = *buffer.get(); cache.get(b); } if (cache.isEmpty()) return TestFail; unsigned int i; for (i = 0; i < buffers.size() - 1; i++) cache.put(i); if (cache.isEmpty()) return TestFail; cache.put(i); if (!cache.isEmpty()) return TestFail; return TestPass; } int init() override { std::random_device rd; unsigned int seed = rd(); std::cout << "Random seed is " << seed << std::endl; generator_.seed(seed); return TestPass; } int run() override { const unsigned int numBuffers = 8; StreamConfiguration cfg; cfg.pixelFormat = formats::YUYV; cfg.size = Size(600, 800); cfg.bufferCount = numBuffers; BufferSource source; int ret = source.allocate(cfg); if (ret != TestPass) return ret; const std::vector<std::unique_ptr<FrameBuffer>> &buffers = source.buffers(); if (buffers.size() != numBuffers) { std::cout << "Got " << buffers.size() << " buffers, expected " << numBuffers << std::endl; return TestFail; } /* * Test cache of same size as there are buffers, the cache is * created from a list of buffers and will be pre-populated. */ V4L2BufferCache cacheFromBuffers(buffers); if (testSequential(&cacheFromBuffers, buffers) != TestPass) return TestFail; if (testRandom(&cacheFromBuffers, buffers) != TestPass) return TestFail; if (testHot(&cacheFromBuffers, buffers, numBuffers) != TestPass) return TestFail; /* * Test cache of same size as there are buffers, the cache is * not pre-populated. */ V4L2BufferCache cacheFromNumbers(numBuffers); if (testSequential(&cacheFromNumbers, buffers) != TestPass) return TestFail; if (testRandom(&cacheFromNumbers, buffers) != TestPass) return TestFail; if (testHot(&cacheFromNumbers, buffers, numBuffers) != TestPass) return TestFail; /* * Test cache half the size of number of buffers used, the cache * is not pre-populated. */ V4L2BufferCache cacheHalf(numBuffers / 2); if (testRandom(&cacheHalf, buffers) != TestPass) return TestFail; if (testHot(&cacheHalf, buffers, numBuffers / 2) != TestPass) return TestFail; /* * Test that the isEmpty function reports the correct result at * various levels of cache fullness. */ if (testIsEmpty(buffers) != TestPass) return TestFail; return TestPass; } private: std::mt19937 generator_; }; } /* namespace */ TEST_REGISTER(BufferCacheTest)