至顶网›软件频道 ›应用软件›用Python编写程序修复"恶邮差"变种U病毒感染文件。

用Python编写程序修复"恶邮差"变种U病毒感染文件。

扫一扫
分享文章到微信
扫一扫
关注官方公众号
至顶头条

将该文件置于某个目录（比如c盘根目录）运行时，可以修复该目录下所有病毒感染文件。

作者： lgfei 来源：CSDN 2008年5月26日

#recover all infected .ZMX file
import os
count=0         # Global 变量
os.system(r'attrib -s -h *.zmx /s') #所有感染文件感染后的属性是系统文件和隐藏文件
#括号内的参数实际上是windows的shell命令，其作用为修改感染文件属性为为普通属性
cwd=os.getcwd()#获得当前文件夹
#core函数是一个递归函数，将所有感染的ZMX文件后缀改回到.exe
def core(n_dir):
    global count
    print 'now in'+n_dir
    os.chdir(n_dir)
    li=os.listdir(n_dir)
    lifile=[x for x in li if '.ZMX' in x]
    if len(lifile)!=0:
print lifile
    for f in lifile:
        os.rename(f,f.replace('.ZMX','.exe'))
        print 'file'+f+'is cured successfully!'
        count+=1
    lidir=[x for x in li if os.path.isdir(x)]
    if len(lidir)==0:
        return
    print lidir #just for debug
    for d in lidir:
        core(n_dir+'\\'+d)

if __name__ == "__main__":
core(cwd)
print "congratulations!!"+str(count)+" infected files r found in"+cwd+' and they r all successfully cured!!!'
raw_input("Press any key to end program") #该语句可以避免命令行窗口自动关闭

将该文件置于某个目录（比如c盘根目录）运行时，可以修复该目录下所有病毒感染文件。可以稍作修改，先获取系统的分区，然后从每个分区根目录运行上面函数。值得注意的是当文件数目比较多，目录层次较深时，该程序可能比较费时（例如，我用该程序修复本人电脑F盘内感染文件时，程序运行花了两分钟左右。而该分区含文件个数69751个，文件夹数目是1216个，该分区的大小是28.9G）。该程序的递归算法显然效率上不是很令人满意。

实际上，可以全部利用shell命令来解决问题，将下列命令保存为.bat文件，或者利用python的os.system()函数来执行下面的语句
@dir /b /s *.zmx>zmx_file.txt
@for /f "delims==" %%a in (zmx_file.txt) do ren "%%a" *.exe
@del zmx_file.txt

VIP专区

VIP用户

普通用户

邮件订阅

如果您非常迫切的想了解IT领域最新产品与技术信息，那么订阅至顶网技术邮件将是您的最佳途径之一。

重磅专题

往期文章

用Python编写程序修复"恶邮差"变种U病毒感染文件。

业界热点: