#include <iostream>
#include "AES.h"
#include "md5.h"

int main()
{
	cout << "输入数字1我将用AES下EBC方式为你加密\n";
	cout << "输入数字2我将用MD5方式为你加密\n";
	cout << "输入数字3我将用AES下EBC方式为你解密\n\n\n";

	AES aes;
	md5 m5;
	int code = 0;
	string key, value;
	while(1)
	{
		cout << "请输入加/解密方式\n"; 
		cin >> code;
		switch (code)
		{
		case 1:
			cout << "请输入要加密的内容\n";
			cin >> value;
			cout << "请输入密钥\n";
			cin >> key;
			cout <<"加密后：" << aes.EncryptionAES(value, key.c_str()) << "\n\n";
			break;
		case 2:
			cout << "请输入要加密的内容\n";
			cin >> value;
			cout << "加密后：" << m5.StringToMD5(value) << "\n\n";
			break;
		case 3:
			cout << "请输入要解密的内容\n";
			cin >> value;
			cout << "请输入密钥\n";
			cin >> key;
			cout << "解密后：" << aes.DecryptionAES(value, key.c_str()) << "\n\n";
			break;
		default:
			return 0;
		}
	}
}