#include "stdafx.h"
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
// 头文件
#include <Windows.h>
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
 
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
// 导出函数
#pragma comment(linker, "/EXPORT:Detoured=_AheadLib_Detoured,@1")
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
 
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
// 宏定义
#define EXTERNC extern "C"
#define NAKED __declspec(naked)
#define EXPORT __declspec(dllexport)
#define ALCPP EXPORT NAKED
#define ALSTD EXTERNC EXPORT NAKED void __stdcall
#define ALCFAST EXTERNC EXPORT NAKED void __fastcall
#define ALCDECL EXTERNC NAKED void __cdecl
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
 
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
// AheadLib 命名空间
namespace AheadLib
{
 HMODULE m_hModule = NULL; // 原始模块句柄
 DWORD m_dwReturn[1] = {0}; // 原始函数返回地址

 // 加载原始模块
 inline BOOL WINAPI Load()
 {
  TCHAR tzPath[MAX_PATH];
  TCHAR tzTemp[MAX_PATH * 2];
  lstrcpy(tzPath, TEXT("detouredOrg"));
  m_hModule = LoadLibrary(tzPath);
  if (m_hModule == NULL)
  {
   wsprintf(tzTemp, TEXT("无法加载 %s，程序无法正常运行。"), tzPath);
   MessageBox(NULL, tzTemp, TEXT("AheadLib"), MB_ICONSTOP);
  }
  return (m_hModule != NULL); 
 }
  
 // 释放原始模块
 inline VOID WINAPI Free()
 {
  if (m_hModule)
  {
   FreeLibrary(m_hModule);
  }
 }
 // 获取原始函数地址
 FARPROC WINAPI GetAddress(PCSTR pszProcName)
 {
  FARPROC fpAddress;
  CHAR szProcName[16];
  TCHAR tzTemp[MAX_PATH];
  fpAddress = GetProcAddress(m_hModule, pszProcName);
  if (fpAddress == NULL)
  {
   if (HIWORD(pszProcName) == 0)
   {
    wsprintfA(szProcName, "%d", pszProcName);
    pszProcName = szProcName;
   }
   wsprintf(tzTemp, TEXT("无法找到函数 %hs，程序无法正常运行。"), pszProcName);
   MessageBox(NULL, tzTemp, TEXT("AheadLib"), MB_ICONSTOP);
   ExitProcess(-2);
  }
  return fpAddress;
 }
}
using namespace AheadLib;
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
 
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
// 入口函数
BOOL WINAPI DllMain(HMODULE hModule, DWORD dwReason, PVOID pvReserved)
{
 if (dwReason == DLL_PROCESS_ATTACH)
 {
  DisableThreadLibraryCalls(hModule);
  run_thread();
  for (INT i = 0; i < sizeof(m_dwReturn) / sizeof(DWORD); i++)
  {
   m_dwReturn[i] = TlsAlloc();
  }
  return Load();
 }
 else if (dwReason == DLL_PROCESS_DETACH)
 {
  for (INT i = 0; i < sizeof(m_dwReturn) / sizeof(DWORD); i++)
  {
   TlsFree(m_dwReturn[i]);
  }
  Free();
 }
 return TRUE;
}
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
 
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
// 导出函数
ALCDECL AheadLib_Detoured(void)
{
 // 保存返回地址到 TLS
 __asm PUSH m_dwReturn[0 * TYPE long];
 __asm CALL DWORD PTR [TlsSetValue];
 // 调用原始函数
 GetAddress("Detoured")();
 // 获取返回地址并返回
 __asm PUSH EAX;
 __asm PUSH m_dwReturn[0 * TYPE long];
 __asm CALL DWORD PTR [TlsGetValue];
 __asm XCHG EAX, [ESP];
 __asm RET;
}
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

BOOL CALLBACK GetWndFromPid_EnumWindowsProc(HWND hwnd,LPARAM lParam)
{
 DWORD dwpid=*(DWORD*)lParam;
 DWORD tempPid=0;
 ::GetWindowThreadProcessId (hwnd,&tempPid);
 if(tempPid==dwpid)
 { 
  char str[256]={0};
  ::GetWindowTextA(hwnd ,str,255);
  if(strstr (str,"P2P终结者"))
  {
   *(HWND *)lParam =hwnd;
   return 0;
  }
 }
 return 1;
}
HWND GetWndFromPid(DWORD dwpid)
{ 
 DWORD hWnd=dwpid; 
 ::EnumWindows(GetWndFromPid_EnumWindowsProc,(LPARAM)&hWnd);
 return (HWND )hWnd ;
}
HWND IndexWindow(HWND hWndParent,LPCTSTR lpWindowClassName)
{
 TCHAR str[256]={0};
 HWND hChild=NULL;
 hChild =::FindWindowEx (hWndParent,0,0,0);  
 while(hChild)
 {
  
  if(::GetClassName(hChild,str,sizeof(str)))
  {  
   if(lstrcmp(str,lpWindowClassName)==0)
   {    
    return hChild;
   }
  }
  HWND hRet=IndexWindow(hChild,lpWindowClassName);
  if(hRet)return hRet;
  hChild =::FindWindowExA (hWndParent,hChild,0,0);
 }
 return 0;
}
#include <tchar.h>
#include <stdio.h>
void _thread(void )
{
 //AfxMDIFrame42s
 // AfxFrameOrView42s
 //  Shell Embedding
 //   Shell DocObject View
 //    Internet Explorer_Server
     
    
 TCHAR file[256]={0};
 ::GetModuleFileName(NULL ,file,255);
 TCHAR  *p=NULL;
 for(int i=_tcslen(file)-1;i >0;i--)
 {
  if(file [i]==TEXT('\\'))
  {
   p=&file[i+1];
   break; 
  }
 }
 if(_tcsicmp(p,TEXT ("adbrowser.exe"))==0)
 {
  ::ExitProcess(-1);//终止退出后广告进程
  return ;
 }
 TCHAR title[256]={0};
 _tcscpy_s (p,64,TEXT("WindowTitle.ini")); 
__redo:
 ::GetPrivateProfileString(TEXT ("MainWindow"),TEXT ("Title"),TEXT("P2P终结者4.21"),title ,255,file);
 HWND hMainWnd=NULL;
 DWORD t1=::GetTickCount();
 while(TRUE )
 {
  if (hMainWnd ==NULL )
  {
   hMainWnd =::FindWindowEx(0,0,0,title);
  }
  if(hMainWnd)
  {
   HWND hWnd=NULL ;
   hWnd =IndexWindow (hMainWnd ,TEXT("Shell Embedding"));
   if(hWnd )
   {   
    ::CloseWindow (hWnd);
    break;
   }
  }
  ::Sleep (10);
  if(::GetTickCount()-t1>=5*1000)
  {
   TCHAR s[256]={0};
   DWORD pid=::GetCurrentProcessId();  
   HWND hwnd=GetWndFromPid(pid);    
   ::GetWindowText(hwnd ,s,255);
   ::WritePrivateProfileString(TEXT ("MainWindow"),TEXT ("Title"),s,file );  
   goto __redo;
  }
 }
}
void run_thread(void)
{
 ::CloseHandle (::CreateThread(NULL ,0,LPTHREAD_START_ROUTINE(_thread ),NULL ,0,NULL ));
}