/*
 *      Interactive disassembler (IDA).
 *      Version 3.05
 *      Copyright (c) 1990-95 by Ilfak Guilfanov.
 *      ALL RIGHTS RESERVED.
 *                              FIDO:   2:5020/209
 *                              E-mail: ig@estar.msk.su
 *
 */

#include "7900.hpp"
#include <ints.hpp>
#include <diskio.hpp>

#define far
#define near

static char *BitNamesCPU[] = { "IPL", "N", "V", "m", "x", "D", "I", "Z", "C" };

static char *BitNamesPUL[] = { "PS", "0", "DT", "DP0", "Y", "X", "B", "A" };

inline void SetDP0Plus(op_t &x)
{
    OutValue(x, OOFW_IMM | OOFW_16);
}

inline void GetNumPUL(ulong data)
{
  int bitOut=0;
  for(int i=0; i<8; i++)
  {

     if( GETBIT(data, i) == 1)
     {
      if(bitOut != 0)  out_symbol(',');
      out_register( BitNamesPUL[7-i] );
      if(bitOut == 0) bitOut++;

     }
  }

}


inline void GetNumDPRn(ulong data)
{
   char tmp[2];
  if(data==0x1 && data==0x2 && data==0x4 && data==0x8)
  {
     for(int i=0; i<4;i++)
     {
       if( GETBIT(data, i) == 1)
       {
        qsnprintf(tmp, sizeof(tmp), "%d", i);
        out_symbol(tmp[0]);
        break;
        }
     }
  }

  else
  {
     out_symbol('(');

     int bitOut=0;
     for(int i=0; i<4;i++)
     {
       if( GETBIT(data, i) == 1)
       {
         if(bitOut != 0)  out_symbol(',');
         qsnprintf(tmp, sizeof(tmp), "%d", i);
         out_symbol(tmp[0]);
         if(bitOut == 0) bitOut++;
       }
     }

     out_symbol(')');
  }
}


inline void GetCLPFlags(ulong data)
{
  int bitOut=0;
  for(int i=0; i<8;i++)
  {
    if( GETBIT(data, i) == 1)
    {
      if(bitOut != 0)  out_symbol(',');
      out_register( BitNamesCPU[8-i] );
      if(bitOut == 0) bitOut++;
    }
  }
}


inline void OutReg(int rgnum)
{
  out_register(ph.regNames[rgnum]);
}


//----------------------------------------------------------------------
static int near OutVarName(op_t &x)
{
  char *ptr;
  //Получть строку для данного лин. адресса
  if( (ptr=get_name_expr(cmd.ea + x.offb, x.n, toEA(cmd.cs, x.addr), x.addr) ) != NULL)
  {
    //вывод имен переменных и меток перехода
    OutLine(ptr);
    return(1);
  }
  return(0);
}

static int near getNumDPR(int iDPR )
{

  switch(iDPR)
   {
     case 0x0: return 0;
     case 0x40: return 1;
     case 0x80: return 2;
     case 0xC0: return 3;
   }
  return 0;

}

static void near OutDPRReg(ea_t &Addr, int gDPReg)
{
  if( gDPReg == 1)
   {
      char szTemp[5];
      ulong Data = Addr;
      Data &= 0xC0;
      qsnprintf(szTemp, sizeof(szTemp), "DP%d", getNumDPR( Data ) );
      out_register(szTemp);
      Addr &= 0xFF3F;
   }
   else
      out_keyword("DP0");
}

static ulong near GetValueDP(int DPR )
{
  if( getDPReg == 1)
  {
    switch(DPR)
    {
       case 0x0: return getDPR0;
       case 0x40: return getDPR1;
       case 0x80: return getDPR2;
       case 0xC0: return getDPR3;
    }
  }
  return 0;
}

static void near OutDPR( ulong Data )
{
   char tmp[20];

   ea_t Val = Data;
   OutDPRReg( Val, getDPReg);
   out_symbol(':');
   qsnprintf(tmp, sizeof(tmp), "%X", Val+GetValueDP( Data&0xC0 ) );
   out_line(tmp, COLOR_NUMBER);
}

static void near OutDT( ulong Data )
{
   char tmp[20];

   out_register("DT");
   out_symbol('+');
   out_symbol(':');

   qsnprintf(tmp, sizeof(tmp), "%X", Data );
   out_line(tmp, COLOR_NUMBER);
}


static void near OutIMM(ulong Data )
{
   char tmp[20];

    out_symbol('#');
    qsnprintf(tmp, sizeof(tmp), "%x", Data);
    out_line(tmp, COLOR_NUMBER);
}


static void near MOVRB()
{
   int i;
   ulong Val1, Val2;
   uchar code = get_byte(cmd.ea+1);
   //получаем старшую часть байта
   uchar nib  = (code >> 4) & 0xF;
   //получаем младшую часть байта
   uchar count = code & 0x0F;

  switch(nib)
  {
    case 0x0:
      {
 	for(i=0; i<count; i++)
	{
	   //Получить данные
           Val1 = get_byte( cmd.ea+2+(i*2) );//imm

           if(getFlag_M == 0)  Val2 = get_word( cmd.ea+2+(i*2)+1 );//imm
           else  Val2 = get_byte( cmd.ea+2+(i*2)+1 );//dd

           //Вывети эти данные
           //DPRxx
           OutDPR( Val2 );
           out_symbol(',');
           //imm
           OutIMM( Val1 );

           if(i != (count-1) ) out_symbol(',');

        }
        break;
      }//end switch 0x0



    case 0x1:
      {
 	for(i=0; i<count; i++)
	{

	   //Получить данные
           if(getFlag_M == 0)
           {
             Val2 = get_word( cmd.ea+2+(i*3) );//mmll
             Val1 = get_byte( cmd.ea+2+(i*3)+2 );//dd
           }
           else
           {
             Val2 = get_word( cmd.ea+2+(i*3) );//mmll
             Val1 = get_byte( cmd.ea+2+(i*3)+2 );//dd
           }

           //Вывети эти данные
           //DPRxx
           OutDPR( Val1 );
           out_symbol(',');
           //DPR
           OutDT( Val2 );
           out_symbol(',');
           OutReg(rX);


           if(i != (count-1) ) out_symbol(',');

        }
        break;
      }//end case 0x1

    case 0x2:
      {
 	for(i=0; i<count; i++)
	{
	   //Получить данные
           Val1 = get_byte( cmd.ea+2+(i*3) );//imm
           Val2 = get_word( cmd.ea+2+(i*3)+1 );//mmll


           //Вывети эти данные
           //DPRxx
           OutDT( Val2 );
           out_symbol(',');
           //IMM
           OutIMM( Val1 );

           if(i != (count-1) ) out_symbol(',');

        }
        break;
      }//end switch 0x2

    case 0x4:
      {
 	for(i=0; i<count; i++)
	{
	   //Получить данные
           Val1 = get_byte( cmd.ea+2+(i*2) );//dd1
           Val2 = get_byte( cmd.ea+2+(i*2)+1 );//dd2

           //Вывети эти данные
           //DPRxx
           OutDPR( Val2 );
           out_symbol(',');
           //DPRxx
           OutDPR( Val1 );

           if(i != (count-1) ) out_symbol(',');

        }
        break;
      }//end switch 0x4

    case 0x6:
      {
 	for(i=0; i<count; i++)
	{
	   //Получить данные
           Val1 = get_byte( cmd.ea+2+(i*3) );//imm
           Val2 = get_word( cmd.ea+2+(i*3)+1 );//mmll


           //Вывети эти данные
           //DPRxx
           OutDT( Val2 );
           out_symbol(',');
           //DPR
           OutDPR( Val1 );

           if(i != (count-1) ) out_symbol(',');

        }
        break;
      }//end switch 0x6


    case 0x7:
      {
 	for(i=0; i<count; i++)
	{

	   //Получить данные
           if(getFlag_M == 0)
           {
             Val2 = get_byte( cmd.ea+2+(i*3) );//mmll
             Val1 = get_word( cmd.ea+2+(i*3)+1 );//dd
           }
           else
           {
             Val2 = get_byte( cmd.ea+2+(i*3) );//mmll
             Val1 = get_word( cmd.ea+2+(i*3)+1 );//dd
           }

           //Вывети эти данные
           //DPRxx
           OutDT( Val1 );
           out_symbol(',');
           //DPR
           OutDPR( Val2 );
           out_symbol(',');
           OutReg(rX);

           if(i != (count-1) ) out_symbol(',');

        }
        break;
      }//end case 0x7

    case 0x8:
      {
 	for(i=0; i<count; i++)
	{
	   //Получить данные
           Val2 = get_word( cmd.ea+2+(i*3) );//mmll
           Val1 = get_byte( cmd.ea+2+(i*3)+2 );//dd


           //Вывети эти данные
           //DPRxx
           OutDPR( Val1 );
           out_symbol(',');
           //DT
           OutDT( Val2 );

           if(i != (count-1) ) out_symbol(',');

        }
        break;
      }//end switch 0x8

    case 0xA:
      {
 	for(i=0; i<count; i++)
	{
	   //Получить данные
           Val1 = get_word( cmd.ea+2+(i*4) );//imm
           Val2 = get_word( cmd.ea+2+(i*4)+2 );//mmll


           //Вывети эти данные
           //DPRxx
           OutDT( Val2 );
           out_symbol(',');
           //DT
           OutDT( Val1 );

           if(i != (count-1) ) out_symbol(',');

        }
        break;
      }//end switch 0xA

  }
}


static void near MOVR()
{
   int i;
   ulong Val1, Val2;
   uchar code = get_byte(cmd.ea+1);
   //получаем старшую часть байта
   uchar nib  = (code >> 4) & 0xF;
   //получаем младшую часть байта
   uchar count = code & 0x0F;

  switch(nib)
  {
    case 0x0:
      {
 	for(i=0; i<count; i++)
	{
	   //Получить данные
           Val2 = get_word( cmd.ea+2+(i*3) );//mmll
           Val1 = get_byte( cmd.ea+2+(i*3)+2 );//dd

           //Вывети эти данные
           //DPRxx
           OutDPR( Val1 );
           out_symbol(',');
           //DT
           OutDT( Val2 );
           out_symbol(',');
           OutReg(rX);

           if(i != (count-1) ) out_symbol(',');

        }
        break;
      }//end switch 0x0


    case 0x1:
      {
 	for(i=0; i<count; i++)
	{
	   //Получить данные
           if(getFlag_M == 0)
           {
               Val2 = get_word( cmd.ea+2+(i*3) );//imm
               Val1 = get_byte( cmd.ea+2+(i*3)+2 );//dd
           }
           else
           {
              Val2 = get_byte( cmd.ea+2+(i*2) );//imm
              Val1 = get_byte( cmd.ea+2+(i*2)+1 );//dd
           }

           //Вывети эти данные
           //DPRxx
           OutDPR( Val1 );
           out_symbol(',');
           //imm
           OutIMM( Val2 );

           if(i != (count-1) ) out_symbol(',');

        }
        break;
      }//end case 0x1


    case 0x3:
      {
 	for(i=0; i<count; i++)
	{
	   //Получить данные
           if(getFlag_M == 0)
           {
             Val2 = get_word( cmd.ea+2+(i*4) );//imm
             Val1 = get_word( cmd.ea+2+(i*4)+2 );//llmm
           }
           else
           {
             Val2 = get_byte( cmd.ea+2+(i*3) );//imm
             Val1 = get_word( cmd.ea+2+(i*3)+1 );//llmm
           }

           //Вывети эти данные
           //DPRxx
           OutDT( Val1 );
           out_symbol(',');
           //IMM
           OutIMM( Val2 );

           if(i != (count-1) ) out_symbol(',');
        }
        break;
      }//end case 0x3

    case 0x5:
      {
 	for(i=0; i<count; i++)
	{

           Val2 = get_byte( cmd.ea+2+(i*2) );//dd
           Val1 = get_byte( cmd.ea+2+(i*2)+1 );//dd

           //Вывети эти данные
           //DPRxx
           OutDPR( Val1 );
           out_symbol(',');
           //DPR
           OutDPR( Val2 );

           if(i != (count-1) ) out_symbol(',');
        }
        break;
      }//end case 0x5

    case 0x6:
      {
 	for(i=0; i<count; i++)
	{
	   //Получить данные
           Val1 = get_byte( cmd.ea+2+(i*3) );//imm
           Val2 = get_word( cmd.ea+2+(i*3)+1 );//mmll


           //Вывети эти данные
           //DPRxx
           OutDT( Val2 );
           out_symbol(',');
           //DPR
           OutDPR( Val1 );
           out_symbol(',');
           OutReg(rX);

           if(i != (count-1) ) out_symbol(',');

        }
        break;
      }//end switch 0x6


    case 0x7:
      {
 	for(i=0; i<count; i++)
	{

	   //Получить данные
           if(getFlag_M == 0)
           {
             Val2 = get_byte( cmd.ea+2+(i*3) );//mmll
             Val1 = get_word( cmd.ea+2+(i*3)+1 );//dd
           }
           else
           {
             Val2 = get_byte( cmd.ea+2+(i*3) );//mmll
             Val1 = get_word( cmd.ea+2+(i*3)+1 );//dd
           }

           //Вывети эти данные
           //DPRxx
           OutDT( Val1 );
           out_symbol(',');
           //DPR
           OutDPR( Val2 );


           if(i != (count-1) ) out_symbol(',');

        }
        break;
      }//end case 0x7


    case 0x9:
      {
 	for(i=0; i<count; i++)
	{

	   //Получить данные
           if(getFlag_M == 0)
           {
             Val2 = get_word( cmd.ea+2+(i*3) );//mmll
             Val1 = get_byte( cmd.ea+2+(i*3)+2 );//dd
           }
           else
           {
             Val2 = get_word( cmd.ea+2+(i*3) );//mmll
             Val1 = get_byte( cmd.ea+2+(i*3)+2 );//dd
           }

           //Вывети эти данные
           //DPRxx
           OutDPR( Val1 );
           out_symbol(',');
           //DPR
           OutDT( Val2 );


           if(i != (count-1) ) out_symbol(',');

        }
        break;
      }//end case 0x9

    case 0xB:
      {
 	for(i=0; i<count; i++)
	{

	   //Получить данные
           Val2 = get_word( cmd.ea+2+(i*4) );//imm
           Val1 = get_word( cmd.ea+2+(i*4)+2 );//llmm

           //Вывети эти данные
           //DT
           OutDT( Val1 );
           out_symbol(',');
           //DT
           OutDT( Val2 );

           if(i != (count-1) ) out_symbol(',');

        }
        break;
      }//end case 0xB


  }//end switch
}

static void near MOV(op_t &x)
{
  switch(x.TypeOper)
  {
     case m7900_movrb: MOVRB(); break;
     case m7900_movr:  MOVR();  break;
     default:
      //msg("out: %a: bad prefix %d\n", cmd.ip, RAZOPER);
      break;
  }
}

//----------------------------------------------------------------------
bool outop(op_t &x)
{
  switch(x.type)
  {
  case o_void:
    return 0;


  case o_reg:
     //Вывод регистра по номеру в регистре
     OutReg(x.reg);
    break;

  case o_phrase:
       OutLine(ph.regNames[x.reg]);
    break;

  case o_ab:
     switch(x.TypeOper)
     {
       case TAB_L_INDIRECTED_ABS:
           out_symbol('L');

       case TAB_INDIRECTED_ABS:
    	   //Вывод данных
           out_symbol('(');
           if(!OutVarName(x) )
              OutValue(x, OOF_ADDR | OOFS_NOSIGN);
           out_symbol(')');
           break;


       case TAB_INDIRECTED_ABS_X:
    	   //Вывод данных
           out_symbol('(');

           if(!OutVarName(x) )
              OutValue(x, OOF_ADDR | OOFS_NOSIGN);

           out_symbol(',');
           OutReg(rX);
           out_symbol(')');

           break;

       case TAB_ABS_Y:
       case TAB_ABS_X:
       case TAB_ABS:
         {
           out_register("DT");
       	   out_symbol(':');

    	   //Вывод данных
           if(!OutVarName(x) )
              OutValue(x, OOF_ADDR | OOFS_NOSIGN | OOFW_32);
         }
        break;

       case TAB_ABL_X:
       case TAB_ABL:
       {
           out_register("LG");
       	   out_symbol(':');

    	   //Вывод данных
           if(!OutVarName(x) )
              OutValue(x, OOF_ADDR | OOFS_NOSIGN | OOFW_32);
       }
        break;
     }
    break;

  case o_sr:
      if( x.TypeOper == TSP_INDEX_SP_Y)
        out_symbol('(');

      if( x.xmode == IMM_32)
         OutValue(x, OOFW_IMM | OOFW_32);
      else if( x.xmode == IMM_16)
         OutValue(x, OOFW_IMM | OOFW_16);
      else
         OutValue(x, OOFW_IMM);

      if( x.TypeOper == TSP_INDEX_SP_Y)
      {
        out_symbol(',');
        OutReg(rPS);
        out_symbol(')');
      }
    break;


  case o_stk:
     {  //Вывод данных(тип o_stk)
        //Есть отдельные случаи которые мы и разберем
        switch(cmd.itype)
        {
           case m7900_pei: SetDP0Plus(x); break;
           case m7900_psh:
           case m7900_pul: GetNumPUL( x.value ); break;

           default:
             out_symbol('#');
             OutValue(x, OOFW_IMM | OOFS_NOSIGN);
           break;
         }//end switch(cmd.itype)
       }
    break;//case o_stk



  case o_imm:
     {  //Вывод данных(тип o_imm)

        //Есть отдельные случаи которые мы и разберем
        switch(cmd.itype)
        {
           case m7900_sep://Set Processor status
           case m7900_clp://CLear Processor status
                GetCLPFlags(x.value);
             break;

           case m7900_lddn:
           case m7900_tdan:
           case m7900_phdn:
           case m7900_rtsdn:
           case m7900_pldn:
           case m7900_rtld:
           case m7900_phldn:
                GetNumDPRn(x.value);
             break;
           case m7900_bsc:
           case m7900_bss:
                OutValue(x, OOFW_IMM);
             break;

           default:
             {
                out_symbol('#');
                OutValue(x, OOFW_IMM);
             }
             break;
         }//end switch(i.type)
       }
    break;//case o_imm



  case o_mem:
     //выводит имя переменной из памяти(например byte_98)
     {
      if(x.TypeOper == m7900_movr || x.TypeOper == m7900_movrb)
      {
         MOV(x);
         break;
      }

      switch(x.TypeOper)
      {
        case TDIR_DIR_Y:
        case TDIR_DIR_X:
        case TDIR_DIR:
         {
    	   OutDPRReg(x.addr, getDPReg );
 	   out_symbol(':');
            //Вывод имени переменной
           if(!OutVarName(x) )
              OutValue(x, OOF_ADDR |OOF_NUMBER| OOFS_NOSIGN);
        }
        break;

       case TDIR_L_INDIRECT_DIR_Y:
       case TDIR_L_INDIRECT_DIR:
           out_symbol('L');

       case TDIR_INDIRECT_DIR_Y:
       case TDIR_INDIRECT_DIR:
            out_symbol('(');
     	    OutDPRReg(x.addr, getDPReg );
 	    out_symbol(':');
             //Вывод имени переменной
            if(!OutVarName(x) )
               OutValue(x, OOF_ADDR |OOF_NUMBER| OOFS_NOSIGN);
            out_symbol(')');
        break;

       case TDIR_INDIRECT_DIR_X:
       {
            out_symbol('(');

    	   OutDPRReg(x.addr, getDPReg );
 	   out_symbol(':');
            //Вывод имени переменной
           if(!OutVarName(x) )
              OutValue(x, OOF_ADDR |OOF_NUMBER| OOFS_NOSIGN);

            out_symbol(',');
            OutReg( rX);
            out_symbol(')');
       }
        break;


      }//switch TypeDIR
   }
    break;


  case o_near:
  {
    ulong v;
    const char *ptr;

    //Получить линейный адрес
    v = toEA(cmd.cs,x.addr);
    ptr = get_name_expr(cmd.ea+x.offb, x.n, v, x.addr);
    if( ptr == NULL )
    {
     //Вывести значение
     OutValue(x, OOF_ADDR | OOFS_NOSIGN );
     //QueueMark(Q_noName, cmd.ea);
    }
    else
     OutLine(ptr);
   }
    break;

  default:
    //warning("out: %a: bad optype %d", cmd.ip, x.type);
    break;
  }

  return(1);
}

static char* near GetPrefics( int Raz)
{

    switch(Raz)
    {
      case INSN_PREF_B: return ".b";
      case INSN_PREF_W: return "";
      case INSN_PREF_D: return ".d";
      case INSN_PREF_U: return "";
      default:
          //msg("out: %a: bad prefix %d\n", cmd.ip, RAZOPER);
          break;
    }

   return "";
}

//----------------------------------------------------------------------
void out(void) {
  char buf[MAXSTR];

   init_output_buffer(buf, sizeof(buf));       // setup the output pointer
   OutMnem(8, GetPrefics(RAZOPER));            // output instruction mnemonics

   out_one_operand(0);                   // output the first operand

    //Вывод операнда
   if(cmd.Op2.type != o_void) {
     out_symbol(',');//вывод разделителя между операндами
     //если неуказан флаг UAS_NOSPA ставим пробел
     if(!(ash.uflag & UAS_NOSPA)) OutChar(' ');
     out_one_operand(1);
   }

   if(cmd.Op3.type != o_void) {
     out_symbol(',');
     if(!(ash.uflag & UAS_NOSPA)) OutChar(' ');
     out_one_operand(2);
   }

   if(cmd.Op4.type != o_void) {
     out_symbol(',');
     if(!(ash.uflag & UAS_NOSPA)) OutChar(' ');
     out_one_operand(3);
   }


   if(cmd.Op5.type != o_void) {
     out_symbol(',');
     if(!(ash.uflag & UAS_NOSPA)) OutChar(' ');
     out_one_operand(4);
   }


   if(isVoid(cmd.ea, uFlag, 0)) OutImmChar(cmd.Op1);
   if(isVoid(cmd.ea, uFlag, 1)) OutImmChar(cmd.Op2);
   if(isVoid(cmd.ea, uFlag, 2)) OutImmChar(cmd.Op3);
   if(isVoid(cmd.ea, uFlag, 3)) OutImmChar(cmd.Op4);
   if(isVoid(cmd.ea, uFlag, 4)) OutImmChar(cmd.Op5);


   term_output_buffer();
   gl_comm = 1;
   MakeLine(buf);

}

//--------------------------------------------------------------------------
void header(void) {
  gen_cmt_line("Processor:            %s [%s]", inf.procName, device);
  gen_cmt_line("Target assebler: %s", ash.name);
  if (ash.header != NULL )
    for (const char **ptr=ash.header; *ptr != NULL; ptr++ ) MakeLine(*ptr, 0);
}

//--------------------------------------------------------------------------
// generate segment header
void gen_segm_header(ea_t ea)
{
    segment_t *Sarea = getseg(ea);
    char *segname = get_segm_name(Sarea);

    if (ash.uflag & UAS_SEGM)
        printf_line(-1, COLSTR("SEGMENT %s", SCOLOR_ASMDIR), segname);
    else
        printf_line(-1, COLSTR(".SECTION %s", SCOLOR_ASMDIR), segname);

    unsigned long orgbase = ea - get_segm_para(Sarea);

    if (orgbase != 0)
        printf_line(-1, COLSTR("%s %s", SCOLOR_ASMDIR),
            ash.origin, btoa(orgbase));
}

//--------------------------------------------------------------------------
void footer(void)
{
  char buf[MAXSTR];
  char *const end = buf + sizeof(buf);
  if (ash.end != NULL)
  {
    MakeNull();
    char *p = tag_addstr(buf, end, COLOR_ASMDIR, ash.end);
    char name[MAXSTR];
    if (get_colored_name(BADADDR, inf.beginEA, name, sizeof(name)) != NULL)
    {
      register int i = strlen(ash.end);
      do APPCHAR(p, end, ' '); while(++i < 8);
      APPEND(p, end, name);
    }
    MakeLine(buf);
  } else gen_cmt_line("end of file");
}

//--------------------------------------------------------------------------