#include <iostream>
#include <fstream>
#include <time.h>
#include <string>
#include "des.h"
using namespace std;


void line_box();
void random_box();
void cry();	
void Des(char *plaint, char *crypt,int innum1,int innum2,bool *k);
void Byte_to_Bit(char *In,bool *Out,int m);
void round(bool *In, bool *Ki);

void change(bool *In, int m,char *table);
void xor(bool *In_1,bool *In_2, int m);
void Sbox(bool *Out, bool *In);
void Bit_to_Byte(bool *In, char *Out, int m);
void get_key();
void l_shift(bool *In, int n);
void xuebeng_test();
void tongji_test();

void main()
{
	srand(time(NULL));
	int choose;
	cout<<"選擇原sbox請輸入1，線性產生sbox請輸入2，隨機生成輸入3"<<endl;
		cin>>choose;
	if(choose!=1&&choose!=2&&choose!=3)
	{
		cout<<"輸入錯誤"<<endl;
		return;
	}
	switch(choose)
    {
    case 1:
        break;
    case 2:
        line_box();
        break;
    case 3:
        random_box();
        break;
     }
	int x;
	cout<<""<<endl;
	cin>>x;
	switch(x)
	{
	case 1:
		xuebeng_test();
		break;
	case 2:
        tongji_test();
		break;
	case 3:
		break;
	case 4:
		get_key();
		cry();

	}

}
void line_box()
{
	for(int i=0;i<8;i++)
		for(int j=0;j<4;j++)
			for(int k=0;k<16;k++)
				sbox[i][j][k]=i+j+k;
}
void random_box()
{
	for(int i=0;i<8;i++)
		for(int j=0;j<4;j++)
			for(int k=0;k<16;k++)
				sbox[i][j][k]=rand()%16;
}
void cry()
{
	int innum1,innum2;
	char head[65536];
	char plaint[8],crypt[8];
	unsigned int file_size,head_size,data_size;
	string file_in,file_out;
	cout<<"選擇1:ECB  2:CBC"<<endl;
	cin>>innum1;
	cout<<"選擇1:加密 2:解密"<<endl;
	cin>>innum2;
	cout<<"請輸入想要加密或解密的文件名稱"<<endl;
	cin>>file_in;
	cout<<"請輸入新建的文件名稱"<<endl;
	cin>>file_out;
	ifstream infile(file_in.c_str(),ios::binary);
	ofstream outfile(file_out.c_str(),ios::binary); 
	if(!infile ||!outfile)
	{
		cout<<"error"<<endl;
		return;
	}
	infile.seekg(2);  //BM
	infile.read((char *)&file_size,4);
	infile.seekg(4,ios_base::cur);
	infile.read((char *)&head_size,4);
	data_size=file_size-head_size;
	infile.seekg(ios::beg);
    infile.read(head,head_size);
    outfile.write(head,head_size);
	char K[]={'a','b','c','d','e','f','g','h'};
	bool k[64];
	Byte_to_Bit(K,k,64);
	for(int i=0;i<data_size/8;i++)
	{
		infile.read(plaint,8);
		Des(plaint,crypt,innum1,innum2,k);
		outfile.write(crypt,8);
	}
	infile.close;
	outfile.close;
}

void Des(char *plaint, char *crypt,int innum1,int innum2,bool *k)
{
	bool M[64], Tmp[32], *Li = &M[0], *Ri = &M[32];
    Byte_to_Bit(plaint, M,64);
	if (innum1== 2&&innum2==1)
        xor(M,k, 64);
	bool buffer[64];
	memcpy(buffer,M,64);
    change(M,64,ip);
	if(innum2==1)
	{
		for(int i=0; i<16; i++) 
		{
			memcpy(Tmp, Ri, 32);
		    round(Ri,sub_key[i]);
		    xor(Ri, Li, 32);
		    memcpy(Li, Tmp, 32);
		 }
	}
	else
	{
		for(int i=15; i>=0; i--)
		{
			memcpy(Tmp, Li, 32);
		    round(Li,sub_key[i]);
		    xor(Li, Ri, 32);
		    memcpy(Ri, Tmp, 32);
		}
	}
	change(M,64,dip);

	if (innum1== 2&&innum2==2)
	{
		xor(M,k, 64);
		memcpy(k,buffer,64);
	}
	else if(innum1==2&&innum2==1)
		memcpy(k,M,64);
	Bit_to_Byte(M,crypt, 64);
}


void round(bool *In, bool *Ki)
{
    bool MR[48];
	memcpy(MR,In,32);
    change(MR, 48,expansion);
    xor(MR, Ki, 48);
    Sbox(In,MR); 
    change(In,32,p);
}
void Byte_to_Bit(char *In,bool *Out,int m)
{
	for(int i=0; i<m; i++)
        Out[i] = (In[i/8]>>(i%8)) & 1;
}

void Bit_to_Byte(bool *In, char *Out, int m)
{
    memset(Out, 0, (m+7)/8);
    for(int i=0; i<m; i++)
        Out[i/8] |= In[i]<<(i%8);
}

  
void change(bool *In, int m,char *table)
{
	bool tem[64];
	for(int i=0;i<m;i++)
		tem[i]=In[table[i]-1];
	memcpy(In,tem,m);
}
void xor(bool *In_1,bool *In_2, int m)
{
	for(int i=0;i<m;i++)
		In_1[i] = In_2[i] ^In_1[i];
} 


void Sbox(bool *Out, bool *In)
{
    for(char i=0,j,k; i<8; i++,In+=6,Out+=4) {
        j = (In[0]<<1) + In[5];
        k = (In[1]<<3) + (In[2]<<2) + (In[3]<<1) + In[4];
        Byte_to_Bit( &sbox[i][j][k], Out,4);
    }

}

void get_key()
{
	bool k[64];
	char key[]={'s','u','n',' ',' ',' ','q','i'};
	bool buffer[56],buffer_l[28], buffer_r[28];
	Byte_to_Bit(key,k,64);
	change(k,56,pc1);
	memcpy(buffer_l,k,28);
	for(int h=0;h<28;h++)
		buffer_r[h]=k[h+28];
	
	for(int i=0;i<16;i++)
	{
		l_shift(buffer_l,loop[i]);
		l_shift(buffer_r,loop[i]);
		memcpy(buffer,buffer_l,28);
		for(int j=0;j<28;j++)
			buffer[j+28]=buffer_r[j];
		change(buffer,48,pc2);
		memcpy(sub_key[i],buffer,48);
	}

}
void l_shift(bool *In, int n)
{
	bool tem;
	for(int i=0;i<n;i++)
	{	
		tem=In[0];
		for(int j=0;j<27;j++)
			In[j]=In[j+1];
		In[27]=tem;
	}
}


void xuebeng_test()
{
	bool a[64]={0},b[64]={0},Tmpa[32],Tmpb[32];
	bool *al=&a[0],*ar=&a[32],*bl=&b[0],*br=&b[32];
	int x=rand()%16;
	int count=0;
	a[x]=0;
	b[x]=1;
	for(int i=0; i<16; i++) 
	{
		memcpy(Tmpa, ar, 32);
	    round(ar,sub_key[i]);
	    xor(ar, al, 32);
	    memcpy(al, Tmpa, 32);

		memcpy(Tmpb, br, 32);
	    round(br,sub_key[i]);
	    xor(br, bl, 32);
	    memcpy(bl, Tmpb, 32);
		for(int j=0;j<64;j++)
			if(a[j]!=b[j])
				count++;
		cout<<"經過"<<i+1<<"次加密的不同位數是"<<count<<"  "<<endl;
		count=0;
	 }
}
void tongji_test()
{
	bool a[64]={0},Tmpa[32];
	bool *al=&a[0],*ar=&a[32];
	int x=rand()%16;
	int count[2]={0};;
	for(int j=0;j<64;j++)
	{
		for(int i=0;i<64;i++)
			a[i]=rand()%2;
		a[x]=1;
		for(i=0; i<16; i++) 
		{
			memcpy(Tmpa, ar, 32);
		    round(ar,sub_key[i]);
		    xor(ar, al, 32);
		    memcpy(al, Tmpa, 32);
		}
		for(i=0;i<64;i++)
		{
			if(a[i]==0)
				count[0]++;
			else
				count[1]++;
		}
	}
	cout<<"0的個數是"<<count[0]<<endl;
	cout<<"1的個數是"<<count[1]<<endl;
}