开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 游戏 > 外挂编程 > 查看源码
ReKeyDlg.cpp
资源名称：MFCReKey(1).rar　[点击查看]
上传用户：jinbin898
上传日期：2022-06-24
资源大小：89k
文件大小：18k
源码类别：

外挂编程

开发平台：

Visual C++

ReKeyDlg.cpp：源码内容
  2. // ReKeyDlg.cpp : 实现文件
  3. //
  5. #include "stdafx.h"
  6. #include "ReKey.h"
  7. #include "ReKeyDlg.h"
  8. #include "tlhelp32.h"
  10. #ifdef _DEBUG
  11. #define new DEBUG_NEW
  12. #endif
  15. // 用于应用程序“关于”菜单项的 CAboutDlg 对话框
  17. class CAboutDlg : public CDialog
  18. {
  19. public:
  20. CAboutDlg();
  22. // 对话框数据
  23. enum { IDD = IDD_ABOUTBOX };
  25. protected:
  26. virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV 支持
  28. // 实现
  29. protected:
  30. DECLARE_MESSAGE_MAP()
  31. };
  33. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  34. {
  35. }
  37. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  38. {
  39. CDialog::DoDataExchange(pDX);
  40. }
  42. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  43. END_MESSAGE_MAP()
  46. // CReKeyDlg 对话框
  47. #define WM_SHOWTASK WM_USER+101
  49. TCHAR g_szConfigFile[MAX_PATH];
  50. HWND hWar3;
  51. HHOOK hHook;
  52. void* g_GameDllbase; //进程war3中Game.dll的加载地址
  53. CString g_war3InstallPath;
  54. TCHAR g_szOutDebugFile[MAX_PATH] = _T("Rekey_OutDebug.txt");
  55. CStdioFile g_DebugFile;
  57. unsigned char g_numPAD[6][2] = {
  58. 0,VK_NUMPAD1,
  59. 0,VK_NUMPAD2,
  60. 0,VK_NUMPAD4,
  61. 0,VK_NUMPAD5,
  62. 0,VK_NUMPAD7,
  63. 0,VK_NUMPAD8};
  65. void WriteDebugFile(const CString& str);
  66. void* GetProcessDllBase(DWORD dwPID, const CString& moduleName);
  68. LRESULT CALLBACK LowKeyboardHookProc(          int code,
  69.   WPARAM wParam,
  70.   LPARAM lParam
  71.   )
  72. {
  73. HWND topWnd = ::GetForegroundWindow();
  75. DWORD processid;
  76. /*DWORD threadid = */::GetWindowThreadProcessId(hWar3,&processid);
  77. HANDLE hProcess = ::OpenProcess(PROCESS_ALL_ACCESS,false,processid);
  79. //  War3 各版本聊天判别地址如下
  80. //  1.20、1.21：
  81. //  0x0045CB8C
  82. // 
  83. //  1.22：
  84. //  0x6FABDFE0
  85. // 
  86. //  1.23：
  87. //  0x6FAD6E30
  88. // 
  89. //  1.24
  90. //  0x6FAE8450
  91. //
  92. // 其中，后三个的 6F 是 Game.dll 的基址。但是 DLL 的基址并不是总是固定不变的，
  93. // 所以实际中，可靠的方法是获取 hDll，然后加上 0x00XXXXXX
  95. BOOL bchatSuccess = FALSE;
  96. LPCVOID pchatbase = (LPCVOID)0x0045CB8C; //判断当前是否为聊天状态
  97. int chatStatus=0;
  98. bchatSuccess = ::ReadProcessMemory(hProcess,pchatbase,&chatStatus,4,NULL);
  100. CloseHandle(hProcess);
  102. if (topWnd == hWar3)
  103. {
  104. //对话框上开启debug的情况下，输出debug信息
  105. CReKeyDlg *dlg = (CReKeyDlg*)AfxGetMainWnd();
  106. if (dlg->m_bIsDebug && WM_KEYDOWN==(DWORD)wParam)
  107. {
  108. CString outLog;
  109. TCHAR keyname[10];
  110. DWORD param = ((PKBDLLHOOKSTRUCT)lParam)->scanCode<<16;
  111. GetKeyNameText(param,keyname,_countof(keyname));
  112. outLog.Format(_T(" -- War3=0x%08X bChat=[%d]%d Key=[%s]n"),hProcess,bchatSuccess,chatStatus,keyname);
  113. WriteDebugFile(outLog);
  114. }
  116. //聊天状态，不做改建。
  117. if ( bchatSuccess && chatStatus!=0 )
  118. {
  119. TRACE0("chating...n");
  120. return CallNextHookEx(hHook,code,wParam,lParam);
  121. }
  123. //根据键映射更改按键
  124. PKBDLLHOOKSTRUCT kbStruct;
  125. if (WM_KEYDOWN==(DWORD)wParam)
  126. {
  127. kbStruct = (PKBDLLHOOKSTRUCT)lParam;
  128. for (int i=0;i<6;i++)
  129. {
  130. //TRACE2("g[i][0] = %d, vkCode = %dn", g_numPAD[i][0], kbStruct->vkCode);
  131. if (g_numPAD[i][0] == kbStruct->vkCode)
  132. {
  133. keybd_event(g_numPAD[i][1],0,0,0);
  134. keybd_event(g_numPAD[i][1],0,KEYEVENTF_KEYUP,0);
  136. return 1;
  137. }
  138. }
  139. }
  140. }
  142. return CallNextHookEx(hHook,code,wParam,lParam);
  144. }
  146. CReKeyDlg::CReKeyDlg(CWnd* pParent /*=NULL*/)
  147. : CDialog(CReKeyDlg::IDD, pParent)
  148. , m_bIsDebug(FALSE)
  149. {
  150. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  151. }
  153. void CReKeyDlg::DoDataExchange(CDataExchange* pDX)
  154. {
  155. CDialog::DoDataExchange(pDX);
  156. DDX_Control(pDX, IDC_STATUS, m_status);
  157. DDX_Control(pDX, IDC_EDIT7, m_edit7);
  158. DDX_Control(pDX, IDC_EDIT8, m_edit8);
  159. DDX_Control(pDX, IDC_EDIT4, m_edit4);
  160. DDX_Control(pDX, IDC_EDIT5, m_edit5);
  161. DDX_Control(pDX, IDC_EDIT1, m_edit1);
  162. DDX_Control(pDX, IDC_EDIT2, m_edit2);
  163. DDX_Control(pDX, IDC_WAR3TOOLTIP, m_staticTooltip);
  164. DDX_Check(pDX, IDC_DEBUGSWITCH, m_bIsDebug);
  165. DDX_Control(pDX, IDC_STARTWARIII, m_StartWar3Ctrl);
  166. }
  168. BEGIN_MESSAGE_MAP(CReKeyDlg, CDialog)
  169. ON_WM_SYSCOMMAND()
  170. ON_WM_PAINT()
  171. ON_WM_QUERYDRAGICON()
  172. //}}AFX_MSG_MAP
  173. ON_WM_CLOSE()
  174. ON_WM_TIMER()
  175. ON_EN_CHANGE(IDC_EDIT1, &CReKeyDlg::OnEnChangeEdit1)
  176. ON_EN_CHANGE(IDC_EDIT2, &CReKeyDlg::OnEnChangeEdit2)
  177. ON_EN_CHANGE(IDC_EDIT4, &CReKeyDlg::OnEnChangeEdit4)
  178. ON_EN_CHANGE(IDC_EDIT5, &CReKeyDlg::OnEnChangeEdit5)
  179. ON_EN_CHANGE(IDC_EDIT7, &CReKeyDlg::OnEnChangeEdit7)
  180. ON_EN_CHANGE(IDC_EDIT8, &CReKeyDlg::OnEnChangeEdit8)
  181. ON_MESSAGE(WM_SHOWTASK,&CReKeyDlg::OnShowTask)
  182. ON_STN_CLICKED(IDC_STARTWARIII, &CReKeyDlg::OnStnClickedStartwariii)
  183. ON_COMMAND(IDM_NF_MENU_EXIT,&CReKeyDlg::OnNFMenuExit)
  184. ON_COMMAND(IDM_SETWAR3PATH,&CReKeyDlg::OnSetWar3Path)
  185. ON_BN_CLICKED(IDC_DEBUGSWITCH, &CReKeyDlg::OnBnClickedDebugswitch)
  186. END_MESSAGE_MAP()
  189. // CReKeyDlg 消息处理程序
  191. BOOL CReKeyDlg::OnInitDialog()
  192. {
  193. CDialog::OnInitDialog();
  195. // 将“关于...”菜单项添加到系统菜单中。
  197. // IDM_ABOUTBOX 必须在系统命令范围内。
  198. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  199. ASSERT(IDM_ABOUTBOX < 0xF000);
  201. CMenu* pSysMenu = GetSystemMenu(FALSE);
  202. if (pSysMenu != NULL)
  203. {
  204. BOOL bNameValid;
  205. CString strAboutMenu;
  206. bNameValid = strAboutMenu.LoadString(IDS_ABOUTBOX);
  207. ASSERT(bNameValid);
  208. if (!strAboutMenu.IsEmpty())
  209. {
  210. pSysMenu->AppendMenu(MF_SEPARATOR);
  211. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  212. }
  213. }
  215. // 设置此对话框的图标。当应用程序主窗口不是对话框时，框架将自动
  216. //  执行此操作
  217. SetIcon(m_hIcon, TRUE); // 设置大图标
  218. SetIcon(m_hIcon, FALSE); // 设置小图标
  220. // TODO: 在此添加额外的初始化代码
  222. //输出调试信息 --开关
  225. //提升权限
  226. HANDLE hProcess = OpenProcess(PROCESS_QUERY_INFORMATION,FALSE,GetCurrentProcessId());
  227. if (hProcess)
  228. {
  229. HANDLE hToken;
  230. if (OpenProcessToken(hProcess,TOKEN_ADJUST_PRIVILEGES|TOKEN_QUERY,&hToken))
  231. {
  232. LUID luid;
  233. if (LookupPrivilegeValue(NULL,SE_DEBUG_NAME,&luid))
  234. {
  235. TOKEN_PRIVILEGES tp;
  236. tp.PrivilegeCount = 1;
  237. tp.Privileges[0].Luid = luid;
  238. tp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
  240. if(!AdjustTokenPrivileges(hToken,FALSE,&tp,sizeof(TOKEN_PRIVILEGES),NULL,NULL)) //AdjustTokenPrivileges
  241. MessageBox(_T("AdjustTokenPrivileges失败。"),_T("提升权限"));
  242. }
  243. else //LookupPrivilegeValue
  244. {
  245. MessageBox(_T("LookupPrivilegeValue失败。"),_T("提升权限"));
  246. }
  247. CloseHandle(hToken);
  248. }
  249. else //OpenProcessToken
  250. {
  251. MessageBox(_T("OpenProcessToken失败。"),_T("提升权限"));
  252. }
  253. }
  255. //得到应用程序全路径
  256. TCHAR fullPath[MAX_PATH];
  257. GetModuleFileName(NULL,fullPath,MAX_PATH);
  259. //解析路径，得到应用程序目录
  260. TCHAR drive[_MAX_DRIVE];
  261. TCHAR dir[_MAX_DIR];
  262. _tsplitpath_s(fullPath,drive,_countof(drive),dir,_countof(dir),NULL,0,NULL,0);
  264. CString strAppPath;
  265. strAppPath.Format(_T("%s%s"),drive,dir);
  267. //生成配置路径
  268. CString configFile = _T("ReKey.conf"); //配置文件名
  269. CString configPath = strAppPath+configFile;
  271. TCHAR* szConfigFile = configPath.GetBuffer(configPath.GetLength());
  272. _tcscpy_s(g_szConfigFile,_countof(g_szConfigFile),szConfigFile);
  273. configPath.ReleaseBuffer();
  275. m_nid.cbSize = sizeof(NOTIFYICONDATA);
  276. m_nid.hWnd = this->m_hWnd;
  277. m_nid.uID = IDR_MAINFRAME;
  278. m_nid.uFlags = NIF_ICON | NIF_MESSAGE | NIF_TIP;
  279. m_nid.uCallbackMessage = WM_SHOWTASK;  //user defined
  280. m_nid.hIcon = LoadIcon(AfxGetInstanceHandle(),MAKEINTRESOURCE(IDR_MAINFRAME));
  281. _tcscpy_s(m_nid.szTip,_countof(m_nid.szTip),L"War3改键工具");
  282. Shell_NotifyIcon(NIM_ADD,&m_nid);
  284. m_edit1.SetLimitText(1);
  285. m_edit2.SetLimitText(1);
  286. m_edit4.SetLimitText(1);
  287. m_edit5.SetLimitText(1);
  288. m_edit7.SetLimitText(1);
  289. m_edit8.SetLimitText(1);
  291. m_hTimer = SetTimer(1,2000,NULL);
  293. CFile fileRead;
  294. if (fileRead.Open(g_szConfigFile,CFile::modeRead))
  295. {
  296. CArchive arLoad(&fileRead,CArchive::load);
  297. arLoad>>g_war3InstallPath;
  298. for (int i=0;i<6;i++)
  299. {
  300. arLoad>>g_numPAD[i][0];
  301. }
  303. fileRead.Close();
  305. m_edit1.SetWindowText(CString((char)g_numPAD[0][0]));
  306. m_edit2.SetWindowText(CString((char)g_numPAD[1][0]));
  307. m_edit4.SetWindowText(CString((char)g_numPAD[2][0]));
  308. m_edit5.SetWindowText(CString((char)g_numPAD[3][0]));
  309. m_edit7.SetWindowText(CString((char)g_numPAD[4][0]));
  310. m_edit8.SetWindowText(CString((char)g_numPAD[5][0]));
  311. }
  313. if ( 0 == g_war3InstallPath.GetLength() )
  314. {
  315. m_staticTooltip.SetWindowText(_T("<-点击图标设置War3路径"));
  316. }
  317. else
  318. {
  319. m_staticTooltip.SetWindowText(_T("<-点击图标启动War3"));
  320. }
  322. //设置war3图标
  323. CWar3Container *war3IconContainer =  (CWar3Container*)GetDlgItem(IDC_STARTWARIII);
  324. HICON icon = (HICON)LoadImage(theApp.m_hInstance,MAKEINTRESOURCE(IDI_WAR3X),IMAGE_ICON,48,48,LR_DEFAULTCOLOR);
  325. war3IconContainer->SetIcon(icon);
  327. GetDlgItem(IDC_DEBUGSWITCH)->ShowWindow(SW_HIDE);
  329. if (__argc == 2)
  330. {
  331. CString debugFlag = _T("/debug");
  332. LPCTSTR flag= __targv[1];
  333. if (0 == debugFlag.CompareNoCase(flag)) //有debug参数
  334. {
  335. GetDlgItem(IDC_DEBUGSWITCH)->ShowWindow(SW_SHOW);
  336. }
  337. }
  339. return TRUE;  // 除非将焦点设置到控件，否则返回 TRUE
  340. }
  342. void CReKeyDlg::OnSysCommand(UINT nID, LPARAM lParam)
  343. {
  344. if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  345. {
  346. CAboutDlg dlgAbout;
  347. dlgAbout.DoModal();
  348. }
  349. //  else if ((nID & 0xFFF0) == SC_CLOSE/*SC_ICON*/)
  350. //  {
  351. //  ShowWindow(SW_HIDE);
  352. //  }
  353. else
  354. {
  355. CDialog::OnSysCommand(nID, lParam);
  356. }
  357. }
  359. // 如果向对话框添加最小化按钮，则需要下面的代码
  360. //  来绘制该图标。对于使用文档/视图模型的 MFC 应用程序，
  361. //  这将由框架自动完成。
  363. void CReKeyDlg::OnPaint()
  364. {
  365. if (IsIconic())
  366. {
  367. CPaintDC dc(this); // 用于绘制的设备上下文
  369. SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);
  371. // 使图标在工作区矩形中居中
  372. int cxIcon = GetSystemMetrics(SM_CXICON);
  373. int cyIcon = GetSystemMetrics(SM_CYICON);
  374. CRect rect;
  375. GetClientRect(&rect);
  376. int x = (rect.Width() - cxIcon + 1) / 2;
  377. int y = (rect.Height() - cyIcon + 1) / 2;
  379. // 绘制图标
  380. dc.DrawIcon(x, y, m_hIcon);
  381. }
  382. else
  383. {
  384. CDialog::OnPaint();
  385. }
  386. }
  388. //当用户拖动最小化窗口时系统调用此函数取得光标
  389. //显示。
  390. HCURSOR CReKeyDlg::OnQueryDragIcon()
  391. {
  392. return static_cast<HCURSOR>(m_hIcon);
  393. }
  396. void CReKeyDlg::OnClose()
  397. {
  398. // TODO: 在此添加消息处理程序代码和/或调用默认值
  399. if (hHook)
  400. {
  401. if (!UnhookWindowsHookEx(hHook))
  402. {
  403. MessageBox(L"UnHook Failed!");
  404. }
  405. }
  407. if ( g_DebugFile.m_pStream )
  408. {
  409. g_DebugFile.Close();
  410. }
  412. Shell_NotifyIcon(NIM_DELETE,&m_nid);
  414. //序列化设置
  415. CFile fileStroe;
  416. if (!fileStroe.Open(g_szConfigFile,CFile::modeWrite|CFile::modeCreate))
  417. {
  418. MessageBox(_T("序列化失败，文件没有找到！"));
  419. CDialog::OnClose();
  420. return ;
  421. }
  422. CArchive arStore(&fileStroe,CArchive::store);
  423. //保存war3运行路径
  424. arStore<<g_war3InstallPath;
  426. //保存按键设置
  427. for (int i=0;i<6;i++)
  428. {
  429. arStore<<g_numPAD[i][0];
  430. }
  431. arStore.Flush(); //不flush总是出现错误
  432. fileStroe.Flush();
  434. fileStroe.Close();
  436. CDialog::OnClose();
  437. }
  439. void CReKeyDlg::OnTimer(UINT_PTR nIDEvent)
  440. {
  441. // TODO: 在此添加消息处理程序代码和/或调用默认值
  442. hWar3 = ::FindWindow(NULL,L"Warcraft III");
  443. if (hWar3)
  444. {
  445. ::SendMessage(hWar3,WM_KEYDOWN,VK_OEM_4,0);
  446. ::SendMessage(hWar3,WM_KEYDOWN,VK_OEM_6,0);
  448. m_status.SetWindowText(L"War3状态：运行中");
  449. if (hHook == NULL)
  450. {
  451. hHook = SetWindowsHookEx(WH_KEYBOARD_LL,LowKeyboardHookProc,GetModuleHandle(NULL),0);
  453. HWND topWnd = ::GetForegroundWindow();
  454. DWORD processid;
  455. ::GetWindowThreadProcessId(hWar3,&processid);
  456. CString str = L"Game.dll";
  457. g_GameDllbase = ::GetProcessDllBase(processid,str);
  458. }
  459. }
  460. else
  461. {
  462. m_status.SetWindowText(L"War3状态：未启动");
  463. if (hHook != NULL)
  464. {
  465. if (UnhookWindowsHookEx(hHook))
  466. {
  467. hHook = NULL;
  468. }
  469. }
  470. }
  471. CDialog::OnTimer(nIDEvent);
  472. }
  474. void CReKeyDlg::OnEnChangeEdit1()
  475. {
  476. UniFormKey(m_edit1);
  477. CString key;
  478. m_edit1.GetWindowText(key);
  479. char *str = (char*)(LPCTSTR)key;
  480. g_numPAD[0][0] = *str;
  481. }
  483. void CReKeyDlg::OnEnChangeEdit2()
  484. {
  485. UniFormKey(m_edit2);
  486. CString key;
  487. m_edit2.GetWindowText(key);
  488. char *str = (char*)(LPCTSTR)key;
  489. g_numPAD[1][0] = *str;
  490. }
  492. void CReKeyDlg::OnEnChangeEdit4()
  493. {
  494. UniFormKey(m_edit4);
  495. CString key;
  496. m_edit4.GetWindowText(key);
  497. char *str = (char*)(LPCTSTR)key;
  498. g_numPAD[2][0] = *str;
  499. }
  501. void CReKeyDlg::OnEnChangeEdit5()
  502. {
  503. UniFormKey(m_edit5);
  504. CString key;
  505. m_edit5.GetWindowText(key);
  506. char *str = (char*)(LPCTSTR)key;
  507. g_numPAD[3][0] = *str;
  508. }
  510. void CReKeyDlg::OnEnChangeEdit7()
  511. {
  512. UniFormKey(m_edit7);
  513. CString key;
  514. m_edit7.GetWindowText(key);
  515. char *str = (char*)(LPCTSTR)key;
  516. g_numPAD[4][0] = *str;
  517. }
  519. void CReKeyDlg::OnEnChangeEdit8()
  520. {
  521. // TODO:  如果该控件是 RICHEDIT 控件，它将不
  522. // 发送此通知，除非重写 CDialog::OnInitDialog()
  523. // 函数并调用 CRichEditCtrl().SetEventMask()，
  524. // 同时将 ENM_CHANGE 标志“或”运算到掩码中。
  526. // TODO:  在此添加控件通知处理程序代码
  527. UniFormKey(m_edit8);
  528. CString key;
  529. m_edit8.GetWindowText(key);
  530. char *str = (char*)(LPCTSTR)key;
  531. g_numPAD[5][0] = *str;
  532. }
  534. //使各键不至于重复
  535. bool CReKeyDlg::UniFormKey(CEdit& edit)
  536. {
  537. CString strc;
  538. edit.GetWindowText(strc);
  540. //空表示无设置，此时不应检查其唯一性，这是可能还有好几个按键是无设置的呢
  541. if (strc.Compare(_T(""))==0)
  542. {
  543. return true;
  544. }
  546. //由于TCHAR字符是2位，而且它按小尾的字节顺序排列，所以第一字节是char值
  547. //考虑 TCHAR tc = 'T';
  548. // char c = (char)(int)tc;
  549. char * c= (char*)(LPCTSTR)strc;
  551. HWND hWndEdit = edit.GetSafeHwnd();
  552. int iEdit = -1;
  553. if (hWndEdit == m_edit1.GetSafeHwnd())
  554. {
  555. iEdit = 0;
  556. }else if (hWndEdit == m_edit2.GetSafeHwnd())
  557. {
  558. iEdit = 1;
  559. }else if (hWndEdit == m_edit4.GetSafeHwnd())
  560. {
  561. iEdit = 2;
  562. }else if (hWndEdit == m_edit5.GetSafeHwnd())
  563. {
  564. iEdit = 3;
  565. }else if (hWndEdit == m_edit7.GetSafeHwnd())
  566. {
  567. iEdit = 4;
  568. }else if (hWndEdit == m_edit8.GetSafeHwnd())
  569. {
  570. iEdit = 5;
  571. }
  573. if (iEdit==-1)
  574. {
  575. return false;
  576. }
  578. int i=0;
  579. for (;i<6;i++)
  580. {
  581. if (*c == g_numPAD[i][0])
  582. {
  583. if (iEdit==i)
  584. continue;
  585. else
  586. break;
  587. }
  588. }
  590. switch (i)
  591. {
  592. case 0:
  593. m_edit1.SetWindowText(_T(""));
  594. break;
  595. case 1:
  596. m_edit2.SetWindowText(_T(""));
  597. break;
  598. case 2:
  599. m_edit4.SetWindowText(_T(""));
  600. break;
  601. case 3:
  602. m_edit5.SetWindowText(_T(""));
  603. break;
  604. case 4:
  605. m_edit7.SetWindowText(_T(""));
  606. break;
  607. case 5:
  608. m_edit8.SetWindowText(_T(""));
  609. break;
  610. default:
  611. break;
  612. }
  613. return true;
  614. }
  616. LRESULT CReKeyDlg::OnShowTask(WPARAM wParam, LPARAM lParam)
  617. {
  618. if(wParam != IDR_MAINFRAME)
  619. return 1;
  621. CMenu menu;
  622. LPPOINT lpoint;
  624. switch (lParam)
  625. {
  626. case WM_RBUTTONUP:
  627. lpoint = new tagPOINT;
  628. ::GetCursorPos(lpoint);
  630. menu.CreatePopupMenu();
  631. menu.AppendMenu(MF_STRING,IDM_NF_MENU_EXIT,L"退出ReKey");
  632. menu.TrackPopupMenu(TPM_LEFTALIGN,lpoint->x,lpoint->y,this);
  633. menu.Detach();
  635. menu.DestroyMenu();
  636. delete lpoint;
  638. break;
  640. case WM_LBUTTONDBLCLK:
  641. this->ShowWindow(SW_SHOWNORMAL);
  642. SetForegroundWindow();
  643. break;
  645. default:
  646. break;
  647. }
  649. return 0;
  650. }
  652. //托盘菜单退出选项处理
  653. void CReKeyDlg::OnNFMenuExit()
  654. {
  655. this->SendMessage(WM_CLOSE);
  656. }
  658. void CReKeyDlg::OnSetWar3Path()
  659. {
  660. CFileDialog fileDLg(TRUE,NULL,NULL,OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT,_T("可执行文件(*.exe)|*.exe||"));
  661. if (fileDLg.DoModal() == IDOK)
  662. {
  663. //更改war3路径
  664. m_staticTooltip.SetWindowText(_T("<-点击图标启动War3"));
  665. g_war3InstallPath = fileDLg.GetPathName();
  666. }
  667. //维持g_war3InstallPath不变
  668. }
  670. void* GetProcessDllBase(DWORD dwPID, const CString& moduleName)
  671. {
  672. HANDLE hModuleSnap = INVALID_HANDLE_VALUE;
  673. MODULEENTRY32 me32;
  675. hModuleSnap = ::CreateToolhelp32Snapshot(TH32CS_SNAPMODULE,dwPID);
  676. if (hModuleSnap == INVALID_HANDLE_VALUE)
  677. {
  678. DWORD dwErr = GetLastError();
  679. return 0;
  680. }
  682. me32.dwSize = sizeof(MODULEENTRY32);
  684. if (!Module32First(hModuleSnap,&me32))
  685. {
  686. CloseHandle(hModuleSnap);
  687. return 0;
  688. }
  690. CString strModule;
  691. void* base=0;
  692. do 
  693. {
  694. strModule = CString(me32.szModule);
  695. if (moduleName == strModule)
  696. {
  697. base = (void*)me32.modBaseAddr;
  698. break;
  699. }
  700. } while (Module32Next(hModuleSnap,&me32));
  702. CloseHandle(hModuleSnap);
  704. return base;
  705. }
  707. void CReKeyDlg::OnStnClickedStartwariii()
  708. {
  709. // 检查war3路径并设置
  710. if ( !g_war3InstallPath.GetLength() )
  711. {
  712. CFileDialog fileDLg(TRUE,NULL,NULL,OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT,_T("可执行文件(*.exe)|*.exe||"));
  713. if (fileDLg.DoModal() == IDOK)
  714. {
  715. m_staticTooltip.SetWindowText(_T("<-点击图标启动War3"));
  716. g_war3InstallPath = fileDLg.GetPathName();
  717. return ; //设置完毕不直接启动war3，感觉这样好些
  718. }
  719. else
  720. {
  721. return ; //没有设置路径，返回
  722. }
  723. }
  725. //war3已经启动，切换过去
  726. hWar3 = ::FindWindow(NULL,L"Warcraft III");
  727. if (hWar3)
  728. {
  729. ::ShowWindow(hWar3,SW_SHOWNORMAL);
  730. ::SetForegroundWindow(hWar3);
  731. return ;
  732. }
  734. //启动war3
  735. //g_war3InstallPath += _T(" -window");
  737. TCHAR szPath[MAX_PATH];
  738. TCHAR *pathBuff = g_war3InstallPath.GetBuffer(g_war3InstallPath.GetLength());
  739. _tcscpy_s(szPath,_countof(szPath),pathBuff);
  740. g_war3InstallPath.ReleaseBuffer();
  742. STARTUPINFO si;
  743. PROCESS_INFORMATION ppiProcess;
  744. memset(&si,0,sizeof(STARTUPINFO));
  746. si.cb = sizeof(STARTUPINFO);
  747. si.wShowWindow = SW_SHOWNORMAL;
  748. si.lpDesktop = NULL;
  749. si.dwFlags = STARTF_USESHOWWINDOW;
  751. if(!CreateProcess(NULL,szPath,NULL,NULL,TRUE,0,NULL,NULL,&si,&ppiProcess))
  752. {
  753. MessageBox(_T("War3进程创建失败!n"));
  754. return ;
  755. }
  757. CloseHandle(ppiProcess.hThread);
  758. CloseHandle(ppiProcess.hProcess);
  759. }
  761. void CReKeyDlg::OnBnClickedDebugswitch()
  762. {
  763. // TODO: 在此添加控件通知处理程序代码
  764. UpdateData(TRUE);
  766. //取得系统当前时间
  767. CString sTime;
  768. SYSTEMTIME st;
  769. GetLocalTime(&st);
  770. sTime.Format(_T("%d-%d-%d %d:%d:%d   "),st.wYear,st.wMonth,st.wDay,st.wHour,st.wMinute,st.wSecond);
  771. CString outLog;
  773. CString strValue;
  775. //创建输出信息
  776. if ( m_bIsDebug )
  777. {
  778. outLog = sTime + _T("Start Debug Outputn");
  779. strValue.Format(_T("%20s: 0x%08Xn%20s: 0x%08Xn%20s: 0x%08Xn"),
  780. _T("War3 Window Handle"),hWar3,
  781. _T("Hook Handle"),hHook,
  782. _T("Game.dll Addr"),(DWORD)g_GameDllbase);
  783. outLog += strValue;
  785. //负责打开文件
  786. g_DebugFile.Open(g_szOutDebugFile,CFile::modeWrite|CFile::modeCreate|CFile::modeNoTruncate|CFile::shareDenyWrite);
  787. g_DebugFile.SeekToEnd();
  789. //写入信息
  790. //g_DebugFile.WriteString((LPCTSTR)outLog);
  791. WriteDebugFile(outLog);
  792. }
  793. else
  794. {
  795. outLog = sTime + _T("Stop Debug Output.nn");
  797. //写入信息
  798. //g_DebugFile.WriteString((LPCTSTR)outLog);
  799. WriteDebugFile(outLog);
  801. //负责关闭文件
  802. if (g_DebugFile.m_pStream)
  803. {
  804. g_DebugFile.Close();
  805. }
  806. }
  807. }
  809. void WriteDebugFile(const CString &str)
  810. {
  811. //没有打开文件，返回
  812. if (NULL == g_DebugFile.m_pStream)
  813. return ;
  815. g_DebugFile.WriteString((LPCTSTR)str);
  816. g_DebugFile.Flush();
  818. }