算法运算

【实例简介】autorun病毒免疫程序
【实例截图】
【核心代码】void CAutoRunDlg::InitComboDrive()
{
    char szDriveStr[MAXBYTE] = { 0 };
    char *pTmp = NULL;

    SetDlgItemText(IDC_COMBO_DRIVE, "请选择欲免疫的磁盘盘符");
    
    GetLogicalDriveStrings(MAXBYTE, szDriveStr);
    pTmp = szDriveStr;

    while ( *pTmp )
    {
        m_CbDrive.AddString(pTmp);
        pTmp = 4;

    }
}
// 创建autorun.inf文件夹
#define AUTORUN "autorun.inf"
// 创建无法删除的文件夹
#define ANTI    "\\anti...\\"

void CAutoRunDlg::OnBtnImmunity() 
{
// TODO: Add your control notification handler code here
char szPath[MAX_PATH] = { 0 };
    GetDlgItemText(IDC_COMBO_DRIVE, szPath, MAX_PATH);

    // 创建autorun.inf文件夹
    strcat(szPath, AUTORUN);
    BOOL bRet = CreateDirectory(szPath, NULL);
    if ( !bRet )
    {
        AfxMessageBox("无法免疫该盘符！ \
            可能已经免疫，或者该磁盘为不可写状态！");
        return ;
    }

    // 创建无法删除的文件夹
    strcat(szPath, ANTI);
    bRet = CreateDirectory(szPath, NULL);
    if ( !bRet )
    {
        AfxMessageBox("无法免疫该盘符！ \
            可能已经免疫，或者该磁盘为不可写状态！");
    }
}

void CAutoRunDlg::OnBtnCancel() 
{
// TODO: Add your control notification handler code here
    
    char szPath[MAX_PATH] = { 0 };
    // 删除ANTI...\目录
    GetDlgItemText(IDC_COMBO_DRIVE, szPath, MAX_PATH);
    strcat(szPath, AUTORUN);
    strcat(szPath, ANTI);
    RemoveDirectory(szPath);
    
    ZeroMemory(szPath, MAX_PATH);
    // 删除autorun.inf目录
    GetDlgItemText(IDC_COMBO_DRIVE, szPath, MAX_PATH);
    strcat(szPath, AUTORUN);
    RemoveDirectory(szPath);
}