diff options
Diffstat (limited to '')
| -rw-r--r-- | mpi/mpicoder.c | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/mpi/mpicoder.c b/mpi/mpicoder.c index f796888c8..8cb1538c0 100644 --- a/mpi/mpicoder.c +++ b/mpi/mpicoder.c @@ -332,12 +332,15 @@ do_get_buffer( MPI a, unsigned *nbytes, int *sign, int force_secure ) byte *p, *buffer; mpi_limb_t alimb; int i; + unsigned int n; if( sign ) *sign = a->sign; - *nbytes = a->nlimbs * BYTES_PER_MPI_LIMB; - p = buffer = force_secure || mpi_is_secure(a) ? m_alloc_secure( *nbytes) - : m_alloc( *nbytes ); + *nbytes = n = a->nlimbs * BYTES_PER_MPI_LIMB; + if (!n) + n++; /* avoid zero length allocation */ + p = buffer = force_secure || mpi_is_secure(a) ? m_alloc_secure(n) + : m_alloc(n); for(i=a->nlimbs-1; i >= 0; i-- ) { alimb = a->d[i]; |