security/manager/ssl/tests/unit/test_content_signing/pysign.py

   1 #!/usr/bin/env python
   2 #
   3 # This Source Code Form is subject to the terms of the Mozilla Public
   4 # License, v. 2.0. If a copy of the MPL was not distributed with this
   5 # file, You can obtain one at http://mozilla.org/MPL/2.0/.
   6
   7 """
   8 Create an ECDSA signature on the P-384 curve using the SHA-384 hash of data from
   9 stdin. The key used for the signature is the secp384r1Encoded key used in pykey
  10 and pycert.
  11
  12 The certificates for the content signature tests make use of this program.
  13 You can use pysign.py like this:
  14
  15 cat test.txt | python pysign.py > test.txt.signature
  16 """
  17
  18 import base64
  19 import binascii
  20 import hashlib
  21 import pathlib
  22 import six
  23 import sys
  24
  25 import ecdsa
  26
  27 # For pykey, find the relative file location and add it to path
  28 toolsDir = (pathlib.Path(__file__).parents[4] / "tools").resolve()
  29 sys.path.append(str(toolsDir))
  30 import pykey
  31
  32 data = sys.stdin.buffer.read()
  33
  34 key = pykey.ECCKey("secp384r1")
  35 sig = key.signRaw(b"Content-Signature:\00" + data, pykey.HASH_SHA384)
  36 print(str(base64.b64encode(sig)).replace("+", "-").replace("/", "_"))