#include <stdio.h> 
#include <stdlib.h> 
#include <string.h>
#include <stdint.h>


int keysize=0; 
int ordered[256]; 
int reversed[256]; 
int streamout[256]; 

/*PRNG*/ 
int streambuddy() 
{ 
  int mixpad[256]; 
  int slot_1,slot_2,slot_3,slot_4; 
  int i; 


  for(i=0;i<256;i++) 
  { 
    if(i==255){ 
      /*WRAPAROUND CASE*/ 
      slot_1 = ordered[0]; 
      slot_2 = ordered[slot_1]; 
      ordered[slot_1] = ordered[i]; 
      ordered[i] = slot_2; 
      slot_3 = reversed[0]; 
      slot_4 = reversed[slot_3]; 
      reversed[slot_3] = reversed[i]; 
      reversed[i] = slot_4; 
    } else { 
      slot_1 = ordered[i+1]; 
      slot_2 = ordered[slot_1]; 
      ordered[slot_1] = ordered[i]; 
      ordered[i] = slot_2; 
      slot_3 = reversed[i+1]; 
      slot_4 = reversed[slot_3]; 
      reversed[slot_3] = reversed[i]; 
      reversed[i] = slot_4; 
    } 
  } 


  for(i=0;i<256;i++) 
  { 
    mixpad[i] = reversed[i] ^ ordered[i]; 
  } 


  for(i=0;i<256;i++) 
  { 
    streamout[i] ^= mixpad[i]; 
  } 
  return 0; 

} 


/*KEYXPAND*/ 
int keyexpander(int key[], int keysize) 
{ 
  int serie[256]; 
  int i=0,j=0,k=0; 
  int stop=0; 
  int found=0; 

  for(i=0;i<256;i++) 
  { 
    stop=0; 
    while(!stop) 
    { 
      found = 0; 
      for(k=0;k<i;k++) 
      { 
        if(key[j]==serie[k]){found=1;} 
      } 
      if(!found) 
      { 
        serie[i]=key[j];stop=1; 
      } else 
      { 
        key[j] = key[j]+1; 
      } 
      if(key[j] > 255) { 
        key[j]=0; 
      } 
    } 
    j++; 
    if(j > keysize-1) {j=0;} 
  } 


  for(k=0;k<42;k++) 
  { 
    for(i=0;i<255;i++) 
    { 
      int slot_1,slot_2; 
      slot_1 = serie[i+1]; 
      slot_2 = serie[slot_1]; 
      serie[slot_1] = serie[i]; 
      serie[i] = slot_2; 
    } 
  } 


  for(i=0;i<256;i++) 
  { 
    ordered[i]=serie[i]; 
  } 
  return 0; 

} 


int ireverse() 
{ 
  int i;int j=0;int slot1;int slot2; 
  for(i=255;i>-1;i--) 
  { 
    reversed[j] = ordered[i]; 
    j++; 
  } 
  /*A mechanism to make the reversed block out of phase*/ 
  for(i=0;i<7;i++){ 
    slot1 = reversed[i+1]; 
    slot2 = reversed[slot1]; 
    reversed[slot1] = reversed[i]; 
    reversed[i] = slot2; 
  } 
  return 0; 

} 


int saltIV(int keyread[]){ 
  int i; 
  srand(clock()); 
  for (i=0;i<4;i++){ 
    keyread[i]=rand()%256; 
  } 
  return 0; 
} 


int IDkeyfix(int keyread[]){ 
int s,r,j,k; 
int sum=128;
int dup; 
r=keysize; 
for(k=0;k<r;k++){ 
   dup=0; 
   s=keyread[k]; 
   for(j=0;j<r;j++){ 
   if ((s==keyread[j])&&(j!=k)&&(dup!=1)){
    dup=1;
    sum=sum+s*k;
    keysize=keysize+1;
    keyread[keysize]=sum%256; 
      //printf("%d",keyread[keysize]);
      //printf(",");
    } 
   } 
 } 
 return 0; 
} 


int encrypt(char *Keyword, int CheckByte)
{ 
  int keyread[256]; 
  int text[256]; 
  int i,j,k; 

  // keysize=4; 
  /*CREATE KEY ->IV+PASSWORD*/ 
  // saltIV(keyread); 

  keysize = strlen(Keyword);

  for (i=0; i<keysize; i++) keyread[i] = Keyword[i];


  /*EXPAND KEY AND CREATE REVERSED SHADOW*/ 
  IDkeyfix(keyread); 
  keyexpander(keyread,keysize); 
  ireverse(); 

  printf ("ordered[]   = ");
  for(i=0;i<CheckByte;i++){
    printf ("%02X ", ordered[i]);
  }
  printf ("\n");
  
  printf ("reversed[]  = ");
  for(i=0;i<CheckByte;i++){
    printf ("%02X ", reversed[i]);
  }
  printf ("\n");


  for(i=0;i<256;i++){streamout[i]=0;}

  streambuddy();

  printf ("streamout[] = ");
  for(i=0;i<CheckByte;i++){
    printf ("%02X ", streamout[i]);
  }
  printf ("\n");

  return 0; 
} 


int main(int argc, char *argv[]) {

  char* KeyWord;
  int CheckByte = 32; 
  
  KeyWord   = argv[1];
  CheckByte = atoi(argv[2]);

  encrypt(KeyWord, CheckByte);

  // fprintf(stderr,"\n");
  
  exit(0);
}



/*

./streambody_new_keysearch baaaaaaaaaaaababc 16
B0:D6:3D:1A:E1:54:77:C0:C1:BF:14:B8:67:83:61:49:

./streambody_new_keysearch baaaaaaaaaaaababd 16
B0:D6:3D:1A:E1:54:77:C0:C1:BF:14:B8:67:83:61:49:





###

rm keysearch.result.txt; c="abcdefghijklmnopqrstuvwxyz"; for ((i=0; i<${#c}; i++)); do cc="aaaaaaaaaaaaaahj6sfE${c:i:1}"; echo -en "$cc => " >> keysearch.result.txt; ./streambody_new_keysearch $cc 16 >> keysearch.result.txt; echo "" >> keysearch.result.txt; done

rm keysearch.result.txt; c="abcdefghijklmnopqrstuvwxyz"; for ((i=0; i<${#c}; i++)); do cc="baaaaaaaaaaaabab${c:i:1}"; echo -en "$cc => " >> keysearch.result.txt; ./streambody_new_keysearch $cc 16 >> keysearch.result.txt; echo "" >> keysearch.result.txt; done

rm keysearch.result.txt; c="abcdefghijklmnopqrstuvwxyz"; for ((i=0; i<${#c}; i++)); do cc="bje8erh:98#_bab${c:i:1}"; echo -en "$cc => " >> keysearch.result.txt; ./streambody_new_keysearch $cc 16 >> keysearch.result.txt; echo "" >> keysearch.result.txt; done


*/