Subversion Repositories or2k

/*

  File for analysis of an architecture's instructions in binary format

  Julius Baxter, julius.baxter@orsoc.se

*/

#include <ctype.h>

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include <stdint.h>

#include <unistd.h>

#include <arpa/inet.h> // for htonl()

#include "insnanalysis.h"

// Access to list management functions

#include "insn-lists.h"

int analyse_insn(instruction insn,

                 instruction_properties *insn_props)

{

  // For now no other instruction sets supported

  return or1k_32_analyse_insn(insn, insn_props);

}

void print_insn(instruction_properties *insn_props)

{

  if (insn_props->insn_string != NULL)

    printf("%s", insn_props->insn_string);

}

void collect_stats(instruction insn,

                   instruction_properties *insn_props, int record_unique)

{

  or1k_32_collect_stats(insn, insn_props, record_unique);

}

void generate_stats(FILE * stream)

{

  or1k_32_generate_stats(stream);

}

int main(int argc, char *argv[])

{

  FILE *ifp, *ofp;

  char insn_buff[INSN_SIZE_BYTES]; // Buffer for instruction data

  int insns_seen_total = 0; // Keep track of total number of instructions read

  char *infilename = NULL;

  char *outfilename = NULL;

  int outfile_flag = 0; // default is not to use an output file

  int opterr = 0;

  int uniquecount_flag = 0;

  int c;

  // Look for:

  // -f filename : Input file name (required)

  // [ -o filename] : Output file name (optional)

  // [ -u ] : Do unique instruction analysis (option)

  while ((c = getopt(argc, argv, "f:o:u::")) != -1)

    switch (c)

      {

      case 'f':

        infilename = optarg;

        break;

      case 'o':

        outfile_flag = 1;

        outfilename = optarg;

        break;

      case 'u':

        uniquecount_flag = 1;

        break;

      default:

        abort();

      }

  // Try to open the input file

  if((ifp = fopen(infilename, "rb"))==NULL) {

    printf("Cannot open input file, %s\n", infilename);

    exit(1);

  }

  int filesize_bytes, filesize_insns;

  // Determine filesize

  if ( fseek(ifp, 0, SEEK_END))

    {

      fclose(ifp);

      fprintf(stderr, "Error detecting input file size\n");

      return -1;

    }

  filesize_bytes = ftell(ifp);

  filesize_insns = filesize_bytes / INSN_SIZE_BYTES;

#ifdef DISPLAY_STRING  

  printf("\nAnalysing file:\t%s\tSize: %d MB\n",

         infilename,((filesize_bytes/1024)/1024));

#endif

  // Reset pointer

  rewind(ifp);

  instruction * insn = (instruction *)insn_buff;

  instruction_properties insn_props;

  // Go through the file, collect stats about instructions

  // What is one-percent of instructions

  float file_one_percent = ((float)filesize_insns / 100.0f );

  float percent_of_percent=0; int percent=0;

  insn_lists_init();

  while(!feof(ifp)) {

    if (fread(insn_buff, INSN_SIZE_BYTES, 1, ifp) != 1)

      break;

    // Endianness is little when read in from binary file created with 

    // or32-elf-objcopy, so swap;

    *insn = htonl(*insn);

    if (*insn == 0) // most probably dead space in binary, skip

      continue;

    reset_instruction_properties(&insn_props);

    if (analyse_insn(*insn, &insn_props) == 0)

      {

        insns_seen_total++;

        collect_stats(*insn, &insn_props, uniquecount_flag);

      }

    else

      {

        // Non-zero return from analyse_insn(): problem analysing instruction.

        // Is a NOP for now, but do something here if needed

        do{ } while(0);

      }

    // Progress indicator

    percent_of_percent += 1.0f;

    if (percent_of_percent >= file_one_percent)

      {

        percent++;

        fprintf(stderr, "\r%d%%", percent);

        percent_of_percent = 0;

      }

  }

  fclose(ifp);

  fprintf(stderr, "\rDone\n", percent);

  // Try to open the output file

  if (outfile_flag)

    {

      if((ofp = fopen(outfilename, "wb+"))==NULL) {

        printf("Cannot open output file.\n");

        exit(1);

      }

    }

  else

    // Otherwise we'll use stdout

    ofp = (FILE *)stdout;

#ifdef DISPLAY_STRING

  fprintf(ofp, "Saw %d instructions\n", insns_seen_total);

#endif

#ifdef DISPLAY_CSV

  fprintf(ofp, "\"File:\",\"%s\",\"Num insns:\",%d,\n",

          infilename, insns_seen_total);

#endif

  // Generate output

  generate_stats(ofp);

  insn_lists_free();

  return 0;

}