车联网数据共享方案 - Shamir 门限密钥共享算法实现 - 常规

一、实验目的

掌握车联网中的数据共享方案原理，并利用 Python3.9 实现 Shamir 门限秘密共享算法。

二、实验环境

Python 3.9
Crypto 库

三、实验内容

3.1 Shamir 门限密钥共享方案原理

Shamir(t, w) 门限密钥共享方案是通过构造一个 t-1 次多项式，将需要共享的主密钥 S 作为常数项，碎片密钥分成 w 部分给 w 个参与者，当碎片密钥数量大于或者等于 t 的话，就可以求解出这个主密钥 S。

子密钥生成：

构造多项式

F(x) = s + a_1 * x + a_2 * x^2 + ... + a_{t-1} * x^{t-1}

其中 s 为密钥，p 为素数 (s 取 w 个不相等的 x 带入 F(x) 中，得到 w 组子密钥，分发给 w 个人保管，将 p 公开，销毁子密钥生成多项式，每个人保管自己的子密钥)。

恢复密钥：

构造多项式

F(x) = s + a_1 * x + a_2 * x^2 + ... + a_{t-1} * x^{t-1}

取 x=0，代入 t 组密钥可以求解出 F(0)，也就是主密钥 s。

3.2 代码实现

import Crypto.Util.number as numb
import random


# 求逆的函数，之前的版本用python2写的，这次用的python3，只把整除符号改了一下
def oj(a, n):
    a = a % n
    s = [0, 1]
    while a != 1:
        if a == 0:
            return 0
        q = n // a
        t = n % a
        n = a
        a = t
        s += [s[-2] - q * s[-1]]
    return s[-1]


# max_length 为p的长度，同时也是秘密的最大长度
# secret_is_text =0 默认输入时文本， 非0时认为是数字
# p 默认为0， 会根据max_length 自动生成，不为0时直接使用，需要保证p为素数， 函数内没有素性检验
def create(max_length=513, secret_is_text=0, p=0):
    if not p:
        p = numb.getPrime(max_length)

    w = int(input('请输入秘密保存人数：'))
    t = int(input('请输入秘密恢复所需人数：'))
    while not (t > 0 and t <= w):
        t = int(input('请重新输入：'))
    s = input('请输入你的秘密:')

    if secret_is_text:
        s = numb.bytes_to_long(s.encode('utf-8'))
    else:
        try:
            s = int(s)
        except Exception as e:
            s = numb.bytes_to_long(s.encode('utf-8'))

    x_list = list()
    a_list = list()
    i = w
    while i > 0:
        x = random.randint(p // 2, p)  # 该范围没有特定限制，如果想让xi,yi取小一点儿的话可把范围写小点儿，但是要大于w
        if x not in x_list:
            x_list.append(x)
            i -= 1

    for a in range(t):
        a_list.append(random.randint(p // 2, p))  # 同上

    result = list()
    for x in x_list:
        y = s
        for a_n in range(t):
            a = a_list[i]
            y += a * pow(x, i + 1, p)
        result.append((x, y))
    return t, p, result


# get_text=1 默认恢复为字符串，若想得到数字填0
def restore(p, information, get_text=1):
    x_list = list()
    y_list = list()
    for x, y in information:
        x_list.append(x)
        y_list.append(y)

    s = 0
    for x_i in range(len(x_list)):
        tmp_num = y_list[x_i]
        x_i_j = 1
        for x_j in range(len(x_list)):
            if x_i != x_j:
                tmp_num = tmp_num * (0 - x_list[x_j]) % p
                x_i_j *= x_list[x_i] - x_list[x_j]
        tmp_num = tmp_num * oj(x_i_j, p) % p
        s += tmp_num

    s = s % p
    print(s)
    if get_text:
        try:
            s = numb.long_to_bytes(s)
            s = s.decode('utf-8')
        except Exception as e:
            print(e)

    return s


t, p, result = create()  # result为秘密碎片的列表

print(result)
print('还原出最初的秘密S:')
print(restore(p, result[:t], 0))

四、实验总结

在 Python3.9 中成功实现了 Shamir 门限秘密共享算法。在编写函数时遇到了参数输入错误的问题，通过调试和修改代码最终解决了问题，成功生成了秘密碎片并恢复了最初的秘密 S。通过本次实验，我掌握了车联网中的数据共享方案原理，加深了对门限密钥共享方案的理解，并提升了编程能力。