#include <windows.h>
#include "PeLibrary.h"

typedef BOOL (WINAPI* LPFNPLOPENFILE)(DWORD,LPTSTR);
typedef DWORD (WINAPI* LPFNPLADDIMPORTS)(LPTSTR,LPTSTR);
typedef BOOL (WINAPI* LPFNPLCLOSEFILE)();

//-------------------------------------------------------------------------------------
bool AddImport( const char* FilePath, const char* LibraryName, const char* FuncName )
{
   if ( FilePath == NULL || LibraryName == NULL || FuncName == NULL )
   {
      return false;
   }

   HMODULE hModule = NULL;
   bool RetVal = false;

   try
   {
      hModule = LoadLibrary( "PeLibrary.dll" );
      if ( hModule != NULL )
      {
         LPFNPLOPENFILE   pOpenFile  = (LPFNPLOPENFILE)GetProcAddress( hModule, "plOpenFile" );
         LPFNPLCLOSEFILE  pCloseFile = (LPFNPLCLOSEFILE)GetProcAddress( hModule, "plCloseFile" );
         LPFNPLADDIMPORTS pAddImp    = (LPFNPLADDIMPORTS)GetProcAddress( hModule, "plAddImportFunction" );
         
         if ( pOpenFile != NULL && pCloseFile != NULL && pAddImp != NULL  )
         {            
            if ( pOpenFile( PL_NO_OPEN_DIALOG, (char*)FilePath ) )
            {               
               if ( pAddImp( (char*)LibraryName, (char*)FuncName ) )
               {
                  RetVal = true;
               }               
            }
            
            pCloseFile();
         }
      }
   }
   catch( ... )
   {
   }
   
   if ( hModule != NULL )
   {
      FreeLibrary( hModule );
   }
   
   return RetVal;
}