/**
 * FailT32 -- Fault Injection on the Lauterbach Trace32 System
 *
 * 1. Invoke t32 executable with appropriate Lauterbach Skript
 *  - Script has to load binary
 *  - and let system run until entry point
 *  -> Then we have a readily configured system
 *
 * @author Martin Hoffmann <hoffmann@cs.fau.de>
 * @date 15.02.2013
 */

#include <t32.h>
#include <iostream>
#include <string.h>
#include <stdlib.h>
#include <stdio.h>
#include <vector>
#include "config/VariantConfig.hpp"
#include "sal/SALInst.hpp"
#include "util/optionparser/optionparser.h"
#include "util/optionparser/optionparser_ext.hpp"

#include "T32Connector.hpp"
#include "T32Tracer.hpp"
#include "t32config.hpp"
#include "sal/MemoryInstruction.hpp"
#include "util/Disassembler.hpp"

using namespace std;
using namespace fail;

static T32Connector t32;

//!< Program options
enum  optionIndex { UNKNOWN, HELP, RUN, T32HOST, PORT, PACKLEN };
const option::Descriptor usage[] =
{
 {UNKNOWN, 0,"" , ""    ,Arg::None, "USAGE: fail-client [options]\n\nATTENTION:\nOnly setup TCP Port/Packlen here, if you really know what you are doing!\nIt is safer use the  CMake configuration (ccmake <BUILDDIR>).\n\n"
                                            "Options:" },
 {HELP,    0,"" , "help",Arg::None, "  --help  \tPrint usage and exit." },
 {RUN,    0,"r", "run",Arg::Required, "  --run, -r  \tLauterbach script to startup system." },
 {T32HOST,    0,"t", "trace32-host",Arg::Required, "  --trace32-host, -t <hostname>  \tHostname/IP Address of the Trace32 instance. (default: localhost)" },
 {PORT,    0,"p", "port",Arg::Required, "  --port <NUM>, -p <NUM> \tTCP Port. (default: " T32_PORTNUM  ")" },
 {PACKLEN,    0,"l", "packet-length",Arg::Required, "  --packet-length, -l <NUM> \tPacket length. (default: " T32_PACKLEN  ")" },
 {0,0,0,0,0,0}
};

/* Here we go... */
int main(int argc, char** argv){
  //----------------------------------------------
  // Evaluate arguments
  argc-=(argc>0); argv+=(argc>0); // skip program name argv[0] if present
  option::Stats stats(usage, argc, argv);
  option::Option options[stats.options_max], buffer[stats.buffer_max];
  option::Parser parse(usage, argc, argv, options, buffer);

  if (parse.error()){
    cerr << "Error parsing arguments." << endl;
    return 1;
  }

  if ( options[HELP] ) // || argc == 0 || options[RUN].count() == 0) // to enforce -s
  {
    int columns = getenv("COLUMNS")? atoi(getenv("COLUMNS")) : 80;
    option::printUsage(fwrite, stdout, usage, columns);
    return 0;
  }

  if(options[RUN].count()){
    t32.setScript(options[RUN].first()->arg);
  }

  if(options[T32HOST].count()){
    t32.setHostname(options[T32HOST].first()->arg);
  }

  if(options[PORT].count()){
    t32.setPort(options[PORT].first()->arg);
  }

  if(options[PACKLEN].count()){
    t32.setPacketlength(options[PACKLEN].first()->arg);
  }
  //----------------------------------------------
  // Initialize T32
  if(t32.startup() == false){
    cout << "Could not connect to Lauterbach :(" << endl;
    return -1;
  }


  // Let the SimulatorController do the dirty work.
  fail::simulator.startup();

  // Here, we come back after any experiment called a resume
  // Start execution of the SUT.
  // The experiments/traces hopefully set some Breakpoints, we can react on.
  // We may also provide a timeout, if a TimerListener was set wanted.

   address_t ip;

   // Enable T32 Tracer (if available)
   T32Tracer tr("/tmp/tr.x"); // TODO configurable trace file path
   tr.setup(); // enable and configure tracing.

   while(1) {
        // Start execution (with next timeout, if any)
        t32.go();
        // Wait for debugger to stop.
        while( t32.isRunning() ) {}

        // Call appropriate callback of the SimulatorController.
        ip = fail::simulator.getCPU(0).getInstructionPointer();
        fail::simulator.onBreakpoint(&fail::simulator.getCPU(0), ip , fail::ANY_ADDR);

        // Evaluate tracing result, handle memory access event..
        if((tr.evaluate() > 0) && (tr.wasDataAccess())){
          MemoryInstruction mem;
          T32Tracer::const_record_iterator it = tr.begin();

          // step back to current PC
          for( ; it != tr.end(); ++it) {
            if(it->isProgram() && (it->getAddress() == ip)){
              break;
            }
          }
          // find latest memory access
          for( ; it != tr.end(); ++it){
            // iterate records till there is a memory access..
            if(it->isProgram() && meminstruction.eval(it->getAddress(), mem)){ // skip data access in between
              ip = it->getAddress(); // store PC of the mem access.
              break; // yay, we found a memory access.
            }
          }
          if(mem.isValid()){ // mem is not valid, if we did not leave the for loop via the break statement.. -> no mem access found :( XXX this should not happen.
            fail::simulator.onMemoryAccess(&fail::simulator.getCPU(0), tr.getLatestRecord().getAddress(), mem.getWidth(), tr.getLatestRecord().isDataWrite(), ip );
          }
        }
    }

  cout << "[T32 Backend] After startup" << endl;
  return 0;
}