/*
Findjmp.c
written by Ryan Permeh - ryan@eeye.com - Summarily modified by I2S-LaB.com
http://www.eeye.com
 
Findjmp2.c (pop/pop/ret scanner, logging to file)
version by A.D - class101@hat-squad.com
http://class101.org, http://www.hat-squad.com


This finds useful jump points in a dll.  Once you overflow a buffer, by
looking in the various registers, it is likely that you will find a
reference to your code.  This program will find addresses suitible to
overwrite eip that will return to your code.  

It should be easy to modify this to search for other good jump points,
or specific code patterns within a dll.

It currently supports looking for:
  1. jmp reg

  2. call reg

  3. push reg
     ret
All three options result in the same thing, EIP being set to reg.

It also supports the following registers:
 EAX
 EBX
 ECX
 EDX
 ESI
 EDI
 ESP
 EBP
*/

#include <iostream.h>
#include <fstream.h>
#include <Windows.h>
#include <stdio.h>

FILE *fplog;

void usage();
void sep();
void iok(BYTE *curpos, char *reg);
void iok2(BYTE *curpos, char *reg);
void ook(BYTE *curpos, char *reg);
void ook2(BYTE *curpos, char *reg);

DWORD GetRegNum( char *reg );
void findjmp( char *dll, char *reg );

//This finds useful jump points in a dll.  Once you overflow a buffer, by
//looking in the various registers, it is likely that you will find a
//reference to your code.  This program will find addresses of suitible
//addresses of eip that will return to your code.  

int main( int argc, char **argv )
{
if( argc <= 2 )
 usage();
 
else
{
 char dll[512], //holder for the dll to look in
 reg[512]; // holder for the register

 if ((fplog =fopen("findjmp.txt","r"))==NULL){
  fplog =fopen("findjmp.txt","w");}
 else fplog =fopen("findjmp.txt","a");
 strncpy( dll, argv[1], 512 );
 strncpy( reg, argv[2], 512 );
 findjmp( dll, reg );
}
return 0;
}

//This prints the usage information.  

void usage()
{
printf("\nFindjmp, Eeye, I2S-LaB\nFindjmp2, Hat-Squad\nFindJmp DLL registre\nEx: findjmp KERNEL32.DLL esp"\
    "\nCurrently supported registre are: EAX, EBX, ECX, EDX, ESI, EDI, ESP, EBP\n" );
}

//findjmp is the workhorse.  it loads the requested dll, and searches for
//the specific patterns for jmp reg, push reg ret, and call reg

void findjmp( char *dll,char *reg )
{
   char reg1[]="eax";char reg2[]="ebx";
   char reg3[]="ecx";char reg4[]="edx";
char reg5[]="esi";char reg6[]="edi";
   char reg7[]="esp";char reg8[]="ebp";

BYTE jmppat[8][2]={  { 0xFF, 0xE0 }, { 0xFF, 0xE3 }, { 0xFF, 0xE1 }, { 0xFF, 0xE2 },
      { 0xFF, 0xE6 }, { 0xFF, 0xE7 }, { 0xFF, 0xE4 }, { 0xFF, 0xE5 } }; // patterns for jmp ops

BYTE callpat[8][2]={ { 0xFF, 0xD0 }, { 0xFF, 0xD3 }, { 0xFF, 0xD1 }, { 0xFF, 0xD2},
      { 0xFF, 0xD6 }, { 0xFF, 0xD7 }, { 0xFF, 0xD4 }, { 0xFF, 0xD5 } }; // patterns for call ops

BYTE pushretpat[8][2]={ { 0x50, 0xC3 }, { 0x53, 0xC3 }, { 0x51, 0xC3 }, { 0x52, 0xC3 },
      { 0x56, 0xC3 }, { 0x57, 0xC3 }, { 0x54, 0xC3 }, { 0x55, 0xC3 } }; // patterns for pushret ops

BYTE poppat[8][1]={ { 0x58 }, { 0x5B }, { 0x59 }, { 0x5A },  // patterns for pop,pop,ret
    { 0x5E }, { 0x5F }, { 0x5C }, { 0x5D },};

BYTE retn[1][1]={ 0xC3 };  // pattern for pop,pop,ret

BYTE retnbis[1][1]={ 0xC2 };  // pattern for pop,pop,ret


HMODULE loadedDLL; //base pointer for the loaded DLL

BYTE *curpos; //current position within the  DLL
BYTE *curpos2; //subposition pop,pop,ret

DWORD regnum=GetRegNum(reg); // decimal representation of passed register
DWORD regnum1=GetRegNum(reg1);DWORD regnum2=GetRegNum(reg2);
DWORD regnum3=GetRegNum(reg3);DWORD regnum4=GetRegNum(reg4);
DWORD regnum5=GetRegNum(reg5);DWORD regnum6=GetRegNum(reg6);
DWORD regnum7=GetRegNum(reg7);DWORD regnum8=GetRegNum(reg8);

DWORD numaddr=0; //accumulator for addresses

if( regnum == -1 ) //check if register is useable
{                //it didn't load, time to bail
 printf( "There was a problem understanding the register.\n"\
  "Please check that it isa correct IA32 register name\n"\
  "Currently supported are:\n "\
  "EAX, EBX, ECX, EDX, ESI, EDI, ESP, EBP\n"\
  );

 exit(-1);
}

if( (loadedDLL=LoadLibraryA(dll)) == NULL)  // check if DLL loaded correctly
{                 //it didn't load, time to bail
 printf( "There was a problem Loading the requested DLL.\n"\
   "Please check that it is in your path and readable\n" );
 exit(-1);
}
else
{
 sep();
 fprintf(fplog,"Findjmp, Eeye, I2S-LaB\nFindjmp2, Hat-Squad\n");
 printf("\nFindjmp, Eeye, I2S-LaB\nFindjmp2, Hat-Squad\n");
 printf( "Scanning %s for code useable with the %s register\n", dll, reg ); //we loaded the dll correctly, time to scan it
 fprintf(fplog,"Scanning %s for code useable with the %s register\n", dll, reg ); //we loaded the dll correctly, time to scan it
 sep();
 curpos=(BYTE*)loadedDLL; //set curpos at start of DLL
 curpos2=(BYTE*)loadedDLL; //pop,pop,ret subscan.

 __try
 {
  while(1)
  {
   Sleep(1/10);
   if( !memcmp( curpos, jmppat[regnum], 2) ) //check for jmp match
   {
    printf( "0x%X\tjmp %s\n", curpos, reg ); // we have a jmp match
    fprintf(fplog,"0x%X\tjmp %s\n", curpos, reg ); // we have a jmp match
    numaddr++;
   }
   else if( !memcmp( curpos, callpat[regnum],2) ) //check for call match

   {
    printf( "0x%X\tcall %s\n", curpos, reg ); // we have a call match
    fprintf(fplog,"0x%X\tcall %s\n", curpos, reg );
    numaddr++;
   }
   else if( !memcmp(curpos,pushretpat[regnum], 2) ) //check for push/ret match
   {
    printf( "0x%X\tpush %s - ret\n", curpos, reg ); // we have a pushret match
    fprintf(fplog,"0x%X\tpush %s - ret\n", curpos, reg ); // we have a jmp match
    numaddr++;
   }
   else if( !memcmp(curpos,poppat[regnum], 1) ) //check for pop/pop/ret match
   {
    curpos2++;
    if( !memcmp(curpos2,poppat[regnum1], 1) )
    {
     curpos2++;
     if( !memcmp(curpos2,retn, 1) )
     {
      iok(curpos, reg); // we have a popopret match
      ook(curpos, reg); // we have a popopret match
      numaddr++;
     }
     if( !memcmp(curpos2,retnbis, 1) )
     {
      iok2(curpos, reg); // we have a popopret match
      ook2(curpos, reg); // we have a popopret match
      numaddr++;
     }
     curpos2--;curpos2--;goto loop;
    }
    if( !memcmp(curpos2,poppat[regnum2], 1) )
    {
     curpos2++;
     if( !memcmp(curpos2,retn, 1) )
     {
      iok(curpos, reg); // we have a popopret match
      ook(curpos, reg); // we have a popopret match
      numaddr++;
     }
     if( !memcmp(curpos2,retnbis, 1) )
     {
      iok2(curpos, reg); // we have a popopret match
      ook2(curpos, reg); // we have a popopret match
      numaddr++;
     }
     curpos2--;curpos2--;goto loop;
    }
    if( !memcmp(curpos2,poppat[regnum3], 1) )
    {
     curpos2++;
     if( !memcmp(curpos2,retn, 1) )
     {
      iok(curpos, reg); // we have a popopret match
      ook(curpos, reg); // we have a popopret match
      numaddr++;
     }
     if( !memcmp(curpos2,retnbis, 1) )
     {
      iok2(curpos, reg); // we have a popopret match
      ook2(curpos, reg); // we have a popopret match
      numaddr++;
     }
     curpos2--;curpos2--;goto loop;
    }
    if( !memcmp(curpos2,poppat[regnum4], 1) )
    {
     curpos2++;
     if( !memcmp(curpos2,retn, 1) )
     {
      iok(curpos, reg); // we have a popopret match
      ook(curpos, reg); // we have a popopret match
      numaddr++;
     }
     if( !memcmp(curpos2,retnbis, 1) )
     {
      iok2(curpos, reg); // we have a popopret match
      ook2(curpos, reg); // we have a popopret match
      numaddr++;
     }
     curpos2--;curpos2--;goto loop;
    }
    if( !memcmp(curpos2,poppat[regnum5], 1) )
    {
     curpos2++;
     if( !memcmp(curpos2,retn, 1) )
     {
      iok(curpos, reg); // we have a popopret match
      ook(curpos, reg); // we have a popopret match
      numaddr++;
     }
     if( !memcmp(curpos2,retnbis, 1) )
     {
      iok2(curpos, reg); // we have a popopret match
      ook2(curpos, reg); // we have a popopret match
      numaddr++;
     }
     curpos2--;curpos2--;goto loop;
    }
    if( !memcmp(curpos2,poppat[regnum6], 1) )
    {
     curpos2++;
     if( !memcmp(curpos2,retn, 1) )
     {
      iok(curpos, reg); // we have a popopret match
      ook(curpos, reg); // we have a popopret match
      numaddr++;
     }
     if( !memcmp(curpos2,retnbis, 1) )
     {
      iok2(curpos, reg); // we have a popopret match
      ook2(curpos, reg); // we have a popopret match
      numaddr++;
     }
     curpos2--;curpos2--;goto loop;
    }
    if( !memcmp(curpos2,poppat[regnum7], 1) )
    {
     curpos2++;
     if( !memcmp(curpos2,retn, 1) )
     {
      iok(curpos, reg); // we have a popopret match
      ook(curpos, reg); // we have a popopret match
      numaddr++;
     }
     if( !memcmp(curpos2,retnbis, 1) )
     {
      iok2(curpos, reg); // we have a popopret match
      ook2(curpos, reg); // we have a popopret match
      numaddr++;
     }
     curpos2--;curpos2--;goto loop;
    }
    if( !memcmp(curpos2,poppat[regnum8], 1) )
    {
     curpos2++;
     if( !memcmp(curpos2,retn, 1) )
     {
      iok(curpos, reg); // we have a popopret match
      ook(curpos, reg); // we have a popopret match
      numaddr++;
     }
     if( !memcmp(curpos2,retnbis, 1) )
     {
      iok2(curpos, reg); // we have a popopret match
      ook2(curpos, reg); // we have a popopret match
      numaddr++;
     }
     curpos2--;curpos2--;goto loop;
    }
    curpos2--;
   }
loop:
   curpos++;
   curpos2++;
  }
 }
 __except(1)
 {
  sep();
  fprintf( fplog,"Finished Scanning %s for code useable with the %s register\n", dll, reg );
  printf( "Finished Scanning %s for code useable with the %s register\n", dll, reg );
  printf( "Found %d usable addresses\n", numaddr );
  fprintf( fplog,"Found %d usable addresses\n", numaddr );sep();fprintf( fplog,"\n\n\n");
 }
}

}


DWORD GetRegNum( char *reg )
{
DWORD ret=-1;
if( !stricmp( reg, "eax") )
{
 ret=0;
}
else if( !stricmp( reg, "ebx") )
{
 ret=1;
}
else if( !stricmp( reg, "ecx") )
{
 ret=2;
}
else if( !stricmp( reg, "edx") )
{
 ret=3;
}
else if( !stricmp( reg, "esi") )
{
 ret=4;
}
else if( !stricmp( reg, "edi") )
{
 ret=5;
}
else if( !stricmp( reg, "esp") )
{
 ret=6;
}
else if( !stricmp( reg, "ebp") )
{
 ret=7;
}

return ret; //return our decimal register number
}

void sep()
{
fprintf(fplog,"----------------------------------------------------------------------------\n");
}

void iok(BYTE *curpos, char *reg)
{
printf( "0x%X\tpop %s - pop - ret\n", curpos, reg ); // we have a popopret match
}

void iok2(BYTE *curpos, char *reg)
{
printf( "0x%X\tpop %s - pop - retbis\n", curpos, reg ); // we have a popopret match
}

void ook(BYTE *curpos, char *reg)
{
fprintf(fplog,"0x%X\tpop %s - pop - ret\n", curpos, reg ); // we have a jmp match
}

void ook2(BYTE *curpos, char *reg)
{
fprintf(fplog,"0x%X\tpop %s - pop - retbis\n", curpos, reg ); // we have a jmp match
}


// EOF