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

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

远程控制编程

开发平台：

WINDOWS

BOServerDlg.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. // BOServerDlg.cpp : implementation file
  21. //
  23. #include "stdafx.h"
  24. #include "bo2kgui.h"
  25. #include "BOServerDlg.h"
  26. #include "BODialogBar.h"
  27. #include "MainFrm.h"
  28. #include "QueryDlg.h"
  29. #include "NewMachineDlg.h"
  30. #include "resource.h"
  31. #include <auth.h>
  32. #include <iohandler.h>
  33. #include <commnet.h>
  34. #include <comm_native.h>
  35. #include <strhandle.h>
  37. #ifdef _DEBUG
  38. #define new DEBUG_NEW
  39. #undef THIS_FILE
  40. static char THIS_FILE[] = __FILE__;
  41. #endif
  43. #define HISTORY_LENGTH 16
  45. int g_nBOServerColor=0;
  47. /////////////////////////////////////////////////////////////////////////////
  48. // CBOServerDlg dialog
  51. CBOServerDlg::CBOServerDlg(CWnd* pParent /*=NULL*/)
  52. : CGradientDialog(CBOServerDlg::IDD, pParent)
  53. {
  54. //{{AFX_DATA_INIT(CBOServerDlg)
  55. // NOTE: the ClassWizard will add member initialization here
  56. //}}AFX_DATA_INIT
  57. m_bGrad=FALSE;
  58. m_bGenerated=FALSE;
  59. m_bIsConnected=FALSE;
  60. m_pAuthSocket=NULL;
  61. m_bHandleNetwork=TRUE;
  62. m_cwtListenThread=NULL;
  63. InitializeCriticalSection(&m_NetCritSec);
  64. }
  66. CBOServerDlg::~CBOServerDlg()
  67. {
  68. DeleteCriticalSection(&m_NetCritSec);
  69. if(m_cwtListenThread!=NULL) delete m_cwtListenThread;
  71. }
  74. void CBOServerDlg::DoDataExchange(CDataExchange* pDX)
  75. {
  76. CGradientDialog::DoDataExchange(pDX);
  77. //{{AFX_DATA_MAP(CBOServerDlg)
  78. // NOTE: the ClassWizard will add DDX and DDV calls here
  79. //}}AFX_DATA_MAP
  80. }
  83. BEGIN_MESSAGE_MAP(CBOServerDlg, CGradientDialog)
  84. //{{AFX_MSG_MAP(CBOServerDlg)
  85. ON_BN_CLICKED(IDC_DOCKING, OnAllowDocking)
  86. ON_BN_CLICKED(IDC_CONNECTCHECK, OnConnectcheck)
  87. ON_NOTIFY(TVN_SELCHANGED, IDC_SERVERCOMMANDS, OnSelchangedServercommands)
  88. ON_BN_CLICKED(IDC_SENDCOMMAND, OnSendcommand)
  89. ON_BN_CLICKED(IDC_CLEARRESPONSES, OnClearResponses)
  90. ON_WM_SIZE()
  91. ON_WM_SHOWWINDOW()
  92. ON_BN_CLICKED(IDC_SETTINGS, OnSettings)
  93. //}}AFX_MSG_MAP
  94. END_MESSAGE_MAP()
  96. /////////////////////////////////////////////////////////////////////////////
  97. // CBOServerDlg message handlers
  99. void CBOServerDlg::OnOK()
  100. {
  102. }
  104. void CBOServerDlg::OnCancel()
  105. {
  107. }
  109. void CBOServerDlg::OnAllowDocking() 
  110. {
  111. CButton *btn;
  112. btn=(CButton *)GetDlgItem(IDC_DOCKING);
  113. if(btn->GetCheck()) {
  114. ((CBODialogBar *)GetParent())->EnableDocking(CBRS_ALIGN_ANY);
  115. } else {
  116. ((CBODialogBar *)GetParent())->EnableDocking(0);
  117. ((CBODialogBar *)GetParent())->m_pDockContext->ToggleDocking();
  118. }
  119. }
  121. BOOL CBOServerDlg::OnInitDialog() 
  122. {
  123. CGradientDialog::OnInitDialog();
  125. m_HistFont.CreatePointFont(86,"Courier New",NULL);
  126. // Subclass dialog items
  127. m_HistEdit.SubclassDlgItem(IDC_SERVERRESPONSE,this);
  128. m_HistEdit.AllowSelection(TRUE);
  129. m_HistEdit.SetFont(&m_HistFont,TRUE);
  131. // Initalize command list
  132. m_CmdDescList.SetNativeCommands();
  134. CButton *dok=(CButton *)GetDlgItem(IDC_DOCKING);
  135. dok->SetCheck(0);
  136. ((CBODialogBar *)GetParent())->EnableDocking(0);
  139. m_TreeImageList.Create(IDB_CMDTREE_IMAGELIST,16,8,RGB(0,255,0));
  141. CTreeCtrl *pTree=(CTreeCtrl *)GetDlgItem(IDC_SERVERCOMMANDS);
  142. pTree->SetImageList(&m_TreeImageList,TVSIL_NORMAL);
  144. m_CmdDescList.FillTreeCtrl(pTree);
  146. UpdateDialog();
  148. // Set Checks
  150. CButton *cqc=(CButton *)GetDlgItem(IDC_QUERYONCONNECT);
  151. cqc->SetCheck(1);
  153. // Set icons
  154. CButton *csb=(CButton *)GetDlgItem(IDC_SETTINGS);
  155. csb->SetIcon(AfxGetApp()->LoadIcon(IDI_EDITMACHINE));
  157. return TRUE;  // return TRUE unless you set the focus to a control
  158.               // EXCEPTION: OCX Property Pages should return FALSE
  159. }
  161. BOOL CBOServerDlg::DestroyWindow() 
  162. {
  163. m_bIsConnected=FALSE;
  164. if(m_cwtListenThread!=NULL) {
  165. WaitForSingleObject(m_cwtListenThread->m_hThread,INFINITE);
  166. delete m_cwtListenThread;
  167. m_cwtListenThread=NULL;
  168. }
  170. if(m_pAuthSocket) {
  171. if(m_pAuthSocket->Close()==-1) {
  172. AfxMessageBox("Could not close IOSocket.nTerminating Anyway.n");
  173. }
  174. delete m_pAuthSocket;
  175. }
  177. return CGradientDialog::DestroyWindow();
  178. }
  180. void CBOServerDlg::UpdateDialog(void)
  181. {
  182. CString txt;
  183. txt+="  Name: ";
  184. txt+=m_ServerInfo.svName;
  185. txt+="rn";
  186. txt+="  Addr: ";
  187. txt+=m_ServerInfo.svAddress;
  189. GetDlgItem(IDC_MACHINEINFO)->SetWindowText(txt);
  190. }
  192. void CBOServerDlg::UseGradient(BOOL bGrad)
  193. {
  194. m_bGrad=bGrad;
  195. if(bGrad) {
  196. if(!m_bGenerated) {
  197. COLORREF dwFace=GetSysColor(COLOR_3DFACE);
  198. COLORREF dwLight=GetSysColor(COLOR_3DHILIGHT);
  199. COLORREF dwShadow=GetSysColor(COLOR_3DSHADOW);
  200. COLORREF dwFadeFace=dwFace;
  201. COLORREF dwFadeLight=RGB(255-((255-GetRValue(dwLight))/2),
  202. 255-((255-GetGValue(dwLight))/2),
  203. 255-((255-GetBValue(dwLight))/2));
  204. COLORREF dwFadeShadow=RGB(GetRValue(dwShadow)/2,
  205. GetGValue(dwShadow)/2,
  206. GetBValue(dwShadow)/2);
  209. int nOrient;
  210. if(g_nBOServerColor==0)      { m_c1=dwFace; m_c2=dwLight; nOrient=1; }
  211. else if(g_nBOServerColor==1) { m_c1=dwShadow; m_c2=dwFace; nOrient=1; }
  212. else if(g_nBOServerColor==2) { m_c1=dwFadeFace; m_c2=dwFadeLight; nOrient=1;}
  213. else { m_c1=dwFadeShadow; m_c2=dwFadeFace; nOrient=1; }
  214. g_nBOServerColor++;
  215. if(g_nBOServerColor>3) g_nBOServerColor=0;
  216. SetGradient(m_c1,m_c2,nOrient);
  217. m_bGenerated=TRUE;
  218. }
  219. else m_bGradient=TRUE;
  220. }
  221. else m_bGradient=FALSE;
  222. }
  225. void CBOServerDlg::OnConnectcheck() 
  226. {
  227. EnterCriticalSection(&m_NetCritSec);
  229. CButton *pbCC=(CButton *)GetDlgItem(IDC_CONNECTCHECK);
  230. CButton *pbSC=(CButton *)GetDlgItem(IDC_SENDCOMMAND);
  231. if(pbCC->GetCheck()) {
  232. pbCC->EnableWindow(FALSE);
  233. pbCC->SetWindowText("Connecting...");
  234. // Create socket and connect
  236. m_pAuthSocket=ConnectAuthSocket(NULL,0,GetSafeHwnd(),
  237. m_ServerInfo.svAddress,
  238. m_ServerInfo.svConnectionType,
  239. m_ServerInfo.svEncryption,
  240. m_ServerInfo.svAuthentication);
  241. if(m_pAuthSocket==NULL) {
  242. pbCC->SetWindowText("Connect");
  243. pbCC->SetCheck(0);
  244. pbCC->EnableWindow();
  245. LeaveCriticalSection(&m_NetCritSec);
  246. return;
  247. }
  248. if(m_pAuthSocket==(CAuthSocket *)0xFFFFFFFF) {
  249. m_pAuthSocket=NULL;
  250. pbCC->SetWindowText("Connect");
  251. pbCC->SetCheck(0);
  252. pbCC->EnableWindow();
  253. MessageBox("Could not connect to remote machine.n","Connection Error",MB_ICONWARNING|MB_OK|MB_TOPMOST|MB_SETFOREGROUND);
  254. LeaveCriticalSection(&m_NetCritSec);
  255. return;
  256. }
  258. // Create socket listener thread to handle asynchronous receives.
  259. m_bIsConnected=TRUE;
  260. m_cwtListenThread=AfxBeginThread(ServerListenThread,this);
  261. m_cwtListenThread->m_bAutoDelete=FALSE;
  262. m_cwtListenThread->SetThreadPriority(THREAD_PRIORITY_LOWEST);
  264. if(m_cwtListenThread==NULL) {
  265. m_pAuthSocket->Close();
  266. m_bIsConnected=FALSE;
  267. delete m_pAuthSocket;
  268. m_pAuthSocket=NULL;
  269. pbCC->SetWindowText("Connect");
  270. pbCC->SetCheck(0);
  271. pbCC->EnableWindow();
  272. MessageBox("Could not create thread.nYour system may be misbehaving.n","System Error",MB_ICONWARNING|MB_OK|MB_TOPMOST|MB_SETFOREGROUND);
  273. LeaveCriticalSection(&m_NetCritSec);
  274. return;
  275. }
  277. // Report that we are happy
  278. pbCC->SetWindowText("Disconnect");
  279. pbCC->EnableWindow();
  280. pbSC->EnableWindow();
  282. // Check for 'Auto-Query'
  283. CButton *cqc=(CButton *)GetDlgItem(IDC_QUERYONCONNECT);
  284. if(cqc->GetCheck()==1) {
  285. if(PerformQuery()<0) {
  286. MessageBox("Error sending command. Connection to remote machine may be broken.","Network Poll Error",MB_ICONSTOP|MB_OK|MB_SETFOREGROUND|MB_TOPMOST);
  287. }
  288. }
  289. } else {
  290. pbCC->EnableWindow(FALSE);
  291. pbCC->SetWindowText("Disconnecting...");
  293. // Close listening thread
  294. m_bIsConnected=FALSE;
  295. if(m_cwtListenThread!=NULL) {
  296. LeaveCriticalSection(&m_NetCritSec);
  297. WaitForSingleObject(m_cwtListenThread->m_hThread,INFINITE);
  298. EnterCriticalSection(&m_NetCritSec);
  299. delete m_cwtListenThread;
  300. m_cwtListenThread=NULL;
  301. }
  303. // Disconnect socket and destroy
  304. if(m_pAuthSocket->Close()==-1) {
  305. pbCC->SetWindowText("Disconnect");
  306. pbCC->SetCheck(1);
  307. pbCC->EnableWindow();
  308. MessageBox("Could not close connection.nYour network may be misbehaving, ornthe server has become unresponsive.n","Network Error",MB_ICONWARNING|MB_OK|MB_TOPMOST|MB_SETFOREGROUND);
  309. LeaveCriticalSection(&m_NetCritSec);
  310. return;
  311. }
  313. delete m_pAuthSocket;
  314. m_pAuthSocket=NULL;
  316. pbCC->SetWindowText("Connect");
  317. pbCC->EnableWindow();
  318. pbSC->EnableWindow(FALSE);
  319. }
  320. LeaveCriticalSection(&m_NetCritSec);
  321. }
  324. void CBOServerDlg::OnSelchangedServercommands(NMHDR* pNMHDR, LRESULT* pResult) 
  325. {
  326. NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR;
  328. int command;
  330. command=pNMTreeView->itemNew.lParam;
  331. if(command==-1) {
  332. GetDlgItem(IDC_STATIC_OPT1)->ShowWindow(SW_HIDE);
  333. GetDlgItem(IDC_STATIC_OPT2)->ShowWindow(SW_HIDE);
  334. GetDlgItem(IDC_STATIC_OPT3)->ShowWindow(SW_HIDE);
  335. GetDlgItem(IDC_OPT1)->ShowWindow(SW_HIDE);
  336. GetDlgItem(IDC_OPT2)->ShowWindow(SW_HIDE);
  337. GetDlgItem(IDC_OPT3)->ShowWindow(SW_HIDE);
  338. }
  339. else {
  340. char *arg;
  341. arg=m_CmdDescList.GetArgDesc1(command);
  342. if(arg) {
  343. GetDlgItem(IDC_STATIC_OPT1)->ShowWindow(SW_HIDE);
  344. GetDlgItem(IDC_STATIC_OPT1)->SetWindowText(arg);
  345. GetDlgItem(IDC_STATIC_OPT1)->ShowWindow(SW_SHOW);
  346. GetDlgItem(IDC_OPT1)->ShowWindow(SW_SHOW);
  347. } else {
  348. GetDlgItem(IDC_STATIC_OPT1)->ShowWindow(SW_HIDE);
  349. GetDlgItem(IDC_OPT1)->ShowWindow(SW_HIDE);
  350. }
  352. arg=m_CmdDescList.GetArgDesc2(command);
  353. if(arg) {
  354. GetDlgItem(IDC_STATIC_OPT2)->ShowWindow(SW_HIDE);
  355. GetDlgItem(IDC_STATIC_OPT2)->SetWindowText(arg);
  356. GetDlgItem(IDC_STATIC_OPT2)->ShowWindow(SW_SHOW);
  357. GetDlgItem(IDC_OPT2)->ShowWindow(SW_SHOW);
  358. } else {
  359. GetDlgItem(IDC_STATIC_OPT2)->ShowWindow(SW_HIDE);
  360. GetDlgItem(IDC_OPT2)->ShowWindow(SW_HIDE);
  361. }
  363. arg=m_CmdDescList.GetArgDesc3(command);
  364. if(arg) {
  365. GetDlgItem(IDC_STATIC_OPT3)->ShowWindow(SW_HIDE);
  366. GetDlgItem(IDC_STATIC_OPT3)->SetWindowText(arg);
  367. GetDlgItem(IDC_STATIC_OPT3)->ShowWindow(SW_SHOW);
  368. GetDlgItem(IDC_OPT3)->ShowWindow(SW_SHOW);
  369. } else {
  370. GetDlgItem(IDC_STATIC_OPT3)->ShowWindow(SW_HIDE);
  371. GetDlgItem(IDC_OPT3)->ShowWindow(SW_HIDE);
  372. }
  373. }
  375. *pResult = 0;
  376. }
  378. void CBOServerDlg::SetServerInfo(SERVER_INFO *si)
  379. {
  380. m_ServerInfo=*si;
  381. }
  383. void CBOServerDlg::OnSendcommand() 
  384. {
  385. EnterCriticalSection(&m_NetCritSec);
  386. if(m_bIsConnected) {
  387. CTreeCtrl *pTree;
  388. int nCommand,ret;
  389. HTREEITEM hti;
  390. pTree=(CTreeCtrl *)GetDlgItem(IDC_SERVERCOMMANDS);
  391. if(pTree==NULL) {
  392. LeaveCriticalSection(&m_NetCritSec);
  393. return;
  394. }
  396. hti=pTree->GetSelectedItem();
  397. if(hti==NULL) {
  398. AfxMessageBox("You need to select an appropriate command.n",MB_OK|MB_ICONWARNING|MB_TOPMOST|MB_SETFOREGROUND);
  399. LeaveCriticalSection(&m_NetCritSec);
  400. return;
  401. }
  402. nCommand=pTree->GetItemData(hti);
  403. if(nCommand==-1) {
  404. AfxMessageBox("You need to select an appropriate command.n",MB_OK|MB_ICONWARNING|MB_TOPMOST|MB_SETFOREGROUND);
  405. LeaveCriticalSection(&m_NetCritSec);
  406. return;
  407. }
  410. CEdit *pEBArg1;
  411. CComboBox *pCBArg2,*pCBArg3;
  412. int nArg1;
  413. char svArg1[1024],svArg2[1024],svArg3[1024];
  415. pEBArg1=(CEdit *)GetDlgItem(IDC_OPT1);
  416. pCBArg2=(CComboBox *)GetDlgItem(IDC_OPT2);
  417. pCBArg3=(CComboBox *)GetDlgItem(IDC_OPT3);
  419. // Store entries into list
  420. if(pEBArg1->IsWindowVisible() && pEBArg1->GetWindowTextLength()!=0) {
  421. pEBArg1->GetWindowText(svArg1,1024);
  422. nArg1=atoi(svArg1);
  423. }
  424. else nArg1=0;
  426. if(pCBArg2->IsWindowVisible() && pCBArg2->GetWindowTextLength()!=0) {
  427. if(pCBArg2->GetCount()==HISTORY_LENGTH) {
  428. pCBArg2->DeleteString(HISTORY_LENGTH-1);
  429. }
  430. pCBArg2->GetWindowText(svArg2,1024);
  431. pCBArg2->InsertString(0,svArg2);
  432. }
  433. else svArg2[0]='';
  435. if(pCBArg3->IsWindowVisible()) {
  436. if(pCBArg3->GetCount()==HISTORY_LENGTH) {
  437. pCBArg3->DeleteString(HISTORY_LENGTH-1);
  438. }
  439. pCBArg3->GetWindowText(svArg3,1024);
  440. pCBArg3->InsertString(0,svArg3);
  441. }
  442. else svArg3[0]='';
  444. // Send command
  446. switch(nCommand) {
  447. case BO_PING:
  448. ret=IssueAuthCommandRequest(m_pAuthSocket,BO_PING,GetTickCount(),0,m_ServerInfo.svAddress,NULL);
  449. break;
  450. case BO_QUERY:
  451. ret=PerformQuery();
  452. break;
  453. default:
  454. ret=IssueAuthCommandRequest(m_pAuthSocket,nCommand,GetTickCount(),nArg1,svArg2,svArg3);
  455. break;
  456. }
  457. if(ret<0) {
  458. AfxMessageBox("Error sending command. Connection to remote machine may be broken.",MB_OK|MB_ICONWARNING|MB_SETFOREGROUND|MB_TOPMOST);
  459. }
  460. }
  461. LeaveCriticalSection(&m_NetCritSec);
  462. }
  464. void CBOServerDlg::OnClearResponses() 
  465. {
  466. // Get edit control to display response
  467. CHistoryEdit *pEdt=(CHistoryEdit *)GetDlgItem(IDC_SERVERRESPONSE);
  468. if(pEdt==NULL) return;
  470. pEdt->SetWindowText("");
  471. }
  473. UINT ServerListenThread(LPVOID pParam)
  474. {
  475. CBOServerDlg *pThis=(CBOServerDlg *)pParam;
  476. while(pThis->m_bIsConnected) {
  477. pThis->HandleNetworkPoll();
  478. Sleep(20);
  479. }
  481. return 0;
  482. }
  484. void CBOServerDlg::HandleNetworkPoll(void)
  485. {
  486. BYTE *pBuffer;
  487. int nBufLen,ret;
  489. // Check connected socket
  490. do {
  491. EnterCriticalSection(&m_NetCritSec);
  492. ret=m_pAuthSocket->Recv(&pBuffer,&nBufLen);
  493. LeaveCriticalSection(&m_NetCritSec);
  494. if(ret<0) {
  495. CButton *pbCC=(CButton *)GetDlgItem(IDC_CONNECTCHECK);
  496. pbCC->PostMessage(BM_CLICK);
  497. }
  498. if(ret>0) {
  499. // Break down command into response
  500. int cmdlen,command,comid,nArg1;
  501. char *svArg2,*svArg3;
  503. BreakDownCommand(pBuffer,&cmdlen,&command,&comid,&nArg1,&svArg2,&svArg3);
  504. if(svArg2!=NULL) {
  505. HandleCommandResponse(svArg2);
  506. }
  507. if(nArg1==2) {
  508. CButton *cqc=(CButton *)GetDlgItem(IDC_QUERYONCONNECT);
  509. if(cqc->GetCheck()==1) {
  510. EnterCriticalSection(&m_NetCritSec);
  511. if(PerformQuery()<0) {
  512. MessageBox("Error sending command. Connection to remote machine may be broken.","Network Poll Error",MB_ICONSTOP|MB_OK|MB_SETFOREGROUND|MB_TOPMOST);
  513. }
  514. LeaveCriticalSection(&m_NetCritSec);
  515. }
  516. }
  517. m_pAuthSocket->Free(pBuffer);
  518. }
  519. } while(ret>0);
  521. }
  523. void CBOServerDlg::HandleCommandResponse(char *svArg2)
  524. {
  525. // Get edit control to display response
  526. CHistoryEdit *pEdt=(CHistoryEdit *)GetDlgItem(IDC_SERVERRESPONSE);
  527. if(pEdt==NULL) return;
  529. // Get number of lines originally
  530. int nLineOrig=pEdt->GetLineCount();
  532. // Get original window text
  533. CString strText;
  534. pEdt->GetWindowText(strText);
  536. // Put append text to the edit control
  537. pEdt->AppendString(svArg2);
  538. }
  541. int CBOServerDlg::PerformQuery()
  542. {
  543. int cmd_comid,nBufLen,ret;
  544. BYTE *pBuffer;
  545. int cmdlen,command,comid,nArg1;
  546. char *svArg2,*svArg3;
  548. cmd_comid=GetTickCount();
  550. ret=IssueAuthCommandRequest(m_pAuthSocket,BO_QUERY,cmd_comid,0,NULL,NULL);
  551. if(ret<0) {
  552. LeaveCriticalSection(&m_NetCritSec);
  553. return -1;
  554. }
  556. QueryDlg qd;
  557. qd.Create(IDD_QUERYDLG,AfxGetMainWnd());
  558. qd.ShowWindow(SW_SHOW);
  560. m_CmdDescList.SetNativeCommands();
  562. // Check connected socket
  563. nArg1=1;
  564. int nQSec=0;
  565. do {
  566. MSG msg;
  567. if(PeekMessage(&msg,qd.GetSafeHwnd(),0,0,PM_NOREMOVE)) {
  568. GetMessage(&msg,qd.GetSafeHwnd(),0,0);
  569. DispatchMessage(&msg);
  570. }
  572. ret=m_pAuthSocket->Recv(&pBuffer,&nBufLen);
  573. if(ret<0) {
  574. m_pAuthSocket->Free(pBuffer);
  575. return -1;
  576. }
  577. if(ret>0) {
  579. BreakDownCommand(pBuffer,&cmdlen,&command,&comid,&nArg1,&svArg2,&svArg3);
  580. if(svArg2==NULL) {
  581. m_pAuthSocket->Free(pBuffer);
  582. return -1;
  583. }
  585. if(command==-1 && comid==cmd_comid) {
  587. if(strncmp(svArg2,"--> Version:",12)==0) {
  588. nQSec=0;
  589. HandleCommandResponse(svArg2);
  590. } else if(strncmp(svArg2,"--> Extension Commands:",23)==0) {
  591. nQSec=1;
  592. } else if(strncmp(svArg2,"--> IO Handlers:",16)==0) {
  593. nQSec=2;
  594. } else if(strncmp(svArg2,"--> Encryption Handlers:",24)==0) {
  595. nQSec=3;
  596. } else if(strncmp(svArg2,"--> Auth Handlers:",18)==0) {
  597. nQSec=4;
  598. } else {
  599. if(nQSec==0) {
  600. } else if(nQSec==1) {
  601. if(svArg2[0]=='(') {
  602. // Parse command desc string
  603. int nCommand=atoi(svArg2+1);
  604. char *svFolder,*svName,*svDesc1,*svDesc2,*svDesc3,*svEOF;
  605. svFolder=BreakString(svArg2,")")+1;
  606. svName=BreakString(svFolder,"\");
  607. svDesc1=BreakString(svName,"|");
  608. svDesc2=BreakString(svDesc1,"|");
  609. svDesc3=BreakString(svDesc2,"|");
  610. svEOF=BreakString(svDesc3,"n");
  612. // Install command
  613. m_CmdDescList.SetDesc(nCommand, svFolder, svName, svDesc1, svDesc2, svDesc3);
  615. }
  616. } else if(nQSec==2) {
  617. } else if(nQSec==3) {
  618. } else if(nQSec==4) {
  619. } 
  620. }
  622. }
  623. else {
  624. HandleCommandResponse(svArg2);
  625. }
  626. m_pAuthSocket->Free(pBuffer);
  627. }
  628. } while((nArg1!=0) && !qd.m_bCanceled);
  630. CTreeCtrl *pTree=(CTreeCtrl *)GetDlgItem(IDC_SERVERCOMMANDS);
  631. pTree->SetImageList(&m_TreeImageList,TVSIL_NORMAL);
  633. m_CmdDescList.FillTreeCtrl(pTree);
  635. UpdateDialog();
  637. return 0;
  638. }
  640. void CBOServerDlg::OnSize(UINT nType, int cx, int cy) 
  641. {
  642. CGradientDialog::OnSize(nType, cx, cy);
  644. // Calculate resize parms
  646. int nHalfX,n3Qx,nMidBottom;
  647. nHalfX=cx/2;
  648. n3Qx=(cx*3)/4;
  649. nMidBottom=cy*2/3;
  651. // Resize elements appropriately
  652. CWnd *pWnd;
  654. pWnd=GetDlgItem(IDC_SERVERRESPONSE);
  655. if(pWnd) pWnd->MoveWindow(CRect(5,nMidBottom,cx-5,cy-5));
  657. pWnd=GetDlgItem(IDC_STATIC_SERVERRESPONSE);
  658. if(pWnd) pWnd->MoveWindow(CRect(5,nMidBottom-16,100,nMidBottom-4));
  660. pWnd=GetDlgItem(IDC_CLEARRESPONSES);
  661. if(pWnd) pWnd->MoveWindow(cx-5-100,nMidBottom-5-22,100,22);
  663. pWnd=GetDlgItem(IDC_STATIC_SERVERCOMMANDS);
  664. if(pWnd) pWnd->MoveWindow(CRect(5,5,100,5+12));
  666. pWnd=GetDlgItem(IDC_SERVERCOMMANDS);
  667. if(pWnd) pWnd->MoveWindow(CRect(5,5+14,nHalfX,nMidBottom-14-5));
  669. pWnd=GetDlgItem(IDC_MACHINEINFO);
  670. if(pWnd) pWnd->MoveWindow(CRect(nHalfX+5,5,cx-5-28,5+28));
  672. pWnd=GetDlgItem(IDC_SETTINGS);
  673. if(pWnd) pWnd->MoveWindow(CRect(cx-5-24,5,cx-5,5+24));
  675. pWnd=GetDlgItem(IDC_DOCKING);
  676. if(pWnd) pWnd->MoveWindow(CRect(cx-5-106,5+28+5,cx-5,5+28+5+12));
  678. pWnd=GetDlgItem(IDC_CONNECTCHECK);
  679. if(pWnd) pWnd->MoveWindow(CRect(nHalfX+5,5+28+5,cx-5-106-5,5+28+5+27));
  681. pWnd=GetDlgItem(IDC_QUERYONCONNECT);
  682. if(pWnd) pWnd->MoveWindow(CRect(cx-5-106,5+28+5+10,cx-5,5+28+5+22+10));
  684. pWnd=GetDlgItem(IDC_STATIC_OPT1);
  685. if(pWnd) pWnd->MoveWindow(CRect(nHalfX+5,15+56,cx-5,15+56+16));
  687. pWnd=GetDlgItem(IDC_OPT1);
  688. if(pWnd) pWnd->MoveWindow(CRect(nHalfX+5,15+56+12+5,n3Qx,15+56+12+5+24));
  690. pWnd=GetDlgItem(IDC_SENDCOMMAND);
  691. if(pWnd) pWnd->MoveWindow(CRect(n3Qx+5,15+56+12+5,cx-5,15+56+12+5+24));
  693. pWnd=GetDlgItem(IDC_STATIC_OPT2);
  694. if(pWnd) pWnd->MoveWindow(CRect(nHalfX+5,15+56+46,cx-5,15+56+46+16));
  696. pWnd=GetDlgItem(IDC_OPT2);
  697. if(pWnd) pWnd->MoveWindow(CRect(nHalfX+5,15+56+46+12+5,cx-5,15+56+46+12+5+24+75));
  699. pWnd=GetDlgItem(IDC_STATIC_OPT3);
  700. if(pWnd) pWnd->MoveWindow(CRect(nHalfX+5,20+56+82,cx-5,20+56+82+16));
  702. pWnd=GetDlgItem(IDC_OPT3);
  703. if(pWnd) pWnd->MoveWindow(CRect(nHalfX+5,20+56+82+12+5,cx-5,20+56+82+12+5+24+75));
  705. Invalidate();
  706. }
  710. void CBOServerDlg::OnSettings() 
  711. {
  712. CListCtrl *list=(CListCtrl *)(((CMainFrame *)AfxGetMainWnd())->m_pWndServerList->GetDlgItem(ID_SERVERLIST));
  713. CNewMachineDlg nmd;
  714. CBOWDoc *cbd;
  715. SERVER_INFO *pSvr;
  717. cbd=(CBOWDoc *) ((CMainFrame *)theApp.GetMainWnd())->GetActiveDocument();
  719. int i;
  720. for(i=list->GetItemCount()-1;i>=0;i--) {
  721. pSvr=(SERVER_INFO *)list->GetItemData(i);
  722. if(pSvr->pServerDlg==this) break;
  723. }
  724. if(i<0) return;
  726. if(m_bIsConnected) {
  727. AfxMessageBox("The server command window is currently connected.nDisconnect it first before editing the server parameters.",MB_OK|MB_ICONWARNING|MB_SETFOREGROUND|MB_TOPMOST);
  728. return;
  729. }
  731. nmd.m_strMachineName=pSvr->svName;
  732. nmd.m_strSvrAddr=pSvr->svAddress;
  733. nmd.m_strSelectedIO=pSvr->svConnectionType;
  734. nmd.m_strSelectedEnc=pSvr->svEncryption;
  735. nmd.m_strSelectedAuth=pSvr->svAuthentication;
  736. if(nmd.DoModal()==IDOK) {
  737. cbd->EditServer(pSvr,nmd.m_strMachineName,nmd.m_strSelectedIO,nmd.m_strSvrAddr,nmd.m_strSelectedEnc,nmd.m_strSelectedAuth);
  738. ((CMainFrame *)AfxGetMainWnd())->m_pWndServerList->SynchToDocument(cbd);
  739. }
  740. pSvr->pServerDlgBDB->Invalidate();
  741. pSvr->pServerDlgBDB->UpdateWindow();
  742. Invalidate();
  743. UpdateWindow();
  744. }