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

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

远程控制编程

开发平台：

WINDOWS

ServerList.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. // ServerList.cpp : implementation file
  21. //
  23. #include "stdafx.h"
  24. #include "bo2kgui.h"
  25. #include "ServerList.h"
  26. #include "NewMachineDlg.h"
  27. #include "MainFrm.h"
  28. #include "BOWDoc.h"
  29. #include "BODialogBar.h"
  30. #include "BOServerDlg.h"
  32. #define TIMER_UPDATE 1
  34. #ifdef _DEBUG
  35. #define new DEBUG_NEW
  36. #undef THIS_FILE
  37. static char THIS_FILE[] = __FILE__;
  38. #endif
  40. /////////////////////////////////////////////////////////////////////////////
  41. // CServerList dialog
  44. CServerList::CServerList(CWnd* pParent /*=NULL*/)
  45. : CGradientDialog(CServerList::IDD, pParent)
  46. {
  47. //{{AFX_DATA_INIT(CServerList)
  48. // NOTE: the ClassWizard will add member initialization here
  49. //}}AFX_DATA_INIT
  51. m_nUpdateTimer=0;
  52. }
  54. CServerList::~CServerList()
  55. {
  57. }
  59. void CServerList::CleanUpAndClose(void)
  60. {
  61. KillTimer(m_nUpdateTimer);
  62. DeleteAllItems();
  63. DestroyWindow();
  64. }
  67. void CServerList::DoDataExchange(CDataExchange* pDX)
  68. {
  69. CGradientDialog::DoDataExchange(pDX);
  70. //{{AFX_DATA_MAP(CServerList)
  71. // NOTE: the ClassWizard will add DDX and DDV calls here
  72. //}}AFX_DATA_MAP
  73. }
  77. BEGIN_MESSAGE_MAP(CServerList, CGradientDialog)
  78. //{{AFX_MSG_MAP(CServerList)
  79. ON_BN_CLICKED(ID_DETAILS, OnDetails)
  80. ON_BN_CLICKED(ID_NEWMACHINE, OnNewmachine)
  81. ON_BN_CLICKED(ID_DELETEMACHINE, OnDeletemachine)
  82. ON_WM_SIZE()
  83. ON_NOTIFY(LVN_ITEMCHANGED, ID_SERVERLIST, OnItemchangedServerlist)
  84. ON_NOTIFY(LVN_KEYDOWN, ID_SERVERLIST, OnKeydownServerlist)
  85. ON_NOTIFY(NM_DBLCLK, ID_SERVERLIST, OnDblclkServerlist)
  86. ON_BN_CLICKED(IDC_DOCKING, OnAllowDocking)
  87. ON_WM_TIMER()
  88. ON_BN_CLICKED(ID_EDITMACHINE, OnEditmachine)
  89. ON_WM_MOVE()
  90. //}}AFX_MSG_MAP
  91. END_MESSAGE_MAP()
  93. /////////////////////////////////////////////////////////////////////////////
  94. // CServerList message handlers
  97. void CServerList::OnCancel()
  98. {
  100. }
  102. void CServerList::OnOK()
  103. {
  105. }
  108. void CServerList::OnDetails() 
  109. {
  110. CListCtrl *list=(CListCtrl *)GetDlgItem(ID_SERVERLIST);
  111. CButton *details=(CButton *)GetDlgItem(ID_DETAILS);
  113. DWORD dwStyle=GetWindowLong(list->GetSafeHwnd(),GWL_STYLE);
  115. dwStyle &= ~LVS_TYPEMASK;
  116. if(details->GetCheck()) dwStyle |= LVS_REPORT;
  117. else dwStyle |= LVS_ICON;
  119. SetWindowLong(list->GetSafeHwnd(),GWL_STYLE,dwStyle);
  120. list->RedrawWindow();
  121. }
  123. void CServerList::OnNewmachine() 
  124. {
  125. CString strSvrAddr;
  126. CNewMachineDlg nmd;
  127. CListCtrl *list=(CListCtrl *)GetDlgItem(ID_SERVERLIST);
  129. if(nmd.DoModal()==IDOK) {
  130. // Add to CBOWDoc
  132. CBOWDoc *cbd;
  133. SERVER_INFO *pSvr;
  134.   cbd=(CBOWDoc *) ((CMainFrame *)theApp.GetMainWnd())->GetActiveDocument();
  136. pSvr=(SERVER_INFO *) cbd->InsertServer(nmd.m_strMachineName,nmd.m_strSelectedIO,nmd.m_strSvrAddr,nmd.m_strSelectedEnc,nmd.m_strSelectedAuth);
  137. if(pSvr==NULL) return;
  139. // Synchronize List Control
  140. SynchToDocument(cbd);
  141. }
  143. }
  145. void CServerList::OnEditmachine() 
  146. {
  147. CListCtrl *list=(CListCtrl *)GetDlgItem(ID_SERVERLIST);
  148. CNewMachineDlg nmd;
  149. CBOWDoc *cbd;
  150. SERVER_INFO *pSvr;
  151. char textbuf[512];
  153. cbd=(CBOWDoc *) ((CMainFrame *)theApp.GetMainWnd())->GetActiveDocument();
  155. int i;
  156. for(i=list->GetItemCount()-1;i>=0;i--) {
  157. if(list->GetItemState(i,LVIS_SELECTED)==LVIS_SELECTED) break;
  158. }
  159. if(i<0) return;
  162. LVITEM lvi;
  163. lvi.iItem=i;
  164. lvi.iSubItem=0;
  165. lvi.mask=LVIF_TEXT | LVIF_PARAM;
  166. lvi.pszText=textbuf;
  167. lvi.cchTextMax=512;
  168. list->GetItem(&lvi);
  169. pSvr=(SERVER_INFO *)lvi.lParam;
  171. if(pSvr->pServerDlg->m_bIsConnected) {
  172. AfxMessageBox("The server command window is currently connected.nDisconnect it first before editing the server parameters.",MB_OK|MB_ICONWARNING|MB_SETFOREGROUND|MB_TOPMOST);
  173. return;
  174. }
  176. nmd.m_strMachineName=pSvr->svName;
  177. nmd.m_strSvrAddr=pSvr->svAddress;
  178. nmd.m_strSelectedIO=pSvr->svConnectionType;
  179. nmd.m_strSelectedEnc=pSvr->svEncryption;
  180. nmd.m_strSelectedAuth=pSvr->svAuthentication;
  181. if(nmd.DoModal()==IDOK) {
  182. cbd->EditServer(pSvr,nmd.m_strMachineName,nmd.m_strSelectedIO,nmd.m_strSvrAddr,nmd.m_strSelectedEnc,nmd.m_strSelectedAuth);
  183. SynchToDocument(cbd);
  184. }
  186. }
  188. void CServerList::OnDeletemachine() 
  189. {
  190. CListCtrl *list=(CListCtrl *)GetDlgItem(ID_SERVERLIST);
  191. CBOWDoc *cbd;
  192. SERVER_INFO *pSvr;
  194. cbd=(CBOWDoc *) ((CMainFrame *)theApp.GetMainWnd())->GetActiveDocument();
  196. int i;
  197. for(i=list->GetItemCount()-1;i>=0;i--) {
  198. if(list->GetItemState(i,LVIS_SELECTED)==LVIS_SELECTED) {
  199. pSvr=(SERVER_INFO *)list->GetItemData(i);
  200. DeleteItem(i);
  201. cbd->DeleteServer(pSvr);
  202. }
  203. }
  204. }
  206. void CServerList::DeleteItem(int idx)
  207. {
  208. CListCtrl *list=(CListCtrl *)GetDlgItem(ID_SERVERLIST);
  210. SERVER_INFO *pSvr;
  211. pSvr=(SERVER_INFO *) list->GetItemData(idx);
  213. pSvr->pServerDlgBDB->Hide();
  215. pSvr->pServerDlg->DestroyWindow();
  216. delete pSvr->pServerDlg;
  217. pSvr->pServerDlg=NULL;
  219. pSvr->pServerDlgBDB->DestroyWindow();
  220. delete pSvr->pServerDlgBDB;
  221. pSvr->pServerDlgBDB=NULL;
  223. list->DeleteItem(idx);
  224. }
  226. void CServerList::DeleteAllItems()
  227. {
  228. CListCtrl *list=(CListCtrl *)GetDlgItem(ID_SERVERLIST);
  229. int i;
  230. if(list!=NULL) {
  231. for(i=list->GetItemCount()-1;i>=0;i--) {
  232. DeleteItem(i);
  233. }
  234. }
  235. }
  237. void CServerList::OnSize(UINT nType, int cx, int cy) 
  238. {
  239. CGradientDialog::OnSize(nType, cx, cy);
  241. CListCtrl *list=(CListCtrl *)GetDlgItem(ID_SERVERLIST);
  242. if(list) {
  243. RECT r;
  244. int newsize;
  245. int i;
  247. // Move window
  248. GetClientRect(&r);
  249. list->MoveWindow(36,26,(r.right-r.left)-40,(r.bottom-r.top)-30);
  251. // Resize columns
  252. list->GetClientRect(&r);
  253. newsize=r.right-r.left;
  255. for(i=0;i<5;i++) {
  256. int newwidth=newsize/5;
  257. if(newwidth<100) newwidth=100;
  258. list->SetColumnWidth(i,newwidth);
  259. }
  260. }
  261. }
  263. BOOL CServerList::OnInitDialog() 
  264. {
  265. CGradientDialog::OnInitDialog();
  267. CListCtrl *list=(CListCtrl *)GetDlgItem(ID_SERVERLIST);
  269. m_NormalImageList.Create(IDB_NORMAL_BOCOMPS,32,1,RGB(255,0,0));
  270. list->SetImageList(&m_NormalImageList,LVSIL_NORMAL);
  272. m_SmallImageList.Create(IDB_LITTLE_BOCOMPS,16,1,RGB(255,0,0));
  273. list->SetImageList(&m_SmallImageList,LVSIL_SMALL);
  275. // Size columns
  276. RECT r;
  277. list->GetClientRect(&r);
  278. int newwidth=(r.right-r.left)/5;
  279. if(newwidth<100) newwidth=100;
  281. list->InsertColumn(0,"Machine",LVCFMT_LEFT,newwidth,0);
  282. list->InsertColumn(1,"Address",LVCFMT_LEFT,newwidth,1);
  283. list->InsertColumn(2,"Connection",LVCFMT_LEFT,newwidth,2);
  284. list->InsertColumn(3,"Encryption",LVCFMT_LEFT,newwidth,3);
  285. list->InsertColumn(4,"Authentication",LVCFMT_LEFT,newwidth,3);
  286. list->SetExtendedStyle(LVS_EX_FULLROWSELECT | LVS_EX_HEADERDRAGDROP);
  288. CButton *nmb=(CButton *)GetDlgItem(ID_NEWMACHINE);
  289. nmb->SetIcon(theApp.LoadIcon(IDI_NEWMACHINE));
  291. CButton *dmb=(CButton *)GetDlgItem(ID_DELETEMACHINE);
  292. dmb->SetIcon(theApp.LoadIcon(IDI_DELETEMACHINE));
  293. dmb->EnableWindow(FALSE);
  295. CButton *emb=(CButton *)GetDlgItem(ID_EDITMACHINE);
  296. emb->SetIcon(theApp.LoadIcon(IDI_EDITMACHINE));
  297. emb->EnableWindow(FALSE);
  299. CButton *dok=(CButton *)GetDlgItem(IDC_DOCKING);
  300. dok->SetCheck(1);
  302. CButton *ddt=(CButton *)GetDlgItem(ID_DETAILS);
  303. ddt->SetCheck(1);
  305. m_nUpdateTimer=SetTimer(TIMER_UPDATE,500,NULL);
  306. if(m_nUpdateTimer==0) return FALSE;
  308. return TRUE;  // return TRUE unless you set the focus to a control
  309.               // EXCEPTION: OCX Property Pages should return FALSE
  310. }
  312. void CServerList::OnItemchangedServerlist(NMHDR* pNMHDR, LRESULT* pResult) 
  313. {
  314. NM_LISTVIEW* pNMListView = (NM_LISTVIEW*)pNMHDR;
  316. CListCtrl *list=(CListCtrl *)GetDlgItem(ID_SERVERLIST);
  318. CButton *del=(CButton *)GetDlgItem(ID_DELETEMACHINE);
  319. if(list->GetSelectedCount()==0) del->EnableWindow(FALSE);
  320. else del->EnableWindow(TRUE);
  322. CButton *edt=(CButton *)GetDlgItem(ID_EDITMACHINE);
  323. if(list->GetSelectedCount()==1) edt->EnableWindow(TRUE);
  324. else edt->EnableWindow(FALSE);
  326. *pResult = 0;
  327. }
  330. void CServerList::OnKeydownServerlist(NMHDR* pNMHDR, LRESULT* pResult) 
  331. {
  332. NMLVKEYDOWN* pLVKeyDown = (NMLVKEYDOWN*)pNMHDR;
  333. CButton *butt;
  335. switch(pLVKeyDown->wVKey) {
  336. case VK_DELETE:
  337. butt=(CButton *)GetDlgItem(ID_DELETEMACHINE);
  338. if(butt->IsWindowEnabled()) OnDeletemachine();
  339. break;
  340. case VK_INSERT:
  341. OnNewmachine();
  342. break;
  343. }
  345. *pResult = 0;
  346. }
  348. void CServerList::OnDblclkServerlist(NMHDR* pNMHDR, LRESULT* pResult) 
  349. {
  350. CListCtrl *list=(CListCtrl *)GetDlgItem(pNMHDR->idFrom);
  352. int i;
  353. if(list->GetSelectedCount()>0) {
  354. for(i=(list->GetItemCount()-1);i>=0;i--) {
  355. if(list->GetItemState(i,LVIS_SELECTED)==LVIS_SELECTED) {
  356. LVITEM lvi;
  357. SERVER_INFO *si;
  358. char svText[512];
  360. // Get item information
  361. lvi.iItem=i;
  362. lvi.iSubItem=0;
  363. lvi.mask=LVIF_PARAM | LVIF_IMAGE | LVIF_TEXT;
  364. lvi.pszText=svText;
  365. lvi.cchTextMax=512;
  366. list->GetItem(&lvi);
  368. si=(SERVER_INFO *) lvi.lParam;
  369. if(si->pServerDlgBDB->IsWindowVisible()) {
  370. si->pServerDlgBDB->Hide();
  371. } else {
  372. si->pServerDlgBDB->Show();
  373. }
  375. }
  376. }
  377. }
  379. *pResult = 0;
  380. }
  382. void CServerList::OnAllowDocking() 
  383. {
  384. CButton *btn;
  385. btn=(CButton *)GetDlgItem(IDC_DOCKING);
  386. if(btn->GetCheck()) {
  387. ((CBODialogBar *)GetParent())->EnableDocking(CBRS_ALIGN_ANY);
  388. } else {
  389. ((CBODialogBar *)GetParent())->EnableDocking(0);
  390. ((CBODialogBar *)GetParent())->m_pDockContext->ToggleDocking();
  391. }
  392. }
  394. void CServerList::SynchToDocument(CBOWDoc *cbd)
  395. {
  396. SERVER_INFO *si;
  397. CListCtrl *pList;
  398. int idx;
  400. pList=(CListCtrl *) GetDlgItem(ID_SERVERLIST);
  402. // Add items that should be there and update those that are already there
  403. si=cbd->m_pHeadSI;
  404. while(si!=NULL) {
  405. BOOL bNew;
  406. LVITEM lvi;
  407. LVFINDINFO lvfi;
  408. lvfi.flags=LVFI_PARAM;
  409. lvfi.lParam=(LPARAM) si;
  411. idx=pList->FindItem(&lvfi,-1);
  412. if(idx==-1) {
  413. idx=0;
  414. lvi.mask=LVIF_TEXT | LVIF_IMAGE | LVIF_PARAM;
  415. lvi.pszText=si->svName;
  416. lvi.iImage=0;
  417. lvi.iItem=idx;
  418. lvi.iSubItem=0;
  419. lvi.lParam=(LPARAM) si;
  420. pList->InsertItem(&lvi);
  421. bNew=TRUE;
  422. } else {
  423. lvi.mask=LVIF_TEXT;
  424. lvi.pszText=si->svName;
  425. lvi.iItem=idx;
  426. lvi.iSubItem=0;
  428. pList->SetItem(&lvi);
  429. bNew=FALSE;
  430. }
  432. lvi.mask=LVIF_TEXT;
  433. lvi.pszText=si->svAddress;
  434. lvi.iItem=idx;
  435. lvi.iSubItem=1;
  436. pList->SetItem(&lvi);
  438. lvi.mask=LVIF_TEXT;
  439. lvi.pszText=si->svConnectionType;
  440. lvi.iItem=idx;
  441. lvi.iSubItem=2;
  442. pList->SetItem(&lvi);
  444. lvi.mask=LVIF_TEXT;
  445. lvi.pszText=si->svEncryption;
  446. lvi.iItem=idx;
  447. lvi.iSubItem=3;
  448. pList->SetItem(&lvi);
  450. lvi.mask=LVIF_TEXT;
  451. lvi.pszText=si->svAuthentication;
  452. lvi.iItem=idx;
  453. lvi.iSubItem=4;
  454. pList->SetItem(&lvi);
  457. // Create cool dialog bar
  458. if(bNew) {
  459. CSize sizeNormal(422,399);
  460. CString title("Server Command Client");
  461. si->pServerDlgBDB=new CBODialogBar;
  462. si->pServerDlgBDB->SetNormalSize(sizeNormal);
  463. si->pServerDlgBDB->Create(this->GetParentFrame(),IDD_BOSERVERDLG,title,sizeNormal);
  464. si->pServerDlgBDB->SetBarStyle((si->pServerDlgBDB->GetBarStyle() & ~CBRS_ALIGN_ANY) | CBRS_TOOLTIPS | CBRS_BORDER_ANY | CBRS_FLYBY | CBRS_SIZE_DYNAMIC);
  465. si->pServerDlgBDB->EnableDockingOnSizeBar(CBRS_ALIGN_ANY);
  467. si->pServerDlg=new CBOServerDlg();
  468. si->pServerDlg->SetServerInfo(si);
  469. si->pServerDlg->Create(CBOServerDlg::IDD,si->pServerDlgBDB);
  470. si->pServerDlg->UseGradient(m_bGradient);
  472. si->pServerDlgBDB->SetDialog(si->pServerDlg);
  474. si->pServerDlgBDB->GetDockingFrame()->FloatControlBar(si->pServerDlgBDB, CPoint((GetSystemMetrics(SM_CXSCREEN)-422)/2,(GetSystemMetrics(SM_CYSCREEN)-399)/2));
  475. //si->pServerDlgBDB->Hide();
  476. }
  477. else {
  478. si->pServerDlg->SetServerInfo(si);
  479. si->pServerDlg->UpdateDialog();
  480. si->pServerDlg->Invalidate();
  481. si->pServerDlg->UpdateWindow();
  482. }
  484. si=si->next;
  485. }
  486. }
  488. void CServerList::OnTimer(UINT nIDEvent) 
  489. {
  490. if(nIDEvent==TIMER_UPDATE) {
  491. UpdateListStatus();
  492. }
  494. CGradientDialog::OnTimer(nIDEvent);
  495. }
  497. void CServerList::UpdateListStatus(void)
  498. {
  499. CListCtrl *list;
  501. list=(CListCtrl *)GetDlgItem(ID_SERVERLIST);
  503. int i;
  504. for(i=(list->GetItemCount()-1);i>=0;i--) {
  505. LVITEM lvi;
  506. SERVER_INFO *si;
  507. char svText[512];
  509. // Get item information
  510. lvi.iItem=i;
  511. lvi.iSubItem=0;
  512. lvi.mask=LVIF_PARAM | LVIF_IMAGE | LVIF_TEXT;
  513. lvi.pszText=svText;
  514. lvi.cchTextMax=512;
  515. list->GetItem(&lvi);
  517. si=(SERVER_INFO *) lvi.lParam;
  518. lvi.mask=LVIF_IMAGE;
  519. if(!si->pServerDlgBDB->IsFloating()) {
  520. if(si->pServerDlgBDB->m_hWnd!=NULL) {
  521. if(si->pServerDlgBDB->IsWindowVisible()) lvi.iImage=1;
  522. else lvi.iImage=0;
  523. }
  524. } else {
  525. if(si->pServerDlg->m_hWnd!=NULL) {
  526. if(si->pServerDlg->GetParent()->GetParent()->GetParent()->IsWindowVisible()) lvi.iImage=1;
  527. else lvi.iImage=0;
  528. }
  529. }
  530. list->SetItem(&lvi);
  531. }
  532. }
  534. void CServerList::UseGradient(BOOL bUse) 
  535. {
  536. CListCtrl *list;
  537. list=(CListCtrl *)GetDlgItem(ID_SERVERLIST);
  539. int i;
  540. for(i=(list->GetItemCount()-1);i>=0;i--) {
  541. LVITEM lvi;
  542. SERVER_INFO *si;
  543. char svText[512];
  545. // Get item information
  546. lvi.iItem=i;
  547. lvi.iSubItem=0;
  548. lvi.mask=LVIF_PARAM | LVIF_IMAGE | LVIF_TEXT;
  549. lvi.pszText=svText;
  550. lvi.cchTextMax=512;
  551. list->GetItem(&lvi);
  553. si=(SERVER_INFO *) lvi.lParam;
  555. si->pServerDlg->UseGradient(bUse);
  556. si->pServerDlg->Invalidate();
  557. si->pServerDlg->UpdateWindow();
  558. }
  560. if(bUse)
  561. SetGradient(GetSysColor(COLOR_3DFACE),GetSysColor(COLOR_3DHILIGHT),2);
  562. else
  563. SetGradient();
  565. Invalidate();
  566. UpdateWindow();
  567. }