• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1#!/usr/bin/env python3
2# coding=utf-8
3# Copyright (c) 2020, Google Inc.
4#
5# Permission to use, copy, modify, and/or distribute this software for any
6# purpose with or without fee is hereby granted, provided that the above
7# copyright notice and this permission notice appear in all copies.
8#
9# THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
10# WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
11# MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
12# SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
13# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION
14# OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
15# CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
16
17"""This script translates JSON test vectors to BoringSSL's "FileTest" format.
18
19Usage: translate_test_vectors.py TEST_VECTORS_JSON_FILE
20
21The TEST_VECTORS_JSON_FILE is expected to come from the JSON copy of
22RFC 9180's test vectors, linked from its [TestVectors] citation.
23The output is written to "hpke_test_vectors.txt".
24"""
25
26import collections
27import json
28import sys
29
30HPKE_MODE_BASE = 0
31HPKE_MODE_PSK = 1
32HPKE_MODE_AUTH = 2
33HPKE_DHKEM_X25519_SHA256 = 0x0020
34HPKE_HKDF_SHA256 = 0x0001
35HPKE_AEAD_EXPORT_ONLY = 0xffff
36
37
38def read_test_vectors_and_generate_code(json_file_in_path, test_file_out_path):
39  """Translates JSON test vectors into BoringSSL's FileTest language.
40
41    Args:
42      json_file_in_path: Path to the JSON test vectors file.
43      test_file_out_path: Path to output file.
44  """
45
46  # Load the JSON file into |test_vecs|.
47  with open(json_file_in_path) as file_in:
48    test_vecs = json.load(file_in)
49
50  lines = []
51  for test in test_vecs:
52    # Filter out test cases that we don't use.
53    if (test["mode"] not in (HPKE_MODE_BASE, HPKE_MODE_AUTH) or
54        test["kem_id"] != HPKE_DHKEM_X25519_SHA256 or
55        test["aead_id"] == HPKE_AEAD_EXPORT_ONLY or
56        test["kdf_id"] != HPKE_HKDF_SHA256):
57      continue
58
59    keys = ["mode", "kdf_id", "aead_id", "info", "skRm", "skEm", "pkRm", "pkEm"]
60
61    if test["mode"] == HPKE_MODE_AUTH:
62      keys.append("pkSm")
63      keys.append("skSm")
64
65    for key in keys:
66      lines.append("{} = {}".format(key, str(test[key])))
67
68    for i, enc in enumerate(test["encryptions"]):
69      lines.append("# encryptions[{}]".format(i))
70      for key in ("aad", "ct", "pt"):
71        lines.append("{} = {}".format(key, str(enc[key])))
72
73    for i, exp in enumerate(test["exports"]):
74      lines.append("# exports[{}]".format(i))
75      for key in ("exporter_context", "L", "exported_value"):
76        lines.append("{} = {}".format(key, str(exp[key])))
77
78    lines.append("")
79
80  with open(test_file_out_path, "w") as file_out:
81    file_out.write("\n".join(lines))
82
83
84def main(argv):
85  if len(argv) != 2:
86    print(__doc__)
87    sys.exit(1)
88
89  read_test_vectors_and_generate_code(argv[1], "hpke_test_vectors.txt")
90
91
92if __name__ == "__main__":
93  main(sys.argv)
94