Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Exercise MFTINDX.py with test image and extracted MFT #83

Merged
merged 10 commits into from
Nov 1, 2023
Merged

Exercise MFTINDX.py with test image and extracted MFT #83

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
46e9421
Replace array.array type annotations with MutableSequence[int]
ajnelson-nist Oct 18, 2023
779f46c
Fix usage of types.MethodType
ajnelson-nist Oct 19, 2023
ebc3052
Initialize empty "B" arrays from empty bytestring, not empty characte…
ajnelson-nist Oct 19, 2023
08794ea
Exercise MFTINDX.py
ajnelson-nist Oct 19, 2023
b263c97
Regenerate Make-managed files
ajnelson-nist Oct 19, 2023
44e1a9e
Fix MutableSequence specialization in Python 3.8
ajnelson-nist Oct 19, 2023
fbe6592
Use type signatures to designate some functions as non-mutative
ajnelson-nist Oct 19, 2023
2205d8e
Merge branch 'master' into exercise_mftindx_py
ajnelson-nist Oct 25, 2023
df0d122
Remove `bytes()` calls that are redundant with type signatures
ajnelson-nist Oct 25, 2023
08a5e02
Replace bytes() calls with typing.cast()
ajnelson-nist Oct 25, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
95 changes: 50 additions & 45 deletions indxparse/BinaryParser.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,13 @@
import sys
import types
from datetime import datetime
from typing import Dict, List, Union
from typing import Dict, List, Union, cast

if sys.version_info >= (3, 9):
from collections.abc import MutableSequence

else:
from typing import MutableSequence

verbose = False

Expand Down Expand Up @@ -161,7 +167,7 @@ def __call__(self, *args, **kwargs):
except KeyError:
# We have an entry not in the cache
self.misses += 1
func = types.MethodType(self.func, self.obj, self.name)
func = types.MethodType(self.func, self.obj)
value = func(*args, **kwargs)
lru = self.mru.newer # Always true
# If we haven't reached capacity
Expand Down Expand Up @@ -229,7 +235,7 @@ def align(offset, alignment):
return offset + (alignment - (offset % alignment))


def dosdate(dosdate: array.array, dostime: array.array) -> datetime:
def dosdate(dosdate: bytes, dostime: bytes) -> datetime:
"""
`dosdate`: 2 bytes, little endian.
`dostime`: 2 bytes, little endian.
Expand Down Expand Up @@ -323,7 +329,7 @@ def __str__(self):


def read_byte(
buf: array.array,
buf: bytes,
offset: int,
) -> int:
"""
Expand All @@ -341,7 +347,7 @@ def read_byte(


def read_word(
buf: array.array,
buf: bytes,
offset: int,
) -> int:
"""
Expand All @@ -359,7 +365,7 @@ def read_word(


def read_dword(
buf: array.array,
buf: bytes,
offset: int,
) -> int:
"""
Expand All @@ -382,7 +388,7 @@ class Block(object):
A block is associated with a offset into a byte-string.
"""

def __init__(self, buf: array.array, offset: int) -> None:
def __init__(self, buf: MutableSequence[int], offset: int) -> None:
"""
Constructor.
Arguments:
Expand Down Expand Up @@ -675,7 +681,7 @@ def unpack_byte(self, offset: int) -> int:
Throws:
- `OverrunBufferException`
"""
return read_byte(self._buf, self._offset + offset)
return read_byte(cast(bytes, self._buf), self._offset + offset)
Copy link
Owner

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

looks great!


def unpack_int8(self, offset: int) -> int:
"""
Expand All @@ -687,7 +693,7 @@ def unpack_int8(self, offset: int) -> int:
"""
o = self._offset + offset
try:
return struct.unpack_from("<b", self._buf, o)[0]
return struct.unpack_from("<b", cast(bytes, self._buf), o)[0]
except struct.error:
raise OverrunBufferException(o, len(self._buf))

Expand All @@ -700,7 +706,7 @@ def unpack_word(self, offset: int) -> int:
Throws:
- `OverrunBufferException`
"""
return read_word(self._buf, self._offset + offset)
return read_word(cast(bytes, self._buf), self._offset + offset)

def unpack_word_be(self, offset: int) -> int:
"""
Expand All @@ -713,7 +719,7 @@ def unpack_word_be(self, offset: int) -> int:
"""
o = self._offset + offset
try:
return struct.unpack_from(">H", self._buf, o)[0]
return struct.unpack_from(">H", cast(bytes, self._buf), o)[0]
except struct.error:
raise OverrunBufferException(o, len(self._buf))

Expand All @@ -728,7 +734,7 @@ def unpack_int16(self, offset: int) -> int:
"""
o = self._offset + offset
try:
return struct.unpack_from("<h", self._buf, o)[0]
return struct.unpack_from("<h", cast(bytes, self._buf), o)[0]
except struct.error:
raise OverrunBufferException(o, len(self._buf))

Expand All @@ -750,7 +756,7 @@ def unpack_dword(self, offset: int) -> int:
Throws:
- `OverrunBufferException`
"""
return read_dword(self._buf, self._offset + offset)
return read_dword(cast(bytes, self._buf), self._offset + offset)

def unpack_dword_be(self, offset: int) -> int:
"""
Expand All @@ -762,7 +768,7 @@ def unpack_dword_be(self, offset: int) -> int:
"""
o = self._offset + offset
try:
return struct.unpack_from(">I", self._buf, o)[0]
return struct.unpack_from(">I", cast(bytes, self._buf), o)[0]
except struct.error:
raise OverrunBufferException(o, len(self._buf))

Expand All @@ -777,7 +783,7 @@ def unpack_int32(self, offset: int) -> int:
"""
o = self._offset + offset
try:
return struct.unpack_from("<i", self._buf, o)[0]
return struct.unpack_from("<i", cast(bytes, self._buf), o)[0]
except struct.error:
raise OverrunBufferException(o, len(self._buf))

Expand All @@ -791,7 +797,7 @@ def unpack_qword(self, offset: int) -> int:
"""
o = self._offset + offset
try:
return struct.unpack_from("<Q", self._buf, o)[0]
return struct.unpack_from("<Q", cast(bytes, self._buf), o)[0]
except struct.error:
raise OverrunBufferException(o, len(self._buf))

Expand All @@ -806,7 +812,7 @@ def unpack_int64(self, offset: int) -> int:
"""
o = self._offset + offset
try:
return struct.unpack_from("<q", self._buf, o)[0]
return struct.unpack_from("<q", cast(bytes, self._buf), o)[0]
except struct.error:
raise OverrunBufferException(o, len(self._buf))

Expand All @@ -821,7 +827,7 @@ def unpack_float(self, offset: int) -> float:
"""
o = self._offset + offset
try:
return struct.unpack_from("<f", self._buf, o)[0]
return struct.unpack_from("<f", cast(bytes, self._buf), o)[0]
except struct.error:
raise OverrunBufferException(o, len(self._buf))

Expand All @@ -836,7 +842,7 @@ def unpack_double(self, offset: int) -> float:
"""
o = self._offset + offset
try:
return struct.unpack_from("<d", self._buf, o)[0]
return struct.unpack_from("<d", cast(bytes, self._buf), o)[0]
except struct.error:
raise OverrunBufferException(o, len(self._buf))

Expand All @@ -855,7 +861,7 @@ def unpack_binary(self, offset: int, length=0) -> array.array:
o = self._offset + offset
try:
return array.array(
"B", struct.unpack_from("<%ds" % (length), self._buf, o)[0]
"B", struct.unpack_from("<%ds" % (length), cast(bytes, self._buf), o)[0]
)
except struct.error:
raise OverrunBufferException(o, len(self._buf))
Expand All @@ -881,11 +887,9 @@ def unpack_wstring(self, offset: int, length: int) -> str:
Throws:
- `UnicodeDecodeError`
"""
return (
return bytes(
self._buf[self._offset + offset : self._offset + offset + 2 * length]
.tobytes()
.decode("utf-16le")
)
).decode("utf-16le")

def unpack_dosdate(self, offset: int) -> datetime:
"""
Expand All @@ -898,7 +902,9 @@ def unpack_dosdate(self, offset: int) -> datetime:
"""
try:
o = self._offset + offset
return dosdate(self._buf[o : o + 2], self._buf[o + 2 : o + 4])
return dosdate(
cast(bytes, self._buf[o : o + 2]), cast(bytes, self._buf[o + 2 : o + 4])
)
except struct.error:
raise OverrunBufferException(o, len(self._buf))

Expand All @@ -925,7 +931,7 @@ def unpack_systemtime(self, offset: int) -> datetime:
"""
o = self._offset + offset
try:
parts = struct.unpack_from("<WWWWWWWW", self._buf, o)
parts = struct.unpack_from("<WWWWWWWW", cast(bytes, self._buf), o)
except struct.error:
raise OverrunBufferException(o, len(self._buf))
return datetime(
Expand Down Expand Up @@ -954,26 +960,25 @@ def unpack_guid(self, offset: int) -> str:
raise OverrunBufferException(o, len(self._buf))

# Yeah, this is ugly
h = list(map(ord, _bin))
return (
"%02x%02x%02x%02x-%02x%02x-%02x%02x-%02x%02x-%02x%02x%02x%02x%02x%02x"
% (
h[3],
h[2],
h[1],
h[0],
h[5],
h[4],
h[7],
h[6],
h[8],
h[9],
h[10],
h[11],
h[12],
h[13],
h[14],
h[15],
_bin[3],
_bin[2],
_bin[1],
_bin[0],
_bin[5],
_bin[4],
_bin[7],
_bin[6],
_bin[8],
_bin[9],
_bin[10],
_bin[11],
_bin[12],
_bin[13],
_bin[14],
_bin[15],
)
)

Expand Down Expand Up @@ -1003,14 +1008,14 @@ class Nestable(object):

def __init__(
self,
buf: array.array,
buf: MutableSequence[int],
offset: int,
) -> None:
super(Nestable, self).__init__()

@staticmethod
def structure_size(
buf: array.array,
buf: MutableSequence[int],
offset: int,
parent,
) -> int:
Expand Down
Loading