40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
# Copyright 2017 The Chromium Authors
|
|
# Use of this source code is governed by a BSD-style license that can be
|
|
# found in the LICENSE file.
|
|
|
|
"""Generate C/C++ source from certificate.
|
|
|
|
Usage:
|
|
generate-spdy-session-fuzzer-includes.py input_filename output_filename
|
|
|
|
Load the PEM block from `input_filename` certificate, perform base64 decoding
|
|
and hex encoding, and save it in `output_filename` so that it can be directly
|
|
included from C/C++ source as a char array.
|
|
"""
|
|
|
|
import base64
|
|
import re
|
|
import sys
|
|
import textwrap
|
|
|
|
input_filename = sys.argv[1]
|
|
output_filename = sys.argv[2]
|
|
|
|
# Load PEM block.
|
|
with open(input_filename, 'r') as f:
|
|
match = re.search(
|
|
r"-----BEGIN CERTIFICATE-----\n(.+)-----END CERTIFICATE-----\n", f.read(),
|
|
re.DOTALL)
|
|
text = match.group(1)
|
|
|
|
# Perform Base64 decoding.
|
|
data = base64.b64decode(text)
|
|
|
|
# Hex format data.
|
|
hex_encoded = ", ".join("0x{:02x}".format(c) for c in bytearray(data))
|
|
|
|
# Write into |output_filename| wrapped at 80 columns.
|
|
with open(output_filename, 'w') as f:
|
|
f.write(textwrap.fill(hex_encoded, 80))
|
|
f.write("\n")
|