boschme7x.cpp

/*
 *  This is a sample plugin module
 *
 *  It can be compiled by any of the supported compilers:
 *
 *      - Borland C++, CBuilder, free C++
 *      - Watcom C++ for DOS32
 *      - Watcom C++ for OS/2
 *      - Visual C++
 *
 */

//Standard Defs
typedef int     BOOL;
#define FALSE   0
#define TRUE    1
#define NULL    0

#include <ida.hpp>
#include <idp.hpp>
#include <expr.hpp>
#include <loader.hpp>
#include <name.hpp>

//#include <bytes.hpp>
//#include <kernwin.hpp>
//#include <name.hpp>
//#include <offset.hpp>
//#include <search.hpp>

#include "BoschHelper.h"

extern plugin_t PLUGIN;

BoschHelper	boschcode; //our class

//--------------------------------------------------------------------------
// Example of a user-defined IDC function in C++

static const char myBoschME7xfunc5_args[] = { VT_LONG, VT_STR, 0 };
static error_t idaapi myfunc5(idc_value_t *argv, idc_value_t *res)
{
	msg("myBoschME7xfunc is called with arg0=%x and arg1=%s\n", argv[0].num, argv[1].c_str());
  res->num = 5;     // let's return 5
  return eOk;
}

//--------------------------------------------------------------------------
// This callback is called for UI notification events
static int idaapi BoschME7x_callback(void * /*user_data*/, int event_id, va_list /*va*/)
{
  if ( event_id != ui_msg )     // avoid recursion
	  if (event_id != ui_refreshmarked ) // ignore uninteresting events
                    msg("ui_callback %d\n", event_id);
  return 0;                     // 0 means "process the event"
                                // otherwise the event would be ignored
}

//--------------------------------------------------------------------------
// A sample how to generate user-defined line prefixes
static const int prefix_width = 8;

static void get_user_defined_prefix(ea_t ea,
                                    int lnnum,
                                    int indent,
                                    const char *line,
                                    char *buf,
                                    size_t bufsize)
{
  buf[0] = '\0';        // empty prefix by default

  // We want to display the prefix only the lines which
  // contain the instruction itself

  if ( indent != -1 ) return;           // a directive
  if ( line[0] == '\0' ) return;        // empty line
  if ( tag_advance(line,1)[-1] == ash.cmnt[0] ) return; // comment line...

  // We don't want the prefix to be printed again for other lines of the
  // same instruction/data. For that we remember the line number
  // and compare it before generating the prefix

  static ea_t old_ea = BADADDR;
  static int old_lnnum;
  if ( old_ea == ea && old_lnnum == lnnum ) return;

  // Ok, seems that we found an instruction line.

  // Let's display the size of the current item as the user-defined prefix
  asize_t our_size = get_item_size(ea);

  // We don't bother about the width of the prefix
  // because it will be padded with spaces by the kernel

  qsnprintf(buf, bufsize, " %d", our_size);

  // Remember the address and line number we produced the line prefix for:
  old_ea = ea;
  old_lnnum = lnnum;

}

//--------------------------------------------------------------------------
//
//      Initialize.
//
//      IDA will call this function only once.
//      If this function returns PLGUIN_SKIP, IDA will never load it again.
//      If this function returns PLUGIN_OK, IDA will unload the plugin but
//      remember that the plugin agreed to work with the database.
//      The plugin will be loaded again if the user invokes it by
//      pressing the hotkey or selecting it from the menu.
//      After the second load the plugin will stay on memory.
//      If this function returns PLUGIN_KEEP, IDA will keep the plugin
//      in the memory. In this case the initialization function can hook
//      into the processor module and user interface notification points.
//      See the hook_to_notification_point() function.
//
//      In this example we check the input file format and make the decision.
//      You may or may not check any other conditions to decide what you do:
//      whether you agree to work with the database or not.
//
int idaapi init(void)
{
  if ( inf.filetype == f_ELF ) return PLUGIN_SKIP;

// Please uncomment the following line to see how the notification works
//  hook_to_notification_point(HT_UI, sample_callback, NULL);
//  PLUGIN.flags &= ~PLUGIN_UNL;

// Please uncomment the following line to see how to the user-defined prefix works
//  set_user_defined_prefix(prefix_width, get_user_defined_prefix);

// Please uncomment the following line to see how to define IDC functions
//  set_idc_func("MyBoschME7xFunc5", myfunc5, myfunc5_args);

  const char *options = get_plugin_options("BoschME7x");
  if ( options != NULL )
    warning("command line options: %s", options);

  return (PLUGIN.flags & PLUGIN_UNL) ? PLUGIN_OK : PLUGIN_KEEP;
}

//--------------------------------------------------------------------------
//      Terminate.
//      Usually this callback is empty.
//      The plugin should unhook from the notification lists if
//      hook_to_notification_point() was used.
//
//      IDA will call this function when the user asks to exit.
//      This function won't be called in the case of emergency exits.

void idaapi term(void)
{
  //unhook_from_notification_point(HT_UI, BoschME7x_callback);
  set_user_defined_prefix(0, NULL);

  del_idc_func("MyBoschME7xFunc5");
}

//--------------------------------------------------------------------------
//
//      The plugin method
//
//      This is the main function of plugin.
//
//      It will be called when the user selects the plugin.
//
//              arg - the input argument, it can be specified in
//                    plugins.cfg file. The default is zero.
//
//

bool idaapi run(size_t arg)
{
	unsigned short BOSCH = 0x01;

	unsigned short newBosch = 0x01;
	unsigned short defSegs = 0x02;
	unsigned short findSandFuncs =0x04;
	unsigned short DissAsm = 0x08;
	unsigned short findDTC =0x10;
	unsigned short createOffsets=0x20;
	unsigned short cfg_box_2_answer=0x00;
	char label[MAXSTR] = "";
	BOOL bNewME711 = FALSE;

	static const char cfg_box_2 []=
		"Select your option for BOSCH Processor\n"
		"<Is this a new Bosch ME7.1.1? (Data table at 0xe00000):C>\n"
		"<Define segments? (WARNING this Will delete your existing disassembly!!!):C>\n"
		"<Find Standard Functions and Comment Them?:C >\n"
		"<Disassemble into asm code :C>\n"
		"<Find DTC Flag Settings:C>\n"
		"<Find and Create Offsets:C>>\n";

	if (BOSCH)
	{
		if (! ask_form(cfg_box_2, &cfg_box_2_answer) )
	    {
			return false;
		}
	}



//  if ( inf.filetype != f_PE ) return PLUGIN_SKIP; // only for PE files
//  ph.id = PLFM_C166
	msg("myBoschME7xfunc - processor is %s, inf.filetype is %d, ph.id is %d\n", inf.procname, inf.filetype, ph.id);
    msg("just fyi: the current screen address is: %a\n", get_screen_ea());
	msg("newBosch =x%x\n ",newBosch);

 // if ( !autoIsOk()&& askyn_c(-1, "HIDECANCEL\n"
//       "The autoanalysis has not finished yet.\n"
//       "The result might be incomplete. Do you want to continue?") < 0 )//
//	   return;



	if(newBosch & cfg_box_2_answer)
		bNewME711 = TRUE;
	else
		bNewME711 = FALSE;

	using namespace std;

	string sECU  = "";

	if (bNewME711)
	{
		sECU = "bNewME711";
		//sECU ="ME761Astra";
		msg("ME711\n");
	}

	if (defSegs & cfg_box_2_answer)
		{
		msg("Calling MakeSegments()\n");
		boschcode.MakeSegments(sECU);
		}

	if(findSandFuncs & cfg_box_2_answer)
		{
		msg("Calling SearchForFuncSigsAndThenCmt()\n");
		boschcode.SearchForFuncSigsAndThenCmt(sECU);
		}
	
	if(DissAsm & cfg_box_2_answer)
		{
		msg("Calling MakeDissCode()\n");
		boschcode.MakeDissCode(sECU);
		}
	
	if (findDTC & cfg_box_2_answer)
		{
		msg("Calling SearchForDTCFlagSetting()\n");
		boschcode.SearchForDTCFlagSetting(sECU);
		}
	
	if (createOffsets & cfg_box_2_answer)
		{
		msg("Calling SearchForArrayOffsetsAndThenCreate()\n");
		boschcode.SearchForArrayOffsetsAndThenCreate(sECU);
		}
	

	msg("BoschMe7x finished.\n");
	return true;
}

//--------------------------------------------------------------------------
char comment[] = "BoschME7x v2.1 - Assists in the disassembly of ME7.x ECUs";

char help[] =
        "BoschME7x plugin module\n"
        "\n"
        "This module assists the user in disassembling Bosch ME7.x ECUs.\n"
        "\n"
        "It correctly sets up IDA with the ECU addresses and segments. Additionally,\n"
        "it auto disassembles and identifies key routines within the binary.\n";


//--------------------------------------------------------------------------
// This is the preferred name of the plugin module in the menu system
// The preferred name may be overriden in plugins.cfg file

char wanted_name[] = "BoschME7x";


// This is the preferred hotkey for the plugin module
// The preferred hotkey may be overriden in plugins.cfg file
// Note: IDA won't tell you if the hotkey is not correct
//       It will just disable the hotkey.

char wanted_hotkey[] = "Alt-1";


//--------------------------------------------------------------------------
//
//      PLUGIN DESCRIPTION BLOCK
//
//--------------------------------------------------------------------------
plugin_t PLUGIN =
{
  IDP_INTERFACE_VERSION,
  PLUGIN_UNL,           // plugin flags
  init,                 // initialize

  term,                 // terminate. this pointer may be NULL.

  run,                  // invoke plugin

  comment,              // long comment about the plugin
                        // it could appear in the status line
                        // or as a hint

  help,                 // multiline help about the plugin

  wanted_name,          // the preferred short name of the plugin
  wanted_hotkey         // the preferred hotkey to run the plugin
};