| Index: tools/sslroots/generate_sslroots.py
|
| diff --git a/tools/sslroots/generate_sslroots.py b/tools/sslroots/generate_sslroots.py
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..65751f1f1a204aa75e793e773c7317a6e79b984e
|
| --- /dev/null
|
| +++ b/tools/sslroots/generate_sslroots.py
|
| @@ -0,0 +1,190 @@
|
| +# -*- coding:utf-8 -*-
|
| +# Copyright (c) 2015 The WebRTC project authors. All Rights Reserved.
|
| +#
|
| +# Use of this source code is governed by a BSD-style license
|
| +# that can be found in the LICENSE file in the root of the source
|
| +# tree. An additional intellectual property rights grant can be found
|
| +# in the file PATENTS. All contributing project authors may
|
| +# be found in the AUTHORS file in the root of the source tree.
|
| +
|
| +
|
| +"""This is a tool to transform a crt file into a C/C++ header.
|
| +
|
| +Usage:
|
| +generate_sslroots.py cert_file.crt [--verbose | -v] [--full_cert | -f]
|
| +
|
| +Arguments:
|
| + -v Print output while running.
|
| + -f Add public key and certificate name. Default is to skip and reduce
|
| + generated file size.
|
| +"""
|
| +
|
| +import commands
|
| +from optparse import OptionParser
|
| +import os
|
| +import re
|
| +import string
|
| +
|
| +_GENERATED_FILE = 'sslroots.h'
|
| +_PREFIX = '__generated__'
|
| +_EXTENSION = '.crt'
|
| +_SUBJECT_NAME_ARRAY = 'subject_name'
|
| +_SUBJECT_NAME_VARIABLE = 'SubjectName'
|
| +_PUBLIC_KEY_ARRAY = 'public_key'
|
| +_PUBLIC_KEY_VARIABLE = 'PublicKey'
|
| +_CERTIFICATE_ARRAY = 'certificate'
|
| +_CERTIFICATE_VARIABLE = 'Certificate'
|
| +_CERTIFICATE_SIZE_VARIABLE = 'CertificateSize'
|
| +_INT_TYPE = 'size_t'
|
| +_CHAR_TYPE = 'const unsigned char*'
|
| +_VERBOSE = 'verbose'
|
| +
|
| +
|
| +def main():
|
| + """The main entrypoint."""
|
| + parser = OptionParser('usage %prog FILE')
|
| + parser.add_option('-v', '--verbose', dest='verbose', action='store_true')
|
| + parser.add_option('-f', '--full_cert', dest='full_cert', action='store_true')
|
| + options, args = parser.parse_args()
|
| + if len(args) < 1:
|
| + parser.error('No crt file specified.')
|
| + return
|
| + root_dir = _SplitCrt(args[0], options)
|
| + _GenCFiles(root_dir, options)
|
| + _Cleanup(root_dir)
|
| +
|
| +
|
| +def _SplitCrt(source_file, options):
|
| + sub_file_blocks = []
|
| + label_name = ''
|
| + root_dir = os.path.dirname(os.path.abspath(source_file)) + '/'
|
| + _PrintOutput(root_dir, options)
|
| + f = open(source_file)
|
| + for line in f:
|
| + if line.startswith('# Label: '):
|
| + sub_file_blocks.append(line)
|
| + label = re.search(r'\".*\"', line)
|
| + temp_label = label.group(0)
|
| + end = len(temp_label)-1
|
| + label_name = _SafeName(temp_label[1:end])
|
| + elif line.startswith('-----END CERTIFICATE-----'):
|
| + sub_file_blocks.append(line)
|
| + new_file_name = root_dir + _PREFIX + label_name + _EXTENSION
|
| + _PrintOutput('Generating: ' + new_file_name, options)
|
| + new_file = open(new_file_name, 'w')
|
| + for out_line in sub_file_blocks:
|
| + new_file.write(out_line)
|
| + new_file.close()
|
| + sub_file_blocks = []
|
| + else:
|
| + sub_file_blocks.append(line)
|
| + f.close()
|
| + return root_dir
|
| +
|
| +
|
| +def _GenCFiles(root_dir, options):
|
| + output_header_file = open(root_dir + _GENERATED_FILE, 'w')
|
| + output_header_file.write(_CreateOutputHeader())
|
| + if options.full_cert:
|
| + subject_name_list = _CreateArraySectionHeader(_SUBJECT_NAME_VARIABLE,
|
| + _CHAR_TYPE, options)
|
| + public_key_list = _CreateArraySectionHeader(_PUBLIC_KEY_VARIABLE,
|
| + _CHAR_TYPE, options)
|
| + certificate_list = _CreateArraySectionHeader(_CERTIFICATE_VARIABLE,
|
| + _CHAR_TYPE, options)
|
| + certificate_size_list = _CreateArraySectionHeader(_CERTIFICATE_SIZE_VARIABLE,
|
| + _INT_TYPE, options)
|
| +
|
| + for _, _, files in os.walk(root_dir):
|
| + for current_file in files:
|
| + if current_file.startswith(_PREFIX):
|
| + prefix_length = len(_PREFIX)
|
| + length = len(current_file) - len(_EXTENSION)
|
| + label = current_file[prefix_length:length]
|
| + filtered_output, cert_size = _CreateCertSection(root_dir, current_file,
|
| + label, options)
|
| + output_header_file.write(filtered_output + '\n\n\n')
|
| + if options.full_cert:
|
| + subject_name_list += _AddLabelToArray(label, _SUBJECT_NAME_ARRAY)
|
| + public_key_list += _AddLabelToArray(label, _PUBLIC_KEY_ARRAY)
|
| + certificate_list += _AddLabelToArray(label, _CERTIFICATE_ARRAY)
|
| + certificate_size_list += (' %s,\n') %(cert_size)
|
| +
|
| + if options.full_cert:
|
| + subject_name_list += _CreateArraySectionFooter()
|
| + output_header_file.write(subject_name_list)
|
| + public_key_list += _CreateArraySectionFooter()
|
| + output_header_file.write(public_key_list)
|
| + certificate_list += _CreateArraySectionFooter()
|
| + output_header_file.write(certificate_list)
|
| + certificate_size_list += _CreateArraySectionFooter()
|
| + output_header_file.write(certificate_size_list)
|
| + output_header_file.close()
|
| +
|
| +
|
| +def _Cleanup(root_dir):
|
| + for f in os.listdir(root_dir):
|
| + if f.startswith(_PREFIX):
|
| + os.remove(root_dir + f)
|
| +
|
| +
|
| +def _CreateCertSection(root_dir, source_file, label, options):
|
| + command = 'openssl x509 -in %s%s -noout -C' %(root_dir, source_file)
|
| + _PrintOutput(command, options)
|
| + output = commands.getstatusoutput(command)[1]
|
| + renamed_output = output.replace('unsigned char XXX_',
|
| + 'const unsigned char ' + label + '_')
|
| + filtered_output = ''
|
| + cert_block = '^const unsigned char.*?};$'
|
| + prog = re.compile(cert_block, re.IGNORECASE | re.MULTILINE | re.DOTALL)
|
| + if not options.full_cert:
|
| + filtered_output = prog.sub('', renamed_output, count=2)
|
| + else:
|
| + filtered_output = renamed_output
|
| +
|
| + cert_size_block = r'\d\d\d+'
|
| + prog2 = re.compile(cert_size_block, re.MULTILINE | re.VERBOSE)
|
| + result = prog2.findall(renamed_output)
|
| + cert_size = result[len(result) - 1]
|
| +
|
| + return filtered_output, cert_size
|
| +
|
| +
|
| +def _CreateOutputHeader():
|
| + output = ('// This file is the root certificates in C form that are needed to'
|
| + ' connect to\n// Google.\n\n'
|
| + '// It was generated with the following command line:\n'
|
| + '// > python tools/certs/generate_sslroots.py'
|
| + '\n// https://pki.google.com/roots.pem\n\n')
|
| + return output
|
| +
|
| +
|
| +def _CreateArraySectionHeader(type_name, type_type, options):
|
| + output = ('const %s kSSLCert%sList[] = {\n') %(type_type, type_name)
|
| + _PrintOutput(output, options)
|
| + return output
|
| +
|
| +
|
| +def _AddLabelToArray(label, type_name):
|
| + return ' %s_%s,\n' %(label, type_name)
|
| +
|
| +
|
| +def _CreateArraySectionFooter():
|
| + return '};\n\n'
|
| +
|
| +
|
| +def _SafeName(original_file_name):
|
| + bad_chars = ' -./\\()áéíőú'
|
| + replacement_chars = ''
|
| + for _ in bad_chars:
|
| + replacement_chars += '_'
|
| + translation_table = string.maketrans(bad_chars, replacement_chars)
|
| + return original_file_name.translate(translation_table)
|
| +
|
| +
|
| +def _PrintOutput(output, options):
|
| + if options.verbose:
|
| + print output
|
| +
|
| +if __name__ == '__main__':
|
| + main()
|
|
|