mirror of
https://github.com/python-pillow/Pillow.git
synced 2024-12-26 01:46:18 +03:00
Various health fixes
This commit is contained in:
Andrew Murray
parent
b213f63c59
commit
f55f2d13cb
196
PIL/OleFileIO.py
196
PIL/OleFileIO.py
|
|
@ -399,12 +399,12 @@ WORD_CLSID = "00020900-0000-0000-C000-000000000046"
|
||||||
#TODO: check Excel, PPT, ...
|
#TODO: check Excel, PPT, ...
|
||||||
|
|
||||||
#[PL]: Defect levels to classify parsing errors - see OleFileIO._raise_defect()
|
#[PL]: Defect levels to classify parsing errors - see OleFileIO._raise_defect()
|
||||||
DEFECT_UNSURE = 10 # a case which looks weird, but not sure it's a defect
|
DEFECT_UNSURE = 10 # a case which looks weird, but not sure it's a defect
|
||||||
DEFECT_POTENTIAL = 20 # a potential defect
|
DEFECT_POTENTIAL = 20 # a potential defect
|
||||||
DEFECT_INCORRECT = 30 # an error according to specifications, but parsing
|
DEFECT_INCORRECT = 30 # an error according to specifications, but parsing
|
||||||
# can go on
|
# can go on
|
||||||
DEFECT_FATAL = 40 # an error which cannot be ignored, parsing is
|
DEFECT_FATAL = 40 # an error which cannot be ignored, parsing is
|
||||||
# impossible
|
# impossible
|
||||||
|
|
||||||
# Minimal size of an empty OLE file, with 512-bytes sectors = 1536 bytes
|
# Minimal size of an empty OLE file, with 512-bytes sectors = 1536 bytes
|
||||||
# (this is used in isOleFile and OleFile.open)
|
# (this is used in isOleFile and OleFile.open)
|
||||||
|
|
@ -504,20 +504,20 @@ def _clsid(clsid):
|
||||||
|
|
||||||
|
|
||||||
def filetime2datetime(filetime):
|
def filetime2datetime(filetime):
|
||||||
"""
|
"""
|
||||||
convert FILETIME (64 bits int) to Python datetime.datetime
|
convert FILETIME (64 bits int) to Python datetime.datetime
|
||||||
"""
|
"""
|
||||||
# TODO: manage exception when microseconds is too large
|
# TODO: manage exception when microseconds is too large
|
||||||
# inspired from http://code.activestate.com/recipes/511425-filetime-to-datetime/
|
# inspired from http://code.activestate.com/recipes/511425-filetime-to-datetime/
|
||||||
_FILETIME_null_date = datetime.datetime(1601, 1, 1, 0, 0, 0)
|
_FILETIME_null_date = datetime.datetime(1601, 1, 1, 0, 0, 0)
|
||||||
#debug('timedelta days=%d' % (filetime//(10*1000000*3600*24)))
|
#debug('timedelta days=%d' % (filetime//(10*1000000*3600*24)))
|
||||||
return _FILETIME_null_date + datetime.timedelta(microseconds=filetime//10)
|
return _FILETIME_null_date + datetime.timedelta(microseconds=filetime//10)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#=== CLASSES ==================================================================
|
#=== CLASSES ==================================================================
|
||||||
|
|
||||||
class OleMetadata:
|
class OleMetadata(object):
|
||||||
"""
|
"""
|
||||||
class to parse and store metadata from standard properties of OLE files.
|
class to parse and store metadata from standard properties of OLE files.
|
||||||
|
|
||||||
|
|
@ -803,7 +803,7 @@ class _OleStream(io.BytesIO):
|
||||||
|
|
||||||
#--- _OleDirectoryEntry -------------------------------------------------------
|
#--- _OleDirectoryEntry -------------------------------------------------------
|
||||||
|
|
||||||
class _OleDirectoryEntry:
|
class _OleDirectoryEntry(object):
|
||||||
|
|
||||||
"""
|
"""
|
||||||
OLE2 Directory Entry
|
OLE2 Directory Entry
|
||||||
|
|
@ -1064,7 +1064,7 @@ class _OleDirectoryEntry:
|
||||||
|
|
||||||
#--- OleFileIO ----------------------------------------------------------------
|
#--- OleFileIO ----------------------------------------------------------------
|
||||||
|
|
||||||
class OleFileIO:
|
class OleFileIO(object):
|
||||||
"""
|
"""
|
||||||
OLE container object
|
OLE container object
|
||||||
|
|
||||||
|
|
@ -1935,12 +1935,12 @@ class OleFileIO:
|
||||||
nb_sectors = (size + (self.sectorsize-1)) // self.sectorsize
|
nb_sectors = (size + (self.sectorsize-1)) // self.sectorsize
|
||||||
debug('nb_sectors = %d' % nb_sectors)
|
debug('nb_sectors = %d' % nb_sectors)
|
||||||
for i in range(nb_sectors):
|
for i in range(nb_sectors):
|
||||||
## try:
|
# try:
|
||||||
## self.fp.seek(offset + self.sectorsize * sect)
|
# self.fp.seek(offset + self.sectorsize * sect)
|
||||||
## except:
|
# except:
|
||||||
## debug('sect=%d, seek=%d' %
|
# debug('sect=%d, seek=%d' %
|
||||||
## (sect, offset+self.sectorsize*sect))
|
# (sect, offset+self.sectorsize*sect))
|
||||||
## raise IOError('OLE sector index out of range')
|
# raise IOError('OLE sector index out of range')
|
||||||
# extract one sector from data, the last one being smaller:
|
# extract one sector from data, the last one being smaller:
|
||||||
if i<(nb_sectors-1):
|
if i<(nb_sectors-1):
|
||||||
data_sector = data [i*self.sectorsize : (i+1)*self.sectorsize]
|
data_sector = data [i*self.sectorsize : (i+1)*self.sectorsize]
|
||||||
|
|
@ -2071,7 +2071,7 @@ class OleFileIO:
|
||||||
"""
|
"""
|
||||||
#REFERENCE: [MS-OLEPS] https://msdn.microsoft.com/en-us/library/dd942421.aspx
|
#REFERENCE: [MS-OLEPS] https://msdn.microsoft.com/en-us/library/dd942421.aspx
|
||||||
# make sure no_conversion is a list, just to simplify code below:
|
# make sure no_conversion is a list, just to simplify code below:
|
||||||
if no_conversion == None:
|
if no_conversion is None:
|
||||||
no_conversion = []
|
no_conversion = []
|
||||||
# stream path as a string to report exceptions:
|
# stream path as a string to report exceptions:
|
||||||
streampath = filename
|
streampath = filename
|
||||||
|
|
@ -2226,8 +2226,6 @@ class OleFileIO:
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
|
|
||||||
import sys
|
|
||||||
|
|
||||||
# [PL] display quick usage info if launched from command-line
|
# [PL] display quick usage info if launched from command-line
|
||||||
if len(sys.argv) <= 1:
|
if len(sys.argv) <= 1:
|
||||||
print('olefile version %s %s - %s' % (__version__, __date__, __author__))
|
print('olefile version %s %s - %s' % (__version__, __date__, __author__))
|
||||||
|
|
@ -2247,55 +2245,55 @@ For more information, see http://www.decalage.info/olefile
|
||||||
|
|
||||||
check_streams = False
|
check_streams = False
|
||||||
for filename in sys.argv[1:]:
|
for filename in sys.argv[1:]:
|
||||||
## try:
|
# try:
|
||||||
# OPTIONS:
|
# OPTIONS:
|
||||||
if filename == '-d':
|
if filename == '-d':
|
||||||
# option to switch debug mode on:
|
# option to switch debug mode on:
|
||||||
set_debug_mode(True)
|
set_debug_mode(True)
|
||||||
continue
|
continue
|
||||||
if filename == '-c':
|
if filename == '-c':
|
||||||
# option to switch check streams mode on:
|
# option to switch check streams mode on:
|
||||||
check_streams = True
|
check_streams = True
|
||||||
continue
|
continue
|
||||||
|
|
||||||
ole = OleFileIO(filename)#, raise_defects=DEFECT_INCORRECT)
|
ole = OleFileIO(filename)#, raise_defects=DEFECT_INCORRECT)
|
||||||
print("-" * 68)
|
print("-" * 68)
|
||||||
print(filename)
|
print(filename)
|
||||||
print("-" * 68)
|
print("-" * 68)
|
||||||
ole.dumpdirectory()
|
ole.dumpdirectory()
|
||||||
|
for streamname in ole.listdir():
|
||||||
|
if streamname[-1][0] == "\005":
|
||||||
|
print(streamname, ": properties")
|
||||||
|
props = ole.getproperties(streamname, convert_time=True)
|
||||||
|
props = sorted(props.items())
|
||||||
|
for k, v in props:
|
||||||
|
#[PL]: avoid to display too large or binary values:
|
||||||
|
if isinstance(v, (basestring, bytes)):
|
||||||
|
if len(v) > 50:
|
||||||
|
v = v[:50]
|
||||||
|
if isinstance(v, bytes):
|
||||||
|
# quick and dirty binary check:
|
||||||
|
for c in (1,2,3,4,5,6,7,11,12,14,15,16,17,18,19,20,
|
||||||
|
21,22,23,24,25,26,27,28,29,30,31):
|
||||||
|
if c in bytearray(v):
|
||||||
|
v = '(binary data)'
|
||||||
|
break
|
||||||
|
print(" ", k, v)
|
||||||
|
|
||||||
|
if check_streams:
|
||||||
|
# Read all streams to check if there are errors:
|
||||||
|
print('\nChecking streams...')
|
||||||
for streamname in ole.listdir():
|
for streamname in ole.listdir():
|
||||||
if streamname[-1][0] == "\005":
|
# print name using repr() to convert binary chars to \xNN:
|
||||||
print(streamname, ": properties")
|
print('-', repr('/'.join(streamname)),'-', end=' ')
|
||||||
props = ole.getproperties(streamname, convert_time=True)
|
st_type = ole.get_type(streamname)
|
||||||
props = sorted(props.items())
|
if st_type == STGTY_STREAM:
|
||||||
for k, v in props:
|
print('size %d' % ole.get_size(streamname))
|
||||||
#[PL]: avoid to display too large or binary values:
|
# just try to read stream in memory:
|
||||||
if isinstance(v, (basestring, bytes)):
|
ole.openstream(streamname)
|
||||||
if len(v) > 50:
|
else:
|
||||||
v = v[:50]
|
print('NOT a stream : type=%d' % st_type)
|
||||||
if isinstance(v, bytes):
|
print()
|
||||||
# quick and dirty binary check:
|
|
||||||
for c in (1,2,3,4,5,6,7,11,12,14,15,16,17,18,19,20,
|
|
||||||
21,22,23,24,25,26,27,28,29,30,31):
|
|
||||||
if c in bytearray(v):
|
|
||||||
v = '(binary data)'
|
|
||||||
break
|
|
||||||
print(" ", k, v)
|
|
||||||
|
|
||||||
if check_streams:
|
|
||||||
# Read all streams to check if there are errors:
|
|
||||||
print('\nChecking streams...')
|
|
||||||
for streamname in ole.listdir():
|
|
||||||
# print name using repr() to convert binary chars to \xNN:
|
|
||||||
print('-', repr('/'.join(streamname)),'-', end=' ')
|
|
||||||
st_type = ole.get_type(streamname)
|
|
||||||
if st_type == STGTY_STREAM:
|
|
||||||
print('size %d' % ole.get_size(streamname))
|
|
||||||
# just try to read stream in memory:
|
|
||||||
ole.openstream(streamname)
|
|
||||||
else:
|
|
||||||
print('NOT a stream : type=%d' % st_type)
|
|
||||||
print()
|
|
||||||
|
|
||||||
## for streamname in ole.listdir():
|
## for streamname in ole.listdir():
|
||||||
## # print name using repr() to convert binary chars to \xNN:
|
## # print name using repr() to convert binary chars to \xNN:
|
||||||
|
|
@ -2303,34 +2301,34 @@ For more information, see http://www.decalage.info/olefile
|
||||||
## print(ole.getmtime(streamname))
|
## print(ole.getmtime(streamname))
|
||||||
## print()
|
## print()
|
||||||
|
|
||||||
print('Modification/Creation times of all directory entries:')
|
print('Modification/Creation times of all directory entries:')
|
||||||
for entry in ole.direntries:
|
for entry in ole.direntries:
|
||||||
if entry is not None:
|
if entry is not None:
|
||||||
print('- %s: mtime=%s ctime=%s' % (entry.name,
|
print('- %s: mtime=%s ctime=%s' % (entry.name,
|
||||||
entry.getmtime(), entry.getctime()))
|
entry.getmtime(), entry.getctime()))
|
||||||
print()
|
print()
|
||||||
|
|
||||||
# parse and display metadata:
|
# parse and display metadata:
|
||||||
meta = ole.get_metadata()
|
meta = ole.get_metadata()
|
||||||
meta.dump()
|
meta.dump()
|
||||||
print()
|
print()
|
||||||
#[PL] Test a few new methods:
|
#[PL] Test a few new methods:
|
||||||
root = ole.get_rootentry_name()
|
root = ole.get_rootentry_name()
|
||||||
print('Root entry name: "%s"' % root)
|
print('Root entry name: "%s"' % root)
|
||||||
if ole.exists('worddocument'):
|
if ole.exists('worddocument'):
|
||||||
print("This is a Word document.")
|
print("This is a Word document.")
|
||||||
print("type of stream 'WordDocument':", ole.get_type('worddocument'))
|
print("type of stream 'WordDocument':", ole.get_type('worddocument'))
|
||||||
print("size :", ole.get_size('worddocument'))
|
print("size :", ole.get_size('worddocument'))
|
||||||
if ole.exists('macros/vba'):
|
if ole.exists('macros/vba'):
|
||||||
print("This document may contain VBA macros.")
|
print("This document may contain VBA macros.")
|
||||||
|
|
||||||
# print parsing issues:
|
# print parsing issues:
|
||||||
print('\nNon-fatal issues raised during parsing:')
|
print('\nNon-fatal issues raised during parsing:')
|
||||||
if ole.parsing_issues:
|
if ole.parsing_issues:
|
||||||
for exctype, msg in ole.parsing_issues:
|
for exctype, msg in ole.parsing_issues:
|
||||||
print('- %s: %s' % (exctype.__name__, msg))
|
print('- %s: %s' % (exctype.__name__, msg))
|
||||||
else:
|
else:
|
||||||
print('None')
|
print('None')
|
||||||
## except IOError as v:
|
## except IOError as v:
|
||||||
## print("***", "cannot read", file, "-", v)
|
## print("***", "cannot read", file, "-", v)
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue
Block a user