Exemplo n.º 1
0
def test_keys_generation_2048_with_password(tmpdir):
    priv_key = generate_rsa_private_key()
    pub_key = generate_rsa_public_key(priv_key)
    save_rsa_private_key(priv_key, path.join(tmpdir, "priv_2048_password.pem"), 'abc')
    save_rsa_public_key(pub_key, path.join(tmpdir, "pub_2048_2.pem"))
    assert path.isfile(path.join(tmpdir, "priv_2048_password.pem"))
    assert path.isfile(path.join(tmpdir, "pub_2048_2.pem"))
Exemplo n.º 2
0
def test_keys_generation_4096(tmpdir):
    priv_key = generate_rsa_private_key(key_size=4096)
    pub_key = generate_rsa_public_key(priv_key)
    save_rsa_private_key(priv_key, path.join(tmpdir, "priv_4096.pem"))
    save_rsa_public_key(pub_key, path.join(tmpdir, "pub_4096.pem"))
    assert path.isfile(path.join(tmpdir, "priv_4096.pem"))
    assert path.isfile(path.join(tmpdir, "pub_4096.pem"))
Exemplo n.º 3
0
def test_keys_generation_2048(tmpdir):
    priv_key = generate_rsa_private_key()
    pub_key = generate_rsa_public_key(priv_key)
    save_rsa_private_key(priv_key, path.join(tmpdir, "priv_2048.pem"))
    save_rsa_public_key(pub_key, path.join(tmpdir, "pub_2048.pem"))
    assert path.isfile(path.join(tmpdir, "priv_2048.pem"))
    assert path.isfile(path.join(tmpdir, "pub_2048.pem"))
Exemplo n.º 4
0
def main() -> None:
    """Main function."""
    # Set the folder for data (certificates, keys)
    data_dir = path.join(path.dirname(__file__), 'data')
    os.makedirs(data_dir, exist_ok=True)

    # Generate and save rsa keys (size 2048) - pem format (for usage of CA certificate)
    priv_key_2048 = generate_rsa_private_key(key_size=2048)
    pub_key_2048 = generate_rsa_public_key(priv_key_2048)
    save_rsa_private_key(priv_key_2048,
                         path.join(data_dir, "ca_privatekey_rsa2048.pem"))
    save_rsa_public_key(pub_key_2048,
                        path.join(data_dir, "ca_publickey_rsa2048.pem"))
    print(
        "The pair of private (ca_privatekey_rsa2048.pem) and public (ca_publickey_rsa2048.pem) key was generated."
    )

    # Generate and save rsa keys (size 2048) - pem format (for usage of chain of certificate)
    priv_key_2048 = generate_rsa_private_key(key_size=2048)
    pub_key_2048 = generate_rsa_public_key(priv_key_2048)
    save_rsa_private_key(priv_key_2048,
                         path.join(data_dir, "crt_privatekey_rsa2048.pem"))
    save_rsa_public_key(pub_key_2048,
                        path.join(data_dir, "crt_publickey_rsa2048.pem"))
    print(
        "The pair of private (crt_privatekey_rsa2048.pem) and public (crt_publickey_rsa2048.pem) key was generated."
    )

    # Generate and save rsa keys (size 2048) - pem format (for usage of chain of certificate)
    priv_key_2048 = generate_rsa_private_key(key_size=2048)
    pub_key_2048 = generate_rsa_public_key(priv_key_2048)
    save_rsa_private_key(priv_key_2048,
                         path.join(data_dir, "chain_privatekey_rsa2048.pem"))
    save_rsa_public_key(pub_key_2048,
                        path.join(data_dir, "chain_publickey_rsa2048.pem"))
    print(
        "The pair of private (chain_privatekey_rsa2048.pem) and public (chain_publickey_rsa2048.pem) key was "
        "generated.")

    # Generate and save rsa keys (size 2048) - pem format (for usage of chain of certificate)
    priv_key_2048 = generate_rsa_private_key(key_size=2048)
    pub_key_2048 = generate_rsa_public_key(priv_key_2048)
    save_rsa_private_key(
        priv_key_2048, path.join(data_dir,
                                 "chain_crt2_privatekey_rsa2048.pem"))
    save_rsa_public_key(
        pub_key_2048, path.join(data_dir, "chain_crt2_publickey_rsa2048.pem"))
    print(
        "The pair of private (chain_crt2_privatekey_rsa2048.pem) and public (chain_crt2_publickey_rsa2048.pem) key "
        "was generated.")

    # Generate and save rsa keys (size 3072) - pem format
    priv_key_3072 = generate_rsa_private_key(key_size=3072)
    pub_key_3072 = generate_rsa_public_key(priv_key_3072)
    save_rsa_private_key(priv_key_3072,
                         path.join(data_dir, "private_rsa3072.pem"))
    save_rsa_public_key(pub_key_3072, path.join(data_dir,
                                                "public_rsa3072.pem"))
    print(
        "The pair of private (private_rsa3072.pem) and public (public_rsa3072.pem) key was generated."
    )

    # Generate and save rsa keys (size 4096) - pem format
    priv_key_4096 = generate_rsa_private_key(key_size=4096)
    pub_key_4096 = generate_rsa_public_key(priv_key_4096)
    save_rsa_private_key(priv_key_4096,
                         path.join(data_dir, "private_rsa4096.pem"))
    save_rsa_public_key(pub_key_4096, path.join(data_dir,
                                                "public_rsa4096.pem"))
    print(
        "The pair of private (private_rsa4096.pem) and public (public_rsa4096.pem) key was generated."
    )

    # Generate and save rsa keys (size 2048) - der format
    priv_key_2048 = generate_rsa_private_key(key_size=2048)
    pub_key_2048 = generate_rsa_public_key(priv_key_2048)
    save_rsa_private_key(priv_key_2048,
                         path.join(data_dir, "private_rsa2048.der"),
                         encoding=Encoding.DER)
    save_rsa_public_key(pub_key_2048,
                        path.join(data_dir, "public_rsa2048.der"),
                        encoding=Encoding.DER)
    print(
        "The pair of private (private_rsa2048.der) and public (public_rsa2048.der) key was generated."
    )

    # Generate and save rsa keys (size 3072) - der format
    priv_key_3072 = generate_rsa_private_key(key_size=3072)
    pub_key_3072 = generate_rsa_public_key(priv_key_3072)
    save_rsa_private_key(priv_key_3072,
                         path.join(data_dir, "private_rsa3072.der"),
                         encoding=Encoding.DER)
    save_rsa_public_key(pub_key_3072,
                        path.join(data_dir, "public_rsa3072.der"),
                        encoding=Encoding.DER)
    print(
        "The pair of private (private_rsa3072.der) and public (public_rsa3072.der) key was generated."
    )

    # Generate and save rsa keys (size 4096) - der format
    priv_key_4096 = generate_rsa_private_key(key_size=4096)
    pub_key_4096 = generate_rsa_public_key(priv_key_4096)
    save_rsa_private_key(priv_key_4096,
                         path.join(data_dir, "private_rsa4096.der"),
                         encoding=Encoding.DER)
    save_rsa_public_key(pub_key_4096,
                        path.join(data_dir, "public_rsa4096.der"),
                        encoding=Encoding.DER)
    print(
        "The pair of private (private_rsa4096.der) and public (public_rsa4096.der) key was generated."
    )

    # Generate and save ECC keys (curve P-256) - pem format
    priv_key_p256 = generate_ecc_private_key(curve_name='P-256')
    pub_key_p256 = generate_ecc_public_key(priv_key_p256)
    save_ecc_private_key(priv_key_p256,
                         path.join(data_dir, "ecc_privatekey_p256.pem"))
    save_ecc_public_key(pub_key_p256,
                        path.join(data_dir, "ecc_publickey_p256.pem"))
    print(
        "The pair of private (ecc_privatekey_p256.pem) and public (ecc_publickey_p256.pem) key was generated."
    )