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

源码开发语言/平台
当前位置：首页 > 源码/资料 > 加密解密 > CA认证 > 查看源码
SetupTree.cpp
资源名称：MiniCA2.rar　[点击查看]
上传用户：dengkfang
上传日期：2008-12-30
资源大小：5233k
文件大小：14k
源码类别：

CA认证

开发平台：

Visual C++

SetupTree.cpp：源码内容
  1. // SetupTree.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "....minica.h"
  6. #include "SetupTree.h"
  7. #include "....CertDbPage.h"
  8. #include "....MiniMainDlg.h"
  10. #define SPACE 80
  11. #ifdef _DEBUG
  12. #define new DEBUG_NEW
  13. #undef THIS_FILE
  14. static char THIS_FILE[] = __FILE__;
  15. #endif
  17. /////////////////////////////////////////////////////////////////////////////
  18. // CSetupTree
  20. CSetupTree::CSetupTree()
  21. {
  22. m_pEdit = NULL;
  23. m_pButton = NULL;
  24. m_hControlItem = NULL;
  25. m_CdbPath = ((CMiniCaApp *)AfxGetApp())->GetAppPath() + "\CertDB.CDB";
  26. }
  28. CSetupTree::~CSetupTree()
  29. {
  30. m_HtreeList.RemoveAll();
  31. }
  34. BEGIN_MESSAGE_MAP(CSetupTree, CTreeCtrl)
  35. //{{AFX_MSG_MAP(CSetupTree)
  36. ON_WM_VSCROLL()
  37. ON_WM_HSCROLL()
  38. ON_NOTIFY_REFLECT(NM_CLICK, OnClick)
  39. ON_BN_CLICKED(TREE_OPTIONS_BUTTON_ID, OnOptionsButtonId)
  40. ON_WM_DESTROY()
  41. ON_NOTIFY_REFLECT(TVN_SELCHANGED, OnSelchanged)
  42. //}}AFX_MSG_MAP
  43. END_MESSAGE_MAP()
  45. /////////////////////////////////////////////////////////////////////////////
  46. // CSetupTree message handlers
  47. void CSetupTree::ExpandParent(HTREEITEM item,UINT nCode )////展开,收缩
  48. {
  49. SetRedraw(FALSE);
  50. if(item)
  51. Expand( item, nCode );
  52. else
  53. return;
  54. HTREEITEM hParentItem = GetParentItem( item );
  55. if(hParentItem)//是否有父
  56. {
  57. Expand( hParentItem, nCode );
  58. ExpandParent(hParentItem, nCode);
  59. }
  60. else
  61. return;
  62. // EnsureVisible( GetSelectedItem());
  63. SetRedraw();
  64. }
  66. void CSetupTree::OnVScroll(UINT nSBCode, UINT nPos, CScrollBar* pScrollBar) 
  67. {
  68. // TODO: Add your message handler code here and/or call default
  69. UpdateTreeControlTextFromChildControl();
  70. CTreeCtrl::OnVScroll(nSBCode, nPos, pScrollBar);
  71. }
  73. void CSetupTree::OnHScroll(UINT nSBCode, UINT nPos, CScrollBar* pScrollBar) 
  74. {
  75. // TODO: Add your message handler code here and/or call default
  76. UpdateTreeControlTextFromChildControl();
  77. CTreeCtrl::OnHScroll(nSBCode, nPos, pScrollBar);
  78. }
  80. void CSetupTree::OnClick(NMHDR* pNMHDR, LRESULT* pResult) 
  81. {
  82. // TODO: Add your control notification handler code here
  83. CPoint point;
  84. GetCursorPos(&point); // 当前鼠标坐标
  85. CPoint PointInTree = point;
  86. ScreenToClient( &PointInTree);
  88. HTREEITEM hPItem;
  89. UINT nFlag;
  90. HTREEITEM hItem = HitTest( PointInTree, &nFlag);
  91. if(hItem == NULL)
  92. {
  93. UpdateTreeControlTextFromChildControl();
  94. return;
  95. }
  97. if (nFlag&TVHT_ONITEMICON) //点种图标,则选择或不选
  98. {
  99. int nImage,nSelectedImage,nState;
  100. GetItemImage(hItem,nImage,nSelectedImage);
  101. if(nImage <3)
  102. {
  103. nState = !nImage;
  104. SetItemImage(hItem,nState,nState);
  105. TravelChild(hItem, nState);
  106. hPItem = GetParentItem(hItem);
  107. if(hPItem!=NULL)
  108. TravelSiblingAndParent(hItem,nState);
  109. }
  110. }
  112. else if(nFlag&TVHT_ONITEMLABEL) //点种标签,增加证书,密码
  113. {
  114. if ((nFlag & TVHT_ONITEM) && GetItemData(hItem) && m_hControlItem == NULL)
  115. {
  116. CreateNewChildControl(hItem);
  117. }
  118. }
  119. else
  120. {
  121. UpdateTreeControlTextFromChildControl();
  122. }
  124. *pResult = 0;
  125. }
  127. void CSetupTree::OnSelchanged(NMHDR* pNMHDR, LRESULT* pResult) 
  128. {
  129. NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR;
  130. // TODO: Add your control notification handler code here
  131. if (m_hControlItem)
  132. {
  133. UpdateTreeControlTextFromChildControl();
  134. }
  135. // //Create the new combo if need be
  136. if(pNMTreeView->itemOld.hItem)
  137. CreateNewChildControl(pNMTreeView->itemNew.hItem);
  139. *pResult = 0;
  140. }
  142. void CSetupTree::TravelChild(HTREEITEM hItem, int nState)
  143. {
  144. HTREEITEM hChildItem,hBrotherItem;
  145. DWORD dData = GetItemData(hItem);
  146. if(dData>=10)
  147. return;
  149. //查找子节点，没有就结束
  150. hChildItem = GetChildItem(hItem);
  151. if(hChildItem!=NULL)
  152. {
  153. //设置子节点的状态与当前节点的状态一致
  154. int temp;
  155. GetItemImage( hChildItem,temp ,temp );
  156. if(temp<3)
  157. SetItemImage( hChildItem,nState ,nState );
  158. //再递归处理子节点的子节点和兄弟节点
  159. TravelChild(hChildItem, nState);
  161. //处理子节点的兄弟节点和其子节点
  162. hBrotherItem = GetNextSiblingItem(hChildItem);
  163. while (hBrotherItem)
  164. {
  165. //设置子节点的兄弟节点状态与当前节点的状态一致
  166. SetItemImage( hBrotherItem,nState ,nState );
  167. //再递归处理子节点的兄弟节点的子节点和兄弟节点
  168. TravelChild(hBrotherItem, nState);
  169. hBrotherItem = GetNextSiblingItem(hBrotherItem);
  170. }
  171. }
  172. }
  174. void CSetupTree::TravelSiblingAndParent(HTREEITEM hItem, int nState)
  175. {
  176. HTREEITEM hNextSiblingItem,hPrevSiblingItem,hParentItem;
  177. //查找父节点，没有就结束
  178. hParentItem = GetParentItem(hItem);
  179. if(hParentItem==NULL)
  180. return;
  182. int nImage,nSelectedImage;
  183. BOOL bParent = TRUE;//需要继续比较
  184. //查找当前节点下面的兄弟节点的图像,如果nState!=nImage,则父节点为灰色
  185. hNextSiblingItem = GetNextSiblingItem(hItem);
  186. while(hNextSiblingItem!=NULL)
  187. {
  188. GetItemImage(hNextSiblingItem,nImage,nSelectedImage);
  189. if(nImage!=nState)
  190. {
  191. bParent = FALSE;//此次比较已经确认父为灰色
  192. break;
  193. }
  194. else 
  195. hNextSiblingItem = GetNextSiblingItem(hNextSiblingItem);
  196. }
  197. if(bParent)//继续比较兄弟节点
  198. {
  199. //查找当前节点上面的兄弟节点的状态
  200. hPrevSiblingItem = GetPrevSiblingItem(hItem);
  201. while(hPrevSiblingItem!=NULL)
  202. {
  203. GetItemImage(hPrevSiblingItem,nImage,nSelectedImage);
  204. if(nImage!=nState) 
  205. {
  206. bParent = FALSE;//此次比较已经确认父为灰色
  207. break;
  208. }
  209. else hPrevSiblingItem = GetPrevSiblingItem(hPrevSiblingItem);
  210. }
  211. }
  212. if(!bParent)//设置父节点为灰色,上一级为灰
  213. {
  214. SetItemImage(hParentItem, 2, 2);
  215. hParentItem = GetParentItem(hItem);
  216. while(hParentItem!=NULL)
  217. {
  218. SetItemImage( hParentItem, 2, 2);
  219. hParentItem = GetParentItem(hParentItem);
  220. }
  221. }
  222. else//设置父节点为当前选中状态
  223. SetItemImage(hParentItem,nState,nState );
  225. //再递归处理父节点的兄弟节点和其父节点
  226. TravelSiblingAndParent(hParentItem,nState);
  227. }
  229. void CSetupTree::OnOptionsButtonId() 
  230. {
  231. // TODO: Add your control notification handler code here
  232. SelectFile();
  233. }
  235. void CSetupTree::SelectFile() 
  236. {
  237. // TODO: Add your control notification handler code here
  238. HTREEITEM hItem = GetSelectedItem ();
  239. UINT m_SelectID = GetItemData(hItem);
  240. CString sFilter,sTitle;//过滤器,标题条
  241. if(m_SelectID<100)
  242. return;
  243. switch(m_SelectID%100)
  244. {
  245. case 0://公钥
  246. sFilter = "公钥证书(*.pem;*.der;*.cer)|*.pem;*.der;*.cer|所有文件(*.*)|*.*||";
  247. sTitle = "公钥证书文件";
  248. break;
  249. case 10://私钥
  250. sFilter = "私钥证书(*.pem;*.der;*.cer)|*.pem;*.der;*.cer|所有文件(*.*)|*.*||";
  251. sTitle = "私钥证书文件";
  252. break;
  253. case 20://pfx
  254. sFilter = "个人信息交换(*.pfx;*.p12)|*.pfx;*.p12|所有文件(*.*)|*.*||";
  255. sTitle = "个人信息交换文件";
  256. break;
  257. default:
  258. return;
  259. }
  260. CFileDialog dlgOpen(true,NULL,NULL,OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT,
  261. CMiniCaApp::NormalCode(sFilter),NULL);
  263. CString strTitle = CMiniCaApp::NormalCode(sTitle);
  264. dlgOpen.m_ofn.lpstrTitle = strTitle;//标题条
  266. if(dlgOpen.DoModal()==IDOK) 
  267. {
  268. CString sName = dlgOpen.GetPathName();
  269. CString strCause,strPwd;
  270. // CString sName = GetItemText(hItem);
  271. // sName += ": " + dlgOpen.GetPathName();
  272. // DWORD dData = GetItemData(hItem);
  273. // stuCertDB CertDB;
  274. // if(CertDB.LoadFromCert(sName.GetBuffer(0),dData,strPwd.GetBuffer(0),0,strCause))
  275. SetItemText(hItem,sName);
  276. UpdateTreeControlTextFromChildControl();
  278. // int nImage,nSelectedImage,nState;
  279. // HTREEITEM hPItem;
  280. // SetItemImage(hItem,1,1);
  281. // TravelChild(hItem, 1);
  282. // hPItem = GetParentItem(hItem);
  283. // if(hPItem!=NULL)
  284. // TravelSiblingAndParent(hItem,1);
  286. }
  287. }
  290. void CSetupTree::LoadCdb() //加载证书库
  291. {
  292. UpdateTreeControlTextFromChildControl();
  293. CFile file;
  294. if(!file.Open(m_CdbPath,CFile::modeRead))
  295. return;
  296. for(;;)
  297. {
  298. stuCertDB CertDB;
  299. int len = file.Read(&CertDB,sizeof(stuCertDB));
  300. if(len==0)
  301. break;
  302. CString str;
  303. if(CertDB.LoadFromDB(str))
  304. {
  305. //查找链表,增加数据
  306. DWORD dData;
  307. HTREEITEM hTempItem;
  308. for(POSITION pos = m_HtreeList.GetHeadPosition(); pos!=NULL; )
  309. {
  310. hTempItem = (HTREEITEM)m_HtreeList.GetNext(pos);
  311. dData = GetItemData(hTempItem);
  312. if(dData == CertDB._uID)
  313. {
  314. SetItemText(hTempItem, CMiniCaApp::NormalCode(str));
  315. SetItemImage(hTempItem,4,4);
  316. if(dData%100 == 10)//私钥
  317. {
  318. SetItemText(GetNextSiblingItem(hTempItem),"********");
  319. if(strcmp(CertDB._chPWD,"")!=0)//密码有效
  320. {
  321. SetItemImage(GetNextSiblingItem(hTempItem),6,6);
  322. }
  323. else
  324. {
  325. SetItemImage(GetNextSiblingItem(hTempItem),5,5);
  326. }
  328. }
  330. SetItemImage(GetParentItem(hTempItem),CertDB._bUSED,CertDB._bUSED);
  331. TravelSiblingAndParent(GetParentItem(hTempItem),CertDB._bUSED);
  332. //展开有效项目
  333. ExpandParent(GetParentItem(hTempItem),TVE_EXPAND);
  334. break;
  335. }
  337. }
  338. }
  339. }
  340. file.Close();
  341. }
  343. void CSetupTree::Savecdb() //保存证书库
  344. {
  345. // TODO: Add your control notification handler code here
  346. // ResetContent();
  347. //遍历数
  348. UpdateTreeControlTextFromChildControl();
  349. CFile file;
  350. if(!file.Open(m_CdbPath,CFile::modeReadWrite | CFile::modeNoTruncate | CFile::modeCreate))
  351. return;
  352. HTREEITEM hTempItem = NULL;
  353. int nImage,nSelectedImage;
  354. CString strPwd,strName,strCause;
  355. HTREEITEM hPItem;
  356. DWORD dData,len = 0;
  357. for(POSITION pos = m_HtreeList.GetHeadPosition(); pos!=NULL;)
  358. {
  359. stuCertDB CertDB;
  360. hTempItem = (HTREEITEM)m_HtreeList.GetNext(pos);
  361. hPItem = GetParentItem(hTempItem);
  362. dData = GetItemData(hTempItem);
  363. GetItemImage(hPItem,nImage,nSelectedImage);//需要直接父窗口的图像
  364. strName = GetItemText(hTempItem);//得到文件或者证书名成
  365. if(strName == "公钥文件" || strName == "私钥文件")//表示此项没有加载
  366. continue;
  368. switch(dData%100)
  369. {
  370. case 0://公钥
  371. strPwd = "";
  372. break;
  373. case 10://私钥
  374. strPwd = GetItemText(GetNextSiblingItem(hTempItem));
  375. break;
  376. default:
  377. break;
  378. }
  380. if(strName.Find(":\",0) == -1)//已经是证书库中内容,非文件,不包含路径,就是证书名
  381. {
  382. //判断使用情况
  383. file.SeekToBegin();
  384. for(int i = 0;;i++)
  385. {
  386. len = file.Read(&CertDB,sizeof(stuCertDB));
  387. if(len == 0)
  388. break;
  389. if(CertDB._uID == dData)
  390. {
  391. file.Seek(i*sizeof(stuCertDB),CFile::begin);
  392. CertDB._bUSED = nImage;
  393. if(dData%100 == 10)
  394. {
  395. if(strPwd.Find("********") != -1 && strcmp(CertDB._chPWD,"") == 0) //发现********
  396. {
  397. strPwd = "";
  398. }
  399. strcpy(CertDB._chPWD,strPwd.GetBuffer(0));//密码
  400. }
  401. //修改CRC
  402. CertDB.SetCRC();
  403. file.Write(&CertDB,sizeof(CertDB));
  404. break;
  405. }
  407. }
  408. continue;
  409. }
  410. else//文件
  411. {
  412. //新文件
  413. if(CertDB.LoadFromCert(strName.GetBuffer(0),dData,strPwd.GetBuffer(0),
  414. nSelectedImage,strCause))
  415. {
  416. BOOL bAlread = FALSE;
  417. len = 0;
  418. //判断证书库中是否已经存在
  419. file.SeekToBegin();
  420. for(int i = 0;;i++)
  421. {
  422. stuCertDB CertTemp;
  423. len = file.Read(&CertTemp,sizeof(stuCertDB));
  424. if(len == 0)
  425. break;
  426. if(CertTemp._uID == dData)
  427. {
  428. file.Seek(i*sizeof(stuCertDB),CFile::begin);
  429. CertDB._bUSED = nImage;
  430. if(dData%100 == 10)// && strPwd.Find("********")==-1
  431. strcpy(CertDB._chPWD,strPwd.GetBuffer(0));//密码
  432. //修改CRC
  433. CertDB.SetCRC();
  434. file.Write(&CertDB,sizeof(CertDB));
  435. bAlread = TRUE;
  436. break;
  437. }
  439. }
  440. if(!bAlread) //新文件
  441. {
  442. file.SeekToEnd();
  443. file.Write(&CertDB,sizeof(stuCertDB));
  444. }
  445. }
  446. else
  447. {
  448. strName = GetItemText(hPItem);
  449. strName += strCause;
  450. //得到父窗口
  451. CCertDbPage * pSetupPage =  (CCertDbPage *)GetParent();
  452. if(::IsWindow(pSetupPage->m_hWnd))
  453. pSetupPage->AddMsg(strName,M_WARING);
  454. }
  455. }
  456. }
  457. file.Close();
  458. LoadCdb();
  459. }
  461. void CSetupTree::UpdateTreeControlTextFromChildControl()
  462. {
  463. HTREEITEM hItem = m_hControlItem;
  465. if (m_pEdit)
  466. {
  467. CString sText;
  468.     m_pEdit->GetWindowText(sText);
  469. if(hItem)
  470. SetItemText(hItem, sText);
  471. m_pEdit->DestroyWindow();
  472. delete m_pEdit;
  473. m_pEdit = NULL;
  474. }
  476. if (m_pButton)
  477. {
  478. m_pButton->DestroyWindow();
  479. delete m_pButton;
  480. m_pButton = NULL;
  481. }
  482. m_hControlItem = NULL;
  483. }
  485. void CSetupTree::OnDestroy() 
  486. {
  487. CTreeCtrl::OnDestroy();
  488. // TODO: Add your message handler code here
  489. UpdateTreeControlTextFromChildControl();
  491. }
  493. void CSetupTree::CreateNewChildControl(HTREEITEM hItem)
  494. {
  495. UINT m_SelectID = GetItemData(hItem);
  496. // SelectItem(hItem);
  497. CString str;
  498. str.Format("%d",m_SelectID);
  499. // AfxMessageBox(str);
  501. if(m_SelectID >= 100)
  502. {
  503. UINT uTemp = m_SelectID%100;
  504. if((uTemp == 11)||(uTemp == 21))//密码 增加新控件
  505. {
  506. // ModifyStyle(0, GetStyle() | TVS_EDITLABELS);
  507. //work out the rect for the combo box
  508. CString sEditText = GetItemText(hItem);
  510. CFont* pFont = GetFont();
  512. //work out the rect for the combo box
  513. CRect rText;
  514. GetItemRect(hItem, &rText, TRUE);
  516. CRect r;
  517. r.top = rText.top;
  518. r.left = rText.left ;
  519. r.right = r.left + 60; //Ask the edit box for the width to use
  520. r.bottom = rText.bottom + 5;
  522. //Create the new edit box
  523. m_pEdit = new CEdit;
  524. ASSERT(m_pEdit);
  525. m_pEdit->CreateEx(WS_EX_CLIENTEDGE, _T("Edit"), sEditText,
  526. WS_VISIBLE | WS_CHILD | ES_LEFT | ES_AUTOHSCROLL | ES_PASSWORD,
  527. r, this, TREE_OPTIONS_EDITBOX_ID);
  528. m_pEdit->SetLimitText(8);
  530. //set the font the edit box should use based on the font in the tree control
  531. m_pEdit->SetFont(pFont);
  532. m_pEdit->SendMessage(EM_SETSEL, 0, -1);
  533. m_pEdit->SetFocus();
  534. }
  535. else if((uTemp == 0)||(uTemp == 10)||(uTemp == 20)) //文件
  536. {
  537. m_pButton = new CXPStyleButtonST;
  538. ASSERT(m_pButton);
  539. CRect rText;
  540. GetItemRect(hItem, &rText, TRUE);
  542. CRect r;
  543. r.top = rText.top;
  544. r.left = rText.left + SPACE;
  545. r.right = r.left + 16; //Ask the edit box for the width to use
  546. r.bottom = rText.top + 16;
  548. m_pButton->Create("",WS_VISIBLE | WS_CHILD | BS_PUSHBUTTON, r, 
  549. this, TREE_OPTIONS_BUTTON_ID);
  550. m_pButton->SetIcon(IDI_ICON_BUTTON);
  551. // m_pButton->OffsetColor(CButtonST::BTNST_COLOR_BK_IN, 30);
  552. // m_pButton->SetFlat(FALSE);
  554. }
  555. }
  556. m_hControlItem = hItem;
  557. }