sangge/tpre-python
2
1
Fork 1
Code Issues 6 Pull Requests Packages Projects 1 Activity

feat: update c extension
All checks were successful
Test CI / test speed (push) Successful in 18s
Details

Browse Source
This commit is contained in:
sangge 2023-12-09 16:33:16 +08:00
parent 8eda13d4f5
commit 8a4059906c
3 changed files with 39 additions and 47 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
include/tpre.h
View File

@ -1,32 +0,0 @@
#ifndef tpre_SM2_H
#define tpre_SM2_H
#include <Python.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
// define TPRE Big Number
typedef uint64_t TPRE_BN[8]
// GF(p)
typedef TPRE_BN SM2_Fp;
// GF(n)
typedef TPRE_BN SM2_Fn;
// 定义一个结构体来表示雅各比坐标系的点
typedef struct
{
TPRE_BN X;
TPRE_BN Y;
TPRE_BN Z;
} JACOBIAN_POINT;
// 定义一个结构体来表示点
typedef struct
{
uint8_t x[32];
uint8_t y[32];
} TPRE_POINT;

9
src/setup.py
View File

@ -1,11 +1,14 @@
from setuptools import setup, Extension
# 定义您的扩展
ext = Extension("tpre", sources=["tpre.c"])
ext = Extension(
"tpreECC",
sources=["tpreECC.c"],
)
setup(
name="tpre",
name="tpreECC",
version="1.0",
description="tpre written in C",
description="basic ECC written in C",
ext_modules=[ext],
)

45
src/tpre.c → src/tpreECC.c
View File

@ -2,9 +2,32 @@
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include "../include/tpre.h"
const SM2_BN SM2_P = {
// define TPRE Big Number
typedef uint64_t TPRE_BN[8]
// GF(p)
typedef TPRE_BN SM2_Fp;
// GF(n)
typedef TPRE_BN SM2_Fn;
// 定义一个结构体来表示雅各比坐标系的点
typedef struct
{
TPRE_BN X;
TPRE_BN Y;
TPRE_BN Z;
} JACOBIAN_POINT;
// 定义一个结构体来表示点
typedef struct
{
uint8_t x[32];
uint8_t y[32];
} TPRE_POINT;
const TPRE_BN SM2_P = {
0xffffffff,
0xffffffff,
0x00000000,
@ -15,7 +38,7 @@ const SM2_BN SM2_P = {
0xfffffffe,
};
const SM2_BN SM2_B = {
const TPRE_BN SM2_B = {
0x4d940e93,
0xddbcbd41,
0x15ab8f92,
@ -26,6 +49,7 @@ const SM2_BN SM2_B = {
0x28e9fa9e,
};
// 生成元GX, GY
const SM2_JACOBIAN_POINT _SM2_G = {
{
0x334c74c7,
@ -58,9 +82,10 @@ const SM2_JACOBIAN_POINT _SM2_G = {
0,
},
};
const SM2_JACOBIAN_POINT *SM2_G = &_SM2_G;
const SM2_BN SM2_N = {
const TPRE_BN SM2_N = {
0x39d54123,
0x53bbf409,
0x21c6052b,
@ -72,7 +97,7 @@ const SM2_BN SM2_N = {
};
// u = (p - 1)/4, u + 1 = (p + 1)/4
const SM2_BN SM2_U_PLUS_ONE = {
const TPRE_BN SM2_U_PLUS_ONE = {
0x00000000,
0x40000000,
0xc0000000,
@ -83,13 +108,9 @@ const SM2_BN SM2_U_PLUS_ONE = {
0x3fffffff,
};
const SM2_BN SM2_ONE = {1, 0, 0, 0, 0, 0, 0, 0};
const SM2_BN SM2_TWO = {2, 0, 0, 0, 0, 0, 0, 0};
const SM2_BN SM2_THREE = {3, 0, 0, 0, 0, 0, 0, 0};
const TPRE_BN SM2_ONE = {1, 0, 0, 0, 0, 0, 0, 0};
const TPRE_BN SM2_TWO = {2, 0, 0, 0, 0, 0, 0, 0};
const TPRE_BN SM2_THREE = {3, 0, 0, 0, 0, 0, 0, 0};
// 将Python中的multiply函数转换为C
static Point multiply(Point a, int64_t n)