Francois Berder / BigInt

This library provides a way to easily handle arbitrary large integers.

This library provides the following operations :

  • addition, substraction, multiplication, division and modulo
  • bits operators (AND, OR, XOR, left and right shifts)
  • boolean operators
  • modular exponentiation (using montgomery algorithm)
  • modular inverse

Example

In this example, we use a 1024 bits long RSA key to encrypt and decrypt a message. We first encrypt the value 0x41 (65 in decimal) and then decrypt it. At the end, m should be equal to 0x41. The encryption is fast (0, 4 second) while the decryption is really slow. This code will take between 30 seconds and 2 minutes to execute depending on the compiler and optimization flags.

main.cpp

#include "mbed.h"
#include "BigInt.h"
#include <stdlib.h>
#include <stdio.h>

uint8_t modbits[] = {
0xd9, 0x4d, 0x88, 0x9e, 0x88, 0x85, 0x3d, 0xd8, 0x97, 0x69, 0xa1, 0x80, 0x15, 0xa0, 0xa2, 0xe6,
0xbf, 0x82, 0xbf, 0x35, 0x6f, 0xe1, 0x4f, 0x25, 0x1f, 0xb4, 0xf5, 0xe2, 0xdf, 0x0d, 0x9f, 0x9a,
0x94, 0xa6, 0x8a, 0x30, 0xc4, 0x28, 0xb3, 0x9e, 0x33, 0x62, 0xfb, 0x37, 0x79, 0xa4, 0x97, 0xec,
0xea, 0xea, 0x37, 0x10, 0x0f, 0x26, 0x4d, 0x7f, 0xb9, 0xfb, 0x1a, 0x97, 0xfb, 0xf6, 0x21, 0x13,
0x3d, 0xe5, 0x5f, 0xdc, 0xb9, 0xb1, 0xad, 0x0d, 0x7a, 0x31, 0xb3, 0x79, 0x21, 0x6d, 0x79, 0x25,
0x2f, 0x5c, 0x52, 0x7b, 0x9b, 0xc6, 0x3d, 0x83, 0xd4, 0xec, 0xf4, 0xd1, 0xd4, 0x5c, 0xbf, 0x84,
0x3e, 0x84, 0x74, 0xba, 0xbc, 0x65, 0x5e, 0x9b, 0xb6, 0x79, 0x9c, 0xba, 0x77, 0xa4, 0x7e, 0xaf,
0xa8, 0x38, 0x29, 0x64, 0x74, 0xaf, 0xc2, 0x4b, 0xeb, 0x9c, 0x82, 0x5b, 0x73, 0xeb, 0xf5, 0x49
};

uint8_t dbits[] = {
0x04, 0x7b, 0x9c, 0xfd, 0xe8, 0x43, 0x17, 0x6b, 0x88, 0x74, 0x1d, 0x68, 0xcf, 0x09, 0x69, 0x52,
0xe9, 0x50, 0x81, 0x31, 0x51, 0x05, 0x8c, 0xe4, 0x6f, 0x2b, 0x04, 0x87, 0x91, 0xa2, 0x6e, 0x50,
0x7a, 0x10, 0x95, 0x79, 0x3c, 0x12, 0xba, 0xe1, 0xe0, 0x9d, 0x82, 0x21, 0x3a, 0xd9, 0x32, 0x69,
0x28, 0xcf, 0x7c, 0x23, 0x50, 0xac, 0xb1, 0x9c, 0x98, 0xf1, 0x9d, 0x32, 0xd5, 0x77, 0xd6, 0x66,
0xcd, 0x7b, 0xb8, 0xb2, 0xb5, 0xba, 0x62, 0x9d, 0x25, 0xcc, 0xf7, 0x2a, 0x5c, 0xeb, 0x8a, 0x8d,
0xa0, 0x38, 0x90, 0x6c, 0x84, 0xdc, 0xdb, 0x1f, 0xe6, 0x77, 0xdf, 0xfb, 0x2c, 0x02, 0x9f, 0xd8,
0x92, 0x63, 0x18, 0xee, 0xde, 0x1b, 0x58, 0x27, 0x2a, 0xf2, 0x2b, 0xda, 0x5c, 0x52, 0x32, 0xbe,
0x06, 0x68, 0x39, 0x39, 0x8e, 0x42, 0xf5, 0x35, 0x2d, 0xf5, 0x88, 0x48, 0xad, 0xad, 0x11, 0xa1
};

int main() 
{
    BigInt e = 65537, mod, d;
    mod.importData(modbits, sizeof(modbits));
    d.importData(dbits, sizeof(dbits));

    BigInt c = modPow(0x41,e,mod);
    c.print();
    BigInt m = modPow(c,d,mod);
    m.print();
    printf("done\n");
    
    return 0;
}

BigInt.cpp

Committer:
feb11
Date:
2013-10-04
Revision:
6:29e78b169f40
Parent:
5:beeb31f340a7
Child:
7:1aad58757705

File content as of revision 6:29e78b169f40:

#include "BigInt.h"
#include <string.h>
#include <stdio.h>
#include <stdlib.h>
#include <iostream>
#include <climits>


static uint32_t max(const uint32_t a, const uint32_t b)
{
    return a < b ? b : a;
}

static uint32_t num(const uint32_t a)
{
    return a/4 + (a%4 ? 1:0); 
}
    
BigInt::BigInt():
size(0),
bits(0)
{
}

BigInt::BigInt(const uint32_t a)
{
    if(a >> 24)
        size = 4;
    else if(a >> 16)
        size = 3;
    else if(a >> 8)
        size = 2;
    else
        size = 1;
    bits = new uint32_t[1];
    bits[0] = a;
}

BigInt::BigInt(const BigInt &a):
size(a.size)
{
    uint32_t l = num(size);
    bits = new uint32_t[l];
    for(int i = 0; i < l; ++i)
        bits[i] = a.bits[i];
}


BigInt::~BigInt()
{
    if(bits)
        delete[] bits;
    bits = 0;
}

BigInt& BigInt::operator=(const BigInt& a)
{
    size = a.size;
    uint32_t l = num(size);
    if(bits)
        delete[] bits;
    bits = new uint32_t[l];
    for(int i = 0; i < l; ++i)
        bits[i] = a.bits[i];    
    return *this;
}

void BigInt::import(uint8_t *data, uint32_t length)
{
    size = length;
    size_t l = size/4;
    if(size % 4 != 0)
        l++;
    if(bits)
        delete[] bits;
    bits = new uint32_t[l];
    memset(bits, 0, sizeof(uint32_t)*l);
    for(int i = length-1; i >=0; --i)
        bits[i/4] |= data[i] << ((i%4)*8);
}

BigInt operator+(const BigInt &a, const BigInt& b)
{
    BigInt result;
    
    result.size = a.size > b.size ? a.size : b.size;
    size_t l = result.size/4;
    if(result.size % 4 != 0)
        l++;
    result.bits = new uint32_t[l];
    memset(result.bits, 0, sizeof(uint32_t)*l);
    uint32_t al = num(a.size);
    uint32_t bl = num(b.size);
    uint32_t carry = 0;
    for(int i = 0; i < l; ++i)
    {
        uint32_t tmpA = 0, tmpB = 0;
        if(i < al)
            tmpA = a.bits[i];
        if(i < bl)
            tmpB = b.bits[i];
        result.bits[i] = tmpA + tmpB + carry;
        carry = result.bits[i] < max(tmpA, tmpB);
    }
    if(carry)
    {
        result.size++;
        if(result.size > l*4)
        { 
            l++;
            result.bits = (uint32_t*)realloc(result.bits, l * sizeof(uint32_t));
            result.bits[l-1] = 0x00000001;
        }
        else
        {
            result.bits[l-1] += 1 << (8 *((result.size-1)%4));
        }
    }
    
    return result;
}

BigInt& BigInt::operator+=(const BigInt &b)
{
    return (*this = (*this) + b);
}

BigInt& BigInt::operator++()
{
    return (*this += 1);
}

BigInt BigInt::operator++(int)
{
    BigInt t = *this;
    *this += 1;
    return t;
}

// a - b, if b >= a, returns 0
// No negative number allowed
BigInt operator-(const BigInt &a, const BigInt& b)
{
    BigInt result;

    if(b >= a)
    {
        return result = 0;
    }
    else
    {
        result.size = a.size;
        uint32_t l = num(a.size);
        result.bits = new uint32_t[l];
        memset(result.bits, 0, sizeof(uint32_t)*l);
        uint32_t bl = num(b.size);
        uint8_t borrow = 0;
        for(uint32_t i = 0; i < l; ++i)
        {
            uint32_t tmpA = a.bits[i], tmpB = 0;
            if(i < bl)
                tmpB = b.bits[i];
            if(borrow)
            {
                if(tmpA == 0)
                    tmpA = ULONG_MAX;
                else
                    --tmpA;
                    
                if(tmpA < tmpB)            
                    result.bits[i] = tmpA + 1 + (ULONG_MAX - tmpB);
                else
                    result.bits[i] = tmpA - tmpB;

                if(a.bits[i] != 0 && tmpA > tmpB)
                    borrow = 0;
            }
            else
            {
                if(tmpA < tmpB)            
                    result.bits[i] = tmpA + 1 + (ULONG_MAX - tmpB);
                else
                    result.bits[i] = tmpA - tmpB;
                borrow = tmpA < tmpB;
            }
        }
        
        // trim result
        uint8_t *tmp = (uint8_t*)result.bits;
        uint32_t newSize = result.size;
        while(tmp[newSize-1] == 0 && newSize > 0)
            newSize--;
        result.size = newSize;
        uint32_t tmpL = num(result.size);
        if(tmpL < l)
            result.bits = (uint32_t*)realloc(result.bits, sizeof(uint32_t)*tmpL); 
        
        return result;
    }
}

BigInt& BigInt::operator-=(const BigInt &b)
{
    return (*this = (*this) - b);
}

BigInt& BigInt::operator--()
{
    return (*this -= 1);
}

BigInt BigInt::operator--(int)
{
    BigInt t = *this;
    *this -= 1;
    return t;
}

BigInt operator*(const BigInt &a, const BigInt& b)
{
    BigInt result;   
    
    // if a == 0 or b == 0 then result = 0
    if(!a || !b)
        return result = 0;
    
    // if a == 1, then result = b
    if(a == 1)
        return (result = b);
    
    // if b == 1, then result = a
    if(b == 1)
        return (result = a);
        
    
    result.size = a.size + b.size;
    result.bits = new uint32_t[num(result.size)];
    memset(result.bits, 0, sizeof(uint32_t)*num(result.size));
    uint32_t carry = 0;
    for(int i = 0; i < num(result.size); ++i)
    {
        uint32_t tmpA = 0, tmpB = 0;
        if(i < num(a.size))
            tmpA = a.bits[i];

        if(i < num(b.size))
            tmpB = b.bits[i];
 
        uint64_t tmp = (uint64_t)tmpA * (uint64_t)tmpB + (uint64_t)carry;        
        result.bits[i] = tmp;
        carry = tmp >> 32;
    }
    
    // trim result
    uint8_t *tmp = (uint8_t*)result.bits;
    uint32_t newSize = result.size;
    while(tmp[newSize-1] == 0 && newSize > 0)
        newSize--;
    if(num(newSize) < num(result.size))
        result.bits = (uint32_t*)realloc(result.bits, sizeof(uint32_t)*num(newSize)); 
    result.size = newSize;
    
    return result;
}

BigInt& BigInt::operator*=(const BigInt &b)
{
    return (*this = (*this) * b);
}

BigInt operator>>(const BigInt &a, const uint32_t m)
{
    BigInt result;
    if(m == 0)
        return result = a;
    if(m/8 >= a.size)
        return result = 0;
    
    result.size = a.size - m/8;
    result.bits = new uint32_t[num(result.size)];
    uint8_t s = m%32;
    for(uint32_t i = 0; i < num(result.size); ++i)
    {
        if(m/32+i+1 < num(a.size))
            result.bits[i] =  (a.bits[m/32+i+1] << (32-s)) |  (a.bits[m/32+i] >> s);
        else
            result.bits[i] = (a.bits[m/32+i] >> s);
    }
    
    
    return result;
}

BigInt& BigInt::operator>>=(const uint32_t m)
{
    return *this = *this >> m;
}
 
BigInt operator<<(const BigInt &a, const uint32_t m)
{
    BigInt result;

    if(m == 0)
        return result = a;

    result.size = m/8 + a.size;
    uint32_t h = a.bits[num(a.size)-1];
    if((h << (m%32)) < h)
        ++result.size;
    uint32_t l = num(result.size);
    result.bits = new uint32_t[l];
    memset(result.bits, 0, sizeof(uint32_t)*l);
    uint32_t s = m % 32;
    for(uint32_t i = 0; i < num(a.size); ++i)
    {
        if(i == 0)
            result.bits[m/32+i] = a.bits[i] << s;
        else
            result.bits[m/32+i] = (a.bits[i] << s) | (a.bits[i-1] >> (32-s));
    }
    if(a.bits[num(a.size)-1] << s < a.bits[num(a.size)-1])
        result.bits[num(result.size)-1] = a.bits[num(a.size)-1] >> (32-s);

    
    return result;
}

BigInt& BigInt::operator<<=(const uint32_t m)
{
    return (*this = *this << m);
}

BigInt operator%(const BigInt &a, const BigInt &b)
{
    BigInt result = a;
    while(result > b)
        result -= b;

    // trim result
    uint8_t *tmp = (uint8_t*)result.bits;
    uint32_t newSize = result.size;
    while(tmp[newSize-1] == 0 && newSize > 0)
        newSize--;
    if(num(newSize) < num(result.size))
        result.bits = (uint32_t*)realloc(result.bits, sizeof(uint32_t)*num(newSize)); 
    result.size = newSize;
    
    return result;
}

BigInt& BigInt::operator%=(const BigInt &a)
{
    return (*this = *this % a);
}

bool operator==(const BigInt &a, const BigInt &b)
{
    if(a.size != b.size)
        return false;
        
    uint32_t l = num(a.size);
    for(int i = 0; i < l; ++i)
        if(a.bits[i] != b.bits[i])
            return false;
    return true;
}

bool operator!=(const BigInt &a, const BigInt &b)
{
    return ! (a==b);
}

bool operator<(const BigInt &a, const BigInt &b)
{
    if(a.size < b.size)
        return true;
    if(a.size > b.size)
        return false;
    uint32_t l = num(a.size);
    for(int i = l-1; i >= 0; --i)
        if(a.bits[i] < b.bits[i])
            return true;
    return false;
}

bool operator<=(const BigInt &a, const BigInt &b)
{
    return (a < b) || (a == b);
}

bool operator>(const BigInt &a, const BigInt &b)
{
    if(a.size > b.size)
        return true;
    if(a.size < b.size)
        return false;
    uint32_t l = num(a.size);
    for(int i = l-1; i >= 0; --i)
        if(a.bits[i] > b.bits[i])
            return true;
    return false;
}

bool operator>=(const BigInt &a, const BigInt &b)
{
    return (a > b) || (a == b);
}

bool operator!(const BigInt &a)
{
    if(a.size != 1)
        return false;
    return a.bits[0] == 0;
}
 
void BigInt::print()
{
    printf("size: %d bytes\n", size);
    uint32_t n = num(size);
    for(int i = n-1; i >= 0; --i)
    {
        printf("%08x ", bits[i]);
    }
    printf("\n");
}