/*! @file CoverageReaderSkyeye.cc
 *  @brief CoverageReaderSkyeye Implementation
 *
 *  This file contains the implementation of the functions supporting
 *  reading the Skyeye coverage data files.
 */

#include <stdio.h>
#include <stdlib.h>
#include <sys/stat.h>

#include <iostream>
#include <fstream>
#include <iomanip>

#include "CoverageReaderSkyeye.h"
#include "CoverageMap.h"
#include "ExecutableInfo.h"
#include "skyeye_header.h"

namespace Coverage {

  CoverageReaderSkyeye::CoverageReaderSkyeye()
  {
  }

  CoverageReaderSkyeye::~CoverageReaderSkyeye()
  {
  }

  void CoverageReaderSkyeye::processFile(
    const std::string&    file,
    ExecutableInfo* const executableInformation
  )
  {
    CoverageMapBase* aCoverageMap = NULL;
    uintptr_t        baseAddress;
    uint8_t          cover;
    std::ifstream    coverageFile;
    prof_header_t    header;
    uintptr_t        i;
    uintptr_t        length;

    //
    // Open the coverage file and read the header.
    //
    coverageFile.open( file );
    if ( !coverageFile ) {
      std::ostringstream what;
      what << "Unable to open " << file;
      throw rld::error( what, "CoverageReaderSkyeye::processFile" );
    }

    coverageFile.read( (char *) &header, sizeof( header ) );
    if ( coverageFile.fail() ) {
      std::ostringstream what;
      what << "Unable to read header from " << file;
      throw rld::error( what, "CoverageReaderSkyeye::processFile" );
    }

    baseAddress = header.prof_start;
    length      = header.prof_end - header.prof_start;

    //
    // Read and process each line of the coverage file.
    //
    for ( i = 0; i < length; i += 8 ) {
      coverageFile.read( (char *) &cover, sizeof( uint8_t ) );
      if ( coverageFile.gcount() != sizeof( uint8_t ) ) {
        std::cerr << "CoverageReaderSkyeye::ProcessFile - breaking after 0x"
                  << std::hex << std::setfill( '0' )
                  << std::setw( 8 ) << i
                  << std::setfill( ' ' ) << std::dec
                  << " in " << file
                  << std::endl;
        break;
      }

      //
      // Obtain the coverage map containing the address and
      // mark the address as executed.
      //
      // NOTE: This method ONLY works for Skyeye in 32-bit mode.
      //
      if ( cover & 0x01 ) {
        aCoverageMap = executableInformation->getCoverageMap(
          baseAddress + i
        );

        if ( aCoverageMap ) {
          aCoverageMap->setWasExecuted( baseAddress + i );
          aCoverageMap->setWasExecuted( baseAddress + i + 1 );
          aCoverageMap->setWasExecuted( baseAddress + i + 2 );
          aCoverageMap->setWasExecuted( baseAddress + i + 3 );
        }
      }

      if ( cover & 0x10 ) {
        aCoverageMap = executableInformation->getCoverageMap(
          baseAddress + i + 4
        );
        if ( aCoverageMap ) {
          aCoverageMap->setWasExecuted( baseAddress + i + 4 );
          aCoverageMap->setWasExecuted( baseAddress + i + 5 );
          aCoverageMap->setWasExecuted( baseAddress + i + 6 );
          aCoverageMap->setWasExecuted( baseAddress + i + 7 );
        }
      }
    }

  }
}