KaneAI - World’s First E2E Software Testing Agent.Book a Demo 
How to use _raise_exception method in Testify
Best Python code snippet using Testify_python
pyole.py
Source:pyole.py 
...42 43 ole_logger = logging.getLogger('ole.logger')44 def __init__(self):45 pass46 def _raise_exception(self, error):47 #self.ole_logger.error(error)48 self.ole_logger.warning(error)49 raise Exception(error)50 def _filetime_to_datetime(self, microseconds):51 seconds, microseconds = divmod(microseconds/10, 1000000)52 days, seconds = divmod(seconds, 86400)53 date_time = datetime.datetime(1601, 1, 1, 0, 0, 0) + datetime.timedelta(days, seconds, microseconds)54 return str(date_time)55class OLEHeader(OLEBase):56 Signature = ''57 CLSID = ''58 MinorVersion = 059 MajorVersion = 060 ByteOrder = 061 SectorShift = 062 MiniSectorShift = 063 Reserved = ''64 NumberOfDirectorySectors = 065 NumberOfFATSectors = 066 FirstDirecotrySector = 067 TransactionSignatureNumber = 068 MiniStreamCutoffSize = 069 FirstMiniFATSector = 070 NumberOfMiniFATSectors = 071 FirstDIFATSector = 072 NumberOfDIFATSectors = 073 DIFAT = list()74 def __init__(self, data):75 76 self.Signature = ''77 self.CLSID = ''78 self.MinorVersion = 079 self.MajorVersion = 080 self.ByteOrder = 081 self.SectorShift = 082 self.MiniSectorShift = 083 self.Reserved = ''84 self.NumberOfDirectorySectors = 085 self.NumberOfFATSectors = 086 self.FirstDirecotrySector = 087 self.TransactionSignatureNumber = 088 self.MiniStreamCutoffSize = 089 self.FirstMiniFATSector = 090 self.NumberOfMiniFATSectors = 091 self.FirstDIFATSector = 092 self.NumberOfDIFATSectors = 093 self.DIFAT = list()94 95 self.Signature = data[0x00:0x08]96 self.ole_logger.debug('OLEHeader.Signature: ' + self.Signature.encode('hex').upper())97 if self.Signature != '\xD0\xCF\x11\xE0\xA1\xB1\x1A\xE1':98 self._raise_exception('OLEHeader.Signature verify failed.')99 100 self.CLSID = data[0x08:0x18]101 self.ole_logger.debug('OLEHeader.CLSID: ' + self.CLSID.encode('hex').upper())102 if self.CLSID != '\x00' * 16:103 self.ole_logger.warning('OLEHeader.CLSID is not null.')104 self.MinorVersion = struct.unpack('<H', data[0x18:0x1A])[0]105 self.ole_logger.debug('OLEHeader.MinorVersion: ' + str(hex(self.MinorVersion)))106 self.MajorVersion = struct.unpack('<H', data[0x1A:0x1C])[0]107 self.ole_logger.debug('OLEHeader.MajorVersion: ' + str(hex(self.MajorVersion)))108 if self.MajorVersion != 0x03 and self.MajorVersion != 0x04:109 self._raise_exception('OLEHeader.MajorVersion has an abnormal value.')110 self.ByteOrder = struct.unpack('<H', data[0x1C:0x1E])[0]111 if self.ByteOrder == 0xFFFE:112 self.ole_logger.debug('OLEHeader.ByteOrder: ' + str(hex(self.ByteOrder)) + ' (little-endian)')113 else:114 self.ole_logger.debug('OLEHeader.ByteOrder: ' + str(hex(self.ByteOrder)))115 self._raise_exception('OLEHeader.ByteOrder has an abnormal value.')116 self.SectorShift = struct.unpack('<H', data[0x1E:0x20])[0]117 if self.SectorShift == 0x09:118 self.ole_logger.debug('OLEHeader.SectorShift: ' + str(hex(self.SectorShift)) + ' (512 bytes)')119 elif self.SectorShift == 0x0C:120 self.ole_logger.debug('OLEHeader.SectorShift: ' + str(hex(self.SectorShift)) + ' (4096 bytes)')121 else:122 self.ole_logger.debug('OLEHeader.SectorShift: ' + str(hex(self.SectorShift)))123 self._raise_exception('OLEHeader.SectorShift has an abnormal value.')124 self.MiniSectorShift = struct.unpack('<H', data[0x20:0x22])[0]125 if self.MiniSectorShift == 0x06:126 self.ole_logger.debug('OLEHeader.MiniSectorShift: ' + str(hex(self.MiniSectorShift)) + ' (64 bytes)')127 else:128 self.ole_logger.debug('OLEHeader.MiniSectorShift: ' + str(hex(self.MiniSectorShift)))129 self._raise_exception('OLEHeader.MiniSectorShift has an abnormal value.')130 self.Reserved = data[0x22:0x28]131 self.ole_logger.debug('OLEHeader.Reserved: ' + self.Reserved.encode('hex').upper())132 if self.Reserved != '\x00' * 6:133 self.ole_logger.warning('OLEHeader.Reserved is not all zeros.')134 self.NumberOfDirectorySectors = struct.unpack('<I', data[0x28:0x2C])[0]135 self.ole_logger.debug('OLEHeader.NumberOfDirectorySectors: ' + str(hex(self.NumberOfDirectorySectors)))136 if self.NumberOfDirectorySectors != 0x0 and self.MajorVersion != 0x04:137 self._raise_exception('OLEHeader.NumberOfDirectorySectors has an abnormal value.')138 139 self.NumberOfFATSectors = struct.unpack('<I', data[0x2C:0x30])[0]140 self.ole_logger.debug('OLEHeader.NumberOfFATSectors: ' + str(hex(self.NumberOfFATSectors)))141 self.FirstDirecotrySector = struct.unpack('<I', data[0x30:0x34])[0]142 self.ole_logger.debug('OLEHeader.FirstDirecotrySector: ' + str(hex(self.FirstDirecotrySector)))143 if self.FirstDirecotrySector == 0:144 self._raise_exception('OLEHeader.FirstDirecotrySector is zero.')145 self.TransactionSignatureNumber = struct.unpack('<I', data[0x34:0x38])[0]146 self.ole_logger.debug('OLEHeader.TransactionSignatureNumber: ' + str(hex(self.TransactionSignatureNumber)))147 self.MiniStreamCutoffSize = struct.unpack('<I', data[0x38:0x3C])[0]148 self.ole_logger.debug('OLEHeader.MiniStreamCutoffSize: ' + str(hex(self.MiniStreamCutoffSize)))149 if self.MiniStreamCutoffSize != 0x1000:150 self._raise_exception('OLEHeader.MiniStreamCutoffSize has an abnormal value.')151 self.FirstMiniFATSector = struct.unpack('<I', data[0x3C:0x40])[0]152 self.ole_logger.debug('OLEHeader.FirstMiniFATSector: ' + str(hex(self.FirstMiniFATSector)))153 self.NumberOfMiniFATSectors = struct.unpack('<I', data[0x40:0x44])[0]154 self.ole_logger.debug('OLEHeader.NumberOfMiniFATSectors: ' + str(hex(self.NumberOfMiniFATSectors)))155 if self.NumberOfMiniFATSectors > 0 and self.FirstMiniFATSector == 0xFFFFFFFE:156 self._raise_exception('OLEHeader.NumberOfMiniFATSectors or OLEHeader.FirstMiniFATSector has an abnormal value.')157 self.FirstDIFATSector = struct.unpack('<I', data[0x44:0x48])[0]158 self.ole_logger.debug('OLEHeader.FirstDIFATSector: ' + str(hex(self.FirstDIFATSector)))159 self.NumberOfDIFATSectors = struct.unpack('<I', data[0x48:0x4C])[0]160 self.ole_logger.debug('OLEHeader.NumberOfDIFATSectors: ' + str(hex(self.NumberOfDIFATSectors)))161 if self.NumberOfDIFATSectors > 0 and self.FirstDIFATSector == 0xFFFFFFFE:162 self._raise_exception('OLEHeader.NumberOfDIFATSectors or OLEHeader.FirstDIFATSector has an abnormal value.')163 164 for i in range(0, 109):165 difat = struct.unpack('<I', data[0x4C+i*4:0x4C+i*4+4])[0]166 if difat == 0xFFFFFFFF:167 break168 self.ole_logger.debug('OLEHeader.DIFAT[' + str(i) + '] :' + str(hex(difat)))169 self.DIFAT.append(difat)170 i += 1171 for j in range(i, 109):172 difat = struct.unpack('<I', data[0x4C+j*4:0x4C+j*4+4])[0]173 if difat != 0xFFFFFFFF:174 self._raise_exception('OLEHeader.DIFAT[' + str(j) + '] has an abnormal value.')175class Directory(OLEBase):176 Name = ''177 NameLength = 0178 ObjectType = 0179 ColorFlag = 0180 LeftSiblingID = 0181 RightSiblingID = 0182 ChildID = 0183 CLSID = ''184 StateBits = 0185 CreationTime = ''186 ModifiedTime = ''187 StartingSector = 0188 StreamSize = 0189 def __init__(self, data):190 191 self.Name = ''192 self.NameLength = 0193 self.ObjectType = 0194 self.ColorFlag = 0195 self.LeftSiblingID = 0196 self.RightSiblingID = 0197 self.ChildID = 0198 self.CLSID = ''199 self.StateBits = 0200 self.CreationTime = ''201 self.ModifiedTime = ''202 self.StartingSector = 0203 self.StreamSize = 0204 self.Name = data[0:0x40].decode('utf-16').strip('\x00')205 self.ole_logger.debug('Dir.Name: ' + self.Name)206 self.NameLength = struct.unpack('<H', data[0x40:0x42])[0]207 self.ole_logger.debug('Dir.NameLength: ' + str(self.NameLength))208 209 if self.NameLength != len(self.Name)*2+2:210 self._raise_exception('DirectoryEntry.NameLength has a wrong value.')211 212 self.ObjectType = ord(data[0x42])213 if self.ObjectType == 0x00: 214 self.ole_logger.debug('Dir.ObjectType: ' + str(self.ObjectType) + ' (unallocated)')215 elif self.ObjectType == 0x01:216 self.ole_logger.debug('Dir.ObjectType: ' + str(self.ObjectType) + ' (storage object)')217 elif self.ObjectType == 0x02:218 self.ole_logger.debug('Dir.ObjectType: ' + str(self.ObjectType) + ' (stream object)')219 elif self.ObjectType == 0x05:220 self.ole_logger.debug('Dir.ObjectType: ' + str(self.ObjectType) + ' (root storage object)')221 else:222 self._raise_exception('DirectoryEntry.ObjectType has an abnormal value.')223 self.ColorFlag = ord(data[0x43])224 if self.ColorFlag == 0x00: 225 self.ole_logger.debug('Dir.ColorFlag: ' + str(self.ColorFlag) + ' (red)')226 elif self.ColorFlag == 0x01:227 self.ole_logger.debug('Dir.ColorFlag: ' + str(self.ColorFlag) + ' (black)')228 else:229 self._raise_exception('DirectoryEntry.ColorFlag has an abnormal value.')230 self.LeftSiblingID = struct.unpack('<I', data[0x44:0x48])[0]231 if self.LeftSiblingID >= 0 and self.LeftSiblingID <= 0xFFFFFFF9:232 self.ole_logger.debug('Dir.LeftSiblingID: ' + str(hex(self.LeftSiblingID)) + ' (REGSID)')233 elif self.LeftSiblingID == 0xFFFFFFFF:234 self.ole_logger.debug('Dir.LeftSiblingID: ' + str(hex(self.LeftSiblingID)) + ' (NOSTREAM)')235 else:236 self._raise_exception('DirectoryEntry.LeftSiblingID has an abnormal value.')237 self.RightSiblingID = struct.unpack('<I', data[0x48:0x4C])[0]238 if self.RightSiblingID >= 0 and self.RightSiblingID <= 0xFFFFFFF9:239 self.ole_logger.debug('Dir.RightSiblingID: ' + str(hex(self.RightSiblingID)) + ' (REGSID)')240 elif self.RightSiblingID == 0xFFFFFFFF:241 self.ole_logger.debug('Dir.LeftSiblingID: ' + str(hex(self.RightSiblingID)) + ' (NOSTREAM)')242 else:243 self._raise_exception('DirectoryEntry.RightSiblingID has an abnormal value.')244 self.ChildID = struct.unpack('<I', data[0x4C:0x50])[0]245 if self.ChildID >= 0 and self.ChildID <= 0xFFFFFFF9:246 self.ole_logger.debug('Dir.ChildID: ' + str(hex(self.ChildID)) + ' (REGSID)')247 elif self.ChildID == 0xFFFFFFFF:248 self.ole_logger.debug('Dir.ChildID: ' + str(hex(self.ChildID)) + ' (NOSTREAM)')249 else:250 self._raise_exception('DirectoryEntry.ChildID has an abnormal value.')251 self.CLSID = data[0x50:0x60]252 self.ole_logger.debug('Dir.CLSID: ' + self.CLSID.encode('hex'))253 self.StateBits = struct.unpack('<I', data[0x60:0x64])[0]254 self.ole_logger.debug('Dir.StateBits: ' + str(hex(self.StateBits)))255 self.CreationTime = struct.unpack('<Q', data[0x64:0x6C])[0]256 self.ole_logger.debug('Dir.CreationTime: ' + self._filetime_to_datetime(self.CreationTime))257 self.ModifiedTime = struct.unpack('<Q', data[0x6C:0x74])[0]258 self.ole_logger.debug('Dir.ModifiedTime: ' + self._filetime_to_datetime(self.ModifiedTime))259 self.StartingSector = struct.unpack('<I', data[0x74:0x78])[0]260 self.ole_logger.debug('Dir.StartingSector: ' + str(hex(self.StartingSector)))261 self.StreamSize = struct.unpack('<Q', data[0x78:0x80])[0]262 self.ole_logger.debug('Dir.StreamSize: ' + str(hex(self.StreamSize)))263class PropertyIdentifierAndOffset(OLEBase):264 PropertyIdentifier = 0265 Offset = 0266 def __init__(self, data):267 self.PropertyIdentifier = 0268 self.Offset = 0269 self.PropertyIdentifier = struct.unpack('<I', data[0:4])[0]270 self.ole_logger.debug('PropertyIdentifierAndOffset.PropertyIdentifier: ' + str(hex(self.PropertyIdentifier)))271 self.Offset = struct.unpack('<I', data[4:8])[0]272 self.ole_logger.debug('PropertyIdentifierAndOffset.Offset: ' + str(hex(self.Offset)))273class DocSummaryInfoPropertySet(OLEBase):274 Size = 0275 NumProperties = 0276 PropertyIdentifierAndOffset = list()277 Property = list()278 def __init__(self, data):279 280 self.Size = 0281 self.NumProperties = 0282 self.PropertyIdentifierAndOffset = list()283 self.Property = list()284 self.Size = struct.unpack('<I', data[0x00:0x04])[0]285 self.ole_logger.debug('DocSummaryInfoPropertySet.Size: ' + str(hex(self.Size)))286 self.NumProperties = struct.unpack('<I', data[0x04:0x08])[0]287 self.ole_logger.debug('DocSummaryInfoPropertySet.NumProperties: ' + str(hex(self.NumProperties)))288 for i in range(0, self.NumProperties):289 piao = PropertyIdentifierAndOffset(data[0x08+i*8:0x08+i*8+8])290 self.PropertyIdentifierAndOffset.append(piao)291 for i in range(0, self.NumProperties):292 if (i+1) < self.NumProperties:293 if self.PropertyIdentifierAndOffset[i].Offset < self.PropertyIdentifierAndOffset[i+1].Offset:294 property = data[self.PropertyIdentifierAndOffset[i].Offset:self.PropertyIdentifierAndOffset[i+1].Offset]295 else:296 self.ole_logger.warning('DocSummaryInfoPropertySet.PropertyIdentifierAndOffset.Offset is not in increasing order.')297 property = data[self.PropertyIdentifierAndOffset[i].Offset:self.Size]298 else:299 property = data[self.PropertyIdentifierAndOffset[i].Offset:self.Size]300 self.Property.append(property)301 302 for i in range(0, self.NumProperties):303 if self.PropertyIdentifierAndOffset[i].PropertyIdentifier == PIDDSI['GKPIDDSI_CODEPAGE']:304 type = struct.unpack('<H', self.Property[i][0x00:0x02])[0]305 self.ole_logger.debug('Property.GKPIDDSI_CODEPAGE.type: ' + str(hex(type)))306 if type != PropertyType['VT_I2']:307 self._raise_exception('Property.GKPIDDSI_CODEPAGE has an abnormal value.')308 codepage = struct.unpack('<H', self.Property[i][0x04:0x06])[0]309 self.ole_logger.debug('Property.GKPIDDSI_CODEPAGE: ' + str(hex(codepage)))310 continue311 if self.PropertyIdentifierAndOffset[i].PropertyIdentifier == PIDDSI['GKPIDDSI_COMPANY']:312 type = struct.unpack('<H', self.Property[i][0x00:0x02])[0]313 self.ole_logger.debug('Property.GKPIDDSI_COMPANY.type: ' + str(hex(type)))314 cch = struct.unpack('<I', self.Property[i][0x04:0x08])[0]315 self.ole_logger.debug('Property.GKPIDDSI_COMPANY.cch: ' + str(hex(cch)))316 if cch > 0x0000FFFF:317 self._raise_exception('Property.GKPIDDSI_COMPANY.cch has an abnormal value.')318 if type == PropertyType['VT_LPSTR']:319 company = self.Property[i][0x08:0x08+cch]320 elif type == PropertyType['VT_LPWSTR']:321 company = self.Property[i][0x08:0x08+cch*2].decode('utf-16')322 else:323 self._raise_exception('Property.GKPIDDSI_COMPANY has an abnormal value.')324 self.ole_logger.debug('Property.GKPIDDSI_COMPANY: ' + company)325 continue326 if self.PropertyIdentifierAndOffset[i].PropertyIdentifier == PIDDSI['GKPIDDSI_LINECOUNT']:327 type = struct.unpack('<H', self.Property[i][0x00:0x02])[0]328 self.ole_logger.debug('Property.GKPIDDSI_LINECOUNT.type: ' + str(hex(type)))329 if type != PropertyType['VT_I4']:330 self._raise_exception('Property.GKPIDDSI_LINECOUNT has an abnormal value.')331 linecount = struct.unpack('<I', self.Property[i][0x04:0x08])[0]332 self.ole_logger.debug('Property.GKPIDDSI_LINECOUNT: ' + str(hex(linecount)))333 continue334 if self.PropertyIdentifierAndOffset[i].PropertyIdentifier == PIDDSI['GKPIDDSI_PARACOUNT']:335 type = struct.unpack('<H', self.Property[i][0x00:0x02])[0]336 self.ole_logger.debug('Property.GKPIDDSI_PARACOUNT.type: ' + str(hex(type)))337 if type != PropertyType['VT_I4']:338 self._raise_exception('Property.GKPIDDSI_PARACOUNT has an abnormal value.')339 pagecount = struct.unpack('<I', self.Property[i][0x04:0x08])[0]340 self.ole_logger.debug('Property.GKPIDDSI_PARACOUNT: ' + str(hex(pagecount)))341 continue342 if self.PropertyIdentifierAndOffset[i].PropertyIdentifier == PIDDSI['GKPIDDSI_CCHWITHSPACES']:343 type = struct.unpack('<H', self.Property[i][0x00:0x02])[0]344 self.ole_logger.debug('Property.GKPIDDSI_CCHWITHSPACES.type: ' + str(hex(type)))345 if type != PropertyType['VT_I4']:346 self._raise_exception('Property.GKPIDDSI_CCHWITHSPACES has an abnormal value.')347 pagecount = struct.unpack('<I', self.Property[i][0x04:0x08])[0]348 self.ole_logger.debug('Property.GKPIDDSI_CCHWITHSPACES: ' + str(hex(pagecount)))349 continue350 if self.PropertyIdentifierAndOffset[i].PropertyIdentifier == PIDDSI['GKPIDDSI_VERSION']:351 type = struct.unpack('<H', self.Property[i][0x00:0x02])[0]352 self.ole_logger.debug('Property.GKPIDDSI_VERSION.type: ' + str(hex(type)))353 if type != PropertyType['VT_I4']:354 self._raise_exception('Property.GKPIDDSI_VERSION has an abnormal value.')355 minorversion = struct.unpack('<H', self.Property[i][0x04:0x06])[0]356 majorverson= struct.unpack('<H', self.Property[i][0x06:0x08])[0]357 if majorverson == 0:358 self._raise_exception('Property.GKPIDDSI_VERSION.MajorVersion has an abnormal value.')359 self.ole_logger.debug('Property.GKPIDDSI_VERSION.MajorVersion: ' + str(hex(majorverson)))360 self.ole_logger.debug('Property.GKPIDDSI_VERSION.MinorVersion: ' + str(hex(minorversion)))361 continue362 if self.PropertyIdentifierAndOffset[i].PropertyIdentifier == PIDDSI['GKPIDDSI_DOCPARTS']:363 type = struct.unpack('<H', self.Property[i][0x00:0x02])[0]364 self.ole_logger.debug('Property.GKPIDDSI_DOCPARTS.type: ' + str(hex(type)))365 celements = struct.unpack('<I', self.Property[i][0x04:0x08])[0]366 self.ole_logger.debug('Property.GKPIDDSI_DOCPARTS.vtValue.cElements: ' + str(hex(celements)))367 if type == (PropertyType['VT_VECTOR'] | PropertyType['VT_LPSTR']): 368 offset = 0369 for j in range(0, celements):370 cch = struct.unpack('<I', self.Property[i][0x08+offset:0x0C+offset])[0]371 self.ole_logger.debug('Property.GKPIDDSI_DOCPARTS.vtValue.rgString[' + str(j) + '].cch: ' + str(hex(cch)))372 if cch > 0x0000FFFF:373 self._raise_exception('Property.GKPIDDSI_DOCPARTS.vtValue.rgString[' + str(j) + '].cch has an abnormal value.')374 value = self.Property[i][0x0C+offset:0x0C+offset+cch]375 self.ole_logger.debug('Property.GKPIDDSI_DOCPARTS.vtValue.rgString[' + str(j) + ']: ' + value.encode('hex'))376 offset = offset + 4 + cch377 elif type == (PropertyType['VT_VECTOR'] | PropertyType['VT_LPWSTR']):378 offset = 0379 for j in range(0, celements):380 cch = struct.unpack('<I', self.Property[i][0x08+offset:0x0C+offset])[0]381 self.ole_logger.debug('Property.GKPIDDSI_DOCPARTS.vtValue.rgString[' + str(j) + '].cch: ' + str(hex(cch)))382 if cch > 0x0000FFFF:383 self._raise_exception('Property.GKPIDDSI_DOCPARTS.vtValue.rgString[' + str(j) + '].cch has an abnormal value.')384 value = self.Property[i][0x0C+offset:0x0C+offset+cch*2].decode('utf-16')385 self.ole_logger.debug('Property.GKPIDDSI_DOCPARTS.vtValue.rgString[' + str(j) + ']: ' + value.encode('hex'))386 offset = offset + 4 + cch*2387 else:388 self._raise_exception('Property.GKPIDDSI_DOCPARTS.type has an abnormal value.')389 continue390 391 if self.PropertyIdentifierAndOffset[i].PropertyIdentifier == PIDDSI['GKPIDDSI_HEADINGPAIR']:392 type = struct.unpack('<H', self.Property[i][0x00:0x02])[0]393 self.ole_logger.debug('Property.GKPIDDSI_HEADINGPAIR.type: ' + str(hex(type)))394 if type != (PropertyType['VT_VECTOR'] | PropertyType['VT_VARIANT']):395 self._raise_exception('Property.GKPIDDSI_HEADINGPAIR.type has an abnormal value.')396 celements = struct.unpack('<I', self.Property[i][0x04:0x08])[0]397 self.ole_logger.debug('Property.GKPIDDSI_HEADINGPAIR.vtValue.cElements: ' + str(hex(celements)))398 offset = 0399 for j in range(0, celements/2):400 strtype = struct.unpack('<H', self.Property[i][0x08+offset:0x0A+offset])[0]401 self.ole_logger.debug('Property.GKPIDDSI_HEADINGPAIR.vtValue.rgHeadingPairs[' + str(j) + '].headingString.type: ' + str(hex(strtype)))402 cch = struct.unpack('<I', self.Property[i][0x0C+offset:0x10+offset])[0]403 self.ole_logger.debug('Property.GKPIDDSI_HEADINGPAIR.vtValue.rgHeadingPairs[' + str(j) + '].headingString.cch: ' + str(hex(cch)))404 if cch > 0x0000FFFF:405 self._raise_exception('Property.GKPIDDSI_HEADINGPAIR.vtValue.rgHeadingPairs[' + str(j) + '].headingString.cch has an abnormal value.')406 if strtype == PropertyType['VT_LPSTR']:407 value = self.Property[i][0x10+offset:0x10+offset+cch]408 self.ole_logger.debug('Property.GKPIDDSI_HEADINGPAIR.vtValue.rgHeadingPairs[' + str(j) + '].headingString: ' + value)409 partstype = struct.unpack('<H', self.Property[i][0x10+offset+cch:0x10+offset+cch+0x02])[0]410 self.ole_logger.debug('Property.GKPIDDSI_HEADINGPAIR.vtValue.rgHeadingPairs[' + str(j) + '].headerParts.type: ' + str(hex(partstype)))411 if partstype != PropertyType['VT_I4']:412 self._raise_exception('Property.GKPIDDSI_HEADINGPAIR.vtValue.rgHeadingPairs[' + str(j) + '].headerParts.type has an abnormal value.')413 parts = struct.unpack('<I', self.Property[i][0x10+offset+cch+0x04:0x10+offset+cch+0x08])[0]414 self.ole_logger.debug('Property.GKPIDDSI_HEADINGPAIR.vtValue.rgHeadingPairs[' + str(j) + '].headerParts: ' + str(hex(parts)))415 offset = offset + 0x10 + cch416 elif strtype == PropertyType['VT_LPWSTR']:417 value = self.Property[i][0x10+offset:0x10+offset+cch*2].decode('utf-16')418 self.ole_logger.debug('Property.GKPIDDSI_HEADINGPAIR.vtValue.rgHeadingPairs[' + str(j) + '].headingString: ' + value)419 partstype = struct.unpack('<H', self.Property[i][0x10+offset+cch*2:0x10+offset+cch*2+0x02])[0]420 self.ole_logger.debug('Property.GKPIDDSI_HEADINGPAIR.vtValue.rgHeadingPairs[' + str(j) + '].headerParts.type: ' + str(hex(partstype)))421 if partstype != PropertyType['VT_I4']:422 self._raise_exception('Property.GKPIDDSI_HEADINGPAIR.vtValue.rgHeadingPairs[' + str(j) + '].headerParts.type has an abnormal value.')423 parts = struct.unpack('<I', self.Property[i][0x10+offset+cch*2+0x04:0x10+offset+cch*2+0x08])[0]424 self.ole_logger.debug('Property.GKPIDDSI_HEADINGPAIR.vtValue.rgHeadingPairs[' + str(j) + '].headerParts: ' + str(hex(parts)))425 offset = offset + 0x10 + cch*2426 else:427 self._raise_exception('Property.GKPIDDSI_HEADINGPAIR.vtValue.rgHeadingPairs[' + str(j) + '].headingString.type has an abnormal value.')428 continue429class DocSummaryInfo(OLEBase):430 byteOrder = 0431 version = 0432 sysId = 0433 OSMajorVersion = 0434 OSMinorVersion = 0435 OSType = 0436 applicationClsid = ''437 cSections = 0438 formatId1 = ''439 sectionOffset1 = 0440 formatId2 = ''441 sectionOffset2 = 0442 DocumentSummaryInfoPropertySet = None443 def __init__(self, data):444 self.byteOrder = 0445 self.version = 0446 self.sysId = 0447 self.OSMajorVersion = 0448 self.OSMinorVersion = 0449 self.OSType = 0450 self.applicationClsid = ''451 self.cSections = 0452 self.formatId1 = ''453 self.sectionOffset1 = 0454 self.formatId2 = ''455 self.sectionOffset2 = 0456 self.DocumentSummaryInfoPropertySet = None457 self.ole_logger.debug('######## DocumentSummaryInfo ########')458 self.byteOrder = struct.unpack('<H', data[0x00:0x02])[0]459 self.ole_logger.debug('DocumentSummaryInfo.byteOrder: ' + str(hex(self.byteOrder)))460 if self.byteOrder != 0xFFFE:461 self._raise_exception('DocumentSummaryInfo.byteOrder has an abnormal value.')462 self.version = struct.unpack('<H', data[0x02:0x04])[0]463 self.ole_logger.debug('DocumentSummaryInfo.version: ' + str(hex(self.version)))464 if self.version != 0 and self.version != 1:465 self._raise_exception('DocumentSummaryInfo.version has an abnormal value.')466 self.sysId = struct.unpack('<I', data[0x04:0x08])[0]467 self.OSMajorVersion = ord(data[0x04])468 self.ole_logger.debug('DocumentSummaryInfo.sysId.OSMajorVersion: ' + str(hex(self.OSMajorVersion)))469 self.OSMinorVersion = ord(data[0x05])470 self.ole_logger.debug('DocumentSummaryInfo.sysId.OSMinorVersion: ' + str(hex(self.OSMinorVersion)))471 self.OSType = struct.unpack('<H', data[0x06:0x08])[0]472 self.ole_logger.debug('DocumentSummaryInfo.sysId.OSType: ' + str(hex(self.OSType)))473 self.applicationClsid = data[0x08:0x18]474 self.ole_logger.debug('DocumentSummaryInfo.applicationClsid: ' + self.applicationClsid.encode('hex'))475 if self.applicationClsid != '\x00' * 0x10:476 self._raise_exception('DocumentSummaryInfo.applicationClsid has an abnormal value.')477 self.cSections = struct.unpack('<I', data[0x18:0x1C])[0]478 self.ole_logger.debug('DocumentSummaryInfo.cSections: ' + str(hex(self.cSections)))479 if self.cSections != 1 and self.cSections != 2:480 self._raise_exception('DocumentSummaryInfo.cSections has an abnormal value.')481 self.formatId1 = data[0x1C:0x2C]482 self.ole_logger.debug('DocumentSummaryInfo.rgIdOffset.IdOffsetElement-1.formatId: ' + self.formatId1.encode('hex'))483 if self.formatId1 != '\x02\xD5\xCD\xD5\x9C\x2E\x1B\x10\x93\x97\x08\x00\x2B\x2C\xF9\xAE':484 self._raise_exception('DocumentSummaryInfo.rgIdOffset.IdOffsetElement-1.formatId has an abnormal value.')485 self.sectionOffset1 = struct.unpack('<I', data[0x2C:0x30])[0]486 self.ole_logger.debug('DocumentSummaryInfo.rgIdOffset.IdOffsetElement-1.sectionOffset: ' + str(hex(self.sectionOffset1)))487 if self.cSections == 2:488 self.formatId2 = data[0x30:0x40]489 self.ole_logger.debug('DocumentSummaryInfo.rgIdOffset.IdOffsetElement-2.formatId: ' + self.formatId2.encode('hex'))490 if self.formatId2 != '\x05\xD5\xCD\xD5\x9C\x2E\x1B\x10\x93\x97\x08\x00\x2B\x2C\xF9\xAE':491 self._raise_exception('DocumentSummaryInfo.rgIdOffset.IdOffsetElement-2.formatId has an abnormal value.')492 self.sectionOffset2 = struct.unpack('<I', data[0x40:0x44])[0]493 self.ole_logger.debug('DocumentSummaryInfo.rgIdOffset.IdOffsetElement-2.sectionOffset: ' + str(hex(self.sectionOffset2)))494 self.DocumentSummaryInfoPropertySet = DocSummaryInfoPropertySet(data[self.sectionOffset1:])495class SummaryInfoPropertySet(OLEBase):496 Size = 0497 NumProperties = 0498 PropertyIdentifierAndOffset = list()499 Property = list()500 def __init__(self, data):501 502 self.Size = 0503 self.NumProperties = 0504 self.PropertyIdentifierAndOffset = list()505 self.Property = list()506 self.Size = struct.unpack('<I', data[0x00:0x04])[0]507 self.ole_logger.debug('SummaryInfoPropertySet.Size: ' + str(hex(self.Size)))508 self.NumProperties = struct.unpack('<I', data[0x04:0x08])[0]509 self.ole_logger.debug('SummaryInfoPropertySet.NumProperties: ' + str(hex(self.NumProperties)))510 for i in range(0, self.NumProperties):511 piao = PropertyIdentifierAndOffset(data[0x08+i*8:0x08+i*8+8])512 self.PropertyIdentifierAndOffset.append(piao)513 for i in range(0, self.NumProperties):514 if (i+1) < self.NumProperties:515 if self.PropertyIdentifierAndOffset[i].Offset < self.PropertyIdentifierAndOffset[i+1].Offset:516 property = data[self.PropertyIdentifierAndOffset[i].Offset:self.PropertyIdentifierAndOffset[i+1].Offset]517 else:518 self.ole_logger.warning('SummaryInfoPropertySet.PropertyIdentifierAndOffset.Offset is not in increasing order.')519 property = data[self.PropertyIdentifierAndOffset[i].Offset:self.Size]520 else:521 property = data[self.PropertyIdentifierAndOffset[i].Offset:self.Size]522 self.Property.append(property)523 524 for i in range(0, self.NumProperties):525 if self.PropertyIdentifierAndOffset[i].PropertyIdentifier == PIDSI['PIDSI_CODEPAGE']:526 type = struct.unpack('<H', self.Property[i][0x00:0x02])[0]527 self.ole_logger.debug('Property.PIDSI_CODEPAGE.type: ' + str(hex(type)))528 if type != PropertyType['VT_I2']:529 self._raise_exception('Property.PIDSI_CODEPAGE has an abnormal value.')530 codepage = struct.unpack('<H', self.Property[i][0x04:0x06])[0]531 self.ole_logger.debug('Property.PIDSI_CODEPAGE: ' + str(hex(codepage)))532 continue533 if self.PropertyIdentifierAndOffset[i].PropertyIdentifier == PIDSI['PIDSI_TITLE']:534 type = struct.unpack('<H', self.Property[i][0x00:0x02])[0]535 self.ole_logger.debug('Property.PIDSI_TITLE.type: ' + str(hex(type)))536 cch = struct.unpack('<I', self.Property[i][0x04:0x08])[0]537 self.ole_logger.debug('Property.PIDSI_TITLE.cch: ' + str(hex(cch)))538 if cch > 0x0000FFFF:539 self._raise_exception('Property.PIDSI_TITLE.cch has an abnormal value.')540 if type == PropertyType['VT_LPSTR']:541 data = self.Property[i][0x08:0x08+cch]542 elif type == PropertyType['VT_LPWSTR']:543 data = self.Property[i][0x08:0x08+cch*2].decode('utf-16')544 else:545 self._raise_exception('Property.PIDSI_TITLE has an abnormal value.')546 self.ole_logger.debug('Property.PIDSI_TITLE: ' + data)547 continue548 549 if self.PropertyIdentifierAndOffset[i].PropertyIdentifier == PIDSI['PIDSI_SUBJECT']:550 type = struct.unpack('<H', self.Property[i][0x00:0x02])[0]551 self.ole_logger.debug('Property.PIDSI_SUBJECT.type: ' + str(hex(type)))552 cch = struct.unpack('<I', self.Property[i][0x04:0x08])[0]553 self.ole_logger.debug('Property.PIDSI_SUBJECT.cch: ' + str(hex(cch)))554 if cch > 0x0000FFFF:555 self._raise_exception('Property.PIDSI_SUBJECT.cch has an abnormal value.')556 if type == PropertyType['VT_LPSTR']:557 data = self.Property[i][0x08:0x08+cch]558 elif type == PropertyType['VT_LPWSTR']:559 data = self.Property[i][0x08:0x08+cch*2].decode('utf-16')560 else:561 self._raise_exception('Property.PIDSI_SUBJECT has an abnormal value.')562 self.ole_logger.debug('Property.PIDSI_SUBJECT: ' + data)563 continue564 if self.PropertyIdentifierAndOffset[i].PropertyIdentifier == PIDSI['PIDSI_AUTHOR']:565 type = struct.unpack('<H', self.Property[i][0x00:0x02])[0]566 self.ole_logger.debug('Property.PIDSI_AUTHOR.type: ' + str(hex(type)))567 cch = struct.unpack('<I', self.Property[i][0x04:0x08])[0]568 self.ole_logger.debug('Property.PIDSI_AUTHOR.cch: ' + str(hex(cch)))569 if cch > 0x0000FFFF:570 self._raise_exception('Property.PIDSI_AUTHOR.cch has an abnormal value.')571 if type == PropertyType['VT_LPSTR']:572 data = self.Property[i][0x08:0x08+cch]573 elif type == PropertyType['VT_LPWSTR']:574 data = self.Property[i][0x08:0x08+cch*2].decode('utf-16')575 else:576 self._raise_exception('Property.PIDSI_AUTHOR has an abnormal value.')577 self.ole_logger.debug('Property.PIDSI_AUTHOR: ' + data)578 continue579 if self.PropertyIdentifierAndOffset[i].PropertyIdentifier == PIDSI['PIDSI_KEYWORDS']:580 type = struct.unpack('<H', self.Property[i][0x00:0x02])[0]581 self.ole_logger.debug('Property.PIDSI_KEYWORDS.type: ' + str(hex(type)))582 cch = struct.unpack('<I', self.Property[i][0x04:0x08])[0]583 self.ole_logger.debug('Property.PIDSI_KEYWORDS.cch: ' + str(hex(cch)))584 if cch > 0x0000FFFF:585 self._raise_exception('Property.PIDSI_KEYWORDS.cch has an abnormal value.')586 if type == PropertyType['VT_LPSTR']:587 data = self.Property[i][0x08:0x08+cch]588 elif type == PropertyType['VT_LPWSTR']:589 data = self.Property[i][0x08:0x08+cch*2].decode('utf-16')590 else:591 self._raise_exception('Property.PIDSI_KEYWORDS has an abnormal value.')592 self.ole_logger.debug('Property.PIDSI_KEYWORDS: ' + data)593 continue594 if self.PropertyIdentifierAndOffset[i].PropertyIdentifier == PIDSI['PIDSI_COMMENTS']:595 type = struct.unpack('<H', self.Property[i][0x00:0x02])[0]596 self.ole_logger.debug('Property.PIDSI_COMMENTS.type: ' + str(hex(type)))597 cch = struct.unpack('<I', self.Property[i][0x04:0x08])[0]598 self.ole_logger.debug('Property.PIDSI_COMMENTS.cch: ' + str(hex(cch)))599 if cch > 0x0000FFFF:600 self._raise_exception('Property.PIDSI_COMMENTS.cch has an abnormal value.')601 if type == PropertyType['VT_LPSTR']:602 data = self.Property[i][0x08:0x08+cch]603 elif type == PropertyType['VT_LPWSTR']:604 data = self.Property[i][0x08:0x08+cch*2].decode('utf-16')605 else:606 self._raise_exception('Property.PIDSI_COMMENTS has an abnormal value.')607 self.ole_logger.debug('Property.PIDSI_COMMENTS: ' + data)608 continue609 if self.PropertyIdentifierAndOffset[i].PropertyIdentifier == PIDSI['PIDSI_TEMPLATE']:610 type = struct.unpack('<H', self.Property[i][0x00:0x02])[0]611 self.ole_logger.debug('Property.PIDSI_TEMPLATE.type: ' + str(hex(type)))612 cch = struct.unpack('<I', self.Property[i][0x04:0x08])[0]613 self.ole_logger.debug('Property.PIDSI_TEMPLATE.cch: ' + str(hex(cch)))614 if cch > 0x0000FFFF:615 self._raise_exception('Property.PIDSI_TEMPLATE.cch has an abnormal value.')616 if type == PropertyType['VT_LPSTR']:617 data = self.Property[i][0x08:0x08+cch]618 elif type == PropertyType['VT_LPWSTR']:619 data = self.Property[i][0x08:0x08+cch*2].decode('utf-16')620 else:621 self._raise_exception('Property.PIDSI_TEMPLATE has an abnormal value.')622 self.ole_logger.debug('Property.PIDSI_TEMPLATE: ' + data)623 continue624 if self.PropertyIdentifierAndOffset[i].PropertyIdentifier == PIDSI['PIDSI_LASTAUTHOR']:625 type = struct.unpack('<H', self.Property[i][0x00:0x02])[0]626 self.ole_logger.debug('Property.PIDSI_LASTAUTHOR.type: ' + str(hex(type)))627 cch = struct.unpack('<I', self.Property[i][0x04:0x08])[0]628 self.ole_logger.debug('Property.PIDSI_LASTAUTHOR.cch: ' + str(hex(cch)))629 if cch > 0x0000FFFF:630 self._raise_exception('Property.PIDSI_LASTAUTHOR.cch has an abnormal value.')631 if type == PropertyType['VT_LPSTR']:632 data = self.Property[i][0x08:0x08+cch]633 elif type == PropertyType['VT_LPWSTR']:634 data = self.Property[i][0x08:0x08+cch*2].decode('utf-16')635 else:636 self._raise_exception('Property.PIDSI_LASTAUTHOR has an abnormal value.')637 self.ole_logger.debug('Property.PIDSI_LASTAUTHOR: ' + data)638 continue639 if self.PropertyIdentifierAndOffset[i].PropertyIdentifier == PIDSI['PIDSI_REVNUMBER']:640 type = struct.unpack('<H', self.Property[i][0x00:0x02])[0]641 self.ole_logger.debug('Property.PIDSI_REVNUMBER.type: ' + str(hex(type)))642 cch = struct.unpack('<I', self.Property[i][0x04:0x08])[0]643 self.ole_logger.debug('Property.PIDSI_REVNUMBER.cch: ' + str(hex(cch)))644 if cch > 0x0000FFFF:645 self._raise_exception('Property.PIDSI_REVNUMBER.cch has an abnormal value.')646 if type == PropertyType['VT_LPSTR']:647 data = self.Property[i][0x08:0x08+cch]648 elif type == PropertyType['VT_LPWSTR']:649 data = self.Property[i][0x08:0x08+cch*2].decode('utf-16')650 else:651 self._raise_exception('Property.PIDSI_REVNUMBER has an abnormal value.')652 self.ole_logger.debug('Property.PIDSI_REVNUMBER: ' + data)653 continue654 655 if self.PropertyIdentifierAndOffset[i].PropertyIdentifier == PIDSI['PIDSI_APPNAME']:656 type = struct.unpack('<H', self.Property[i][0x00:0x02])[0]657 self.ole_logger.debug('Property.PIDSI_APPNAME.type: ' + str(hex(type)))658 cch = struct.unpack('<I', self.Property[i][0x04:0x08])[0]659 self.ole_logger.debug('Property.PIDSI_APPNAME.cch: ' + str(hex(cch)))660 if cch > 0x0000FFFF:661 self._raise_exception('Property.PIDSI_APPNAME.cch has an abnormal value.')662 if type == PropertyType['VT_LPSTR']:663 data = self.Property[i][0x08:0x08+cch]664 elif type == PropertyType['VT_LPWSTR']:665 data = self.Property[i][0x08:0x08+cch*2].decode('utf-16')666 else:667 self._raise_exception('Property.PIDSI_APPNAME has an abnormal value.')668 self.ole_logger.debug('Property.PIDSI_APPNAME: ' + data)669 continue670 if self.PropertyIdentifierAndOffset[i].PropertyIdentifier == PIDSI['PIDSI_EDITTIME']:671 type = struct.unpack('<H', self.Property[i][0x00:0x02])[0]672 self.ole_logger.debug('Property.PIDSI_APPNAME.type: ' + str(hex(type)))673 if type != PropertyType['VT_FILETIME']:674 self._raise_exception('Property.PIDSI_EDITTIME has an abnormal value.')675 time = struct.unpack('<Q', self.Property[i][0x04:0x0C])[0]676 self.ole_logger.debug('Property.PIDSI_EDITTIME: ' + str(hex(time)))677 continue678 if self.PropertyIdentifierAndOffset[i].PropertyIdentifier == PIDSI['PIDSI_LASTPRINTED']:679 type = struct.unpack('<H', self.Property[i][0x00:0x02])[0]680 self.ole_logger.debug('Property.PIDSI_LASTPRINTED.type: ' + str(hex(type)))681 if type != PropertyType['VT_FILETIME']:682 self._raise_exception('Property.PIDSI_LASTPRINTED has an abnormal value.')683 time = struct.unpack('<Q', self.Property[i][0x04:0x0C])[0]684 self.ole_logger.debug('Property.PIDSI_LASTPRINTED: ' + self._filetime_to_datetime(time))685 continue686 if self.PropertyIdentifierAndOffset[i].PropertyIdentifier == PIDSI['PIDSI_CREATE_DTM']:687 type = struct.unpack('<H', self.Property[i][0x00:0x02])[0]688 self.ole_logger.debug('Property.PIDSI_CREATE_DTM.type: ' + str(hex(type)))689 if type != PropertyType['VT_FILETIME']:690 self._raise_exception('Property.PIDSI_CREATE_DTM has an abnormal value.')691 time = struct.unpack('<Q', self.Property[i][0x04:0x0C])[0]692 self.ole_logger.debug('Property.PIDSI_CREATE_DTM: ' + self._filetime_to_datetime(time))693 continue694 if self.PropertyIdentifierAndOffset[i].PropertyIdentifier == PIDSI['PIDSI_LASTSAVE_DTM']:695 type = struct.unpack('<H', self.Property[i][0x00:0x02])[0]696 self.ole_logger.debug('Property.PIDSI_LASTSAVE_DTM.type: ' + str(hex(type)))697 if type != PropertyType['VT_FILETIME']:698 self._raise_exception('Property.PIDSI_LASTSAVE_DTM has an abnormal value.')699 time = struct.unpack('<Q', self.Property[i][0x04:0x0C])[0]700 self.ole_logger.debug('Property.PIDSI_LASTSAVE_DTM: ' + self._filetime_to_datetime(time))701 continue702 703 if self.PropertyIdentifierAndOffset[i].PropertyIdentifier == PIDSI['PIDSI_PAGECOUNT']:704 type = struct.unpack('<H', self.Property[i][0x00:0x02])[0]705 self.ole_logger.debug('Property.PIDSI_PAGECOUNT.type: ' + str(hex(type)))706 if type != PropertyType['VT_I4']:707 self._raise_exception('Property.PIDSI_PAGECOUNT has an abnormal value.')708 count = struct.unpack('<I', self.Property[i][0x04:0x08])[0]709 self.ole_logger.debug('Property.PIDSI_PAGECOUNT: ' + str(hex(count)))710 continue711 if self.PropertyIdentifierAndOffset[i].PropertyIdentifier == PIDSI['PIDSI_WORDCOUNT']:712 type = struct.unpack('<H', self.Property[i][0x00:0x02])[0]713 self.ole_logger.debug('Property.PIDSI_WORDCOUNT.type: ' + str(hex(type)))714 if type != PropertyType['VT_I4']:715 self._raise_exception('Property.PIDSI_WORDCOUNT has an abnormal value.')716 count = struct.unpack('<I', self.Property[i][0x04:0x08])[0]717 self.ole_logger.debug('Property.PIDSI_WORDCOUNT: ' + str(hex(count)))718 continue719 if self.PropertyIdentifierAndOffset[i].PropertyIdentifier == PIDSI['PIDSI_CHARCOUNT']:720 type = struct.unpack('<H', self.Property[i][0x00:0x02])[0]721 self.ole_logger.debug('Property.PIDSI_CHARCOUNT.type: ' + str(hex(type)))722 if type != PropertyType['VT_I4']:723 self._raise_exception('Property.PIDSI_CHARCOUNT has an abnormal value.')724 count = struct.unpack('<I', self.Property[i][0x04:0x08])[0]725 self.ole_logger.debug('Property.PIDSI_CHARCOUNT: ' + str(hex(count)))726 continue727 if self.PropertyIdentifierAndOffset[i].PropertyIdentifier == PIDSI['PIDSI_DOC_SECURITY']:728 type = struct.unpack('<H', self.Property[i][0x00:0x02])[0]729 self.ole_logger.debug('Property.PIDSI_DOC_SECURITY.type: ' + str(hex(type)))730 if type != PropertyType['VT_I4']:731 self._raise_exception('Property.PIDSI_DOC_SECURITY has an abnormal value.')732 security = struct.unpack('<I', self.Property[i][0x04:0x08])[0]733 self.ole_logger.debug('Property.PIDSI_DOC_SECURITY: ' + str(hex(security)))734 continue735class SummaryInfo(OLEBase):736 byteOrder = 0737 version = 0738 sysId = 0739 OSMajorVersion = 0740 OSMinorVersion = 0741 OSType = 0742 applicationClsid = ''743 cSections = 0744 formatId1 = ''745 sectionOffset1 = 0746 formatId2 = ''747 sectionOffset2 = 0748 SummaryInfoPropertySet = None749 def __init__(self, data):750 self.byteOrder = 0751 self.version = 0752 self.sysId = 0753 self.OSMajorVersion = 0754 self.OSMinorVersion = 0755 self.OSType = 0756 self.applicationClsid = ''757 self.cSections = 0758 self.formatId1 = ''759 self.sectionOffset1 = 0760 self.formatId2 = ''761 self.sectionOffset2 = 0762 self.SummaryInfoPropertySet = None763 self.ole_logger.debug('######## SummaryInfo ########')764 self.byteOrder = struct.unpack('<H', data[0x00:0x02])[0]765 self.ole_logger.debug('SummaryInfo.byteOrder: ' + str(hex(self.byteOrder)))766 if self.byteOrder != 0xFFFE:767 self._raise_exception('DocumentSummaryInfo.byteOrder has an abnormal value.')768 self.version = struct.unpack('<H', data[0x02:0x04])[0]769 self.ole_logger.debug('SummaryInfo.version: ' + str(hex(self.version)))770 if self.version != 0 and self.version != 1:771 self._raise_exception('SummaryInfo.version has an abnormal value.')772 self.sysId = struct.unpack('<I', data[0x04:0x08])[0]773 self.ole_logger.debug('SummaryInfo.sysId: ' + str(hex(self.sysId)))774 self.clsid = data[0x08:0x18]775 self.ole_logger.debug('SummaryInfo.clsid: ' + self.clsid.encode('hex'))776 if self.clsid != '\x00' * 0x10:777 self._raise_exception('SummaryInfo.clsid has an abnormal value.')778 self.cSections = struct.unpack('<I', data[0x18:0x1C])[0]779 self.ole_logger.debug('SummaryInfo.cSections: ' + str(hex(self.cSections)))780 if self.cSections != 1 and self.cSections != 2:781 self._raise_exception('SummaryInfo.cSections has an abnormal value.')782 self.formatId1 = data[0x1C:0x2C]783 self.ole_logger.debug('SummaryInfo.rgIdOffset.IdOffsetElement-1.formatId: ' + self.formatId1.encode('hex'))784 if self.formatId1 != '\xE0\x85\x9F\xF2\xF9\x4F\x68\x10\xAB\x91\x08\x00\x2B\x27\xB3\xD9':785 self._raise_exception('SummaryInfo.rgIdOffset.IdOffsetElement-1.formatId has an abnormal value.')786 self.sectionOffset1 = struct.unpack('<I', data[0x2C:0x30])[0]787 self.ole_logger.debug('DocumentSummaryInfo.rgIdOffset.IdOffsetElement-1.sectionOffset: ' + str(hex(self.sectionOffset1)))788 if self.cSections == 2:789 self.formatId2 = data[0x30:0x40]790 self.ole_logger.debug('SummaryInfo.rgIdOffset.IdOffsetElement-2.formatId: ' + self.formatId2.encode('hex'))791 if self.formatId2 != '\x05\xD5\xCD\xD5\x9C\x2E\x1B\x10\x93\x97\x08\x00\x2B\x2C\xF9\xAE':792 self._raise_exception('SummaryInfo.rgIdOffset.IdOffsetElement-2.formatId has an abnormal value.')793 self.sectionOffset2 = struct.unpack('<I', data[0x40:0x44])[0]794 self.ole_logger.debug('SummaryInfo.rgIdOffset.IdOffsetElement-2.sectionOffset: ' + str(hex(self.sectionOffset2)))795 self.SummaryInfoPropertySet = SummaryInfoPropertySet(data[self.sectionOffset1:])796class OLEFile(OLEBase):797 file_data = None798 sector_size = 0799 mini_sector_size = 0800 OLEHeader = None801 DIFAT = list()802 FAT = list()803 MiniFAT = list()804 Directory = list()805 SummaryInfo = None806 DocumentSummaryInfo = None807 def __init__(self, filename):808 809 self.file_data = None810 self.sector_size = 0811 self.mini_sector_size = 0812 813 self.OLEHeader = None814 self.DIFAT = list()815 self.FAT = list()816 self.MiniFAT = list()817 self.Directory = list()818 self.SummaryInfo = None819 self.DocumentSummaryInfo = None820 821 822 if os.path.isfile(filename):823 self.file_data = open(filename, 'rb').read()824 self.ole_logger.debug('Load file: ' + filename)825 self.OLEHeader = OLEHeader(self.file_data)826 if self.OLEHeader.SectorShift == 0x09:827 self.sector_size = 512828 elif self.OLEHeader.SectorShift == 0x0C:829 self.sector_size = 4096830 else:831 self._raise_exception('Invalid Sector Size.')832 if self.OLEHeader.MiniSectorShift == 0x06:833 self.mini_sector_size = 64834 else:835 self._raise_exception('Invalid MiniSector Size.')836 self._init_fat_chain()837 838 if self.OLEHeader.NumberOfMiniFATSectors > 0:839 self._init_minifat_chain()840 self._init_dir_entry()841 for i in range(0, len(self.Directory)):842 if self.Directory[i].Name == '\x05SummaryInformation':843 self.SummaryInfo = SummaryInfo(self.find_object_by_index(i))844 if self.Directory[i].Name == '\x05DocumentSummaryInformation':845 self.DocumentSummaryInfo = DocSummaryInfo(self.find_object_by_index(i))846 else:847 self._raise_exception('Invalid file: ' + filename)848 def _init_fat_chain(self):849 self.DIFAT = list(self.OLEHeader.DIFAT)850 if self.OLEHeader.NumberOfDIFATSectors > 0:851 difat_sector_index = self.OLEHeader.FirstDIFATSector852 for i in range(0, self.OLEHeader.NumberOfDIFATSectors):853 difat_sector_offset = (difat_sector_index+1) * self.sector_size854 self.ole_logger.debug('DIFAT sector #' + str(i) + ' at offset: ' + str(hex(difat_sector_offset)))855 for j in range(0, self.sector_size/4-1):856 difat = struct.unpack('<I', self.file_data[difat_sector_offset+j*4:difat_sector_offset+j*4+4])[0]857 if difat == 0xFFFFFFFF:858 if i+1 == self.OLEHeader.NumberOfDIFATSectors:859 break860 else:861 _raise_exception('Encounter an invalid DIFAT value when parsing DIFAT chain.')862 self.ole_logger.debug('DIFT[' + str(len(self.DIFAT)) + ']: ' + str(hex(difat)))863 self.DIFAT.append(difat) 864 difat_sector_index = struct.unpack('<I', self.file_data[difat_sector_offset+j*4:difat_sector_offset+j*4+4])[0]865 866 if len(self.DIFAT) != self.OLEHeader.NumberOfFATSectors:867 self.ole_logger.warn('OLEHeader.NumberOfFATSectors does not mahtch the number of the DIFAT entries.')868 869 for i in range(0, self.OLEHeader.NumberOfFATSectors):870 fat_sector_index = self.DIFAT[i]871 fat_sector_offset = (fat_sector_index+1) * self.sector_size872 self.ole_logger.debug('FAT sector #' + str(i) + ' at offset: ' + str(hex(fat_sector_offset)))873 for j in range(0, self.sector_size/4):874 fat = struct.unpack('<I', self.file_data[fat_sector_offset+j*4:fat_sector_offset+j*4+4])[0]875 self.FAT.append(fat)...
pyvba.py
Source:pyvba.py
...117 self.ole_logger.debug('PROJECTwm.NameMap[' + str(i) + '].MBCS: ' + namemap_mbcs)118 namemap_utf16 = namemaps[i][index+1:]+'\x00'119 self.ole_logger.debug('PROJECTwm.NameMap[' + str(i) + '].UTF16: ' + namemap_utf16)120 else:121 self._raise_exception('PROJECTwm.NameMap[' + str(i) + '] has an abnormal values.')122 else:123 self._raise_exception('PROJECTwm stream contains abnormal values.')124class VBAProject(VBABase):125 Reserved1 = 0126 Version = 0127 Reserved2 = 0128 Reserved3 = 0129 PerformanceCache = ''130 131 def __init__(self, data):132 self.Reserved1 = 0133 self.Version = 0134 self.Reserved2 = 0135 self.Reserved3 = 0136 self.PerformanceCache = ''137 self.ole_logger.debug('######## _VBA_PROJECTStream ########')138 self.Reserved1 = struct.unpack('<H', data[0x00:0x02])[0]139 self.ole_logger.debug('_VBA_PROJECT.Reserved1: ' + str(hex(self.Reserved1)))140 if self.Reserved1 != 0x61CC:141 self._raise_exception('_VBA_PROJECT.Reserved1 has an abnormal values.')142 self.Version = struct.unpack('<H', data[0x02:0x04])[0]143 self.ole_logger.debug('_VBA_PROJECT.Version: ' + str(hex(self.Version)))144 self.Reserved2 = ord(data[0x04])145 self.ole_logger.debug('_VBA_PROJECT.Reserved2: ' + str(hex(self.Reserved2)))146 if self.Reserved2 != 0x00:147 self._raise_exception('_VBA_PROJECT.Reserved2 has an abnormal values.')148 self.Reserved3 = struct.unpack('<H', data[0x05:0x07])[0]149 self.ole_logger.debug('_VBA_PROJECT.Reserved3: ' + str(hex(self.Reserved3)))150 self.PerformanceCache = data[0x07:]151class ProjectSysKindRecord(VBABase):152 Id = 0153 Size = 0154 SysKind = 0155 def __init__(self, data):156 self.Id = 0157 self.Size = 0158 self.SysKind = 0159 self.Id = struct.unpack('<H', data[0x00:0x02])[0]160 self.ole_logger.debug('ProjectSysKindRecord.Id: ' + str(hex(self.Id)))161 if self.Id != 0x01:162 self._raise_exception('ProjectSysKindRecord.Id has an abnormal value.')163 self.Size = struct.unpack('<I', data[0x02:0x06])[0]164 self.ole_logger.debug('ProjectSysKindRecord.Size: ' + str(hex(self.Size)))165 if self.Size != 0x04:166 self._raise_exception('ProjectSysKindRecord.Size has an abnormal value.')167 self.SysKind = struct.unpack('<I', data[0x06:0x0A])[0]168 if self.SysKind == 0x00:169 self.ole_logger.debug('ProjectSysKindRecord.SysKind: ' + str(hex(self.SysKind)) + ' (16-bit Windows Platforms)')170 elif self.SysKind == 0x01:171 self.ole_logger.debug('ProjectSysKindRecord.SysKind: ' + str(hex(self.SysKind)) + ' (32-bit Windows Platforms)')172 elif self.SysKind == 0x02:173 self.ole_logger.debug('ProjectSysKindRecord.SysKind: ' + str(hex(self.SysKind)) + ' (Macintosh Platforms)')174 elif self.SysKind == 0x03:175 self.ole_logger.debug('ProjectSysKindRecord.SysKind: ' + str(hex(self.SysKind)) + ' (64-bit Windows Platforms)')176 else:177 self._raise_exception('ProjectSysKindRecord.SysKind has an abnormal value.')178class ProjectLcidRecord(VBABase):179 180 Id = 0181 Size = 0182 Lcid = 0183 def __init__(self, data):184 self.Id = 0185 self.Size = 0186 self.Lcid = 0187 self.Id = struct.unpack('<H', data[0x00:0x02])[0]188 self.ole_logger.debug('ProjectLcidRecord.Id: ' + str(hex(self.Id)))189 if self.Id != 0x02:190 self._raise_exception('ProjectLcidRecord.Id has an abnormal value.')191 self.Size = struct.unpack('<I', data[0x02:0x06])[0]192 self.ole_logger.debug('ProjectLcidRecord.Size: ' + str(hex(self.Size)))193 if self.Size != 0x04:194 self._raise_exception('ProjectLcidRecord.Size has an abnormal value.')195 self.Lcid = struct.unpack('<I', data[0x06:0x0A])[0]196 self.ole_logger.debug('ProjectLcidRecord.Lcid: ' + str(hex(self.Lcid)))197 if self.Lcid != 0x409:198 self._raise_exception('ProjectLcidRecord.Lcid has an abnormal value.')199class ProjectLcidInvokeRecord(VBABase):200 201 Id = 0202 Size = 0203 LcidInvoke = 0204 def __init__(self, data):205 self.Id = 0206 self.Size = 0207 self.LcidInvoke = 0208 self.Id = struct.unpack('<H', data[0x00:0x02])[0]209 self.ole_logger.debug('ProjectLcidInvokeRecord.Id: ' + str(hex(self.Id)))210 if self.Id != 0x14:211 self._raise_exception('ProjectLcidInvokeRecord.Id has an abnormal value.')212 self.Size = struct.unpack('<I', data[0x02:0x06])[0]213 self.ole_logger.debug('ProjectLcidInvokeRecord.Size: ' + str(hex(self.Size)))214 if self.Size != 0x04:215 self._raise_exception('ProjectLcidInvokeRecord.Size has an abnormal value.')216 self.LcidInvoke = struct.unpack('<I', data[0x06:0x0A])[0]217 self.ole_logger.debug('ProjectLcidInvokeRecord.LcidInvoke: ' + str(hex(self.LcidInvoke)))218 if self.LcidInvoke != 0x409:219 self._raise_exception('ProjectLcidInvokeRecord.LcidInvoke has an abnormal value.')220class ProjectCodePageRecord(VBABase):221 222 Id = 0223 Size = 0224 CodePage = 0225 def __init__(self, data):226 self.Id = 0227 self.Size = 0228 self.CodePage = 0229 self.Id = struct.unpack('<H', data[0x00:0x02])[0]230 self.ole_logger.debug('ProjectCodePageRecord.Id: ' + str(hex(self.Id)))231 if self.Id != 0x03:232 self._raise_exception('ProjectCodePageRecord.Id has an abnormal value.')233 self.Size = struct.unpack('<I', data[0x02:0x06])[0]234 self.ole_logger.debug('ProjectCodePageRecord.Size: ' + str(hex(self.Size)))235 if self.Size != 0x02:236 self._raise_exception('ProjectCodePageRecord.Size has an abnormal value.')237 self.CodePage = struct.unpack('<H', data[0x06:0x08])[0]238 self.ole_logger.debug('ProjectCodePageRecord.CodePage: ' + str(hex(self.CodePage)))239class ProjectNameRecord(VBABase):240 241 Id = 0242 Size = 0243 ProjectName = 0244 def __init__(self, data):245 self.Id = 0246 self.Size = 0247 self.ProjectName = 0248 self.Id = struct.unpack('<H', data[0x00:0x02])[0]249 self.ole_logger.debug('ProjectNameRecord.Id: ' + str(hex(self.Id)))250 if self.Id != 0x04:251 self._raise_exception('ProjectNameRecord.Id has an abnormal value.')252 self.Size = struct.unpack('<I', data[0x02:0x06])[0]253 self.ole_logger.debug('ProjectNameRecord.Size: ' + str(hex(self.Size)))254 if self.Size < 0x01 or self.Size > 0x80:255 self._raise_exception('ProjectNameRecord.Size has an abnormal value.')256 self.ProjectName = data[0x06:0x06+self.Size]257 self.ole_logger.debug('ProjectNameRecord.ProjectName: ' + self.ProjectName)258class ProjectDocStringRecord(VBABase):259 260 Id = 0261 SizeOfDocString = 0262 DocString = ''263 Reserved = 0264 SizeOfDocStringUnicode = 0265 DocStringUnicode = ''266 def __init__(self, data):267 self.Id = 0268 self.SizeOfDocString = 0269 self.DocString = ''270 self.Reserved = 0271 self.SizeOfDocStringUnicode = 0272 self.DocStringUnicode = ''273 self.Id = struct.unpack('<H', data[0x00:0x02])[0]274 self.ole_logger.debug('ProjectDocStringRecord.Id: ' + str(hex(self.Id)))275 if self.Id != 0x05:276 self._raise_exception('ProjectDocStringRecord.Id has an abnormal value.')277 self.SizeOfDocString = struct.unpack('<I', data[0x02:0x06])[0]278 self.ole_logger.debug('ProjectDocStringRecord.SizeOfDocString: ' + str(hex(self.SizeOfDocString)))279 if self.SizeOfDocString > 2000:280 self._raise_exception('ProjectDocStringRecord.SizeOfDocString has an abnormal value.')281 self.DocString = data[0x06:0x06+self.SizeOfDocString]282 self.ole_logger.debug('ProjectDocStringRecord.DocString: ' + self.DocString)283 current = 0x06 + self.SizeOfDocString284 self.Reserved = struct.unpack('<H', data[current:current+0x02])[0]285 self.ole_logger.debug('ProjectDocStringRecord.Reserved: ' + str(hex(self.Reserved)))286 if self.Reserved != 0x40:287 self._raise_exception('ProjectDocStringRecord.Reserved has an abnormal value.')288 current = current + 0x02289 self.SizeOfDocStringUnicode = struct.unpack('<I', data[current:current+0x04])[0]290 self.ole_logger.debug('ProjectDocStringRecord.SizeOfDocStringUnicode: ' + str(hex(self.SizeOfDocStringUnicode)))291 if self.SizeOfDocStringUnicode > 2000*2:292 self._raise_exception('ProjectDocStringRecord.SizeOfDocStringUnicode has an abnormal value.')293 current = current + 0x04294 self.DocStringUnicode = data[current:current+self.SizeOfDocStringUnicode]295 self.ole_logger.debug('ProjectDocStringRecord.DocStringUnicode: ' + self.DocStringUnicode)296class ProjectHelpFilePathRecord(VBABase):297 298 Id = 0299 SizeOfHelpFile1 = 0300 HelpFile1 = ''301 Reserved = 0302 SizeOfHelpFile2 = 0303 HelpFile2 = ''304 def __init__(self, data):305 self.Id = 0306 self.SizeOfHelpFile1 = 0307 self.HelpFile1 = ''308 self.Reserved = 0309 self.SizeOfHelpFile2 = 0310 self.HelpFile2 = ''311 self.Id = struct.unpack('<H', data[0x00:0x02])[0]312 self.ole_logger.debug('ProjectHelpFilePathRecord.Id: ' + str(hex(self.Id)))313 if self.Id != 0x06:314 self._raise_exception('ProjectHelpFilePathRecord.Id has an abnormal value.')315 self.SizeOfHelpFile1 = struct.unpack('<I', data[0x02:0x06])[0]316 self.ole_logger.debug('ProjectHelpFilePathRecord.SizeOfHelpFile1: ' + str(hex(self.SizeOfHelpFile1)))317 if self.SizeOfHelpFile1 > 260:318 self._raise_exception('ProjectHelpFilePathRecord.SizeOfHelpFile1 has an abnormal value.')319 self.HelpFile1 = data[0x06:0x06+self.SizeOfHelpFile1]320 self.ole_logger.debug('ProjectHelpFilePathRecord.HelpFile1: ' + self.HelpFile1)321 current = 0x06 + self.SizeOfHelpFile1322 self.Reserved = struct.unpack('<H', data[current:current+0x02])[0]323 self.ole_logger.debug('ProjectHelpFilePathRecord.Reserved: ' + str(hex(self.Reserved)))324 if self.Reserved != 0x3D:325 self._raise_exception('ProjectHelpFilePathRecord.Reserved has an abnormal value.')326 current = current + 0x02327 self.SizeOfHelpFile2 = struct.unpack('<I', data[current:current+0x04])[0]328 self.ole_logger.debug('ProjectHelpFilePathRecord.SizeOfHelpFile2: ' + str(hex(self.SizeOfHelpFile2)))329 if self.SizeOfHelpFile2 > 260:330 self._raise_exception('ProjectHelpFilePathRecord.SizeOfHelpFile2 has an abnormal value.')331 current = current + 0x04332 self.HelpFile2 = data[current:current+self.SizeOfHelpFile2]333 self.ole_logger.debug('ProjectHelpFilePathRecord.HelpFile2: ' + self.HelpFile2)334class ProjectHelpContextRecord(VBABase):335 336 Id = 0337 Size = 0338 HelpContext = 0339 def __init__(self, data):340 self.Id = 0341 self.Size = 0342 self.HelpContext = 0343 self.Id = struct.unpack('<H', data[0x00:0x02])[0]344 self.ole_logger.debug('ProjectHelpContextRecord.Id: ' + str(hex(self.Id)))345 if self.Id != 0x07:346 self._raise_exception('ProjectHelpContextRecord.Id has an abnormal value.')347 self.Size = struct.unpack('<I', data[0x02:0x06])[0]348 self.ole_logger.debug('ProjectHelpContextRecord.Size: ' + str(hex(self.Size)))349 if self.Size != 0x04:350 self._raise_exception('ProjectHelpContextRecord.Size has an abnormal value.')351 self.HelpContext = struct.unpack('<I', data[0x06:0x0A])[0]352 self.ole_logger.debug('ProjectHelpContextRecord.HelpContext: ' + str(hex(self.HelpContext)))353class ProjectLibFlagsRecord(VBABase):354 355 Id = 0356 Size = 0357 LibFlags = 0358 def __init__(self, data):359 self.Id = 0360 self.Size = 0361 self.LibFlags = 0362 self.Id = struct.unpack('<H', data[0x00:0x02])[0]363 self.ole_logger.debug('ProjectLibFlagsRecord.Id: ' + str(hex(self.Id)))364 if self.Id != 0x8:365 self._raise_exception('ProjectLibFlagsRecord.Id has an abnormal value.')366 self.Size = struct.unpack('<I', data[0x02:0x06])[0]367 self.ole_logger.debug('ProjectLibFlagsRecord.Size: ' + str(hex(self.Size)))368 if self.Size != 0x04:369 self._raise_exception('ProjectLibFlagsRecord.Size has an abnormal value.')370 self.LibFlags = struct.unpack('<I', data[0x06:0x0A])[0]371 self.ole_logger.debug('ProjectLibFlagsRecord.LibFlags: ' + str(hex(self.LibFlags)))372 if self.LibFlags != 0x00:373 self._raise_exception('ProjectLibFlagsRecord.LibFlags has an abnormal value.')374class ProjectVersionRecord(VBABase):375 376 Id = 0377 Size = 0378 MajorVersion = 0379 MinorVersion = 0380 def __init__(self, data):381 self.Id = 0382 self.Reserved = 0383 self.MajorVersion = 0384 self.MinorVersion = 0385 self.Id = struct.unpack('<H', data[0x00:0x02])[0]386 self.ole_logger.debug('ProjectVersionRecord.Id: ' + str(hex(self.Id)))387 if self.Id != 0x9:388 self._raise_exception('ProjectVersionRecord.Id has an abnormal value.')389 self.Reserved = struct.unpack('<I', data[0x02:0x06])[0]390 self.ole_logger.debug('ProjectVersionRecord.Reserved: ' + str(hex(self.Reserved)))391 if self.Reserved != 0x04:392 self._raise_exception('ProjectVersionRecord.Reserved has an abnormal value.')393 self.MajorVersion = struct.unpack('<I', data[0x06:0x0A])[0]394 self.ole_logger.debug('ProjectVersionRecord.MajorVersion: ' + str(hex(self.MajorVersion)))395 self.MinorVersion = struct.unpack('<H', data[0x0A:0x0C])[0]396 self.ole_logger.debug('ProjectVersionRecord.MinorVersion: ' + str(hex(self.MinorVersion)))397class ProjectConstantsRecord(VBABase):398 399 Id = 0400 SizeOfConstants = 0401 Constants = ''402 Reserved = 0403 SizeOfConstantsUnicode = 0404 ConstantsUnicode = ''405 def __init__(self, data):406 self.Id = 0407 self.SizeOfConstants = 0408 self.Constants = ''409 self.Reserved = 0410 self.SizeOfConstantsUnicode = 0411 self.ConstantsUnicode = ''412 self.Id = struct.unpack('<H', data[0x00:0x02])[0]413 self.ole_logger.debug('ProjectConstantsRecord.Id: ' + str(hex(self.Id)))414 if self.Id != 0x0C:415 self._raise_exception('ProjectConstantsRecord.Id has an abnormal value.')416 self.SizeOfConstants = struct.unpack('<I', data[0x02:0x06])[0]417 self.ole_logger.debug('ProjectConstantsRecord.SizeOfConstants: ' + str(hex(self.SizeOfConstants)))418 if self.SizeOfConstants > 1015:419 self._raise_exception('ProjectConstantsRecord.SizeOfConstants has an abnormal value.')420 self.Constants = data[0x06:0x06+self.SizeOfConstants]421 self.ole_logger.debug('ProjectConstantsRecord.Constants: ' + self.Constants)422 current = 0x06 + self.SizeOfConstants423 self.Reserved = struct.unpack('<H', data[current:current+0x02])[0]424 self.ole_logger.debug('ProjectConstantsRecord.Reserved: ' + str(hex(self.Reserved)))425 if self.Reserved != 0x3C:426 self._raise_exception('ProjectConstantsRecord.Reserved has an abnormal value.')427 current = current + 0x02428 self.SizeOfConstantsUnicode = struct.unpack('<I', data[current:current+0x04])[0]429 self.ole_logger.debug('ProjectConstantsRecord.SizeOfConstantsUnicode: ' + str(hex(self.SizeOfConstantsUnicode)))430 if self.SizeOfConstantsUnicode > 1015*2:431 self._raise_exception('ProjectConstantsRecord.SizeOfConstantsUnicode has an abnormal value.')432 current = current + 0x04433 self.ConstantsUnicode = data[current:current+self.SizeOfConstantsUnicode]434 self.ole_logger.debug('ProjectConstantsRecord.ConstantsUnicode: ' + self.ConstantsUnicode)435class ProjectInformationRecord(VBABase):436 SysKindRecord = None437 LcidRecord = None438 LcidInvokeRecord = None439 CodePageRecord = None440 NameRecord = None441 DocStringRecord = None442 HelpFilePathRecord = None443 HelpContextRecord = None444 LibFlagsRecord = None445 VersionRecord = None446 ConstantsRecord = None447 Size = 0448 def __init__(self, data):449 self.SysKindRecord = None450 self.LcidRecord = None451 self.LcidInvokeRecord = None452 self.NameRecord = None453 self.DocStringRecord = None454 self.HelpFilePathRecord = None455 self.HelpContextRecord = None456 self.LibFlagsRecord = None457 self.VersionRecord = None458 self.ConstantsRecord = None459 self.Size = 0460 self.SysKindRecord = ProjectSysKindRecord(data[0x00:0x0A])461 462 self.LcidRecord = ProjectLcidRecord(data[0x0A:0x14])463 464 self.LcidInvokeRecord = ProjectLcidInvokeRecord(data[0x14:0x1E])465 466 self.CodePageRecord = ProjectCodePageRecord(data[0x1E:0x26])467 468 self.NameRecord = ProjectNameRecord(data[0x26:])469 current = 0x26 + 0x06 + self.NameRecord.Size470 self.DocStringRecord = ProjectDocStringRecord(data[current:])471 current = current + 0x0C + self.DocStringRecord.SizeOfDocString + self.DocStringRecord.SizeOfDocStringUnicode472 self.HelpFilePathRecord = ProjectHelpFilePathRecord(data[current:])473 current = current + 0x0C + self.HelpFilePathRecord.SizeOfHelpFile1 + self.HelpFilePathRecord.SizeOfHelpFile2474 self.HelpContextRecord = ProjectHelpContextRecord(data[current:])475 current = current + 0x0A476 self.LibFlagsRecord = ProjectLibFlagsRecord(data[current:])477 current = current + 0x0A478 self.VersionRecord = ProjectVersionRecord(data[current:])479 current = current + 0x0C480 self.ConstantsRecord = ProjectConstantsRecord(data[current:])481 self.Size = current + 0x0C + self.ConstantsRecord.SizeOfConstants + self.ConstantsRecord.SizeOfConstantsUnicode482class ReferenceNameRecord(VBABase):483 Id = 0484 SizeOfName = 0485 Name = ''486 Reserved = 0487 SizeOfNameUnicode = 0488 NameUnicode = ''489 def __init__(self, data):490 self.Id = 0491 self.SizeOfName = 0492 self.Name = ''493 self.Reserved = 0494 self.SizeOfNameUnicode = 0495 self.NameUnicode = ''496 self.Id = struct.unpack('<H', data[0x00:0x02])[0]497 self.ole_logger.debug('ReferenceNameRecord.Id: ' + str(hex(self.Id)))498 if self.Id != 0x16:499 self._raise_exception('ReferenceNameRecord.Id has an abnormal value.')500 self.SizeOfName = struct.unpack('<I', data[0x02:0x06])[0]501 self.ole_logger.debug('ReferenceNameRecord.SizeOfName: ' + str(hex(self.SizeOfName)))502 self.Name = data[0x06:0x06+self.SizeOfName]503 self.ole_logger.debug('ReferenceNameRecord.Name: ' + self.Name)504 current = 0x06 + self.SizeOfName505 self.Reserved = struct.unpack('<H', data[current:current+0x02])[0]506 self.ole_logger.debug('ReferenceNameRecord.Reserved: ' + str(hex(self.Reserved)))507 if self.Reserved != 0x3E:508 self.ole_logger.warn('ReferenceNameRecord.Reserved has an abnormal value.')509 current = current + 0x02510 self.SizeOfNameUnicode = struct.unpack('<I', data[current:current+0x04])[0]511 self.ole_logger.debug('ReferenceNameRecord.SizeOfNameUnicode: ' + str(hex(self.SizeOfNameUnicode)))512 current = current + 0x04513 self.NameUnicode = data[current:current+self.SizeOfNameUnicode]514 self.ole_logger.debug('ReferenceNameRecord.NameUnicode: ' + self.NameUnicode)515class ReferenceOriginalRecord(VBABase):516 Id = 0517 SizeOfLibidOriginal = 0518 LibidOriginal = ''519 def __init__(self, data):520 self.Id = 0521 self.SizeOfLibidOriginal = 0522 self.LibidOriginal = ''523 self.id = struct.unpack('<H', data[0x00:0x02])[0]524 if self.id != 0x33:525 self._raise_exception('ReferenceOriginalRecord.Id has an abnormal value.')526 self.SizeOfLibidOriginal = struct.unpack('<I', data[0x02:0x06])[0]527 self.ole_logger.debug('ReferenceOriginalRecord.SizeOfLibidOriginal: ' + str(hex(self.SizeOfLibidOriginal)))528 self.LibidOriginal = data[0x06:0x06+self.SizeOfLibidOriginal]529 self.ole_logger.debug('ReferenceOriginalRecord.LibidOriginal: ' + self.LibidOriginal)530class ReferenceControlRecord(VBABase):531 OriginalRecord = None532 Id = 0533 SizeTwiddled = 0534 SizeOfLibidTwiddled = 0535 LibidTwiddled = ''536 Reserved1 = 0537 Reserved2 = 0538 NameRecordExtended = None539 Reserved3 = 0540 SizeExtended = 0541 SizeOfLibidExtended = 0542 LibidExtended = ''543 Reserved4 = 0544 Reserved5 = 0545 OriginalTypeLib = ''546 Cookie = 0547 Size = 0548 def __init__(self, data):549 self.OriginalRecord = None550 self.Id = 0551 self.SizeTwiddled = 0552 self.SizeOfLibidTwiddled = 0553 self.LibidTwiddled = ''554 self.Reserved1 = 0555 self.Reserved2 = 0556 self.NameRecordExtended = None557 self.Reserved3 = 0558 self.SizeExtended = 0559 self.SizeOfLibidExtended = 0560 self.LibidExtended = ''561 self.Reserved4 = 0562 self.Reserved5 = 0563 self.OriginalTypeLib = ''564 self.Cookie = 0565 self.Size = 0566 current = 0567 id = struct.unpack('<H', data[current:current+0x02])[0]568 if id == 0x33:569 self.OriginalRecord = ReferenceOriginalRecord(data)570 current = current + 0x06 + self.OriginalRecord.SizeOfLibidOriginal571 self.Id = struct.unpack('<H', data[current:current+0x02])[0]572 self.ole_logger.debug('ReferenceControlRecord.Id: ' + str(hex(self.Id)))573 if self.Id != 0x2F:574 self._raise_exception('ReferenceControlRecord.Id has an abnormal value.')575 current = current + 0x02576 self.SizeTwiddled = struct.unpack('<I', data[current:current+0x04])[0]577 self.ole_logger.debug('ReferenceControlRecord.SizeTwiddled: ' + str(hex(self.SizeTwiddled)))578 current = current + 0x04579 self.SizeOfLibidTwiddled = struct.unpack('<I', data[current:current+0x04])[0]580 self.ole_logger.debug('ReferenceControlRecord.SizeOfLibidTwiddled: ' + str(hex(self.SizeOfLibidTwiddled)))581 current = current + 0x04 582 self.LibidTwiddled = data[current:current+self.SizeOfLibidTwiddled]583 self.ole_logger.debug('ReferenceControlRecord.LibidTwiddled: ' + self.LibidTwiddled)584 current = current + self.SizeOfLibidTwiddled585 self.Reserved1 = struct.unpack('<I', data[current:current+0x04])[0]586 self.ole_logger.debug('ReferenceControlRecord.Reserved1: ' + str(hex(self.Reserved1)))587 if self.Reserved1 != 0x00:588 self._raise_exception('ReferenceControlRecord.Reserved1 has an abnormal value.')589 current = current + 0x04590 self.Reserved2 = struct.unpack('<H', data[current:current+0x02])[0]591 self.ole_logger.debug('ReferenceControlRecord.Reserved2: ' + str(hex(self.Reserved2)))592 if self.Reserved2 != 0x00:593 self._raise_exception('ReferenceControlRecord.Reserved2 has an abnormal value.')594 current = current + 0x02595 id = struct.unpack('<H', data[current:current+0x02])[0]596 if id == 0x16:597 self.NameRecordExtended = ReferenceNameRecord(data[current:])598 current = current + 0x0C + self.NameRecordExtended.SizeOfName + self.NameRecordExtended.SizeOfNameUnicode599 self.Reserved3 = struct.unpack('<H', data[current:current+0x02])[0]600 self.ole_logger.debug('ReferenceControlRecord.Reserved3: ' + str(hex(self.Reserved3)))601 if self.Reserved3 != 0x30:602 self._raise_exception('ReferenceControlRecord.Reserved3 has an abnormal value.')603 current = current + 0x02604 self.SizeExtended = struct.unpack('<I', data[current:current+0x04])[0]605 self.ole_logger.debug('ReferenceControlRecord.SizeExtended: ' + str(hex(self.SizeExtended)))606 current = current + 0x04607 self.SizeOfLibidExtended = struct.unpack('<I', data[current:current+0x04])[0]608 self.ole_logger.debug('ReferenceControlRecord.SizeOfLibidExtended: ' + str(hex(self.SizeOfLibidExtended)))609 current = current + 0x04610 self.LibidExtended = data[current:current+self.SizeOfLibidExtended]611 self.ole_logger.debug('ReferenceControlRecord.LibidExtended: ' + self.LibidExtended)612 current = current + self.SizeOfLibidExtended613 self.Reserved4 = struct.unpack('<I', data[current:current+0x04])[0]614 self.ole_logger.debug('ReferenceControlRecord.Reserved4: ' + str(hex(self.Reserved4)))615 if self.Reserved4 != 0x00:616 self._raise_exception('ReferenceControlRecord.Reserved4 has an abnormal value.')617 current = current + 0x04618 self.Reserved5 = struct.unpack('<H', data[current:current+0x02])[0]619 self.ole_logger.debug('ReferenceControlRecord.Reserved5: ' + str(hex(self.Reserved5)))620 if self.Reserved5 != 0x00:621 self._raise_exception('ReferenceControlRecord.Reserved5 has an abnormal value.')622 current = current + 0x02623 self.OriginalTypeLib = data[current:current+0x10]624 self.ole_logger.debug('ReferenceControlRecord.OriginalTypeLib: ' + self.OriginalTypeLib.encode('hex'))625 current = current + 0x10626 self.Cookie = struct.unpack('<I', data[current:current+0x04])[0]627 self.ole_logger.debug('ReferenceControlRecord.Cookie: ' + str(hex(self.Cookie)))628 self.Size = current + 0x04629class ReferenceRegisteredRecord(VBABase):630 Id = 0631 Size = 0632 SizeOfLibid = 0633 Libid = ''634 Reserved1 = 0635 Reserved2 = 0636 def __init__(self, data):637 self.Id = 0638 self.Size = 0639 self.SizeOfLibid = 0640 self.Libid = ''641 self.Reserved1 = 0642 self.Reserved2 = 0643 current = 0644 self.Id = struct.unpack('<H', data[current:current+0x02])[0]645 self.ole_logger.debug('ReferenceRegisteredRecord.Id: ' + str(hex(self.Id)))646 if self.Id != 0x0D:647 self._raise_exception('ReferenceRegisteredRecord.Id has an abnormal value.')648 current = current + 0x02649 self.Size = struct.unpack('<I', data[current:current+0x04])[0]650 self.ole_logger.debug('ReferenceRegisteredRecord.Size: ' + str(hex(self.Size)))651 current = current + 0x04652 self.SizeOfLibid = struct.unpack('<I', data[current:current+0x04])[0]653 self.ole_logger.debug('ReferenceRegisteredRecord.SizeOfLibid: ' + str(hex(self.SizeOfLibid)))654 current = current + 0x04 655 self.Libid = data[current:current+self.SizeOfLibid]656 self.ole_logger.debug('ReferenceRegisteredRecord.Libid: ' + self.Libid)657 current = current + self.SizeOfLibid658 self.Reserved1 = struct.unpack('<I', data[current:current+0x04])[0]659 self.ole_logger.debug('ReferenceRegisteredRecord.Reserved1: ' + str(hex(self.Reserved1)))660 if self.Reserved1 != 0x00:661 self._raise_exception('ReferenceRegisteredRecord.Reserved1 has an abnormal value.')662 current = current + 0x04663 self.Reserved2 = struct.unpack('<H', data[current:current+0x02])[0]664 self.ole_logger.debug('ReferenceRegisteredRecord.Reserved2: ' + str(hex(self.Reserved2)))665 if self.Reserved2 != 0x00:666 self._raise_exception('ReferenceRegisteredRecord.Reserved2 has an abnormal value.')667class ReferenceProjectRecord(VBABase):668 Id = 0669 Size = 0670 SizeOfLibidAbsolute = 0671 LibidAbsolute = ''672 SizeOfLibidRelative = 0673 LibidRelative = ''674 MajorVersion = 0675 MinorVersion = 0676 def __init__(self, data):677 self.Id = 0678 self.Size = 0679 self.SizeOfLibidAbsolute = 0680 self.LibidAbsolute = ''681 self.SizeOfLibidRelative = 0682 self.LibidRelative = ''683 self.MajorVersion = 0684 self.MinorVersion = 0685 current = 0686 self.Id = struct.unpack('<H', data[current:current+0x02])[0]687 self.ole_logger.debug('ReferenceProjectRecord.Id: ' + str(hex(self.Id)))688 if self.Id != 0x0E:689 self._raise_exception('ReferenceProjectRecord.Id has an abnormal value.')690 current = current + 0x02691 self.Size = struct.unpack('<I', data[current:current+0x04])[0]692 self.ole_logger.debug('ReferenceProjectRecord.Size: ' + str(hex(self.Size)))693 current = current + 0x04694 self.SizeOfLibidAbsolute = struct.unpack('<I', data[current:current+0x04])[0]695 self.ole_logger.debug('ReferenceProjectRecord.SizeOfLibidAbsolute: ' + str(hex(self.SizeOfLibidAbsolute)))696 current = current + 0x04 697 self.LibidAbsolute = data[current:current+self.SizeOfLibidAbsolute]698 self.ole_logger.debug('ReferenceProjectRecord.LibidAbsolute: ' + self.LibidAbsolute)699 current = current + self.SizeOfLibidAbsolute700 self.SizeOfLibidRelative = struct.unpack('<I', data[current:current+0x04])[0]701 self.ole_logger.debug('ReferenceProjectRecord.SizeOfLibidRelative: ' + str(hex(self.SizeOfLibidRelative)))702 current = current + 0x04 703 self.LibidRelative = data[current:current+self.SizeOfLibidRelative]704 self.ole_logger.debug('ReferenceProjectRecord.LibidRelative: ' + self.LibidRelative)705 current = current + self.SizeOfLibidRelative706 self.MajorVersion = struct.unpack('<I', data[current:current+0x04])[0]707 self.ole_logger.debug('ReferenceProjectRecord.MajorVersion: ' + str(hex(self.MajorVersion)))708 current = current + 0x04709 self.MinorVersion = struct.unpack('<H', data[current:current+0x02])[0]710 self.ole_logger.debug('ReferenceProjectRecord.MinorVersion: ' + str(hex(self.MinorVersion)))711class ProjectReferencesRecord(VBABase):712 ReferenceArray = list()713 Size = 0714 def __init__(self, data):715 self.ReferenceArray = list()716 self.Size = 0717 current = 0718 NameRecord = None719 ControlRecord = None720 RegisteredRecord = None721 ProjectRecord = None722 while True:723 id = struct.unpack('<H', data[current:current+2])[0]724 if id == 0x0F:725 self.Size = current726 break727 elif id == 0x16:728 NameRecord = ReferenceNameRecord(data[current:])729 current = current + 0x0C + NameRecord.SizeOfName + NameRecord.SizeOfNameUnicode730 id = struct.unpack('<H', data[current:current+2])[0]731 if id == 0x2F or id == 0x33:732 ControlRecord = ReferenceControlRecord(data[current:])733 current = current + ControlRecord.Size734 self.ReferenceArray.append([NameRecord, ControlRecord])735 elif id == 0x0D:736 RegisteredRecord = ReferenceRegisteredRecord(data[current:])737 current = current + 0x06 + RegisteredRecord.Size738 self.ReferenceArray.append([NameRecord, RegisteredRecord])739 elif id == 0x0E:740 ProjectRecord = ReferenceProjectRecord(data[current:])741 current = current + 0x06 + ProjectRecord.Size742 self.ReferenceArray.append([NameRecord, ProjectRecord])743 744 else:745 self.ole_logger.warn('ReferencesRecord.Id has an abnormal value.')746 elif id == 0x2F or id == 0x33:747 ControlRecord = ReferenceControlRecord(data[current:])748 current = current + ControlRecord.Size749 self.ReferenceArray.append([None, ControlRecord])750 elif id == 0x0D:751 RegisteredRecord = ReferenceRegisteredRecord(data[current:])752 current = current + 0x06 + RegisteredRecord.Size753 self.ReferenceArray.append([None, RegisteredRecord])754 elif id == 0x0E:755 ProjectRecord = ReferenceProjectRecord(data[current:])756 current = current + 0x06 + ProjectRecord.Size757 self.ReferenceArray.append([None, ProjectRecord])758 759 else:760 self._raise_exception('ReferencesRecord.Id has an abnormal value.')761class ProjectCookieRecord(VBABase):762 763 Id = 0764 Size = 0765 Cookie = 0766 def __init__(self, data):767 self.Id = 0768 self.Size = 0769 self.CodePage = 0770 self.Id = struct.unpack('<H', data[0x00:0x02])[0]771 self.ole_logger.debug('ProjectCookieRecord.Id: ' + str(hex(self.Id)))772 if self.Id != 0x13:773 self._raise_exception('ProjectCookieRecord.Id has an abnormal value.')774 self.Size = struct.unpack('<I', data[0x02:0x06])[0]775 self.ole_logger.debug('ProjectCookieRecord.Size: ' + str(hex(self.Size)))776 if self.Size != 0x02:777 self._raise_exception('ProjectCookieRecord.Size has an abnormal value.')778 self.Cookie = struct.unpack('<H', data[0x06:0x08])[0]779 self.ole_logger.debug('ProjectCookieRecord.Cookie: ' + str(hex(self.Cookie)))780class ModuleNameRecord(VBABase):781 Id = 0782 SizeOfModuleName = 0783 ModuleName = ''784 def __init__(self, data):785 self.Id = 0786 self.SizeOfModuleName = 0787 self.ModuleName = ''788 self.Id = struct.unpack('<H', data[0x00:0x02])[0]789 self.ole_logger.debug('ModuleNameRecord.Id: ' + str(hex(self.Id)))790 if self.Id != 0x19:791 self._raise_exception('ModuleNameRecord.Id has an abnormal value.')792 self.SizeOfModuleName = struct.unpack('<I', data[0x02:0x06])[0]793 self.ole_logger.debug('ModuleNameRecord.Size: ' + str(hex(self.SizeOfModuleName)))794 self.ModuleName = data[0x06:0x06+self.SizeOfModuleName]795 self.ole_logger.debug('ModuleNameRecord.ModuleName: ' + self.ModuleName)796class ModuleNameUnicodeRecord(VBABase):797 Id = 0798 SizeOfModuleNameUnicode = 0799 ModuleNameUnicode = ''800 def __init__(self, data):801 self.Id = 0802 self.SizeOfModuleName = 0803 self.ModuleNameUnicode = ''804 self.Id = struct.unpack('<H', data[0x00:0x02])[0]805 self.ole_logger.debug('ModuleNameUnicodeRecord.Id: ' + str(hex(self.Id)))806 if self.Id != 0x47:807 self._raise_exception('ModuleNameUnicodeRecord.Id has an abnormal value.')808 self.SizeOfModuleNameUnicode = struct.unpack('<I', data[0x02:0x06])[0]809 self.ole_logger.debug('ModuleNameUnicodeRecord.SizeOfModuleNameUnicode: ' + str(hex(self.SizeOfModuleNameUnicode)))810 self.ModuleNameUnicode = data[0x06:0x06+self.SizeOfModuleNameUnicode]811 self.ole_logger.debug('ModuleNameUnicodeRecord.ModuleName: ' + self.ModuleNameUnicode)812class ModuleStreamNameRecord(VBABase):813 814 Id = 0815 SizeOfStreamName = 0816 StreamName = ''817 Reserved = 0818 SizeOfStreamNameUnicode = 0819 StreamNameUnicode = ''820 def __init__(self, data):821 self.Id = 0822 self.SizeOfStreamName = 0823 self.StreamName = ''824 self.Reserved = 0825 self.SizeOfStreamNameUnicode = 0826 self.StreamNameUnicode = ''827 self.Id = struct.unpack('<H', data[0x00:0x02])[0]828 self.ole_logger.debug('ModuleStreamNameRecord.Id: ' + str(hex(self.Id)))829 if self.Id != 0x1A:830 self._raise_exception('ModuleStreamNameRecord.Id has an abnormal value.')831 self.SizeOfStreamName = struct.unpack('<I', data[0x02:0x06])[0]832 self.ole_logger.debug('ModuleStreamNameRecord.SizeOfStreamName: ' + str(hex(self.SizeOfStreamName)))833 self.StreamName = data[0x06:0x06+self.SizeOfStreamName]834 self.ole_logger.debug('ModuleStreamNameRecord.StreamName: ' + self.StreamName)835 current = 0x06 + self.SizeOfStreamName836 self.Reserved = struct.unpack('<H', data[current:current+0x02])[0]837 self.ole_logger.debug('ModuleStreamNameRecord.Reserved: ' + str(hex(self.Reserved)))838 if self.Reserved != 0x32:839 self._raise_exception('ModuleStreamNameRecord.Reserved has an abnormal value.')840 current = current + 0x02841 self.SizeOfStreamNameUnicode = struct.unpack('<I', data[current:current+0x04])[0]842 self.ole_logger.debug('ModuleStreamNameRecord.SizeOfStreamNameUnicode: ' + str(hex(self.SizeOfStreamNameUnicode)))843 current = current + 0x04844 self.StreamNameUnicode = data[current:current+self.SizeOfStreamNameUnicode]845 self.ole_logger.debug('ModuleStreamNameRecord.StreamNameUnicode: ' + self.StreamNameUnicode)846class ModuleDocStringRecord(VBABase):847 848 Id = 0849 SizeOfDocString = 0850 DocString = ''851 Reserved = 0852 SizeOfDocStringUnicode = 0853 DocStringUnicode = ''854 def __init__(self, data):855 self.Id = 0856 self.SizeOfDocString = 0857 self.DocString = ''858 self.Reserved = 0859 self.SizeOfDocStringUnicode = 0860 self.DocStringUnicode = ''861 self.Id = struct.unpack('<H', data[0x00:0x02])[0]862 self.ole_logger.debug('ModuleDocStringRecord.Id: ' + str(hex(self.Id)))863 if self.Id != 0x1C:864 self._raise_exception('ModuleDocStringRecord.Id has an abnormal value.')865 self.SizeOfDocString = struct.unpack('<I', data[0x02:0x06])[0]866 self.ole_logger.debug('ModuleDocStringRecord.SizeOfDocString: ' + str(hex(self.SizeOfDocString)))867 self.DocString = data[0x06:0x06+self.SizeOfDocString]868 self.ole_logger.debug('ModuleDocStringRecord.DocString: ' + self.DocString)869 current = 0x06 + self.SizeOfDocString870 self.Reserved = struct.unpack('<H', data[current:current+0x02])[0]871 self.ole_logger.debug('ModuleDocStringRecord.Reserved: ' + str(hex(self.Reserved)))872 if self.Reserved != 0x48:873 self._raise_exception('ModuleDocStringRecord.Reserved has an abnormal value.')874 current = current + 0x02875 self.SizeOfDocStringUnicode = struct.unpack('<I', data[current:current+0x04])[0]876 self.ole_logger.debug('ModuleDocStringRecord.SizeOfDocStringUnicode: ' + str(hex(self.SizeOfDocStringUnicode)))877 current = current + 0x04878 self.DocStringUnicode = data[current:current+self.SizeOfDocStringUnicode]879 self.ole_logger.debug('ModuleDocStringRecord.DocStringUnicode: ' + self.DocStringUnicode) 880class ModuleOffsetRecord(VBABase):881 882 Id = 0883 Size = 0884 TextOffset = 0885 def __init__(self, data):886 self.Id = 0887 self.Size = 0888 self.TextOffset = 0889 self.Id = struct.unpack('<H', data[0x00:0x02])[0]890 self.ole_logger.debug('ModuleOffsetRecord.Id: ' + str(hex(self.Id)))891 if self.Id != 0x31:892 self._raise_exception('ModuleOffsetRecord.Id has an abnormal value.')893 self.Size = struct.unpack('<I', data[0x02:0x06])[0]894 self.ole_logger.debug('ModuleOffsetRecord.Size: ' + str(hex(self.Size)))895 if self.Size != 0x04:896 self._raise_exception('ModuleOffsetRecord.Size has an abnormal value.')897 self.TextOffset = struct.unpack('<I', data[0x06:0x0A])[0]898 self.ole_logger.debug('ModuleOffsetRecord.TextOffset: ' + str(hex(self.TextOffset))) 899class ModuleHelpContextRecord(VBABase):900 901 Id = 0902 Size = 0903 HelpContext = 0904 def __init__(self, data):905 self.Id = 0906 self.Size = 0907 self.HelpContext = 0908 self.Id = struct.unpack('<H', data[0x00:0x02])[0]909 self.ole_logger.debug('ModuleHelpContextRecord.Id: ' + str(hex(self.Id)))910 if self.Id != 0x1E:911 self._raise_exception('ModuleHelpContextRecord.Id has an abnormal value.')912 self.Size = struct.unpack('<I', data[0x02:0x06])[0]913 self.ole_logger.debug('ModuleHelpContextRecord.Size: ' + str(hex(self.Size)))914 if self.Size != 0x04:915 self._raise_exception('ModuleHelpContextRecord.Size has an abnormal value.')916 self.HelpContext = struct.unpack('<I', data[0x06:0x0A])[0]917 self.ole_logger.debug('ModuleHelpContextRecord.HelpContext: ' + str(hex(self.HelpContext))) 918class ModuleCookieRecord(VBABase):919 920 Id = 0921 Size = 0922 Cookie = 0923 def __init__(self, data):924 self.Id = 0925 self.Size = 0926 self.Cookie = 0927 self.Id = struct.unpack('<H', data[0x00:0x02])[0]928 self.ole_logger.debug('ModuleCookieRecord.Id: ' + str(hex(self.Id)))929 if self.Id != 0x2C:930 self._raise_exception('ModuleCookieRecord.Id has an abnormal value.')931 self.Size = struct.unpack('<I', data[0x02:0x06])[0]932 self.ole_logger.debug('ModuleCookieRecord.Size: ' + str(hex(self.Size)))933 if self.Size != 0x02:934 self._raise_exception('ModuleCookieRecord.Size has an abnormal value.')935 self.Cookie = struct.unpack('<H', data[0x06:0x08])[0]936 self.ole_logger.debug('ModuleCookieRecord.Cookie: ' + str(hex(self.Cookie))) 937class ModuleTypeRecord(VBABase):938 939 Id = 0940 Reserved = 0941 def __init__(self, data):942 self.Id = 0943 self.Size = 0944 self.Id = struct.unpack('<H', data[0x00:0x02])[0]945 self.ole_logger.debug('ModuleTypeRecord.Id: ' + str(hex(self.Id)))946 if self.Id != 0x21 and self.Id != 0x22:947 self._raise_exception('ModuleTypeRecord.Id has an abnormal value.')948 self.Reserved = struct.unpack('<I', data[0x02:0x06])[0]949 self.ole_logger.debug('ModuleTypeRecord.Reserved: ' + str(hex(self.Reserved)))950 if self.Reserved != 0x00:951 self._raise_exception('ModuleTypeRecord.Reserved has an abnormal value.')952class ModuleReadOnlyRecord(VBABase):953 954 Id = 0955 Reserved = 0956 def __init__(self, data):957 self.Id = 0958 self.Size = 0959 self.Id = struct.unpack('<H', data[0x00:0x02])[0]960 self.ole_logger.debug('ModuleReadOnlyRecord.Id: ' + str(hex(self.Id)))961 if self.Id != 0x25:962 self._raise_exception('ModuleReadOnlyRecord.Id has an abnormal value.')963 self.Reserved = struct.unpack('<I', data[0x02:0x06])[0]964 self.ole_logger.debug('ModuleReadOnlyRecord.Reserved: ' + str(hex(self.Reserved)))965 if self.Reserved != 0x00:966 self._raise_exception('ModuleReadOnlyRecord.Reserved has an abnormal value.')967class ModulePrivateRecord(VBABase):968 969 Id = 0970 Reserved = 0971 def __init__(self, data):972 self.Id = 0973 self.Size = 0974 self.Id = struct.unpack('<H', data[0x00:0x02])[0]975 self.ole_logger.debug('ModulePrivateRecord.Id: ' + str(hex(self.Id)))976 if self.Id != 0x28:977 self._raise_exception('ModulePrivateRecord.Id has an abnormal value.')978 self.Reserved = struct.unpack('<I', data[0x02:0x06])[0]979 self.ole_logger.debug('ModulePrivateRecord.Reserved: ' + str(hex(self.Reserved)))980 if self.Reserved != 0x00:981 self._raise_exception('ModulePrivateRecord.Reserved has an abnormal value.')982class ModuleRecord(VBABase):983 NameRecord = None984 NameUnicodeRecord = None985 StreamNameRecord = None986 DocStringRecord = None987 OffsetRecord = None988 HelpContextRecord = None989 CookieRecord = None990 TypeRecord = None991 ReadOnlyRecord = None992 PrivateRecord = None993 Terminator = 0994 Reserved = 0995 Size = 0996 997 def __init__(self, data):998 self.NameRecord = None999 self.NameUnicodeRecord = None1000 self.StreamNameRecord = None1001 self.DocStringRecord = None1002 self.OffsetRecord = None1003 self.HelpContextRecord = None1004 self.CookieRecord = None1005 self.TypeRecord = None1006 self.ReadOnlyRecord = None1007 self.PrivateRecord = None1008 self.Terminator = 01009 self.Reserved = 01010 self.Size = 01011 current = 01012 self.NameRecord = ModuleNameRecord(data)1013 current = current + 0x06 + self.NameRecord.SizeOfModuleName1014 self.NameUnicodeRecord = ModuleNameUnicodeRecord(data[current:])1015 current = current + 0x06 + self.NameUnicodeRecord.SizeOfModuleNameUnicode1016 self.StreamNameRecord = ModuleStreamNameRecord(data[current:])1017 current = current + 0x0C + self.StreamNameRecord.SizeOfStreamName + self.StreamNameRecord.SizeOfStreamNameUnicode1018 self.DocStringRecord = ModuleDocStringRecord(data[current:])1019 current = current + 0x0C + self.DocStringRecord.SizeOfDocString + self.DocStringRecord.SizeOfDocStringUnicode1020 self.OffsetRecord = ModuleOffsetRecord(data[current:current+0x0A])1021 current = current + 0x0A1022 self.HelpContextRecord = ModuleHelpContextRecord(data[current:current+0x0A])1023 current = current + 0x0A1024 self.CookieRecord = ModuleCookieRecord(data[current:current+0x08])1025 current = current + 0x081026 self.TypeRecord = ModuleTypeRecord(data[current:current+0x06])1027 while True:1028 current = current + 0x061029 id = struct.unpack('<H', data[current:current+0x02])[0]1030 if id == 0x25:1031 self.ReadOnlyRecord = ModuleReadOnlyRecord(data[current:current+0x06])1032 elif id == 0x28:1033 self.PrivateRecord = ModulePrivateRecord(data[current:current+0x06])1034 elif id == 0x2B:1035 self.Terminator = struct.unpack('<H', data[current:current+0x02])[0]1036 break1037 else:1038 self._raise_exception('ModuleRecord contains an abnormal record id.')1039 current = current + 0x021040 self.Reserved = struct.unpack('<I', data[current:current+0x04])[0]1041 if self.Size != 0x00:1042 self._raise_exception('ModuleRecord.Reserved has an abnormal value.')1043 1044 self.Size = current + 0x041045class ProjectModulesRecord(VBABase):1046 Id = 01047 Size = 01048 Count = 01049 CookieRecord = None1050 ModuleArray = list()1051 def __init__(self, data):1052 self.Id = 01053 self.Size = 01054 self.Count = 01055 self.CookieRecord = None1056 self.ModuleArray = list()1057 self.Id = struct.unpack('<H', data[0x00:0x02])[0]1058 self.ole_logger.debug('ModulesRecord.Id: ' + str(hex(self.Id)))1059 if self.Id != 0x0F:1060 self._raise_exception('ModulesRecord.Id has an abnormal value.')1061 self.Size = struct.unpack('<I', data[0x02:0x06])[0]1062 self.ole_logger.debug('ModulesRecord.Size: ' + str(hex(self.Size)))1063 if self.Size != 0x02:1064 self._raise_exception('ModulesRecord.Size has an abnormal value.')1065 1066 self.Count = struct.unpack('<H', data[0x06:0x08])[0]1067 self.ole_logger.debug('ModulesRecord.Count: ' + str(hex(self.Count)))1068 self.CookieRecord = ProjectCookieRecord(data[0x08:0x10])1069 current = 0x101070 for i in range(0, self.Count):1071 Module = ModuleRecord(data[current:])1072 self.ModuleArray.append(Module)1073 current = current + Module.Size1074 1075 1076class DirStream(VBABase):1077 InformationRecord = None1078 ReferencesRecord = None1079 ModulesRecord = None1080 def __init__(self, data):1081 self.InformationRecord = None1082 self.ReferencesRecord = None1083 self.ModulesRecord = None1084 1085 self.ole_logger.debug('######## dirStream ########')1086 1087 data = self._decompress(data)1088 self.InformationRecord = ProjectInformationRecord(data)1089 current = self.InformationRecord.Size1090 self.ReferencesRecord = ProjectReferencesRecord(data[current:])1091 current = current + self.ReferencesRecord.Size1092 self.ModulesRecord = ProjectModulesRecord(data[current:])1093class VBAFile(VBABase):1094 OLE = None1095 PROJECT = None1096 PROJECTwm = None1097 VBA_PROJECT = None1098 dir = None1099 def __init__(self, filename):1100 self.OLE = None1101 self.PROJECT = None1102 self.PROJECTwm = None1103 self.VBA_PROJECT = None1104 self.dir = None1105 self.OLE = OLEFile(filename)1106 project_data = self.OLE.find_object_by_name('PROJECT')1107 if project_data is not None:1108 self.PROJECT = ProjectStream(project_data)1109 else:1110 self.ole_logger.warn('VBA project does not contain the PROJECT stream.')1111 1112 projectwm_data = self.OLE.find_object_by_name('PROJECTwm')1113 if projectwm_data is not None:1114 self.PROJECTwm = Projectwm(projectwm_data)1115 1116 vba_project_data = self.OLE.find_object_by_name('_VBA_PROJECT')1117 if vba_project_data is not None:1118 self.VBA_PROJECT = VBAProject(vba_project_data)1119 else:1120 self.ole_logger.warn('VBA project does not contain the _VBA_PROJECT stream.')1121 dir_data = self.OLE.find_object_by_name('dir')1122 if dir_data is not None:1123 self.dir = DirStream(dir_data)1124 else:1125 self._raise_exception('VBA project does not contain the dir stream.')1126if __name__ == '__main__':1127 init_logging(True)1128 #init_logging(False)1129 try:1130 vba = VBAFile('oletest1.doc')1131 except Exception as e:...
test_traceback.py
Source:test_traceback.py
1# Licensed to the .NET Foundation under one or more agreements.2# The .NET Foundation licenses this file to you under the Apache 2.0 License.3# See the LICENSE file in the project root for more information.4# !!! DO NOT MOVE OR CHANGE THE FOLLOWING LINES5def _raise_exception():6 raise Exception()7_retb = (18, 0, 'test_traceback.py', '_raise_exception')8from iptest.assert_util import *9from iptest.file_util import *10if not is_cli: import os11def _raise_exception_with_finally():12 try:13 raise Exception()14 finally:15 pass16_rewftb= (27, 0, 'test_traceback.py', '_raise_exception_with_finally')17def assert_traceback(expected):18 import sys19 tb = sys.exc_info()[2]20 21 if expected is None:22 AreEqual(None, expected)23 else:24 tb_list = []25 while tb is not None :26 f = tb.tb_frame27 co = f.f_code28 filename = co.co_filename.lower()29 name = co.co_name30 tb_list.append((tb.tb_lineno, tb.tb_lasti, filename, name))31 tb = tb.tb_next32 33 #print tb_list34 35 AreEqual(len(tb_list), len(expected))36 37 for x in range(len(expected)):38 AreEqual(tb_list[x][0], expected[x][0])39 AreEqual(tb_list[x][2:], expected[x][2:])40 41def test_no_traceback():42 #assert_traceback(None)43 try:44 _raise_exception()45 except:46 pass47 assert_traceback(None)48FILE="test_traceback.py"49LINE100 = 7050def test_catch_others_exception():51 try:52 _raise_exception()53 except:54 assert_traceback([(LINE100 + 2, 0, FILE, 'test_catch_others_exception'), _retb])55LINE110 = 7856def test_catch_its_own_exception():57 try:58 raise Exception()59 except:60 assert_traceback([(LINE110 + 2, 0, FILE, 'test_catch_its_own_exception')])61LINE120 = 8662def test_catch_others_exception_with_finally():63 try:64 _raise_exception_with_finally()65 except:66 assert_traceback([(LINE120 + 2, 0, FILE, 'test_catch_others_exception_with_finally'), _rewftb])67LINE130 = 9468def test_nested_caught_outside():69 try:70 x = 271 try:72 _raise_exception()73 except NameError:74 Assert(False, "unhittable")75 y = 276 except:77 assert_traceback([(LINE130 + 4, 0, FILE, 'test_nested_caught_outside'), _retb])78LINE140 = 10879def test_nested_caught_inside():80 try:81 x = 282 try:83 _raise_exception()84 except:85 assert_traceback([(LINE140 + 4, 0, FILE, 'test_nested_caught_inside'), _retb])86 y = 287 except:88 assert_traceback(None)89LINE150 = 12090def test_throw_in_except():91 try:92 _raise_exception()93 except:94 assert_traceback([(LINE150+2, 0, FILE, 'test_throw_in_except'), _retb])95 try:96 assert_traceback([(LINE150+2, 0, FILE, 'test_throw_in_except'), _retb])97 _raise_exception()98 except:99 assert_traceback([(LINE150+7, 0, FILE, 'test_throw_in_except'), _retb])100 assert_traceback([(LINE150+7, 0, FILE, 'test_throw_in_except'), _retb])101LINE160 = 134102class C1:103 def M(self):104 try:105 _raise_exception()106 except:107 assert_traceback([(LINE160 + 3, 0, FILE, 'M'), _retb])108def test_throw_in_method():109 c = C1()110 c.M()111LINE170 = 147112def test_throw_when_defining_class():113 class C2(object):114 try:115 _raise_exception()116 except:117 assert_traceback([(LINE170 + 3, 0, FILE, 'C2'), _retb])118def throw_when_defining_class_directly():119 class C3(C1):120 _raise_exception()121LINE180 = 160122def test_throw_when_defining_class_directly():123 try:124 throw_when_defining_class_directly()125 except:126 assert_traceback([(LINE180 + 2, 0, FILE, 'test_throw_when_defining_class_directly'),127 (LINE180 - 5, 0, FILE, 'throw_when_defining_class_directly'),128 (LINE180 - 4, 0, FILE, 'C3'), _retb])129LINE200 = 169130def test_compiled_code():131 try:132 codeobj = compile('\nraise Exception()', '<mycode>', 'exec')133 exec(codeobj, {})134 except:135 assert_traceback([(LINE200+3, 0, FILE, 'test_compiled_code'), (2, 0, '<mycode>', '<module>')])136def generator_throw_before_yield():137 _raise_exception()138 yield 1139 140LINE210 = 181141def test_throw_before_yield():142 try:143 for x in generator_throw_before_yield():144 pass145 except:146 assert_traceback([(LINE210+3, 0, FILE, 'test_throw_before_yield'), (LINE210-4, 2, 'test_traceback.py', 'generator_throw_before_yield'), _retb])147def generator_throw_after_yield():148 yield 1149 _raise_exception()150LINE220 = 194151def test_throw_while_yield():152 try:153 for x in generator_throw_while_yield():154 pass155 except:156 assert_traceback([(LINE220+3, 0, FILE, 'test_throw_while_yield')])157def generator_yield_inside_try():158 try:159 yield 1160 yield 2161 _raise_exception()162 except NameError:163 pass164LINE230 = 211165def test_yield_inside_try():166 try:167 for x in generator_yield_inside_try():168 pass169 except:170 assert_traceback([(LINE230+3, 0, FILE, 'test_yield_inside_try'), (LINE230-5, 2, 'test_traceback.py', 'generator_yield_inside_try'), _retb])171LINE240 = 221172def test_throw_and_throw():173 try:174 _raise_exception()175 except:176 assert_traceback([(LINE240 + 2, 0, FILE, 'test_throw_and_throw'), _retb])177 try:178 _raise_exception()179 except:180 assert_traceback([(LINE240 + 6, 0, FILE, 'test_throw_and_throw'), _retb])181LINE250 = 233182def test_throw_in_another_file():183 if is_cli: _f_file = path_combine(get_full_dir_name(testpath.public_testdir), 'foo.py')184 else: _f_file = os.getcwd() + '\\foo.py'185 write_to_file(_f_file, '''186def another_raise():187 raise Exception()188''');189 try:190 import foo191 foo.another_raise()192 except:193 assert_traceback([(LINE250 + 8, 0, FILE, 'test_throw_in_another_file'), (3, 0, _f_file.lower(), 'another_raise')])194 finally:195 os.remove(_f_file)196class MyException(Exception): pass197Line260 = 250198def catch_MyException():199 try:200 _raise_exception()201 except MyException:202 assert_traceback([]) # UNREACABLE. THIS TRICK SIMPLIFIES THE CHECK203def test_catch_MyException():204 try:205 catch_MyException()206 except:207 assert_traceback([(Line260+8, 0, FILE, 'test_catch_MyException'), (Line260+2, 0, FILE, 'catch_MyException'), _retb])208Line263 = 263209def test_cp11923_first():210 try:211 _t_test = path_combine(testpath.public_testdir, "cp11923.py")212 write_to_file(_t_test, """def f():213 x = 'something bad'214 raise Exception(x)""")...
Automation Testing Tutorials
Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.
LambdaTest Learning Hubs:
- JUnit Tutorial
- TestNG Tutorial
- Webdriver Tutorial
- WebDriverIO Tutorial
- Protractor Tutorial
- Selenium 4 Tutorial
- Jenkins Tutorial
- NUnit Tutorial
- Jest Tutorial
- Playwright Tutorial
- Cypress Tutorial
- PyTest Tutorial
YouTube
You could also refer to video tutorials over LambdaTest YouTube channel to get step by step demonstration from industry experts.
Run Testify automation tests on LambdaTest cloud grid
Perform automation testing on 3000+ real desktop and mobile devices online.
Try LambdaTest Now !!
Get 100 minutes of automation test minutes FREE!!
Was this article helpful?
