mona_installer

import os,sys,urllib,zipfile,subprocess,fnmatch

def find_location(pattern,path):
        print "[+] Locating %s in %s" % (pattern,path)
        for (path, dirs, files) in os.walk(path):
                if os.path.isfile(os.path.join(path,pattern)) and path.lower().find("x64") == -1:
                        return path

def unzip_file():
        try:
                print "[+] Unzipping download"
                zip = zipfile.ZipFile(r"pykd.zip")
                zip.extractall("unzipped_pykd")
                zip.close()
        except:
                print "[-] Error Unzipping"

def download_files():
        try:
                print "[+] Downloading PYKD zip file from Corelan, this may take a few moments (4mb)"
                urllib.urlretrieve ("https://github.com/corelan/windbglib/raw/master/pykd/pykd.zip", "pykd.zip")
                unzip_file()
                print "[+] Downloading mona.py & windbglib.py from Corelan"
                urllib.urlretrieve("https://github.com/corelan/mona/raw/master/mona.py", "unzipped_pykd\\mona.py")
                urllib.urlretrieve("https://github.com/corelan/windbglib/raw/master/windbglib.py", "unzipped_pykd\\windbglib.py")
        except:
                print "[-] Error Downloading"

def install_vc_file():
        try:
                print "[+] Installing VC redistributable"
                p = subprocess.Popen("unzipped_pykd\\vcredist_x86.exe").wait()
        except:
                print "[-] Error installing redistributable"

def register_dll():
        try:
                dll = find_location("msdia90.dll","C:\\")
                print ("[+] Registering 'msdia90.dll' with system located in:\n\t[*] %s" %dll)
                p = subprocess.Popen('regsvr32 "%s\\msdia90.dll"' %dll).wait()
        except:
                print "[-] Error registering DLL"

def copy_files():
        try:
                windbg = find_location("windbg.exe","C:\\")
                print ("[+] Copy pykd.pyd & windbglib.py & mona.py into Windbg folder:\n\t[*] %s" %windbg)
                os.system('move "unzipped_pykd\\pykd.pyd" "%s\\winext\\pykd.pyd"' %windbg)
                os.system('move "unzipped_pykd\\mona.py" "%s\\"' %windbg)
                os.system('move "unzipped_pykd\\windbglib.py" "%s\\"' %windbg)
        except:
                print "[-] Error copying file"

if __name__ == '__main__':
        try:
                        download_files()
                        install_vc_file()
                        register_dll()
                        copy_files()
                        print "[+] Complete. Open Windbg and test"
        except:
                        print "Error running"
                        sys.exit(0)