#include<iostream>
#include<string>
#include<iomanip>
using namespace std;
#include"predefine.h"
#include"rand.h"
#include"rsa.h"

void clearintNode()
{
	intNode i;
	i.clear();
}
void printMenu()
{
	cout << "__________________________\n";
	cout << "請選擇：\n";
	cout << "1.產(chǎn)生兩個隨機大素數(shù)并寫入文件(16進制)!\n";
	cout << "2.產(chǎn)生兩個隨機大素數(shù)并寫入文件(10進制)!\n";
	cout << "3.從文件讀入素數(shù)和公鑰并解密(私鑰自動生成)(16進制)!\n";
	cout << "4.從文件讀入素數(shù)和公鑰并解密(私鑰自動生成)(10進制)!\n";
	cout << "5.從文件讀入模數(shù)和公鑰并加密(16進制)!\n";
	cout << "6.從文件讀入模數(shù)和公鑰并加密(10進制)!\n";
	cout << "0.退出!\n";
}

void creatPrime(int base)
{
	Rand r;
	char c[30];
	cout << "請輸入用DES產(chǎn)生隨機素數(shù)所需密鑰所在的文件名!\n";
	cin >> c;
	r.setKeyFile(c);

	cout <<"請輸用產(chǎn)生隨機數(shù)的種子所在的文件名!\n";
	cin >> c;
	r.setZVFile(c);

	cout << "請輸入用來保存素數(shù)的文件名!\n";
	cin >> c;
	fstream out;
	out.open(c, ios::out);

	char* ch;
	int ch_l;
	BigInt p, q, n;

	p = r.randStrPrime();
	if(base == 10)
		ch = p.outputDEC(ch_l);
	else
		ch = p.outputHEX(ch_l);
	for(int i = 0; i < ch_l; ++i)
		out << ch[i];
	out << endl;
	delete[] ch;

	q = r.randStrPrime();
	if(base == 10)
		ch = q.outputDEC(ch_l);
	else
		ch = q.outputHEX(ch_l);
	for(i = 0; i < ch_l; ++i)
		out << ch[i];
	out << endl;

	cout << "要在文件后面加入加密時所要的公鑰嗎?\n";
	char judge;
	cin >> judge;
	if(judge == 'y' || judge == 'Y')
	{
		cout << "請輸入公鑰( " << base <<" 進制)!\n";
		cin >> ch;
		out << ch << endl;
		out.close();
		cout << "要新建一個文件并保存公鑰和模數(shù)嗎?";
		cin >> judge;
		if(judge == 'y' || judge == 'Y')
		{
			cout << "請輸入要保存公鑰的文件名:";
			cin >> c;
			out.open(c, ios::out);
			out << ch << endl;
			delete[] ch;
			n = p * q;
			if(base == 10)
				ch = n.outputDEC(ch_l);
			else
				ch = n.outputHEX(ch_l);
			for(i = 0; i < ch_l; ++i)
				out << ch[i];
			out << endl;
			cout << "寫入成功!" << endl;
			out.close();
			delete[] ch;
		}
	}

	q.clear();
	p.clear();
	n.clear();
}

void decodingFile(int base)
{
	rsa r;
	char name[30];
	cout << "請輸入素數(shù)和公鑰所在的文件名:\n";
	cin >> name;

	fstream in;
	in.open(name, ios::in);
	BigInt p, q, pub;
	char num[150];

	in >> num;
	if(base == 10)
		p.loadDEC(num, strlen(num));
	else
		p.loadHEX(num, strlen(num));

	in >> num;
	if(base == 10)
		q.loadDEC(num, strlen(num));
	else
		q.loadHEX(num, strlen(num));

	in >> num;
	if(base == 10)
		pub.loadDEC(num, strlen(num));
	else
		pub.loadHEX(num, strlen(num));
	r.setkey(p, q, pub);
	in.close();

	cout << "請輸入要解密的文件名：\n";
	cin >> name;
	r.setCFileName(name);

	cout << "請輸入要保存明文的文件名：\n";
	cin >> name;
	r.setMFileName(name);
	
	cout << "請選擇:\n"
		 << "1.讀入文件的ASCII碼解密!\n"
		 << "2.讀入文件的ASCII碼用中國剩余定理解密!\n"
		 << "3.以a = 00, b = 01, .........A = 26, B = 27..........解密!\n"
		 << "4.以a = 00, b = 01, .........A = 26, B = 27..........用中國剩余定理解密!\n";
	cout << "0.返回!" << endl;
	int choose;
	cin >> choose;
	if(choose != 0)
	{
		r.DeCode(choose, base);
		cout << "解密成功!" << endl;
	}
	p.clear();
	q.clear();
	pub.clear();
}
void encodingFile(int base)
{
	rsa r;
	char name[30];
	cout << "請輸入公鑰和模數(shù)所在的文件名:\n";
	cin >> name;

	fstream in;
	in.open(name, ios::in);
	BigInt pub, n;
	char num[300];

	in >> num;
	if(base == 10)
		pub.loadDEC(num, strlen(num));
	else
		pub.loadHEX(num, strlen(num));

	in >> num;
	if(base == 10)
		n.loadDEC(num, strlen(num));
	else
		n.loadHEX(num, strlen(num));

	r.setpubkey(pub, n);
	in.close();

	cout << "請輸入要加密的文件名：\n";
	cin >> name;
	r.setMFileName(name);

	cout << "請輸入要保存密文的文件名：\n";
	cin >> name;
	r.setCFileName(name);
	
	cout << "請選擇:\n"
		 << "1.讀入文件的ASCII碼加密!\n"
		 << "2.以a = 00, b = 01, .........A = 26, B = 27..........加密!\n";
	cout << "0.返回!" << endl;
	int choose;
	cin >> choose;
	if(choose != 0)
	{
		r.EnCode(choose, base);
		cout << "加密成功!" << endl;
	}

	n.clear();
	pub.clear();
	n.clear();
}
	
int main()
{
	int choose;
	while(true)
	{
		printMenu();
		cin >> choose;
		switch(choose)
		{
		case 1:
			creatPrime(16);
			break;
		case 2:
			creatPrime(10);
			break;
		case 3:
			decodingFile(16);
			break;
		case 4:
			decodingFile(10);
			break;
		case 5:
			encodingFile(16);
			break;
		case 6:
			encodingFile(10);
			break;
		default:
			break;
		}
		if(choose == 0)
			break;
	}

	clearintNode();
	return 0;
}