security/apps/gen_cert_header.py

   1 # This Source Code Form is subject to the terms of the Mozilla Public
   2 # License, v. 2.0. If a copy of the MPL was not distributed with this
   3 # file, You can obtain one at http://mozilla.org/MPL/2.0/.
   4
   5 import sys
   6 import binascii
   7
   8 def file_byte_generator(filename, block_size = 512):
   9   with open(filename, "rb") as f:
  10     while True:
  11       block = f.read(block_size)
  12       if block:
  13         for byte in block:
  14           yield byte
  15       else:
  16         break
  17
  18 def create_header(array_name, in_filename):
  19   hexified = ["0x" + binascii.hexlify(byte) for byte in file_byte_generator(in_filename)]
  20   print "const uint8_t " + array_name + "[] = {"
  21   print ", ".join(hexified)
  22   print "};"
  23   return 0
  24
  25 def create_empty_header(array_name):
  26   # mfbt/ArrayUtils.h will not be able to pick up the
  27   # correct specialization for ArrayLength(const array[0])
  28   # so add a value of 0 which will fail cert verification
  29   # just the same as an empty array
  30   print "const uint8_t " + array_name + "[] = { 0x0 };"
  31   return 0
  32
  33 if __name__ == '__main__':
  34   if len(sys.argv) < 2:
  35     print 'ERROR: usage: gen_cert_header.py array_name in_filename'
  36     sys.exit(1);
  37   if len(sys.argv) == 2:
  38     sys.exit(create_empty_header(sys.argv[1]))
  39   sys.exit(create_header(sys.argv[1], sys.argv[2]))