mirror of
https://github.com/thecompu/Driv3rs.git
synced 2024-06-11 13:29:32 +00:00
58 lines
1.8 KiB
Python
58 lines
1.8 KiB
Python
from struct import unpack
|
|
import os
|
|
|
|
def readUnpack(bytes, **options):
|
|
if options.get("type") == 't':
|
|
print 'DEBUG: In t'
|
|
SOS = SOSfile.read(bytes)
|
|
return unpack('< %ss' % bytes, SOS)
|
|
|
|
if options.get("type") == 'b':
|
|
print 'DEBUG: In b'
|
|
SOS = SOSfile.read(bytes)
|
|
offset_unpacked = unpack ('< H', SOS)
|
|
return int('.'.join(str(x) for x in offset_unpacked))
|
|
|
|
#Clear SCREEN
|
|
print("\033c");
|
|
|
|
#Is File a SOS DRIVER file?
|
|
|
|
SOSfile = open('SOSCFFA.DRIVER', 'rb')
|
|
filetype = readUnpack(8, type = 't')
|
|
|
|
if 'SOS DRVR' in filetype:
|
|
print "This is a proper SOS file."
|
|
print "Filetype is: %s." % (filetype)
|
|
else:
|
|
print "Filetype is: %s." % (filetype)
|
|
print "This is not a proper SOS file"
|
|
exit()
|
|
|
|
|
|
### At this point, we need the first offset to tell us where to jump to
|
|
### find the first actual driver.
|
|
|
|
# Read immediate two bytes after SOS DRVR to establish first offset value.
|
|
offset = readUnpack(2, type = 'b')
|
|
print "The first offset value is", hex(offset)
|
|
drivers = 0 ## This is to keep a running total of drivers.
|
|
drivers_dict = dict() ## Intialize a dictionary to hold the drivers.
|
|
|
|
### Begin an indefinite loop that will come around until we
|
|
### encounter FF which indicates the last driver.
|
|
|
|
while offset != 65535 :
|
|
SOSfile.seek(offset,1)
|
|
## print "DEBUG: This is our new position in the file: ", hex(SOSfile.tell())
|
|
offset = readUnpack(2, type = 'b')
|
|
if offset == 0 : ## Check to see if we're at the beginning of a new driver.
|
|
drivers = drivers + 1 ## And add to count of found drivers.
|
|
offset = readUnpack(2, type = 'b')
|
|
drivers_dict ['Driver_'+str(drivers)] = dict([('Offset', hex(offset))])
|
|
## print 'DEBUG: New offset is: ' , hex(offset)
|
|
|
|
SOSfile.close()
|
|
print 'Total drivers encountered: ', drivers
|
|
print drivers_dict
|