#include <string>
#include <iostream>
#include <vector>
#include <fstream> 
#include "Vigenere.h"
#include "DES.h"
#include "RSA.h"
#include "MD5.h"
#include "DSS.h"
using namespace std;
void printHead();
void dss();
void vigenere();
void md5();
void rsa();
void des();
string getText(string filename);
void writeCiphertext(string ciphertext, string fileName);
int main()
{
	printHead();
	int a;
	while (cin>>a)
	{
		switch (a)
		{
		case 1:vigenere(); break;
		case 2:des(); break;
		case 3:rsa(); break;
		case 4:md5(); break;
		case 5:dss(); break;
		default: 
			break;
		}
		if (a==6)
		{
			cout << "谢谢使用" << endl;
			break;
		}
		system("pause");
		system("cls");
		printHead();
	}
	system("pause");
	return 0;
}

void printHead() {
	cout << "---------------------------------------" << endl;
	cout << "-            1.Vigenere加密算法       -" << endl;
	cout << "-            2.DES加密算法            -" << endl;
	cout << "-            3.RSA加密算法            -" << endl;
	cout << "-            4.MD5加密算法            -" << endl;
	cout << "-            5.DSS签名算法            -" << endl;
	cout << "-            6.退出                   -" << endl;
	cout << "---------------------------------------" << endl;
	cout << "明文内容为："+ getText("plaintext.txt") +"\n" << endl;
	cout << "选择：";
}

void dss() {
	cout << "\n***  DSS加密算法  ***" << endl;
	cout << "随机生产公钥私钥对:" << endl;
	DSS d(1,"1");
	d.generateKey();
	d.signature(getText("plaintext.txt"));
	cout << "公钥签名后签名为(r,s)：" << "(" << d.r << "," << d.s << ")" << endl;
	writeCiphertext(d.sig, "ciphertext.dss");
	cout << "签名验证：" << endl;
	d.verify(getText("plaintext.txt"));
	cout << "签名已写入 ciphertext.dss 文件" << endl;
}

void vigenere() {
	string keyword;
	cout << "\n***  Vigenere加密算法  ***" << endl;
	cout << "输入密钥(一个英文单词):" << endl;
	cin >> keyword;
	Vigenere v(keyword);
	v.Encrypt(getText("plaintext.txt"));
	cout << "加密后密文为：" << v.ciphertext << endl;
	writeCiphertext(v.ciphertext, "ciphertext.vig");
	v.Decrypt(getText("ciphertext.vig"));
	cout << "解密后明文为：" << v.plaintext << endl;
	cout << "密文已写入 ciphertext.vig 文件" << endl;
}

void md5() {
	string keyword;
	cout << "\n***  MD5加密算法  ***" << endl;
	MD5 m;
	m.Encrypt(getText("plaintext.txt"));
	cout << "加密后密文为：" << m.ciphertext << endl;
	writeCiphertext(m.ciphertext, "ciphertext.md5");
	cout << "密文已写入 ciphertext.md5 文件" << endl;
}

void rsa() {
	string keyword;
	cout << "\n***  RSA加密算法  ***" << endl;
	cout << "随机生产公钥私钥对:" << endl;
	RSA r(1,1);
	r.generateKey();
	r.Encrypt(getText("plaintext.txt"));
	cout << "公钥加密后密文为：" << r.ciphertext << endl;
	writeCiphertext(r.ciphertext, "ciphertext.rsa");
	r.Decrypt(getText("ciphertext.rsa"));
	cout << "私钥解密后明文为：" << r.plaintext << endl;
	cout << "密文已写入 ciphertext.rsa 文件" << endl;
	
}

void des() {
	string keyword;
	cout << "\n***  DES加密算法  ***" << endl;
	cout << "输入密钥:" << endl;
	cin >> keyword;
	DES d(keyword);
	d.Encrypt(getText("plaintext.txt"));
	cout << "加密后密文为：" << d.ciphertext << endl;
	writeCiphertext(d.ciphertext, "ciphertext.des");
	d.Decrypt(getText("ciphertext.des"));
	cout << "解密后明文为：" << d.plaintext << endl;
	cout << "密文已写入 ciphertext.des 文件" << endl;
}

//读入文本
string getText(string filename) {
	ifstream infile(filename);
	string plaintext;
	getline(infile, plaintext);
	infile.close();
	return plaintext;
}

//创建并写入文本
void writeCiphertext(string ciphertext, string fileName) {
	ofstream outfile(fileName, ofstream::out);
	outfile << ciphertext;
	outfile.close();
}