/*



      Gaara virus disinfector type 1
  --------------------------------------
  by Piotr Bania <bania.piotr@gmail.com>
  http://www.piotrbania.com
  
  
  Details:

  Firs of all, the disinfector checks if the GAARA is already resident
  if so you may want to clear the RAM memory. And this seems to be 
  the best way of disinfection but you will loose all your RAM programs
  unless they are archived. Anyway i made an another approach.
  Which i hope can teach you how to clear the virus.

  This little file, is going prevent Gaara virus from execution.
  It simply checks all variables (files) for the marker occurency
  then if it is found, we are searching for the const bytes of gaara
  body within the virus execution starts. Then we patch those bytes
  with orginal bytes from the host. Due to that fact, Gaara will not
  get executed, and the execution flow returns to the orginal program.
  Finally the GAARA marker is erased from the orginal file. Well
  i could also re-design the heap memory and change the size but i
  was too lazy to do so. Nothing more like few minutes code.

*/



#include <tigcclib.h>

#define get_romb(x)          ({(long)x = *(long*)0xC8;}) 
#define LOW_GAARA_SIZE      400



/* signatures */
unsigned char   sig_gaara_mark[]  = {'G','A','A','R','A','X'};
unsigned char   sig_gaara_start[] = {0x4E, 0x71, 0x4E, 0x71, 0x48, 0xE7, 0xFF, 0xFE};


/* repair */
unsigned char   repair_bytes[]    =  {0x4E, 0x5E, 0x4E, 0x75};

unsigned short  gaara_m_offset     = 0;
int              d_files            = 0;


void is_infected(void)
{
  unsigned long a;
  get_romb(a);
    
  if (!(a & 0xE00000))
  {
    printf("Disg: Seems the ROMCALL-JUMP-TABLE is redirected\n");
    printf("Disg: Please reset RAM to make it clean.\n");

    /* well i should rewrite the ROMCALL-JUMP-TABLE ptr here by hand
    but i guess it would be a cool job for you :)*/

    ngetchx();    
    return;
  }
  
  printf("Disg: Seems the RAM is clean\n");  
}


unsigned short get_sig_off(void *mem, unsigned short size, unsigned char *sig, int sig_s)
{
  
  int i,ii, found;
  for (i = 0; i < (size-3); i++)
  {  
    found = 1;
    for (ii = 0; ii < sig_s; ii++)
    {
      if (*(unsigned char*)(mem+i+ii) != sig[ii])
      {
        found = 0;
        goto next;        
      }
    }    

next:    
    if (found == 1)
    {
      if (*(char*)(mem+i+ii) == 'X')
        return 0;      // to avoid fail on itself
      return i;    
    }
      
  }  
  
  return 0;
}


int is_gaara(void *mem, unsigned short size)
{
  int i;
  unsigned long offset;
  
  printf("Disg: Scaning for the infection mark\n");

  gaara_m_offset = get_sig_off(mem,size,&sig_gaara_mark,sizeof(sig_gaara_mark)-1);
  if (gaara_m_offset != 0)
  {
          printf("Disg: File is infected with Gaara, trying to repair\n");
          return 1;      
  }

  printf("Disg: File is clean\n");
  return 0;  
}


void disinfection_one(void *mem, unsigned short size)
{
  void *where;
  unsigned short offset;
    
  offset = get_sig_off(mem,size,&sig_gaara_start,7);
  
  if (!offset)
  {
    printf("Disg: Disinfection failed, cannot find second signature\n");
    return;    
  }

  where = (void*)(mem + offset);
  printf("Disg: Gaara entry found at 0x%lx\n",where);
  
  /* lets set orginal bytes just after the virus entry point, so
  the body will not be executed, and the execution flow will return to the host at start*/
  
  memcpy((void*)(where+4),(void*)&repair_bytes,sizeof(repair_bytes));
  
  
  /* and now destroy the Gaara marker, pad it with 0 */
  memset((void*)(mem + gaara_m_offset),0,5);
  
  printf("Disg: File is repaired\n");
  d_files++;
  return;
}




void in_infected_file(SYM_ENTRY *se)
{
  unsigned short size;
  void *mem = HeapDeref(se->handle);
  
  printf("Disg: Checking %s\n",se->name);
  
  if (!mem)
  {
    printf("Disg: Error, cannot HeapDeref();\n");
    return;    
  }
  
  size = *(unsigned short*)mem;
  
  printf("Disg: Variable size is %d bytes \n",size);
  
  if (size < LOW_GAARA_SIZE)
  {
    printf("Disg: The file is not infected (size is too low)\n");
    return;    
  }
    
  if (!is_gaara(mem,size))
    return;
        
  disinfection_one(mem,size);
  
}




void dis_loop(void)
{

  SYM_ENTRY *se;

  if (!FolderOp(NULL, FOP_LOCK | FOP_ALL_FOLDERS))
  {
    printf("Disg: Locking folder table failed\n");
    return;    
  }
    
  se = SymFindFirst(NULL, FO_RECURSE | FO_SKIP_TEMPS);
  
  while (se)
  {
      in_infected_file(se);      
      se = SymFindNext();    
  }
    
  FolderOp(NULL,FOP_UNLOCK | FOP_ALL_FOLDERS);
  
}


// Main Function
void _main(void)
{
  clrscr();
  printf("*** DISG - Gaara virus file disinfector ready.\n");
  printf("*** by Piotr Bania - http://www.piotrbania.com\n");

  is_infected();
  dis_loop();
  
  
  printf("Disg: Disinfection ended, %d file were disinfected\n",d_files);
  ngetchx();
}