//----------------------------------------------------------------------------
//
//  ise_test_21.cpp
//
//  Authors: Joe Jarchow, Geoffrey Griffith, Joseph Kadhim, Shinya Daigaku
//           Andrew Pouzeshi
//
//  Sponsor: Tom Lookabaugh, Assistant Professor of Computer Science
//           University of Colorado
//
//  Senior Project: Team ISE (Image Selective Encryption)
//----------------------------------------------------------------------------


// This file is designed to test the ISE production code with test21.sh

#include <stdlib.h>     
#include <iostream>
#include <fstream> 
#include <string>
#include "rijndael-api-fst.h"
#include "ise.h"

using namespace std;

int main(int argc, char * argv[])
{
    /* Dencrypt_File Function with Output File Name Noty Set */
    size_t length = 0;
    int err = 0;
    char * MyKey;
    char * MyISE;	

    // Set the key
    if (argc > 0)
    {
        length = strlen(argv[1]);
        MyKey = new char [length + 1];
        strcpy (MyKey,argv[1]);
    }
    else
    {
        length = strlen("Bad_Key");
        MyKey = new char [length + 1];
        strcpy (MyKey,"Bad_Key");
    }

    // Set ISE file name
    if (argc > 1)
    {
        length = strlen(argv[2]);
        MyISE = new char [length + 1];
        strcpy (MyISE,argv[2]);
    }
    else
    {
        length = strlen("images/ralphie.jpg");
        MyISE = new char [length + 1];
        strcpy (MyISE,"images/ralphie.jpg");
    }

    /* Call the constructor with key and jpeg file */
    jpeg_ise decrypt_test(MyKey, MyISE);

    /* Dencryption */
    err = decrypt_test.decrypt_file();
    if (err != 0) 
    {
	cout << "Test 21: Decryption failed on file: " << MyISE << endl;
    }
	
    delete [] MyKey;
    delete [] MyISE;

    return 0;
}
