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() |