如何在python

时间:2018-05-30 15:05:16

标签: python-3.x ssl openssl cryptography sha256

背景

我使用openssl SHA256和私钥签署了一个文件,如下所示:

with subprocess.Popen(
        # Pipe the signature to openssl to convert it from raw binary encoding to base64 encoding.
        # This will prevent any potential corruption due to line ending conversions, and also allows 
        # a human to read and copy the signature (e.g. for manual verification).
        'openssl dgst -sha256 -sign private.key sign_me.zip | openssl base64 > signature.sha256',
        stdout=subprocess.PIPE,
        stderr=subprocess.STDOUT,
        shell=True,
) as proc:
    out, _ = proc.communicate()

要求

  1. 我需要使用signature.sha256public_key.crt来验证sign_me.zip是否未被修改。
  2. 与Python 3.2 - 3.4
  3. 兼容
  4. 需要在Windows和Redhat上工作,并且无法保证OpenSSL将位于路径上或位于已知位置。理想情况下,我想使用核心Python模块,但如果降低复杂性,我会考虑使用第三方模块。
  5. 我尝试过什么

    我已经做了大量的搜索,试图弄清楚如何做到这一点,但我找不到满意的答案。以下列出了我尝试和/或研究的内容:

    • 我可以通过以下shell命令手动验证签名。由于要求3,这不能作为永久解决方案。

      openssl dgst -sha256 -verify <(openssl x509 -in public_key.crt -pubkey -noout) -signature signature.sha256 sign_me.zip

    • 我找到了this question,这几乎就是我想做的事情。近两年来,它尚未得到回答甚至评论。它提到了ssl python library,它主要处理客户端/服务器证书和套接字。

    • This question似乎使用加密库来验证“SHA256withRSA和PKCS1填充”签名。不幸的是,它针对Python 2.7,另外我无法在问题引用的Python 2.7 crypto模块中找到verify()方法文档。
    • 我还发现了名为cryptography的第三方模块。 Stack Overflow上的共识似乎是这是加密等最新/最好的模块,但我无法找到符合我要求的文档。

    也许我错过了一些明显的东西?我没有做过很多安全/加密/散列工作,所以欢迎反馈。

1 个答案:

答案 0 :(得分:4)

感谢Patrick Mevzek向我指出了正确的方向。最终,我使用Cryptography模块为我的问题找到了以下解决方案。我最终改变了我对文件签名的方式,以匹配以后将对其进行验证的方式。

密钥生成:

from cryptography.hazmat.backends import default_backend
from cryptography.hazmat.primitives import serialization
from cryptography.hazmat.primitives.asymmetric import rsa

# Generate the public/private key pair.
private_key = rsa.generate_private_key(
    public_exponent = 65537,
    key_size = 4096,
    backend = default_backend(),
)

# Save the private key to a file.
with open('private.key', 'wb') as f:
    f.write(
        private_key.private_bytes(
            encoding=serialization.Encoding.PEM,
            format=serialization.PrivateFormat.TraditionalOpenSSL,
            encryption_algorithm=serialization.NoEncryption(),
        )
    )

# Save the public key to a file.
with open('public.pem', 'wb') as f:
    f.write(
        private_key.public_key().public_bytes(
            encoding = serialization.Encoding.PEM,
            format = serialization.PublicFormat.SubjectPublicKeyInfo,
        )
    )

签名:

import base64
from cryptography.hazmat.backends import default_backend
from cryptography.hazmat.primitives import hashes
from cryptography.hazmat.primitives import serialization
from cryptography.hazmat.primitives.asymmetric import padding

# Load the private key. 
with open('private.key', 'rb') as key_file: 
    private_key = serialization.load_pem_private_key(
        key_file.read(),
        password = None,
        backend = default_backend(),
    )

# Load the contents of the file to be signed.
with open('payload.dat', 'rb') as f:
    payload = f.read()

# Sign the payload file.
signature = base64.b64encode(
    private_key.sign(
        payload,
        padding.PSS(
            mgf = padding.MGF1(hashes.SHA256()),
            salt_length = padding.PSS.MAX_LENGTH,
        ),
        hashes.SHA256(),
    )
)
with open('signature.sig', 'wb') as f:
    f.write(signature)

验证:

import base64
import cryptography.exceptions
from cryptography.hazmat.backends import default_backend
from cryptography.hazmat.primitives import hashes
from cryptography.hazmat.primitives.asymmetric import padding
from cryptography.hazmat.primitives.serialization import load_pem_public_key

# Load the public key.
with open('public.pem', 'rb') as f:
    public_key = load_pem_public_key(f.read(), default_backend())

# Load the payload contents and the signature.
with open('payload.dat', 'rb') as f:
    payload_contents = f.read(payload_file)
with open('signature.sig', 'rb') as f:
    signature = base64.b64decode(f.read())

# Perform the verification.
try:
    public_key.verify(
        signature,
        payload_contents,
        padding.PSS(
            mgf = padding.MGF1(hashes.SHA256()),
            salt_length = padding.PSS.MAX_LENGTH,
        ),
        hashes.SHA256(),
    )
except cryptography.exceptions.InvalidSignature as e:
    print('ERROR: Payload and/or signature files failed verification!')