repos_assist/天使插件源码/TSPlug/dllmain.cpp

// dllmain.cpp : DllMain 的实现。

/*本源码由TC简单软件科技有限公司开源,功能可以自由修改、发布、
长沙简单软件科技有限公司对于源码不做后期维护,,请大家在使用过程中遵循开源协议
*/



#include "stdafx.h"
#include "resource.h"
#include "TSPlug_i.h"
#include "dllmain.h"
#include "DXBind.h"
#include "TSRuntime.h"
#include <psapi.h> 
#pragma comment ( lib, "psapi.lib" )


CTSPlugModule _AtlModule;
TCHAR gDLLFolder[MAX_PATH + 1];
HHOOK g_hSetWindowsHook=NULL;
HMODULE g_hInstance=NULL;
extern bool g_Unbind;
extern HWND g_InjectHWND;
extern HWND g_currentHwnd;
HANDLE g_hthread=NULL;
extern int SySTpye;

//卸载线程
void IMEUnLoadThread(void *para)
{
	while(1)
	{
		if(TSRuntime::pData->InjectType==204)//203注入
		{
			TSRuntime::pData->InjectType=205;
			TSRuntime::MyLoadLibrary();
			FreeLibraryAndExitThread(g_hInstance,0);  //卸载DLL
			return;
		}
		if(g_Unbind==true)	 //等待窗口解绑
			{
			//TSRuntime::add_log( "卸载DLL,InjectType:%d",TSRuntime::pData->InjectType);
			if(TSRuntime::pData->InjectType==1||TSRuntime::pData->InjectType==202||TSRuntime::pData->InjectType==205)
				{
					//TSRuntime::add_log( "卸载DLL,InjectType:%d",TSRuntime::pData->InjectType);
					FreeLibraryAndExitThread(g_hInstance,0);  //卸载DLL
				}
			   return ;
			}
		//如果注入方进程异常退出,自我解绑,卸载DLL
		if(::IsWindow(g_InjectHWND)==false&&g_InjectHWND!=NULL)
			{
			if(TSRuntime::pData->InjectType==0)
				{
					//通知自身窗口解绑
					SendMessage(g_currentHwnd,TS_UNBIND,0,0);
					::UnhookWindowsHookEx(g_hSetWindowsHook);
				}
			else
				{
					//通知自身窗口解绑
					SendMessage(g_currentHwnd,TS_UNBIND,0,0);
					FreeLibraryAndExitThread(g_hInstance,0);  //卸载DLL
				}
				return ;
			}
		Sleep(10);

		//CString scd;
	}

}

DWORD CALLBACK CBFunA(DWORD calldata1, DWORD calldata2,DWORD calldata3)	//输入法注入回调函数
{
	HINSTANCE my_hInstance=(HINSTANCE)calldata1;  //输入法传入自身DLL基址和自身得到的基址验证后才开启线程
	//TSRuntime::add_log( "IME注入");
	if(my_hInstance)
		{
			if(my_hInstance==g_hInstance)
				g_hthread=(HANDLE)_beginthread(IMEUnLoadThread, 0 ,0 );//启动线程等待解绑卸载DLL
		}
	return 0;
}

static HMODULE ModuleFromAddress(PVOID pv) 
{
	MEMORY_BASIC_INFORMATION mbi;
	if(::VirtualQuery(pv, &mbi, sizeof(mbi)) != 0)
		{
			return (HMODULE)mbi.AllocationBase;
		}
	else
		{
			return NULL;
		}
}

static LRESULT WINAPI GetMsgProc(int code, WPARAM wParam, LPARAM lParam) 
{
	
	return ::CallNextHookEx(g_hSetWindowsHook, code, wParam, lParam);
}

BOOL WINAPI CBFunB(BOOL bInstall, DWORD dwThreadId)
{
	BOOL bOk=FALSE;
	if(bInstall)
		{
		g_hSetWindowsHook = ::SetWindowsHookEx(WH_CALLWNDPROC, GetMsgProc,ModuleFromAddress(GetMsgProc), dwThreadId);
		//TSRuntime::add_log( "g_hSetWindowsHook:%x,PID:%d",GetCurrentProcessId());
		if(g_hSetWindowsHook != NULL)
			{
			   bOk = true;
			}
			
		} 
	else 
		{
		 
		 if(g_hSetWindowsHook)
			 {
				//::MessageBox(0,L"Dll Main:UnhookWindowsHookEx",L"TS",0);
				bOk = ::UnhookWindowsHookEx(g_hSetWindowsHook);
			 }
		}
	return bOk;
}

// DLL 入口点
extern "C" BOOL WINAPI DllMain(HINSTANCE hInstance, DWORD dwReason, LPVOID lpReserved)
{
	switch(dwReason)
		{
		case DLL_PROCESS_ATTACH:
			{

				TSRuntime::InitKeyPressCharMap();
				TSRuntime::InitVirtualToASCIIMap();
				TSRuntime::InitKeyMap();
				TSRuntime::InitKeyPressMap();
				SySTpye=TSRuntime::InitialWindowsVersion();
				g_hInstance=hInstance;
				//wcscpy(gDLLFolder,TSRuntime::GetComPath()); 
				TSRuntime::GetComPath(gDLLFolder);
				//::MessageBox(0,gDLLFolder,L"TS",0);
				char pszMapName[MAX_PATH]={0};
				sprintf( pszMapName,"%s%d",TS_MAPVIEW_NAME,GetCurrentProcessId());
				HANDLE hFileMap = OpenFileMappingA(FILE_MAP_ALL_ACCESS, FALSE, pszMapName);
					//如果hFileMap句柄不为空说明DLL被注入,准备启动注入线程
				if(hFileMap!=NULL)                           
					{
					  //::MessageBox(0,L"Dll Main:DLL_PROCESS_ATTACH",L"TS",0);
					 //// 这里先打开共享内存,共享内存和程序是一对一的关系
				      CShareMemory *sm=new CShareMemory(pszMapName);
					  TSRuntime::pData=(CMessageData*)sm->GetBuffer();
					 //// 共享内存的初始化数据是不能为空的,为空就不正常
					  if(TSRuntime::pData!=NULL)       
						{
							if(TSRuntime::pData->InjectType==BIND_201||TSRuntime::pData->InjectType==BIND_203)//201模式注入
							{
								if(TSRuntime::pData->InjectType==BIND_201)
									TSRuntime::pData->InjectType=202;
								else if(TSRuntime::pData->InjectType==BIND_203)
									TSRuntime::pData->InjectType=204;
								//TSRuntime::add_log("201模式注入");
								DWORD InternalCallWinProc_Addr=(DWORD)::GetModuleHandle(L"user32.dll");
								if(SySTpye==1)//WinXP
									InternalCallWinProc_Addr+=USER32InternalCallWinProcXPoffse;
								else if(SySTpye==2)//Win2003
									InternalCallWinProc_Addr+=USER32InternalCallWinProcWin2003offse;
								else if(SySTpye==4&&TSRuntime::IsWin7X64)//WIN7X64
									InternalCallWinProc_Addr+=USER32InternalCallProcWin7x64offse;
								else if(SySTpye==4)//WIN7X86
									InternalCallWinProc_Addr+=USER32InternalCallProcWin7offse;
								else if(SySTpye==5&&TSRuntime::IsWin8X64)//WIN8X64
									InternalCallWinProc_Addr+=USER32InternalCallProcWin8x64offse;
								else if(SySTpye==5)//WIN8X86
									InternalCallWinProc_Addr+=USER32InternalCallProcWin8offse;

								BYTE ori[5]={0x55,0x8b,0xec,0x56,0x57};

								//注入完成还原钩子
								memcpy((void*)InternalCallWinProc_Addr,ori,5);
								FlushInstructionCache(GetCurrentProcess(),(void*)InternalCallWinProc_Addr,5);

								////TS_BIND201_NAME
								wchar_t pszEventName[MAX_PATH]={0};
								::wsprintf(pszEventName,L"%s%d",TS_BIND201_NAME,::GetCurrentProcessId());
								HANDLE picEvent=::CreateEvent(NULL,TRUE,FALSE,pszEventName);
								::WaitForSingleObject(picEvent,INFINITE);
								::CloseHandle(picEvent);
								g_Unbind=false;

								_beginthread(IMEUnLoadThread, 0 ,0 );
								if(TSRuntime::pData->InjectType==202)
									TSRuntime::g_DxObj.hookApi();
							}
							else if(TSRuntime::pData->InjectType!=202&&TSRuntime::pData->InjectType!=204) //// 这里根据传入的模式进行函数拦截,兵起一个检测线程进行检测
							{
								if(TSRuntime::pData->InjectType==205)//203绑定
									_beginthread(IMEUnLoadThread, 0 ,0 );	
								TSRuntime::g_DxObj.hookApi();
							}
						}

					}
				
			break;
			}

		}
	return _AtlModule.DllMain(dwReason, lpReserved); 
}

TsMutex::TsMutex(char *pszEventName)
	{
		//InitializeCriticalSection(&m_mutex);
		hEvent=OpenEventA(EVENT_ALL_ACCESS,false,pszEventName);
		if(hEvent==NULL)
			{
				hEvent=CreateEventA(NULL,FALSE,FALSE,pszEventName);
				::SetEvent(hEvent);
			}
		
	}

TsMutex::~TsMutex()
	{
		//DeleteCriticalSection(&m_mutex);
		CloseHandle(hEvent);
	}

void TsMutex::lock()
	{
		//::WaitForSingleObject(hEvent,INFINITE);
		::WaitForSingleObject(hEvent,10000);
		//EnterCriticalSection(&m_mutex);
	}
void TsMutex::unlock()
	{
		::SetEvent(hEvent);
		//LeaveCriticalSection(&m_mutex);
	}

TsMutexlock::TsMutexlock(TsMutex *ptcmutex)
	{
		m_ptcmutex=	 ptcmutex;
		m_ptcmutex->lock();
	}

TsMutexlock::~TsMutexlock()
	{
		m_ptcmutex->unlock();
	}