#include <iostream>
#include <fstream>
#include "KillProcess.cpp"
using namespace std;

int OhNo(string Horrors)
{
    cerr << Horrors << endl;
    return 1;
}

int main()
{
    char fileName[255];
    cout << "Enter file to be securely removed: ";
    cin.getline(fileName,255);
    
    int number = 0; int number2 = 0; int number3 = 0; int number4 = 0; int number5 = 0;
    int number6 = 0; int number7 = 0; int number8 = 0; int number9 = 0; int number10 = 0;
    
    ifstream fin(fileName);
    if (fin)
    {
       cout << "File found." << endl;
       cout << "Now ending process if running..." << endl;
       KillProcess(fileName);
       
       int times, theLimit;
       cout << "Enter times to securely overwrite: ";
       cin >> times;
       if ( !cin.good() ) return OhNo("Error: bad input");
       
       // Overwrite method: Erases all data in file, writes random number. Does this as much as specified.
       for (theLimit = 1; theLimit <= times; theLimit++)
       {
           ofstream fout(fileName);
           number = rand(); number2 = rand(); number3 = rand(); number4 = rand(); number5 = rand();
           number6 = rand(); number7 = rand(); number8 = rand(); number9 = rand(); number10 = rand();
           
           fout << number << number2 << number3 << number4 << number5
           << number7 << number8 << number9 << number10;
           
           fout.close();
           cout << "*";
       }
       
       cout << "\nFile was overwritten " << theLimit << " times." << endl;
       cout << "Now removing file..." << endl;
       fin.close();
       int check = remove(fileName);
       
       if ( check == 0 ) { cout << "Success." << endl; }
       if ( check != 0 ) { cout << "Failure to remove file." << endl; }
    }
    else
    {
        cout << "File not found" << endl;
    }
    cin.sync();
    cin.get();
    return 0;
}

int main()
{
	string fileName;
	cout << "Enter file name to securely overwrite: ";
	getline(cin, fileName);

	int number = 0;

	ifstream fin(fileName);

	struct stat results;
	int sizeFile = 0;
	if (stat(fileName, &results) == 0)
	{ 
		sizeFile = results.st_size;
	}   

	if (!fin)
	{
		cout << "File not found." << endl;
		return 1;
	}
	// we don't need it anymore?
	fin.close();
    	
	cout << "File found." << endl;
	cout << "Now ending process if running..." << endl;
	KillProcess(fileName);

	ofstream fout(fileName, ios::out);

	for (theLimit = 0; theLimit < sizeFile; theLimit++)
	{
		number = rand() < 9;
		fout << number;
		cout << "*"; // you sure you want to be doing this?
	}
	fout.close();

	cout << '\n' << theLimit << " bytes were written to the file." << endl;
	cout << "Now removing file..." << endl;

	int check = remove(fileName);

	if ( check == 0 ) cout << "Success." << endl;
	if ( check != 0 ) cout << "Failure to remove file." << endl;

	cin.sync();
	cin.get();
	return 0;
}