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

源码开发语言/平台
当前位置：首页 > 源码/资料 > 磁盘/存储 > SCSI/ASPI > 查看源码
FPDUMPWDlg.cpp
资源名称：fpdumpw.zip　[点击查看]
上传用户：wandash
上传日期：2007-01-07
资源大小：79k
文件大小：11k
源码类别：

SCSI/ASPI

开发平台：

Visual C++

FPDUMPWDlg.cpp：源码内容
  1. // FPDUMPWDlg.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "FPDUMPW.h"
  6. #include "FPDUMPWDlg.h"
  7. #include "aspiw32.h"
  8. #include "DialogSettings.h"
  9. #include <io.h>
  10. #include <direct.h>
  12. #ifdef _DEBUG
  13. #define new DEBUG_NEW
  14. #undef THIS_FILE
  15. static char THIS_FILE[] = __FILE__;
  16. #endif
  17. CString strFileName;
  18. BYTE HostAdapterID = 0;
  19. BYTE TargetID = 3;
  20. BYTE LUN = 0;
  21. // 后台线程
  22. UINT DumpWatcher( LPVOID pParam )
  23. {
  24. IsWatcherRunning = TRUE;
  25. bCancelWatcher = FALSE;
  26. CWnd *pDialog;
  27. CButton *pbtnSetting;
  28. CButton *pbtnClose;
  29. pDialog = AfxGetMainWnd();
  30. pbtnSetting = (CButton*)(pDialog->GetDlgItem(IDC_DUMP_SETTINGS));
  31. pbtnSetting->EnableWindow(FALSE);
  32. pbtnClose = (CButton*)(pDialog->GetDlgItem(IDC_DUMP_CLOSE));
  33. pbtnClose->EnableWindow(FALSE);
  35. // ::MessageBox(NULL,strFileName,"MESSAGE",MB_OK);
  36. // 1. Wait until tape is ready...
  37. pDialog->SetDlgItemText(IDC_DUMPSTATUS,"等待磁带机就绪...");
  38. for (;;)
  39. {
  40. if (bCancelWatcher) goto ExitLoop;
  41. if (IO_ALL_TEST_UNIT_READY(HostAdapterID,TargetID,LUN)==SS_COMP) break;
  42. }
  43. pDialog->SetDlgItemText(IDC_DUMPSTATUS,"磁带机已经就绪");
  45. // 2. Load the tape
  46. if (bCancelWatcher) goto ExitLoop;
  47. pDialog->SetDlgItemText(IDC_DUMPSTATUS,"正在装载磁带...");
  48. if (IO_TAPE_LOAD(HostAdapterID,TargetID,LUN,0,1,1)!=SS_COMP)
  49. {
  50. pDialog->SetDlgItemText(IDC_DUMPSTATUS,"装载磁带出错");
  51. goto ExitLoop;
  52. }
  53. pDialog->SetDlgItemText(IDC_DUMPSTATUS,"装载磁带完成");
  55. // 3. read files
  56. if (bCancelWatcher) goto ExitLoop;
  57. {
  58. CString strFileNameEx;
  59. char szTag[12];
  60. int nStatus = 1; // 1 头，2 身体，3 尾
  61. int n = 1;
  62. int block = 0;
  63. FILE *fp = NULL;
  65. for (;;)
  66. {
  67. BYTE buf[2048];
  68. int nBlockLen;
  69. BOOLEAN bFileMark;
  70. BOOLEAN bILI;
  71. BOOLEAN bEOM;
  72. BOOLEAN bValid;
  73. BYTE bSenseKey;
  74. for (;;)
  75. {
  76. if (bCancelWatcher) goto ExitLoop;
  77. if (IO_ALL_TEST_UNIT_READY(HostAdapterID,TargetID,LUN)==SS_COMP) break;
  78. }
  79. if (bCancelWatcher) goto ExitLoop;
  80. IO_TAPE_READ(HostAdapterID,TargetID,LUN,2048,&(buf[0]),&nBlockLen,&bFileMark,&bEOM,&bILI,&bValid,&bSenseKey);
  81. if (bValid==0)
  82. {
  83. pDialog->SetDlgItemText(IDC_DUMPSTATUS,"读磁带的过程中发生了无法预料的错误");
  84. fclose(fp);
  85. fp = NULL;
  86. goto ExitLoop;
  87. }
  88. if (nBlockLen>0)
  89. {
  90. if (nBlockLen<=2048)
  91. {
  92. if (fp==NULL)
  93. {
  94. switch(nStatus)
  95. {
  96. case 1:
  97. sprintf(szTag,".H%2.2d",n);
  98. break;
  99. case 2:
  100. sprintf(szTag,".D%2.2d",n);
  101. break;
  102. case 3:
  103. sprintf(szTag,".E%2.2d",n);
  104. break;
  105. }
  106. strFileNameEx = strFileName+ CString(szTag);
  108. pDialog->SetDlgItemText(IDC_DUMPSTATUS,CString("正在创建文件")+strFileNameEx);
  110. fp = fopen(strFileNameEx,"wb");
  111. if (fp==NULL)
  112. {
  113. pDialog->SetDlgItemText(IDC_DUMPSTATUS,CString("无法创建文件")+strFileNameEx);
  114. goto ExitLoop2;
  115. }
  116. }
  117. fwrite(&(buf[0]),1,nBlockLen,fp);
  118. } else
  119. {
  120. pDialog->SetDlgItemText(IDC_DUMPSTATUS,"磁带数据缓冲区太小");
  121. fclose(fp);
  122. fp = NULL;
  123. goto ExitLoop;
  124. }
  125. }
  126. if (bSenseKey==8)
  127. {
  128. if (fp!=NULL)
  129. {
  130. fclose(fp);
  131. fp = NULL;
  132. }
  133. pDialog->SetDlgItemText(IDC_DUMPSTATUS,"读磁带顺利结束");
  134. goto ExitLoop2;
  135. }
  136. if ((bFileMark)&&(fp!=NULL))
  137. {
  138. nStatus = (nStatus % 3) + 1;
  139. fclose(fp);
  140. fp = NULL;
  141. }
  142. }
  143. }
  144. ExitLoop:
  145. pDialog->SetDlgItemText(IDC_DUMPSTATUS,"用户中断了读磁带");
  146. ExitLoop2:
  147. pbtnSetting = (CButton*)(pDialog->GetDlgItem(IDC_DUMP_SETTINGS));
  148. pbtnSetting->EnableWindow(TRUE);
  149. pbtnClose = (CButton*)(pDialog->GetDlgItem(IDC_DUMP_CLOSE));
  150. pbtnClose->EnableWindow(TRUE);
  151. IsWatcherRunning = FALSE;
  152. return 0;
  153. }
  155. // 信号灯
  156. BOOLEAN  bCancelWatcher;
  157. CWinThread *pDumpWatcher;
  158. BOOLEAN  IsWatcherRunning = FALSE;
  159. /////////////////////////////////////////////////////////////////////////////
  160. // CAboutDlg dialog used for App About
  162. class CAboutDlg : public CDialog
  163. {
  164. public:
  165. CAboutDlg();
  167. // Dialog Data
  168. //{{AFX_DATA(CAboutDlg)
  169. enum { IDD = IDD_ABOUTBOX };
  170. //}}AFX_DATA
  172. // ClassWizard generated virtual function overrides
  173. //{{AFX_VIRTUAL(CAboutDlg)
  174. protected:
  175. virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
  176. //}}AFX_VIRTUAL
  178. // Implementation
  179. protected:
  180. //{{AFX_MSG(CAboutDlg)
  181. //}}AFX_MSG
  182. DECLARE_MESSAGE_MAP()
  183. };
  185. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  186. {
  187. //{{AFX_DATA_INIT(CAboutDlg)
  188. //}}AFX_DATA_INIT
  189. }
  191. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  192. {
  193. CDialog::DoDataExchange(pDX);
  194. //{{AFX_DATA_MAP(CAboutDlg)
  195. //}}AFX_DATA_MAP
  196. }
  198. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  199. //{{AFX_MSG_MAP(CAboutDlg)
  200. // No message handlers
  201. //}}AFX_MSG_MAP
  202. END_MESSAGE_MAP()
  204. /////////////////////////////////////////////////////////////////////////////
  205. // CFPDUMPWDlg dialog
  207. CFPDUMPWDlg::CFPDUMPWDlg(CWnd* pParent /*=NULL*/)
  208. : CDialog(CFPDUMPWDlg::IDD, pParent)
  209. {
  210. //{{AFX_DATA_INIT(CFPDUMPWDlg)
  211. // NOTE: the ClassWizard will add member initialization here
  212. //}}AFX_DATA_INIT
  213. // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
  214. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  215. }
  217. void CFPDUMPWDlg::DoDataExchange(CDataExchange* pDX)
  218. {
  219. CDialog::DoDataExchange(pDX);
  220. //{{AFX_DATA_MAP(CFPDUMPWDlg)
  221. // NOTE: the ClassWizard will add DDX and DDV calls here
  222. //}}AFX_DATA_MAP
  223. }
  225. BEGIN_MESSAGE_MAP(CFPDUMPWDlg, CDialog)
  226. //{{AFX_MSG_MAP(CFPDUMPWDlg)
  227. ON_WM_SYSCOMMAND()
  228. ON_WM_PAINT()
  229. ON_WM_QUERYDRAGICON()
  230. ON_BN_CLICKED(IDC_DUMP_START, OnDumpStart)
  231. ON_BN_CLICKED(IDC_DUMP_CANCEL, OnDumpCancel)
  232. ON_WM_CLOSE()
  233. ON_BN_CLICKED(IDC_DUMP_SETTINGS, OnDumpSettings)
  234. ON_BN_CLICKED(IDC_DUMP_CLOSE, OnDumpClose)
  235. //}}AFX_MSG_MAP
  236. END_MESSAGE_MAP()
  238. /////////////////////////////////////////////////////////////////////////////
  239. // CFPDUMPWDlg message handlers
  241. BOOL CFPDUMPWDlg::OnInitDialog()
  242. {
  243. CDialog::OnInitDialog();
  245. // Add "About..." menu item to system menu.
  247. // IDM_ABOUTBOX must be in the system command range.
  248. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  249. ASSERT(IDM_ABOUTBOX < 0xF000);
  251. CMenu* pSysMenu = GetSystemMenu(FALSE);
  252. if (pSysMenu != NULL)
  253. {
  254. CString strAboutMenu;
  255. strAboutMenu.LoadString(IDS_ABOUTBOX);
  256. if (!strAboutMenu.IsEmpty())
  257. {
  258. pSysMenu->AppendMenu(MF_SEPARATOR);
  259. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  260. }
  261. }
  263. // Set the icon for this dialog.  The framework does this automatically
  264. //  when the application's main window is not a dialog
  265. SetIcon(m_hIcon, TRUE); // Set big icon
  266. SetIcon(m_hIcon, FALSE); // Set small icon
  268. // TODO: Add extra initialization here
  269. pDumpWatcher = NULL;
  270. if (!LoadASPIManager())
  271. {
  272. MessageBox("无法载入ASPI管理模块，程序无法继续","严重错误",MB_ICONERROR);
  273. EndDialog(FALSE);
  274. }
  275. return TRUE;  // return TRUE  unless you set the focus to a control
  276. }
  278. void CFPDUMPWDlg::OnSysCommand(UINT nID, LPARAM lParam)
  279. {
  280. if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  281. {
  282. CAboutDlg dlgAbout;
  283. dlgAbout.DoModal();
  284. }
  285. else
  286. {
  287. CDialog::OnSysCommand(nID, lParam);
  288. }
  289. }
  291. // If you add a minimize button to your dialog, you will need the code below
  292. //  to draw the icon.  For MFC applications using the document/view model,
  293. //  this is automatically done for you by the framework.
  295. void CFPDUMPWDlg::OnPaint() 
  296. {
  297. if (IsIconic())
  298. {
  299. CPaintDC dc(this); // device context for painting
  301. SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
  303. // Center icon in client rectangle
  304. int cxIcon = GetSystemMetrics(SM_CXICON);
  305. int cyIcon = GetSystemMetrics(SM_CYICON);
  306. CRect rect;
  307. GetClientRect(&rect);
  308. int x = (rect.Width() - cxIcon + 1) / 2;
  309. int y = (rect.Height() - cyIcon + 1) / 2;
  311. // Draw the icon
  312. dc.DrawIcon(x, y, m_hIcon);
  313. }
  314. else
  315. {
  316. CDialog::OnPaint();
  317. }
  318. }
  320. // The system calls this to obtain the cursor to display while the user drags
  321. //  the minimized window.
  322. HCURSOR CFPDUMPWDlg::OnQueryDragIcon()
  323. {
  324. return (HCURSOR) m_hIcon;
  325. }
  327. void CFPDUMPWDlg::OnDumpStart() 
  328. {
  329. // TODO: Add your control notification handler code here
  330. CString strDir;
  331. CString strName;
  332. GetDlgItemText(IDC_EDIT1,strDir);
  333. GetDlgItemText(IDC_EDIT2,strName);
  334. strDir.TrimLeft();
  335. strDir.TrimRight();
  336. strName.TrimLeft();
  337. strName.TrimRight();
  338. if ((strDir.GetLength()!=0)&&(strDir.Right(1)!=CString("\")))
  339. {
  340. strDir += CString("\");
  341. }
  343. strFileName = strDir + strName;
  344. if (strFileName.GetLength()==0)
  345. {
  346. MessageBox("子目录名和存盘名不能全部为空","参数错误",MB_ICONERROR);
  347. return ;
  348. }
  349. if (strDir.GetLength()>0)
  350. {
  351. CString strSearch = strDir + "*.*";
  352. struct _finddata_t fileinfo;
  353. CString strPath = strDir.Left(strDir.GetLength()-1);
  354. long handle;
  355. // 判断子目录是否存在
  356. handle = _findfirst(strSearch,&fileinfo);
  357. if (handle==-1)
  358. {
  359. int nRes = MessageBox("子目录不存在,是否要创建?","参数错误",MB_ICONERROR|MB_YESNO);
  360. if (nRes==IDYES)
  361. {
  362. if (-1==_mkdir(strPath))
  363. {
  364. MessageBox("无法创建子目录?","参数错误",MB_ICONERROR|MB_OK);
  365. }
  366. }
  367. } else
  368. {
  369. _findclose(handle);
  370. }
  371. }
  373. // 判断文件是否存在
  374. FILE *fp;
  375. CString strFirstFileName;
  376. strFirstFileName = strFileName + ".D01";
  377. fp = fopen((LPCSTR)strFirstFileName,"rb");
  378. if (fp!=NULL) 
  379. {
  380. fclose(fp);
  381. if (IDNO==MessageBox("文件已存在,要覆盖吗?","文件名重复",MB_YESNO))
  382. return ;
  383. }
  385. pDumpWatcher = AfxBeginThread(DumpWatcher,NULL);
  386. }
  388. void CFPDUMPWDlg::OnDumpCancel() 
  389. {
  390. // TODO: Add your control notification handler code here
  391. if (!IsWatcherRunning)
  392. return;
  393. bCancelWatcher = TRUE;
  394. // WaitForSingleObject(pDumpWatcher->m_hThread,INFINITE);
  395. // pDumpWatcher=NULL;
  396. }
  398. void CFPDUMPWDlg::OnClose() 
  399. {
  400. // TODO: Add your message handler code here and/or call default
  402. CDialog::OnClose();
  403. }
  405. void CFPDUMPWDlg::OnDumpSettings() 
  406. {
  407. // TODO: Add your control notification handler code here
  409. CDialogSettings myDialogSettings;
  410. myDialogSettings.m_HostAdapterID = HostAdapterID;
  411. myDialogSettings.m_TargetID  = TargetID;
  412. myDialogSettings.m_LUN  = LUN;
  414. if (IDOK==myDialogSettings.DoModal())
  415. {
  416. HostAdapterID = myDialogSettings.m_HostAdapterID;
  417. TargetID      = myDialogSettings.m_TargetID;
  418. LUN   = myDialogSettings.m_LUN;
  419. }
  420. }
  422. void CFPDUMPWDlg::OnDumpClose() 
  423. {
  424. // TODO: Add your control notification handler code here
  425. EndDialog(0);
  426. }
  428. void CFPDUMPWDlg::OnOK() 
  429. {
  430. // TODO: Add extra validation here
  431. return;
  432. // CDialog::OnOK();
  433. }
  435. void CFPDUMPWDlg::OnCancel() 
  436. {
  437. // TODO: Add extra cleanup here
  438. if (IsWatcherRunning)
  439. return;
  440. CDialog::OnCancel();
  441. }