Skip to content

Commit a8ea8bd

Browse files
uudiinherbertx
uudiin
authored and
herbertx
committed
lib/mpi: Extend the MPI library
Expand the mpi library based on libgcrypt, and the ECC algorithm of mpi based on libgcrypt requires these functions. Some other algorithms will be developed based on mpi ecc, such as SM2. Signed-off-by: Tianjia Zhang <[email protected]> Tested-by: Xufeng Zhang <[email protected]> Signed-off-by: Herbert Xu <[email protected]>
1 parent f492828 commit a8ea8bd

File tree

14 files changed

+2076
-10
lines changed

14 files changed

+2076
-10
lines changed

‎include/linux/mpi.h‎

Lines changed: 87 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,35 +40,122 @@ struct gcry_mpi {
4040
typedef struct gcry_mpi *MPI;
4141

4242
#define mpi_get_nlimbs(a) ((a)->nlimbs)
43+
#define mpi_has_sign(a) ((a)->sign)
4344

4445
/*-- mpiutil.c --*/
4546
MPI mpi_alloc(unsigned nlimbs);
47+
void mpi_clear(MPI a);
4648
void mpi_free(MPI a);
4749
int mpi_resize(MPI a, unsigned nlimbs);
4850

51+
static inline MPI mpi_new(unsigned int nbits)
52+
{
53+
return mpi_alloc((nbits + BITS_PER_MPI_LIMB - 1) / BITS_PER_MPI_LIMB);
54+
}
55+
56+
MPI mpi_copy(MPI a);
57+
MPI mpi_alloc_like(MPI a);
58+
void mpi_snatch(MPI w, MPI u);
59+
MPI mpi_set(MPI w, MPI u);
60+
MPI mpi_set_ui(MPI w, unsigned long u);
61+
MPI mpi_alloc_set_ui(unsigned long u);
62+
void mpi_swap_cond(MPI a, MPI b, unsigned long swap);
63+
64+
/* Constants used to return constant MPIs. See mpi_init if you
65+
* want to add more constants.
66+
*/
67+
#define MPI_NUMBER_OF_CONSTANTS 6
68+
enum gcry_mpi_constants {
69+
MPI_C_ZERO,
70+
MPI_C_ONE,
71+
MPI_C_TWO,
72+
MPI_C_THREE,
73+
MPI_C_FOUR,
74+
MPI_C_EIGHT
75+
};
76+
77+
MPI mpi_const(enum gcry_mpi_constants no);
78+
4979
/*-- mpicoder.c --*/
80+
81+
/* Different formats of external big integer representation. */
82+
enum gcry_mpi_format {
83+
GCRYMPI_FMT_NONE = 0,
84+
GCRYMPI_FMT_STD = 1, /* Twos complement stored without length. */
85+
GCRYMPI_FMT_PGP = 2, /* As used by OpenPGP (unsigned only). */
86+
GCRYMPI_FMT_SSH = 3, /* As used by SSH (like STD but with length). */
87+
GCRYMPI_FMT_HEX = 4, /* Hex format. */
88+
GCRYMPI_FMT_USG = 5, /* Like STD but unsigned. */
89+
GCRYMPI_FMT_OPAQUE = 8 /* Opaque format (some functions only). */
90+
};
91+
5092
MPI mpi_read_raw_data(const void *xbuffer, size_t nbytes);
5193
MPI mpi_read_from_buffer(const void *buffer, unsigned *ret_nread);
94+
int mpi_fromstr(MPI val, const char *str);
95+
MPI mpi_scanval(const char *string);
5296
MPI mpi_read_raw_from_sgl(struct scatterlist *sgl, unsigned int len);
5397
void *mpi_get_buffer(MPI a, unsigned *nbytes, int *sign);
5498
int mpi_read_buffer(MPI a, uint8_t *buf, unsigned buf_len, unsigned *nbytes,
5599
int *sign);
56100
int mpi_write_to_sgl(MPI a, struct scatterlist *sg, unsigned nbytes,
57101
int *sign);
102+
int mpi_print(enum gcry_mpi_format format, unsigned char *buffer,
103+
size_t buflen, size_t *nwritten, MPI a);
104+
105+
/*-- mpi-mod.c --*/
106+
void mpi_mod(MPI rem, MPI dividend, MPI divisor);
107+
108+
/* Context used with Barrett reduction. */
109+
struct barrett_ctx_s;
110+
typedef struct barrett_ctx_s *mpi_barrett_t;
111+
112+
mpi_barrett_t mpi_barrett_init(MPI m, int copy);
113+
void mpi_barrett_free(mpi_barrett_t ctx);
114+
void mpi_mod_barrett(MPI r, MPI x, mpi_barrett_t ctx);
115+
void mpi_mul_barrett(MPI w, MPI u, MPI v, mpi_barrett_t ctx);
58116

59117
/*-- mpi-pow.c --*/
60118
int mpi_powm(MPI res, MPI base, MPI exp, MPI mod);
61119

62120
/*-- mpi-cmp.c --*/
63121
int mpi_cmp_ui(MPI u, ulong v);
64122
int mpi_cmp(MPI u, MPI v);
123+
int mpi_cmpabs(MPI u, MPI v);
65124

66125
/*-- mpi-sub-ui.c --*/
67126
int mpi_sub_ui(MPI w, MPI u, unsigned long vval);
68127

69128
/*-- mpi-bit.c --*/
70129
void mpi_normalize(MPI a);
71130
unsigned mpi_get_nbits(MPI a);
131+
int mpi_test_bit(MPI a, unsigned int n);
132+
void mpi_set_bit(MPI a, unsigned int n);
133+
void mpi_set_highbit(MPI a, unsigned int n);
134+
void mpi_clear_highbit(MPI a, unsigned int n);
135+
void mpi_clear_bit(MPI a, unsigned int n);
136+
void mpi_rshift_limbs(MPI a, unsigned int count);
137+
void mpi_rshift(MPI x, MPI a, unsigned int n);
138+
void mpi_lshift_limbs(MPI a, unsigned int count);
139+
void mpi_lshift(MPI x, MPI a, unsigned int n);
140+
141+
/*-- mpi-add.c --*/
142+
void mpi_add_ui(MPI w, MPI u, unsigned long v);
143+
void mpi_add(MPI w, MPI u, MPI v);
144+
void mpi_sub(MPI w, MPI u, MPI v);
145+
void mpi_addm(MPI w, MPI u, MPI v, MPI m);
146+
void mpi_subm(MPI w, MPI u, MPI v, MPI m);
147+
148+
/*-- mpi-mul.c --*/
149+
void mpi_mul(MPI w, MPI u, MPI v);
150+
void mpi_mulm(MPI w, MPI u, MPI v, MPI m);
151+
152+
/*-- mpi-div.c --*/
153+
void mpi_tdiv_r(MPI rem, MPI num, MPI den);
154+
void mpi_fdiv_r(MPI rem, MPI dividend, MPI divisor);
155+
void mpi_fdiv_q(MPI quot, MPI dividend, MPI divisor);
156+
157+
/*-- mpi-inv.c --*/
158+
int mpi_invm(MPI x, MPI a, MPI n);
72159

73160
/* inline functions */
74161

‎lib/mpi/Makefile‎

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,9 +14,14 @@ mpi-y = \
1414
generic_mpih-sub1.o \
1515
generic_mpih-add1.o \
1616
mpicoder.o \
17+
mpi-add.o \
1718
mpi-bit.o \
1819
mpi-cmp.o \
1920
mpi-sub-ui.o \
21+
mpi-div.o \
22+
mpi-inv.o \
23+
mpi-mod.o \
24+
mpi-mul.o \
2025
mpih-cmp.o \
2126
mpih-div.o \
2227
mpih-mul.o \

‎lib/mpi/mpi-add.c‎

Lines changed: 155 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,155 @@
1+
/* mpi-add.c - MPI functions
2+
* Copyright (C) 1994, 1996, 1998, 2001, 2002,
3+
* 2003 Free Software Foundation, Inc.
4+
*
5+
* This file is part of Libgcrypt.
6+
*
7+
* Note: This code is heavily based on the GNU MP Library.
8+
* Actually it's the same code with only minor changes in the
9+
* way the data is stored; this is to support the abstraction
10+
* of an optional secure memory allocation which may be used
11+
* to avoid revealing of sensitive data due to paging etc.
12+
*/
13+
14+
#include "mpi-internal.h"
15+
16+
/****************
17+
* Add the unsigned integer V to the mpi-integer U and store the
18+
* result in W. U and V may be the same.
19+
*/
20+
void mpi_add_ui(MPI w, MPI u, unsigned long v)
21+
{
22+
mpi_ptr_t wp, up;
23+
mpi_size_t usize, wsize;
24+
int usign, wsign;
25+
26+
usize = u->nlimbs;
27+
usign = u->sign;
28+
wsign = 0;
29+
30+
/* If not space for W (and possible carry), increase space. */
31+
wsize = usize + 1;
32+
if (w->alloced < wsize)
33+
mpi_resize(w, wsize);
34+
35+
/* These must be after realloc (U may be the same as W). */
36+
up = u->d;
37+
wp = w->d;
38+
39+
if (!usize) { /* simple */
40+
wp[0] = v;
41+
wsize = v ? 1:0;
42+
} else if (!usign) { /* mpi is not negative */
43+
mpi_limb_t cy;
44+
cy = mpihelp_add_1(wp, up, usize, v);
45+
wp[usize] = cy;
46+
wsize = usize + cy;
47+
} else {
48+
/* The signs are different. Need exact comparison to determine
49+
* which operand to subtract from which.
50+
*/
51+
if (usize == 1 && up[0] < v) {
52+
wp[0] = v - up[0];
53+
wsize = 1;
54+
} else {
55+
mpihelp_sub_1(wp, up, usize, v);
56+
/* Size can decrease with at most one limb. */
57+
wsize = usize - (wp[usize-1] == 0);
58+
wsign = 1;
59+
}
60+
}
61+
62+
w->nlimbs = wsize;
63+
w->sign = wsign;
64+
}
65+
66+
67+
void mpi_add(MPI w, MPI u, MPI v)
68+
{
69+
mpi_ptr_t wp, up, vp;
70+
mpi_size_t usize, vsize, wsize;
71+
int usign, vsign, wsign;
72+
73+
if (u->nlimbs < v->nlimbs) { /* Swap U and V. */
74+
usize = v->nlimbs;
75+
usign = v->sign;
76+
vsize = u->nlimbs;
77+
vsign = u->sign;
78+
wsize = usize + 1;
79+
RESIZE_IF_NEEDED(w, wsize);
80+
/* These must be after realloc (u or v may be the same as w). */
81+
up = v->d;
82+
vp = u->d;
83+
} else {
84+
usize = u->nlimbs;
85+
usign = u->sign;
86+
vsize = v->nlimbs;
87+
vsign = v->sign;
88+
wsize = usize + 1;
89+
RESIZE_IF_NEEDED(w, wsize);
90+
/* These must be after realloc (u or v may be the same as w). */
91+
up = u->d;
92+
vp = v->d;
93+
}
94+
wp = w->d;
95+
wsign = 0;
96+
97+
if (!vsize) { /* simple */
98+
MPN_COPY(wp, up, usize);
99+
wsize = usize;
100+
wsign = usign;
101+
} else if (usign != vsign) { /* different sign */
102+
/* This test is right since USIZE >= VSIZE */
103+
if (usize != vsize) {
104+
mpihelp_sub(wp, up, usize, vp, vsize);
105+
wsize = usize;
106+
MPN_NORMALIZE(wp, wsize);
107+
wsign = usign;
108+
} else if (mpihelp_cmp(up, vp, usize) < 0) {
109+
mpihelp_sub_n(wp, vp, up, usize);
110+
wsize = usize;
111+
MPN_NORMALIZE(wp, wsize);
112+
if (!usign)
113+
wsign = 1;
114+
} else {
115+
mpihelp_sub_n(wp, up, vp, usize);
116+
wsize = usize;
117+
MPN_NORMALIZE(wp, wsize);
118+
if (usign)
119+
wsign = 1;
120+
}
121+
} else { /* U and V have same sign. Add them. */
122+
mpi_limb_t cy = mpihelp_add(wp, up, usize, vp, vsize);
123+
wp[usize] = cy;
124+
wsize = usize + cy;
125+
if (usign)
126+
wsign = 1;
127+
}
128+
129+
w->nlimbs = wsize;
130+
w->sign = wsign;
131+
}
132+
EXPORT_SYMBOL_GPL(mpi_add);
133+
134+
void mpi_sub(MPI w, MPI u, MPI v)
135+
{
136+
MPI vv = mpi_copy(v);
137+
vv->sign = !vv->sign;
138+
mpi_add(w, u, vv);
139+
mpi_free(vv);
140+
}
141+
142+
143+
void mpi_addm(MPI w, MPI u, MPI v, MPI m)
144+
{
145+
mpi_add(w, u, v);
146+
mpi_mod(w, w, m);
147+
}
148+
EXPORT_SYMBOL_GPL(mpi_addm);
149+
150+
void mpi_subm(MPI w, MPI u, MPI v, MPI m)
151+
{
152+
mpi_sub(w, u, v);
153+
mpi_mod(w, w, m);
154+
}
155+
EXPORT_SYMBOL_GPL(mpi_subm);

0 commit comments

Comments
 (0)