new ida

[Cipher_code.git] / OneRoundIoT / openssl / openssl_evp.c

diff --git a/OneRoundIoT/openssl/openssl_evp.c b/OneRoundIoT/openssl/openssl_evp.c
index 3c1416ffb16a0a1c8c78d053717aaa7cb4d6307a..0a4b9497cb5af73c3638dd7be5eb43e4117fa7d8 100644 (file)
--- a/OneRoundIoT/openssl/openssl_evp.c
+++ b/OneRoundIoT/openssl/openssl_evp.c
@@ -16,6 +16,9 @@ typedef unsigned char   uchar;
 int nb_test=1;
 int ctr=0;
 
 int nb_test=1;
 int ctr=0;
 

+double time_encrypt=0;
+double time_decrypt=0;
+

 double TimeStart()
 {
   struct timeval tstart;
 double TimeStart()
 {
   struct timeval tstart;


@@ -41,7 +44,7 @@ void handleErrors(void)
 
 
 int encrypt(unsigned char *plaintext, int plaintext_len, unsigned char *key,
 
 
 int encrypt(unsigned char *plaintext, int plaintext_len, unsigned char *key,

-           unsigned char *iv, unsigned char *ciphertext, int ctr)
+           unsigned char *iv, unsigned char *ciphertext, int ctr, int index)

 {
   EVP_CIPHER_CTX *ctx;
 
 {
   EVP_CIPHER_CTX *ctx;
 


@@ -49,6 +52,9 @@ int encrypt(unsigned char *plaintext, int plaintext_len, unsigned char *key,
 
   int ciphertext_len;
 
 
   int ciphertext_len;
 

+
+
+  

   /* Create and initialise the context */
   if(!(ctx = EVP_CIPHER_CTX_new())) handleErrors();
 
   /* Create and initialise the context */
   if(!(ctx = EVP_CIPHER_CTX_new())) handleErrors();
 


@@ -58,39 +64,46 @@ int encrypt(unsigned char *plaintext, int plaintext_len, unsigned char *key,
    * IV size for *most* modes is the same as the block size. For AES this
    * is 128 bits */
 
    * IV size for *most* modes is the same as the block size. For AES this
    * is 128 bits */
 

+  double t=TimeStart();
+  

   //256
   //avant ecb
   //256
   //avant ecb

-  if(ctr)
-    if(1 != EVP_EncryptInit_ex(ctx, EVP_aes_128_ctr(), NULL, key, iv))
-      handleErrors();
+  for(int i=0;i<nb_test;i++)
+  {  
+    
+    if(ctr) {
+      if(1 != EVP_EncryptInit_ex(ctx, EVP_aes_128_ctr(), NULL, key, iv))
+       handleErrors();
+    }

     else
       if(1 != EVP_EncryptInit_ex(ctx, EVP_aes_128_cbc(), NULL, key, iv))
        handleErrors();
     else
       if(1 != EVP_EncryptInit_ex(ctx, EVP_aes_128_cbc(), NULL, key, iv))
        handleErrors();

-
+    
+    //time+=TimeStop(t);
+    //printf("Time init %f\n",time);
+    
+    

 //  int cipherBlockSize = EVP_CIPHER_CTX_block_size(ctx);  
 //  printf("INFO(evp_encrypt): block size: %d\n", cipherBlockSize);
 //  int cipherBlockSize = EVP_CIPHER_CTX_block_size(ctx);  
 //  printf("INFO(evp_encrypt): block size: %d\n", cipherBlockSize);

-
-  
-  /* Provide the message to be encrypted, and obtain the encrypted output.
-   * EVP_EncryptUpdate can be called multiple times if necessary
-   */
-
-  if(1 != EVP_EncryptUpdate(ctx, ciphertext, &len, plaintext, plaintext_len))
-    handleErrors();
-  ciphertext_len = len;
-
+    
+    
+    /* Provide the message to be encrypted, and obtain the encrypted output.
+     * EVP_EncryptUpdate can be called multiple times if necessary
+     */
+    
+    if(1 != EVP_EncryptUpdate(ctx, ciphertext, &len, plaintext, plaintext_len))
+      handleErrors();
+    ciphertext_len = len;
+    
+    /* Finalise the encryption. Further ciphertext bytes may be written at
+     * this stage.
+     */
+    if(1 != EVP_EncryptFinal_ex(ctx, ciphertext + len, &len)) handleErrors();
+    ciphertext_len += len;
+  }

   
   

-
-
-
-
+  time_encrypt+=TimeStop(t);

   
   

-  /* Finalise the encryption. Further ciphertext bytes may be written at
-   * this stage.
-   */
-  if(1 != EVP_EncryptFinal_ex(ctx, ciphertext + len, &len)) handleErrors();
-  ciphertext_len += len;
-

   /* Clean up */
   EVP_CIPHER_CTX_free(ctx);
 
   /* Clean up */
   EVP_CIPHER_CTX_free(ctx);
 


@@ -98,7 +111,7 @@ int encrypt(unsigned char *plaintext, int plaintext_len, unsigned char *key,
 }
 
 int decrypt(unsigned char *ciphertext, int ciphertext_len, unsigned char *key,
 }
 
 int decrypt(unsigned char *ciphertext, int ciphertext_len, unsigned char *key,

-           unsigned char *iv, unsigned char *plaintext, int ctr)
+           unsigned char *iv, unsigned char *plaintext, int ctr, int index)

 {
   EVP_CIPHER_CTX *ctx;
 
 {
   EVP_CIPHER_CTX *ctx;
 


@@ -117,11 +130,17 @@ int decrypt(unsigned char *ciphertext, int ciphertext_len, unsigned char *key,
 
   //256
 
 
   //256
 

+  double t=TimeStart();
+  
+  for(int i=0;i<nb_test;i++)
+  {  
+  

   //avant => ecb
   //avant => ecb

-  if(ctr)
+  if(ctr) {

     if(1 != EVP_DecryptInit_ex(ctx, EVP_aes_128_ctr(), NULL, key, iv))
       handleErrors();
     if(1 != EVP_DecryptInit_ex(ctx, EVP_aes_128_ctr(), NULL, key, iv))
       handleErrors();

-    else
+  }
+  else

       if(1 != EVP_DecryptInit_ex(ctx, EVP_aes_128_cbc(), NULL, key, iv))
     handleErrors();
 
       if(1 != EVP_DecryptInit_ex(ctx, EVP_aes_128_cbc(), NULL, key, iv))
     handleErrors();
 


@@ -129,17 +148,32 @@ int decrypt(unsigned char *ciphertext, int ciphertext_len, unsigned char *key,
    * EVP_DecryptUpdate can be called multiple times if necessary
    */
   
    * EVP_DecryptUpdate can be called multiple times if necessary
    */
   

- 
-  if(1 != EVP_DecryptUpdate(ctx, plaintext, &len, ciphertext, ciphertext_len))
-    handleErrors();
-  plaintext_len = len;
+/*  static double time=0;
+  double t=0;
+  t=TimeStart();
+*/

 
 

+    plaintext_len = 0;
+    if(1 != EVP_DecryptUpdate(ctx, plaintext, &len, ciphertext, ciphertext_len))
+      handleErrors();
+    plaintext_len = len;
+
+/*  time+=TimeStop(t);
+//  if(index==nb_test-1)
+    printf("Time decrypt %f\n",time);
+*/
+
+  

   /* Finalise the decryption. Further plaintext bytes may be written at
    * this stage.
    */
   if(1 != EVP_DecryptFinal_ex(ctx, plaintext + len, &len)) handleErrors();
   plaintext_len += len;
   /* Finalise the decryption. Further plaintext bytes may be written at
    * this stage.
    */
   if(1 != EVP_DecryptFinal_ex(ctx, plaintext + len, &len)) handleErrors();
   plaintext_len += len;

+  }
+

 
 

+  time_decrypt+=TimeStop(t);
+  

   /* Clean up */
   EVP_CIPHER_CTX_free(ctx);
 
   /* Clean up */
   EVP_CIPHER_CTX_free(ctx);
 


@@ -153,15 +187,24 @@ int main (int argc, char** argv)
    * real application? :-)
    */
 
    * real application? :-)
    */
 

-   for(int i=1; i<argc; i++){
+  int size_buf=1;
+  int lena=0;
+
+   
+  for(int i=1; i<argc; i++){

     if(strncmp(argv[i],"nb",2)==0)    nb_test = atoi(&(argv[i][2]));    //nb of test         
     if(strncmp(argv[i],"ctr",3)==0) ctr = atoi(&(argv[i][3]));          //CTR ? 1  otherwise CBC like
     if(strncmp(argv[i],"nb",2)==0)    nb_test = atoi(&(argv[i][2]));    //nb of test         
     if(strncmp(argv[i],"ctr",3)==0) ctr = atoi(&(argv[i][3]));          //CTR ? 1  otherwise CBC like

+    if(strncmp(argv[i],"sizebuf",7)==0) size_buf = atoi(&(argv[i][7]));          //SIZE of the buffer
+    if(strncmp(argv[i],"lena",4)==0) lena = atoi(&(argv[i][4]));          //Use Lena or buffer

   }
 
   }
 

-  printf("nb times %d\n",nb_test);
+/*  printf("nb times %d\n",nb_test);

   printf("ctr %d\n",ctr);
   printf("ctr %d\n",ctr);

+  printf("lena %d\n",lena);
+  printf("size_buf %d\n",size_buf);
+*/
+

 
 

-  

 
   
   /* A 256 bit key */
 
   
   /* A 256 bit key */


@@ -181,29 +224,49 @@ int main (int argc, char** argv)
   int width;
   int height;
   uchar *data_R, *data_G, *data_B;
   int width;
   int height;
   uchar *data_R, *data_G, *data_B;

-  load_RGB_pixmap("lena.ppm", &width, &height, &data_R, &data_G, &data_B);
+  int imsize;
+  uchar *buffer;

 
 
 
 

+  if(lena==1) {
+    load_RGB_pixmap("lena.ppm", &width, &height, &data_R, &data_G, &data_B);
+    imsize=width*height*3;

 //  load_RGB_pixmap("No_ecb_mode_picture.ppm", &width, &height, &data_R, &data_G, &data_B);
 //  load_RGB_pixmap("No_ecb_mode_picture.ppm", &width, &height, &data_R, &data_G, &data_B);

-//  load_RGB_pixmap("4096.ppm", &width, &height, &data_R, &data_G, &data_B);
-  int size=width*height*3;
+  }
+  else {
+    width=size_buf;
+    height=size_buf;
+    imsize=width*height;
+    buffer=malloc(imsize*sizeof(uchar));
+    for(int i=0;i<imsize;i++) {
+      buffer[i]=lrand48();
+    }
+  }
+  

 
 

-  int oneD=width*height;
-  uchar *plaintext = malloc(size);

 
 

-  
-  for(int i=0;i<oneD;i++) {
-    plaintext[i]=data_R[i];
-    plaintext[oneD+i]=data_G[i];
-    plaintext[2*oneD+i]=data_B[i];
+  int oneD=width*height;
+  uchar *plaintext = malloc(imsize+1000);   //add that for cbc
+  if(lena) {
+    for(int i=0;i<oneD;i++) {
+      plaintext[i]=data_R[i];
+      plaintext[oneD+i]=data_G[i];
+      plaintext[2*oneD+i]=data_B[i];
+    }
+  }
+  else
+  {
+     for(int i=0;i<oneD;i++) {
+       plaintext[i]=buffer[i];
+    }

   }
 
   
 
   }
 
   
 

-  uchar *ciphertext = malloc(size);
+  uchar *ciphertext = malloc(imsize+1000); //add that for cbc

 
   /* Buffer for the decrypted text */
 
   /* Buffer for the decrypted text */

-  uchar *decryptedtext = malloc(size);
+  uchar *decryptedtext = malloc(imsize+1000); //add that for cbc

 
   int decryptedtext_len, ciphertext_len;
 
 
   int decryptedtext_len, ciphertext_len;
 


@@ -212,8 +275,10 @@ int main (int argc, char** argv)
   OpenSSL_add_all_algorithms();
   OPENSSL_config(NULL);
 */
   OpenSSL_add_all_algorithms();
   OPENSSL_config(NULL);
 */

-   double time=0;
-  double t=TimeStart();
+
+
+
+

 
   
   /* Encrypt the plaintext */
 
   
   /* Encrypt the plaintext */


@@ -221,48 +286,55 @@ int main (int argc, char** argv)
 
   int i;
 
 
   int i;
 

-  for(i=0;i<nb_test;i++)
+//  for(i=0;i<nb_test;i++)

   {  
   {  

-    ciphertext_len = encrypt (plaintext, size, key, iv,
-                             ciphertext, ctr);
+    ciphertext_len = encrypt (plaintext, imsize, key, iv,
+                             ciphertext, ctr, i );

   }
 
   }
 

- time+=TimeStop(t);

 
 

- printf("Time encrypt %f\n",time);

 
 

+// printf("Time encrypt %f\n",time);
+ printf("%e\t",(double)imsize*nb_test/time_encrypt);

 
 

- for(int i=0;i<oneD;i++) {
-    data_R[i]=ciphertext[i];
-    data_G[i]=ciphertext[oneD+i];
-    data_B[i]=ciphertext[2*oneD+i];
-  }
-  store_RGB_pixmap("lena2.ppm", data_R, data_G, data_B, width, height);                   
+ if(lena) {
+   for(int i=0;i<oneD;i++) {
+     data_R[i]=ciphertext[i];
+     data_G[i]=ciphertext[oneD+i];
+     data_B[i]=ciphertext[2*oneD+i];
+   }
+   store_RGB_pixmap("lena2.ppm", data_R, data_G, data_B, width, height);
+ }

  
   
  
   

-  time=0;
-  t=0;
-  t=TimeStart();
-
-  for(int i=0;i<nb_test;i++)
+  //for(int i=0;i<nb_test;i++)

   {  
     /* Decrypt the ciphertext */
     decryptedtext_len = decrypt(ciphertext, ciphertext_len, key, iv,
   {  
     /* Decrypt the ciphertext */
     decryptedtext_len = decrypt(ciphertext, ciphertext_len, key, iv,

-                               decryptedtext,ctr);
-  }
-
- time+=TimeStop(t);
-
- printf("Time decrypt %f\n",time);
-
-
-  for(int i=0;i<oneD;i++) {
-    data_R[i]=decryptedtext[i];
-    data_G[i]=decryptedtext[oneD+i];
-    data_B[i]=decryptedtext[2*oneD+i];
+                               decryptedtext,ctr, i);

   }
   }

-  store_RGB_pixmap("lena3.ppm", data_R, data_G, data_B, width, height);                    

 
 

+ //printf("Time decrypt %f\n",time);
+ printf("%e\t",(double)imsize*nb_test/time_decrypt);
+
+ if(lena) {
+   for(int i=0;i<oneD;i++) {
+     data_R[i]=decryptedtext[i];
+     data_G[i]=decryptedtext[oneD+i];
+     data_B[i]=decryptedtext[2*oneD+i];
+   }
+   store_RGB_pixmap("lena3.ppm", data_R, data_G, data_B, width, height);
+ }
+ else {
+   int equal=1;
+   for(int i=0;i<imsize;i++) {
+     //cout<<(int)buffer[i]<<endl;
+     if(buffer[i]!=decryptedtext[i]) {
+       equal=0;
+     }
+   }
+//   printf("RESULT CORRECT: %d\n",equal);
+ }

   
 
   /* Clean up */
   
 
   /* Clean up */