Code:
#include "..\header\SecSuite.h"
int main()
{
std::string Dir,File,Option,Restart;
DisplayOptions();
getline(std::cin,Option,'\n');
if(Option=="SDW")
{
std::cout<<"Sorry,This option is not yet available."<<std::endl;
}
if(Option=="SDC")
{
std::string Dir=GetObject("DiskName");
if(Dir.length()!=3)
{
CheckLength(Dir);
}
File=Dir+"SecureDiskCleanse.txt";
bool FileExist=true;/*NEED TO WRITE THIS FUNCTION...*/
if(FileExist==true)
{
std::cout<<"Do you want to start where you left off?(Y or N):";
getline(std::cin,Restart,'\n');
if(Option=="Y"||"y")
{
RestartMainFunction(Dir,File);
}
else
{
NoRestartMainFunction(Dir,File);
}
}
else
{
MainFunction(Dir,File);
}
Delete(File);
std::cout<<"Disk Cleansed."<<std::endl;
std::cin.get();
}
if(Option=="SFD")
{
int Loops,SizeOfFile,OrigSizeOfFile;
File=GetObject("File location");
OrigSizeOfFile=GetFileSize(File.c_str());
std::ofstream a_file(File.c_str(), std::ios::trunc);
Loops=GetFileSize(File.c_str());
while(Loops>0,Loops--)
{
a_file<<"1";
}
Delete(File);
std::cin.get();
}
else
{
std::cin.get();
}
return 0;
}
And then GUI
Code:
#define WIN32_LEAN_AND_MEAN
#include <windows.h>
HINSTANCE g_hInst;
LRESULT CALLBACK WndProc(HWND,UINT,WPARAM,LPARAM);
int WINAPI WinMain(HINSTANCE hInstance,HINSTANCE hPrevInstance,
LPSTR lpCmdLine,int nCmdShow)
{
MSG Msg;
HWND hwnd;
WNDCLASSEX wincl;
g_hInst=hInstance;
TCHAR chClassName[]=TEXT("SecuritySuite");
wincl.cbClsExtra=0;
wincl.cbWndExtra=0;
wincl.style=CS_HREDRAW|CS_VREDRAW;
wincl.lpszMenuName=NULL;
wincl.hInstance=hInstance;
wincl.lpszClassName=chClassName;
wincl.cbSize=sizeof(WNDCLASSEX);
wincl.lpfnWndProc=(WNDPROC)WndProc;
wincl.hCursor=LoadCursor(NULL,IDC_ARROW);
wincl.hIcon=LoadIcon(NULL,IDI_APPLICATION);
wincl.hIconSm=LoadIcon(NULL,IDI_APPLICATION);
wincl.hbrBackground=(HBRUSH)(COLOR_WINDOW+1);
if(!RegisterClassEx(&wincl))
{
MessageBox(0,"Window Registration Failed!","Error!",MB_ICONSTOP|MB_OK);
return 0;
}
hwnd=CreateWindowEx(0,chClassName,"Security Suite",WS_OVERLAPPEDWINDOW,
CW_USEDEFAULT,CW_USEDEFAULT,330,240,HWND_DESKTOP,NULL,
hInstance,NULL);
if(hwnd==NULL)
{
MessageBox(0,"Window Creation Failed!","Error",MB_ICONSTOP|MB_OK);
return 0;
}
ShowWindow(hwnd,nCmdShow);
UpdateWindow(hwnd);
while(GetMessage(&Msg,NULL,0,0))
{
TranslateMessage(&Msg);
DispatchMessage(&Msg);
}
return Msg.wParam;
}
LRESULT CALLBACK WndProc(HWND hwnd,UINT msg,WPARAM wParam,LPARAM lParam)
{
HDC hdc;
PAINTSTRUCT ps;
TCHAR chTxt[32];
HWND hButton,hCombo;
LPSTR Greeting,Question,Option;
Option="Option:";
Greeting="Welcome to C.J.'s SecuritySuite!";
Question="Please select what you want to do.";
switch(msg)
{
case WM_CREATE:
hCombo=CreateWindowEx(NULL,"COMBOBOX","Options",
WS_CHILD|WS_VISIBLE|CBS_DROPDOWNLIST,
60,120,180,100,hwnd,NULL,g_hInst,NULL);
lstrcpy(chTxt,"Secure Disk Wipe");
SendMessage(hCombo,CB_ADDSTRING,0,(LPARAM)chTxt);
lstrcpy(chTxt,"Secure File Deletion");
SendMessage(hCombo,CB_ADDSTRING,1,(LPARAM)chTxt);
lstrcpy(chTxt,"Secure Disk Cleanse");
SendMessage(hCombo,CB_ADDSTRING,2,(LPARAM)chTxt);
hButton=CreateWindowEx(NULL,"Button","Submit",
WS_BORDER|WS_CHILD|WS_VISIBLE|BS_PUSHBUTTON,
250,120,60,25,hwnd,NULL,g_hInst,NULL);
break;
case WM_PAINT:
hdc=BeginPaint(hwnd,&ps);
TextOut(hdc,53,20,Greeting,strlen(Greeting));
TextOut(hdc,49,40,Question,strlen(Question));
TextOut(hdc,10,120,Option,strlen(Option));
EndPaint(hwnd,&ps);
break;
case WM_CLOSE:
DestroyWindow(hwnd);
break;
case WM_DESTROY:
PostQuitMessage (0);
break;
default:
return DefWindowProc(hwnd,msg,wParam,lParam);
}
return 0;
}