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

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

远程控制编程

开发平台：

WINDOWS

pluginconfig.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. // pluginconfig.cpp : implementation file
  21. //
  23. #include "stdafx.h"
  24. #include "bo2kgui.h"
  25. #include "MainFrm.h"
  26. #include "pluginconfig.h"
  27. #include "dll_load.h"
  29. #ifdef _DEBUG
  30. #define new DEBUG_NEW
  31. #undef THIS_FILE
  32. static char THIS_FILE[] = __FILE__;
  33. #endif
  35. /////////////////////////////////////////////////////////////////////////////
  36. // CPluginConfig dialog
  39. CPluginConfig::CPluginConfig(CWnd* pParent /*=NULL*/)
  40. : CDialog(CPluginConfig::IDD, pParent)
  41. {
  42. //{{AFX_DATA_INIT(CPluginConfig)
  43. // NOTE: the ClassWizard will add member initialization here
  44. //}}AFX_DATA_INIT
  45. }
  48. void CPluginConfig::DoDataExchange(CDataExchange* pDX)
  49. {
  50. CDialog::DoDataExchange(pDX);
  51. //{{AFX_DATA_MAP(CPluginConfig)
  52. // NOTE: the ClassWizard will add DDX and DDV calls here
  53. //}}AFX_DATA_MAP
  54. }
  57. BEGIN_MESSAGE_MAP(CPluginConfig, CDialog)
  58. //{{AFX_MSG_MAP(CPluginConfig)
  59. ON_BN_CLICKED(IDC_INSERT, OnInsert)
  60. ON_BN_CLICKED(IDC_REMOVE, OnRemove)
  61. ON_NOTIFY(TVN_SELCHANGED, IDC_OPTIONVARIABLES, OnSelchangedOptionvariables)
  62. ON_BN_CLICKED(IDC_BOOL_DISABLED, OnBoolDisabled)
  63. ON_BN_CLICKED(IDC_BOOL_ENABLED, OnBoolEnabled)
  64. ON_BN_CLICKED(IDC_SETVALUE, OnSetvalue)
  65. //}}AFX_MSG_MAP
  66. END_MESSAGE_MAP()
  68. /////////////////////////////////////////////////////////////////////////////
  69. // CPluginConfig message handlers
  71. BOOL CPluginConfig::OnInitDialog() 
  72. {
  73. CDialog::OnInitDialog();
  75. m_ImgList.Create(IDB_IMGLIST,16,1,RGB(0,255,0));
  77. CListCtrl *pLC=((CListCtrl *)GetDlgItem(IDC_PLUGINS));
  78. pLC->InsertColumn(0,"Plugin Name",LVCFMT_LEFT,100,-1);
  79. pLC->InsertColumn(1,"Version",LVCFMT_LEFT,48,-1);
  80. pLC->InsertColumn(2,"BO2K Ver",LVCFMT_LEFT,60,-1);
  81. pLC->InsertColumn(3,"Description",LVCFMT_LEFT,200,-1);
  82. pLC->SetImageList(&m_ImgList,LVSIL_SMALL);
  83. pLC->SetExtendedStyle(LVS_EX_FULLROWSELECT | LVS_EX_HEADERDRAGDROP);
  85. CTreeCtrl *pTC=(CTreeCtrl *)GetDlgItem(IDC_OPTIONVARIABLES);
  86. pTC->SetImageList(&m_ImgList,TVSIL_NORMAL);
  88. int i,count;
  89. CMainFrame *pFrm=(CMainFrame *)AfxGetMainWnd();
  90. count=pFrm->m_arrPluginInfo.GetSize();
  91. for(i=0;i<count;i++) {
  92. CString str;
  93. PLUGIN_INFO pi;
  95. pi=pFrm->m_arrPluginInfo[i];
  97. pLC->InsertItem(i,pi.svFilename,0);
  98. str.Format("%d.%d",pi.wVersionHi,pi.wVersionLo);
  99. pLC->SetItem(i,1,LVIF_TEXT,str,0,0,0,0);
  100. str.Format("%d.%d",pi.wBOVersionHi,pi.wBOVersionLo);
  101. pLC->SetItem(i,2,LVIF_TEXT,str,0,0,0,0);
  102. pLC->SetItem(i,3,LVIF_TEXT,pi.svDescription,0,0,0,0);
  103. }
  105. UpdateVariableList();
  107. EnableControls(FALSE);
  109. return TRUE;  // return TRUE unless you set the focus to a control
  110.               // EXCEPTION: OCX Property Pages should return FALSE
  111. }
  113. void CPluginConfig::EnableControls(BOOL bEnable)
  114. {
  115. // Deselect All List/Tree Controls
  117. ((CTreeCtrl *)GetDlgItem(IDC_OPTIONVARIABLES))->SelectItem(NULL);
  118. ((CListCtrl *)GetDlgItem(IDC_PLUGINS))->SetSelectionMark(-1);
  120. // Enable/Disable Controls
  122. GetDlgItem(IDC_STATIC_NEWVAL)->EnableWindow(FALSE);
  123. GetDlgItem(IDC_STATIC_CURVAL)->EnableWindow(FALSE);
  124. GetDlgItem(IDC_STR_NEWVALUE)->EnableWindow(FALSE);
  125. GetDlgItem(IDC_STR_CURRENTVALUE)->EnableWindow(FALSE);
  126. GetDlgItem(IDC_SETVALUE)->EnableWindow(FALSE);
  127. GetDlgItem(IDC_BOOL_ENABLED)->EnableWindow(FALSE);
  128. GetDlgItem(IDC_BOOL_DISABLED)->EnableWindow(FALSE);
  129. GetDlgItem(IDC_BOOL_GROUP)->EnableWindow(FALSE);
  130. }
  132. BOOL CPluginConfig::IsValidPlugin(LPCSTR svPath)
  133. {
  134. HMODULE hMod;
  135. BOOL bAltLoad=FALSE;
  137. hMod=LoadLibrary(svPath);
  138. if(hMod==NULL) {
  139. bAltLoad=TRUE;
  140. hMod=LoadDLL(svPath);
  141. if(hMod==NULL) return FALSE;
  142. }
  144. FARPROC proc=GetDLLProcAddress(hMod,"InstallPlugin");
  146. if(bAltLoad) FreeDLL(hMod);
  147. else FreeLibrary(hMod);
  149. return (proc==NULL)?FALSE:TRUE;
  150. }
  152. void CPluginConfig::UpdateVariableList()
  153. {
  154. CTreeCtrl *pTC=(CTreeCtrl *)GetDlgItem(IDC_OPTIONVARIABLES);
  155. pTC->DeleteAllItems();
  157. CMainFrame *pFrm=(CMainFrame *)AfxGetMainWnd();
  159. // Add options for executable
  161. pFrm->m_arrVarInfo.RemoveAll();
  163. char *base=(char *)AfxGetInstanceHandle();
  164. PIMAGE_OPTIONAL_HEADER poh=(PIMAGE_OPTIONAL_HEADER)OPTHDROFFSET(base);
  165. pFrm->AddVariables(base, poh->SizeOfImage, -1);
  167. // Add options for all plugins
  169. int i,count;
  170. count=pFrm->m_arrPluginInfo.GetSize();
  171. for(i=0;i<count;i++) {
  172. pFrm->AddVariables((char *)pFrm->m_arrPluginInfo[i].pPluginImage,pFrm->m_arrPluginInfo[i].dwPluginLen,i);
  173. }
  176. // Add option variables to tree control
  177. count=pFrm->m_arrVarInfo.GetSize();
  178. for(i=0;i<count;i++) {
  179. VARIABLE_INFO vi=pFrm->m_arrVarInfo.GetAt(i);
  180. HTREEITEM hti;
  182. // Find category item
  183. hti=pTC->GetRootItem();
  184. while(hti!=NULL) {
  185. if(pTC->GetItemText(hti).Compare(vi.svCategory)==0) break;
  186. hti=pTC->GetNextSiblingItem(hti);
  187. }
  188. if(hti==NULL) {
  189. if(vi.nPlugin==-1) {
  190. hti=pTC->InsertItem(vi.svCategory,4,5);
  191. } else {
  192. hti=pTC->InsertItem(vi.svCategory,6,7);
  193. }
  194. pTC->SetItemData(hti,0xFFFFFFFF);
  195. }
  196. int nImg;
  197. if(vi.VarType=='B') nImg=3;
  198. else if(vi.VarType=='N') nImg=2;
  199. else if(vi.VarType=='S') nImg=1;
  201. hti=pTC->InsertItem(vi.svVarName,nImg,nImg,hti);
  202. pTC->SetItemData(hti,i);
  203. }
  204. }
  206. void CPluginConfig::OnInsert() 
  207. {
  208. CMainFrame *pFrm=(CMainFrame *)AfxGetMainWnd();
  209. CFileDialog cfd(TRUE,".dll",NULL,OFN_PATHMUSTEXIST | OFN_HIDEREADONLY,"DLL Files (*.dll)|*.dll|All Files (*.*)|*.*||",this);
  210. cfd.m_ofn.lpstrTitle="Insert BO2K Plugin";
  212. if(cfd.DoModal()==IDOK) {
  213. if(IsValidPlugin(cfd.GetPathName())) {
  214. int nNum;
  215. CListCtrl *pLC=((CListCtrl *)GetDlgItem(IDC_PLUGINS));
  216. CString str;
  217. PLUGIN_INFO pi;
  219. nNum=pFrm->AddPlugin(cfd.GetPathName());
  221. pi=pFrm->m_arrPluginInfo[nNum];
  223. pLC->InsertItem(nNum,pi.svFilename,0);
  224. str.Format("%d.%d",pi.wVersionHi,pi.wVersionLo);
  225. pLC->SetItem(nNum,1,LVIF_TEXT,str,0,0,0,0);
  226. str.Format("%d.%d",pi.wBOVersionHi,pi.wBOVersionLo);
  227. pLC->SetItem(nNum,2,LVIF_TEXT,str,0,0,0,0);
  228. pLC->SetItem(nNum,3,LVIF_TEXT,pi.svDescription,0,0,0,0);
  230. UpdateVariableList();
  231. } else {
  232. AfxMessageBox("The specified file is not a valid BO2K plugin");
  233. }
  234. }
  235. }
  237. void CPluginConfig::OnRemove() 
  238. {
  239. int nNum;
  240. CMainFrame *pFrm=(CMainFrame *)AfxGetMainWnd();
  241. CListCtrl *pLC=((CListCtrl *)GetDlgItem(IDC_PLUGINS));
  243. nNum=pLC->GetSelectionMark();
  244. if(nNum<0) return;
  246. CString str;
  248. pLC->DeleteItem(nNum);
  249. nNum=pFrm->RemovePlugin(nNum);
  251. UpdateVariableList();
  252. }
  257. void CPluginConfig::OnSelchangedOptionvariables(NMHDR* pNMHDR, LRESULT* pResult) 
  258. {
  259. NMTREEVIEW *pNMTreeView = (NMTREEVIEW*)pNMHDR;
  260. CMainFrame *pFrm=(CMainFrame *)AfxGetMainWnd();
  262. // Hide all option controls
  263. GetDlgItem(IDC_STATIC_NEWVAL)->EnableWindow(FALSE);
  264. GetDlgItem(IDC_STATIC_CURVAL)->EnableWindow(FALSE);
  265. GetDlgItem(IDC_STR_NEWVALUE)->EnableWindow(FALSE);
  266. GetDlgItem(IDC_STR_CURRENTVALUE)->EnableWindow(FALSE);
  267. GetDlgItem(IDC_SETVALUE)->EnableWindow(FALSE);
  268. GetDlgItem(IDC_BOOL_ENABLED)->EnableWindow(FALSE);
  269. GetDlgItem(IDC_BOOL_DISABLED)->EnableWindow(FALSE);
  270. GetDlgItem(IDC_BOOL_GROUP)->EnableWindow(FALSE);
  272. if(pNMTreeView->itemNew.hItem!=NULL) {
  273. int nVar=pNMTreeView->itemNew.lParam;
  274. if(nVar!=0xFFFFFFFF) {
  275. if(pFrm->m_arrVarInfo[nVar].VarType=='B') {
  276. GetDlgItem(IDC_BOOL_GROUP)->EnableWindow();
  277. GetDlgItem(IDC_BOOL_ENABLED)->EnableWindow();
  278. GetDlgItem(IDC_BOOL_DISABLED)->EnableWindow();
  280. if(atoi(pFrm->m_arrVarInfo[nVar].svVarValue)) {
  281. ((CButton *)GetDlgItem(IDC_BOOL_ENABLED))->SetCheck(1);
  282. ((CButton *)GetDlgItem(IDC_BOOL_DISABLED))->SetCheck(0);
  283. } else {
  284. ((CButton *)GetDlgItem(IDC_BOOL_DISABLED))->SetCheck(1);
  285. ((CButton *)GetDlgItem(IDC_BOOL_ENABLED))->SetCheck(0);
  286. }
  288. } else if(pFrm->m_arrVarInfo[nVar].VarType=='N') {
  289. GetDlgItem(IDC_STATIC_NEWVAL)->EnableWindow();
  290. GetDlgItem(IDC_STATIC_CURVAL)->EnableWindow();
  291. GetDlgItem(IDC_STR_NEWVALUE)->EnableWindow();
  292. GetDlgItem(IDC_STR_CURRENTVALUE)->EnableWindow();
  293. GetDlgItem(IDC_SETVALUE)->EnableWindow();
  294. GetDlgItem(IDC_STR_CURRENTVALUE)->SetWindowText(pFrm->m_arrVarInfo[nVar].svVarValue);
  295. GetDlgItem(IDC_STR_NEWVALUE)->SetWindowText("");
  296. ((CEdit *)GetDlgItem(IDC_STR_CURRENTVALUE))->SetLimitText(pFrm->m_arrVarInfo[nVar].nStrLen);
  297. ((CEdit *)GetDlgItem(IDC_STR_NEWVALUE))->SetLimitText(pFrm->m_arrVarInfo[nVar].nStrLen);
  298. GetDlgItem(IDC_STR_CURRENTVALUE)->ModifyStyle(0,ES_NUMBER);
  299. GetDlgItem(IDC_STR_NEWVALUE)->ModifyStyle(0,ES_NUMBER);
  300. } else if(pFrm->m_arrVarInfo[nVar].VarType=='S') {
  301. GetDlgItem(IDC_STATIC_NEWVAL)->EnableWindow();
  302. GetDlgItem(IDC_STATIC_CURVAL)->EnableWindow();
  303. GetDlgItem(IDC_STR_NEWVALUE)->EnableWindow();
  304. GetDlgItem(IDC_STR_CURRENTVALUE)->EnableWindow();
  305. GetDlgItem(IDC_SETVALUE)->EnableWindow();
  306. GetDlgItem(IDC_STR_CURRENTVALUE)->SetWindowText(pFrm->m_arrVarInfo[nVar].svVarValue);
  307. GetDlgItem(IDC_STR_NEWVALUE)->SetWindowText("");
  308. ((CEdit *)GetDlgItem(IDC_STR_CURRENTVALUE))->SetLimitText(pFrm->m_arrVarInfo[nVar].nStrLen);
  309. ((CEdit *)GetDlgItem(IDC_STR_NEWVALUE))->SetLimitText(pFrm->m_arrVarInfo[nVar].nStrLen);
  310. GetDlgItem(IDC_STR_CURRENTVALUE)->ModifyStyle(ES_NUMBER,0);
  311. GetDlgItem(IDC_STR_NEWVALUE)->ModifyStyle(ES_NUMBER,0);
  312. }
  313. }
  314. }
  316. *pResult = 0;
  317. }
  319. void CPluginConfig::OnBoolDisabled() 
  320. {
  321. CTreeCtrl *pTC=(CTreeCtrl *)GetDlgItem(IDC_OPTIONVARIABLES);
  322. CMainFrame *pFrm=(CMainFrame *)AfxGetMainWnd();
  324. HTREEITEM hti=pTC->GetSelectedItem();
  325. int nVar=pTC->GetItemData(hti);
  326. if(nVar!=0xFFFFFFFF) {
  327. pFrm->m_arrVarInfo[nVar].svVarValue[0]='0';
  329. // Set into image
  330. VARIABLE_INFO vi=pFrm->m_arrVarInfo[nVar];
  331. char *ptr;
  332. DWORD pos;
  333. if(vi.nPlugin>=0) {
  334. PLUGIN_INFO pi=pFrm->m_arrPluginInfo[vi.nPlugin];
  335. ptr=(char *)pi.pPluginImage;
  336. } else {
  337. ptr=(char *)AfxGetInstanceHandle();
  338. }
  339. pos=vi.dwPos;
  340. memset(ptr+pos,0,vi.nStrLen+1);
  341. memcpy(ptr+pos,vi.svVarValue,lstrlen(vi.svVarValue));
  343. }
  344. }
  346. void CPluginConfig::OnBoolEnabled() 
  347. {
  348. CTreeCtrl *pTC=(CTreeCtrl *)GetDlgItem(IDC_OPTIONVARIABLES);
  349. CMainFrame *pFrm=(CMainFrame *)AfxGetMainWnd();
  351. HTREEITEM hti=pTC->GetSelectedItem();
  352. int nVar=pTC->GetItemData(hti);
  353. if(nVar!=0xFFFFFFFF) {
  354. pFrm->m_arrVarInfo[nVar].svVarValue[0]='1';
  356. // Set into image
  357. VARIABLE_INFO vi=pFrm->m_arrVarInfo[nVar];
  358. char *ptr;
  359. DWORD pos;
  360. if(vi.nPlugin>=0) {
  361. PLUGIN_INFO pi=pFrm->m_arrPluginInfo[vi.nPlugin];
  362. ptr=(char *)pi.pPluginImage;
  363. } else {
  364. ptr=(char *)AfxGetInstanceHandle();
  365. }
  366. pos=vi.dwPos;
  367. memset(ptr+pos,0,vi.nStrLen+1);
  368. memcpy(ptr+pos,vi.svVarValue,lstrlen(vi.svVarValue));
  369. }
  370. }
  372. void CPluginConfig::OnSetvalue()
  373. {
  374. CMainFrame *pFrm=(CMainFrame *)AfxGetMainWnd();
  375. CTreeCtrl *pTC=(CTreeCtrl *)GetDlgItem(IDC_OPTIONVARIABLES);
  377. HTREEITEM hti=pTC->GetSelectedItem();
  378. int nVar=pTC->GetItemData(hti);
  379. if(nVar!=0xFFFFFFFF) {
  380. CString str;
  381. GetDlgItem(IDC_STR_NEWVALUE)->GetWindowText(str);
  382. if(pFrm->m_arrVarInfo[nVar].VarType=='N') {
  383. int nValue=atoi(str);
  384. if((nValue>pFrm->m_arrVarInfo[nVar].nNumHi) ||
  385.    (nValue<pFrm->m_arrVarInfo[nVar].nNumLo)) {
  386. AfxMessageBox("Value is out of range.");
  387. return;
  388. }
  389. }
  390. lstrcpyn(pFrm->m_arrVarInfo[nVar].svVarValue,str,256);
  391. GetDlgItem(IDC_STR_CURRENTVALUE)->SetWindowText(str);
  393. // Set into image
  394. VARIABLE_INFO vi=pFrm->m_arrVarInfo[nVar];
  395. char *ptr;
  396. DWORD pos;
  397. if(vi.nPlugin>=0) {
  398. PLUGIN_INFO pi=pFrm->m_arrPluginInfo[vi.nPlugin];
  399. ptr=(char *)pi.pPluginImage;
  400. } else {
  401. ptr=(char *)AfxGetInstanceHandle();
  402. }
  403. pos=vi.dwPos;
  404. memset(ptr+pos,0,vi.nStrLen+1);
  405. memcpy(ptr+pos,vi.svVarValue,lstrlen(vi.svVarValue));
  406. }
  407. }