# 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")