Python实现RSA加解密算法

闲人编程

于 2024-08-28 11:25:06 发布

阅读量953

点赞数 12

分类专栏：密码学文章标签： python 算法网络 RSA J密码学加密解密

本文链接：https://blog.csdn.net/qq_42568323/article/details/141636174

版权

密码学专栏收录该内容

16 篇文章 0 订阅

订阅专栏

深入了解RSA加密算法

RSA（Rivest-Shamir-Adleman）是一种非对称加密算法，由Ron Rivest、Adi Shamir和Leonard Adleman于1977年提出。作为非对称加密算法的典型代表，RSA被广泛用于数据加密和数字签名。本文将深入介绍RSA算法的工作原理，并展示如何使用Python从头实现RSA加密和解密，而不依赖任何第三方加密库。

一、RSA算法概述

RSA是一种基于数论的非对称加密算法，利用了大数分解的计算复杂性。与对称加密算法不同，RSA使用一对密钥：公钥用于加密，私钥用于解密。这意味着即使加密密钥被公开，只有拥有解密密钥的人才能解密密文。

1.1 关键步骤

RSA加密算法的核心可以分为以下几个步骤：

密钥生成:
- 选择两个大素数 $p$ 和 $q$ ，计算它们的乘积 $\times q$ 。这个值 $n$ 将被用作公钥和私钥的一个组成部分。
- 计算欧拉函数 $\phi(n) = (p-1) \times (q-1)$ 。
- 选择一个与 $\phi(n)$ 互质的整数 $e$ ，使得 $\phi(n)$ 。 这个 $e$ 是公钥的一部分。
- 计算私钥 $d$ ，满足 $\times e \equiv 1 \ (\text{mod} \ \phi(n))$ 。这个 $d$ 是私钥的一部分。
加密:
- 加密时，将明文 $M$ 转换为整数形式 $m$ ，然后使用公钥 $(e, n)$ 计算密文 $c$ ：
  $m^e \ (\text{mod} \ n)$
解密:
- 解密时，使用私钥 $(d, n)$ 计算明文 $m$ ：
  $c^d \ (\text{mod} \ n)$

1.2 安全性分析

RSA的安全性基于大数分解的困难性。随着密钥长度的增加，因数分解 (n) 的难度成指数级增长，这使得通过已知公钥推算私钥在计算上变得不可行。

二、RSA算法的Python实现

下面我们将一步步实现一个基本的RSA加密和解密程序，不使用任何第三方加密库。我们将从密钥生成开始，然后实现加密和解密。

2.1 辅助函数

我们首先定义几个基本的数学辅助函数：

import random

# 计算最大公约数
def gcd(a, b):
    while b != 0:
        a, b = b, a % b
    return a

# 扩展欧几里得算法，用于计算模逆
def extended_gcd(a, b):
    if b == 0:
        return a, 1, 0
    g, x1, y1 = extended_gcd(b, a % b)
    x = y1
    y = x1 - (a // b) * y1
    return g, x, y

# 计算模逆
def mod_inverse(e, phi):
    g, x, _ = extended_gcd(e, phi)
    if g != 1:
        raise ValueError("模逆不存在")
    return x % phi

# 快速幂算法
def power_mod(base, exponent, modulus):
    result = 1
    base = base % modulus
    while exponent > 0:
        if exponent % 2 == 1:
            result = (result * base) % modulus
        exponent = exponent >> 1
        base = (base * base) % modulus
    return result

2.2 密钥生成

接下来，我们定义密钥生成函数。为了简化实现，我们将使用一个简单的素数生成方法。

# 判断一个数是否为素数
def is_prime(n):
    if n <= 1:
        return False
    if n <= 3:
        return True
    if n % 2 == 0 or n % 3 == 0:
        return False
    i = 5
    while i * i <= n:
        if n % i == 0 or n % (i + 2) == 0:
            return False
        i += 6
    return True

# 生成随机大素数
def generate_prime_candidate(length):
    p = 4
    while not is_prime(p):
        p = random.getrandbits(length)
    return p

# 生成RSA密钥对
def generate_keys(bit_length):
    p = generate_prime_candidate(bit_length // 2)
    q = generate_prime_candidate(bit_length // 2)
    n = p * q
    phi = (p - 1) * (q - 1)
    e = random.randrange(1, phi)
    g = gcd(e, phi)
    while g != 1:
        e = random.randrange(1, phi)
        g = gcd(e, phi)
    d = mod_inverse(e, phi)
    return ((e, n), (d, n))  # 公钥和私钥

2.3 加密与解密

一旦密钥生成，我们可以实现加密和解密功能。

# RSA加密
def rsa_encrypt(plaintext, public_key):
    e, n = public_key
    # 将明文转换为整数
    plaintext_int = int.from_bytes(plaintext.encode('utf-8'), byteorder='big')
    # 加密
    ciphertext = power_mod(plaintext_int, e, n)
    return ciphertext

# RSA解密
def rsa_decrypt(ciphertext, private_key):
    d, n = private_key
    # 解密
    plaintext_int = power_mod(ciphertext, d, n)
    # 将整数转换为明文
    plaintext = plaintext_int.to_bytes((plaintext_int.bit_length() + 7) // 8, byteorder='big').decode('utf-8')
    return plaintext

2.4 使用示例

最后，我们展示如何使用上述函数生成密钥，并对消息进行加密和解密。

if __name__ == "__main__":
    # 生成密钥对（1024位）
    public_key, private_key = generate_keys(1024)
    
    # 原始消息
    message = "Hello, RSA!"
    print("Original Message:", message)
    
    # 加密
    ciphertext = rsa_encrypt(message, public_key)
    print("Encrypted Message:", ciphertext)
    
    # 解密
    decrypted_message = rsa_decrypt(ciphertext, private_key)
    print("Decrypted Message:", decrypted_message)
    
    # 检查解密结果是否与原始消息一致
    assert decrypted_message == message