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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > 远程控制编程 > 查看源码
MainFrm.cpp
资源名称：bo2ksrc.zip　[点击查看]
上传用户：jinandeyu
上传日期：2007-01-05
资源大小：620k
文件大小：25k
源码类别：

远程控制编程

开发平台：

WINDOWS

MainFrm.cpp：源码内容
  1. /*  Back Orifice 2000 - Remote Administration Suite
  2.     Copyright (C) 1999, Cult Of The Dead Cow
  4.     This program is free software; you can redistribute it and/or modify
  5.     it under the terms of the GNU General Public License as published by
  6.     the Free Software Foundation; either version 2 of the License, or
  7.     (at your option) any later version.
  9.     This program is distributed in the hope that it will be useful,
  10.     but WITHOUT ANY WARRANTY; without even the implied warranty of
  11.     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12.     GNU General Public License for more details.
  14.     You should have received a copy of the GNU General Public License
  15.     along with this program; if not, write to the Free Software
  16.     Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  18. The author of this program may be contacted at dildog@l0pht.com. */
  20. // MainFrm.cpp : implementation of the CMainFrame class
  21. //
  23. #include "stdafx.h"
  24. #include "bo2kgui.h"
  25. #include "MainFrm.h"
  26. #include "pluginconfig.h"
  27. #include <auth.h>
  28. #include <nullauth.h>
  29. #include <iohandler.h>
  30. #include <io_simpleudp.h>
  31. #include <io_simpletcp.h>
  32. #include <encryption.h>
  33. #include <xorencrypt.h>
  34. #include <deshash.h>
  35. #include <dll_load.h>
  36. #include <plugins.h>
  37. #include <interactiveconnect.h>
  38. #include <interactivelisten.h>
  39. #include "Splash.h"
  41. #ifdef _DEBUG
  42. #define new DEBUG_NEW
  43. #undef THIS_FILE
  44. static char THIS_FILE[] = __FILE__;
  45. #endif
  48. #define WM_CDIALOG_DO_MODAL (WM_USER+123)
  49. #define WM_TASKBAR_CALLBACK (WM_USER+124)
  51. typedef struct {
  52. CDialog *pDialog;
  53. int nRet;
  54. HANDLE hTrigger;
  55. } CDIALOGDOMODALPARAM;
  58. /////////////////////////////////////////////////////////////////////////////
  59. // CMainFrame
  61. IMPLEMENT_DYNCREATE(CMainFrame, CCJFrameWnd)
  63. BEGIN_MESSAGE_MAP(CMainFrame, CCJFrameWnd)
  64. //{{AFX_MSG_MAP(CMainFrame)
  65. ON_WM_CREATE()
  66. ON_COMMAND(IDM_SERVERLIST, OnServerlist)
  67. ON_UPDATE_COMMAND_UI(IDM_SERVERLIST, OnUpdateServerlist)
  68. ON_WM_ERASEBKGND()
  69. ON_COMMAND(IDM_PLUGINS, OnPlugins)
  70. ON_COMMAND(IDM_GRADIENTS, OnGradients)
  71. ON_UPDATE_COMMAND_UI(IDM_GRADIENTS, OnUpdateGradients)
  72. ON_COMMAND(IDM_MENUTYPE, OnMenuType)
  73. ON_UPDATE_COMMAND_UI(IDM_MENUTYPE, OnUpdateMenuType)
  74. ON_COMMAND(ID_SERVER_NEW, OnServerNew)
  75. ON_UPDATE_COMMAND_UI(ID_SERVER_NEW, OnUpdateServerNew)
  76. ON_COMMAND(ID_SERVER_EDIT, OnServerEdit)
  77. ON_UPDATE_COMMAND_UI(ID_SERVER_EDIT, OnUpdateServerEdit)
  78. ON_COMMAND(ID_SERVER_DELETE, OnServerDelete)
  79. ON_UPDATE_COMMAND_UI(ID_SERVER_DELETE, OnUpdateServerDelete)
  80. ON_WM_ACTIVATE()
  81. // ON_COMMAND(IDM_SYSTRAY, OnSysTray)
  82. // ON_UPDATE_COMMAND_UI(IDM_SYSTRAY, OnUpdateSysTray)
  83. // ON_MESSAGE(WM_TASKBAR_CALLBACK, OnTrayNotification)
  84. ON_COMMAND_RANGE(IDM_PLUGIN_0,IDM_PLUGIN_4095,OnPluginMenu)
  85. ON_WM_TIMER()
  86. ON_WM_SHOWWINDOW()
  87. //}}AFX_MSG_MAP
  88. END_MESSAGE_MAP()
  90. static UINT indicators[] =
  91. {
  92. ID_SEPARATOR,           // status line indicator
  93. ID_INDICATOR_CAPS,
  94. ID_INDICATOR_NUM,
  95. ID_INDICATOR_SCRL,
  96. };
  98. /////////////////////////////////////////////////////////////////////////////
  99. // The global handler objects
  101. CIOHandler *g_pIOHandler;
  102. CEncryptionHandler *g_pEncryptionHandler;
  103. CAuthHandler *g_pAuthHandler;
  105. /////////////////////////////////////////////////////////////////////////////
  106. // CMainFrame construction/destruction
  108. CMainFrame::CMainFrame()
  109. {
  110. // TODO: add member initialization code here
  111. m_pWndServerList=NULL;
  112. m_bGradients=FALSE;
  113. // m_bSysTray=FALSE;
  114. // m_pTrayIcon=NULL;
  116. g_pIOHandler=NULL;
  117. g_pEncryptionHandler=NULL;
  118. g_pAuthHandler=NULL;
  120. m_arrPluginInfo.RemoveAll();
  121. m_arrVarInfo.RemoveAll();
  123. }
  125. CMainFrame::~CMainFrame()
  126. {
  127. }
  129. int CMainFrame::OnCreate(LPCREATESTRUCT lpCreateStruct)
  130. {
  131. if (CCJFrameWnd::OnCreate(lpCreateStruct) == -1)
  132. return -1;
  134. if (!m_wndStatusBar.Create(this) ||
  135. !m_wndStatusBar.SetIndicators(indicators,
  136.   sizeof(indicators)/sizeof(UINT)))
  137. {
  138. TRACE0("Failed to create status barn");
  139. return -1;      // fail to create
  140. }
  142. // Get config
  143. m_bGradients=AfxGetApp()->GetProfileInt("Config","Gradients",FALSE);
  144. m_bScreenReader=AfxGetApp()->GetProfileInt("Config","Compatible Menus",TRUE);
  145. // m_bSysTray=AfxGetApp()->GetProfileInt("Config","System Tray",0);
  147. // Start up toolbars and docking windows
  148. CSize defaultSize(GetSystemMetrics(SM_CXSCREEN)/4,GetSystemMetrics(SM_CYSCREEN)/5);
  149. SetInitialSize(defaultSize.cy,defaultSize.cy,defaultSize.cx,defaultSize.cx);
  150. EnableDocking(CBRS_ALIGN_ANY);
  151. EnableDockingSizeBar(CBRS_ALIGN_ANY);
  153. if(InitializeToolbars()==-1) return FALSE;
  154. if(InitializeDockingWindows()==-1) return FALSE;
  156. // Load up built-in IO and plugins
  157. if(InitializePlugins()==-1) return FALSE;
  159. CSplash spl;
  160. spl.DoModal();
  162. return 0;
  163. }
  166. int CMainFrame::InitializeToolbars()
  167. {
  168. // Create main toolbar
  169. if (!m_wndTBMain.Create(this) ||
  170. !m_wndTBMain.LoadToolBar(IDR_MAINFRAME)) {
  171. TRACE0("Failed to create main toolbarn");
  172. return -1;      // fail to create
  173. }
  174. m_wndTBMain.SetWindowText(_T("Main toolbar"));
  175. m_wndTBMain.SetBarStyle(m_wndTBMain.GetBarStyle() | CBRS_TOOLTIPS | CBRS_FLYBY | CBRS_SIZE_DYNAMIC);
  176. m_wndTBMain.EnableDocking(CBRS_ALIGN_ANY);
  177. m_wndTBMain.SetButtonSize(CSize(24,22));
  179. // Create main menu bar
  180. m_menuBar.m_bAutoRemoveFrameMenu=FALSE;
  181. m_menuBar.Create(this, WS_CHILD | WS_VISIBLE | CBRS_TOP, IDR_MAINFRAME);
  182. m_menuBar.LoadMenu(GetMenu()->GetSafeHmenu());
  184. m_menuBar.SetWindowText(_T("Menu"));
  185. m_menuBar.SetBarStyle(m_menuBar.GetBarStyle() | CBRS_TOOLTIPS | CBRS_FLYBY | CBRS_SIZE_DYNAMIC);
  186. m_menuBar.EnableDocking(CBRS_ALIGN_TOP | CBRS_ALIGN_BOTTOM);
  187. m_menuBar.SetButtonSize(CSize(24,22));
  189. m_menuManager.Install(this);
  190. m_menuManager.LoadToolbar(IDR_MAINFRAME);
  192. if(m_bScreenReader) {
  193. m_menuManager.m_bShowButtons=FALSE;
  194. m_menuManager.Refresh();
  195. m_menuBar.ShowWindow(SW_HIDE);
  196. DockControlBar(&m_wndTBMain);
  197. } else {
  198. m_menuManager.m_bShowButtons=TRUE;
  199. m_menuManager.Refresh();
  201. m_menuBar.ShowWindow(SW_SHOW);
  202. SetMenu(NULL);
  203. DockControlBar(&m_menuBar);
  205. DockControlBarLeftOf(&m_wndTBMain,&m_menuBar);
  206. DockControlBar(&m_wndTBMain);
  207. }
  209. return 0;
  210. }
  212. int CMainFrame::InitializeDockingWindows()
  213. {
  214. // Create workspace dockbar
  215. CString title("Server List");
  216. if (!m_wndServerListBDB.Create(this,  IDD_SERVERLIST, title)) {
  217. TRACE0("Failed to create dialogbarn");
  218. return -1;      // fail to create
  219. }
  220. m_wndServerListBDB.SetBarStyle((m_wndServerListBDB.GetBarStyle() & ~CBRS_ALIGN_ANY) |
  221. CBRS_TOOLTIPS | CBRS_BOTTOM | CBRS_FLYBY | CBRS_SIZE_DYNAMIC);
  222. m_wndServerListBDB.EnableDockingOnSizeBar(CBRS_ALIGN_ANY);
  225. m_pWndServerList=new CServerList();
  226. m_pWndServerList->Create(CServerList::IDD,&m_wndServerListBDB);
  227. m_wndServerListBDB.SetDialog(m_pWndServerList);
  228. m_pWndServerList->UseGradient(m_bGradients);
  230. DockSizeBar(&m_wndServerListBDB);
  232. CMenu menu;
  233. menu.Attach(m_menuBar.GetMenu());
  234. menu.CheckMenuItem(IDD_SERVERLIST, MF_CHECKED);
  235. menu.Detach();
  237. return 0;
  238. }
  243. /////////////////////////////////////////////////////////////////////////////
  244. // CMainFrame diagnostics
  246. #ifdef _DEBUG
  247. void CMainFrame::AssertValid() const
  248. {
  249. CCJFrameWnd::AssertValid();
  250. }
  252. void CMainFrame::Dump(CDumpContext& dc) const
  253. {
  254. CCJFrameWnd::Dump(dc);
  255. }
  257. #endif //_DEBUG
  259. /////////////////////////////////////////////////////////////////////////////
  260. // CMainFrame message handlers
  263. void CMainFrame::OnServerlist() 
  264. {
  265. if(m_wndServerListBDB.IsWindowVisible()) {
  266. ShowControlBar(&m_wndServerListBDB, FALSE, FALSE);
  267. } else {
  268. ShowControlBar(&m_wndServerListBDB, TRUE, FALSE);
  269. }
  270. }
  272. void CMainFrame::OnUpdateServerlist(CCmdUI* pCmdUI) 
  273. {
  274. if(m_wndServerListBDB.IsWindowVisible()) {
  275. pCmdUI->SetCheck(1);
  276. } else {
  277. pCmdUI->SetCheck(0);
  278. }
  279. }
  281. BOOL CMainFrame::DestroyWindow() 
  282. {
  284. // Clean up server list
  285. if(m_pWndServerList) {
  286. m_pWndServerList->CleanUpAndClose();
  287. delete m_pWndServerList;
  288. m_pWndServerList=NULL;
  289. }
  291. // Kill all the plugins
  292. TerminatePlugins();
  294. // AfxGetApp()->WriteProfileInt("Config","System Tray",m_bSysTray);
  295. AfxGetApp()->WriteProfileInt("Config","Compatible Menus",m_bScreenReader);
  296. AfxGetApp()->WriteProfileInt("Config","Gradients",m_bGradients);
  298. // Delete the tray icon
  299. // if(m_pTrayIcon) {
  300. // delete m_pTrayIcon;
  301. // m_pTrayIcon=NULL;
  302. // }
  304. return CCJFrameWnd::DestroyWindow();
  305. }
  307. BOOL CMainFrame::OnEraseBkgnd(CDC* pDC) 
  308. {
  309. return TRUE;
  310. //return CCJFrameWnd::OnEraseBkgnd(pDC);
  311. }
  313. void CMainFrame::OnPlugins() 
  314. {
  315. CPluginConfig cpg;
  317. cpg.DoModal();
  318. }
  320. void CMainFrame::OnGradients() 
  321. {
  322. m_bGradients=!m_bGradients;
  324. m_pWndServerList->UseGradient(m_bGradients);
  325. }
  327. void CMainFrame::OnUpdateGradients(CCmdUI* pCmdUI) 
  328. {
  329. if(m_bGradients) pCmdUI->SetCheck(1);
  330. else pCmdUI->SetCheck(0);
  331. }
  333. void CMainFrame::OnMenuType() 
  334. {
  335. m_bScreenReader=!m_bScreenReader;
  337. if(m_bScreenReader) {
  338. m_menuManager.m_bShowButtons=FALSE;
  339. m_menuManager.Refresh();
  341. DockControlBar(&m_menuBar);
  342. m_menuBar.ShowWindow(FALSE);
  343. SetMenu(CMenu::FromHandle(m_menuBar.GetMenu()));
  344. FloatControlBar(&m_wndTBMain,CPoint(0,0));
  345. DockControlBar(&m_wndTBMain);
  346. } else {
  347. m_menuManager.m_bShowButtons=TRUE;
  348. m_menuManager.Refresh();
  350. SetMenu(NULL);
  351. m_menuBar.ShowWindow(TRUE);
  352. DockControlBar(&m_menuBar);
  353. DockControlBarLeftOf(&m_wndTBMain,&m_menuBar);
  354. DockControlBar(&m_wndTBMain);
  355. }
  358. }
  360. void CMainFrame::OnUpdateMenuType(CCmdUI* pCmdUI) 
  361. {
  362. if(m_bScreenReader)
  363. pCmdUI->SetCheck(1);
  364. else 
  365. pCmdUI->SetCheck(0);
  366. }
  368. void CMainFrame::OnServerNew()
  369. {
  370. m_pWndServerList->OnNewmachine();
  371. }
  373. void CMainFrame::OnUpdateServerNew(CCmdUI* pCmdUI)
  374. {
  375. pCmdUI->Enable(m_pWndServerList->GetDlgItem(ID_NEWMACHINE)->IsWindowEnabled());
  376. }
  378. void CMainFrame::OnServerEdit()
  379. {
  380. m_pWndServerList->OnEditmachine();
  381. }
  383. void CMainFrame::OnUpdateServerEdit(CCmdUI* pCmdUI)
  384. {
  385. pCmdUI->Enable(m_pWndServerList->GetDlgItem(ID_EDITMACHINE)->IsWindowEnabled());
  386. }
  388. void CMainFrame::OnServerDelete()
  389. {
  390. m_pWndServerList->OnDeletemachine();
  391. }
  393. void CMainFrame::OnUpdateServerDelete(CCmdUI* pCmdUI)
  394. {
  395. pCmdUI->Enable(m_pWndServerList->GetDlgItem(ID_DELETEMACHINE)->IsWindowEnabled());
  396. }
  398. void CMainFrame::OnSysTray()
  399. {
  400. /* m_bSysTray=!m_bSysTray;
  401. if(m_bSysTray) {
  402. m_pTrayIcon=new CTrayIcon(IDR_TRAYICON);
  403. m_pTrayIcon->SetNotificationWnd(this,WM_TASKBAR_CALLBACK);
  404. } else {
  405. delete m_pTrayIcon;
  406. m_pTrayIcon=NULL;
  407. }*/
  408. }
  410. void CMainFrame::OnUpdateSysTray(CCmdUI* pCmdUI)
  411. {
  412. // pCmdUI->SetCheck(m_bSysTray);
  413. }
  417. int CMainFrame::RegisterClientMenu(LPCSTR szCategory, LPCSTR szComName, TYPEOF_ClientMenu *pProc)
  418. {
  419. CMenu menu, *pMenu, *pSubMenu;
  421. // Add to plugins menu
  422. menu.Attach(m_menuBar.GetMenu());
  423. pMenu=menu.GetSubMenu(3);
  424. if(pMenu==NULL) return -1;
  425. menu.Detach();
  427. // Find/create submenu for plugin
  428. int i,count;
  429. count=pMenu->GetMenuItemCount();
  430. for(i=2;i<count;i++) {
  431. CString str;
  432. pMenu->GetMenuString(i,str,MF_BYPOSITION);
  433. if(str.CompareNoCase(szCategory)==0) break;
  434. }
  435. if(i==count) {
  436. pSubMenu=new CMenu();
  437. if(pSubMenu==NULL) return -1;
  438. pSubMenu->CreatePopupMenu();
  440. pMenu->InsertMenu(i,MF_POPUP|MF_STRING|MF_ENABLED,(UINT)pSubMenu->GetSafeHmenu(),szCategory);
  441. } else {
  442. pSubMenu=pMenu->GetSubMenu(i);
  443. }
  445. // Find empty id;
  446. int nId;
  447. nId=0;
  448. count=m_arrPluginMenuItems.GetSize();
  450. i=0;
  451. while(i!=count) {
  452. for(i=0;i<count;i++) {
  453. if(m_arrPluginMenuItems[i].nId==nId) {
  454. nId++;
  455. break;
  456. }
  457. }
  458. }
  460. // Insert item for plugin
  461. PLUGINMENUITEM pmi;
  462. pmi.nId=nId;
  463. pmi.pProc=pProc;
  465. m_arrPluginMenuItems.Add(pmi);
  466. pSubMenu->AppendMenu(MF_STRING|MF_ENABLED,IDM_PLUGIN_0+nId,szComName);
  468. return 0;
  469. }
  471. int CMainFrame::UnregisterClientMenu(LPCSTR szCategory, LPCSTR szComName)
  472. {
  473. CMenu menu, *pMenu, *pSubMenu;
  475. // Remove from plugins menu
  476. menu.Attach(m_menuBar.GetMenu());
  477. pMenu=menu.GetSubMenu(3);
  478. if(pMenu==NULL) return -1;
  479. menu.Detach();
  482. // Find submenu for plugin
  483. int k,count;
  484. count=pMenu->GetMenuItemCount();
  485. for(k=2;k<count;k++) {
  486. CString str;
  487. pMenu->GetMenuString(k,str,MF_BYPOSITION);
  488. if(str.CompareNoCase(szCategory)==0) break;
  489. }
  490. if(k==count) {
  491. return -1;
  492. } else {
  493. pSubMenu=pMenu->GetSubMenu(k);
  494. }
  496. // Find menu item
  497. int i;
  498. count=pSubMenu->GetMenuItemCount();
  499. for(i=0;i<count;i++) {
  500. CString str;
  501. pSubMenu->GetMenuString(i,str,MF_BYPOSITION);
  502. if(str.CompareNoCase(szComName)==0) break;
  503. }
  504. if(i==count) return -1;
  506. // Remove menu item
  507. int j,pcount,nId;
  508. nId=pSubMenu->GetMenuItemID(i)-IDM_PLUGIN_0;
  509. pSubMenu->RemoveMenu(i,MF_BYPOSITION);
  510. if(count==1) {
  511. pMenu->RemoveMenu(k,MF_BYPOSITION);
  512. delete pSubMenu;
  513. }
  515. pcount=m_arrPluginMenuItems.GetSize();
  516. for(j=pcount-1;j>=0;j--) {
  517. if(m_arrPluginMenuItems[j].nId==nId) {
  518. m_arrPluginMenuItems.RemoveAt(j);
  519. break;
  520. }
  521. }
  522. if(j==-1) return -1;
  525. return 0;
  526. }
  528. int CMainFrame::InitializePlugins()
  529. {
  530. // Start DLL Loader
  531. InitializeDLLLoad();
  533. // Start IO Subsystem
  534. g_pIOHandler=new CIOHandler();
  535. if(g_pIOHandler==NULL) return -1;
  537. // Start Encryption Subsystem
  538. g_pEncryptionHandler=new CEncryptionHandler();
  539. if(g_pEncryptionHandler==NULL) return -1;
  541. // Start Authentication Subsystem
  542. g_pAuthHandler=new CAuthHandler();
  543. if(g_pAuthHandler==NULL) return -1;
  545. // Add built-in stuff
  546. g_pIOHandler->Insert(GetSimpleUdpIOHandler());
  547. g_pIOHandler->Insert(GetSimpleTcpIOHandler());
  548. g_pEncryptionHandler->Insert(GetXOREncryptionEngine());
  549. //g_pEncryptionHandler->Insert(GetDESEncryptionEngine());
  550. g_pAuthHandler->Insert(GetNullAuthHandler());
  552. // Install plugins
  553. int i,count;
  554. count=AfxGetApp()->GetProfileInt("Plugins","Count",0);
  555. for(i=0;i<count;i++) {
  556. CString strKey,strValue;
  557. strKey.Format("Plugin_%d",i);
  558. strValue=AfxGetApp()->GetProfileString("Plugins",strKey,NULL);
  560. AddPlugin(strValue);
  561. }
  563. // Add variables for client executable
  564. char *base=(char *)AfxGetInstanceHandle();
  565. PIMAGE_OPTIONAL_HEADER poh=(PIMAGE_OPTIONAL_HEADER)OPTHDROFFSET(base);
  566. AddVariables(base, poh->SizeOfImage, -1);
  567. LoadPluginConfig(-1);
  569. return 0;
  570. }
  572. int CMainFrame::TerminatePlugins()
  573. {
  574. // Save client config
  575. SavePluginConfig(-1);
  577. // Unload plugins
  578. int i,count;
  579. count=m_arrPluginInfo.GetSize();
  580. AfxGetApp()->WriteProfileInt("Plugins","Count",count);
  581. for(i=(count-1);i>=0;i--) {
  582. CString strKey;
  583. strKey.Format("Plugin_%d",i);
  584. AfxGetApp()->WriteProfileString("Plugins",strKey,m_arrPluginInfo[i].svFilename);
  586. RemovePlugin(i);
  587. }
  589. // Purge extra registry keys
  590. i=count; 
  591. while(1) {
  592. CString strKey;
  593. strKey.Format("Plugin_%d",i);
  594. if(AfxGetApp()->GetProfileString("Plugins",strKey).IsEmpty()) break;
  595. AfxGetApp()->WriteProfileString("Plugins",strKey,NULL);
  596. i++;
  597. }
  599. // Terminate Auth Subsystem
  600. if(g_pAuthHandler!=NULL) {
  601. for(i=0;i<MAX_AUTH_HANDLERS;i++) {
  602. g_pAuthHandler->Remove(i);
  603. }
  604. delete g_pAuthHandler;
  605. }
  607. // Terminate Encryption Subsystem
  608. if(g_pEncryptionHandler!=NULL) {
  609. for(i=0;i<MAX_ENCRYPTION_ENGINES;i++) {
  610. g_pEncryptionHandler->Remove(i);
  611. }
  612. delete g_pEncryptionHandler;
  613. }
  615. // Terminate IO Subsystem
  616. if(g_pIOHandler!=NULL) {
  617. for(i=0;i<MAX_IO_HANDLERS;i++) {
  618. g_pIOHandler->Remove(i);
  619. }
  620. delete g_pIOHandler;
  621. }
  623. // Stop DLL Loader
  624. KillDLLLoad();
  626. return 0;
  627. }
  629. int CMainFrame::AddPlugin(CString strPath)
  630. {
  631. PLUGIN_INFO pi;
  633. pi.pPluginImage=NULL;
  634. pi.dwPluginLen=0;
  635. pi.svFilename[0]=0;
  636. pi.svDescription[0]=0;
  637. pi.wVersionLo=0;
  638. pi.wVersionHi=0;
  639. pi.wBOVersionLo=0;
  640. pi.wBOVersionHi=0;
  643. // Load Plugin DLL 
  644. HINSTANCE hDLL;
  645. hDLL=LoadLibrary(strPath);
  646. if(hDLL==NULL)
  647. hDLL=LoadDLL(strPath);
  649. if(hDLL!=NULL) {
  650. pi.pPluginImage=(BYTE *)hDLL;
  651. PIMAGE_OPTIONAL_HEADER poh=(PIMAGE_OPTIONAL_HEADER)OPTHDROFFSET(pi.pPluginImage);
  652. pi.dwPluginLen=poh->SizeOfImage;
  654. // Call plugin version information function
  655. PLUGIN_VERSION pv;
  656. TYPEOF_PluginVersion *PluginVersion=(TYPEOF_PluginVersion *)GetDLLProcAddress(hDLL,"PluginVersion");
  657. if(PluginVersion!=NULL) {
  658. if(PluginVersion(&pv)) {
  659. // Copy plugin version information
  660. lstrcpyn(pi.svFilename,strPath,MAX_PATH+1);
  661. lstrcpyn(pi.svDescription,pv.svDescription,256);
  662. pi.wVersionLo=pv.wVersionLo;
  663. pi.wVersionHi=pv.wVersionHi;
  664. pi.wBOVersionLo=pv.wBOVersionLo;
  665. pi.wBOVersionHi=pv.wBOVersionHi;
  666. }
  667. }
  669. // Extract variable list
  670. int nPlugin=m_arrPluginInfo.Add(pi);
  671. if(nPlugin>=0) {
  672. AddVariables((char *)pi.pPluginImage,pi.dwPluginLen,nPlugin);
  673. LoadPluginConfig(nPlugin);
  674. } else {
  675. FreeDLL(hDLL);
  676. FreeLibrary(hDLL);
  677. }
  679. // Start plugin
  680. PLUGIN_LINKAGE pl;
  682. pl.pDispatchCommand=NULL;
  683. pl.pEncryptionHandler=g_pEncryptionHandler;
  684. pl.pIOHandler=g_pIOHandler;
  685. pl.pIssueAuthCommandRequest=&IssueAuthCommandRequest;
  686. pl.pIssueAuthCommandReply=&IssueAuthCommandReply;
  687. pl.pRegisterClientMenu=&::RegisterClientMenu;
  688. pl.pRegisterCommand=NULL;
  689. pl.pUnregisterClientMenu=&::UnregisterClientMenu;
  690. pl.pUnregisterCommand=NULL;
  691. pl.pListenAuthSocket=&ListenAuthSocket;
  692. pl.pConnectAuthSocket=&ConnectAuthSocket;
  693. pl.pInteractiveConnect=&InteractiveConnect;
  694. pl.pInteractiveListen=&InteractiveListen;
  696. TYPEOF_InstallPlugin *InstallPlugin=(TYPEOF_InstallPlugin *)GetDLLProcAddress(hDLL,"InstallPlugin");
  697. if(InstallPlugin(pl)==-1) {
  698. AfxMessageBox("Error installing plugin.",MB_OK|MB_ICONWARNING|MB_SETFOREGROUND|MB_TOPMOST);
  699. RemovePlugin(nPlugin);
  700. return -1;
  701. }
  703. return nPlugin;
  704. }
  706. return -1;
  707. }
  709. int CMainFrame::RemovePlugin(int nPlugin)
  710. {
  711. if(nPlugin>=m_arrPluginInfo.GetSize()) return -1;
  712. if(nPlugin<0) return -1;
  714. HMODULE hDLL=(HMODULE)m_arrPluginInfo[nPlugin].pPluginImage;
  716. // Terminate plugin
  717. TYPEOF_TerminatePlugin *TerminatePlugin=(TYPEOF_TerminatePlugin *)GetDLLProcAddress(hDLL,"TerminatePlugin");
  718. TerminatePlugin();
  720. // Save plugin configuration to registry
  721. if(SavePluginConfig(nPlugin)==-1) {
  722. AfxMessageBox("Unable to save plugin config to registry",MB_OK|MB_SETFOREGROUND|MB_TOPMOST|MB_ICONWARNING);
  723. }
  725. // Remove from memory and plugin info list
  726. FreeDLL(hDLL);
  727. FreeLibrary(hDLL);
  729. m_arrPluginInfo.RemoveAt(nPlugin);
  731. return 0;
  732. }
  736. void CMainFrame::AddVariables(char *pBuffer, DWORD dwLen, int nPlugin)
  737. {
  738. DWORD pos;
  739. for(pos=0;pos<dwLen-10;pos++) {
  740. if(memcmp(pBuffer+pos,"<**CFG**>",9)==0) {
  741. pos+=9;
  742. // Get category name
  743. VARIABLE_INFO vi;
  744. lstrcpyn(vi.svCategory,pBuffer+pos,256);
  745. pos+=(lstrlen(pBuffer+pos)+1);
  747. // Go through all variables
  748. while(*(pBuffer+pos)!='') {
  749. DWORD dwValPos;
  750. vi.VarType=*(pBuffer+pos);
  751. if(vi.VarType=='B') {
  752. pos+=2;
  753. // Get Name and Value
  754. DWORD dwStart=pos;
  755. while(*(pBuffer+pos)!='=') pos++;
  756. lstrcpyn(vi.svVarName,pBuffer+dwStart,min(pos-dwStart+1,256));
  757. pos++;
  758. dwValPos=pos;
  759. lstrcpyn(vi.svVarValue,pBuffer+pos,256);
  760. vi.nStrLen=1;
  761. pos+=2;
  762. } else if(vi.VarType=='N') {
  763. // Get Number Range
  764. pos+=2;
  765. vi.nNumLo=atoi(pBuffer+pos);
  766. while(*(pBuffer+pos)!=',') pos++;
  767. pos++;
  768. vi.nStrLen=pos;
  769. vi.nNumHi=atoi(pBuffer+pos);
  770. while(*(pBuffer+pos)!=']') pos++;
  771. vi.nStrLen=pos-vi.nStrLen;
  772. pos+=2;
  774. // Get Name and Value
  775. DWORD dwStart=pos;
  776. while(*(pBuffer+pos)!='=') pos++;
  777. lstrcpyn(vi.svVarName,pBuffer+dwStart,min(pos-dwStart+1,256));
  778. pos++;
  779. dwValPos=pos;
  780. lstrcpyn(vi.svVarValue,pBuffer+pos,256);
  781. pos+=(vi.nStrLen+1);
  782. } else if(vi.VarType=='S') {
  783. // Get Number Range
  784. pos+=2;
  785. vi.nStrLen=atoi(pBuffer+pos);
  786. while(*(pBuffer+pos)!=']') pos++;
  787. pos+=2;
  789. // Get Name and Value
  790. DWORD dwStart=pos;
  791. while(*(pBuffer+pos)!='=') pos++;
  792. lstrcpyn(vi.svVarName,pBuffer+dwStart,min(pos-dwStart+1,256));
  793. pos++;
  794. lstrcpyn(vi.svVarValue,pBuffer+pos,256);
  795. dwValPos=pos;
  796. pos+=(vi.nStrLen+1);
  797. } else {
  798. AfxMessageBox("Error parsing variables. Variable list could not be determined.");
  799. return;
  800. }
  802. // Add to variable info array
  803. vi.nPlugin=nPlugin;
  804. vi.dwPos=dwValPos;
  805. m_arrVarInfo.Add(vi);
  807. }
  808. }
  809. }
  810. }
  813. int CMainFrame::LoadPluginConfig(int nNum)
  814. {
  815. int i,count;
  817. count=m_arrVarInfo.GetSize();
  819. char *base;
  820. if(nNum==-1) {
  821. base=(char *)AfxGetInstanceHandle();
  822. } else {
  823. base=(char *)m_arrPluginInfo[nNum].pPluginImage;
  824. }
  826. for(i=0;i<count;i++) {
  827. VARIABLE_INFO vi=m_arrVarInfo[i];
  828. if(vi.nPlugin==nNum) {
  829. CString str;
  830. str=AfxGetApp()->GetProfileString(vi.svCategory,vi.svVarName,base+vi.dwPos);
  831. lstrcpyn(base+vi.dwPos,str,vi.nStrLen+1);
  832. }
  833. }
  835. return 0;
  836. }
  838. int CMainFrame::SavePluginConfig(int nNum)
  839. {
  840. int i,count;
  842. count=m_arrVarInfo.GetSize();
  844. char *base;
  845. if(nNum==-1) {
  846. base=(char *)AfxGetInstanceHandle();
  847. } else {
  848. base=(char *)m_arrPluginInfo[nNum].pPluginImage;
  849. }
  851. for(i=0;i<count;i++) {
  852. VARIABLE_INFO vi=m_arrVarInfo[i];
  853. if(vi.nPlugin==nNum) {
  854. AfxGetApp()->WriteProfileString(vi.svCategory,vi.svVarName,base+vi.dwPos);
  855. }
  856. }
  858. return 0;
  859. }
  862. void CMainFrame::OnPluginMenu(UINT nID) 
  863. {
  864. // Find id
  865. int i,count;
  867. count=m_arrPluginMenuItems.GetSize();
  868. for(i=0;i<count;i++) {
  869. if(m_arrPluginMenuItems[i].nId==(int)(nID-IDM_PLUGIN_0)) break;
  870. }
  871. if(i==count) return;
  873. // Call plugin procedure
  875. int nRet=m_arrPluginMenuItems[i].pProc(GetSafeHwnd());
  877. if(nRet==-1) {
  878. AfxMessageBox("An error occurred while starting up the plugin.nThings may not proceed as planned.n",MB_OK|MB_ICONWARNING|MB_SETFOREGROUND|MB_TOPMOST);
  879. }
  881. }
  883. BOOL CMainFrame::PreTranslateMessage(MSG *pMsg)
  884. {
  885. m_menuBar.TranslateFrameMessage(pMsg);
  887. return CCJFrameWnd::PreTranslateMessage(pMsg);
  888. }
  891. LRESULT CMainFrame::WindowProc(UINT message, WPARAM wParam, LPARAM lParam) 
  892. {
  893. switch(message) {
  894. case WM_CDIALOG_DO_MODAL: 
  895. {
  896. CDialog *cdlg=((CDIALOGDOMODALPARAM *)lParam)->pDialog;
  897. int nRet=cdlg->DoModal();
  898. ((CDIALOGDOMODALPARAM *)lParam)->nRet=nRet;
  899. SetEvent(((CDIALOGDOMODALPARAM *)lParam)->hTrigger);
  900. return TRUE;
  901. }
  902. }
  904. return CCJFrameWnd::WindowProc(message, wParam, lParam);
  905. }
  907. int CMainFrame::DoDialogModal(CDialog *dlg)
  908. {
  909. MSG msg;
  910. CDIALOGDOMODALPARAM cddmp;
  911. cddmp.pDialog=dlg;
  912. cddmp.nRet=-1;
  913. cddmp.hTrigger=CreateEvent(NULL,TRUE,FALSE,NULL);
  914. PostMessage(WM_CDIALOG_DO_MODAL,0,(LPARAM)&cddmp);
  915. while(WaitForSingleObject(cddmp.hTrigger,0)!=WAIT_OBJECT_0) {
  916. if(PeekMessage(&msg,NULL,0,0,PM_REMOVE)) {
  917. TranslateMessage(&msg);
  918. DispatchMessage(&msg);
  919. }
  920. Sleep(20);
  921. }
  922. CloseHandle(cddmp.hTrigger);
  923. return cddmp.nRet;
  924. }
  926. // Global callbacks for plugins
  928. int RegisterClientMenu(LPCSTR szCategory, LPCSTR szComName, TYPEOF_ClientMenu *pProc)
  929. {
  930. return ((CMainFrame *)AfxGetMainWnd())->RegisterClientMenu(szCategory,szComName, pProc);
  931. }
  933. int UnregisterClientMenu(LPCSTR szCategory, LPCSTR szComName)
  934. {
  935. return ((CMainFrame *)AfxGetMainWnd())->UnregisterClientMenu(szCategory,szComName);
  936. }
  939. int InteractiveConnect(HWND hParent, LPCSTR svBindStr,LPCSTR svNetMod,LPCSTR svEncryption,LPCSTR svAuth,char *svRBindStr,char *svRNetMod,char *svREncryption,char *svRAuth)
  940. {
  941. CMainFrame *pcmf=(CMainFrame *)AfxGetMainWnd();
  942. CWnd *pWnd=CWnd::FromHandle(hParent);
  943. CInteractiveConnect cic(pWnd);
  945. cic.m_strSvrAddr=svBindStr;
  946. cic.m_strSelectedIO=svNetMod;
  947. cic.m_strSelectedEnc=svEncryption;
  948. cic.m_strSelectedAuth=svAuth;
  950. if(pcmf->DoDialogModal(&cic)==IDOK) {
  951. lstrcpyn(svRBindStr,cic.m_strSvrAddr,256);
  952. lstrcpyn(svRNetMod,cic.m_strSelectedIO,256);
  953. lstrcpyn(svREncryption,cic.m_strSelectedEnc,256);
  954. lstrcpyn(svRAuth,cic.m_strSelectedAuth,256);
  955. return 0;
  956. }
  957. return -1;
  958. }
  960. int InteractiveListen(HWND hParent, LPCSTR svBindStr,LPCSTR svNetMod,LPCSTR svEncryption,LPCSTR svAuth,char *svRBindStr,char *svRNetMod,char *svREncryption,char *svRAuth)
  961. {
  962. CMainFrame *pcmf=(CMainFrame *)AfxGetMainWnd();
  963. CWnd *pWnd=CWnd::FromHandle(hParent);
  964. CInteractiveListen cil(pWnd);
  965.  
  966. cil.m_strSvrAddr=svBindStr;
  967. cil.m_strSelectedIO=svNetMod;
  968. cil.m_strSelectedEnc=svEncryption;
  969. cil.m_strSelectedAuth=svAuth;
  970. if(pcmf->DoDialogModal(&cil)==IDOK) {
  971. lstrcpyn(svRBindStr,cil.m_strSvrAddr,256);
  972. lstrcpyn(svRNetMod,cil.m_strSelectedIO,256);
  973. lstrcpyn(svREncryption,cil.m_strSelectedEnc,256);
  974. lstrcpyn(svRAuth,cil.m_strSelectedAuth,256);
  975. return 0;
  976. }
  977. return -1;
  978. }
  981. void CMainFrame::OnActivate(UINT nState, CWnd* pWndOther, BOOL bMinimized) 
  982. {
  983. CCJFrameWnd::OnActivate(nState, pWndOther, bMinimized);
  984. }
  987. /*LRESULT CMainFrame::OnTrayNotification(WPARAM uID, LPARAM lEvent)
  988. {
  989.    // let tray icon do default stuff
  990.    return m_pTrayIcon->OnTrayNotification(uID, lEvent);
  991. }
  992. */