Driv3rs/3Slurp.py

147 lines
9.3 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)
text_unpacked = unpack('%ss' % bytes, SOS)
return ''.join(text_unpacked)
if options.get("type") == 'b':
#print 'DEBUG: In b', bytes
SOS = SOSfile.read(bytes)
#print 'DEBUG: In b2', bytes, SOS
offset_unpacked = unpack ('< H', SOS)
#print 'DEBUG: In b3', bytes, SOS
return int('.'.join(str(x) for x in offset_unpacked))
if options.get("type") == '1':
SOS = SOSfile.read(bytes)
offset_unpacked = unpack ('< B', SOS)
return int(ord(SOS))
def nibblize(byte, **options):
if options.get("direction") == 'high':
return str(int(hex(byte >> 4), 0))
if options.get("direction") == 'low':
return str(int(hex(byte & 0x0F), 0))
dev_types ={273: 'Character Device, Write-Only, Formatter',
321: 'Character Device, Write-Only, RS232 Printer', # dictionary for types and subtypes
577: 'Character Device, Write-Only, Silentype',
833: 'Character Device, Write-Only, Parallel Printer',
323: 'Character Device, Write-Only, Sound Port',
353: 'Character Device, Read-Write, System Console',
354: 'Character Device, Read-Write, Graphics Screen',
355: 'Character Device, Read-Write, Onboard RS232',
356: 'Character Device, Read-Write, Parallel Card',
481: 'Block Device, Disk ///',
721: 'Block Device, PROFile',
4337: 'Block Device, CFFA3000'}
mfgs = {17491: 'David Schmidt'} # dictionary for manufacturers
#Clear SCREEN
print("\033c");
#Is File a SOS DRIVER file?
SOSfile = open('SOSCFFA.DRIVER', 'rb')
filetype = readUnpack(8, type = 't')
if filetype == 'SOS DRVR':
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()
rel_offset = readUnpack(2, type = 'b') # read two bytes after SOS DRVR to establish first rel_offset value
print "The first relative offset value is", rel_offset, hex(rel_offset) # print out for debug
drivers = 0 # this is to keep a running total of drivers.
drivers_list=[] # initialize a list to hold dictionaries.
loop = True # set True for major driver loop
while loop : # establish loop to find major drivers
driver = {} # intialize a dictionary to hold vaules as we loop
SOSfile.seek(rel_offset,1) # jump to location of first driver
driver['location'] = SOSfile.tell() # add to driver dictionary 52c
rel_offset = readUnpack(2, type = 'b') # 0000 comment length
if rel_offset == 0xFFFF: # if we encounter FFFF, no more drivers
loop = False # set loop to False to cancel while
else :
drivers_list.append(driver) # add to drivers_list list
SOSfile.seek(rel_offset,1) # seek forward based on contents of rel_offset
rel_offset = readUnpack(2, type = 'b') # read 2 bytes and stick into rel_offset
SOSfile.seek(rel_offset,1) # seek forward based on contents of rel_offset
rel_offset = readUnpack(2, type = 'b') # read 2 bytes and stick into rel_offset
for i in range(0,len(drivers_list)): # begin loop to grab information from DIB
SOSfile.seek(drivers_list[i]['location'],0) # reference SOF and seek to beginning of major driver
comment_len = readUnpack(2, type = 'b') # grab comment length (2 bytes)
drivers_list[i]['comment_len'] = comment_len # store comment length
if comment_len != 0: # if comment length is not 0000
comment_txt = readUnpack(comment_len, type = 't') # comment_len bytes as text
drivers_list[i]['comment_txt'] = comment_txt # place comment in dictionary
else:
drivers_list[i]['comment_txt'] = '' # else enter comment as nothing
SOSfile.seek(2,1) # skip two bytes immediately following comment or 0000
link_ptr = readUnpack(2, type = 'b') # grab link pointer
drivers_list[i]['link_ptr'] = link_ptr # store link pointer
entry = readUnpack(2, type = 'b') # grab entry field
drivers_list[i]['entry'] = entry # store entry field
name_len = readUnpack(1, type = '1') # grab one byte for name length
drivers_list[i]['name_len'] = name_len # store name length
name = readUnpack(name_len, type = 't') # read name based on length
drivers_list[i]['name'] = name # store name
SOSfile.seek(15 - name_len,1) # skip past remainder of 15-byte long name field
flag = readUnpack(1, type = '1') # grab Flag byte
if flag == 192: # is flag 0xc0?
drivers_list[i]['flag'] = 'ACTIVE, Load on Boundary' # yes, ACTIVE and load on boundary
elif flag == 128: # is flag 0x80?
drivers_list[i]['flag'] = 'ACTIVE' # insert Flag into dictionary with value ACTIVE
else: # otherwise...
drivers_list[i]['flag'] = 'INACTIVE' # insert Flag into dictionary with value INACTIVE
slot_num = readUnpack(1, type = '1') # grab slot number
if slot_num == 0: # check if device has no slot number...
drivers_list[i]['slot_num'] = 'None' # ... and indicate so.
else: # otherwise...
drivers_list[i]['slot_num'] = slot_num # ... enter slot number into the dictionary
unit = readUnpack(1, type = '1') # get the unit byte
drivers_list[i]['unit'] = unit # store unit in dictionary
dev_type = readUnpack(2, type ='b')
try:
drivers_list[i]['dev_type'] = dev_types[dev_type] # insert device type into dictionary
except:
drivers_list[i]['dev_type'] = 'Unknown'
SOSfile.seek(1,1)
block_num = readUnpack(2, type = 'b') # get block_num
if block_num != 0: # is block_num not zero?
drivers_list[i]['block_num'] = block_num # yes, store retrieved block_num value
else: # otherwise...
drivers_list[i]['block_num'] = 'Character Device or Undefined' # log field as char device or undefined
mfg = readUnpack(2, type = 'b') # read mfg bytes
try:
drivers_list[i]['mfg'] = mfgs[mfg] # try to match key and place in dictionary
except:
if 1 <= mfg <= 31: # if exception, check if mfg is between 1 and 31
drivers_list[i]['mfg'] = 'Apple Computer' # yes, this is Apple Computer
else:
drivers_list[i]['mfg'] = 'Unknown' # no, this is unknown. log as such
atell = SOSfile.tell()
ver_byte0 = readUnpack(1, type = '1') # grab the V/v0 numbers | V = high-nibble, v0 = low-nibble
ver_byte1 = readUnpack(1, type = '1') # grab the v1/Q numbers | v1 = high-nibble, Q = low-nibble
V = nibblize(ver_byte1, direction = 'high') # grab major version number as string
v0 = nibblize(ver_byte1, direction = 'low') # grab first minor version number (v0) as string
v1 = nibblize(ver_byte0, direction = 'high') # grab second minor version number (v1) as string
Q = nibblize(ver_byte0, direction = 'low') # grab qualifier number. We only care about A (Alpha), B (Beta), or E (Experimental)
drivers_list[i]['version'] = V + '.' + v0 + v1 + Q
#print drivers_list[i]
print drivers_list[i]['version']
SOSfile.close()