summaryrefslogtreecommitdiff
path: root/chromium/net/data/ssl/scripts/generate-fuzzer-cert-include.py
blob: 0639b5c27dd444ac5a940d7a8e61a71cfa4dd971 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
# Copyright 2017 The Chromium Authors. All rights reserved.
# 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(ord(c)) for c in 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")