304 lines
8.0 KiB
C++
304 lines
8.0 KiB
C++
// KeyStringDlg.cpp : implementation file
|
|
//
|
|
|
|
#include "stdafx.h"
|
|
#include "KeyString.h"
|
|
#include "KeyStringDlg.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()
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// CKeyStringDlg dialog
|
|
|
|
CKeyStringDlg::CKeyStringDlg(CWnd* pParent /*=NULL*/)
|
|
: CDialog(CKeyStringDlg::IDD, pParent)
|
|
{
|
|
//{{AFX_DATA_INIT(CKeyStringDlg)
|
|
// NOTE: the ClassWizard will add member initialization here
|
|
//}}AFX_DATA_INIT
|
|
m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
|
|
m_hFileIcon = AfxGetApp()->LoadIcon(IDI_FILEOPEN );
|
|
m_hSaveIcon = AfxGetApp()->LoadIcon(IDI_FILESAVE );
|
|
}
|
|
|
|
void CKeyStringDlg::DoDataExchange(CDataExchange* pDX)
|
|
{
|
|
CDialog::DoDataExchange(pDX);
|
|
//{{AFX_DATA_MAP(CKeyStringDlg)
|
|
// NOTE: the ClassWizard will add DDX and DDV calls here
|
|
//}}AFX_DATA_MAP
|
|
}
|
|
|
|
BEGIN_MESSAGE_MAP(CKeyStringDlg, CDialog)
|
|
//{{AFX_MSG_MAP(CKeyStringDlg)
|
|
ON_WM_SYSCOMMAND()
|
|
ON_WM_PAINT()
|
|
ON_WM_QUERYDRAGICON()
|
|
ON_BN_CLICKED(IDB_SEARCH, OnSearch)
|
|
ON_BN_CLICKED(IDB_FILESELECT, OnFileselect)
|
|
ON_EN_UPDATE(IDC_FILE, OnUpdateFile)
|
|
ON_BN_CLICKED(IDB_LOGFILESELECT, OnLogfileselect)
|
|
ON_EN_UPDATE(IDC_LOGFILE, OnUpdateLogfile)
|
|
//}}AFX_MSG_MAP
|
|
END_MESSAGE_MAP()
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// CKeyStringDlg message handlers
|
|
|
|
BOOL CKeyStringDlg::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);
|
|
}
|
|
}
|
|
|
|
SetIcon(m_hIcon, TRUE); // Set big icon
|
|
SetIcon(m_hIcon, FALSE); // Set small icon
|
|
|
|
// TODO: Add extra initialization here
|
|
((CButton *)GetDlgItem( IDB_FILESELECT ) )->SetIcon( m_hFileIcon );
|
|
((CButton *)GetDlgItem( IDB_LOGFILESELECT ) )->SetIcon( m_hSaveIcon );
|
|
((CWnd *)GetDlgItem( IDC_PROGRESS ) )->ShowWindow( SW_HIDE );
|
|
((CWnd *)GetDlgItem( IDC_VIEW ) )->ShowWindow( SW_SHOW );
|
|
((CButton *)GetDlgItem( IDC_VIEW ) )->SetCheck( TRUE );
|
|
|
|
|
|
|
|
return TRUE; // return TRUE unless you set the focus to a control
|
|
}
|
|
|
|
void CKeyStringDlg::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 CKeyStringDlg::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();
|
|
}
|
|
}
|
|
|
|
HCURSOR CKeyStringDlg::OnQueryDragIcon()
|
|
{
|
|
return (HCURSOR) m_hIcon;
|
|
}
|
|
|
|
void CKeyStringDlg::OnFileselect()
|
|
{
|
|
CFileDialog oFileDlg( TRUE );
|
|
|
|
oFileDlg.DoModal();
|
|
|
|
m_szFileName = oFileDlg.GetPathName();
|
|
GetDlgItem( IDC_FILE )->SetWindowText( m_szFileName );
|
|
}
|
|
|
|
void InsertTabs( FILE * pFile, int Nb )
|
|
{
|
|
int i;
|
|
|
|
for( i=0 ; i<Nb ; i++ )
|
|
fprintf( pFile, " " );
|
|
}
|
|
|
|
void CKeyStringDlg::OnSearch()
|
|
{
|
|
FILE * pDataFile;
|
|
FILE * pLogFile;
|
|
char szSequence[ 256 ];
|
|
long d_lPositions[ 256 ];
|
|
char iSeqLength, iSeqPos;
|
|
long iPos, iCpt;
|
|
|
|
CProgressCtrl * pProgress;
|
|
|
|
//--- Opening files ---
|
|
if( pDataFile = fopen( m_szFileName, "rb" ) )
|
|
{
|
|
if( pLogFile = fopen( m_szLogFileName, "w" ) )
|
|
{
|
|
((CWnd *) GetDlgItem( IDC_PROGRESS ) )->ShowWindow( SW_SHOW );
|
|
((CWnd *) GetDlgItem( IDC_VIEW ) )->ShowWindow( SW_HIDE );
|
|
|
|
fprintf( pLogFile, "--------------------------------------------\n" );
|
|
fprintf( pLogFile, " Key Positions \n" );
|
|
fprintf( pLogFile, "--------------------------------------------\n" );
|
|
fprintf( pLogFile, "[File ]> %s\n", m_szFileName );
|
|
fprintf( pLogFile, "[Key String ]> " );
|
|
|
|
((CEdit *) GetDlgItem( IDE_STRING ))->GetWindowText( szSequence, 255);
|
|
pProgress = ((CProgressCtrl *) GetDlgItem( IDC_PROGRESS ));
|
|
iSeqLength = strlen( szSequence );
|
|
iPos = 0;
|
|
iSeqPos = 0;
|
|
|
|
pProgress->SetRange( 0, iSeqLength );
|
|
|
|
fseek( pDataFile, -1, SEEK_END );
|
|
iPos = ftell( pDataFile );
|
|
|
|
while( (iSeqPos < iSeqLength) && (iPos > 0) )
|
|
{
|
|
if( fgetc( pDataFile ) == szSequence[ iSeqPos ] )
|
|
{
|
|
fprintf( pLogFile, "%c", szSequence[ iSeqPos ] );
|
|
d_lPositions[ iSeqPos ] = iPos;
|
|
iSeqPos ++;
|
|
pProgress->SetPos( iSeqPos );
|
|
}
|
|
|
|
fseek( pDataFile, -2, SEEK_CUR );
|
|
iPos = ftell( pDataFile );
|
|
}
|
|
fprintf( pLogFile, "\n--------------------------------------------\n" );
|
|
fprintf( pLogFile, "\n" );
|
|
|
|
//--- Creating IA-Like file ---
|
|
fprintf( pLogFile, "TestProtection := faux\n" );
|
|
for( iCpt = 0 ; iCpt < iSeqLength ; iCpt++ )
|
|
{
|
|
fprintf( pLogFile, "si ( PRO_LitValeurPositionCle( %i ) = %i )\n", d_lPositions[ iCpt ], szSequence[ iCpt ] );
|
|
InsertTabs( pLogFile, iCpt );
|
|
fprintf( pLogFile, "alors\t" );
|
|
}
|
|
fprintf( pLogFile, "TestProtection := vrai\n" );
|
|
|
|
for( iCpt = iSeqLength ; iCpt > 0 ; iCpt-- )
|
|
{
|
|
InsertTabs( pLogFile, iCpt-1 );
|
|
fprintf( pLogFile, "finsi\n" );
|
|
}
|
|
|
|
if( iSeqPos < iSeqLength )
|
|
MessageBox( "Unable to find the full KeyString in file.", "Error", MB_OK | MB_ICONSTOP );
|
|
else
|
|
MessageBox( "The full KeyString has been found.", "Operation Successful", MB_OK | MB_ICONEXCLAMATION );
|
|
|
|
fclose( pLogFile );
|
|
fclose( pDataFile );
|
|
pProgress->SetPos( 0 );
|
|
|
|
((CWnd *) GetDlgItem( IDC_PROGRESS ) )->ShowWindow( SW_HIDE );
|
|
((CWnd *) GetDlgItem( IDC_VIEW ) )->ShowWindow( SW_SHOW );
|
|
|
|
if( ((CButton *)GetDlgItem( IDC_VIEW ) )->GetCheck() )
|
|
ShellExecute( NULL, NULL, "Notepad.exe", m_szLogFileName, NULL, SW_SHOW );
|
|
|
|
}
|
|
}
|
|
}
|
|
|
|
void CKeyStringDlg::OnUpdateFile()
|
|
{
|
|
((CEdit *)GetDlgItem(IDC_FILE))->GetWindowText( m_szFileName );
|
|
}
|
|
|
|
void CKeyStringDlg::OnLogfileselect()
|
|
{
|
|
CFileDialog oFileDlg( FALSE, "*.log", "KeyString.log", OFN_OVERWRITEPROMPT, "Log Files (*.log) |*.log| All Files (*.*) |*.*||", NULL );
|
|
oFileDlg.DoModal();
|
|
|
|
m_szLogFileName = oFileDlg.GetPathName();
|
|
GetDlgItem( IDC_LOGFILE )->SetWindowText( m_szLogFileName );
|
|
}
|
|
|
|
void CKeyStringDlg::OnUpdateLogfile()
|
|
{
|
|
((CEdit *)GetDlgItem(IDC_LOGFILE))->GetWindowText( m_szLogFileName );
|
|
}
|