TeaSpeak/tommath
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

+ allow mp_shrink to shrink initialized, but empty MPI's

Browse Source
This commit is contained in:
Steffen Jaeckel 2010-06-02 15:09:36 +02:00
parent eec4e5c546
commit 7d14fd029c
2 changed files with 16 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
bn_mp_shrink.c
View File

@ -19,12 +19,17 @@
int mp_shrink (mp_int * a) int mp_shrink (mp_int * a)
{ {
mp_digit *tmp; mp_digit *tmp;
if (a->alloc != a->used && a->used > 0) { int used = 1;
if ((tmp = OPT_CAST(mp_digit) XREALLOC (a->dp, sizeof (mp_digit) * a->used)) == NULL) {
if(a->used > 0)
used = a->used;
if (a->alloc != used) {
if ((tmp = OPT_CAST(mp_digit) XREALLOC (a->dp, sizeof (mp_digit) * used)) == NULL) {
return MP_MEM; return MP_MEM;
} }
a->dp = tmp; a->dp = tmp;
a->alloc = a->used; a->alloc = used;
} }
return MP_OKAY; return MP_OKAY;
} }

11
pre_gen/mpi.c
View File

@ -7217,12 +7217,17 @@ int mp_set_int (mp_int * a, unsigned long b)
int mp_shrink (mp_int * a) int mp_shrink (mp_int * a)
{ {
mp_digit *tmp; mp_digit *tmp;
if (a->alloc != a->used && a->used > 0) { int used = 1;
if ((tmp = OPT_CAST(mp_digit) XREALLOC (a->dp, sizeof (mp_digit) * a->used)) == NULL) {
if(a->used > 0)
used = a->used;
if (a->alloc != used) {
if ((tmp = OPT_CAST(mp_digit) XREALLOC (a->dp, sizeof (mp_digit) * used)) == NULL) {
return MP_MEM; return MP_MEM;
} }
a->dp = tmp; a->dp = tmp;
a->alloc = a->used; a->alloc = used;
} }
return MP_OKAY; return MP_OKAY;
} }