diff --git a/CMakeLists.txt b/CMakeLists.txt index e47e3e6..434cca9 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -13,4 +13,5 @@ project(romulus ) set(CMAKE_POSITION_INDEPENDENT_CODE on) +add_subdirectory(ISAAC/) add_subdirectory(Romulus-M/) \ No newline at end of file diff --git a/ISAAC/CMakeLists.txt b/ISAAC/CMakeLists.txt new file mode 100644 index 0000000..8fe90ed --- /dev/null +++ b/ISAAC/CMakeLists.txt @@ -0,0 +1,6 @@ +set(SRC + rand.c +) + +add_library(isaac STATIC ${SRC}) + diff --git a/ISAAC/rand.c b/ISAAC/rand.c new file mode 100644 index 0000000..63babc7 --- /dev/null +++ b/ISAAC/rand.c @@ -0,0 +1,137 @@ +/* +------------------------------------------------------------------------------ +rand.c: By Bob Jenkins. My random number generator, ISAAC. Public Domain. +MODIFIED: + 960327: Creation (addition of randinit, really) + 970719: use context, not global variables, for internal state + 980324: added main (ifdef'ed out), also rearranged randinit() + 010626: Note that this is public domain +------------------------------------------------------------------------------ +*/ +#ifndef STANDARD +#include "standard.h" +#endif +#ifndef RAND +#include "rand.h" +#endif + + +#define ind(mm,x) (*(ub4 *)((ub1 *)(mm) + ((x) & ((RANDSIZ-1)<<2)))) +#define rngstep(mix,a,b,mm,m,m2,r,x) \ +{ \ + x = *m; \ + a = (a^(mix)) + *(m2++); \ + *(m++) = y = ind(mm,x) + a + b; \ + *(r++) = b = ind(mm,y>>RANDSIZL) + x; \ +} + +void isaac(ctx) +randctx *ctx; +{ + register ub4 a,b,x,y,*m,*mm,*m2,*r,*mend; + mm=ctx->randmem; r=ctx->randrsl; + a = ctx->randa; b = ctx->randb + (++ctx->randc); + for (m = mm, mend = m2 = m+(RANDSIZ/2); m>6 , a, b, mm, m, m2, r, x); + rngstep( a<<2 , a, b, mm, m, m2, r, x); + rngstep( a>>16, a, b, mm, m, m2, r, x); + } + for (m2 = mm; m2>6 , a, b, mm, m, m2, r, x); + rngstep( a<<2 , a, b, mm, m, m2, r, x); + rngstep( a>>16, a, b, mm, m, m2, r, x); + } + ctx->randb = b; ctx->randa = a; +} + + +#define mix(a,b,c,d,e,f,g,h) \ +{ \ + a^=b<<11; d+=a; b+=c; \ + b^=c>>2; e+=b; c+=d; \ + c^=d<<8; f+=c; d+=e; \ + d^=e>>16; g+=d; e+=f; \ + e^=f<<10; h+=e; f+=g; \ + f^=g>>4; a+=f; g+=h; \ + g^=h<<8; b+=g; h+=a; \ + h^=a>>9; c+=h; a+=b; \ +} + +/* if (flag==TRUE), then use the contents of randrsl[] to initialize mm[]. */ +void randinit(ctx, flag) +randctx *ctx; +word flag; +{ + word i; + ub4 a,b,c,d,e,f,g,h; + ub4 *m,*r; + ctx->randa = ctx->randb = ctx->randc = 0; + m=ctx->randmem; + r=ctx->randrsl; + a=b=c=d=e=f=g=h=0x9e3779b9; /* the golden ratio */ + + for (i=0; i<4; ++i) /* scramble it */ + { + mix(a,b,c,d,e,f,g,h); + } + + if (flag) + { + /* initialize using the contents of r[] as the seed */ + for (i=0; irandcnt=RANDSIZ; /* prepare to use the first set of results */ +} + + +#ifdef NEVER +int main() +{ + ub4 i,j; + randctx ctx; + ctx.randa=ctx.randb=ctx.randc=(ub4)0; + for (i=0; i<256; ++i) ctx.randrsl[i]=(ub4)0; + randinit(&ctx, TRUE); + for (i=0; i<2; ++i) + { + isaac(&ctx); + for (j=0; j<256; ++j) + { + printf("%.8lx",ctx.randrsl[j]); + if ((j&7)==7) printf("\n"); + } + } +} +#endif diff --git a/ISAAC/rand.h b/ISAAC/rand.h new file mode 100644 index 0000000..73f6e3e --- /dev/null +++ b/ISAAC/rand.h @@ -0,0 +1,56 @@ +/* +------------------------------------------------------------------------------ +rand.h: definitions for a random number generator +By Bob Jenkins, 1996, Public Domain +MODIFIED: + 960327: Creation (addition of randinit, really) + 970719: use context, not global variables, for internal state + 980324: renamed seed to flag + 980605: recommend RANDSIZL=4 for noncryptography. + 010626: note this is public domain +------------------------------------------------------------------------------ +*/ +#ifndef STANDARD +#include "standard.h" +#endif + +#ifndef RAND +#define RAND +#define RANDSIZL (8) +#define RANDSIZ (1<randcnt-- ? \ + (isaac(r), (r)->randcnt=RANDSIZ-1, (r)->randrsl[(r)->randcnt]) : \ + (r)->randrsl[(r)->randcnt]) + +#endif /* RAND */ + + diff --git a/ISAAC/standard.h b/ISAAC/standard.h new file mode 100644 index 0000000..202a5d6 --- /dev/null +++ b/ISAAC/standard.h @@ -0,0 +1,57 @@ +/* +------------------------------------------------------------------------------ +Standard definitions and types, Bob Jenkins +------------------------------------------------------------------------------ +*/ +#ifndef STANDARD +# define STANDARD +# ifndef STDIO +# include +# define STDIO +# endif +# ifndef STDDEF +# include +# define STDDEF +# endif +typedef unsigned long long ub8; +#define UB8MAXVAL 0xffffffffffffffffLL +#define UB8BITS 64 +typedef signed long long sb8; +#define SB8MAXVAL 0x7fffffffffffffffLL +typedef unsigned long int ub4; /* unsigned 4-byte quantities */ +#define UB4MAXVAL 0xffffffff +typedef signed long int sb4; +#define UB4BITS 32 +#define SB4MAXVAL 0x7fffffff +typedef unsigned short int ub2; +#define UB2MAXVAL 0xffff +#define UB2BITS 16 +typedef signed short int sb2; +#define SB2MAXVAL 0x7fff +typedef unsigned char ub1; +#define UB1MAXVAL 0xff +#define UB1BITS 8 +typedef signed char sb1; /* signed 1-byte quantities */ +#define SB1MAXVAL 0x7f +typedef int word; /* fastest type available */ + +#define bis(target,mask) ((target) |= (mask)) +#define bic(target,mask) ((target) &= ~(mask)) +#define bit(target,mask) ((target) & (mask)) +#ifndef min +# define min(a,b) (((a)<(b)) ? (a) : (b)) +#endif /* min */ +#ifndef max +# define max(a,b) (((a)<(b)) ? (b) : (a)) +#endif /* max */ +#ifndef align +# define align(a) (((ub4)a+(sizeof(void *)-1))&(~(sizeof(void *)-1))) +#endif /* align */ +#ifndef abs +# define abs(a) (((a)>0) ? (a) : -(a)) +#endif +#define TRUE 1 +#define FALSE 0 +#define SUCCESS 0 /* 1 on VAX */ + +#endif /* STANDARD */ diff --git a/Romulus-M/CMakeLists.txt b/Romulus-M/CMakeLists.txt index 8a7555f..98e0c29 100644 --- a/Romulus-M/CMakeLists.txt +++ b/Romulus-M/CMakeLists.txt @@ -6,5 +6,6 @@ set(SRC encrypt_wrapper.c ) +include_directories(../ISAAC) +link_libraries(isaac) add_library(romulus-m STATIC ${SRC}) -install(TARGETS romulus-m) \ No newline at end of file diff --git a/Romulus-M/encrypt_wrapper.c b/Romulus-M/encrypt_wrapper.c index f84814c..4d969a4 100644 --- a/Romulus-M/encrypt_wrapper.c +++ b/Romulus-M/encrypt_wrapper.c @@ -3,64 +3,67 @@ #include #ifdef _MSC_VER -#define _CRT_RAND_S #define __bswap_16 _byteswap_ushort #define __bswap_32 _byteswap_ulong #else #include -#include #endif - #include #include "romulus_m.h" #include "api.h" +#include +#ifdef _DEBUG +#define DEBUG_PRINT printf +#else +#define DEBUG_PRINT +#endif -int encrypt(packet *packet, unsigned char* outBuf, UINT16 *len, unsigned char* key){ - if(sizeof(UINT16) + sizeof(UINT16) + packet->Size > *len){ +#define MIN(a,b) ((a) < (b) ? (a) : (b)) + +randctx rCtx = {0}; + +int encrypt(packet *packet, unsigned char *outBuf, UINT16 *len, unsigned char *key) +{ + randinit(); + if (sizeof(UINT16) + sizeof(UINT16) + packet->Size > *len) + { return 1; } - int packetSize = packet->Size; + int packetSize = packet->Size; // enc // ciphertext - out, MAX 16 bytes larger than plaintext - defined as CRYPTO_ABYTES unsigned long long clen = packetSize + CRYPTO_ABYTES; - unsigned char* c = (unsigned char*) calloc(clen, 1); - + unsigned char *c = (unsigned char *)calloc(clen, 1); + // plaintext - in int mlen = packetSize; - unsigned char* m = packet->Data; - + unsigned char *m = packet->Data; + // additional text int adlen = sizeof(packet->DataType); - unsigned char* ad = (unsigned char*)&packet->DataType; + unsigned char *ad = (unsigned char *)&packet->DataType; // n = nonce, CRYPTO_NPUBBYTES size 16 unsigned char npub[CRYPTO_NPUBBYTES]; - #ifdef _MSC_VER - UINT32 secRandom; - for (size_t i = 0; i < CRYPTO_NPUBBYTES; i+=sizeof(UINT32)) - { - rand_s(&secRandom); - memcpy(&npub[i], &secRandom, sizeof(UINT32)); - } - #else - getrandom(&npub[0], sizeof(npub), 0); - #endif + + fill_random(npub, CRYPTO_NPUBBYTES); int ret = romulus_m_encrypt(c, &clen, m, mlen, ad, adlen, 0, &npub[0], key); - - if(clen > *len + CRYPTO_NPUBBYTES || ret != 0){ + + if (clen > *len + CRYPTO_NPUBBYTES || ret != 0) + { free(c); return 1; } - + // type 0 1 // Swap after adding it as additional data - packet->DataType = __bswap_16(packet->DataType); + packet->DataType = __bswap_16(packet->DataType); int loc = 0; - memcpy(&outBuf[loc], &packet->DataType,sizeof(UINT16)); + memcpy(&outBuf[loc], &packet->DataType, sizeof(UINT16)); loc += sizeof(UINT16); // data size 2 3 @@ -84,11 +87,14 @@ int encrypt(packet *packet, unsigned char* outBuf, UINT16 *len, unsigned char* k return 0; } -int decrypt(packetEx *packet, unsigned char* inBuf, UINT16 len, unsigned char* key){ +int decrypt(packetEx *packet, unsigned char *inBuf, UINT16 len, unsigned char *key) +{ int loc = 0; - if(inBuf == 0 ){ + if (inBuf == 0) + { return 1; } + DEBUG_PRINT("line %d\n", 95); memcpy(&packet->DataType, &inBuf[loc], sizeof(UINT16)); loc += sizeof(UINT16); memcpy(&packet->Id, &inBuf[loc], sizeof(UINT32)); @@ -100,38 +106,52 @@ int decrypt(packetEx *packet, unsigned char* inBuf, UINT16 len, unsigned char* k packet->Size = __bswap_16(packet->Size); packet->Id = __bswap_32(packet->Id); - if(packet->Size < CRYPTO_NPUBBYTES){ + if ((len - loc) < 0 || packet->Size < CRYPTO_NPUBBYTES || packet->Size > 1000) + { return 1; } + DEBUG_PRINT("line %d\n", 111); // Data // Nonce|Data // dec - // n = nonce, CRYPTO_NPUBBYTES size 16 - unsigned char* npub = &inBuf[loc]; + // n = nonce, CRYPTO_NPUBBYTES size 16 + unsigned char *npub = &inBuf[loc]; loc += CRYPTO_NPUBBYTES; - // ciphertext - out, MAX 16 bytes larger than plaintext - defined as CRYPTO_ABYTES - int clen = packet->Size - CRYPTO_NPUBBYTES; - unsigned char* c = &inBuf[loc]; - - // plaintext - in + // ciphertext - in, MAX 16 bytes larger than plaintext - defined as CRYPTO_ABYTES + int clen = len - loc; + unsigned char *c = &inBuf[loc]; + DEBUG_PRINT("line %d len:%d loc:%d\n", 132, len, loc); + + // plaintext - out unsigned long long mlen = clen; - unsigned char* m = (unsigned char*)calloc(mlen, 1); - + unsigned char *m = (unsigned char *)calloc(clen, 1); + // additional text int adlen = sizeof(packet->DataType); - unsigned char* ad = (unsigned char*)&packet->DataType; + unsigned char *ad = (unsigned char *)&packet->DataType; - int ret = romulus_m_decrypt(m, &mlen, 0, c, clen, ad, adlen, npub, key); - if(mlen <= 1000 && ret == 0){ + DEBUG_PRINT("line %d mlen:%d clen:%d\n", 132, mlen, clen); + int ret = romulus_m_decrypt(m, &mlen, 0, c, clen, ad, adlen, npub, key); + DEBUG_PRINT("line %d packet:%d, mlen %d\n", 134, packet->Size, mlen); + if (ret == 0) + { + DEBUG_PRINT("line %d\n", 138); memcpy(packet->Data, m, mlen); packet->Size = (UINT16)mlen; } - else{ - free(m); - return -1; - } + free(m); - return 0; + DEBUG_PRINT("line %d\n", 143); + return ret; +} + +void fill_random(unsigned char* buffer, int length){ + UINT32 secRandom; + for (size_t i = 0; i < length; i += sizeof(UINT32)) + { + secRandom = rand(&rCtx); + memcpy(&buffer[i], &secRandom, MIN(sizeof(UINT32), length-i)); + } } \ No newline at end of file diff --git a/Romulus-M/encrypt_wrapper.h b/Romulus-M/encrypt_wrapper.h index ca8ccce..50d0634 100644 --- a/Romulus-M/encrypt_wrapper.h +++ b/Romulus-M/encrypt_wrapper.h @@ -26,4 +26,6 @@ typedef struct { EXPORT int encrypt(packet *packet, unsigned char* outBuf, UINT16 *len, unsigned char* key); -EXPORT int decrypt(packetEx *packet, unsigned char* inBuf, UINT16 len, unsigned char* key); \ No newline at end of file +EXPORT int decrypt(packetEx *packet, unsigned char* inBuf, UINT16 len, unsigned char* key); + +void fill_random(unsigned char* buffer, int length); \ No newline at end of file