rapid7/metasploit-framework

/* DLLHijackAuditKit (C) 2010 Rapid7, Inc */

function print_status(msg) {
    try {
        WScript.StdOut.WriteLine("[*] "+ msg);
    } catch(e) {}
}

function process_list() {
    var res = new Array();
    var wbemFlagReturnImmediately = 0x10;
    var wbemFlagForwardOnly = 0x20;
    var oWMI = GetObject("winmgmts:\\\\localhost\\root\\CIMV2");
    var cPID = oWMI.ExecQuery("SELECT * FROM Win32_Process", "WQL", wbemFlagReturnImmediately | wbemFlagForwardOnly);
    var enumItems = new Enumerator(cPID);
    for (; !enumItems.atEnd(); enumItems.moveNext()) {
        var p = enumItems.item();
        if (p.ExecutablePath && p.ExecutablePath.toLowerCase().indexOf("taskmgr") != -1) continue;
        res.push(p.ProcessId);
    }
    return res;
}



var pause_interval = 100000;

var oFso = new ActiveXObject("Scripting.FileSystemObject");
var oShl = new ActiveXObject("WScript.Shell");
var oLoc = new ActiveXObject("WbemScripting.SWbemLocator");
var oSvc = oLoc.ConnectServer(null, "root\\default");
var oReg = oSvc.Get("StdRegProv");

var oCWD = oShl.CurrentDirectory + "";

var oMethod = oReg.Methods_.Item("EnumKey");
var oInParam = oMethod.InParameters.SpawnInstance_();
oInParam.hDefKey = 0x80000002;
oInParam.sSubKeyName = "Software\\Classes";

var oOutParam = oReg.ExecMethod_(oMethod.Name, oInParam);
var aNames = oOutParam.sNames.toArray();


try { oFso.CreateFolder("DLLAudit"); } catch(e) { }
try { oFso.CreateFolder("DLLAudit\\ext"); } catch(e) { }


if (! oFso.FileExists("procmon.exe")) {
    print_status("Downloading procmon.exe from \\\\live.sysinternals.com ...")
    try { oFso.CopyFile("\\\\live.sysinternals.com\\Tools\\procmon.exe", "procmon.exe"); } catch(e) {}
}

if (! oFso.FileExists("procmon.exe")) {
    print_status("Failed to download procmon.exe, copy here manually.");
    WScript.Quit();
}


print_status("Starting the process monitor...");
oShl.Run("procmon.exe /AcceptEULA /Quiet /LoadConfig DLLAudit.pmc", 10);
WScript.Sleep(5000);

var total = 0;
print_status("Creating test cases for each file extension...");

for (var i = 0; i < aNames.length; i++) {
    if (aNames[i].substr(0,1) != ".") continue;
    var ext = aNames[i].substr(1,32).toLowerCase();

    if (ext == "com") continue;
    if (ext == "pif") continue;
    if (ext == "exe") continue;
    if (ext == "bat") continue;
    if (ext == "scr") continue;
    if (ext == "dos") continue;
    if (ext == "386") continue;
    if (ext == "cpl") continue;
    if (ext == "sys") continue;
    if (ext == "dll") continue;
    if (ext == "drv") continue;
    if (ext == "rb") continue;
    if (ext == "py") continue;
    if (ext == "pl") continue;
    if (ext == "crds") continue;
    if (ext == "crd") continue;
    if (ext == "pml") continue;
    if (ext == "pmc") continue;

    try { oFso.CreateFolder("DLLAudit\\ext\\" + ext); } catch(e) { }
    try {
        var a = oFso.CreateTextFile("DLLAudit\\ext\\" + ext + "\\exploit." + ext);
        a.WriteLine("HOWDY!");
        a.Close();
    } catch(e) { }

    total++;
}

print_status("Created " + total + " test cases");
var procs = process_list();
print_status("Protecting " + procs.length + " processes");

var tries = 0;

var base = oFso.GetFolder("DLLAudit\\ext");
var subs = new Enumerator(base.SubFolders);
for (; !subs.atEnd(); subs.moveNext()) {
    var path = subs.item() + "";
    var bits = path.split("\\");
    var ext  = bits[bits.length - 1];

    print_status("Auditing extension: " + ext);
    oShl.CurrentDirectory = path + "\\";

    oShl.Run("cmd.exe /c start exploit." + ext, 0);
    WScript.Sleep(500);

    var nprocs = process_list();
    var cnt = 0;
    while(nprocs.length == procs.length && cnt < 2) {
        cnt++;
        WScript.Sleep(500);
        nprocs = process_list();
    }

    // If an application spawned, give it three seconds
    // This helps with ProcMon memory usage as well
    if (nprocs.length > procs.length) {
        WScript.Sleep(3000);
    }

    var killer = "taskkill /F ";
    for (var i=0; i < nprocs.length; i++) {
        var found = false;
        for (var x=0; x < procs.length; x++) {
            if (nprocs[i] == procs[x]) {
                found = true;
                break;
            }
        }
        if (found) continue;
        killer = killer + "/PID " + nprocs[i] + " ";
    }
    oShl.Run(killer, 0, true);

    tries++;

    if (tries % pause_interval == 0) {
        print_status("Completed " + tries + " extensions, hit enter to continue.")
        WScript.Stdin.ReadLine();
        print_status("Continuing...")
    }
}

print_status("Data collection phase complete, export Logfile.CSV from ProcMon.")