// RsaDlg.cpp : implementation file
//

#include "stdafx.h"
#include "Rsa.h"
#include "RsaDlg.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CAboutDlg dialog used for App About

class CAboutDlg : public CDialog
{
public:
	CAboutDlg();

// Dialog Data
	//{{AFX_DATA(CAboutDlg)
	enum { IDD = IDD_ABOUTBOX };
	//}}AFX_DATA

	// ClassWizard generated virtual function overrides
	//{{AFX_VIRTUAL(CAboutDlg)
	protected:
	virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
	//}}AFX_VIRTUAL

// Implementation
protected:
	//{{AFX_MSG(CAboutDlg)
	//}}AFX_MSG
	DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
	//{{AFX_DATA_INIT(CAboutDlg)
	//}}AFX_DATA_INIT
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CAboutDlg)
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
	//{{AFX_MSG_MAP(CAboutDlg)
		// No message handlers
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CRsaDlg dialog

CRsaDlg::CRsaDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CRsaDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CRsaDlg)
	m_pnum = _T("");
	m_qnum = _T("");
	m_nnum = _T("");
	m_fnum = _T("");
	m_enum = _T("");
	m_dnum = _T("");
	m_input = _T("");
	m_inputnum = _T("");
	m_orignum = _T("");
	m_encnum = _T("");
	//}}AFX_DATA_INIT
	// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CRsaDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CRsaDlg)
	DDX_Control(pDX, IDC_NOTES, m_notes);
	DDX_Text(pDX, IDC_PNUM, m_pnum);
	DDX_Text(pDX, IDC_QNUM, m_qnum);
	DDX_Text(pDX, IDC_NNUM, m_nnum);
	DDX_Text(pDX, IDC_FNUM, m_fnum);
	DDX_Text(pDX, IDC_ENUM, m_enum);
	DDX_Text(pDX, IDC_DNUM, m_dnum);
	DDX_Text(pDX, IDC_INPUT, m_input);
	DDX_Text(pDX, IDC_INPUTNUM, m_inputnum);
	DDX_Text(pDX, IDC_DECTEXT, m_orignum);
	DDX_Text(pDX, IDC_ENCTEXT, m_encnum);
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CRsaDlg, CDialog)
	//{{AFX_MSG_MAP(CRsaDlg)
	ON_WM_SYSCOMMAND()
	ON_WM_PAINT()
	ON_WM_QUERYDRAGICON()
	ON_BN_CLICKED(IDC_GenPrimes, OnGenPrimes)
	ON_BN_CLICKED(IDC_Convert, OnConvert)
	ON_BN_CLICKED(IDC_GenE, OnGenE)
	ON_BN_CLICKED(IDC_CalcD, OnCalcD)
	ON_BN_CLICKED(IDC_Encrypt, OnEncrypt)
	ON_BN_CLICKED(IDC_Decrypt, OnDecrypt)
	ON_BN_CLICKED(IDC_AUTO, OnAuto)
	ON_BN_CLICKED(IDC_CLEAR, OnClear)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CRsaDlg message handlers

BOOL CRsaDlg::OnInitDialog()
{
	CDialog::OnInitDialog();

	// Add "About..." menu item to system menu.

	// IDM_ABOUTBOX must be in the system command range.
	ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
	ASSERT(IDM_ABOUTBOX < 0xF000);

	CMenu* pSysMenu = GetSystemMenu(FALSE);
	if (pSysMenu != NULL)
	{
		CString strAboutMenu;
		strAboutMenu.LoadString(IDS_ABOUTBOX);
		if (!strAboutMenu.IsEmpty())
		{
			pSysMenu->AppendMenu(MF_SEPARATOR);
			pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
		}
	}

	// Set the icon for this dialog.  The framework does this automatically
	//  when the application's main window is not a dialog
	SetIcon(m_hIcon, TRUE);			// Set big icon
	SetIcon(m_hIcon, FALSE);		// Set small icon
	
	// TODO: Add extra initialization here
	
	return TRUE;  // return TRUE  unless you set the focus to a control
}

void CRsaDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
	if ((nID & 0xFFF0) == IDM_ABOUTBOX)
	{
		CAboutDlg dlgAbout;
		dlgAbout.DoModal();
	}
	else
	{
		CDialog::OnSysCommand(nID, lParam);
	}
}

// If you add a minimize button to your dialog, you will need the code below
//  to draw the icon.  For MFC applications using the document/view model,
//  this is automatically done for you by the framework.

void CRsaDlg::OnPaint() 
{
	if (IsIconic())
	{
		CPaintDC dc(this); // device context for painting

		SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);

		// Center icon in client rectangle
		int cxIcon = GetSystemMetrics(SM_CXICON);
		int cyIcon = GetSystemMetrics(SM_CYICON);
		CRect rect;
		GetClientRect(&rect);
		int x = (rect.Width() - cxIcon + 1) / 2;
		int y = (rect.Height() - cyIcon + 1) / 2;

		// Draw the icon
		dc.DrawIcon(x, y, m_hIcon);
	}
	else
	{
		CDialog::OnPaint();
	}
}

// The system calls this to obtain the cursor to display while the user drags
//  the minimized window.
HCURSOR CRsaDlg::OnQueryDragIcon()
{
	return (HCURSOR) m_hIcon;
}

void CRsaDlg::OnGenPrimes() 
{
	char buffer[20];

	//Make sure no problems arise if p,q are recalculated
	m_enum = "" ;
	m_dnum = "" ;
	m_encnum = "";
	m_orignum= "";

	p = findprime(0);
	m_pnum = CString(p.ToString().c_str());
	m_pnum = m_pnum + "     (" + CString(_itoa(p.DigitsNum(),buffer,10)) + " Digits)";

	q = findprime(p);
	m_qnum = CString(q.ToString().c_str());
	m_qnum = m_qnum + "     (" + CString(_itoa(q.DigitsNum(),buffer,10)) + " Digits)";
	

	n=p*q;
	m_nnum=CString(n.ToString().c_str());
	m_nnum = m_nnum + "     (" + CString(_itoa(n.DigitsNum(),buffer,10)) + " Digits)";

	f=(p-1)*(q-1);
	m_fnum=CString(f.ToString().c_str());
	m_fnum = m_fnum + "     (" + CString(_itoa(f.DigitsNum(),buffer,10)) + " Digits)";


	UpdateData(false);


}

void CRsaDlg::OnConvert() 
{
	CString tmp;
	char buffer[20];
	int asciival;

	tmp = "";
	UpdateData(true);
	for (int i=0;i<m_input.GetLength();i++)
	{
		asciival = (int)m_input[i]-31;
		if (asciival<10) tmp=tmp + "0";
		tmp = tmp + _itoa(asciival,buffer,10);
	}
	inputnum = tmp.operator LPCTSTR();

	tmp = tmp + "    (" + CString(_itoa(tmp.GetLength(),buffer,10)) + " Digits)";

	m_encnum = "";
	m_orignum = "";
	m_inputnum = tmp;
	UpdateData(false);
	
}

void CRsaDlg::OnGenE() 
{
	char buffer[20];
	CString tmp;
	//Make sure no problems arise if e is recalculated
	m_enum = "" ;
	m_dnum = "" ;
	m_encnum = "" ;
	m_orignum = "" ;

	if (f!=0) 
		e = selecte(f,2000);
	else
		e = "0";

	m_enum = CString(e.ToString().c_str());
	m_enum = m_enum + "     (" + CString(_itoa(e.DigitsNum(),buffer,10)) + " Digits)";
	UpdateData(false);

	tmp = "\nPublic  Key = (" + CString(e.ToString().c_str()) + ", " ;
	tmp = tmp + CString(n.ToString().c_str()) + ")";
	WriteNotes(tmp);	
		
}

void CRsaDlg::OnCalcD() 
{
	char buffer[20];
	CString tmp;

	d = selectd(e,f);

	m_dnum = CString(d.ToString().c_str());
	m_dnum = m_dnum + "     (" + CString(_itoa(d.DigitsNum(),buffer,10)) + " Digits)";

	UpdateData(false);

	tmp = "\nPrivate Key = (" + CString(d.ToString().c_str()) + ", " ;
	tmp = tmp + CString(n.ToString().c_str()) + ")";
	WriteNotes(tmp);
}

void CRsaDlg::OnEncrypt() 
{
	char buffer[20];
	CString tmp;
	encnum =fastexponent(inputnum,e,n);
	m_encnum= CString(encnum.ToString().c_str());

	m_encnum = m_encnum + "     (" + CString(_itoa(encnum.DigitsNum(),buffer,10)) + " Digits)";
	UpdateData(false);

	WriteNotes("\n");
	tmp = "\nEncrypt Converted Number Using ";
	tmp = tmp + "Public  Key = (" + CString(e.ToString().c_str()) + ", " ;
	tmp = tmp + CString(n.ToString().c_str()) + ")";
	WriteNotes(tmp);	
		
}

void CRsaDlg::OnDecrypt() 
{
	char buffer[20];
	CString tmp;
	orignum =fastexponent(encnum,d,n);
	m_orignum= CString(orignum.ToString().c_str());

	m_orignum = m_orignum + "     (" + CString(_itoa(orignum.DigitsNum(),buffer,10)) + " Digits)";
	UpdateData(false);
	
	WriteNotes("\n");
	tmp = "\nDecrypt Message Using ";
	tmp = tmp + "Private Key = (" + CString(d.ToString().c_str()) + ", " ;
	tmp = tmp + CString(n.ToString().c_str()) + ")";
	WriteNotes(tmp);
	WriteNotes("\n");
}

void CRsaDlg::OnAuto() 
{
	OnConvert();
	OnGenPrimes();
	OnGenE();
	OnCalcD();
	OnEncrypt();
	OnDecrypt();
}

void CRsaDlg::OnClear() 
{

	m_inputnum = ""; inputnum="0";
	m_pnum = ""; p = "0";
	m_qnum = ""; q = "0" ;
	m_nnum = ""; n = "0"; 
	m_fnum = ""; f = "0";
	m_enum = ""; e = "0";
	m_dnum = ""; d = "0";
	m_encnum = "";  encnum = "";
	m_orignum = ""; orignum = "";
	UpdateData(false);

}

void CRsaDlg::WriteNotes(CString str)
{
	CString winStr;
	
	m_notes.GetWindowText(winStr);
	m_notes.SetSel(winStr.GetLength(),winStr.GetLength());

	winStr = "";
	if(str[0] == 0x0A)
		winStr = 0x0D;

	winStr += str;
	m_notes.ReplaceSel(winStr);
}

BigInt CRsaDlg::findprime(BigInt ignore_prime)
{
   int tmprandom;
   BigInt odd;
   CString tmp;
   srand((unsigned)time(NULL) + ignore_prime.ToInt());

   while(1)
   {
      odd= (int)0;
      
      tmprandom = random_range(1000,9999);
      for (int i=1;i<=5;i++)
      {
         odd += tmprandom;
         odd.ShiftLeft(4);
         tmprandom = random_range(0,9999);
      }
      odd.ShiftRight(4);

      if(odd%2==0)
      odd++;

      if(ignore_prime == odd) continue;
	   
	  tmp = "\nTrying number : " + CString(odd.ToString().c_str());
	  WriteNotes(tmp);
      //elegse an o odd einai protos arithmos 
      if ( fastexponent(2,odd,odd)==2 ) 
	  {
		  WriteNotes(CString("\t\t\tPRIME FOUND").operator LPCTSTR ());
		  break;
	  }
	  WriteNotes(CString("\t\tNOT PRIME").operator LPCTSTR ());
   }
   
   return odd;
}