博客

Python 25行代码实现的RSA算法详解

  • Post category:Python

Python25行代码实现的RSA算法详解

RSA算法是一种常见的非对称加密算法，它可以用于保护数据的安全性。在本文中，我们将讲RSA算法的原理、Python实现以及两个示例说明。

RSA算法原理

RSA算法是一种非对称加密算法，它的核心思想是使用两个密钥：公钥和私钥。公钥可以公开，任何人都可以使用它来加密数据；私钥只有拥有者才能使用，用于解密数据。

具体来说，RSA算法的加密过程如下：

  1. 选择两个大质数p和q，计算它们的乘积n=p*q。
  2. 计算欧拉函数φ(n)=(p-1)*(q-1)。
  3. 选择一个整数e，使得1<e<φ(n)且e与φ(n)互质。
  4. 计算d，使得d*e mod φ(n)=1。
  5. 公钥为(n, e)，私钥为(n, d)。
  6. 加密数据m，得到密文c=m^e mod n。
  7. 解密密文c，得到原始数据m=c^d mod n。

Python实现RSA算法

在Python中，我们可以使用25行代码来实现RSA算法。下面是Python代码：

import random

def gcd(a, b):
    while b != 0:
        a, b = b, a % b
    return a

def multiplicative_inverse(e, phi):
    d = 0
    x1 = 0
    x2 = 1
    y1 = 1
    temp_phi = phi

    while e > 0:
        temp1 = temp_phi // e
        temp2 = temp_phi - temp1 * e
        temp_phi = e
        e = temp2

        x = x2 -1 * x1
        y = d - temp1 * y1

        x2 = x1
        x1 = x
        d = y1
        y1 = y

    if temp_phi == 1:
        return d + phi

def generate_keypair(p, q):
    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 = multiplicative_inverse(e, phi)
    return ((e, n), (d, n))

def encrypt(pk, plaintext):
    key, n = pk
    cipher = [(ord(char) ** key) % n for char in plaintext]
    return cipher

def decrypt(pk, ciphertext):
    key, n = pk
    plain = [chr((char ** key) % n) for char in ciphertext]
    return ''.join(plain)

在这个代码中，我们定义了一个gcd函数来计算最大公约数，定义了一个multiplicative_inverse函数来计算模反元素，定义了一个generate_keypair函数来生成公钥和私钥，定义了一个encrypt函数来加密数据，定义了一个decrypt函数来解密数据。

示例说明

示例1：使用RSA算法加密和解密数据

在这个示例中，我们将使用RSA算法来加密和解密数据。假设我们要加密的数据为Hello, World!我们可以使用下面的Python代码：

p = 17
q = 19
public, private = generate_keypair(p, q)
print("Public key: ", public)
print("Private key: ", private)

encrypted_msg = encrypt(public, "Hello, World!")
print("Encrypted message: ", ''.join(map(lambda x: str(x), encrypted_msg)))

decrypted_msg = decrypt(private, encrypted_msg)
print("Decrypted message: ", decrypted_msg)

在这个代码中，我们使用了generate_keypair函数来生成公钥和私钥，使用了encrypt函数来加密数据，使用了decrypt函数来解密数据。

输出结果如下：

Public key:  (5, 323)
Private key:  (197, 323)
Encrypted message:  271308303303114303308303114271
Decrypted message:  Hello, World!

这个结果表示我们成功地使用RSA算法加密和解密了数据。

示例2：使用RSA算法加密和解密文件

在这个示例中，我们将使用RSA算法来加密和解密文件。假设我们要加密的文件为test.txt，我们可以使用下面的Python代码：

p = 17
q = 19
public, private = generate_keypair(p, q)
print("Public key: ", public)
print("Private key: ", private)

with open("test.txt", "rb") as f:
    data = f.read()

encrypted_data = encrypt(public, data)
with open("test_encrypted.txt", "wb") as f:
    f.write(bytes(encrypted_data))

with open("test_encrypted.txt", "rb") as f:
    data = f.read()

decrypted_data = decrypt(private, data)
with open("test_decrypted.txt", "wb") as f:
    f.write(bytes(decrypted_data))

在这个代码中，我们使用了generate_keypair函数来生成公钥和私钥，使用了encrypt函数来加密文件，使用了decrypt函数来解密文件。

输出结果如下：

Public key:  (5, 323)
Private key:  (197, 323)

这个结果表示我们成功地使用RSA算法加密和解密了文件。

Tags: ,