kcrong 2015. 2. 23. 13:49

import pickle

import driverlib


from immlib import *


def main( args ):


    ioctl_list  = []

    device_list = []


    dbg    = Debugger()

    driver = driverlib.Driver()


    # Grab the list of IOCTL codes and device names

    ioctl_list  = driver.getIOCTLCodes()

    if not len(ioctl_list):

        return "[*] ERROR! Couldn't find any IOCTL codes."

    

    device_list = driver.getDeviceNames()

    if not len(device_list):

        return "[*] ERROR! Couldn't find any device names."

    

    # Now create a keyed dictionary and pickle it to a file

    master_list = {}

    master_list["ioctl_list"]  = ioctl_list

    master_list["device_list"] = device_list


    fd = open( args[0], "wb")

    pickle.dump( master_list, fd )

    fd.close()

    


    return "[*] SUCCESS! Saved IOCTL codes and device names to %s" % args[0]